[mirror_ubuntu-hirsute-kernel.git] / block / bsg-lib.c

/*
 *  BSG helper library
 *
 *  Copyright (C) 2008   James Smart, Emulex Corporation
 *  Copyright (C) 2011   Red Hat, Inc.  All rights reserved.
 *  Copyright (C) 2011   Mike Christie
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */
#include <linux/slab.h>
#include <linux/blk-mq.h>
#include <linux/delay.h>
#include <linux/scatterlist.h>
#include <linux/bsg-lib.h>
#include <linux/export.h>
#include <scsi/scsi_cmnd.h>
#include <scsi/sg.h>

#define uptr64(val) ((void __user *)(uintptr_t)(val))

static int bsg_transport_check_proto(struct sg_io_v4 *hdr)
{
	if (hdr->protocol != BSG_PROTOCOL_SCSI  ||
	    hdr->subprotocol != BSG_SUB_PROTOCOL_SCSI_TRANSPORT)
		return -EINVAL;
	if (!capable(CAP_SYS_RAWIO))
		return -EPERM;
	return 0;
}

static int bsg_transport_fill_hdr(struct request *rq, struct sg_io_v4 *hdr,
		fmode_t mode)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(rq);

	job->request_len = hdr->request_len;
	job->request = memdup_user(uptr64(hdr->request), hdr->request_len);

	return PTR_ERR_OR_ZERO(job->request);
}

static int bsg_transport_complete_rq(struct request *rq, struct sg_io_v4 *hdr)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(rq);
	int ret = 0;

	/*
	 * The assignments below don't make much sense, but are kept for
	 * bug by bug backwards compatibility:
	 */
	hdr->device_status = job->result & 0xff;
	hdr->transport_status = host_byte(job->result);
	hdr->driver_status = driver_byte(job->result);
	hdr->info = 0;
	if (hdr->device_status || hdr->transport_status || hdr->driver_status)
		hdr->info |= SG_INFO_CHECK;
	hdr->response_len = 0;

	if (job->result < 0) {
		/* we're only returning the result field in the reply */
		job->reply_len = sizeof(u32);
		ret = job->result;
	}

	if (job->reply_len && hdr->response) {
		int len = min(hdr->max_response_len, job->reply_len);

		if (copy_to_user(uptr64(hdr->response), job->reply, len))
			ret = -EFAULT;
		else
			hdr->response_len = len;
	}

	/* we assume all request payload was transferred, residual == 0 */
	hdr->dout_resid = 0;

	if (rq->next_rq) {
		unsigned int rsp_len = job->reply_payload.payload_len;

		if (WARN_ON(job->reply_payload_rcv_len > rsp_len))
			hdr->din_resid = 0;
		else
			hdr->din_resid = rsp_len - job->reply_payload_rcv_len;
	} else {
		hdr->din_resid = 0;
	}

	return ret;
}

static void bsg_transport_free_rq(struct request *rq)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(rq);

	kfree(job->request);
}

static const struct bsg_ops bsg_transport_ops = {
	.check_proto		= bsg_transport_check_proto,
	.fill_hdr		= bsg_transport_fill_hdr,
	.complete_rq		= bsg_transport_complete_rq,
	.free_rq		= bsg_transport_free_rq,
};

/**
 * bsg_teardown_job - routine to teardown a bsg job
 * @kref: kref inside bsg_job that is to be torn down
 */
static void bsg_teardown_job(struct kref *kref)
{
	struct bsg_job *job = container_of(kref, struct bsg_job, kref);
	struct request *rq = blk_mq_rq_from_pdu(job);

	put_device(job->dev);	/* release reference for the request */

	kfree(job->request_payload.sg_list);
	kfree(job->reply_payload.sg_list);

	blk_mq_end_request(rq, BLK_STS_OK);
}

void bsg_job_put(struct bsg_job *job)
{
	kref_put(&job->kref, bsg_teardown_job);
}
EXPORT_SYMBOL_GPL(bsg_job_put);

int bsg_job_get(struct bsg_job *job)
{
	return kref_get_unless_zero(&job->kref);
}
EXPORT_SYMBOL_GPL(bsg_job_get);

/**
 * bsg_job_done - completion routine for bsg requests
 * @job: bsg_job that is complete
 * @result: job reply result
 * @reply_payload_rcv_len: length of payload recvd
 *
 * The LLD should call this when the bsg job has completed.
 */
void bsg_job_done(struct bsg_job *job, int result,
		  unsigned int reply_payload_rcv_len)
{
	job->result = result;
	job->reply_payload_rcv_len = reply_payload_rcv_len;
	blk_mq_complete_request(blk_mq_rq_from_pdu(job));
}
EXPORT_SYMBOL_GPL(bsg_job_done);

/**
 * bsg_complete - softirq done routine for destroying the bsg requests
 * @rq: BSG request that holds the job to be destroyed
 */
static void bsg_complete(struct request *rq)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(rq);

	bsg_job_put(job);
}

static int bsg_map_buffer(struct bsg_buffer *buf, struct request *req)
{
	size_t sz = (sizeof(struct scatterlist) * req->nr_phys_segments);

	BUG_ON(!req->nr_phys_segments);

	buf->sg_list = kzalloc(sz, GFP_KERNEL);
	if (!buf->sg_list)
		return -ENOMEM;
	sg_init_table(buf->sg_list, req->nr_phys_segments);
	buf->sg_cnt = blk_rq_map_sg(req->q, req, buf->sg_list);
	buf->payload_len = blk_rq_bytes(req);
	return 0;
}

/**
 * bsg_prepare_job - create the bsg_job structure for the bsg request
 * @dev: device that is being sent the bsg request
 * @req: BSG request that needs a job structure
 */
static bool bsg_prepare_job(struct device *dev, struct request *req)
{
	struct request *rsp = req->next_rq;
	struct bsg_job *job = blk_mq_rq_to_pdu(req);
	int ret;

	job->timeout = req->timeout;

	if (req->bio) {
		ret = bsg_map_buffer(&job->request_payload, req);
		if (ret)
			goto failjob_rls_job;
	}
	if (rsp && rsp->bio) {
		ret = bsg_map_buffer(&job->reply_payload, rsp);
		if (ret)
			goto failjob_rls_rqst_payload;
	}
	job->dev = dev;
	/* take a reference for the request */
	get_device(job->dev);
	kref_init(&job->kref);
	return true;

failjob_rls_rqst_payload:
	kfree(job->request_payload.sg_list);
failjob_rls_job:
	job->result = -ENOMEM;
	return false;
}

/**
 * bsg_queue_rq - generic handler for bsg requests
 * @hctx: hardware queue
 * @bd: queue data
 *
 * On error the create_bsg_job function should return a -Exyz error value
 * that will be set to ->result.
 *
 * Drivers/subsys should pass this to the queue init function.
 */
static blk_status_t bsg_queue_rq(struct blk_mq_hw_ctx *hctx,
				 const struct blk_mq_queue_data *bd)
{
	struct request_queue *q = hctx->queue;
	struct device *dev = q->queuedata;
	struct request *req = bd->rq;
	int ret;

	blk_mq_start_request(req);

	if (!get_device(dev))
		return BLK_STS_IOERR;

	if (!bsg_prepare_job(dev, req))
		return BLK_STS_IOERR;

	ret = q->bsg_job_fn(blk_mq_rq_to_pdu(req));
	if (ret)
		return BLK_STS_IOERR;

	put_device(dev);
	return BLK_STS_OK;
}

/* called right after the request is allocated for the request_queue */
static int bsg_init_rq(struct blk_mq_tag_set *set, struct request *req,
		       unsigned int hctx_idx, unsigned int numa_node)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(req);

	job->reply = kzalloc(SCSI_SENSE_BUFFERSIZE, GFP_KERNEL);
	if (!job->reply)
		return -ENOMEM;
	return 0;
}

/* called right before the request is given to the request_queue user */
static void bsg_initialize_rq(struct request *req)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(req);
	void *reply = job->reply;

	memset(job, 0, sizeof(*job));
	job->reply = reply;
	job->reply_len = SCSI_SENSE_BUFFERSIZE;
	job->dd_data = job + 1;
}

static void bsg_exit_rq(struct blk_mq_tag_set *set, struct request *req,
		       unsigned int hctx_idx)
{
	struct bsg_job *job = blk_mq_rq_to_pdu(req);

	kfree(job->reply);
}

void bsg_remove_queue(struct request_queue *q)
{
	if (q) {
		struct blk_mq_tag_set *set = q->tag_set;

		bsg_unregister_queue(q);
		blk_cleanup_queue(q);
		blk_mq_free_tag_set(set);
		kfree(set);
	}
}
EXPORT_SYMBOL_GPL(bsg_remove_queue);

static enum blk_eh_timer_return bsg_timeout(struct request *rq, bool reserved)
{
	enum blk_eh_timer_return ret = BLK_EH_DONE;
	struct request_queue *q = rq->q;

	if (q->bsg_job_timeout_fn)
		ret = q->bsg_job_timeout_fn(rq);

	return ret;
}

static const struct blk_mq_ops bsg_mq_ops = {
	.queue_rq		= bsg_queue_rq,
	.init_request		= bsg_init_rq,
	.exit_request		= bsg_exit_rq,
	.initialize_rq_fn	= bsg_initialize_rq,
	.complete		= bsg_complete,
	.timeout		= bsg_timeout,
};

/**
 * bsg_setup_queue - Create and add the bsg hooks so we can receive requests
 * @dev: device to attach bsg device to
 * @name: device to give bsg device
 * @job_fn: bsg job handler
 * @dd_job_size: size of LLD data needed for each job
 */
struct request_queue *bsg_setup_queue(struct device *dev, const char *name,
		bsg_job_fn *job_fn, rq_timed_out_fn *timeout, int dd_job_size)
{
	struct blk_mq_tag_set *set;
	struct request_queue *q;
	int ret = -ENOMEM;

	set = kzalloc(sizeof(*set), GFP_KERNEL);
	if (!set)
		return ERR_PTR(-ENOMEM);

	set->ops = &bsg_mq_ops,
	set->nr_hw_queues = 1;
	set->queue_depth = 128;
	set->numa_node = NUMA_NO_NODE;
	set->cmd_size = sizeof(struct bsg_job) + dd_job_size;
	set->flags = BLK_MQ_F_NO_SCHED | BLK_MQ_F_BLOCKING;
	if (blk_mq_alloc_tag_set(set))
		goto out_tag_set;

	q = blk_mq_init_queue(set);
	if (IS_ERR(q)) {
		ret = PTR_ERR(q);
		goto out_queue;
	}

	q->queuedata = dev;
	q->bsg_job_fn = job_fn;
	q->bsg_job_timeout_fn = timeout;
	blk_queue_flag_set(QUEUE_FLAG_BIDI, q);
	blk_queue_rq_timeout(q, BLK_DEFAULT_SG_TIMEOUT);

	ret = bsg_register_queue(q, dev, name, &bsg_transport_ops);
	if (ret) {
		printk(KERN_ERR "%s: bsg interface failed to "
		       "initialize - register queue\n", dev->kobj.name);
		goto out_cleanup_queue;
	}

	return q;
out_cleanup_queue:
	blk_cleanup_queue(q);
out_queue:
	blk_mq_free_tag_set(set);
out_tag_set:
	kfree(set);
	return ERR_PTR(ret);
}
EXPORT_SYMBOL_GPL(bsg_setup_queue);
Commit	Line	Data
aa387cc8 MC	1	/*
	2	* BSG helper library
	3	*
	4	* Copyright (C) 2008 James Smart, Emulex Corporation
	5	* Copyright (C) 2011 Red Hat, Inc. All rights reserved.
	6	* Copyright (C) 2011 Mike Christie
	7	*
	8	* This program is free software; you can redistribute it and/or modify
	9	* it under the terms of the GNU General Public License as published by
	10	* the Free Software Foundation; either version 2 of the License, or
	11	* (at your option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	21	*
	22	*/
	23	#include <linux/slab.h>
cd2f076f	24	#include <linux/blk-mq.h>
aa387cc8 MC	25	#include <linux/delay.h>
	26	#include <linux/scatterlist.h>
	27	#include <linux/bsg-lib.h>
6adb1236	28	#include <linux/export.h>
aa387cc8	29	#include <scsi/scsi_cmnd.h>
17cb960f CH	30	#include <scsi/sg.h>
	31
	32	#define uptr64(val) ((void __user *)(uintptr_t)(val))
	33
	34	static int bsg_transport_check_proto(struct sg_io_v4 *hdr)
	35	{
	36	if (hdr->protocol != BSG_PROTOCOL_SCSI \|\|
	37	hdr->subprotocol != BSG_SUB_PROTOCOL_SCSI_TRANSPORT)
	38	return -EINVAL;
	39	if (!capable(CAP_SYS_RAWIO))
	40	return -EPERM;
	41	return 0;
	42	}
	43
	44	static int bsg_transport_fill_hdr(struct request rq, struct sg_io_v4 hdr,
	45	fmode_t mode)
	46	{
	47	struct bsg_job *job = blk_mq_rq_to_pdu(rq);
	48
	49	job->request_len = hdr->request_len;
	50	job->request = memdup_user(uptr64(hdr->request), hdr->request_len);
47255491	51
47255491	52	return PTR_ERR_OR_ZERO(job->request);
17cb960f CH	53	}
	54
	55	static int bsg_transport_complete_rq(struct request rq, struct sg_io_v4 hdr)
	56	{
	57	struct bsg_job *job = blk_mq_rq_to_pdu(rq);
	58	int ret = 0;
	59
	60	/*
	61	* The assignments below don't make much sense, but are kept for
	62	* bug by bug backwards compatibility:
	63	*/
	64	hdr->device_status = job->result & 0xff;
	65	hdr->transport_status = host_byte(job->result);
	66	hdr->driver_status = driver_byte(job->result);
	67	hdr->info = 0;
	68	if (hdr->device_status \|\| hdr->transport_status \|\| hdr->driver_status)
	69	hdr->info \|= SG_INFO_CHECK;
	70	hdr->response_len = 0;
	71
	72	if (job->result < 0) {
	73	/* we're only returning the result field in the reply */
	74	job->reply_len = sizeof(u32);
	75	ret = job->result;
	76	}
	77
	78	if (job->reply_len && hdr->response) {
	79	int len = min(hdr->max_response_len, job->reply_len);
	80
	81	if (copy_to_user(uptr64(hdr->response), job->reply, len))
	82	ret = -EFAULT;
	83	else
	84	hdr->response_len = len;
	85	}
	86
	87	/* we assume all request payload was transferred, residual == 0 */
	88	hdr->dout_resid = 0;
	89
	90	if (rq->next_rq) {
	91	unsigned int rsp_len = job->reply_payload.payload_len;
	92
	93	if (WARN_ON(job->reply_payload_rcv_len > rsp_len))
	94	hdr->din_resid = 0;
	95	else
	96	hdr->din_resid = rsp_len - job->reply_payload_rcv_len;
	97	} else {
	98	hdr->din_resid = 0;
	99	}
	100
	101	return ret;
	102	}
	103
	104	static void bsg_transport_free_rq(struct request *rq)
	105	{
	106	struct bsg_job *job = blk_mq_rq_to_pdu(rq);
	107
	108	kfree(job->request);
	109	}
	110
	111	static const struct bsg_ops bsg_transport_ops = {
	112	.check_proto = bsg_transport_check_proto,
	113	.fill_hdr = bsg_transport_fill_hdr,
	114	.complete_rq = bsg_transport_complete_rq,
	115	.free_rq = bsg_transport_free_rq,
	116	};
aa387cc8 MC	117
aa387cc8 MC	118	/**
50b4d485	119	* bsg_teardown_job - routine to teardown a bsg job
aa98192d	120	* @kref: kref inside bsg_job that is to be torn down
aa387cc8	121	*/
50b4d485	122	static void bsg_teardown_job(struct kref *kref)
aa387cc8	123	{
bf0f2d38	124	struct bsg_job *job = container_of(kref, struct bsg_job, kref);
ef6fa64f	125	struct request *rq = blk_mq_rq_from_pdu(job);
c00da4c9	126
aa387cc8 MC	127	put_device(job->dev); /* release reference for the request */
	128
	129	kfree(job->request_payload.sg_list);
	130	kfree(job->reply_payload.sg_list);
50b4d485	131
cd2f076f	132	blk_mq_end_request(rq, BLK_STS_OK);
aa387cc8 MC	133	}
aa387cc8 MC	134
fb6f7c8d JT	135	void bsg_job_put(struct bsg_job *job)
fb6f7c8d JT	136	{
50b4d485	137	kref_put(&job->kref, bsg_teardown_job);
fb6f7c8d JT	138	}
	139	EXPORT_SYMBOL_GPL(bsg_job_put);
	140
	141	int bsg_job_get(struct bsg_job *job)
	142	{
	143	return kref_get_unless_zero(&job->kref);
	144	}
	145	EXPORT_SYMBOL_GPL(bsg_job_get);
aa387cc8 MC	146
	147	/**
	148	* bsg_job_done - completion routine for bsg requests
	149	* @job: bsg_job that is complete
	150	* @result: job reply result
	151	* @reply_payload_rcv_len: length of payload recvd
	152	*
	153	* The LLD should call this when the bsg job has completed.
	154	*/
	155	void bsg_job_done(struct bsg_job *job, int result,
	156	unsigned int reply_payload_rcv_len)
	157	{
17cb960f CH	158	job->result = result;
17cb960f CH	159	job->reply_payload_rcv_len = reply_payload_rcv_len;
cd2f076f	160	blk_mq_complete_request(blk_mq_rq_from_pdu(job));
aa387cc8 MC	161	}
	162	EXPORT_SYMBOL_GPL(bsg_job_done);
	163
	164	/**
cd2f076f	165	* bsg_complete - softirq done routine for destroying the bsg requests
aa387cc8 MC	166	* @rq: BSG request that holds the job to be destroyed
aa387cc8 MC	167	*/
cd2f076f	168	static void bsg_complete(struct request *rq)
aa387cc8	169	{
50b4d485	170	struct bsg_job *job = blk_mq_rq_to_pdu(rq);
aa387cc8	171
fb6f7c8d	172	bsg_job_put(job);
aa387cc8 MC	173	}
	174
	175	static int bsg_map_buffer(struct bsg_buffer buf, struct request req)
	176	{
	177	size_t sz = (sizeof(struct scatterlist) * req->nr_phys_segments);
	178
	179	BUG_ON(!req->nr_phys_segments);
	180
	181	buf->sg_list = kzalloc(sz, GFP_KERNEL);
	182	if (!buf->sg_list)
	183	return -ENOMEM;
	184	sg_init_table(buf->sg_list, req->nr_phys_segments);
	185	buf->sg_cnt = blk_rq_map_sg(req->q, req, buf->sg_list);
	186	buf->payload_len = blk_rq_bytes(req);
	187	return 0;
	188	}
	189
	190	/**
50b4d485	191	* bsg_prepare_job - create the bsg_job structure for the bsg request
aa387cc8 MC	192	* @dev: device that is being sent the bsg request
	193	* @req: BSG request that needs a job structure
	194	*/
17cb960f	195	static bool bsg_prepare_job(struct device dev, struct request req)
aa387cc8 MC	196	{
aa387cc8 MC	197	struct request *rsp = req->next_rq;
50b4d485	198	struct bsg_job *job = blk_mq_rq_to_pdu(req);
aa387cc8 MC	199	int ret;
aa387cc8 MC	200
31156ec3	201	job->timeout = req->timeout;
50b4d485	202
aa387cc8 MC	203	if (req->bio) {
	204	ret = bsg_map_buffer(&job->request_payload, req);
	205	if (ret)
	206	goto failjob_rls_job;
	207	}
	208	if (rsp && rsp->bio) {
	209	ret = bsg_map_buffer(&job->reply_payload, rsp);
	210	if (ret)
	211	goto failjob_rls_rqst_payload;
	212	}
	213	job->dev = dev;
	214	/* take a reference for the request */
	215	get_device(job->dev);
bf0f2d38	216	kref_init(&job->kref);
17cb960f	217	return true;
aa387cc8 MC	218
	219	failjob_rls_rqst_payload:
	220	kfree(job->request_payload.sg_list);
	221	failjob_rls_job:
17cb960f CH	222	job->result = -ENOMEM;
17cb960f CH	223	return false;
aa387cc8 MC	224	}
aa387cc8 MC	225
aa387cc8	226	/**
cd2f076f JA	227	* bsg_queue_rq - generic handler for bsg requests
	228	* @hctx: hardware queue
	229	* @bd: queue data
aa387cc8 MC	230	*
aa387cc8 MC	231	* On error the create_bsg_job function should return a -Exyz error value
17d5363b	232	* that will be set to ->result.
aa387cc8 MC	233	*
	234	* Drivers/subsys should pass this to the queue init function.
	235	*/
cd2f076f JA	236	static blk_status_t bsg_queue_rq(struct blk_mq_hw_ctx *hctx,
cd2f076f JA	237	const struct blk_mq_queue_data *bd)
aa387cc8	238	{
cd2f076f	239	struct request_queue *q = hctx->queue;
aa387cc8	240	struct device *dev = q->queuedata;
cd2f076f	241	struct request *req = bd->rq;
aa387cc8 MC	242	int ret;
aa387cc8 MC	243
cd2f076f JA	244	blk_mq_start_request(req);
cd2f076f JA	245
aa387cc8	246	if (!get_device(dev))
cd2f076f JA	247	return BLK_STS_IOERR;
	248
	249	if (!bsg_prepare_job(dev, req))
	250	return BLK_STS_IOERR;
	251
	252	ret = q->bsg_job_fn(blk_mq_rq_to_pdu(req));
	253	if (ret)
	254	return BLK_STS_IOERR;
aa387cc8	255
aa387cc8	256	put_device(dev);
cd2f076f	257	return BLK_STS_OK;
aa387cc8	258	}
aa387cc8	259
17cb960f	260	/* called right after the request is allocated for the request_queue */
cd2f076f JA	261	static int bsg_init_rq(struct blk_mq_tag_set set, struct request req,
cd2f076f JA	262	unsigned int hctx_idx, unsigned int numa_node)
50b4d485 BB	263	{
50b4d485 BB	264	struct bsg_job *job = blk_mq_rq_to_pdu(req);
eab40cf3	265
cd2f076f	266	job->reply = kzalloc(SCSI_SENSE_BUFFERSIZE, GFP_KERNEL);
17cb960f	267	if (!job->reply)
eab40cf3	268	return -ENOMEM;
eab40cf3 BB	269	return 0;
	270	}
	271
17cb960f	272	/* called right before the request is given to the request_queue user */
eab40cf3 BB	273	static void bsg_initialize_rq(struct request *req)
	274	{
	275	struct bsg_job *job = blk_mq_rq_to_pdu(req);
17cb960f	276	void *reply = job->reply;
eab40cf3	277
50b4d485	278	memset(job, 0, sizeof(*job));
17cb960f CH	279	job->reply = reply;
17cb960f CH	280	job->reply_len = SCSI_SENSE_BUFFERSIZE;
50b4d485	281	job->dd_data = job + 1;
50b4d485 BB	282	}
50b4d485 BB	283
cd2f076f JA	284	static void bsg_exit_rq(struct blk_mq_tag_set set, struct request req,
cd2f076f JA	285	unsigned int hctx_idx)
50b4d485 BB	286	{
50b4d485 BB	287	struct bsg_job *job = blk_mq_rq_to_pdu(req);
50b4d485	288
17cb960f	289	kfree(job->reply);
50b4d485 BB	290	}
50b4d485 BB	291
5e28b8d8 JA	292	void bsg_remove_queue(struct request_queue *q)
	293	{
	294	if (q) {
cd2f076f JA	295	struct blk_mq_tag_set *set = q->tag_set;
cd2f076f JA	296
5e28b8d8 JA	297	bsg_unregister_queue(q);
5e28b8d8 JA	298	blk_cleanup_queue(q);
cd2f076f JA	299	blk_mq_free_tag_set(set);
cd2f076f JA	300	kfree(set);
5e28b8d8 JA	301	}
	302	}
	303	EXPORT_SYMBOL_GPL(bsg_remove_queue);
	304
cd2f076f JA	305	static enum blk_eh_timer_return bsg_timeout(struct request *rq, bool reserved)
	306	{
	307	enum blk_eh_timer_return ret = BLK_EH_DONE;
	308	struct request_queue *q = rq->q;
	309
4316b79e JA	310	if (q->bsg_job_timeout_fn)
4316b79e JA	311	ret = q->bsg_job_timeout_fn(rq);
cd2f076f JA	312
	313	return ret;
	314	}
	315
	316	static const struct blk_mq_ops bsg_mq_ops = {
	317	.queue_rq = bsg_queue_rq,
	318	.init_request = bsg_init_rq,
	319	.exit_request = bsg_exit_rq,
	320	.initialize_rq_fn = bsg_initialize_rq,
	321	.complete = bsg_complete,
	322	.timeout = bsg_timeout,
	323	};
	324
aa387cc8 MC	325	/**
	326	* bsg_setup_queue - Create and add the bsg hooks so we can receive requests
	327	* @dev: device to attach bsg device to
aa387cc8 MC	328	* @name: device to give bsg device
	329	* @job_fn: bsg job handler
	330	* @dd_job_size: size of LLD data needed for each job
aa387cc8	331	*/
c1225f01	332	struct request_queue bsg_setup_queue(struct device dev, const char *name,
aae3b069	333	bsg_job_fn job_fn, rq_timed_out_fn timeout, int dd_job_size)
aa387cc8	334	{
cd2f076f	335	struct blk_mq_tag_set *set;
8ae94eb6	336	struct request_queue *q;
cd2f076f	337	int ret = -ENOMEM;
aa387cc8	338
cd2f076f JA	339	set = kzalloc(sizeof(*set), GFP_KERNEL);
cd2f076f JA	340	if (!set)
8ae94eb6	341	return ERR_PTR(-ENOMEM);
82ed4db4	342
cd2f076f JA	343	set->ops = &bsg_mq_ops,
	344	set->nr_hw_queues = 1;
	345	set->queue_depth = 128;
	346	set->numa_node = NUMA_NO_NODE;
	347	set->cmd_size = sizeof(struct bsg_job) + dd_job_size;
	348	set->flags = BLK_MQ_F_NO_SCHED \| BLK_MQ_F_BLOCKING;
	349	if (blk_mq_alloc_tag_set(set))
	350	goto out_tag_set;
	351
	352	q = blk_mq_init_queue(set);
	353	if (IS_ERR(q)) {
	354	ret = PTR_ERR(q);
	355	goto out_queue;
	356	}
8ae94eb6	357
aa387cc8	358	q->queuedata = dev;
aa387cc8	359	q->bsg_job_fn = job_fn;
4316b79e	360	q->bsg_job_timeout_fn = timeout;
8b904b5b	361	blk_queue_flag_set(QUEUE_FLAG_BIDI, q);
aa387cc8 MC	362	blk_queue_rq_timeout(q, BLK_DEFAULT_SG_TIMEOUT);
aa387cc8 MC	363
5de815a7	364	ret = bsg_register_queue(q, dev, name, &bsg_transport_ops);
aa387cc8 MC	365	if (ret) {
	366	printk(KERN_ERR "%s: bsg interface failed to "
	367	"initialize - register queue\n", dev->kobj.name);
82ed4db4	368	goto out_cleanup_queue;
aa387cc8 MC	369	}
aa387cc8 MC	370
8ae94eb6	371	return q;
82ed4db4 CH	372	out_cleanup_queue:
82ed4db4 CH	373	blk_cleanup_queue(q);
cd2f076f JA	374	out_queue:
	375	blk_mq_free_tag_set(set);
	376	out_tag_set:
	377	kfree(set);
82ed4db4	378	return ERR_PTR(ret);
aa387cc8 MC	379	}
aa387cc8 MC	380	EXPORT_SYMBOL_GPL(bsg_setup_queue);