null_blk: use ppa_cache pool
[deliverable/linux.git] / drivers / block / null_blk.c
CommitLineData
f2298c04 1#include <linux/module.h>
fc1bc354 2
f2298c04
JA
3#include <linux/moduleparam.h>
4#include <linux/sched.h>
5#include <linux/fs.h>
6#include <linux/blkdev.h>
7#include <linux/init.h>
8#include <linux/slab.h>
9#include <linux/blk-mq.h>
10#include <linux/hrtimer.h>
b2b7e001 11#include <linux/lightnvm.h>
f2298c04
JA
12
13struct nullb_cmd {
14 struct list_head list;
15 struct llist_node ll_list;
16 struct call_single_data csd;
17 struct request *rq;
18 struct bio *bio;
19 unsigned int tag;
20 struct nullb_queue *nq;
21};
22
23struct nullb_queue {
24 unsigned long *tag_map;
25 wait_queue_head_t wait;
26 unsigned int queue_depth;
27
28 struct nullb_cmd *cmds;
29};
30
31struct nullb {
32 struct list_head list;
33 unsigned int index;
34 struct request_queue *q;
35 struct gendisk *disk;
24d2f903 36 struct blk_mq_tag_set tag_set;
f2298c04
JA
37 struct hrtimer timer;
38 unsigned int queue_depth;
39 spinlock_t lock;
40
41 struct nullb_queue *queues;
42 unsigned int nr_queues;
b2b7e001 43 char disk_name[DISK_NAME_LEN];
f2298c04
JA
44};
45
46static LIST_HEAD(nullb_list);
47static struct mutex lock;
48static int null_major;
49static int nullb_indexes;
6bb9535b 50static struct kmem_cache *ppa_cache;
f2298c04
JA
51
52struct completion_queue {
53 struct llist_head list;
54 struct hrtimer timer;
55};
56
57/*
58 * These are per-cpu for now, they will need to be configured by the
59 * complete_queues parameter and appropriately mapped.
60 */
61static DEFINE_PER_CPU(struct completion_queue, completion_queues);
62
63enum {
64 NULL_IRQ_NONE = 0,
65 NULL_IRQ_SOFTIRQ = 1,
66 NULL_IRQ_TIMER = 2,
ce2c350b 67};
f2298c04 68
ce2c350b 69enum {
f2298c04
JA
70 NULL_Q_BIO = 0,
71 NULL_Q_RQ = 1,
72 NULL_Q_MQ = 2,
73};
74
2d263a78 75static int submit_queues;
f2298c04
JA
76module_param(submit_queues, int, S_IRUGO);
77MODULE_PARM_DESC(submit_queues, "Number of submission queues");
78
79static int home_node = NUMA_NO_NODE;
80module_param(home_node, int, S_IRUGO);
81MODULE_PARM_DESC(home_node, "Home node for the device");
82
83static int queue_mode = NULL_Q_MQ;
709c8667
MB
84
85static int null_param_store_val(const char *str, int *val, int min, int max)
86{
87 int ret, new_val;
88
89 ret = kstrtoint(str, 10, &new_val);
90 if (ret)
91 return -EINVAL;
92
93 if (new_val < min || new_val > max)
94 return -EINVAL;
95
96 *val = new_val;
97 return 0;
98}
99
100static int null_set_queue_mode(const char *str, const struct kernel_param *kp)
101{
102 return null_param_store_val(str, &queue_mode, NULL_Q_BIO, NULL_Q_MQ);
103}
104
9c27847d 105static const struct kernel_param_ops null_queue_mode_param_ops = {
709c8667
MB
106 .set = null_set_queue_mode,
107 .get = param_get_int,
108};
109
110device_param_cb(queue_mode, &null_queue_mode_param_ops, &queue_mode, S_IRUGO);
54ae81cd 111MODULE_PARM_DESC(queue_mode, "Block interface to use (0=bio,1=rq,2=multiqueue)");
f2298c04
JA
112
113static int gb = 250;
114module_param(gb, int, S_IRUGO);
115MODULE_PARM_DESC(gb, "Size in GB");
116
117static int bs = 512;
118module_param(bs, int, S_IRUGO);
119MODULE_PARM_DESC(bs, "Block size (in bytes)");
120
121static int nr_devices = 2;
122module_param(nr_devices, int, S_IRUGO);
123MODULE_PARM_DESC(nr_devices, "Number of devices to register");
124
b2b7e001
MB
125static bool use_lightnvm;
126module_param(use_lightnvm, bool, S_IRUGO);
127MODULE_PARM_DESC(use_lightnvm, "Register as a LightNVM device");
128
f2298c04 129static int irqmode = NULL_IRQ_SOFTIRQ;
709c8667
MB
130
131static int null_set_irqmode(const char *str, const struct kernel_param *kp)
132{
133 return null_param_store_val(str, &irqmode, NULL_IRQ_NONE,
134 NULL_IRQ_TIMER);
135}
136
9c27847d 137static const struct kernel_param_ops null_irqmode_param_ops = {
709c8667
MB
138 .set = null_set_irqmode,
139 .get = param_get_int,
140};
141
142device_param_cb(irqmode, &null_irqmode_param_ops, &irqmode, S_IRUGO);
f2298c04
JA
143MODULE_PARM_DESC(irqmode, "IRQ completion handler. 0-none, 1-softirq, 2-timer");
144
145static int completion_nsec = 10000;
146module_param(completion_nsec, int, S_IRUGO);
147MODULE_PARM_DESC(completion_nsec, "Time in ns to complete a request in hardware. Default: 10,000ns");
148
149static int hw_queue_depth = 64;
150module_param(hw_queue_depth, int, S_IRUGO);
151MODULE_PARM_DESC(hw_queue_depth, "Queue depth for each hardware queue. Default: 64");
152
20005244 153static bool use_per_node_hctx = false;
f2298c04 154module_param(use_per_node_hctx, bool, S_IRUGO);
20005244 155MODULE_PARM_DESC(use_per_node_hctx, "Use per-node allocation for hardware context queues. Default: false");
f2298c04
JA
156
157static void put_tag(struct nullb_queue *nq, unsigned int tag)
158{
159 clear_bit_unlock(tag, nq->tag_map);
160
161 if (waitqueue_active(&nq->wait))
162 wake_up(&nq->wait);
163}
164
165static unsigned int get_tag(struct nullb_queue *nq)
166{
167 unsigned int tag;
168
169 do {
170 tag = find_first_zero_bit(nq->tag_map, nq->queue_depth);
171 if (tag >= nq->queue_depth)
172 return -1U;
173 } while (test_and_set_bit_lock(tag, nq->tag_map));
174
175 return tag;
176}
177
178static void free_cmd(struct nullb_cmd *cmd)
179{
180 put_tag(cmd->nq, cmd->tag);
181}
182
183static struct nullb_cmd *__alloc_cmd(struct nullb_queue *nq)
184{
185 struct nullb_cmd *cmd;
186 unsigned int tag;
187
188 tag = get_tag(nq);
189 if (tag != -1U) {
190 cmd = &nq->cmds[tag];
191 cmd->tag = tag;
192 cmd->nq = nq;
193 return cmd;
194 }
195
196 return NULL;
197}
198
199static struct nullb_cmd *alloc_cmd(struct nullb_queue *nq, int can_wait)
200{
201 struct nullb_cmd *cmd;
202 DEFINE_WAIT(wait);
203
204 cmd = __alloc_cmd(nq);
205 if (cmd || !can_wait)
206 return cmd;
207
208 do {
209 prepare_to_wait(&nq->wait, &wait, TASK_UNINTERRUPTIBLE);
210 cmd = __alloc_cmd(nq);
211 if (cmd)
212 break;
213
214 io_schedule();
215 } while (1);
216
217 finish_wait(&nq->wait, &wait);
218 return cmd;
219}
220
221static void end_cmd(struct nullb_cmd *cmd)
222{
ce2c350b
CH
223 switch (queue_mode) {
224 case NULL_Q_MQ:
c8a446ad 225 blk_mq_end_request(cmd->rq, 0);
ce2c350b
CH
226 return;
227 case NULL_Q_RQ:
228 INIT_LIST_HEAD(&cmd->rq->queuelist);
229 blk_end_request_all(cmd->rq, 0);
230 break;
231 case NULL_Q_BIO:
4246a0b6 232 bio_endio(cmd->bio);
ce2c350b
CH
233 break;
234 }
f2298c04 235
ce2c350b 236 free_cmd(cmd);
f2298c04
JA
237}
238
239static enum hrtimer_restart null_cmd_timer_expired(struct hrtimer *timer)
240{
241 struct completion_queue *cq;
242 struct llist_node *entry;
243 struct nullb_cmd *cmd;
244
245 cq = &per_cpu(completion_queues, smp_processor_id());
246
247 while ((entry = llist_del_all(&cq->list)) != NULL) {
d7790b92 248 entry = llist_reverse_order(entry);
f2298c04 249 do {
21974061
MK
250 struct request_queue *q = NULL;
251
f2298c04 252 cmd = container_of(entry, struct nullb_cmd, ll_list);
f2298c04 253 entry = entry->next;
21974061
MK
254 if (cmd->rq)
255 q = cmd->rq->q;
fc27691f 256 end_cmd(cmd);
8b70f45e 257
21974061
MK
258 if (q && !q->mq_ops && blk_queue_stopped(q)) {
259 spin_lock(q->queue_lock);
260 if (blk_queue_stopped(q))
261 blk_start_queue(q);
262 spin_unlock(q->queue_lock);
8b70f45e 263 }
f2298c04
JA
264 } while (entry);
265 }
266
267 return HRTIMER_NORESTART;
268}
269
270static void null_cmd_end_timer(struct nullb_cmd *cmd)
271{
272 struct completion_queue *cq = &per_cpu(completion_queues, get_cpu());
273
274 cmd->ll_list.next = NULL;
275 if (llist_add(&cmd->ll_list, &cq->list)) {
276 ktime_t kt = ktime_set(0, completion_nsec);
277
419c21a3 278 hrtimer_start(&cq->timer, kt, HRTIMER_MODE_REL_PINNED);
f2298c04
JA
279 }
280
281 put_cpu();
282}
283
284static void null_softirq_done_fn(struct request *rq)
285{
d891fa70
JA
286 if (queue_mode == NULL_Q_MQ)
287 end_cmd(blk_mq_rq_to_pdu(rq));
288 else
289 end_cmd(rq->special);
f2298c04
JA
290}
291
f2298c04
JA
292static inline void null_handle_cmd(struct nullb_cmd *cmd)
293{
294 /* Complete IO by inline, softirq or timer */
295 switch (irqmode) {
f2298c04 296 case NULL_IRQ_SOFTIRQ:
ce2c350b
CH
297 switch (queue_mode) {
298 case NULL_Q_MQ:
f4829a9b 299 blk_mq_complete_request(cmd->rq, cmd->rq->errors);
ce2c350b
CH
300 break;
301 case NULL_Q_RQ:
302 blk_complete_request(cmd->rq);
303 break;
304 case NULL_Q_BIO:
305 /*
306 * XXX: no proper submitting cpu information available.
307 */
308 end_cmd(cmd);
309 break;
310 }
311 break;
312 case NULL_IRQ_NONE:
f2298c04 313 end_cmd(cmd);
f2298c04
JA
314 break;
315 case NULL_IRQ_TIMER:
316 null_cmd_end_timer(cmd);
317 break;
318 }
319}
320
321static struct nullb_queue *nullb_to_queue(struct nullb *nullb)
322{
323 int index = 0;
324
325 if (nullb->nr_queues != 1)
326 index = raw_smp_processor_id() / ((nr_cpu_ids + nullb->nr_queues - 1) / nullb->nr_queues);
327
328 return &nullb->queues[index];
329}
330
dece1635 331static blk_qc_t null_queue_bio(struct request_queue *q, struct bio *bio)
f2298c04
JA
332{
333 struct nullb *nullb = q->queuedata;
334 struct nullb_queue *nq = nullb_to_queue(nullb);
335 struct nullb_cmd *cmd;
336
337 cmd = alloc_cmd(nq, 1);
338 cmd->bio = bio;
339
340 null_handle_cmd(cmd);
dece1635 341 return BLK_QC_T_NONE;
f2298c04
JA
342}
343
344static int null_rq_prep_fn(struct request_queue *q, struct request *req)
345{
346 struct nullb *nullb = q->queuedata;
347 struct nullb_queue *nq = nullb_to_queue(nullb);
348 struct nullb_cmd *cmd;
349
350 cmd = alloc_cmd(nq, 0);
351 if (cmd) {
352 cmd->rq = req;
353 req->special = cmd;
354 return BLKPREP_OK;
355 }
8b70f45e 356 blk_stop_queue(q);
f2298c04
JA
357
358 return BLKPREP_DEFER;
359}
360
361static void null_request_fn(struct request_queue *q)
362{
363 struct request *rq;
364
365 while ((rq = blk_fetch_request(q)) != NULL) {
366 struct nullb_cmd *cmd = rq->special;
367
368 spin_unlock_irq(q->queue_lock);
369 null_handle_cmd(cmd);
370 spin_lock_irq(q->queue_lock);
371 }
372}
373
74c45052
JA
374static int null_queue_rq(struct blk_mq_hw_ctx *hctx,
375 const struct blk_mq_queue_data *bd)
f2298c04 376{
74c45052 377 struct nullb_cmd *cmd = blk_mq_rq_to_pdu(bd->rq);
f2298c04 378
74c45052 379 cmd->rq = bd->rq;
f2298c04
JA
380 cmd->nq = hctx->driver_data;
381
74c45052 382 blk_mq_start_request(bd->rq);
e2490073 383
f2298c04
JA
384 null_handle_cmd(cmd);
385 return BLK_MQ_RQ_QUEUE_OK;
386}
387
2d263a78
MB
388static void null_init_queue(struct nullb *nullb, struct nullb_queue *nq)
389{
390 BUG_ON(!nullb);
391 BUG_ON(!nq);
392
393 init_waitqueue_head(&nq->wait);
394 nq->queue_depth = nullb->queue_depth;
395}
396
f2298c04
JA
397static int null_init_hctx(struct blk_mq_hw_ctx *hctx, void *data,
398 unsigned int index)
399{
400 struct nullb *nullb = data;
401 struct nullb_queue *nq = &nullb->queues[index];
402
f2298c04 403 hctx->driver_data = nq;
2d263a78
MB
404 null_init_queue(nullb, nq);
405 nullb->nr_queues++;
f2298c04
JA
406
407 return 0;
408}
409
410static struct blk_mq_ops null_mq_ops = {
411 .queue_rq = null_queue_rq,
412 .map_queue = blk_mq_map_queue,
413 .init_hctx = null_init_hctx,
ce2c350b 414 .complete = null_softirq_done_fn,
f2298c04
JA
415};
416
de65d2d2
MB
417static void cleanup_queue(struct nullb_queue *nq)
418{
419 kfree(nq->tag_map);
420 kfree(nq->cmds);
421}
422
423static void cleanup_queues(struct nullb *nullb)
424{
425 int i;
426
427 for (i = 0; i < nullb->nr_queues; i++)
428 cleanup_queue(&nullb->queues[i]);
429
430 kfree(nullb->queues);
431}
432
f2298c04
JA
433static void null_del_dev(struct nullb *nullb)
434{
435 list_del_init(&nullb->list);
436
b2b7e001
MB
437 if (use_lightnvm)
438 nvm_unregister(nullb->disk->disk_name);
f2298c04 439 del_gendisk(nullb->disk);
518d00b7 440 blk_cleanup_queue(nullb->q);
24d2f903
CH
441 if (queue_mode == NULL_Q_MQ)
442 blk_mq_free_tag_set(&nullb->tag_set);
f2298c04 443 put_disk(nullb->disk);
de65d2d2 444 cleanup_queues(nullb);
f2298c04
JA
445 kfree(nullb);
446}
447
b2b7e001
MB
448#ifdef CONFIG_NVM
449
450static void null_lnvm_end_io(struct request *rq, int error)
451{
452 struct nvm_rq *rqd = rq->end_io_data;
453 struct nvm_dev *dev = rqd->dev;
454
455 dev->mt->end_io(rqd, error);
456
457 blk_put_request(rq);
458}
459
460static int null_lnvm_submit_io(struct request_queue *q, struct nvm_rq *rqd)
461{
462 struct request *rq;
463 struct bio *bio = rqd->bio;
464
465 rq = blk_mq_alloc_request(q, bio_rw(bio), GFP_KERNEL, 0);
466 if (IS_ERR(rq))
467 return -ENOMEM;
468
469 rq->cmd_type = REQ_TYPE_DRV_PRIV;
470 rq->__sector = bio->bi_iter.bi_sector;
471 rq->ioprio = bio_prio(bio);
472
473 if (bio_has_data(bio))
474 rq->nr_phys_segments = bio_phys_segments(q, bio);
475
476 rq->__data_len = bio->bi_iter.bi_size;
477 rq->bio = rq->biotail = bio;
478
479 rq->end_io_data = rqd;
480
481 blk_execute_rq_nowait(q, NULL, rq, 0, null_lnvm_end_io);
482
483 return 0;
484}
485
486static int null_lnvm_id(struct request_queue *q, struct nvm_id *id)
487{
488 sector_t size = gb * 1024 * 1024 * 1024ULL;
489 struct nvm_id_group *grp;
490
491 id->ver_id = 0x1;
492 id->vmnt = 0;
493 id->cgrps = 1;
494 id->cap = 0x3;
495 id->dom = 0x1;
496 id->ppat = NVM_ADDRMODE_LINEAR;
497
498 do_div(size, bs); /* convert size to pages */
499 grp = &id->groups[0];
500 grp->mtype = 0;
501 grp->fmtype = 1;
502 grp->num_ch = 1;
503 grp->num_lun = 1;
504 grp->num_pln = 1;
505 grp->num_blk = size / 256;
506 grp->num_pg = 256;
507 grp->fpg_sz = bs;
508 grp->csecs = bs;
509 grp->trdt = 25000;
510 grp->trdm = 25000;
511 grp->tprt = 500000;
512 grp->tprm = 500000;
513 grp->tbet = 1500000;
514 grp->tbem = 1500000;
515 grp->mpos = 0x010101; /* single plane rwe */
516 grp->cpar = hw_queue_depth;
517
518 return 0;
519}
520
521static void *null_lnvm_create_dma_pool(struct request_queue *q, char *name)
522{
523 mempool_t *virtmem_pool;
524
6bb9535b 525 virtmem_pool = mempool_create_slab_pool(64, ppa_cache);
b2b7e001
MB
526 if (!virtmem_pool) {
527 pr_err("null_blk: Unable to create virtual memory pool\n");
528 return NULL;
529 }
530
531 return virtmem_pool;
532}
533
534static void null_lnvm_destroy_dma_pool(void *pool)
535{
536 mempool_destroy(pool);
537}
538
539static void *null_lnvm_dev_dma_alloc(struct request_queue *q, void *pool,
540 gfp_t mem_flags, dma_addr_t *dma_handler)
541{
542 return mempool_alloc(pool, mem_flags);
543}
544
545static void null_lnvm_dev_dma_free(void *pool, void *entry,
546 dma_addr_t dma_handler)
547{
548 mempool_free(entry, pool);
549}
550
551static struct nvm_dev_ops null_lnvm_dev_ops = {
552 .identity = null_lnvm_id,
553 .submit_io = null_lnvm_submit_io,
554
555 .create_dma_pool = null_lnvm_create_dma_pool,
556 .destroy_dma_pool = null_lnvm_destroy_dma_pool,
557 .dev_dma_alloc = null_lnvm_dev_dma_alloc,
558 .dev_dma_free = null_lnvm_dev_dma_free,
559
560 /* Simulate nvme protocol restriction */
561 .max_phys_sect = 64,
562};
563#else
564static struct nvm_dev_ops null_lnvm_dev_ops;
565#endif /* CONFIG_NVM */
566
f2298c04
JA
567static int null_open(struct block_device *bdev, fmode_t mode)
568{
569 return 0;
570}
571
572static void null_release(struct gendisk *disk, fmode_t mode)
573{
574}
575
576static const struct block_device_operations null_fops = {
577 .owner = THIS_MODULE,
578 .open = null_open,
579 .release = null_release,
580};
581
582static int setup_commands(struct nullb_queue *nq)
583{
584 struct nullb_cmd *cmd;
585 int i, tag_size;
586
587 nq->cmds = kzalloc(nq->queue_depth * sizeof(*cmd), GFP_KERNEL);
588 if (!nq->cmds)
2d263a78 589 return -ENOMEM;
f2298c04
JA
590
591 tag_size = ALIGN(nq->queue_depth, BITS_PER_LONG) / BITS_PER_LONG;
592 nq->tag_map = kzalloc(tag_size * sizeof(unsigned long), GFP_KERNEL);
593 if (!nq->tag_map) {
594 kfree(nq->cmds);
2d263a78 595 return -ENOMEM;
f2298c04
JA
596 }
597
598 for (i = 0; i < nq->queue_depth; i++) {
599 cmd = &nq->cmds[i];
600 INIT_LIST_HEAD(&cmd->list);
601 cmd->ll_list.next = NULL;
602 cmd->tag = -1U;
603 }
604
605 return 0;
606}
607
f2298c04
JA
608static int setup_queues(struct nullb *nullb)
609{
2d263a78
MB
610 nullb->queues = kzalloc(submit_queues * sizeof(struct nullb_queue),
611 GFP_KERNEL);
f2298c04 612 if (!nullb->queues)
2d263a78 613 return -ENOMEM;
f2298c04
JA
614
615 nullb->nr_queues = 0;
616 nullb->queue_depth = hw_queue_depth;
617
2d263a78
MB
618 return 0;
619}
620
621static int init_driver_queues(struct nullb *nullb)
622{
623 struct nullb_queue *nq;
624 int i, ret = 0;
f2298c04
JA
625
626 for (i = 0; i < submit_queues; i++) {
627 nq = &nullb->queues[i];
2d263a78
MB
628
629 null_init_queue(nullb, nq);
630
631 ret = setup_commands(nq);
632 if (ret)
31f9690e 633 return ret;
f2298c04
JA
634 nullb->nr_queues++;
635 }
2d263a78 636 return 0;
f2298c04
JA
637}
638
639static int null_add_dev(void)
640{
641 struct gendisk *disk;
642 struct nullb *nullb;
643 sector_t size;
dc501dc0 644 int rv;
f2298c04
JA
645
646 nullb = kzalloc_node(sizeof(*nullb), GFP_KERNEL, home_node);
dc501dc0
RE
647 if (!nullb) {
648 rv = -ENOMEM;
24d2f903 649 goto out;
dc501dc0 650 }
f2298c04
JA
651
652 spin_lock_init(&nullb->lock);
653
57053d8c
MB
654 if (queue_mode == NULL_Q_MQ && use_per_node_hctx)
655 submit_queues = nr_online_nodes;
656
dc501dc0
RE
657 rv = setup_queues(nullb);
658 if (rv)
24d2f903 659 goto out_free_nullb;
f2298c04
JA
660
661 if (queue_mode == NULL_Q_MQ) {
cdef54dd 662 nullb->tag_set.ops = &null_mq_ops;
24d2f903
CH
663 nullb->tag_set.nr_hw_queues = submit_queues;
664 nullb->tag_set.queue_depth = hw_queue_depth;
665 nullb->tag_set.numa_node = home_node;
666 nullb->tag_set.cmd_size = sizeof(struct nullb_cmd);
667 nullb->tag_set.flags = BLK_MQ_F_SHOULD_MERGE;
668 nullb->tag_set.driver_data = nullb;
669
dc501dc0
RE
670 rv = blk_mq_alloc_tag_set(&nullb->tag_set);
671 if (rv)
24d2f903
CH
672 goto out_cleanup_queues;
673
674 nullb->q = blk_mq_init_queue(&nullb->tag_set);
35b489d3 675 if (IS_ERR(nullb->q)) {
dc501dc0 676 rv = -ENOMEM;
24d2f903 677 goto out_cleanup_tags;
dc501dc0 678 }
f2298c04
JA
679 } else if (queue_mode == NULL_Q_BIO) {
680 nullb->q = blk_alloc_queue_node(GFP_KERNEL, home_node);
dc501dc0
RE
681 if (!nullb->q) {
682 rv = -ENOMEM;
24d2f903 683 goto out_cleanup_queues;
dc501dc0 684 }
f2298c04 685 blk_queue_make_request(nullb->q, null_queue_bio);
31f9690e
JK
686 rv = init_driver_queues(nullb);
687 if (rv)
688 goto out_cleanup_blk_queue;
f2298c04
JA
689 } else {
690 nullb->q = blk_init_queue_node(null_request_fn, &nullb->lock, home_node);
dc501dc0
RE
691 if (!nullb->q) {
692 rv = -ENOMEM;
24d2f903 693 goto out_cleanup_queues;
dc501dc0 694 }
f2298c04 695 blk_queue_prep_rq(nullb->q, null_rq_prep_fn);
24d2f903 696 blk_queue_softirq_done(nullb->q, null_softirq_done_fn);
31f9690e
JK
697 rv = init_driver_queues(nullb);
698 if (rv)
699 goto out_cleanup_blk_queue;
f2298c04
JA
700 }
701
f2298c04
JA
702 nullb->q->queuedata = nullb;
703 queue_flag_set_unlocked(QUEUE_FLAG_NONROT, nullb->q);
b277da0a 704 queue_flag_clear_unlocked(QUEUE_FLAG_ADD_RANDOM, nullb->q);
f2298c04 705
f2298c04
JA
706
707 mutex_lock(&lock);
708 list_add_tail(&nullb->list, &nullb_list);
709 nullb->index = nullb_indexes++;
710 mutex_unlock(&lock);
711
712 blk_queue_logical_block_size(nullb->q, bs);
713 blk_queue_physical_block_size(nullb->q, bs);
714
b2b7e001
MB
715 sprintf(nullb->disk_name, "nullb%d", nullb->index);
716
717 if (use_lightnvm) {
718 rv = nvm_register(nullb->q, nullb->disk_name,
719 &null_lnvm_dev_ops);
720 if (rv)
721 goto out_cleanup_blk_queue;
722 goto done;
723 }
724
725 disk = nullb->disk = alloc_disk_node(1, home_node);
726 if (!disk) {
727 rv = -ENOMEM;
728 goto out_cleanup_lightnvm;
729 }
f2298c04 730 size = gb * 1024 * 1024 * 1024ULL;
5fdb7e1b 731 set_capacity(disk, size >> 9);
f2298c04 732
227290b4 733 disk->flags |= GENHD_FL_EXT_DEVT | GENHD_FL_SUPPRESS_PARTITION_INFO;
f2298c04
JA
734 disk->major = null_major;
735 disk->first_minor = nullb->index;
736 disk->fops = &null_fops;
737 disk->private_data = nullb;
738 disk->queue = nullb->q;
b2b7e001
MB
739 strncpy(disk->disk_name, nullb->disk_name, DISK_NAME_LEN);
740
f2298c04 741 add_disk(disk);
b2b7e001 742done:
f2298c04 743 return 0;
24d2f903 744
b2b7e001
MB
745out_cleanup_lightnvm:
746 if (use_lightnvm)
747 nvm_unregister(nullb->disk_name);
24d2f903
CH
748out_cleanup_blk_queue:
749 blk_cleanup_queue(nullb->q);
750out_cleanup_tags:
751 if (queue_mode == NULL_Q_MQ)
752 blk_mq_free_tag_set(&nullb->tag_set);
753out_cleanup_queues:
754 cleanup_queues(nullb);
755out_free_nullb:
756 kfree(nullb);
757out:
dc501dc0 758 return rv;
f2298c04
JA
759}
760
761static int __init null_init(void)
762{
763 unsigned int i;
764
9967d8ac
R
765 if (bs > PAGE_SIZE) {
766 pr_warn("null_blk: invalid block size\n");
767 pr_warn("null_blk: defaults block size to %lu\n", PAGE_SIZE);
768 bs = PAGE_SIZE;
769 }
f2298c04 770
6bb9535b
MB
771 if (use_lightnvm && bs != 4096) {
772 pr_warn("null_blk: LightNVM only supports 4k block size\n");
773 pr_warn("null_blk: defaults block size to 4k\n");
774 bs = 4096;
775 }
776
b2b7e001
MB
777 if (use_lightnvm && queue_mode != NULL_Q_MQ) {
778 pr_warn("null_blk: LightNVM only supported for blk-mq\n");
779 pr_warn("null_blk: defaults queue mode to blk-mq\n");
780 queue_mode = NULL_Q_MQ;
781 }
782
d15ee6b1 783 if (queue_mode == NULL_Q_MQ && use_per_node_hctx) {
fc1bc354 784 if (submit_queues < nr_online_nodes) {
d15ee6b1
MB
785 pr_warn("null_blk: submit_queues param is set to %u.",
786 nr_online_nodes);
fc1bc354
MB
787 submit_queues = nr_online_nodes;
788 }
d15ee6b1 789 } else if (submit_queues > nr_cpu_ids)
f2298c04
JA
790 submit_queues = nr_cpu_ids;
791 else if (!submit_queues)
792 submit_queues = 1;
793
794 mutex_init(&lock);
795
796 /* Initialize a separate list for each CPU for issuing softirqs */
797 for_each_possible_cpu(i) {
798 struct completion_queue *cq = &per_cpu(completion_queues, i);
799
800 init_llist_head(&cq->list);
801
802 if (irqmode != NULL_IRQ_TIMER)
803 continue;
804
805 hrtimer_init(&cq->timer, CLOCK_MONOTONIC, HRTIMER_MODE_REL);
806 cq->timer.function = null_cmd_timer_expired;
807 }
808
809 null_major = register_blkdev(0, "nullb");
810 if (null_major < 0)
811 return null_major;
812
6bb9535b
MB
813 if (use_lightnvm) {
814 ppa_cache = kmem_cache_create("ppa_cache", 64 * sizeof(u64),
815 0, 0, NULL);
816 if (!ppa_cache) {
817 pr_err("null_blk: unable to create ppa cache\n");
818 return -ENOMEM;
819 }
820 }
821
f2298c04
JA
822 for (i = 0; i < nr_devices; i++) {
823 if (null_add_dev()) {
824 unregister_blkdev(null_major, "nullb");
6bb9535b 825 goto err_ppa;
f2298c04
JA
826 }
827 }
828
829 pr_info("null: module loaded\n");
830 return 0;
6bb9535b
MB
831err_ppa:
832 kmem_cache_destroy(ppa_cache);
833 return -EINVAL;
f2298c04
JA
834}
835
836static void __exit null_exit(void)
837{
838 struct nullb *nullb;
839
840 unregister_blkdev(null_major, "nullb");
841
842 mutex_lock(&lock);
843 while (!list_empty(&nullb_list)) {
844 nullb = list_entry(nullb_list.next, struct nullb, list);
845 null_del_dev(nullb);
846 }
847 mutex_unlock(&lock);
6bb9535b
MB
848
849 kmem_cache_destroy(ppa_cache);
f2298c04
JA
850}
851
852module_init(null_init);
853module_exit(null_exit);
854
855MODULE_AUTHOR("Jens Axboe <jaxboe@fusionio.com>");
856MODULE_LICENSE("GPL");
This page took 0.271772 seconds and 5 git commands to generate.