1 /* 2 * Copyright (C) 2013 Fusion IO. 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 #include "btrfs-tests.h" 20 #include "../ctree.h" 21 #include "../btrfs_inode.h" 22 #include "../disk-io.h" 23 #include "../extent_io.h" 24 #include "../volumes.h" 25 26 static struct btrfs_fs_info *alloc_dummy_fs_info(void) 27 { 28 struct btrfs_fs_info *fs_info = kzalloc(sizeof(struct btrfs_fs_info), 29 GFP_NOFS); 30 if (!fs_info) 31 return fs_info; 32 fs_info->fs_devices = kzalloc(sizeof(struct btrfs_fs_devices), 33 GFP_NOFS); 34 if (!fs_info->fs_devices) { 35 kfree(fs_info); 36 return NULL; 37 } 38 return fs_info; 39 } 40 static void free_dummy_root(struct btrfs_root *root) 41 { 42 if (!root) 43 return; 44 if (root->fs_info) { 45 kfree(root->fs_info->fs_devices); 46 kfree(root->fs_info); 47 } 48 if (root->node) 49 free_extent_buffer(root->node); 50 kfree(root); 51 } 52 53 static void insert_extent(struct btrfs_root *root, u64 start, u64 len, 54 u64 ram_bytes, u64 offset, u64 disk_bytenr, 55 u64 disk_len, u32 type, u8 compression, int slot) 56 { 57 struct btrfs_path path; 58 struct btrfs_file_extent_item *fi; 59 struct extent_buffer *leaf = root->node; 60 struct btrfs_key key; 61 u32 value_len = sizeof(struct btrfs_file_extent_item); 62 63 if (type == BTRFS_FILE_EXTENT_INLINE) 64 value_len += len; 65 memset(&path, 0, sizeof(path)); 66 67 path.nodes[0] = leaf; 68 path.slots[0] = slot; 69 70 key.objectid = BTRFS_FIRST_FREE_OBJECTID; 71 key.type = BTRFS_EXTENT_DATA_KEY; 72 key.offset = start; 73 74 setup_items_for_insert(root, &path, &key, &value_len, value_len, 75 value_len + sizeof(struct btrfs_item), 1); 76 fi = btrfs_item_ptr(leaf, slot, struct btrfs_file_extent_item); 77 btrfs_set_file_extent_generation(leaf, fi, 1); 78 btrfs_set_file_extent_type(leaf, fi, type); 79 btrfs_set_file_extent_disk_bytenr(leaf, fi, disk_bytenr); 80 btrfs_set_file_extent_disk_num_bytes(leaf, fi, disk_len); 81 btrfs_set_file_extent_offset(leaf, fi, offset); 82 btrfs_set_file_extent_num_bytes(leaf, fi, len); 83 btrfs_set_file_extent_ram_bytes(leaf, fi, ram_bytes); 84 btrfs_set_file_extent_compression(leaf, fi, compression); 85 btrfs_set_file_extent_encryption(leaf, fi, 0); 86 btrfs_set_file_extent_other_encoding(leaf, fi, 0); 87 } 88 89 static void insert_inode_item_key(struct btrfs_root *root) 90 { 91 struct btrfs_path path; 92 struct extent_buffer *leaf = root->node; 93 struct btrfs_key key; 94 u32 value_len = 0; 95 96 memset(&path, 0, sizeof(path)); 97 98 path.nodes[0] = leaf; 99 path.slots[0] = 0; 100 101 key.objectid = BTRFS_INODE_ITEM_KEY; 102 key.type = BTRFS_INODE_ITEM_KEY; 103 key.offset = 0; 104 105 setup_items_for_insert(root, &path, &key, &value_len, value_len, 106 value_len + sizeof(struct btrfs_item), 1); 107 } 108 109 /* 110 * Build the most complicated map of extents the earth has ever seen. We want 111 * this so we can test all of the corner cases of btrfs_get_extent. Here is a 112 * diagram of how the extents will look though this may not be possible we still 113 * want to make sure everything acts normally (the last number is not inclusive) 114 * 115 * [0 - 5][5 - 6][6 - 10][10 - 4096][ 4096 - 8192 ][8192 - 12288] 116 * [hole ][inline][ hole ][ regular ][regular1 split][ hole ] 117 * 118 * [ 12288 - 20480][20480 - 24576][ 24576 - 28672 ][28672 - 36864][36864 - 45056] 119 * [regular1 split][ prealloc1 ][prealloc1 written][ prealloc1 ][ compressed ] 120 * 121 * [45056 - 49152][49152-53248][53248-61440][61440-65536][ 65536+81920 ] 122 * [ compressed1 ][ regular ][compressed1][ regular ][ hole but no extent] 123 * 124 * [81920-86016] 125 * [ regular ] 126 */ 127 static void setup_file_extents(struct btrfs_root *root) 128 { 129 int slot = 0; 130 u64 disk_bytenr = 1 * 1024 * 1024; 131 u64 offset = 0; 132 133 /* First we want a hole */ 134 insert_extent(root, offset, 5, 5, 0, 0, 0, BTRFS_FILE_EXTENT_REG, 0, 135 slot); 136 slot++; 137 offset += 5; 138 139 /* 140 * Now we want an inline extent, I don't think this is possible but hey 141 * why not? Also keep in mind if we have an inline extent it counts as 142 * the whole first page. If we were to expand it we would have to cow 143 * and we wouldn't have an inline extent anymore. 144 */ 145 insert_extent(root, offset, 1, 1, 0, 0, 0, BTRFS_FILE_EXTENT_INLINE, 0, 146 slot); 147 slot++; 148 offset = 4096; 149 150 /* Now another hole */ 151 insert_extent(root, offset, 4, 4, 0, 0, 0, BTRFS_FILE_EXTENT_REG, 0, 152 slot); 153 slot++; 154 offset += 4; 155 156 /* Now for a regular extent */ 157 insert_extent(root, offset, 4095, 4095, 0, disk_bytenr, 4096, 158 BTRFS_FILE_EXTENT_REG, 0, slot); 159 slot++; 160 disk_bytenr += 4096; 161 offset += 4095; 162 163 /* 164 * Now for 3 extents that were split from a hole punch so we test 165 * offsets properly. 166 */ 167 insert_extent(root, offset, 4096, 16384, 0, disk_bytenr, 16384, 168 BTRFS_FILE_EXTENT_REG, 0, slot); 169 slot++; 170 offset += 4096; 171 insert_extent(root, offset, 4096, 4096, 0, 0, 0, BTRFS_FILE_EXTENT_REG, 172 0, slot); 173 slot++; 174 offset += 4096; 175 insert_extent(root, offset, 8192, 16384, 8192, disk_bytenr, 16384, 176 BTRFS_FILE_EXTENT_REG, 0, slot); 177 slot++; 178 offset += 8192; 179 disk_bytenr += 16384; 180 181 /* Now for a unwritten prealloc extent */ 182 insert_extent(root, offset, 4096, 4096, 0, disk_bytenr, 4096, 183 BTRFS_FILE_EXTENT_PREALLOC, 0, slot); 184 slot++; 185 offset += 4096; 186 187 /* 188 * We want to jack up disk_bytenr a little more so the em stuff doesn't 189 * merge our records. 190 */ 191 disk_bytenr += 8192; 192 193 /* 194 * Now for a partially written prealloc extent, basically the same as 195 * the hole punch example above. Ram_bytes never changes when you mark 196 * extents written btw. 197 */ 198 insert_extent(root, offset, 4096, 16384, 0, disk_bytenr, 16384, 199 BTRFS_FILE_EXTENT_PREALLOC, 0, slot); 200 slot++; 201 offset += 4096; 202 insert_extent(root, offset, 4096, 16384, 4096, disk_bytenr, 16384, 203 BTRFS_FILE_EXTENT_REG, 0, slot); 204 slot++; 205 offset += 4096; 206 insert_extent(root, offset, 8192, 16384, 8192, disk_bytenr, 16384, 207 BTRFS_FILE_EXTENT_PREALLOC, 0, slot); 208 slot++; 209 offset += 8192; 210 disk_bytenr += 16384; 211 212 /* Now a normal compressed extent */ 213 insert_extent(root, offset, 8192, 8192, 0, disk_bytenr, 4096, 214 BTRFS_FILE_EXTENT_REG, BTRFS_COMPRESS_ZLIB, slot); 215 slot++; 216 offset += 8192; 217 /* No merges */ 218 disk_bytenr += 8192; 219 220 /* Now a split compressed extent */ 221 insert_extent(root, offset, 4096, 16384, 0, disk_bytenr, 4096, 222 BTRFS_FILE_EXTENT_REG, BTRFS_COMPRESS_ZLIB, slot); 223 slot++; 224 offset += 4096; 225 insert_extent(root, offset, 4096, 4096, 0, disk_bytenr + 4096, 4096, 226 BTRFS_FILE_EXTENT_REG, 0, slot); 227 slot++; 228 offset += 4096; 229 insert_extent(root, offset, 8192, 16384, 8192, disk_bytenr, 4096, 230 BTRFS_FILE_EXTENT_REG, BTRFS_COMPRESS_ZLIB, slot); 231 slot++; 232 offset += 8192; 233 disk_bytenr += 8192; 234 235 /* Now extents that have a hole but no hole extent */ 236 insert_extent(root, offset, 4096, 4096, 0, disk_bytenr, 4096, 237 BTRFS_FILE_EXTENT_REG, 0, slot); 238 slot++; 239 offset += 16384; 240 disk_bytenr += 4096; 241 insert_extent(root, offset, 4096, 4096, 0, disk_bytenr, 4096, 242 BTRFS_FILE_EXTENT_REG, 0, slot); 243 } 244 245 static unsigned long prealloc_only = 0; 246 static unsigned long compressed_only = 0; 247 static unsigned long vacancy_only = 0; 248 249 static noinline int test_btrfs_get_extent(void) 250 { 251 struct inode *inode = NULL; 252 struct btrfs_root *root = NULL; 253 struct extent_map *em = NULL; 254 u64 orig_start; 255 u64 disk_bytenr; 256 u64 offset; 257 int ret = -ENOMEM; 258 259 inode = btrfs_new_test_inode(); 260 if (!inode) { 261 test_msg("Couldn't allocate inode\n"); 262 return ret; 263 } 264 265 BTRFS_I(inode)->location.type = BTRFS_INODE_ITEM_KEY; 266 BTRFS_I(inode)->location.objectid = BTRFS_FIRST_FREE_OBJECTID; 267 BTRFS_I(inode)->location.offset = 0; 268 269 root = btrfs_alloc_dummy_root(); 270 if (IS_ERR(root)) { 271 test_msg("Couldn't allocate root\n"); 272 goto out; 273 } 274 275 /* 276 * We do this since btrfs_get_extent wants to assign em->bdev to 277 * root->fs_info->fs_devices->latest_bdev. 278 */ 279 root->fs_info = alloc_dummy_fs_info(); 280 if (!root->fs_info) { 281 test_msg("Couldn't allocate dummy fs info\n"); 282 goto out; 283 } 284 285 root->node = alloc_dummy_extent_buffer(0, 4096); 286 if (!root->node) { 287 test_msg("Couldn't allocate dummy buffer\n"); 288 goto out; 289 } 290 291 /* 292 * We will just free a dummy node if it's ref count is 2 so we need an 293 * extra ref so our searches don't accidently release our page. 294 */ 295 extent_buffer_get(root->node); 296 btrfs_set_header_nritems(root->node, 0); 297 btrfs_set_header_level(root->node, 0); 298 ret = -EINVAL; 299 300 /* First with no extents */ 301 BTRFS_I(inode)->root = root; 302 em = btrfs_get_extent(inode, NULL, 0, 0, 4096, 0); 303 if (IS_ERR(em)) { 304 em = NULL; 305 test_msg("Got an error when we shouldn't have\n"); 306 goto out; 307 } 308 if (em->block_start != EXTENT_MAP_HOLE) { 309 test_msg("Expected a hole, got %llu\n", em->block_start); 310 goto out; 311 } 312 if (!test_bit(EXTENT_FLAG_VACANCY, &em->flags)) { 313 test_msg("Vacancy flag wasn't set properly\n"); 314 goto out; 315 } 316 free_extent_map(em); 317 btrfs_drop_extent_cache(inode, 0, (u64)-1, 0); 318 319 /* 320 * All of the magic numbers are based on the mapping setup in 321 * setup_file_extents, so if you change anything there you need to 322 * update the comment and update the expected values below. 323 */ 324 setup_file_extents(root); 325 326 em = btrfs_get_extent(inode, NULL, 0, 0, (u64)-1, 0); 327 if (IS_ERR(em)) { 328 test_msg("Got an error when we shouldn't have\n"); 329 goto out; 330 } 331 if (em->block_start != EXTENT_MAP_HOLE) { 332 test_msg("Expected a hole, got %llu\n", em->block_start); 333 goto out; 334 } 335 if (em->start != 0 || em->len != 5) { 336 test_msg("Unexpected extent wanted start 0 len 5, got start " 337 "%llu len %llu\n", em->start, em->len); 338 goto out; 339 } 340 if (em->flags != 0) { 341 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 342 goto out; 343 } 344 offset = em->start + em->len; 345 free_extent_map(em); 346 347 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 348 if (IS_ERR(em)) { 349 test_msg("Got an error when we shouldn't have\n"); 350 goto out; 351 } 352 if (em->block_start != EXTENT_MAP_INLINE) { 353 test_msg("Expected an inline, got %llu\n", em->block_start); 354 goto out; 355 } 356 if (em->start != offset || em->len != 4091) { 357 test_msg("Unexpected extent wanted start %llu len 1, got start " 358 "%llu len %llu\n", offset, em->start, em->len); 359 goto out; 360 } 361 if (em->flags != 0) { 362 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 363 goto out; 364 } 365 /* 366 * We don't test anything else for inline since it doesn't get set 367 * unless we have a page for it to write into. Maybe we should change 368 * this? 369 */ 370 offset = em->start + em->len; 371 free_extent_map(em); 372 373 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 374 if (IS_ERR(em)) { 375 test_msg("Got an error when we shouldn't have\n"); 376 goto out; 377 } 378 if (em->block_start != EXTENT_MAP_HOLE) { 379 test_msg("Expected a hole, got %llu\n", em->block_start); 380 goto out; 381 } 382 if (em->start != offset || em->len != 4) { 383 test_msg("Unexpected extent wanted start %llu len 4, got start " 384 "%llu len %llu\n", offset, em->start, em->len); 385 goto out; 386 } 387 if (em->flags != 0) { 388 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 389 goto out; 390 } 391 offset = em->start + em->len; 392 free_extent_map(em); 393 394 /* Regular extent */ 395 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 396 if (IS_ERR(em)) { 397 test_msg("Got an error when we shouldn't have\n"); 398 goto out; 399 } 400 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 401 test_msg("Expected a real extent, got %llu\n", em->block_start); 402 goto out; 403 } 404 if (em->start != offset || em->len != 4095) { 405 test_msg("Unexpected extent wanted start %llu len 4095, got " 406 "start %llu len %llu\n", offset, em->start, em->len); 407 goto out; 408 } 409 if (em->flags != 0) { 410 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 411 goto out; 412 } 413 if (em->orig_start != em->start) { 414 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 415 em->orig_start); 416 goto out; 417 } 418 offset = em->start + em->len; 419 free_extent_map(em); 420 421 /* The next 3 are split extents */ 422 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 423 if (IS_ERR(em)) { 424 test_msg("Got an error when we shouldn't have\n"); 425 goto out; 426 } 427 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 428 test_msg("Expected a real extent, got %llu\n", em->block_start); 429 goto out; 430 } 431 if (em->start != offset || em->len != 4096) { 432 test_msg("Unexpected extent wanted start %llu len 4096, got " 433 "start %llu len %llu\n", offset, em->start, em->len); 434 goto out; 435 } 436 if (em->flags != 0) { 437 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 438 goto out; 439 } 440 if (em->orig_start != em->start) { 441 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 442 em->orig_start); 443 goto out; 444 } 445 disk_bytenr = em->block_start; 446 orig_start = em->start; 447 offset = em->start + em->len; 448 free_extent_map(em); 449 450 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 451 if (IS_ERR(em)) { 452 test_msg("Got an error when we shouldn't have\n"); 453 goto out; 454 } 455 if (em->block_start != EXTENT_MAP_HOLE) { 456 test_msg("Expected a hole, got %llu\n", em->block_start); 457 goto out; 458 } 459 if (em->start != offset || em->len != 4096) { 460 test_msg("Unexpected extent wanted start %llu len 4096, got " 461 "start %llu len %llu\n", offset, em->start, em->len); 462 goto out; 463 } 464 if (em->flags != 0) { 465 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 466 goto out; 467 } 468 offset = em->start + em->len; 469 free_extent_map(em); 470 471 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 472 if (IS_ERR(em)) { 473 test_msg("Got an error when we shouldn't have\n"); 474 goto out; 475 } 476 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 477 test_msg("Expected a real extent, got %llu\n", em->block_start); 478 goto out; 479 } 480 if (em->start != offset || em->len != 8192) { 481 test_msg("Unexpected extent wanted start %llu len 8192, got " 482 "start %llu len %llu\n", offset, em->start, em->len); 483 goto out; 484 } 485 if (em->flags != 0) { 486 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 487 goto out; 488 } 489 if (em->orig_start != orig_start) { 490 test_msg("Wrong orig offset, want %llu, have %llu\n", 491 orig_start, em->orig_start); 492 goto out; 493 } 494 disk_bytenr += (em->start - orig_start); 495 if (em->block_start != disk_bytenr) { 496 test_msg("Wrong block start, want %llu, have %llu\n", 497 disk_bytenr, em->block_start); 498 goto out; 499 } 500 offset = em->start + em->len; 501 free_extent_map(em); 502 503 /* Prealloc extent */ 504 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 505 if (IS_ERR(em)) { 506 test_msg("Got an error when we shouldn't have\n"); 507 goto out; 508 } 509 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 510 test_msg("Expected a real extent, got %llu\n", em->block_start); 511 goto out; 512 } 513 if (em->start != offset || em->len != 4096) { 514 test_msg("Unexpected extent wanted start %llu len 4096, got " 515 "start %llu len %llu\n", offset, em->start, em->len); 516 goto out; 517 } 518 if (em->flags != prealloc_only) { 519 test_msg("Unexpected flags set, want %lu have %lu\n", 520 prealloc_only, em->flags); 521 goto out; 522 } 523 if (em->orig_start != em->start) { 524 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 525 em->orig_start); 526 goto out; 527 } 528 offset = em->start + em->len; 529 free_extent_map(em); 530 531 /* The next 3 are a half written prealloc extent */ 532 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 533 if (IS_ERR(em)) { 534 test_msg("Got an error when we shouldn't have\n"); 535 goto out; 536 } 537 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 538 test_msg("Expected a real extent, got %llu\n", em->block_start); 539 goto out; 540 } 541 if (em->start != offset || em->len != 4096) { 542 test_msg("Unexpected extent wanted start %llu len 4096, got " 543 "start %llu len %llu\n", offset, em->start, em->len); 544 goto out; 545 } 546 if (em->flags != prealloc_only) { 547 test_msg("Unexpected flags set, want %lu have %lu\n", 548 prealloc_only, em->flags); 549 goto out; 550 } 551 if (em->orig_start != em->start) { 552 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 553 em->orig_start); 554 goto out; 555 } 556 disk_bytenr = em->block_start; 557 orig_start = em->start; 558 offset = em->start + em->len; 559 free_extent_map(em); 560 561 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 562 if (IS_ERR(em)) { 563 test_msg("Got an error when we shouldn't have\n"); 564 goto out; 565 } 566 if (em->block_start >= EXTENT_MAP_HOLE) { 567 test_msg("Expected a real extent, got %llu\n", em->block_start); 568 goto out; 569 } 570 if (em->start != offset || em->len != 4096) { 571 test_msg("Unexpected extent wanted start %llu len 4096, got " 572 "start %llu len %llu\n", offset, em->start, em->len); 573 goto out; 574 } 575 if (em->flags != 0) { 576 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 577 goto out; 578 } 579 if (em->orig_start != orig_start) { 580 test_msg("Unexpected orig offset, wanted %llu, have %llu\n", 581 orig_start, em->orig_start); 582 goto out; 583 } 584 if (em->block_start != (disk_bytenr + (em->start - em->orig_start))) { 585 test_msg("Unexpected block start, wanted %llu, have %llu\n", 586 disk_bytenr + (em->start - em->orig_start), 587 em->block_start); 588 goto out; 589 } 590 offset = em->start + em->len; 591 free_extent_map(em); 592 593 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 594 if (IS_ERR(em)) { 595 test_msg("Got an error when we shouldn't have\n"); 596 goto out; 597 } 598 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 599 test_msg("Expected a real extent, got %llu\n", em->block_start); 600 goto out; 601 } 602 if (em->start != offset || em->len != 8192) { 603 test_msg("Unexpected extent wanted start %llu len 8192, got " 604 "start %llu len %llu\n", offset, em->start, em->len); 605 goto out; 606 } 607 if (em->flags != prealloc_only) { 608 test_msg("Unexpected flags set, want %lu have %lu\n", 609 prealloc_only, em->flags); 610 goto out; 611 } 612 if (em->orig_start != orig_start) { 613 test_msg("Wrong orig offset, want %llu, have %llu\n", orig_start, 614 em->orig_start); 615 goto out; 616 } 617 if (em->block_start != (disk_bytenr + (em->start - em->orig_start))) { 618 test_msg("Unexpected block start, wanted %llu, have %llu\n", 619 disk_bytenr + (em->start - em->orig_start), 620 em->block_start); 621 goto out; 622 } 623 offset = em->start + em->len; 624 free_extent_map(em); 625 626 /* Now for the compressed extent */ 627 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 628 if (IS_ERR(em)) { 629 test_msg("Got an error when we shouldn't have\n"); 630 goto out; 631 } 632 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 633 test_msg("Expected a real extent, got %llu\n", em->block_start); 634 goto out; 635 } 636 if (em->start != offset || em->len != 8192) { 637 test_msg("Unexpected extent wanted start %llu len 8192, got " 638 "start %llu len %llu\n", offset, em->start, em->len); 639 goto out; 640 } 641 if (em->flags != compressed_only) { 642 test_msg("Unexpected flags set, want %lu have %lu\n", 643 compressed_only, em->flags); 644 goto out; 645 } 646 if (em->orig_start != em->start) { 647 test_msg("Wrong orig offset, want %llu, have %llu\n", 648 em->start, em->orig_start); 649 goto out; 650 } 651 if (em->compress_type != BTRFS_COMPRESS_ZLIB) { 652 test_msg("Unexpected compress type, wanted %d, got %d\n", 653 BTRFS_COMPRESS_ZLIB, em->compress_type); 654 goto out; 655 } 656 offset = em->start + em->len; 657 free_extent_map(em); 658 659 /* Split compressed extent */ 660 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 661 if (IS_ERR(em)) { 662 test_msg("Got an error when we shouldn't have\n"); 663 goto out; 664 } 665 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 666 test_msg("Expected a real extent, got %llu\n", em->block_start); 667 goto out; 668 } 669 if (em->start != offset || em->len != 4096) { 670 test_msg("Unexpected extent wanted start %llu len 4096, got " 671 "start %llu len %llu\n", offset, em->start, em->len); 672 goto out; 673 } 674 if (em->flags != compressed_only) { 675 test_msg("Unexpected flags set, want %lu have %lu\n", 676 compressed_only, em->flags); 677 goto out; 678 } 679 if (em->orig_start != em->start) { 680 test_msg("Wrong orig offset, want %llu, have %llu\n", 681 em->start, em->orig_start); 682 goto out; 683 } 684 if (em->compress_type != BTRFS_COMPRESS_ZLIB) { 685 test_msg("Unexpected compress type, wanted %d, got %d\n", 686 BTRFS_COMPRESS_ZLIB, em->compress_type); 687 goto out; 688 } 689 disk_bytenr = em->block_start; 690 orig_start = em->start; 691 offset = em->start + em->len; 692 free_extent_map(em); 693 694 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 695 if (IS_ERR(em)) { 696 test_msg("Got an error when we shouldn't have\n"); 697 goto out; 698 } 699 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 700 test_msg("Expected a real extent, got %llu\n", em->block_start); 701 goto out; 702 } 703 if (em->start != offset || em->len != 4096) { 704 test_msg("Unexpected extent wanted start %llu len 4096, got " 705 "start %llu len %llu\n", offset, em->start, em->len); 706 goto out; 707 } 708 if (em->flags != 0) { 709 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 710 goto out; 711 } 712 if (em->orig_start != em->start) { 713 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 714 em->orig_start); 715 goto out; 716 } 717 offset = em->start + em->len; 718 free_extent_map(em); 719 720 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 721 if (IS_ERR(em)) { 722 test_msg("Got an error when we shouldn't have\n"); 723 goto out; 724 } 725 if (em->block_start != disk_bytenr) { 726 test_msg("Block start does not match, want %llu got %llu\n", 727 disk_bytenr, em->block_start); 728 goto out; 729 } 730 if (em->start != offset || em->len != 8192) { 731 test_msg("Unexpected extent wanted start %llu len 8192, got " 732 "start %llu len %llu\n", offset, em->start, em->len); 733 goto out; 734 } 735 if (em->flags != compressed_only) { 736 test_msg("Unexpected flags set, want %lu have %lu\n", 737 compressed_only, em->flags); 738 goto out; 739 } 740 if (em->orig_start != orig_start) { 741 test_msg("Wrong orig offset, want %llu, have %llu\n", 742 em->start, orig_start); 743 goto out; 744 } 745 if (em->compress_type != BTRFS_COMPRESS_ZLIB) { 746 test_msg("Unexpected compress type, wanted %d, got %d\n", 747 BTRFS_COMPRESS_ZLIB, em->compress_type); 748 goto out; 749 } 750 offset = em->start + em->len; 751 free_extent_map(em); 752 753 /* A hole between regular extents but no hole extent */ 754 em = btrfs_get_extent(inode, NULL, 0, offset + 6, 4096, 0); 755 if (IS_ERR(em)) { 756 test_msg("Got an error when we shouldn't have\n"); 757 goto out; 758 } 759 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 760 test_msg("Expected a real extent, got %llu\n", em->block_start); 761 goto out; 762 } 763 if (em->start != offset || em->len != 4096) { 764 test_msg("Unexpected extent wanted start %llu len 4096, got " 765 "start %llu len %llu\n", offset, em->start, em->len); 766 goto out; 767 } 768 if (em->flags != 0) { 769 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 770 goto out; 771 } 772 if (em->orig_start != em->start) { 773 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 774 em->orig_start); 775 goto out; 776 } 777 offset = em->start + em->len; 778 free_extent_map(em); 779 780 em = btrfs_get_extent(inode, NULL, 0, offset, 4096 * 1024, 0); 781 if (IS_ERR(em)) { 782 test_msg("Got an error when we shouldn't have\n"); 783 goto out; 784 } 785 if (em->block_start != EXTENT_MAP_HOLE) { 786 test_msg("Expected a hole extent, got %llu\n", em->block_start); 787 goto out; 788 } 789 /* 790 * Currently we just return a length that we requested rather than the 791 * length of the actual hole, if this changes we'll have to change this 792 * test. 793 */ 794 if (em->start != offset || em->len != 12288) { 795 test_msg("Unexpected extent wanted start %llu len 12288, got " 796 "start %llu len %llu\n", offset, em->start, em->len); 797 goto out; 798 } 799 if (em->flags != vacancy_only) { 800 test_msg("Unexpected flags set, want %lu have %lu\n", 801 vacancy_only, em->flags); 802 goto out; 803 } 804 if (em->orig_start != em->start) { 805 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 806 em->orig_start); 807 goto out; 808 } 809 offset = em->start + em->len; 810 free_extent_map(em); 811 812 em = btrfs_get_extent(inode, NULL, 0, offset, 4096, 0); 813 if (IS_ERR(em)) { 814 test_msg("Got an error when we shouldn't have\n"); 815 goto out; 816 } 817 if (em->block_start >= EXTENT_MAP_LAST_BYTE) { 818 test_msg("Expected a real extent, got %llu\n", em->block_start); 819 goto out; 820 } 821 if (em->start != offset || em->len != 4096) { 822 test_msg("Unexpected extent wanted start %llu len 4096, got " 823 "start %llu len %llu\n", offset, em->start, em->len); 824 goto out; 825 } 826 if (em->flags != 0) { 827 test_msg("Unexpected flags set, want 0 have %lu\n", em->flags); 828 goto out; 829 } 830 if (em->orig_start != em->start) { 831 test_msg("Wrong orig offset, want %llu, have %llu\n", em->start, 832 em->orig_start); 833 goto out; 834 } 835 ret = 0; 836 out: 837 if (!IS_ERR(em)) 838 free_extent_map(em); 839 iput(inode); 840 free_dummy_root(root); 841 return ret; 842 } 843 844 static int test_hole_first(void) 845 { 846 struct inode *inode = NULL; 847 struct btrfs_root *root = NULL; 848 struct extent_map *em = NULL; 849 int ret = -ENOMEM; 850 851 inode = btrfs_new_test_inode(); 852 if (!inode) { 853 test_msg("Couldn't allocate inode\n"); 854 return ret; 855 } 856 857 BTRFS_I(inode)->location.type = BTRFS_INODE_ITEM_KEY; 858 BTRFS_I(inode)->location.objectid = BTRFS_FIRST_FREE_OBJECTID; 859 BTRFS_I(inode)->location.offset = 0; 860 861 root = btrfs_alloc_dummy_root(); 862 if (IS_ERR(root)) { 863 test_msg("Couldn't allocate root\n"); 864 goto out; 865 } 866 867 root->fs_info = alloc_dummy_fs_info(); 868 if (!root->fs_info) { 869 test_msg("Couldn't allocate dummy fs info\n"); 870 goto out; 871 } 872 873 root->node = alloc_dummy_extent_buffer(0, 4096); 874 if (!root->node) { 875 test_msg("Couldn't allocate dummy buffer\n"); 876 goto out; 877 } 878 879 extent_buffer_get(root->node); 880 btrfs_set_header_nritems(root->node, 0); 881 btrfs_set_header_level(root->node, 0); 882 BTRFS_I(inode)->root = root; 883 ret = -EINVAL; 884 885 /* 886 * Need a blank inode item here just so we don't confuse 887 * btrfs_get_extent. 888 */ 889 insert_inode_item_key(root); 890 insert_extent(root, 4096, 4096, 4096, 0, 4096, 4096, 891 BTRFS_FILE_EXTENT_REG, 0, 1); 892 em = btrfs_get_extent(inode, NULL, 0, 0, 8192, 0); 893 if (IS_ERR(em)) { 894 test_msg("Got an error when we shouldn't have\n"); 895 goto out; 896 } 897 if (em->block_start != EXTENT_MAP_HOLE) { 898 test_msg("Expected a hole, got %llu\n", em->block_start); 899 goto out; 900 } 901 if (em->start != 0 || em->len != 4096) { 902 test_msg("Unexpected extent wanted start 0 len 4096, got start " 903 "%llu len %llu\n", em->start, em->len); 904 goto out; 905 } 906 if (em->flags != vacancy_only) { 907 test_msg("Wrong flags, wanted %lu, have %lu\n", vacancy_only, 908 em->flags); 909 goto out; 910 } 911 free_extent_map(em); 912 913 em = btrfs_get_extent(inode, NULL, 0, 4096, 8192, 0); 914 if (IS_ERR(em)) { 915 test_msg("Got an error when we shouldn't have\n"); 916 goto out; 917 } 918 if (em->block_start != 4096) { 919 test_msg("Expected a real extent, got %llu\n", em->block_start); 920 goto out; 921 } 922 if (em->start != 4096 || em->len != 4096) { 923 test_msg("Unexpected extent wanted start 4096 len 4096, got " 924 "start %llu len %llu\n", em->start, em->len); 925 goto out; 926 } 927 if (em->flags != 0) { 928 test_msg("Unexpected flags set, wanted 0 got %lu\n", 929 em->flags); 930 goto out; 931 } 932 ret = 0; 933 out: 934 if (!IS_ERR(em)) 935 free_extent_map(em); 936 iput(inode); 937 free_dummy_root(root); 938 return ret; 939 } 940 941 int btrfs_test_inodes(void) 942 { 943 int ret; 944 945 set_bit(EXTENT_FLAG_COMPRESSED, &compressed_only); 946 set_bit(EXTENT_FLAG_VACANCY, &vacancy_only); 947 set_bit(EXTENT_FLAG_PREALLOC, &prealloc_only); 948 949 test_msg("Running btrfs_get_extent tests\n"); 950 ret = test_btrfs_get_extent(); 951 if (ret) 952 return ret; 953 test_msg("Running hole first btrfs_get_extent test\n"); 954 return test_hole_first(); 955 } 956