core.c 25 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183
  1. /*
  2. * Copyright (C) 2015 IT University of Copenhagen. All rights reserved.
  3. * Initial release: Matias Bjorling <m@bjorling.me>
  4. *
  5. * This program is free software; you can redistribute it and/or
  6. * modify it under the terms of the GNU General Public License version
  7. * 2 as published by the Free Software Foundation.
  8. *
  9. * This program is distributed in the hope that it will be useful, but
  10. * WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU General Public License
  15. * along with this program; see the file COPYING. If not, write to
  16. * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139,
  17. * USA.
  18. *
  19. */
  20. #include <linux/blkdev.h>
  21. #include <linux/blk-mq.h>
  22. #include <linux/list.h>
  23. #include <linux/types.h>
  24. #include <linux/sem.h>
  25. #include <linux/bitmap.h>
  26. #include <linux/module.h>
  27. #include <linux/miscdevice.h>
  28. #include <linux/lightnvm.h>
  29. #include <linux/sched/sysctl.h>
  30. #include <uapi/linux/lightnvm.h>
  31. static LIST_HEAD(nvm_targets);
  32. static LIST_HEAD(nvm_mgrs);
  33. static LIST_HEAD(nvm_devices);
  34. static DECLARE_RWSEM(nvm_lock);
  35. static struct nvm_tgt_type *nvm_find_target_type(const char *name)
  36. {
  37. struct nvm_tgt_type *tt;
  38. list_for_each_entry(tt, &nvm_targets, list)
  39. if (!strcmp(name, tt->name))
  40. return tt;
  41. return NULL;
  42. }
  43. int nvm_register_target(struct nvm_tgt_type *tt)
  44. {
  45. int ret = 0;
  46. down_write(&nvm_lock);
  47. if (nvm_find_target_type(tt->name))
  48. ret = -EEXIST;
  49. else
  50. list_add(&tt->list, &nvm_targets);
  51. up_write(&nvm_lock);
  52. return ret;
  53. }
  54. EXPORT_SYMBOL(nvm_register_target);
  55. void nvm_unregister_target(struct nvm_tgt_type *tt)
  56. {
  57. if (!tt)
  58. return;
  59. down_write(&nvm_lock);
  60. list_del(&tt->list);
  61. up_write(&nvm_lock);
  62. }
  63. EXPORT_SYMBOL(nvm_unregister_target);
  64. void *nvm_dev_dma_alloc(struct nvm_dev *dev, gfp_t mem_flags,
  65. dma_addr_t *dma_handler)
  66. {
  67. return dev->ops->dev_dma_alloc(dev, dev->ppalist_pool, mem_flags,
  68. dma_handler);
  69. }
  70. EXPORT_SYMBOL(nvm_dev_dma_alloc);
  71. void nvm_dev_dma_free(struct nvm_dev *dev, void *ppa_list,
  72. dma_addr_t dma_handler)
  73. {
  74. dev->ops->dev_dma_free(dev->ppalist_pool, ppa_list, dma_handler);
  75. }
  76. EXPORT_SYMBOL(nvm_dev_dma_free);
  77. static struct nvmm_type *nvm_find_mgr_type(const char *name)
  78. {
  79. struct nvmm_type *mt;
  80. list_for_each_entry(mt, &nvm_mgrs, list)
  81. if (!strcmp(name, mt->name))
  82. return mt;
  83. return NULL;
  84. }
  85. struct nvmm_type *nvm_init_mgr(struct nvm_dev *dev)
  86. {
  87. struct nvmm_type *mt;
  88. int ret;
  89. lockdep_assert_held(&nvm_lock);
  90. list_for_each_entry(mt, &nvm_mgrs, list) {
  91. if (strncmp(dev->sb.mmtype, mt->name, NVM_MMTYPE_LEN))
  92. continue;
  93. ret = mt->register_mgr(dev);
  94. if (ret < 0) {
  95. pr_err("nvm: media mgr failed to init (%d) on dev %s\n",
  96. ret, dev->name);
  97. return NULL; /* initialization failed */
  98. } else if (ret > 0)
  99. return mt;
  100. }
  101. return NULL;
  102. }
  103. int nvm_register_mgr(struct nvmm_type *mt)
  104. {
  105. struct nvm_dev *dev;
  106. int ret = 0;
  107. down_write(&nvm_lock);
  108. if (nvm_find_mgr_type(mt->name)) {
  109. ret = -EEXIST;
  110. goto finish;
  111. } else {
  112. list_add(&mt->list, &nvm_mgrs);
  113. }
  114. /* try to register media mgr if any device have none configured */
  115. list_for_each_entry(dev, &nvm_devices, devices) {
  116. if (dev->mt)
  117. continue;
  118. dev->mt = nvm_init_mgr(dev);
  119. }
  120. finish:
  121. up_write(&nvm_lock);
  122. return ret;
  123. }
  124. EXPORT_SYMBOL(nvm_register_mgr);
  125. void nvm_unregister_mgr(struct nvmm_type *mt)
  126. {
  127. if (!mt)
  128. return;
  129. down_write(&nvm_lock);
  130. list_del(&mt->list);
  131. up_write(&nvm_lock);
  132. }
  133. EXPORT_SYMBOL(nvm_unregister_mgr);
  134. static struct nvm_dev *nvm_find_nvm_dev(const char *name)
  135. {
  136. struct nvm_dev *dev;
  137. list_for_each_entry(dev, &nvm_devices, devices)
  138. if (!strcmp(name, dev->name))
  139. return dev;
  140. return NULL;
  141. }
  142. struct nvm_block *nvm_get_blk_unlocked(struct nvm_dev *dev, struct nvm_lun *lun,
  143. unsigned long flags)
  144. {
  145. return dev->mt->get_blk_unlocked(dev, lun, flags);
  146. }
  147. EXPORT_SYMBOL(nvm_get_blk_unlocked);
  148. /* Assumes that all valid pages have already been moved on release to bm */
  149. void nvm_put_blk_unlocked(struct nvm_dev *dev, struct nvm_block *blk)
  150. {
  151. return dev->mt->put_blk_unlocked(dev, blk);
  152. }
  153. EXPORT_SYMBOL(nvm_put_blk_unlocked);
  154. struct nvm_block *nvm_get_blk(struct nvm_dev *dev, struct nvm_lun *lun,
  155. unsigned long flags)
  156. {
  157. return dev->mt->get_blk(dev, lun, flags);
  158. }
  159. EXPORT_SYMBOL(nvm_get_blk);
  160. /* Assumes that all valid pages have already been moved on release to bm */
  161. void nvm_put_blk(struct nvm_dev *dev, struct nvm_block *blk)
  162. {
  163. return dev->mt->put_blk(dev, blk);
  164. }
  165. EXPORT_SYMBOL(nvm_put_blk);
  166. int nvm_submit_io(struct nvm_dev *dev, struct nvm_rq *rqd)
  167. {
  168. return dev->mt->submit_io(dev, rqd);
  169. }
  170. EXPORT_SYMBOL(nvm_submit_io);
  171. int nvm_erase_blk(struct nvm_dev *dev, struct nvm_block *blk)
  172. {
  173. return dev->mt->erase_blk(dev, blk, 0);
  174. }
  175. EXPORT_SYMBOL(nvm_erase_blk);
  176. void nvm_addr_to_generic_mode(struct nvm_dev *dev, struct nvm_rq *rqd)
  177. {
  178. int i;
  179. if (rqd->nr_pages > 1) {
  180. for (i = 0; i < rqd->nr_pages; i++)
  181. rqd->ppa_list[i] = dev_to_generic_addr(dev,
  182. rqd->ppa_list[i]);
  183. } else {
  184. rqd->ppa_addr = dev_to_generic_addr(dev, rqd->ppa_addr);
  185. }
  186. }
  187. EXPORT_SYMBOL(nvm_addr_to_generic_mode);
  188. void nvm_generic_to_addr_mode(struct nvm_dev *dev, struct nvm_rq *rqd)
  189. {
  190. int i;
  191. if (rqd->nr_pages > 1) {
  192. for (i = 0; i < rqd->nr_pages; i++)
  193. rqd->ppa_list[i] = generic_to_dev_addr(dev,
  194. rqd->ppa_list[i]);
  195. } else {
  196. rqd->ppa_addr = generic_to_dev_addr(dev, rqd->ppa_addr);
  197. }
  198. }
  199. EXPORT_SYMBOL(nvm_generic_to_addr_mode);
  200. int nvm_set_rqd_ppalist(struct nvm_dev *dev, struct nvm_rq *rqd,
  201. struct ppa_addr *ppas, int nr_ppas)
  202. {
  203. int i, plane_cnt, pl_idx;
  204. if (dev->plane_mode == NVM_PLANE_SINGLE && nr_ppas == 1) {
  205. rqd->nr_pages = 1;
  206. rqd->ppa_addr = ppas[0];
  207. return 0;
  208. }
  209. plane_cnt = dev->plane_mode;
  210. rqd->nr_pages = plane_cnt * nr_ppas;
  211. if (dev->ops->max_phys_sect < rqd->nr_pages)
  212. return -EINVAL;
  213. rqd->ppa_list = nvm_dev_dma_alloc(dev, GFP_KERNEL, &rqd->dma_ppa_list);
  214. if (!rqd->ppa_list) {
  215. pr_err("nvm: failed to allocate dma memory\n");
  216. return -ENOMEM;
  217. }
  218. for (pl_idx = 0; pl_idx < plane_cnt; pl_idx++) {
  219. for (i = 0; i < nr_ppas; i++) {
  220. ppas[i].g.pl = pl_idx;
  221. rqd->ppa_list[(pl_idx * nr_ppas) + i] = ppas[i];
  222. }
  223. }
  224. return 0;
  225. }
  226. EXPORT_SYMBOL(nvm_set_rqd_ppalist);
  227. void nvm_free_rqd_ppalist(struct nvm_dev *dev, struct nvm_rq *rqd)
  228. {
  229. if (!rqd->ppa_list)
  230. return;
  231. nvm_dev_dma_free(dev, rqd->ppa_list, rqd->dma_ppa_list);
  232. }
  233. EXPORT_SYMBOL(nvm_free_rqd_ppalist);
  234. int nvm_erase_ppa(struct nvm_dev *dev, struct ppa_addr *ppas, int nr_ppas)
  235. {
  236. struct nvm_rq rqd;
  237. int ret;
  238. if (!dev->ops->erase_block)
  239. return 0;
  240. memset(&rqd, 0, sizeof(struct nvm_rq));
  241. ret = nvm_set_rqd_ppalist(dev, &rqd, ppas, nr_ppas);
  242. if (ret)
  243. return ret;
  244. nvm_generic_to_addr_mode(dev, &rqd);
  245. ret = dev->ops->erase_block(dev, &rqd);
  246. nvm_free_rqd_ppalist(dev, &rqd);
  247. return ret;
  248. }
  249. EXPORT_SYMBOL(nvm_erase_ppa);
  250. void nvm_end_io(struct nvm_rq *rqd, int error)
  251. {
  252. rqd->error = error;
  253. rqd->end_io(rqd);
  254. }
  255. EXPORT_SYMBOL(nvm_end_io);
  256. static void nvm_end_io_sync(struct nvm_rq *rqd)
  257. {
  258. struct completion *waiting = rqd->wait;
  259. rqd->wait = NULL;
  260. complete(waiting);
  261. }
  262. int nvm_submit_ppa(struct nvm_dev *dev, struct ppa_addr *ppa, int nr_ppas,
  263. int opcode, int flags, void *buf, int len)
  264. {
  265. DECLARE_COMPLETION_ONSTACK(wait);
  266. struct nvm_rq rqd;
  267. struct bio *bio;
  268. int ret;
  269. unsigned long hang_check;
  270. bio = bio_map_kern(dev->q, buf, len, GFP_KERNEL);
  271. if (IS_ERR_OR_NULL(bio))
  272. return -ENOMEM;
  273. memset(&rqd, 0, sizeof(struct nvm_rq));
  274. ret = nvm_set_rqd_ppalist(dev, &rqd, ppa, nr_ppas);
  275. if (ret) {
  276. bio_put(bio);
  277. return ret;
  278. }
  279. rqd.opcode = opcode;
  280. rqd.bio = bio;
  281. rqd.wait = &wait;
  282. rqd.dev = dev;
  283. rqd.end_io = nvm_end_io_sync;
  284. rqd.flags = flags;
  285. nvm_generic_to_addr_mode(dev, &rqd);
  286. ret = dev->ops->submit_io(dev, &rqd);
  287. /* Prevent hang_check timer from firing at us during very long I/O */
  288. hang_check = sysctl_hung_task_timeout_secs;
  289. if (hang_check)
  290. while (!wait_for_completion_io_timeout(&wait, hang_check * (HZ/2)));
  291. else
  292. wait_for_completion_io(&wait);
  293. nvm_free_rqd_ppalist(dev, &rqd);
  294. return rqd.error;
  295. }
  296. EXPORT_SYMBOL(nvm_submit_ppa);
  297. static int nvm_init_slc_tbl(struct nvm_dev *dev, struct nvm_id_group *grp)
  298. {
  299. int i;
  300. dev->lps_per_blk = dev->pgs_per_blk;
  301. dev->lptbl = kcalloc(dev->lps_per_blk, sizeof(int), GFP_KERNEL);
  302. if (!dev->lptbl)
  303. return -ENOMEM;
  304. /* Just a linear array */
  305. for (i = 0; i < dev->lps_per_blk; i++)
  306. dev->lptbl[i] = i;
  307. return 0;
  308. }
  309. static int nvm_init_mlc_tbl(struct nvm_dev *dev, struct nvm_id_group *grp)
  310. {
  311. int i, p;
  312. struct nvm_id_lp_mlc *mlc = &grp->lptbl.mlc;
  313. if (!mlc->num_pairs)
  314. return 0;
  315. dev->lps_per_blk = mlc->num_pairs;
  316. dev->lptbl = kcalloc(dev->lps_per_blk, sizeof(int), GFP_KERNEL);
  317. if (!dev->lptbl)
  318. return -ENOMEM;
  319. /* The lower page table encoding consists of a list of bytes, where each
  320. * has a lower and an upper half. The first half byte maintains the
  321. * increment value and every value after is an offset added to the
  322. * previous incrementation value */
  323. dev->lptbl[0] = mlc->pairs[0] & 0xF;
  324. for (i = 1; i < dev->lps_per_blk; i++) {
  325. p = mlc->pairs[i >> 1];
  326. if (i & 0x1) /* upper */
  327. dev->lptbl[i] = dev->lptbl[i - 1] + ((p & 0xF0) >> 4);
  328. else /* lower */
  329. dev->lptbl[i] = dev->lptbl[i - 1] + (p & 0xF);
  330. }
  331. return 0;
  332. }
  333. static int nvm_core_init(struct nvm_dev *dev)
  334. {
  335. struct nvm_id *id = &dev->identity;
  336. struct nvm_id_group *grp = &id->groups[0];
  337. /* device values */
  338. dev->nr_chnls = grp->num_ch;
  339. dev->luns_per_chnl = grp->num_lun;
  340. dev->pgs_per_blk = grp->num_pg;
  341. dev->blks_per_lun = grp->num_blk;
  342. dev->nr_planes = grp->num_pln;
  343. dev->sec_size = grp->csecs;
  344. dev->oob_size = grp->sos;
  345. dev->sec_per_pg = grp->fpg_sz / grp->csecs;
  346. dev->mccap = grp->mccap;
  347. memcpy(&dev->ppaf, &id->ppaf, sizeof(struct nvm_addr_format));
  348. dev->plane_mode = NVM_PLANE_SINGLE;
  349. dev->max_rq_size = dev->ops->max_phys_sect * dev->sec_size;
  350. if (grp->mtype != 0) {
  351. pr_err("nvm: memory type not supported\n");
  352. return -EINVAL;
  353. }
  354. switch (grp->fmtype) {
  355. case NVM_ID_FMTYPE_SLC:
  356. if (nvm_init_slc_tbl(dev, grp))
  357. return -ENOMEM;
  358. break;
  359. case NVM_ID_FMTYPE_MLC:
  360. if (nvm_init_mlc_tbl(dev, grp))
  361. return -ENOMEM;
  362. break;
  363. default:
  364. pr_err("nvm: flash type not supported\n");
  365. return -EINVAL;
  366. }
  367. if (!dev->lps_per_blk)
  368. pr_info("nvm: lower page programming table missing\n");
  369. if (grp->mpos & 0x020202)
  370. dev->plane_mode = NVM_PLANE_DOUBLE;
  371. if (grp->mpos & 0x040404)
  372. dev->plane_mode = NVM_PLANE_QUAD;
  373. /* calculated values */
  374. dev->sec_per_pl = dev->sec_per_pg * dev->nr_planes;
  375. dev->sec_per_blk = dev->sec_per_pl * dev->pgs_per_blk;
  376. dev->sec_per_lun = dev->sec_per_blk * dev->blks_per_lun;
  377. dev->nr_luns = dev->luns_per_chnl * dev->nr_chnls;
  378. dev->total_secs = dev->nr_luns * dev->sec_per_lun;
  379. INIT_LIST_HEAD(&dev->online_targets);
  380. mutex_init(&dev->mlock);
  381. spin_lock_init(&dev->lock);
  382. return 0;
  383. }
  384. static void nvm_free(struct nvm_dev *dev)
  385. {
  386. if (!dev)
  387. return;
  388. if (dev->mt)
  389. dev->mt->unregister_mgr(dev);
  390. kfree(dev->lptbl);
  391. }
  392. static int nvm_init(struct nvm_dev *dev)
  393. {
  394. int ret = -EINVAL;
  395. if (!dev->q || !dev->ops)
  396. return ret;
  397. if (dev->ops->identity(dev, &dev->identity)) {
  398. pr_err("nvm: device could not be identified\n");
  399. goto err;
  400. }
  401. pr_debug("nvm: ver:%x nvm_vendor:%x groups:%u\n",
  402. dev->identity.ver_id, dev->identity.vmnt,
  403. dev->identity.cgrps);
  404. if (dev->identity.ver_id != 1) {
  405. pr_err("nvm: device not supported by kernel.");
  406. goto err;
  407. }
  408. if (dev->identity.cgrps != 1) {
  409. pr_err("nvm: only one group configuration supported.");
  410. goto err;
  411. }
  412. ret = nvm_core_init(dev);
  413. if (ret) {
  414. pr_err("nvm: could not initialize core structures.\n");
  415. goto err;
  416. }
  417. pr_info("nvm: registered %s [%u/%u/%u/%u/%u/%u]\n",
  418. dev->name, dev->sec_per_pg, dev->nr_planes,
  419. dev->pgs_per_blk, dev->blks_per_lun, dev->nr_luns,
  420. dev->nr_chnls);
  421. return 0;
  422. err:
  423. pr_err("nvm: failed to initialize nvm\n");
  424. return ret;
  425. }
  426. static void nvm_exit(struct nvm_dev *dev)
  427. {
  428. if (dev->ppalist_pool)
  429. dev->ops->destroy_dma_pool(dev->ppalist_pool);
  430. nvm_free(dev);
  431. pr_info("nvm: successfully unloaded\n");
  432. }
  433. int nvm_register(struct request_queue *q, char *disk_name,
  434. struct nvm_dev_ops *ops)
  435. {
  436. struct nvm_dev *dev;
  437. int ret;
  438. if (!ops->identity)
  439. return -EINVAL;
  440. dev = kzalloc(sizeof(struct nvm_dev), GFP_KERNEL);
  441. if (!dev)
  442. return -ENOMEM;
  443. dev->q = q;
  444. dev->ops = ops;
  445. strncpy(dev->name, disk_name, DISK_NAME_LEN);
  446. ret = nvm_init(dev);
  447. if (ret)
  448. goto err_init;
  449. if (dev->ops->max_phys_sect > 256) {
  450. pr_info("nvm: max sectors supported is 256.\n");
  451. ret = -EINVAL;
  452. goto err_init;
  453. }
  454. if (dev->ops->max_phys_sect > 1) {
  455. dev->ppalist_pool = dev->ops->create_dma_pool(dev, "ppalist");
  456. if (!dev->ppalist_pool) {
  457. pr_err("nvm: could not create ppa pool\n");
  458. ret = -ENOMEM;
  459. goto err_init;
  460. }
  461. }
  462. if (dev->identity.cap & NVM_ID_DCAP_BBLKMGMT) {
  463. ret = nvm_get_sysblock(dev, &dev->sb);
  464. if (!ret)
  465. pr_err("nvm: device not initialized.\n");
  466. else if (ret < 0)
  467. pr_err("nvm: err (%d) on device initialization\n", ret);
  468. }
  469. /* register device with a supported media manager */
  470. down_write(&nvm_lock);
  471. if (ret > 0)
  472. dev->mt = nvm_init_mgr(dev);
  473. list_add(&dev->devices, &nvm_devices);
  474. up_write(&nvm_lock);
  475. return 0;
  476. err_init:
  477. kfree(dev);
  478. return ret;
  479. }
  480. EXPORT_SYMBOL(nvm_register);
  481. void nvm_unregister(char *disk_name)
  482. {
  483. struct nvm_dev *dev;
  484. down_write(&nvm_lock);
  485. dev = nvm_find_nvm_dev(disk_name);
  486. if (!dev) {
  487. pr_err("nvm: could not find device %s to unregister\n",
  488. disk_name);
  489. up_write(&nvm_lock);
  490. return;
  491. }
  492. list_del(&dev->devices);
  493. up_write(&nvm_lock);
  494. nvm_exit(dev);
  495. kfree(dev);
  496. }
  497. EXPORT_SYMBOL(nvm_unregister);
  498. static const struct block_device_operations nvm_fops = {
  499. .owner = THIS_MODULE,
  500. };
  501. static int nvm_create_target(struct nvm_dev *dev,
  502. struct nvm_ioctl_create *create)
  503. {
  504. struct nvm_ioctl_create_simple *s = &create->conf.s;
  505. struct request_queue *tqueue;
  506. struct gendisk *tdisk;
  507. struct nvm_tgt_type *tt;
  508. struct nvm_target *t;
  509. void *targetdata;
  510. if (!dev->mt) {
  511. pr_info("nvm: device has no media manager registered.\n");
  512. return -ENODEV;
  513. }
  514. down_write(&nvm_lock);
  515. tt = nvm_find_target_type(create->tgttype);
  516. if (!tt) {
  517. pr_err("nvm: target type %s not found\n", create->tgttype);
  518. up_write(&nvm_lock);
  519. return -EINVAL;
  520. }
  521. list_for_each_entry(t, &dev->online_targets, list) {
  522. if (!strcmp(create->tgtname, t->disk->disk_name)) {
  523. pr_err("nvm: target name already exists.\n");
  524. up_write(&nvm_lock);
  525. return -EINVAL;
  526. }
  527. }
  528. up_write(&nvm_lock);
  529. t = kmalloc(sizeof(struct nvm_target), GFP_KERNEL);
  530. if (!t)
  531. return -ENOMEM;
  532. tqueue = blk_alloc_queue_node(GFP_KERNEL, dev->q->node);
  533. if (!tqueue)
  534. goto err_t;
  535. blk_queue_make_request(tqueue, tt->make_rq);
  536. tdisk = alloc_disk(0);
  537. if (!tdisk)
  538. goto err_queue;
  539. sprintf(tdisk->disk_name, "%s", create->tgtname);
  540. tdisk->flags = GENHD_FL_EXT_DEVT;
  541. tdisk->major = 0;
  542. tdisk->first_minor = 0;
  543. tdisk->fops = &nvm_fops;
  544. tdisk->queue = tqueue;
  545. targetdata = tt->init(dev, tdisk, s->lun_begin, s->lun_end);
  546. if (IS_ERR(targetdata))
  547. goto err_init;
  548. tdisk->private_data = targetdata;
  549. tqueue->queuedata = targetdata;
  550. blk_queue_max_hw_sectors(tqueue, 8 * dev->ops->max_phys_sect);
  551. set_capacity(tdisk, tt->capacity(targetdata));
  552. add_disk(tdisk);
  553. t->type = tt;
  554. t->disk = tdisk;
  555. down_write(&nvm_lock);
  556. list_add_tail(&t->list, &dev->online_targets);
  557. up_write(&nvm_lock);
  558. return 0;
  559. err_init:
  560. put_disk(tdisk);
  561. err_queue:
  562. blk_cleanup_queue(tqueue);
  563. err_t:
  564. kfree(t);
  565. return -ENOMEM;
  566. }
  567. static void nvm_remove_target(struct nvm_target *t)
  568. {
  569. struct nvm_tgt_type *tt = t->type;
  570. struct gendisk *tdisk = t->disk;
  571. struct request_queue *q = tdisk->queue;
  572. lockdep_assert_held(&nvm_lock);
  573. del_gendisk(tdisk);
  574. blk_cleanup_queue(q);
  575. if (tt->exit)
  576. tt->exit(tdisk->private_data);
  577. put_disk(tdisk);
  578. list_del(&t->list);
  579. kfree(t);
  580. }
  581. static int __nvm_configure_create(struct nvm_ioctl_create *create)
  582. {
  583. struct nvm_dev *dev;
  584. struct nvm_ioctl_create_simple *s;
  585. down_write(&nvm_lock);
  586. dev = nvm_find_nvm_dev(create->dev);
  587. up_write(&nvm_lock);
  588. if (!dev) {
  589. pr_err("nvm: device not found\n");
  590. return -EINVAL;
  591. }
  592. if (create->conf.type != NVM_CONFIG_TYPE_SIMPLE) {
  593. pr_err("nvm: config type not valid\n");
  594. return -EINVAL;
  595. }
  596. s = &create->conf.s;
  597. if (s->lun_begin > s->lun_end || s->lun_end > dev->nr_luns) {
  598. pr_err("nvm: lun out of bound (%u:%u > %u)\n",
  599. s->lun_begin, s->lun_end, dev->nr_luns);
  600. return -EINVAL;
  601. }
  602. return nvm_create_target(dev, create);
  603. }
  604. static int __nvm_configure_remove(struct nvm_ioctl_remove *remove)
  605. {
  606. struct nvm_target *t = NULL;
  607. struct nvm_dev *dev;
  608. int ret = -1;
  609. down_write(&nvm_lock);
  610. list_for_each_entry(dev, &nvm_devices, devices)
  611. list_for_each_entry(t, &dev->online_targets, list) {
  612. if (!strcmp(remove->tgtname, t->disk->disk_name)) {
  613. nvm_remove_target(t);
  614. ret = 0;
  615. break;
  616. }
  617. }
  618. up_write(&nvm_lock);
  619. if (ret) {
  620. pr_err("nvm: target \"%s\" doesn't exist.\n", remove->tgtname);
  621. return -EINVAL;
  622. }
  623. return 0;
  624. }
  625. #ifdef CONFIG_NVM_DEBUG
  626. static int nvm_configure_show(const char *val)
  627. {
  628. struct nvm_dev *dev;
  629. char opcode, devname[DISK_NAME_LEN];
  630. int ret;
  631. ret = sscanf(val, "%c %32s", &opcode, devname);
  632. if (ret != 2) {
  633. pr_err("nvm: invalid command. Use \"opcode devicename\".\n");
  634. return -EINVAL;
  635. }
  636. down_write(&nvm_lock);
  637. dev = nvm_find_nvm_dev(devname);
  638. up_write(&nvm_lock);
  639. if (!dev) {
  640. pr_err("nvm: device not found\n");
  641. return -EINVAL;
  642. }
  643. if (!dev->mt)
  644. return 0;
  645. dev->mt->lun_info_print(dev);
  646. return 0;
  647. }
  648. static int nvm_configure_remove(const char *val)
  649. {
  650. struct nvm_ioctl_remove remove;
  651. char opcode;
  652. int ret;
  653. ret = sscanf(val, "%c %256s", &opcode, remove.tgtname);
  654. if (ret != 2) {
  655. pr_err("nvm: invalid command. Use \"d targetname\".\n");
  656. return -EINVAL;
  657. }
  658. remove.flags = 0;
  659. return __nvm_configure_remove(&remove);
  660. }
  661. static int nvm_configure_create(const char *val)
  662. {
  663. struct nvm_ioctl_create create;
  664. char opcode;
  665. int lun_begin, lun_end, ret;
  666. ret = sscanf(val, "%c %256s %256s %48s %u:%u", &opcode, create.dev,
  667. create.tgtname, create.tgttype,
  668. &lun_begin, &lun_end);
  669. if (ret != 6) {
  670. pr_err("nvm: invalid command. Use \"opcode device name tgttype lun_begin:lun_end\".\n");
  671. return -EINVAL;
  672. }
  673. create.flags = 0;
  674. create.conf.type = NVM_CONFIG_TYPE_SIMPLE;
  675. create.conf.s.lun_begin = lun_begin;
  676. create.conf.s.lun_end = lun_end;
  677. return __nvm_configure_create(&create);
  678. }
  679. /* Exposes administrative interface through /sys/module/lnvm/configure_by_str */
  680. static int nvm_configure_by_str_event(const char *val,
  681. const struct kernel_param *kp)
  682. {
  683. char opcode;
  684. int ret;
  685. ret = sscanf(val, "%c", &opcode);
  686. if (ret != 1) {
  687. pr_err("nvm: string must have the format of \"cmd ...\"\n");
  688. return -EINVAL;
  689. }
  690. switch (opcode) {
  691. case 'a':
  692. return nvm_configure_create(val);
  693. case 'd':
  694. return nvm_configure_remove(val);
  695. case 's':
  696. return nvm_configure_show(val);
  697. default:
  698. pr_err("nvm: invalid command\n");
  699. return -EINVAL;
  700. }
  701. return 0;
  702. }
  703. static int nvm_configure_get(char *buf, const struct kernel_param *kp)
  704. {
  705. int sz;
  706. struct nvm_dev *dev;
  707. sz = sprintf(buf, "available devices:\n");
  708. down_write(&nvm_lock);
  709. list_for_each_entry(dev, &nvm_devices, devices) {
  710. if (sz > 4095 - DISK_NAME_LEN - 2)
  711. break;
  712. sz += sprintf(buf + sz, " %32s\n", dev->name);
  713. }
  714. up_write(&nvm_lock);
  715. return sz;
  716. }
  717. static const struct kernel_param_ops nvm_configure_by_str_event_param_ops = {
  718. .set = nvm_configure_by_str_event,
  719. .get = nvm_configure_get,
  720. };
  721. #undef MODULE_PARAM_PREFIX
  722. #define MODULE_PARAM_PREFIX "lnvm."
  723. module_param_cb(configure_debug, &nvm_configure_by_str_event_param_ops, NULL,
  724. 0644);
  725. #endif /* CONFIG_NVM_DEBUG */
  726. static long nvm_ioctl_info(struct file *file, void __user *arg)
  727. {
  728. struct nvm_ioctl_info *info;
  729. struct nvm_tgt_type *tt;
  730. int tgt_iter = 0;
  731. if (!capable(CAP_SYS_ADMIN))
  732. return -EPERM;
  733. info = memdup_user(arg, sizeof(struct nvm_ioctl_info));
  734. if (IS_ERR(info))
  735. return -EFAULT;
  736. info->version[0] = NVM_VERSION_MAJOR;
  737. info->version[1] = NVM_VERSION_MINOR;
  738. info->version[2] = NVM_VERSION_PATCH;
  739. down_write(&nvm_lock);
  740. list_for_each_entry(tt, &nvm_targets, list) {
  741. struct nvm_ioctl_info_tgt *tgt = &info->tgts[tgt_iter];
  742. tgt->version[0] = tt->version[0];
  743. tgt->version[1] = tt->version[1];
  744. tgt->version[2] = tt->version[2];
  745. strncpy(tgt->tgtname, tt->name, NVM_TTYPE_NAME_MAX);
  746. tgt_iter++;
  747. }
  748. info->tgtsize = tgt_iter;
  749. up_write(&nvm_lock);
  750. if (copy_to_user(arg, info, sizeof(struct nvm_ioctl_info))) {
  751. kfree(info);
  752. return -EFAULT;
  753. }
  754. kfree(info);
  755. return 0;
  756. }
  757. static long nvm_ioctl_get_devices(struct file *file, void __user *arg)
  758. {
  759. struct nvm_ioctl_get_devices *devices;
  760. struct nvm_dev *dev;
  761. int i = 0;
  762. if (!capable(CAP_SYS_ADMIN))
  763. return -EPERM;
  764. devices = kzalloc(sizeof(struct nvm_ioctl_get_devices), GFP_KERNEL);
  765. if (!devices)
  766. return -ENOMEM;
  767. down_write(&nvm_lock);
  768. list_for_each_entry(dev, &nvm_devices, devices) {
  769. struct nvm_ioctl_device_info *info = &devices->info[i];
  770. sprintf(info->devname, "%s", dev->name);
  771. if (dev->mt) {
  772. info->bmversion[0] = dev->mt->version[0];
  773. info->bmversion[1] = dev->mt->version[1];
  774. info->bmversion[2] = dev->mt->version[2];
  775. sprintf(info->bmname, "%s", dev->mt->name);
  776. } else {
  777. sprintf(info->bmname, "none");
  778. }
  779. i++;
  780. if (i > 31) {
  781. pr_err("nvm: max 31 devices can be reported.\n");
  782. break;
  783. }
  784. }
  785. up_write(&nvm_lock);
  786. devices->nr_devices = i;
  787. if (copy_to_user(arg, devices,
  788. sizeof(struct nvm_ioctl_get_devices))) {
  789. kfree(devices);
  790. return -EFAULT;
  791. }
  792. kfree(devices);
  793. return 0;
  794. }
  795. static long nvm_ioctl_dev_create(struct file *file, void __user *arg)
  796. {
  797. struct nvm_ioctl_create create;
  798. if (!capable(CAP_SYS_ADMIN))
  799. return -EPERM;
  800. if (copy_from_user(&create, arg, sizeof(struct nvm_ioctl_create)))
  801. return -EFAULT;
  802. create.dev[DISK_NAME_LEN - 1] = '\0';
  803. create.tgttype[NVM_TTYPE_NAME_MAX - 1] = '\0';
  804. create.tgtname[DISK_NAME_LEN - 1] = '\0';
  805. if (create.flags != 0) {
  806. pr_err("nvm: no flags supported\n");
  807. return -EINVAL;
  808. }
  809. return __nvm_configure_create(&create);
  810. }
  811. static long nvm_ioctl_dev_remove(struct file *file, void __user *arg)
  812. {
  813. struct nvm_ioctl_remove remove;
  814. if (!capable(CAP_SYS_ADMIN))
  815. return -EPERM;
  816. if (copy_from_user(&remove, arg, sizeof(struct nvm_ioctl_remove)))
  817. return -EFAULT;
  818. remove.tgtname[DISK_NAME_LEN - 1] = '\0';
  819. if (remove.flags != 0) {
  820. pr_err("nvm: no flags supported\n");
  821. return -EINVAL;
  822. }
  823. return __nvm_configure_remove(&remove);
  824. }
  825. static void nvm_setup_nvm_sb_info(struct nvm_sb_info *info)
  826. {
  827. info->seqnr = 1;
  828. info->erase_cnt = 0;
  829. info->version = 1;
  830. }
  831. static long __nvm_ioctl_dev_init(struct nvm_ioctl_dev_init *init)
  832. {
  833. struct nvm_dev *dev;
  834. struct nvm_sb_info info;
  835. int ret;
  836. down_write(&nvm_lock);
  837. dev = nvm_find_nvm_dev(init->dev);
  838. up_write(&nvm_lock);
  839. if (!dev) {
  840. pr_err("nvm: device not found\n");
  841. return -EINVAL;
  842. }
  843. nvm_setup_nvm_sb_info(&info);
  844. strncpy(info.mmtype, init->mmtype, NVM_MMTYPE_LEN);
  845. info.fs_ppa.ppa = -1;
  846. if (dev->identity.cap & NVM_ID_DCAP_BBLKMGMT) {
  847. ret = nvm_init_sysblock(dev, &info);
  848. if (ret)
  849. return ret;
  850. }
  851. memcpy(&dev->sb, &info, sizeof(struct nvm_sb_info));
  852. down_write(&nvm_lock);
  853. dev->mt = nvm_init_mgr(dev);
  854. up_write(&nvm_lock);
  855. return 0;
  856. }
  857. static long nvm_ioctl_dev_init(struct file *file, void __user *arg)
  858. {
  859. struct nvm_ioctl_dev_init init;
  860. if (!capable(CAP_SYS_ADMIN))
  861. return -EPERM;
  862. if (copy_from_user(&init, arg, sizeof(struct nvm_ioctl_dev_init)))
  863. return -EFAULT;
  864. if (init.flags != 0) {
  865. pr_err("nvm: no flags supported\n");
  866. return -EINVAL;
  867. }
  868. init.dev[DISK_NAME_LEN - 1] = '\0';
  869. return __nvm_ioctl_dev_init(&init);
  870. }
  871. static long nvm_ioctl_dev_factory(struct file *file, void __user *arg)
  872. {
  873. struct nvm_ioctl_dev_factory fact;
  874. struct nvm_dev *dev;
  875. if (!capable(CAP_SYS_ADMIN))
  876. return -EPERM;
  877. if (copy_from_user(&fact, arg, sizeof(struct nvm_ioctl_dev_factory)))
  878. return -EFAULT;
  879. fact.dev[DISK_NAME_LEN - 1] = '\0';
  880. if (fact.flags & ~(NVM_FACTORY_NR_BITS - 1))
  881. return -EINVAL;
  882. down_write(&nvm_lock);
  883. dev = nvm_find_nvm_dev(fact.dev);
  884. up_write(&nvm_lock);
  885. if (!dev) {
  886. pr_err("nvm: device not found\n");
  887. return -EINVAL;
  888. }
  889. if (dev->mt) {
  890. dev->mt->unregister_mgr(dev);
  891. dev->mt = NULL;
  892. }
  893. if (dev->identity.cap & NVM_ID_DCAP_BBLKMGMT)
  894. return nvm_dev_factory(dev, fact.flags);
  895. return 0;
  896. }
  897. static long nvm_ctl_ioctl(struct file *file, uint cmd, unsigned long arg)
  898. {
  899. void __user *argp = (void __user *)arg;
  900. switch (cmd) {
  901. case NVM_INFO:
  902. return nvm_ioctl_info(file, argp);
  903. case NVM_GET_DEVICES:
  904. return nvm_ioctl_get_devices(file, argp);
  905. case NVM_DEV_CREATE:
  906. return nvm_ioctl_dev_create(file, argp);
  907. case NVM_DEV_REMOVE:
  908. return nvm_ioctl_dev_remove(file, argp);
  909. case NVM_DEV_INIT:
  910. return nvm_ioctl_dev_init(file, argp);
  911. case NVM_DEV_FACTORY:
  912. return nvm_ioctl_dev_factory(file, argp);
  913. }
  914. return 0;
  915. }
  916. static const struct file_operations _ctl_fops = {
  917. .open = nonseekable_open,
  918. .unlocked_ioctl = nvm_ctl_ioctl,
  919. .owner = THIS_MODULE,
  920. .llseek = noop_llseek,
  921. };
  922. static struct miscdevice _nvm_misc = {
  923. .minor = MISC_DYNAMIC_MINOR,
  924. .name = "lightnvm",
  925. .nodename = "lightnvm/control",
  926. .fops = &_ctl_fops,
  927. };
  928. MODULE_ALIAS_MISCDEV(MISC_DYNAMIC_MINOR);
  929. static int __init nvm_mod_init(void)
  930. {
  931. int ret;
  932. ret = misc_register(&_nvm_misc);
  933. if (ret)
  934. pr_err("nvm: misc_register failed for control device");
  935. return ret;
  936. }
  937. static void __exit nvm_mod_exit(void)
  938. {
  939. misc_deregister(&_nvm_misc);
  940. }
  941. MODULE_AUTHOR("Matias Bjorling <m@bjorling.me>");
  942. MODULE_LICENSE("GPL v2");
  943. MODULE_VERSION("0.1");
  944. module_init(nvm_mod_init);
  945. module_exit(nvm_mod_exit);