convert/source-ext2.c

   1 /*
   2  * This program is free software; you can redistribute it and/or
   3  * modify it under the terms of the GNU General Public
   4  * License v2 as published by the Free Software Foundation.
   5  *
   6  * This program is distributed in the hope that it will be useful,
   7  * but WITHOUT ANY WARRANTY; without even the implied warranty of
   8  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   9  * General Public License for more details.
  10  *
  11  * You should have received a copy of the GNU General Public
  12  * License along with this program; if not, write to the
  13  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  14  * Boston, MA 021110-1307, USA.
  15  */
  16
  17 #if BTRFSCONVERT_EXT2
  18
  19 #include "kerncompat.h"
  20 #include <linux/limits.h>
  21 #include <pthread.h>
  22 #include "disk-io.h"
  23 #include "transaction.h"
  24 #include "utils.h"
  25 #include "convert/common.h"
  26 #include "convert/source-ext2.h"
  27
  28 /*
  29  * Open Ext2fs in readonly mode, read block allocation bitmap and
  30  * inode bitmap into memory.
  31  */
  32 static int ext2_open_fs(struct btrfs_convert_context *cctx, const char *name)
  33 {
  34         errcode_t ret;
  35         ext2_filsys ext2_fs;
  36         ext2_ino_t ino;
  37         u32 ro_feature;
  38         int open_flag = EXT2_FLAG_SOFTSUPP_FEATURES | EXT2_FLAG_64BITS;
  39
  40         ret = ext2fs_open(name, open_flag, 0, 0, unix_io_manager, &ext2_fs);
  41         if (ret) {
  42                 if (ret != EXT2_ET_BAD_MAGIC)
  43                         fprintf(stderr, "ext2fs_open: %s\n", error_message(ret));
  44                 return -1;
  45         }
  46         /*
  47          * We need to know exactly the used space, some RO compat flags like
  48          * BIGALLOC will affect how used space is present.
  49          * So we need manuall check any unsupported RO compat flags
  50          */
  51         ro_feature = ext2_fs->super->s_feature_ro_compat;
  52         if (ro_feature & ~EXT2_LIB_FEATURE_RO_COMPAT_SUPP) {
  53                 error(
  54 "unsupported RO features detected: %x, abort convert to avoid possible corruption",
  55                       ro_feature & ~EXT2_LIB_FEATURE_COMPAT_SUPP);
  56                 goto fail;
  57         }
  58         ret = ext2fs_read_inode_bitmap(ext2_fs);
  59         if (ret) {
  60                 fprintf(stderr, "ext2fs_read_inode_bitmap: %s\n",
  61                         error_message(ret));
  62                 goto fail;
  63         }
  64         ret = ext2fs_read_block_bitmap(ext2_fs);
  65         if (ret) {
  66                 fprintf(stderr, "ext2fs_read_block_bitmap: %s\n",
  67                         error_message(ret));
  68                 goto fail;
  69         }
  70         /*
  71          * search each block group for a free inode. this set up
  72          * uninit block/inode bitmaps appropriately.
  73          */
  74         ino = 1;
  75         while (ino <= ext2_fs->super->s_inodes_count) {
  76                 ext2_ino_t foo;
  77                 ext2fs_new_inode(ext2_fs, ino, 0, NULL, &foo);
  78                 ino += EXT2_INODES_PER_GROUP(ext2_fs->super);
  79         }
  80
  81         if (!(ext2_fs->super->s_feature_incompat &
  82               EXT2_FEATURE_INCOMPAT_FILETYPE)) {
  83                 error("filetype feature is missing");
  84                 goto fail;
  85         }
  86
  87         cctx->fs_data = ext2_fs;
  88         cctx->blocksize = ext2_fs->blocksize;
  89         cctx->block_count = ext2_fs->super->s_blocks_count;
  90         cctx->total_bytes = ext2_fs->blocksize * ext2_fs->super->s_blocks_count;
  91         cctx->volume_name = strndup(ext2_fs->super->s_volume_name, 16);
  92         cctx->first_data_block = ext2_fs->super->s_first_data_block;
  93         cctx->inodes_count = ext2_fs->super->s_inodes_count;
  94         cctx->free_inodes_count = ext2_fs->super->s_free_inodes_count;
  95         return 0;
  96 fail:
  97         ext2fs_close(ext2_fs);
  98         return -1;
  99 }
 100
 101 static int __ext2_add_one_block(ext2_filsys fs, char *bitmap,
 102                                 unsigned long group_nr, struct cache_tree *used)
 103 {
 104         unsigned long offset;
 105         unsigned i;
 106         int ret = 0;
 107
 108         offset = fs->super->s_first_data_block;
 109         offset /= EXT2FS_CLUSTER_RATIO(fs);
 110         offset += group_nr * EXT2_CLUSTERS_PER_GROUP(fs->super);
 111         for (i = 0; i < EXT2_CLUSTERS_PER_GROUP(fs->super); i++) {
 112                 if ((i + offset) >= ext2fs_blocks_count(fs->super))
 113                         break;
 114
 115                 if (ext2fs_test_bit(i, bitmap)) {
 116                         u64 start;
 117
 118                         start = (i + offset) * EXT2FS_CLUSTER_RATIO(fs);
 119                         start *= fs->blocksize;
 120                         ret = add_merge_cache_extent(used, start,
 121                                                      fs->blocksize);
 122                         if (ret < 0)
 123                                 break;
 124                 }
 125         }
 126         return ret;
 127 }
 128
 129 /*
 130  * Read all used ext2 space into cctx->used cache tree
 131  */
 132 static int ext2_read_used_space(struct btrfs_convert_context *cctx)
 133 {
 134         ext2_filsys fs = (ext2_filsys)cctx->fs_data;
 135         blk64_t blk_itr = EXT2FS_B2C(fs, fs->super->s_first_data_block);
 136         struct cache_tree *used_tree = &cctx->used_space;
 137         char *block_bitmap = NULL;
 138         unsigned long i;
 139         int block_nbytes;
 140         int ret = 0;
 141
 142         block_nbytes = EXT2_CLUSTERS_PER_GROUP(fs->super) / 8;
 143         if (!block_nbytes) {
 144                 error("EXT2_CLUSTERS_PER_GROUP too small: %llu",
 145                         (unsigned long long)(EXT2_CLUSTERS_PER_GROUP(fs->super)));
 146                 return -EINVAL;
 147         }
 148
 149         block_bitmap = malloc(block_nbytes);
 150         if (!block_bitmap)
 151                 return -ENOMEM;
 152
 153         for (i = 0; i < fs->group_desc_count; i++) {
 154                 ret = ext2fs_get_block_bitmap_range2(fs->block_map, blk_itr,
 155                                                 block_nbytes * 8, block_bitmap);
 156                 if (ret) {
 157                         error("fail to get bitmap from ext2, %s",
 158                                 error_message(ret));
 159                         ret = -EINVAL;
 160                         break;
 161                 }
 162                 ret = __ext2_add_one_block(fs, block_bitmap, i, used_tree);
 163                 if (ret < 0) {
 164                         error("fail to build used space tree, %s",
 165                               strerror(-ret));
 166                         break;
 167                 }
 168                 blk_itr += EXT2_CLUSTERS_PER_GROUP(fs->super);
 169         }
 170
 171         free(block_bitmap);
 172         return ret;
 173 }
 174
 175 static void ext2_close_fs(struct btrfs_convert_context *cctx)
 176 {
 177         if (cctx->volume_name) {
 178                 free(cctx->volume_name);
 179                 cctx->volume_name = NULL;
 180         }
 181         ext2fs_close(cctx->fs_data);
 182 }
 183
 184 static u8 ext2_filetype_conversion_table[EXT2_FT_MAX] = {
 185         [EXT2_FT_UNKNOWN]       = BTRFS_FT_UNKNOWN,
 186         [EXT2_FT_REG_FILE]      = BTRFS_FT_REG_FILE,
 187         [EXT2_FT_DIR]           = BTRFS_FT_DIR,
 188         [EXT2_FT_CHRDEV]        = BTRFS_FT_CHRDEV,
 189         [EXT2_FT_BLKDEV]        = BTRFS_FT_BLKDEV,
 190         [EXT2_FT_FIFO]          = BTRFS_FT_FIFO,
 191         [EXT2_FT_SOCK]          = BTRFS_FT_SOCK,
 192         [EXT2_FT_SYMLINK]       = BTRFS_FT_SYMLINK,
 193 };
 194
 195 static int ext2_dir_iterate_proc(ext2_ino_t dir, int entry,
 196                             struct ext2_dir_entry *dirent,
 197                             int offset, int blocksize,
 198                             char *buf,void *priv_data)
 199 {
 200         int ret;
 201         int file_type;
 202         u64 objectid;
 203         char dotdot[] = "..";
 204         struct dir_iterate_data *idata = (struct dir_iterate_data *)priv_data;
 205         int name_len;
 206
 207         name_len = dirent->name_len & 0xFF;
 208
 209         objectid = dirent->inode + INO_OFFSET;
 210         if (!strncmp(dirent->name, dotdot, name_len)) {
 211                 if (name_len == 2) {
 212                         BUG_ON(idata->parent != 0);
 213                         idata->parent = objectid;
 214                 }
 215                 return 0;
 216         }
 217         if (dirent->inode < EXT2_GOOD_OLD_FIRST_INO)
 218                 return 0;
 219
 220         file_type = dirent->name_len >> 8;
 221         BUG_ON(file_type > EXT2_FT_SYMLINK);
 222
 223         ret = convert_insert_dirent(idata->trans, idata->root, dirent->name,
 224                                     name_len, idata->objectid, objectid,
 225                                     ext2_filetype_conversion_table[file_type],
 226                                     idata->index_cnt, idata->inode);
 227         if (ret < 0) {
 228                 idata->errcode = ret;
 229                 return BLOCK_ABORT;
 230         }
 231
 232         idata->index_cnt++;
 233         return 0;
 234 }
 235
 236 static int ext2_create_dir_entries(struct btrfs_trans_handle *trans,
 237                               struct btrfs_root *root, u64 objectid,
 238                               struct btrfs_inode_item *btrfs_inode,
 239                               ext2_filsys ext2_fs, ext2_ino_t ext2_ino)
 240 {
 241         int ret;
 242         errcode_t err;
 243         struct dir_iterate_data data = {
 244                 .trans          = trans,
 245                 .root           = root,
 246                 .inode          = btrfs_inode,
 247                 .objectid       = objectid,
 248                 .index_cnt      = 2,
 249                 .parent         = 0,
 250                 .errcode        = 0,
 251         };
 252
 253         err = ext2fs_dir_iterate2(ext2_fs, ext2_ino, 0, NULL,
 254                                   ext2_dir_iterate_proc, &data);
 255         if (err)
 256                 goto error;
 257         ret = data.errcode;
 258         if (ret == 0 && data.parent == objectid) {
 259                 ret = btrfs_insert_inode_ref(trans, root, "..", 2,
 260                                              objectid, objectid, 0);
 261         }
 262         return ret;
 263 error:
 264         fprintf(stderr, "ext2fs_dir_iterate2: %s\n", error_message(err));
 265         return -1;
 266 }
 267
 268 static int ext2_block_iterate_proc(ext2_filsys fs, blk_t *blocknr,
 269                                 e2_blkcnt_t blockcnt, blk_t ref_block,
 270                                 int ref_offset, void *priv_data)
 271 {
 272         int ret;
 273         struct blk_iterate_data *idata;
 274         idata = (struct blk_iterate_data *)priv_data;
 275         ret = block_iterate_proc(*blocknr, blockcnt, idata);
 276         if (ret) {
 277                 idata->errcode = ret;
 278                 return BLOCK_ABORT;
 279         }
 280         return 0;
 281 }
 282
 283 /*
 284  * traverse file's data blocks, record these data blocks as file extents.
 285  */
 286 static int ext2_create_file_extents(struct btrfs_trans_handle *trans,
 287                                struct btrfs_root *root, u64 objectid,
 288                                struct btrfs_inode_item *btrfs_inode,
 289                                ext2_filsys ext2_fs, ext2_ino_t ext2_ino,
 290                                u32 convert_flags)
 291 {
 292         int ret;
 293         char *buffer = NULL;
 294         errcode_t err;
 295         u32 last_block;
 296         u32 sectorsize = root->fs_info->sectorsize;
 297         u64 inode_size = btrfs_stack_inode_size(btrfs_inode);
 298         struct blk_iterate_data data;
 299
 300         init_blk_iterate_data(&data, trans, root, btrfs_inode, objectid,
 301                         convert_flags & CONVERT_FLAG_DATACSUM);
 302
 303         err = ext2fs_block_iterate2(ext2_fs, ext2_ino, BLOCK_FLAG_DATA_ONLY,
 304                                     NULL, ext2_block_iterate_proc, &data);
 305         if (err)
 306                 goto error;
 307         ret = data.errcode;
 308         if (ret)
 309                 goto fail;
 310         if ((convert_flags & CONVERT_FLAG_INLINE_DATA) && data.first_block == 0
 311             && data.num_blocks > 0
 312             && inode_size <= BTRFS_MAX_INLINE_DATA_SIZE(root)) {
 313                 u64 num_bytes = data.num_blocks * sectorsize;
 314                 u64 disk_bytenr = data.disk_block * sectorsize;
 315                 u64 nbytes;
 316
 317                 buffer = malloc(num_bytes);
 318                 if (!buffer)
 319                         return -ENOMEM;
 320                 ret = read_disk_extent(root, disk_bytenr, num_bytes, buffer);
 321                 if (ret)
 322                         goto fail;
 323                 if (num_bytes > inode_size)
 324                         num_bytes = inode_size;
 325                 ret = btrfs_insert_inline_extent(trans, root, objectid,
 326                                                  0, buffer, num_bytes);
 327                 if (ret)
 328                         goto fail;
 329                 nbytes = btrfs_stack_inode_nbytes(btrfs_inode) + num_bytes;
 330                 btrfs_set_stack_inode_nbytes(btrfs_inode, nbytes);
 331         } else if (data.num_blocks > 0) {
 332                 ret = record_file_blocks(&data, data.first_block,
 333                                          data.disk_block, data.num_blocks);
 334                 if (ret)
 335                         goto fail;
 336         }
 337         data.first_block += data.num_blocks;
 338         last_block = (inode_size + sectorsize - 1) / sectorsize;
 339         if (last_block > data.first_block) {
 340                 ret = record_file_blocks(&data, data.first_block, 0,
 341                                          last_block - data.first_block);
 342         }
 343 fail:
 344         free(buffer);
 345         return ret;
 346 error:
 347         fprintf(stderr, "ext2fs_block_iterate2: %s\n", error_message(err));
 348         return -1;
 349 }
 350
 351 static int ext2_create_symlink(struct btrfs_trans_handle *trans,
 352                               struct btrfs_root *root, u64 objectid,
 353                               struct btrfs_inode_item *btrfs_inode,
 354                               ext2_filsys ext2_fs, ext2_ino_t ext2_ino,
 355                               struct ext2_inode *ext2_inode)
 356 {
 357         int ret;
 358         char *pathname;
 359         u64 inode_size = btrfs_stack_inode_size(btrfs_inode);
 360         if (ext2fs_inode_data_blocks2(ext2_fs, ext2_inode)) {
 361                 btrfs_set_stack_inode_size(btrfs_inode, inode_size + 1);
 362                 ret = ext2_create_file_extents(trans, root, objectid,
 363                                 btrfs_inode, ext2_fs, ext2_ino,
 364                                 CONVERT_FLAG_DATACSUM |
 365                                 CONVERT_FLAG_INLINE_DATA);
 366                 btrfs_set_stack_inode_size(btrfs_inode, inode_size);
 367                 return ret;
 368         }
 369
 370         pathname = (char *)&(ext2_inode->i_block[0]);
 371         BUG_ON(pathname[inode_size] != 0);
 372         ret = btrfs_insert_inline_extent(trans, root, objectid, 0,
 373                                          pathname, inode_size + 1);
 374         btrfs_set_stack_inode_nbytes(btrfs_inode, inode_size + 1);
 375         return ret;
 376 }
 377
 378 /*
 379  * Following xattr/acl related codes are based on codes in
 380  * fs/ext3/xattr.c and fs/ext3/acl.c
 381  */
 382 #define EXT2_XATTR_BHDR(ptr) ((struct ext2_ext_attr_header *)(ptr))
 383 #define EXT2_XATTR_BFIRST(ptr) \
 384         ((struct ext2_ext_attr_entry *)(EXT2_XATTR_BHDR(ptr) + 1))
 385 #define EXT2_XATTR_IHDR(inode) \
 386         ((struct ext2_ext_attr_header *) ((void *)(inode) + \
 387                 EXT2_GOOD_OLD_INODE_SIZE + (inode)->i_extra_isize))
 388 #define EXT2_XATTR_IFIRST(inode) \
 389         ((struct ext2_ext_attr_entry *) ((void *)EXT2_XATTR_IHDR(inode) + \
 390                 sizeof(EXT2_XATTR_IHDR(inode)->h_magic)))
 391
 392 static int ext2_xattr_check_names(struct ext2_ext_attr_entry *entry,
 393                                   const void *end)
 394 {
 395         struct ext2_ext_attr_entry *next;
 396
 397         while (!EXT2_EXT_IS_LAST_ENTRY(entry)) {
 398                 next = EXT2_EXT_ATTR_NEXT(entry);
 399                 if ((void *)next >= end)
 400                         return -EIO;
 401                 entry = next;
 402         }
 403         return 0;
 404 }
 405
 406 static int ext2_xattr_check_block(const char *buf, size_t size)
 407 {
 408         int error;
 409         struct ext2_ext_attr_header *header = EXT2_XATTR_BHDR(buf);
 410
 411         if (header->h_magic != EXT2_EXT_ATTR_MAGIC ||
 412             header->h_blocks != 1)
 413                 return -EIO;
 414         error = ext2_xattr_check_names(EXT2_XATTR_BFIRST(buf), buf + size);
 415         return error;
 416 }
 417
 418 static int ext2_xattr_check_entry(struct ext2_ext_attr_entry *entry,
 419                                   size_t size)
 420 {
 421         size_t value_size = entry->e_value_size;
 422
 423         if (entry->e_value_block != 0 || value_size > size ||
 424             entry->e_value_offs + value_size > size)
 425                 return -EIO;
 426         return 0;
 427 }
 428
 429 static int ext2_acl_to_xattr(void *dst, const void *src,
 430                              size_t dst_size, size_t src_size)
 431 {
 432         int i, count;
 433         const void *end = src + src_size;
 434         acl_ea_header *ext_acl = (acl_ea_header *)dst;
 435         acl_ea_entry *dst_entry = ext_acl->a_entries;
 436         ext2_acl_entry *src_entry;
 437
 438         if (src_size < sizeof(ext2_acl_header))
 439                 goto fail;
 440         if (((ext2_acl_header *)src)->a_version !=
 441             cpu_to_le32(EXT2_ACL_VERSION))
 442                 goto fail;
 443         src += sizeof(ext2_acl_header);
 444         count = ext2_acl_count(src_size);
 445         if (count <= 0)
 446                 goto fail;
 447
 448         BUG_ON(dst_size < acl_ea_size(count));
 449         ext_acl->a_version = cpu_to_le32(ACL_EA_VERSION);
 450         for (i = 0; i < count; i++, dst_entry++) {
 451                 src_entry = (ext2_acl_entry *)src;
 452                 if (src + sizeof(ext2_acl_entry_short) > end)
 453                         goto fail;
 454                 dst_entry->e_tag = src_entry->e_tag;
 455                 dst_entry->e_perm = src_entry->e_perm;
 456                 switch (le16_to_cpu(src_entry->e_tag)) {
 457                 case ACL_USER_OBJ:
 458                 case ACL_GROUP_OBJ:
 459                 case ACL_MASK:
 460                 case ACL_OTHER:
 461                         src += sizeof(ext2_acl_entry_short);
 462                         dst_entry->e_id = cpu_to_le32(ACL_UNDEFINED_ID);
 463                         break;
 464                 case ACL_USER:
 465                 case ACL_GROUP:
 466                         src += sizeof(ext2_acl_entry);
 467                         if (src > end)
 468                                 goto fail;
 469                         dst_entry->e_id = src_entry->e_id;
 470                         break;
 471                 default:
 472                         goto fail;
 473                 }
 474         }
 475         if (src != end)
 476                 goto fail;
 477         return 0;
 478 fail:
 479         return -EINVAL;
 480 }
 481
 482 static char *xattr_prefix_table[] = {
 483         [1] =   "user.",
 484         [2] =   "system.posix_acl_access",
 485         [3] =   "system.posix_acl_default",
 486         [4] =   "trusted.",
 487         [6] =   "security.",
 488 };
 489
 490 static int ext2_copy_single_xattr(struct btrfs_trans_handle *trans,
 491                              struct btrfs_root *root, u64 objectid,
 492                              struct ext2_ext_attr_entry *entry,
 493                              const void *data, u32 datalen)
 494 {
 495         int ret = 0;
 496         int name_len;
 497         int name_index;
 498         void *databuf = NULL;
 499         char namebuf[XATTR_NAME_MAX + 1];
 500
 501         name_index = entry->e_name_index;
 502         if (name_index >= ARRAY_SIZE(xattr_prefix_table) ||
 503             xattr_prefix_table[name_index] == NULL)
 504                 return -EOPNOTSUPP;
 505         name_len = strlen(xattr_prefix_table[name_index]) +
 506                    entry->e_name_len;
 507         if (name_len >= sizeof(namebuf))
 508                 return -ERANGE;
 509
 510         if (name_index == 2 || name_index == 3) {
 511                 size_t bufsize = acl_ea_size(ext2_acl_count(datalen));
 512                 databuf = malloc(bufsize);
 513                 if (!databuf)
 514                        return -ENOMEM;
 515                 ret = ext2_acl_to_xattr(databuf, data, bufsize, datalen);
 516                 if (ret)
 517                         goto out;
 518                 data = databuf;
 519                 datalen = bufsize;
 520         }
 521         strncpy(namebuf, xattr_prefix_table[name_index], XATTR_NAME_MAX);
 522         strncat(namebuf, EXT2_EXT_ATTR_NAME(entry), entry->e_name_len);
 523         if (name_len + datalen > BTRFS_LEAF_DATA_SIZE(root->fs_info) -
 524             sizeof(struct btrfs_item) - sizeof(struct btrfs_dir_item)) {
 525                 fprintf(stderr, "skip large xattr on inode %Lu name %.*s\n",
 526                         objectid - INO_OFFSET, name_len, namebuf);
 527                 goto out;
 528         }
 529         ret = btrfs_insert_xattr_item(trans, root, namebuf, name_len,
 530                                       data, datalen, objectid);
 531 out:
 532         free(databuf);
 533         return ret;
 534 }
 535
 536 static int ext2_copy_extended_attrs(struct btrfs_trans_handle *trans,
 537                                struct btrfs_root *root, u64 objectid,
 538                                struct btrfs_inode_item *btrfs_inode,
 539                                ext2_filsys ext2_fs, ext2_ino_t ext2_ino)
 540 {
 541         int ret = 0;
 542         int inline_ea = 0;
 543         errcode_t err;
 544         u32 datalen;
 545         u32 block_size = ext2_fs->blocksize;
 546         u32 inode_size = EXT2_INODE_SIZE(ext2_fs->super);
 547         struct ext2_inode_large *ext2_inode;
 548         struct ext2_ext_attr_entry *entry;
 549         void *data;
 550         char *buffer = NULL;
 551         char inode_buf[EXT2_GOOD_OLD_INODE_SIZE];
 552
 553         if (inode_size <= EXT2_GOOD_OLD_INODE_SIZE) {
 554                 ext2_inode = (struct ext2_inode_large *)inode_buf;
 555         } else {
 556                 ext2_inode = (struct ext2_inode_large *)malloc(inode_size);
 557                 if (!ext2_inode)
 558                        return -ENOMEM;
 559         }
 560         err = ext2fs_read_inode_full(ext2_fs, ext2_ino, (void *)ext2_inode,
 561                                      inode_size);
 562         if (err) {
 563                 fprintf(stderr, "ext2fs_read_inode_full: %s\n",
 564                         error_message(err));
 565                 ret = -1;
 566                 goto out;
 567         }
 568
 569         if (ext2_ino > ext2_fs->super->s_first_ino &&
 570             inode_size > EXT2_GOOD_OLD_INODE_SIZE) {
 571                 if (EXT2_GOOD_OLD_INODE_SIZE +
 572                     ext2_inode->i_extra_isize > inode_size) {
 573                         ret = -EIO;
 574                         goto out;
 575                 }
 576                 if (ext2_inode->i_extra_isize != 0 &&
 577                     EXT2_XATTR_IHDR(ext2_inode)->h_magic ==
 578                     EXT2_EXT_ATTR_MAGIC) {
 579                         inline_ea = 1;
 580                 }
 581         }
 582         if (inline_ea) {
 583                 int total;
 584                 void *end = (void *)ext2_inode + inode_size;
 585                 entry = EXT2_XATTR_IFIRST(ext2_inode);
 586                 total = end - (void *)entry;
 587                 ret = ext2_xattr_check_names(entry, end);
 588                 if (ret)
 589                         goto out;
 590                 while (!EXT2_EXT_IS_LAST_ENTRY(entry)) {
 591                         ret = ext2_xattr_check_entry(entry, total);
 592                         if (ret)
 593                                 goto out;
 594                         data = (void *)EXT2_XATTR_IFIRST(ext2_inode) +
 595                                 entry->e_value_offs;
 596                         datalen = entry->e_value_size;
 597                         ret = ext2_copy_single_xattr(trans, root, objectid,
 598                                                 entry, data, datalen);
 599                         if (ret)
 600                                 goto out;
 601                         entry = EXT2_EXT_ATTR_NEXT(entry);
 602                 }
 603         }
 604
 605         if (ext2_inode->i_file_acl == 0)
 606                 goto out;
 607
 608         buffer = malloc(block_size);
 609         if (!buffer) {
 610                 ret = -ENOMEM;
 611                 goto out;
 612         }
 613         err = ext2fs_read_ext_attr2(ext2_fs, ext2_inode->i_file_acl, buffer);
 614         if (err) {
 615                 fprintf(stderr, "ext2fs_read_ext_attr2: %s\n",
 616                         error_message(err));
 617                 ret = -1;
 618                 goto out;
 619         }
 620         ret = ext2_xattr_check_block(buffer, block_size);
 621         if (ret)
 622                 goto out;
 623
 624         entry = EXT2_XATTR_BFIRST(buffer);
 625         while (!EXT2_EXT_IS_LAST_ENTRY(entry)) {
 626                 ret = ext2_xattr_check_entry(entry, block_size);
 627                 if (ret)
 628                         goto out;
 629                 data = buffer + entry->e_value_offs;
 630                 datalen = entry->e_value_size;
 631                 ret = ext2_copy_single_xattr(trans, root, objectid,
 632                                         entry, data, datalen);
 633                 if (ret)
 634                         goto out;
 635                 entry = EXT2_EXT_ATTR_NEXT(entry);
 636         }
 637 out:
 638         free(buffer);
 639         if ((void *)ext2_inode != inode_buf)
 640                 free(ext2_inode);
 641         return ret;
 642 }
 643
 644 static inline dev_t old_decode_dev(u16 val)
 645 {
 646         return MKDEV((val >> 8) & 255, val & 255);
 647 }
 648
 649 static void ext2_copy_inode_item(struct btrfs_inode_item *dst,
 650                            struct ext2_inode *src, u32 blocksize)
 651 {
 652         btrfs_set_stack_inode_generation(dst, 1);
 653         btrfs_set_stack_inode_sequence(dst, 0);
 654         btrfs_set_stack_inode_transid(dst, 1);
 655         btrfs_set_stack_inode_size(dst, src->i_size);
 656         btrfs_set_stack_inode_nbytes(dst, 0);
 657         btrfs_set_stack_inode_block_group(dst, 0);
 658         btrfs_set_stack_inode_nlink(dst, src->i_links_count);
 659         btrfs_set_stack_inode_uid(dst, src->i_uid | (src->i_uid_high << 16));
 660         btrfs_set_stack_inode_gid(dst, src->i_gid | (src->i_gid_high << 16));
 661         btrfs_set_stack_inode_mode(dst, src->i_mode);
 662         btrfs_set_stack_inode_rdev(dst, 0);
 663         btrfs_set_stack_inode_flags(dst, 0);
 664         btrfs_set_stack_timespec_sec(&dst->atime, src->i_atime);
 665         btrfs_set_stack_timespec_nsec(&dst->atime, 0);
 666         btrfs_set_stack_timespec_sec(&dst->ctime, src->i_ctime);
 667         btrfs_set_stack_timespec_nsec(&dst->ctime, 0);
 668         btrfs_set_stack_timespec_sec(&dst->mtime, src->i_mtime);
 669         btrfs_set_stack_timespec_nsec(&dst->mtime, 0);
 670         btrfs_set_stack_timespec_sec(&dst->otime, 0);
 671         btrfs_set_stack_timespec_nsec(&dst->otime, 0);
 672
 673         if (S_ISDIR(src->i_mode)) {
 674                 btrfs_set_stack_inode_size(dst, 0);
 675                 btrfs_set_stack_inode_nlink(dst, 1);
 676         }
 677         if (S_ISREG(src->i_mode)) {
 678                 btrfs_set_stack_inode_size(dst, (u64)src->i_size_high << 32 |
 679                                            (u64)src->i_size);
 680         }
 681         if (!S_ISREG(src->i_mode) && !S_ISDIR(src->i_mode) &&
 682             !S_ISLNK(src->i_mode)) {
 683                 if (src->i_block[0]) {
 684                         btrfs_set_stack_inode_rdev(dst,
 685                                 old_decode_dev(src->i_block[0]));
 686                 } else {
 687                         btrfs_set_stack_inode_rdev(dst,
 688                                 decode_dev(src->i_block[1]));
 689                 }
 690         }
 691         memset(&dst->reserved, 0, sizeof(dst->reserved));
 692 }
 693 static int ext2_check_state(struct btrfs_convert_context *cctx)
 694 {
 695         ext2_filsys fs = cctx->fs_data;
 696
 697         if (!(fs->super->s_state & EXT2_VALID_FS))
 698                 return 1;
 699         else if (fs->super->s_state & EXT2_ERROR_FS)
 700                 return 1;
 701         else
 702                 return 0;
 703 }
 704
 705 /* EXT2_*_FL to BTRFS_INODE_FLAG_* stringification helper */
 706 #define COPY_ONE_EXT2_FLAG(flags, ext2_inode, name) ({                  \
 707         if (ext2_inode->i_flags & EXT2_##name##_FL)                     \
 708                 flags |= BTRFS_INODE_##name;                            \
 709 })
 710
 711 /*
 712  * Convert EXT2_*_FL to corresponding BTRFS_INODE_* flags
 713  *
 714  * Only a subset of EXT_*_FL is supported in btrfs.
 715  */
 716 static void ext2_convert_inode_flags(struct btrfs_inode_item *dst,
 717                                      struct ext2_inode *src)
 718 {
 719         u64 flags = btrfs_stack_inode_flags(dst);
 720
 721         COPY_ONE_EXT2_FLAG(flags, src, APPEND);
 722         COPY_ONE_EXT2_FLAG(flags, src, SYNC);
 723         COPY_ONE_EXT2_FLAG(flags, src, IMMUTABLE);
 724         COPY_ONE_EXT2_FLAG(flags, src, NODUMP);
 725         COPY_ONE_EXT2_FLAG(flags, src, NOATIME);
 726         COPY_ONE_EXT2_FLAG(flags, src, DIRSYNC);
 727         btrfs_set_stack_inode_flags(dst, flags);
 728 }
 729
 730 /*
 731  * copy a single inode. do all the required works, such as cloning
 732  * inode item, creating file extents and creating directory entries.
 733  */
 734 static int ext2_copy_single_inode(struct btrfs_trans_handle *trans,
 735                              struct btrfs_root *root, u64 objectid,
 736                              ext2_filsys ext2_fs, ext2_ino_t ext2_ino,
 737                              struct ext2_inode *ext2_inode,
 738                              u32 convert_flags)
 739 {
 740         int ret;
 741         struct btrfs_inode_item btrfs_inode;
 742
 743         if (ext2_inode->i_links_count == 0)
 744                 return 0;
 745
 746         ext2_copy_inode_item(&btrfs_inode, ext2_inode, ext2_fs->blocksize);
 747         if (!(convert_flags & CONVERT_FLAG_DATACSUM)
 748             && S_ISREG(ext2_inode->i_mode)) {
 749                 u32 flags = btrfs_stack_inode_flags(&btrfs_inode) |
 750                             BTRFS_INODE_NODATASUM;
 751                 btrfs_set_stack_inode_flags(&btrfs_inode, flags);
 752         }
 753         ext2_convert_inode_flags(&btrfs_inode, ext2_inode);
 754
 755         switch (ext2_inode->i_mode & S_IFMT) {
 756         case S_IFREG:
 757                 ret = ext2_create_file_extents(trans, root, objectid,
 758                         &btrfs_inode, ext2_fs, ext2_ino, convert_flags);
 759                 break;
 760         case S_IFDIR:
 761                 ret = ext2_create_dir_entries(trans, root, objectid,
 762                                 &btrfs_inode, ext2_fs, ext2_ino);
 763                 break;
 764         case S_IFLNK:
 765                 ret = ext2_create_symlink(trans, root, objectid,
 766                                 &btrfs_inode, ext2_fs, ext2_ino, ext2_inode);
 767                 break;
 768         default:
 769                 ret = 0;
 770                 break;
 771         }
 772         if (ret)
 773                 return ret;
 774
 775         if (convert_flags & CONVERT_FLAG_XATTR) {
 776                 ret = ext2_copy_extended_attrs(trans, root, objectid,
 777                                 &btrfs_inode, ext2_fs, ext2_ino);
 778                 if (ret)
 779                         return ret;
 780         }
 781         return btrfs_insert_inode(trans, root, objectid, &btrfs_inode);
 782 }
 783
 784 static int ext2_is_special_inode(ext2_ino_t ino)
 785 {
 786         if (ino < EXT2_GOOD_OLD_FIRST_INO && ino != EXT2_ROOT_INO)
 787                 return 1;
 788         return 0;
 789 }
 790
 791 /*
 792  * scan ext2's inode bitmap and copy all used inodes.
 793  */
 794 static int ext2_copy_inodes(struct btrfs_convert_context *cctx,
 795                             struct btrfs_root *root,
 796                             u32 convert_flags, struct task_ctx *p)
 797 {
 798         ext2_filsys ext2_fs = cctx->fs_data;
 799         int ret;
 800         errcode_t err;
 801         ext2_inode_scan ext2_scan;
 802         struct ext2_inode ext2_inode;
 803         ext2_ino_t ext2_ino;
 804         u64 objectid;
 805         struct btrfs_trans_handle *trans;
 806
 807         trans = btrfs_start_transaction(root, 1);
 808         if (IS_ERR(trans))
 809                 return PTR_ERR(trans);
 810         err = ext2fs_open_inode_scan(ext2_fs, 0, &ext2_scan);
 811         if (err) {
 812                 fprintf(stderr, "ext2fs_open_inode_scan: %s\n", error_message(err));
 813                 return -1;
 814         }
 815         while (!(err = ext2fs_get_next_inode(ext2_scan, &ext2_ino,
 816                                              &ext2_inode))) {
 817                 /* no more inodes */
 818                 if (ext2_ino == 0)
 819                         break;
 820                 if (ext2_is_special_inode(ext2_ino))
 821                         continue;
 822                 objectid = ext2_ino + INO_OFFSET;
 823                 ret = ext2_copy_single_inode(trans, root,
 824                                         objectid, ext2_fs, ext2_ino,
 825                                         &ext2_inode, convert_flags);
 826                 pthread_mutex_lock(&p->mutex);
 827                 p->cur_copy_inodes++;
 828                 pthread_mutex_unlock(&p->mutex);
 829                 if (ret)
 830                         return ret;
 831                 if (trans->blocks_used >= 4096) {
 832                         ret = btrfs_commit_transaction(trans, root);
 833                         BUG_ON(ret);
 834                         trans = btrfs_start_transaction(root, 1);
 835                         BUG_ON(IS_ERR(trans));
 836                 }
 837         }
 838         if (err) {
 839                 fprintf(stderr, "ext2fs_get_next_inode: %s\n", error_message(err));
 840                 return -1;
 841         }
 842         ret = btrfs_commit_transaction(trans, root);
 843         BUG_ON(ret);
 844         ext2fs_close_inode_scan(ext2_scan);
 845
 846         return ret;
 847 }
 848
 849 const struct btrfs_convert_operations ext2_convert_ops = {
 850         .name                   = "ext2",
 851         .open_fs                = ext2_open_fs,
 852         .read_used_space        = ext2_read_used_space,
 853         .copy_inodes            = ext2_copy_inodes,
 854         .close_fs               = ext2_close_fs,
 855         .check_state            = ext2_check_state,
 856 };
 857
 858 #endif  /* BTRFSCONVERT_EXT2 */