btrfs-progs: check the free space tree in btrfsck
[platform/upstream/btrfs-progs.git] / utils.c
diff --git a/utils.c b/utils.c
index e89707c..03648db 100644 (file)
--- a/utils.c
+++ b/utils.c
@@ -35,6 +35,8 @@
 #include <limits.h>
 #include <blkid/blkid.h>
 #include <sys/vfs.h>
+#include <sys/statfs.h>
+#include <linux/magic.h>
 
 #include "kerncompat.h"
 #include "radix-tree.h"
@@ -152,7 +154,7 @@ int test_uuid_unique(char *fs_uuid)
        blkid_dev dev = NULL;
        blkid_cache cache = NULL;
 
-       if (blkid_get_cache(&cache, 0) < 0) {
+       if (blkid_get_cache(&cache, NULL) < 0) {
                printf("ERROR: lblkid cache get failed\n");
                return 1;
        }
@@ -177,12 +179,10 @@ int test_uuid_unique(char *fs_uuid)
 /*
  * @fs_uuid - if NULL, generates a UUID, returns back the new filesystem UUID
  */
-int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
-              u64 blocks[7], u64 num_bytes, u32 nodesize,
-              u32 sectorsize, u32 stripesize, u64 features)
+int make_btrfs(int fd, struct btrfs_mkfs_config *cfg)
 {
        struct btrfs_super_block super;
-       struct extent_buffer *buf = NULL;
+       struct extent_buffer *buf;
        struct btrfs_root_item root_item;
        struct btrfs_disk_key disk_key;
        struct btrfs_extent_item *extent_item;
@@ -200,59 +200,63 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        u64 ref_root;
        u32 array_size;
        u32 item_size;
-       int skinny_metadata = !!(features &
+       int skinny_metadata = !!(cfg->features &
                                 BTRFS_FEATURE_INCOMPAT_SKINNY_METADATA);
+       u64 num_bytes;
 
-       first_free = BTRFS_SUPER_INFO_OFFSET + sectorsize * 2 - 1;
-       first_free &= ~((u64)sectorsize - 1);
+       buf = malloc(sizeof(*buf) + max(cfg->sectorsize, cfg->nodesize));
+       if (!buf)
+               return -ENOMEM;
+
+       first_free = BTRFS_SUPER_INFO_OFFSET + cfg->sectorsize * 2 - 1;
+       first_free &= ~((u64)cfg->sectorsize - 1);
 
        memset(&super, 0, sizeof(super));
 
-       num_bytes = (num_bytes / sectorsize) * sectorsize;
-       if (fs_uuid && *fs_uuid) {
-               if (uuid_parse(fs_uuid, super.fsid) != 0) {
-                       fprintf(stderr, "could not parse UUID: %s\n", fs_uuid);
+       num_bytes = (cfg->num_bytes / cfg->sectorsize) * cfg->sectorsize;
+       if (cfg->fs_uuid && *cfg->fs_uuid) {
+               if (uuid_parse(cfg->fs_uuid, super.fsid) != 0) {
+                       fprintf(stderr, "could not parse UUID: %s\n",
+                               cfg->fs_uuid);
                        ret = -EINVAL;
                        goto out;
                }
-               if (!test_uuid_unique(fs_uuid)) {
-                       fprintf(stderr, "non-unique UUID: %s\n", fs_uuid);
+               if (!test_uuid_unique(cfg->fs_uuid)) {
+                       fprintf(stderr, "non-unique UUID: %s\n", cfg->fs_uuid);
                        ret = -EBUSY;
                        goto out;
                }
        } else {
                uuid_generate(super.fsid);
-               if (fs_uuid)
-                       uuid_unparse(super.fsid, fs_uuid);
+               if (cfg->fs_uuid)
+                       uuid_unparse(super.fsid, cfg->fs_uuid);
        }
        uuid_generate(super.dev_item.uuid);
        uuid_generate(chunk_tree_uuid);
 
-       btrfs_set_super_bytenr(&super, blocks[0]);
+       btrfs_set_super_bytenr(&super, cfg->blocks[0]);
        btrfs_set_super_num_devices(&super, 1);
        btrfs_set_super_magic(&super, BTRFS_MAGIC);
        btrfs_set_super_generation(&super, 1);
-       btrfs_set_super_root(&super, blocks[1]);
-       btrfs_set_super_chunk_root(&super, blocks[3]);
+       btrfs_set_super_root(&super, cfg->blocks[1]);
+       btrfs_set_super_chunk_root(&super, cfg->blocks[3]);
        btrfs_set_super_total_bytes(&super, num_bytes);
-       btrfs_set_super_bytes_used(&super, 6 * nodesize);
-       btrfs_set_super_sectorsize(&super, sectorsize);
-       btrfs_set_super_leafsize(&super, nodesize);
-       btrfs_set_super_nodesize(&super, nodesize);
-       btrfs_set_super_stripesize(&super, stripesize);
+       btrfs_set_super_bytes_used(&super, 6 * cfg->nodesize);
+       btrfs_set_super_sectorsize(&super, cfg->sectorsize);
+       btrfs_set_super_leafsize(&super, cfg->nodesize);
+       btrfs_set_super_nodesize(&super, cfg->nodesize);
+       btrfs_set_super_stripesize(&super, cfg->stripesize);
        btrfs_set_super_csum_type(&super, BTRFS_CSUM_TYPE_CRC32);
        btrfs_set_super_chunk_root_generation(&super, 1);
        btrfs_set_super_cache_generation(&super, -1);
-       btrfs_set_super_incompat_flags(&super, features);
-       if (label)
-               strncpy(super.label, label, BTRFS_LABEL_SIZE - 1);
-
-       buf = malloc(sizeof(*buf) + max(sectorsize, nodesize));
+       btrfs_set_super_incompat_flags(&super, cfg->features);
+       if (cfg->label)
+               strncpy(super.label, cfg->label, BTRFS_LABEL_SIZE - 1);
 
        /* create the tree of root objects */
-       memset(buf->data, 0, nodesize);
-       buf->len = nodesize;
-       btrfs_set_header_bytenr(buf, blocks[1]);
+       memset(buf->data, 0, cfg->nodesize);
+       buf->len = cfg->nodesize;
+       btrfs_set_header_bytenr(buf, cfg->blocks[1]);
        btrfs_set_header_nritems(buf, 4);
        btrfs_set_header_generation(buf, 1);
        btrfs_set_header_backref_rev(buf, BTRFS_MIXED_BACKREF_REV);
@@ -270,10 +274,10 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        btrfs_set_stack_inode_generation(inode_item, 1);
        btrfs_set_stack_inode_size(inode_item, 3);
        btrfs_set_stack_inode_nlink(inode_item, 1);
-       btrfs_set_stack_inode_nbytes(inode_item, nodesize);
+       btrfs_set_stack_inode_nbytes(inode_item, cfg->nodesize);
        btrfs_set_stack_inode_mode(inode_item, S_IFDIR | 0755);
        btrfs_set_root_refs(&root_item, 1);
-       btrfs_set_root_used(&root_item, nodesize);
+       btrfs_set_root_used(&root_item, cfg->nodesize);
        btrfs_set_root_generation(&root_item, 1);
 
        memset(&disk_key, 0, sizeof(disk_key));
@@ -281,8 +285,8 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        btrfs_set_disk_key_offset(&disk_key, 0);
        nritems = 0;
 
-       itemoff = __BTRFS_LEAF_DATA_SIZE(nodesize) - sizeof(root_item);
-       btrfs_set_root_bytenr(&root_item, blocks[2]);
+       itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize) - sizeof(root_item);
+       btrfs_set_root_bytenr(&root_item, cfg->blocks[2]);
        btrfs_set_disk_key_objectid(&disk_key, BTRFS_EXTENT_TREE_OBJECTID);
        btrfs_set_item_key(buf, &disk_key, nritems);
        btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
@@ -293,7 +297,7 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        nritems++;
 
        itemoff = itemoff - sizeof(root_item);
-       btrfs_set_root_bytenr(&root_item, blocks[4]);
+       btrfs_set_root_bytenr(&root_item, cfg->blocks[4]);
        btrfs_set_disk_key_objectid(&disk_key, BTRFS_DEV_TREE_OBJECTID);
        btrfs_set_item_key(buf, &disk_key, nritems);
        btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
@@ -305,7 +309,7 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        nritems++;
 
        itemoff = itemoff - sizeof(root_item);
-       btrfs_set_root_bytenr(&root_item, blocks[5]);
+       btrfs_set_root_bytenr(&root_item, cfg->blocks[5]);
        btrfs_set_disk_key_objectid(&disk_key, BTRFS_FS_TREE_OBJECTID);
        btrfs_set_item_key(buf, &disk_key, nritems);
        btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
@@ -317,7 +321,7 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        nritems++;
 
        itemoff = itemoff - sizeof(root_item);
-       btrfs_set_root_bytenr(&root_item, blocks[6]);
+       btrfs_set_root_bytenr(&root_item, cfg->blocks[6]);
        btrfs_set_disk_key_objectid(&disk_key, BTRFS_CSUM_TREE_OBJECTID);
        btrfs_set_item_key(buf, &disk_key, nritems);
        btrfs_set_item_offset(buf, btrfs_item_nr(nritems), itemoff);
@@ -330,28 +334,28 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
 
 
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, nodesize, blocks[1]);
-       if (ret != nodesize) {
+       ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[1]);
+       if (ret != cfg->nodesize) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
 
        /* create the items for the extent tree */
        memset(buf->data + sizeof(struct btrfs_header), 0,
-               nodesize - sizeof(struct btrfs_header));
+               cfg->nodesize - sizeof(struct btrfs_header));
        nritems = 0;
-       itemoff = __BTRFS_LEAF_DATA_SIZE(nodesize);
+       itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize);
        for (i = 1; i < 7; i++) {
                item_size = sizeof(struct btrfs_extent_item);
                if (!skinny_metadata)
                        item_size += sizeof(struct btrfs_tree_block_info);
 
-               BUG_ON(blocks[i] < first_free);
-               BUG_ON(blocks[i] < blocks[i - 1]);
+               BUG_ON(cfg->blocks[i] < first_free);
+               BUG_ON(cfg->blocks[i] < cfg->blocks[i - 1]);
 
                /* create extent item */
                itemoff -= item_size;
-               btrfs_set_disk_key_objectid(&disk_key, blocks[i]);
+               btrfs_set_disk_key_objectid(&disk_key, cfg->blocks[i]);
                if (skinny_metadata) {
                        btrfs_set_disk_key_type(&disk_key,
                                                BTRFS_METADATA_ITEM_KEY);
@@ -359,7 +363,7 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
                } else {
                        btrfs_set_disk_key_type(&disk_key,
                                                BTRFS_EXTENT_ITEM_KEY);
-                       btrfs_set_disk_key_offset(&disk_key, nodesize);
+                       btrfs_set_disk_key_offset(&disk_key, cfg->nodesize);
                }
                btrfs_set_item_key(buf, &disk_key, nritems);
                btrfs_set_item_offset(buf, btrfs_item_nr(nritems),
@@ -376,7 +380,7 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
 
                /* create extent ref */
                ref_root = reference_root_table[i];
-               btrfs_set_disk_key_objectid(&disk_key, blocks[i]);
+               btrfs_set_disk_key_objectid(&disk_key, cfg->blocks[i]);
                btrfs_set_disk_key_offset(&disk_key, ref_root);
                btrfs_set_disk_key_type(&disk_key, BTRFS_TREE_BLOCK_REF_KEY);
                btrfs_set_item_key(buf, &disk_key, nritems);
@@ -385,22 +389,22 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
                btrfs_set_item_size(buf, btrfs_item_nr(nritems), 0);
                nritems++;
        }
-       btrfs_set_header_bytenr(buf, blocks[2]);
+       btrfs_set_header_bytenr(buf, cfg->blocks[2]);
        btrfs_set_header_owner(buf, BTRFS_EXTENT_TREE_OBJECTID);
        btrfs_set_header_nritems(buf, nritems);
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, nodesize, blocks[2]);
-       if (ret != nodesize) {
+       ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[2]);
+       if (ret != cfg->nodesize) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
 
        /* create the chunk tree */
        memset(buf->data + sizeof(struct btrfs_header), 0,
-               nodesize - sizeof(struct btrfs_header));
+               cfg->nodesize - sizeof(struct btrfs_header));
        nritems = 0;
        item_size = sizeof(*dev_item);
-       itemoff = __BTRFS_LEAF_DATA_SIZE(nodesize) - item_size;
+       itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize) - item_size;
 
        /* first device 1 (there is no device 0) */
        btrfs_set_disk_key_objectid(&disk_key, BTRFS_DEV_ITEMS_OBJECTID);
@@ -416,9 +420,9 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        btrfs_set_device_total_bytes(buf, dev_item, num_bytes);
        btrfs_set_device_bytes_used(buf, dev_item,
                                    BTRFS_MKFS_SYSTEM_GROUP_SIZE);
-       btrfs_set_device_io_align(buf, dev_item, sectorsize);
-       btrfs_set_device_io_width(buf, dev_item, sectorsize);
-       btrfs_set_device_sector_size(buf, dev_item, sectorsize);
+       btrfs_set_device_io_align(buf, dev_item, cfg->sectorsize);
+       btrfs_set_device_io_width(buf, dev_item, cfg->sectorsize);
+       btrfs_set_device_sector_size(buf, dev_item, cfg->sectorsize);
        btrfs_set_device_type(buf, dev_item, 0);
 
        write_extent_buffer(buf, super.dev_item.uuid,
@@ -447,9 +451,9 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        btrfs_set_chunk_owner(buf, chunk, BTRFS_EXTENT_TREE_OBJECTID);
        btrfs_set_chunk_stripe_len(buf, chunk, 64 * 1024);
        btrfs_set_chunk_type(buf, chunk, BTRFS_BLOCK_GROUP_SYSTEM);
-       btrfs_set_chunk_io_align(buf, chunk, sectorsize);
-       btrfs_set_chunk_io_width(buf, chunk, sectorsize);
-       btrfs_set_chunk_sector_size(buf, chunk, sectorsize);
+       btrfs_set_chunk_io_align(buf, chunk, cfg->sectorsize);
+       btrfs_set_chunk_io_width(buf, chunk, cfg->sectorsize);
+       btrfs_set_chunk_sector_size(buf, chunk, cfg->sectorsize);
        btrfs_set_chunk_num_stripes(buf, chunk, 1);
        btrfs_set_stripe_devid_nr(buf, chunk, 0, 1);
        btrfs_set_stripe_offset_nr(buf, chunk, 0, 0);
@@ -472,21 +476,21 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
        ptr += item_size;
        btrfs_set_super_sys_array_size(&super, array_size);
 
-       btrfs_set_header_bytenr(buf, blocks[3]);
+       btrfs_set_header_bytenr(buf, cfg->blocks[3]);
        btrfs_set_header_owner(buf, BTRFS_CHUNK_TREE_OBJECTID);
        btrfs_set_header_nritems(buf, nritems);
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, nodesize, blocks[3]);
-       if (ret != nodesize) {
+       ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[3]);
+       if (ret != cfg->nodesize) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
 
        /* create the device tree */
        memset(buf->data + sizeof(struct btrfs_header), 0,
-               nodesize - sizeof(struct btrfs_header));
+               cfg->nodesize - sizeof(struct btrfs_header));
        nritems = 0;
-       itemoff = __BTRFS_LEAF_DATA_SIZE(nodesize) -
+       itemoff = __BTRFS_LEAF_DATA_SIZE(cfg->nodesize) -
                sizeof(struct btrfs_dev_extent);
 
        btrfs_set_disk_key_objectid(&disk_key, 1);
@@ -511,49 +515,49 @@ int make_btrfs(int fd, const char *device, const char *label, char *fs_uuid,
                                    BTRFS_MKFS_SYSTEM_GROUP_SIZE);
        nritems++;
 
-       btrfs_set_header_bytenr(buf, blocks[4]);
+       btrfs_set_header_bytenr(buf, cfg->blocks[4]);
        btrfs_set_header_owner(buf, BTRFS_DEV_TREE_OBJECTID);
        btrfs_set_header_nritems(buf, nritems);
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, nodesize, blocks[4]);
-       if (ret != nodesize) {
+       ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[4]);
+       if (ret != cfg->nodesize) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
 
        /* create the FS root */
        memset(buf->data + sizeof(struct btrfs_header), 0,
-               nodesize - sizeof(struct btrfs_header));
-       btrfs_set_header_bytenr(buf, blocks[5]);
+               cfg->nodesize - sizeof(struct btrfs_header));
+       btrfs_set_header_bytenr(buf, cfg->blocks[5]);
        btrfs_set_header_owner(buf, BTRFS_FS_TREE_OBJECTID);
        btrfs_set_header_nritems(buf, 0);
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, nodesize, blocks[5]);
-       if (ret != nodesize) {
+       ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[5]);
+       if (ret != cfg->nodesize) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
        /* finally create the csum root */
        memset(buf->data + sizeof(struct btrfs_header), 0,
-               nodesize - sizeof(struct btrfs_header));
-       btrfs_set_header_bytenr(buf, blocks[6]);
+               cfg->nodesize - sizeof(struct btrfs_header));
+       btrfs_set_header_bytenr(buf, cfg->blocks[6]);
        btrfs_set_header_owner(buf, BTRFS_CSUM_TREE_OBJECTID);
        btrfs_set_header_nritems(buf, 0);
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, nodesize, blocks[6]);
-       if (ret != nodesize) {
+       ret = pwrite(fd, buf->data, cfg->nodesize, cfg->blocks[6]);
+       if (ret != cfg->nodesize) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
 
        /* and write out the super block */
-       BUG_ON(sizeof(super) > sectorsize);
-       memset(buf->data, 0, sectorsize);
+       BUG_ON(sizeof(super) > cfg->sectorsize);
+       memset(buf->data, 0, BTRFS_SUPER_INFO_SIZE);
        memcpy(buf->data, &super, sizeof(super));
-       buf->len = sectorsize;
+       buf->len = BTRFS_SUPER_INFO_SIZE;
        csum_tree_block_size(buf, BTRFS_CRC32_SIZE, 0);
-       ret = pwrite(fd, buf->data, sectorsize, blocks[0]);
-       if (ret != sectorsize) {
+       ret = pwrite(fd, buf->data, BTRFS_SUPER_INFO_SIZE, cfg->blocks[0]);
+       if (ret != BTRFS_SUPER_INFO_SIZE) {
                ret = (ret < 0 ? -errno : -EIO);
                goto out;
        }
@@ -722,28 +726,27 @@ static int zero_dev_clamped(int fd, off_t start, ssize_t len, u64 dev_size)
 
 int btrfs_add_to_fsid(struct btrfs_trans_handle *trans,
                      struct btrfs_root *root, int fd, char *path,
-                     u64 block_count, u32 io_width, u32 io_align,
+                     u64 device_total_bytes, u32 io_width, u32 io_align,
                      u32 sectorsize)
 {
        struct btrfs_super_block *disk_super;
        struct btrfs_super_block *super = root->fs_info->super_copy;
        struct btrfs_device *device;
        struct btrfs_dev_item *dev_item;
-       char *buf;
-       u64 total_bytes;
+       char *buf = NULL;
+       u64 fs_total_bytes;
        u64 num_devs;
        int ret;
 
+       device_total_bytes = (device_total_bytes / sectorsize) * sectorsize;
+
        device = kzalloc(sizeof(*device), GFP_NOFS);
        if (!device)
-               return -ENOMEM;
-       buf = kmalloc(sectorsize, GFP_NOFS);
-       if (!buf) {
-               kfree(device);
-               return -ENOMEM;
-       }
+               goto err_nomem;
+       buf = kzalloc(sectorsize, GFP_NOFS);
+       if (!buf)
+               goto err_nomem;
        BUG_ON(sizeof(*disk_super) > sectorsize);
-       memset(buf, 0, sectorsize);
 
        disk_super = (struct btrfs_super_block *)buf;
        dev_item = &disk_super->dev_item;
@@ -756,17 +759,20 @@ int btrfs_add_to_fsid(struct btrfs_trans_handle *trans,
        device->sector_size = sectorsize;
        device->fd = fd;
        device->writeable = 1;
-       device->total_bytes = block_count;
+       device->total_bytes = device_total_bytes;
        device->bytes_used = 0;
        device->total_ios = 0;
        device->dev_root = root->fs_info->dev_root;
        device->name = strdup(path);
+       if (!device->name)
+               goto err_nomem;
 
+       INIT_LIST_HEAD(&device->dev_list);
        ret = btrfs_add_device(trans, root, device);
        BUG_ON(ret);
 
-       total_bytes = btrfs_super_total_bytes(super) + block_count;
-       btrfs_set_super_total_bytes(super, total_bytes);
+       fs_total_bytes = btrfs_super_total_bytes(super) + device_total_bytes;
+       btrfs_set_super_total_bytes(super, fs_total_bytes);
 
        num_devs = btrfs_super_num_devices(super) + 1;
        btrfs_set_super_num_devices(super, num_devs);
@@ -790,46 +796,63 @@ int btrfs_add_to_fsid(struct btrfs_trans_handle *trans,
        list_add(&device->dev_list, &root->fs_info->fs_devices->devices);
        device->fs_devices = root->fs_info->fs_devices;
        return 0;
+
+err_nomem:
+       kfree(device);
+       kfree(buf);
+       return -ENOMEM;
 }
 
-static void btrfs_wipe_existing_sb(int fd)
+static int btrfs_wipe_existing_sb(int fd)
 {
        const char *off = NULL;
        size_t len = 0;
        loff_t offset;
        char buf[BUFSIZ];
-       int rc = 0;
+       int ret = 0;
        blkid_probe pr = NULL;
 
        pr = blkid_new_probe();
        if (!pr)
-               return;
+               return -1;
 
-       if (blkid_probe_set_device(pr, fd, 0, 0))
+       if (blkid_probe_set_device(pr, fd, 0, 0)) {
+               ret = -1;
                goto out;
+       }
 
-       rc = blkid_probe_lookup_value(pr, "SBMAGIC_OFFSET", &off, NULL);
-       if (!rc)
-               rc = blkid_probe_lookup_value(pr, "SBMAGIC", NULL, &len);
+       ret = blkid_probe_lookup_value(pr, "SBMAGIC_OFFSET", &off, NULL);
+       if (!ret)
+               ret = blkid_probe_lookup_value(pr, "SBMAGIC", NULL, &len);
 
-       if (rc || len == 0 || off == NULL)
+       if (ret || len == 0 || off == NULL) {
+               /*
+                * If lookup fails, the probe did not find any values, eg. for
+                * a file image or a loop device. Soft error.
+                */
+               ret = 1;
                goto out;
+       }
 
        offset = strtoll(off, NULL, 10);
        if (len > sizeof(buf))
                len = sizeof(buf);
 
        memset(buf, 0, len);
-       rc = pwrite(fd, buf, len, offset);
+       ret = pwrite(fd, buf, len, offset);
+       if (ret != len) {
+               fprintf(stderr, "ERROR: cannot wipe existing superblock\n");
+               ret = -1;
+       }
        fsync(fd);
 
 out:
        blkid_free_probe(pr);
-       return;
+       return ret;
 }
 
 int btrfs_prepare_device(int fd, char *file, int zero_end, u64 *block_count_ret,
-                          u64 max_block_count, int *mixed, int discard)
+                          u64 max_block_count, int discard)
 {
        u64 block_count;
        struct stat st;
@@ -849,9 +872,6 @@ int btrfs_prepare_device(int fd, char *file, int zero_end, u64 *block_count_ret,
        if (max_block_count)
                block_count = min(block_count, max_block_count);
 
-       if (block_count < BTRFS_MKFS_SMALL_VOLUME_SIZE && !(*mixed))
-               *mixed = 1;
-
        if (discard) {
                /*
                 * We intentionally ignore errors from the discard ioctl.  It
@@ -879,7 +899,12 @@ int btrfs_prepare_device(int fd, char *file, int zero_end, u64 *block_count_ret,
                return 1;
        }
 
-       btrfs_wipe_existing_sb(fd);
+       ret = btrfs_wipe_existing_sb(fd);
+       if (ret < 0) {
+               fprintf(stderr, "ERROR: cannot wipe superblocks on '%s'\n",
+                               file);
+               return 1;
+       }
 
        *block_count_ret = block_count;
        return 0;
@@ -936,7 +961,7 @@ int is_block_device(const char *path)
        if (stat(path, &statbuf) < 0)
                return -errno;
 
-       return S_ISBLK(statbuf.st_mode);
+       return !!S_ISBLK(statbuf.st_mode);
 }
 
 /*
@@ -1058,26 +1083,82 @@ out:
  *
  * On error, return -1, errno should be set.
  */
-int open_path_or_dev_mnt(const char *path, DIR **dirstream)
+int open_path_or_dev_mnt(const char *path, DIR **dirstream, int verbose)
 {
-       char mp[BTRFS_PATH_NAME_MAX + 1];
-       int fdmnt;
+       char mp[PATH_MAX];
+       int ret;
 
        if (is_block_device(path)) {
-               int ret;
-
                ret = get_btrfs_mount(path, mp, sizeof(mp));
                if (ret < 0) {
                        /* not a mounted btrfs dev */
+                       error_on(verbose, "'%s' is not a mounted btrfs device",
+                                path);
                        errno = EINVAL;
                        return -1;
                }
-               fdmnt = open_file_or_dir(mp, dirstream);
+               ret = open_file_or_dir(mp, dirstream);
+               error_on(verbose && ret < 0, "can't access '%s': %s",
+                        path, strerror(errno));
        } else {
-               fdmnt = open_file_or_dir(path, dirstream);
+               ret = btrfs_open_dir(path, dirstream, 1);
        }
 
-       return fdmnt;
+       return ret;
+}
+
+/*
+ * Do the following checks before calling open_file_or_dir():
+ * 1: path is in a btrfs filesystem
+ * 2: path is a directory
+ */
+int btrfs_open_dir(const char *path, DIR **dirstream, int verbose)
+{
+       struct statfs stfs;
+       struct stat st;
+       int ret;
+
+       if (statfs(path, &stfs) != 0) {
+               if (verbose)
+                       fprintf(stderr,
+                               "ERROR: can't access '%s': %s\n",
+                               path, strerror(errno));
+               return -1;
+       }
+
+       if (stfs.f_type != BTRFS_SUPER_MAGIC) {
+               if (verbose)
+                       fprintf(stderr,
+                               "ERROR: not a btrfs filesystem: %s\n",
+                               path);
+               return -2;
+       }
+
+       if (stat(path, &st) != 0) {
+               if (verbose)
+                       fprintf(stderr,
+                               "ERROR: can't access '%s': %s\n",
+                               path, strerror(errno));
+               return -1;
+       }
+
+       if (!S_ISDIR(st.st_mode)) {
+               if (verbose)
+                       fprintf(stderr,
+                               "ERROR: not a directory: %s\n",
+                               path);
+               return -3;
+       }
+
+       ret = open_file_or_dir(path, dirstream);
+       if (ret < 0) {
+               if (verbose)
+                       fprintf(stderr,
+                               "ERROR: can't access '%s': %s\n",
+                               path, strerror(errno));
+       }
+
+       return ret;
 }
 
 /* checks if a device is a loop device */
@@ -1091,6 +1172,34 @@ static int is_loop_device (const char* device) {
                MAJOR(statbuf.st_rdev) == LOOP_MAJOR);
 }
 
+/*
+ * Takes a loop device path (e.g. /dev/loop0) and returns
+ * the associated file (e.g. /images/my_btrfs.img) using
+ * loopdev API
+ */
+static int resolve_loop_device_with_loopdev(const char* loop_dev, char* loop_file)
+{
+       int fd;
+       int ret;
+       struct loop_info64 lo64;
+
+       fd = open(loop_dev, O_RDONLY | O_NONBLOCK);
+       if (fd < 0)
+               return -errno;
+       ret = ioctl(fd, LOOP_GET_STATUS64, &lo64);
+       if (ret < 0) {
+               ret = -errno;
+               goto out;
+       }
+
+       memcpy(loop_file, lo64.lo_file_name, sizeof(lo64.lo_file_name));
+       loop_file[sizeof(lo64.lo_file_name)] = 0;
+
+out:
+       close(fd);
+
+       return ret;
+}
 
 /* Takes a loop device path (e.g. /dev/loop0) and returns
  * the associated file (e.g. /images/my_btrfs.img) */
@@ -1106,8 +1215,15 @@ static int resolve_loop_device(const char* loop_dev, char* loop_file,
        if (!realpath(loop_dev, real_loop_dev))
                return -errno;
        snprintf(p, PATH_MAX, "/sys/block/%s/loop/backing_file", strrchr(real_loop_dev, '/'));
-       if (!(f = fopen(p, "r")))
+       if (!(f = fopen(p, "r"))) {
+               if (errno == ENOENT)
+                       /*
+                        * It's possibly a partitioned loop device, which is
+                        * resolvable with loopdev API.
+                        */
+                       return resolve_loop_device_with_loopdev(loop_dev, loop_file);
                return -errno;
+       }
 
        snprintf(fmt, 20, "%%%i[^\n]", max_len-1);
        ret = fscanf(f, fmt, loop_file);
@@ -1409,8 +1525,8 @@ int btrfs_register_one_device(const char *fname)
                        strerror(errno));
                return -errno;
        }
-       strncpy(args.name, fname, BTRFS_PATH_NAME_MAX);
-       args.name[BTRFS_PATH_NAME_MAX-1] = 0;
+       memset(&args, 0, sizeof(args));
+       strncpy_null(args.name, fname);
        ret = ioctl(fd, BTRFS_IOC_SCAN_DEV, &args);
        e = errno;
        if (ret < 0) {
@@ -1437,7 +1553,7 @@ int btrfs_register_all_devices(void)
 
        list_for_each_entry(fs_devices, all_uuids, list) {
                list_for_each_entry(device, &fs_devices->devices, dev_list) {
-                       if (strlen(device->name) != 0) {
+                       if (*device->name) {
                                err = btrfs_register_one_device(device->name);
                                if (err < 0)
                                        return err;
@@ -1663,7 +1779,7 @@ static int set_label_mounted(const char *mount_path, const char *label)
        return 0;
 }
 
-static int get_label_unmounted(const char *dev, char *label)
+int get_label_unmounted(const char *dev, char *label)
 {
        struct btrfs_root *root;
        int ret;
@@ -1673,11 +1789,6 @@ static int get_label_unmounted(const char *dev, char *label)
               fprintf(stderr, "FATAL: error checking %s mount status\n", dev);
               return -1;
        }
-       if (ret > 0) {
-               fprintf(stderr, "ERROR: dev %s is mounted, use mount point\n",
-                       dev);
-               return -1;
-       }
 
        /* Open the super_block at the default location
         * and as read-only.
@@ -1702,6 +1813,7 @@ int get_label_mounted(const char *mount_path, char *labelp)
 {
        char label[BTRFS_LABEL_SIZE];
        int fd;
+       int ret;
 
        fd = open(mount_path, O_RDONLY | O_NOATIME);
        if (fd < 0) {
@@ -1710,10 +1822,14 @@ int get_label_mounted(const char *mount_path, char *labelp)
        }
 
        memset(label, '\0', sizeof(label));
-       if (ioctl(fd, BTRFS_IOC_GET_FSLABEL, label) < 0) {
-               fprintf(stderr, "ERROR: unable get label %s\n", strerror(errno));
+       ret = ioctl(fd, BTRFS_IOC_GET_FSLABEL, label);
+       if (ret < 0) {
+               if (errno != ENOTTY)
+                       fprintf(stderr, "ERROR: unable to get label %s\n",
+                                       strerror(errno));
+               ret = -errno;
                close(fd);
-               return -1;
+               return ret;
        }
 
        strncpy(labelp, label, sizeof(label));
@@ -2056,12 +2172,12 @@ int get_fs_info(char *path, struct btrfs_ioctl_fs_info_args *fi_args,
        struct btrfs_fs_devices *fs_devices_mnt = NULL;
        struct btrfs_ioctl_dev_info_args *di_args;
        struct btrfs_ioctl_dev_info_args tmp;
-       char mp[BTRFS_PATH_NAME_MAX + 1];
+       char mp[PATH_MAX];
        DIR *dirstream = NULL;
 
        memset(fi_args, 0, sizeof(*fi_args));
 
-       if (is_block_device(path)) {
+       if (is_block_device(path) == 1) {
                struct btrfs_super_block *disk_super;
                char buf[BTRFS_SUPER_INFO_SIZE];
                u64 devid;
@@ -2347,7 +2463,7 @@ static int group_profile_devs_min(u64 flag)
 }
 
 int test_num_disk_vs_raid(u64 metadata_profile, u64 data_profile,
-       u64 dev_cnt, int mixed)
+       u64 dev_cnt, int mixed, int ssd)
 {
        u64 allowed = 0;
 
@@ -2388,11 +2504,9 @@ int test_num_disk_vs_raid(u64 metadata_profile, u64 data_profile,
                return 1;
        }
 
-       if (!mixed && (data_profile & BTRFS_BLOCK_GROUP_DUP)) {
-               fprintf(stderr,
-                       "ERROR: DUP for data is allowed only in mixed mode");
-               return 1;
-       }
+       warning_on(!mixed && (data_profile & BTRFS_BLOCK_GROUP_DUP) && ssd,
+                  "DUP may not actually lead to 2 copies on the device, see manual page");
+
        return 0;
 }
 
@@ -2473,7 +2587,7 @@ int test_dev_for_mkfs(char *file, int force_overwrite)
        return 0;
 }
 
-int btrfs_scan_lblkid()
+int btrfs_scan_lblkid(void)
 {
        int fd = -1;
        int ret;
@@ -2487,7 +2601,7 @@ int btrfs_scan_lblkid()
        if (btrfs_scan_done)
                return 0;
 
-       if (blkid_get_cache(&cache, 0) < 0) {
+       if (blkid_get_cache(&cache, NULL) < 0) {
                printf("ERROR: lblkid cache get failed\n");
                return 1;
        }
@@ -2851,7 +2965,7 @@ int btrfs_tree_search2_ioctl_supported(int fd)
        return v2_supported;
 }
 
-int btrfs_check_nodesize(u32 nodesize, u32 sectorsize)
+int btrfs_check_nodesize(u32 nodesize, u32 sectorsize, u64 features)
 {
        if (nodesize < sectorsize) {
                fprintf(stderr,
@@ -2868,6 +2982,144 @@ int btrfs_check_nodesize(u32 nodesize, u32 sectorsize)
                        "ERROR: Illegal nodesize %u (not aligned to %u)\n",
                        nodesize, sectorsize);
                return -1;
+       } else if (features & BTRFS_FEATURE_INCOMPAT_MIXED_GROUPS &&
+                  nodesize != sectorsize) {
+               fprintf(stderr,
+                       "ERROR: Illegal nodesize %u (not equal to %u for mixed block group)\n",
+                       nodesize, sectorsize);
+               return -1;
        }
        return 0;
 }
+
+/*
+ * Copy a path argument from SRC to DEST and check the SRC length if it's at
+ * most PATH_MAX and fits into DEST. DESTLEN is supposed to be exact size of
+ * the buffer.
+ * The destination buffer is zero terminated.
+ * Return < 0 for error, 0 otherwise.
+ */
+int arg_copy_path(char *dest, const char *src, int destlen)
+{
+       size_t len = strlen(src);
+
+       if (len >= PATH_MAX || len >= destlen)
+               return -ENAMETOOLONG;
+
+       __strncpy__null(dest, src, destlen);
+
+       return 0;
+}
+
+unsigned int get_unit_mode_from_arg(int *argc, char *argv[], int df_mode)
+{
+       unsigned int unit_mode = UNITS_DEFAULT;
+       int arg_i;
+       int arg_end;
+
+       for (arg_i = 0; arg_i < *argc; arg_i++) {
+               if (!strcmp(argv[arg_i], "--raw")) {
+                       unit_mode = UNITS_RAW;
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "--human-readable")) {
+                       unit_mode = UNITS_HUMAN_BINARY;
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+
+               if (!strcmp(argv[arg_i], "--iec")) {
+                       units_set_mode(&unit_mode, UNITS_BINARY);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "--si")) {
+                       units_set_mode(&unit_mode, UNITS_DECIMAL);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+
+               if (!strcmp(argv[arg_i], "--kbytes")) {
+                       units_set_base(&unit_mode, UNITS_KBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "--mbytes")) {
+                       units_set_base(&unit_mode, UNITS_MBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "--gbytes")) {
+                       units_set_base(&unit_mode, UNITS_GBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "--tbytes")) {
+                       units_set_base(&unit_mode, UNITS_TBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+
+               if (!df_mode)
+                       continue;
+
+               if (!strcmp(argv[arg_i], "-b")) {
+                       unit_mode = UNITS_RAW;
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "-h")) {
+                       unit_mode = UNITS_HUMAN_BINARY;
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "-H")) {
+                       unit_mode = UNITS_HUMAN_DECIMAL;
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "-k")) {
+                       units_set_base(&unit_mode, UNITS_KBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "-m")) {
+                       units_set_base(&unit_mode, UNITS_MBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "-g")) {
+                       units_set_base(&unit_mode, UNITS_GBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+               if (!strcmp(argv[arg_i], "-t")) {
+                       units_set_base(&unit_mode, UNITS_TBYTES);
+                       argv[arg_i] = NULL;
+                       continue;
+               }
+       }
+
+       for (arg_i = 0, arg_end = 0; arg_i < *argc; arg_i++) {
+               if (!argv[arg_i])
+                       continue;
+               argv[arg_end] = argv[arg_i];
+               arg_end++;
+       }
+
+       *argc = arg_end;
+
+       return unit_mode;
+}
+
+int string_is_numerical(const char *str)
+{
+       if (!(*str >= '0' && *str <= '9'))
+               return 0;
+       while (*str >= '0' && *str <= '9')
+               str++;
+       if (*str != '\0')
+               return 0;
+       return 1;
+}