706 static void put_tag(struct nullb_queue *nq, unsigned int tag)  in put_tag()  argument
708 	clear_bit_unlock(tag, nq->tag_map);  in put_tag()
710 	if (waitqueue_active(&nq->wait))  in put_tag()
711 		wake_up(&nq->wait);  in put_tag()
714 static unsigned int get_tag(struct nullb_queue *nq)  in get_tag()  argument
719 		tag = find_first_zero_bit(nq->tag_map, nq->queue_depth);  in get_tag()
720 		if (tag >= nq->queue_depth)  in get_tag()
722 	} while (test_and_set_bit_lock(tag, nq->tag_map));  in get_tag()
729 	put_tag(cmd->nq, cmd->tag);  in free_cmd()
734 static struct nullb_cmd *__alloc_cmd(struct nullb_queue *nq)  in __alloc_cmd()  argument
739 	tag = get_tag(nq);  in __alloc_cmd()
741 		cmd = &nq->cmds[tag];  in __alloc_cmd()
744 		cmd->nq = nq;  in __alloc_cmd()
745 		if (nq->dev->irqmode == NULL_IRQ_TIMER) {  in __alloc_cmd()
756 static struct nullb_cmd *alloc_cmd(struct nullb_queue *nq, struct bio *bio)  in alloc_cmd()  argument
766 		cmd = __alloc_cmd(nq);  in alloc_cmd()
771 		prepare_to_wait(&nq->wait, &wait, TASK_UNINTERRUPTIBLE);  in alloc_cmd()
773 		finish_wait(&nq->wait, &wait);  in alloc_cmd()
779 	int queue_mode = cmd->nq->dev->queue_mode;  in end_cmd()
803 	ktime_t kt = cmd->nq->dev->completion_nsec;  in null_cmd_end_timer()
1243 	struct nullb *nullb = cmd->nq->dev->nullb;  in null_handle_rq()
1270 	struct nullb *nullb = cmd->nq->dev->nullb;  in null_handle_bio()
1311 	struct nullb_device *dev = cmd->nq->dev;  in null_handle_throttled()
1334 	struct badblocks *bb = &cmd->nq->dev->badblocks;  in null_handle_badblocks()
1349 	struct nullb_device *dev = cmd->nq->dev;  in null_handle_memory_backed()
1365 	struct nullb_device *dev = cmd->nq->dev;  in nullb_zero_read_cmd_buffer()
1392 	switch (cmd->nq->dev->irqmode) {  in nullb_complete_cmd()
1394 		switch (cmd->nq->dev->queue_mode) {  in nullb_complete_cmd()
1419 	struct nullb_device *dev = cmd->nq->dev;  in null_process_cmd()
1437 	struct nullb_device *dev = cmd->nq->dev;  in null_handle_cmd()
1508 	struct nullb_queue *nq = nullb_to_queue(nullb);  in null_submit_bio()  local
1510 	null_handle_cmd(alloc_cmd(nq, bio), sector, nr_sectors, bio_op(bio));  in null_submit_bio()
1586 	struct nullb_queue *nq = hctx->driver_data;  in null_poll()  local
1590 	spin_lock(&nq->poll_lock);  in null_poll()
1591 	list_splice_init(&nq->poll_list, &list);  in null_poll()
1592 	spin_unlock(&nq->poll_lock);  in null_poll()
1620 		struct nullb_queue *nq = hctx->driver_data;  in null_timeout_rq()  local
1622 		spin_lock(&nq->poll_lock);  in null_timeout_rq()
1624 		spin_unlock(&nq->poll_lock);  in null_timeout_rq()
1644 	struct nullb_queue *nq = hctx->driver_data;  in null_queue_rq()  local
1651 	if (!is_poll && nq->dev->irqmode == NULL_IRQ_TIMER) {  in null_queue_rq()
1657 	cmd->nq = nq;  in null_queue_rq()
1667 		nq->requeue_selection++;  in null_queue_rq()
1668 		if (nq->requeue_selection & 1)  in null_queue_rq()
1677 		spin_lock(&nq->poll_lock);  in null_queue_rq()
1678 		list_add_tail(&bd->rq->queuelist, &nq->poll_list);  in null_queue_rq()
1679 		spin_unlock(&nq->poll_lock);  in null_queue_rq()
1688 static void cleanup_queue(struct nullb_queue *nq)  in cleanup_queue()  argument
1690 	bitmap_free(nq->tag_map);  in cleanup_queue()
1691 	kfree(nq->cmds);  in cleanup_queue()
1706 	struct nullb_queue *nq = hctx->driver_data;  in null_exit_hctx()  local
1707 	struct nullb *nullb = nq->dev->nullb;  in null_exit_hctx()
1712 static void null_init_queue(struct nullb *nullb, struct nullb_queue *nq)  in null_init_queue()  argument
1714 	init_waitqueue_head(&nq->wait);  in null_init_queue()
1715 	nq->queue_depth = nullb->queue_depth;  in null_init_queue()
1716 	nq->dev = nullb->dev;  in null_init_queue()
1717 	INIT_LIST_HEAD(&nq->poll_list);  in null_init_queue()
1718 	spin_lock_init(&nq->poll_lock);  in null_init_queue()
1725 	struct nullb_queue *nq;  in null_init_hctx()  local
1732 	nq = &nullb->queues[hctx_idx];  in null_init_hctx()
1733 	hctx->driver_data = nq;  in null_init_hctx()
1734 	null_init_queue(nullb, nq);  in null_init_hctx()
1814 static int setup_commands(struct nullb_queue *nq)  in setup_commands()  argument
1819 	nq->cmds = kcalloc(nq->queue_depth, sizeof(*cmd), GFP_KERNEL);  in setup_commands()
1820 	if (!nq->cmds)  in setup_commands()
1823 	nq->tag_map = bitmap_zalloc(nq->queue_depth, GFP_KERNEL);  in setup_commands()
1824 	if (!nq->tag_map) {  in setup_commands()
1825 		kfree(nq->cmds);  in setup_commands()
1829 	for (i = 0; i < nq->queue_depth; i++) {  in setup_commands()
1830 		cmd = &nq->cmds[i];  in setup_commands()
1855 	struct nullb_queue *nq;  in init_driver_queues()  local
1859 		nq = &nullb->queues[i];  in init_driver_queues()
1861 		null_init_queue(nullb, nq);  in init_driver_queues()
1863 		ret = setup_commands(nq);  in init_driver_queues()