?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Drop quick-test from the list of programs built by default
[btrfs-progs-unstable/devel.git] / convert.c
blob01464549e562fd232c722f4eb93021b493fa31a3
1 /*
2 * Copyright (C) 2007 Oracle. All rights reserved.
3 *
4 * This program is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU General Public
6 * License v2 as published by the Free Software Foundation.
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 * General Public License for more details.
12 *
13 * You should have received a copy of the GNU General Public
14 * License along with this program; if not, write to the
15 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
16 * Boston, MA 021110-1307, USA.
17 */
18
19 #define _XOPEN_SOURCE 500
20 #ifndef __CHECKER__
21 #include <sys/ioctl.h>
22 #include <sys/mount.h>
23 #endif
24 #include <stdio.h>
25 #include <stdlib.h>
26 #include <sys/types.h>
27 #include <sys/stat.h>
28 #include <sys/acl.h>
29 #include <fcntl.h>
30 #include <unistd.h>
31 #include <uuid/uuid.h>
32 #include <linux/fs.h>
33 #include "kerncompat.h"
34 #include "ctree.h"
35 #include "disk-io.h"
36 #include "volumes.h"
37 #include "transaction.h"
38 #include "crc32c.h"
39 #include "utils.h"
40 #include <ext2fs/ext2_fs.h>
41 #include <ext2fs/ext2fs.h>
42 #include <ext2fs/ext2_ext_attr.h>
43 #define INO_OFFSET (BTRFS_FIRST_FREE_OBJECTID - EXT2_ROOT_INO)
44 /*
45 * Open Ext2fs in readonly mode, read block allocation bitmap and
46 * inode bitmap into memory.
47 */
48 static int open_ext2fs(const char *name, ext2_filsys *ret_fs)
49 {
50 errcode_t ret;
51 ext2_filsys ext2_fs;
52 ret = ext2fs_open(name, 0, 0, 0, unix_io_manager, &ext2_fs);
53 if (ret) {
54 fprintf(stderr, "ext2fs_open: %s\n", error_message(ret));
55 goto fail;
56 }
57 ret = ext2fs_read_inode_bitmap(ext2_fs);
58 if (ret) {
59 fprintf(stderr, "ext2fs_read_inode_bitmap: %s\n",
60 error_message(ret));
61 goto fail;
62 }
63 ret = ext2fs_read_block_bitmap(ext2_fs);
64 if (ret) {
65 fprintf(stderr, "ext2fs_read_block_bitmap: %s\n",
66 error_message(ret));
67 goto fail;
68 }
69 *ret_fs = ext2_fs;
70 return 0;
71 fail:
72 return -1;
73 }
74
75 static int close_ext2fs(ext2_filsys fs)
76 {
77 ext2fs_close(fs);
78 return 0;
79 }
80
81 static int ext2_alloc_block(ext2_filsys fs, u64 goal, u64 *block_ret)
82 {
83 blk_t block;
84
85 if (!ext2fs_new_block(fs, goal, NULL, &block)) {
86 ext2fs_fast_mark_block_bitmap(fs->block_map, block);
87 *block_ret = block;
88 return 0;
89 }
90 return -ENOSPC;
91 }
92
93 static int ext2_free_block(ext2_filsys fs, u64 block)
94 {
95 BUG_ON(block != (blk_t)block);
96 ext2fs_fast_unmark_block_bitmap(fs->block_map, block);
97 return 0;
98 }
99
100 static int custom_alloc_extent(struct btrfs_root *root, u64 num_bytes,
101 u64 hint_byte, struct btrfs_key *ins)
102 {
103 struct btrfs_block_group_cache *cache;
104 ext2_filsys ext2_fs = (ext2_filsys)root->fs_info->priv_data;
105 u32 blocksize = ext2_fs->blocksize;
106 u64 bytenr;
107 blk_t block;
108 int wrapped = 0;
109 int ret;
110
111 while (1) {
112 block = hint_byte / blocksize;
113 ret = ext2fs_new_block(ext2_fs, block, NULL, &block);
114 if (ret)
115 goto fail;
116
117 bytenr = (u64)block * blocksize;
118 if (bytenr < hint_byte && ++wrapped > 1)
119 goto fail;
120
121 cache = btrfs_lookup_block_group(root->fs_info, bytenr);
122 BUG_ON(!cache);
123 if (cache->flags & BTRFS_BLOCK_GROUP_SYSTEM) {
124 hint_byte = cache->key.objectid + cache->key.offset;
125 continue;
126 }
127
128 if (test_range_bit(&root->fs_info->pinned_extents, bytenr,
129 bytenr + blocksize - 1, EXTENT_DIRTY, 0)) {
130 hint_byte = bytenr + blocksize;
131 continue;
132 }
133
134 ext2fs_fast_mark_block_bitmap(ext2_fs->block_map, block);
135 break;
136 }
137 ins->objectid = bytenr;
138 ins->offset = blocksize;
139 btrfs_set_key_type(ins, BTRFS_EXTENT_ITEM_KEY);
140 return 0;
141 fail:
142 fprintf(stderr, "not enough free space\n");
143 return -ENOSPC;
144 }
145
146 static int custom_free_extent(struct btrfs_root *root, u64 bytenr,
147 u64 num_bytes)
148 {
149 u64 block;
150 ext2_filsys fs = (ext2_filsys)root->fs_info->priv_data;
151
152 BUG_ON(bytenr & (fs->blocksize - 1));
153 block = bytenr / fs->blocksize;
154 while (num_bytes > 0) {
155 ext2_free_block(fs, block);
156 block++;
157 num_bytes -= fs->blocksize;
158 }
159 return 0;
160 }
161
162 struct btrfs_extent_ops extent_ops = {
163 .alloc_extent = custom_alloc_extent,
164 .free_extent = custom_free_extent,
165 };
166
167 struct dir_iterate_data {
168 struct btrfs_trans_handle *trans;
169 struct btrfs_root *root;
170 struct btrfs_inode_item *inode;
171 u64 objectid;
172 u64 parent;
173 int errcode;
174 };
175
176 static u8 filetype_conversion_table[EXT2_FT_MAX] = {
177 [EXT2_FT_UNKNOWN] = BTRFS_FT_UNKNOWN,
178 [EXT2_FT_REG_FILE] = BTRFS_FT_REG_FILE,
179 [EXT2_FT_DIR] = BTRFS_FT_DIR,
180 [EXT2_FT_CHRDEV] = BTRFS_FT_CHRDEV,
181 [EXT2_FT_BLKDEV] = BTRFS_FT_BLKDEV,
182 [EXT2_FT_FIFO] = BTRFS_FT_FIFO,
183 [EXT2_FT_SOCK] = BTRFS_FT_SOCK,
184 [EXT2_FT_SYMLINK] = BTRFS_FT_SYMLINK,
185 };
186
187 static int dir_iterate_proc(ext2_ino_t dir, int entry,
188 struct ext2_dir_entry *old,
189 int offset, int blocksize,
190 char *buf,void *priv_data)
191 {
192 int ret;
193 int file_type;
194 u64 objectid;
195 u64 inode_size;
196 char dotdot[] = "..";
197 struct btrfs_key location;
198 struct ext2_dir_entry_2 *dirent = (struct ext2_dir_entry_2 *)old;
199 struct dir_iterate_data *idata = (struct dir_iterate_data *)priv_data;
200
201 objectid = dirent->inode + INO_OFFSET;
202 if (!strncmp(dirent->name, dotdot, dirent->name_len)) {
203 if (dirent->name_len == 2) {
204 BUG_ON(idata->parent != 0);
205 idata->parent = objectid;
206 }
207 return 0;
208 }
209 if (dirent->inode < EXT2_GOOD_OLD_FIRST_INO)
210 return 0;
211
212 location.objectid = objectid;
213 location.offset = 0;
214 btrfs_set_key_type(&location, BTRFS_INODE_ITEM_KEY);
215
216 file_type = dirent->file_type;
217 BUG_ON(file_type > EXT2_FT_SYMLINK);
218 ret = btrfs_insert_dir_item(idata->trans, idata->root,
219 dirent->name, dirent->name_len,
220 idata->objectid, &location,
221 filetype_conversion_table[file_type]);
222 if (ret)
223 goto fail;
224 ret = btrfs_insert_inode_ref(idata->trans, idata->root,
225 dirent->name, dirent->name_len,
226 objectid, idata->objectid);
227 if (ret)
228 goto fail;
229 inode_size = btrfs_stack_inode_size(idata->inode) +
230 dirent->name_len * 2;
231 btrfs_set_stack_inode_size(idata->inode, inode_size);
232 return 0;
233 fail:
234 idata->errcode = ret;
235 return BLOCK_ABORT;
236 }
237
238 static int create_dir_entries(struct btrfs_trans_handle *trans,
239 struct btrfs_root *root, u64 objectid,
240 struct btrfs_inode_item *btrfs_inode,
241 ext2_filsys ext2_fs, ext2_ino_t ext2_ino)
242 {
243 int ret;
244 errcode_t err;
245 struct dir_iterate_data data = {
246 .trans = trans,
247 .root = root,
248 .inode = btrfs_inode,
249 .objectid = objectid,
250 .parent = 0,
251 .errcode = 0,
252 };
253
254 err = ext2fs_dir_iterate2(ext2_fs, ext2_ino, 0, NULL,
255 dir_iterate_proc, &data);
256 if (err)
257 goto error;
258 ret = data.errcode;
259 if (ret == 0 && data.parent == objectid) {
260 ret = btrfs_insert_inode_ref(trans, root, "..", 2,
261 objectid, objectid);
262 }
263 return ret;
264 error:
265 fprintf(stderr, "ext2fs_dir_iterate2: %s\n", error_message(err));
266 return -1;
267 }
268
269 static int read_disk_extent(struct btrfs_root *root, u64 bytenr,
270 u32 num_bytes, char *buffer)
271 {
272 int ret;
273 struct btrfs_fs_devices *fs_devs = root->fs_info->fs_devices;
274
275 ret = pread(fs_devs->latest_bdev, buffer, num_bytes, bytenr);
276 if (ret != num_bytes)
277 goto fail;
278 ret = 0;
279 fail:
280 if (ret > 0)
281 ret = -1;
282 return ret;
283 }
284 /*
285 * Record a file extent. Do all the required works, such as inserting
286 * file extent item, inserting extent item and backref item into extent
287 * tree and updating block accounting.
288 */
289 static int record_file_extent(struct btrfs_trans_handle *trans,
290 struct btrfs_root *root, u64 objectid,
291 struct btrfs_inode_item *inode,
292 u64 file_pos, u64 disk_bytenr,
293 u64 num_bytes, int checksum)
294 {
295 int ret;
296 struct btrfs_fs_info *info = root->fs_info;
297 struct btrfs_root *extent_root = info->extent_root;
298 struct btrfs_key ins_key;
299 struct btrfs_path path;
300 struct btrfs_extent_item extent_item;
301 u32 blocksize = root->sectorsize;
302 u64 nblocks;
303 u64 bytes_used;
304
305 ret = btrfs_insert_file_extent(trans, root, objectid, file_pos,
306 disk_bytenr, num_bytes, num_bytes);
307 if (ret || disk_bytenr == 0)
308 return ret;
309
310 nblocks = btrfs_stack_inode_nblocks(inode) + num_bytes / 512;
311 btrfs_set_stack_inode_nblocks(inode, nblocks);
312 if (checksum) {
313 u64 offset;
314 char *buffer;
315
316 ret = -ENOMEM;
317 buffer = malloc(blocksize);
318 if (!buffer)
319 goto fail;
320 for (offset = 0; offset < num_bytes; offset += blocksize) {
321 ret = read_disk_extent(root, disk_bytenr + offset,
322 blocksize, buffer);
323 if (ret)
324 break;
325 ret = btrfs_csum_file_block(trans, root, inode,
326 objectid, file_pos + offset,
327 buffer, blocksize);
328 if (ret)
329 break;
330 }
331 free(buffer);
332 if (ret)
333 goto fail;
334 }
335
336 bytes_used = btrfs_root_used(&root->root_item);
337 btrfs_set_root_used(&root->root_item, bytes_used + num_bytes);
338 ins_key.objectid = disk_bytenr;
339 ins_key.offset = num_bytes;
340 btrfs_set_key_type(&ins_key, BTRFS_EXTENT_ITEM_KEY);
341 btrfs_set_stack_extent_refs(&extent_item, 1);
342 ret = btrfs_insert_item(trans, extent_root, &ins_key,
343 &extent_item, sizeof(extent_item));
344 if (ret == 0) {
345 bytes_used = btrfs_super_bytes_used(&info->super_copy);
346 btrfs_set_super_bytes_used(&info->super_copy, bytes_used +
347 num_bytes);
348 btrfs_init_path(&path);
349 ret = btrfs_insert_extent_backref(trans, extent_root, &path,
350 disk_bytenr, root->root_key.objectid,
351 trans->transid, objectid, file_pos);
352 if (ret)
353 goto fail;
354 ret = btrfs_update_block_group(trans, root, disk_bytenr,
355 num_bytes, 1, 0);
356 } else if (ret == -EEXIST) {
357 ret = btrfs_inc_extent_ref(trans, root, disk_bytenr, num_bytes,
358 root->root_key.objectid,
359 trans->transid, objectid, file_pos);
360 }
361 if (ret)
362 goto fail;
363 btrfs_extent_post_op(trans, extent_root);
364 return 0;
365 fail:
366 return ret;
367 }
368
369 static int record_file_blocks(struct btrfs_trans_handle *trans,
370 struct btrfs_root *root, u64 objectid,
371 struct btrfs_inode_item *inode,
372 u64 file_block, u64 disk_block,
373 u64 num_blocks, int checksum)
374 {
375 u64 file_pos = file_block * root->sectorsize;
376 u64 disk_bytenr = disk_block * root->sectorsize;
377 u64 num_bytes = num_blocks * root->sectorsize;
378 return record_file_extent(trans, root, objectid, inode, file_pos,
379 disk_bytenr, num_bytes, checksum);
380 }
381
382 struct blk_iterate_data {
383 struct btrfs_trans_handle *trans;
384 struct btrfs_root *root;
385 struct btrfs_inode_item *inode;
386 u64 objectid;
387 u64 first_block;
388 u64 disk_block;
389 u64 num_blocks;
390 u64 boundary;
391 int checksum;
392 int errcode;
393 };
394
395 static int block_iterate_proc(ext2_filsys ext2_fs,
396 u64 disk_block, u64 file_block,
397 struct blk_iterate_data *idata)
398 {
399 int ret;
400 struct btrfs_root *root = idata->root;
401 struct btrfs_trans_handle *trans = idata->trans;
402 struct btrfs_block_group_cache *cache;
403 u64 bytenr;
404
405 BUG_ON(disk_block == 0);
406 if ((idata->num_blocks > 0 && disk_block >= idata->boundary) ||
407 (file_block > idata->first_block + idata->num_blocks) ||
408 (disk_block != idata->disk_block + idata->num_blocks)) {
409 if (idata->num_blocks > 0) {
410 ret = record_file_blocks(trans, root, idata->objectid,
411 idata->inode, idata->first_block,
412 idata->disk_block, idata->num_blocks,
413 idata->checksum);
414 if (ret)
415 goto fail;
416 idata->first_block += idata->num_blocks;
417 idata->num_blocks = 0;
418 }
419 if (file_block > idata->first_block) {
420 ret = record_file_blocks(trans, root, idata->objectid,
421 idata->inode, idata->first_block,
422 0, file_block - idata->first_block,
423 idata->checksum);
424 if (ret)
425 goto fail;
426 }
427
428 bytenr = disk_block * root->sectorsize;
429 cache = btrfs_lookup_block_group(root->fs_info, bytenr);
430 BUG_ON(!cache);
431 bytenr = cache->key.objectid + cache->key.offset;
432
433 idata->first_block = file_block;
434 idata->disk_block = disk_block;
435 idata->boundary = bytenr / root->sectorsize;
436 }
437 idata->num_blocks++;
438 return 0;
439 fail:
440 idata->errcode = ret;
441 return BLOCK_ABORT;
442 }
443
444 static int __block_iterate_proc(ext2_filsys fs, blk_t *blocknr,
445 e2_blkcnt_t blockcnt, blk_t ref_block,
446 int ref_offset, void *priv_data)
447 {
448 struct blk_iterate_data *idata;
449 idata = (struct blk_iterate_data *)priv_data;
450 return block_iterate_proc(fs, *blocknr, blockcnt, idata);
451 }
452
453 /*
454 * traverse file's data blocks, record these data blocks as file extents.
455 */
456 static int create_file_extents(struct btrfs_trans_handle *trans,
457 struct btrfs_root *root, u64 objectid,
458 struct btrfs_inode_item *btrfs_inode,
459 ext2_filsys ext2_fs, ext2_ino_t ext2_ino,
460 int datacsum, int packing)
461 {
462 int ret;
463 char *buffer = NULL;
464 errcode_t err;
465 u32 last_block;
466 u32 sectorsize = root->sectorsize;
467 u64 inode_size = btrfs_stack_inode_size(btrfs_inode);
468 struct blk_iterate_data data = {
469 .trans = trans,
470 .root = root,
471 .inode = btrfs_inode,
472 .objectid = objectid,
473 .first_block = 0,
474 .disk_block = 0,
475 .num_blocks = 0,
476 .boundary = (u64)-1,
477 .checksum = datacsum,
478 .errcode = 0,
479 };
480 err = ext2fs_block_iterate2(ext2_fs, ext2_ino, BLOCK_FLAG_DATA_ONLY,
481 NULL, __block_iterate_proc, &data);
482 if (err)
483 goto error;
484 ret = data.errcode;
485 if (ret)
486 goto fail;
487 if (packing && data.first_block == 0 && data.num_blocks > 0 &&
488 inode_size <= BTRFS_MAX_INLINE_DATA_SIZE(root)) {
489 u64 num_bytes = data.num_blocks * sectorsize;
490 u64 disk_bytenr = data.disk_block * sectorsize;
491
492 buffer = malloc(num_bytes);
493 if (!buffer)
494 return -ENOMEM;
495 ret = read_disk_extent(root, disk_bytenr, num_bytes, buffer);
496 if (ret)
497 goto fail;
498 if (num_bytes > inode_size)
499 num_bytes = inode_size;
500 ret = btrfs_insert_inline_extent(trans, root, objectid,
501 0, buffer, num_bytes);
502 if (ret)
503 goto fail;
504 } else if (data.num_blocks > 0) {
505 ret = record_file_blocks(trans, root, objectid, btrfs_inode,
506 data.first_block, data.disk_block,
507 data.num_blocks, data.checksum);
508 if (ret)
509 goto fail;
510 }
511 data.first_block += data.num_blocks;
512 last_block = (inode_size + sectorsize - 1) / sectorsize;
513 if (last_block > data.first_block) {
514 ret = record_file_blocks(trans, root, objectid, btrfs_inode,
515 data.first_block, 0, last_block -
516 data.first_block, data.checksum);
517 }
518 fail:
519 if (buffer)
520 free(buffer);
521 return ret;
522 error:
523 fprintf(stderr, "ext2fs_block_iterate2: %s\n", error_message(err));
524 return -1;
525 }
526
527 static int create_symbol_link(struct btrfs_trans_handle *trans,
528 struct btrfs_root *root, u64 objectid,
529 struct btrfs_inode_item *btrfs_inode,
530 ext2_filsys ext2_fs, ext2_ino_t ext2_ino,
531 struct ext2_inode *ext2_inode)
532 {
533 int ret;
534 char *pathname;
535 u64 inode_size = btrfs_stack_inode_size(btrfs_inode);
536 if (ext2fs_inode_data_blocks(ext2_fs, ext2_inode)) {
537 btrfs_set_stack_inode_size(btrfs_inode, inode_size + 1);
538 ret = create_file_extents(trans, root, objectid, btrfs_inode,
539 ext2_fs, ext2_ino, 1, 1);
540 btrfs_set_stack_inode_size(btrfs_inode, inode_size);
541 return ret;
542 }
543
544 pathname = (char *)&(ext2_inode->i_block[0]);
545 BUG_ON(pathname[inode_size] != 0);
546 ret = btrfs_insert_inline_extent(trans, root, objectid, 0,
547 pathname, inode_size + 1);
548 return ret;
549 }
550
551 /*
552 * Following xattr/acl related codes are based on codes in
553 * fs/ext3/xattr.c and fs/ext3/acl.c
554 */
555 #define EXT2_XATTR_BHDR(ptr) ((struct ext2_ext_attr_header *)(ptr))
556 #define EXT2_XATTR_BFIRST(ptr) \
557 ((struct ext2_ext_attr_entry *)(EXT2_XATTR_BHDR(ptr) + 1))
558 #define EXT2_XATTR_IHDR(inode) \
559 ((struct ext2_ext_attr_header *) ((void *)(inode) + \
560 EXT2_GOOD_OLD_INODE_SIZE + (inode)->i_extra_isize))
561 #define EXT2_XATTR_IFIRST(inode) \
562 ((struct ext2_ext_attr_entry *) ((void *)EXT2_XATTR_IHDR(inode) + \
563 sizeof(EXT2_XATTR_IHDR(inode)->h_magic)))
564
565 static int ext2_xattr_check_names(struct ext2_ext_attr_entry *entry,
566 const void *end)
567 {
568 struct ext2_ext_attr_entry *next;
569
570 while (!EXT2_EXT_IS_LAST_ENTRY(entry)) {
571 next = EXT2_EXT_ATTR_NEXT(entry);
572 if ((void *)next >= end)
573 return -EIO;
574 entry = next;
575 }
576 return 0;
577 }
578
579 static int ext2_xattr_check_block(const char *buf, size_t size)
580 {
581 int error;
582 struct ext2_ext_attr_header *header = EXT2_XATTR_BHDR(buf);
583
584 if (header->h_magic != EXT2_EXT_ATTR_MAGIC ||
585 header->h_blocks != 1)
586 return -EIO;
587 error = ext2_xattr_check_names(EXT2_XATTR_BFIRST(buf), buf + size);
588 return error;
589 }
590
591 static int ext2_xattr_check_entry(struct ext2_ext_attr_entry *entry,
592 size_t size)
593 {
594 size_t value_size = entry->e_value_size;
595
596 if (entry->e_value_block != 0 || value_size > size ||
597 entry->e_value_offs + value_size > size)
598 return -EIO;
599 return 0;
600 }
601
602 #define EXT2_ACL_VERSION 0x0001
603
604 typedef struct {
605 __le16 e_tag;
606 __le16 e_perm;
607 __le32 e_id;
608 } ext2_acl_entry;
609
610 typedef struct {
611 __le16 e_tag;
612 __le16 e_perm;
613 } ext2_acl_entry_short;
614
615 typedef struct {
616 __le32 a_version;
617 } ext2_acl_header;
618
619 static inline int ext2_acl_count(size_t size)
620 {
621 ssize_t s;
622 size -= sizeof(ext2_acl_header);
623 s = size - 4 * sizeof(ext2_acl_entry_short);
624 if (s < 0) {
625 if (size % sizeof(ext2_acl_entry_short))
626 return -1;
627 return size / sizeof(ext2_acl_entry_short);
628 } else {
629 if (s % sizeof(ext2_acl_entry))
630 return -1;
631 return s / sizeof(ext2_acl_entry) + 4;
632 }
633 }
634
635 #define ACL_EA_VERSION 0x0002
636
637 typedef struct {
638 __le16 e_tag;
639 __le16 e_perm;
640 __le32 e_id;
641 } acl_ea_entry;
642
643 typedef struct {
644 __le32 a_version;
645 acl_ea_entry a_entries[0];
646 } acl_ea_header;
647
648 static inline size_t acl_ea_size(int count)
649 {
650 return sizeof(acl_ea_header) + count * sizeof(acl_ea_entry);
651 }
652
653 static int ext2_acl_to_xattr(void *dst, const void *src,
654 size_t dst_size, size_t src_size)
655 {
656 int i, count;
657 const void *end = src + src_size;
658 acl_ea_header *ext_acl = (acl_ea_header *)dst;
659 acl_ea_entry *dst_entry = ext_acl->a_entries;
660 ext2_acl_entry *src_entry;
661
662 if (src_size < sizeof(ext2_acl_header))
663 goto fail;
664 if (((ext2_acl_header *)src)->a_version !=
665 cpu_to_le32(EXT2_ACL_VERSION))
666 goto fail;
667 src += sizeof(ext2_acl_header);
668 count = ext2_acl_count(src_size);
669 if (count <= 0)
670 goto fail;
671
672 BUG_ON(dst_size < acl_ea_size(count));
673 ext_acl->a_version = cpu_to_le32(ACL_EA_VERSION);
674 for (i = 0; i < count; i++, dst_entry++) {
675 src_entry = (ext2_acl_entry *)src;
676 if (src + sizeof(ext2_acl_entry_short) > end)
677 goto fail;
678 dst_entry->e_tag = src_entry->e_tag;
679 dst_entry->e_perm = src_entry->e_perm;
680 switch (le16_to_cpu(src_entry->e_tag)) {
681 case ACL_USER_OBJ:
682 case ACL_GROUP_OBJ:
683 case ACL_MASK:
684 case ACL_OTHER:
685 src += sizeof(ext2_acl_entry_short);
686 dst_entry->e_id = cpu_to_le32(ACL_UNDEFINED_ID);
687 break;
688 case ACL_USER:
689 case ACL_GROUP:
690 src += sizeof(ext2_acl_entry);
691 if (src > end)
692 goto fail;
693 dst_entry->e_id = src_entry->e_id;
694 break;
695 default:
696 goto fail;
697 }
698 }
699 if (src != end)
700 goto fail;
701 return 0;
702 fail:
703 return -EINVAL;
704 }
705
706 static char *xattr_prefix_table[] = {
707 [1] = "user.",
708 [2] = "system.posix_acl_access",
709 [3] = "system.posix_acl_default",
710 [4] = "trusted.",
711 [6] = "security.",
712 };
713
714 static int copy_single_xattr(struct btrfs_trans_handle *trans,
715 struct btrfs_root *root, u64 objectid,
716 struct ext2_ext_attr_entry *entry,
717 const void *data, u32 datalen)
718 {
719 int ret = 0;
720 int name_len;
721 int name_index;
722 void *databuf = NULL;
723 char namebuf[XATTR_NAME_MAX + 1];
724
725 name_index = entry->e_name_index;
726 if (name_index >= ARRAY_SIZE(xattr_prefix_table) ||
727 xattr_prefix_table[name_index] == NULL)
728 return -EOPNOTSUPP;
729 name_len = strlen(xattr_prefix_table[name_index]) +
730 entry->e_name_len;
731 if (name_len >= sizeof(namebuf))
732 return -ERANGE;
733
734 if (name_index == 2 || name_index == 3) {
735 size_t bufsize = acl_ea_size(ext2_acl_count(datalen));
736 databuf = malloc(bufsize);
737 if (!databuf)
738 return -ENOMEM;
739 ret = ext2_acl_to_xattr(databuf, data, bufsize, datalen);
740 if (ret)
741 goto out;
742 data = databuf;
743 datalen = bufsize;
744 }
745 strcpy(namebuf, xattr_prefix_table[name_index]);
746 strncat(namebuf, EXT2_EXT_ATTR_NAME(entry), entry->e_name_len);
747 if (name_len + datalen > BTRFS_LEAF_DATA_SIZE(root) -
748 sizeof(struct btrfs_item) - sizeof(struct btrfs_dir_item)) {
749 fprintf(stderr, "skip large xattr on inode %Lu name %.*s\n",
750 objectid - INO_OFFSET, name_len, namebuf);
751 goto out;
752 }
753 ret = btrfs_insert_xattr_item(trans, root, namebuf, name_len,
754 data, datalen, objectid);
755 out:
756 if (databuf)
757 free(databuf);
758 return ret;
759 }
760
761 static int copy_extended_attrs(struct btrfs_trans_handle *trans,
762 struct btrfs_root *root, u64 objectid,
763 struct btrfs_inode_item *btrfs_inode,
764 ext2_filsys ext2_fs, ext2_ino_t ext2_ino)
765 {
766 int ret = 0;
767 int inline_ea = 0;
768 errcode_t err;
769 u32 datalen;
770 u32 block_size = ext2_fs->blocksize;
771 u32 inode_size = EXT2_INODE_SIZE(ext2_fs->super);
772 struct ext2_inode_large *ext2_inode;
773 struct ext2_ext_attr_entry *entry;
774 void *data;
775 char *buffer = NULL;
776 char inode_buf[EXT2_GOOD_OLD_INODE_SIZE];
777
778 if (inode_size <= EXT2_GOOD_OLD_INODE_SIZE) {
779 ext2_inode = (struct ext2_inode_large *)inode_buf;
780 } else {
781 ext2_inode = (struct ext2_inode_large *)malloc(inode_size);
782 if (!ext2_inode)
783 return -ENOMEM;
784 }
785 err = ext2fs_read_inode_full(ext2_fs, ext2_ino, (void *)ext2_inode,
786 inode_size);
787 if (err) {
788 fprintf(stderr, "ext2fs_read_inode_full: %s\n",
789 error_message(err));
790 ret = -1;
791 goto out;
792 }
793
794 if (ext2_ino > ext2_fs->super->s_first_ino &&
795 inode_size > EXT2_GOOD_OLD_INODE_SIZE) {
796 if (EXT2_GOOD_OLD_INODE_SIZE +
797 ext2_inode->i_extra_isize > inode_size) {
798 ret = -EIO;
799 goto out;
800 }
801 if (ext2_inode->i_extra_isize != 0 &&
802 EXT2_XATTR_IHDR(ext2_inode)->h_magic ==
803 EXT2_EXT_ATTR_MAGIC) {
804 inline_ea = 1;
805 }
806 }
807 if (inline_ea) {
808 int total;
809 void *end = (void *)ext2_inode + inode_size;
810 entry = EXT2_XATTR_IFIRST(ext2_inode);
811 total = end - (void *)entry;
812 ret = ext2_xattr_check_names(entry, end);
813 if (ret)
814 goto out;
815 while (!EXT2_EXT_IS_LAST_ENTRY(entry)) {
816 ret = ext2_xattr_check_entry(entry, total);
817 if (ret)
818 goto out;
819 data = (void *)EXT2_XATTR_IFIRST(ext2_inode) +
820 entry->e_value_offs;
821 datalen = entry->e_value_size;
822 ret = copy_single_xattr(trans, root, objectid,
823 entry, data, datalen);
824 if (ret)
825 goto out;
826 entry = EXT2_EXT_ATTR_NEXT(entry);
827 }
828 }
829
830 if (ext2_inode->i_file_acl == 0)
831 goto out;
832
833 buffer = malloc(block_size);
834 if (!buffer) {
835 ret = -ENOMEM;
836 goto out;
837 }
838 err = ext2fs_read_ext_attr(ext2_fs, ext2_inode->i_file_acl, buffer);
839 if (err) {
840 fprintf(stderr, "ext2fs_read_ext_attr: %s\n",
841 error_message(err));
842 ret = -1;
843 goto out;
844 }
845 ret = ext2_xattr_check_block(buffer, block_size);
846 if (ret)
847 goto out;
848
849 entry = EXT2_XATTR_BFIRST(buffer);
850 while (!EXT2_EXT_IS_LAST_ENTRY(entry)) {
851 ret = ext2_xattr_check_entry(entry, block_size);
852 if (ret)
853 goto out;
854 data = buffer + entry->e_value_offs;
855 datalen = entry->e_value_size;
856 ret = copy_single_xattr(trans, root, objectid,
857 entry, data, datalen);
858 if (ret)
859 goto out;
860 entry = EXT2_EXT_ATTR_NEXT(entry);
861 }
862 out:
863 if (buffer != NULL)
864 free(buffer);
865 if ((void *)ext2_inode != inode_buf)
866 free(ext2_inode);
867 return ret;
868 }
869 #define MINORBITS 20
870 #define MKDEV(ma, mi) (((ma) << MINORBITS) | (mi))
871
872 static inline dev_t old_decode_dev(u16 val)
873 {
874 return MKDEV((val >> 8) & 255, val & 255);
875 }
876
877 static inline dev_t new_decode_dev(u32 dev)
878 {
879 unsigned major = (dev & 0xfff00) >> 8;
880 unsigned minor = (dev & 0xff) | ((dev >> 12) & 0xfff00);
881 return MKDEV(major, minor);
882 }
883
884 static int copy_inode_item(struct btrfs_inode_item *dst,
885 struct ext2_inode *src)
886 {
887 btrfs_set_stack_inode_generation(dst, 1);
888 btrfs_set_stack_inode_size(dst, src->i_size);
889 btrfs_set_stack_inode_nblocks(dst, src->i_blocks);
890 btrfs_set_stack_inode_block_group(dst, 0);
891 btrfs_set_stack_inode_nblocks(dst, 0);
892 btrfs_set_stack_inode_nlink(dst, src->i_links_count);
893 btrfs_set_stack_inode_uid(dst, src->i_uid | (src->i_uid_high << 16));
894 btrfs_set_stack_inode_gid(dst, src->i_gid | (src->i_gid_high << 16));
895 btrfs_set_stack_inode_mode(dst, src->i_mode);
896 btrfs_set_stack_inode_rdev(dst, 0);
897 btrfs_set_stack_inode_flags(dst, 0);
898 btrfs_set_stack_inode_compat_flags(dst, 0);
899 btrfs_set_stack_timespec_sec(&dst->atime, src->i_atime);
900 btrfs_set_stack_timespec_nsec(&dst->atime, 0);
901 btrfs_set_stack_timespec_sec(&dst->ctime, src->i_ctime);
902 btrfs_set_stack_timespec_nsec(&dst->ctime, 0);
903 btrfs_set_stack_timespec_sec(&dst->mtime, src->i_mtime);
904 btrfs_set_stack_timespec_nsec(&dst->mtime, 0);
905 btrfs_set_stack_timespec_sec(&dst->otime, 0);
906 btrfs_set_stack_timespec_nsec(&dst->otime, 0);
907
908 if (S_ISDIR(src->i_mode)) {
909 btrfs_set_stack_inode_size(dst, 0);
910 btrfs_set_stack_inode_nlink(dst, 1);
911 }
912 if (!S_ISREG(src->i_mode) && !S_ISDIR(src->i_mode) &&
913 !S_ISLNK(src->i_mode)) {
914 if (src->i_block[0]) {
915 btrfs_set_stack_inode_rdev(dst,
916 old_decode_dev(src->i_block[0]));
917 } else {
918 btrfs_set_stack_inode_rdev(dst,
919 new_decode_dev(src->i_block[1]));
920 }
921 }
922 return 0;
923 }
924
925 /*
926 * copy a single inode. do all the required works, such as cloning
927 * inode item, creating file extents and creating directory entries.
928 */
929 static int copy_single_inode(struct btrfs_trans_handle *trans,
930 struct btrfs_root *root, u64 objectid,
931 ext2_filsys ext2_fs, ext2_ino_t ext2_ino,
932 int datacsum, int packing, int noxattr)
933 {
934 int ret;
935 errcode_t err;
936 struct ext2_inode ext2_inode;
937 struct btrfs_key inode_key;
938 struct btrfs_inode_item btrfs_inode;
939
940 err = ext2fs_read_inode(ext2_fs, ext2_ino, &ext2_inode);
941 if (err)
942 goto error;
943
944 if (!ext2_inode.i_links_count &&
945 (!ext2_inode.i_mode || ext2_inode.i_dtime)) {
946 printf("skip inode %u\n", ext2_ino);
947 return 0;
948 }
949 copy_inode_item(&btrfs_inode, &ext2_inode);
950 if (!datacsum && S_ISREG(ext2_inode.i_mode)) {
951 u32 flags = btrfs_stack_inode_flags(&btrfs_inode) |
952 BTRFS_INODE_NODATASUM;
953 btrfs_set_stack_inode_flags(&btrfs_inode, flags);
954 }
955
956 switch (ext2_inode.i_mode & S_IFMT) {
957 case S_IFREG:
958 ret = create_file_extents(trans, root, objectid, &btrfs_inode,
959 ext2_fs, ext2_ino, datacsum, packing);
960 break;
961 case S_IFDIR:
962 ret = create_dir_entries(trans, root, objectid, &btrfs_inode,
963 ext2_fs, ext2_ino);
964 break;
965 case S_IFLNK:
966 ret = create_symbol_link(trans, root, objectid, &btrfs_inode,
967 ext2_fs, ext2_ino, &ext2_inode);
968 break;
969 default:
970 ret = 0;
971 break;
972 }
973 if (ret)
974 return ret;
975
976 if (!noxattr) {
977 ret = copy_extended_attrs(trans, root, objectid, &btrfs_inode,
978 ext2_fs, ext2_ino);
979 if (ret)
980 return ret;
981 }
982 inode_key.objectid = objectid;
983 inode_key.offset = 0;
984 btrfs_set_key_type(&inode_key, BTRFS_INODE_ITEM_KEY);
985 ret = btrfs_insert_inode(trans, root, objectid, &btrfs_inode);
986 return ret;
987 error:
988 fprintf(stderr, "ext2fs_read_inode: %s\n", error_message(err));
989 return -1;
990 }
991
992 static int copy_disk_extent(struct btrfs_root *root, u64 dst_bytenr,
993 u64 src_bytenr, u32 num_bytes)
994 {
995 int ret;
996 char *buffer;
997 struct btrfs_fs_devices *fs_devs = root->fs_info->fs_devices;
998
999 buffer = malloc(num_bytes);
1000 if (!buffer)
1001 return -ENOMEM;
1002 ret = pread(fs_devs->latest_bdev, buffer, num_bytes, src_bytenr);
1003 if (ret != num_bytes)
1004 goto fail;
1005 ret = pwrite(fs_devs->latest_bdev, buffer, num_bytes, dst_bytenr);
1006 if (ret != num_bytes)
1007 goto fail;
1008 ret = 0;
1009 fail:
1010 free(buffer);
1011 if (ret > 0)
1012 ret = -1;
1013 return ret;
1014 }
1015 /*
1016 * scan ext2's inode bitmap and copy all used inode.
1017 */
1018 static int copy_inodes(struct btrfs_root *root, ext2_filsys ext2_fs,
1019 int datacsum, int packing, int noxattr)
1020 {
1021 int ret;
1022 ext2_ino_t ext2_ino;
1023 u64 objectid;
1024 struct btrfs_trans_handle *trans;
1025
1026 trans = btrfs_start_transaction(root, 1);
1027 if (!trans)
1028 return -ENOMEM;
1029 ext2_ino = ext2_fs->inode_map->start;
1030 for (; ext2_ino <= ext2_fs->inode_map->end; ext2_ino++) {
1031 if (ext2fs_fast_test_inode_bitmap(ext2_fs->inode_map,
1032 ext2_ino)) {
1033 /* skip special inode in ext2fs */
1034 if (ext2_ino < EXT2_GOOD_OLD_FIRST_INO &&
1035 ext2_ino != EXT2_ROOT_INO)
1036 continue;
1037 objectid = ext2_ino + INO_OFFSET;
1038 ret = copy_single_inode(trans, root,
1039 objectid, ext2_fs, ext2_ino,
1040 datacsum, packing, noxattr);
1041 if (ret)
1042 return ret;
1043 }
1044 if (trans->blocks_used >= 4096) {
1045 ret = btrfs_commit_transaction(trans, root);
1046 BUG_ON(ret);
1047 trans = btrfs_start_transaction(root, 1);
1048 BUG_ON(!trans);
1049 }
1050 }
1051 ret = btrfs_commit_transaction(trans, root);
1052 BUG_ON(ret);
1053
1054 return ret;
1055 }
1056
1057 static int lookup_extent_item(struct btrfs_trans_handle *trans,
1058 struct btrfs_root *root,
1059 u64 bytenr, u64 num_bytes)
1060 {
1061 int ret;
1062 struct btrfs_key key;
1063 struct btrfs_path path;
1064
1065 btrfs_init_path(&path);
1066 key.objectid = bytenr;
1067 key.offset = (u64)-1;
1068 btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY);
1069 ret = btrfs_search_slot(trans, root->fs_info->extent_root,
1070 &key, &path, 0, 0);
1071 if (ret < 0)
1072 goto out;
1073 BUG_ON(ret == 0);
1074
1075 ret = btrfs_previous_item(root, &path, 0, BTRFS_EXTENT_ITEM_KEY);
1076 if (ret != 0)
1077 goto out;
1078
1079 btrfs_item_key_to_cpu(path.nodes[0], &key, path.slots[0]);
1080 if (key.objectid + key.offset < bytenr + num_bytes)
1081 ret = 1;
1082 out:
1083 btrfs_release_path(root, &path);
1084 return ret;
1085 }
1086 /*
1087 * Construct a range of ext2fs image file.
1088 * scan block allocation bitmap, find all blocks used by the ext2fs
1089 * in this range and create file extents that point to these blocks.
1090 *
1091 * Note: Before calling the function, no file extent points to blocks
1092 * in this range
1093 */
1094 static int create_image_file_range(struct btrfs_trans_handle *trans,
1095 struct btrfs_root *root, u64 objectid,
1096 struct btrfs_inode_item *inode,
1097 u64 start_byte, u64 end_byte,
1098 ext2_filsys ext2_fs)
1099 {
1100 u64 bytenr;
1101 u32 blocksize = ext2_fs->blocksize;
1102 u32 block = start_byte / blocksize;
1103 u32 last_block = (end_byte + blocksize - 1) / blocksize;
1104 int ret;
1105 struct blk_iterate_data data = {
1106 .trans = trans,
1107 .root = root,
1108 .inode = inode,
1109 .objectid = objectid,
1110 .first_block = block,
1111 .disk_block = 0,
1112 .num_blocks = 0,
1113 .boundary = (u64)-1,
1114 .checksum = 0,
1115 .errcode = 0,
1116 };
1117 for (; start_byte < end_byte; block++, start_byte += blocksize) {
1118 if (!ext2fs_fast_test_block_bitmap(ext2_fs->block_map, block))
1119 continue;
1120 /* the bit may be set by us, check extent tree */
1121 bytenr = (u64)block * blocksize;
1122 ret = lookup_extent_item(trans, root, bytenr, blocksize);
1123 if (ret < 0)
1124 goto fail;
1125 if (ret == 0)
1126 continue;
1127
1128 ret = block_iterate_proc(NULL, block, block, &data);
1129 if (ret & BLOCK_ABORT) {
1130 ret = data.errcode;
1131 goto fail;
1132 }
1133 }
1134 if (data.num_blocks > 0) {
1135 ret = record_file_blocks(trans, root, objectid, inode,
1136 data.first_block, data.disk_block,
1137 data.num_blocks, 0);
1138 if (ret)
1139 goto fail;
1140 data.first_block += data.num_blocks;
1141 }
1142 if (last_block > data.first_block) {
1143 ret = record_file_blocks(trans, root, objectid, inode,
1144 data.first_block, 0, last_block -
1145 data.first_block, 0);
1146 if (ret)
1147 goto fail;
1148 }
1149 fail:
1150 return ret;
1151 }
1152 /*
1153 * Create the ext2fs image file.
1154 */
1155 static int create_ext2_image(struct btrfs_root *root, ext2_filsys ext2_fs,
1156 const char *name)
1157 {
1158 int ret;
1159 struct btrfs_key key;
1160 struct btrfs_key location;
1161 struct btrfs_path path;
1162 struct btrfs_inode_item btrfs_inode;
1163 struct btrfs_inode_item *inode_item;
1164 struct extent_buffer *leaf;
1165 struct btrfs_fs_info *fs_info = root->fs_info;
1166 struct btrfs_root *extent_root = fs_info->extent_root;
1167 struct btrfs_trans_handle *trans;
1168 struct btrfs_extent_ref *ref_item;
1169 u64 bytenr;
1170 u64 num_bytes;
1171 u64 ref_root;
1172 u64 ref_owner;
1173 u64 objectid;
1174 u64 last_byte;
1175 u64 first_free;
1176 u64 total_bytes;
1177 u32 sectorsize = root->sectorsize;
1178 int file_extent;
1179
1180 total_bytes = btrfs_super_total_bytes(&fs_info->super_copy);
1181 first_free = BTRFS_SUPER_INFO_OFFSET + sectorsize * 2 - 1;
1182 first_free &= ~((u64)sectorsize - 1);
1183
1184 memset(&btrfs_inode, 0, sizeof(btrfs_inode));
1185 btrfs_set_stack_inode_generation(&btrfs_inode, 1);
1186 btrfs_set_stack_inode_size(&btrfs_inode, total_bytes);
1187 btrfs_set_stack_inode_nlink(&btrfs_inode, 1);
1188 btrfs_set_stack_inode_nblocks(&btrfs_inode, 0);
1189 btrfs_set_stack_inode_mode(&btrfs_inode, S_IFREG | 0400);
1190 btrfs_set_stack_inode_flags(&btrfs_inode, BTRFS_INODE_NODATASUM |
1191 BTRFS_INODE_READONLY);
1192 btrfs_init_path(&path);
1193 trans = btrfs_start_transaction(root, 1);
1194 BUG_ON(!trans);
1195
1196 objectid = btrfs_root_dirid(&root->root_item);
1197 ret = btrfs_find_free_objectid(trans, root, objectid, &objectid);
1198 if (ret)
1199 goto fail;
1200
1201 /*
1202 * copy blocks covered by extent #0 to new positions. extent #0 is
1203 * special, we can't rely on relocate_extents_range to relocate it.
1204 */
1205 for (last_byte = 0; last_byte < first_free; last_byte += sectorsize) {
1206 ret = custom_alloc_extent(root, sectorsize, 0, &key);
1207 if (ret)
1208 goto fail;
1209 ret = copy_disk_extent(root, key.objectid, last_byte,
1210 sectorsize);
1211 if (ret)
1212 goto fail;
1213 ret = record_file_extent(trans, root, objectid,
1214 &btrfs_inode, last_byte,
1215 key.objectid, sectorsize, 0);
1216 if (ret)
1217 goto fail;
1218 }
1219
1220 while(1) {
1221 key.objectid = last_byte;
1222 key.offset = 0;
1223 btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY);
1224 ret = btrfs_search_slot(trans, fs_info->extent_root,
1225 &key, &path, 0, 0);
1226 if (ret < 0)
1227 goto fail;
1228 next:
1229 leaf = path.nodes[0];
1230 if (path.slots[0] >= btrfs_header_nritems(leaf)) {
1231 ret = btrfs_next_leaf(extent_root, &path);
1232 if (ret < 0)
1233 goto fail;
1234 if (ret > 0)
1235 break;
1236 leaf = path.nodes[0];
1237 }
1238 btrfs_item_key_to_cpu(leaf, &key, path.slots[0]);
1239 if (last_byte > key.objectid ||
1240 key.type != BTRFS_EXTENT_ITEM_KEY) {
1241 path.slots[0]++;
1242 goto next;
1243 }
1244 /*
1245 * Check backref to distinguish extent items for normal
1246 * files (files that correspond to files in Ext2fs) from
1247 * extent items for ctree blocks.
1248 */
1249 bytenr = key.objectid;
1250 num_bytes = key.offset;
1251 file_extent = 0;
1252 while (1) {
1253 if (path.slots[0] >= btrfs_header_nritems(leaf)) {
1254 ret = btrfs_next_leaf(extent_root, &path);
1255 if (ret > 0)
1256 break;
1257 if (ret < 0)
1258 goto fail;
1259 leaf = path.nodes[0];
1260 }
1261 btrfs_item_key_to_cpu(leaf, &key, path.slots[0]);
1262 if (key.objectid != bytenr)
1263 break;
1264 if (key.type != BTRFS_EXTENT_REF_KEY) {
1265 path.slots[0]++;
1266 continue;
1267 }
1268 ref_item = btrfs_item_ptr(leaf, path.slots[0],
1269 struct btrfs_extent_ref);
1270 ref_root = btrfs_ref_root(leaf, ref_item);
1271 ref_owner = btrfs_ref_objectid(leaf, ref_item);
1272 if ((ref_root == BTRFS_FS_TREE_OBJECTID) &&
1273 (ref_owner >= BTRFS_FIRST_FREE_OBJECTID)) {
1274 file_extent = 1;
1275 break;
1276 }
1277 path.slots[0]++;
1278 }
1279 if (!file_extent)
1280 goto next;
1281
1282 if (bytenr > last_byte) {
1283 ret = create_image_file_range(trans, root, objectid,
1284 &btrfs_inode, last_byte,
1285 bytenr, ext2_fs);
1286 if (ret)
1287 goto fail;
1288 }
1289 ret = record_file_extent(trans, root, objectid, &btrfs_inode,
1290 bytenr, bytenr, num_bytes, 0);
1291 if (ret)
1292 goto fail;
1293 last_byte = bytenr + num_bytes;
1294 btrfs_release_path(extent_root, &path);
1295
1296 if (trans->blocks_used >= 4096) {
1297 ret = btrfs_commit_transaction(trans, root);
1298 BUG_ON(ret);
1299 trans = btrfs_start_transaction(root, 1);
1300 BUG_ON(!trans);
1301 }
1302 }
1303 btrfs_release_path(root, &path);
1304 if (total_bytes > last_byte) {
1305 ret = create_image_file_range(trans, root, objectid,
1306 &btrfs_inode, last_byte,
1307 total_bytes, ext2_fs);
1308 if (ret)
1309 goto fail;
1310 }
1311
1312 ret = btrfs_insert_inode(trans, root, objectid, &btrfs_inode);
1313 if (ret)
1314 goto fail;
1315
1316 location.objectid = objectid;
1317 location.offset = 0;
1318 btrfs_set_key_type(&location, BTRFS_INODE_ITEM_KEY);
1319 ret = btrfs_insert_dir_item(trans, root, name, strlen(name),
1320 btrfs_root_dirid(&root->root_item),
1321 &location, EXT2_FT_REG_FILE);
1322 if (ret)
1323 goto fail;
1324 ret = btrfs_insert_inode_ref(trans, root, name, strlen(name),
1325 objectid,
1326 btrfs_root_dirid(&root->root_item));
1327 if (ret)
1328 goto fail;
1329 location.objectid = btrfs_root_dirid(&root->root_item);
1330 location.offset = 0;
1331 btrfs_set_key_type(&location, BTRFS_INODE_ITEM_KEY);
1332 ret = btrfs_lookup_inode(trans, root, &path, &location, 1);
1333 if (ret)
1334 goto fail;
1335 leaf = path.nodes[0];
1336 inode_item = btrfs_item_ptr(leaf, path.slots[0],
1337 struct btrfs_inode_item);
1338 btrfs_set_inode_size(leaf, inode_item, strlen(name) * 2 +
1339 btrfs_inode_size(leaf, inode_item));
1340 btrfs_mark_buffer_dirty(leaf);
1341 btrfs_release_path(root, &path);
1342 ret = btrfs_commit_transaction(trans, root);
1343 BUG_ON(ret);
1344 fail:
1345 btrfs_release_path(root, &path);
1346 return ret;
1347 }
1348
1349 struct btrfs_root *create_subvol(struct btrfs_root *root, const char *name)
1350 {
1351 int ret;
1352 u64 objectid;
1353 struct btrfs_key location;
1354 struct btrfs_root_item root_item;
1355 struct btrfs_trans_handle *trans;
1356 struct btrfs_fs_info *fs_info = root->fs_info;
1357 struct btrfs_root *tree_root = fs_info->tree_root;
1358 struct btrfs_root *new_root;
1359 struct extent_buffer *tmp;
1360
1361 trans = btrfs_start_transaction(root, 1);
1362 BUG_ON(!trans);
1363
1364 objectid = btrfs_super_root_dir(&fs_info->super_copy);
1365 ret = btrfs_find_free_objectid(trans, root, objectid, &objectid);
1366 if (ret)
1367 goto fail;
1368 ret = btrfs_copy_root(trans, root, root->node, &tmp, objectid);
1369 if (ret)
1370 goto fail;
1371 memcpy(&root_item, &root->root_item, sizeof(root_item));
1372 btrfs_set_root_bytenr(&root_item, tmp->start);
1373 btrfs_set_root_level(&root_item, btrfs_header_level(tmp));
1374 free_extent_buffer(tmp);
1375
1376 location.objectid = objectid;
1377 location.offset = 1;
1378 btrfs_set_key_type(&location, BTRFS_ROOT_ITEM_KEY);
1379 ret = btrfs_insert_root(trans, root->fs_info->tree_root,
1380 &location, &root_item);
1381 if (ret)
1382 goto fail;
1383 location.offset = (u64)-1;
1384 ret = btrfs_insert_dir_item(trans, tree_root, name, strlen(name),
1385 btrfs_super_root_dir(&fs_info->super_copy),
1386 &location, BTRFS_FT_DIR);
1387 if (ret)
1388 goto fail;
1389 ret = btrfs_insert_inode_ref(trans, tree_root, name, strlen(name),
1390 objectid,
1391 btrfs_super_root_dir(&fs_info->super_copy));
1392 if (ret)
1393 goto fail;
1394 ret = btrfs_commit_transaction(trans, root);
1395 BUG_ON(ret);
1396 new_root = btrfs_read_fs_root(fs_info, &location);
1397 if (!new_root || IS_ERR(new_root))
1398 goto fail;
1399 trans = btrfs_start_transaction(new_root, 1);
1400 BUG_ON(!trans);
1401 ret = btrfs_make_root_dir(trans, new_root, BTRFS_FIRST_FREE_OBJECTID);
1402 if (ret)
1403 goto fail;
1404 ret = btrfs_commit_transaction(trans, new_root);
1405 BUG_ON(ret);
1406 return new_root;
1407 fail:
1408 return NULL;
1409 }
1410 /*
1411 * Fixup block accounting. The initial block accounting created by
1412 * make_block_groups isn't accuracy in this case.
1413 */
1414 static int fixup_block_accounting(struct btrfs_trans_handle *trans,
1415 struct btrfs_root *root)
1416 {
1417 int ret;
1418 int slot;
1419 u64 start = 0;
1420 u64 bytes_used = 0;
1421 struct btrfs_path path;
1422 struct btrfs_key key;
1423 struct extent_buffer *leaf;
1424 struct btrfs_block_group_cache *cache;
1425 struct btrfs_fs_info *fs_info = root->fs_info;
1426
1427 while(1) {
1428 cache = btrfs_lookup_block_group(fs_info, start);
1429 if (!cache)
1430 break;
1431 start = cache->key.objectid + cache->key.offset;
1432 btrfs_set_block_group_used(&cache->item, 0);
1433 cache->space_info->bytes_used = 0;
1434 }
1435
1436 btrfs_init_path(&path);
1437 key.offset = 0;
1438 key.objectid = 0;
1439 btrfs_set_key_type(&key, BTRFS_EXTENT_ITEM_KEY);
1440 ret = btrfs_search_slot(trans, root->fs_info->extent_root,
1441 &key, &path, 0, 0);
1442 if (ret < 0)
1443 return ret;
1444 while(1) {
1445 leaf = path.nodes[0];
1446 slot = path.slots[0];
1447 if (slot >= btrfs_header_nritems(leaf)) {
1448 ret = btrfs_next_leaf(root, &path);
1449 if (ret < 0)
1450 return ret;
1451 if (ret > 0)
1452 break;
1453 leaf = path.nodes[0];
1454 slot = path.slots[0];
1455 }
1456 btrfs_item_key_to_cpu(leaf, &key, slot);
1457 if (key.type == BTRFS_EXTENT_ITEM_KEY) {
1458 bytes_used += key.offset;
1459 ret = btrfs_update_block_group(trans, root,
1460 key.objectid, key.offset, 1, 0);
1461 BUG_ON(ret);
1462 }
1463 path.slots[0]++;
1464 }
1465 btrfs_set_super_bytes_used(&root->fs_info->super_copy, bytes_used);
1466 btrfs_release_path(root, &path);
1467 return 0;
1468 }
1469
1470 static int create_chunk_mapping(struct btrfs_trans_handle *trans,
1471 struct btrfs_root *root)
1472 {
1473 struct btrfs_fs_info *info = root->fs_info;
1474 struct btrfs_root *chunk_root = info->chunk_root;
1475 struct btrfs_root *extent_root = info->extent_root;
1476 struct btrfs_device *device;
1477 struct btrfs_block_group_cache *cache;
1478 struct btrfs_dev_extent *extent;
1479 struct extent_buffer *leaf;
1480 struct btrfs_chunk chunk;
1481 struct btrfs_key key;
1482 struct btrfs_path path;
1483 u64 cur_start;
1484 u64 total_bytes;
1485 u64 chunk_objectid;
1486 int ret;
1487
1488 btrfs_init_path(&path);
1489
1490 total_bytes = btrfs_super_total_bytes(&root->fs_info->super_copy);
1491 chunk_objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID;
1492
1493 BUG_ON(list_empty(&info->fs_devices->devices));
1494 device = list_entry(info->fs_devices->devices.next,
1495 struct btrfs_device, dev_list);
1496 BUG_ON(device->devid != info->fs_devices->latest_devid);
1497
1498 /* delete device extent created by make_btrfs */
1499 key.objectid = device->devid;
1500 key.offset = 0;
1501 key.type = BTRFS_DEV_EXTENT_KEY;
1502 ret = btrfs_search_slot(trans, device->dev_root, &key, &path, -1, 1);
1503 if (ret < 0)
1504 goto err;
1505
1506 BUG_ON(ret > 0);
1507 ret = btrfs_del_item(trans, device->dev_root, &path);
1508 if (ret)
1509 goto err;
1510 btrfs_release_path(device->dev_root, &path);
1511
1512 /* delete chunk item created by make_btrfs */
1513 key.objectid = chunk_objectid;
1514 key.offset = 0;
1515 key.type = BTRFS_CHUNK_ITEM_KEY;
1516 ret = btrfs_search_slot(trans, chunk_root, &key, &path, -1, 1);
1517 if (ret < 0)
1518 goto err;
1519
1520 BUG_ON(ret > 0);
1521 ret = btrfs_del_item(trans, chunk_root, &path);
1522 if (ret)
1523 goto err;
1524 btrfs_release_path(chunk_root, &path);
1525
1526 /* for each block group, create device extent and chunk item */
1527 cur_start = 0;
1528 while (cur_start < total_bytes) {
1529 cache = btrfs_lookup_block_group(root->fs_info, cur_start);
1530 BUG_ON(!cache);
1531
1532 /* insert device extent */
1533 key.objectid = device->devid;
1534 key.offset = cache->key.objectid;
1535 key.type = BTRFS_DEV_EXTENT_KEY;
1536 ret = btrfs_insert_empty_item(trans, device->dev_root, &path,
1537 &key, sizeof(*extent));
1538 if (ret)
1539 goto err;
1540
1541 leaf = path.nodes[0];
1542 extent = btrfs_item_ptr(leaf, path.slots[0],
1543 struct btrfs_dev_extent);
1544
1545 btrfs_set_dev_extent_chunk_tree(leaf, extent,
1546 chunk_root->root_key.objectid);
1547 btrfs_set_dev_extent_chunk_objectid(leaf, extent,
1548 chunk_objectid);
1549 btrfs_set_dev_extent_chunk_offset(leaf, extent,
1550 cache->key.objectid);
1551 btrfs_set_dev_extent_length(leaf, extent, cache->key.offset);
1552 write_extent_buffer(leaf, root->fs_info->chunk_tree_uuid,
1553 (unsigned long)btrfs_dev_extent_chunk_tree_uuid(extent),
1554 BTRFS_UUID_SIZE);
1555 btrfs_mark_buffer_dirty(leaf);
1556 btrfs_release_path(device->dev_root, &path);
1557
1558 /* insert chunk item */
1559 btrfs_set_stack_chunk_length(&chunk, cache->key.offset);
1560 btrfs_set_stack_chunk_owner(&chunk,
1561 extent_root->root_key.objectid);
1562 btrfs_set_stack_chunk_stripe_len(&chunk, 64 * 1024);
1563 btrfs_set_stack_chunk_type(&chunk, cache->flags);
1564 btrfs_set_stack_chunk_io_align(&chunk, device->io_align);
1565 btrfs_set_stack_chunk_io_width(&chunk, device->io_width);
1566 btrfs_set_stack_chunk_sector_size(&chunk, device->sector_size);
1567 btrfs_set_stack_chunk_num_stripes(&chunk, 1);
1568 btrfs_set_stack_chunk_sub_stripes(&chunk, 0);
1569 btrfs_set_stack_stripe_devid(&chunk.stripe, device->devid);
1570 btrfs_set_stack_stripe_offset(&chunk.stripe,
1571 cache->key.objectid);
1572 memcpy(&chunk.stripe.dev_uuid, device->uuid, BTRFS_UUID_SIZE);
1573
1574 key.objectid = chunk_objectid;
1575 key.offset = cache->key.objectid;
1576 key.type = BTRFS_CHUNK_ITEM_KEY;
1577
1578 ret = btrfs_insert_item(trans, chunk_root, &key, &chunk,
1579 btrfs_chunk_item_size(1));
1580 if (ret)
1581 goto err;
1582
1583 cur_start = cache->key.objectid + cache->key.offset;
1584 }
1585
1586 device->bytes_used = total_bytes;
1587 ret = btrfs_update_device(trans, device);
1588 err:
1589 btrfs_release_path(device->dev_root, &path);
1590 return ret;
1591 }
1592
1593 static int init_btrfs(struct btrfs_root *root)
1594 {
1595 int ret;
1596 struct btrfs_key location;
1597 struct btrfs_trans_handle *trans;
1598 struct btrfs_fs_info *fs_info = root->fs_info;
1599
1600 trans = btrfs_start_transaction(root, 1);
1601 BUG_ON(!trans);
1602 ret = btrfs_make_block_groups(trans, root);
1603 if (ret)
1604 goto err;
1605 ret = fixup_block_accounting(trans, root);
1606 if (ret)
1607 goto err;
1608 ret = create_chunk_mapping(trans, root);
1609 if (ret)
1610 goto err;
1611 ret = btrfs_make_root_dir(trans, fs_info->tree_root,
1612 BTRFS_ROOT_TREE_DIR_OBJECTID);
1613 if (ret)
1614 goto err;
1615 memcpy(&location, &root->root_key, sizeof(location));
1616 location.offset = (u64)-1;
1617 ret = btrfs_insert_dir_item(trans, fs_info->tree_root, "default", 7,
1618 btrfs_super_root_dir(&fs_info->super_copy),
1619 &location, BTRFS_FT_DIR);
1620 if (ret)
1621 goto err;
1622 ret = btrfs_insert_inode_ref(trans, fs_info->tree_root, "default", 7,
1623 location.objectid,
1624 btrfs_super_root_dir(&fs_info->super_copy));
1625 if (ret)
1626 goto err;
1627 btrfs_set_root_dirid(&fs_info->fs_root->root_item,
1628 BTRFS_FIRST_FREE_OBJECTID);
1629 ret = btrfs_commit_transaction(trans, root);
1630 BUG_ON(ret);
1631 err:
1632 return ret;
1633 }
1634
1635 /*
1636 * Migrate super block to it's default position and zero 0 ~ 16k
1637 */
1638 static int migrate_super_block(int fd, u64 old_bytenr, u32 sectorsize)
1639 {
1640 int ret;
1641 struct extent_buffer *buf;
1642 struct btrfs_super_block *super;
1643 u32 len;
1644 u32 bytenr;
1645
1646 BUG_ON(sectorsize < sizeof(*super));
1647 buf = malloc(sizeof(*buf) + sectorsize);
1648 if (!buf)
1649 return -ENOMEM;
1650
1651 buf->len = sectorsize;
1652 ret = pread(fd, buf->data, sectorsize, old_bytenr);
1653 if (ret != sectorsize)
1654 goto fail;
1655
1656 super = (struct btrfs_super_block *)buf->data;
1657 BUG_ON(btrfs_super_bytenr(super) != old_bytenr);
1658 btrfs_set_super_bytenr(super, BTRFS_SUPER_INFO_OFFSET);
1659
1660 csum_tree_block(NULL, buf, 0);
1661 ret = pwrite(fd, buf->data, sectorsize, BTRFS_SUPER_INFO_OFFSET);
1662 if (ret != sectorsize)
1663 goto fail;
1664
1665 ret = fsync(fd);
1666 if (ret)
1667 goto fail;
1668
1669 memset(buf->data, 0, sectorsize);
1670 for (bytenr = 0; bytenr < BTRFS_SUPER_INFO_OFFSET; ) {
1671 len = BTRFS_SUPER_INFO_OFFSET - bytenr;
1672 if (len > sectorsize)
1673 len = sectorsize;
1674 ret = pwrite(fd, buf->data, len, bytenr);
1675 if (ret != len) {
1676 fprintf(stderr, "unable to zero fill device\n");
1677 break;
1678 }
1679 bytenr += len;
1680 }
1681 ret = 0;
1682 fsync(fd);
1683 fail:
1684 free(buf);
1685 if (ret > 0)
1686 ret = -1;
1687 return ret;
1688 }
1689
1690 static int prepare_system_chunk_sb(struct btrfs_super_block *super)
1691 {
1692 struct btrfs_chunk *chunk;
1693 struct btrfs_disk_key *key;
1694 u32 sectorsize = btrfs_super_sectorsize(super);
1695
1696 key = (struct btrfs_disk_key *)(super->sys_chunk_array);
1697 chunk = (struct btrfs_chunk *)(super->sys_chunk_array +
1698 sizeof(struct btrfs_disk_key));
1699
1700 btrfs_set_disk_key_objectid(key, BTRFS_FIRST_CHUNK_TREE_OBJECTID);
1701 btrfs_set_disk_key_type(key, BTRFS_CHUNK_ITEM_KEY);
1702 btrfs_set_disk_key_offset(key, 0);
1703
1704 btrfs_set_stack_chunk_length(chunk, btrfs_super_total_bytes(super));
1705 btrfs_set_stack_chunk_owner(chunk, BTRFS_EXTENT_TREE_OBJECTID);
1706 btrfs_set_stack_chunk_stripe_len(chunk, 64 * 1024);
1707 btrfs_set_stack_chunk_type(chunk, BTRFS_BLOCK_GROUP_SYSTEM);
1708 btrfs_set_stack_chunk_io_align(chunk, sectorsize);
1709 btrfs_set_stack_chunk_io_width(chunk, sectorsize);
1710 btrfs_set_stack_chunk_sector_size(chunk, sectorsize);
1711 btrfs_set_stack_chunk_num_stripes(chunk, 1);
1712 btrfs_set_stack_chunk_sub_stripes(chunk, 0);
1713 chunk->stripe.devid = super->dev_item.devid;
1714 chunk->stripe.offset = cpu_to_le64(0);
1715 memcpy(chunk->stripe.dev_uuid, super->dev_item.uuid, BTRFS_UUID_SIZE);
1716 btrfs_set_super_sys_array_size(super, sizeof(*key) + sizeof(*chunk));
1717 return 0;
1718 }
1719
1720 static int prepare_system_chunk(int fd, u64 sb_bytenr, u32 sectorsize)
1721 {
1722 int ret;
1723 struct extent_buffer *buf;
1724 struct btrfs_super_block *super;
1725
1726 BUG_ON(sectorsize < sizeof(*super));
1727 buf = malloc(sizeof(*buf) + sectorsize);
1728 if (!buf)
1729 return -ENOMEM;
1730
1731 buf->len = sectorsize;
1732 ret = pread(fd, buf->data, sectorsize, sb_bytenr);
1733 if (ret != sectorsize)
1734 goto fail;
1735
1736 super = (struct btrfs_super_block *)buf->data;
1737 BUG_ON(btrfs_super_bytenr(super) != sb_bytenr);
1738 BUG_ON(btrfs_super_num_devices(super) != 1);
1739
1740 ret = prepare_system_chunk_sb(super);
1741 if (ret)
1742 goto fail;
1743
1744 csum_tree_block(NULL, buf, 0);
1745 ret = pwrite(fd, buf->data, sectorsize, sb_bytenr);
1746 if (ret != sectorsize)
1747 goto fail;
1748
1749 ret = 0;
1750 fail:
1751 free(buf);
1752 if (ret > 0)
1753 ret = -1;
1754 return ret;
1755 }
1756
1757 static int relocate_one_reference(struct btrfs_trans_handle *trans,
1758 struct btrfs_root *root,
1759 u64 objectid, u64 offset,
1760 struct extent_io_tree *reloc_tree)
1761 {
1762 struct extent_buffer *leaf;
1763 struct btrfs_file_extent_item *fi;
1764 struct btrfs_key key;
1765 struct btrfs_path path;
1766 struct btrfs_inode_item inode;
1767 struct blk_iterate_data data;
1768 u64 bytenr;
1769 u64 num_bytes;
1770 u64 cur_offset;
1771 u64 extent_start;
1772 u64 extent_size;
1773 u64 new_pos;
1774 u64 nblocks;
1775 u64 root_gen;
1776 u64 root_owner;
1777 u64 sector_end;
1778 u32 sectorsize = root->sectorsize;
1779 unsigned long ptr;
1780 int ret;
1781
1782 btrfs_init_path(&path);
1783
1784 key.objectid = objectid;
1785 key.offset = 0;
1786 key.type = BTRFS_INODE_ITEM_KEY;
1787 ret = btrfs_lookup_inode(trans, root, &path, &key, 0);
1788 if (ret)
1789 goto fail;
1790
1791 leaf = path.nodes[0];
1792 ptr = btrfs_item_ptr_offset(leaf, path.slots[0]);
1793 read_extent_buffer(leaf, &inode, ptr, sizeof(inode));
1794 btrfs_release_path(root, &path);
1795
1796 ret = btrfs_lookup_file_extent(trans, root, &path,
1797 objectid, offset, -1);
1798 if (ret)
1799 goto fail;
1800
1801 leaf = path.nodes[0];
1802 fi = btrfs_item_ptr(leaf, path.slots[0],
1803 struct btrfs_file_extent_item);
1804
1805 root_gen = btrfs_header_generation(leaf);
1806 root_owner = btrfs_header_owner(leaf);
1807
1808 extent_start = btrfs_file_extent_disk_bytenr(leaf, fi);
1809 extent_size = btrfs_file_extent_disk_num_bytes(leaf, fi);
1810 bytenr = extent_start + btrfs_file_extent_offset(leaf, fi);
1811 num_bytes = btrfs_file_extent_num_bytes(leaf, fi);
1812
1813 ret = btrfs_del_item(trans, root, &path);
1814 if (ret)
1815 goto fail;
1816 btrfs_release_path(root, &path);
1817
1818 BUG_ON(num_bytes & (sectorsize - 1));
1819 nblocks = btrfs_stack_inode_nblocks(&inode) - num_bytes / 512;
1820 btrfs_set_stack_inode_nblocks(&inode, nblocks);
1821
1822 data = (struct blk_iterate_data) {
1823 .trans = trans,
1824 .root = root,
1825 .inode = &inode,
1826 .objectid = objectid,
1827 .first_block = offset / sectorsize,
1828 .disk_block = 0,
1829 .num_blocks = 0,
1830 .boundary = (u64)-1,
1831 .checksum = 0,
1832 .errcode = 0,
1833 };
1834
1835 cur_offset = offset;
1836 while (num_bytes > 0) {
1837 sector_end = bytenr + sectorsize - 1;
1838 if (test_range_bit(reloc_tree, bytenr, sector_end,
1839 EXTENT_LOCKED, 1)) {
1840 ret = get_state_private(reloc_tree, bytenr, &new_pos);
1841 BUG_ON(ret);
1842 } else {
1843 ret = custom_alloc_extent(root, sectorsize, 0, &key);
1844 if (ret)
1845 goto fail;
1846 new_pos = key.objectid;
1847 ret = copy_disk_extent(root, new_pos, bytenr,
1848 sectorsize);
1849 if (ret)
1850 goto fail;
1851 ret = set_extent_bits(reloc_tree, bytenr, sector_end,
1852 EXTENT_LOCKED, GFP_NOFS);
1853 BUG_ON(ret);
1854 ret = set_state_private(reloc_tree, bytenr, new_pos);
1855 BUG_ON(ret);
1856 }
1857
1858 ret = block_iterate_proc(NULL, new_pos / sectorsize,
1859 cur_offset / sectorsize, &data);
1860 if (ret & BLOCK_ABORT) {
1861 ret = data.errcode;
1862 goto fail;
1863 }
1864
1865 cur_offset += sectorsize;
1866 bytenr += sectorsize;
1867 num_bytes -= sectorsize;
1868 }
1869
1870 if (data.num_blocks > 0) {
1871 ret = record_file_blocks(trans, root, objectid, &inode,
1872 data.first_block, data.disk_block,
1873 data.num_blocks, 0);
1874 if (ret)
1875 goto fail;
1876 }
1877
1878 key.objectid = objectid;
1879 key.offset = 0;
1880 key.type = BTRFS_INODE_ITEM_KEY;
1881 ret = btrfs_lookup_inode(trans, root, &path, &key, 1);
1882 if (ret)
1883 goto fail;
1884
1885 leaf = path.nodes[0];
1886 ptr = btrfs_item_ptr_offset(leaf, path.slots[0]);
1887 write_extent_buffer(leaf, &inode, ptr, sizeof(inode));
1888 btrfs_mark_buffer_dirty(leaf);
1889 btrfs_release_path(root, &path);
1890
1891 ret = btrfs_free_extent(trans, root, extent_start, extent_size,
1892 root_owner, root_gen, objectid, offset, 0);
1893 fail:
1894 btrfs_release_path(root, &path);
1895 return ret;
1896 }
1897
1898 static int relocate_extents_range(struct btrfs_root *fs_root,
1899 struct btrfs_root *ext2_root,
1900 u64 start_byte, u64 end_byte)
1901 {
1902 struct btrfs_fs_info *info = fs_root->fs_info;
1903 struct btrfs_root *extent_root = info->extent_root;
1904 struct btrfs_root *cur_root;
1905 struct btrfs_trans_handle *trans;
1906 struct btrfs_extent_ref *ref_item;
1907 struct extent_buffer *leaf;
1908 struct btrfs_key key;
1909 struct btrfs_path path;
1910 struct extent_io_tree reloc_tree;
1911 u64 cur_byte;
1912 u64 num_bytes;
1913 u64 ref_root;
1914 u64 ref_owner;
1915 u64 ref_offset;
1916 u64 num_refs;
1917 int pass = 0;
1918 int ret;
1919 int found;
1920
1921 btrfs_init_path(&path);
1922 extent_io_tree_init(&reloc_tree);
1923 again:
1924 cur_root = (pass % 2 == 0) ? ext2_root : fs_root;
1925 num_refs = 0;
1926
1927 trans = btrfs_start_transaction(cur_root, 1);
1928 BUG_ON(!trans);
1929
1930 cur_byte = start_byte;
1931 while (1) {
1932 key.objectid = cur_byte;
1933 key.offset = 0;
1934 key.type = BTRFS_EXTENT_ITEM_KEY;
1935 ret = btrfs_search_slot(trans, extent_root,
1936 &key, &path, 0, 0);
1937 if (ret < 0)
1938 goto fail;
1939 next:
1940 leaf = path.nodes[0];
1941 if (path.slots[0] >= btrfs_header_nritems(leaf)) {
1942 ret = btrfs_next_leaf(extent_root, &path);
1943 if (ret < 0)
1944 goto fail;
1945 if (ret > 0)
1946 break;
1947 leaf = path.nodes[0];
1948 }
1949
1950 btrfs_item_key_to_cpu(leaf, &key, path.slots[0]);
1951 if (key.objectid < cur_byte ||
1952 key.type != BTRFS_EXTENT_ITEM_KEY) {
1953 path.slots[0]++;
1954 goto next;
1955 }
1956 if (key.objectid >= end_byte)
1957 break;
1958
1959 cur_byte = key.objectid;
1960 num_bytes = key.offset;
1961 found = 0;
1962 while (1) {
1963 if (path.slots[0] >= btrfs_header_nritems(leaf)) {
1964 ret = btrfs_next_leaf(extent_root, &path);
1965 if (ret > 0)
1966 break;
1967 if (ret < 0)
1968 goto fail;
1969 leaf = path.nodes[0];
1970 }
1971
1972 btrfs_item_key_to_cpu(leaf, &key, path.slots[0]);
1973 if (key.objectid != cur_byte)
1974 break;
1975 if (key.type != BTRFS_EXTENT_REF_KEY) {
1976 path.slots[0]++;
1977 continue;
1978 }
1979 ref_item = btrfs_item_ptr(leaf, path.slots[0],
1980 struct btrfs_extent_ref);
1981 ref_root = btrfs_ref_root(leaf, ref_item);
1982 ref_owner = btrfs_ref_objectid(leaf, ref_item);
1983 ref_offset = btrfs_ref_offset(leaf, ref_item);
1984 num_refs++;
1985
1986 BUG_ON(ref_owner < BTRFS_FIRST_FREE_OBJECTID);
1987 if (ref_root == cur_root->root_key.objectid)
1988 found = 1;
1989
1990 path.slots[0]++;
1991 }
1992 if (!found)
1993 goto next;
1994
1995 cur_byte += num_bytes;
1996 btrfs_release_path(extent_root, &path);
1997
1998 ret = relocate_one_reference(trans, cur_root, ref_owner,
1999 ref_offset, &reloc_tree);
2000 if (ret)
2001 goto fail;
2002
2003 if (trans->blocks_used >= 4096) {
2004 ret = btrfs_commit_transaction(trans, cur_root);
2005 BUG_ON(ret);
2006 trans = btrfs_start_transaction(cur_root, 1);
2007 BUG_ON(!trans);
2008 }
2009 }
2010 btrfs_release_path(cur_root, &path);
2011
2012 ret = btrfs_commit_transaction(trans, cur_root);
2013 BUG_ON(ret);
2014
2015 if (num_refs > 0 && pass++ < 4)
2016 goto again;
2017
2018 ret = (num_refs > 0) ? -1 : 0;
2019 fail:
2020 btrfs_release_path(cur_root, &path);
2021 extent_io_tree_cleanup(&reloc_tree);
2022 return ret;
2023 }
2024
2025 /*
2026 * relocate data in system chunk
2027 */
2028 static int cleanup_sys_chunk(struct btrfs_root *fs_root,
2029 struct btrfs_root *ext2_root)
2030 {
2031 struct btrfs_block_group_cache *cache;
2032 int ret = 0;
2033 u64 offset = 0;
2034 u64 end_byte;
2035
2036 while(1) {
2037 cache = btrfs_lookup_block_group(fs_root->fs_info, offset);
2038 if (!cache)
2039 break;
2040
2041 end_byte = cache->key.objectid + cache->key.offset;
2042 if (cache->flags & BTRFS_BLOCK_GROUP_SYSTEM) {
2043 ret = relocate_extents_range(fs_root, ext2_root,
2044 cache->key.objectid,
2045 end_byte);
2046 if (ret)
2047 goto fail;
2048 }
2049 offset = end_byte;
2050 }
2051 fail:
2052 return ret;
2053 }
2054
2055 static int fixup_chunk_mapping(struct btrfs_root *root)
2056 {
2057 struct btrfs_trans_handle *trans;
2058 struct btrfs_fs_info *info = root->fs_info;
2059 struct btrfs_root *chunk_root = info->chunk_root;
2060 struct extent_buffer *leaf;
2061 struct btrfs_key key;
2062 struct btrfs_path path;
2063 struct btrfs_chunk chunk;
2064 unsigned long ptr;
2065 u32 size;
2066 u64 type;
2067 int ret;
2068
2069 btrfs_init_path(&path);
2070
2071 trans = btrfs_start_transaction(root, 1);
2072 BUG_ON(!trans);
2073
2074 /*
2075 * recow the whole chunk tree. this will move all chunk tree blocks
2076 * into system block group.
2077 */
2078 memset(&key, 0, sizeof(key));
2079 while (1) {
2080 ret = btrfs_search_slot(trans, chunk_root, &key, &path, 0, 1);
2081 if (ret < 0)
2082 goto err;
2083
2084 ret = btrfs_next_leaf(chunk_root, &path);
2085 if (ret < 0)
2086 goto err;
2087 if (ret > 0)
2088 break;
2089
2090 btrfs_item_key_to_cpu(path.nodes[0], &key, path.slots[0]);
2091 btrfs_release_path(chunk_root, &path);
2092 }
2093 btrfs_release_path(chunk_root, &path);
2094
2095 /* fixup the system chunk array in super block */
2096 btrfs_set_super_sys_array_size(&info->super_copy, 0);
2097
2098 key.objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID;
2099 key.offset = 0;
2100 key.type = BTRFS_CHUNK_ITEM_KEY;
2101
2102 ret = btrfs_search_slot(trans, chunk_root, &key, &path, 0, 0);
2103 if (ret < 0)
2104 goto err;
2105 BUG_ON(ret != 0);
2106 while(1) {
2107 leaf = path.nodes[0];
2108 if (path.slots[0] >= btrfs_header_nritems(leaf)) {
2109 ret = btrfs_next_leaf(chunk_root, &path);
2110 if (ret < 0)
2111 goto err;
2112 if (ret > 0)
2113 break;
2114 leaf = path.nodes[0];
2115 }
2116 btrfs_item_key_to_cpu(leaf, &key, path.slots[0]);
2117 if (key.type != BTRFS_CHUNK_ITEM_KEY)
2118 goto next;
2119
2120 ptr = btrfs_item_ptr_offset(leaf, path.slots[0]);
2121 size = btrfs_item_size_nr(leaf, path.slots[0]);
2122 BUG_ON(size != sizeof(chunk));
2123 read_extent_buffer(leaf, &chunk, ptr, size);
2124 type = btrfs_stack_chunk_type(&chunk);
2125
2126 if (!(type & BTRFS_BLOCK_GROUP_SYSTEM))
2127 goto next;
2128
2129 ret = btrfs_add_system_chunk(trans, chunk_root, &key,
2130 &chunk, size);
2131 if (ret)
2132 goto err;
2133 next:
2134 path.slots[0]++;
2135 }
2136
2137 ret = btrfs_commit_transaction(trans, root);
2138 BUG_ON(ret);
2139 err:
2140 btrfs_release_path(chunk_root, &path);
2141 return ret;
2142 }
2143
2144 int do_convert(const char *devname, int datacsum, int packing, int noxattr)
2145 {
2146 int i, fd, ret;
2147 u32 blocksize;
2148 u64 blocks[6];
2149 u64 total_bytes;
2150 u64 super_bytenr;
2151 ext2_filsys ext2_fs;
2152 struct btrfs_root *root;
2153 struct btrfs_root *ext2_root;
2154
2155 ret = open_ext2fs(devname, &ext2_fs);
2156 if (ret) {
2157 fprintf(stderr, "unable to open the Ext2fs\n");
2158 goto fail;
2159 }
2160 blocksize = ext2_fs->blocksize;
2161 total_bytes = (u64)ext2_fs->super->s_blocks_count * blocksize;
2162 if (blocksize < 4096) {
2163 fprintf(stderr, "block size is too small\n");
2164 goto fail;
2165 }
2166 if (!(ext2_fs->super->s_feature_incompat &
2167 EXT2_FEATURE_INCOMPAT_FILETYPE)) {
2168 fprintf(stderr, "filetype feature is missing\n");
2169 goto fail;
2170 }
2171 for (i = 0; i < 6; i++) {
2172 ret = ext2_alloc_block(ext2_fs, 0, blocks + i);
2173 if (ret) {
2174 fprintf(stderr, "not enough free space\n");
2175 goto fail;
2176 }
2177 blocks[i] *= blocksize;
2178 }
2179 super_bytenr = blocks[0];
2180 fd = open(devname, O_RDWR);
2181 if (fd < 0) {
2182 fprintf(stderr, "unable to open %s\n", devname);
2183 goto fail;
2184 }
2185 ret = make_btrfs(fd, devname, ext2_fs->super->s_volume_name,
2186 blocks, total_bytes, blocksize, blocksize,
2187 blocksize, blocksize);
2188 if (ret) {
2189 fprintf(stderr, "unable to create initial ctree\n");
2190 goto fail;
2191 }
2192 /* create a system chunk that maps the whole device */
2193 ret = prepare_system_chunk(fd, super_bytenr, blocksize);
2194 if (ret) {
2195 fprintf(stderr, "unable to update system chunk\n");
2196 goto fail;
2197 }
2198 root = open_ctree_fd(fd, devname, super_bytenr);
2199 if (!root) {
2200 fprintf(stderr, "unable to open ctree\n");
2201 goto fail;
2202 }
2203
2204 root->fs_info->priv_data = ext2_fs;
2205 root->fs_info->extent_ops = &extent_ops;
2206 ret = init_btrfs(root);
2207 if (ret) {
2208 fprintf(stderr, "unable to setup the root tree\n");
2209 goto fail;
2210 }
2211 ext2_root = create_subvol(root, "ext2_saved");
2212 if (!ext2_root) {
2213 fprintf(stderr, "unable to create subvol\n");
2214 goto fail;
2215 }
2216 printf("creating btrfs metadata.\n");
2217 ret = copy_inodes(root, ext2_fs, datacsum, packing, noxattr);
2218 if (ret) {
2219 fprintf(stderr, "error during copy_inodes %d\n", ret);
2220 goto fail;
2221 }
2222 printf("creating ext2fs image file.\n");
2223 ret = create_ext2_image(ext2_root, ext2_fs, "image");
2224 if (ret) {
2225 fprintf(stderr, "error during create_ext2_image %d\n", ret);
2226 goto fail;
2227 }
2228 printf("cleaning up system chunk.\n");
2229 ret = cleanup_sys_chunk(root, ext2_root);
2230 if (ret) {
2231 fprintf(stderr, "error during cleanup_sys_chunk %d\n", ret);
2232 goto fail;
2233 }
2234 btrfs_free_fs_root(ext2_root->fs_info, ext2_root);
2235 ret = close_ctree(root);
2236 if (ret) {
2237 fprintf(stderr, "error during close_ctree %d\n", ret);
2238 goto fail;
2239 }
2240 close_ext2fs(ext2_fs);
2241
2242 /*
2243 * If this step succeed, we get a mountable btrfs. Otherwise
2244 * the ext2fs is left unchanged.
2245 */
2246 ret = migrate_super_block(fd, super_bytenr, blocksize);
2247 if (ret) {
2248 fprintf(stderr, "unable to migrate super block\n");
2249 goto fail;
2250 }
2251
2252 root = open_ctree_fd(fd, devname, 0);
2253 if (!root) {
2254 fprintf(stderr, "unable to open ctree\n");
2255 goto fail;
2256 }
2257 /* move chunk tree into system chunk. */
2258 ret = fixup_chunk_mapping(root);
2259 if (ret) {
2260 fprintf(stderr, "error during fixup_chunk_tree\n");
2261 goto fail;
2262 }
2263 ret = close_ctree(root);
2264 close(fd);
2265
2266 printf("conversion complete.\n");
2267 return 0;
2268 fail:
2269 fprintf(stderr, "conversion aborted.\n");
2270 return -1;
2271 }
2272
2273 static int may_rollback(struct btrfs_root *root)
2274 {
2275 struct btrfs_fs_info *info = root->fs_info;
2276 struct btrfs_multi_bio *multi = NULL;
2277 u64 bytenr;
2278 u64 length;
2279 u64 physical;
2280 u64 total_bytes;
2281 int num_stripes;
2282 int ret;
2283
2284 if (btrfs_super_num_devices(&info->super_copy) != 1)
2285 goto fail;
2286
2287 bytenr = BTRFS_SUPER_INFO_OFFSET;
2288 total_bytes = btrfs_super_total_bytes(&root->fs_info->super_copy);
2289
2290 while (1) {
2291 ret = btrfs_map_block(&info->mapping_tree, WRITE, bytenr,
2292 &length, &multi, 0);
2293 if (ret)
2294 goto fail;
2295
2296 num_stripes = multi->num_stripes;
2297 physical = multi->stripes[0].physical;
2298 kfree(multi);
2299
2300 if (num_stripes != 1 || physical != bytenr)
2301 goto fail;
2302
2303 bytenr += length;
2304 if (bytenr >= total_bytes)
2305 break;
2306 }
2307 return 0;
2308 fail:
2309 return -1;
2310 }
2311
2312 int do_rollback(const char *devname, int force)
2313 {
2314 int fd;
2315 int ret;
2316 struct btrfs_root *root;
2317 struct btrfs_root *ext2_root;
2318 struct btrfs_root *chunk_root;
2319 struct btrfs_dir_item *dir;
2320 struct btrfs_inode_item *inode;
2321 struct btrfs_file_extent_item *fi;
2322 struct btrfs_trans_handle *trans;
2323 struct extent_buffer *leaf;
2324 struct btrfs_block_group_cache *cache1;
2325 struct btrfs_block_group_cache *cache2;
2326 struct btrfs_key key;
2327 struct btrfs_path path;
2328 struct extent_io_tree io_tree;
2329 char *buf;
2330 char *name;
2331 u64 bytenr;
2332 u64 num_bytes;
2333 u64 root_dir;
2334 u64 objectid;
2335 u64 offset;
2336 u64 start;
2337 u64 end;
2338 u64 sb_bytenr;
2339 u64 first_free;
2340 u64 total_bytes;
2341 u32 sectorsize;
2342
2343 extent_io_tree_init(&io_tree);
2344
2345 fd = open(devname, O_RDWR);
2346 if (fd < 0) {
2347 fprintf(stderr, "unable to open %s\n", devname);
2348 goto fail;
2349 }
2350 root = open_ctree_fd(fd, devname, 0);
2351 if (!root) {
2352 fprintf(stderr, "unable to open ctree\n");
2353 goto fail;
2354 }
2355 ret = may_rollback(root);
2356 if (ret < 0) {
2357 fprintf(stderr, "unable to do rollback\n");
2358 goto fail;
2359 }
2360
2361 sectorsize = root->sectorsize;
2362 buf = malloc(sectorsize);
2363 if (!buf) {
2364 fprintf(stderr, "unable to allocate memory\n");
2365 goto fail;
2366 }
2367
2368 btrfs_init_path(&path);
2369 name = "ext2_saved";
2370 root_dir = btrfs_super_root_dir(&root->fs_info->super_copy);
2371 dir = btrfs_lookup_dir_item(NULL, root->fs_info->tree_root, &path,
2372 root_dir, name, strlen(name), 0);
2373 if (!dir || IS_ERR(dir)) {
2374 fprintf(stderr, "unable to find subvol %s\n", name);
2375 goto fail;
2376 }
2377 leaf = path.nodes[0];
2378 btrfs_dir_item_key_to_cpu(leaf, dir, &key);
2379 btrfs_release_path(root->fs_info->tree_root, &path);
2380
2381 ext2_root = btrfs_read_fs_root(root->fs_info, &key);
2382 if (!ext2_root || IS_ERR(ext2_root)) {
2383 fprintf(stderr, "unable to open subvol %s\n", name);
2384 goto fail;
2385 }
2386
2387 name = "image";
2388 root_dir = btrfs_root_dirid(&root->root_item);
2389 dir = btrfs_lookup_dir_item(NULL, ext2_root, &path,
2390 root_dir, name, strlen(name), 0);
2391 if (!dir || IS_ERR(dir)) {
2392 fprintf(stderr, "unable to find file %s\n", name);
2393 goto fail;
2394 }
2395 leaf = path.nodes[0];
2396 btrfs_dir_item_key_to_cpu(leaf, dir, &key);
2397 btrfs_release_path(ext2_root, &path);
2398
2399 objectid = key.objectid;
2400
2401 ret = btrfs_lookup_inode(NULL, ext2_root, &path, &key, 0);
2402 if (ret) {
2403 fprintf(stderr, "unable to find inode item\n");
2404 goto fail;
2405 }
2406 leaf = path.nodes[0];
2407 inode = btrfs_item_ptr(leaf, path.slots[0], struct btrfs_inode_item);
2408 total_bytes = btrfs_inode_size(leaf, inode);
2409 btrfs_release_path(ext2_root, &path);
2410
2411 key.objectid = objectid;
2412 key.offset = 0;
2413 btrfs_set_key_type(&key, BTRFS_EXTENT_DATA_KEY);
2414 ret = btrfs_search_slot(NULL, ext2_root, &key, &path, 0, 0);
2415 if (ret != 0) {
2416 fprintf(stderr, "unable to find first file extent\n");
2417 btrfs_release_path(ext2_root, &path);
2418 goto fail;
2419 }
2420
2421 /* build mapping tree for the relocated blocks */
2422 for (offset = 0; offset < total_bytes; ) {
2423 leaf = path.nodes[0];
2424 if (path.slots[0] >= btrfs_header_nritems(leaf)) {
2425 ret = btrfs_next_leaf(root, &path);
2426 if (ret != 0)
2427 break;
2428 continue;
2429 }
2430
2431 btrfs_item_key_to_cpu(leaf, &key, path.slots[0]);
2432 if (key.objectid != objectid || key.offset != offset ||
2433 btrfs_key_type(&key) != BTRFS_EXTENT_DATA_KEY)
2434 break;
2435
2436 fi = btrfs_item_ptr(leaf, path.slots[0],
2437 struct btrfs_file_extent_item);
2438 if (btrfs_file_extent_type(leaf, fi) != BTRFS_FILE_EXTENT_REG)
2439 break;
2440
2441 bytenr = btrfs_file_extent_disk_bytenr(leaf, fi);
2442 /* skip holes and direct mapped extents */
2443 if (bytenr == 0 || bytenr == offset)
2444 goto next_extent;
2445
2446 bytenr += btrfs_file_extent_offset(leaf, fi);
2447 num_bytes = btrfs_file_extent_num_bytes(leaf, fi);
2448
2449 cache1 = btrfs_lookup_block_group(root->fs_info, offset);
2450 cache2 = btrfs_lookup_block_group(root->fs_info,
2451 offset + num_bytes - 1);
2452 if (!cache1 || cache1 != cache2 ||
2453 !(cache1->flags & BTRFS_BLOCK_GROUP_SYSTEM))
2454 break;
2455
2456 set_extent_bits(&io_tree, offset, offset + num_bytes - 1,
2457 EXTENT_LOCKED, GFP_NOFS);
2458 set_state_private(&io_tree, offset, bytenr);
2459 next_extent:
2460 offset += btrfs_file_extent_num_bytes(leaf, fi);
2461 path.slots[0]++;
2462 }
2463 btrfs_release_path(ext2_root, &path);
2464 btrfs_free_fs_root(ext2_root->fs_info, ext2_root);
2465
2466 if (offset < total_bytes) {
2467 fprintf(stderr, "unable to build extent mapping\n");
2468 goto fail;
2469 }
2470
2471 first_free = BTRFS_SUPER_INFO_OFFSET + 2 * sectorsize - 1;
2472 first_free &= ~((u64)sectorsize - 1);
2473 /* backup for extent #0 should exist */
2474 if(!test_range_bit(&io_tree, 0, first_free - 1, EXTENT_LOCKED, 1)) {
2475 fprintf(stderr, "no backup for the first extent\n");
2476 goto fail;
2477 }
2478 /* force no allocation from system block group */
2479 root->fs_info->system_allocs = -1;
2480 trans = btrfs_start_transaction(root, 1);
2481 BUG_ON(!trans);
2482 /*
2483 * recow the whole chunk tree, this will remove all chunk tree blocks
2484 * from system block group
2485 */
2486 chunk_root = root->fs_info->chunk_root;
2487 memset(&key, 0, sizeof(key));
2488 while (1) {
2489 ret = btrfs_search_slot(trans, chunk_root, &key, &path, 0, 1);
2490 if (ret < 0)
2491 break;
2492
2493 ret = btrfs_next_leaf(chunk_root, &path);
2494 if (ret)
2495 break;
2496
2497 btrfs_item_key_to_cpu(path.nodes[0], &key, path.slots[0]);
2498 btrfs_release_path(chunk_root, &path);
2499 }
2500 btrfs_release_path(chunk_root, &path);
2501
2502 offset = 0;
2503 num_bytes = 0;
2504 while(1) {
2505 cache1 = btrfs_lookup_block_group(root->fs_info, offset);
2506 if (!cache1)
2507 break;
2508
2509 if (cache1->flags & BTRFS_BLOCK_GROUP_SYSTEM)
2510 num_bytes += btrfs_block_group_used(&cache1->item);
2511
2512 offset = cache1->key.objectid + cache1->key.offset;
2513 }
2514 /* only extent #0 left in system block group? */
2515 if (num_bytes > first_free) {
2516 fprintf(stderr, "unable to empty system block group\n");
2517 goto fail;
2518 }
2519 /* create a system chunk that maps the whole device */
2520 ret = prepare_system_chunk_sb(&root->fs_info->super_copy);
2521 if (ret) {
2522 fprintf(stderr, "unable to update system chunk\n");
2523 goto fail;
2524 }
2525
2526 ret = btrfs_commit_transaction(trans, root);
2527 BUG_ON(ret);
2528
2529 ret = close_ctree(root);
2530 if (ret) {
2531 fprintf(stderr, "error during close_ctree %d\n", ret);
2532 goto fail;
2533 }
2534
2535 sb_bytenr = (u64)-1;
2536 /* copy all relocated blocks back */
2537 while(1) {
2538 ret = find_first_extent_bit(&io_tree, 0, &start, &end,
2539 EXTENT_LOCKED);
2540 if (ret)
2541 break;
2542
2543 ret = get_state_private(&io_tree, start, &bytenr);
2544 BUG_ON(ret);
2545
2546 clear_extent_bits(&io_tree, start, end, EXTENT_LOCKED,
2547 GFP_NOFS);
2548
2549 while (start <= end) {
2550 if (start == BTRFS_SUPER_INFO_OFFSET) {
2551 sb_bytenr = bytenr;
2552 goto next_sector;
2553 }
2554 ret = pread(fd, buf, sectorsize, bytenr);
2555 if (ret < 0) {
2556 fprintf(stderr, "error during pread %d\n", ret);
2557 goto fail;
2558 }
2559 BUG_ON(ret != sectorsize);
2560 ret = pwrite(fd, buf, sectorsize, start);
2561 if (ret < 0) {
2562 fprintf(stderr, "error during pwrite %d\n", ret);
2563 goto fail;
2564 }
2565 BUG_ON(ret != sectorsize);
2566 next_sector:
2567 start += sectorsize;
2568 bytenr += sectorsize;
2569 }
2570 }
2571
2572 ret = fsync(fd);
2573 if (ret) {
2574 fprintf(stderr, "error during fsync %d\n", ret);
2575 goto fail;
2576 }
2577 /*
2578 * finally, overwrite btrfs super block.
2579 */
2580 ret = pread(fd, buf, sectorsize, sb_bytenr);
2581 if (ret < 0) {
2582 fprintf(stderr, "error during pread %d\n", ret);
2583 goto fail;
2584 }
2585 BUG_ON(ret != sectorsize);
2586 ret = pwrite(fd, buf, sectorsize, BTRFS_SUPER_INFO_OFFSET);
2587 if (ret < 0) {
2588 fprintf(stderr, "error during pwrite %d\n", ret);
2589 goto fail;
2590 }
2591 BUG_ON(ret != sectorsize);
2592 ret = fsync(fd);
2593 if (ret) {
2594 fprintf(stderr, "error during fsync %d\n", ret);
2595 goto fail;
2596 }
2597
2598 close(fd);
2599 free(buf);
2600 extent_io_tree_cleanup(&io_tree);
2601 printf("rollback complete.\n");
2602 return 0;
2603 fail:
2604 fprintf(stderr, "rollback aborted.\n");
2605 return -1;
2606 }
2607
2608 static void print_usage(void)
2609 {
2610 printf("usage: btrfs-convert [-d] [-i] [-n] [-r] device\n");
2611 printf("\t-d disable data checksum\n");
2612 printf("\t-i ignore xattrs and ACLs\n");
2613 printf("\t-n disable packing of small files\n");
2614 printf("\t-r roll back to ext2fs\n");
2615 }
2616
2617 int main(int argc, char *argv[])
2618 {
2619 int ret;
2620 int packing = 1;
2621 int noxattr = 0;
2622 int datacsum = 1;
2623 int rollback = 0;
2624 char *file;
2625 while(1) {
2626 int c = getopt(argc, argv, "dinr");
2627 if (c < 0)
2628 break;
2629 switch(c) {
2630 case 'd':
2631 datacsum = 0;
2632 break;
2633 case 'i':
2634 noxattr = 1;
2635 break;
2636 case 'n':
2637 packing = 0;
2638 break;
2639 case 'r':
2640 rollback = 1;
2641 break;
2642 default:
2643 print_usage();
2644 }
2645 }
2646 argc = argc - optind;
2647 if (argc != 1) {
2648 print_usage();
2649 return 1;
2650 }
2651
2652 file = argv[optind];
2653 if (check_mounted(file)) {
2654 fprintf(stderr, "%s is mounted\n", file);
2655 return 1;
2656 }
2657
2658 if (rollback) {
2659 ret = do_rollback(file, 0);
2660 } else {
2661 ret = do_convert(file, datacsum, packing, noxattr);
2662 }
2663 if (ret)
2664 return 1;
2665 return 0;
2666 }
(deprecated) Btrfs progs developments and patch integration