1 // SPDX-License-Identifier: GPL-2.0 2 /* 3 * Copyright (c) 2018 Red Hat, Inc. 4 * 5 * This is a test "dust" device, which fails reads on specified 6 * sectors, emulating the behavior of a hard disk drive sending 7 * a "Read Medium Error" sense. 8 * 9 */ 10 11 #include <linux/device-mapper.h> 12 #include <linux/module.h> 13 #include <linux/rbtree.h> 14 15 #define DM_MSG_PREFIX "dust" 16 17 struct badblock { 18 struct rb_node node; 19 sector_t bb; 20 unsigned char wr_fail_cnt; 21 }; 22 23 struct dust_device { 24 struct dm_dev *dev; 25 struct rb_root badblocklist; 26 unsigned long long badblock_count; 27 spinlock_t dust_lock; 28 unsigned int blksz; 29 int sect_per_block_shift; 30 unsigned int sect_per_block; 31 sector_t start; 32 bool fail_read_on_bb:1; 33 bool quiet_mode:1; 34 }; 35 36 static struct badblock *dust_rb_search(struct rb_root *root, sector_t blk) 37 { 38 struct rb_node *node = root->rb_node; 39 40 while (node) { 41 struct badblock *bblk = rb_entry(node, struct badblock, node); 42 43 if (bblk->bb > blk) 44 node = node->rb_left; 45 else if (bblk->bb < blk) 46 node = node->rb_right; 47 else 48 return bblk; 49 } 50 51 return NULL; 52 } 53 54 static bool dust_rb_insert(struct rb_root *root, struct badblock *new) 55 { 56 struct badblock *bblk; 57 struct rb_node **link = &root->rb_node, *parent = NULL; 58 sector_t value = new->bb; 59 60 while (*link) { 61 parent = *link; 62 bblk = rb_entry(parent, struct badblock, node); 63 64 if (bblk->bb > value) 65 link = &(*link)->rb_left; 66 else if (bblk->bb < value) 67 link = &(*link)->rb_right; 68 else 69 return false; 70 } 71 72 rb_link_node(&new->node, parent, link); 73 rb_insert_color(&new->node, root); 74 75 return true; 76 } 77 78 static int dust_remove_block(struct dust_device *dd, unsigned long long block) 79 { 80 struct badblock *bblock; 81 unsigned long flags; 82 83 spin_lock_irqsave(&dd->dust_lock, flags); 84 bblock = dust_rb_search(&dd->badblocklist, block); 85 86 if (bblock == NULL) { 87 if (!dd->quiet_mode) { 88 DMERR("%s: block %llu not found in badblocklist", 89 __func__, block); 90 } 91 spin_unlock_irqrestore(&dd->dust_lock, flags); 92 return -EINVAL; 93 } 94 95 rb_erase(&bblock->node, &dd->badblocklist); 96 dd->badblock_count--; 97 if (!dd->quiet_mode) 98 DMINFO("%s: badblock removed at block %llu", __func__, block); 99 kfree(bblock); 100 spin_unlock_irqrestore(&dd->dust_lock, flags); 101 102 return 0; 103 } 104 105 static int dust_add_block(struct dust_device *dd, unsigned long long block, 106 unsigned char wr_fail_cnt) 107 { 108 struct badblock *bblock; 109 unsigned long flags; 110 111 bblock = kmalloc(sizeof(*bblock), GFP_KERNEL); 112 if (bblock == NULL) { 113 if (!dd->quiet_mode) 114 DMERR("%s: badblock allocation failed", __func__); 115 return -ENOMEM; 116 } 117 118 spin_lock_irqsave(&dd->dust_lock, flags); 119 bblock->bb = block; 120 bblock->wr_fail_cnt = wr_fail_cnt; 121 if (!dust_rb_insert(&dd->badblocklist, bblock)) { 122 if (!dd->quiet_mode) { 123 DMERR("%s: block %llu already in badblocklist", 124 __func__, block); 125 } 126 spin_unlock_irqrestore(&dd->dust_lock, flags); 127 kfree(bblock); 128 return -EINVAL; 129 } 130 131 dd->badblock_count++; 132 if (!dd->quiet_mode) { 133 DMINFO("%s: badblock added at block %llu with write fail count %u", 134 __func__, block, wr_fail_cnt); 135 } 136 spin_unlock_irqrestore(&dd->dust_lock, flags); 137 138 return 0; 139 } 140 141 static int dust_query_block(struct dust_device *dd, unsigned long long block, char *result, 142 unsigned int maxlen, unsigned int *sz_ptr) 143 { 144 struct badblock *bblock; 145 unsigned long flags; 146 unsigned int sz = *sz_ptr; 147 148 spin_lock_irqsave(&dd->dust_lock, flags); 149 bblock = dust_rb_search(&dd->badblocklist, block); 150 if (bblock != NULL) 151 DMEMIT("%s: block %llu found in badblocklist", __func__, block); 152 else 153 DMEMIT("%s: block %llu not found in badblocklist", __func__, block); 154 spin_unlock_irqrestore(&dd->dust_lock, flags); 155 156 return 1; 157 } 158 159 static int __dust_map_read(struct dust_device *dd, sector_t thisblock) 160 { 161 struct badblock *bblk = dust_rb_search(&dd->badblocklist, thisblock); 162 163 if (bblk) 164 return DM_MAPIO_KILL; 165 166 return DM_MAPIO_REMAPPED; 167 } 168 169 static int dust_map_read(struct dust_device *dd, sector_t thisblock, 170 bool fail_read_on_bb) 171 { 172 unsigned long flags; 173 int r = DM_MAPIO_REMAPPED; 174 175 if (fail_read_on_bb) { 176 thisblock >>= dd->sect_per_block_shift; 177 spin_lock_irqsave(&dd->dust_lock, flags); 178 r = __dust_map_read(dd, thisblock); 179 spin_unlock_irqrestore(&dd->dust_lock, flags); 180 } 181 182 return r; 183 } 184 185 static int __dust_map_write(struct dust_device *dd, sector_t thisblock) 186 { 187 struct badblock *bblk = dust_rb_search(&dd->badblocklist, thisblock); 188 189 if (bblk && bblk->wr_fail_cnt > 0) { 190 bblk->wr_fail_cnt--; 191 return DM_MAPIO_KILL; 192 } 193 194 if (bblk) { 195 rb_erase(&bblk->node, &dd->badblocklist); 196 dd->badblock_count--; 197 kfree(bblk); 198 if (!dd->quiet_mode) { 199 sector_div(thisblock, dd->sect_per_block); 200 DMINFO("block %llu removed from badblocklist by write", 201 (unsigned long long)thisblock); 202 } 203 } 204 205 return DM_MAPIO_REMAPPED; 206 } 207 208 static int dust_map_write(struct dust_device *dd, sector_t thisblock, 209 bool fail_read_on_bb) 210 { 211 unsigned long flags; 212 int r = DM_MAPIO_REMAPPED; 213 214 if (fail_read_on_bb) { 215 thisblock >>= dd->sect_per_block_shift; 216 spin_lock_irqsave(&dd->dust_lock, flags); 217 r = __dust_map_write(dd, thisblock); 218 spin_unlock_irqrestore(&dd->dust_lock, flags); 219 } 220 221 return r; 222 } 223 224 static int dust_map(struct dm_target *ti, struct bio *bio) 225 { 226 struct dust_device *dd = ti->private; 227 int r; 228 229 bio_set_dev(bio, dd->dev->bdev); 230 bio->bi_iter.bi_sector = dd->start + dm_target_offset(ti, bio->bi_iter.bi_sector); 231 232 if (bio_data_dir(bio) == READ) 233 r = dust_map_read(dd, bio->bi_iter.bi_sector, dd->fail_read_on_bb); 234 else 235 r = dust_map_write(dd, bio->bi_iter.bi_sector, dd->fail_read_on_bb); 236 237 return r; 238 } 239 240 static bool __dust_clear_badblocks(struct rb_root *tree, 241 unsigned long long count) 242 { 243 struct rb_node *node = NULL, *nnode = NULL; 244 245 nnode = rb_first(tree); 246 if (nnode == NULL) { 247 BUG_ON(count != 0); 248 return false; 249 } 250 251 while (nnode) { 252 node = nnode; 253 nnode = rb_next(node); 254 rb_erase(node, tree); 255 count--; 256 kfree(node); 257 } 258 BUG_ON(count != 0); 259 BUG_ON(tree->rb_node != NULL); 260 261 return true; 262 } 263 264 static int dust_clear_badblocks(struct dust_device *dd, char *result, unsigned int maxlen, 265 unsigned int *sz_ptr) 266 { 267 unsigned long flags; 268 struct rb_root badblocklist; 269 unsigned long long badblock_count; 270 unsigned int sz = *sz_ptr; 271 272 spin_lock_irqsave(&dd->dust_lock, flags); 273 badblocklist = dd->badblocklist; 274 badblock_count = dd->badblock_count; 275 dd->badblocklist = RB_ROOT; 276 dd->badblock_count = 0; 277 spin_unlock_irqrestore(&dd->dust_lock, flags); 278 279 if (!__dust_clear_badblocks(&badblocklist, badblock_count)) 280 DMEMIT("%s: no badblocks found", __func__); 281 else 282 DMEMIT("%s: badblocks cleared", __func__); 283 284 return 1; 285 } 286 287 static int dust_list_badblocks(struct dust_device *dd, char *result, unsigned int maxlen, 288 unsigned int *sz_ptr) 289 { 290 unsigned long flags; 291 struct rb_root badblocklist; 292 struct rb_node *node; 293 struct badblock *bblk; 294 unsigned int sz = *sz_ptr; 295 unsigned long long num = 0; 296 297 spin_lock_irqsave(&dd->dust_lock, flags); 298 badblocklist = dd->badblocklist; 299 for (node = rb_first(&badblocklist); node; node = rb_next(node)) { 300 bblk = rb_entry(node, struct badblock, node); 301 DMEMIT("%llu\n", bblk->bb); 302 num++; 303 } 304 305 spin_unlock_irqrestore(&dd->dust_lock, flags); 306 if (!num) 307 DMEMIT("No blocks in badblocklist"); 308 309 return 1; 310 } 311 312 /* 313 * Target parameters: 314 * 315 * <device_path> <offset> <blksz> 316 * 317 * device_path: path to the block device 318 * offset: offset to data area from start of device_path 319 * blksz: block size (minimum 512, maximum 1073741824, must be a power of 2) 320 */ 321 static int dust_ctr(struct dm_target *ti, unsigned int argc, char **argv) 322 { 323 struct dust_device *dd; 324 unsigned long long tmp; 325 char dummy; 326 unsigned int blksz; 327 unsigned int sect_per_block; 328 sector_t DUST_MAX_BLKSZ_SECTORS = 2097152; 329 sector_t max_block_sectors = min(ti->len, DUST_MAX_BLKSZ_SECTORS); 330 331 if (argc != 3) { 332 ti->error = "Invalid argument count"; 333 return -EINVAL; 334 } 335 336 if (kstrtouint(argv[2], 10, &blksz) || !blksz) { 337 ti->error = "Invalid block size parameter"; 338 return -EINVAL; 339 } 340 341 if (blksz < 512) { 342 ti->error = "Block size must be at least 512"; 343 return -EINVAL; 344 } 345 346 if (!is_power_of_2(blksz)) { 347 ti->error = "Block size must be a power of 2"; 348 return -EINVAL; 349 } 350 351 if (to_sector(blksz) > max_block_sectors) { 352 ti->error = "Block size is too large"; 353 return -EINVAL; 354 } 355 356 sect_per_block = (blksz >> SECTOR_SHIFT); 357 358 if (sscanf(argv[1], "%llu%c", &tmp, &dummy) != 1 || tmp != (sector_t)tmp) { 359 ti->error = "Invalid device offset sector"; 360 return -EINVAL; 361 } 362 363 dd = kzalloc(sizeof(struct dust_device), GFP_KERNEL); 364 if (dd == NULL) { 365 ti->error = "Cannot allocate context"; 366 return -ENOMEM; 367 } 368 369 if (dm_get_device(ti, argv[0], dm_table_get_mode(ti->table), &dd->dev)) { 370 ti->error = "Device lookup failed"; 371 kfree(dd); 372 return -EINVAL; 373 } 374 375 dd->sect_per_block = sect_per_block; 376 dd->blksz = blksz; 377 dd->start = tmp; 378 379 dd->sect_per_block_shift = __ffs(sect_per_block); 380 381 /* 382 * Whether to fail a read on a "bad" block. 383 * Defaults to false; enabled later by message. 384 */ 385 dd->fail_read_on_bb = false; 386 387 /* 388 * Initialize bad block list rbtree. 389 */ 390 dd->badblocklist = RB_ROOT; 391 dd->badblock_count = 0; 392 spin_lock_init(&dd->dust_lock); 393 394 dd->quiet_mode = false; 395 396 BUG_ON(dm_set_target_max_io_len(ti, dd->sect_per_block) != 0); 397 398 ti->num_discard_bios = 1; 399 ti->num_flush_bios = 1; 400 ti->private = dd; 401 402 return 0; 403 } 404 405 static void dust_dtr(struct dm_target *ti) 406 { 407 struct dust_device *dd = ti->private; 408 409 __dust_clear_badblocks(&dd->badblocklist, dd->badblock_count); 410 dm_put_device(ti, dd->dev); 411 kfree(dd); 412 } 413 414 static int dust_message(struct dm_target *ti, unsigned int argc, char **argv, 415 char *result, unsigned int maxlen) 416 { 417 struct dust_device *dd = ti->private; 418 sector_t size = bdev_nr_sectors(dd->dev->bdev); 419 bool invalid_msg = false; 420 int r = -EINVAL; 421 unsigned long long tmp, block; 422 unsigned char wr_fail_cnt; 423 unsigned int tmp_ui; 424 unsigned long flags; 425 unsigned int sz = 0; 426 char dummy; 427 428 if (argc == 1) { 429 if (!strcasecmp(argv[0], "addbadblock") || 430 !strcasecmp(argv[0], "removebadblock") || 431 !strcasecmp(argv[0], "queryblock")) { 432 DMERR("%s requires an additional argument", argv[0]); 433 } else if (!strcasecmp(argv[0], "disable")) { 434 DMINFO("disabling read failures on bad sectors"); 435 dd->fail_read_on_bb = false; 436 r = 0; 437 } else if (!strcasecmp(argv[0], "enable")) { 438 DMINFO("enabling read failures on bad sectors"); 439 dd->fail_read_on_bb = true; 440 r = 0; 441 } else if (!strcasecmp(argv[0], "countbadblocks")) { 442 spin_lock_irqsave(&dd->dust_lock, flags); 443 DMEMIT("countbadblocks: %llu badblock(s) found", 444 dd->badblock_count); 445 spin_unlock_irqrestore(&dd->dust_lock, flags); 446 r = 1; 447 } else if (!strcasecmp(argv[0], "clearbadblocks")) { 448 r = dust_clear_badblocks(dd, result, maxlen, &sz); 449 } else if (!strcasecmp(argv[0], "quiet")) { 450 if (!dd->quiet_mode) 451 dd->quiet_mode = true; 452 else 453 dd->quiet_mode = false; 454 r = 0; 455 } else if (!strcasecmp(argv[0], "listbadblocks")) { 456 r = dust_list_badblocks(dd, result, maxlen, &sz); 457 } else { 458 invalid_msg = true; 459 } 460 } else if (argc == 2) { 461 if (sscanf(argv[1], "%llu%c", &tmp, &dummy) != 1) 462 return r; 463 464 block = tmp; 465 sector_div(size, dd->sect_per_block); 466 if (block > size) { 467 DMERR("selected block value out of range"); 468 return r; 469 } 470 471 if (!strcasecmp(argv[0], "addbadblock")) 472 r = dust_add_block(dd, block, 0); 473 else if (!strcasecmp(argv[0], "removebadblock")) 474 r = dust_remove_block(dd, block); 475 else if (!strcasecmp(argv[0], "queryblock")) 476 r = dust_query_block(dd, block, result, maxlen, &sz); 477 else 478 invalid_msg = true; 479 480 } else if (argc == 3) { 481 if (sscanf(argv[1], "%llu%c", &tmp, &dummy) != 1) 482 return r; 483 484 if (sscanf(argv[2], "%u%c", &tmp_ui, &dummy) != 1) 485 return r; 486 487 block = tmp; 488 if (tmp_ui > 255) { 489 DMERR("selected write fail count out of range"); 490 return r; 491 } 492 wr_fail_cnt = tmp_ui; 493 sector_div(size, dd->sect_per_block); 494 if (block > size) { 495 DMERR("selected block value out of range"); 496 return r; 497 } 498 499 if (!strcasecmp(argv[0], "addbadblock")) 500 r = dust_add_block(dd, block, wr_fail_cnt); 501 else 502 invalid_msg = true; 503 504 } else 505 DMERR("invalid number of arguments '%d'", argc); 506 507 if (invalid_msg) 508 DMERR("unrecognized message '%s' received", argv[0]); 509 510 return r; 511 } 512 513 static void dust_status(struct dm_target *ti, status_type_t type, 514 unsigned int status_flags, char *result, unsigned int maxlen) 515 { 516 struct dust_device *dd = ti->private; 517 unsigned int sz = 0; 518 519 switch (type) { 520 case STATUSTYPE_INFO: 521 DMEMIT("%s %s %s", dd->dev->name, 522 dd->fail_read_on_bb ? "fail_read_on_bad_block" : "bypass", 523 dd->quiet_mode ? "quiet" : "verbose"); 524 break; 525 526 case STATUSTYPE_TABLE: 527 DMEMIT("%s %llu %u", dd->dev->name, 528 (unsigned long long)dd->start, dd->blksz); 529 break; 530 531 case STATUSTYPE_IMA: 532 *result = '\0'; 533 break; 534 } 535 } 536 537 static int dust_prepare_ioctl(struct dm_target *ti, struct block_device **bdev) 538 { 539 struct dust_device *dd = ti->private; 540 struct dm_dev *dev = dd->dev; 541 542 *bdev = dev->bdev; 543 544 /* 545 * Only pass ioctls through if the device sizes match exactly. 546 */ 547 if (dd->start || ti->len != bdev_nr_sectors(dev->bdev)) 548 return 1; 549 550 return 0; 551 } 552 553 static int dust_iterate_devices(struct dm_target *ti, iterate_devices_callout_fn fn, 554 void *data) 555 { 556 struct dust_device *dd = ti->private; 557 558 return fn(ti, dd->dev, dd->start, ti->len, data); 559 } 560 561 static struct target_type dust_target = { 562 .name = "dust", 563 .version = {1, 0, 0}, 564 .module = THIS_MODULE, 565 .ctr = dust_ctr, 566 .dtr = dust_dtr, 567 .iterate_devices = dust_iterate_devices, 568 .map = dust_map, 569 .message = dust_message, 570 .status = dust_status, 571 .prepare_ioctl = dust_prepare_ioctl, 572 }; 573 574 static int __init dm_dust_init(void) 575 { 576 int r = dm_register_target(&dust_target); 577 578 if (r < 0) 579 DMERR("dm_register_target failed %d", r); 580 581 return r; 582 } 583 584 static void __exit dm_dust_exit(void) 585 { 586 dm_unregister_target(&dust_target); 587 } 588 589 module_init(dm_dust_init); 590 module_exit(dm_dust_exit); 591 592 MODULE_DESCRIPTION(DM_NAME " dust test target"); 593 MODULE_AUTHOR("Bryan Gurney <dm-devel@redhat.com>"); 594 MODULE_LICENSE("GPL"); 595