2 * Copyright (C) 2015 IT University of Copenhagen. All rights reserved.
3 * Initial release: Matias Bjorling <m@bjorling.me>
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.
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.
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,
21 #include <linux/blkdev.h>
22 #include <linux/blk-mq.h>
23 #include <linux/list.h>
24 #include <linux/types.h>
25 #include <linux/sem.h>
26 #include <linux/bitmap.h>
27 #include <linux/module.h>
28 #include <linux/miscdevice.h>
29 #include <linux/lightnvm.h>
30 #include <uapi/linux/lightnvm.h>
32 static LIST_HEAD(nvm_targets
);
33 static LIST_HEAD(nvm_mgrs
);
34 static LIST_HEAD(nvm_devices
);
35 static DECLARE_RWSEM(nvm_lock
);
37 static struct nvm_tgt_type
*nvm_find_target_type(const char *name
)
39 struct nvm_tgt_type
*tt
;
41 list_for_each_entry(tt
, &nvm_targets
, list
)
42 if (!strcmp(name
, tt
->name
))
48 int nvm_register_target(struct nvm_tgt_type
*tt
)
52 down_write(&nvm_lock
);
53 if (nvm_find_target_type(tt
->name
))
56 list_add(&tt
->list
, &nvm_targets
);
61 EXPORT_SYMBOL(nvm_register_target
);
63 void nvm_unregister_target(struct nvm_tgt_type
*tt
)
68 down_write(&nvm_lock
);
72 EXPORT_SYMBOL(nvm_unregister_target
);
74 void *nvm_dev_dma_alloc(struct nvm_dev
*dev
, gfp_t mem_flags
,
75 dma_addr_t
*dma_handler
)
77 return dev
->ops
->dev_dma_alloc(dev
, dev
->ppalist_pool
, mem_flags
,
80 EXPORT_SYMBOL(nvm_dev_dma_alloc
);
82 void nvm_dev_dma_free(struct nvm_dev
*dev
, void *ppa_list
,
83 dma_addr_t dma_handler
)
85 dev
->ops
->dev_dma_free(dev
->ppalist_pool
, ppa_list
, dma_handler
);
87 EXPORT_SYMBOL(nvm_dev_dma_free
);
89 static struct nvmm_type
*nvm_find_mgr_type(const char *name
)
93 list_for_each_entry(mt
, &nvm_mgrs
, list
)
94 if (!strcmp(name
, mt
->name
))
100 struct nvmm_type
*nvm_init_mgr(struct nvm_dev
*dev
)
102 struct nvmm_type
*mt
;
105 lockdep_assert_held(&nvm_lock
);
107 list_for_each_entry(mt
, &nvm_mgrs
, list
) {
108 ret
= mt
->register_mgr(dev
);
110 pr_err("nvm: media mgr failed to init (%d) on dev %s\n",
112 return NULL
; /* initialization failed */
120 int nvm_register_mgr(struct nvmm_type
*mt
)
125 down_write(&nvm_lock
);
126 if (nvm_find_mgr_type(mt
->name
)) {
130 list_add(&mt
->list
, &nvm_mgrs
);
133 /* try to register media mgr if any device have none configured */
134 list_for_each_entry(dev
, &nvm_devices
, devices
) {
138 dev
->mt
= nvm_init_mgr(dev
);
145 EXPORT_SYMBOL(nvm_register_mgr
);
147 void nvm_unregister_mgr(struct nvmm_type
*mt
)
152 down_write(&nvm_lock
);
156 EXPORT_SYMBOL(nvm_unregister_mgr
);
158 static struct nvm_dev
*nvm_find_nvm_dev(const char *name
)
162 list_for_each_entry(dev
, &nvm_devices
, devices
)
163 if (!strcmp(name
, dev
->name
))
169 struct nvm_block
*nvm_get_blk(struct nvm_dev
*dev
, struct nvm_lun
*lun
,
172 return dev
->mt
->get_blk(dev
, lun
, flags
);
174 EXPORT_SYMBOL(nvm_get_blk
);
176 /* Assumes that all valid pages have already been moved on release to bm */
177 void nvm_put_blk(struct nvm_dev
*dev
, struct nvm_block
*blk
)
179 return dev
->mt
->put_blk(dev
, blk
);
181 EXPORT_SYMBOL(nvm_put_blk
);
183 int nvm_submit_io(struct nvm_dev
*dev
, struct nvm_rq
*rqd
)
185 return dev
->mt
->submit_io(dev
, rqd
);
187 EXPORT_SYMBOL(nvm_submit_io
);
189 int nvm_erase_blk(struct nvm_dev
*dev
, struct nvm_block
*blk
)
191 return dev
->mt
->erase_blk(dev
, blk
, 0);
193 EXPORT_SYMBOL(nvm_erase_blk
);
195 void nvm_addr_to_generic_mode(struct nvm_dev
*dev
, struct nvm_rq
*rqd
)
199 if (rqd
->nr_pages
> 1) {
200 for (i
= 0; i
< rqd
->nr_pages
; i
++)
201 rqd
->ppa_list
[i
] = dev_to_generic_addr(dev
,
204 rqd
->ppa_addr
= dev_to_generic_addr(dev
, rqd
->ppa_addr
);
207 EXPORT_SYMBOL(nvm_addr_to_generic_mode
);
209 void nvm_generic_to_addr_mode(struct nvm_dev
*dev
, struct nvm_rq
*rqd
)
213 if (rqd
->nr_pages
> 1) {
214 for (i
= 0; i
< rqd
->nr_pages
; i
++)
215 rqd
->ppa_list
[i
] = generic_to_dev_addr(dev
,
218 rqd
->ppa_addr
= generic_to_dev_addr(dev
, rqd
->ppa_addr
);
221 EXPORT_SYMBOL(nvm_generic_to_addr_mode
);
223 int nvm_erase_ppa(struct nvm_dev
*dev
, struct ppa_addr ppa
)
225 int plane_cnt
= 0, pl_idx
, ret
;
228 if (!dev
->ops
->erase_block
)
231 if (dev
->plane_mode
== NVM_PLANE_SINGLE
) {
235 plane_cnt
= (1 << dev
->plane_mode
);
236 rqd
.nr_pages
= plane_cnt
;
238 rqd
.ppa_list
= nvm_dev_dma_alloc(dev
, GFP_KERNEL
,
241 pr_err("nvm: failed to allocate dma memory\n");
245 for (pl_idx
= 0; pl_idx
< plane_cnt
; pl_idx
++) {
247 rqd
.ppa_list
[pl_idx
] = ppa
;
251 nvm_generic_to_addr_mode(dev
, &rqd
);
253 ret
= dev
->ops
->erase_block(dev
, &rqd
);
256 nvm_dev_dma_free(dev
, rqd
.ppa_list
, rqd
.dma_ppa_list
);
260 EXPORT_SYMBOL(nvm_erase_ppa
);
262 static int nvm_core_init(struct nvm_dev
*dev
)
264 struct nvm_id
*id
= &dev
->identity
;
265 struct nvm_id_group
*grp
= &id
->groups
[0];
268 dev
->nr_chnls
= grp
->num_ch
;
269 dev
->luns_per_chnl
= grp
->num_lun
;
270 dev
->pgs_per_blk
= grp
->num_pg
;
271 dev
->blks_per_lun
= grp
->num_blk
;
272 dev
->nr_planes
= grp
->num_pln
;
273 dev
->sec_size
= grp
->csecs
;
274 dev
->oob_size
= grp
->sos
;
275 dev
->sec_per_pg
= grp
->fpg_sz
/ grp
->csecs
;
276 memcpy(&dev
->ppaf
, &id
->ppaf
, sizeof(struct nvm_addr_format
));
278 dev
->plane_mode
= NVM_PLANE_SINGLE
;
279 dev
->max_rq_size
= dev
->ops
->max_phys_sect
* dev
->sec_size
;
281 if (grp
->mtype
!= 0) {
282 pr_err("nvm: memory type not supported\n");
286 if (grp
->fmtype
!= 0 && grp
->fmtype
!= 1) {
287 pr_err("nvm: flash type not supported\n");
291 if (grp
->mpos
& 0x020202)
292 dev
->plane_mode
= NVM_PLANE_DOUBLE
;
293 if (grp
->mpos
& 0x040404)
294 dev
->plane_mode
= NVM_PLANE_QUAD
;
296 /* calculated values */
297 dev
->sec_per_pl
= dev
->sec_per_pg
* dev
->nr_planes
;
298 dev
->sec_per_blk
= dev
->sec_per_pl
* dev
->pgs_per_blk
;
299 dev
->sec_per_lun
= dev
->sec_per_blk
* dev
->blks_per_lun
;
300 dev
->nr_luns
= dev
->luns_per_chnl
* dev
->nr_chnls
;
302 dev
->total_blocks
= dev
->nr_planes
*
306 dev
->total_pages
= dev
->total_blocks
* dev
->pgs_per_blk
;
307 INIT_LIST_HEAD(&dev
->online_targets
);
312 static void nvm_free(struct nvm_dev
*dev
)
318 dev
->mt
->unregister_mgr(dev
);
321 static int nvm_init(struct nvm_dev
*dev
)
325 if (!dev
->q
|| !dev
->ops
)
328 if (dev
->ops
->identity(dev
, &dev
->identity
)) {
329 pr_err("nvm: device could not be identified\n");
333 pr_debug("nvm: ver:%x nvm_vendor:%x groups:%u\n",
334 dev
->identity
.ver_id
, dev
->identity
.vmnt
,
335 dev
->identity
.cgrps
);
337 if (dev
->identity
.ver_id
!= 1) {
338 pr_err("nvm: device not supported by kernel.");
342 if (dev
->identity
.cgrps
!= 1) {
343 pr_err("nvm: only one group configuration supported.");
347 ret
= nvm_core_init(dev
);
349 pr_err("nvm: could not initialize core structures.\n");
353 pr_info("nvm: registered %s [%u/%u/%u/%u/%u/%u]\n",
354 dev
->name
, dev
->sec_per_pg
, dev
->nr_planes
,
355 dev
->pgs_per_blk
, dev
->blks_per_lun
, dev
->nr_luns
,
359 pr_err("nvm: failed to initialize nvm\n");
363 static void nvm_exit(struct nvm_dev
*dev
)
365 if (dev
->ppalist_pool
)
366 dev
->ops
->destroy_dma_pool(dev
->ppalist_pool
);
369 pr_info("nvm: successfully unloaded\n");
372 int nvm_register(struct request_queue
*q
, char *disk_name
,
373 struct nvm_dev_ops
*ops
)
381 dev
= kzalloc(sizeof(struct nvm_dev
), GFP_KERNEL
);
387 strncpy(dev
->name
, disk_name
, DISK_NAME_LEN
);
393 if (dev
->ops
->max_phys_sect
> 256) {
394 pr_info("nvm: max sectors supported is 256.\n");
399 if (dev
->ops
->max_phys_sect
> 1) {
400 dev
->ppalist_pool
= dev
->ops
->create_dma_pool(dev
, "ppalist");
401 if (!dev
->ppalist_pool
) {
402 pr_err("nvm: could not create ppa pool\n");
408 /* register device with a supported media manager */
409 down_write(&nvm_lock
);
410 dev
->mt
= nvm_init_mgr(dev
);
411 list_add(&dev
->devices
, &nvm_devices
);
419 EXPORT_SYMBOL(nvm_register
);
421 void nvm_unregister(char *disk_name
)
425 down_write(&nvm_lock
);
426 dev
= nvm_find_nvm_dev(disk_name
);
428 pr_err("nvm: could not find device %s to unregister\n",
434 list_del(&dev
->devices
);
440 EXPORT_SYMBOL(nvm_unregister
);
442 static const struct block_device_operations nvm_fops
= {
443 .owner
= THIS_MODULE
,
446 static int nvm_create_target(struct nvm_dev
*dev
,
447 struct nvm_ioctl_create
*create
)
449 struct nvm_ioctl_create_simple
*s
= &create
->conf
.s
;
450 struct request_queue
*tqueue
;
451 struct gendisk
*tdisk
;
452 struct nvm_tgt_type
*tt
;
453 struct nvm_target
*t
;
457 pr_info("nvm: device has no media manager registered.\n");
461 down_write(&nvm_lock
);
462 tt
= nvm_find_target_type(create
->tgttype
);
464 pr_err("nvm: target type %s not found\n", create
->tgttype
);
469 list_for_each_entry(t
, &dev
->online_targets
, list
) {
470 if (!strcmp(create
->tgtname
, t
->disk
->disk_name
)) {
471 pr_err("nvm: target name already exists.\n");
478 t
= kmalloc(sizeof(struct nvm_target
), GFP_KERNEL
);
482 tqueue
= blk_alloc_queue_node(GFP_KERNEL
, dev
->q
->node
);
485 blk_queue_make_request(tqueue
, tt
->make_rq
);
487 tdisk
= alloc_disk(0);
491 sprintf(tdisk
->disk_name
, "%s", create
->tgtname
);
492 tdisk
->flags
= GENHD_FL_EXT_DEVT
;
494 tdisk
->first_minor
= 0;
495 tdisk
->fops
= &nvm_fops
;
496 tdisk
->queue
= tqueue
;
498 targetdata
= tt
->init(dev
, tdisk
, s
->lun_begin
, s
->lun_end
);
499 if (IS_ERR(targetdata
))
502 tdisk
->private_data
= targetdata
;
503 tqueue
->queuedata
= targetdata
;
505 blk_queue_max_hw_sectors(tqueue
, 8 * dev
->ops
->max_phys_sect
);
507 set_capacity(tdisk
, tt
->capacity(targetdata
));
513 down_write(&nvm_lock
);
514 list_add_tail(&t
->list
, &dev
->online_targets
);
521 blk_cleanup_queue(tqueue
);
527 static void nvm_remove_target(struct nvm_target
*t
)
529 struct nvm_tgt_type
*tt
= t
->type
;
530 struct gendisk
*tdisk
= t
->disk
;
531 struct request_queue
*q
= tdisk
->queue
;
533 lockdep_assert_held(&nvm_lock
);
536 blk_cleanup_queue(q
);
539 tt
->exit(tdisk
->private_data
);
547 static int __nvm_configure_create(struct nvm_ioctl_create
*create
)
550 struct nvm_ioctl_create_simple
*s
;
552 down_write(&nvm_lock
);
553 dev
= nvm_find_nvm_dev(create
->dev
);
556 pr_err("nvm: device not found\n");
560 if (create
->conf
.type
!= NVM_CONFIG_TYPE_SIMPLE
) {
561 pr_err("nvm: config type not valid\n");
566 if (s
->lun_begin
> s
->lun_end
|| s
->lun_end
> dev
->nr_luns
) {
567 pr_err("nvm: lun out of bound (%u:%u > %u)\n",
568 s
->lun_begin
, s
->lun_end
, dev
->nr_luns
);
572 return nvm_create_target(dev
, create
);
575 static int __nvm_configure_remove(struct nvm_ioctl_remove
*remove
)
577 struct nvm_target
*t
= NULL
;
581 down_write(&nvm_lock
);
582 list_for_each_entry(dev
, &nvm_devices
, devices
)
583 list_for_each_entry(t
, &dev
->online_targets
, list
) {
584 if (!strcmp(remove
->tgtname
, t
->disk
->disk_name
)) {
585 nvm_remove_target(t
);
593 pr_err("nvm: target \"%s\" doesn't exist.\n", remove
->tgtname
);
600 #ifdef CONFIG_NVM_DEBUG
601 static int nvm_configure_show(const char *val
)
604 char opcode
, devname
[DISK_NAME_LEN
];
607 ret
= sscanf(val
, "%c %32s", &opcode
, devname
);
609 pr_err("nvm: invalid command. Use \"opcode devicename\".\n");
613 down_write(&nvm_lock
);
614 dev
= nvm_find_nvm_dev(devname
);
617 pr_err("nvm: device not found\n");
624 dev
->mt
->lun_info_print(dev
);
629 static int nvm_configure_remove(const char *val
)
631 struct nvm_ioctl_remove remove
;
635 ret
= sscanf(val
, "%c %256s", &opcode
, remove
.tgtname
);
637 pr_err("nvm: invalid command. Use \"d targetname\".\n");
643 return __nvm_configure_remove(&remove
);
646 static int nvm_configure_create(const char *val
)
648 struct nvm_ioctl_create create
;
650 int lun_begin
, lun_end
, ret
;
652 ret
= sscanf(val
, "%c %256s %256s %48s %u:%u", &opcode
, create
.dev
,
653 create
.tgtname
, create
.tgttype
,
654 &lun_begin
, &lun_end
);
656 pr_err("nvm: invalid command. Use \"opcode device name tgttype lun_begin:lun_end\".\n");
661 create
.conf
.type
= NVM_CONFIG_TYPE_SIMPLE
;
662 create
.conf
.s
.lun_begin
= lun_begin
;
663 create
.conf
.s
.lun_end
= lun_end
;
665 return __nvm_configure_create(&create
);
669 /* Exposes administrative interface through /sys/module/lnvm/configure_by_str */
670 static int nvm_configure_by_str_event(const char *val
,
671 const struct kernel_param
*kp
)
676 ret
= sscanf(val
, "%c", &opcode
);
678 pr_err("nvm: string must have the format of \"cmd ...\"\n");
684 return nvm_configure_create(val
);
686 return nvm_configure_remove(val
);
688 return nvm_configure_show(val
);
690 pr_err("nvm: invalid command\n");
697 static int nvm_configure_get(char *buf
, const struct kernel_param
*kp
)
700 char *buf_start
= buf
;
703 buf
+= sprintf(buf
, "available devices:\n");
704 down_write(&nvm_lock
);
705 list_for_each_entry(dev
, &nvm_devices
, devices
) {
706 if (sz
> 4095 - DISK_NAME_LEN
)
708 buf
+= sprintf(buf
, " %32s\n", dev
->name
);
712 return buf
- buf_start
- 1;
715 static const struct kernel_param_ops nvm_configure_by_str_event_param_ops
= {
716 .set
= nvm_configure_by_str_event
,
717 .get
= nvm_configure_get
,
720 #undef MODULE_PARAM_PREFIX
721 #define MODULE_PARAM_PREFIX "lnvm."
723 module_param_cb(configure_debug
, &nvm_configure_by_str_event_param_ops
, NULL
,
726 #endif /* CONFIG_NVM_DEBUG */
728 static long nvm_ioctl_info(struct file
*file
, void __user
*arg
)
730 struct nvm_ioctl_info
*info
;
731 struct nvm_tgt_type
*tt
;
734 if (!capable(CAP_SYS_ADMIN
))
737 info
= memdup_user(arg
, sizeof(struct nvm_ioctl_info
));
741 info
->version
[0] = NVM_VERSION_MAJOR
;
742 info
->version
[1] = NVM_VERSION_MINOR
;
743 info
->version
[2] = NVM_VERSION_PATCH
;
745 down_write(&nvm_lock
);
746 list_for_each_entry(tt
, &nvm_targets
, list
) {
747 struct nvm_ioctl_info_tgt
*tgt
= &info
->tgts
[tgt_iter
];
749 tgt
->version
[0] = tt
->version
[0];
750 tgt
->version
[1] = tt
->version
[1];
751 tgt
->version
[2] = tt
->version
[2];
752 strncpy(tgt
->tgtname
, tt
->name
, NVM_TTYPE_NAME_MAX
);
757 info
->tgtsize
= tgt_iter
;
760 if (copy_to_user(arg
, info
, sizeof(struct nvm_ioctl_info
))) {
769 static long nvm_ioctl_get_devices(struct file
*file
, void __user
*arg
)
771 struct nvm_ioctl_get_devices
*devices
;
775 if (!capable(CAP_SYS_ADMIN
))
778 devices
= kzalloc(sizeof(struct nvm_ioctl_get_devices
), GFP_KERNEL
);
782 down_write(&nvm_lock
);
783 list_for_each_entry(dev
, &nvm_devices
, devices
) {
784 struct nvm_ioctl_device_info
*info
= &devices
->info
[i
];
786 sprintf(info
->devname
, "%s", dev
->name
);
788 info
->bmversion
[0] = dev
->mt
->version
[0];
789 info
->bmversion
[1] = dev
->mt
->version
[1];
790 info
->bmversion
[2] = dev
->mt
->version
[2];
791 sprintf(info
->bmname
, "%s", dev
->mt
->name
);
793 sprintf(info
->bmname
, "none");
798 pr_err("nvm: max 31 devices can be reported.\n");
804 devices
->nr_devices
= i
;
806 if (copy_to_user(arg
, devices
,
807 sizeof(struct nvm_ioctl_get_devices
))) {
816 static long nvm_ioctl_dev_create(struct file
*file
, void __user
*arg
)
818 struct nvm_ioctl_create create
;
820 if (!capable(CAP_SYS_ADMIN
))
823 if (copy_from_user(&create
, arg
, sizeof(struct nvm_ioctl_create
)))
826 create
.dev
[DISK_NAME_LEN
- 1] = '\0';
827 create
.tgttype
[NVM_TTYPE_NAME_MAX
- 1] = '\0';
828 create
.tgtname
[DISK_NAME_LEN
- 1] = '\0';
830 if (create
.flags
!= 0) {
831 pr_err("nvm: no flags supported\n");
835 return __nvm_configure_create(&create
);
838 static long nvm_ioctl_dev_remove(struct file
*file
, void __user
*arg
)
840 struct nvm_ioctl_remove remove
;
842 if (!capable(CAP_SYS_ADMIN
))
845 if (copy_from_user(&remove
, arg
, sizeof(struct nvm_ioctl_remove
)))
848 remove
.tgtname
[DISK_NAME_LEN
- 1] = '\0';
850 if (remove
.flags
!= 0) {
851 pr_err("nvm: no flags supported\n");
855 return __nvm_configure_remove(&remove
);
858 static long nvm_ctl_ioctl(struct file
*file
, uint cmd
, unsigned long arg
)
860 void __user
*argp
= (void __user
*)arg
;
864 return nvm_ioctl_info(file
, argp
);
865 case NVM_GET_DEVICES
:
866 return nvm_ioctl_get_devices(file
, argp
);
868 return nvm_ioctl_dev_create(file
, argp
);
870 return nvm_ioctl_dev_remove(file
, argp
);
875 static const struct file_operations _ctl_fops
= {
876 .open
= nonseekable_open
,
877 .unlocked_ioctl
= nvm_ctl_ioctl
,
878 .owner
= THIS_MODULE
,
879 .llseek
= noop_llseek
,
882 static struct miscdevice _nvm_misc
= {
883 .minor
= MISC_DYNAMIC_MINOR
,
885 .nodename
= "lightnvm/control",
889 MODULE_ALIAS_MISCDEV(MISC_DYNAMIC_MINOR
);
891 static int __init
nvm_mod_init(void)
895 ret
= misc_register(&_nvm_misc
);
897 pr_err("nvm: misc_register failed for control device");
902 static void __exit
nvm_mod_exit(void)
904 misc_deregister(&_nvm_misc
);
907 MODULE_AUTHOR("Matias Bjorling <m@bjorling.me>");
908 MODULE_LICENSE("GPL v2");
909 MODULE_VERSION("0.1");
910 module_init(nvm_mod_init
);
911 module_exit(nvm_mod_exit
);