linux - drivers/target/target_core

3403 lines · 86.2 KiB

1	// SPDX-License-Identifier: GPL-2.0-only
2	/*
3	* Copyright (C) 2013 Shaohua Li <[email protected]>
4	* Copyright (C) 2014 Red Hat, Inc.
5	* Copyright (C) 2015 Arrikto, Inc.
6	* Copyright (C) 2017 Chinamobile, Inc.
7	*/
8
9	#include <linux/spinlock.h>
10	#include <linux/module.h>
11	#include <linux/kernel.h>
12	#include <linux/timer.h>
13	#include <linux/parser.h>
14	#include <linux/vmalloc.h>
15	#include <linux/uio_driver.h>
16	#include <linux/xarray.h>
17	#include <linux/stringify.h>
18	#include <linux/bitops.h>
19	#include <linux/highmem.h>
20	#include <linux/configfs.h>
21	#include <linux/mutex.h>
22	#include <linux/workqueue.h>
23	#include <linux/pagemap.h>
24	#include <net/genetlink.h>
25	#include <scsi/scsi_common.h>
26	#include <scsi/scsi_proto.h>
27	#include <target/target_core_base.h>
28	#include <target/target_core_fabric.h>
29	#include <target/target_core_backend.h>
30
31	#include <linux/target_core_user.h>
32
33	/**
34	* DOC: Userspace I/O
35	* Userspace I/O
36	* -------------
37	*
38	* Define a shared-memory interface for LIO to pass SCSI commands and
39	* data to userspace for processing. This is to allow backends that
40	* are too complex for in-kernel support to be possible.
41	*
42	* It uses the UIO framework to do a lot of the device-creation and
43	* introspection work for us.
44	*
45	* See the .h file for how the ring is laid out. Note that while the
46	* command ring is defined, the particulars of the data area are
47	* not. Offset values in the command entry point to other locations
48	* internal to the mmap-ed area. There is separate space outside the
49	* command ring for data buffers. This leaves maximum flexibility for
50	* moving buffer allocations, or even page flipping or other
51	* allocation techniques, without altering the command ring layout.
52	*
53	* SECURITY:
54	* The user process must be assumed to be malicious. There's no way to
55	* prevent it breaking the command ring protocol if it wants, but in
56	* order to prevent other issues we must only ever read data from
57	* the shared memory area, not offsets or sizes. This applies to
58	* command ring entries as well as the mailbox. Extra code needed for
59	* this may have a 'UAM' comment.
60	*/
61
62	#define TCMU_TIME_OUT (30 * MSEC_PER_SEC)
63
64	/* For mailbox plus cmd ring, the size is fixed 8MB */
65	#define MB_CMDR_SIZE_DEF (8 * 1024 * 1024)
66	/* Offset of cmd ring is size of mailbox */
67	#define CMDR_OFF ((__u32)sizeof(struct tcmu_mailbox))
68	#define CMDR_SIZE_DEF (MB_CMDR_SIZE_DEF - CMDR_OFF)
69
70	/*
71	* For data area, the default block size is PAGE_SIZE and
72	* the default total size is 256K * PAGE_SIZE.
73	*/
74	#define DATA_PAGES_PER_BLK_DEF 1
75	#define DATA_AREA_PAGES_DEF (256 * 1024)
76
77	#define TCMU_MBS_TO_PAGES(_mbs) ((size_t)_mbs << (20 - PAGE_SHIFT))
78	#define TCMU_PAGES_TO_MBS(_pages) (_pages >> (20 - PAGE_SHIFT))
79
80	/*
81	* Default number of global data blocks(512K * PAGE_SIZE)
82	* when the unmap thread will be started.
83	*/
84	#define TCMU_GLOBAL_MAX_PAGES_DEF (512 * 1024)
85
86	static u8 tcmu_kern_cmd_reply_supported;
87	static u8 tcmu_netlink_blocked;
88
89	static struct device *tcmu_root_device;
90
91	struct tcmu_hba {
92	u32 host_id;
93	};
94
95	#define TCMU_CONFIG_LEN 256
96
97	static DEFINE_MUTEX(tcmu_nl_cmd_mutex);
98	static LIST_HEAD(tcmu_nl_cmd_list);
99
100	struct tcmu_dev;
101
102	struct tcmu_nl_cmd {
103	/* wake up thread waiting for reply */
104	struct completion complete;
105	struct list_head nl_list;
106	struct tcmu_dev *udev;
107	int cmd;
108	int status;
109	};
110
111	struct tcmu_dev {
112	struct list_head node;
113	struct kref kref;
114
115	struct se_device se_dev;
116	struct se_dev_plug se_plug;
117
118	char *name;
119	struct se_hba *hba;
120
121	#define TCMU_DEV_BIT_OPEN 0
122	#define TCMU_DEV_BIT_BROKEN 1
123	#define TCMU_DEV_BIT_BLOCKED 2
124	#define TCMU_DEV_BIT_TMR_NOTIFY 3
125	#define TCMU_DEV_BIT_PLUGGED 4
126	unsigned long flags;
127
128	struct uio_info uio_info;
129
130	struct inode *inode;
131
132	uint64_t dev_size;
133
134	struct tcmu_mailbox *mb_addr;
135	void *cmdr;
136	u32 cmdr_size;
137	u32 cmdr_last_cleaned;
138	/* Offset of data area from start of mb */
139	/* Must add data_off and mb_addr to get the address */
140	size_t data_off;
141	int data_area_mb;
142	uint32_t max_blocks;
143	size_t mmap_pages;
144
145	struct mutex cmdr_lock;
146	struct list_head qfull_queue;
147	struct list_head tmr_queue;
148
149	uint32_t dbi_max;
150	uint32_t dbi_thresh;
151	unsigned long *data_bitmap;
152	struct xarray data_pages;
153	uint32_t data_pages_per_blk;
154	uint32_t data_blk_size;
155
156	struct xarray commands;
157
158	struct timer_list cmd_timer;
159	unsigned int cmd_time_out;
160	struct list_head inflight_queue;
161
162	struct timer_list qfull_timer;
163	int qfull_time_out;
164
165	struct list_head timedout_entry;
166
167	struct tcmu_nl_cmd curr_nl_cmd;
168
169	char dev_config[TCMU_CONFIG_LEN];
170
171	int nl_reply_supported;
172	};
173
174	#define TCMU_DEV(_se_dev) container_of(_se_dev, struct tcmu_dev, se_dev)
175
176	struct tcmu_cmd {
177	struct se_cmd *se_cmd;
178	struct tcmu_dev *tcmu_dev;
179	struct list_head queue_entry;
180
181	uint16_t cmd_id;
182
183	/* Can't use se_cmd when cleaning up expired cmds, because if
184	cmd has been completed then accessing se_cmd is off limits */
185	uint32_t dbi_cnt;
186	uint32_t dbi_bidi_cnt;
187	uint32_t dbi_cur;
188	uint32_t *dbi;
189
190	uint32_t data_len_bidi;
191
192	unsigned long deadline;
193
194	#define TCMU_CMD_BIT_EXPIRED 0
195	#define TCMU_CMD_BIT_KEEP_BUF 1
196	unsigned long flags;
197	};
198
199	struct tcmu_tmr {
200	struct list_head queue_entry;
201
202	uint8_t tmr_type;
203	uint32_t tmr_cmd_cnt;
204	int16_t tmr_cmd_ids[] __counted_by(tmr_cmd_cnt);
205	};
206
207	/*
208	* To avoid dead lock the mutex lock order should always be:
209	*
210	* mutex_lock(&root_udev_mutex);
211	* ...
212	* mutex_lock(&tcmu_dev->cmdr_lock);
213	* mutex_unlock(&tcmu_dev->cmdr_lock);
214	* ...
215	* mutex_unlock(&root_udev_mutex);
216	*/
217	static DEFINE_MUTEX(root_udev_mutex);
218	static LIST_HEAD(root_udev);
219
220	static DEFINE_SPINLOCK(timed_out_udevs_lock);
221	static LIST_HEAD(timed_out_udevs);
222
223	static struct kmem_cache *tcmu_cmd_cache;
224
225	static atomic_t global_page_count = ATOMIC_INIT(0);
226	static struct delayed_work tcmu_unmap_work;
227	static int tcmu_global_max_pages = TCMU_GLOBAL_MAX_PAGES_DEF;
228
229	static int tcmu_set_global_max_data_area(const char *str,
230	const struct kernel_param *kp)
231	{
232	int ret, max_area_mb;
233
234	ret = kstrtoint(str, 10, &max_area_mb);
235	if (ret)
236	return -EINVAL;
237
238	if (max_area_mb <= 0) {
239	pr_err("global_max_data_area must be larger than 0.\n");
240	return -EINVAL;
241	}
242
243	tcmu_global_max_pages = TCMU_MBS_TO_PAGES(max_area_mb);
244	if (atomic_read(&global_page_count) > tcmu_global_max_pages)
245	schedule_delayed_work(&tcmu_unmap_work, 0);
246	else
247	cancel_delayed_work_sync(&tcmu_unmap_work);
248
249	return 0;
250	}
251
252	static int tcmu_get_global_max_data_area(char *buffer,
253	const struct kernel_param *kp)
254	{
255	return sprintf(buffer, "%d\n", TCMU_PAGES_TO_MBS(tcmu_global_max_pages));
256	}
257
258	static const struct kernel_param_ops tcmu_global_max_data_area_op = {
259	.set = tcmu_set_global_max_data_area,
260	.get = tcmu_get_global_max_data_area,
261	};
262
263	module_param_cb(global_max_data_area_mb, &tcmu_global_max_data_area_op, NULL,
264	S_IWUSR \| S_IRUGO);
265	MODULE_PARM_DESC(global_max_data_area_mb,
266	"Max MBs allowed to be allocated to all the tcmu device's "
267	"data areas.");
268
269	static int tcmu_get_block_netlink(char *buffer,
270	const struct kernel_param *kp)
271	{
272	return sprintf(buffer, "%s\n", tcmu_netlink_blocked ?
273	"blocked" : "unblocked");
274	}
275
276	static int tcmu_set_block_netlink(const char *str,
277	const struct kernel_param *kp)
278	{
279	int ret;
280	u8 val;
281
282	ret = kstrtou8(str, 0, &val);
283	if (ret < 0)
284	return ret;
285
286	if (val > 1) {
287	pr_err("Invalid block netlink value %u\n", val);
288	return -EINVAL;
289	}
290
291	tcmu_netlink_blocked = val;
292	return 0;
293	}
294
295	static const struct kernel_param_ops tcmu_block_netlink_op = {
296	.set = tcmu_set_block_netlink,
297	.get = tcmu_get_block_netlink,
298	};
299
300	module_param_cb(block_netlink, &tcmu_block_netlink_op, NULL, S_IWUSR \| S_IRUGO);
301	MODULE_PARM_DESC(block_netlink, "Block new netlink commands.");
302
303	static int tcmu_fail_netlink_cmd(struct tcmu_nl_cmd *nl_cmd)
304	{
305	struct tcmu_dev *udev = nl_cmd->udev;
306
307	if (!tcmu_netlink_blocked) {
308	pr_err("Could not reset device's netlink interface. Netlink is not blocked.\n");
309	return -EBUSY;
310	}
311
312	if (nl_cmd->cmd != TCMU_CMD_UNSPEC) {
313	pr_debug("Aborting nl cmd %d on %s\n", nl_cmd->cmd, udev->name);
314	nl_cmd->status = -EINTR;
315	list_del(&nl_cmd->nl_list);
316	complete(&nl_cmd->complete);
317	}
318	return 0;
319	}
320
321	static int tcmu_set_reset_netlink(const char *str,
322	const struct kernel_param *kp)
323	{
324	struct tcmu_nl_cmd nl_cmd, tmp_cmd;
325	int ret;
326	u8 val;
327
328	ret = kstrtou8(str, 0, &val);
329	if (ret < 0)
330	return ret;
331
332	if (val != 1) {
333	pr_err("Invalid reset netlink value %u\n", val);
334	return -EINVAL;
335	}
336
337	mutex_lock(&tcmu_nl_cmd_mutex);
338	list_for_each_entry_safe(nl_cmd, tmp_cmd, &tcmu_nl_cmd_list, nl_list) {
339	ret = tcmu_fail_netlink_cmd(nl_cmd);
340	if (ret)
341	break;
342	}
343	mutex_unlock(&tcmu_nl_cmd_mutex);
344
345	return ret;
346	}
347
348	static const struct kernel_param_ops tcmu_reset_netlink_op = {
349	.set = tcmu_set_reset_netlink,
350	};
351
352	module_param_cb(reset_netlink, &tcmu_reset_netlink_op, NULL, S_IWUSR);
353	MODULE_PARM_DESC(reset_netlink, "Reset netlink commands.");
354
355	/* multicast group */
356	enum tcmu_multicast_groups {
357	TCMU_MCGRP_CONFIG,
358	};
359
360	static const struct genl_multicast_group tcmu_mcgrps[] = {
361	[TCMU_MCGRP_CONFIG] = { .name = "config", },
362	};
363
364	static const struct nla_policy tcmu_attr_policy[TCMU_ATTR_MAX + 1] = {
365	[TCMU_ATTR_DEVICE] = { .type = NLA_STRING },
366	[TCMU_ATTR_MINOR] = { .type = NLA_U32 },
367	[TCMU_ATTR_CMD_STATUS] = { .type = NLA_S32 },
368	[TCMU_ATTR_DEVICE_ID] = { .type = NLA_U32 },
369	[TCMU_ATTR_SUPP_KERN_CMD_REPLY] = { .type = NLA_U8 },
370	};
371
372	static int tcmu_genl_cmd_done(struct genl_info *info, int completed_cmd)
373	{
374	struct tcmu_dev *udev = NULL;
375	struct tcmu_nl_cmd *nl_cmd;
376	int dev_id, rc, ret = 0;
377
378	if (!info->attrs[TCMU_ATTR_CMD_STATUS] \|\|
379	!info->attrs[TCMU_ATTR_DEVICE_ID]) {
380	printk(KERN_ERR "TCMU_ATTR_CMD_STATUS or TCMU_ATTR_DEVICE_ID not set, doing nothing\n");
381	return -EINVAL;
382	}
383
384	dev_id = nla_get_u32(info->attrs[TCMU_ATTR_DEVICE_ID]);
385	rc = nla_get_s32(info->attrs[TCMU_ATTR_CMD_STATUS]);
386
387	mutex_lock(&tcmu_nl_cmd_mutex);
388	list_for_each_entry(nl_cmd, &tcmu_nl_cmd_list, nl_list) {
389	if (nl_cmd->udev->se_dev.dev_index == dev_id) {
390	udev = nl_cmd->udev;
391	break;
392	}
393	}
394
395	if (!udev) {
396	pr_err("tcmu nl cmd %u/%d completion could not find device with dev id %u.\n",
397	completed_cmd, rc, dev_id);
398	ret = -ENODEV;
399	goto unlock;
400	}
401	list_del(&nl_cmd->nl_list);
402
403	pr_debug("%s genl cmd done got id %d curr %d done %d rc %d stat %d\n",
404	udev->name, dev_id, nl_cmd->cmd, completed_cmd, rc,
405	nl_cmd->status);
406
407	if (nl_cmd->cmd != completed_cmd) {
408	pr_err("Mismatched commands on %s (Expecting reply for %d. Current %d).\n",
409	udev->name, completed_cmd, nl_cmd->cmd);
410	ret = -EINVAL;
411	goto unlock;
412	}
413
414	nl_cmd->status = rc;
415	complete(&nl_cmd->complete);
416	unlock:
417	mutex_unlock(&tcmu_nl_cmd_mutex);
418	return ret;
419	}
420
421	static int tcmu_genl_rm_dev_done(struct sk_buff skb, struct genl_info info)
422	{
423	return tcmu_genl_cmd_done(info, TCMU_CMD_REMOVED_DEVICE);
424	}
425
426	static int tcmu_genl_add_dev_done(struct sk_buff skb, struct genl_info info)
427	{
428	return tcmu_genl_cmd_done(info, TCMU_CMD_ADDED_DEVICE);
429	}
430
431	static int tcmu_genl_reconfig_dev_done(struct sk_buff *skb,
432	struct genl_info *info)
433	{
434	return tcmu_genl_cmd_done(info, TCMU_CMD_RECONFIG_DEVICE);
435	}
436
437	static int tcmu_genl_set_features(struct sk_buff skb, struct genl_info info)
438	{
439	if (info->attrs[TCMU_ATTR_SUPP_KERN_CMD_REPLY]) {
440	tcmu_kern_cmd_reply_supported =
441	nla_get_u8(info->attrs[TCMU_ATTR_SUPP_KERN_CMD_REPLY]);
442	printk(KERN_INFO "tcmu daemon: command reply support %u.\n",
443	tcmu_kern_cmd_reply_supported);
444	}
445
446	return 0;
447	}
448
449	static const struct genl_small_ops tcmu_genl_ops[] = {
450	{
451	.cmd = TCMU_CMD_SET_FEATURES,
452	.validate = GENL_DONT_VALIDATE_STRICT \| GENL_DONT_VALIDATE_DUMP,
453	.flags = GENL_ADMIN_PERM,
454	.doit = tcmu_genl_set_features,
455	},
456	{
457	.cmd = TCMU_CMD_ADDED_DEVICE_DONE,
458	.validate = GENL_DONT_VALIDATE_STRICT \| GENL_DONT_VALIDATE_DUMP,
459	.flags = GENL_ADMIN_PERM,
460	.doit = tcmu_genl_add_dev_done,
461	},
462	{
463	.cmd = TCMU_CMD_REMOVED_DEVICE_DONE,
464	.validate = GENL_DONT_VALIDATE_STRICT \| GENL_DONT_VALIDATE_DUMP,
465	.flags = GENL_ADMIN_PERM,
466	.doit = tcmu_genl_rm_dev_done,
467	},
468	{
469	.cmd = TCMU_CMD_RECONFIG_DEVICE_DONE,
470	.validate = GENL_DONT_VALIDATE_STRICT \| GENL_DONT_VALIDATE_DUMP,
471	.flags = GENL_ADMIN_PERM,
472	.doit = tcmu_genl_reconfig_dev_done,
473	},
474	};
475
476	/* Our generic netlink family */
477	static struct genl_family tcmu_genl_family __ro_after_init = {
478	.module = THIS_MODULE,
479	.hdrsize = 0,
480	.name = "TCM-USER",
481	.version = 2,
482	.maxattr = TCMU_ATTR_MAX,
483	.policy = tcmu_attr_policy,
484	.mcgrps = tcmu_mcgrps,
485	.n_mcgrps = ARRAY_SIZE(tcmu_mcgrps),
486	.netnsok = true,
487	.small_ops = tcmu_genl_ops,
488	.n_small_ops = ARRAY_SIZE(tcmu_genl_ops),
489	.resv_start_op = TCMU_CMD_SET_FEATURES + 1,
490	};
491
492	#define tcmu_cmd_set_dbi_cur(cmd, index) ((cmd)->dbi_cur = (index))
493	#define tcmu_cmd_reset_dbi_cur(cmd) tcmu_cmd_set_dbi_cur(cmd, 0)
494	#define tcmu_cmd_set_dbi(cmd, index) ((cmd)->dbi[(cmd)->dbi_cur++] = (index))
495	#define tcmu_cmd_get_dbi(cmd) ((cmd)->dbi[(cmd)->dbi_cur++])
496
497	static void tcmu_cmd_free_data(struct tcmu_cmd *tcmu_cmd, uint32_t len)
498	{
499	struct tcmu_dev *udev = tcmu_cmd->tcmu_dev;
500	uint32_t i;
501
502	for (i = 0; i < len; i++)
503	clear_bit(tcmu_cmd->dbi[i], udev->data_bitmap);
504	}
505
506	static inline int tcmu_get_empty_block(struct tcmu_dev *udev,
507	struct tcmu_cmd *tcmu_cmd,
508	int prev_dbi, int length, int *iov_cnt)
509	{
510	XA_STATE(xas, &udev->data_pages, 0);
511	struct page *page;
512	int i, cnt, dbi, dpi;
513	int page_cnt = DIV_ROUND_UP(length, PAGE_SIZE);
514
515	dbi = find_first_zero_bit(udev->data_bitmap, udev->dbi_thresh);
516	if (dbi == udev->dbi_thresh)
517	return -1;
518
519	dpi = dbi * udev->data_pages_per_blk;
520	/* Count the number of already allocated pages */
521	xas_set(&xas, dpi);
522	rcu_read_lock();
523	for (cnt = 0; xas_next(&xas) && cnt < page_cnt;)
524	cnt++;
525	rcu_read_unlock();
526
527	for (i = cnt; i < page_cnt; i++) {
528	/* try to get new zeroed page from the mm */
529	page = alloc_page(GFP_NOIO \| __GFP_ZERO);
530	if (!page)
531	break;
532
533	if (xa_store(&udev->data_pages, dpi + i, page, GFP_NOIO)) {
534	__free_page(page);
535	break;
536	}
537	}
538	if (atomic_add_return(i - cnt, &global_page_count) >
539	tcmu_global_max_pages)
540	schedule_delayed_work(&tcmu_unmap_work, 0);
541
542	if (i && dbi > udev->dbi_max)
543	udev->dbi_max = dbi;
544
545	set_bit(dbi, udev->data_bitmap);
546	tcmu_cmd_set_dbi(tcmu_cmd, dbi);
547
548	if (dbi != prev_dbi + 1)
549	*iov_cnt += 1;
550
551	return i == page_cnt ? dbi : -1;
552	}
553
554	static int tcmu_get_empty_blocks(struct tcmu_dev *udev,
555	struct tcmu_cmd *tcmu_cmd, int length)
556	{
557	/* start value of dbi + 1 must not be a valid dbi */
558	int dbi = -2;
559	int blk_data_len, iov_cnt = 0;
560	uint32_t blk_size = udev->data_blk_size;
561
562	for (; length > 0; length -= blk_size) {
563	blk_data_len = min_t(uint32_t, length, blk_size);
564	dbi = tcmu_get_empty_block(udev, tcmu_cmd, dbi, blk_data_len,
565	&iov_cnt);
566	if (dbi < 0)
567	return -1;
568	}
569	return iov_cnt;
570	}
571
572	static inline void tcmu_free_cmd(struct tcmu_cmd *tcmu_cmd)
573	{
574	kfree(tcmu_cmd->dbi);
575	kmem_cache_free(tcmu_cmd_cache, tcmu_cmd);
576	}
577
578	static inline void tcmu_cmd_set_block_cnts(struct tcmu_cmd *cmd)
579	{
580	int i, len;
581	struct se_cmd *se_cmd = cmd->se_cmd;
582	uint32_t blk_size = cmd->tcmu_dev->data_blk_size;
583
584	cmd->dbi_cnt = DIV_ROUND_UP(se_cmd->data_length, blk_size);
585
586	if (se_cmd->se_cmd_flags & SCF_BIDI) {
587	BUG_ON(!(se_cmd->t_bidi_data_sg && se_cmd->t_bidi_data_nents));
588	for (i = 0, len = 0; i < se_cmd->t_bidi_data_nents; i++)
589	len += se_cmd->t_bidi_data_sg[i].length;
590	cmd->dbi_bidi_cnt = DIV_ROUND_UP(len, blk_size);
591	cmd->dbi_cnt += cmd->dbi_bidi_cnt;
592	cmd->data_len_bidi = len;
593	}
594	}
595
596	static int new_block_to_iov(struct tcmu_dev udev, struct tcmu_cmd cmd,
597	struct iovec **iov, int prev_dbi, int len)
598	{
599	/* Get the next dbi */
600	int dbi = tcmu_cmd_get_dbi(cmd);
601
602	/* Do not add more than udev->data_blk_size to iov */
603	len = min_t(int, len, udev->data_blk_size);
604
605	/*
606	* The following code will gather and map the blocks to the same iovec
607	* when the blocks are all next to each other.
608	*/
609	if (dbi != prev_dbi + 1) {
610	/* dbi is not next to previous dbi, so start new iov */
611	if (prev_dbi >= 0)
612	(*iov)++;
613	/* write offset relative to mb_addr */
614	(iov)->iov_base = (void __user )
615	(udev->data_off + dbi * udev->data_blk_size);
616	}
617	(*iov)->iov_len += len;
618
619	return dbi;
620	}
621
622	static void tcmu_setup_iovs(struct tcmu_dev udev, struct tcmu_cmd cmd,
623	struct iovec **iov, int data_length)
624	{
625	/* start value of dbi + 1 must not be a valid dbi */
626	int dbi = -2;
627
628	/* We prepare the IOVs for DMA_FROM_DEVICE transfer direction */
629	for (; data_length > 0; data_length -= udev->data_blk_size)
630	dbi = new_block_to_iov(udev, cmd, iov, dbi, data_length);
631	}
632
633	static struct tcmu_cmd tcmu_alloc_cmd(struct se_cmd se_cmd)
634	{
635	struct se_device *se_dev = se_cmd->se_dev;
636	struct tcmu_dev *udev = TCMU_DEV(se_dev);
637	struct tcmu_cmd *tcmu_cmd;
638
639	tcmu_cmd = kmem_cache_zalloc(tcmu_cmd_cache, GFP_NOIO);
640	if (!tcmu_cmd)
641	return NULL;
642
643	INIT_LIST_HEAD(&tcmu_cmd->queue_entry);
644	tcmu_cmd->se_cmd = se_cmd;
645	tcmu_cmd->tcmu_dev = udev;
646
647	tcmu_cmd_set_block_cnts(tcmu_cmd);
648	tcmu_cmd->dbi = kcalloc(tcmu_cmd->dbi_cnt, sizeof(uint32_t),
649	GFP_NOIO);
650	if (!tcmu_cmd->dbi) {
651	kmem_cache_free(tcmu_cmd_cache, tcmu_cmd);
652	return NULL;
653	}
654
655	return tcmu_cmd;
656	}
657
658	static inline void tcmu_flush_dcache_range(void *vaddr, size_t size)
659	{
660	unsigned long offset = offset_in_page(vaddr);
661	void *start = vaddr - offset;
662
663	size = round_up(size+offset, PAGE_SIZE);
664
665	while (size) {
666	flush_dcache_page(vmalloc_to_page(start));
667	start += PAGE_SIZE;
668	size -= PAGE_SIZE;
669	}
670	}
671
672	/*
673	* Some ring helper functions. We don't assume size is a power of 2 so
674	* we can't use circ_buf.h.
675	*/
676	static inline size_t spc_used(size_t head, size_t tail, size_t size)
677	{
678	int diff = head - tail;
679
680	if (diff >= 0)
681	return diff;
682	else
683	return size + diff;
684	}
685
686	static inline size_t spc_free(size_t head, size_t tail, size_t size)
687	{
688	/* Keep 1 byte unused or we can't tell full from empty */
689	return (size - spc_used(head, tail, size) - 1);
690	}
691
692	static inline size_t head_to_end(size_t head, size_t size)
693	{
694	return size - head;
695	}
696
697	#define UPDATE_HEAD(head, used, size) smp_store_release(&head, ((head % size) + used) % size)
698
699	#define TCMU_SG_TO_DATA_AREA 1
700	#define TCMU_DATA_AREA_TO_SG 2
701
702	static inline void tcmu_copy_data(struct tcmu_dev *udev,
703	struct tcmu_cmd *tcmu_cmd, uint32_t direction,
704	struct scatterlist *sg, unsigned int sg_nents,
705	struct iovec **iov, size_t data_len)
706	{
707	/* start value of dbi + 1 must not be a valid dbi */
708	int dbi = -2;
709	size_t page_remaining, cp_len;
710	int page_cnt, page_inx, dpi;
711	struct sg_mapping_iter sg_iter;
712	unsigned int sg_flags;
713	struct page *page;
714	void data_page_start, data_addr;
715
716	if (direction == TCMU_SG_TO_DATA_AREA)
717	sg_flags = SG_MITER_ATOMIC \| SG_MITER_FROM_SG;
718	else
719	sg_flags = SG_MITER_ATOMIC \| SG_MITER_TO_SG;
720	sg_miter_start(&sg_iter, sg, sg_nents, sg_flags);
721
722	while (data_len) {
723	if (direction == TCMU_SG_TO_DATA_AREA)
724	dbi = new_block_to_iov(udev, tcmu_cmd, iov, dbi,
725	data_len);
726	else
727	dbi = tcmu_cmd_get_dbi(tcmu_cmd);
728
729	page_cnt = DIV_ROUND_UP(data_len, PAGE_SIZE);
730	if (page_cnt > udev->data_pages_per_blk)
731	page_cnt = udev->data_pages_per_blk;
732
733	dpi = dbi * udev->data_pages_per_blk;
734	for (page_inx = 0; page_inx < page_cnt && data_len;
735	page_inx++, dpi++) {
736	page = xa_load(&udev->data_pages, dpi);
737
738	if (direction == TCMU_DATA_AREA_TO_SG)
739	flush_dcache_page(page);
740	data_page_start = kmap_atomic(page);
741	page_remaining = PAGE_SIZE;
742
743	while (page_remaining && data_len) {
744	if (!sg_miter_next(&sg_iter)) {
745	/* set length to 0 to abort outer loop */
746	data_len = 0;
747	pr_debug("%s: aborting data copy due to exhausted sg_list\n",
748	__func__);
749	break;
750	}
751	cp_len = min3(sg_iter.length, page_remaining,
752	data_len);
753
754	data_addr = data_page_start +
755	PAGE_SIZE - page_remaining;
756	if (direction == TCMU_SG_TO_DATA_AREA)
757	memcpy(data_addr, sg_iter.addr, cp_len);
758	else
759	memcpy(sg_iter.addr, data_addr, cp_len);
760
761	data_len -= cp_len;
762	page_remaining -= cp_len;
763	sg_iter.consumed = cp_len;
764	}
765	sg_miter_stop(&sg_iter);
766
767	kunmap_atomic(data_page_start);
768	if (direction == TCMU_SG_TO_DATA_AREA)
769	flush_dcache_page(page);
770	}
771	}
772	}
773
774	static void scatter_data_area(struct tcmu_dev udev, struct tcmu_cmd tcmu_cmd,
775	struct iovec **iov)
776	{
777	struct se_cmd *se_cmd = tcmu_cmd->se_cmd;
778
779	tcmu_copy_data(udev, tcmu_cmd, TCMU_SG_TO_DATA_AREA, se_cmd->t_data_sg,
780	se_cmd->t_data_nents, iov, se_cmd->data_length);
781	}
782
783	static void gather_data_area(struct tcmu_dev udev, struct tcmu_cmd tcmu_cmd,
784	bool bidi, uint32_t read_len)
785	{
786	struct se_cmd *se_cmd = tcmu_cmd->se_cmd;
787	struct scatterlist *data_sg;
788	unsigned int data_nents;
789
790	if (!bidi) {
791	data_sg = se_cmd->t_data_sg;
792	data_nents = se_cmd->t_data_nents;
793	} else {
794	/*
795	* For bidi case, the first count blocks are for Data-Out
796	* buffer blocks, and before gathering the Data-In buffer
797	* the Data-Out buffer blocks should be skipped.
798	*/
799	tcmu_cmd_set_dbi_cur(tcmu_cmd,
800	tcmu_cmd->dbi_cnt - tcmu_cmd->dbi_bidi_cnt);
801
802	data_sg = se_cmd->t_bidi_data_sg;
803	data_nents = se_cmd->t_bidi_data_nents;
804	}
805
806	tcmu_copy_data(udev, tcmu_cmd, TCMU_DATA_AREA_TO_SG, data_sg,
807	data_nents, NULL, read_len);
808	}
809
810	static inline size_t spc_bitmap_free(unsigned long *bitmap, uint32_t thresh)
811	{
812	return thresh - bitmap_weight(bitmap, thresh);
813	}
814
815	/*
816	* We can't queue a command until we have space available on the cmd ring.
817	*
818	* Called with ring lock held.
819	*/
820	static bool is_ring_space_avail(struct tcmu_dev *udev, size_t cmd_size)
821	{
822	struct tcmu_mailbox *mb = udev->mb_addr;
823	size_t space, cmd_needed;
824	u32 cmd_head;
825
826	tcmu_flush_dcache_range(mb, sizeof(*mb));
827
828	cmd_head = mb->cmd_head % udev->cmdr_size; /* UAM */
829
830	/*
831	* If cmd end-of-ring space is too small then we need space for a NOP plus
832	* original cmd - cmds are internally contiguous.
833	*/
834	if (head_to_end(cmd_head, udev->cmdr_size) >= cmd_size)
835	cmd_needed = cmd_size;
836	else
837	cmd_needed = cmd_size + head_to_end(cmd_head, udev->cmdr_size);
838
839	space = spc_free(cmd_head, udev->cmdr_last_cleaned, udev->cmdr_size);
840	if (space < cmd_needed) {
841	pr_debug("no cmd space: %u %u %u\n", cmd_head,
842	udev->cmdr_last_cleaned, udev->cmdr_size);
843	return false;
844	}
845	return true;
846	}
847
848	/*
849	* We have to allocate data buffers before we can queue a command.
850	* Returns -1 on error (not enough space) or number of needed iovs on success
851	*
852	* Called with ring lock held.
853	*/
854	static int tcmu_alloc_data_space(struct tcmu_dev udev, struct tcmu_cmd cmd,
855	int *iov_bidi_cnt)
856	{
857	int space, iov_cnt = 0, ret = 0;
858
859	if (!cmd->dbi_cnt)
860	goto wr_iov_cnts;
861
862	/* try to check and get the data blocks as needed */
863	space = spc_bitmap_free(udev->data_bitmap, udev->dbi_thresh);
864	if (space < cmd->dbi_cnt) {
865	unsigned long blocks_left =
866	(udev->max_blocks - udev->dbi_thresh) + space;
867
868	if (blocks_left < cmd->dbi_cnt) {
869	pr_debug("no data space: only %lu available, but ask for %u\n",
870	blocks_left * udev->data_blk_size,
871	cmd->dbi_cnt * udev->data_blk_size);
872	return -1;
873	}
874
875	udev->dbi_thresh += cmd->dbi_cnt;
876	if (udev->dbi_thresh > udev->max_blocks)
877	udev->dbi_thresh = udev->max_blocks;
878	}
879
880	iov_cnt = tcmu_get_empty_blocks(udev, cmd, cmd->se_cmd->data_length);
881	if (iov_cnt < 0)
882	return -1;
883
884	if (cmd->dbi_bidi_cnt) {
885	ret = tcmu_get_empty_blocks(udev, cmd, cmd->data_len_bidi);
886	if (ret < 0)
887	return -1;
888	}
889	wr_iov_cnts:
890	*iov_bidi_cnt = ret;
891	return iov_cnt + ret;
892	}
893
894	static inline size_t tcmu_cmd_get_base_cmd_size(size_t iov_cnt)
895	{
896	return max(offsetof(struct tcmu_cmd_entry, req.iov[iov_cnt]),
897	sizeof(struct tcmu_cmd_entry));
898	}
899
900	static inline size_t tcmu_cmd_get_cmd_size(struct tcmu_cmd *tcmu_cmd,
901	size_t base_command_size)
902	{
903	struct se_cmd *se_cmd = tcmu_cmd->se_cmd;
904	size_t command_size;
905
906	command_size = base_command_size +
907	round_up(scsi_command_size(se_cmd->t_task_cdb),
908	TCMU_OP_ALIGN_SIZE);
909
910	WARN_ON(command_size & (TCMU_OP_ALIGN_SIZE-1));
911
912	return command_size;
913	}
914
915	static void tcmu_setup_cmd_timer(struct tcmu_cmd *tcmu_cmd, unsigned int tmo,
916	struct timer_list *timer)
917	{
918	if (!tmo)
919	return;
920
921	tcmu_cmd->deadline = round_jiffies_up(jiffies + msecs_to_jiffies(tmo));
922	if (!timer_pending(timer))
923	mod_timer(timer, tcmu_cmd->deadline);
924
925	pr_debug("Timeout set up for cmd %p, dev = %s, tmo = %lu\n", tcmu_cmd,
926	tcmu_cmd->tcmu_dev->name, tmo / MSEC_PER_SEC);
927	}
928
929	static int add_to_qfull_queue(struct tcmu_cmd *tcmu_cmd)
930	{
931	struct tcmu_dev *udev = tcmu_cmd->tcmu_dev;
932	unsigned int tmo;
933
934	/*
935	* For backwards compat if qfull_time_out is not set use
936	* cmd_time_out and if that's not set use the default time out.
937	*/
938	if (!udev->qfull_time_out)
939	return -ETIMEDOUT;
940	else if (udev->qfull_time_out > 0)
941	tmo = udev->qfull_time_out;
942	else if (udev->cmd_time_out)
943	tmo = udev->cmd_time_out;
944	else
945	tmo = TCMU_TIME_OUT;
946
947	tcmu_setup_cmd_timer(tcmu_cmd, tmo, &udev->qfull_timer);
948
949	list_add_tail(&tcmu_cmd->queue_entry, &udev->qfull_queue);
950	pr_debug("adding cmd %p on dev %s to ring space wait queue\n",
951	tcmu_cmd, udev->name);
952	return 0;
953	}
954
955	static uint32_t ring_insert_padding(struct tcmu_dev *udev, size_t cmd_size)
956	{
957	struct tcmu_cmd_entry_hdr *hdr;
958	struct tcmu_mailbox *mb = udev->mb_addr;
959	uint32_t cmd_head = mb->cmd_head % udev->cmdr_size; /* UAM */
960
961	/* Insert a PAD if end-of-ring space is too small */
962	if (head_to_end(cmd_head, udev->cmdr_size) < cmd_size) {
963	size_t pad_size = head_to_end(cmd_head, udev->cmdr_size);
964
965	hdr = udev->cmdr + cmd_head;
966	tcmu_hdr_set_op(&hdr->len_op, TCMU_OP_PAD);
967	tcmu_hdr_set_len(&hdr->len_op, pad_size);
968	hdr->cmd_id = 0; /* not used for PAD */
969	hdr->kflags = 0;
970	hdr->uflags = 0;
971	tcmu_flush_dcache_range(hdr, sizeof(*hdr));
972
973	UPDATE_HEAD(mb->cmd_head, pad_size, udev->cmdr_size);
974	tcmu_flush_dcache_range(mb, sizeof(*mb));
975
976	cmd_head = mb->cmd_head % udev->cmdr_size; /* UAM */
977	WARN_ON(cmd_head != 0);
978	}
979
980	return cmd_head;
981	}
982
983	static void tcmu_unplug_device(struct se_dev_plug *se_plug)
984	{
985	struct se_device *se_dev = se_plug->se_dev;
986	struct tcmu_dev *udev = TCMU_DEV(se_dev);
987
988	clear_bit(TCMU_DEV_BIT_PLUGGED, &udev->flags);
989	uio_event_notify(&udev->uio_info);
990	}
991
992	static struct se_dev_plug tcmu_plug_device(struct se_device se_dev)
993	{
994	struct tcmu_dev *udev = TCMU_DEV(se_dev);
995
996	if (!test_and_set_bit(TCMU_DEV_BIT_PLUGGED, &udev->flags))
997	return &udev->se_plug;
998
999	return NULL;
1000	}
1001
1002	/**
1003	* queue_cmd_ring - queue cmd to ring or internally
1004	* @tcmu_cmd: cmd to queue
1005	* @scsi_err: TCM error code if failure (-1) returned.
1006	*
1007	* Returns:
1008	* -1 we cannot queue internally or to the ring.
1009	* 0 success
1010	* 1 internally queued to wait for ring memory to free.
1011	*/
1012	static int queue_cmd_ring(struct tcmu_cmd tcmu_cmd, sense_reason_t scsi_err)
1013	{
1014	struct tcmu_dev *udev = tcmu_cmd->tcmu_dev;
1015	struct se_cmd *se_cmd = tcmu_cmd->se_cmd;
1016	size_t base_command_size, command_size;
1017	struct tcmu_mailbox *mb = udev->mb_addr;
1018	struct tcmu_cmd_entry *entry;
1019	struct iovec *iov;
1020	int iov_cnt, iov_bidi_cnt;
1021	uint32_t cmd_id, cmd_head;
1022	uint64_t cdb_off;
1023	uint32_t blk_size = udev->data_blk_size;
1024	/* size of data buffer needed */
1025	size_t data_length = (size_t)tcmu_cmd->dbi_cnt * blk_size;
1026
1027	*scsi_err = TCM_NO_SENSE;
1028
1029	if (test_bit(TCMU_DEV_BIT_BLOCKED, &udev->flags)) {
1030	*scsi_err = TCM_LUN_BUSY;
1031	return -1;
1032	}
1033
1034	if (test_bit(TCMU_DEV_BIT_BROKEN, &udev->flags)) {
1035	*scsi_err = TCM_LOGICAL_UNIT_COMMUNICATION_FAILURE;
1036	return -1;
1037	}
1038
1039	if (!list_empty(&udev->qfull_queue))
1040	goto queue;
1041
1042	if (data_length > (size_t)udev->max_blocks * blk_size) {
1043	pr_warn("TCMU: Request of size %zu is too big for %zu data area\n",
1044	data_length, (size_t)udev->max_blocks * blk_size);
1045	*scsi_err = TCM_INVALID_CDB_FIELD;
1046	return -1;
1047	}
1048
1049	iov_cnt = tcmu_alloc_data_space(udev, tcmu_cmd, &iov_bidi_cnt);
1050	if (iov_cnt < 0)
1051	goto free_and_queue;
1052
1053	/*
1054	* Must be a certain minimum size for response sense info, but
1055	* also may be larger if the iov array is large.
1056	*/
1057	base_command_size = tcmu_cmd_get_base_cmd_size(iov_cnt);
1058	command_size = tcmu_cmd_get_cmd_size(tcmu_cmd, base_command_size);
1059
1060	if (command_size > (udev->cmdr_size / 2)) {
1061	pr_warn("TCMU: Request of size %zu is too big for %u cmd ring\n",
1062	command_size, udev->cmdr_size);
1063	tcmu_cmd_free_data(tcmu_cmd, tcmu_cmd->dbi_cur);
1064	*scsi_err = TCM_INVALID_CDB_FIELD;
1065	return -1;
1066	}
1067
1068	if (!is_ring_space_avail(udev, command_size))
1069	/*
1070	* Don't leave commands partially setup because the unmap
1071	* thread might need the blocks to make forward progress.
1072	*/
1073	goto free_and_queue;
1074
1075	if (xa_alloc(&udev->commands, &cmd_id, tcmu_cmd, XA_LIMIT(1, 0xffff),
1076	GFP_NOWAIT) < 0) {
1077	pr_err("tcmu: Could not allocate cmd id.\n");
1078
1079	tcmu_cmd_free_data(tcmu_cmd, tcmu_cmd->dbi_cnt);
1080	*scsi_err = TCM_OUT_OF_RESOURCES;
1081	return -1;
1082	}
1083	tcmu_cmd->cmd_id = cmd_id;
1084
1085	pr_debug("allocated cmd id %u for cmd %p dev %s\n", tcmu_cmd->cmd_id,
1086	tcmu_cmd, udev->name);
1087
1088	cmd_head = ring_insert_padding(udev, command_size);
1089
1090	entry = udev->cmdr + cmd_head;
1091	memset(entry, 0, command_size);
1092	tcmu_hdr_set_op(&entry->hdr.len_op, TCMU_OP_CMD);
1093
1094	/* prepare iov list and copy data to data area if necessary */
1095	tcmu_cmd_reset_dbi_cur(tcmu_cmd);
1096	iov = &entry->req.iov[0];
1097
1098	if (se_cmd->data_direction == DMA_TO_DEVICE \|\|
1099	se_cmd->se_cmd_flags & SCF_BIDI)
1100	scatter_data_area(udev, tcmu_cmd, &iov);
1101	else
1102	tcmu_setup_iovs(udev, tcmu_cmd, &iov, se_cmd->data_length);
1103
1104	entry->req.iov_cnt = iov_cnt - iov_bidi_cnt;
1105
1106	/* Handle BIDI commands */
1107	if (se_cmd->se_cmd_flags & SCF_BIDI) {
1108	iov++;
1109	tcmu_setup_iovs(udev, tcmu_cmd, &iov, tcmu_cmd->data_len_bidi);
1110	entry->req.iov_bidi_cnt = iov_bidi_cnt;
1111	}
1112
1113	tcmu_setup_cmd_timer(tcmu_cmd, udev->cmd_time_out, &udev->cmd_timer);
1114
1115	entry->hdr.cmd_id = tcmu_cmd->cmd_id;
1116
1117	tcmu_hdr_set_len(&entry->hdr.len_op, command_size);
1118
1119	/* All offsets relative to mb_addr, not start of entry! */
1120	cdb_off = CMDR_OFF + cmd_head + base_command_size;
1121	memcpy((void *) mb + cdb_off, se_cmd->t_task_cdb, scsi_command_size(se_cmd->t_task_cdb));
1122	entry->req.cdb_off = cdb_off;
1123	tcmu_flush_dcache_range(entry, command_size);
1124
1125	UPDATE_HEAD(mb->cmd_head, command_size, udev->cmdr_size);
1126	tcmu_flush_dcache_range(mb, sizeof(*mb));
1127
1128	list_add_tail(&tcmu_cmd->queue_entry, &udev->inflight_queue);
1129
1130	if (!test_bit(TCMU_DEV_BIT_PLUGGED, &udev->flags))
1131	uio_event_notify(&udev->uio_info);
1132
1133	return 0;
1134
1135	free_and_queue:
1136	tcmu_cmd_free_data(tcmu_cmd, tcmu_cmd->dbi_cur);
1137	tcmu_cmd_reset_dbi_cur(tcmu_cmd);
1138
1139	queue:
1140	if (add_to_qfull_queue(tcmu_cmd)) {
1141	*scsi_err = TCM_OUT_OF_RESOURCES;
1142	return -1;
1143	}
1144
1145	return 1;
1146	}
1147
1148	/**
1149	* queue_tmr_ring - queue tmr info to ring or internally
1150	* @udev: related tcmu_dev
1151	* @tmr: tcmu_tmr containing tmr info to queue
1152	*
1153	* Returns:
1154	* 0 success
1155	* 1 internally queued to wait for ring memory to free.
1156	*/
1157	static int
1158	queue_tmr_ring(struct tcmu_dev udev, struct tcmu_tmr tmr)
1159	{
1160	struct tcmu_tmr_entry *entry;
1161	int cmd_size;
1162	int id_list_sz;
1163	struct tcmu_mailbox *mb = udev->mb_addr;
1164	uint32_t cmd_head;
1165
1166	if (test_bit(TCMU_DEV_BIT_BROKEN, &udev->flags))
1167	goto out_free;
1168
1169	id_list_sz = sizeof(tmr->tmr_cmd_ids[0]) * tmr->tmr_cmd_cnt;
1170	cmd_size = round_up(sizeof(*entry) + id_list_sz, TCMU_OP_ALIGN_SIZE);
1171
1172	if (!list_empty(&udev->tmr_queue) \|\|
1173	!is_ring_space_avail(udev, cmd_size)) {
1174	list_add_tail(&tmr->queue_entry, &udev->tmr_queue);
1175	pr_debug("adding tmr %p on dev %s to TMR ring space wait queue\n",
1176	tmr, udev->name);
1177	return 1;
1178	}
1179
1180	cmd_head = ring_insert_padding(udev, cmd_size);
1181
1182	entry = udev->cmdr + cmd_head;
1183	memset(entry, 0, cmd_size);
1184	tcmu_hdr_set_op(&entry->hdr.len_op, TCMU_OP_TMR);
1185	tcmu_hdr_set_len(&entry->hdr.len_op, cmd_size);
1186	entry->tmr_type = tmr->tmr_type;
1187	entry->cmd_cnt = tmr->tmr_cmd_cnt;
1188	memcpy(&entry->cmd_ids[0], &tmr->tmr_cmd_ids[0], id_list_sz);
1189	tcmu_flush_dcache_range(entry, cmd_size);
1190
1191	UPDATE_HEAD(mb->cmd_head, cmd_size, udev->cmdr_size);
1192	tcmu_flush_dcache_range(mb, sizeof(*mb));
1193
1194	uio_event_notify(&udev->uio_info);
1195
1196	out_free:
1197	kfree(tmr);
1198
1199	return 0;
1200	}
1201
1202	static sense_reason_t
1203	tcmu_queue_cmd(struct se_cmd *se_cmd)
1204	{
1205	struct se_device *se_dev = se_cmd->se_dev;
1206	struct tcmu_dev *udev = TCMU_DEV(se_dev);
1207	struct tcmu_cmd *tcmu_cmd;
1208	sense_reason_t scsi_ret = TCM_CHECK_CONDITION_ABORT_CMD;
1209	int ret = -1;
1210
1211	tcmu_cmd = tcmu_alloc_cmd(se_cmd);
1212	if (!tcmu_cmd)
1213	return TCM_LOGICAL_UNIT_COMMUNICATION_FAILURE;
1214
1215	mutex_lock(&udev->cmdr_lock);
1216	if (!(se_cmd->transport_state & CMD_T_ABORTED))
1217	ret = queue_cmd_ring(tcmu_cmd, &scsi_ret);
1218	if (ret < 0)
1219	tcmu_free_cmd(tcmu_cmd);
1220	else
1221	se_cmd->priv = tcmu_cmd;
1222	mutex_unlock(&udev->cmdr_lock);
1223	return scsi_ret;
1224	}
1225
1226	static void tcmu_set_next_deadline(struct list_head *queue,
1227	struct timer_list *timer)
1228	{
1229	struct tcmu_cmd *cmd;
1230
1231	if (!list_empty(queue)) {
1232	cmd = list_first_entry(queue, struct tcmu_cmd, queue_entry);
1233	mod_timer(timer, cmd->deadline);
1234	} else
1235	timer_delete(timer);
1236	}
1237
1238	static int
1239	tcmu_tmr_type(enum tcm_tmreq_table tmf)
1240	{
1241	switch (tmf) {
1242	case TMR_ABORT_TASK: return TCMU_TMR_ABORT_TASK;
1243	case TMR_ABORT_TASK_SET: return TCMU_TMR_ABORT_TASK_SET;
1244	case TMR_CLEAR_ACA: return TCMU_TMR_CLEAR_ACA;
1245	case TMR_CLEAR_TASK_SET: return TCMU_TMR_CLEAR_TASK_SET;
1246	case TMR_LUN_RESET: return TCMU_TMR_LUN_RESET;
1247	case TMR_TARGET_WARM_RESET: return TCMU_TMR_TARGET_WARM_RESET;
1248	case TMR_TARGET_COLD_RESET: return TCMU_TMR_TARGET_COLD_RESET;
1249	case TMR_LUN_RESET_PRO: return TCMU_TMR_LUN_RESET_PRO;
1250	default: return TCMU_TMR_UNKNOWN;
1251	}
1252	}
1253
1254	static void
1255	tcmu_tmr_notify(struct se_device *se_dev, enum tcm_tmreq_table tmf,
1256	struct list_head *cmd_list)
1257	{
1258	int i = 0, cmd_cnt = 0;
1259	bool unqueued = false;
1260	struct tcmu_cmd *cmd;
1261	struct se_cmd *se_cmd;
1262	struct tcmu_tmr *tmr;
1263	struct tcmu_dev *udev = TCMU_DEV(se_dev);
1264
1265	mutex_lock(&udev->cmdr_lock);
1266
1267	/* First we check for aborted commands in qfull_queue */
1268	list_for_each_entry(se_cmd, cmd_list, state_list) {
1269	i++;
1270	if (!se_cmd->priv)
1271	continue;
1272	cmd = se_cmd->priv;
1273	/* Commands on qfull queue have no id yet */
1274	if (cmd->cmd_id) {
1275	cmd_cnt++;
1276	continue;
1277	}
1278	pr_debug("Removing aborted command %p from queue on dev %s.\n",
1279	cmd, udev->name);
1280
1281	list_del_init(&cmd->queue_entry);
1282	tcmu_free_cmd(cmd);
1283	se_cmd->priv = NULL;
1284	target_complete_cmd(se_cmd, SAM_STAT_TASK_ABORTED);
1285	unqueued = true;
1286	}
1287	if (unqueued)
1288	tcmu_set_next_deadline(&udev->qfull_queue, &udev->qfull_timer);
1289
1290	if (!test_bit(TCMU_DEV_BIT_TMR_NOTIFY, &udev->flags))
1291	goto unlock;
1292
1293	pr_debug("TMR event %d on dev %s, aborted cmds %d, afflicted cmd_ids %d\n",
1294	tcmu_tmr_type(tmf), udev->name, i, cmd_cnt);
1295
1296	tmr = kmalloc_flex(*tmr, tmr_cmd_ids, cmd_cnt, GFP_NOIO);
1297	if (!tmr)
1298	goto unlock;
1299
1300	tmr->tmr_type = tcmu_tmr_type(tmf);
1301	tmr->tmr_cmd_cnt = cmd_cnt;
1302
1303	if (cmd_cnt != 0) {
1304	cmd_cnt = 0;
1305	list_for_each_entry(se_cmd, cmd_list, state_list) {
1306	if (!se_cmd->priv)
1307	continue;
1308	cmd = se_cmd->priv;
1309	if (cmd->cmd_id)
1310	tmr->tmr_cmd_ids[cmd_cnt++] = cmd->cmd_id;
1311	}
1312	}
1313
1314	queue_tmr_ring(udev, tmr);
1315
1316	unlock:
1317	mutex_unlock(&udev->cmdr_lock);
1318	}
1319
1320	static bool tcmu_handle_completion(struct tcmu_cmd *cmd,
1321	struct tcmu_cmd_entry *entry, bool keep_buf)
1322	{
1323	struct se_cmd *se_cmd = cmd->se_cmd;
1324	struct tcmu_dev *udev = cmd->tcmu_dev;
1325	bool read_len_valid = false;
1326	bool ret = true;
1327	uint32_t read_len;
1328
1329	/*
1330	* cmd has been completed already from timeout, just reclaim
1331	* data area space and free cmd
1332	*/
1333	if (test_bit(TCMU_CMD_BIT_EXPIRED, &cmd->flags)) {
1334	WARN_ON_ONCE(se_cmd);
1335	goto out;
1336	}
1337	if (test_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags)) {
1338	pr_err("cmd_id %u already completed with KEEP_BUF, ring is broken\n",
1339	entry->hdr.cmd_id);
1340	set_bit(TCMU_DEV_BIT_BROKEN, &udev->flags);
1341	ret = false;
1342	goto out;
1343	}
1344
1345	list_del_init(&cmd->queue_entry);
1346
1347	tcmu_cmd_reset_dbi_cur(cmd);
1348
1349	if (entry->hdr.uflags & TCMU_UFLAG_UNKNOWN_OP) {
1350	pr_warn("TCMU: Userspace set UNKNOWN_OP flag on se_cmd %p\n",
1351	cmd->se_cmd);
1352	entry->rsp.scsi_status = SAM_STAT_CHECK_CONDITION;
1353	goto done;
1354	}
1355
1356	read_len = se_cmd->data_length;
1357	if (se_cmd->data_direction == DMA_FROM_DEVICE &&
1358	(entry->hdr.uflags & TCMU_UFLAG_READ_LEN) && entry->rsp.read_len) {
1359	read_len_valid = true;
1360	if (entry->rsp.read_len < read_len)
1361	read_len = entry->rsp.read_len;
1362	}
1363
1364	if (entry->rsp.scsi_status == SAM_STAT_CHECK_CONDITION) {
1365	transport_copy_sense_to_cmd(se_cmd, entry->rsp.sense_buffer);
1366	if (!read_len_valid )
1367	goto done;
1368	else
1369	se_cmd->se_cmd_flags \|= SCF_TREAT_READ_AS_NORMAL;
1370	}
1371	if (se_cmd->se_cmd_flags & SCF_BIDI) {
1372	/* Get Data-In buffer before clean up */
1373	gather_data_area(udev, cmd, true, read_len);
1374	} else if (se_cmd->data_direction == DMA_FROM_DEVICE) {
1375	gather_data_area(udev, cmd, false, read_len);
1376	} else if (se_cmd->data_direction == DMA_TO_DEVICE) {
1377	/* TODO: */
1378	} else if (se_cmd->data_direction != DMA_NONE) {
1379	pr_warn("TCMU: data direction was %d!\n",
1380	se_cmd->data_direction);
1381	}
1382
1383	done:
1384	se_cmd->priv = NULL;
1385	if (read_len_valid) {
1386	pr_debug("read_len = %d\n", read_len);
1387	target_complete_cmd_with_length(cmd->se_cmd,
1388	entry->rsp.scsi_status, read_len);
1389	} else
1390	target_complete_cmd(cmd->se_cmd, entry->rsp.scsi_status);
1391
1392	out:
1393	if (!keep_buf) {
1394	tcmu_cmd_free_data(cmd, cmd->dbi_cnt);
1395	tcmu_free_cmd(cmd);
1396	} else {
1397	/*
1398	* Keep this command after completion, since userspace still
1399	* needs the data buffer. Mark it with TCMU_CMD_BIT_KEEP_BUF
1400	* and reset potential TCMU_CMD_BIT_EXPIRED, so we don't accept
1401	* a second completion later.
1402	* Userspace can free the buffer later by writing the cmd_id
1403	* to new action attribute free_kept_buf.
1404	*/
1405	clear_bit(TCMU_CMD_BIT_EXPIRED, &cmd->flags);
1406	set_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags);
1407	}
1408	return ret;
1409	}
1410
1411	static int tcmu_run_tmr_queue(struct tcmu_dev *udev)
1412	{
1413	struct tcmu_tmr tmr, tmp;
1414	LIST_HEAD(tmrs);
1415
1416	if (list_empty(&udev->tmr_queue))
1417	return 1;
1418
1419	pr_debug("running %s's tmr queue\n", udev->name);
1420
1421	list_splice_init(&udev->tmr_queue, &tmrs);
1422
1423	list_for_each_entry_safe(tmr, tmp, &tmrs, queue_entry) {
1424	list_del_init(&tmr->queue_entry);
1425
1426	pr_debug("removing tmr %p on dev %s from queue\n",
1427	tmr, udev->name);
1428
1429	if (queue_tmr_ring(udev, tmr)) {
1430	pr_debug("ran out of space during tmr queue run\n");
1431	/*
1432	* tmr was requeued, so just put all tmrs back in
1433	* the queue
1434	*/
1435	list_splice_tail(&tmrs, &udev->tmr_queue);
1436	return 0;
1437	}
1438	}
1439
1440	return 1;
1441	}
1442
1443	static bool tcmu_handle_completions(struct tcmu_dev *udev)
1444	{
1445	struct tcmu_mailbox *mb;
1446	struct tcmu_cmd *cmd;
1447	bool free_space = false;
1448
1449	if (test_bit(TCMU_DEV_BIT_BROKEN, &udev->flags)) {
1450	pr_err("ring broken, not handling completions\n");
1451	return false;
1452	}
1453
1454	mb = udev->mb_addr;
1455	tcmu_flush_dcache_range(mb, sizeof(*mb));
1456
1457	while (udev->cmdr_last_cleaned != READ_ONCE(mb->cmd_tail)) {
1458
1459	struct tcmu_cmd_entry *entry = udev->cmdr + udev->cmdr_last_cleaned;
1460	bool keep_buf;
1461
1462	/*
1463	* Flush max. up to end of cmd ring since current entry might
1464	* be a padding that is shorter than sizeof(*entry)
1465	*/
1466	size_t ring_left = head_to_end(udev->cmdr_last_cleaned,
1467	udev->cmdr_size);
1468	tcmu_flush_dcache_range(entry, ring_left < sizeof(*entry) ?
1469	ring_left : sizeof(*entry));
1470
1471	free_space = true;
1472
1473	if (tcmu_hdr_get_op(entry->hdr.len_op) == TCMU_OP_PAD \|\|
1474	tcmu_hdr_get_op(entry->hdr.len_op) == TCMU_OP_TMR) {
1475	UPDATE_HEAD(udev->cmdr_last_cleaned,
1476	tcmu_hdr_get_len(entry->hdr.len_op),
1477	udev->cmdr_size);
1478	continue;
1479	}
1480	WARN_ON(tcmu_hdr_get_op(entry->hdr.len_op) != TCMU_OP_CMD);
1481
1482	keep_buf = !!(entry->hdr.uflags & TCMU_UFLAG_KEEP_BUF);
1483	if (keep_buf)
1484	cmd = xa_load(&udev->commands, entry->hdr.cmd_id);
1485	else
1486	cmd = xa_erase(&udev->commands, entry->hdr.cmd_id);
1487	if (!cmd) {
1488	pr_err("cmd_id %u not found, ring is broken\n",
1489	entry->hdr.cmd_id);
1490	set_bit(TCMU_DEV_BIT_BROKEN, &udev->flags);
1491	return false;
1492	}
1493
1494	if (!tcmu_handle_completion(cmd, entry, keep_buf))
1495	break;
1496
1497	UPDATE_HEAD(udev->cmdr_last_cleaned,
1498	tcmu_hdr_get_len(entry->hdr.len_op),
1499	udev->cmdr_size);
1500	}
1501	if (free_space)
1502	free_space = tcmu_run_tmr_queue(udev);
1503
1504	if (atomic_read(&global_page_count) > tcmu_global_max_pages &&
1505	xa_empty(&udev->commands) && list_empty(&udev->qfull_queue)) {
1506	/*
1507	* Allocated blocks exceeded global block limit, currently no
1508	* more pending or waiting commands so try to reclaim blocks.
1509	*/
1510	schedule_delayed_work(&tcmu_unmap_work, 0);
1511	}
1512	if (udev->cmd_time_out)
1513	tcmu_set_next_deadline(&udev->inflight_queue, &udev->cmd_timer);
1514
1515	return free_space;
1516	}
1517
1518	static void tcmu_check_expired_ring_cmd(struct tcmu_cmd *cmd)
1519	{
1520	struct se_cmd *se_cmd;
1521
1522	if (!time_after_eq(jiffies, cmd->deadline))
1523	return;
1524
1525	set_bit(TCMU_CMD_BIT_EXPIRED, &cmd->flags);
1526	list_del_init(&cmd->queue_entry);
1527	se_cmd = cmd->se_cmd;
1528	se_cmd->priv = NULL;
1529	cmd->se_cmd = NULL;
1530
1531	pr_debug("Timing out inflight cmd %u on dev %s.\n",
1532	cmd->cmd_id, cmd->tcmu_dev->name);
1533
1534	target_complete_cmd(se_cmd, SAM_STAT_CHECK_CONDITION);
1535	}
1536
1537	static void tcmu_check_expired_queue_cmd(struct tcmu_cmd *cmd)
1538	{
1539	struct se_cmd *se_cmd;
1540
1541	if (!time_after_eq(jiffies, cmd->deadline))
1542	return;
1543
1544	pr_debug("Timing out queued cmd %p on dev %s.\n",
1545	cmd, cmd->tcmu_dev->name);
1546
1547	list_del_init(&cmd->queue_entry);
1548	se_cmd = cmd->se_cmd;
1549	tcmu_free_cmd(cmd);
1550
1551	se_cmd->priv = NULL;
1552	target_complete_cmd(se_cmd, SAM_STAT_TASK_SET_FULL);
1553	}
1554
1555	static void tcmu_device_timedout(struct tcmu_dev *udev)
1556	{
1557	spin_lock(&timed_out_udevs_lock);
1558	if (list_empty(&udev->timedout_entry))
1559	list_add_tail(&udev->timedout_entry, &timed_out_udevs);
1560	spin_unlock(&timed_out_udevs_lock);
1561
1562	schedule_delayed_work(&tcmu_unmap_work, 0);
1563	}
1564
1565	static void tcmu_cmd_timedout(struct timer_list *t)
1566	{
1567	struct tcmu_dev *udev = timer_container_of(udev, t, cmd_timer);
1568
1569	pr_debug("%s cmd timeout has expired\n", udev->name);
1570	tcmu_device_timedout(udev);
1571	}
1572
1573	static void tcmu_qfull_timedout(struct timer_list *t)
1574	{
1575	struct tcmu_dev *udev = timer_container_of(udev, t, qfull_timer);
1576
1577	pr_debug("%s qfull timeout has expired\n", udev->name);
1578	tcmu_device_timedout(udev);
1579	}
1580
1581	static int tcmu_attach_hba(struct se_hba *hba, u32 host_id)
1582	{
1583	struct tcmu_hba *tcmu_hba;
1584
1585	tcmu_hba = kzalloc_obj(struct tcmu_hba);
1586	if (!tcmu_hba)
1587	return -ENOMEM;
1588
1589	tcmu_hba->host_id = host_id;
1590	hba->hba_ptr = tcmu_hba;
1591
1592	return 0;
1593	}
1594
1595	static void tcmu_detach_hba(struct se_hba *hba)
1596	{
1597	kfree(hba->hba_ptr);
1598	hba->hba_ptr = NULL;
1599	}
1600
1601	static struct se_device tcmu_alloc_device(struct se_hba hba, const char *name)
1602	{
1603	struct tcmu_dev *udev;
1604
1605	udev = kzalloc_obj(struct tcmu_dev);
1606	if (!udev)
1607	return NULL;
1608	kref_init(&udev->kref);
1609
1610	udev->name = kstrdup(name, GFP_KERNEL);
1611	if (!udev->name) {
1612	kfree(udev);
1613	return NULL;
1614	}
1615
1616	udev->hba = hba;
1617	udev->cmd_time_out = TCMU_TIME_OUT;
1618	udev->qfull_time_out = -1;
1619
1620	udev->data_pages_per_blk = DATA_PAGES_PER_BLK_DEF;
1621	udev->max_blocks = DATA_AREA_PAGES_DEF / udev->data_pages_per_blk;
1622	udev->cmdr_size = CMDR_SIZE_DEF;
1623	udev->data_area_mb = TCMU_PAGES_TO_MBS(DATA_AREA_PAGES_DEF);
1624
1625	mutex_init(&udev->cmdr_lock);
1626
1627	INIT_LIST_HEAD(&udev->node);
1628	INIT_LIST_HEAD(&udev->timedout_entry);
1629	INIT_LIST_HEAD(&udev->qfull_queue);
1630	INIT_LIST_HEAD(&udev->tmr_queue);
1631	INIT_LIST_HEAD(&udev->inflight_queue);
1632	xa_init_flags(&udev->commands, XA_FLAGS_ALLOC1);
1633
1634	timer_setup(&udev->qfull_timer, tcmu_qfull_timedout, 0);
1635	timer_setup(&udev->cmd_timer, tcmu_cmd_timedout, 0);
1636
1637	xa_init(&udev->data_pages);
1638
1639	return &udev->se_dev;
1640	}
1641
1642	static void tcmu_dev_call_rcu(struct rcu_head *p)
1643	{
1644	struct se_device *dev = container_of(p, struct se_device, rcu_head);
1645	struct tcmu_dev *udev = TCMU_DEV(dev);
1646
1647	kfree(udev->uio_info.name);
1648	kfree(udev->name);
1649	kfree(udev);
1650	}
1651
1652	static int tcmu_check_and_free_pending_cmd(struct tcmu_cmd *cmd)
1653	{
1654	if (test_bit(TCMU_CMD_BIT_EXPIRED, &cmd->flags) \|\|
1655	test_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags)) {
1656	kmem_cache_free(tcmu_cmd_cache, cmd);
1657	return 0;
1658	}
1659	return -EINVAL;
1660	}
1661
1662	static u32 tcmu_blocks_release(struct tcmu_dev *udev, unsigned long first,
1663	unsigned long last)
1664	{
1665	struct page *page;
1666	unsigned long dpi;
1667	u32 pages_freed = 0;
1668
1669	first = first * udev->data_pages_per_blk;
1670	last = (last + 1) * udev->data_pages_per_blk - 1;
1671	xa_for_each_range(&udev->data_pages, dpi, page, first, last) {
1672	xa_erase(&udev->data_pages, dpi);
1673	/*
1674	* While reaching here there may be page faults occurring on
1675	* the to-be-released pages. A race condition may occur if
1676	* unmap_mapping_range() is called before page faults on these
1677	* pages have completed; a valid but stale map is created.
1678	*
1679	* If another command subsequently runs and needs to extend
1680	* dbi_thresh, it may reuse the slot corresponding to the
1681	* previous page in data_bitmap. Though we will allocate a new
1682	* page for the slot in data_area, no page fault will happen
1683	* because we have a valid map. Therefore the command's data
1684	* will be lost.
1685	*
1686	* We lock and unlock pages that are to be released to ensure
1687	* all page faults have completed. This way
1688	* unmap_mapping_range() can ensure stale maps are cleanly
1689	* removed.
1690	*/
1691	lock_page(page);
1692	unlock_page(page);
1693	__free_page(page);
1694	pages_freed++;
1695	}
1696
1697	atomic_sub(pages_freed, &global_page_count);
1698
1699	return pages_freed;
1700	}
1701
1702	static void tcmu_remove_all_queued_tmr(struct tcmu_dev *udev)
1703	{
1704	struct tcmu_tmr tmr, tmp;
1705
1706	list_for_each_entry_safe(tmr, tmp, &udev->tmr_queue, queue_entry) {
1707	list_del_init(&tmr->queue_entry);
1708	kfree(tmr);
1709	}
1710	}
1711
1712	static void tcmu_dev_kref_release(struct kref *kref)
1713	{
1714	struct tcmu_dev *udev = container_of(kref, struct tcmu_dev, kref);
1715	struct se_device *dev = &udev->se_dev;
1716	struct tcmu_cmd *cmd;
1717	bool all_expired = true;
1718	unsigned long i;
1719
1720	vfree(udev->mb_addr);
1721	udev->mb_addr = NULL;
1722
1723	spin_lock_bh(&timed_out_udevs_lock);
1724	if (!list_empty(&udev->timedout_entry))
1725	list_del(&udev->timedout_entry);
1726	spin_unlock_bh(&timed_out_udevs_lock);
1727
1728	/* Upper layer should drain all requests before calling this */
1729	mutex_lock(&udev->cmdr_lock);
1730	xa_for_each(&udev->commands, i, cmd) {
1731	if (tcmu_check_and_free_pending_cmd(cmd) != 0)
1732	all_expired = false;
1733	}
1734	/* There can be left over TMR cmds. Remove them. */
1735	tcmu_remove_all_queued_tmr(udev);
1736	if (!list_empty(&udev->qfull_queue))
1737	all_expired = false;
1738	xa_destroy(&udev->commands);
1739	WARN_ON(!all_expired);
1740
1741	tcmu_blocks_release(udev, 0, udev->dbi_max);
1742	bitmap_free(udev->data_bitmap);
1743	mutex_unlock(&udev->cmdr_lock);
1744
1745	pr_debug("dev_kref_release\n");
1746
1747	call_rcu(&dev->rcu_head, tcmu_dev_call_rcu);
1748	}
1749
1750	static void run_qfull_queue(struct tcmu_dev *udev, bool fail)
1751	{
1752	struct tcmu_cmd tcmu_cmd, tmp_cmd;
1753	LIST_HEAD(cmds);
1754	sense_reason_t scsi_ret;
1755	int ret;
1756
1757	if (list_empty(&udev->qfull_queue))
1758	return;
1759
1760	pr_debug("running %s's cmdr queue forcefail %d\n", udev->name, fail);
1761
1762	list_splice_init(&udev->qfull_queue, &cmds);
1763
1764	list_for_each_entry_safe(tcmu_cmd, tmp_cmd, &cmds, queue_entry) {
1765	list_del_init(&tcmu_cmd->queue_entry);
1766
1767	pr_debug("removing cmd %p on dev %s from queue\n",
1768	tcmu_cmd, udev->name);
1769
1770	if (fail) {
1771	/*
1772	* We were not able to even start the command, so
1773	* fail with busy to allow a retry in case runner
1774	* was only temporarily down. If the device is being
1775	* removed then LIO core will do the right thing and
1776	* fail the retry.
1777	*/
1778	tcmu_cmd->se_cmd->priv = NULL;
1779	target_complete_cmd(tcmu_cmd->se_cmd, SAM_STAT_BUSY);
1780	tcmu_free_cmd(tcmu_cmd);
1781	continue;
1782	}
1783
1784	ret = queue_cmd_ring(tcmu_cmd, &scsi_ret);
1785	if (ret < 0) {
1786	pr_debug("cmd %p on dev %s failed with %u\n",
1787	tcmu_cmd, udev->name, scsi_ret);
1788	/*
1789	* Ignore scsi_ret for now. target_complete_cmd
1790	* drops it.
1791	*/
1792	tcmu_cmd->se_cmd->priv = NULL;
1793	target_complete_cmd(tcmu_cmd->se_cmd,
1794	SAM_STAT_CHECK_CONDITION);
1795	tcmu_free_cmd(tcmu_cmd);
1796	} else if (ret > 0) {
1797	pr_debug("ran out of space during cmdr queue run\n");
1798	/*
1799	* cmd was requeued, so just put all cmds back in
1800	* the queue
1801	*/
1802	list_splice_tail(&cmds, &udev->qfull_queue);
1803	break;
1804	}
1805	}
1806
1807	tcmu_set_next_deadline(&udev->qfull_queue, &udev->qfull_timer);
1808	}
1809
1810	static int tcmu_irqcontrol(struct uio_info *info, s32 irq_on)
1811	{
1812	struct tcmu_dev *udev = container_of(info, struct tcmu_dev, uio_info);
1813
1814	mutex_lock(&udev->cmdr_lock);
1815	if (tcmu_handle_completions(udev))
1816	run_qfull_queue(udev, false);
1817	mutex_unlock(&udev->cmdr_lock);
1818
1819	return 0;
1820	}
1821
1822	/*
1823	* mmap code from uio.c. Copied here because we want to hook mmap()
1824	* and this stuff must come along.
1825	*/
1826	static int tcmu_find_mem_index(struct vm_area_struct *vma)
1827	{
1828	struct tcmu_dev *udev = vma->vm_private_data;
1829	struct uio_info *info = &udev->uio_info;
1830
1831	if (vma->vm_pgoff < MAX_UIO_MAPS) {
1832	if (info->mem[vma->vm_pgoff].size == 0)
1833	return -1;
1834	return (int)vma->vm_pgoff;
1835	}
1836	return -1;
1837	}
1838
1839	static struct page tcmu_try_get_data_page(struct tcmu_dev udev, uint32_t dpi)
1840	{
1841	struct page *page;
1842
1843	mutex_lock(&udev->cmdr_lock);
1844	page = xa_load(&udev->data_pages, dpi);
1845	if (likely(page)) {
1846	get_page(page);
1847	lock_page(page);
1848	mutex_unlock(&udev->cmdr_lock);
1849	return page;
1850	}
1851
1852	/*
1853	* Userspace messed up and passed in a address not in the
1854	* data iov passed to it.
1855	*/
1856	pr_err("Invalid addr to data page mapping (dpi %u) on device %s\n",
1857	dpi, udev->name);
1858	mutex_unlock(&udev->cmdr_lock);
1859
1860	return NULL;
1861	}
1862
1863	static int tcmu_vma_mapped(unsigned long start, unsigned long end, pgoff_t pgoff,
1864	const struct file file, void *vm_private_data)
1865	{
1866	struct tcmu_dev udev = vm_private_data;
1867
1868	pr_debug("vma_mapped\n");
1869
1870	kref_get(&udev->kref);
1871	return 0;
1872	}
1873
1874	static void tcmu_vma_open(struct vm_area_struct *vma)
1875	{
1876	struct tcmu_dev *udev = vma->vm_private_data;
1877
1878	pr_debug("vma_open\n");
1879
1880	kref_get(&udev->kref);
1881	}
1882
1883	static void tcmu_vma_close(struct vm_area_struct *vma)
1884	{
1885	struct tcmu_dev *udev = vma->vm_private_data;
1886
1887	pr_debug("vma_close\n");
1888
1889	/* release ref from tcmu_vma_open */
1890	kref_put(&udev->kref, tcmu_dev_kref_release);
1891	}
1892
1893	static vm_fault_t tcmu_vma_fault(struct vm_fault *vmf)
1894	{
1895	struct tcmu_dev *udev = vmf->vma->vm_private_data;
1896	struct uio_info *info = &udev->uio_info;
1897	struct page *page;
1898	unsigned long offset;
1899	void *addr;
1900	vm_fault_t ret = 0;
1901
1902	int mi = tcmu_find_mem_index(vmf->vma);
1903	if (mi < 0)
1904	return VM_FAULT_SIGBUS;
1905
1906	/*
1907	* We need to subtract mi because userspace uses offset = N*PAGE_SIZE
1908	* to use mem[N].
1909	*/
1910	offset = (vmf->pgoff - mi) << PAGE_SHIFT;
1911
1912	if (offset < udev->data_off) {
1913	/* For the vmalloc()ed cmd area pages */
1914	addr = (void *)(unsigned long)info->mem[mi].addr + offset;
1915	page = vmalloc_to_page(addr);
1916	get_page(page);
1917	} else {
1918	uint32_t dpi;
1919
1920	/* For the dynamically growing data area pages */
1921	dpi = (offset - udev->data_off) / PAGE_SIZE;
1922	page = tcmu_try_get_data_page(udev, dpi);
1923	if (!page)
1924	return VM_FAULT_SIGBUS;
1925	ret = VM_FAULT_LOCKED;
1926	}
1927
1928	vmf->page = page;
1929	return ret;
1930	}
1931
1932	static const struct vm_operations_struct tcmu_vm_ops = {
1933	.mapped = tcmu_vma_mapped,
1934	.open = tcmu_vma_open,
1935	.close = tcmu_vma_close,
1936	.fault = tcmu_vma_fault,
1937	};
1938
1939	static int tcmu_mmap_prepare(struct uio_info info, struct vm_area_desc desc)
1940	{
1941	struct tcmu_dev *udev = container_of(info, struct tcmu_dev, uio_info);
1942
1943	vma_desc_set_flags(desc, VMA_DONTEXPAND_BIT, VMA_DONTDUMP_BIT);
1944	desc->vm_ops = &tcmu_vm_ops;
1945
1946	desc->private_data = udev;
1947
1948	/* Ensure the mmap is exactly the right size */
1949	if (vma_desc_pages(desc) != udev->mmap_pages)
1950	return -EINVAL;
1951
1952	return 0;
1953	}
1954
1955	static int tcmu_open(struct uio_info info, struct inode inode)
1956	{
1957	struct tcmu_dev *udev = container_of(info, struct tcmu_dev, uio_info);
1958
1959	/* O_EXCL not supported for char devs, so fake it? */
1960	if (test_and_set_bit(TCMU_DEV_BIT_OPEN, &udev->flags))
1961	return -EBUSY;
1962
1963	udev->inode = inode;
1964
1965	pr_debug("open\n");
1966
1967	return 0;
1968	}
1969
1970	static int tcmu_release(struct uio_info info, struct inode inode)
1971	{
1972	struct tcmu_dev *udev = container_of(info, struct tcmu_dev, uio_info);
1973	struct tcmu_cmd *cmd;
1974	unsigned long i;
1975	bool freed = false;
1976
1977	mutex_lock(&udev->cmdr_lock);
1978
1979	xa_for_each(&udev->commands, i, cmd) {
1980	/* Cmds with KEEP_BUF set are no longer on the ring, but
1981	* userspace still holds the data buffer. If userspace closes
1982	* we implicitly free these cmds and buffers, since after new
1983	* open the (new ?) userspace cannot find the cmd in the ring
1984	* and thus never will release the buffer by writing cmd_id to
1985	* free_kept_buf action attribute.
1986	*/
1987	if (!test_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags))
1988	continue;
1989	pr_debug("removing KEEP_BUF cmd %u on dev %s from ring\n",
1990	cmd->cmd_id, udev->name);
1991	freed = true;
1992
1993	xa_erase(&udev->commands, i);
1994	tcmu_cmd_free_data(cmd, cmd->dbi_cnt);
1995	tcmu_free_cmd(cmd);
1996	}
1997	/*
1998	* We only freed data space, not ring space. Therefore we dont call
1999	* run_tmr_queue, but call run_qfull_queue if tmr_list is empty.
2000	*/
2001	if (freed && list_empty(&udev->tmr_queue))
2002	run_qfull_queue(udev, false);
2003
2004	mutex_unlock(&udev->cmdr_lock);
2005
2006	clear_bit(TCMU_DEV_BIT_OPEN, &udev->flags);
2007
2008	pr_debug("close\n");
2009
2010	return 0;
2011	}
2012
2013	static int tcmu_init_genl_cmd_reply(struct tcmu_dev *udev, int cmd)
2014	{
2015	struct tcmu_nl_cmd *nl_cmd = &udev->curr_nl_cmd;
2016
2017	if (!tcmu_kern_cmd_reply_supported)
2018	return 0;
2019
2020	if (udev->nl_reply_supported <= 0)
2021	return 0;
2022
2023	mutex_lock(&tcmu_nl_cmd_mutex);
2024
2025	if (tcmu_netlink_blocked) {
2026	mutex_unlock(&tcmu_nl_cmd_mutex);
2027	pr_warn("Failing nl cmd %d on %s. Interface is blocked.\n", cmd,
2028	udev->name);
2029	return -EAGAIN;
2030	}
2031
2032	if (nl_cmd->cmd != TCMU_CMD_UNSPEC) {
2033	mutex_unlock(&tcmu_nl_cmd_mutex);
2034	pr_warn("netlink cmd %d already executing on %s\n",
2035	nl_cmd->cmd, udev->name);
2036	return -EBUSY;
2037	}
2038
2039	memset(nl_cmd, 0, sizeof(*nl_cmd));
2040	nl_cmd->cmd = cmd;
2041	nl_cmd->udev = udev;
2042	init_completion(&nl_cmd->complete);
2043	INIT_LIST_HEAD(&nl_cmd->nl_list);
2044
2045	list_add_tail(&nl_cmd->nl_list, &tcmu_nl_cmd_list);
2046
2047	mutex_unlock(&tcmu_nl_cmd_mutex);
2048	return 0;
2049	}
2050
2051	static void tcmu_destroy_genl_cmd_reply(struct tcmu_dev *udev)
2052	{
2053	struct tcmu_nl_cmd *nl_cmd = &udev->curr_nl_cmd;
2054
2055	if (!tcmu_kern_cmd_reply_supported)
2056	return;
2057
2058	if (udev->nl_reply_supported <= 0)
2059	return;
2060
2061	mutex_lock(&tcmu_nl_cmd_mutex);
2062
2063	list_del(&nl_cmd->nl_list);
2064	memset(nl_cmd, 0, sizeof(*nl_cmd));
2065
2066	mutex_unlock(&tcmu_nl_cmd_mutex);
2067	}
2068
2069	static int tcmu_wait_genl_cmd_reply(struct tcmu_dev *udev)
2070	{
2071	struct tcmu_nl_cmd *nl_cmd = &udev->curr_nl_cmd;
2072	int ret;
2073
2074	if (!tcmu_kern_cmd_reply_supported)
2075	return 0;
2076
2077	if (udev->nl_reply_supported <= 0)
2078	return 0;
2079
2080	pr_debug("sleeping for nl reply\n");
2081	wait_for_completion(&nl_cmd->complete);
2082
2083	mutex_lock(&tcmu_nl_cmd_mutex);
2084	nl_cmd->cmd = TCMU_CMD_UNSPEC;
2085	ret = nl_cmd->status;
2086	mutex_unlock(&tcmu_nl_cmd_mutex);
2087
2088	return ret;
2089	}
2090
2091	static int tcmu_netlink_event_init(struct tcmu_dev *udev,
2092	enum tcmu_genl_cmd cmd,
2093	struct sk_buff buf, void hdr)
2094	{
2095	struct sk_buff *skb;
2096	void *msg_header;
2097	int ret = -ENOMEM;
2098
2099	skb = genlmsg_new(NLMSG_GOODSIZE, GFP_KERNEL);
2100	if (!skb)
2101	return ret;
2102
2103	msg_header = genlmsg_put(skb, 0, 0, &tcmu_genl_family, 0, cmd);
2104	if (!msg_header)
2105	goto free_skb;
2106
2107	ret = nla_put_string(skb, TCMU_ATTR_DEVICE, udev->uio_info.name);
2108	if (ret < 0)
2109	goto free_skb;
2110
2111	ret = nla_put_u32(skb, TCMU_ATTR_MINOR, udev->uio_info.uio_dev->minor);
2112	if (ret < 0)
2113	goto free_skb;
2114
2115	ret = nla_put_u32(skb, TCMU_ATTR_DEVICE_ID, udev->se_dev.dev_index);
2116	if (ret < 0)
2117	goto free_skb;
2118
2119	*buf = skb;
2120	*hdr = msg_header;
2121	return ret;
2122
2123	free_skb:
2124	nlmsg_free(skb);
2125	return ret;
2126	}
2127
2128	static int tcmu_netlink_event_send(struct tcmu_dev *udev,
2129	enum tcmu_genl_cmd cmd,
2130	struct sk_buff skb, void msg_header)
2131	{
2132	int ret;
2133
2134	genlmsg_end(skb, msg_header);
2135
2136	ret = tcmu_init_genl_cmd_reply(udev, cmd);
2137	if (ret) {
2138	nlmsg_free(skb);
2139	return ret;
2140	}
2141
2142	ret = genlmsg_multicast_allns(&tcmu_genl_family, skb, 0,
2143	TCMU_MCGRP_CONFIG);
2144
2145	/* Wait during an add as the listener may not be up yet */
2146	if (ret == 0 \|\|
2147	(ret == -ESRCH && cmd == TCMU_CMD_ADDED_DEVICE))
2148	return tcmu_wait_genl_cmd_reply(udev);
2149	else
2150	tcmu_destroy_genl_cmd_reply(udev);
2151
2152	return ret;
2153	}
2154
2155	static int tcmu_send_dev_add_event(struct tcmu_dev *udev)
2156	{
2157	struct sk_buff *skb = NULL;
2158	void *msg_header = NULL;
2159	int ret = 0;
2160
2161	ret = tcmu_netlink_event_init(udev, TCMU_CMD_ADDED_DEVICE, &skb,
2162	&msg_header);
2163	if (ret < 0)
2164	return ret;
2165	return tcmu_netlink_event_send(udev, TCMU_CMD_ADDED_DEVICE, skb,
2166	msg_header);
2167	}
2168
2169	static int tcmu_send_dev_remove_event(struct tcmu_dev *udev)
2170	{
2171	struct sk_buff *skb = NULL;
2172	void *msg_header = NULL;
2173	int ret = 0;
2174
2175	ret = tcmu_netlink_event_init(udev, TCMU_CMD_REMOVED_DEVICE,
2176	&skb, &msg_header);
2177	if (ret < 0)
2178	return ret;
2179	return tcmu_netlink_event_send(udev, TCMU_CMD_REMOVED_DEVICE,
2180	skb, msg_header);
2181	}
2182
2183	static int tcmu_update_uio_info(struct tcmu_dev *udev)
2184	{
2185	struct tcmu_hba *hba = udev->hba->hba_ptr;
2186	struct uio_info *info;
2187	char *str;
2188
2189	info = &udev->uio_info;
2190
2191	if (udev->dev_config[0])
2192	str = kasprintf(GFP_KERNEL, "tcm-user/%u/%s/%s", hba->host_id,
2193	udev->name, udev->dev_config);
2194	else
2195	str = kasprintf(GFP_KERNEL, "tcm-user/%u/%s", hba->host_id,
2196	udev->name);
2197	if (!str)
2198	return -ENOMEM;
2199
2200	/* If the old string exists, free it */
2201	kfree(info->name);
2202	info->name = str;
2203
2204	return 0;
2205	}
2206
2207	static int tcmu_configure_device(struct se_device *dev)
2208	{
2209	struct tcmu_dev *udev = TCMU_DEV(dev);
2210	struct uio_info *info;
2211	struct tcmu_mailbox *mb;
2212	size_t data_size;
2213	int ret = 0;
2214
2215	ret = tcmu_update_uio_info(udev);
2216	if (ret)
2217	return ret;
2218
2219	info = &udev->uio_info;
2220
2221	mutex_lock(&udev->cmdr_lock);
2222	udev->data_bitmap = bitmap_zalloc(udev->max_blocks, GFP_KERNEL);
2223	mutex_unlock(&udev->cmdr_lock);
2224	if (!udev->data_bitmap) {
2225	ret = -ENOMEM;
2226	goto err_bitmap_alloc;
2227	}
2228
2229	mb = vzalloc(udev->cmdr_size + CMDR_OFF);
2230	if (!mb) {
2231	ret = -ENOMEM;
2232	goto err_vzalloc;
2233	}
2234
2235	/* mailbox fits in first part of CMDR space */
2236	udev->mb_addr = mb;
2237	udev->cmdr = (void *)mb + CMDR_OFF;
2238	udev->data_off = udev->cmdr_size + CMDR_OFF;
2239	data_size = TCMU_MBS_TO_PAGES(udev->data_area_mb) << PAGE_SHIFT;
2240	udev->mmap_pages = (data_size + udev->cmdr_size + CMDR_OFF) >> PAGE_SHIFT;
2241	udev->data_blk_size = udev->data_pages_per_blk * PAGE_SIZE;
2242	udev->dbi_thresh = 0; /* Default in Idle state */
2243
2244	/* Initialise the mailbox of the ring buffer */
2245	mb->version = TCMU_MAILBOX_VERSION;
2246	mb->flags = TCMU_MAILBOX_FLAG_CAP_OOOC \|
2247	TCMU_MAILBOX_FLAG_CAP_READ_LEN \|
2248	TCMU_MAILBOX_FLAG_CAP_TMR \|
2249	TCMU_MAILBOX_FLAG_CAP_KEEP_BUF;
2250	mb->cmdr_off = CMDR_OFF;
2251	mb->cmdr_size = udev->cmdr_size;
2252
2253	WARN_ON(!PAGE_ALIGNED(udev->data_off));
2254	WARN_ON(data_size % PAGE_SIZE);
2255
2256	info->version = __stringify(TCMU_MAILBOX_VERSION);
2257
2258	info->mem[0].name = "tcm-user command & data buffer";
2259	info->mem[0].addr = (phys_addr_t)(uintptr_t)udev->mb_addr;
2260	info->mem[0].size = data_size + udev->cmdr_size + CMDR_OFF;
2261	info->mem[0].memtype = UIO_MEM_NONE;
2262
2263	info->irqcontrol = tcmu_irqcontrol;
2264	info->irq = UIO_IRQ_CUSTOM;
2265
2266	info->mmap_prepare = tcmu_mmap_prepare;
2267	info->open = tcmu_open;
2268	info->release = tcmu_release;
2269
2270	ret = uio_register_device(tcmu_root_device, info);
2271	if (ret)
2272	goto err_register;
2273
2274	/* User can set hw_block_size before enable the device */
2275	if (dev->dev_attrib.hw_block_size == 0)
2276	dev->dev_attrib.hw_block_size = 512;
2277	/* Other attributes can be configured in userspace */
2278	if (!dev->dev_attrib.hw_max_sectors)
2279	dev->dev_attrib.hw_max_sectors = 128;
2280	if (!dev->dev_attrib.emulate_write_cache)
2281	dev->dev_attrib.emulate_write_cache = 0;
2282	dev->dev_attrib.hw_queue_depth = 128;
2283
2284	/* If user didn't explicitly disable netlink reply support, use
2285	* module scope setting.
2286	*/
2287	if (udev->nl_reply_supported >= 0)
2288	udev->nl_reply_supported = tcmu_kern_cmd_reply_supported;
2289
2290	/*
2291	* Get a ref incase userspace does a close on the uio device before
2292	* LIO has initiated tcmu_free_device.
2293	*/
2294	kref_get(&udev->kref);
2295
2296	ret = tcmu_send_dev_add_event(udev);
2297	if (ret)
2298	goto err_netlink;
2299
2300	mutex_lock(&root_udev_mutex);
2301	list_add(&udev->node, &root_udev);
2302	mutex_unlock(&root_udev_mutex);
2303
2304	return 0;
2305
2306	err_netlink:
2307	kref_put(&udev->kref, tcmu_dev_kref_release);
2308	uio_unregister_device(&udev->uio_info);
2309	err_register:
2310	vfree(udev->mb_addr);
2311	udev->mb_addr = NULL;
2312	err_vzalloc:
2313	bitmap_free(udev->data_bitmap);
2314	udev->data_bitmap = NULL;
2315	err_bitmap_alloc:
2316	kfree(info->name);
2317	info->name = NULL;
2318
2319	return ret;
2320	}
2321
2322	static void tcmu_free_device(struct se_device *dev)
2323	{
2324	struct tcmu_dev *udev = TCMU_DEV(dev);
2325
2326	/* release ref from init */
2327	kref_put(&udev->kref, tcmu_dev_kref_release);
2328	}
2329
2330	static void tcmu_destroy_device(struct se_device *dev)
2331	{
2332	struct tcmu_dev *udev = TCMU_DEV(dev);
2333
2334	timer_delete_sync(&udev->cmd_timer);
2335	timer_delete_sync(&udev->qfull_timer);
2336
2337	mutex_lock(&root_udev_mutex);
2338	list_del(&udev->node);
2339	mutex_unlock(&root_udev_mutex);
2340
2341	tcmu_send_dev_remove_event(udev);
2342
2343	uio_unregister_device(&udev->uio_info);
2344
2345	/* release ref from configure */
2346	kref_put(&udev->kref, tcmu_dev_kref_release);
2347	}
2348
2349	static void tcmu_unblock_dev(struct tcmu_dev *udev)
2350	{
2351	mutex_lock(&udev->cmdr_lock);
2352	clear_bit(TCMU_DEV_BIT_BLOCKED, &udev->flags);
2353	mutex_unlock(&udev->cmdr_lock);
2354	}
2355
2356	static void tcmu_block_dev(struct tcmu_dev *udev)
2357	{
2358	mutex_lock(&udev->cmdr_lock);
2359
2360	if (test_and_set_bit(TCMU_DEV_BIT_BLOCKED, &udev->flags))
2361	goto unlock;
2362
2363	/* complete IO that has executed successfully */
2364	tcmu_handle_completions(udev);
2365	/* fail IO waiting to be queued */
2366	run_qfull_queue(udev, true);
2367
2368	unlock:
2369	mutex_unlock(&udev->cmdr_lock);
2370	}
2371
2372	static void tcmu_reset_ring(struct tcmu_dev *udev, u8 err_level)
2373	{
2374	struct tcmu_mailbox *mb;
2375	struct tcmu_cmd *cmd;
2376	unsigned long i;
2377
2378	mutex_lock(&udev->cmdr_lock);
2379
2380	xa_for_each(&udev->commands, i, cmd) {
2381	pr_debug("removing cmd %u on dev %s from ring %s\n",
2382	cmd->cmd_id, udev->name,
2383	test_bit(TCMU_CMD_BIT_EXPIRED, &cmd->flags) ?
2384	"(is expired)" :
2385	(test_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags) ?
2386	"(is keep buffer)" : ""));
2387
2388	xa_erase(&udev->commands, i);
2389	if (!test_bit(TCMU_CMD_BIT_EXPIRED, &cmd->flags) &&
2390	!test_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags)) {
2391	WARN_ON(!cmd->se_cmd);
2392	list_del_init(&cmd->queue_entry);
2393	cmd->se_cmd->priv = NULL;
2394	if (err_level == 1) {
2395	/*
2396	* Userspace was not able to start the
2397	* command or it is retryable.
2398	*/
2399	target_complete_cmd(cmd->se_cmd, SAM_STAT_BUSY);
2400	} else {
2401	/* hard failure */
2402	target_complete_cmd(cmd->se_cmd,
2403	SAM_STAT_CHECK_CONDITION);
2404	}
2405	}
2406	tcmu_cmd_free_data(cmd, cmd->dbi_cnt);
2407	tcmu_free_cmd(cmd);
2408	}
2409
2410	mb = udev->mb_addr;
2411	tcmu_flush_dcache_range(mb, sizeof(*mb));
2412	pr_debug("mb last %u head %u tail %u\n", udev->cmdr_last_cleaned,
2413	mb->cmd_tail, mb->cmd_head);
2414
2415	udev->cmdr_last_cleaned = 0;
2416	mb->cmd_tail = 0;
2417	mb->cmd_head = 0;
2418	tcmu_flush_dcache_range(mb, sizeof(*mb));
2419	clear_bit(TCMU_DEV_BIT_BROKEN, &udev->flags);
2420
2421	timer_delete(&udev->cmd_timer);
2422
2423	/*
2424	* ring is empty and qfull queue never contains aborted commands.
2425	* So TMRs in tmr queue do not contain relevant cmd_ids.
2426	* After a ring reset userspace should do a fresh start, so
2427	* even LUN RESET message is no longer relevant.
2428	* Therefore remove all TMRs from qfull queue
2429	*/
2430	tcmu_remove_all_queued_tmr(udev);
2431
2432	run_qfull_queue(udev, false);
2433
2434	mutex_unlock(&udev->cmdr_lock);
2435	}
2436
2437	enum {
2438	Opt_dev_config, Opt_dev_size, Opt_hw_block_size, Opt_hw_max_sectors,
2439	Opt_nl_reply_supported, Opt_max_data_area_mb, Opt_data_pages_per_blk,
2440	Opt_cmd_ring_size_mb, Opt_err,
2441	};
2442
2443	static const match_table_t tokens = {
2444	{Opt_dev_config, "dev_config=%s"},
2445	{Opt_dev_size, "dev_size=%s"},
2446	{Opt_hw_block_size, "hw_block_size=%d"},
2447	{Opt_hw_max_sectors, "hw_max_sectors=%d"},
2448	{Opt_nl_reply_supported, "nl_reply_supported=%d"},
2449	{Opt_max_data_area_mb, "max_data_area_mb=%d"},
2450	{Opt_data_pages_per_blk, "data_pages_per_blk=%d"},
2451	{Opt_cmd_ring_size_mb, "cmd_ring_size_mb=%d"},
2452	{Opt_err, NULL}
2453	};
2454
2455	static int tcmu_set_dev_attrib(substring_t arg, u32 dev_attrib)
2456	{
2457	int val, ret;
2458
2459	ret = match_int(arg, &val);
2460	if (ret < 0) {
2461	pr_err("match_int() failed for dev attrib. Error %d.\n",
2462	ret);
2463	return ret;
2464	}
2465
2466	if (val <= 0) {
2467	pr_err("Invalid dev attrib value %d. Must be greater than zero.\n",
2468	val);
2469	return -EINVAL;
2470	}
2471	*dev_attrib = val;
2472	return 0;
2473	}
2474
2475	static int tcmu_set_max_blocks_param(struct tcmu_dev udev, substring_t arg)
2476	{
2477	int val, ret;
2478	uint32_t pages_per_blk = udev->data_pages_per_blk;
2479
2480	ret = match_int(arg, &val);
2481	if (ret < 0) {
2482	pr_err("match_int() failed for max_data_area_mb=. Error %d.\n",
2483	ret);
2484	return ret;
2485	}
2486	if (val <= 0) {
2487	pr_err("Invalid max_data_area %d.\n", val);
2488	return -EINVAL;
2489	}
2490	if (val > TCMU_PAGES_TO_MBS(tcmu_global_max_pages)) {
2491	pr_err("%d is too large. Adjusting max_data_area_mb to global limit of %u\n",
2492	val, TCMU_PAGES_TO_MBS(tcmu_global_max_pages));
2493	val = TCMU_PAGES_TO_MBS(tcmu_global_max_pages);
2494	}
2495	if (TCMU_MBS_TO_PAGES(val) < pages_per_blk) {
2496	pr_err("Invalid max_data_area %d (%zu pages): smaller than data_pages_per_blk (%u pages).\n",
2497	val, TCMU_MBS_TO_PAGES(val), pages_per_blk);
2498	return -EINVAL;
2499	}
2500
2501	mutex_lock(&udev->cmdr_lock);
2502	if (udev->data_bitmap) {
2503	pr_err("Cannot set max_data_area_mb after it has been enabled.\n");
2504	ret = -EINVAL;
2505	goto unlock;
2506	}
2507
2508	udev->data_area_mb = val;
2509	udev->max_blocks = TCMU_MBS_TO_PAGES(val) / pages_per_blk;
2510
2511	unlock:
2512	mutex_unlock(&udev->cmdr_lock);
2513	return ret;
2514	}
2515
2516	static int tcmu_set_data_pages_per_blk(struct tcmu_dev udev, substring_t arg)
2517	{
2518	int val, ret;
2519
2520	ret = match_int(arg, &val);
2521	if (ret < 0) {
2522	pr_err("match_int() failed for data_pages_per_blk=. Error %d.\n",
2523	ret);
2524	return ret;
2525	}
2526
2527	if (val > TCMU_MBS_TO_PAGES(udev->data_area_mb)) {
2528	pr_err("Invalid data_pages_per_blk %d: greater than max_data_area_mb %d -> %zd pages).\n",
2529	val, udev->data_area_mb,
2530	TCMU_MBS_TO_PAGES(udev->data_area_mb));
2531	return -EINVAL;
2532	}
2533
2534	mutex_lock(&udev->cmdr_lock);
2535	if (udev->data_bitmap) {
2536	pr_err("Cannot set data_pages_per_blk after it has been enabled.\n");
2537	ret = -EINVAL;
2538	goto unlock;
2539	}
2540
2541	udev->data_pages_per_blk = val;
2542	udev->max_blocks = TCMU_MBS_TO_PAGES(udev->data_area_mb) / val;
2543
2544	unlock:
2545	mutex_unlock(&udev->cmdr_lock);
2546	return ret;
2547	}
2548
2549	static int tcmu_set_cmd_ring_size(struct tcmu_dev udev, substring_t arg)
2550	{
2551	int val, ret;
2552
2553	ret = match_int(arg, &val);
2554	if (ret < 0) {
2555	pr_err("match_int() failed for cmd_ring_size_mb=. Error %d.\n",
2556	ret);
2557	return ret;
2558	}
2559
2560	if (val <= 0) {
2561	pr_err("Invalid cmd_ring_size_mb %d.\n", val);
2562	return -EINVAL;
2563	}
2564
2565	mutex_lock(&udev->cmdr_lock);
2566	if (udev->data_bitmap) {
2567	pr_err("Cannot set cmd_ring_size_mb after it has been enabled.\n");
2568	ret = -EINVAL;
2569	goto unlock;
2570	}
2571
2572	udev->cmdr_size = (val << 20) - CMDR_OFF;
2573	if (val > (MB_CMDR_SIZE_DEF >> 20)) {
2574	pr_err("%d is too large. Adjusting cmd_ring_size_mb to global limit of %u\n",
2575	val, (MB_CMDR_SIZE_DEF >> 20));
2576	udev->cmdr_size = CMDR_SIZE_DEF;
2577	}
2578
2579	unlock:
2580	mutex_unlock(&udev->cmdr_lock);
2581	return ret;
2582	}
2583
2584	static ssize_t tcmu_set_configfs_dev_params(struct se_device *dev,
2585	const char *page, ssize_t count)
2586	{
2587	struct tcmu_dev *udev = TCMU_DEV(dev);
2588	char orig, ptr, *opts;
2589	substring_t args[MAX_OPT_ARGS];
2590	int ret = 0, token;
2591
2592	opts = kstrdup(page, GFP_KERNEL);
2593	if (!opts)
2594	return -ENOMEM;
2595
2596	orig = opts;
2597
2598	while ((ptr = strsep(&opts, ",\n")) != NULL) {
2599	if (!*ptr)
2600	continue;
2601
2602	token = match_token(ptr, tokens, args);
2603	switch (token) {
2604	case Opt_dev_config:
2605	if (match_strlcpy(udev->dev_config, &args[0],
2606	TCMU_CONFIG_LEN) == 0) {
2607	ret = -EINVAL;
2608	break;
2609	}
2610	pr_debug("TCMU: Referencing Path: %s\n", udev->dev_config);
2611	break;
2612	case Opt_dev_size:
2613	ret = match_u64(&args[0], &udev->dev_size);
2614	if (ret < 0)
2615	pr_err("match_u64() failed for dev_size=. Error %d.\n",
2616	ret);
2617	break;
2618	case Opt_hw_block_size:
2619	ret = tcmu_set_dev_attrib(&args[0],
2620	&(dev->dev_attrib.hw_block_size));
2621	break;
2622	case Opt_hw_max_sectors:
2623	ret = tcmu_set_dev_attrib(&args[0],
2624	&(dev->dev_attrib.hw_max_sectors));
2625	break;
2626	case Opt_nl_reply_supported:
2627	ret = match_int(&args[0], &udev->nl_reply_supported);
2628	if (ret < 0)
2629	pr_err("match_int() failed for nl_reply_supported=. Error %d.\n",
2630	ret);
2631	break;
2632	case Opt_max_data_area_mb:
2633	ret = tcmu_set_max_blocks_param(udev, &args[0]);
2634	break;
2635	case Opt_data_pages_per_blk:
2636	ret = tcmu_set_data_pages_per_blk(udev, &args[0]);
2637	break;
2638	case Opt_cmd_ring_size_mb:
2639	ret = tcmu_set_cmd_ring_size(udev, &args[0]);
2640	break;
2641	default:
2642	break;
2643	}
2644
2645	if (ret)
2646	break;
2647	}
2648
2649	kfree(orig);
2650	return (!ret) ? count : ret;
2651	}
2652
2653	static ssize_t tcmu_show_configfs_dev_params(struct se_device dev, char b)
2654	{
2655	struct tcmu_dev *udev = TCMU_DEV(dev);
2656	ssize_t bl = 0;
2657
2658	bl = sprintf(b + bl, "Config: %s ",
2659	udev->dev_config[0] ? udev->dev_config : "NULL");
2660	bl += sprintf(b + bl, "Size: %llu ", udev->dev_size);
2661	bl += sprintf(b + bl, "MaxDataAreaMB: %u ", udev->data_area_mb);
2662	bl += sprintf(b + bl, "DataPagesPerBlk: %u ", udev->data_pages_per_blk);
2663	bl += sprintf(b + bl, "CmdRingSizeMB: %u\n",
2664	(udev->cmdr_size + CMDR_OFF) >> 20);
2665
2666	return bl;
2667	}
2668
2669	static sector_t tcmu_get_blocks(struct se_device *dev)
2670	{
2671	struct tcmu_dev *udev = TCMU_DEV(dev);
2672
2673	return div_u64(udev->dev_size - dev->dev_attrib.block_size,
2674	dev->dev_attrib.block_size);
2675	}
2676
2677	static sense_reason_t
2678	tcmu_parse_cdb(struct se_cmd *cmd)
2679	{
2680	return passthrough_parse_cdb(cmd, tcmu_queue_cmd);
2681	}
2682
2683	static ssize_t tcmu_cmd_time_out_show(struct config_item item, char page)
2684	{
2685	struct se_dev_attrib *da = container_of(to_config_group(item),
2686	struct se_dev_attrib, da_group);
2687	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2688
2689	return snprintf(page, PAGE_SIZE, "%lu\n", udev->cmd_time_out / MSEC_PER_SEC);
2690	}
2691
2692	static ssize_t tcmu_cmd_time_out_store(struct config_item item, const char page,
2693	size_t count)
2694	{
2695	struct se_dev_attrib *da = container_of(to_config_group(item),
2696	struct se_dev_attrib, da_group);
2697	struct tcmu_dev *udev = container_of(da->da_dev,
2698	struct tcmu_dev, se_dev);
2699	u32 val;
2700	int ret;
2701
2702	if (da->da_dev->export_count) {
2703	pr_err("Unable to set tcmu cmd_time_out while exports exist\n");
2704	return -EINVAL;
2705	}
2706
2707	ret = kstrtou32(page, 0, &val);
2708	if (ret < 0)
2709	return ret;
2710
2711	udev->cmd_time_out = val * MSEC_PER_SEC;
2712	return count;
2713	}
2714	CONFIGFS_ATTR(tcmu_, cmd_time_out);
2715
2716	static ssize_t tcmu_qfull_time_out_show(struct config_item item, char page)
2717	{
2718	struct se_dev_attrib *da = container_of(to_config_group(item),
2719	struct se_dev_attrib, da_group);
2720	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2721
2722	return snprintf(page, PAGE_SIZE, "%ld\n", udev->qfull_time_out <= 0 ?
2723	udev->qfull_time_out :
2724	udev->qfull_time_out / MSEC_PER_SEC);
2725	}
2726
2727	static ssize_t tcmu_qfull_time_out_store(struct config_item *item,
2728	const char *page, size_t count)
2729	{
2730	struct se_dev_attrib *da = container_of(to_config_group(item),
2731	struct se_dev_attrib, da_group);
2732	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2733	s32 val;
2734	int ret;
2735
2736	ret = kstrtos32(page, 0, &val);
2737	if (ret < 0)
2738	return ret;
2739
2740	if (val >= 0) {
2741	udev->qfull_time_out = val * MSEC_PER_SEC;
2742	} else if (val == -1) {
2743	udev->qfull_time_out = val;
2744	} else {
2745	printk(KERN_ERR "Invalid qfull timeout value %d\n", val);
2746	return -EINVAL;
2747	}
2748	return count;
2749	}
2750	CONFIGFS_ATTR(tcmu_, qfull_time_out);
2751
2752	static ssize_t tcmu_max_data_area_mb_show(struct config_item item, char page)
2753	{
2754	struct se_dev_attrib *da = container_of(to_config_group(item),
2755	struct se_dev_attrib, da_group);
2756	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2757
2758	return snprintf(page, PAGE_SIZE, "%u\n", udev->data_area_mb);
2759	}
2760	CONFIGFS_ATTR_RO(tcmu_, max_data_area_mb);
2761
2762	static ssize_t tcmu_data_pages_per_blk_show(struct config_item *item,
2763	char *page)
2764	{
2765	struct se_dev_attrib *da = container_of(to_config_group(item),
2766	struct se_dev_attrib, da_group);
2767	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2768
2769	return snprintf(page, PAGE_SIZE, "%u\n", udev->data_pages_per_blk);
2770	}
2771	CONFIGFS_ATTR_RO(tcmu_, data_pages_per_blk);
2772
2773	static ssize_t tcmu_cmd_ring_size_mb_show(struct config_item item, char page)
2774	{
2775	struct se_dev_attrib *da = container_of(to_config_group(item),
2776	struct se_dev_attrib, da_group);
2777	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2778
2779	return snprintf(page, PAGE_SIZE, "%u\n",
2780	(udev->cmdr_size + CMDR_OFF) >> 20);
2781	}
2782	CONFIGFS_ATTR_RO(tcmu_, cmd_ring_size_mb);
2783
2784	static ssize_t tcmu_dev_config_show(struct config_item item, char page)
2785	{
2786	struct se_dev_attrib *da = container_of(to_config_group(item),
2787	struct se_dev_attrib, da_group);
2788	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2789
2790	return snprintf(page, PAGE_SIZE, "%s\n", udev->dev_config);
2791	}
2792
2793	static int tcmu_send_dev_config_event(struct tcmu_dev *udev,
2794	const char *reconfig_data)
2795	{
2796	struct sk_buff *skb = NULL;
2797	void *msg_header = NULL;
2798	int ret = 0;
2799
2800	ret = tcmu_netlink_event_init(udev, TCMU_CMD_RECONFIG_DEVICE,
2801	&skb, &msg_header);
2802	if (ret < 0)
2803	return ret;
2804	ret = nla_put_string(skb, TCMU_ATTR_DEV_CFG, reconfig_data);
2805	if (ret < 0) {
2806	nlmsg_free(skb);
2807	return ret;
2808	}
2809	return tcmu_netlink_event_send(udev, TCMU_CMD_RECONFIG_DEVICE,
2810	skb, msg_header);
2811	}
2812
2813
2814	static ssize_t tcmu_dev_config_store(struct config_item item, const char page,
2815	size_t count)
2816	{
2817	struct se_dev_attrib *da = container_of(to_config_group(item),
2818	struct se_dev_attrib, da_group);
2819	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2820	int ret, len;
2821
2822	len = strlen(page);
2823	if (!len \|\| len > TCMU_CONFIG_LEN - 1)
2824	return -EINVAL;
2825
2826	/* Check if device has been configured before */
2827	if (target_dev_configured(&udev->se_dev)) {
2828	ret = tcmu_send_dev_config_event(udev, page);
2829	if (ret) {
2830	pr_err("Unable to reconfigure device\n");
2831	return ret;
2832	}
2833	strscpy(udev->dev_config, page, TCMU_CONFIG_LEN);
2834
2835	ret = tcmu_update_uio_info(udev);
2836	if (ret)
2837	return ret;
2838	return count;
2839	}
2840	strscpy(udev->dev_config, page, TCMU_CONFIG_LEN);
2841
2842	return count;
2843	}
2844	CONFIGFS_ATTR(tcmu_, dev_config);
2845
2846	static ssize_t tcmu_dev_size_show(struct config_item item, char page)
2847	{
2848	struct se_dev_attrib *da = container_of(to_config_group(item),
2849	struct se_dev_attrib, da_group);
2850	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2851
2852	return snprintf(page, PAGE_SIZE, "%llu\n", udev->dev_size);
2853	}
2854
2855	static int tcmu_send_dev_size_event(struct tcmu_dev *udev, u64 size)
2856	{
2857	struct sk_buff *skb = NULL;
2858	void *msg_header = NULL;
2859	int ret = 0;
2860
2861	ret = tcmu_netlink_event_init(udev, TCMU_CMD_RECONFIG_DEVICE,
2862	&skb, &msg_header);
2863	if (ret < 0)
2864	return ret;
2865	ret = nla_put_u64_64bit(skb, TCMU_ATTR_DEV_SIZE,
2866	size, TCMU_ATTR_PAD);
2867	if (ret < 0) {
2868	nlmsg_free(skb);
2869	return ret;
2870	}
2871	return tcmu_netlink_event_send(udev, TCMU_CMD_RECONFIG_DEVICE,
2872	skb, msg_header);
2873	}
2874
2875	static ssize_t tcmu_dev_size_store(struct config_item item, const char page,
2876	size_t count)
2877	{
2878	struct se_dev_attrib *da = container_of(to_config_group(item),
2879	struct se_dev_attrib, da_group);
2880	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2881	u64 val;
2882	int ret;
2883
2884	ret = kstrtou64(page, 0, &val);
2885	if (ret < 0)
2886	return ret;
2887
2888	/* Check if device has been configured before */
2889	if (target_dev_configured(&udev->se_dev)) {
2890	ret = tcmu_send_dev_size_event(udev, val);
2891	if (ret) {
2892	pr_err("Unable to reconfigure device\n");
2893	return ret;
2894	}
2895	}
2896	udev->dev_size = val;
2897	return count;
2898	}
2899	CONFIGFS_ATTR(tcmu_, dev_size);
2900
2901	static ssize_t tcmu_nl_reply_supported_show(struct config_item *item,
2902	char *page)
2903	{
2904	struct se_dev_attrib *da = container_of(to_config_group(item),
2905	struct se_dev_attrib, da_group);
2906	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2907
2908	return snprintf(page, PAGE_SIZE, "%d\n", udev->nl_reply_supported);
2909	}
2910
2911	static ssize_t tcmu_nl_reply_supported_store(struct config_item *item,
2912	const char *page, size_t count)
2913	{
2914	struct se_dev_attrib *da = container_of(to_config_group(item),
2915	struct se_dev_attrib, da_group);
2916	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2917	s8 val;
2918	int ret;
2919
2920	ret = kstrtos8(page, 0, &val);
2921	if (ret < 0)
2922	return ret;
2923
2924	udev->nl_reply_supported = val;
2925	return count;
2926	}
2927	CONFIGFS_ATTR(tcmu_, nl_reply_supported);
2928
2929	static ssize_t tcmu_emulate_write_cache_show(struct config_item *item,
2930	char *page)
2931	{
2932	struct se_dev_attrib *da = container_of(to_config_group(item),
2933	struct se_dev_attrib, da_group);
2934
2935	return snprintf(page, PAGE_SIZE, "%i\n", da->emulate_write_cache);
2936	}
2937
2938	static int tcmu_send_emulate_write_cache(struct tcmu_dev *udev, u8 val)
2939	{
2940	struct sk_buff *skb = NULL;
2941	void *msg_header = NULL;
2942	int ret = 0;
2943
2944	ret = tcmu_netlink_event_init(udev, TCMU_CMD_RECONFIG_DEVICE,
2945	&skb, &msg_header);
2946	if (ret < 0)
2947	return ret;
2948	ret = nla_put_u8(skb, TCMU_ATTR_WRITECACHE, val);
2949	if (ret < 0) {
2950	nlmsg_free(skb);
2951	return ret;
2952	}
2953	return tcmu_netlink_event_send(udev, TCMU_CMD_RECONFIG_DEVICE,
2954	skb, msg_header);
2955	}
2956
2957	static ssize_t tcmu_emulate_write_cache_store(struct config_item *item,
2958	const char *page, size_t count)
2959	{
2960	struct se_dev_attrib *da = container_of(to_config_group(item),
2961	struct se_dev_attrib, da_group);
2962	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2963	u8 val;
2964	int ret;
2965
2966	ret = kstrtou8(page, 0, &val);
2967	if (ret < 0)
2968	return ret;
2969
2970	/* Check if device has been configured before */
2971	if (target_dev_configured(&udev->se_dev)) {
2972	ret = tcmu_send_emulate_write_cache(udev, val);
2973	if (ret) {
2974	pr_err("Unable to reconfigure device\n");
2975	return ret;
2976	}
2977	}
2978
2979	da->emulate_write_cache = val;
2980	return count;
2981	}
2982	CONFIGFS_ATTR(tcmu_, emulate_write_cache);
2983
2984	static ssize_t tcmu_tmr_notification_show(struct config_item item, char page)
2985	{
2986	struct se_dev_attrib *da = container_of(to_config_group(item),
2987	struct se_dev_attrib, da_group);
2988	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
2989
2990	return snprintf(page, PAGE_SIZE, "%i\n",
2991	test_bit(TCMU_DEV_BIT_TMR_NOTIFY, &udev->flags));
2992	}
2993
2994	static ssize_t tcmu_tmr_notification_store(struct config_item *item,
2995	const char *page, size_t count)
2996	{
2997	struct se_dev_attrib *da = container_of(to_config_group(item),
2998	struct se_dev_attrib, da_group);
2999	struct tcmu_dev *udev = TCMU_DEV(da->da_dev);
3000	u8 val;
3001	int ret;
3002
3003	ret = kstrtou8(page, 0, &val);
3004	if (ret < 0)
3005	return ret;
3006	if (val > 1)
3007	return -EINVAL;
3008
3009	if (val)
3010	set_bit(TCMU_DEV_BIT_TMR_NOTIFY, &udev->flags);
3011	else
3012	clear_bit(TCMU_DEV_BIT_TMR_NOTIFY, &udev->flags);
3013	return count;
3014	}
3015	CONFIGFS_ATTR(tcmu_, tmr_notification);
3016
3017	static ssize_t tcmu_block_dev_show(struct config_item item, char page)
3018	{
3019	struct se_device *se_dev = container_of(to_config_group(item),
3020	struct se_device,
3021	dev_action_group);
3022	struct tcmu_dev *udev = TCMU_DEV(se_dev);
3023
3024	if (test_bit(TCMU_DEV_BIT_BLOCKED, &udev->flags))
3025	return snprintf(page, PAGE_SIZE, "%s\n", "blocked");
3026	else
3027	return snprintf(page, PAGE_SIZE, "%s\n", "unblocked");
3028	}
3029
3030	static ssize_t tcmu_block_dev_store(struct config_item item, const char page,
3031	size_t count)
3032	{
3033	struct se_device *se_dev = container_of(to_config_group(item),
3034	struct se_device,
3035	dev_action_group);
3036	struct tcmu_dev *udev = TCMU_DEV(se_dev);
3037	u8 val;
3038	int ret;
3039
3040	if (!target_dev_configured(&udev->se_dev)) {
3041	pr_err("Device is not configured.\n");
3042	return -EINVAL;
3043	}
3044
3045	ret = kstrtou8(page, 0, &val);
3046	if (ret < 0)
3047	return ret;
3048
3049	if (val > 1) {
3050	pr_err("Invalid block value %d\n", val);
3051	return -EINVAL;
3052	}
3053
3054	if (!val)
3055	tcmu_unblock_dev(udev);
3056	else
3057	tcmu_block_dev(udev);
3058	return count;
3059	}
3060	CONFIGFS_ATTR(tcmu_, block_dev);
3061
3062	static ssize_t tcmu_reset_ring_store(struct config_item item, const char page,
3063	size_t count)
3064	{
3065	struct se_device *se_dev = container_of(to_config_group(item),
3066	struct se_device,
3067	dev_action_group);
3068	struct tcmu_dev *udev = TCMU_DEV(se_dev);
3069	u8 val;
3070	int ret;
3071
3072	if (!target_dev_configured(&udev->se_dev)) {
3073	pr_err("Device is not configured.\n");
3074	return -EINVAL;
3075	}
3076
3077	ret = kstrtou8(page, 0, &val);
3078	if (ret < 0)
3079	return ret;
3080
3081	if (val != 1 && val != 2) {
3082	pr_err("Invalid reset ring value %d\n", val);
3083	return -EINVAL;
3084	}
3085
3086	tcmu_reset_ring(udev, val);
3087	return count;
3088	}
3089	CONFIGFS_ATTR_WO(tcmu_, reset_ring);
3090
3091	static ssize_t tcmu_free_kept_buf_store(struct config_item item, const char page,
3092	size_t count)
3093	{
3094	struct se_device *se_dev = container_of(to_config_group(item),
3095	struct se_device,
3096	dev_action_group);
3097	struct tcmu_dev *udev = TCMU_DEV(se_dev);
3098	struct tcmu_cmd *cmd;
3099	u16 cmd_id;
3100	int ret;
3101
3102	if (!target_dev_configured(&udev->se_dev)) {
3103	pr_err("Device is not configured.\n");
3104	return -EINVAL;
3105	}
3106
3107	ret = kstrtou16(page, 0, &cmd_id);
3108	if (ret < 0)
3109	return ret;
3110
3111	mutex_lock(&udev->cmdr_lock);
3112
3113	{
3114	XA_STATE(xas, &udev->commands, cmd_id);
3115
3116	xas_lock(&xas);
3117	cmd = xas_load(&xas);
3118	if (!cmd) {
3119	pr_err("free_kept_buf: cmd_id %d not found\n", cmd_id);
3120	count = -EINVAL;
3121	xas_unlock(&xas);
3122	goto out_unlock;
3123	}
3124	if (!test_bit(TCMU_CMD_BIT_KEEP_BUF, &cmd->flags)) {
3125	pr_err("free_kept_buf: cmd_id %d was not completed with KEEP_BUF\n",
3126	cmd_id);
3127	count = -EINVAL;
3128	xas_unlock(&xas);
3129	goto out_unlock;
3130	}
3131	xas_store(&xas, NULL);
3132	xas_unlock(&xas);
3133	}
3134
3135	tcmu_cmd_free_data(cmd, cmd->dbi_cnt);
3136	tcmu_free_cmd(cmd);
3137	/*
3138	* We only freed data space, not ring space. Therefore we dont call
3139	* run_tmr_queue, but call run_qfull_queue if tmr_list is empty.
3140	*/
3141	if (list_empty(&udev->tmr_queue))
3142	run_qfull_queue(udev, false);
3143
3144	out_unlock:
3145	mutex_unlock(&udev->cmdr_lock);
3146	return count;
3147	}
3148	CONFIGFS_ATTR_WO(tcmu_, free_kept_buf);
3149
3150	static struct configfs_attribute *tcmu_attrib_attrs[] = {
3151	&tcmu_attr_cmd_time_out,
3152	&tcmu_attr_qfull_time_out,
3153	&tcmu_attr_max_data_area_mb,
3154	&tcmu_attr_data_pages_per_blk,
3155	&tcmu_attr_cmd_ring_size_mb,
3156	&tcmu_attr_dev_config,
3157	&tcmu_attr_dev_size,
3158	&tcmu_attr_emulate_write_cache,
3159	&tcmu_attr_tmr_notification,
3160	&tcmu_attr_nl_reply_supported,
3161	NULL,
3162	};
3163
3164	static struct configfs_attribute **tcmu_attrs;
3165
3166	static struct configfs_attribute *tcmu_action_attrs[] = {
3167	&tcmu_attr_block_dev,
3168	&tcmu_attr_reset_ring,
3169	&tcmu_attr_free_kept_buf,
3170	NULL,
3171	};
3172
3173	static struct target_backend_ops tcmu_ops = {
3174	.name = "user",
3175	.owner = THIS_MODULE,
3176	.transport_flags_default = TRANSPORT_FLAG_PASSTHROUGH,
3177	.transport_flags_changeable = TRANSPORT_FLAG_PASSTHROUGH_PGR \|
3178	TRANSPORT_FLAG_PASSTHROUGH_ALUA,
3179	.attach_hba = tcmu_attach_hba,
3180	.detach_hba = tcmu_detach_hba,
3181	.alloc_device = tcmu_alloc_device,
3182	.configure_device = tcmu_configure_device,
3183	.destroy_device = tcmu_destroy_device,
3184	.free_device = tcmu_free_device,
3185	.unplug_device = tcmu_unplug_device,
3186	.plug_device = tcmu_plug_device,
3187	.parse_cdb = tcmu_parse_cdb,
3188	.tmr_notify = tcmu_tmr_notify,
3189	.set_configfs_dev_params = tcmu_set_configfs_dev_params,
3190	.show_configfs_dev_params = tcmu_show_configfs_dev_params,
3191	.get_device_type = sbc_get_device_type,
3192	.get_blocks = tcmu_get_blocks,
3193	.tb_dev_action_attrs = tcmu_action_attrs,
3194	};
3195
3196	static void find_free_blocks(void)
3197	{
3198	struct tcmu_dev *udev;
3199	loff_t off;
3200	u32 pages_freed, total_pages_freed = 0;
3201	u32 start, end, block, total_blocks_freed = 0;
3202
3203	if (atomic_read(&global_page_count) <= tcmu_global_max_pages)
3204	return;
3205
3206	mutex_lock(&root_udev_mutex);
3207	list_for_each_entry(udev, &root_udev, node) {
3208	mutex_lock(&udev->cmdr_lock);
3209
3210	if (!target_dev_configured(&udev->se_dev)) {
3211	mutex_unlock(&udev->cmdr_lock);
3212	continue;
3213	}
3214
3215	/* Try to complete the finished commands first */
3216	if (tcmu_handle_completions(udev))
3217	run_qfull_queue(udev, false);
3218
3219	/* Skip the udevs in idle */
3220	if (!udev->dbi_thresh) {
3221	mutex_unlock(&udev->cmdr_lock);
3222	continue;
3223	}
3224
3225	end = udev->dbi_max + 1;
3226	block = find_last_bit(udev->data_bitmap, end);
3227	if (block == udev->dbi_max) {
3228	/*
3229	* The last bit is dbi_max, so it is not possible
3230	* reclaim any blocks.
3231	*/
3232	mutex_unlock(&udev->cmdr_lock);
3233	continue;
3234	} else if (block == end) {
3235	/* The current udev will goto idle state */
3236	udev->dbi_thresh = start = 0;
3237	udev->dbi_max = 0;
3238	} else {
3239	udev->dbi_thresh = start = block + 1;
3240	udev->dbi_max = block;
3241	}
3242
3243	/*
3244	* Release the block pages.
3245	*
3246	* Also note that since tcmu_vma_fault() gets an extra page
3247	* refcount, tcmu_blocks_release() won't free pages if pages
3248	* are mapped. This means it is safe to call
3249	* tcmu_blocks_release() before unmap_mapping_range() which
3250	* drops the refcount of any pages it unmaps and thus releases
3251	* them.
3252	*/
3253	pages_freed = tcmu_blocks_release(udev, start, end - 1);
3254
3255	/* Here will truncate the data area from off */
3256	off = udev->data_off + (loff_t)start * udev->data_blk_size;
3257	unmap_mapping_range(udev->inode->i_mapping, off, 0, 1);
3258
3259	mutex_unlock(&udev->cmdr_lock);
3260
3261	total_pages_freed += pages_freed;
3262	total_blocks_freed += end - start;
3263	pr_debug("Freed %u pages (total %u) from %u blocks (total %u) from %s.\n",
3264	pages_freed, total_pages_freed, end - start,
3265	total_blocks_freed, udev->name);
3266	}
3267	mutex_unlock(&root_udev_mutex);
3268
3269	if (atomic_read(&global_page_count) > tcmu_global_max_pages)
3270	schedule_delayed_work(&tcmu_unmap_work, msecs_to_jiffies(5000));
3271	}
3272
3273	static void check_timedout_devices(void)
3274	{
3275	struct tcmu_dev udev, tmp_dev;
3276	struct tcmu_cmd cmd, tmp_cmd;
3277	LIST_HEAD(devs);
3278
3279	spin_lock_bh(&timed_out_udevs_lock);
3280	list_splice_init(&timed_out_udevs, &devs);
3281
3282	list_for_each_entry_safe(udev, tmp_dev, &devs, timedout_entry) {
3283	list_del_init(&udev->timedout_entry);
3284	spin_unlock_bh(&timed_out_udevs_lock);
3285
3286	mutex_lock(&udev->cmdr_lock);
3287
3288	/*
3289	* If cmd_time_out is disabled but qfull is set deadline
3290	* will only reflect the qfull timeout. Ignore it.
3291	*/
3292	if (udev->cmd_time_out) {
3293	list_for_each_entry_safe(cmd, tmp_cmd,
3294	&udev->inflight_queue,
3295	queue_entry) {
3296	tcmu_check_expired_ring_cmd(cmd);
3297	}
3298	tcmu_set_next_deadline(&udev->inflight_queue,
3299	&udev->cmd_timer);
3300	}
3301	list_for_each_entry_safe(cmd, tmp_cmd, &udev->qfull_queue,
3302	queue_entry) {
3303	tcmu_check_expired_queue_cmd(cmd);
3304	}
3305	tcmu_set_next_deadline(&udev->qfull_queue, &udev->qfull_timer);
3306
3307	mutex_unlock(&udev->cmdr_lock);
3308
3309	spin_lock_bh(&timed_out_udevs_lock);
3310	}
3311
3312	spin_unlock_bh(&timed_out_udevs_lock);
3313	}
3314
3315	static void tcmu_unmap_work_fn(struct work_struct *work)
3316	{
3317	check_timedout_devices();
3318	find_free_blocks();
3319	}
3320
3321	static int __init tcmu_module_init(void)
3322	{
3323	int ret, i, k, len = 0;
3324
3325	BUILD_BUG_ON((sizeof(struct tcmu_cmd_entry) % TCMU_OP_ALIGN_SIZE) != 0);
3326
3327	INIT_DELAYED_WORK(&tcmu_unmap_work, tcmu_unmap_work_fn);
3328
3329	tcmu_cmd_cache = kmem_cache_create("tcmu_cmd_cache",
3330	sizeof(struct tcmu_cmd),
3331	__alignof__(struct tcmu_cmd),
3332	0, NULL);
3333	if (!tcmu_cmd_cache)
3334	return -ENOMEM;
3335
3336	tcmu_root_device = root_device_register("tcm_user");
3337	if (IS_ERR(tcmu_root_device)) {
3338	ret = PTR_ERR(tcmu_root_device);
3339	goto out_free_cache;
3340	}
3341
3342	ret = genl_register_family(&tcmu_genl_family);
3343	if (ret < 0) {
3344	goto out_unreg_device;
3345	}
3346
3347	for (i = 0; passthrough_attrib_attrs[i] != NULL; i++)
3348	len += sizeof(struct configfs_attribute *);
3349	for (i = 0; passthrough_pr_attrib_attrs[i] != NULL; i++)
3350	len += sizeof(struct configfs_attribute *);
3351	for (i = 0; tcmu_attrib_attrs[i] != NULL; i++)
3352	len += sizeof(struct configfs_attribute *);
3353	len += sizeof(struct configfs_attribute *);
3354
3355	tcmu_attrs = kzalloc(len, GFP_KERNEL);
3356	if (!tcmu_attrs) {
3357	ret = -ENOMEM;
3358	goto out_unreg_genl;
3359	}
3360
3361	for (i = 0; passthrough_attrib_attrs[i] != NULL; i++)
3362	tcmu_attrs[i] = passthrough_attrib_attrs[i];
3363	for (k = 0; passthrough_pr_attrib_attrs[k] != NULL; k++)
3364	tcmu_attrs[i++] = passthrough_pr_attrib_attrs[k];
3365	for (k = 0; tcmu_attrib_attrs[k] != NULL; k++)
3366	tcmu_attrs[i++] = tcmu_attrib_attrs[k];
3367	tcmu_ops.tb_dev_attrib_attrs = tcmu_attrs;
3368
3369	ret = transport_backend_register(&tcmu_ops);
3370	if (ret)
3371	goto out_attrs;
3372
3373	return 0;
3374
3375	out_attrs:
3376	kfree(tcmu_attrs);
3377	out_unreg_genl:
3378	genl_unregister_family(&tcmu_genl_family);
3379	out_unreg_device:
3380	root_device_unregister(tcmu_root_device);
3381	out_free_cache:
3382	kmem_cache_destroy(tcmu_cmd_cache);
3383
3384	return ret;
3385	}
3386
3387	static void __exit tcmu_module_exit(void)
3388	{
3389	cancel_delayed_work_sync(&tcmu_unmap_work);
3390	target_backend_unregister(&tcmu_ops);
3391	kfree(tcmu_attrs);
3392	genl_unregister_family(&tcmu_genl_family);
3393	root_device_unregister(tcmu_root_device);
3394	kmem_cache_destroy(tcmu_cmd_cache);
3395	}
3396
3397	MODULE_DESCRIPTION("TCM USER subsystem plugin");
3398	MODULE_AUTHOR("Shaohua Li <[email protected]>");
3399	MODULE_AUTHOR("Andy Grover <[email protected]>");
3400	MODULE_LICENSE("GPL");
3401
3402	module_init(tcmu_module_init);
3403	module_exit(tcmu_module_exit);