drivers/lightnvm/gennvm.c

   1 /*
   2  * Copyright (C) 2015 Matias Bjorling <m@bjorling.me>
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License version
   6  * 2 as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful, but
   9  * 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 License
  14  * along with this program; see the file COPYING.  If not, write to
  15  * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139,
  16  * USA.
  17  *
  18  * Implementation of a general nvm manager for Open-Channel SSDs.
  19  */
  20
  21 #include "gennvm.h"
  22
  23 static struct nvm_target *gen_find_target(struct gen_dev *gn, const char *name)
  24 {
  25         struct nvm_target *tgt;
  26
  27         list_for_each_entry(tgt, &gn->targets, list)
  28                 if (!strcmp(name, tgt->disk->disk_name))
  29                         return tgt;
  30
  31         return NULL;
  32 }
  33
  34 static const struct block_device_operations gen_fops = {
  35         .owner          = THIS_MODULE,
  36 };
  37
  38 static int gen_create_tgt(struct nvm_dev *dev, struct nvm_ioctl_create *create)
  39 {
  40         struct gen_dev *gn = dev->mp;
  41         struct nvm_ioctl_create_simple *s = &create->conf.s;
  42         struct request_queue *tqueue;
  43         struct gendisk *tdisk;
  44         struct nvm_tgt_type *tt;
  45         struct nvm_target *t;
  46         void *targetdata;
  47
  48         tt = nvm_find_target_type(create->tgttype, 1);
  49         if (!tt) {
  50                 pr_err("nvm: target type %s not found\n", create->tgttype);
  51                 return -EINVAL;
  52         }
  53
  54         mutex_lock(&gn->lock);
  55         t = gen_find_target(gn, create->tgtname);
  56         if (t) {
  57                 pr_err("nvm: target name already exists.\n");
  58                 mutex_unlock(&gn->lock);
  59                 return -EINVAL;
  60         }
  61         mutex_unlock(&gn->lock);
  62
  63         t = kmalloc(sizeof(struct nvm_target), GFP_KERNEL);
  64         if (!t)
  65                 return -ENOMEM;
  66
  67         tqueue = blk_alloc_queue_node(GFP_KERNEL, dev->q->node);
  68         if (!tqueue)
  69                 goto err_t;
  70         blk_queue_make_request(tqueue, tt->make_rq);
  71
  72         tdisk = alloc_disk(0);
  73         if (!tdisk)
  74                 goto err_queue;
  75
  76         sprintf(tdisk->disk_name, "%s", create->tgtname);
  77         tdisk->flags = GENHD_FL_EXT_DEVT;
  78         tdisk->major = 0;
  79         tdisk->first_minor = 0;
  80         tdisk->fops = &gen_fops;
  81         tdisk->queue = tqueue;
  82
  83         targetdata = tt->init(dev, tdisk, s->lun_begin, s->lun_end);
  84         if (IS_ERR(targetdata))
  85                 goto err_init;
  86
  87         tdisk->private_data = targetdata;
  88         tqueue->queuedata = targetdata;
  89
  90         blk_queue_max_hw_sectors(tqueue, 8 * dev->ops->max_phys_sect);
  91
  92         set_capacity(tdisk, tt->capacity(targetdata));
  93         add_disk(tdisk);
  94
  95         t->type = tt;
  96         t->disk = tdisk;
  97         t->dev = dev;
  98
  99         mutex_lock(&gn->lock);
 100         list_add_tail(&t->list, &gn->targets);
 101         mutex_unlock(&gn->lock);
 102
 103         return 0;
 104 err_init:
 105         put_disk(tdisk);
 106 err_queue:
 107         blk_cleanup_queue(tqueue);
 108 err_t:
 109         kfree(t);
 110         return -ENOMEM;
 111 }
 112
 113 static void __gen_remove_target(struct nvm_target *t)
 114 {
 115         struct nvm_tgt_type *tt = t->type;
 116         struct gendisk *tdisk = t->disk;
 117         struct request_queue *q = tdisk->queue;
 118
 119         del_gendisk(tdisk);
 120         blk_cleanup_queue(q);
 121
 122         if (tt->exit)
 123                 tt->exit(tdisk->private_data);
 124
 125         put_disk(tdisk);
 126
 127         list_del(&t->list);
 128         kfree(t);
 129 }
 130
 131 /**
 132  * gen_remove_tgt - Removes a target from the media manager
 133  * @dev:        device
 134  * @remove:     ioctl structure with target name to remove.
 135  *
 136  * Returns:
 137  * 0: on success
 138  * 1: on not found
 139  * <0: on error
 140  */
 141 static int gen_remove_tgt(struct nvm_dev *dev, struct nvm_ioctl_remove *remove)
 142 {
 143         struct gen_dev *gn = dev->mp;
 144         struct nvm_target *t;
 145
 146         if (!gn)
 147                 return 1;
 148
 149         mutex_lock(&gn->lock);
 150         t = gen_find_target(gn, remove->tgtname);
 151         if (!t) {
 152                 mutex_unlock(&gn->lock);
 153                 return 1;
 154         }
 155         __gen_remove_target(t);
 156         mutex_unlock(&gn->lock);
 157
 158         return 0;
 159 }
 160
 161 static int gen_get_area(struct nvm_dev *dev, sector_t *lba, sector_t len)
 162 {
 163         struct gen_dev *gn = dev->mp;
 164         struct gen_area *area, *prev, *next;
 165         sector_t begin = 0;
 166         sector_t max_sectors = (dev->sec_size * dev->total_secs) >> 9;
 167
 168         if (len > max_sectors)
 169                 return -EINVAL;
 170
 171         area = kmalloc(sizeof(struct gen_area), GFP_KERNEL);
 172         if (!area)
 173                 return -ENOMEM;
 174
 175         prev = NULL;
 176
 177         spin_lock(&dev->lock);
 178         list_for_each_entry(next, &gn->area_list, list) {
 179                 if (begin + len > next->begin) {
 180                         begin = next->end;
 181                         prev = next;
 182                         continue;
 183                 }
 184                 break;
 185         }
 186
 187         if ((begin + len) > max_sectors) {
 188                 spin_unlock(&dev->lock);
 189                 kfree(area);
 190                 return -EINVAL;
 191         }
 192
 193         area->begin = *lba = begin;
 194         area->end = begin + len;
 195
 196         if (prev) /* insert into sorted order */
 197                 list_add(&area->list, &prev->list);
 198         else
 199                 list_add(&area->list, &gn->area_list);
 200         spin_unlock(&dev->lock);
 201
 202         return 0;
 203 }
 204
 205 static void gen_put_area(struct nvm_dev *dev, sector_t begin)
 206 {
 207         struct gen_dev *gn = dev->mp;
 208         struct gen_area *area;
 209
 210         spin_lock(&dev->lock);
 211         list_for_each_entry(area, &gn->area_list, list) {
 212                 if (area->begin != begin)
 213                         continue;
 214
 215                 list_del(&area->list);
 216                 spin_unlock(&dev->lock);
 217                 kfree(area);
 218                 return;
 219         }
 220         spin_unlock(&dev->lock);
 221 }
 222
 223 static void gen_blocks_free(struct nvm_dev *dev)
 224 {
 225         struct gen_dev *gn = dev->mp;
 226         struct gen_lun *lun;
 227         int i;
 228
 229         gen_for_each_lun(gn, lun, i) {
 230                 if (!lun->vlun.blocks)
 231                         break;
 232                 vfree(lun->vlun.blocks);
 233         }
 234 }
 235
 236 static void gen_luns_free(struct nvm_dev *dev)
 237 {
 238         struct gen_dev *gn = dev->mp;
 239
 240         kfree(gn->luns);
 241 }
 242
 243 static int gen_luns_init(struct nvm_dev *dev, struct gen_dev *gn)
 244 {
 245         struct gen_lun *lun;
 246         int i;
 247
 248         gn->luns = kcalloc(dev->nr_luns, sizeof(struct gen_lun), GFP_KERNEL);
 249         if (!gn->luns)
 250                 return -ENOMEM;
 251
 252         gen_for_each_lun(gn, lun, i) {
 253                 spin_lock_init(&lun->vlun.lock);
 254                 INIT_LIST_HEAD(&lun->free_list);
 255                 INIT_LIST_HEAD(&lun->used_list);
 256                 INIT_LIST_HEAD(&lun->bb_list);
 257
 258                 lun->reserved_blocks = 2; /* for GC only */
 259                 lun->vlun.id = i;
 260                 lun->vlun.lun_id = i % dev->luns_per_chnl;
 261                 lun->vlun.chnl_id = i / dev->luns_per_chnl;
 262                 lun->vlun.nr_free_blocks = dev->blks_per_lun;
 263         }
 264         return 0;
 265 }
 266
 267 static int gen_block_bb(struct gen_dev *gn, struct ppa_addr ppa,
 268                                                         u8 *blks, int nr_blks)
 269 {
 270         struct nvm_dev *dev = gn->dev;
 271         struct gen_lun *lun;
 272         struct nvm_block *blk;
 273         int i;
 274
 275         nr_blks = nvm_bb_tbl_fold(dev, blks, nr_blks);
 276         if (nr_blks < 0)
 277                 return nr_blks;
 278
 279         lun = &gn->luns[(dev->luns_per_chnl * ppa.g.ch) + ppa.g.lun];
 280
 281         for (i = 0; i < nr_blks; i++) {
 282                 if (blks[i] == 0)
 283                         continue;
 284
 285                 blk = &lun->vlun.blocks[i];
 286                 list_move_tail(&blk->list, &lun->bb_list);
 287                 lun->vlun.nr_free_blocks--;
 288         }
 289
 290         return 0;
 291 }
 292
 293 static int gen_block_map(u64 slba, u32 nlb, __le64 *entries, void *private)
 294 {
 295         struct nvm_dev *dev = private;
 296         struct gen_dev *gn = dev->mp;
 297         u64 elba = slba + nlb;
 298         struct gen_lun *lun;
 299         struct nvm_block *blk;
 300         u64 i;
 301         int lun_id;
 302
 303         if (unlikely(elba > dev->total_secs)) {
 304                 pr_err("gen: L2P data from device is out of bounds!\n");
 305                 return -EINVAL;
 306         }
 307
 308         for (i = 0; i < nlb; i++) {
 309                 u64 pba = le64_to_cpu(entries[i]);
 310
 311                 if (unlikely(pba >= dev->total_secs && pba != U64_MAX)) {
 312                         pr_err("gen: L2P data entry is out of bounds!\n");
 313                         return -EINVAL;
 314                 }
 315
 316                 /* Address zero is a special one. The first page on a disk is
 317                  * protected. It often holds internal device boot
 318                  * information.
 319                  */
 320                 if (!pba)
 321                         continue;
 322
 323                 /* resolve block from physical address */
 324                 lun_id = div_u64(pba, dev->sec_per_lun);
 325                 lun = &gn->luns[lun_id];
 326
 327                 /* Calculate block offset into lun */
 328                 pba = pba - (dev->sec_per_lun * lun_id);
 329                 blk = &lun->vlun.blocks[div_u64(pba, dev->sec_per_blk)];
 330
 331                 if (!blk->state) {
 332                         /* at this point, we don't know anything about the
 333                          * block. It's up to the FTL on top to re-etablish the
 334                          * block state. The block is assumed to be open.
 335                          */
 336                         list_move_tail(&blk->list, &lun->used_list);
 337                         blk->state = NVM_BLK_ST_TGT;
 338                         lun->vlun.nr_free_blocks--;
 339                 }
 340         }
 341
 342         return 0;
 343 }
 344
 345 static int gen_blocks_init(struct nvm_dev *dev, struct gen_dev *gn)
 346 {
 347         struct gen_lun *lun;
 348         struct nvm_block *block;
 349         sector_t lun_iter, blk_iter, cur_block_id = 0;
 350         int ret, nr_blks;
 351         u8 *blks;
 352
 353         nr_blks = dev->blks_per_lun * dev->plane_mode;
 354         blks = kmalloc(nr_blks, GFP_KERNEL);
 355         if (!blks)
 356                 return -ENOMEM;
 357
 358         gen_for_each_lun(gn, lun, lun_iter) {
 359                 lun->vlun.blocks = vzalloc(sizeof(struct nvm_block) *
 360                                                         dev->blks_per_lun);
 361                 if (!lun->vlun.blocks) {
 362                         kfree(blks);
 363                         return -ENOMEM;
 364                 }
 365
 366                 for (blk_iter = 0; blk_iter < dev->blks_per_lun; blk_iter++) {
 367                         block = &lun->vlun.blocks[blk_iter];
 368
 369                         INIT_LIST_HEAD(&block->list);
 370
 371                         block->lun = &lun->vlun;
 372                         block->id = cur_block_id++;
 373
 374                         list_add_tail(&block->list, &lun->free_list);
 375                 }
 376
 377                 if (dev->ops->get_bb_tbl) {
 378                         struct ppa_addr ppa;
 379
 380                         ppa.ppa = 0;
 381                         ppa.g.ch = lun->vlun.chnl_id;
 382                         ppa.g.lun = lun->vlun.lun_id;
 383
 384                         ret = nvm_get_bb_tbl(dev, ppa, blks);
 385                         if (ret)
 386                                 pr_err("gen: could not get BB table\n");
 387
 388                         ret = gen_block_bb(gn, ppa, blks, nr_blks);
 389                         if (ret)
 390                                 pr_err("gen: BB table map failed\n");
 391                 }
 392         }
 393
 394         if ((dev->identity.dom & NVM_RSP_L2P) && dev->ops->get_l2p_tbl) {
 395                 ret = dev->ops->get_l2p_tbl(dev, 0, dev->total_secs,
 396                                                         gen_block_map, dev);
 397                 if (ret) {
 398                         pr_err("gen: could not read L2P table.\n");
 399                         pr_warn("gen: default block initialization");
 400                 }
 401         }
 402
 403         kfree(blks);
 404         return 0;
 405 }
 406
 407 static void gen_free(struct nvm_dev *dev)
 408 {
 409         gen_blocks_free(dev);
 410         gen_luns_free(dev);
 411         kfree(dev->mp);
 412         dev->mp = NULL;
 413 }
 414
 415 static int gen_register(struct nvm_dev *dev)
 416 {
 417         struct gen_dev *gn;
 418         int ret;
 419
 420         if (!try_module_get(THIS_MODULE))
 421                 return -ENODEV;
 422
 423         gn = kzalloc(sizeof(struct gen_dev), GFP_KERNEL);
 424         if (!gn)
 425                 return -ENOMEM;
 426
 427         gn->dev = dev;
 428         gn->nr_luns = dev->nr_luns;
 429         INIT_LIST_HEAD(&gn->area_list);
 430         mutex_init(&gn->lock);
 431         INIT_LIST_HEAD(&gn->targets);
 432         dev->mp = gn;
 433
 434         ret = gen_luns_init(dev, gn);
 435         if (ret) {
 436                 pr_err("gen: could not initialize luns\n");
 437                 goto err;
 438         }
 439
 440         ret = gen_blocks_init(dev, gn);
 441         if (ret) {
 442                 pr_err("gen: could not initialize blocks\n");
 443                 goto err;
 444         }
 445
 446         return 1;
 447 err:
 448         gen_free(dev);
 449         module_put(THIS_MODULE);
 450         return ret;
 451 }
 452
 453 static void gen_unregister(struct nvm_dev *dev)
 454 {
 455         struct gen_dev *gn = dev->mp;
 456         struct nvm_target *t, *tmp;
 457
 458         mutex_lock(&gn->lock);
 459         list_for_each_entry_safe(t, tmp, &gn->targets, list) {
 460                 if (t->dev != dev)
 461                         continue;
 462                 __gen_remove_target(t);
 463         }
 464         mutex_unlock(&gn->lock);
 465
 466         gen_free(dev);
 467         module_put(THIS_MODULE);
 468 }
 469
 470 static struct nvm_block *gen_get_blk(struct nvm_dev *dev,
 471                                 struct nvm_lun *vlun, unsigned long flags)
 472 {
 473         struct gen_lun *lun = container_of(vlun, struct gen_lun, vlun);
 474         struct nvm_block *blk = NULL;
 475         int is_gc = flags & NVM_IOTYPE_GC;
 476
 477         spin_lock(&vlun->lock);
 478         if (list_empty(&lun->free_list)) {
 479                 pr_err_ratelimited("gen: lun %u have no free pages available",
 480                                                                 lun->vlun.id);
 481                 goto out;
 482         }
 483
 484         if (!is_gc && lun->vlun.nr_free_blocks < lun->reserved_blocks)
 485                 goto out;
 486
 487         blk = list_first_entry(&lun->free_list, struct nvm_block, list);
 488
 489         list_move_tail(&blk->list, &lun->used_list);
 490         blk->state = NVM_BLK_ST_TGT;
 491         lun->vlun.nr_free_blocks--;
 492 out:
 493         spin_unlock(&vlun->lock);
 494         return blk;
 495 }
 496
 497 static void gen_put_blk(struct nvm_dev *dev, struct nvm_block *blk)
 498 {
 499         struct nvm_lun *vlun = blk->lun;
 500         struct gen_lun *lun = container_of(vlun, struct gen_lun, vlun);
 501
 502         spin_lock(&vlun->lock);
 503         if (blk->state & NVM_BLK_ST_TGT) {
 504                 list_move_tail(&blk->list, &lun->free_list);
 505                 lun->vlun.nr_free_blocks++;
 506                 blk->state = NVM_BLK_ST_FREE;
 507         } else if (blk->state & NVM_BLK_ST_BAD) {
 508                 list_move_tail(&blk->list, &lun->bb_list);
 509                 blk->state = NVM_BLK_ST_BAD;
 510         } else {
 511                 WARN_ON_ONCE(1);
 512                 pr_err("gen: erroneous block type (%lu -> %u)\n",
 513                                                         blk->id, blk->state);
 514                 list_move_tail(&blk->list, &lun->bb_list);
 515         }
 516         spin_unlock(&vlun->lock);
 517 }
 518
 519 static void gen_mark_blk(struct nvm_dev *dev, struct ppa_addr ppa, int type)
 520 {
 521         struct gen_dev *gn = dev->mp;
 522         struct gen_lun *lun;
 523         struct nvm_block *blk;
 524
 525         pr_debug("gen: ppa  (ch: %u lun: %u blk: %u pg: %u) -> %u\n",
 526                         ppa.g.ch, ppa.g.lun, ppa.g.blk, ppa.g.pg, type);
 527
 528         if (unlikely(ppa.g.ch > dev->nr_chnls ||
 529                                         ppa.g.lun > dev->luns_per_chnl ||
 530                                         ppa.g.blk > dev->blks_per_lun)) {
 531                 WARN_ON_ONCE(1);
 532                 pr_err("gen: ppa broken (ch: %u > %u lun: %u > %u blk: %u > %u",
 533                                 ppa.g.ch, dev->nr_chnls,
 534                                 ppa.g.lun, dev->luns_per_chnl,
 535                                 ppa.g.blk, dev->blks_per_lun);
 536                 return;
 537         }
 538
 539         lun = &gn->luns[(dev->luns_per_chnl * ppa.g.ch) + ppa.g.lun];
 540         blk = &lun->vlun.blocks[ppa.g.blk];
 541
 542         /* will be moved to bb list on put_blk from target */
 543         blk->state = type;
 544 }
 545
 546 static void gen_end_io(struct nvm_rq *rqd)
 547 {
 548         struct nvm_tgt_instance *ins = rqd->ins;
 549
 550         ins->tt->end_io(rqd);
 551 }
 552
 553 static int gen_submit_io(struct nvm_dev *dev, struct nvm_rq *rqd)
 554 {
 555         if (!dev->ops->submit_io)
 556                 return -ENODEV;
 557
 558         /* Convert address space */
 559         nvm_generic_to_addr_mode(dev, rqd);
 560
 561         rqd->dev = dev;
 562         rqd->end_io = gen_end_io;
 563         return dev->ops->submit_io(dev, rqd);
 564 }
 565
 566 static int gen_erase_blk(struct nvm_dev *dev, struct nvm_block *blk, int flags)
 567 {
 568         struct ppa_addr addr = block_to_ppa(dev, blk);
 569
 570         return nvm_erase_ppa(dev, &addr, 1, flags);
 571 }
 572
 573 static int gen_reserve_lun(struct nvm_dev *dev, int lunid)
 574 {
 575         return test_and_set_bit(lunid, dev->lun_map);
 576 }
 577
 578 static void gen_release_lun(struct nvm_dev *dev, int lunid)
 579 {
 580         WARN_ON(!test_and_clear_bit(lunid, dev->lun_map));
 581 }
 582
 583 static struct nvm_lun *gen_get_lun(struct nvm_dev *dev, int lunid)
 584 {
 585         struct gen_dev *gn = dev->mp;
 586
 587         if (unlikely(lunid >= dev->nr_luns))
 588                 return NULL;
 589
 590         return &gn->luns[lunid].vlun;
 591 }
 592
 593 static void gen_lun_info_print(struct nvm_dev *dev)
 594 {
 595         struct gen_dev *gn = dev->mp;
 596         struct gen_lun *lun;
 597         unsigned int i;
 598
 599
 600         gen_for_each_lun(gn, lun, i) {
 601                 spin_lock(&lun->vlun.lock);
 602
 603                 pr_info("%s: lun%8u\t%u\n", dev->name, i,
 604                                                 lun->vlun.nr_free_blocks);
 605
 606                 spin_unlock(&lun->vlun.lock);
 607         }
 608 }
 609
 610 static struct nvmm_type gen = {
 611         .name                   = "gennvm",
 612         .version                = {0, 1, 0},
 613
 614         .register_mgr           = gen_register,
 615         .unregister_mgr         = gen_unregister,
 616
 617         .create_tgt             = gen_create_tgt,
 618         .remove_tgt             = gen_remove_tgt,
 619
 620         .get_blk                = gen_get_blk,
 621         .put_blk                = gen_put_blk,
 622
 623         .submit_io              = gen_submit_io,
 624         .erase_blk              = gen_erase_blk,
 625
 626         .mark_blk               = gen_mark_blk,
 627
 628         .get_lun                = gen_get_lun,
 629         .reserve_lun            = gen_reserve_lun,
 630         .release_lun            = gen_release_lun,
 631         .lun_info_print         = gen_lun_info_print,
 632
 633         .get_area               = gen_get_area,
 634         .put_area               = gen_put_area,
 635
 636 };
 637
 638 static int __init gen_module_init(void)
 639 {
 640         return nvm_register_mgr(&gen);
 641 }
 642
 643 static void gen_module_exit(void)
 644 {
 645         nvm_unregister_mgr(&gen);
 646 }
 647
 648 module_init(gen_module_init);
 649 module_exit(gen_module_exit);
 650 MODULE_LICENSE("GPL v2");
 651 MODULE_DESCRIPTION("General media manager for Open-Channel SSDs");