utils.c

   1 /*
   2  * Copyright (C) 2007 Oracle.  All rights reserved.
   3  * Copyright (C) 2008 Morey Roof.  All rights reserved.
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public
   7  * License v2 as published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public
  15  * License along with this program; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  * Boston, MA 021110-1307, USA.
  18  */
  19
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <sys/ioctl.h>
  24 #include <sys/mount.h>
  25 #include <sys/types.h>
  26 #include <sys/stat.h>
  27 #include <uuid/uuid.h>
  28 #include <fcntl.h>
  29 #include <unistd.h>
  30 #include <mntent.h>
  31 #include <ctype.h>
  32 #include <linux/loop.h>
  33 #include <linux/major.h>
  34 #include <linux/kdev_t.h>
  35 #include <limits.h>
  36 #include <blkid/blkid.h>
  37 #include <sys/vfs.h>
  38 #include <sys/statfs.h>
  39 #include <linux/magic.h>
  40 #include <getopt.h>
  41 #include <sys/utsname.h>
  42 #include <linux/version.h>
  43
  44 #include "kerncompat.h"
  45 #include "radix-tree.h"
  46 #include "ctree.h"
  47 #include "disk-io.h"
  48 #include "transaction.h"
  49 #include "crc32c.h"
  50 #include "utils.h"
  51 #include "volumes.h"
  52 #include "ioctl.h"
  53 #include "commands.h"
  54
  55 #ifndef BLKDISCARD
  56 #define BLKDISCARD      _IO(0x12,119)
  57 #endif
  58
  59 static int btrfs_scan_done = 0;
  60
  61 static char argv0_buf[ARGV0_BUF_SIZE] = "btrfs";
  62
  63 static int rand_seed_initlized = 0;
  64 static unsigned short rand_seed[3];
  65
  66 const char *get_argv0_buf(void)
  67 {
  68         return argv0_buf;
  69 }
  70
  71 void fixup_argv0(char **argv, const char *token)
  72 {
  73         int len = strlen(argv0_buf);
  74
  75         snprintf(argv0_buf + len, sizeof(argv0_buf) - len, " %s", token);
  76         argv[0] = argv0_buf;
  77 }
  78
  79 void set_argv0(char **argv)
  80 {
  81         strncpy(argv0_buf, argv[0], sizeof(argv0_buf));
  82         argv0_buf[sizeof(argv0_buf) - 1] = 0;
  83 }
  84
  85 int check_argc_exact(int nargs, int expected)
  86 {
  87         if (nargs < expected)
  88                 fprintf(stderr, "%s: too few arguments\n", argv0_buf);
  89         if (nargs > expected)
  90                 fprintf(stderr, "%s: too many arguments\n", argv0_buf);
  91
  92         return nargs != expected;
  93 }
  94
  95 int check_argc_min(int nargs, int expected)
  96 {
  97         if (nargs < expected) {
  98                 fprintf(stderr, "%s: too few arguments\n", argv0_buf);
  99                 return 1;
 100         }
 101
 102         return 0;
 103 }
 104
 105 int check_argc_max(int nargs, int expected)
 106 {
 107         if (nargs > expected) {
 108                 fprintf(stderr, "%s: too many arguments\n", argv0_buf);
 109                 return 1;
 110         }
 111
 112         return 0;
 113 }
 114
 115
 116 /*
 117  * Discard the given range in one go
 118  */
 119 static int discard_range(int fd, u64 start, u64 len)
 120 {
 121         u64 range[2] = { start, len };
 122
 123         if (ioctl(fd, BLKDISCARD, &range) < 0)
 124                 return errno;
 125         return 0;
 126 }
 127
 128 /*
 129  * Discard blocks in the given range in 1G chunks, the process is interruptible
 130  */
 131 static int discard_blocks(int fd, u64 start, u64 len)
 132 {
 133         while (len > 0) {
 134                 /* 1G granularity */
 135                 u64 chunk_size = min_t(u64, len, 1*1024*1024*1024);
 136                 int ret;
 137
 138                 ret = discard_range(fd, start, chunk_size);
 139                 if (ret)
 140                         return ret;
 141                 len -= chunk_size;
 142                 start += chunk_size;
 143         }
 144
 145         return 0;
 146 }
 147
 148 static u64 reference_root_table[] = {
 149         [1] =   BTRFS_ROOT_TREE_OBJECTID,
 150         [2] =   BTRFS_EXTENT_TREE_OBJECTID,
 151         [3] =   BTRFS_CHUNK_TREE_OBJECTID,
 152         [4] =   BTRFS_DEV_TREE_OBJECTID,
 153         [5] =   BTRFS_FS_TREE_OBJECTID,
 154         [6] =   BTRFS_CSUM_TREE_OBJECTID,
 155 };
 156
 157 int test_uuid_unique(char *fs_uuid)
 158 {
 159         int unique = 1;
 160         blkid_dev_iterate iter = NULL;
 161         blkid_dev dev = NULL;
 162         blkid_cache cache = NULL;
 163
 164         if (blkid_get_cache(&cache, NULL) < 0) {
 165                 printf("ERROR: lblkid cache get failed\n");
 166                 return 1;
 167         }
 168         blkid_probe_all(cache);
 169         iter = blkid_dev_iterate_begin(cache);
 170         blkid_dev_set_search(iter, "UUID", fs_uuid);
 171
 172         while (blkid_dev_next(iter, &dev) == 0) {
 173                 dev = blkid_verify(cache, dev);
 174                 if (dev) {
 175                         unique = 0;
 176                         break;
 177                 }
 178         }
 179
 180         blkid_dev_iterate_end(iter);
 181         blkid_put_cache(cache);
 182
 183         return unique;
 184 }
 185
 186 /*
 187  * Reserve space from free_tree.
 188  * The algorithm is very simple, find the first cache_extent with enough space
 189  * and allocate from its beginning.
 190  */
 191 static int reserve_free_space(struct cache_tree *free_tree, u64 len,
 192                               u64 *ret_start)
 193 {
 194         struct cache_extent *cache;
 195         int found = 0;
 196
 197         ASSERT(ret_start != NULL);
 198         cache = first_cache_extent(free_tree);
 199         while (cache) {
 200                 if (cache->size > len) {
 201                         found = 1;
 202                         *ret_start = cache->start;
 203
 204                         cache->size -= len;
 205                         if (cache->size == 0) {
 206                                 remove_cache_extent(free_tree, cache);
 207                                 free(cache);
 208                         } else {
 209                                 cache->start += len;
 210                         }
 211                         break;
 212                 }
 213                 cache = next_cache_extent(cache);
 214         }
 215         if (!found)
 216                 return -ENOSPC;
 217         return 0;
 218 }
 219
 220 static inline int write_temp_super(int fd, struct btrfs_super_block *sb,
 221                                    u64 sb_bytenr)
 222 {
 223         u32 crc = ~(u32)0;
 224         int ret;
 225
 226         crc = btrfs_csum_data(NULL, (char *)sb + BTRFS_CSUM_SIZE, crc,
 227                               BTRFS_SUPER_INFO_SIZE - BTRFS_CSUM_SIZE);
 228         btrfs_csum_final(crc, &sb->csum[0]);
 229         ret = pwrite(fd, sb, BTRFS_SUPER_INFO_SIZE, sb_bytenr);
 230         if (ret < BTRFS_SUPER_INFO_SIZE)
 231                 ret = (ret < 0 ? -errno : -EIO);
 232         else
 233                 ret = 0;
 234         return ret;
 235 }
 236
 237 /*
 238  * Setup temporary superblock at cfg->super_bynter
 239  * Needed info are extracted from cfg, and root_bytenr, chunk_bytenr
 240  *
 241  * For now sys chunk array will be empty and dev_item is empty too.
 242  * They will be re-initialized at temp chunk tree setup.
 243  *
 244  * The superblock signature is not valid, denotes a partially created
 245  * filesystem, needs to be finalized.
 246  */
 247 static int setup_temp_super(int fd, struct btrfs_mkfs_config *cfg,
 248                             u64 root_bytenr, u64 chunk_bytenr)
 249 {
 250         unsigned char chunk_uuid[BTRFS_UUID_SIZE];
 251         char super_buf[BTRFS_SUPER_INFO_SIZE];
 252         struct btrfs_super_block *super = (struct btrfs_super_block *)super_buf;
 253         int ret;
 254
 255         memset(super_buf, 0, BTRFS_SUPER_INFO_SIZE);
 256         cfg->num_bytes = round_down(cfg->num_bytes, cfg->sectorsize);
 257
 258         if (*cfg->fs_uuid) {
 259                 if (uuid_parse(cfg->fs_uuid, super->fsid) != 0) {
 260                         error("cound not parse UUID: %s", cfg->fs_uuid);
 261                         ret = -EINVAL;
 262                         goto out;
 263                 }
 264                 if (!test_uuid_unique(cfg->fs_uuid)) {
 265                         error("non-unique UUID: %s", cfg->fs_uuid);
 266                         ret = -EINVAL;
 267                         goto out;
 268                 }
 269         } else {
 270                 uuid_generate(super->fsid);
 271                 uuid_unparse(super->fsid, cfg->fs_uuid);
 272         }
 273         uuid_generate(chunk_uuid);
 274         uuid_unparse(chunk_uuid, cfg->chunk_uuid);
 275
 276         btrfs_set_super_bytenr(super, cfg->super_bytenr);
 277         btrfs_set_super_num_devices(super, 1);
 278         btrfs_set_super_magic(super, BTRFS_MAGIC_PARTIAL);
 279         btrfs_set_super_generation(super, 1);
 280         btrfs_set_super_root(super, root_bytenr);
 281         btrfs_set_super_chunk_root(super, chunk_bytenr);
 282         btrfs_set_super_total_bytes(super, cfg->num_bytes);
 283         /*
 284          * Temporary filesystem will only have 6 tree roots:
 285          * chunk tree, root tree, extent_tree, device tree, fs tree
 286          * and csum tree.
 287          */
 288         btrfs_set_super_bytes_used(super, 6 * cfg->nodesize);
 289         btrfs_set_super_sectorsize(super, cfg->sectorsize);
 290         btrfs_set_super_leafsize(super, cfg->nodesize);
 291         btrfs_set_super_nodesize(super, cfg->nodesize);
 292         btrfs_set_super_stripesize(super, cfg->stripesize);
 293         btrfs_set_super_csum_type(super, BTRFS_CSUM_TYPE_CRC32);
 294         btrfs_set_super_chunk_root(super, chunk_bytenr);
 295         btrfs_set_super_cache_generation(super, -1);
 296         btrfs_set_super_incompat_flags(super, cfg->features);
 297         if (cfg->label)
 298                 __strncpy_null(super->label, cfg->label, BTRFS_LABEL_SIZE - 1);
 299
 300         /* Sys chunk array will be re-initialized at chunk tree init time */
 301         super->sys_chunk_array_size = 0;
 302
 303         ret = write_temp_super(fd, super, cfg->super_bytenr);
 304 out:
 305         return ret;
 306 }
 307
 308 /*
 309  * Setup an extent buffer for tree block.
 310  */
 311 static int setup_temp_extent_buffer(struct extent_buffer *buf,
 312                                     struct btrfs_mkfs_config *cfg,
 313                                     u64 bytenr, u64 owner)
 314 {
 315         unsigned char fsid[BTRFS_FSID_SIZE];
 316         unsigned char chunk_uuid[BTRFS_UUID_SIZE];
 317         int ret;
 318
 319         ret = uuid_parse(cfg->fs_uuid, fsid);
 320         if (ret)
 321                 return -EINVAL;
 322         ret = uuid_parse(cfg->chunk_uuid, chunk_uuid);
 323         if (ret)
 324                 return -EINVAL;
 325
 326         memset(buf->data, 0, cfg->nodesize);
 327         buf->len = cfg->nodesize;
 328         btrfs_set_header_bytenr(buf, bytenr);
 329         btrfs_set_header_generation(buf, 1);
 330         btrfs_set_header_backref_rev(buf, BTRFS_MIXED_BACKREF_REV);
 331         btrfs_set_header_owner(buf, owner);
 332         btrfs_set_header_flags(buf, BTRFS_HEADER_FLAG_WRITTEN);
 333         write_extent_buffer(buf, chunk_uuid, btrfs_header_chunk_tree_uuid(buf),
 334                             BTRFS_UUID_SIZE);
 335         write_extent_buffer(buf, fsid, btrfs_header_fsid(), BTRFS_FSID_SIZE);
 336         return 0;
 337 }
 338
 339 static inline int write_temp_extent_buffer(int fd, struct extent_buffer *buf,
 340                                            u64 bytenr)
 341 {
 342         int ret;
 343
 344         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
 345
 346         /* Temporary extent buffer is always mapped 1:1 on disk */
 347         ret = pwrite(fd, buf->data, buf->len, bytenr);
 348         if (ret < buf->len)
 349                 ret = (ret < 0 ? ret : -EIO);
 350         else
 351                 ret = 0;
 352         return ret;
 353 }
 354
 355 /*
 356  * Insert a root item for temporary tree root
 357  *
 358  * Only used in make_btrfs_v2().
 359  */
 360 static void insert_temp_root_item(struct extent_buffer *buf,
 361                                   struct btrfs_mkfs_config *cfg,
 362                                   int *slot, u32 *itemoff, u64 objectid,
 363                                   u64 bytenr)
 364 {
 365         struct btrfs_root_item root_item;
 366         struct btrfs_inode_item *inode_item;
 367         struct btrfs_disk_key disk_key;
 368
 369         btrfs_set_header_nritems(buf, *slot + 1);
 370         (*itemoff) -= sizeof(root_item);
 371         memset(&root_item, 0, sizeof(root_item));
 372         inode_item = &root_item.inode;
 373         btrfs_set_stack_inode_generation(inode_item, 1);
 374         btrfs_set_stack_inode_size(inode_item, 3);
 375         btrfs_set_stack_inode_nlink(inode_item, 1);
 376         btrfs_set_stack_inode_nbytes(inode_item, cfg->nodesize);
 377         btrfs_set_stack_inode_mode(inode_item, S_IFDIR | 0755);
 378         btrfs_set_root_refs(&root_item, 1);
 379         btrfs_set_root_used(&root_item, cfg->nodesize);
 380         btrfs_set_root_generation(&root_item, 1);
 381         btrfs_set_root_bytenr(&root_item, bytenr);
 382
 383         memset(&disk_key, 0, sizeof(disk_key));
 384         btrfs_set_disk_key_type(&disk_key, BTRFS_ROOT_ITEM_KEY);
 385         btrfs_set_disk_key_objectid(&disk_key, objectid);
 386         btrfs_set_disk_key_offset(&disk_key, 0);
 387
 388         btrfs_set_item_key(buf, &disk_key, *slot);
 389         btrfs_set_item_offset(buf, btrfs_item_nr(*slot), *itemoff);
 390         btrfs_set_item_size(buf, btrfs_item_nr(*slot), sizeof(root_item));
 391         write_extent_buffer(buf, &root_item,
 392                             btrfs_item_ptr_offset(buf, *slot),
 393                             sizeof(root_item));
 394         (*slot)++;
 395 }
 396
 397 static int setup_temp_root_tree(int fd, struct btrfs_mkfs_config *cfg,
 398                                 u64 root_bytenr, u64 extent_bytenr,
 399                                 u64 dev_bytenr, u64 fs_bytenr, u64 csum_bytenr)
 400 {
 401         struct extent_buffer *buf = NULL;
 402         u32 itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize);
 403         int slot = 0;
 404         int ret;
 405
 406         /*
 407          * Provided bytenr must in ascending order, or tree root will have a
 408          * bad key order.
 409          */
 410         if (!(root_bytenr < extent_bytenr && extent_bytenr < dev_bytenr &&
 411               dev_bytenr < fs_bytenr && fs_bytenr < csum_bytenr)) {
 412                 error("bad tree bytenr order: "
 413                                 "root < extent %llu < %llu, "
 414                                 "extent < dev %llu < %llu, "
 415                                 "dev < fs %llu < %llu, "
 416                                 "fs < csum %llu < %llu",
 417                                 (unsigned long long)root_bytenr,
 418                                 (unsigned long long)extent_bytenr,
 419                                 (unsigned long long)extent_bytenr,
 420                                 (unsigned long long)dev_bytenr,
 421                                 (unsigned long long)dev_bytenr,
 422                                 (unsigned long long)fs_bytenr,
 423                                 (unsigned long long)fs_bytenr,
 424                                 (unsigned long long)csum_bytenr);
 425                 return -EINVAL;
 426         }
 427         buf = malloc(sizeof(*buf) + cfg->nodesize);
 428         if (!buf)
 429                 return -ENOMEM;
 430
 431         ret = setup_temp_extent_buffer(buf, cfg, root_bytenr,
 432                                        BTRFS_ROOT_TREE_OBJECTID);
 433         if (ret < 0)
 434                 goto out;
 435
 436         insert_temp_root_item(buf, cfg, &slot, &itemoff,
 437                               BTRFS_EXTENT_TREE_OBJECTID, extent_bytenr);
 438         insert_temp_root_item(buf, cfg, &slot, &itemoff,
 439                               BTRFS_DEV_TREE_OBJECTID, dev_bytenr);
 440         insert_temp_root_item(buf, cfg, &slot, &itemoff,
 441                               BTRFS_FS_TREE_OBJECTID, fs_bytenr);
 442         insert_temp_root_item(buf, cfg, &slot, &itemoff,
 443                               BTRFS_CSUM_TREE_OBJECTID, csum_bytenr);
 444
 445         ret = write_temp_extent_buffer(fd, buf, root_bytenr);
 446 out:
 447         free(buf);
 448         return ret;
 449 }
 450
 451 static int insert_temp_dev_item(int fd, struct extent_buffer *buf,
 452                                 struct btrfs_mkfs_config *cfg,
 453                                 int *slot, u32 *itemoff)
 454 {
 455         struct btrfs_disk_key disk_key;
 456         struct btrfs_dev_item *dev_item;
 457         char super_buf[BTRFS_SUPER_INFO_SIZE];
 458         unsigned char dev_uuid[BTRFS_UUID_SIZE];
 459         unsigned char fsid[BTRFS_FSID_SIZE];
 460         struct btrfs_super_block *super = (struct btrfs_super_block *)super_buf;
 461         int ret;
 462
 463         ret = pread(fd, super_buf, BTRFS_SUPER_INFO_SIZE, cfg->super_bytenr);
 464         if (ret < BTRFS_SUPER_INFO_SIZE) {
 465                 ret = (ret < 0 ? -errno : -EIO);
 466                 goto out;
 467         }
 468
 469         btrfs_set_header_nritems(buf, *slot + 1);
 470         (*itemoff) -= sizeof(*dev_item);
 471         /* setup device item 1, 0 is for replace case */
 472         btrfs_set_disk_key_type(&disk_key, BTRFS_DEV_ITEM_KEY);
 473         btrfs_set_disk_key_objectid(&disk_key, BTRFS_DEV_ITEMS_OBJECTID);
 474         btrfs_set_disk_key_offset(&disk_key, 1);
 475         btrfs_set_item_key(buf, &disk_key, *slot);
 476         btrfs_set_item_offset(buf, btrfs_item_nr(*slot), *itemoff);
 477         btrfs_set_item_size(buf, btrfs_item_nr(*slot), sizeof(*dev_item));
 478
 479         dev_item = btrfs_item_ptr(buf, *slot, struct btrfs_dev_item);
 480         /* Generate device uuid */
 481         uuid_generate(dev_uuid);
 482         write_extent_buffer(buf, dev_uuid,
 483                         (unsigned long)btrfs_device_uuid(dev_item),
 484                         BTRFS_UUID_SIZE);
 485         uuid_parse(cfg->fs_uuid, fsid);
 486         write_extent_buffer(buf, fsid,
 487                         (unsigned long)btrfs_device_fsid(dev_item),
 488                         BTRFS_FSID_SIZE);
 489         btrfs_set_device_id(buf, dev_item, 1);
 490         btrfs_set_device_generation(buf, dev_item, 0);
 491         btrfs_set_device_total_bytes(buf, dev_item, cfg->num_bytes);
 492         /*
 493          * The number must match the initial SYSTEM and META chunk size
 494          */
 495         btrfs_set_device_bytes_used(buf, dev_item,
 496                         BTRFS_MKFS_SYSTEM_GROUP_SIZE +
 497                         BTRFS_CONVERT_META_GROUP_SIZE);
 498         btrfs_set_device_io_align(buf, dev_item, cfg->sectorsize);
 499         btrfs_set_device_io_width(buf, dev_item, cfg->sectorsize);
 500         btrfs_set_device_sector_size(buf, dev_item, cfg->sectorsize);
 501         btrfs_set_device_type(buf, dev_item, 0);
 502
 503         /* Super dev_item is not complete, copy the complete one to sb */
 504         read_extent_buffer(buf, &super->dev_item, (unsigned long)dev_item,
 505                            sizeof(*dev_item));
 506         ret = write_temp_super(fd, super, cfg->super_bytenr);
 507         (*slot)++;
 508 out:
 509         return ret;
 510 }
 511
 512 static int insert_temp_chunk_item(int fd, struct extent_buffer *buf,
 513                                   struct btrfs_mkfs_config *cfg,
 514                                   int *slot, u32 *itemoff, u64 start, u64 len,
 515                                   u64 type)
 516 {
 517         struct btrfs_chunk *chunk;
 518         struct btrfs_disk_key disk_key;
 519         char super_buf[BTRFS_SUPER_INFO_SIZE];
 520         struct btrfs_super_block *sb = (struct btrfs_super_block *)super_buf;
 521         int ret = 0;
 522
 523         ret = pread(fd, super_buf, BTRFS_SUPER_INFO_SIZE,
 524                     cfg->super_bytenr);
 525         if (ret < BTRFS_SUPER_INFO_SIZE) {
 526                 ret = (ret < 0 ? ret : -EIO);
 527                 return ret;
 528         }
 529
 530         btrfs_set_header_nritems(buf, *slot + 1);
 531         (*itemoff) -= btrfs_chunk_item_size(1);
 532         btrfs_set_disk_key_type(&disk_key, BTRFS_CHUNK_ITEM_KEY);
 533         btrfs_set_disk_key_objectid(&disk_key, BTRFS_FIRST_CHUNK_TREE_OBJECTID);
 534         btrfs_set_disk_key_offset(&disk_key, start);
 535         btrfs_set_item_key(buf, &disk_key, *slot);
 536         btrfs_set_item_offset(buf, btrfs_item_nr(*slot), *itemoff);
 537         btrfs_set_item_size(buf, btrfs_item_nr(*slot),
 538                             btrfs_chunk_item_size(1));
 539
 540         chunk = btrfs_item_ptr(buf, *slot, struct btrfs_chunk);
 541         btrfs_set_chunk_length(buf, chunk, len);
 542         btrfs_set_chunk_owner(buf, chunk, BTRFS_EXTENT_TREE_OBJECTID);
 543         btrfs_set_chunk_stripe_len(buf, chunk, 64 * 1024);
 544         btrfs_set_chunk_type(buf, chunk, type);
 545         btrfs_set_chunk_io_align(buf, chunk, cfg->sectorsize);
 546         btrfs_set_chunk_io_width(buf, chunk, cfg->sectorsize);
 547         btrfs_set_chunk_sector_size(buf, chunk, cfg->sectorsize);
 548         btrfs_set_chunk_num_stripes(buf, chunk, 1);
 549         /* TODO: Support DUP profile for system chunk */
 550         btrfs_set_stripe_devid_nr(buf, chunk, 0, 1);
 551         /* We are doing 1:1 mapping, so start is its dev offset */
 552         btrfs_set_stripe_offset_nr(buf, chunk, 0, start);
 553         write_extent_buffer(buf, &sb->dev_item.uuid,
 554                             (unsigned long)btrfs_stripe_dev_uuid_nr(chunk, 0),
 555                             BTRFS_UUID_SIZE);
 556         (*slot)++;
 557
 558         /*
 559          * If it's system chunk, also copy it to super block.
 560          */
 561         if (type & BTRFS_BLOCK_GROUP_SYSTEM) {
 562                 char *cur;
 563                 u32 array_size;
 564
 565                 cur = (char *)sb->sys_chunk_array
 566                         + btrfs_super_sys_array_size(sb);
 567                 memcpy(cur, &disk_key, sizeof(disk_key));
 568                 cur += sizeof(disk_key);
 569                 read_extent_buffer(buf, cur, (unsigned long int)chunk,
 570                                    btrfs_chunk_item_size(1));
 571                 array_size = btrfs_super_sys_array_size(sb);
 572                 array_size += btrfs_chunk_item_size(1) +
 573                                             sizeof(disk_key);
 574                 btrfs_set_super_sys_array_size(sb, array_size);
 575
 576                 ret = write_temp_super(fd, sb, cfg->super_bytenr);
 577         }
 578         return ret;
 579 }
 580
 581 static int setup_temp_chunk_tree(int fd, struct btrfs_mkfs_config *cfg,
 582                                  u64 sys_chunk_start, u64 meta_chunk_start,
 583                                  u64 chunk_bytenr)
 584 {
 585         struct extent_buffer *buf = NULL;
 586         u32 itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize);
 587         int slot = 0;
 588         int ret;
 589
 590         /* Must ensure SYS chunk starts before META chunk */
 591         if (meta_chunk_start < sys_chunk_start) {
 592                 error("wrong chunk order: meta < system %llu < %llu",
 593                                 (unsigned long long)meta_chunk_start,
 594                                 (unsigned long long)sys_chunk_start);
 595                 return -EINVAL;
 596         }
 597         buf = malloc(sizeof(*buf) + cfg->nodesize);
 598         if (!buf)
 599                 return -ENOMEM;
 600         ret = setup_temp_extent_buffer(buf, cfg, chunk_bytenr,
 601                                        BTRFS_CHUNK_TREE_OBJECTID);
 602         if (ret < 0)
 603                 goto out;
 604
 605         ret = insert_temp_dev_item(fd, buf, cfg, &slot, &itemoff);
 606         if (ret < 0)
 607                 goto out;
 608         ret = insert_temp_chunk_item(fd, buf, cfg, &slot, &itemoff,
 609                                      sys_chunk_start,
 610                                      BTRFS_MKFS_SYSTEM_GROUP_SIZE,
 611                                      BTRFS_BLOCK_GROUP_SYSTEM);
 612         if (ret < 0)
 613                 goto out;
 614         ret = insert_temp_chunk_item(fd, buf, cfg, &slot, &itemoff,
 615                                      meta_chunk_start,
 616                                      BTRFS_CONVERT_META_GROUP_SIZE,
 617                                      BTRFS_BLOCK_GROUP_METADATA);
 618         if (ret < 0)
 619                 goto out;
 620         ret = write_temp_extent_buffer(fd, buf, chunk_bytenr);
 621
 622 out:
 623         free(buf);
 624         return ret;
 625 }
 626
 627 static void insert_temp_dev_extent(struct extent_buffer *buf,
 628                                    int *slot, u32 *itemoff, u64 start, u64 len)
 629 {
 630         struct btrfs_dev_extent *dev_extent;
 631         struct btrfs_disk_key disk_key;
 632
 633         btrfs_set_header_nritems(buf, *slot + 1);
 634         (*itemoff) -= sizeof(*dev_extent);
 635         btrfs_set_disk_key_type(&disk_key, BTRFS_DEV_EXTENT_KEY);
 636         btrfs_set_disk_key_objectid(&disk_key, 1);
 637         btrfs_set_disk_key_offset(&disk_key, start);
 638         btrfs_set_item_key(buf, &disk_key, *slot);
 639         btrfs_set_item_offset(buf, btrfs_item_nr(*slot), *itemoff);
 640         btrfs_set_item_size(buf, btrfs_item_nr(*slot), sizeof(*dev_extent));
 641
 642         dev_extent = btrfs_item_ptr(buf, *slot, struct btrfs_dev_extent);
 643         btrfs_set_dev_extent_chunk_objectid(buf, dev_extent,
 644                                             BTRFS_FIRST_CHUNK_TREE_OBJECTID);
 645         btrfs_set_dev_extent_length(buf, dev_extent, len);
 646         btrfs_set_dev_extent_chunk_offset(buf, dev_extent, start);
 647         btrfs_set_dev_extent_chunk_tree(buf, dev_extent,
 648                                         BTRFS_CHUNK_TREE_OBJECTID);
 649         (*slot)++;
 650 }
 651
 652 static int setup_temp_dev_tree(int fd, struct btrfs_mkfs_config *cfg,
 653                                u64 sys_chunk_start, u64 meta_chunk_start,
 654                                u64 dev_bytenr)
 655 {
 656         struct extent_buffer *buf = NULL;
 657         u32 itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize);
 658         int slot = 0;
 659         int ret;
 660
 661         /* Must ensure SYS chunk starts before META chunk */
 662         if (meta_chunk_start < sys_chunk_start) {
 663                 error("wrong chunk order: meta < system %llu < %llu",
 664                                 (unsigned long long)meta_chunk_start,
 665                                 (unsigned long long)sys_chunk_start);
 666                 return -EINVAL;
 667         }
 668         buf = malloc(sizeof(*buf) + cfg->nodesize);
 669         if (!buf)
 670                 return -ENOMEM;
 671         ret = setup_temp_extent_buffer(buf, cfg, dev_bytenr,
 672                                        BTRFS_DEV_TREE_OBJECTID);
 673         if (ret < 0)
 674                 goto out;
 675         insert_temp_dev_extent(buf, &slot, &itemoff, sys_chunk_start,
 676                                BTRFS_MKFS_SYSTEM_GROUP_SIZE);
 677         insert_temp_dev_extent(buf, &slot, &itemoff, meta_chunk_start,
 678                                BTRFS_CONVERT_META_GROUP_SIZE);
 679         ret = write_temp_extent_buffer(fd, buf, dev_bytenr);
 680 out:
 681         free(buf);
 682         return ret;
 683 }
 684
 685 static int setup_temp_fs_tree(int fd, struct btrfs_mkfs_config *cfg,
 686                               u64 fs_bytenr)
 687 {
 688         struct extent_buffer *buf = NULL;
 689         int ret;
 690
 691         buf = malloc(sizeof(*buf) + cfg->nodesize);
 692         if (!buf)
 693                 return -ENOMEM;
 694         ret = setup_temp_extent_buffer(buf, cfg, fs_bytenr,
 695                                        BTRFS_FS_TREE_OBJECTID);
 696         if (ret < 0)
 697                 goto out;
 698         /*
 699          * Temporary fs tree is completely empty.
 700          */
 701         ret = write_temp_extent_buffer(fd, buf, fs_bytenr);
 702 out:
 703         free(buf);
 704         return ret;
 705 }
 706
 707 static int setup_temp_csum_tree(int fd, struct btrfs_mkfs_config *cfg,
 708                                 u64 csum_bytenr)
 709 {
 710         struct extent_buffer *buf = NULL;
 711         int ret;
 712
 713         buf = malloc(sizeof(*buf) + cfg->nodesize);
 714         if (!buf)
 715                 return -ENOMEM;
 716         ret = setup_temp_extent_buffer(buf, cfg, csum_bytenr,
 717                                        BTRFS_CSUM_TREE_OBJECTID);
 718         if (ret < 0)
 719                 goto out;
 720         /*
 721          * Temporary csum tree is completely empty.
 722          */
 723         ret = write_temp_extent_buffer(fd, buf, csum_bytenr);
 724 out:
 725         free(buf);
 726         return ret;
 727 }
 728
 729 /*
 730  * Insert one temporary extent item.
 731  *
 732  * NOTE: if skinny_metadata is not enabled, this function must be called
 733  * after all other trees are initialized.
 734  * Or fs without skinny-metadata will be screwed up.
 735  */
 736 static int insert_temp_extent_item(int fd, struct extent_buffer *buf,
 737                                    struct btrfs_mkfs_config *cfg,
 738                                    int *slot, u32 *itemoff, u64 bytenr,
 739                                    u64 ref_root)
 740 {
 741         struct extent_buffer *tmp;
 742         struct btrfs_extent_item *ei;
 743         struct btrfs_extent_inline_ref *iref;
 744         struct btrfs_disk_key disk_key;
 745         struct btrfs_disk_key tree_info_key;
 746         struct btrfs_tree_block_info *info;
 747         int itemsize;
 748         int skinny_metadata = cfg->features &
 749                               BTRFS_FEATURE_INCOMPAT_SKINNY_METADATA;
 750         int ret;
 751
 752         if (skinny_metadata)
 753                 itemsize = sizeof(*ei) + sizeof(*iref);
 754         else
 755                 itemsize = sizeof(*ei) + sizeof(*iref) +
 756                            sizeof(struct btrfs_tree_block_info);
 757
 758         btrfs_set_header_nritems(buf, *slot + 1);
 759         *(itemoff) -= itemsize;
 760
 761         if (skinny_metadata) {
 762                 btrfs_set_disk_key_type(&disk_key, BTRFS_METADATA_ITEM_KEY);
 763                 btrfs_set_disk_key_offset(&disk_key, 0);
 764         } else {
 765                 btrfs_set_disk_key_type(&disk_key, BTRFS_EXTENT_ITEM_KEY);
 766                 btrfs_set_disk_key_offset(&disk_key, cfg->nodesize);
 767         }
 768         btrfs_set_disk_key_objectid(&disk_key, bytenr);
 769
 770         btrfs_set_item_key(buf, &disk_key, *slot);
 771         btrfs_set_item_offset(buf, btrfs_item_nr(*slot), *itemoff);
 772         btrfs_set_item_size(buf, btrfs_item_nr(*slot), itemsize);
 773
 774         ei = btrfs_item_ptr(buf, *slot, struct btrfs_extent_item);
 775         btrfs_set_extent_refs(buf, ei, 1);
 776         btrfs_set_extent_generation(buf, ei, 1);
 777         btrfs_set_extent_flags(buf, ei, BTRFS_EXTENT_FLAG_TREE_BLOCK);
 778
 779         if (skinny_metadata) {
 780                 iref = (struct btrfs_extent_inline_ref *)(ei + 1);
 781         } else {
 782                 info = (struct btrfs_tree_block_info *)(ei + 1);
 783                 iref = (struct btrfs_extent_inline_ref *)(info + 1);
 784         }
 785         btrfs_set_extent_inline_ref_type(buf, iref,
 786                                          BTRFS_TREE_BLOCK_REF_KEY);
 787         btrfs_set_extent_inline_ref_offset(buf, iref, ref_root);
 788
 789         (*slot)++;
 790         if (skinny_metadata)
 791                 return 0;
 792
 793         /*
 794          * Lastly, check the tree block key by read the tree block
 795          * Since we do 1:1 mapping for convert case, we can directly
 796          * read the bytenr from disk
 797          */
 798         tmp = malloc(sizeof(*tmp) + cfg->nodesize);
 799         if (!tmp)
 800                 return -ENOMEM;
 801         ret = setup_temp_extent_buffer(tmp, cfg, bytenr, ref_root);
 802         if (ret < 0)
 803                 goto out;
 804         ret = pread(fd, tmp->data, cfg->nodesize, bytenr);
 805         if (ret < cfg->nodesize) {
 806                 ret = (ret < 0 ? -errno : -EIO);
 807                 goto out;
 808         }
 809         if (btrfs_header_nritems(tmp) == 0) {
 810                 btrfs_set_disk_key_type(&tree_info_key, 0);
 811                 btrfs_set_disk_key_objectid(&tree_info_key, 0);
 812                 btrfs_set_disk_key_offset(&tree_info_key, 0);
 813         } else {
 814                 btrfs_item_key(tmp, &tree_info_key, 0);
 815         }
 816         btrfs_set_tree_block_key(buf, info, &tree_info_key);
 817
 818 out:
 819         free(tmp);
 820         return ret;
 821 }
 822
 823 static void insert_temp_block_group(struct extent_buffer *buf,
 824                                    struct btrfs_mkfs_config *cfg,
 825                                    int *slot, u32 *itemoff,
 826                                    u64 bytenr, u64 len, u64 used, u64 flag)
 827 {
 828         struct btrfs_block_group_item bgi;
 829         struct btrfs_disk_key disk_key;
 830
 831         btrfs_set_header_nritems(buf, *slot + 1);
 832         (*itemoff) -= sizeof(bgi);
 833         btrfs_set_disk_key_type(&disk_key, BTRFS_BLOCK_GROUP_ITEM_KEY);
 834         btrfs_set_disk_key_objectid(&disk_key, bytenr);
 835         btrfs_set_disk_key_offset(&disk_key, len);
 836         btrfs_set_item_key(buf, &disk_key, *slot);
 837         btrfs_set_item_offset(buf, btrfs_item_nr(*slot), *itemoff);
 838         btrfs_set_item_size(buf, btrfs_item_nr(*slot), sizeof(bgi));
 839
 840         btrfs_set_block_group_flags(&bgi, flag);
 841         btrfs_set_block_group_used(&bgi, used);
 842         btrfs_set_block_group_chunk_objectid(&bgi,
 843                         BTRFS_FIRST_CHUNK_TREE_OBJECTID);
 844         write_extent_buffer(buf, &bgi, btrfs_item_ptr_offset(buf, *slot),
 845                             sizeof(bgi));
 846         (*slot)++;
 847 }
 848
 849 static int setup_temp_extent_tree(int fd, struct btrfs_mkfs_config *cfg,
 850                                   u64 chunk_bytenr, u64 root_bytenr,
 851                                   u64 extent_bytenr, u64 dev_bytenr,
 852                                   u64 fs_bytenr, u64 csum_bytenr)
 853 {
 854         struct extent_buffer *buf = NULL;
 855         u32 itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize);
 856         int slot = 0;
 857         int ret;
 858
 859         /*
 860          * We must ensure provided bytenr are in ascending order,
 861          * or extent tree key order will be broken.
 862          */
 863         if (!(chunk_bytenr < root_bytenr && root_bytenr < extent_bytenr &&
 864               extent_bytenr < dev_bytenr && dev_bytenr < fs_bytenr &&
 865               fs_bytenr < csum_bytenr)) {
 866                 error("bad tree bytenr order: "
 867                                 "chunk < root %llu < %llu, "
 868                                 "root < extent %llu < %llu, "
 869                                 "extent < dev %llu < %llu, "
 870                                 "dev < fs %llu < %llu, "
 871                                 "fs < csum %llu < %llu",
 872                                 (unsigned long long)chunk_bytenr,
 873                                 (unsigned long long)root_bytenr,
 874                                 (unsigned long long)root_bytenr,
 875                                 (unsigned long long)extent_bytenr,
 876                                 (unsigned long long)extent_bytenr,
 877                                 (unsigned long long)dev_bytenr,
 878                                 (unsigned long long)dev_bytenr,
 879                                 (unsigned long long)fs_bytenr,
 880                                 (unsigned long long)fs_bytenr,
 881                                 (unsigned long long)csum_bytenr);
 882                 return -EINVAL;
 883         }
 884         buf = malloc(sizeof(*buf) + cfg->nodesize);
 885         if (!buf)
 886                 return -ENOMEM;
 887
 888         ret = setup_temp_extent_buffer(buf, cfg, extent_bytenr,
 889                                        BTRFS_EXTENT_TREE_OBJECTID);
 890         if (ret < 0)
 891                 goto out;
 892
 893         ret = insert_temp_extent_item(fd, buf, cfg, &slot, &itemoff,
 894                         chunk_bytenr, BTRFS_CHUNK_TREE_OBJECTID);
 895         if (ret < 0)
 896                 goto out;
 897
 898         insert_temp_block_group(buf, cfg, &slot, &itemoff, chunk_bytenr,
 899                         BTRFS_MKFS_SYSTEM_GROUP_SIZE, cfg->nodesize,
 900                         BTRFS_BLOCK_GROUP_SYSTEM);
 901
 902         ret = insert_temp_extent_item(fd, buf, cfg, &slot, &itemoff,
 903                         root_bytenr, BTRFS_ROOT_TREE_OBJECTID);
 904         if (ret < 0)
 905                 goto out;
 906
 907         /* 5 tree block used, root, extent, dev, fs and csum*/
 908         insert_temp_block_group(buf, cfg, &slot, &itemoff, root_bytenr,
 909                         BTRFS_CONVERT_META_GROUP_SIZE, cfg->nodesize * 5,
 910                         BTRFS_BLOCK_GROUP_METADATA);
 911
 912         ret = insert_temp_extent_item(fd, buf, cfg, &slot, &itemoff,
 913                         extent_bytenr, BTRFS_EXTENT_TREE_OBJECTID);
 914         if (ret < 0)
 915                 goto out;
 916         ret = insert_temp_extent_item(fd, buf, cfg, &slot, &itemoff,
 917                         dev_bytenr, BTRFS_DEV_TREE_OBJECTID);
 918         if (ret < 0)
 919                 goto out;
 920         ret = insert_temp_extent_item(fd, buf, cfg, &slot, &itemoff,
 921                         fs_bytenr, BTRFS_FS_TREE_OBJECTID);
 922         if (ret < 0)
 923                 goto out;
 924         ret = insert_temp_extent_item(fd, buf, cfg, &slot, &itemoff,
 925                         csum_bytenr, BTRFS_CSUM_TREE_OBJECTID);
 926         if (ret < 0)
 927                 goto out;
 928
 929         ret = write_temp_extent_buffer(fd, buf, extent_bytenr);
 930 out:
 931         free(buf);
 932         return ret;
 933 }
 934
 935 /*
 936  * Improved version of make_btrfs().
 937  *
 938  * This one will
 939  * 1) Do chunk allocation to avoid used data
 940  *    And after this function, extent type matches chunk type
 941  * 2) Better structured code
 942  *    No super long hand written codes to initialized all tree blocks
 943  *    Split into small blocks and reuse codes.
 944  *    TODO: Reuse tree operation facilities by introducing new flags
 945  */
 946 static int make_convert_btrfs(int fd, struct btrfs_mkfs_config *cfg,
 947                               struct btrfs_convert_context *cctx)
 948 {
 949         struct cache_tree *free = &cctx->free;
 950         struct cache_tree *used = &cctx->used;
 951         u64 sys_chunk_start;
 952         u64 meta_chunk_start;
 953         /* chunk tree bytenr, in system chunk */
 954         u64 chunk_bytenr;
 955         /* metadata trees bytenr, in metadata chunk */
 956         u64 root_bytenr;
 957         u64 extent_bytenr;
 958         u64 dev_bytenr;
 959         u64 fs_bytenr;
 960         u64 csum_bytenr;
 961         int ret;
 962
 963         /* Shouldn't happen */
 964         BUG_ON(cache_tree_empty(used));
 965
 966         /*
 967          * reserve space for temporary superblock first
 968          * Here we allocate a little larger space, to keep later
 969          * free space will be STRIPE_LEN aligned
 970          */
 971         ret = reserve_free_space(free, BTRFS_STRIPE_LEN,
 972                                  &cfg->super_bytenr);
 973         if (ret < 0)
 974                 goto out;
 975
 976         /*
 977          * Then reserve system chunk space
 978          * TODO: Change system group size depending on cctx->total_bytes.
 979          * If using current 4M, it can only handle less than one TB for
 980          * worst case and then run out of sys space.
 981          */
 982         ret = reserve_free_space(free, BTRFS_MKFS_SYSTEM_GROUP_SIZE,
 983                                  &sys_chunk_start);
 984         if (ret < 0)
 985                 goto out;
 986         ret = reserve_free_space(free, BTRFS_CONVERT_META_GROUP_SIZE,
 987                                  &meta_chunk_start);
 988         if (ret < 0)
 989                 goto out;
 990
 991         /*
 992          * Allocated meta/sys chunks will be mapped 1:1 with device offset.
 993          *
 994          * Inside the allocated metadata chunk, the layout will be:
 995          *  | offset            | contents      |
 996          *  -------------------------------------
 997          *  | +0                | tree root     |
 998          *  | +nodesize         | extent root   |
 999          *  | +nodesize * 2     | device root   |
1000          *  | +nodesize * 3     | fs tree       |
1001          *  | +nodesize * 4     | csum tree     |
1002          *  -------------------------------------
1003          * Inside the allocated system chunk, the layout will be:
1004          *  | offset            | contents      |
1005          *  -------------------------------------
1006          *  | +0                | chunk root    |
1007          *  -------------------------------------
1008          */
1009         chunk_bytenr = sys_chunk_start;
1010         root_bytenr = meta_chunk_start;
1011         extent_bytenr = meta_chunk_start + cfg->nodesize;
1012         dev_bytenr = meta_chunk_start + cfg->nodesize * 2;
1013         fs_bytenr = meta_chunk_start + cfg->nodesize * 3;
1014         csum_bytenr = meta_chunk_start + cfg->nodesize * 4;
1015
1016         ret = setup_temp_super(fd, cfg, root_bytenr, chunk_bytenr);
1017         if (ret < 0)
1018                 goto out;
1019
1020         ret = setup_temp_root_tree(fd, cfg, root_bytenr, extent_bytenr,
1021                                    dev_bytenr, fs_bytenr, csum_bytenr);
1022         if (ret < 0)
1023                 goto out;
1024         ret = setup_temp_chunk_tree(fd, cfg, sys_chunk_start, meta_chunk_start,
1025                                     chunk_bytenr);
1026         if (ret < 0)
1027                 goto out;
1028         ret = setup_temp_dev_tree(fd, cfg, sys_chunk_start, meta_chunk_start,
1029                                   dev_bytenr);
1030         if (ret < 0)
1031                 goto out;
1032         ret = setup_temp_fs_tree(fd, cfg, fs_bytenr);
1033         if (ret < 0)
1034                 goto out;
1035         ret = setup_temp_csum_tree(fd, cfg, csum_bytenr);
1036         if (ret < 0)
1037                 goto out;
1038         /*
1039          * Setup extent tree last, since it may need to read tree block key
1040          * for non-skinny metadata case.
1041          */
1042         ret = setup_temp_extent_tree(fd, cfg, chunk_bytenr, root_bytenr,
1043                                      extent_bytenr, dev_bytenr, fs_bytenr,
1044                                      csum_bytenr);
1045 out:
1046         return ret;
1047 }
1048
1049 /*
1050  * @fs_uuid - if NULL, generates a UUID, returns back the new filesystem UUID
1051  *
1052  * The superblock signature is not valid, denotes a partially created
1053  * filesystem, needs to be finalized.
1054  */
1055 int make_btrfs(int fd, struct btrfs_mkfs_config *cfg,
1056                 struct btrfs_convert_context *cctx)
1057 {
1058         struct btrfs_super_block super;
1059         struct extent_buffer *buf;
1060         struct btrfs_root_item root_item;
1061         struct btrfs_disk_key disk_key;
1062         struct btrfs_extent_item *extent_item;
1063         struct btrfs_inode_item *inode_item;
1064         struct btrfs_chunk *chunk;
1065         struct btrfs_dev_item *dev_item;
1066         struct btrfs_dev_extent *dev_extent;
1067         u8 chunk_tree_uuid[BTRFS_UUID_SIZE];
1068         u8 *ptr;
1069         int i;
1070         int ret;
1071         u32 itemoff;
1072         u32 nritems = 0;
1073         u64 first_free;
1074         u64 ref_root;
1075         u32 array_size;
1076         u32 item_size;
1077         int skinny_metadata = !!(cfg->features &
1078                                  BTRFS_FEATURE_INCOMPAT_SKINNY_METADATA);
1079         u64 num_bytes;
1080
1081         if (cctx)
1082                 return make_convert_btrfs(fd, cfg, cctx);
1083         buf = malloc(sizeof(*buf) + max(cfg->sectorsize, cfg->nodesize));
1084         if (!buf)
1085                 return -ENOMEM;
1086
1087         first_free = BTRFS_SUPER_INFO_OFFSET + cfg->sectorsize * 2 - 1;
1088         first_free &= ~((u64)cfg->sectorsize - 1);
1089
1090         memset(&super, 0, sizeof(super));
1091
1092         num_bytes = (cfg->num_bytes / cfg->sectorsize) * cfg->sectorsize;
1093         if (*cfg->fs_uuid) {
1094                 if (uuid_parse(cfg->fs_uuid, super.fsid) != 0) {
1095                         error("cannot not parse UUID: %s", cfg->fs_uuid);
1096                         ret = -EINVAL;
1097                         goto out;
1098                 }
1099                 if (!test_uuid_unique(cfg->fs_uuid)) {
1100                         error("non-unique UUID: %s", cfg->fs_uuid);
1101                         ret = -EBUSY;
1102                         goto out;
1103                 }
1104         } else {
1105                 uuid_generate(super.fsid);
1106                 uuid_unparse(super.fsid, cfg->fs_uuid);
1107         }
1108         uuid_generate(super.dev_item.uuid);
1109         uuid_generate(chunk_tree_uuid);
1110
1111         btrfs_set_super_bytenr(&super, cfg->blocks[0]);
1112         btrfs_set_super_num_devices(&super, 1);
1113         btrfs_set_super_magic(&super, BTRFS_MAGIC_PARTIAL);
1114         btrfs_set_super_generation(&super, 1);
1115         btrfs_set_super_root(&super, cfg->blocks[1]);
1116         btrfs_set_super_chunk_root(&super, cfg->blocks[3]);
1117         btrfs_set_super_total_bytes(&super, num_bytes);
1118         btrfs_set_super_bytes_used(&super, 6 * cfg->nodesize);
1119         btrfs_set_super_sectorsize(&super, cfg->sectorsize);
1120         btrfs_set_super_leafsize(&super, cfg->nodesize);
1121         btrfs_set_super_nodesize(&super, cfg->nodesize);
1122         btrfs_set_super_stripesize(&super, cfg->stripesize);
1123         btrfs_set_super_csum_type(&super, BTRFS_CSUM_TYPE_CRC32);
1124         btrfs_set_super_chunk_root_generation(&super, 1);
1125         btrfs_set_super_cache_generation(&super, -1);
1126         btrfs_set_super_incompat_flags(&super, cfg->features);
1127         if (cfg->label)
1128                 __strncpy_null(super.label, cfg->label, BTRFS_LABEL_SIZE - 1);
1129
1130         /* create the tree of root objects */
1131         memset(buf->data, 0, cfg->nodesize);
1132         buf->len = cfg->nodesize;
1133         btrfs_set_header_bytenr(buf, cfg->blocks[1]);
1134         btrfs_set_header_nritems(buf, 4);
1135         btrfs_set_header_generation(buf, 1);
1136         btrfs_set_header_backref_rev(buf, BTRFS_MIXED_BACKREF_REV);
1137         btrfs_set_header_owner(buf, BTRFS_ROOT_TREE_OBJECTID);
1138         write_extent_buffer(buf, super.fsid, btrfs_header_fsid(),
1139                             BTRFS_FSID_SIZE);
1140
1141         write_extent_buffer(buf, chunk_tree_uuid,
1142                             btrfs_header_chunk_tree_uuid(buf),
1143                             BTRFS_UUID_SIZE);
1144
1145         /* create the items for the root tree */
1146         memset(&root_item, 0, sizeof(root_item));
1147         inode_item = &root_item.inode;
1148         btrfs_set_stack_inode_generation(inode_item, 1);
1149         btrfs_set_stack_inode_size(inode_item, 3);
1150         btrfs_set_stack_inode_nlink(inode_item, 1);
1151         btrfs_set_stack_inode_nbytes(inode_item, cfg->nodesize);
1152         btrfs_set_stack_inode_mode(inode_item, S_IFDIR | 0755);
1153         btrfs_set_root_refs(&root_item, 1);
1154         btrfs_set_root_used(&root_item, cfg->nodesize);
1155         btrfs_set_root_generation(&root_item, 1);
1156
1157         memset(&disk_key, 0, sizeof(disk_key));
1158         btrfs_set_disk_key_type(&disk_key, BTRFS_ROOT_ITEM_KEY);
1159         btrfs_set_disk_key_offset(&disk_key, 0);
1160         nritems = 0;
1161
1162         itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize) - sizeof(root_item);
1163         btrfs_set_root_bytenr(&root_item, cfg->blocks[2]);
1164         btrfs_set_disk_key_objectid(&disk_key, BTRFS_EXTENT_TREE_OBJECTID);
1165         btrfs_set_item_key(buf, &disk_key, nritems);
1166         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1167         btrfs_set_item_size(buf, btrfs_item_nr(nritems),
1168                             sizeof(root_item));
1169         write_extent_buffer(buf, &root_item, btrfs_item_ptr_offset(buf,
1170                             nritems), sizeof(root_item));
1171         nritems++;
1172
1173         itemoff = itemoff - sizeof(root_item);
1174         btrfs_set_root_bytenr(&root_item, cfg->blocks[4]);
1175         btrfs_set_disk_key_objectid(&disk_key, BTRFS_DEV_TREE_OBJECTID);
1176         btrfs_set_item_key(buf, &disk_key, nritems);
1177         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1178         btrfs_set_item_size(buf, btrfs_item_nr(nritems),
1179                             sizeof(root_item));
1180         write_extent_buffer(buf, &root_item,
1181                             btrfs_item_ptr_offset(buf, nritems),
1182                             sizeof(root_item));
1183         nritems++;
1184
1185         itemoff = itemoff - sizeof(root_item);
1186         btrfs_set_root_bytenr(&root_item, cfg->blocks[5]);
1187         btrfs_set_disk_key_objectid(&disk_key, BTRFS_FS_TREE_OBJECTID);
1188         btrfs_set_item_key(buf, &disk_key, nritems);
1189         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1190         btrfs_set_item_size(buf, btrfs_item_nr(nritems),
1191                             sizeof(root_item));
1192         write_extent_buffer(buf, &root_item,
1193                             btrfs_item_ptr_offset(buf, nritems),
1194                             sizeof(root_item));
1195         nritems++;
1196
1197         itemoff = itemoff - sizeof(root_item);
1198         btrfs_set_root_bytenr(&root_item, cfg->blocks[6]);
1199         btrfs_set_disk_key_objectid(&disk_key, BTRFS_CSUM_TREE_OBJECTID);
1200         btrfs_set_item_key(buf, &disk_key, nritems);
1201         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1202         btrfs_set_item_size(buf, btrfs_item_nr(nritems),
1203                             sizeof(root_item));
1204         write_extent_buffer(buf, &root_item,
1205                             btrfs_item_ptr_offset(buf, nritems),
1206                             sizeof(root_item));
1207         nritems++;
1208
1209
1210         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1211         ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[1]);
1212         if (ret != cfg->nodesize) {
1213                 ret = (ret < 0 ? -errno : -EIO);
1214                 goto out;
1215         }
1216
1217         /* create the items for the extent tree */
1218         memset(buf->data + sizeof(struct btrfs_header), 0,
1219                 cfg->nodesize - sizeof(struct btrfs_header));
1220         nritems = 0;
1221         itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize);
1222         for (i = 1; i < 7; i++) {
1223                 item_size = sizeof(struct btrfs_extent_item);
1224                 if (!skinny_metadata)
1225                         item_size += sizeof(struct btrfs_tree_block_info);
1226
1227                 if (cfg->blocks[i] < first_free) {
1228                         error("block[%d] below first free: %llu < %llu",
1229                                         i, (unsigned long long)cfg->blocks[i],
1230                                         (unsigned long long)first_free);
1231                         ret = -EINVAL;
1232                         goto out;
1233                 }
1234                 if (cfg->blocks[i] < cfg->blocks[i - 1]) {
1235                         error("blocks %d and %d in reverse order: %llu < %llu",
1236                                 i, i - 1,
1237                                 (unsigned long long)cfg->blocks[i],
1238                                 (unsigned long long)cfg->blocks[i - 1]);
1239                         ret = -EINVAL;
1240                         goto out;
1241                 }
1242
1243                 /* create extent item */
1244                 itemoff -= item_size;
1245                 btrfs_set_disk_key_objectid(&disk_key, cfg->blocks[i]);
1246                 if (skinny_metadata) {
1247                         btrfs_set_disk_key_type(&disk_key,
1248                                                 BTRFS_METADATA_ITEM_KEY);
1249                         btrfs_set_disk_key_offset(&disk_key, 0);
1250                 } else {
1251                         btrfs_set_disk_key_type(&disk_key,
1252                                                 BTRFS_EXTENT_ITEM_KEY);
1253                         btrfs_set_disk_key_offset(&disk_key, cfg->nodesize);
1254                 }
1255                 btrfs_set_item_key(buf, &disk_key, nritems);
1256                 btrfs_set_item_offset(buf, btrfs_item_nr(nritems),
1257                                       itemoff);
1258                 btrfs_set_item_size(buf, btrfs_item_nr(nritems),
1259                                     item_size);
1260                 extent_item = btrfs_item_ptr(buf, nritems,
1261                                              struct btrfs_extent_item);
1262                 btrfs_set_extent_refs(buf, extent_item, 1);
1263                 btrfs_set_extent_generation(buf, extent_item, 1);
1264                 btrfs_set_extent_flags(buf, extent_item,
1265                                        BTRFS_EXTENT_FLAG_TREE_BLOCK);
1266                 nritems++;
1267
1268                 /* create extent ref */
1269                 ref_root = reference_root_table[i];
1270                 btrfs_set_disk_key_objectid(&disk_key, cfg->blocks[i]);
1271                 btrfs_set_disk_key_offset(&disk_key, ref_root);
1272                 btrfs_set_disk_key_type(&disk_key, BTRFS_TREE_BLOCK_REF_KEY);
1273                 btrfs_set_item_key(buf, &disk_key, nritems);
1274                 btrfs_set_item_offset(buf, btrfs_item_nr(nritems),
1275                                       itemoff);
1276                 btrfs_set_item_size(buf, btrfs_item_nr(nritems), 0);
1277                 nritems++;
1278         }
1279         btrfs_set_header_bytenr(buf, cfg->blocks[2]);
1280         btrfs_set_header_owner(buf, BTRFS_EXTENT_TREE_OBJECTID);
1281         btrfs_set_header_nritems(buf, nritems);
1282         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1283         ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[2]);
1284         if (ret != cfg->nodesize) {
1285                 ret = (ret < 0 ? -errno : -EIO);
1286                 goto out;
1287         }
1288
1289         /* create the chunk tree */
1290         memset(buf->data + sizeof(struct btrfs_header), 0,
1291                 cfg->nodesize - sizeof(struct btrfs_header));
1292         nritems = 0;
1293         item_size = sizeof(*dev_item);
1294         itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize) - item_size;
1295
1296         /* first device 1 (there is no device 0) */
1297         btrfs_set_disk_key_objectid(&disk_key, BTRFS_DEV_ITEMS_OBJECTID);
1298         btrfs_set_disk_key_offset(&disk_key, 1);
1299         btrfs_set_disk_key_type(&disk_key, BTRFS_DEV_ITEM_KEY);
1300         btrfs_set_item_key(buf, &disk_key, nritems);
1301         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1302         btrfs_set_item_size(buf, btrfs_item_nr(nritems), item_size);
1303
1304         dev_item = btrfs_item_ptr(buf, nritems, struct btrfs_dev_item);
1305         btrfs_set_device_id(buf, dev_item, 1);
1306         btrfs_set_device_generation(buf, dev_item, 0);
1307         btrfs_set_device_total_bytes(buf, dev_item, num_bytes);
1308         btrfs_set_device_bytes_used(buf, dev_item,
1309                                     BTRFS_MKFS_SYSTEM_GROUP_SIZE);
1310         btrfs_set_device_io_align(buf, dev_item, cfg->sectorsize);
1311         btrfs_set_device_io_width(buf, dev_item, cfg->sectorsize);
1312         btrfs_set_device_sector_size(buf, dev_item, cfg->sectorsize);
1313         btrfs_set_device_type(buf, dev_item, 0);
1314
1315         write_extent_buffer(buf, super.dev_item.uuid,
1316                             (unsigned long)btrfs_device_uuid(dev_item),
1317                             BTRFS_UUID_SIZE);
1318         write_extent_buffer(buf, super.fsid,
1319                             (unsigned long)btrfs_device_fsid(dev_item),
1320                             BTRFS_UUID_SIZE);
1321         read_extent_buffer(buf, &super.dev_item, (unsigned long)dev_item,
1322                            sizeof(*dev_item));
1323
1324         nritems++;
1325         item_size = btrfs_chunk_item_size(1);
1326         itemoff = itemoff - item_size;
1327
1328         /* then we have chunk 0 */
1329         btrfs_set_disk_key_objectid(&disk_key, BTRFS_FIRST_CHUNK_TREE_OBJECTID);
1330         btrfs_set_disk_key_offset(&disk_key, 0);
1331         btrfs_set_disk_key_type(&disk_key, BTRFS_CHUNK_ITEM_KEY);
1332         btrfs_set_item_key(buf, &disk_key, nritems);
1333         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1334         btrfs_set_item_size(buf, btrfs_item_nr(nritems), item_size);
1335
1336         chunk = btrfs_item_ptr(buf, nritems, struct btrfs_chunk);
1337         btrfs_set_chunk_length(buf, chunk, BTRFS_MKFS_SYSTEM_GROUP_SIZE);
1338         btrfs_set_chunk_owner(buf, chunk, BTRFS_EXTENT_TREE_OBJECTID);
1339         btrfs_set_chunk_stripe_len(buf, chunk, 64 * 1024);
1340         btrfs_set_chunk_type(buf, chunk, BTRFS_BLOCK_GROUP_SYSTEM);
1341         btrfs_set_chunk_io_align(buf, chunk, cfg->sectorsize);
1342         btrfs_set_chunk_io_width(buf, chunk, cfg->sectorsize);
1343         btrfs_set_chunk_sector_size(buf, chunk, cfg->sectorsize);
1344         btrfs_set_chunk_num_stripes(buf, chunk, 1);
1345         btrfs_set_stripe_devid_nr(buf, chunk, 0, 1);
1346         btrfs_set_stripe_offset_nr(buf, chunk, 0, 0);
1347         nritems++;
1348
1349         write_extent_buffer(buf, super.dev_item.uuid,
1350                             (unsigned long)btrfs_stripe_dev_uuid(&chunk->stripe),
1351                             BTRFS_UUID_SIZE);
1352
1353         /* copy the key for the chunk to the system array */
1354         ptr = super.sys_chunk_array;
1355         array_size = sizeof(disk_key);
1356
1357         memcpy(ptr, &disk_key, sizeof(disk_key));
1358         ptr += sizeof(disk_key);
1359
1360         /* copy the chunk to the system array */
1361         read_extent_buffer(buf, ptr, (unsigned long)chunk, item_size);
1362         array_size += item_size;
1363         ptr += item_size;
1364         btrfs_set_super_sys_array_size(&super, array_size);
1365
1366         btrfs_set_header_bytenr(buf, cfg->blocks[3]);
1367         btrfs_set_header_owner(buf, BTRFS_CHUNK_TREE_OBJECTID);
1368         btrfs_set_header_nritems(buf, nritems);
1369         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1370         ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[3]);
1371         if (ret != cfg->nodesize) {
1372                 ret = (ret < 0 ? -errno : -EIO);
1373                 goto out;
1374         }
1375
1376         /* create the device tree */
1377         memset(buf->data + sizeof(struct btrfs_header), 0,
1378                 cfg->nodesize - sizeof(struct btrfs_header));
1379         nritems = 0;
1380         itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize) -
1381                 sizeof(struct btrfs_dev_extent);
1382
1383         btrfs_set_disk_key_objectid(&disk_key, 1);
1384         btrfs_set_disk_key_offset(&disk_key, 0);
1385         btrfs_set_disk_key_type(&disk_key, BTRFS_DEV_EXTENT_KEY);
1386         btrfs_set_item_key(buf, &disk_key, nritems);
1387         btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
1388         btrfs_set_item_size(buf, btrfs_item_nr(nritems),
1389                             sizeof(struct btrfs_dev_extent));
1390         dev_extent = btrfs_item_ptr(buf, nritems, struct btrfs_dev_extent);
1391         btrfs_set_dev_extent_chunk_tree(buf, dev_extent,
1392                                         BTRFS_CHUNK_TREE_OBJECTID);
1393         btrfs_set_dev_extent_chunk_objectid(buf, dev_extent,
1394                                         BTRFS_FIRST_CHUNK_TREE_OBJECTID);
1395         btrfs_set_dev_extent_chunk_offset(buf, dev_extent, 0);
1396
1397         write_extent_buffer(buf, chunk_tree_uuid,
1398                     (unsigned long)btrfs_dev_extent_chunk_tree_uuid(dev_extent),
1399                     BTRFS_UUID_SIZE);
1400
1401         btrfs_set_dev_extent_length(buf, dev_extent,
1402                                     BTRFS_MKFS_SYSTEM_GROUP_SIZE);
1403         nritems++;
1404
1405         btrfs_set_header_bytenr(buf, cfg->blocks[4]);
1406         btrfs_set_header_owner(buf, BTRFS_DEV_TREE_OBJECTID);
1407         btrfs_set_header_nritems(buf, nritems);
1408         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1409         ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[4]);
1410         if (ret != cfg->nodesize) {
1411                 ret = (ret < 0 ? -errno : -EIO);
1412                 goto out;
1413         }
1414
1415         /* create the FS root */
1416         memset(buf->data + sizeof(struct btrfs_header), 0,
1417                 cfg->nodesize - sizeof(struct btrfs_header));
1418         btrfs_set_header_bytenr(buf, cfg->blocks[5]);
1419         btrfs_set_header_owner(buf, BTRFS_FS_TREE_OBJECTID);
1420         btrfs_set_header_nritems(buf, 0);
1421         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1422         ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[5]);
1423         if (ret != cfg->nodesize) {
1424                 ret = (ret < 0 ? -errno : -EIO);
1425                 goto out;
1426         }
1427         /* finally create the csum root */
1428         memset(buf->data + sizeof(struct btrfs_header), 0,
1429                 cfg->nodesize - sizeof(struct btrfs_header));
1430         btrfs_set_header_bytenr(buf, cfg->blocks[6]);
1431         btrfs_set_header_owner(buf, BTRFS_CSUM_TREE_OBJECTID);
1432         btrfs_set_header_nritems(buf, 0);
1433         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1434         ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[6]);
1435         if (ret != cfg->nodesize) {
1436                 ret = (ret < 0 ? -errno : -EIO);
1437                 goto out;
1438         }
1439
1440         /* and write out the super block */
1441         memset(buf->data, 0, BTRFS_SUPER_INFO_SIZE);
1442         memcpy(buf->data, &super, sizeof(super));
1443         buf->len = BTRFS_SUPER_INFO_SIZE;
1444         csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
1445         ret = pwrite(fd, buf->data, BTRFS_SUPER_INFO_SIZE, cfg->blocks[0]);
1446         if (ret != BTRFS_SUPER_INFO_SIZE) {
1447                 ret = (ret < 0 ? -errno : -EIO);
1448                 goto out;
1449         }
1450
1451         ret = 0;
1452
1453 out:
1454         free(buf);
1455         return ret;
1456 }
1457
1458 #define VERSION_TO_STRING3(a,b,c)       #a "." #b "." #c, KERNEL_VERSION(a,b,c)
1459 #define VERSION_TO_STRING2(a,b)         #a "." #b, KERNEL_VERSION(a,b,0)
1460
1461 /*
1462  * Feature stability status and versions: compat <= safe <= default
1463  */
1464 static const struct btrfs_fs_feature {
1465         const char *name;
1466         u64 flag;
1467         const char *sysfs_name;
1468         /*
1469          * Compatibility with kernel of given version. Filesystem can be
1470          * mounted.
1471          */
1472         const char *compat_str;
1473         u32 compat_ver;
1474         /*
1475          * Considered safe for use, but is not on by default, even if the
1476          * kernel supports the feature.
1477          */
1478         const char *safe_str;
1479         u32 safe_ver;
1480         /*
1481          * Considered safe for use and will be turned on by default if
1482          * supported by the running kernel.
1483          */
1484         const char *default_str;
1485         u32 default_ver;
1486         const char *desc;
1487 } mkfs_features[] = {
1488         { "mixed-bg", BTRFS_FEATURE_INCOMPAT_MIXED_GROUPS,
1489                 "mixed_groups",
1490                 VERSION_TO_STRING3(2,6,37),
1491                 VERSION_TO_STRING3(2,6,37),
1492                 NULL, 0,
1493                 "mixed data and metadata block groups" },
1494         { "extref", BTRFS_FEATURE_INCOMPAT_EXTENDED_IREF,
1495                 "extended_iref",
1496                 VERSION_TO_STRING2(3,7),
1497                 VERSION_TO_STRING2(3,12),
1498                 VERSION_TO_STRING2(3,12),
1499                 "increased hardlink limit per file to 65536" },
1500         { "raid56", BTRFS_FEATURE_INCOMPAT_RAID56,
1501                 "raid56",
1502                 VERSION_TO_STRING2(3,9),
1503                 NULL, 0,
1504                 NULL, 0,
1505                 "raid56 extended format" },
1506         { "skinny-metadata", BTRFS_FEATURE_INCOMPAT_SKINNY_METADATA,
1507                 "skinny_metadata",
1508                 VERSION_TO_STRING2(3,10),
1509                 VERSION_TO_STRING2(3,18),
1510                 VERSION_TO_STRING2(3,18),
1511                 "reduced-size metadata extent refs" },
1512         { "no-holes", BTRFS_FEATURE_INCOMPAT_NO_HOLES,
1513                 "no_holes",
1514                 VERSION_TO_STRING2(3,14),
1515                 VERSION_TO_STRING2(4,0),
1516                 NULL, 0,
1517                 "no explicit hole extents for files" },
1518         /* Keep this one last */
1519         { "list-all", BTRFS_FEATURE_LIST_ALL, NULL }
1520 };
1521
1522 static int parse_one_fs_feature(const char *name, u64 *flags)
1523 {
1524         int i;
1525         int found = 0;
1526
1527         for (i = 0; i < ARRAY_SIZE(mkfs_features); i++) {
1528                 if (name[0] == '^' &&
1529                         !strcmp(mkfs_features[i].name, name + 1)) {
1530                         *flags &= ~ mkfs_features[i].flag;
1531                         found = 1;
1532                 } else if (!strcmp(mkfs_features[i].name, name)) {
1533                         *flags |= mkfs_features[i].flag;
1534                         found = 1;
1535                 }
1536         }
1537
1538         return !found;
1539 }
1540
1541 void btrfs_parse_features_to_string(char *buf, u64 flags)
1542 {
1543         int i;
1544
1545         buf[0] = 0;
1546
1547         for (i = 0; i < ARRAY_SIZE(mkfs_features); i++) {
1548                 if (flags & mkfs_features[i].flag) {
1549                         if (*buf)
1550                                 strcat(buf, ", ");
1551                         strcat(buf, mkfs_features[i].name);
1552                 }
1553         }
1554 }
1555
1556 void btrfs_process_fs_features(u64 flags)
1557 {
1558         int i;
1559
1560         for (i = 0; i < ARRAY_SIZE(mkfs_features); i++) {
1561                 if (flags & mkfs_features[i].flag) {
1562                         printf("Turning ON incompat feature '%s': %s\n",
1563                                 mkfs_features[i].name,
1564                                 mkfs_features[i].desc);
1565                 }
1566         }
1567 }
1568
1569 void btrfs_list_all_fs_features(u64 mask_disallowed)
1570 {
1571         int i;
1572
1573         fprintf(stderr, "Filesystem features available:\n");
1574         for (i = 0; i < ARRAY_SIZE(mkfs_features) - 1; i++) {
1575                 const struct btrfs_fs_feature *feat = &mkfs_features[i];
1576
1577                 if (feat->flag & mask_disallowed)
1578                         continue;
1579                 fprintf(stderr, "%-20s- %s (0x%llx", feat->name, feat->desc,
1580                                 feat->flag);
1581                 if (feat->compat_ver)
1582                         fprintf(stderr, ", compat=%s", feat->compat_str);
1583                 if (feat->safe_ver)
1584                         fprintf(stderr, ", safe=%s", feat->safe_str);
1585                 if (feat->default_ver)
1586                         fprintf(stderr, ", default=%s", feat->default_str);
1587                 fprintf(stderr, ")\n");
1588         }
1589 }
1590
1591 /*
1592  * Return NULL if all features were parsed fine, otherwise return the name of
1593  * the first unparsed.
1594  */
1595 char* btrfs_parse_fs_features(char *namelist, u64 *flags)
1596 {
1597         char *this_char;
1598         char *save_ptr = NULL; /* Satisfy static checkers */
1599
1600         for (this_char = strtok_r(namelist, ",", &save_ptr);
1601              this_char != NULL;
1602              this_char = strtok_r(NULL, ",", &save_ptr)) {
1603                 if (parse_one_fs_feature(this_char, flags))
1604                         return this_char;
1605         }
1606
1607         return NULL;
1608 }
1609
1610 void print_kernel_version(FILE *stream, u32 version)
1611 {
1612         u32 v[3];
1613
1614         v[0] = version & 0xFF;
1615         v[1] = (version >> 8) & 0xFF;
1616         v[2] = version >> 16;
1617         fprintf(stream, "%u.%u", v[2], v[1]);
1618         if (v[0])
1619                 fprintf(stream, ".%u", v[0]);
1620 }
1621
1622 u32 get_running_kernel_version(void)
1623 {
1624         struct utsname utsbuf;
1625         char *tmp;
1626         char *saveptr = NULL;
1627         u32 version;
1628
1629         uname(&utsbuf);
1630         if (strcmp(utsbuf.sysname, "Linux") != 0) {
1631                 error("unsupported system: %s", utsbuf.sysname);
1632                 exit(1);
1633         }
1634         /* 1.2.3-4-name */
1635         tmp = strchr(utsbuf.release, '-');
1636         if (tmp)
1637                 *tmp = 0;
1638
1639         tmp = strtok_r(utsbuf.release, ".", &saveptr);
1640         if (!string_is_numerical(tmp))
1641                 return (u32)-1;
1642         version = atoi(tmp) << 16;
1643         tmp = strtok_r(NULL, ".", &saveptr);
1644         if (!string_is_numerical(tmp))
1645                 return (u32)-1;
1646         version |= atoi(tmp) << 8;
1647         tmp = strtok_r(NULL, ".", &saveptr);
1648         if (tmp) {
1649                 if (!string_is_numerical(tmp))
1650                         return (u32)-1;
1651                 version |= atoi(tmp);
1652         }
1653
1654         return version;
1655 }
1656
1657 u64 btrfs_device_size(int fd, struct stat *st)
1658 {
1659         u64 size;
1660         if (S_ISREG(st->st_mode)) {
1661                 return st->st_size;
1662         }
1663         if (!S_ISBLK(st->st_mode)) {
1664                 return 0;
1665         }
1666         if (ioctl(fd, BLKGETSIZE64, &size) >= 0) {
1667                 return size;
1668         }
1669         return 0;
1670 }
1671
1672 static int zero_blocks(int fd, off_t start, size_t len)
1673 {
1674         char *buf = malloc(len);
1675         int ret = 0;
1676         ssize_t written;
1677
1678         if (!buf)
1679                 return -ENOMEM;
1680         memset(buf, 0, len);
1681         written = pwrite(fd, buf, len, start);
1682         if (written != len)
1683                 ret = -EIO;
1684         free(buf);
1685         return ret;
1686 }
1687
1688 #define ZERO_DEV_BYTES (2 * 1024 * 1024)
1689
1690 /* don't write outside the device by clamping the region to the device size */
1691 static int zero_dev_clamped(int fd, off_t start, ssize_t len, u64 dev_size)
1692 {
1693         off_t end = max(start, start + len);
1694
1695 #ifdef __sparc__
1696         /* and don't overwrite the disk labels on sparc */
1697         start = max(start, 1024);
1698         end = max(end, 1024);
1699 #endif
1700
1701         start = min_t(u64, start, dev_size);
1702         end = min_t(u64, end, dev_size);
1703
1704         return zero_blocks(fd, start, end - start);
1705 }
1706
1707 int btrfs_add_to_fsid(struct btrfs_trans_handle *trans,
1708                       struct btrfs_root *root, int fd, const char *path,
1709                       u64 device_total_bytes, u32 io_width, u32 io_align,
1710                       u32 sectorsize)
1711 {
1712         struct btrfs_super_block *disk_super;
1713         struct btrfs_super_block *super = root->fs_info->super_copy;
1714         struct btrfs_device *device;
1715         struct btrfs_dev_item *dev_item;
1716         char *buf = NULL;
1717         u64 fs_total_bytes;
1718         u64 num_devs;
1719         int ret;
1720
1721         device_total_bytes = (device_total_bytes / sectorsize) * sectorsize;
1722
1723         device = calloc(1, sizeof(*device));
1724         if (!device) {
1725                 ret = -ENOMEM;
1726                 goto out;
1727         }
1728         buf = calloc(1, sectorsize);
1729         if (!buf) {
1730                 ret = -ENOMEM;
1731                 goto out;
1732         }
1733
1734         disk_super = (struct btrfs_super_block *)buf;
1735         dev_item = &disk_super->dev_item;
1736
1737         uuid_generate(device->uuid);
1738         device->devid = 0;
1739         device->type = 0;
1740         device->io_width = io_width;
1741         device->io_align = io_align;
1742         device->sector_size = sectorsize;
1743         device->fd = fd;
1744         device->writeable = 1;
1745         device->total_bytes = device_total_bytes;
1746         device->bytes_used = 0;
1747         device->total_ios = 0;
1748         device->dev_root = root->fs_info->dev_root;
1749         device->name = strdup(path);
1750         if (!device->name) {
1751                 ret = -ENOMEM;
1752                 goto out;
1753         }
1754
1755         INIT_LIST_HEAD(&device->dev_list);
1756         ret = btrfs_add_device(trans, root, device);
1757         if (ret)
1758                 goto out;
1759
1760         fs_total_bytes = btrfs_super_total_bytes(super) + device_total_bytes;
1761         btrfs_set_super_total_bytes(super, fs_total_bytes);
1762
1763         num_devs = btrfs_super_num_devices(super) + 1;
1764         btrfs_set_super_num_devices(super, num_devs);
1765
1766         memcpy(disk_super, super, sizeof(*disk_super));
1767
1768         btrfs_set_super_bytenr(disk_super, BTRFS_SUPER_INFO_OFFSET);
1769         btrfs_set_stack_device_id(dev_item, device->devid);
1770         btrfs_set_stack_device_type(dev_item, device->type);
1771         btrfs_set_stack_device_io_align(dev_item, device->io_align);
1772         btrfs_set_stack_device_io_width(dev_item, device->io_width);
1773         btrfs_set_stack_device_sector_size(dev_item, device->sector_size);
1774         btrfs_set_stack_device_total_bytes(dev_item, device->total_bytes);
1775         btrfs_set_stack_device_bytes_used(dev_item, device->bytes_used);
1776         memcpy(&dev_item->uuid, device->uuid, BTRFS_UUID_SIZE);
1777
1778         ret = pwrite(fd, buf, sectorsize, BTRFS_SUPER_INFO_OFFSET);
1779         BUG_ON(ret != sectorsize);
1780
1781         free(buf);
1782         list_add(&device->dev_list, &root->fs_info->fs_devices->devices);
1783         device->fs_devices = root->fs_info->fs_devices;
1784         return 0;
1785
1786 out:
1787         free(device);
1788         free(buf);
1789         return ret;
1790 }
1791
1792 static int btrfs_wipe_existing_sb(int fd)
1793 {
1794         const char *off = NULL;
1795         size_t len = 0;
1796         loff_t offset;
1797         char buf[BUFSIZ];
1798         int ret = 0;
1799         blkid_probe pr = NULL;
1800
1801         pr = blkid_new_probe();
1802         if (!pr)
1803                 return -1;
1804
1805         if (blkid_probe_set_device(pr, fd, 0, 0)) {
1806                 ret = -1;
1807                 goto out;
1808         }
1809
1810         ret = blkid_probe_lookup_value(pr, "SBMAGIC_OFFSET", &off, NULL);
1811         if (!ret)
1812                 ret = blkid_probe_lookup_value(pr, "SBMAGIC", NULL, &len);
1813
1814         if (ret || len == 0 || off == NULL) {
1815                 /*
1816                  * If lookup fails, the probe did not find any values, eg. for
1817                  * a file image or a loop device. Soft error.
1818                  */
1819                 ret = 1;
1820                 goto out;
1821         }
1822
1823         offset = strtoll(off, NULL, 10);
1824         if (len > sizeof(buf))
1825                 len = sizeof(buf);
1826
1827         memset(buf, 0, len);
1828         ret = pwrite(fd, buf, len, offset);
1829         if (ret < 0) {
1830                 error("cannot wipe existing superblock: %s", strerror(errno));
1831                 ret = -1;
1832         } else if (ret != len) {
1833                 error("cannot wipe existing superblock: wrote %d of %zd", ret, len);
1834                 ret = -1;
1835         }
1836         fsync(fd);
1837
1838 out:
1839         blkid_free_probe(pr);
1840         return ret;
1841 }
1842
1843 int btrfs_prepare_device(int fd, const char *file, u64 *block_count_ret,
1844                 u64 max_block_count, unsigned opflags)
1845 {
1846         u64 block_count;
1847         struct stat st;
1848         int i, ret;
1849
1850         ret = fstat(fd, &st);
1851         if (ret < 0) {
1852                 error("unable to stat %s: %s", file, strerror(errno));
1853                 return 1;
1854         }
1855
1856         block_count = btrfs_device_size(fd, &st);
1857         if (block_count == 0) {
1858                 error("unable to determine size of %s", file);
1859                 return 1;
1860         }
1861         if (max_block_count)
1862                 block_count = min(block_count, max_block_count);
1863
1864         if (opflags & PREP_DEVICE_DISCARD) {
1865                 /*
1866                  * We intentionally ignore errors from the discard ioctl.  It
1867                  * is not necessary for the mkfs functionality but just an
1868                  * optimization.
1869                  */
1870                 if (discard_range(fd, 0, 0) == 0) {
1871                         if (opflags & PREP_DEVICE_VERBOSE)
1872                                 printf("Performing full device TRIM (%s) ...\n",
1873                                                 pretty_size(block_count));
1874                         discard_blocks(fd, 0, block_count);
1875                 }
1876         }
1877
1878         ret = zero_dev_clamped(fd, 0, ZERO_DEV_BYTES, block_count);
1879         for (i = 0 ; !ret && i < BTRFS_SUPER_MIRROR_MAX; i++)
1880                 ret = zero_dev_clamped(fd, btrfs_sb_offset(i),
1881                                        BTRFS_SUPER_INFO_SIZE, block_count);
1882         if (!ret && (opflags & PREP_DEVICE_ZERO_END))
1883                 ret = zero_dev_clamped(fd, block_count - ZERO_DEV_BYTES,
1884                                        ZERO_DEV_BYTES, block_count);
1885
1886         if (ret < 0) {
1887                 error("failed to zero device '%s': %s", file, strerror(-ret));
1888                 return 1;
1889         }
1890
1891         ret = btrfs_wipe_existing_sb(fd);
1892         if (ret < 0) {
1893                 error("cannot wipe superblocks on %s", file);
1894                 return 1;
1895         }
1896
1897         *block_count_ret = block_count;
1898         return 0;
1899 }
1900
1901 int btrfs_make_root_dir(struct btrfs_trans_handle *trans,
1902                         struct btrfs_root *root, u64 objectid)
1903 {
1904         int ret;
1905         struct btrfs_inode_item inode_item;
1906         time_t now = time(NULL);
1907
1908         memset(&inode_item, 0, sizeof(inode_item));
1909         btrfs_set_stack_inode_generation(&inode_item, trans->transid);
1910         btrfs_set_stack_inode_size(&inode_item, 0);
1911         btrfs_set_stack_inode_nlink(&inode_item, 1);
1912         btrfs_set_stack_inode_nbytes(&inode_item, root->nodesize);
1913         btrfs_set_stack_inode_mode(&inode_item, S_IFDIR | 0755);
1914         btrfs_set_stack_timespec_sec(&inode_item.atime, now);
1915         btrfs_set_stack_timespec_nsec(&inode_item.atime, 0);
1916         btrfs_set_stack_timespec_sec(&inode_item.ctime, now);
1917         btrfs_set_stack_timespec_nsec(&inode_item.ctime, 0);
1918         btrfs_set_stack_timespec_sec(&inode_item.mtime, now);
1919         btrfs_set_stack_timespec_nsec(&inode_item.mtime, 0);
1920         btrfs_set_stack_timespec_sec(&inode_item.otime, now);
1921         btrfs_set_stack_timespec_nsec(&inode_item.otime, 0);
1922
1923         if (root->fs_info->tree_root == root)
1924                 btrfs_set_super_root_dir(root->fs_info->super_copy, objectid);
1925
1926         ret = btrfs_insert_inode(trans, root, objectid, &inode_item);
1927         if (ret)
1928                 goto error;
1929
1930         ret = btrfs_insert_inode_ref(trans, root, "..", 2, objectid, objectid, 0);
1931         if (ret)
1932                 goto error;
1933
1934         btrfs_set_root_dirid(&root->root_item, objectid);
1935         ret = 0;
1936 error:
1937         return ret;
1938 }
1939
1940 /*
1941  * checks if a path is a block device node
1942  * Returns negative errno on failure, otherwise
1943  * returns 1 for blockdev, 0 for not-blockdev
1944  */
1945 int is_block_device(const char *path)
1946 {
1947         struct stat statbuf;
1948
1949         if (stat(path, &statbuf) < 0)
1950                 return -errno;
1951
1952         return !!S_ISBLK(statbuf.st_mode);
1953 }
1954
1955 /*
1956  * check if given path is a mount point
1957  * return 1 if yes. 0 if no. -1 for error
1958  */
1959 int is_mount_point(const char *path)
1960 {
1961         FILE *f;
1962         struct mntent *mnt;
1963         int ret = 0;
1964
1965         f = setmntent("/proc/self/mounts", "r");
1966         if (f == NULL)
1967                 return -1;
1968
1969         while ((mnt = getmntent(f)) != NULL) {
1970                 if (strcmp(mnt->mnt_dir, path))
1971                         continue;
1972                 ret = 1;
1973                 break;
1974         }
1975         endmntent(f);
1976         return ret;
1977 }
1978
1979 static int is_reg_file(const char *path)
1980 {
1981         struct stat statbuf;
1982
1983         if (stat(path, &statbuf) < 0)
1984                 return -errno;
1985         return S_ISREG(statbuf.st_mode);
1986 }
1987
1988 /*
1989  * This function checks if the given input parameter is
1990  * an uuid or a path
1991  * return <0 : some error in the given input
1992  * return BTRFS_ARG_UNKNOWN:    unknown input
1993  * return BTRFS_ARG_UUID:       given input is uuid
1994  * return BTRFS_ARG_MNTPOINT:   given input is path
1995  * return BTRFS_ARG_REG:        given input is regular file
1996  * return BTRFS_ARG_BLKDEV:     given input is block device
1997  */
1998 int check_arg_type(const char *input)
1999 {
2000         uuid_t uuid;
2001         char path[PATH_MAX];
2002
2003         if (!input)
2004                 return -EINVAL;
2005
2006         if (realpath(input, path)) {
2007                 if (is_block_device(path) == 1)
2008                         return BTRFS_ARG_BLKDEV;
2009
2010                 if (is_mount_point(path) == 1)
2011                         return BTRFS_ARG_MNTPOINT;
2012
2013                 if (is_reg_file(path))
2014                         return BTRFS_ARG_REG;
2015
2016                 return BTRFS_ARG_UNKNOWN;
2017         }
2018
2019         if (strlen(input) == (BTRFS_UUID_UNPARSED_SIZE - 1) &&
2020                 !uuid_parse(input, uuid))
2021                 return BTRFS_ARG_UUID;
2022
2023         return BTRFS_ARG_UNKNOWN;
2024 }
2025
2026 /*
2027  * Find the mount point for a mounted device.
2028  * On success, returns 0 with mountpoint in *mp.
2029  * On failure, returns -errno (not mounted yields -EINVAL)
2030  * Is noisy on failures, expects to be given a mounted device.
2031  */
2032 int get_btrfs_mount(const char *dev, char *mp, size_t mp_size)
2033 {
2034         int ret;
2035         int fd = -1;
2036
2037         ret = is_block_device(dev);
2038         if (ret <= 0) {
2039                 if (!ret) {
2040                         error("not a block device: %s", dev);
2041                         ret = -EINVAL;
2042                 } else {
2043                         error("cannot check %s: %s", dev, strerror(-ret));
2044                 }
2045                 goto out;
2046         }
2047
2048         fd = open(dev, O_RDONLY);
2049         if (fd < 0) {
2050                 ret = -errno;
2051                 error("cannot open %s: %s", dev, strerror(errno));
2052                 goto out;
2053         }
2054
2055         ret = check_mounted_where(fd, dev, mp, mp_size, NULL);
2056         if (!ret) {
2057                 ret = -EINVAL;
2058         } else { /* mounted, all good */
2059                 ret = 0;
2060         }
2061 out:
2062         if (fd != -1)
2063                 close(fd);
2064         return ret;
2065 }
2066
2067 /*
2068  * Given a pathname, return a filehandle to:
2069  *      the original pathname or,
2070  *      if the pathname is a mounted btrfs device, to its mountpoint.
2071  *
2072  * On error, return -1, errno should be set.
2073  */
2074 int open_path_or_dev_mnt(const char *path, DIR **dirstream, int verbose)
2075 {
2076         char mp[PATH_MAX];
2077         int ret;
2078
2079         if (is_block_device(path)) {
2080                 ret = get_btrfs_mount(path, mp, sizeof(mp));
2081                 if (ret < 0) {
2082                         /* not a mounted btrfs dev */
2083                         error_on(verbose, "'%s' is not a mounted btrfs device",
2084                                  path);
2085                         errno = EINVAL;
2086                         return -1;
2087                 }
2088                 ret = open_file_or_dir(mp, dirstream);
2089                 error_on(verbose && ret < 0, "can't access '%s': %s",
2090                          path, strerror(errno));
2091         } else {
2092                 ret = btrfs_open_dir(path, dirstream, 1);
2093         }
2094
2095         return ret;
2096 }
2097
2098 /*
2099  * Do the following checks before calling open_file_or_dir():
2100  * 1: path is in a btrfs filesystem
2101  * 2: path is a directory
2102  */
2103 int btrfs_open_dir(const char *path, DIR **dirstream, int verbose)
2104 {
2105         struct statfs stfs;
2106         struct stat st;
2107         int ret;
2108
2109         if (statfs(path, &stfs) != 0) {
2110                 error_on(verbose, "cannot access '%s': %s", path,
2111                                 strerror(errno));
2112                 return -1;
2113         }
2114
2115         if (stfs.f_type != BTRFS_SUPER_MAGIC) {
2116                 error_on(verbose, "not a btrfs filesystem: %s", path);
2117                 return -2;
2118         }
2119
2120         if (stat(path, &st) != 0) {
2121                 error_on(verbose, "cannot access '%s': %s", path,
2122                                 strerror(errno));
2123                 return -1;
2124         }
2125
2126         if (!S_ISDIR(st.st_mode)) {
2127                 error_on(verbose, "not a directory: %s", path);
2128                 return -3;
2129         }
2130
2131         ret = open_file_or_dir(path, dirstream);
2132         if (ret < 0) {
2133                 error_on(verbose, "cannot access '%s': %s", path,
2134                                 strerror(errno));
2135         }
2136
2137         return ret;
2138 }
2139
2140 /* checks if a device is a loop device */
2141 static int is_loop_device (const char* device) {
2142         struct stat statbuf;
2143
2144         if(stat(device, &statbuf) < 0)
2145                 return -errno;
2146
2147         return (S_ISBLK(statbuf.st_mode) &&
2148                 MAJOR(statbuf.st_rdev) == LOOP_MAJOR);
2149 }
2150
2151 /*
2152  * Takes a loop device path (e.g. /dev/loop0) and returns
2153  * the associated file (e.g. /images/my_btrfs.img) using
2154  * loopdev API
2155  */
2156 static int resolve_loop_device_with_loopdev(const char* loop_dev, char* loop_file)
2157 {
2158         int fd;
2159         int ret;
2160         struct loop_info64 lo64;
2161
2162         fd = open(loop_dev, O_RDONLY | O_NONBLOCK);
2163         if (fd < 0)
2164                 return -errno;
2165         ret = ioctl(fd, LOOP_GET_STATUS64, &lo64);
2166         if (ret < 0) {
2167                 ret = -errno;
2168                 goto out;
2169         }
2170
2171         memcpy(loop_file, lo64.lo_file_name, sizeof(lo64.lo_file_name));
2172         loop_file[sizeof(lo64.lo_file_name)] = 0;
2173
2174 out:
2175         close(fd);
2176
2177         return ret;
2178 }
2179
2180 /* Takes a loop device path (e.g. /dev/loop0) and returns
2181  * the associated file (e.g. /images/my_btrfs.img) */
2182 static int resolve_loop_device(const char* loop_dev, char* loop_file,
2183                 int max_len)
2184 {
2185         int ret;
2186         FILE *f;
2187         char fmt[20];
2188         char p[PATH_MAX];
2189         char real_loop_dev[PATH_MAX];
2190
2191         if (!realpath(loop_dev, real_loop_dev))
2192                 return -errno;
2193         snprintf(p, PATH_MAX, "/sys/block/%s/loop/backing_file", strrchr(real_loop_dev, '/'));
2194         if (!(f = fopen(p, "r"))) {
2195                 if (errno == ENOENT)
2196                         /*
2197                          * It's possibly a partitioned loop device, which is
2198                          * resolvable with loopdev API.
2199                          */
2200                         return resolve_loop_device_with_loopdev(loop_dev, loop_file);
2201                 return -errno;
2202         }
2203
2204         snprintf(fmt, 20, "%%%i[^\n]", max_len-1);
2205         ret = fscanf(f, fmt, loop_file);
2206         fclose(f);
2207         if (ret == EOF)
2208                 return -errno;
2209
2210         return 0;
2211 }
2212
2213 /*
2214  * Checks whether a and b are identical or device
2215  * files associated with the same block device
2216  */
2217 static int is_same_blk_file(const char* a, const char* b)
2218 {
2219         struct stat st_buf_a, st_buf_b;
2220         char real_a[PATH_MAX];
2221         char real_b[PATH_MAX];
2222
2223         if (!realpath(a, real_a))
2224                 strncpy_null(real_a, a);
2225
2226         if (!realpath(b, real_b))
2227                 strncpy_null(real_b, b);
2228
2229         /* Identical path? */
2230         if (strcmp(real_a, real_b) == 0)
2231                 return 1;
2232
2233         if (stat(a, &st_buf_a) < 0 || stat(b, &st_buf_b) < 0) {
2234                 if (errno == ENOENT)
2235                         return 0;
2236                 return -errno;
2237         }
2238
2239         /* Same blockdevice? */
2240         if (S_ISBLK(st_buf_a.st_mode) && S_ISBLK(st_buf_b.st_mode) &&
2241             st_buf_a.st_rdev == st_buf_b.st_rdev) {
2242                 return 1;
2243         }
2244
2245         /* Hardlink? */
2246         if (st_buf_a.st_dev == st_buf_b.st_dev &&
2247             st_buf_a.st_ino == st_buf_b.st_ino) {
2248                 return 1;
2249         }
2250
2251         return 0;
2252 }
2253
2254 /* checks if a and b are identical or device
2255  * files associated with the same block device or
2256  * if one file is a loop device that uses the other
2257  * file.
2258  */
2259 static int is_same_loop_file(const char* a, const char* b)
2260 {
2261         char res_a[PATH_MAX];
2262         char res_b[PATH_MAX];
2263         const char* final_a = NULL;
2264         const char* final_b = NULL;
2265         int ret;
2266
2267         /* Resolve a if it is a loop device */
2268         if((ret = is_loop_device(a)) < 0) {
2269                 if (ret == -ENOENT)
2270                         return 0;
2271                 return ret;
2272         } else if (ret) {
2273                 ret = resolve_loop_device(a, res_a, sizeof(res_a));
2274                 if (ret < 0) {
2275                         if (errno != EPERM)
2276                                 return ret;
2277                 } else {
2278                         final_a = res_a;
2279                 }
2280         } else {
2281                 final_a = a;
2282         }
2283
2284         /* Resolve b if it is a loop device */
2285         if ((ret = is_loop_device(b)) < 0) {
2286                 if (ret == -ENOENT)
2287                         return 0;
2288                 return ret;
2289         } else if (ret) {
2290                 ret = resolve_loop_device(b, res_b, sizeof(res_b));
2291                 if (ret < 0) {
2292                         if (errno != EPERM)
2293                                 return ret;
2294                 } else {
2295                         final_b = res_b;
2296                 }
2297         } else {
2298                 final_b = b;
2299         }
2300
2301         return is_same_blk_file(final_a, final_b);
2302 }
2303
2304 /* Checks if a file exists and is a block or regular file*/
2305 static int is_existing_blk_or_reg_file(const char* filename)
2306 {
2307         struct stat st_buf;
2308
2309         if(stat(filename, &st_buf) < 0) {
2310                 if(errno == ENOENT)
2311                         return 0;
2312                 else
2313                         return -errno;
2314         }
2315
2316         return (S_ISBLK(st_buf.st_mode) || S_ISREG(st_buf.st_mode));
2317 }
2318
2319 /* Checks if a file is used (directly or indirectly via a loop device)
2320  * by a device in fs_devices
2321  */
2322 static int blk_file_in_dev_list(struct btrfs_fs_devices* fs_devices,
2323                 const char* file)
2324 {
2325         int ret;
2326         struct list_head *head;
2327         struct list_head *cur;
2328         struct btrfs_device *device;
2329
2330         head = &fs_devices->devices;
2331         list_for_each(cur, head) {
2332                 device = list_entry(cur, struct btrfs_device, dev_list);
2333
2334                 if((ret = is_same_loop_file(device->name, file)))
2335                         return ret;
2336         }
2337
2338         return 0;
2339 }
2340
2341 /*
2342  * Resolve a pathname to a device mapper node to /dev/mapper/<name>
2343  * Returns NULL on invalid input or malloc failure; Other failures
2344  * will be handled by the caller using the input pathame.
2345  */
2346 char *canonicalize_dm_name(const char *ptname)
2347 {
2348         FILE    *f;
2349         size_t  sz;
2350         char    path[PATH_MAX], name[PATH_MAX], *res = NULL;
2351
2352         if (!ptname || !*ptname)
2353                 return NULL;
2354
2355         snprintf(path, sizeof(path), "/sys/block/%s/dm/name", ptname);
2356         if (!(f = fopen(path, "r")))
2357                 return NULL;
2358
2359         /* read <name>\n from sysfs */
2360         if (fgets(name, sizeof(name), f) && (sz = strlen(name)) > 1) {
2361                 name[sz - 1] = '\0';
2362                 snprintf(path, sizeof(path), "/dev/mapper/%s", name);
2363
2364                 if (access(path, F_OK) == 0)
2365                         res = strdup(path);
2366         }
2367         fclose(f);
2368         return res;
2369 }
2370
2371 /*
2372  * Resolve a pathname to a canonical device node, e.g. /dev/sda1 or
2373  * to a device mapper pathname.
2374  * Returns NULL on invalid input or malloc failure; Other failures
2375  * will be handled by the caller using the input pathame.
2376  */
2377 char *canonicalize_path(const char *path)
2378 {
2379         char *canonical, *p;
2380
2381         if (!path || !*path)
2382                 return NULL;
2383
2384         canonical = realpath(path, NULL);
2385         if (!canonical)
2386                 return strdup(path);
2387         p = strrchr(canonical, '/');
2388         if (p && strncmp(p, "/dm-", 4) == 0 && isdigit(*(p + 4))) {
2389                 char *dm = canonicalize_dm_name(p + 1);
2390
2391                 if (dm) {
2392                         free(canonical);
2393                         return dm;
2394                 }
2395         }
2396         return canonical;
2397 }
2398
2399 /*
2400  * returns 1 if the device was mounted, < 0 on error or 0 if everything
2401  * is safe to continue.
2402  */
2403 int check_mounted(const char* file)
2404 {
2405         int fd;
2406         int ret;
2407
2408         fd = open(file, O_RDONLY);
2409         if (fd < 0) {
2410                 error("mount check: cannot open %s: %s", file,
2411                                 strerror(errno));
2412                 return -errno;
2413         }
2414
2415         ret =  check_mounted_where(fd, file, NULL, 0, NULL);
2416         close(fd);
2417
2418         return ret;
2419 }
2420
2421 int check_mounted_where(int fd, const char *file, char *where, int size,
2422                         struct btrfs_fs_devices **fs_dev_ret)
2423 {
2424         int ret;
2425         u64 total_devs = 1;
2426         int is_btrfs;
2427         struct btrfs_fs_devices *fs_devices_mnt = NULL;
2428         FILE *f;
2429         struct mntent *mnt;
2430
2431         /* scan the initial device */
2432         ret = btrfs_scan_one_device(fd, file, &fs_devices_mnt,
2433                     &total_devs, BTRFS_SUPER_INFO_OFFSET, SBREAD_DEFAULT);
2434         is_btrfs = (ret >= 0);
2435
2436         /* scan other devices */
2437         if (is_btrfs && total_devs > 1) {
2438                 ret = btrfs_scan_devices();
2439                 if (ret)
2440                         return ret;
2441         }
2442
2443         /* iterate over the list of currently mounted filesystems */
2444         if ((f = setmntent ("/proc/self/mounts", "r")) == NULL)
2445                 return -errno;
2446
2447         while ((mnt = getmntent (f)) != NULL) {
2448                 if(is_btrfs) {
2449                         if(strcmp(mnt->mnt_type, "btrfs") != 0)
2450                                 continue;
2451
2452                         ret = blk_file_in_dev_list(fs_devices_mnt, mnt->mnt_fsname);
2453                 } else {
2454                         /* ignore entries in the mount table that are not
2455                            associated with a file*/
2456                         if((ret = is_existing_blk_or_reg_file(mnt->mnt_fsname)) < 0)
2457                                 goto out_mntloop_err;
2458                         else if(!ret)
2459                                 continue;
2460
2461                         ret = is_same_loop_file(file, mnt->mnt_fsname);
2462                 }
2463
2464                 if(ret < 0)
2465                         goto out_mntloop_err;
2466                 else if(ret)
2467                         break;
2468         }
2469
2470         /* Did we find an entry in mnt table? */
2471         if (mnt && size && where) {
2472                 strncpy(where, mnt->mnt_dir, size);
2473                 where[size-1] = 0;
2474         }
2475         if (fs_dev_ret)
2476                 *fs_dev_ret = fs_devices_mnt;
2477
2478         ret = (mnt != NULL);
2479
2480 out_mntloop_err:
2481         endmntent (f);
2482
2483         return ret;
2484 }
2485
2486 struct pending_dir {
2487         struct list_head list;
2488         char name[PATH_MAX];
2489 };
2490
2491 int btrfs_register_one_device(const char *fname)
2492 {
2493         struct btrfs_ioctl_vol_args args;
2494         int fd;
2495         int ret;
2496
2497         fd = open("/dev/btrfs-control", O_RDWR);
2498         if (fd < 0) {
2499                 warning(
2500         "failed to open /dev/btrfs-control, skipping device registration: %s",
2501                         strerror(errno));
2502                 return -errno;
2503         }
2504         memset(&args, 0, sizeof(args));
2505         strncpy_null(args.name, fname);
2506         ret = ioctl(fd, BTRFS_IOC_SCAN_DEV, &args);
2507         if (ret < 0) {
2508                 error("device scan failed on '%s': %s", fname,
2509                                 strerror(errno));
2510                 ret = -errno;
2511         }
2512         close(fd);
2513         return ret;
2514 }
2515
2516 /*
2517  * Register all devices in the fs_uuid list created in the user
2518  * space. Ensure btrfs_scan_devices() is called before this func.
2519  */
2520 int btrfs_register_all_devices(void)
2521 {
2522         int err = 0;
2523         int ret = 0;
2524         struct btrfs_fs_devices *fs_devices;
2525         struct btrfs_device *device;
2526         struct list_head *all_uuids;
2527
2528         all_uuids = btrfs_scanned_uuids();
2529
2530         list_for_each_entry(fs_devices, all_uuids, list) {
2531                 list_for_each_entry(device, &fs_devices->devices, dev_list) {
2532                         if (*device->name)
2533                                 err = btrfs_register_one_device(device->name);
2534
2535                         if (err)
2536                                 ret++;
2537                 }
2538         }
2539
2540         return ret;
2541 }
2542
2543 int btrfs_device_already_in_root(struct btrfs_root *root, int fd,
2544                                  int super_offset)
2545 {
2546         struct btrfs_super_block *disk_super;
2547         char *buf;
2548         int ret = 0;
2549
2550         buf = malloc(BTRFS_SUPER_INFO_SIZE);
2551         if (!buf) {
2552                 ret = -ENOMEM;
2553                 goto out;
2554         }
2555         ret = pread(fd, buf, BTRFS_SUPER_INFO_SIZE, super_offset);
2556         if (ret != BTRFS_SUPER_INFO_SIZE)
2557                 goto brelse;
2558
2559         ret = 0;
2560         disk_super = (struct btrfs_super_block *)buf;
2561         /*
2562          * Accept devices from the same filesystem, allow partially created
2563          * structures.
2564          */
2565         if (btrfs_super_magic(disk_super) != BTRFS_MAGIC &&
2566                         btrfs_super_magic(disk_super) != BTRFS_MAGIC_PARTIAL)
2567                 goto brelse;
2568
2569         if (!memcmp(disk_super->fsid, root->fs_info->super_copy->fsid,
2570                     BTRFS_FSID_SIZE))
2571                 ret = 1;
2572 brelse:
2573         free(buf);
2574 out:
2575         return ret;
2576 }
2577
2578 /*
2579  * Note: this function uses a static per-thread buffer. Do not call this
2580  * function more than 10 times within one argument list!
2581  */
2582 const char *pretty_size_mode(s64 size, unsigned mode)
2583 {
2584         static __thread int ps_index = 0;
2585         static __thread char ps_array[10][32];
2586         char *ret;
2587
2588         ret = ps_array[ps_index];
2589         ps_index++;
2590         ps_index %= 10;
2591         (void)pretty_size_snprintf(size, ret, 32, mode);
2592
2593         return ret;
2594 }
2595
2596 static const char* unit_suffix_binary[] =
2597         { "B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB"};
2598 static const char* unit_suffix_decimal[] =
2599         { "B", "kB", "MB", "GB", "TB", "PB", "EB"};
2600
2601 int pretty_size_snprintf(s64 size, char *str, size_t str_size, unsigned unit_mode)
2602 {
2603         int num_divs;
2604         float fraction;
2605         s64 base = 0;
2606         int mult = 0;
2607         const char** suffix = NULL;
2608         s64 last_size;
2609
2610         if (str_size == 0)
2611                 return 0;
2612
2613         if ((unit_mode & ~UNITS_MODE_MASK) == UNITS_RAW) {
2614                 snprintf(str, str_size, "%lld", size);
2615                 return 0;
2616         }
2617
2618         if ((unit_mode & ~UNITS_MODE_MASK) == UNITS_BINARY) {
2619                 base = 1024;
2620                 mult = 1024;
2621                 suffix = unit_suffix_binary;
2622         } else if ((unit_mode & ~UNITS_MODE_MASK) == UNITS_DECIMAL) {
2623                 base = 1000;
2624                 mult = 1000;
2625                 suffix = unit_suffix_decimal;
2626         }
2627
2628         /* Unknown mode */
2629         if (!base) {
2630                 fprintf(stderr, "INTERNAL ERROR: unknown unit base, mode %d\n",
2631                                 unit_mode);
2632                 assert(0);
2633                 return -1;
2634         }
2635
2636         num_divs = 0;
2637         last_size = size;
2638         switch (unit_mode & UNITS_MODE_MASK) {
2639         case UNITS_TBYTES: base *= mult; num_divs++;
2640         case UNITS_GBYTES: base *= mult; num_divs++;
2641         case UNITS_MBYTES: base *= mult; num_divs++;
2642         case UNITS_KBYTES: num_divs++;
2643                            break;
2644         case UNITS_BYTES:
2645                            base = 1;
2646                            num_divs = 0;
2647                            break;
2648         default:
2649                 while ((size < 0 ? -size : size) >= mult) {
2650                         last_size = size;
2651                         size /= mult;
2652                         num_divs++;
2653                 }
2654                 /*
2655                  * If the value is smaller than base, we didn't do any
2656                  * division, in that case, base should be 1, not original
2657                  * base, or the unit will be wrong
2658                  */
2659                 if (num_divs == 0)
2660                         base = 1;
2661         }
2662
2663         if (num_divs >= ARRAY_SIZE(unit_suffix_binary)) {
2664                 str[0] = '\0';
2665                 printf("INTERNAL ERROR: unsupported unit suffix, index %d\n",
2666                                 num_divs);
2667                 assert(0);
2668                 return -1;
2669         }
2670         fraction = (float)last_size / base;
2671
2672         return snprintf(str, str_size, "%.2f%s", fraction, suffix[num_divs]);
2673 }
2674
2675 /*
2676  * __strncpy_null - strncpy with null termination
2677  * @dest:       the target array
2678  * @src:        the source string
2679  * @n:          maximum bytes to copy (size of *dest)
2680  *
2681  * Like strncpy, but ensures destination is null-terminated.
2682  *
2683  * Copies the string pointed to by src, including the terminating null
2684  * byte ('\0'), to the buffer pointed to by dest, up to a maximum
2685  * of n bytes.  Then ensure that dest is null-terminated.
2686  */
2687 char *__strncpy_null(char *dest, const char *src, size_t n)
2688 {
2689         strncpy(dest, src, n);
2690         if (n > 0)
2691                 dest[n - 1] = '\0';
2692         return dest;
2693 }
2694
2695 /*
2696  * Checks to make sure that the label matches our requirements.
2697  * Returns:
2698        0    if everything is safe and usable
2699       -1    if the label is too long
2700  */
2701 static int check_label(const char *input)
2702 {
2703        int len = strlen(input);
2704
2705        if (len > BTRFS_LABEL_SIZE - 1) {
2706                 error("label %s is too long (max %d)", input,
2707                                 BTRFS_LABEL_SIZE - 1);
2708                return -1;
2709        }
2710
2711        return 0;
2712 }
2713
2714 static int set_label_unmounted(const char *dev, const char *label)
2715 {
2716         struct btrfs_trans_handle *trans;
2717         struct btrfs_root *root;
2718         int ret;
2719
2720         ret = check_mounted(dev);
2721         if (ret < 0) {
2722                error("checking mount status of %s failed: %d", dev, ret);
2723                return -1;
2724         }
2725         if (ret > 0) {
2726                 error("device %s is mounted, use mount point", dev);
2727                 return -1;
2728         }
2729
2730         /* Open the super_block at the default location
2731          * and as read-write.
2732          */
2733         root = open_ctree(dev, 0, OPEN_CTREE_WRITES);
2734         if (!root) /* errors are printed by open_ctree() */
2735                 return -1;
2736
2737         trans = btrfs_start_transaction(root, 1);
2738         __strncpy_null(root->fs_info->super_copy->label, label, BTRFS_LABEL_SIZE - 1);
2739
2740         btrfs_commit_transaction(trans, root);
2741
2742         /* Now we close it since we are done. */
2743         close_ctree(root);
2744         return 0;
2745 }
2746
2747 static int set_label_mounted(const char *mount_path, const char *labelp)
2748 {
2749         int fd;
2750         char label[BTRFS_LABEL_SIZE];
2751
2752         fd = open(mount_path, O_RDONLY | O_NOATIME);
2753         if (fd < 0) {
2754                 error("unable to access %s: %s", mount_path, strerror(errno));
2755                 return -1;
2756         }
2757
2758         memset(label, 0, sizeof(label));
2759         __strncpy_null(label, labelp, BTRFS_LABEL_SIZE - 1);
2760         if (ioctl(fd, BTRFS_IOC_SET_FSLABEL, label) < 0) {
2761                 error("unable to set label of %s: %s", mount_path,
2762                                 strerror(errno));
2763                 close(fd);
2764                 return -1;
2765         }
2766
2767         close(fd);
2768         return 0;
2769 }
2770
2771 int get_label_unmounted(const char *dev, char *label)
2772 {
2773         struct btrfs_root *root;
2774         int ret;
2775
2776         ret = check_mounted(dev);
2777         if (ret < 0) {
2778                error("checking mount status of %s failed: %d", dev, ret);
2779                return -1;
2780         }
2781
2782         /* Open the super_block at the default location
2783          * and as read-only.
2784          */
2785         root = open_ctree(dev, 0, 0);
2786         if(!root)
2787                 return -1;
2788
2789         __strncpy_null(label, root->fs_info->super_copy->label,
2790                         BTRFS_LABEL_SIZE - 1);
2791
2792         /* Now we close it since we are done. */
2793         close_ctree(root);
2794         return 0;
2795 }
2796
2797 /*
2798  * If a partition is mounted, try to get the filesystem label via its
2799  * mounted path rather than device.  Return the corresponding error
2800  * the user specified the device path.
2801  */
2802 int get_label_mounted(const char *mount_path, char *labelp)
2803 {
2804         char label[BTRFS_LABEL_SIZE];
2805         int fd;
2806         int ret;
2807
2808         fd = open(mount_path, O_RDONLY | O_NOATIME);
2809         if (fd < 0) {
2810                 error("unable to access %s: %s", mount_path, strerror(errno));
2811                 return -1;
2812         }
2813
2814         memset(label, '\0', sizeof(label));
2815         ret = ioctl(fd, BTRFS_IOC_GET_FSLABEL, label);
2816         if (ret < 0) {
2817                 if (errno != ENOTTY)
2818                         error("unable to get label of %s: %s", mount_path,
2819                                         strerror(errno));
2820                 ret = -errno;
2821                 close(fd);
2822                 return ret;
2823         }
2824
2825         __strncpy_null(labelp, label, BTRFS_LABEL_SIZE - 1);
2826         close(fd);
2827         return 0;
2828 }
2829
2830 int get_label(const char *btrfs_dev, char *label)
2831 {
2832         int ret;
2833
2834         ret = is_existing_blk_or_reg_file(btrfs_dev);
2835         if (!ret)
2836                 ret = get_label_mounted(btrfs_dev, label);
2837         else if (ret > 0)
2838                 ret = get_label_unmounted(btrfs_dev, label);
2839
2840         return ret;
2841 }
2842
2843 int set_label(const char *btrfs_dev, const char *label)
2844 {
2845         int ret;
2846
2847         if (check_label(label))
2848                 return -1;
2849
2850         ret = is_existing_blk_or_reg_file(btrfs_dev);
2851         if (!ret)
2852                 ret = set_label_mounted(btrfs_dev, label);
2853         else if (ret > 0)
2854                 ret = set_label_unmounted(btrfs_dev, label);
2855
2856         return ret;
2857 }
2858
2859 /*
2860  * A not-so-good version fls64. No fascinating optimization since
2861  * no one except parse_size use it
2862  */
2863 static int fls64(u64 x)
2864 {
2865         int i;
2866
2867         for (i = 0; i <64; i++)
2868                 if (x << i & (1ULL << 63))
2869                         return 64 - i;
2870         return 64 - i;
2871 }
2872
2873 u64 parse_size(char *s)
2874 {
2875         char c;
2876         char *endptr;
2877         u64 mult = 1;
2878         u64 ret;
2879
2880         if (!s) {
2881                 error("size value is empty");
2882                 exit(1);
2883         }
2884         if (s[0] == '-') {
2885                 error("size value '%s' is less equal than 0", s);
2886                 exit(1);
2887         }
2888         ret = strtoull(s, &endptr, 10);
2889         if (endptr == s) {
2890                 error("size value '%s' is invalid", s);
2891                 exit(1);
2892         }
2893         if (endptr[0] && endptr[1]) {
2894                 error("illegal suffix contains character '%c' in wrong position",
2895                         endptr[1]);
2896                 exit(1);
2897         }
2898         /*
2899          * strtoll returns LLONG_MAX when overflow, if this happens,
2900          * need to call strtoull to get the real size
2901          */
2902         if (errno == ERANGE && ret == ULLONG_MAX) {
2903                 error("size value '%s' is too large for u64", s);
2904                 exit(1);
2905         }
2906         if (endptr[0]) {
2907                 c = tolower(endptr[0]);
2908                 switch (c) {
2909                 case 'e':
2910                         mult *= 1024;
2911                         /* fallthrough */
2912                 case 'p':
2913                         mult *= 1024;
2914                         /* fallthrough */
2915                 case 't':
2916                         mult *= 1024;
2917                         /* fallthrough */
2918                 case 'g':
2919                         mult *= 1024;
2920                         /* fallthrough */
2921                 case 'm':
2922                         mult *= 1024;
2923                         /* fallthrough */
2924                 case 'k':
2925                         mult *= 1024;
2926                         /* fallthrough */
2927                 case 'b':
2928                         break;
2929                 default:
2930                         error("unknown size descriptor '%c'", c);
2931                         exit(1);
2932                 }
2933         }
2934         /* Check whether ret * mult overflow */
2935         if (fls64(ret) + fls64(mult) - 1 > 64) {
2936                 error("size value '%s' is too large for u64", s);
2937                 exit(1);
2938         }
2939         ret *= mult;
2940         return ret;
2941 }
2942
2943 u64 parse_qgroupid(const char *p)
2944 {
2945         char *s = strchr(p, '/');
2946         const char *ptr_src_end = p + strlen(p);
2947         char *ptr_parse_end = NULL;
2948         u64 level;
2949         u64 id;
2950         int fd;
2951         int ret = 0;
2952
2953         if (p[0] == '/')
2954                 goto path;
2955
2956         /* Numeric format like '0/257' is the primary case */
2957         if (!s) {
2958                 id = strtoull(p, &ptr_parse_end, 10);
2959                 if (ptr_parse_end != ptr_src_end)
2960                         goto path;
2961                 return id;
2962         }
2963         level = strtoull(p, &ptr_parse_end, 10);
2964         if (ptr_parse_end != s)
2965                 goto path;
2966
2967         id = strtoull(s + 1, &ptr_parse_end, 10);
2968         if (ptr_parse_end != ptr_src_end)
2969                 goto  path;
2970
2971         return (level << BTRFS_QGROUP_LEVEL_SHIFT) | id;
2972
2973 path:
2974         /* Path format like subv at 'my_subvol' is the fallback case */
2975         ret = test_issubvolume(p);
2976         if (ret < 0 || !ret)
2977                 goto err;
2978         fd = open(p, O_RDONLY);
2979         if (fd < 0)
2980                 goto err;
2981         ret = lookup_path_rootid(fd, &id);
2982         if (ret)
2983                 error("failed to lookup root id: %s", strerror(-ret));
2984         close(fd);
2985         if (ret < 0)
2986                 goto err;
2987         return id;
2988
2989 err:
2990         error("invalid qgroupid or subvolume path: %s", p);
2991         exit(-1);
2992 }
2993
2994 int open_file_or_dir3(const char *fname, DIR **dirstream, int open_flags)
2995 {
2996         int ret;
2997         struct stat st;
2998         int fd;
2999
3000         ret = stat(fname, &st);
3001         if (ret < 0) {
3002                 return -1;
3003         }
3004         if (S_ISDIR(st.st_mode)) {
3005                 *dirstream = opendir(fname);
3006                 if (!*dirstream)
3007                         return -1;
3008                 fd = dirfd(*dirstream);
3009         } else if (S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) {
3010                 fd = open(fname, open_flags);
3011         } else {
3012                 /*
3013                  * we set this on purpose, in case the caller output
3014                  * strerror(errno) as success
3015                  */
3016                 errno = EINVAL;
3017                 return -1;
3018         }
3019         if (fd < 0) {
3020                 fd = -1;
3021                 if (*dirstream) {
3022                         closedir(*dirstream);
3023                         *dirstream = NULL;
3024                 }
3025         }
3026         return fd;
3027 }
3028
3029 int open_file_or_dir(const char *fname, DIR **dirstream)
3030 {
3031         return open_file_or_dir3(fname, dirstream, O_RDWR);
3032 }
3033
3034 void close_file_or_dir(int fd, DIR *dirstream)
3035 {
3036         if (dirstream)
3037                 closedir(dirstream);
3038         else if (fd >= 0)
3039                 close(fd);
3040 }
3041
3042 int get_device_info(int fd, u64 devid,
3043                 struct btrfs_ioctl_dev_info_args *di_args)
3044 {
3045         int ret;
3046
3047         di_args->devid = devid;
3048         memset(&di_args->uuid, '\0', sizeof(di_args->uuid));
3049
3050         ret = ioctl(fd, BTRFS_IOC_DEV_INFO, di_args);
3051         return ret < 0 ? -errno : 0;
3052 }
3053
3054 static u64 find_max_device_id(struct btrfs_ioctl_search_args *search_args,
3055                               int nr_items)
3056 {
3057         struct btrfs_dev_item *dev_item;
3058         char *buf = search_args->buf;
3059
3060         buf += (nr_items - 1) * (sizeof(struct btrfs_ioctl_search_header)
3061                                        + sizeof(struct btrfs_dev_item));
3062         buf += sizeof(struct btrfs_ioctl_search_header);
3063
3064         dev_item = (struct btrfs_dev_item *)buf;
3065
3066         return btrfs_stack_device_id(dev_item);
3067 }
3068
3069 static int search_chunk_tree_for_fs_info(int fd,
3070                                 struct btrfs_ioctl_fs_info_args *fi_args)
3071 {
3072         int ret;
3073         int max_items;
3074         u64 start_devid = 1;
3075         struct btrfs_ioctl_search_args search_args;
3076         struct btrfs_ioctl_search_key *search_key = &search_args.key;
3077
3078         fi_args->num_devices = 0;
3079
3080         max_items = BTRFS_SEARCH_ARGS_BUFSIZE
3081                / (sizeof(struct btrfs_ioctl_search_header)
3082                                + sizeof(struct btrfs_dev_item));
3083
3084         search_key->tree_id = BTRFS_CHUNK_TREE_OBJECTID;
3085         search_key->min_objectid = BTRFS_DEV_ITEMS_OBJECTID;
3086         search_key->max_objectid = BTRFS_DEV_ITEMS_OBJECTID;
3087         search_key->min_type = BTRFS_DEV_ITEM_KEY;
3088         search_key->max_type = BTRFS_DEV_ITEM_KEY;
3089         search_key->min_transid = 0;
3090         search_key->max_transid = (u64)-1;
3091         search_key->nr_items = max_items;
3092         search_key->max_offset = (u64)-1;
3093
3094 again:
3095         search_key->min_offset = start_devid;
3096
3097         ret = ioctl(fd, BTRFS_IOC_TREE_SEARCH, &search_args);
3098         if (ret < 0)
3099                 return -errno;
3100
3101         fi_args->num_devices += (u64)search_key->nr_items;
3102
3103         if (search_key->nr_items == max_items) {
3104                 start_devid = find_max_device_id(&search_args,
3105                                         search_key->nr_items) + 1;
3106                 goto again;
3107         }
3108
3109         /* get the lastest max_id to stay consistent with the num_devices */
3110         if (search_key->nr_items == 0)
3111                 /*
3112                  * last tree_search returns an empty buf, use the devid of
3113                  * the last dev_item of the previous tree_search
3114                  */
3115                 fi_args->max_id = start_devid - 1;
3116         else
3117                 fi_args->max_id = find_max_device_id(&search_args,
3118                                                 search_key->nr_items);
3119
3120         return 0;
3121 }
3122
3123 /*
3124  * For a given path, fill in the ioctl fs_ and info_ args.
3125  * If the path is a btrfs mountpoint, fill info for all devices.
3126  * If the path is a btrfs device, fill in only that device.
3127  *
3128  * The path provided must be either on a mounted btrfs fs,
3129  * or be a mounted btrfs device.
3130  *
3131  * Returns 0 on success, or a negative errno.
3132  */
3133 int get_fs_info(const char *path, struct btrfs_ioctl_fs_info_args *fi_args,
3134                 struct btrfs_ioctl_dev_info_args **di_ret)
3135 {
3136         int fd = -1;
3137         int ret = 0;
3138         int ndevs = 0;
3139         u64 last_devid = 0;
3140         int replacing = 0;
3141         struct btrfs_fs_devices *fs_devices_mnt = NULL;
3142         struct btrfs_ioctl_dev_info_args *di_args;
3143         struct btrfs_ioctl_dev_info_args tmp;
3144         char mp[PATH_MAX];
3145         DIR *dirstream = NULL;
3146
3147         memset(fi_args, 0, sizeof(*fi_args));
3148
3149         if (is_block_device(path) == 1) {
3150                 struct btrfs_super_block *disk_super;
3151                 char buf[BTRFS_SUPER_INFO_SIZE];
3152
3153                 /* Ensure it's mounted, then set path to the mountpoint */
3154                 fd = open(path, O_RDONLY);
3155                 if (fd < 0) {
3156                         ret = -errno;
3157                         error("cannot open %s: %s", path, strerror(errno));
3158                         goto out;
3159                 }
3160                 ret = check_mounted_where(fd, path, mp, sizeof(mp),
3161                                           &fs_devices_mnt);
3162                 if (!ret) {
3163                         ret = -EINVAL;
3164                         goto out;
3165                 }
3166                 if (ret < 0)
3167                         goto out;
3168                 path = mp;
3169                 /* Only fill in this one device */
3170                 fi_args->num_devices = 1;
3171
3172                 disk_super = (struct btrfs_super_block *)buf;
3173                 ret = btrfs_read_dev_super(fd, disk_super,
3174                                            BTRFS_SUPER_INFO_OFFSET, 0);
3175                 if (ret < 0) {
3176                         ret = -EIO;
3177                         goto out;
3178                 }
3179                 last_devid = btrfs_stack_device_id(&disk_super->dev_item);
3180                 fi_args->max_id = last_devid;
3181
3182                 memcpy(fi_args->fsid, fs_devices_mnt->fsid, BTRFS_FSID_SIZE);
3183                 close(fd);
3184         }
3185
3186         /* at this point path must not be for a block device */
3187         fd = open_file_or_dir(path, &dirstream);
3188         if (fd < 0) {
3189                 ret = -errno;
3190                 goto out;
3191         }
3192
3193         /* fill in fi_args if not just a single device */
3194         if (fi_args->num_devices != 1) {
3195                 ret = ioctl(fd, BTRFS_IOC_FS_INFO, fi_args);
3196                 if (ret < 0) {
3197                         ret = -errno;
3198                         goto out;
3199                 }
3200
3201                 /*
3202                  * The fs_args->num_devices does not include seed devices
3203                  */
3204                 ret = search_chunk_tree_for_fs_info(fd, fi_args);
3205                 if (ret)
3206                         goto out;
3207
3208                 /*
3209                  * search_chunk_tree_for_fs_info() will lacks the devid 0
3210                  * so manual probe for it here.
3211                  */
3212                 ret = get_device_info(fd, 0, &tmp);
3213                 if (!ret) {
3214                         fi_args->num_devices++;
3215                         ndevs++;
3216                         replacing = 1;
3217                         if (last_devid == 0)
3218                                 last_devid++;
3219                 }
3220         }
3221
3222         if (!fi_args->num_devices)
3223                 goto out;
3224
3225         di_args = *di_ret = malloc((fi_args->num_devices) * sizeof(*di_args));
3226         if (!di_args) {
3227                 ret = -errno;
3228                 goto out;
3229         }
3230
3231         if (replacing)
3232                 memcpy(di_args, &tmp, sizeof(tmp));
3233         for (; last_devid <= fi_args->max_id; last_devid++) {
3234                 ret = get_device_info(fd, last_devid, &di_args[ndevs]);
3235                 if (ret == -ENODEV)
3236                         continue;
3237                 if (ret)
3238                         goto out;
3239                 ndevs++;
3240         }
3241
3242         /*
3243         * only when the only dev we wanted to find is not there then
3244         * let any error be returned
3245         */
3246         if (fi_args->num_devices != 1) {
3247                 BUG_ON(ndevs == 0);
3248                 ret = 0;
3249         }
3250
3251 out:
3252         close_file_or_dir(fd, dirstream);
3253         return ret;
3254 }
3255
3256 #define isoctal(c)      (((c) & ~7) == '0')
3257
3258 static inline void translate(char *f, char *t)
3259 {
3260         while (*f != '\0') {
3261                 if (*f == '\\' &&
3262                     isoctal(f[1]) && isoctal(f[2]) && isoctal(f[3])) {
3263                         *t++ = 64*(f[1] & 7) + 8*(f[2] & 7) + (f[3] & 7);
3264                         f += 4;
3265                 } else
3266                         *t++ = *f++;
3267         }
3268         *t = '\0';
3269         return;
3270 }
3271
3272 /*
3273  * Checks if the swap device.
3274  * Returns 1 if swap device, < 0 on error or 0 if not swap device.
3275  */
3276 static int is_swap_device(const char *file)
3277 {
3278         FILE    *f;
3279         struct stat     st_buf;
3280         dev_t   dev;
3281         ino_t   ino = 0;
3282         char    tmp[PATH_MAX];
3283         char    buf[PATH_MAX];
3284         char    *cp;
3285         int     ret = 0;
3286
3287         if (stat(file, &st_buf) < 0)
3288                 return -errno;
3289         if (S_ISBLK(st_buf.st_mode))
3290                 dev = st_buf.st_rdev;
3291         else if (S_ISREG(st_buf.st_mode)) {
3292                 dev = st_buf.st_dev;
3293                 ino = st_buf.st_ino;
3294         } else
3295                 return 0;
3296
3297         if ((f = fopen("/proc/swaps", "r")) == NULL)
3298                 return 0;
3299
3300         /* skip the first line */
3301         if (fgets(tmp, sizeof(tmp), f) == NULL)
3302                 goto out;
3303
3304         while (fgets(tmp, sizeof(tmp), f) != NULL) {
3305                 if ((cp = strchr(tmp, ' ')) != NULL)
3306                         *cp = '\0';
3307                 if ((cp = strchr(tmp, '\t')) != NULL)
3308                         *cp = '\0';
3309                 translate(tmp, buf);
3310                 if (stat(buf, &st_buf) != 0)
3311                         continue;
3312                 if (S_ISBLK(st_buf.st_mode)) {
3313                         if (dev == st_buf.st_rdev) {
3314                                 ret = 1;
3315                                 break;
3316                         }
3317                 } else if (S_ISREG(st_buf.st_mode)) {
3318                         if (dev == st_buf.st_dev && ino == st_buf.st_ino) {
3319                                 ret = 1;
3320                                 break;
3321                         }
3322                 }
3323         }
3324
3325 out:
3326         fclose(f);
3327
3328         return ret;
3329 }
3330
3331 /*
3332  * Check for existing filesystem or partition table on device.
3333  * Returns:
3334  *       1 for existing fs or partition
3335  *       0 for nothing found
3336  *      -1 for internal error
3337  */
3338 static int check_overwrite(const char *device)
3339 {
3340         const char      *type;
3341         blkid_probe     pr = NULL;
3342         int             ret;
3343         blkid_loff_t    size;
3344
3345         if (!device || !*device)
3346                 return 0;
3347
3348         ret = -1; /* will reset on success of all setup calls */
3349
3350         pr = blkid_new_probe_from_filename(device);
3351         if (!pr)
3352                 goto out;
3353
3354         size = blkid_probe_get_size(pr);
3355         if (size < 0)
3356                 goto out;
3357
3358         /* nothing to overwrite on a 0-length device */
3359         if (size == 0) {
3360                 ret = 0;
3361                 goto out;
3362         }
3363
3364         ret = blkid_probe_enable_partitions(pr, 1);
3365         if (ret < 0)
3366                 goto out;
3367
3368         ret = blkid_do_fullprobe(pr);
3369         if (ret < 0)
3370                 goto out;
3371
3372         /*
3373          * Blkid returns 1 for nothing found and 0 when it finds a signature,
3374          * but we want the exact opposite, so reverse the return value here.
3375          *
3376          * In addition print some useful diagnostics about what actually is
3377          * on the device.
3378          */
3379         if (ret) {
3380                 ret = 0;
3381                 goto out;
3382         }
3383
3384         if (!blkid_probe_lookup_value(pr, "TYPE", &type, NULL)) {
3385                 fprintf(stderr,
3386                         "%s appears to contain an existing "
3387                         "filesystem (%s).\n", device, type);
3388         } else if (!blkid_probe_lookup_value(pr, "PTTYPE", &type, NULL)) {
3389                 fprintf(stderr,
3390                         "%s appears to contain a partition "
3391                         "table (%s).\n", device, type);
3392         } else {
3393                 fprintf(stderr,
3394                         "%s appears to contain something weird "
3395                         "according to blkid\n", device);
3396         }
3397         ret = 1;
3398
3399 out:
3400         if (pr)
3401                 blkid_free_probe(pr);
3402         if (ret == -1)
3403                 fprintf(stderr,
3404                         "probe of %s failed, cannot detect "
3405                           "existing filesystem.\n", device);
3406         return ret;
3407 }
3408
3409 static int group_profile_devs_min(u64 flag)
3410 {
3411         switch (flag & BTRFS_BLOCK_GROUP_PROFILE_MASK) {
3412         case 0: /* single */
3413         case BTRFS_BLOCK_GROUP_DUP:
3414                 return 1;
3415         case BTRFS_BLOCK_GROUP_RAID0:
3416         case BTRFS_BLOCK_GROUP_RAID1:
3417         case BTRFS_BLOCK_GROUP_RAID5:
3418                 return 2;
3419         case BTRFS_BLOCK_GROUP_RAID6:
3420                 return 3;
3421         case BTRFS_BLOCK_GROUP_RAID10:
3422                 return 4;
3423         default:
3424                 return -1;
3425         }
3426 }
3427
3428 int test_num_disk_vs_raid(u64 metadata_profile, u64 data_profile,
3429         u64 dev_cnt, int mixed, int ssd)
3430 {
3431         u64 allowed = 0;
3432         u64 profile = metadata_profile | data_profile;
3433
3434         switch (dev_cnt) {
3435         default:
3436         case 4:
3437                 allowed |= BTRFS_BLOCK_GROUP_RAID10;
3438         case 3:
3439                 allowed |= BTRFS_BLOCK_GROUP_RAID6;
3440         case 2:
3441                 allowed |= BTRFS_BLOCK_GROUP_RAID0 | BTRFS_BLOCK_GROUP_RAID1 |
3442                         BTRFS_BLOCK_GROUP_RAID5;
3443         case 1:
3444                 allowed |= BTRFS_BLOCK_GROUP_DUP;
3445         }
3446
3447         if (dev_cnt > 1 && profile & BTRFS_BLOCK_GROUP_DUP) {
3448                 warning("DUP is not recommended on filesystem with multiple devices");
3449         }
3450         if (metadata_profile & ~allowed) {
3451                 fprintf(stderr,
3452                         "ERROR: unable to create FS with metadata profile %s "
3453                         "(have %llu devices but %d devices are required)\n",
3454                         btrfs_group_profile_str(metadata_profile), dev_cnt,
3455                         group_profile_devs_min(metadata_profile));
3456                 return 1;
3457         }
3458         if (data_profile & ~allowed) {
3459                 fprintf(stderr,
3460                         "ERROR: unable to create FS with data profile %s "
3461                         "(have %llu devices but %d devices are required)\n",
3462                         btrfs_group_profile_str(data_profile), dev_cnt,
3463                         group_profile_devs_min(data_profile));
3464                 return 1;
3465         }
3466
3467         if (dev_cnt == 3 && profile & BTRFS_BLOCK_GROUP_RAID6) {
3468                 warning("RAID6 is not recommended on filesystem with 3 devices only");
3469         }
3470         if (dev_cnt == 2 && profile & BTRFS_BLOCK_GROUP_RAID5) {
3471                 warning("RAID5 is not recommended on filesystem with 2 devices only");
3472         }
3473         warning_on(!mixed && (data_profile & BTRFS_BLOCK_GROUP_DUP) && ssd,
3474                    "DUP may not actually lead to 2 copies on the device, see manual page");
3475
3476         return 0;
3477 }
3478
3479 int group_profile_max_safe_loss(u64 flags)
3480 {
3481         switch (flags & BTRFS_BLOCK_GROUP_PROFILE_MASK) {
3482         case 0: /* single */
3483         case BTRFS_BLOCK_GROUP_DUP:
3484         case BTRFS_BLOCK_GROUP_RAID0:
3485                 return 0;
3486         case BTRFS_BLOCK_GROUP_RAID1:
3487         case BTRFS_BLOCK_GROUP_RAID5:
3488         case BTRFS_BLOCK_GROUP_RAID10:
3489                 return 1;
3490         case BTRFS_BLOCK_GROUP_RAID6:
3491                 return 2;
3492         default:
3493                 return -1;
3494         }
3495 }
3496
3497 /*
3498  * Check if a device is suitable for btrfs
3499  * returns:
3500  *  1: something is wrong, an error is printed
3501  *  0: all is fine
3502  */
3503 int test_dev_for_mkfs(const char *file, int force_overwrite)
3504 {
3505         int ret, fd;
3506         struct stat st;
3507
3508         ret = is_swap_device(file);
3509         if (ret < 0) {
3510                 error("checking status of %s: %s", file, strerror(-ret));
3511                 return 1;
3512         }
3513         if (ret == 1) {
3514                 error("%s is a swap device", file);
3515                 return 1;
3516         }
3517         if (!force_overwrite) {
3518                 if (check_overwrite(file)) {
3519                         error("use the -f option to force overwrite of %s",
3520                                         file);
3521                         return 1;
3522                 }
3523         }
3524         ret = check_mounted(file);
3525         if (ret < 0) {
3526                 error("cannot check mount status of %s: %s", file,
3527                                 strerror(-ret));
3528                 return 1;
3529         }
3530         if (ret == 1) {
3531                 error("%s is mounted", file);
3532                 return 1;
3533         }
3534         /* check if the device is busy */
3535         fd = open(file, O_RDWR|O_EXCL);
3536         if (fd < 0) {
3537                 error("unable to open %s: %s", file, strerror(errno));
3538                 return 1;
3539         }
3540         if (fstat(fd, &st)) {
3541                 error("unable to stat %s: %s", file, strerror(errno));
3542                 close(fd);
3543                 return 1;
3544         }
3545         if (!S_ISBLK(st.st_mode)) {
3546                 error("%s is not a block device", file);
3547                 close(fd);
3548                 return 1;
3549         }
3550         close(fd);
3551         return 0;
3552 }
3553
3554 int btrfs_scan_devices(void)
3555 {
3556         int fd = -1;
3557         int ret;
3558         u64 num_devices;
3559         struct btrfs_fs_devices *tmp_devices;
3560         blkid_dev_iterate iter = NULL;
3561         blkid_dev dev = NULL;
3562         blkid_cache cache = NULL;
3563         char path[PATH_MAX];
3564
3565         if (btrfs_scan_done)
3566                 return 0;
3567
3568         if (blkid_get_cache(&cache, NULL) < 0) {
3569                 error("blkid cache get failed");
3570                 return 1;
3571         }
3572         blkid_probe_all(cache);
3573         iter = blkid_dev_iterate_begin(cache);
3574         blkid_dev_set_search(iter, "TYPE", "btrfs");
3575         while (blkid_dev_next(iter, &dev) == 0) {
3576                 dev = blkid_verify(cache, dev);
3577                 if (!dev)
3578                         continue;
3579                 /* if we are here its definitely a btrfs disk*/
3580                 strncpy_null(path, blkid_dev_devname(dev));
3581
3582                 fd = open(path, O_RDONLY);
3583                 if (fd < 0) {
3584                         error("cannot open %s: %s", path, strerror(errno));
3585                         continue;
3586                 }
3587                 ret = btrfs_scan_one_device(fd, path, &tmp_devices,
3588                                 &num_devices, BTRFS_SUPER_INFO_OFFSET,
3589                                 SBREAD_DEFAULT);
3590                 if (ret) {
3591                         error("cannot scan %s: %s", path, strerror(-ret));
3592                         close (fd);
3593                         continue;
3594                 }
3595
3596                 close(fd);
3597         }
3598         blkid_dev_iterate_end(iter);
3599         blkid_put_cache(cache);
3600
3601         btrfs_scan_done = 1;
3602
3603         return 0;
3604 }
3605
3606 int is_vol_small(const char *file)
3607 {
3608         int fd = -1;
3609         int e;
3610         struct stat st;
3611         u64 size;
3612
3613         fd = open(file, O_RDONLY);
3614         if (fd < 0)
3615                 return -errno;
3616         if (fstat(fd, &st) < 0) {
3617                 e = -errno;
3618                 close(fd);
3619                 return e;
3620         }
3621         size = btrfs_device_size(fd, &st);
3622         if (size == 0) {
3623                 close(fd);
3624                 return -1;
3625         }
3626         if (size < BTRFS_MKFS_SMALL_VOLUME_SIZE) {
3627                 close(fd);
3628                 return 1;
3629         } else {
3630                 close(fd);
3631                 return 0;
3632         }
3633 }
3634
3635 /*
3636  * This reads a line from the stdin and only returns non-zero if the
3637  * first whitespace delimited token is a case insensitive match with yes
3638  * or y.
3639  */
3640 int ask_user(const char *question)
3641 {
3642         char buf[30] = {0,};
3643         char *saveptr = NULL;
3644         char *answer;
3645
3646         printf("%s [y/N]: ", question);
3647
3648         return fgets(buf, sizeof(buf) - 1, stdin) &&
3649                (answer = strtok_r(buf, " \t\n\r", &saveptr)) &&
3650                (!strcasecmp(answer, "yes") || !strcasecmp(answer, "y"));
3651 }
3652
3653 /*
3654  * return 0 if a btrfs mount point is found
3655  * return 1 if a mount point is found but not btrfs
3656  * return <0 if something goes wrong
3657  */
3658 int find_mount_root(const char *path, char **mount_root)
3659 {
3660         FILE *mnttab;
3661         int fd;
3662         struct mntent *ent;
3663         int len;
3664         int ret;
3665         int not_btrfs = 1;
3666         int longest_matchlen = 0;
3667         char *longest_match = NULL;
3668
3669         fd = open(path, O_RDONLY | O_NOATIME);
3670         if (fd < 0)
3671                 return -errno;
3672         close(fd);
3673
3674         mnttab = setmntent("/proc/self/mounts", "r");
3675         if (!mnttab)
3676                 return -errno;
3677
3678         while ((ent = getmntent(mnttab))) {
3679                 len = strlen(ent->mnt_dir);
3680                 if (strncmp(ent->mnt_dir, path, len) == 0) {
3681                         /* match found and use the latest match */
3682                         if (longest_matchlen <= len) {
3683                                 free(longest_match);
3684                                 longest_matchlen = len;
3685                                 longest_match = strdup(ent->mnt_dir);
3686                                 not_btrfs = strcmp(ent->mnt_type, "btrfs");
3687                         }
3688                 }
3689         }
3690         endmntent(mnttab);
3691
3692         if (!longest_match)
3693                 return -ENOENT;
3694         if (not_btrfs) {
3695                 free(longest_match);
3696                 return 1;
3697         }
3698
3699         ret = 0;
3700         *mount_root = realpath(longest_match, NULL);
3701         if (!*mount_root)
3702                 ret = -errno;
3703
3704         free(longest_match);
3705         return ret;
3706 }
3707
3708 int test_minimum_size(const char *file, u32 nodesize)
3709 {
3710         int fd;
3711         struct stat statbuf;
3712
3713         fd = open(file, O_RDONLY);
3714         if (fd < 0)
3715                 return -errno;
3716         if (stat(file, &statbuf) < 0) {
3717                 close(fd);
3718                 return -errno;
3719         }
3720         if (btrfs_device_size(fd, &statbuf) < btrfs_min_dev_size(nodesize)) {
3721                 close(fd);
3722                 return 1;
3723         }
3724         close(fd);
3725         return 0;
3726 }
3727
3728
3729 /*
3730  * Test if path is a directory
3731  * Returns:
3732  *   0 - path exists but it is not a directory
3733  *   1 - path exists and it is a directory
3734  * < 0 - error
3735  */
3736 int test_isdir(const char *path)
3737 {
3738         struct stat st;
3739         int ret;
3740
3741         ret = stat(path, &st);
3742         if (ret < 0)
3743                 return -errno;
3744
3745         return !!S_ISDIR(st.st_mode);
3746 }
3747
3748 void units_set_mode(unsigned *units, unsigned mode)
3749 {
3750         unsigned base = *units & UNITS_MODE_MASK;
3751
3752         *units = base | mode;
3753 }
3754
3755 void units_set_base(unsigned *units, unsigned base)
3756 {
3757         unsigned mode = *units & ~UNITS_MODE_MASK;
3758
3759         *units = base | mode;
3760 }
3761
3762 int find_next_key(struct btrfs_path *path, struct btrfs_key *key)
3763 {
3764         int level;
3765
3766         for (level = 0; level < BTRFS_MAX_LEVEL; level++) {
3767                 if (!path->nodes[level])
3768                         break;
3769                 if (path->slots[level] + 1 >=
3770                     btrfs_header_nritems(path->nodes[level]))
3771                         continue;
3772                 if (level == 0)
3773                         btrfs_item_key_to_cpu(path->nodes[level], key,
3774                                               path->slots[level] + 1);
3775                 else
3776                         btrfs_node_key_to_cpu(path->nodes[level], key,
3777                                               path->slots[level] + 1);
3778                 return 0;
3779         }
3780         return 1;
3781 }
3782
3783 const char* btrfs_group_type_str(u64 flag)
3784 {
3785         u64 mask = BTRFS_BLOCK_GROUP_TYPE_MASK |
3786                 BTRFS_SPACE_INFO_GLOBAL_RSV;
3787
3788         switch (flag & mask) {
3789         case BTRFS_BLOCK_GROUP_DATA:
3790                 return "Data";
3791         case BTRFS_BLOCK_GROUP_SYSTEM:
3792                 return "System";
3793         case BTRFS_BLOCK_GROUP_METADATA:
3794                 return "Metadata";
3795         case BTRFS_BLOCK_GROUP_DATA|BTRFS_BLOCK_GROUP_METADATA:
3796                 return "Data+Metadata";
3797         case BTRFS_SPACE_INFO_GLOBAL_RSV:
3798                 return "GlobalReserve";
3799         default:
3800                 return "unknown";
3801         }
3802 }
3803
3804 const char* btrfs_group_profile_str(u64 flag)
3805 {
3806         switch (flag & BTRFS_BLOCK_GROUP_PROFILE_MASK) {
3807         case 0:
3808                 return "single";
3809         case BTRFS_BLOCK_GROUP_RAID0:
3810                 return "RAID0";
3811         case BTRFS_BLOCK_GROUP_RAID1:
3812                 return "RAID1";
3813         case BTRFS_BLOCK_GROUP_RAID5:
3814                 return "RAID5";
3815         case BTRFS_BLOCK_GROUP_RAID6:
3816                 return "RAID6";
3817         case BTRFS_BLOCK_GROUP_DUP:
3818                 return "DUP";
3819         case BTRFS_BLOCK_GROUP_RAID10:
3820                 return "RAID10";
3821         default:
3822                 return "unknown";
3823         }
3824 }
3825
3826 u64 disk_size(const char *path)
3827 {
3828         struct statfs sfs;
3829
3830         if (statfs(path, &sfs) < 0)
3831                 return 0;
3832         else
3833                 return sfs.f_bsize * sfs.f_blocks;
3834 }
3835
3836 u64 get_partition_size(const char *dev)
3837 {
3838         u64 result;
3839         int fd = open(dev, O_RDONLY);
3840
3841         if (fd < 0)
3842                 return 0;
3843         if (ioctl(fd, BLKGETSIZE64, &result) < 0) {
3844                 close(fd);
3845                 return 0;
3846         }
3847         close(fd);
3848
3849         return result;
3850 }
3851
3852 /*
3853  * Check if the BTRFS_IOC_TREE_SEARCH_V2 ioctl is supported on a given
3854  * filesystem, opened at fd
3855  */
3856 int btrfs_tree_search2_ioctl_supported(int fd)
3857 {
3858         struct btrfs_ioctl_search_args_v2 *args2;
3859         struct btrfs_ioctl_search_key *sk;
3860         int args2_size = 1024;
3861         char args2_buf[args2_size];
3862         int ret;
3863
3864         args2 = (struct btrfs_ioctl_search_args_v2 *)args2_buf;
3865         sk = &(args2->key);
3866
3867         /*
3868          * Search for the extent tree item in the root tree.
3869          */
3870         sk->tree_id = BTRFS_ROOT_TREE_OBJECTID;
3871         sk->min_objectid = BTRFS_EXTENT_TREE_OBJECTID;
3872         sk->max_objectid = BTRFS_EXTENT_TREE_OBJECTID;
3873         sk->min_type = BTRFS_ROOT_ITEM_KEY;
3874         sk->max_type = BTRFS_ROOT_ITEM_KEY;
3875         sk->min_offset = 0;
3876         sk->max_offset = (u64)-1;
3877         sk->min_transid = 0;
3878         sk->max_transid = (u64)-1;
3879         sk->nr_items = 1;
3880         args2->buf_size = args2_size - sizeof(struct btrfs_ioctl_search_args_v2);
3881         ret = ioctl(fd, BTRFS_IOC_TREE_SEARCH_V2, args2);
3882         if (ret == -EOPNOTSUPP)
3883                 return 0;
3884         else if (ret == 0)
3885                 return 1;
3886         return ret;
3887 }
3888
3889 int btrfs_check_nodesize(u32 nodesize, u32 sectorsize, u64 features)
3890 {
3891         if (nodesize < sectorsize) {
3892                 error("illegal nodesize %u (smaller than %u)",
3893                                 nodesize, sectorsize);
3894                 return -1;
3895         } else if (nodesize > BTRFS_MAX_METADATA_BLOCKSIZE) {
3896                 error("illegal nodesize %u (larger than %u)",
3897                         nodesize, BTRFS_MAX_METADATA_BLOCKSIZE);
3898                 return -1;
3899         } else if (nodesize & (sectorsize - 1)) {
3900                 error("illegal nodesize %u (not aligned to %u)",
3901                         nodesize, sectorsize);
3902                 return -1;
3903         } else if (features & BTRFS_FEATURE_INCOMPAT_MIXED_GROUPS &&
3904                    nodesize != sectorsize) {
3905                 error("illegal nodesize %u (not equal to %u for mixed block group)",
3906                         nodesize, sectorsize);
3907                 return -1;
3908         }
3909         return 0;
3910 }
3911
3912 /*
3913  * Copy a path argument from SRC to DEST and check the SRC length if it's at
3914  * most PATH_MAX and fits into DEST. DESTLEN is supposed to be exact size of
3915  * the buffer.
3916  * The destination buffer is zero terminated.
3917  * Return < 0 for error, 0 otherwise.
3918  */
3919 int arg_copy_path(char *dest, const char *src, int destlen)
3920 {
3921         size_t len = strlen(src);
3922
3923         if (len >= PATH_MAX || len >= destlen)
3924                 return -ENAMETOOLONG;
3925
3926         __strncpy_null(dest, src, destlen);
3927
3928         return 0;
3929 }
3930
3931 unsigned int get_unit_mode_from_arg(int *argc, char *argv[], int df_mode)
3932 {
3933         unsigned int unit_mode = UNITS_DEFAULT;
3934         int arg_i;
3935         int arg_end;
3936
3937         for (arg_i = 0; arg_i < *argc; arg_i++) {
3938                 if (!strcmp(argv[arg_i], "--"))
3939                         break;
3940
3941                 if (!strcmp(argv[arg_i], "--raw")) {
3942                         unit_mode = UNITS_RAW;
3943                         argv[arg_i] = NULL;
3944                         continue;
3945                 }
3946                 if (!strcmp(argv[arg_i], "--human-readable")) {
3947                         unit_mode = UNITS_HUMAN_BINARY;
3948                         argv[arg_i] = NULL;
3949                         continue;
3950                 }
3951
3952                 if (!strcmp(argv[arg_i], "--iec")) {
3953                         units_set_mode(&unit_mode, UNITS_BINARY);
3954                         argv[arg_i] = NULL;
3955                         continue;
3956                 }
3957                 if (!strcmp(argv[arg_i], "--si")) {
3958                         units_set_mode(&unit_mode, UNITS_DECIMAL);
3959                         argv[arg_i] = NULL;
3960                         continue;
3961                 }
3962
3963                 if (!strcmp(argv[arg_i], "--kbytes")) {
3964                         units_set_base(&unit_mode, UNITS_KBYTES);
3965                         argv[arg_i] = NULL;
3966                         continue;
3967                 }
3968                 if (!strcmp(argv[arg_i], "--mbytes")) {
3969                         units_set_base(&unit_mode, UNITS_MBYTES);
3970                         argv[arg_i] = NULL;
3971                         continue;
3972                 }
3973                 if (!strcmp(argv[arg_i], "--gbytes")) {
3974                         units_set_base(&unit_mode, UNITS_GBYTES);
3975                         argv[arg_i] = NULL;
3976                         continue;
3977                 }
3978                 if (!strcmp(argv[arg_i], "--tbytes")) {
3979                         units_set_base(&unit_mode, UNITS_TBYTES);
3980                         argv[arg_i] = NULL;
3981                         continue;
3982                 }
3983
3984                 if (!df_mode)
3985                         continue;
3986
3987                 if (!strcmp(argv[arg_i], "-b")) {
3988                         unit_mode = UNITS_RAW;
3989                         argv[arg_i] = NULL;
3990                         continue;
3991                 }
3992                 if (!strcmp(argv[arg_i], "-h")) {
3993                         unit_mode = UNITS_HUMAN_BINARY;
3994                         argv[arg_i] = NULL;
3995                         continue;
3996                 }
3997                 if (!strcmp(argv[arg_i], "-H")) {
3998                         unit_mode = UNITS_HUMAN_DECIMAL;
3999                         argv[arg_i] = NULL;
4000                         continue;
4001                 }
4002                 if (!strcmp(argv[arg_i], "-k")) {
4003                         units_set_base(&unit_mode, UNITS_KBYTES);
4004                         argv[arg_i] = NULL;
4005                         continue;
4006                 }
4007                 if (!strcmp(argv[arg_i], "-m")) {
4008                         units_set_base(&unit_mode, UNITS_MBYTES);
4009                         argv[arg_i] = NULL;
4010                         continue;
4011                 }
4012                 if (!strcmp(argv[arg_i], "-g")) {
4013                         units_set_base(&unit_mode, UNITS_GBYTES);
4014                         argv[arg_i] = NULL;
4015                         continue;
4016                 }
4017                 if (!strcmp(argv[arg_i], "-t")) {
4018                         units_set_base(&unit_mode, UNITS_TBYTES);
4019                         argv[arg_i] = NULL;
4020                         continue;
4021                 }
4022         }
4023
4024         for (arg_i = 0, arg_end = 0; arg_i < *argc; arg_i++) {
4025                 if (!argv[arg_i])
4026                         continue;
4027                 argv[arg_end] = argv[arg_i];
4028                 arg_end++;
4029         }
4030
4031         *argc = arg_end;
4032
4033         return unit_mode;
4034 }
4035
4036 int string_is_numerical(const char *str)
4037 {
4038         if (!str)
4039                 return 0;
4040         if (!(*str >= '0' && *str <= '9'))
4041                 return 0;
4042         while (*str >= '0' && *str <= '9')
4043                 str++;
4044         if (*str != '\0')
4045                 return 0;
4046         return 1;
4047 }
4048
4049 /*
4050  * Preprocess @argv with getopt_long to reorder options and consume the "--"
4051  * option separator.
4052  * Unknown short and long options are reported, optionally the @usage is printed
4053  * before exit.
4054  */
4055 void clean_args_no_options(int argc, char *argv[], const char * const *usagestr)
4056 {
4057         static const struct option long_options[] = {
4058                 {NULL, 0, NULL, 0}
4059         };
4060
4061         while (1) {
4062                 int c = getopt_long(argc, argv, "", long_options, NULL);
4063
4064                 if (c < 0)
4065                         break;
4066
4067                 switch (c) {
4068                 default:
4069                         if (usagestr)
4070                                 usage(usagestr);
4071                 }
4072         }
4073 }
4074
4075 /*
4076  * Same as clean_args_no_options but pass through arguments that could look
4077  * like short options. Eg. reisze which takes a negative resize argument like
4078  * '-123M' .
4079  *
4080  * This accepts only two forms:
4081  * - "-- option1 option2 ..."
4082  * - "option1 option2 ..."
4083  */
4084 void clean_args_no_options_relaxed(int argc, char *argv[], const char * const *usagestr)
4085 {
4086         if (argc <= 1)
4087                 return;
4088
4089         if (strcmp(argv[1], "--") == 0)
4090                 optind = 2;
4091 }
4092
4093 /* Subvolume helper functions */
4094 /*
4095  * test if name is a correct subvolume name
4096  * this function return
4097  * 0-> name is not a correct subvolume name
4098  * 1-> name is a correct subvolume name
4099  */
4100 int test_issubvolname(const char *name)
4101 {
4102         return name[0] != '\0' && !strchr(name, '/') &&
4103                 strcmp(name, ".") && strcmp(name, "..");
4104 }
4105
4106 /*
4107  * Test if path is a subvolume
4108  * Returns:
4109  *   0 - path exists but it is not a subvolume
4110  *   1 - path exists and it is  a subvolume
4111  * < 0 - error
4112  */
4113 int test_issubvolume(const char *path)
4114 {
4115         struct stat     st;
4116         struct statfs stfs;
4117         int             res;
4118
4119         res = stat(path, &st);
4120         if (res < 0)
4121                 return -errno;
4122
4123         if (st.st_ino != BTRFS_FIRST_FREE_OBJECTID || !S_ISDIR(st.st_mode))
4124                 return 0;
4125
4126         res = statfs(path, &stfs);
4127         if (res < 0)
4128                 return -errno;
4129
4130         return (int)stfs.f_type == BTRFS_SUPER_MAGIC;
4131 }
4132
4133 const char *subvol_strip_mountpoint(const char *mnt, const char *full_path)
4134 {
4135         int len = strlen(mnt);
4136         if (!len)
4137                 return full_path;
4138
4139         if (mnt[len - 1] != '/')
4140                 len += 1;
4141
4142         return full_path + len;
4143 }
4144
4145 /*
4146  * Returns
4147  * <0: Std error
4148  * 0: All fine
4149  * 1: Error; and error info printed to the terminal. Fixme.
4150  * 2: If the fullpath is root tree instead of subvol tree
4151  */
4152 int get_subvol_info(const char *fullpath, struct root_info *get_ri)
4153 {
4154         u64 sv_id;
4155         int ret = 1;
4156         int fd = -1;
4157         int mntfd = -1;
4158         char *mnt = NULL;
4159         const char *svpath = NULL;
4160         DIR *dirstream1 = NULL;
4161         DIR *dirstream2 = NULL;
4162
4163         ret = test_issubvolume(fullpath);
4164         if (ret < 0)
4165                 return ret;
4166         if (!ret) {
4167                 error("not a subvolume: %s", fullpath);
4168                 return 1;
4169         }
4170
4171         ret = find_mount_root(fullpath, &mnt);
4172         if (ret < 0)
4173                 return ret;
4174         if (ret > 0) {
4175                 error("%s doesn't belong to btrfs mount point", fullpath);
4176                 return 1;
4177         }
4178         ret = 1;
4179         svpath = subvol_strip_mountpoint(mnt, fullpath);
4180
4181         fd = btrfs_open_dir(fullpath, &dirstream1, 1);
4182         if (fd < 0)
4183                 goto out;
4184
4185         ret = btrfs_list_get_path_rootid(fd, &sv_id);
4186         if (ret)
4187                 goto out;
4188
4189         mntfd = btrfs_open_dir(mnt, &dirstream2, 1);
4190         if (mntfd < 0)
4191                 goto out;
4192
4193         memset(get_ri, 0, sizeof(*get_ri));
4194         get_ri->root_id = sv_id;
4195
4196         if (sv_id == BTRFS_FS_TREE_OBJECTID)
4197                 ret = btrfs_get_toplevel_subvol(mntfd, get_ri);
4198         else
4199                 ret = btrfs_get_subvol(mntfd, get_ri);
4200         if (ret)
4201                 error("can't find '%s': %d", svpath, ret);
4202
4203 out:
4204         close_file_or_dir(mntfd, dirstream2);
4205         close_file_or_dir(fd, dirstream1);
4206         free(mnt);
4207
4208         return ret;
4209 }
4210
4211 void init_rand_seed(u64 seed)
4212 {
4213         int i;
4214
4215         /* only use the last 48 bits */
4216         for (i = 0; i < 3; i++) {
4217                 rand_seed[i] = (unsigned short)(seed ^ (unsigned short)(-1));
4218                 seed >>= 16;
4219         }
4220         rand_seed_initlized = 1;
4221 }
4222
4223 static void __init_seed(void)
4224 {
4225         struct timeval tv;
4226         int ret;
4227         int fd;
4228
4229         if(rand_seed_initlized)
4230                 return;
4231         /* Use urandom as primary seed source. */
4232         fd = open("/dev/urandom", O_RDONLY);
4233         if (fd >= 0) {
4234                 ret = read(fd, rand_seed, sizeof(rand_seed));
4235                 close(fd);
4236                 if (ret < sizeof(rand_seed))
4237                         goto fallback;
4238         } else {
4239 fallback:
4240                 /* Use time and pid as fallback seed */
4241                 warning("failed to read /dev/urandom, use time and pid as random seed");
4242                 gettimeofday(&tv, 0);
4243                 rand_seed[0] = getpid() ^ (tv.tv_sec & 0xFFFF);
4244                 rand_seed[1] = getppid() ^ (tv.tv_usec & 0xFFFF);
4245                 rand_seed[2] = (tv.tv_sec ^ tv.tv_usec) >> 16;
4246         }
4247         rand_seed_initlized = 1;
4248 }
4249
4250 u32 rand_u32(void)
4251 {
4252         __init_seed();
4253         /*
4254          * Don't use nrand48, its range is [0,2^31) The highest bit will alwasy
4255          * be 0.  Use jrand48 to include the highest bit.
4256          */
4257         return (u32)jrand48(rand_seed);
4258 }
4259
4260 unsigned int rand_range(unsigned int upper)
4261 {
4262         __init_seed();
4263         /*
4264          * Use the full 48bits to mod, which would be more uniformly
4265          * distributed
4266          */
4267         return (unsigned int)(jrand48(rand_seed) % upper);
4268 }