btrfs-corrupt-block.c

   1 /*
   2  * Copyright (C) 2009 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 #define _GNU_SOURCE 1
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <fcntl.h>
  24 #include <unistd.h>
  25 #include <getopt.h>
  26 #include "kerncompat.h"
  27 #include "ctree.h"
  28 #include "volumes.h"
  29 #include "disk-io.h"
  30 #include "print-tree.h"
  31 #include "transaction.h"
  32 #include "list.h"
  33 #include "version.h"
  34 #include "utils.h"
  35
  36 #define FIELD_BUF_LEN 80
  37
  38 struct extent_buffer *debug_corrupt_block(struct btrfs_root *root, u64 bytenr,
  39                                      u32 blocksize, u64 copy)
  40 {
  41         int ret;
  42         struct extent_buffer *eb;
  43         u64 length;
  44         struct btrfs_multi_bio *multi = NULL;
  45         struct btrfs_device *device;
  46         int num_copies;
  47         int mirror_num = 1;
  48
  49         eb = btrfs_find_create_tree_block(root, bytenr, blocksize);
  50         if (!eb)
  51                 return NULL;
  52
  53         length = blocksize;
  54         while (1) {
  55                 ret = btrfs_map_block(&root->fs_info->mapping_tree, READ,
  56                                       eb->start, &length, &multi,
  57                                       mirror_num, NULL);
  58                 BUG_ON(ret);
  59                 device = multi->stripes[0].dev;
  60                 eb->fd = device->fd;
  61                 device->total_ios++;
  62                 eb->dev_bytenr = multi->stripes[0].physical;
  63
  64                 fprintf(stdout,
  65                         "mirror %d logical %llu physical %llu device %s\n",
  66                         mirror_num, (unsigned long long)bytenr,
  67                         (unsigned long long)eb->dev_bytenr, device->name);
  68                 kfree(multi);
  69
  70                 if (!copy || mirror_num == copy) {
  71                         ret = read_extent_from_disk(eb, 0, eb->len);
  72                         printf("corrupting %llu copy %d\n", eb->start,
  73                                mirror_num);
  74                         memset(eb->data, 0, eb->len);
  75                         write_extent_to_disk(eb);
  76                         fsync(eb->fd);
  77                 }
  78
  79                 num_copies = btrfs_num_copies(&root->fs_info->mapping_tree,
  80                                               eb->start, eb->len);
  81                 if (num_copies == 1)
  82                         break;
  83
  84                 mirror_num++;
  85                 if (mirror_num > num_copies)
  86                         break;
  87         }
  88         return eb;
  89 }
  90
  91 static void print_usage(void)
  92 {
  93         fprintf(stderr, "usage: btrfs-corrupt-block [options] device\n");
  94         fprintf(stderr, "\t-l Logical extent to be corrupted\n");
  95         fprintf(stderr, "\t-c Copy of the extent to be corrupted"
  96                 " (usually 1 or 2, default: 0)\n");
  97         fprintf(stderr, "\t-b Number of bytes to be corrupted\n");
  98         fprintf(stderr, "\t-e Extent to be corrupted\n");
  99         fprintf(stderr, "\t-E The whole extent tree to be corrupted\n");
 100         fprintf(stderr, "\t-u Given chunk item to be corrupted\n");
 101         fprintf(stderr, "\t-U The whole chunk tree to be corrupted\n");
 102         fprintf(stderr, "\t-i The inode item to corrupt (must also specify "
 103                 "the field to corrupt)\n");
 104         fprintf(stderr, "\t-x The file extent item to corrupt (must also "
 105                 "specify -i for the inode and -f for the field to corrupt)\n");
 106         fprintf(stderr, "\t-m The metadata block to corrupt (must also "
 107                 "specify -f for the field to corrupt)\n");
 108         fprintf(stderr, "\t-K The key to corrupt in the format "
 109                 "<num>,<num>,<num> (must also specify -f for the field)\n");
 110         fprintf(stderr, "\t-f The field in the item to corrupt\n");
 111         fprintf(stderr, "\t-I An item to corrupt (must also specify the field "
 112                 "to corrupt and a root+key for the item)\n");
 113         fprintf(stderr, "\t-D Corrupt a dir item, must specify key and field\n");
 114         fprintf(stderr, "\t-d Delete this item (must specify -K)\n");
 115         exit(1);
 116 }
 117
 118 static void corrupt_keys(struct btrfs_trans_handle *trans,
 119                          struct btrfs_root *root,
 120                          struct extent_buffer *eb)
 121 {
 122         int slot;
 123         int bad_slot;
 124         int nr;
 125         struct btrfs_disk_key bad_key;;
 126
 127         nr = btrfs_header_nritems(eb);
 128         if (nr == 0)
 129                 return;
 130
 131         slot = rand() % nr;
 132         bad_slot = rand() % nr;
 133
 134         if (bad_slot == slot)
 135                 return;
 136
 137         fprintf(stderr,
 138                 "corrupting keys in block %llu slot %d swapping with %d\n",
 139                 (unsigned long long)eb->start, slot, bad_slot);
 140
 141         if (btrfs_header_level(eb) == 0) {
 142                 btrfs_item_key(eb, &bad_key, bad_slot);
 143                 btrfs_set_item_key(eb, &bad_key, slot);
 144         } else {
 145                 btrfs_node_key(eb, &bad_key, bad_slot);
 146                 btrfs_set_node_key(eb, &bad_key, slot);
 147         }
 148         btrfs_mark_buffer_dirty(eb);
 149         if (!trans) {
 150                 u16 csum_size =
 151                         btrfs_super_csum_size(root->fs_info->super_copy);
 152                 csum_tree_block_size(eb, csum_size, 0);
 153                 write_extent_to_disk(eb);
 154         }
 155 }
 156
 157
 158 static int corrupt_keys_in_block(struct btrfs_root *root, u64 bytenr)
 159 {
 160         struct extent_buffer *eb;
 161
 162         eb = read_tree_block(root, bytenr, root->leafsize, 0);
 163         if (!eb)
 164                 return -EIO;;
 165
 166         corrupt_keys(NULL, root, eb);
 167         free_extent_buffer(eb);
 168         return 0;
 169 }
 170
 171 static int corrupt_extent(struct btrfs_trans_handle *trans,
 172                           struct btrfs_root *root, u64 bytenr, u64 copy)
 173 {
 174         struct btrfs_key key;
 175         struct extent_buffer *leaf;
 176         u32 item_size;
 177         unsigned long ptr;
 178         struct btrfs_path *path;
 179         int ret;
 180         int slot;
 181         int should_del = rand() % 3;
 182
 183         path = btrfs_alloc_path();
 184         if (!path)
 185                 return -ENOMEM;
 186
 187         key.objectid = bytenr;
 188         key.type = (u8)-1;
 189         key.offset = (u64)-1;
 190
 191         while(1) {
 192                 ret = btrfs_search_slot(trans, root->fs_info->extent_root,
 193                                         &key, path, -1, 1);
 194                 if (ret < 0)
 195                         break;
 196
 197                 if (ret > 0) {
 198                         if (path->slots[0] == 0)
 199                                 break;
 200                         path->slots[0]--;
 201                         ret = 0;
 202                 }
 203                 leaf = path->nodes[0];
 204                 slot = path->slots[0];
 205                 btrfs_item_key_to_cpu(leaf, &key, slot);
 206                 if (key.objectid != bytenr)
 207                         break;
 208
 209                 if (key.type != BTRFS_EXTENT_ITEM_KEY &&
 210                     key.type != BTRFS_TREE_BLOCK_REF_KEY &&
 211                     key.type != BTRFS_EXTENT_DATA_REF_KEY &&
 212                     key.type != BTRFS_EXTENT_REF_V0_KEY &&
 213                     key.type != BTRFS_SHARED_BLOCK_REF_KEY &&
 214                     key.type != BTRFS_SHARED_DATA_REF_KEY)
 215                         goto next;
 216
 217                 if (should_del) {
 218                         fprintf(stderr,
 219                                 "deleting extent record: key %llu %u %llu\n",
 220                                 key.objectid, key.type, key.offset);
 221
 222                         if (key.type == BTRFS_EXTENT_ITEM_KEY) {
 223                                 /* make sure this extent doesn't get
 224                                  * reused for other purposes */
 225                                 btrfs_pin_extent(root->fs_info,
 226                                                  key.objectid, key.offset);
 227                         }
 228
 229                         btrfs_del_item(trans, root, path);
 230                 } else {
 231                         fprintf(stderr,
 232                                 "corrupting extent record: key %llu %u %llu\n",
 233                                 key.objectid, key.type, key.offset);
 234                         ptr = btrfs_item_ptr_offset(leaf, slot);
 235                         item_size = btrfs_item_size_nr(leaf, slot);
 236                         memset_extent_buffer(leaf, 0, ptr, item_size);
 237                         btrfs_mark_buffer_dirty(leaf);
 238                 }
 239 next:
 240                 btrfs_release_path(path);
 241
 242                 if (key.offset > 0)
 243                         key.offset--;
 244                 if (key.offset == 0)
 245                         break;
 246         }
 247
 248         btrfs_free_path(path);
 249         return 0;
 250 }
 251
 252 static void btrfs_corrupt_extent_leaf(struct btrfs_trans_handle *trans,
 253                                       struct btrfs_root *root,
 254                                       struct extent_buffer *eb)
 255 {
 256         u32 nr = btrfs_header_nritems(eb);
 257         u32 victim = rand() % nr;
 258         u64 objectid;
 259         struct btrfs_key key;
 260
 261         btrfs_item_key_to_cpu(eb, &key, victim);
 262         objectid = key.objectid;
 263         corrupt_extent(trans, root, objectid, 1);
 264 }
 265
 266 static void btrfs_corrupt_extent_tree(struct btrfs_trans_handle *trans,
 267                                       struct btrfs_root *root,
 268                                       struct extent_buffer *eb)
 269 {
 270         int i;
 271
 272         if (!eb)
 273                 return;
 274
 275         if (btrfs_is_leaf(eb)) {
 276                 btrfs_corrupt_extent_leaf(trans, root, eb);
 277                 return;
 278         }
 279
 280         if (btrfs_header_level(eb) == 1 && eb != root->node) {
 281                 if (rand() % 5)
 282                         return;
 283         }
 284
 285         for (i = 0; i < btrfs_header_nritems(eb); i++) {
 286                 struct extent_buffer *next;
 287
 288                 next = read_tree_block(root, btrfs_node_blockptr(eb, i),
 289                                        root->leafsize,
 290                                        btrfs_node_ptr_generation(eb, i));
 291                 if (!next)
 292                         continue;
 293                 btrfs_corrupt_extent_tree(trans, root, next);
 294                 free_extent_buffer(next);
 295         }
 296 }
 297
 298 enum btrfs_inode_field {
 299         BTRFS_INODE_FIELD_ISIZE,
 300         BTRFS_INODE_FIELD_BAD,
 301 };
 302
 303 enum btrfs_file_extent_field {
 304         BTRFS_FILE_EXTENT_DISK_BYTENR,
 305         BTRFS_FILE_EXTENT_BAD,
 306 };
 307
 308 enum btrfs_dir_item_field {
 309         BTRFS_DIR_ITEM_NAME,
 310         BTRFS_DIR_ITEM_LOCATION_OBJECTID,
 311         BTRFS_DIR_ITEM_BAD,
 312 };
 313
 314 enum btrfs_metadata_block_field {
 315         BTRFS_METADATA_BLOCK_GENERATION,
 316         BTRFS_METADATA_BLOCK_SHIFT_ITEMS,
 317         BTRFS_METADATA_BLOCK_BAD,
 318 };
 319
 320 enum btrfs_item_field {
 321         BTRFS_ITEM_OFFSET,
 322         BTRFS_ITEM_BAD,
 323 };
 324
 325 enum btrfs_key_field {
 326         BTRFS_KEY_OBJECTID,
 327         BTRFS_KEY_TYPE,
 328         BTRFS_KEY_OFFSET,
 329         BTRFS_KEY_BAD,
 330 };
 331
 332 static enum btrfs_inode_field convert_inode_field(char *field)
 333 {
 334         if (!strncmp(field, "isize", FIELD_BUF_LEN))
 335                 return BTRFS_INODE_FIELD_ISIZE;
 336         return BTRFS_INODE_FIELD_BAD;
 337 }
 338
 339 static enum btrfs_file_extent_field convert_file_extent_field(char *field)
 340 {
 341         if (!strncmp(field, "disk_bytenr", FIELD_BUF_LEN))
 342                 return BTRFS_FILE_EXTENT_DISK_BYTENR;
 343         return BTRFS_FILE_EXTENT_BAD;
 344 }
 345
 346 static enum btrfs_metadata_block_field
 347 convert_metadata_block_field(char *field)
 348 {
 349         if (!strncmp(field, "generation", FIELD_BUF_LEN))
 350                 return BTRFS_METADATA_BLOCK_GENERATION;
 351         if (!strncmp(field, "shift_items", FIELD_BUF_LEN))
 352                 return BTRFS_METADATA_BLOCK_SHIFT_ITEMS;
 353         return BTRFS_METADATA_BLOCK_BAD;
 354 }
 355
 356 static enum btrfs_key_field convert_key_field(char *field)
 357 {
 358         if (!strncmp(field, "objectid", FIELD_BUF_LEN))
 359                 return BTRFS_KEY_OBJECTID;
 360         if (!strncmp(field, "type", FIELD_BUF_LEN))
 361                 return BTRFS_KEY_TYPE;
 362         if (!strncmp(field, "offset", FIELD_BUF_LEN))
 363                 return BTRFS_KEY_OFFSET;
 364         return BTRFS_KEY_BAD;
 365 }
 366
 367 static enum btrfs_item_field convert_item_field(char *field)
 368 {
 369         if (!strncmp(field, "offset", FIELD_BUF_LEN))
 370                 return BTRFS_ITEM_OFFSET;
 371         return BTRFS_ITEM_BAD;
 372 }
 373
 374 static enum btrfs_dir_item_field convert_dir_item_field(char *field)
 375 {
 376         if (!strncmp(field, "name", FIELD_BUF_LEN))
 377                 return BTRFS_DIR_ITEM_NAME;
 378         if (!strncmp(field, "location_objectid", FIELD_BUF_LEN))
 379                 return BTRFS_DIR_ITEM_LOCATION_OBJECTID;
 380         return BTRFS_DIR_ITEM_BAD;
 381 }
 382
 383 static u64 generate_u64(u64 orig)
 384 {
 385         u64 ret;
 386         do {
 387                 ret = rand();
 388         } while (ret == orig);
 389         return ret;
 390 }
 391
 392 static u32 generate_u32(u32 orig)
 393 {
 394         u32 ret;
 395         do {
 396                 ret = rand();
 397         } while (ret == orig);
 398         return ret;
 399 }
 400
 401 static u8 generate_u8(u8 orig)
 402 {
 403         u8 ret;
 404         do {
 405                 ret = rand();
 406         } while (ret == orig);
 407         return ret;
 408 }
 409
 410 static int corrupt_key(struct btrfs_root *root, struct btrfs_key *key,
 411                        char *field)
 412 {
 413         enum btrfs_key_field corrupt_field = convert_key_field(field);
 414         struct btrfs_path *path;
 415         struct btrfs_trans_handle *trans;
 416         int ret;
 417
 418         root = root->fs_info->fs_root;
 419         if (corrupt_field == BTRFS_KEY_BAD) {
 420                 fprintf(stderr, "Invalid field %s\n", field);
 421                 return -EINVAL;
 422         }
 423
 424         path = btrfs_alloc_path();
 425         if (!path)
 426                 return -ENOMEM;
 427
 428         trans = btrfs_start_transaction(root, 1);
 429         if (IS_ERR(trans)) {
 430                 btrfs_free_path(path);
 431                 return PTR_ERR(trans);
 432         }
 433
 434         ret = btrfs_search_slot(trans, root, key, path, 0, 1);
 435         if (ret < 0)
 436                 goto out;
 437         if (ret > 0) {
 438                 fprintf(stderr, "Couldn't find the key to corrupt\n");
 439                 ret = -ENOENT;
 440                 goto out;
 441         }
 442
 443         switch (corrupt_field) {
 444         case BTRFS_KEY_OBJECTID:
 445                 key->objectid = generate_u64(key->objectid);
 446                 break;
 447         case BTRFS_KEY_TYPE:
 448                 key->type = generate_u8(key->type);
 449                 break;
 450         case BTRFS_KEY_OFFSET:
 451                 key->offset = generate_u64(key->objectid);
 452                 break;
 453         default:
 454                 fprintf(stderr, "Invalid field %s, %d\n", field,
 455                         corrupt_field);
 456                 ret = -EINVAL;
 457                 goto out;
 458         }
 459
 460         btrfs_set_item_key_unsafe(root, path, key);
 461 out:
 462         btrfs_free_path(path);
 463         btrfs_commit_transaction(trans, root);
 464         return ret;
 465 }
 466
 467 static int corrupt_dir_item(struct btrfs_root *root, struct btrfs_key *key,
 468                             char *field)
 469 {
 470         struct btrfs_trans_handle *trans;
 471         struct btrfs_dir_item *di;
 472         struct btrfs_path *path;
 473         char *name;
 474         struct btrfs_key location;
 475         struct btrfs_disk_key disk_key;
 476         unsigned long name_ptr;
 477         enum btrfs_dir_item_field corrupt_field =
 478                 convert_dir_item_field(field);
 479         u64 bogus;
 480         u16 name_len;
 481         int ret;
 482
 483         if (corrupt_field == BTRFS_DIR_ITEM_BAD) {
 484                 fprintf(stderr, "Invalid field %s\n", field);
 485                 return -EINVAL;
 486         }
 487
 488         path = btrfs_alloc_path();
 489         if (!path)
 490                 return -ENOMEM;
 491
 492         trans = btrfs_start_transaction(root, 1);
 493         if (IS_ERR(trans)) {
 494                 btrfs_free_path(path);
 495                 return PTR_ERR(trans);
 496         }
 497
 498         ret = btrfs_search_slot(trans, root, key, path, 0, 1);
 499         if (ret) {
 500                 if (ret > 0)
 501                         ret = -ENOENT;
 502                 fprintf(stderr, "Error searching for dir item %d\n", ret);
 503                 goto out;
 504         }
 505
 506         di = btrfs_item_ptr(path->nodes[0], path->slots[0],
 507                             struct btrfs_dir_item);
 508
 509         switch (corrupt_field) {
 510         case BTRFS_DIR_ITEM_NAME:
 511                 name_len = btrfs_dir_name_len(path->nodes[0], di);
 512                 name = malloc(name_len);
 513                 if (!name) {
 514                         ret = -ENOMEM;
 515                         goto out;
 516                 }
 517                 name_ptr = (unsigned long)(di + 1);
 518                 read_extent_buffer(path->nodes[0], name, name_ptr, name_len);
 519                 name[0]++;
 520                 write_extent_buffer(path->nodes[0], name, name_ptr, name_len);
 521                 btrfs_mark_buffer_dirty(path->nodes[0]);
 522                 free(name);
 523                 goto out;
 524         case BTRFS_DIR_ITEM_LOCATION_OBJECTID:
 525                 btrfs_dir_item_key_to_cpu(path->nodes[0], di, &location);
 526                 bogus = generate_u64(location.objectid);
 527                 location.objectid = bogus;
 528                 btrfs_cpu_key_to_disk(&disk_key, &location);
 529                 btrfs_set_dir_item_key(path->nodes[0], di, &disk_key);
 530                 btrfs_mark_buffer_dirty(path->nodes[0]);
 531                 goto out;
 532         default:
 533                 ret = -EINVAL;
 534                 goto out;
 535         }
 536 out:
 537         btrfs_commit_transaction(trans, root);
 538         btrfs_free_path(path);
 539         return ret;
 540 }
 541
 542 static int corrupt_inode(struct btrfs_trans_handle *trans,
 543                          struct btrfs_root *root, u64 inode, char *field)
 544 {
 545         struct btrfs_inode_item *ei;
 546         struct btrfs_path *path;
 547         struct btrfs_key key;
 548         enum btrfs_inode_field corrupt_field = convert_inode_field(field);
 549         u64 bogus;
 550         u64 orig;
 551         int ret;
 552
 553         if (corrupt_field == BTRFS_INODE_FIELD_BAD) {
 554                 fprintf(stderr, "Invalid field %s\n", field);
 555                 return -EINVAL;
 556         }
 557
 558         key.objectid = inode;
 559         key.type = BTRFS_INODE_ITEM_KEY;
 560         key.offset = (u64)-1;
 561
 562         path = btrfs_alloc_path();
 563         if (!path)
 564                 return -ENOMEM;
 565
 566         ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
 567         if (ret < 0)
 568                 goto out;
 569         if (ret) {
 570                 if (!path->slots[0]) {
 571                         fprintf(stderr, "Couldn't find inode %Lu\n", inode);
 572                         ret = -ENOENT;
 573                         goto out;
 574                 }
 575                 path->slots[0]--;
 576                 ret = 0;
 577         }
 578
 579         btrfs_item_key_to_cpu(path->nodes[0], &key, path->slots[0]);
 580         if (key.objectid != inode) {
 581                 fprintf(stderr, "Couldn't find inode %Lu\n", inode);
 582                 ret = -ENOENT;
 583                 goto out;
 584         }
 585
 586         ei = btrfs_item_ptr(path->nodes[0], path->slots[0],
 587                             struct btrfs_inode_item);
 588         switch (corrupt_field) {
 589         case BTRFS_INODE_FIELD_ISIZE:
 590                 orig = btrfs_inode_size(path->nodes[0], ei);
 591                 bogus = generate_u64(orig);
 592                 btrfs_set_inode_size(path->nodes[0], ei, bogus);
 593                 break;
 594         default:
 595                 ret = -EINVAL;
 596                 break;
 597         }
 598         btrfs_mark_buffer_dirty(path->nodes[0]);
 599 out:
 600         btrfs_free_path(path);
 601         return ret;
 602 }
 603
 604 static int corrupt_file_extent(struct btrfs_trans_handle *trans,
 605                                struct btrfs_root *root, u64 inode, u64 extent,
 606                                char *field)
 607 {
 608         struct btrfs_file_extent_item *fi;
 609         struct btrfs_path *path;
 610         struct btrfs_key key;
 611         enum btrfs_file_extent_field corrupt_field;
 612         u64 bogus;
 613         u64 orig;
 614         int ret = 0;
 615
 616         corrupt_field = convert_file_extent_field(field);
 617         if (corrupt_field == BTRFS_FILE_EXTENT_BAD) {
 618                 fprintf(stderr, "Invalid field %s\n", field);
 619                 return -EINVAL;
 620         }
 621
 622         key.objectid = inode;
 623         key.type = BTRFS_EXTENT_DATA_KEY;
 624         key.offset = extent;
 625
 626         path = btrfs_alloc_path();
 627         if (!path)
 628                 return -ENOMEM;
 629
 630         ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
 631         if (ret < 0)
 632                 goto out;
 633         if (ret) {
 634                 fprintf(stderr, "Couldn't find extent %llu for inode %llu\n",
 635                         extent, inode);
 636                 ret = -ENOENT;
 637                 goto out;
 638         }
 639
 640         fi = btrfs_item_ptr(path->nodes[0], path->slots[0],
 641                             struct btrfs_file_extent_item);
 642         switch (corrupt_field) {
 643         case BTRFS_FILE_EXTENT_DISK_BYTENR:
 644                 orig = btrfs_file_extent_disk_bytenr(path->nodes[0], fi);
 645                 bogus = generate_u64(orig);
 646                 btrfs_set_file_extent_disk_bytenr(path->nodes[0], fi, bogus);
 647                 break;
 648         default:
 649                 ret = -EINVAL;
 650                 break;
 651         }
 652         btrfs_mark_buffer_dirty(path->nodes[0]);
 653 out:
 654         btrfs_free_path(path);
 655         return ret;
 656 }
 657
 658 static void shift_items(struct btrfs_root *root, struct extent_buffer *eb)
 659 {
 660         int nritems = btrfs_header_nritems(eb);
 661         int shift_space = btrfs_leaf_free_space(root, eb) / 2;
 662         int slot = nritems / 2;
 663         int i = 0;
 664         unsigned int data_end = btrfs_item_offset_nr(eb, nritems - 1);
 665
 666         /* Shift the item data up to and including slot back by shift space */
 667         memmove_extent_buffer(eb, btrfs_leaf_data(eb) + data_end - shift_space,
 668                               btrfs_leaf_data(eb) + data_end,
 669                               btrfs_item_offset_nr(eb, slot - 1) - data_end);
 670
 671         /* Now update the item pointers. */
 672         for (i = nritems - 1; i >= slot; i--) {
 673                 u32 offset = btrfs_item_offset_nr(eb, i);
 674                 offset -= shift_space;
 675                 btrfs_set_item_offset(eb, btrfs_item_nr(i), offset);
 676         }
 677 }
 678
 679 static int corrupt_metadata_block(struct btrfs_root *root, u64 block,
 680                                   char *field)
 681 {
 682         struct btrfs_trans_handle *trans;
 683         struct btrfs_path *path;
 684         struct extent_buffer *eb;
 685         struct btrfs_key key, root_key;
 686         enum btrfs_metadata_block_field corrupt_field;
 687         u64 root_objectid;
 688         u64 orig, bogus;
 689         u8 level;
 690         int ret;
 691
 692         corrupt_field = convert_metadata_block_field(field);
 693         if (corrupt_field == BTRFS_METADATA_BLOCK_BAD) {
 694                 fprintf(stderr, "Invalid field %s\n", field);
 695                 return -EINVAL;
 696         }
 697
 698         eb = read_tree_block(root, block, root->leafsize, 0);
 699         if (!eb) {
 700                 fprintf(stderr, "Couldn't read in tree block %s\n", field);
 701                 return -EINVAL;
 702         }
 703         root_objectid = btrfs_header_owner(eb);
 704         level = btrfs_header_level(eb);
 705         if (level)
 706                 btrfs_node_key_to_cpu(eb, &key, 0);
 707         else
 708                 btrfs_item_key_to_cpu(eb, &key, 0);
 709         free_extent_buffer(eb);
 710
 711         root_key.objectid = root_objectid;
 712         root_key.type = BTRFS_ROOT_ITEM_KEY;
 713         root_key.offset = (u64)-1;
 714
 715         root = btrfs_read_fs_root(root->fs_info, &root_key);
 716         if (IS_ERR(root)) {
 717                 fprintf(stderr, "Couldn't finde owner root %llu\n",
 718                         key.objectid);
 719                 return PTR_ERR(root);
 720         }
 721
 722         path = btrfs_alloc_path();
 723         if (!path)
 724                 return -ENOMEM;
 725
 726         trans = btrfs_start_transaction(root, 1);
 727         if (IS_ERR(trans)) {
 728                 btrfs_free_path(path);
 729                 fprintf(stderr, "Couldn't start transaction %ld\n",
 730                         PTR_ERR(trans));
 731                 return PTR_ERR(trans);
 732         }
 733
 734         path->lowest_level = level;
 735         ret = btrfs_search_slot(trans, root, &key, path, 0, 1);
 736         if (ret < 0) {
 737                 fprintf(stderr, "Error searching to node %d\n", ret);
 738                 goto out;
 739         }
 740         eb = path->nodes[level];
 741
 742         ret = 0;
 743         switch (corrupt_field) {
 744         case BTRFS_METADATA_BLOCK_GENERATION:
 745                 orig = btrfs_header_generation(eb);
 746                 bogus = generate_u64(orig);
 747                 btrfs_set_header_generation(eb, bogus);
 748                 break;
 749         case BTRFS_METADATA_BLOCK_SHIFT_ITEMS:
 750                 shift_items(root, path->nodes[level]);
 751                 break;
 752         default:
 753                 ret = -EINVAL;
 754                 break;
 755         }
 756         btrfs_mark_buffer_dirty(path->nodes[level]);
 757 out:
 758         btrfs_commit_transaction(trans, root);
 759         btrfs_free_path(path);
 760         return ret;
 761 }
 762
 763 static int corrupt_btrfs_item(struct btrfs_root *root, struct btrfs_key *key,
 764                               char *field)
 765 {
 766         struct btrfs_trans_handle *trans;
 767         struct btrfs_path *path;
 768         enum btrfs_item_field corrupt_field;
 769         u32 orig, bogus;
 770         int ret;
 771
 772         corrupt_field = convert_item_field(field);
 773         if (corrupt_field == BTRFS_ITEM_BAD) {
 774                 fprintf(stderr, "Invalid field %s\n", field);
 775                 return -EINVAL;
 776         }
 777
 778         path = btrfs_alloc_path();
 779         if (!path)
 780                 return -ENOMEM;
 781
 782         trans = btrfs_start_transaction(root, 1);
 783         if (IS_ERR(trans)) {
 784                 btrfs_free_path(path);
 785                 fprintf(stderr, "Couldn't start transaction %ld\n",
 786                         PTR_ERR(trans));
 787                 return PTR_ERR(trans);
 788         }
 789
 790         ret = btrfs_search_slot(trans, root, key, path, 0, 1);
 791         if (ret != 0) {
 792                 fprintf(stderr, "Error searching to node %d\n", ret);
 793                 goto out;
 794         }
 795
 796         ret = 0;
 797         switch (corrupt_field) {
 798         case BTRFS_ITEM_OFFSET:
 799                 orig = btrfs_item_offset_nr(path->nodes[0], path->slots[0]);
 800                 bogus = generate_u32(orig);
 801                 btrfs_set_item_offset(path->nodes[0],
 802                                       btrfs_item_nr(path->slots[0]), bogus);
 803                 break;
 804         default:
 805                 ret = -EINVAL;
 806                 break;
 807         }
 808         btrfs_mark_buffer_dirty(path->nodes[0]);
 809 out:
 810         btrfs_commit_transaction(trans, root);
 811         btrfs_free_path(path);
 812         return ret;
 813 }
 814
 815 static int delete_item(struct btrfs_root *root, struct btrfs_key *key)
 816 {
 817         struct btrfs_trans_handle *trans;
 818         struct btrfs_path *path;
 819         int ret;
 820
 821         path = btrfs_alloc_path();
 822         if (!path)
 823                 return -ENOMEM;
 824
 825         trans = btrfs_start_transaction(root, 1);
 826         if (IS_ERR(trans)) {
 827                 btrfs_free_path(path);
 828                 fprintf(stderr, "Couldn't start transaction %ld\n",
 829                         PTR_ERR(trans));
 830                 return PTR_ERR(trans);
 831         }
 832
 833         ret = btrfs_search_slot(trans, root, key, path, -1, 1);
 834         if (ret) {
 835                 if (ret > 0)
 836                         ret = -ENOENT;
 837                 fprintf(stderr, "Error searching to node %d\n", ret);
 838                 goto out;
 839         }
 840         ret = btrfs_del_item(trans, root, path);
 841         btrfs_mark_buffer_dirty(path->nodes[0]);
 842 out:
 843         btrfs_commit_transaction(trans, root);
 844         btrfs_free_path(path);
 845         return ret;
 846 }
 847
 848 /* corrupt item using NO cow.
 849  * Because chunk recover will recover based on whole partition scaning,
 850  * If using COW, chunk recover will use the old item to recover,
 851  * which is still OK but we want to check the ability to rebuild chunk
 852  * not only restore the old ones */
 853 int corrupt_item_nocow(struct btrfs_trans_handle *trans,
 854                        struct btrfs_root *root, struct btrfs_path *path,
 855                        int del)
 856 {
 857         int ret = 0;
 858         struct btrfs_key key;
 859         struct extent_buffer *leaf;
 860         unsigned long ptr;
 861         int slot;
 862         u32 item_size;
 863
 864         leaf = path->nodes[0];
 865         slot = path->slots[0];
 866         /* Not deleting the first item of a leaf to keep leaf structure */
 867         if (slot == 0)
 868                 del = 0;
 869         /* Only accept valid eb */
 870         BUG_ON(!leaf->data || slot >= btrfs_header_nritems(leaf));
 871         btrfs_item_key_to_cpu(leaf, &key, slot);
 872         if (del) {
 873                 fprintf(stdout, "Deleting key and data [%llu, %u, %llu].\n",
 874                         key.objectid, key.type, key.offset);
 875                 btrfs_del_item(trans, root, path);
 876         } else {
 877                 fprintf(stdout, "Corrupting key and data [%llu, %u, %llu].\n",
 878                         key.objectid, key.type, key.offset);
 879                 ptr = btrfs_item_ptr_offset(leaf, slot);
 880                 item_size = btrfs_item_size_nr(leaf, slot);
 881                 memset_extent_buffer(leaf, 0, ptr, item_size);
 882                 btrfs_mark_buffer_dirty(leaf);
 883         }
 884         return ret;
 885 }
 886 int corrupt_chunk_tree(struct btrfs_trans_handle *trans,
 887                        struct btrfs_root *root)
 888 {
 889         int ret;
 890         int del;
 891         int slot;
 892         struct btrfs_path *path;
 893         struct btrfs_key key;
 894         struct btrfs_key found_key;
 895         struct extent_buffer *leaf;
 896
 897         path = btrfs_alloc_path();
 898         if (!path)
 899                 return -ENOMEM;
 900
 901         key.objectid = (u64)-1;
 902         key.offset = (u64)-1;
 903         key.type = (u8)-1;
 904
 905         /* Here, cow and ins_len must equals 0 for the following reasons:
 906          * 1) chunk recover is based on disk scanning, so COW should be
 907          *    disabled in case the original chunk being scanned and
 908          *    recovered using the old chunk.
 909          * 2) if cow = 0, ins_len must also be set to 0, or BUG_ON will be
 910          *    triggered.
 911          */
 912         ret = btrfs_search_slot(trans, root, &key, path, 0, 0);
 913         BUG_ON(ret == 0);
 914         if (ret < 0) {
 915                 fprintf(stderr, "Error searching tree\n");
 916                 goto free_out;
 917         }
 918         /* corrupt/del dev_item first */
 919         while (!btrfs_previous_item(root, path, 0, BTRFS_DEV_ITEM_KEY)) {
 920                 slot = path->slots[0];
 921                 leaf = path->nodes[0];
 922                 del = rand() % 3;
 923                 /* Never delete the first item to keep the leaf structure */
 924                 if (path->slots[0] == 0)
 925                         del = 0;
 926                 ret = corrupt_item_nocow(trans, root, path, del);
 927                 if (ret)
 928                         goto free_out;
 929         }
 930         btrfs_release_path(path);
 931
 932         /* Here, cow and ins_len must equals 0 for the following reasons:
 933          * 1) chunk recover is based on disk scanning, so COW should be
 934          *    disabled in case the original chunk being scanned and
 935          *    recovered using the old chunk.
 936          * 2) if cow = 0, ins_len must also be set to 0, or BUG_ON will be
 937          *    triggered.
 938          */
 939         ret = btrfs_search_slot(trans, root, &key, path, 0, 0);
 940         BUG_ON(ret == 0);
 941         if (ret < 0) {
 942                 fprintf(stderr, "Error searching tree\n");
 943                 goto free_out;
 944         }
 945         /* corrupt/del chunk then*/
 946         while (!btrfs_previous_item(root, path, 0, BTRFS_CHUNK_ITEM_KEY)) {
 947                 slot = path->slots[0];
 948                 leaf = path->nodes[0];
 949                 del = rand() % 3;
 950                 btrfs_item_key_to_cpu(leaf, &found_key, slot);
 951                 ret = corrupt_item_nocow(trans, root, path, del);
 952                 if (ret)
 953                         goto free_out;
 954         }
 955 free_out:
 956         btrfs_free_path(path);
 957         return ret;
 958 }
 959 int find_chunk_offset(struct btrfs_root *root,
 960                       struct btrfs_path *path, u64 offset)
 961 {
 962         struct btrfs_key key;
 963         int ret;
 964
 965         key.objectid = BTRFS_FIRST_CHUNK_TREE_OBJECTID;
 966         key.type = BTRFS_CHUNK_ITEM_KEY;
 967         key.offset = offset;
 968
 969         /* Here, cow and ins_len must equals 0 for following reasons:
 970          * 1) chunk recover is based on disk scanning, so COW should
 971          *    be disabled in case the original chunk being scanned
 972          *    and recovered using the old chunk.
 973          * 2) if cow = 0, ins_len must also be set to 0, or BUG_ON
 974          *    will be triggered.
 975          */
 976         ret = btrfs_search_slot(NULL, root, &key, path, 0, 0);
 977         if (ret > 0) {
 978                 fprintf(stderr, "Can't find chunk with given offset %llu\n",
 979                         offset);
 980                 goto out;
 981         }
 982         if (ret < 0) {
 983                 fprintf(stderr, "Error searching chunk");
 984                 goto out;
 985         }
 986 out:
 987         return ret;
 988
 989 }
 990 int main(int ac, char **av)
 991 {
 992         struct cache_tree root_cache;
 993         struct btrfs_key key;
 994         struct btrfs_root *root;
 995         struct extent_buffer *eb;
 996         char *dev;
 997         /* chunk offset can be 0,so change to (u64)-1 */
 998         u64 logical = (u64)-1;
 999         int ret = 0;
1000         u64 copy = 0;
1001         u64 bytes = 4096;
1002         int extent_rec = 0;
1003         int extent_tree = 0;
1004         int corrupt_block_keys = 0;
1005         int chunk_rec = 0;
1006         int chunk_tree = 0;
1007         int corrupt_item = 0;
1008         int corrupt_di = 0;
1009         int delete = 0;
1010         u64 metadata_block = 0;
1011         u64 inode = 0;
1012         u64 file_extent = (u64)-1;
1013         char field[FIELD_BUF_LEN];
1014
1015         field[0] = '\0';
1016         srand(128);
1017         memset(&key, 0, sizeof(key));
1018
1019         while(1) {
1020                 int c;
1021                 int option_index = 0;
1022                 static const struct option long_options[] = {
1023                         /* { "byte-count", 1, NULL, 'b' }, */
1024                         { "logical", 1, NULL, 'l' },
1025                         { "copy", 1, NULL, 'c' },
1026                         { "bytes", 1, NULL, 'b' },
1027                         { "extent-record", 0, NULL, 'e' },
1028                         { "extent-tree", 0, NULL, 'E' },
1029                         { "keys", 0, NULL, 'k' },
1030                         { "chunk-record", 0, NULL, 'u' },
1031                         { "chunk-tree", 0, NULL, 'U' },
1032                         { "inode", 1, NULL, 'i'},
1033                         { "file-extent", 1, NULL, 'x'},
1034                         { "metadata-block", 1, NULL, 'm'},
1035                         { "field", 1, NULL, 'f'},
1036                         { "key", 1, NULL, 'K'},
1037                         { "item", 0, NULL, 'I'},
1038                         { "dir-item", 0, NULL, 'D'},
1039                         { "delete", 0, NULL, 'd'},
1040                         { 0, 0, 0, 0}
1041                 };
1042
1043                 c = getopt_long(ac, av, "l:c:b:eEkuUi:f:x:m:K:IDd", long_options,
1044                                 &option_index);
1045                 if (c < 0)
1046                         break;
1047                 switch(c) {
1048                         case 'l':
1049                                 logical = arg_strtou64(optarg);
1050                                 break;
1051                         case 'c':
1052                                 copy = arg_strtou64(optarg);
1053                                 break;
1054                         case 'b':
1055                                 bytes = arg_strtou64(optarg);
1056                                 break;
1057                         case 'e':
1058                                 extent_rec = 1;
1059                                 break;
1060                         case 'E':
1061                                 extent_tree = 1;
1062                                 break;
1063                         case 'k':
1064                                 corrupt_block_keys = 1;
1065                                 break;
1066                         case 'u':
1067                                 chunk_rec = 1;
1068                                 break;
1069                         case 'U':
1070                                 chunk_tree = 1;
1071                                 break;
1072                         case 'i':
1073                                 inode = arg_strtou64(optarg);
1074                                 break;
1075                         case 'f':
1076                                 strncpy(field, optarg, FIELD_BUF_LEN);
1077                                 break;
1078                         case 'x':
1079                                 file_extent = arg_strtou64(optarg);
1080                                 break;
1081                         case 'm':
1082                                 metadata_block = arg_strtou64(optarg);
1083                                 break;
1084                         case 'K':
1085                                 ret = sscanf(optarg, "%llu,%u,%llu",
1086                                              &key.objectid,
1087                                              (unsigned int *)&key.type,
1088                                              &key.offset);
1089                                 if (ret != 3) {
1090                                         fprintf(stderr, "error reading key "
1091                                                 "%d\n", errno);
1092                                         print_usage();
1093                                 }
1094                                 break;
1095                         case 'D':
1096                                 corrupt_di = 1;
1097                                 break;
1098                         case 'I':
1099                                 corrupt_item = 1;
1100                                 break;
1101                         case 'd':
1102                                 delete = 1;
1103                                 break;
1104                         default:
1105                                 print_usage();
1106                 }
1107         }
1108         set_argv0(av);
1109         ac = ac - optind;
1110         if (check_argc_min(ac, 1))
1111                 print_usage();
1112         dev = av[optind];
1113
1114         radix_tree_init();
1115         cache_tree_init(&root_cache);
1116
1117         root = open_ctree(dev, 0, OPEN_CTREE_WRITES);
1118         if (!root) {
1119                 fprintf(stderr, "Open ctree failed\n");
1120                 exit(1);
1121         }
1122         if (extent_rec) {
1123                 struct btrfs_trans_handle *trans;
1124
1125                 if (logical == (u64)-1)
1126                         print_usage();
1127                 trans = btrfs_start_transaction(root, 1);
1128                 ret = corrupt_extent (trans, root, logical, 0);
1129                 btrfs_commit_transaction(trans, root);
1130                 goto out_close;
1131         }
1132         if (extent_tree) {
1133                 struct btrfs_trans_handle *trans;
1134                 trans = btrfs_start_transaction(root, 1);
1135                 btrfs_corrupt_extent_tree(trans, root->fs_info->extent_root,
1136                                           root->fs_info->extent_root->node);
1137                 btrfs_commit_transaction(trans, root);
1138                 goto out_close;
1139         }
1140         if (chunk_rec) {
1141                 struct btrfs_trans_handle *trans;
1142                 struct btrfs_path *path;
1143                 int del;
1144
1145                 if (logical == (u64)-1)
1146                         print_usage();
1147                 del = rand() % 3;
1148                 path = btrfs_alloc_path();
1149                 if (!path) {
1150                         fprintf(stderr, "path allocation failed\n");
1151                         goto out_close;
1152                 }
1153
1154                 if (find_chunk_offset(root->fs_info->chunk_root, path,
1155                                       logical) != 0) {
1156                         btrfs_free_path(path);
1157                         goto out_close;
1158                 }
1159                 trans = btrfs_start_transaction(root, 1);
1160                 ret = corrupt_item_nocow(trans, root->fs_info->chunk_root,
1161                                          path, del);
1162                 if (ret < 0)
1163                         fprintf(stderr, "Failed to corrupt chunk record\n");
1164                 btrfs_commit_transaction(trans, root);
1165                 goto out_close;
1166         }
1167         if (chunk_tree) {
1168                 struct btrfs_trans_handle *trans;
1169                 trans = btrfs_start_transaction(root, 1);
1170                 ret = corrupt_chunk_tree(trans, root->fs_info->chunk_root);
1171                 if (ret < 0)
1172                         fprintf(stderr, "Failed to corrupt chunk tree\n");
1173                 btrfs_commit_transaction(trans, root);
1174                 goto out_close;
1175         }
1176         if (inode) {
1177                 struct btrfs_trans_handle *trans;
1178
1179                 if (!strlen(field))
1180                         print_usage();
1181
1182                 trans = btrfs_start_transaction(root, 1);
1183                 if (file_extent == (u64)-1) {
1184                         printf("corrupting inode\n");
1185                         ret = corrupt_inode(trans, root, inode, field);
1186                 } else {
1187                         printf("corrupting file extent\n");
1188                         ret = corrupt_file_extent(trans, root, inode,
1189                                                   file_extent, field);
1190                 }
1191                 btrfs_commit_transaction(trans, root);
1192                 goto out_close;
1193         }
1194         if (metadata_block) {
1195                 if (!strlen(field))
1196                         print_usage();
1197                 ret = corrupt_metadata_block(root, metadata_block, field);
1198                 goto out_close;
1199         }
1200         if (corrupt_di) {
1201                 if (!key.objectid || !strlen(field))
1202                         print_usage();
1203                 ret = corrupt_dir_item(root, &key, field);
1204                 goto out_close;
1205         }
1206         if (corrupt_item) {
1207                 if (!key.objectid)
1208                         print_usage();
1209                 ret = corrupt_btrfs_item(root, &key, field);
1210         }
1211         if (delete) {
1212                 if (!key.objectid)
1213                         print_usage();
1214                 ret = delete_item(root, &key);
1215                 goto out_close;
1216         }
1217         if (key.objectid || key.offset || key.type) {
1218                 if (!strlen(field))
1219                         print_usage();
1220                 ret = corrupt_key(root, &key, field);
1221                 goto out_close;
1222         }
1223         /*
1224          * If we made it here and we have extent set then we didn't specify
1225          * inode and we're screwed.
1226          */
1227         if (file_extent != (u64)-1)
1228                 print_usage();
1229
1230         if (logical == (u64)-1)
1231                 print_usage();
1232
1233         if (bytes == 0)
1234                 bytes = root->sectorsize;
1235
1236         bytes = (bytes + root->sectorsize - 1) / root->sectorsize;
1237         bytes *= root->sectorsize;
1238
1239         while (bytes > 0) {
1240                 if (corrupt_block_keys) {
1241                         corrupt_keys_in_block(root, logical);
1242                 } else {
1243                         eb = debug_corrupt_block(root, logical,
1244                                                  root->sectorsize, copy);
1245                         free_extent_buffer(eb);
1246                 }
1247                 logical += root->sectorsize;
1248                 bytes -= root->sectorsize;
1249         }
1250         return ret;
1251 out_close:
1252         close_ctree(root);
1253         return ret;
1254 }