disk-io.c

   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 600
  20 #define __USE_XOPEN2K
  21 #define _GNU_SOURCE 1
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <sys/types.h>
  25 #include <sys/stat.h>
  26 #include <fcntl.h>
  27 #include <unistd.h>
  28 #include "kerncompat.h"
  29 #include "radix-tree.h"
  30 #include "ctree.h"
  31 #include "disk-io.h"
  32 #include "volumes.h"
  33 #include "transaction.h"
  34 #include "crc32c.h"
  35 #include "utils.h"
  36 #include "print-tree.h"
  37
  38 static int check_tree_block(struct btrfs_root *root, struct extent_buffer *buf)
  39 {
  40
  41         struct btrfs_fs_devices *fs_devices;
  42         int ret = 1;
  43
  44         if (buf->start != btrfs_header_bytenr(buf)) {
  45                 printk("Check tree block failed, want=%Lu, have=%Lu\n",
  46                        buf->start, btrfs_header_bytenr(buf));
  47                 return ret;
  48         }
  49
  50         fs_devices = root->fs_info->fs_devices;
  51         while (fs_devices) {
  52                 if (!memcmp_extent_buffer(buf, fs_devices->fsid,
  53                                           (unsigned long)btrfs_header_fsid(buf),
  54                                           BTRFS_FSID_SIZE)) {
  55                         ret = 0;
  56                         break;
  57                 }
  58                 fs_devices = fs_devices->seed;
  59         }
  60         return ret;
  61 }
  62
  63 u32 btrfs_csum_data(struct btrfs_root *root, char *data, u32 seed, size_t len)
  64 {
  65         return crc32c(seed, data, len);
  66 }
  67
  68 void btrfs_csum_final(u32 crc, char *result)
  69 {
  70         *(__le32 *)result = ~cpu_to_le32(crc);
  71 }
  72
  73 int csum_tree_block_size(struct extent_buffer *buf, u16 csum_size,
  74                          int verify)
  75 {
  76         char *result;
  77         u32 len;
  78         u32 crc = ~(u32)0;
  79
  80         result = malloc(csum_size * sizeof(char));
  81         if (!result)
  82                 return 1;
  83
  84         len = buf->len - BTRFS_CSUM_SIZE;
  85         crc = crc32c(crc, buf->data + BTRFS_CSUM_SIZE, len);
  86         btrfs_csum_final(crc, result);
  87
  88         if (verify) {
  89                 if (memcmp_extent_buffer(buf, result, 0, csum_size)) {
  90                         printk("checksum verify failed on %llu found %08X "
  91                                "wanted %08X\n", (unsigned long long)buf->start,
  92                                *((u32 *)result), *((u32*)(char *)buf->data));
  93                         free(result);
  94                         return 1;
  95                 }
  96         } else {
  97                 write_extent_buffer(buf, result, 0, csum_size);
  98         }
  99         free(result);
 100         return 0;
 101 }
 102
 103 int csum_tree_block(struct btrfs_root *root, struct extent_buffer *buf,
 104                     int verify)
 105 {
 106         u16 csum_size =
 107                 btrfs_super_csum_size(root->fs_info->super_copy);
 108         return csum_tree_block_size(buf, csum_size, verify);
 109 }
 110
 111 struct extent_buffer *btrfs_find_tree_block(struct btrfs_root *root,
 112                                             u64 bytenr, u32 blocksize)
 113 {
 114         return find_extent_buffer(&root->fs_info->extent_cache,
 115                                   bytenr, blocksize);
 116 }
 117
 118 struct extent_buffer *btrfs_find_create_tree_block(struct btrfs_root *root,
 119                                                  u64 bytenr, u32 blocksize)
 120 {
 121         return alloc_extent_buffer(&root->fs_info->extent_cache, bytenr,
 122                                    blocksize);
 123 }
 124
 125 int readahead_tree_block(struct btrfs_root *root, u64 bytenr, u32 blocksize,
 126                          u64 parent_transid)
 127 {
 128         int ret;
 129         struct extent_buffer *eb;
 130         u64 length;
 131         struct btrfs_multi_bio *multi = NULL;
 132         struct btrfs_device *device;
 133
 134         eb = btrfs_find_tree_block(root, bytenr, blocksize);
 135         if (eb && btrfs_buffer_uptodate(eb, parent_transid)) {
 136                 free_extent_buffer(eb);
 137                 return 0;
 138         }
 139
 140         length = blocksize;
 141         ret = btrfs_map_block(&root->fs_info->mapping_tree, READ,
 142                               bytenr, &length, &multi, 0, NULL);
 143         BUG_ON(ret);
 144         device = multi->stripes[0].dev;
 145         device->total_ios++;
 146         blocksize = min(blocksize, (u32)(64 * 1024));
 147         readahead(device->fd, multi->stripes[0].physical, blocksize);
 148         kfree(multi);
 149         return 0;
 150 }
 151
 152 static int verify_parent_transid(struct extent_io_tree *io_tree,
 153                                  struct extent_buffer *eb, u64 parent_transid,
 154                                  int ignore)
 155 {
 156         int ret;
 157
 158         if (!parent_transid || btrfs_header_generation(eb) == parent_transid)
 159                 return 0;
 160
 161         if (extent_buffer_uptodate(eb) &&
 162             btrfs_header_generation(eb) == parent_transid) {
 163                 ret = 0;
 164                 goto out;
 165         }
 166         printk("parent transid verify failed on %llu wanted %llu found %llu\n",
 167                (unsigned long long)eb->start,
 168                (unsigned long long)parent_transid,
 169                (unsigned long long)btrfs_header_generation(eb));
 170         if (ignore) {
 171                 printk("Ignoring transid failure\n");
 172                 return 0;
 173         }
 174
 175         ret = 1;
 176 out:
 177         clear_extent_buffer_uptodate(io_tree, eb);
 178         return ret;
 179
 180 }
 181
 182
 183 static int read_whole_eb(struct btrfs_fs_info *info, struct extent_buffer *eb, int mirror)
 184 {
 185         unsigned long offset = 0;
 186         struct btrfs_multi_bio *multi = NULL;
 187         struct btrfs_device *device;
 188         int ret = 0;
 189         u64 read_len;
 190         unsigned long bytes_left = eb->len;
 191
 192         while (bytes_left) {
 193                 read_len = bytes_left;
 194                 ret = btrfs_map_block(&info->mapping_tree, READ,
 195                                       eb->start + offset, &read_len, &multi,
 196                                       mirror, NULL);
 197                 if (ret) {
 198                         printk("Couldn't map the block %Lu\n", eb->start + offset);
 199                         kfree(multi);
 200                         return -EIO;
 201                 }
 202                 device = multi->stripes[0].dev;
 203
 204                 if (device->fd == 0) {
 205                         kfree(multi);
 206                         return -EIO;
 207                 }
 208
 209                 eb->fd = device->fd;
 210                 device->total_ios++;
 211                 eb->dev_bytenr = multi->stripes[0].physical;
 212                 kfree(multi);
 213                 multi = NULL;
 214
 215                 if (read_len > bytes_left)
 216                         read_len = bytes_left;
 217
 218                 ret = read_extent_from_disk(eb, offset, read_len);
 219                 if (ret)
 220                         return -EIO;
 221                 offset += read_len;
 222                 bytes_left -= read_len;
 223         }
 224         return 0;
 225 }
 226
 227 struct extent_buffer *read_tree_block(struct btrfs_root *root, u64 bytenr,
 228                                      u32 blocksize, u64 parent_transid)
 229 {
 230         int ret;
 231         struct extent_buffer *eb;
 232         u64 best_transid = 0;
 233         int mirror_num = 0;
 234         int good_mirror = 0;
 235         int num_copies;
 236         int ignore = 0;
 237
 238         eb = btrfs_find_create_tree_block(root, bytenr, blocksize);
 239         if (!eb)
 240                 return NULL;
 241
 242         if (btrfs_buffer_uptodate(eb, parent_transid))
 243                 return eb;
 244
 245         while (1) {
 246                 ret = read_whole_eb(root->fs_info, eb, mirror_num);
 247                 if (ret == 0 && check_tree_block(root, eb) == 0 &&
 248                     csum_tree_block(root, eb, 1) == 0 &&
 249                     verify_parent_transid(eb->tree, eb, parent_transid, ignore)
 250                     == 0) {
 251                         btrfs_set_buffer_uptodate(eb);
 252                         return eb;
 253                 }
 254                 if (ignore) {
 255                         if (check_tree_block(root, eb))
 256                                 printk("read block failed check_tree_block\n");
 257                         else
 258                                 printk("Csum didn't match\n");
 259                         break;
 260                 }
 261                 num_copies = btrfs_num_copies(&root->fs_info->mapping_tree,
 262                                               eb->start, eb->len);
 263                 if (num_copies == 1) {
 264                         ignore = 1;
 265                         continue;
 266                 }
 267                 if (btrfs_header_generation(eb) > best_transid) {
 268                         best_transid = btrfs_header_generation(eb);
 269                         good_mirror = mirror_num;
 270                 }
 271                 mirror_num++;
 272                 if (mirror_num > num_copies) {
 273                         mirror_num = good_mirror;
 274                         ignore = 1;
 275                         continue;
 276                 }
 277         }
 278         free_extent_buffer(eb);
 279         return NULL;
 280 }
 281
 282 static int rmw_eb(struct btrfs_fs_info *info,
 283                   struct extent_buffer *eb, struct extent_buffer *orig_eb)
 284 {
 285         int ret;
 286         unsigned long orig_off = 0;
 287         unsigned long dest_off = 0;
 288         unsigned long copy_len = eb->len;
 289
 290         ret = read_whole_eb(info, eb, 0);
 291         if (ret)
 292                 return ret;
 293
 294         if (eb->start + eb->len <= orig_eb->start ||
 295             eb->start >= orig_eb->start + orig_eb->len)
 296                 return 0;
 297         /*
 298          * | ----- orig_eb ------- |
 299          *         | ----- stripe -------  |
 300          *         | ----- orig_eb ------- |
 301          *              | ----- orig_eb ------- |
 302          */
 303         if (eb->start > orig_eb->start)
 304                 orig_off = eb->start - orig_eb->start;
 305         if (orig_eb->start > eb->start)
 306                 dest_off = orig_eb->start - eb->start;
 307
 308         if (copy_len > orig_eb->len - orig_off)
 309                 copy_len = orig_eb->len - orig_off;
 310         if (copy_len > eb->len - dest_off)
 311                 copy_len = eb->len - dest_off;
 312
 313         memcpy(eb->data + dest_off, orig_eb->data + orig_off, copy_len);
 314         return 0;
 315 }
 316
 317 static void split_eb_for_raid56(struct btrfs_fs_info *info,
 318                                 struct extent_buffer *orig_eb,
 319                                struct extent_buffer **ebs,
 320                                u64 stripe_len, u64 *raid_map,
 321                                int num_stripes)
 322 {
 323         struct extent_buffer *eb;
 324         u64 start = orig_eb->start;
 325         u64 this_eb_start;
 326         int i;
 327         int ret;
 328
 329         for (i = 0; i < num_stripes; i++) {
 330                 if (raid_map[i] >= BTRFS_RAID5_P_STRIPE)
 331                         break;
 332
 333                 eb = malloc(sizeof(struct extent_buffer) + stripe_len);
 334                 if (!eb)
 335                         BUG();
 336                 memset(eb, 0, sizeof(struct extent_buffer) + stripe_len);
 337
 338                 eb->start = raid_map[i];
 339                 eb->len = stripe_len;
 340                 eb->refs = 1;
 341                 eb->flags = 0;
 342                 eb->fd = -1;
 343                 eb->dev_bytenr = (u64)-1;
 344
 345                 this_eb_start = raid_map[i];
 346
 347                 if (start > this_eb_start ||
 348                     start + orig_eb->len < this_eb_start + stripe_len) {
 349                         ret = rmw_eb(info, eb, orig_eb);
 350                         BUG_ON(ret);
 351                 } else {
 352                         memcpy(eb->data, orig_eb->data + eb->start - start, stripe_len);
 353                 }
 354                 ebs[i] = eb;
 355         }
 356 }
 357
 358 static int write_raid56_with_parity(struct btrfs_fs_info *info,
 359                                     struct extent_buffer *eb,
 360                                     struct btrfs_multi_bio *multi,
 361                                     u64 stripe_len, u64 *raid_map)
 362 {
 363         struct extent_buffer *ebs[multi->num_stripes], *p_eb = NULL, *q_eb = NULL;
 364         int i;
 365         int j;
 366         int ret;
 367         int alloc_size = eb->len;
 368
 369         if (stripe_len > alloc_size)
 370                 alloc_size = stripe_len;
 371
 372         split_eb_for_raid56(info, eb, ebs, stripe_len, raid_map,
 373                             multi->num_stripes);
 374
 375         for (i = 0; i < multi->num_stripes; i++) {
 376                 struct extent_buffer *new_eb;
 377                 if (raid_map[i] < BTRFS_RAID5_P_STRIPE) {
 378                         ebs[i]->dev_bytenr = multi->stripes[i].physical;
 379                         ebs[i]->fd = multi->stripes[i].dev->fd;
 380                         multi->stripes[i].dev->total_ios++;
 381                         BUG_ON(ebs[i]->start != raid_map[i]);
 382                         continue;
 383                 }
 384                 new_eb = kmalloc(sizeof(*eb) + alloc_size, GFP_NOFS);
 385                 BUG_ON(!new_eb);
 386                 new_eb->dev_bytenr = multi->stripes[i].physical;
 387                 new_eb->fd = multi->stripes[i].dev->fd;
 388                 multi->stripes[i].dev->total_ios++;
 389                 new_eb->len = stripe_len;
 390
 391                 if (raid_map[i] == BTRFS_RAID5_P_STRIPE)
 392                         p_eb = new_eb;
 393                 else if (raid_map[i] == BTRFS_RAID6_Q_STRIPE)
 394                         q_eb = new_eb;
 395         }
 396         if (q_eb) {
 397                 void *pointers[multi->num_stripes];
 398                 ebs[multi->num_stripes - 2] = p_eb;
 399                 ebs[multi->num_stripes - 1] = q_eb;
 400
 401                 for (i = 0; i < multi->num_stripes; i++)
 402                         pointers[i] = ebs[i]->data;
 403
 404                 raid6_gen_syndrome(multi->num_stripes, stripe_len, pointers);
 405         } else {
 406                 ebs[multi->num_stripes - 1] = p_eb;
 407                 memcpy(p_eb->data, ebs[0]->data, stripe_len);
 408                 for (j = 1; j < multi->num_stripes - 1; j++) {
 409                         for (i = 0; i < stripe_len; i += sizeof(unsigned long)) {
 410                                 *(unsigned long *)(p_eb->data + i) ^=
 411                                         *(unsigned long *)(ebs[j]->data + i);
 412                         }
 413                 }
 414         }
 415
 416         for (i = 0; i < multi->num_stripes; i++) {
 417                 ret = write_extent_to_disk(ebs[i]);
 418                 BUG_ON(ret);
 419                 if (ebs[i] != eb)
 420                         kfree(ebs[i]);
 421         }
 422         return 0;
 423 }
 424
 425 int write_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root,
 426                      struct extent_buffer *eb)
 427 {
 428         int ret;
 429         int dev_nr;
 430         u64 length;
 431         u64 *raid_map = NULL;
 432         struct btrfs_multi_bio *multi = NULL;
 433
 434         if (check_tree_block(root, eb))
 435                 BUG();
 436         if (!btrfs_buffer_uptodate(eb, trans->transid))
 437                 BUG();
 438
 439         btrfs_set_header_flag(eb, BTRFS_HEADER_FLAG_WRITTEN);
 440         csum_tree_block(root, eb, 0);
 441
 442         dev_nr = 0;
 443         length = eb->len;
 444         ret = btrfs_map_block(&root->fs_info->mapping_tree, WRITE,
 445                               eb->start, &length, &multi, 0, &raid_map);
 446
 447         if (raid_map) {
 448                 ret = write_raid56_with_parity(root->fs_info, eb, multi,
 449                                                length, raid_map);
 450                 BUG_ON(ret);
 451         } else while (dev_nr < multi->num_stripes) {
 452                 BUG_ON(ret);
 453                 eb->fd = multi->stripes[dev_nr].dev->fd;
 454                 eb->dev_bytenr = multi->stripes[dev_nr].physical;
 455                 multi->stripes[dev_nr].dev->total_ios++;
 456                 dev_nr++;
 457                 ret = write_extent_to_disk(eb);
 458                 BUG_ON(ret);
 459         }
 460         kfree(multi);
 461         return 0;
 462 }
 463
 464 int __setup_root(u32 nodesize, u32 leafsize, u32 sectorsize,
 465                         u32 stripesize, struct btrfs_root *root,
 466                         struct btrfs_fs_info *fs_info, u64 objectid)
 467 {
 468         root->node = NULL;
 469         root->commit_root = NULL;
 470         root->sectorsize = sectorsize;
 471         root->nodesize = nodesize;
 472         root->leafsize = leafsize;
 473         root->stripesize = stripesize;
 474         root->ref_cows = 0;
 475         root->track_dirty = 0;
 476
 477         root->fs_info = fs_info;
 478         root->objectid = objectid;
 479         root->last_trans = 0;
 480         root->highest_inode = 0;
 481         root->last_inode_alloc = 0;
 482
 483         INIT_LIST_HEAD(&root->dirty_list);
 484         memset(&root->root_key, 0, sizeof(root->root_key));
 485         memset(&root->root_item, 0, sizeof(root->root_item));
 486         root->root_key.objectid = objectid;
 487         return 0;
 488 }
 489
 490 static int update_cowonly_root(struct btrfs_trans_handle *trans,
 491                                struct btrfs_root *root)
 492 {
 493         int ret;
 494         u64 old_root_bytenr;
 495         struct btrfs_root *tree_root = root->fs_info->tree_root;
 496
 497         btrfs_write_dirty_block_groups(trans, root);
 498         while(1) {
 499                 old_root_bytenr = btrfs_root_bytenr(&root->root_item);
 500                 if (old_root_bytenr == root->node->start)
 501                         break;
 502                 btrfs_set_root_bytenr(&root->root_item,
 503                                        root->node->start);
 504                 btrfs_set_root_generation(&root->root_item,
 505                                           trans->transid);
 506                 root->root_item.level = btrfs_header_level(root->node);
 507                 ret = btrfs_update_root(trans, tree_root,
 508                                         &root->root_key,
 509                                         &root->root_item);
 510                 BUG_ON(ret);
 511                 btrfs_write_dirty_block_groups(trans, root);
 512         }
 513         return 0;
 514 }
 515
 516 static int commit_tree_roots(struct btrfs_trans_handle *trans,
 517                              struct btrfs_fs_info *fs_info)
 518 {
 519         struct btrfs_root *root;
 520         struct list_head *next;
 521         struct extent_buffer *eb;
 522         int ret;
 523
 524         if (fs_info->readonly)
 525                 return 0;
 526
 527         eb = fs_info->tree_root->node;
 528         extent_buffer_get(eb);
 529         ret = btrfs_cow_block(trans, fs_info->tree_root, eb, NULL, 0, &eb);
 530         free_extent_buffer(eb);
 531         if (ret)
 532                 return ret;
 533
 534         while(!list_empty(&fs_info->dirty_cowonly_roots)) {
 535                 next = fs_info->dirty_cowonly_roots.next;
 536                 list_del_init(next);
 537                 root = list_entry(next, struct btrfs_root, dirty_list);
 538                 update_cowonly_root(trans, root);
 539                 free_extent_buffer(root->commit_root);
 540                 root->commit_root = NULL;
 541         }
 542
 543         return 0;
 544 }
 545
 546 static int __commit_transaction(struct btrfs_trans_handle *trans,
 547                                 struct btrfs_root *root)
 548 {
 549         u64 start;
 550         u64 end;
 551         struct extent_buffer *eb;
 552         struct extent_io_tree *tree = &root->fs_info->extent_cache;
 553         int ret;
 554
 555         while(1) {
 556                 ret = find_first_extent_bit(tree, 0, &start, &end,
 557                                             EXTENT_DIRTY);
 558                 if (ret)
 559                         break;
 560                 while(start <= end) {
 561                         eb = find_first_extent_buffer(tree, start);
 562                         BUG_ON(!eb || eb->start != start);
 563                         ret = write_tree_block(trans, root, eb);
 564                         BUG_ON(ret);
 565                         start += eb->len;
 566                         clear_extent_buffer_dirty(eb);
 567                         free_extent_buffer(eb);
 568                 }
 569         }
 570         return 0;
 571 }
 572
 573 int btrfs_commit_transaction(struct btrfs_trans_handle *trans,
 574                              struct btrfs_root *root)
 575 {
 576         u64 transid = trans->transid;
 577         int ret = 0;
 578         struct btrfs_fs_info *fs_info = root->fs_info;
 579
 580         if (root->commit_root == root->node)
 581                 goto commit_tree;
 582
 583         free_extent_buffer(root->commit_root);
 584         root->commit_root = NULL;
 585
 586         btrfs_set_root_bytenr(&root->root_item, root->node->start);
 587         btrfs_set_root_generation(&root->root_item, trans->transid);
 588         root->root_item.level = btrfs_header_level(root->node);
 589         ret = btrfs_update_root(trans, root->fs_info->tree_root,
 590                                 &root->root_key, &root->root_item);
 591         BUG_ON(ret);
 592 commit_tree:
 593         ret = commit_tree_roots(trans, fs_info);
 594         BUG_ON(ret);
 595         ret = __commit_transaction(trans, root);
 596         BUG_ON(ret);
 597         write_ctree_super(trans, root);
 598         btrfs_finish_extent_commit(trans, fs_info->extent_root,
 599                                    &fs_info->pinned_extents);
 600         btrfs_free_transaction(root, trans);
 601         free_extent_buffer(root->commit_root);
 602         root->commit_root = NULL;
 603         fs_info->running_transaction = NULL;
 604         fs_info->last_trans_committed = transid;
 605         return 0;
 606 }
 607
 608 static int find_and_setup_root(struct btrfs_root *tree_root,
 609                                struct btrfs_fs_info *fs_info,
 610                                u64 objectid, struct btrfs_root *root)
 611 {
 612         int ret;
 613         u32 blocksize;
 614         u64 generation;
 615
 616         __setup_root(tree_root->nodesize, tree_root->leafsize,
 617                      tree_root->sectorsize, tree_root->stripesize,
 618                      root, fs_info, objectid);
 619         ret = btrfs_find_last_root(tree_root, objectid,
 620                                    &root->root_item, &root->root_key);
 621         if (ret)
 622                 return ret;
 623
 624         blocksize = btrfs_level_size(root, btrfs_root_level(&root->root_item));
 625         generation = btrfs_root_generation(&root->root_item);
 626         root->node = read_tree_block(root, btrfs_root_bytenr(&root->root_item),
 627                                      blocksize, generation);
 628         if (!extent_buffer_uptodate(root->node))
 629                 return -EIO;
 630
 631         return 0;
 632 }
 633
 634 static int find_and_setup_log_root(struct btrfs_root *tree_root,
 635                                struct btrfs_fs_info *fs_info,
 636                                struct btrfs_super_block *disk_super)
 637 {
 638         u32 blocksize;
 639         u64 blocknr = btrfs_super_log_root(disk_super);
 640         struct btrfs_root *log_root = malloc(sizeof(struct btrfs_root));
 641
 642         if (!log_root)
 643                 return -ENOMEM;
 644
 645         if (blocknr == 0) {
 646                 free(log_root);
 647                 return 0;
 648         }
 649
 650         blocksize = btrfs_level_size(tree_root,
 651                              btrfs_super_log_root_level(disk_super));
 652
 653         __setup_root(tree_root->nodesize, tree_root->leafsize,
 654                      tree_root->sectorsize, tree_root->stripesize,
 655                      log_root, fs_info, BTRFS_TREE_LOG_OBJECTID);
 656
 657         log_root->node = read_tree_block(tree_root, blocknr,
 658                                      blocksize,
 659                                      btrfs_super_generation(disk_super) + 1);
 660
 661         fs_info->log_root_tree = log_root;
 662
 663         if (!extent_buffer_uptodate(log_root->node)) {
 664                 free_extent_buffer(log_root->node);
 665                 free(log_root);
 666                 fs_info->log_root_tree = NULL;
 667                 return -EIO;
 668         }
 669
 670         return 0;
 671 }
 672
 673
 674 int btrfs_free_fs_root(struct btrfs_root *root)
 675 {
 676         if (root->node)
 677                 free_extent_buffer(root->node);
 678         if (root->commit_root)
 679                 free_extent_buffer(root->commit_root);
 680         kfree(root);
 681         return 0;
 682 }
 683
 684 static void __free_fs_root(struct cache_extent *cache)
 685 {
 686         struct btrfs_root *root;
 687
 688         root = container_of(cache, struct btrfs_root, cache);
 689         btrfs_free_fs_root(root);
 690 }
 691
 692 FREE_EXTENT_CACHE_BASED_TREE(fs_roots, __free_fs_root);
 693
 694 struct btrfs_root *btrfs_read_fs_root_no_cache(struct btrfs_fs_info *fs_info,
 695                                                struct btrfs_key *location)
 696 {
 697         struct btrfs_root *root;
 698         struct btrfs_root *tree_root = fs_info->tree_root;
 699         struct btrfs_path *path;
 700         struct extent_buffer *l;
 701         u64 generation;
 702         u32 blocksize;
 703         int ret = 0;
 704
 705         root = malloc(sizeof(*root));
 706         if (!root)
 707                 return ERR_PTR(-ENOMEM);
 708         memset(root, 0, sizeof(*root));
 709         if (location->offset == (u64)-1) {
 710                 ret = find_and_setup_root(tree_root, fs_info,
 711                                           location->objectid, root);
 712                 if (ret) {
 713                         free(root);
 714                         return ERR_PTR(ret);
 715                 }
 716                 goto insert;
 717         }
 718
 719         __setup_root(tree_root->nodesize, tree_root->leafsize,
 720                      tree_root->sectorsize, tree_root->stripesize,
 721                      root, fs_info, location->objectid);
 722
 723         path = btrfs_alloc_path();
 724         BUG_ON(!path);
 725         ret = btrfs_search_slot(NULL, tree_root, location, path, 0, 0);
 726         if (ret != 0) {
 727                 if (ret > 0)
 728                         ret = -ENOENT;
 729                 goto out;
 730         }
 731         l = path->nodes[0];
 732         read_extent_buffer(l, &root->root_item,
 733                btrfs_item_ptr_offset(l, path->slots[0]),
 734                sizeof(root->root_item));
 735         memcpy(&root->root_key, location, sizeof(*location));
 736         ret = 0;
 737 out:
 738         btrfs_release_path(root, path);
 739         btrfs_free_path(path);
 740         if (ret) {
 741                 free(root);
 742                 return ERR_PTR(ret);
 743         }
 744         generation = btrfs_root_generation(&root->root_item);
 745         blocksize = btrfs_level_size(root, btrfs_root_level(&root->root_item));
 746         root->node = read_tree_block(root, btrfs_root_bytenr(&root->root_item),
 747                                      blocksize, generation);
 748         BUG_ON(!root->node);
 749 insert:
 750         root->ref_cows = 1;
 751         return root;
 752 }
 753
 754 struct btrfs_root *btrfs_read_fs_root(struct btrfs_fs_info *fs_info,
 755                                       struct btrfs_key *location)
 756 {
 757         struct btrfs_root *root;
 758         struct cache_extent *cache;
 759         int ret;
 760
 761         if (location->objectid == BTRFS_ROOT_TREE_OBJECTID)
 762                 return fs_info->tree_root;
 763         if (location->objectid == BTRFS_EXTENT_TREE_OBJECTID)
 764                 return fs_info->extent_root;
 765         if (location->objectid == BTRFS_CHUNK_TREE_OBJECTID)
 766                 return fs_info->chunk_root;
 767         if (location->objectid == BTRFS_DEV_TREE_OBJECTID)
 768                 return fs_info->dev_root;
 769         if (location->objectid == BTRFS_CSUM_TREE_OBJECTID)
 770                 return fs_info->csum_root;
 771
 772         BUG_ON(location->objectid == BTRFS_TREE_RELOC_OBJECTID ||
 773                location->offset != (u64)-1);
 774
 775         cache = find_cache_extent(&fs_info->fs_root_cache,
 776                                   location->objectid, 1);
 777         if (cache)
 778                 return container_of(cache, struct btrfs_root, cache);
 779
 780         root = btrfs_read_fs_root_no_cache(fs_info, location);
 781         if (IS_ERR(root))
 782                 return root;
 783
 784         root->cache.start = location->objectid;
 785         root->cache.size = 1;
 786         ret = insert_cache_extent(&fs_info->fs_root_cache, &root->cache);
 787         BUG_ON(ret);
 788         return root;
 789 }
 790
 791 void btrfs_free_fs_info(struct btrfs_fs_info *fs_info)
 792 {
 793         free(fs_info->tree_root);
 794         free(fs_info->extent_root);
 795         free(fs_info->chunk_root);
 796         free(fs_info->dev_root);
 797         free(fs_info->csum_root);
 798         free(fs_info->super_copy);
 799         free(fs_info->log_root_tree);
 800         free(fs_info);
 801 }
 802
 803 struct btrfs_fs_info *btrfs_new_fs_info(int writable, u64 sb_bytenr)
 804 {
 805         struct btrfs_fs_info *fs_info;
 806
 807         fs_info = malloc(sizeof(struct btrfs_fs_info));
 808         if (!fs_info)
 809                 return NULL;
 810
 811         memset(fs_info, 0, sizeof(struct btrfs_fs_info));
 812
 813         fs_info->tree_root = malloc(sizeof(struct btrfs_root));
 814         fs_info->extent_root = malloc(sizeof(struct btrfs_root));
 815         fs_info->chunk_root = malloc(sizeof(struct btrfs_root));
 816         fs_info->dev_root = malloc(sizeof(struct btrfs_root));
 817         fs_info->csum_root = malloc(sizeof(struct btrfs_root));
 818         fs_info->super_copy = malloc(BTRFS_SUPER_INFO_SIZE);
 819
 820         if (!fs_info->tree_root || !fs_info->extent_root ||
 821             !fs_info->chunk_root || !fs_info->dev_root ||
 822             !fs_info->csum_root || !fs_info->super_copy)
 823                 goto free_all;
 824
 825         memset(fs_info->super_copy, 0, BTRFS_SUPER_INFO_SIZE);
 826         memset(fs_info->tree_root, 0, sizeof(struct btrfs_root));
 827         memset(fs_info->extent_root, 0, sizeof(struct btrfs_root));
 828         memset(fs_info->chunk_root, 0, sizeof(struct btrfs_root));
 829         memset(fs_info->dev_root, 0, sizeof(struct btrfs_root));
 830         memset(fs_info->csum_root, 0, sizeof(struct btrfs_root));
 831
 832         extent_io_tree_init(&fs_info->extent_cache);
 833         extent_io_tree_init(&fs_info->free_space_cache);
 834         extent_io_tree_init(&fs_info->block_group_cache);
 835         extent_io_tree_init(&fs_info->pinned_extents);
 836         extent_io_tree_init(&fs_info->pending_del);
 837         extent_io_tree_init(&fs_info->extent_ins);
 838
 839         cache_tree_init(&fs_info->fs_root_cache);
 840         cache_tree_init(&fs_info->mapping_tree.cache_tree);
 841
 842         mutex_init(&fs_info->fs_mutex);
 843         INIT_LIST_HEAD(&fs_info->dirty_cowonly_roots);
 844         INIT_LIST_HEAD(&fs_info->space_info);
 845
 846         if (!writable)
 847                 fs_info->readonly = 1;
 848
 849         fs_info->super_bytenr = sb_bytenr;
 850         fs_info->data_alloc_profile = (u64)-1;
 851         fs_info->metadata_alloc_profile = (u64)-1;
 852         fs_info->system_alloc_profile = fs_info->metadata_alloc_profile;
 853         return fs_info;
 854 free_all:
 855         btrfs_free_fs_info(fs_info);
 856         return NULL;
 857 }
 858
 859 int btrfs_check_fs_compatibility(struct btrfs_super_block *sb, int writable)
 860 {
 861         u64 features;
 862
 863         features = btrfs_super_incompat_flags(sb) &
 864                    ~BTRFS_FEATURE_INCOMPAT_SUPP;
 865         if (features) {
 866                 printk("couldn't open because of unsupported "
 867                        "option features (%Lx).\n",
 868                        (unsigned long long)features);
 869                 return -ENOTSUP;
 870         }
 871
 872         features = btrfs_super_incompat_flags(sb);
 873         if (!(features & BTRFS_FEATURE_INCOMPAT_MIXED_BACKREF)) {
 874                 features |= BTRFS_FEATURE_INCOMPAT_MIXED_BACKREF;
 875                 btrfs_set_super_incompat_flags(sb, features);
 876         }
 877
 878         features = btrfs_super_compat_ro_flags(sb) &
 879                 ~BTRFS_FEATURE_COMPAT_RO_SUPP;
 880         if (writable && features) {
 881                 printk("couldn't open RDWR because of unsupported "
 882                        "option features (%Lx).\n",
 883                        (unsigned long long)features);
 884                 return -ENOTSUP;
 885         }
 886         return 0;
 887 }
 888
 889 int btrfs_setup_all_roots(struct btrfs_fs_info *fs_info,
 890                           u64 root_tree_bytenr, int partial)
 891 {
 892         struct btrfs_super_block *sb = fs_info->super_copy;
 893         struct btrfs_root *root;
 894         struct btrfs_key key;
 895         u32 sectorsize;
 896         u32 nodesize;
 897         u32 leafsize;
 898         u32 stripesize;
 899         u64 generation;
 900         u32 blocksize;
 901         int ret;
 902
 903         nodesize = btrfs_super_nodesize(sb);
 904         leafsize = btrfs_super_leafsize(sb);
 905         sectorsize = btrfs_super_sectorsize(sb);
 906         stripesize = btrfs_super_stripesize(sb);
 907
 908         root = fs_info->tree_root;
 909         __setup_root(nodesize, leafsize, sectorsize, stripesize,
 910                      root, fs_info, BTRFS_ROOT_TREE_OBJECTID);
 911         blocksize = btrfs_level_size(root, btrfs_super_root_level(sb));
 912         generation = btrfs_super_generation(sb);
 913
 914         if (!root_tree_bytenr)
 915                 root_tree_bytenr = btrfs_super_root(sb);
 916         root->node = read_tree_block(root, root_tree_bytenr, blocksize,
 917                                      generation);
 918         if (!extent_buffer_uptodate(root->node)) {
 919                 fprintf(stderr, "Couldn't read tree root\n");
 920                 return -EIO;
 921         }
 922
 923         ret = find_and_setup_root(root, fs_info, BTRFS_EXTENT_TREE_OBJECTID,
 924                                   fs_info->extent_root);
 925         if (ret) {
 926                 printk("Couldn't setup extent tree\n");
 927                 return -EIO;
 928         }
 929         fs_info->extent_root->track_dirty = 1;
 930
 931         ret = find_and_setup_root(root, fs_info, BTRFS_DEV_TREE_OBJECTID,
 932                                   fs_info->dev_root);
 933         if (ret) {
 934                 printk("Couldn't setup device tree\n");
 935                 return -EIO;
 936         }
 937         fs_info->dev_root->track_dirty = 1;
 938
 939         ret = find_and_setup_root(root, fs_info, BTRFS_CSUM_TREE_OBJECTID,
 940                                   fs_info->csum_root);
 941         if (ret) {
 942                 printk("Couldn't setup csum tree\n");
 943                 if (!partial)
 944                         return -EIO;
 945         }
 946         fs_info->csum_root->track_dirty = 1;
 947
 948         ret = find_and_setup_log_root(root, fs_info, sb);
 949         if (ret) {
 950                 printk("Couldn't setup log root tree\n");
 951                 return -EIO;
 952         }
 953
 954         fs_info->generation = generation;
 955         fs_info->last_trans_committed = generation;
 956         btrfs_read_block_groups(fs_info->tree_root);
 957
 958         key.objectid = BTRFS_FS_TREE_OBJECTID;
 959         key.type = BTRFS_ROOT_ITEM_KEY;
 960         key.offset = (u64)-1;
 961         fs_info->fs_root = btrfs_read_fs_root(fs_info, &key);
 962
 963         if (!fs_info->fs_root)
 964                 return -EIO;
 965         return 0;
 966 }
 967
 968 void btrfs_release_all_roots(struct btrfs_fs_info *fs_info)
 969 {
 970         if (fs_info->csum_root)
 971                 free_extent_buffer(fs_info->csum_root->node);
 972         if (fs_info->dev_root)
 973                 free_extent_buffer(fs_info->dev_root->node);
 974         if (fs_info->extent_root)
 975                 free_extent_buffer(fs_info->extent_root->node);
 976         if (fs_info->tree_root)
 977                 free_extent_buffer(fs_info->tree_root->node);
 978         if (fs_info->log_root_tree)
 979                 free_extent_buffer(fs_info->log_root_tree->node);
 980         if (fs_info->chunk_root)
 981                 free_extent_buffer(fs_info->chunk_root->node);
 982 }
 983
 984 static void free_map_lookup(struct cache_extent *ce)
 985 {
 986         struct map_lookup *map;
 987
 988         map = container_of(ce, struct map_lookup, ce);
 989         kfree(map);
 990 }
 991
 992 FREE_EXTENT_CACHE_BASED_TREE(mapping_cache, free_map_lookup);
 993
 994 void btrfs_cleanup_all_caches(struct btrfs_fs_info *fs_info)
 995 {
 996         free_mapping_cache_tree(&fs_info->mapping_tree.cache_tree);
 997         extent_io_tree_cleanup(&fs_info->extent_cache);
 998         extent_io_tree_cleanup(&fs_info->free_space_cache);
 999         extent_io_tree_cleanup(&fs_info->block_group_cache);
1000         extent_io_tree_cleanup(&fs_info->pinned_extents);
1001         extent_io_tree_cleanup(&fs_info->pending_del);
1002         extent_io_tree_cleanup(&fs_info->extent_ins);
1003 }
1004
1005 int btrfs_scan_fs_devices(int fd, const char *path,
1006                           struct btrfs_fs_devices **fs_devices)
1007 {
1008         u64 total_devs;
1009         int ret;
1010
1011         ret = btrfs_scan_one_device(fd, path, fs_devices,
1012                                     &total_devs, BTRFS_SUPER_INFO_OFFSET);
1013         if (ret) {
1014                 fprintf(stderr, "No valid Btrfs found on %s\n", path);
1015                 return ret;
1016         }
1017
1018         if (total_devs != 1) {
1019                 ret = btrfs_scan_for_fsid(*fs_devices, total_devs, 1);
1020                 if (ret)
1021                         return ret;
1022         }
1023         return 0;
1024 }
1025
1026 int btrfs_setup_chunk_tree_and_device_map(struct btrfs_fs_info *fs_info)
1027 {
1028         struct btrfs_super_block *sb = fs_info->super_copy;
1029         u32 sectorsize;
1030         u32 nodesize;
1031         u32 leafsize;
1032         u32 blocksize;
1033         u32 stripesize;
1034         u64 generation;
1035         int ret;
1036
1037         nodesize = btrfs_super_nodesize(sb);
1038         leafsize = btrfs_super_leafsize(sb);
1039         sectorsize = btrfs_super_sectorsize(sb);
1040         stripesize = btrfs_super_stripesize(sb);
1041
1042         __setup_root(nodesize, leafsize, sectorsize, stripesize,
1043                      fs_info->chunk_root, fs_info, BTRFS_CHUNK_TREE_OBJECTID);
1044
1045         ret = btrfs_read_sys_array(fs_info->chunk_root);
1046         if (ret)
1047                 return ret;
1048
1049         blocksize = btrfs_level_size(fs_info->chunk_root,
1050                                      btrfs_super_chunk_root_level(sb));
1051         generation = btrfs_super_chunk_root_generation(sb);
1052
1053         fs_info->chunk_root->node = read_tree_block(fs_info->chunk_root,
1054                                                     btrfs_super_chunk_root(sb),
1055                                                     blocksize, generation);
1056         if (!fs_info->chunk_root->node ||
1057             !extent_buffer_uptodate(fs_info->chunk_root->node)) {
1058                 fprintf(stderr, "Couldn't read chunk root\n");
1059                 return ret;
1060         }
1061
1062         if (!(btrfs_super_flags(sb) & BTRFS_SUPER_FLAG_METADUMP)) {
1063                 ret = btrfs_read_chunk_tree(fs_info->chunk_root);
1064                 if (ret) {
1065                         fprintf(stderr, "Couldn't read chunk tree\n");
1066                         return ret;
1067                 }
1068         }
1069         return 0;
1070 }
1071
1072 static struct btrfs_fs_info *__open_ctree_fd(int fp, const char *path,
1073                                              u64 sb_bytenr,
1074                                              u64 root_tree_bytenr, int writes,
1075                                              int partial)
1076 {
1077         struct btrfs_fs_info *fs_info;
1078         struct btrfs_super_block *disk_super;
1079         struct btrfs_fs_devices *fs_devices = NULL;
1080         struct extent_buffer *eb;
1081         int ret;
1082
1083         if (sb_bytenr == 0)
1084                 sb_bytenr = BTRFS_SUPER_INFO_OFFSET;
1085
1086         /* try to drop all the caches */
1087         if (posix_fadvise(fp, 0, 0, POSIX_FADV_DONTNEED))
1088                 fprintf(stderr, "Warning, could not drop caches\n");
1089
1090         fs_info = btrfs_new_fs_info(writes, sb_bytenr);
1091         if (!fs_info) {
1092                 fprintf(stderr, "Failed to allocate memory for fs_info\n");
1093                 return NULL;
1094         }
1095
1096         ret = btrfs_scan_fs_devices(fp, path, &fs_devices);
1097         if (ret)
1098                 goto out;
1099
1100         fs_info->fs_devices = fs_devices;
1101         if (writes)
1102                 ret = btrfs_open_devices(fs_devices, O_RDWR);
1103         else
1104                 ret = btrfs_open_devices(fs_devices, O_RDONLY);
1105         if (ret)
1106                 goto out_devices;
1107
1108
1109         disk_super = fs_info->super_copy;
1110         ret = btrfs_read_dev_super(fs_devices->latest_bdev,
1111                                    disk_super, sb_bytenr);
1112         if (ret) {
1113                 printk("No valid btrfs found\n");
1114                 goto out_devices;
1115         }
1116
1117         memcpy(fs_info->fsid, &disk_super->fsid, BTRFS_FSID_SIZE);
1118
1119         ret = btrfs_check_fs_compatibility(fs_info->super_copy, writes);
1120         if (ret)
1121                 goto out_devices;
1122
1123         ret = btrfs_setup_chunk_tree_and_device_map(fs_info);
1124         if (ret)
1125                 goto out_chunk;
1126
1127         eb = fs_info->chunk_root->node;
1128         read_extent_buffer(eb, fs_info->chunk_tree_uuid,
1129                            (unsigned long)btrfs_header_chunk_tree_uuid(eb),
1130                            BTRFS_UUID_SIZE);
1131
1132         ret = btrfs_setup_all_roots(fs_info, root_tree_bytenr, partial);
1133         if (ret)
1134                 goto out_failed;
1135
1136         return fs_info;
1137
1138 out_failed:
1139         if (partial)
1140                 return fs_info;
1141 out_chunk:
1142         btrfs_release_all_roots(fs_info);
1143         btrfs_cleanup_all_caches(fs_info);
1144 out_devices:
1145         btrfs_close_devices(fs_devices);
1146 out:
1147         btrfs_free_fs_info(fs_info);
1148         return NULL;
1149 }
1150
1151 struct btrfs_fs_info *open_ctree_fs_info(const char *filename,
1152                                          u64 sb_bytenr, u64 root_tree_bytenr,
1153                                          int writes, int partial)
1154 {
1155         int fp;
1156         struct btrfs_fs_info *info;
1157         int flags = O_CREAT | O_RDWR;
1158
1159         if (!writes)
1160                 flags = O_RDONLY;
1161
1162         fp = open(filename, flags, 0600);
1163         if (fp < 0) {
1164                 fprintf (stderr, "Could not open %s\n", filename);
1165                 return NULL;
1166         }
1167         info = __open_ctree_fd(fp, filename, sb_bytenr, root_tree_bytenr,
1168                                writes, partial);
1169         close(fp);
1170         return info;
1171 }
1172
1173 struct btrfs_root *open_ctree(const char *filename, u64 sb_bytenr, int writes)
1174 {
1175         struct btrfs_fs_info *info;
1176
1177         info = open_ctree_fs_info(filename, sb_bytenr, 0, writes, 0);
1178         if (!info)
1179                 return NULL;
1180         return info->fs_root;
1181 }
1182
1183 struct btrfs_root *open_ctree_fd(int fp, const char *path, u64 sb_bytenr,
1184                                  int writes)
1185 {
1186         struct btrfs_fs_info *info;
1187         info = __open_ctree_fd(fp, path, sb_bytenr, 0, writes, 0);
1188         if (!info)
1189                 return NULL;
1190         return info->fs_root;
1191 }
1192
1193 int btrfs_read_dev_super(int fd, struct btrfs_super_block *sb, u64 sb_bytenr)
1194 {
1195         u8 fsid[BTRFS_FSID_SIZE];
1196         int fsid_is_initialized = 0;
1197         struct btrfs_super_block buf;
1198         int i;
1199         int ret;
1200         u64 transid = 0;
1201         u64 bytenr;
1202
1203         if (sb_bytenr != BTRFS_SUPER_INFO_OFFSET) {
1204                 ret = pread64(fd, &buf, sizeof(buf), sb_bytenr);
1205                 if (ret < sizeof(buf))
1206                         return -1;
1207
1208                 if (btrfs_super_bytenr(&buf) != sb_bytenr ||
1209                     buf.magic != cpu_to_le64(BTRFS_MAGIC))
1210                         return -1;
1211
1212                 memcpy(sb, &buf, sizeof(*sb));
1213                 return 0;
1214         }
1215
1216         for (i = 0; i < BTRFS_SUPER_MIRROR_MAX; i++) {
1217                 bytenr = btrfs_sb_offset(i);
1218                 ret = pread64(fd, &buf, sizeof(buf), bytenr);
1219                 if (ret < sizeof(buf))
1220                         break;
1221
1222                 if (btrfs_super_bytenr(&buf) != bytenr )
1223                         continue;
1224                 /* if magic is NULL, the device was removed */
1225                 if (buf.magic == 0 && i == 0)
1226                         return -1;
1227                 if (buf.magic != cpu_to_le64(BTRFS_MAGIC))
1228                         continue;
1229
1230                 if (!fsid_is_initialized) {
1231                         memcpy(fsid, buf.fsid, sizeof(fsid));
1232                         fsid_is_initialized = 1;
1233                 } else if (memcmp(fsid, buf.fsid, sizeof(fsid))) {
1234                         /*
1235                          * the superblocks (the original one and
1236                          * its backups) contain data of different
1237                          * filesystems -> the super cannot be trusted
1238                          */
1239                         continue;
1240                 }
1241
1242                 if (btrfs_super_generation(&buf) > transid) {
1243                         memcpy(sb, &buf, sizeof(*sb));
1244                         transid = btrfs_super_generation(&buf);
1245                 }
1246         }
1247
1248         return transid > 0 ? 0 : -1;
1249 }
1250
1251 int write_dev_supers(struct btrfs_root *root, struct btrfs_super_block *sb,
1252                      struct btrfs_device *device)
1253 {
1254         u64 bytenr;
1255         u32 crc;
1256         int i, ret;
1257
1258         if (root->fs_info->super_bytenr != BTRFS_SUPER_INFO_OFFSET) {
1259                 btrfs_set_super_bytenr(sb, root->fs_info->super_bytenr);
1260                 crc = ~(u32)0;
1261                 crc = btrfs_csum_data(NULL, (char *)sb + BTRFS_CSUM_SIZE, crc,
1262                                       BTRFS_SUPER_INFO_SIZE - BTRFS_CSUM_SIZE);
1263                 btrfs_csum_final(crc, (char *)&sb->csum[0]);
1264
1265                 /*
1266                  * super_copy is BTRFS_SUPER_INFO_SIZE bytes and is
1267                  * zero filled, we can use it directly
1268                  */
1269                 ret = pwrite64(device->fd, root->fs_info->super_copy,
1270                                 BTRFS_SUPER_INFO_SIZE,
1271                                 root->fs_info->super_bytenr);
1272                 BUG_ON(ret != BTRFS_SUPER_INFO_SIZE);
1273                 return 0;
1274         }
1275
1276         for (i = 0; i < BTRFS_SUPER_MIRROR_MAX; i++) {
1277                 bytenr = btrfs_sb_offset(i);
1278                 if (bytenr + BTRFS_SUPER_INFO_SIZE > device->total_bytes)
1279                         break;
1280
1281                 btrfs_set_super_bytenr(sb, bytenr);
1282
1283                 crc = ~(u32)0;
1284                 crc = btrfs_csum_data(NULL, (char *)sb + BTRFS_CSUM_SIZE, crc,
1285                                       BTRFS_SUPER_INFO_SIZE - BTRFS_CSUM_SIZE);
1286                 btrfs_csum_final(crc, (char *)&sb->csum[0]);
1287
1288                 /*
1289                  * super_copy is BTRFS_SUPER_INFO_SIZE bytes and is
1290                  * zero filled, we can use it directly
1291                  */
1292                 ret = pwrite64(device->fd, root->fs_info->super_copy,
1293                                 BTRFS_SUPER_INFO_SIZE, bytenr);
1294                 BUG_ON(ret != BTRFS_SUPER_INFO_SIZE);
1295         }
1296
1297         return 0;
1298 }
1299
1300 int write_all_supers(struct btrfs_root *root)
1301 {
1302         struct list_head *cur;
1303         struct list_head *head = &root->fs_info->fs_devices->devices;
1304         struct btrfs_device *dev;
1305         struct btrfs_super_block *sb;
1306         struct btrfs_dev_item *dev_item;
1307         int ret;
1308         u64 flags;
1309
1310         sb = root->fs_info->super_copy;
1311         dev_item = &sb->dev_item;
1312         list_for_each(cur, head) {
1313                 dev = list_entry(cur, struct btrfs_device, dev_list);
1314                 if (!dev->writeable)
1315                         continue;
1316
1317                 btrfs_set_stack_device_generation(dev_item, 0);
1318                 btrfs_set_stack_device_type(dev_item, dev->type);
1319                 btrfs_set_stack_device_id(dev_item, dev->devid);
1320                 btrfs_set_stack_device_total_bytes(dev_item, dev->total_bytes);
1321                 btrfs_set_stack_device_bytes_used(dev_item, dev->bytes_used);
1322                 btrfs_set_stack_device_io_align(dev_item, dev->io_align);
1323                 btrfs_set_stack_device_io_width(dev_item, dev->io_width);
1324                 btrfs_set_stack_device_sector_size(dev_item, dev->sector_size);
1325                 memcpy(dev_item->uuid, dev->uuid, BTRFS_UUID_SIZE);
1326                 memcpy(dev_item->fsid, dev->fs_devices->fsid, BTRFS_UUID_SIZE);
1327
1328                 flags = btrfs_super_flags(sb);
1329                 btrfs_set_super_flags(sb, flags | BTRFS_HEADER_FLAG_WRITTEN);
1330
1331                 ret = write_dev_supers(root, sb, dev);
1332                 BUG_ON(ret);
1333         }
1334         return 0;
1335 }
1336
1337 int write_ctree_super(struct btrfs_trans_handle *trans,
1338                       struct btrfs_root *root)
1339 {
1340         int ret;
1341         struct btrfs_root *tree_root = root->fs_info->tree_root;
1342         struct btrfs_root *chunk_root = root->fs_info->chunk_root;
1343
1344         if (root->fs_info->readonly)
1345                 return 0;
1346
1347         btrfs_set_super_generation(root->fs_info->super_copy,
1348                                    trans->transid);
1349         btrfs_set_super_root(root->fs_info->super_copy,
1350                              tree_root->node->start);
1351         btrfs_set_super_root_level(root->fs_info->super_copy,
1352                                    btrfs_header_level(tree_root->node));
1353         btrfs_set_super_chunk_root(root->fs_info->super_copy,
1354                                    chunk_root->node->start);
1355         btrfs_set_super_chunk_root_level(root->fs_info->super_copy,
1356                                          btrfs_header_level(chunk_root->node));
1357         btrfs_set_super_chunk_root_generation(root->fs_info->super_copy,
1358                                 btrfs_header_generation(chunk_root->node));
1359
1360         ret = write_all_supers(root);
1361         if (ret)
1362                 fprintf(stderr, "failed to write new super block err %d\n", ret);
1363         return ret;
1364 }
1365
1366 int close_ctree(struct btrfs_root *root)
1367 {
1368         int ret;
1369         struct btrfs_trans_handle *trans;
1370         struct btrfs_fs_info *fs_info = root->fs_info;
1371
1372         if (fs_info->last_trans_committed !=
1373             fs_info->generation) {
1374                 trans = btrfs_start_transaction(root, 1);
1375                 btrfs_commit_transaction(trans, root);
1376                 trans = btrfs_start_transaction(root, 1);
1377                 ret = commit_tree_roots(trans, fs_info);
1378                 BUG_ON(ret);
1379                 ret = __commit_transaction(trans, root);
1380                 BUG_ON(ret);
1381                 write_ctree_super(trans, root);
1382                 btrfs_free_transaction(root, trans);
1383         }
1384         btrfs_free_block_groups(fs_info);
1385
1386         free_fs_roots_tree(&fs_info->fs_root_cache);
1387
1388         btrfs_release_all_roots(fs_info);
1389         btrfs_close_devices(fs_info->fs_devices);
1390         btrfs_cleanup_all_caches(fs_info);
1391         btrfs_free_fs_info(fs_info);
1392         return 0;
1393 }
1394
1395 int clean_tree_block(struct btrfs_trans_handle *trans, struct btrfs_root *root,
1396                      struct extent_buffer *eb)
1397 {
1398         return clear_extent_buffer_dirty(eb);
1399 }
1400
1401 int wait_on_tree_block_writeback(struct btrfs_root *root,
1402                                  struct extent_buffer *eb)
1403 {
1404         return 0;
1405 }
1406
1407 void btrfs_mark_buffer_dirty(struct extent_buffer *eb)
1408 {
1409         set_extent_buffer_dirty(eb);
1410 }
1411
1412 int btrfs_buffer_uptodate(struct extent_buffer *buf, u64 parent_transid)
1413 {
1414         int ret;
1415
1416         ret = extent_buffer_uptodate(buf);
1417         if (!ret)
1418                 return ret;
1419
1420         ret = verify_parent_transid(buf->tree, buf, parent_transid, 1);
1421         return !ret;
1422 }
1423
1424 int btrfs_set_buffer_uptodate(struct extent_buffer *eb)
1425 {
1426         return set_extent_buffer_uptodate(eb);
1427 }