Linux Audio

Check our new training course

Loading...
v6.8
   1// SPDX-License-Identifier: GPL-2.0+
   2/*******************************************************************************
   3 * Vhost kernel TCM fabric driver for virtio SCSI initiators
   4 *
   5 * (C) Copyright 2010-2013 Datera, Inc.
   6 * (C) Copyright 2010-2012 IBM Corp.
   7 *
 
 
   8 * Authors: Nicholas A. Bellinger <nab@daterainc.com>
   9 *          Stefan Hajnoczi <stefanha@linux.vnet.ibm.com>
 
 
 
 
 
 
 
 
 
 
 
  10 ****************************************************************************/
  11
  12#include <linux/module.h>
  13#include <linux/moduleparam.h>
  14#include <generated/utsrelease.h>
  15#include <linux/utsname.h>
  16#include <linux/init.h>
  17#include <linux/slab.h>
  18#include <linux/kthread.h>
  19#include <linux/types.h>
  20#include <linux/string.h>
  21#include <linux/configfs.h>
  22#include <linux/ctype.h>
  23#include <linux/compat.h>
  24#include <linux/eventfd.h>
  25#include <linux/fs.h>
  26#include <linux/vmalloc.h>
  27#include <linux/miscdevice.h>
  28#include <linux/blk_types.h>
  29#include <linux/bio.h>
  30#include <asm/unaligned.h>
  31#include <scsi/scsi_common.h>
  32#include <scsi/scsi_proto.h>
  33#include <target/target_core_base.h>
  34#include <target/target_core_fabric.h>
  35#include <linux/vhost.h>
  36#include <linux/virtio_scsi.h>
  37#include <linux/llist.h>
  38#include <linux/bitmap.h>
  39
  40#include "vhost.h"
  41
  42#define VHOST_SCSI_VERSION  "v0.1"
  43#define VHOST_SCSI_NAMELEN 256
  44#define VHOST_SCSI_MAX_CDB_SIZE 32
  45#define VHOST_SCSI_PREALLOC_SGLS 2048
  46#define VHOST_SCSI_PREALLOC_UPAGES 2048
  47#define VHOST_SCSI_PREALLOC_PROT_SGLS 2048
  48
  49/* Max number of requests before requeueing the job.
  50 * Using this limit prevents one virtqueue from starving others with
  51 * request.
  52 */
  53#define VHOST_SCSI_WEIGHT 256
  54
  55struct vhost_scsi_inflight {
  56	/* Wait for the flush operation to finish */
  57	struct completion comp;
  58	/* Refcount for the inflight reqs */
  59	struct kref kref;
  60};
  61
  62struct vhost_scsi_cmd {
  63	/* Descriptor from vhost_get_vq_desc() for virt_queue segment */
  64	int tvc_vq_desc;
  65	/* virtio-scsi initiator task attribute */
  66	int tvc_task_attr;
  67	/* virtio-scsi response incoming iovecs */
  68	int tvc_in_iovs;
  69	/* virtio-scsi initiator data direction */
  70	enum dma_data_direction tvc_data_direction;
  71	/* Expected data transfer length from virtio-scsi header */
  72	u32 tvc_exp_data_len;
  73	/* The Tag from include/linux/virtio_scsi.h:struct virtio_scsi_cmd_req */
  74	u64 tvc_tag;
  75	/* The number of scatterlists associated with this cmd */
  76	u32 tvc_sgl_count;
  77	u32 tvc_prot_sgl_count;
  78	/* Saved unpacked SCSI LUN for vhost_scsi_target_queue_cmd() */
  79	u32 tvc_lun;
  80	u32 copied_iov:1;
  81	const void *saved_iter_addr;
  82	struct iov_iter saved_iter;
  83	/* Pointer to the SGL formatted memory from virtio-scsi */
  84	struct scatterlist *tvc_sgl;
  85	struct scatterlist *tvc_prot_sgl;
  86	struct page **tvc_upages;
  87	/* Pointer to response header iovec */
  88	struct iovec *tvc_resp_iov;
  89	/* Pointer to vhost_scsi for our device */
  90	struct vhost_scsi *tvc_vhost;
  91	/* Pointer to vhost_virtqueue for the cmd */
  92	struct vhost_virtqueue *tvc_vq;
  93	/* Pointer to vhost nexus memory */
  94	struct vhost_scsi_nexus *tvc_nexus;
  95	/* The TCM I/O descriptor that is accessed via container_of() */
  96	struct se_cmd tvc_se_cmd;
  97	/* Copy of the incoming SCSI command descriptor block (CDB) */
  98	unsigned char tvc_cdb[VHOST_SCSI_MAX_CDB_SIZE];
  99	/* Sense buffer that will be mapped into outgoing status */
 100	unsigned char tvc_sense_buf[TRANSPORT_SENSE_BUFFER];
 101	/* Completed commands list, serviced from vhost worker thread */
 102	struct llist_node tvc_completion_list;
 103	/* Used to track inflight cmd */
 104	struct vhost_scsi_inflight *inflight;
 105};
 106
 107struct vhost_scsi_nexus {
 108	/* Pointer to TCM session for I_T Nexus */
 109	struct se_session *tvn_se_sess;
 110};
 111
 112struct vhost_scsi_tpg {
 113	/* Vhost port target portal group tag for TCM */
 114	u16 tport_tpgt;
 115	/* Used to track number of TPG Port/Lun Links wrt to explict I_T Nexus shutdown */
 116	int tv_tpg_port_count;
 117	/* Used for vhost_scsi device reference to tpg_nexus, protected by tv_tpg_mutex */
 118	int tv_tpg_vhost_count;
 119	/* Used for enabling T10-PI with legacy devices */
 120	int tv_fabric_prot_type;
 121	/* list for vhost_scsi_list */
 122	struct list_head tv_tpg_list;
 123	/* Used to protect access for tpg_nexus */
 124	struct mutex tv_tpg_mutex;
 125	/* Pointer to the TCM VHost I_T Nexus for this TPG endpoint */
 126	struct vhost_scsi_nexus *tpg_nexus;
 127	/* Pointer back to vhost_scsi_tport */
 128	struct vhost_scsi_tport *tport;
 129	/* Returned by vhost_scsi_make_tpg() */
 130	struct se_portal_group se_tpg;
 131	/* Pointer back to vhost_scsi, protected by tv_tpg_mutex */
 132	struct vhost_scsi *vhost_scsi;
 
 133};
 134
 135struct vhost_scsi_tport {
 136	/* SCSI protocol the tport is providing */
 137	u8 tport_proto_id;
 138	/* Binary World Wide unique Port Name for Vhost Target port */
 139	u64 tport_wwpn;
 140	/* ASCII formatted WWPN for Vhost Target port */
 141	char tport_name[VHOST_SCSI_NAMELEN];
 142	/* Returned by vhost_scsi_make_tport() */
 143	struct se_wwn tport_wwn;
 144};
 145
 146struct vhost_scsi_evt {
 147	/* event to be sent to guest */
 148	struct virtio_scsi_event event;
 149	/* event list, serviced from vhost worker thread */
 150	struct llist_node list;
 151};
 152
 153enum {
 154	VHOST_SCSI_VQ_CTL = 0,
 155	VHOST_SCSI_VQ_EVT = 1,
 156	VHOST_SCSI_VQ_IO = 2,
 157};
 158
 159/* Note: can't set VIRTIO_F_VERSION_1 yet, since that implies ANY_LAYOUT. */
 160enum {
 161	VHOST_SCSI_FEATURES = VHOST_FEATURES | (1ULL << VIRTIO_SCSI_F_HOTPLUG) |
 162					       (1ULL << VIRTIO_SCSI_F_T10_PI)
 163};
 164
 165#define VHOST_SCSI_MAX_TARGET	256
 166#define VHOST_SCSI_MAX_IO_VQ	1024
 167#define VHOST_SCSI_MAX_EVENT	128
 168
 169static unsigned vhost_scsi_max_io_vqs = 128;
 170module_param_named(max_io_vqs, vhost_scsi_max_io_vqs, uint, 0644);
 171MODULE_PARM_DESC(max_io_vqs, "Set the max number of IO virtqueues a vhost scsi device can support. The default is 128. The max is 1024.");
 172
 173struct vhost_scsi_virtqueue {
 174	struct vhost_virtqueue vq;
 175	struct vhost_scsi *vs;
 176	/*
 177	 * Reference counting for inflight reqs, used for flush operation. At
 178	 * each time, one reference tracks new commands submitted, while we
 179	 * wait for another one to reach 0.
 180	 */
 181	struct vhost_scsi_inflight inflights[2];
 182	/*
 183	 * Indicate current inflight in use, protected by vq->mutex.
 184	 * Writers must also take dev mutex and flush under it.
 185	 */
 186	int inflight_idx;
 187	struct vhost_scsi_cmd *scsi_cmds;
 188	struct sbitmap scsi_tags;
 189	int max_cmds;
 190
 191	struct vhost_work completion_work;
 192	struct llist_head completion_list;
 193};
 194
 195struct vhost_scsi {
 196	/* Protected by vhost_scsi->dev.mutex */
 197	struct vhost_scsi_tpg **vs_tpg;
 198	char vs_vhost_wwpn[TRANSPORT_IQN_LEN];
 199
 200	struct vhost_dev dev;
 201	struct vhost_scsi_virtqueue *vqs;
 202	struct vhost_scsi_inflight **old_inflight;
 
 
 203
 204	struct vhost_work vs_event_work; /* evt injection work item */
 205	struct llist_head vs_event_list; /* evt injection queue */
 206
 207	bool vs_events_missed; /* any missed events, protected by vq->mutex */
 208	int vs_events_nr; /* num of pending events, protected by vq->mutex */
 209};
 210
 211struct vhost_scsi_tmf {
 212	struct vhost_work vwork;
 
 213	struct vhost_scsi *vhost;
 214	struct vhost_scsi_virtqueue *svq;
 
 215
 216	struct se_cmd se_cmd;
 217	u8 scsi_resp;
 218	struct vhost_scsi_inflight *inflight;
 219	struct iovec resp_iov;
 220	int in_iovs;
 221	int vq_desc;
 222};
 223
 224/*
 225 * Context for processing request and control queue operations.
 226 */
 227struct vhost_scsi_ctx {
 228	int head;
 229	unsigned int out, in;
 230	size_t req_size, rsp_size;
 231	size_t out_size, in_size;
 232	u8 *target, *lunp;
 233	void *req;
 234	struct iov_iter out_iter;
 235};
 236
 237/*
 238 * Global mutex to protect vhost_scsi TPG list for vhost IOCTLs and LIO
 239 * configfs management operations.
 240 */
 241static DEFINE_MUTEX(vhost_scsi_mutex);
 242static LIST_HEAD(vhost_scsi_list);
 243
 244static void vhost_scsi_done_inflight(struct kref *kref)
 245{
 246	struct vhost_scsi_inflight *inflight;
 247
 248	inflight = container_of(kref, struct vhost_scsi_inflight, kref);
 249	complete(&inflight->comp);
 250}
 251
 252static void vhost_scsi_init_inflight(struct vhost_scsi *vs,
 253				    struct vhost_scsi_inflight *old_inflight[])
 254{
 255	struct vhost_scsi_inflight *new_inflight;
 256	struct vhost_virtqueue *vq;
 257	int idx, i;
 258
 259	for (i = 0; i < vs->dev.nvqs;  i++) {
 260		vq = &vs->vqs[i].vq;
 261
 262		mutex_lock(&vq->mutex);
 263
 264		/* store old infight */
 265		idx = vs->vqs[i].inflight_idx;
 266		if (old_inflight)
 267			old_inflight[i] = &vs->vqs[i].inflights[idx];
 268
 269		/* setup new infight */
 270		vs->vqs[i].inflight_idx = idx ^ 1;
 271		new_inflight = &vs->vqs[i].inflights[idx ^ 1];
 272		kref_init(&new_inflight->kref);
 273		init_completion(&new_inflight->comp);
 274
 275		mutex_unlock(&vq->mutex);
 276	}
 277}
 278
 279static struct vhost_scsi_inflight *
 280vhost_scsi_get_inflight(struct vhost_virtqueue *vq)
 281{
 282	struct vhost_scsi_inflight *inflight;
 283	struct vhost_scsi_virtqueue *svq;
 284
 285	svq = container_of(vq, struct vhost_scsi_virtqueue, vq);
 286	inflight = &svq->inflights[svq->inflight_idx];
 287	kref_get(&inflight->kref);
 288
 289	return inflight;
 290}
 291
 292static void vhost_scsi_put_inflight(struct vhost_scsi_inflight *inflight)
 293{
 294	kref_put(&inflight->kref, vhost_scsi_done_inflight);
 295}
 296
 297static int vhost_scsi_check_true(struct se_portal_group *se_tpg)
 298{
 299	return 1;
 300}
 301
 
 
 
 
 
 302static char *vhost_scsi_get_fabric_wwn(struct se_portal_group *se_tpg)
 303{
 304	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
 305				struct vhost_scsi_tpg, se_tpg);
 306	struct vhost_scsi_tport *tport = tpg->tport;
 307
 308	return &tport->tport_name[0];
 309}
 310
 311static u16 vhost_scsi_get_tpgt(struct se_portal_group *se_tpg)
 312{
 313	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
 314				struct vhost_scsi_tpg, se_tpg);
 315	return tpg->tport_tpgt;
 316}
 317
 318static int vhost_scsi_check_prot_fabric_only(struct se_portal_group *se_tpg)
 319{
 320	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
 321				struct vhost_scsi_tpg, se_tpg);
 322
 323	return tpg->tv_fabric_prot_type;
 324}
 325
 
 
 
 
 
 326static void vhost_scsi_release_cmd_res(struct se_cmd *se_cmd)
 327{
 328	struct vhost_scsi_cmd *tv_cmd = container_of(se_cmd,
 329				struct vhost_scsi_cmd, tvc_se_cmd);
 330	struct vhost_scsi_virtqueue *svq = container_of(tv_cmd->tvc_vq,
 331				struct vhost_scsi_virtqueue, vq);
 332	struct vhost_scsi_inflight *inflight = tv_cmd->inflight;
 333	int i;
 334
 335	if (tv_cmd->tvc_sgl_count) {
 336		for (i = 0; i < tv_cmd->tvc_sgl_count; i++) {
 337			if (tv_cmd->copied_iov)
 338				__free_page(sg_page(&tv_cmd->tvc_sgl[i]));
 339			else
 340				put_page(sg_page(&tv_cmd->tvc_sgl[i]));
 341		}
 342		kfree(tv_cmd->saved_iter_addr);
 343	}
 344	if (tv_cmd->tvc_prot_sgl_count) {
 345		for (i = 0; i < tv_cmd->tvc_prot_sgl_count; i++)
 346			put_page(sg_page(&tv_cmd->tvc_prot_sgl[i]));
 347	}
 348
 349	sbitmap_clear_bit(&svq->scsi_tags, se_cmd->map_tag);
 350	vhost_scsi_put_inflight(inflight);
 351}
 352
 353static void vhost_scsi_release_tmf_res(struct vhost_scsi_tmf *tmf)
 354{
 
 355	struct vhost_scsi_inflight *inflight = tmf->inflight;
 356
 357	kfree(tmf);
 
 
 358	vhost_scsi_put_inflight(inflight);
 359}
 360
 361static void vhost_scsi_release_cmd(struct se_cmd *se_cmd)
 362{
 363	if (se_cmd->se_cmd_flags & SCF_SCSI_TMR_CDB) {
 364		struct vhost_scsi_tmf *tmf = container_of(se_cmd,
 365					struct vhost_scsi_tmf, se_cmd);
 366		struct vhost_virtqueue *vq = &tmf->svq->vq;
 367
 368		vhost_vq_work_queue(vq, &tmf->vwork);
 369	} else {
 370		struct vhost_scsi_cmd *cmd = container_of(se_cmd,
 371					struct vhost_scsi_cmd, tvc_se_cmd);
 372		struct vhost_scsi_virtqueue *svq =  container_of(cmd->tvc_vq,
 373					struct vhost_scsi_virtqueue, vq);
 374
 375		llist_add(&cmd->tvc_completion_list, &svq->completion_list);
 376		vhost_vq_work_queue(&svq->vq, &svq->completion_work);
 377	}
 378}
 379
 
 
 
 
 
 380static int vhost_scsi_write_pending(struct se_cmd *se_cmd)
 381{
 382	/* Go ahead and process the write immediately */
 383	target_execute_cmd(se_cmd);
 384	return 0;
 385}
 386
 
 
 
 
 
 
 
 
 
 
 387static int vhost_scsi_queue_data_in(struct se_cmd *se_cmd)
 388{
 389	transport_generic_free_cmd(se_cmd, 0);
 390	return 0;
 391}
 392
 393static int vhost_scsi_queue_status(struct se_cmd *se_cmd)
 394{
 395	transport_generic_free_cmd(se_cmd, 0);
 396	return 0;
 397}
 398
 399static void vhost_scsi_queue_tm_rsp(struct se_cmd *se_cmd)
 400{
 401	struct vhost_scsi_tmf *tmf = container_of(se_cmd, struct vhost_scsi_tmf,
 402						  se_cmd);
 403
 404	tmf->scsi_resp = se_cmd->se_tmr_req->response;
 405	transport_generic_free_cmd(&tmf->se_cmd, 0);
 406}
 407
 408static void vhost_scsi_aborted_task(struct se_cmd *se_cmd)
 409{
 410	return;
 411}
 412
 413static void vhost_scsi_free_evt(struct vhost_scsi *vs, struct vhost_scsi_evt *evt)
 414{
 415	vs->vs_events_nr--;
 416	kfree(evt);
 417}
 418
 419static struct vhost_scsi_evt *
 420vhost_scsi_allocate_evt(struct vhost_scsi *vs,
 421		       u32 event, u32 reason)
 422{
 423	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
 424	struct vhost_scsi_evt *evt;
 425
 426	if (vs->vs_events_nr > VHOST_SCSI_MAX_EVENT) {
 427		vs->vs_events_missed = true;
 428		return NULL;
 429	}
 430
 431	evt = kzalloc(sizeof(*evt), GFP_KERNEL);
 432	if (!evt) {
 433		vq_err(vq, "Failed to allocate vhost_scsi_evt\n");
 434		vs->vs_events_missed = true;
 435		return NULL;
 436	}
 437
 438	evt->event.event = cpu_to_vhost32(vq, event);
 439	evt->event.reason = cpu_to_vhost32(vq, reason);
 440	vs->vs_events_nr++;
 441
 442	return evt;
 443}
 444
 445static int vhost_scsi_check_stop_free(struct se_cmd *se_cmd)
 446{
 447	return target_put_sess_cmd(se_cmd);
 448}
 449
 450static void
 451vhost_scsi_do_evt_work(struct vhost_scsi *vs, struct vhost_scsi_evt *evt)
 452{
 453	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
 454	struct virtio_scsi_event *event = &evt->event;
 455	struct virtio_scsi_event __user *eventp;
 456	unsigned out, in;
 457	int head, ret;
 458
 459	if (!vhost_vq_get_backend(vq)) {
 460		vs->vs_events_missed = true;
 461		return;
 462	}
 463
 464again:
 465	vhost_disable_notify(&vs->dev, vq);
 466	head = vhost_get_vq_desc(vq, vq->iov,
 467			ARRAY_SIZE(vq->iov), &out, &in,
 468			NULL, NULL);
 469	if (head < 0) {
 470		vs->vs_events_missed = true;
 471		return;
 472	}
 473	if (head == vq->num) {
 474		if (vhost_enable_notify(&vs->dev, vq))
 475			goto again;
 476		vs->vs_events_missed = true;
 477		return;
 478	}
 479
 480	if ((vq->iov[out].iov_len != sizeof(struct virtio_scsi_event))) {
 481		vq_err(vq, "Expecting virtio_scsi_event, got %zu bytes\n",
 482				vq->iov[out].iov_len);
 483		vs->vs_events_missed = true;
 484		return;
 485	}
 486
 487	if (vs->vs_events_missed) {
 488		event->event |= cpu_to_vhost32(vq, VIRTIO_SCSI_T_EVENTS_MISSED);
 489		vs->vs_events_missed = false;
 490	}
 491
 492	eventp = vq->iov[out].iov_base;
 493	ret = __copy_to_user(eventp, event, sizeof(*event));
 494	if (!ret)
 495		vhost_add_used_and_signal(&vs->dev, vq, head, 0);
 496	else
 497		vq_err(vq, "Faulted on vhost_scsi_send_event\n");
 498}
 499
 500static void vhost_scsi_evt_work(struct vhost_work *work)
 501{
 502	struct vhost_scsi *vs = container_of(work, struct vhost_scsi,
 503					vs_event_work);
 504	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
 505	struct vhost_scsi_evt *evt, *t;
 506	struct llist_node *llnode;
 507
 508	mutex_lock(&vq->mutex);
 509	llnode = llist_del_all(&vs->vs_event_list);
 510	llist_for_each_entry_safe(evt, t, llnode, list) {
 511		vhost_scsi_do_evt_work(vs, evt);
 512		vhost_scsi_free_evt(vs, evt);
 513	}
 514	mutex_unlock(&vq->mutex);
 515}
 516
 517static int vhost_scsi_copy_sgl_to_iov(struct vhost_scsi_cmd *cmd)
 518{
 519	struct iov_iter *iter = &cmd->saved_iter;
 520	struct scatterlist *sg = cmd->tvc_sgl;
 521	struct page *page;
 522	size_t len;
 523	int i;
 524
 525	for (i = 0; i < cmd->tvc_sgl_count; i++) {
 526		page = sg_page(&sg[i]);
 527		len = sg[i].length;
 528
 529		if (copy_page_to_iter(page, 0, len, iter) != len) {
 530			pr_err("Could not copy data while handling misaligned cmd. Error %zu\n",
 531			       len);
 532			return -1;
 533		}
 534	}
 535
 536	return 0;
 537}
 538
 539/* Fill in status and signal that we are done processing this command
 540 *
 541 * This is scheduled in the vhost work queue so we are called with the owner
 542 * process mm and can access the vring.
 543 */
 544static void vhost_scsi_complete_cmd_work(struct vhost_work *work)
 545{
 546	struct vhost_scsi_virtqueue *svq = container_of(work,
 547				struct vhost_scsi_virtqueue, completion_work);
 
 548	struct virtio_scsi_cmd_resp v_rsp;
 549	struct vhost_scsi_cmd *cmd, *t;
 550	struct llist_node *llnode;
 551	struct se_cmd *se_cmd;
 552	struct iov_iter iov_iter;
 553	bool signal = false;
 554	int ret;
 555
 556	llnode = llist_del_all(&svq->completion_list);
 
 557	llist_for_each_entry_safe(cmd, t, llnode, tvc_completion_list) {
 558		se_cmd = &cmd->tvc_se_cmd;
 559
 560		pr_debug("%s tv_cmd %p resid %u status %#02x\n", __func__,
 561			cmd, se_cmd->residual_count, se_cmd->scsi_status);
 562		memset(&v_rsp, 0, sizeof(v_rsp));
 563
 564		if (cmd->saved_iter_addr && vhost_scsi_copy_sgl_to_iov(cmd)) {
 565			v_rsp.response = VIRTIO_SCSI_S_BAD_TARGET;
 566		} else {
 567			v_rsp.resid = cpu_to_vhost32(cmd->tvc_vq,
 568						     se_cmd->residual_count);
 569			/* TODO is status_qualifier field needed? */
 570			v_rsp.status = se_cmd->scsi_status;
 571			v_rsp.sense_len = cpu_to_vhost32(cmd->tvc_vq,
 572							 se_cmd->scsi_sense_length);
 573			memcpy(v_rsp.sense, cmd->tvc_sense_buf,
 574			       se_cmd->scsi_sense_length);
 575		}
 576
 577		iov_iter_init(&iov_iter, ITER_DEST, cmd->tvc_resp_iov,
 578			      cmd->tvc_in_iovs, sizeof(v_rsp));
 579		ret = copy_to_iter(&v_rsp, sizeof(v_rsp), &iov_iter);
 580		if (likely(ret == sizeof(v_rsp))) {
 581			signal = true;
 582
 583			vhost_add_used(cmd->tvc_vq, cmd->tvc_vq_desc, 0);
 
 
 
 584		} else
 585			pr_err("Faulted on virtio_scsi_cmd_resp\n");
 586
 587		vhost_scsi_release_cmd_res(se_cmd);
 588	}
 589
 590	if (signal)
 591		vhost_signal(&svq->vs->dev, &svq->vq);
 
 
 592}
 593
 594static struct vhost_scsi_cmd *
 595vhost_scsi_get_cmd(struct vhost_virtqueue *vq, struct vhost_scsi_tpg *tpg,
 596		   unsigned char *cdb, u64 scsi_tag, u16 lun, u8 task_attr,
 597		   u32 exp_data_len, int data_direction)
 598{
 599	struct vhost_scsi_virtqueue *svq = container_of(vq,
 600					struct vhost_scsi_virtqueue, vq);
 601	struct vhost_scsi_cmd *cmd;
 602	struct vhost_scsi_nexus *tv_nexus;
 603	struct scatterlist *sg, *prot_sg;
 604	struct iovec *tvc_resp_iov;
 605	struct page **pages;
 606	int tag;
 607
 608	tv_nexus = tpg->tpg_nexus;
 609	if (!tv_nexus) {
 610		pr_err("Unable to locate active struct vhost_scsi_nexus\n");
 611		return ERR_PTR(-EIO);
 612	}
 613
 614	tag = sbitmap_get(&svq->scsi_tags);
 615	if (tag < 0) {
 616		pr_err("Unable to obtain tag for vhost_scsi_cmd\n");
 617		return ERR_PTR(-ENOMEM);
 618	}
 619
 620	cmd = &svq->scsi_cmds[tag];
 621	sg = cmd->tvc_sgl;
 622	prot_sg = cmd->tvc_prot_sgl;
 623	pages = cmd->tvc_upages;
 624	tvc_resp_iov = cmd->tvc_resp_iov;
 625	memset(cmd, 0, sizeof(*cmd));
 626	cmd->tvc_sgl = sg;
 627	cmd->tvc_prot_sgl = prot_sg;
 628	cmd->tvc_upages = pages;
 629	cmd->tvc_se_cmd.map_tag = tag;
 630	cmd->tvc_tag = scsi_tag;
 631	cmd->tvc_lun = lun;
 632	cmd->tvc_task_attr = task_attr;
 633	cmd->tvc_exp_data_len = exp_data_len;
 634	cmd->tvc_data_direction = data_direction;
 635	cmd->tvc_nexus = tv_nexus;
 636	cmd->inflight = vhost_scsi_get_inflight(vq);
 637	cmd->tvc_resp_iov = tvc_resp_iov;
 638
 639	memcpy(cmd->tvc_cdb, cdb, VHOST_SCSI_MAX_CDB_SIZE);
 640
 641	return cmd;
 642}
 643
 644/*
 645 * Map a user memory range into a scatterlist
 646 *
 647 * Returns the number of scatterlist entries used or -errno on error.
 648 */
 649static int
 650vhost_scsi_map_to_sgl(struct vhost_scsi_cmd *cmd,
 651		      struct iov_iter *iter,
 652		      struct scatterlist *sgl,
 653		      bool is_prot)
 654{
 655	struct page **pages = cmd->tvc_upages;
 656	struct scatterlist *sg = sgl;
 657	ssize_t bytes, mapped_bytes;
 658	size_t offset, mapped_offset;
 659	unsigned int npages = 0;
 660
 661	bytes = iov_iter_get_pages2(iter, pages, LONG_MAX,
 662				VHOST_SCSI_PREALLOC_UPAGES, &offset);
 663	/* No pages were pinned */
 664	if (bytes <= 0)
 665		return bytes < 0 ? bytes : -EFAULT;
 666
 667	mapped_bytes = bytes;
 668	mapped_offset = offset;
 669
 670	while (bytes) {
 671		unsigned n = min_t(unsigned, PAGE_SIZE - offset, bytes);
 672		/*
 673		 * The block layer requires bios/requests to be a multiple of
 674		 * 512 bytes, but Windows can send us vecs that are misaligned.
 675		 * This can result in bios and later requests with misaligned
 676		 * sizes if we have to break up a cmd/scatterlist into multiple
 677		 * bios.
 678		 *
 679		 * We currently only break up a command into multiple bios if
 680		 * we hit the vec/seg limit, so check if our sgl_count is
 681		 * greater than the max and if a vec in the cmd has a
 682		 * misaligned offset/size.
 683		 */
 684		if (!is_prot &&
 685		    (offset & (SECTOR_SIZE - 1) || n & (SECTOR_SIZE - 1)) &&
 686		    cmd->tvc_sgl_count > BIO_MAX_VECS) {
 687			WARN_ONCE(true,
 688				  "vhost-scsi detected misaligned IO. Performance may be degraded.");
 689			goto revert_iter_get_pages;
 690		}
 691
 692		sg_set_page(sg++, pages[npages++], n, offset);
 693		bytes -= n;
 694		offset = 0;
 695	}
 696
 697	return npages;
 698
 699revert_iter_get_pages:
 700	iov_iter_revert(iter, mapped_bytes);
 701
 702	npages = 0;
 703	while (mapped_bytes) {
 704		unsigned int n = min_t(unsigned int, PAGE_SIZE - mapped_offset,
 705				       mapped_bytes);
 706
 707		put_page(pages[npages++]);
 708
 709		mapped_bytes -= n;
 710		mapped_offset = 0;
 711	}
 712
 713	return -EINVAL;
 714}
 715
 716static int
 717vhost_scsi_calc_sgls(struct iov_iter *iter, size_t bytes, int max_sgls)
 718{
 719	int sgl_count = 0;
 720
 721	if (!iter || !iter_iov(iter)) {
 722		pr_err("%s: iter->iov is NULL, but expected bytes: %zu"
 723		       " present\n", __func__, bytes);
 724		return -EINVAL;
 725	}
 726
 727	sgl_count = iov_iter_npages(iter, 0xffff);
 728	if (sgl_count > max_sgls) {
 729		pr_err("%s: requested sgl_count: %d exceeds pre-allocated"
 730		       " max_sgls: %d\n", __func__, sgl_count, max_sgls);
 731		return -EINVAL;
 732	}
 733	return sgl_count;
 734}
 735
 736static int
 737vhost_scsi_copy_iov_to_sgl(struct vhost_scsi_cmd *cmd, struct iov_iter *iter,
 738			   struct scatterlist *sg, int sg_count)
 739{
 740	size_t len = iov_iter_count(iter);
 741	unsigned int nbytes = 0;
 742	struct page *page;
 743	int i;
 744
 745	if (cmd->tvc_data_direction == DMA_FROM_DEVICE) {
 746		cmd->saved_iter_addr = dup_iter(&cmd->saved_iter, iter,
 747						GFP_KERNEL);
 748		if (!cmd->saved_iter_addr)
 749			return -ENOMEM;
 750	}
 751
 752	for (i = 0; i < sg_count; i++) {
 753		page = alloc_page(GFP_KERNEL);
 754		if (!page) {
 755			i--;
 756			goto err;
 757		}
 758
 759		nbytes = min_t(unsigned int, PAGE_SIZE, len);
 760		sg_set_page(&sg[i], page, nbytes, 0);
 761
 762		if (cmd->tvc_data_direction == DMA_TO_DEVICE &&
 763		    copy_page_from_iter(page, 0, nbytes, iter) != nbytes)
 764			goto err;
 765
 766		len -= nbytes;
 767	}
 768
 769	cmd->copied_iov = 1;
 770	return 0;
 771
 772err:
 773	pr_err("Could not read %u bytes while handling misaligned cmd\n",
 774	       nbytes);
 775
 776	for (; i >= 0; i--)
 777		__free_page(sg_page(&sg[i]));
 778	kfree(cmd->saved_iter_addr);
 779	return -ENOMEM;
 780}
 781
 782static int
 783vhost_scsi_map_iov_to_sgl(struct vhost_scsi_cmd *cmd, struct iov_iter *iter,
 784			  struct scatterlist *sg, int sg_count, bool is_prot)
 785{
 786	struct scatterlist *p = sg;
 787	size_t revert_bytes;
 788	int ret;
 789
 790	while (iov_iter_count(iter)) {
 791		ret = vhost_scsi_map_to_sgl(cmd, iter, sg, is_prot);
 792		if (ret < 0) {
 793			revert_bytes = 0;
 794
 795			while (p < sg) {
 796				struct page *page = sg_page(p);
 797
 798				if (page) {
 799					put_page(page);
 800					revert_bytes += p->length;
 801				}
 802				p++;
 803			}
 804
 805			iov_iter_revert(iter, revert_bytes);
 806			return ret;
 807		}
 808		sg += ret;
 809	}
 810
 811	return 0;
 812}
 813
 814static int
 815vhost_scsi_mapal(struct vhost_scsi_cmd *cmd,
 816		 size_t prot_bytes, struct iov_iter *prot_iter,
 817		 size_t data_bytes, struct iov_iter *data_iter)
 818{
 819	int sgl_count, ret;
 
 820
 821	if (prot_bytes) {
 822		sgl_count = vhost_scsi_calc_sgls(prot_iter, prot_bytes,
 823						 VHOST_SCSI_PREALLOC_PROT_SGLS);
 824		if (sgl_count < 0)
 825			return sgl_count;
 826
 827		sg_init_table(cmd->tvc_prot_sgl, sgl_count);
 828		cmd->tvc_prot_sgl_count = sgl_count;
 829		pr_debug("%s prot_sg %p prot_sgl_count %u\n", __func__,
 830			 cmd->tvc_prot_sgl, cmd->tvc_prot_sgl_count);
 831
 832		ret = vhost_scsi_map_iov_to_sgl(cmd, prot_iter,
 833						cmd->tvc_prot_sgl,
 834						cmd->tvc_prot_sgl_count, true);
 835		if (ret < 0) {
 836			cmd->tvc_prot_sgl_count = 0;
 837			return ret;
 838		}
 839	}
 840	sgl_count = vhost_scsi_calc_sgls(data_iter, data_bytes,
 841					 VHOST_SCSI_PREALLOC_SGLS);
 842	if (sgl_count < 0)
 843		return sgl_count;
 844
 845	sg_init_table(cmd->tvc_sgl, sgl_count);
 846	cmd->tvc_sgl_count = sgl_count;
 847	pr_debug("%s data_sg %p data_sgl_count %u\n", __func__,
 848		  cmd->tvc_sgl, cmd->tvc_sgl_count);
 849
 850	ret = vhost_scsi_map_iov_to_sgl(cmd, data_iter, cmd->tvc_sgl,
 851					cmd->tvc_sgl_count, false);
 852	if (ret == -EINVAL) {
 853		sg_init_table(cmd->tvc_sgl, cmd->tvc_sgl_count);
 854		ret = vhost_scsi_copy_iov_to_sgl(cmd, data_iter, cmd->tvc_sgl,
 855						 cmd->tvc_sgl_count);
 856	}
 857
 858	if (ret < 0) {
 859		cmd->tvc_sgl_count = 0;
 860		return ret;
 861	}
 862	return 0;
 863}
 864
 865static int vhost_scsi_to_tcm_attr(int attr)
 866{
 867	switch (attr) {
 868	case VIRTIO_SCSI_S_SIMPLE:
 869		return TCM_SIMPLE_TAG;
 870	case VIRTIO_SCSI_S_ORDERED:
 871		return TCM_ORDERED_TAG;
 872	case VIRTIO_SCSI_S_HEAD:
 873		return TCM_HEAD_TAG;
 874	case VIRTIO_SCSI_S_ACA:
 875		return TCM_ACA_TAG;
 876	default:
 877		break;
 878	}
 879	return TCM_SIMPLE_TAG;
 880}
 881
 882static void vhost_scsi_target_queue_cmd(struct vhost_scsi_cmd *cmd)
 883{
 884	struct se_cmd *se_cmd = &cmd->tvc_se_cmd;
 885	struct vhost_scsi_nexus *tv_nexus;
 886	struct scatterlist *sg_ptr, *sg_prot_ptr = NULL;
 887
 888	/* FIXME: BIDI operation */
 889	if (cmd->tvc_sgl_count) {
 890		sg_ptr = cmd->tvc_sgl;
 891
 892		if (cmd->tvc_prot_sgl_count)
 893			sg_prot_ptr = cmd->tvc_prot_sgl;
 894		else
 895			se_cmd->prot_pto = true;
 896	} else {
 897		sg_ptr = NULL;
 898	}
 899	tv_nexus = cmd->tvc_nexus;
 900
 901	se_cmd->tag = 0;
 902	target_init_cmd(se_cmd, tv_nexus->tvn_se_sess, &cmd->tvc_sense_buf[0],
 903			cmd->tvc_lun, cmd->tvc_exp_data_len,
 904			vhost_scsi_to_tcm_attr(cmd->tvc_task_attr),
 905			cmd->tvc_data_direction, TARGET_SCF_ACK_KREF);
 906
 907	if (target_submit_prep(se_cmd, cmd->tvc_cdb, sg_ptr,
 908			       cmd->tvc_sgl_count, NULL, 0, sg_prot_ptr,
 909			       cmd->tvc_prot_sgl_count, GFP_KERNEL))
 910		return;
 911
 912	target_submit(se_cmd);
 913}
 914
 915static void
 916vhost_scsi_send_bad_target(struct vhost_scsi *vs,
 917			   struct vhost_virtqueue *vq,
 918			   int head, unsigned out)
 919{
 920	struct virtio_scsi_cmd_resp __user *resp;
 921	struct virtio_scsi_cmd_resp rsp;
 922	int ret;
 923
 924	memset(&rsp, 0, sizeof(rsp));
 925	rsp.response = VIRTIO_SCSI_S_BAD_TARGET;
 926	resp = vq->iov[out].iov_base;
 927	ret = __copy_to_user(resp, &rsp, sizeof(rsp));
 928	if (!ret)
 929		vhost_add_used_and_signal(&vs->dev, vq, head, 0);
 930	else
 931		pr_err("Faulted on virtio_scsi_cmd_resp\n");
 932}
 933
 934static int
 935vhost_scsi_get_desc(struct vhost_scsi *vs, struct vhost_virtqueue *vq,
 936		    struct vhost_scsi_ctx *vc)
 937{
 938	int ret = -ENXIO;
 939
 940	vc->head = vhost_get_vq_desc(vq, vq->iov,
 941				     ARRAY_SIZE(vq->iov), &vc->out, &vc->in,
 942				     NULL, NULL);
 943
 944	pr_debug("vhost_get_vq_desc: head: %d, out: %u in: %u\n",
 945		 vc->head, vc->out, vc->in);
 946
 947	/* On error, stop handling until the next kick. */
 948	if (unlikely(vc->head < 0))
 949		goto done;
 950
 951	/* Nothing new?  Wait for eventfd to tell us they refilled. */
 952	if (vc->head == vq->num) {
 953		if (unlikely(vhost_enable_notify(&vs->dev, vq))) {
 954			vhost_disable_notify(&vs->dev, vq);
 955			ret = -EAGAIN;
 956		}
 957		goto done;
 958	}
 959
 960	/*
 961	 * Get the size of request and response buffers.
 962	 * FIXME: Not correct for BIDI operation
 963	 */
 964	vc->out_size = iov_length(vq->iov, vc->out);
 965	vc->in_size = iov_length(&vq->iov[vc->out], vc->in);
 966
 967	/*
 968	 * Copy over the virtio-scsi request header, which for a
 969	 * ANY_LAYOUT enabled guest may span multiple iovecs, or a
 970	 * single iovec may contain both the header + outgoing
 971	 * WRITE payloads.
 972	 *
 973	 * copy_from_iter() will advance out_iter, so that it will
 974	 * point at the start of the outgoing WRITE payload, if
 975	 * DMA_TO_DEVICE is set.
 976	 */
 977	iov_iter_init(&vc->out_iter, ITER_SOURCE, vq->iov, vc->out, vc->out_size);
 978	ret = 0;
 979
 980done:
 981	return ret;
 982}
 983
 984static int
 985vhost_scsi_chk_size(struct vhost_virtqueue *vq, struct vhost_scsi_ctx *vc)
 986{
 987	if (unlikely(vc->in_size < vc->rsp_size)) {
 988		vq_err(vq,
 989		       "Response buf too small, need min %zu bytes got %zu",
 990		       vc->rsp_size, vc->in_size);
 991		return -EINVAL;
 992	} else if (unlikely(vc->out_size < vc->req_size)) {
 993		vq_err(vq,
 994		       "Request buf too small, need min %zu bytes got %zu",
 995		       vc->req_size, vc->out_size);
 996		return -EIO;
 997	}
 998
 999	return 0;
1000}
1001
1002static int
1003vhost_scsi_get_req(struct vhost_virtqueue *vq, struct vhost_scsi_ctx *vc,
1004		   struct vhost_scsi_tpg **tpgp)
1005{
1006	int ret = -EIO;
1007
1008	if (unlikely(!copy_from_iter_full(vc->req, vc->req_size,
1009					  &vc->out_iter))) {
1010		vq_err(vq, "Faulted on copy_from_iter_full\n");
1011	} else if (unlikely(*vc->lunp != 1)) {
1012		/* virtio-scsi spec requires byte 0 of the lun to be 1 */
1013		vq_err(vq, "Illegal virtio-scsi lun: %u\n", *vc->lunp);
1014	} else {
1015		struct vhost_scsi_tpg **vs_tpg, *tpg;
1016
1017		vs_tpg = vhost_vq_get_backend(vq);	/* validated at handler entry */
1018
1019		tpg = READ_ONCE(vs_tpg[*vc->target]);
1020		if (unlikely(!tpg)) {
1021			vq_err(vq, "Target 0x%x does not exist\n", *vc->target);
1022		} else {
1023			if (tpgp)
1024				*tpgp = tpg;
1025			ret = 0;
1026		}
1027	}
1028
1029	return ret;
1030}
1031
1032static u16 vhost_buf_to_lun(u8 *lun_buf)
1033{
1034	return ((lun_buf[2] << 8) | lun_buf[3]) & 0x3FFF;
1035}
1036
1037static void
1038vhost_scsi_handle_vq(struct vhost_scsi *vs, struct vhost_virtqueue *vq)
1039{
1040	struct vhost_scsi_tpg **vs_tpg, *tpg;
1041	struct virtio_scsi_cmd_req v_req;
1042	struct virtio_scsi_cmd_req_pi v_req_pi;
1043	struct vhost_scsi_ctx vc;
1044	struct vhost_scsi_cmd *cmd;
1045	struct iov_iter in_iter, prot_iter, data_iter;
1046	u64 tag;
1047	u32 exp_data_len, data_direction;
1048	int ret, prot_bytes, i, c = 0;
1049	u16 lun;
1050	u8 task_attr;
1051	bool t10_pi = vhost_has_feature(vq, VIRTIO_SCSI_F_T10_PI);
1052	void *cdb;
1053
1054	mutex_lock(&vq->mutex);
1055	/*
1056	 * We can handle the vq only after the endpoint is setup by calling the
1057	 * VHOST_SCSI_SET_ENDPOINT ioctl.
1058	 */
1059	vs_tpg = vhost_vq_get_backend(vq);
1060	if (!vs_tpg)
1061		goto out;
1062
1063	memset(&vc, 0, sizeof(vc));
1064	vc.rsp_size = sizeof(struct virtio_scsi_cmd_resp);
1065
1066	vhost_disable_notify(&vs->dev, vq);
1067
1068	do {
1069		ret = vhost_scsi_get_desc(vs, vq, &vc);
1070		if (ret)
1071			goto err;
1072
1073		/*
1074		 * Setup pointers and values based upon different virtio-scsi
1075		 * request header if T10_PI is enabled in KVM guest.
1076		 */
1077		if (t10_pi) {
1078			vc.req = &v_req_pi;
1079			vc.req_size = sizeof(v_req_pi);
1080			vc.lunp = &v_req_pi.lun[0];
1081			vc.target = &v_req_pi.lun[1];
1082		} else {
1083			vc.req = &v_req;
1084			vc.req_size = sizeof(v_req);
1085			vc.lunp = &v_req.lun[0];
1086			vc.target = &v_req.lun[1];
1087		}
1088
1089		/*
1090		 * Validate the size of request and response buffers.
1091		 * Check for a sane response buffer so we can report
1092		 * early errors back to the guest.
1093		 */
1094		ret = vhost_scsi_chk_size(vq, &vc);
1095		if (ret)
1096			goto err;
1097
1098		ret = vhost_scsi_get_req(vq, &vc, &tpg);
1099		if (ret)
1100			goto err;
1101
1102		ret = -EIO;	/* bad target on any error from here on */
1103
1104		/*
1105		 * Determine data_direction by calculating the total outgoing
1106		 * iovec sizes + incoming iovec sizes vs. virtio-scsi request +
1107		 * response headers respectively.
1108		 *
1109		 * For DMA_TO_DEVICE this is out_iter, which is already pointing
1110		 * to the right place.
1111		 *
1112		 * For DMA_FROM_DEVICE, the iovec will be just past the end
1113		 * of the virtio-scsi response header in either the same
1114		 * or immediately following iovec.
1115		 *
1116		 * Any associated T10_PI bytes for the outgoing / incoming
1117		 * payloads are included in calculation of exp_data_len here.
1118		 */
1119		prot_bytes = 0;
1120
1121		if (vc.out_size > vc.req_size) {
1122			data_direction = DMA_TO_DEVICE;
1123			exp_data_len = vc.out_size - vc.req_size;
1124			data_iter = vc.out_iter;
1125		} else if (vc.in_size > vc.rsp_size) {
1126			data_direction = DMA_FROM_DEVICE;
1127			exp_data_len = vc.in_size - vc.rsp_size;
1128
1129			iov_iter_init(&in_iter, ITER_DEST, &vq->iov[vc.out], vc.in,
1130				      vc.rsp_size + exp_data_len);
1131			iov_iter_advance(&in_iter, vc.rsp_size);
1132			data_iter = in_iter;
1133		} else {
1134			data_direction = DMA_NONE;
1135			exp_data_len = 0;
1136		}
1137		/*
1138		 * If T10_PI header + payload is present, setup prot_iter values
1139		 * and recalculate data_iter for vhost_scsi_mapal() mapping to
1140		 * host scatterlists via get_user_pages_fast().
1141		 */
1142		if (t10_pi) {
1143			if (v_req_pi.pi_bytesout) {
1144				if (data_direction != DMA_TO_DEVICE) {
1145					vq_err(vq, "Received non zero pi_bytesout,"
1146						" but wrong data_direction\n");
1147					goto err;
1148				}
1149				prot_bytes = vhost32_to_cpu(vq, v_req_pi.pi_bytesout);
1150			} else if (v_req_pi.pi_bytesin) {
1151				if (data_direction != DMA_FROM_DEVICE) {
1152					vq_err(vq, "Received non zero pi_bytesin,"
1153						" but wrong data_direction\n");
1154					goto err;
1155				}
1156				prot_bytes = vhost32_to_cpu(vq, v_req_pi.pi_bytesin);
1157			}
1158			/*
1159			 * Set prot_iter to data_iter and truncate it to
1160			 * prot_bytes, and advance data_iter past any
1161			 * preceding prot_bytes that may be present.
1162			 *
1163			 * Also fix up the exp_data_len to reflect only the
1164			 * actual data payload length.
1165			 */
1166			if (prot_bytes) {
1167				exp_data_len -= prot_bytes;
1168				prot_iter = data_iter;
1169				iov_iter_truncate(&prot_iter, prot_bytes);
1170				iov_iter_advance(&data_iter, prot_bytes);
1171			}
1172			tag = vhost64_to_cpu(vq, v_req_pi.tag);
1173			task_attr = v_req_pi.task_attr;
1174			cdb = &v_req_pi.cdb[0];
1175			lun = vhost_buf_to_lun(v_req_pi.lun);
1176		} else {
1177			tag = vhost64_to_cpu(vq, v_req.tag);
1178			task_attr = v_req.task_attr;
1179			cdb = &v_req.cdb[0];
1180			lun = vhost_buf_to_lun(v_req.lun);
1181		}
1182		/*
1183		 * Check that the received CDB size does not exceeded our
1184		 * hardcoded max for vhost-scsi, then get a pre-allocated
1185		 * cmd descriptor for the new virtio-scsi tag.
1186		 *
1187		 * TODO what if cdb was too small for varlen cdb header?
1188		 */
1189		if (unlikely(scsi_command_size(cdb) > VHOST_SCSI_MAX_CDB_SIZE)) {
1190			vq_err(vq, "Received SCSI CDB with command_size: %d that"
1191				" exceeds SCSI_MAX_VARLEN_CDB_SIZE: %d\n",
1192				scsi_command_size(cdb), VHOST_SCSI_MAX_CDB_SIZE);
1193				goto err;
1194		}
1195		cmd = vhost_scsi_get_cmd(vq, tpg, cdb, tag, lun, task_attr,
1196					 exp_data_len + prot_bytes,
1197					 data_direction);
1198		if (IS_ERR(cmd)) {
1199			vq_err(vq, "vhost_scsi_get_cmd failed %ld\n",
1200			       PTR_ERR(cmd));
1201			goto err;
1202		}
1203		cmd->tvc_vhost = vs;
1204		cmd->tvc_vq = vq;
1205		for (i = 0; i < vc.in ; i++)
1206			cmd->tvc_resp_iov[i] = vq->iov[vc.out + i];
1207		cmd->tvc_in_iovs = vc.in;
1208
1209		pr_debug("vhost_scsi got command opcode: %#02x, lun: %d\n",
1210			 cmd->tvc_cdb[0], cmd->tvc_lun);
1211		pr_debug("cmd: %p exp_data_len: %d, prot_bytes: %d data_direction:"
1212			 " %d\n", cmd, exp_data_len, prot_bytes, data_direction);
1213
1214		if (data_direction != DMA_NONE) {
1215			if (unlikely(vhost_scsi_mapal(cmd, prot_bytes,
1216						      &prot_iter, exp_data_len,
1217						      &data_iter))) {
1218				vq_err(vq, "Failed to map iov to sgl\n");
1219				vhost_scsi_release_cmd_res(&cmd->tvc_se_cmd);
1220				goto err;
1221			}
1222		}
1223		/*
1224		 * Save the descriptor from vhost_get_vq_desc() to be used to
1225		 * complete the virtio-scsi request in TCM callback context via
1226		 * vhost_scsi_queue_data_in() and vhost_scsi_queue_status()
1227		 */
1228		cmd->tvc_vq_desc = vc.head;
1229		vhost_scsi_target_queue_cmd(cmd);
1230		ret = 0;
1231err:
1232		/*
1233		 * ENXIO:  No more requests, or read error, wait for next kick
1234		 * EINVAL: Invalid response buffer, drop the request
1235		 * EIO:    Respond with bad target
1236		 * EAGAIN: Pending request
1237		 */
1238		if (ret == -ENXIO)
1239			break;
1240		else if (ret == -EIO)
1241			vhost_scsi_send_bad_target(vs, vq, vc.head, vc.out);
1242	} while (likely(!vhost_exceeds_weight(vq, ++c, 0)));
1243out:
1244	mutex_unlock(&vq->mutex);
1245}
1246
1247static void
1248vhost_scsi_send_tmf_resp(struct vhost_scsi *vs, struct vhost_virtqueue *vq,
1249			 int in_iovs, int vq_desc, struct iovec *resp_iov,
1250			 int tmf_resp_code)
1251{
1252	struct virtio_scsi_ctrl_tmf_resp rsp;
1253	struct iov_iter iov_iter;
1254	int ret;
1255
1256	pr_debug("%s\n", __func__);
1257	memset(&rsp, 0, sizeof(rsp));
1258	rsp.response = tmf_resp_code;
1259
1260	iov_iter_init(&iov_iter, ITER_DEST, resp_iov, in_iovs, sizeof(rsp));
1261
1262	ret = copy_to_iter(&rsp, sizeof(rsp), &iov_iter);
1263	if (likely(ret == sizeof(rsp)))
1264		vhost_add_used_and_signal(&vs->dev, vq, vq_desc, 0);
1265	else
1266		pr_err("Faulted on virtio_scsi_ctrl_tmf_resp\n");
1267}
1268
1269static void vhost_scsi_tmf_resp_work(struct vhost_work *work)
1270{
1271	struct vhost_scsi_tmf *tmf = container_of(work, struct vhost_scsi_tmf,
1272						  vwork);
1273	struct vhost_virtqueue *ctl_vq, *vq;
1274	int resp_code, i;
1275
1276	if (tmf->scsi_resp == TMR_FUNCTION_COMPLETE) {
1277		/*
1278		 * Flush IO vqs that don't share a worker with the ctl to make
1279		 * sure they have sent their responses before us.
1280		 */
1281		ctl_vq = &tmf->vhost->vqs[VHOST_SCSI_VQ_CTL].vq;
1282		for (i = VHOST_SCSI_VQ_IO; i < tmf->vhost->dev.nvqs; i++) {
1283			vq = &tmf->vhost->vqs[i].vq;
1284
1285			if (vhost_vq_is_setup(vq) &&
1286			    vq->worker != ctl_vq->worker)
1287				vhost_vq_flush(vq);
1288		}
1289
 
1290		resp_code = VIRTIO_SCSI_S_FUNCTION_SUCCEEDED;
1291	} else {
1292		resp_code = VIRTIO_SCSI_S_FUNCTION_REJECTED;
1293	}
1294
1295	vhost_scsi_send_tmf_resp(tmf->vhost, &tmf->svq->vq, tmf->in_iovs,
1296				 tmf->vq_desc, &tmf->resp_iov, resp_code);
1297	vhost_scsi_release_tmf_res(tmf);
1298}
1299
1300static void
1301vhost_scsi_handle_tmf(struct vhost_scsi *vs, struct vhost_scsi_tpg *tpg,
1302		      struct vhost_virtqueue *vq,
1303		      struct virtio_scsi_ctrl_tmf_req *vtmf,
1304		      struct vhost_scsi_ctx *vc)
1305{
1306	struct vhost_scsi_virtqueue *svq = container_of(vq,
1307					struct vhost_scsi_virtqueue, vq);
1308	struct vhost_scsi_tmf *tmf;
1309
1310	if (vhost32_to_cpu(vq, vtmf->subtype) !=
1311	    VIRTIO_SCSI_T_TMF_LOGICAL_UNIT_RESET)
1312		goto send_reject;
1313
1314	if (!tpg->tpg_nexus || !tpg->tpg_nexus->tvn_se_sess) {
1315		pr_err("Unable to locate active struct vhost_scsi_nexus for LUN RESET.\n");
1316		goto send_reject;
1317	}
1318
1319	tmf = kzalloc(sizeof(*tmf), GFP_KERNEL);
1320	if (!tmf)
 
 
1321		goto send_reject;
 
1322
1323	vhost_work_init(&tmf->vwork, vhost_scsi_tmf_resp_work);
 
 
 
 
 
1324	tmf->vhost = vs;
1325	tmf->svq = svq;
1326	tmf->resp_iov = vq->iov[vc->out];
1327	tmf->vq_desc = vc->head;
1328	tmf->in_iovs = vc->in;
1329	tmf->inflight = vhost_scsi_get_inflight(vq);
1330
1331	if (target_submit_tmr(&tmf->se_cmd, tpg->tpg_nexus->tvn_se_sess, NULL,
1332			      vhost_buf_to_lun(vtmf->lun), NULL,
1333			      TMR_LUN_RESET, GFP_KERNEL, 0,
1334			      TARGET_SCF_ACK_KREF) < 0) {
1335		vhost_scsi_release_tmf_res(tmf);
1336		goto send_reject;
1337	}
1338
1339	return;
1340
1341send_reject:
1342	vhost_scsi_send_tmf_resp(vs, vq, vc->in, vc->head, &vq->iov[vc->out],
1343				 VIRTIO_SCSI_S_FUNCTION_REJECTED);
1344}
1345
1346static void
1347vhost_scsi_send_an_resp(struct vhost_scsi *vs,
1348			struct vhost_virtqueue *vq,
1349			struct vhost_scsi_ctx *vc)
1350{
1351	struct virtio_scsi_ctrl_an_resp rsp;
1352	struct iov_iter iov_iter;
1353	int ret;
1354
1355	pr_debug("%s\n", __func__);
1356	memset(&rsp, 0, sizeof(rsp));	/* event_actual = 0 */
1357	rsp.response = VIRTIO_SCSI_S_OK;
1358
1359	iov_iter_init(&iov_iter, ITER_DEST, &vq->iov[vc->out], vc->in, sizeof(rsp));
1360
1361	ret = copy_to_iter(&rsp, sizeof(rsp), &iov_iter);
1362	if (likely(ret == sizeof(rsp)))
1363		vhost_add_used_and_signal(&vs->dev, vq, vc->head, 0);
1364	else
1365		pr_err("Faulted on virtio_scsi_ctrl_an_resp\n");
1366}
1367
1368static void
1369vhost_scsi_ctl_handle_vq(struct vhost_scsi *vs, struct vhost_virtqueue *vq)
1370{
1371	struct vhost_scsi_tpg *tpg;
1372	union {
1373		__virtio32 type;
1374		struct virtio_scsi_ctrl_an_req an;
1375		struct virtio_scsi_ctrl_tmf_req tmf;
1376	} v_req;
1377	struct vhost_scsi_ctx vc;
1378	size_t typ_size;
1379	int ret, c = 0;
1380
1381	mutex_lock(&vq->mutex);
1382	/*
1383	 * We can handle the vq only after the endpoint is setup by calling the
1384	 * VHOST_SCSI_SET_ENDPOINT ioctl.
1385	 */
1386	if (!vhost_vq_get_backend(vq))
1387		goto out;
1388
1389	memset(&vc, 0, sizeof(vc));
1390
1391	vhost_disable_notify(&vs->dev, vq);
1392
1393	do {
1394		ret = vhost_scsi_get_desc(vs, vq, &vc);
1395		if (ret)
1396			goto err;
1397
1398		/*
1399		 * Get the request type first in order to setup
1400		 * other parameters dependent on the type.
1401		 */
1402		vc.req = &v_req.type;
1403		typ_size = sizeof(v_req.type);
1404
1405		if (unlikely(!copy_from_iter_full(vc.req, typ_size,
1406						  &vc.out_iter))) {
1407			vq_err(vq, "Faulted on copy_from_iter tmf type\n");
1408			/*
1409			 * The size of the response buffer depends on the
1410			 * request type and must be validated against it.
1411			 * Since the request type is not known, don't send
1412			 * a response.
1413			 */
1414			continue;
1415		}
1416
1417		switch (vhost32_to_cpu(vq, v_req.type)) {
1418		case VIRTIO_SCSI_T_TMF:
1419			vc.req = &v_req.tmf;
1420			vc.req_size = sizeof(struct virtio_scsi_ctrl_tmf_req);
1421			vc.rsp_size = sizeof(struct virtio_scsi_ctrl_tmf_resp);
1422			vc.lunp = &v_req.tmf.lun[0];
1423			vc.target = &v_req.tmf.lun[1];
1424			break;
1425		case VIRTIO_SCSI_T_AN_QUERY:
1426		case VIRTIO_SCSI_T_AN_SUBSCRIBE:
1427			vc.req = &v_req.an;
1428			vc.req_size = sizeof(struct virtio_scsi_ctrl_an_req);
1429			vc.rsp_size = sizeof(struct virtio_scsi_ctrl_an_resp);
1430			vc.lunp = &v_req.an.lun[0];
1431			vc.target = NULL;
1432			break;
1433		default:
1434			vq_err(vq, "Unknown control request %d", v_req.type);
1435			continue;
1436		}
1437
1438		/*
1439		 * Validate the size of request and response buffers.
1440		 * Check for a sane response buffer so we can report
1441		 * early errors back to the guest.
1442		 */
1443		ret = vhost_scsi_chk_size(vq, &vc);
1444		if (ret)
1445			goto err;
1446
1447		/*
1448		 * Get the rest of the request now that its size is known.
1449		 */
1450		vc.req += typ_size;
1451		vc.req_size -= typ_size;
1452
1453		ret = vhost_scsi_get_req(vq, &vc, &tpg);
1454		if (ret)
1455			goto err;
1456
1457		if (v_req.type == VIRTIO_SCSI_T_TMF)
1458			vhost_scsi_handle_tmf(vs, tpg, vq, &v_req.tmf, &vc);
1459		else
1460			vhost_scsi_send_an_resp(vs, vq, &vc);
1461err:
1462		/*
1463		 * ENXIO:  No more requests, or read error, wait for next kick
1464		 * EINVAL: Invalid response buffer, drop the request
1465		 * EIO:    Respond with bad target
1466		 * EAGAIN: Pending request
1467		 */
1468		if (ret == -ENXIO)
1469			break;
1470		else if (ret == -EIO)
1471			vhost_scsi_send_bad_target(vs, vq, vc.head, vc.out);
1472	} while (likely(!vhost_exceeds_weight(vq, ++c, 0)));
1473out:
1474	mutex_unlock(&vq->mutex);
1475}
1476
1477static void vhost_scsi_ctl_handle_kick(struct vhost_work *work)
1478{
1479	struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
1480						poll.work);
1481	struct vhost_scsi *vs = container_of(vq->dev, struct vhost_scsi, dev);
1482
1483	pr_debug("%s: The handling func for control queue.\n", __func__);
1484	vhost_scsi_ctl_handle_vq(vs, vq);
1485}
1486
1487static void
1488vhost_scsi_send_evt(struct vhost_scsi *vs, struct vhost_virtqueue *vq,
1489		    struct vhost_scsi_tpg *tpg, struct se_lun *lun,
1490		    u32 event, u32 reason)
 
 
1491{
1492	struct vhost_scsi_evt *evt;
1493
1494	evt = vhost_scsi_allocate_evt(vs, event, reason);
1495	if (!evt)
1496		return;
1497
1498	if (tpg && lun) {
1499		/* TODO: share lun setup code with virtio-scsi.ko */
1500		/*
1501		 * Note: evt->event is zeroed when we allocate it and
1502		 * lun[4-7] need to be zero according to virtio-scsi spec.
1503		 */
1504		evt->event.lun[0] = 0x01;
1505		evt->event.lun[1] = tpg->tport_tpgt;
1506		if (lun->unpacked_lun >= 256)
1507			evt->event.lun[2] = lun->unpacked_lun >> 8 | 0x40 ;
1508		evt->event.lun[3] = lun->unpacked_lun & 0xFF;
1509	}
1510
1511	llist_add(&evt->list, &vs->vs_event_list);
1512	vhost_vq_work_queue(vq, &vs->vs_event_work);
1513}
1514
1515static void vhost_scsi_evt_handle_kick(struct vhost_work *work)
1516{
1517	struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
1518						poll.work);
1519	struct vhost_scsi *vs = container_of(vq->dev, struct vhost_scsi, dev);
1520
1521	mutex_lock(&vq->mutex);
1522	if (!vhost_vq_get_backend(vq))
1523		goto out;
1524
1525	if (vs->vs_events_missed)
1526		vhost_scsi_send_evt(vs, vq, NULL, NULL, VIRTIO_SCSI_T_NO_EVENT,
1527				    0);
1528out:
1529	mutex_unlock(&vq->mutex);
1530}
1531
1532static void vhost_scsi_handle_kick(struct vhost_work *work)
1533{
1534	struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
1535						poll.work);
1536	struct vhost_scsi *vs = container_of(vq->dev, struct vhost_scsi, dev);
1537
1538	vhost_scsi_handle_vq(vs, vq);
1539}
1540
1541/* Callers must hold dev mutex */
1542static void vhost_scsi_flush(struct vhost_scsi *vs)
1543{
 
1544	int i;
1545
1546	/* Init new inflight and remember the old inflight */
1547	vhost_scsi_init_inflight(vs, vs->old_inflight);
1548
1549	/*
1550	 * The inflight->kref was initialized to 1. We decrement it here to
1551	 * indicate the start of the flush operation so that it will reach 0
1552	 * when all the reqs are finished.
1553	 */
1554	for (i = 0; i < vs->dev.nvqs; i++)
1555		kref_put(&vs->old_inflight[i]->kref, vhost_scsi_done_inflight);
1556
1557	/* Flush both the vhost poll and vhost work */
1558	vhost_dev_flush(&vs->dev);
1559
1560	/* Wait for all reqs issued before the flush to be finished */
1561	for (i = 0; i < vs->dev.nvqs; i++)
1562		wait_for_completion(&vs->old_inflight[i]->comp);
1563}
1564
1565static void vhost_scsi_destroy_vq_cmds(struct vhost_virtqueue *vq)
1566{
1567	struct vhost_scsi_virtqueue *svq = container_of(vq,
1568					struct vhost_scsi_virtqueue, vq);
1569	struct vhost_scsi_cmd *tv_cmd;
1570	unsigned int i;
1571
1572	if (!svq->scsi_cmds)
1573		return;
1574
1575	for (i = 0; i < svq->max_cmds; i++) {
1576		tv_cmd = &svq->scsi_cmds[i];
1577
1578		kfree(tv_cmd->tvc_sgl);
1579		kfree(tv_cmd->tvc_prot_sgl);
1580		kfree(tv_cmd->tvc_upages);
1581		kfree(tv_cmd->tvc_resp_iov);
1582	}
1583
1584	sbitmap_free(&svq->scsi_tags);
1585	kfree(svq->scsi_cmds);
1586	svq->scsi_cmds = NULL;
1587}
1588
1589static int vhost_scsi_setup_vq_cmds(struct vhost_virtqueue *vq, int max_cmds)
1590{
1591	struct vhost_scsi_virtqueue *svq = container_of(vq,
1592					struct vhost_scsi_virtqueue, vq);
1593	struct vhost_scsi_cmd *tv_cmd;
1594	unsigned int i;
1595
1596	if (svq->scsi_cmds)
1597		return 0;
1598
1599	if (sbitmap_init_node(&svq->scsi_tags, max_cmds, -1, GFP_KERNEL,
1600			      NUMA_NO_NODE, false, true))
1601		return -ENOMEM;
1602	svq->max_cmds = max_cmds;
1603
1604	svq->scsi_cmds = kcalloc(max_cmds, sizeof(*tv_cmd), GFP_KERNEL);
1605	if (!svq->scsi_cmds) {
1606		sbitmap_free(&svq->scsi_tags);
1607		return -ENOMEM;
1608	}
1609
1610	for (i = 0; i < max_cmds; i++) {
1611		tv_cmd = &svq->scsi_cmds[i];
1612
1613		tv_cmd->tvc_sgl = kcalloc(VHOST_SCSI_PREALLOC_SGLS,
1614					  sizeof(struct scatterlist),
1615					  GFP_KERNEL);
1616		if (!tv_cmd->tvc_sgl) {
1617			pr_err("Unable to allocate tv_cmd->tvc_sgl\n");
1618			goto out;
1619		}
1620
1621		tv_cmd->tvc_upages = kcalloc(VHOST_SCSI_PREALLOC_UPAGES,
1622					     sizeof(struct page *),
1623					     GFP_KERNEL);
1624		if (!tv_cmd->tvc_upages) {
1625			pr_err("Unable to allocate tv_cmd->tvc_upages\n");
1626			goto out;
1627		}
1628
1629		tv_cmd->tvc_resp_iov = kcalloc(UIO_MAXIOV,
1630					       sizeof(struct iovec),
1631					       GFP_KERNEL);
1632		if (!tv_cmd->tvc_resp_iov) {
1633			pr_err("Unable to allocate tv_cmd->tvc_resp_iov\n");
1634			goto out;
1635		}
1636
1637		tv_cmd->tvc_prot_sgl = kcalloc(VHOST_SCSI_PREALLOC_PROT_SGLS,
1638					       sizeof(struct scatterlist),
1639					       GFP_KERNEL);
1640		if (!tv_cmd->tvc_prot_sgl) {
1641			pr_err("Unable to allocate tv_cmd->tvc_prot_sgl\n");
1642			goto out;
1643		}
1644	}
1645	return 0;
1646out:
1647	vhost_scsi_destroy_vq_cmds(vq);
1648	return -ENOMEM;
1649}
1650
1651/*
1652 * Called from vhost_scsi_ioctl() context to walk the list of available
1653 * vhost_scsi_tpg with an active struct vhost_scsi_nexus
1654 *
1655 *  The lock nesting rule is:
1656 *    vs->dev.mutex -> vhost_scsi_mutex -> tpg->tv_tpg_mutex -> vq->mutex
1657 */
1658static int
1659vhost_scsi_set_endpoint(struct vhost_scsi *vs,
1660			struct vhost_scsi_target *t)
1661{
1662	struct se_portal_group *se_tpg;
1663	struct vhost_scsi_tport *tv_tport;
1664	struct vhost_scsi_tpg *tpg;
1665	struct vhost_scsi_tpg **vs_tpg;
1666	struct vhost_virtqueue *vq;
1667	int index, ret, i, len;
1668	bool match = false;
1669
 
1670	mutex_lock(&vs->dev.mutex);
1671
1672	/* Verify that ring has been setup correctly. */
1673	for (index = 0; index < vs->dev.nvqs; ++index) {
1674		/* Verify that ring has been setup correctly. */
1675		if (!vhost_vq_access_ok(&vs->vqs[index].vq)) {
1676			ret = -EFAULT;
1677			goto out;
1678		}
1679	}
1680
1681	len = sizeof(vs_tpg[0]) * VHOST_SCSI_MAX_TARGET;
1682	vs_tpg = kzalloc(len, GFP_KERNEL);
1683	if (!vs_tpg) {
1684		ret = -ENOMEM;
1685		goto out;
1686	}
1687	if (vs->vs_tpg)
1688		memcpy(vs_tpg, vs->vs_tpg, len);
1689
1690	mutex_lock(&vhost_scsi_mutex);
1691	list_for_each_entry(tpg, &vhost_scsi_list, tv_tpg_list) {
1692		mutex_lock(&tpg->tv_tpg_mutex);
1693		if (!tpg->tpg_nexus) {
1694			mutex_unlock(&tpg->tv_tpg_mutex);
1695			continue;
1696		}
1697		if (tpg->tv_tpg_vhost_count != 0) {
1698			mutex_unlock(&tpg->tv_tpg_mutex);
1699			continue;
1700		}
1701		tv_tport = tpg->tport;
1702
1703		if (!strcmp(tv_tport->tport_name, t->vhost_wwpn)) {
1704			if (vs->vs_tpg && vs->vs_tpg[tpg->tport_tpgt]) {
1705				mutex_unlock(&tpg->tv_tpg_mutex);
1706				mutex_unlock(&vhost_scsi_mutex);
1707				ret = -EEXIST;
1708				goto undepend;
1709			}
1710			/*
1711			 * In order to ensure individual vhost-scsi configfs
1712			 * groups cannot be removed while in use by vhost ioctl,
1713			 * go ahead and take an explicit se_tpg->tpg_group.cg_item
1714			 * dependency now.
1715			 */
1716			se_tpg = &tpg->se_tpg;
1717			ret = target_depend_item(&se_tpg->tpg_group.cg_item);
1718			if (ret) {
1719				pr_warn("target_depend_item() failed: %d\n", ret);
1720				mutex_unlock(&tpg->tv_tpg_mutex);
1721				mutex_unlock(&vhost_scsi_mutex);
1722				goto undepend;
1723			}
1724			tpg->tv_tpg_vhost_count++;
1725			tpg->vhost_scsi = vs;
1726			vs_tpg[tpg->tport_tpgt] = tpg;
1727			match = true;
1728		}
1729		mutex_unlock(&tpg->tv_tpg_mutex);
1730	}
1731	mutex_unlock(&vhost_scsi_mutex);
1732
1733	if (match) {
1734		memcpy(vs->vs_vhost_wwpn, t->vhost_wwpn,
1735		       sizeof(vs->vs_vhost_wwpn));
1736
1737		for (i = VHOST_SCSI_VQ_IO; i < vs->dev.nvqs; i++) {
1738			vq = &vs->vqs[i].vq;
1739			if (!vhost_vq_is_setup(vq))
1740				continue;
1741
1742			ret = vhost_scsi_setup_vq_cmds(vq, vq->num);
1743			if (ret)
1744				goto destroy_vq_cmds;
1745		}
1746
1747		for (i = 0; i < vs->dev.nvqs; i++) {
1748			vq = &vs->vqs[i].vq;
1749			mutex_lock(&vq->mutex);
1750			vhost_vq_set_backend(vq, vs_tpg);
1751			vhost_vq_init_access(vq);
1752			mutex_unlock(&vq->mutex);
1753		}
1754		ret = 0;
1755	} else {
1756		ret = -EEXIST;
1757	}
1758
1759	/*
1760	 * Act as synchronize_rcu to make sure access to
1761	 * old vs->vs_tpg is finished.
1762	 */
1763	vhost_scsi_flush(vs);
1764	kfree(vs->vs_tpg);
1765	vs->vs_tpg = vs_tpg;
1766	goto out;
1767
1768destroy_vq_cmds:
1769	for (i--; i >= VHOST_SCSI_VQ_IO; i--) {
1770		if (!vhost_vq_get_backend(&vs->vqs[i].vq))
1771			vhost_scsi_destroy_vq_cmds(&vs->vqs[i].vq);
1772	}
1773undepend:
1774	for (i = 0; i < VHOST_SCSI_MAX_TARGET; i++) {
1775		tpg = vs_tpg[i];
1776		if (tpg) {
1777			mutex_lock(&tpg->tv_tpg_mutex);
1778			tpg->vhost_scsi = NULL;
1779			tpg->tv_tpg_vhost_count--;
1780			mutex_unlock(&tpg->tv_tpg_mutex);
1781			target_undepend_item(&tpg->se_tpg.tpg_group.cg_item);
1782		}
1783	}
1784	kfree(vs_tpg);
1785out:
1786	mutex_unlock(&vs->dev.mutex);
 
1787	return ret;
1788}
1789
1790static int
1791vhost_scsi_clear_endpoint(struct vhost_scsi *vs,
1792			  struct vhost_scsi_target *t)
1793{
1794	struct se_portal_group *se_tpg;
1795	struct vhost_scsi_tport *tv_tport;
1796	struct vhost_scsi_tpg *tpg;
1797	struct vhost_virtqueue *vq;
1798	bool match = false;
1799	int index, ret, i;
1800	u8 target;
1801
 
1802	mutex_lock(&vs->dev.mutex);
1803	/* Verify that ring has been setup correctly. */
1804	for (index = 0; index < vs->dev.nvqs; ++index) {
1805		if (!vhost_vq_access_ok(&vs->vqs[index].vq)) {
1806			ret = -EFAULT;
1807			goto err_dev;
1808		}
1809	}
1810
1811	if (!vs->vs_tpg) {
1812		ret = 0;
1813		goto err_dev;
1814	}
1815
1816	for (i = 0; i < VHOST_SCSI_MAX_TARGET; i++) {
1817		target = i;
1818		tpg = vs->vs_tpg[target];
1819		if (!tpg)
1820			continue;
1821
 
1822		tv_tport = tpg->tport;
1823		if (!tv_tport) {
1824			ret = -ENODEV;
1825			goto err_dev;
1826		}
1827
1828		if (strcmp(tv_tport->tport_name, t->vhost_wwpn)) {
1829			pr_warn("tv_tport->tport_name: %s, tpg->tport_tpgt: %hu"
1830				" does not match t->vhost_wwpn: %s, t->vhost_tpgt: %hu\n",
1831				tv_tport->tport_name, tpg->tport_tpgt,
1832				t->vhost_wwpn, t->vhost_tpgt);
1833			ret = -EINVAL;
1834			goto err_dev;
1835		}
1836		match = true;
1837	}
1838	if (!match)
1839		goto free_vs_tpg;
1840
1841	/* Prevent new cmds from starting and accessing the tpgs/sessions */
1842	for (i = 0; i < vs->dev.nvqs; i++) {
1843		vq = &vs->vqs[i].vq;
1844		mutex_lock(&vq->mutex);
1845		vhost_vq_set_backend(vq, NULL);
1846		mutex_unlock(&vq->mutex);
1847	}
1848	/* Make sure cmds are not running before tearing them down. */
1849	vhost_scsi_flush(vs);
1850
1851	for (i = 0; i < vs->dev.nvqs; i++) {
1852		vq = &vs->vqs[i].vq;
1853		vhost_scsi_destroy_vq_cmds(vq);
1854	}
1855
1856	/*
1857	 * We can now release our hold on the tpg and sessions and userspace
1858	 * can free them after this point.
1859	 */
1860	for (i = 0; i < VHOST_SCSI_MAX_TARGET; i++) {
1861		target = i;
1862		tpg = vs->vs_tpg[target];
1863		if (!tpg)
1864			continue;
1865
1866		mutex_lock(&tpg->tv_tpg_mutex);
1867
1868		tpg->tv_tpg_vhost_count--;
1869		tpg->vhost_scsi = NULL;
1870		vs->vs_tpg[target] = NULL;
1871
1872		mutex_unlock(&tpg->tv_tpg_mutex);
1873
 
 
 
1874		se_tpg = &tpg->se_tpg;
1875		target_undepend_item(&se_tpg->tpg_group.cg_item);
1876	}
 
 
 
 
 
 
 
 
 
1877
1878free_vs_tpg:
 
 
 
 
1879	/*
1880	 * Act as synchronize_rcu to make sure access to
1881	 * old vs->vs_tpg is finished.
1882	 */
1883	vhost_scsi_flush(vs);
1884	kfree(vs->vs_tpg);
1885	vs->vs_tpg = NULL;
1886	WARN_ON(vs->vs_events_nr);
1887	mutex_unlock(&vs->dev.mutex);
 
1888	return 0;
1889
 
 
1890err_dev:
1891	mutex_unlock(&vs->dev.mutex);
 
1892	return ret;
1893}
1894
1895static int vhost_scsi_set_features(struct vhost_scsi *vs, u64 features)
1896{
1897	struct vhost_virtqueue *vq;
1898	int i;
1899
1900	if (features & ~VHOST_SCSI_FEATURES)
1901		return -EOPNOTSUPP;
1902
1903	mutex_lock(&vs->dev.mutex);
1904	if ((features & (1 << VHOST_F_LOG_ALL)) &&
1905	    !vhost_log_access_ok(&vs->dev)) {
1906		mutex_unlock(&vs->dev.mutex);
1907		return -EFAULT;
1908	}
1909
1910	for (i = 0; i < vs->dev.nvqs; i++) {
1911		vq = &vs->vqs[i].vq;
1912		mutex_lock(&vq->mutex);
1913		vq->acked_features = features;
1914		mutex_unlock(&vq->mutex);
1915	}
1916	mutex_unlock(&vs->dev.mutex);
1917	return 0;
1918}
1919
1920static int vhost_scsi_open(struct inode *inode, struct file *f)
1921{
1922	struct vhost_scsi_virtqueue *svq;
1923	struct vhost_scsi *vs;
1924	struct vhost_virtqueue **vqs;
1925	int r = -ENOMEM, i, nvqs = vhost_scsi_max_io_vqs;
1926
1927	vs = kvzalloc(sizeof(*vs), GFP_KERNEL);
1928	if (!vs)
1929		goto err_vs;
1930
1931	if (nvqs > VHOST_SCSI_MAX_IO_VQ) {
1932		pr_err("Invalid max_io_vqs of %d. Using %d.\n", nvqs,
1933		       VHOST_SCSI_MAX_IO_VQ);
1934		nvqs = VHOST_SCSI_MAX_IO_VQ;
1935	} else if (nvqs == 0) {
1936		pr_err("Invalid max_io_vqs of %d. Using 1.\n", nvqs);
1937		nvqs = 1;
1938	}
1939	nvqs += VHOST_SCSI_VQ_IO;
1940
1941	vs->old_inflight = kmalloc_array(nvqs, sizeof(*vs->old_inflight),
1942					 GFP_KERNEL | __GFP_ZERO);
1943	if (!vs->old_inflight)
1944		goto err_inflight;
1945
1946	vs->vqs = kmalloc_array(nvqs, sizeof(*vs->vqs),
1947				GFP_KERNEL | __GFP_ZERO);
1948	if (!vs->vqs)
1949		goto err_vqs;
1950
1951	vqs = kmalloc_array(nvqs, sizeof(*vqs), GFP_KERNEL);
1952	if (!vqs)
1953		goto err_local_vqs;
1954
 
1955	vhost_work_init(&vs->vs_event_work, vhost_scsi_evt_work);
1956
1957	vs->vs_events_nr = 0;
1958	vs->vs_events_missed = false;
1959
1960	vqs[VHOST_SCSI_VQ_CTL] = &vs->vqs[VHOST_SCSI_VQ_CTL].vq;
1961	vqs[VHOST_SCSI_VQ_EVT] = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
1962	vs->vqs[VHOST_SCSI_VQ_CTL].vq.handle_kick = vhost_scsi_ctl_handle_kick;
1963	vs->vqs[VHOST_SCSI_VQ_EVT].vq.handle_kick = vhost_scsi_evt_handle_kick;
1964	for (i = VHOST_SCSI_VQ_IO; i < nvqs; i++) {
1965		svq = &vs->vqs[i];
1966
1967		vqs[i] = &svq->vq;
1968		svq->vs = vs;
1969		init_llist_head(&svq->completion_list);
1970		vhost_work_init(&svq->completion_work,
1971				vhost_scsi_complete_cmd_work);
1972		svq->vq.handle_kick = vhost_scsi_handle_kick;
1973	}
1974	vhost_dev_init(&vs->dev, vqs, nvqs, UIO_MAXIOV,
1975		       VHOST_SCSI_WEIGHT, 0, true, NULL);
1976
1977	vhost_scsi_init_inflight(vs, NULL);
1978
1979	f->private_data = vs;
1980	return 0;
1981
1982err_local_vqs:
1983	kfree(vs->vqs);
1984err_vqs:
1985	kfree(vs->old_inflight);
1986err_inflight:
1987	kvfree(vs);
1988err_vs:
1989	return r;
1990}
1991
1992static int vhost_scsi_release(struct inode *inode, struct file *f)
1993{
1994	struct vhost_scsi *vs = f->private_data;
1995	struct vhost_scsi_target t;
1996
1997	mutex_lock(&vs->dev.mutex);
1998	memcpy(t.vhost_wwpn, vs->vs_vhost_wwpn, sizeof(t.vhost_wwpn));
1999	mutex_unlock(&vs->dev.mutex);
2000	vhost_scsi_clear_endpoint(vs, &t);
2001	vhost_dev_stop(&vs->dev);
2002	vhost_dev_cleanup(&vs->dev);
 
 
2003	kfree(vs->dev.vqs);
2004	kfree(vs->vqs);
2005	kfree(vs->old_inflight);
2006	kvfree(vs);
2007	return 0;
2008}
2009
2010static long
2011vhost_scsi_ioctl(struct file *f,
2012		 unsigned int ioctl,
2013		 unsigned long arg)
2014{
2015	struct vhost_scsi *vs = f->private_data;
2016	struct vhost_scsi_target backend;
2017	void __user *argp = (void __user *)arg;
2018	u64 __user *featurep = argp;
2019	u32 __user *eventsp = argp;
2020	u32 events_missed;
2021	u64 features;
2022	int r, abi_version = VHOST_SCSI_ABI_VERSION;
2023	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
2024
2025	switch (ioctl) {
2026	case VHOST_SCSI_SET_ENDPOINT:
2027		if (copy_from_user(&backend, argp, sizeof backend))
2028			return -EFAULT;
2029		if (backend.reserved != 0)
2030			return -EOPNOTSUPP;
2031
2032		return vhost_scsi_set_endpoint(vs, &backend);
2033	case VHOST_SCSI_CLEAR_ENDPOINT:
2034		if (copy_from_user(&backend, argp, sizeof backend))
2035			return -EFAULT;
2036		if (backend.reserved != 0)
2037			return -EOPNOTSUPP;
2038
2039		return vhost_scsi_clear_endpoint(vs, &backend);
2040	case VHOST_SCSI_GET_ABI_VERSION:
2041		if (copy_to_user(argp, &abi_version, sizeof abi_version))
2042			return -EFAULT;
2043		return 0;
2044	case VHOST_SCSI_SET_EVENTS_MISSED:
2045		if (get_user(events_missed, eventsp))
2046			return -EFAULT;
2047		mutex_lock(&vq->mutex);
2048		vs->vs_events_missed = events_missed;
2049		mutex_unlock(&vq->mutex);
2050		return 0;
2051	case VHOST_SCSI_GET_EVENTS_MISSED:
2052		mutex_lock(&vq->mutex);
2053		events_missed = vs->vs_events_missed;
2054		mutex_unlock(&vq->mutex);
2055		if (put_user(events_missed, eventsp))
2056			return -EFAULT;
2057		return 0;
2058	case VHOST_GET_FEATURES:
2059		features = VHOST_SCSI_FEATURES;
2060		if (copy_to_user(featurep, &features, sizeof features))
2061			return -EFAULT;
2062		return 0;
2063	case VHOST_SET_FEATURES:
2064		if (copy_from_user(&features, featurep, sizeof features))
2065			return -EFAULT;
2066		return vhost_scsi_set_features(vs, features);
2067	case VHOST_NEW_WORKER:
2068	case VHOST_FREE_WORKER:
2069	case VHOST_ATTACH_VRING_WORKER:
2070	case VHOST_GET_VRING_WORKER:
2071		mutex_lock(&vs->dev.mutex);
2072		r = vhost_worker_ioctl(&vs->dev, ioctl, argp);
2073		mutex_unlock(&vs->dev.mutex);
2074		return r;
2075	default:
2076		mutex_lock(&vs->dev.mutex);
2077		r = vhost_dev_ioctl(&vs->dev, ioctl, argp);
2078		/* TODO: flush backend after dev ioctl. */
2079		if (r == -ENOIOCTLCMD)
2080			r = vhost_vring_ioctl(&vs->dev, ioctl, argp);
2081		mutex_unlock(&vs->dev.mutex);
2082		return r;
2083	}
2084}
2085
2086static const struct file_operations vhost_scsi_fops = {
2087	.owner          = THIS_MODULE,
2088	.release        = vhost_scsi_release,
2089	.unlocked_ioctl = vhost_scsi_ioctl,
2090	.compat_ioctl	= compat_ptr_ioctl,
2091	.open           = vhost_scsi_open,
2092	.llseek		= noop_llseek,
2093};
2094
2095static struct miscdevice vhost_scsi_misc = {
2096	MISC_DYNAMIC_MINOR,
2097	"vhost-scsi",
2098	&vhost_scsi_fops,
2099};
2100
2101static int __init vhost_scsi_register(void)
2102{
2103	return misc_register(&vhost_scsi_misc);
2104}
2105
2106static void vhost_scsi_deregister(void)
2107{
2108	misc_deregister(&vhost_scsi_misc);
2109}
2110
2111static char *vhost_scsi_dump_proto_id(struct vhost_scsi_tport *tport)
2112{
2113	switch (tport->tport_proto_id) {
2114	case SCSI_PROTOCOL_SAS:
2115		return "SAS";
2116	case SCSI_PROTOCOL_FCP:
2117		return "FCP";
2118	case SCSI_PROTOCOL_ISCSI:
2119		return "iSCSI";
2120	default:
2121		break;
2122	}
2123
2124	return "Unknown";
2125}
2126
2127static void
2128vhost_scsi_do_plug(struct vhost_scsi_tpg *tpg,
2129		  struct se_lun *lun, bool plug)
2130{
2131
2132	struct vhost_scsi *vs = tpg->vhost_scsi;
2133	struct vhost_virtqueue *vq;
2134	u32 reason;
2135
2136	if (!vs)
2137		return;
2138
 
 
2139	if (plug)
2140		reason = VIRTIO_SCSI_EVT_RESET_RESCAN;
2141	else
2142		reason = VIRTIO_SCSI_EVT_RESET_REMOVED;
2143
2144	vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
2145	mutex_lock(&vq->mutex);
2146	/*
2147	 * We can't queue events if the backend has been cleared, because
2148	 * we could end up queueing an event after the flush.
2149	 */
2150	if (!vhost_vq_get_backend(vq))
2151		goto unlock;
2152
2153	if (vhost_has_feature(vq, VIRTIO_SCSI_F_HOTPLUG))
2154		vhost_scsi_send_evt(vs, vq, tpg, lun,
2155				   VIRTIO_SCSI_T_TRANSPORT_RESET, reason);
2156unlock:
2157	mutex_unlock(&vq->mutex);
 
2158}
2159
2160static void vhost_scsi_hotplug(struct vhost_scsi_tpg *tpg, struct se_lun *lun)
2161{
2162	vhost_scsi_do_plug(tpg, lun, true);
2163}
2164
2165static void vhost_scsi_hotunplug(struct vhost_scsi_tpg *tpg, struct se_lun *lun)
2166{
2167	vhost_scsi_do_plug(tpg, lun, false);
2168}
2169
2170static int vhost_scsi_port_link(struct se_portal_group *se_tpg,
2171			       struct se_lun *lun)
2172{
2173	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2174				struct vhost_scsi_tpg, se_tpg);
 
 
 
 
 
 
 
 
 
2175
2176	mutex_lock(&tpg->tv_tpg_mutex);
2177	tpg->tv_tpg_port_count++;
2178	vhost_scsi_hotplug(tpg, lun);
2179	mutex_unlock(&tpg->tv_tpg_mutex);
2180
 
 
 
 
2181	return 0;
2182}
2183
2184static void vhost_scsi_port_unlink(struct se_portal_group *se_tpg,
2185				  struct se_lun *lun)
2186{
2187	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2188				struct vhost_scsi_tpg, se_tpg);
 
 
 
2189
2190	mutex_lock(&tpg->tv_tpg_mutex);
2191	tpg->tv_tpg_port_count--;
2192	vhost_scsi_hotunplug(tpg, lun);
 
 
 
2193	mutex_unlock(&tpg->tv_tpg_mutex);
 
 
 
 
2194}
2195
2196static ssize_t vhost_scsi_tpg_attrib_fabric_prot_type_store(
2197		struct config_item *item, const char *page, size_t count)
2198{
2199	struct se_portal_group *se_tpg = attrib_to_tpg(item);
2200	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2201				struct vhost_scsi_tpg, se_tpg);
2202	unsigned long val;
2203	int ret = kstrtoul(page, 0, &val);
2204
2205	if (ret) {
2206		pr_err("kstrtoul() returned %d for fabric_prot_type\n", ret);
2207		return ret;
2208	}
2209	if (val != 0 && val != 1 && val != 3) {
2210		pr_err("Invalid vhost_scsi fabric_prot_type: %lu\n", val);
2211		return -EINVAL;
2212	}
2213	tpg->tv_fabric_prot_type = val;
2214
2215	return count;
2216}
2217
2218static ssize_t vhost_scsi_tpg_attrib_fabric_prot_type_show(
2219		struct config_item *item, char *page)
2220{
2221	struct se_portal_group *se_tpg = attrib_to_tpg(item);
2222	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2223				struct vhost_scsi_tpg, se_tpg);
2224
2225	return sysfs_emit(page, "%d\n", tpg->tv_fabric_prot_type);
2226}
2227
2228CONFIGFS_ATTR(vhost_scsi_tpg_attrib_, fabric_prot_type);
2229
2230static struct configfs_attribute *vhost_scsi_tpg_attrib_attrs[] = {
2231	&vhost_scsi_tpg_attrib_attr_fabric_prot_type,
2232	NULL,
2233};
2234
2235static int vhost_scsi_make_nexus(struct vhost_scsi_tpg *tpg,
2236				const char *name)
2237{
2238	struct vhost_scsi_nexus *tv_nexus;
2239
2240	mutex_lock(&tpg->tv_tpg_mutex);
2241	if (tpg->tpg_nexus) {
2242		mutex_unlock(&tpg->tv_tpg_mutex);
2243		pr_debug("tpg->tpg_nexus already exists\n");
2244		return -EEXIST;
2245	}
2246
2247	tv_nexus = kzalloc(sizeof(*tv_nexus), GFP_KERNEL);
2248	if (!tv_nexus) {
2249		mutex_unlock(&tpg->tv_tpg_mutex);
2250		pr_err("Unable to allocate struct vhost_scsi_nexus\n");
2251		return -ENOMEM;
2252	}
2253	/*
2254	 * Since we are running in 'demo mode' this call with generate a
2255	 * struct se_node_acl for the vhost_scsi struct se_portal_group with
2256	 * the SCSI Initiator port name of the passed configfs group 'name'.
2257	 */
2258	tv_nexus->tvn_se_sess = target_setup_session(&tpg->se_tpg, 0, 0,
2259					TARGET_PROT_DIN_PASS | TARGET_PROT_DOUT_PASS,
2260					(unsigned char *)name, tv_nexus, NULL);
2261	if (IS_ERR(tv_nexus->tvn_se_sess)) {
2262		mutex_unlock(&tpg->tv_tpg_mutex);
2263		kfree(tv_nexus);
2264		return -ENOMEM;
2265	}
2266	tpg->tpg_nexus = tv_nexus;
2267
2268	mutex_unlock(&tpg->tv_tpg_mutex);
2269	return 0;
2270}
2271
2272static int vhost_scsi_drop_nexus(struct vhost_scsi_tpg *tpg)
2273{
2274	struct se_session *se_sess;
2275	struct vhost_scsi_nexus *tv_nexus;
2276
2277	mutex_lock(&tpg->tv_tpg_mutex);
2278	tv_nexus = tpg->tpg_nexus;
2279	if (!tv_nexus) {
2280		mutex_unlock(&tpg->tv_tpg_mutex);
2281		return -ENODEV;
2282	}
2283
2284	se_sess = tv_nexus->tvn_se_sess;
2285	if (!se_sess) {
2286		mutex_unlock(&tpg->tv_tpg_mutex);
2287		return -ENODEV;
2288	}
2289
2290	if (tpg->tv_tpg_port_count != 0) {
2291		mutex_unlock(&tpg->tv_tpg_mutex);
2292		pr_err("Unable to remove TCM_vhost I_T Nexus with"
2293			" active TPG port count: %d\n",
2294			tpg->tv_tpg_port_count);
2295		return -EBUSY;
2296	}
2297
2298	if (tpg->tv_tpg_vhost_count != 0) {
2299		mutex_unlock(&tpg->tv_tpg_mutex);
2300		pr_err("Unable to remove TCM_vhost I_T Nexus with"
2301			" active TPG vhost count: %d\n",
2302			tpg->tv_tpg_vhost_count);
2303		return -EBUSY;
2304	}
2305
2306	pr_debug("TCM_vhost_ConfigFS: Removing I_T Nexus to emulated"
2307		" %s Initiator Port: %s\n", vhost_scsi_dump_proto_id(tpg->tport),
2308		tv_nexus->tvn_se_sess->se_node_acl->initiatorname);
2309
2310	/*
2311	 * Release the SCSI I_T Nexus to the emulated vhost Target Port
2312	 */
2313	target_remove_session(se_sess);
2314	tpg->tpg_nexus = NULL;
2315	mutex_unlock(&tpg->tv_tpg_mutex);
2316
2317	kfree(tv_nexus);
2318	return 0;
2319}
2320
2321static ssize_t vhost_scsi_tpg_nexus_show(struct config_item *item, char *page)
2322{
2323	struct se_portal_group *se_tpg = to_tpg(item);
2324	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2325				struct vhost_scsi_tpg, se_tpg);
2326	struct vhost_scsi_nexus *tv_nexus;
2327	ssize_t ret;
2328
2329	mutex_lock(&tpg->tv_tpg_mutex);
2330	tv_nexus = tpg->tpg_nexus;
2331	if (!tv_nexus) {
2332		mutex_unlock(&tpg->tv_tpg_mutex);
2333		return -ENODEV;
2334	}
2335	ret = sysfs_emit(page, "%s\n",
2336			tv_nexus->tvn_se_sess->se_node_acl->initiatorname);
2337	mutex_unlock(&tpg->tv_tpg_mutex);
2338
2339	return ret;
2340}
2341
2342static ssize_t vhost_scsi_tpg_nexus_store(struct config_item *item,
2343		const char *page, size_t count)
2344{
2345	struct se_portal_group *se_tpg = to_tpg(item);
2346	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2347				struct vhost_scsi_tpg, se_tpg);
2348	struct vhost_scsi_tport *tport_wwn = tpg->tport;
2349	unsigned char i_port[VHOST_SCSI_NAMELEN], *ptr, *port_ptr;
2350	int ret;
2351	/*
2352	 * Shutdown the active I_T nexus if 'NULL' is passed..
2353	 */
2354	if (!strncmp(page, "NULL", 4)) {
2355		ret = vhost_scsi_drop_nexus(tpg);
2356		return (!ret) ? count : ret;
2357	}
2358	/*
2359	 * Otherwise make sure the passed virtual Initiator port WWN matches
2360	 * the fabric protocol_id set in vhost_scsi_make_tport(), and call
2361	 * vhost_scsi_make_nexus().
2362	 */
2363	if (strlen(page) >= VHOST_SCSI_NAMELEN) {
2364		pr_err("Emulated NAA Sas Address: %s, exceeds"
2365				" max: %d\n", page, VHOST_SCSI_NAMELEN);
2366		return -EINVAL;
2367	}
2368	snprintf(&i_port[0], VHOST_SCSI_NAMELEN, "%s", page);
2369
2370	ptr = strstr(i_port, "naa.");
2371	if (ptr) {
2372		if (tport_wwn->tport_proto_id != SCSI_PROTOCOL_SAS) {
2373			pr_err("Passed SAS Initiator Port %s does not"
2374				" match target port protoid: %s\n", i_port,
2375				vhost_scsi_dump_proto_id(tport_wwn));
2376			return -EINVAL;
2377		}
2378		port_ptr = &i_port[0];
2379		goto check_newline;
2380	}
2381	ptr = strstr(i_port, "fc.");
2382	if (ptr) {
2383		if (tport_wwn->tport_proto_id != SCSI_PROTOCOL_FCP) {
2384			pr_err("Passed FCP Initiator Port %s does not"
2385				" match target port protoid: %s\n", i_port,
2386				vhost_scsi_dump_proto_id(tport_wwn));
2387			return -EINVAL;
2388		}
2389		port_ptr = &i_port[3]; /* Skip over "fc." */
2390		goto check_newline;
2391	}
2392	ptr = strstr(i_port, "iqn.");
2393	if (ptr) {
2394		if (tport_wwn->tport_proto_id != SCSI_PROTOCOL_ISCSI) {
2395			pr_err("Passed iSCSI Initiator Port %s does not"
2396				" match target port protoid: %s\n", i_port,
2397				vhost_scsi_dump_proto_id(tport_wwn));
2398			return -EINVAL;
2399		}
2400		port_ptr = &i_port[0];
2401		goto check_newline;
2402	}
2403	pr_err("Unable to locate prefix for emulated Initiator Port:"
2404			" %s\n", i_port);
2405	return -EINVAL;
2406	/*
2407	 * Clear any trailing newline for the NAA WWN
2408	 */
2409check_newline:
2410	if (i_port[strlen(i_port)-1] == '\n')
2411		i_port[strlen(i_port)-1] = '\0';
2412
2413	ret = vhost_scsi_make_nexus(tpg, port_ptr);
2414	if (ret < 0)
2415		return ret;
2416
2417	return count;
2418}
2419
2420CONFIGFS_ATTR(vhost_scsi_tpg_, nexus);
2421
2422static struct configfs_attribute *vhost_scsi_tpg_attrs[] = {
2423	&vhost_scsi_tpg_attr_nexus,
2424	NULL,
2425};
2426
2427static struct se_portal_group *
2428vhost_scsi_make_tpg(struct se_wwn *wwn, const char *name)
2429{
2430	struct vhost_scsi_tport *tport = container_of(wwn,
2431			struct vhost_scsi_tport, tport_wwn);
2432
2433	struct vhost_scsi_tpg *tpg;
2434	u16 tpgt;
2435	int ret;
2436
2437	if (strstr(name, "tpgt_") != name)
2438		return ERR_PTR(-EINVAL);
2439	if (kstrtou16(name + 5, 10, &tpgt) || tpgt >= VHOST_SCSI_MAX_TARGET)
2440		return ERR_PTR(-EINVAL);
2441
2442	tpg = kzalloc(sizeof(*tpg), GFP_KERNEL);
2443	if (!tpg) {
2444		pr_err("Unable to allocate struct vhost_scsi_tpg");
2445		return ERR_PTR(-ENOMEM);
2446	}
2447	mutex_init(&tpg->tv_tpg_mutex);
2448	INIT_LIST_HEAD(&tpg->tv_tpg_list);
 
2449	tpg->tport = tport;
2450	tpg->tport_tpgt = tpgt;
2451
2452	ret = core_tpg_register(wwn, &tpg->se_tpg, tport->tport_proto_id);
2453	if (ret < 0) {
2454		kfree(tpg);
2455		return NULL;
2456	}
2457	mutex_lock(&vhost_scsi_mutex);
2458	list_add_tail(&tpg->tv_tpg_list, &vhost_scsi_list);
2459	mutex_unlock(&vhost_scsi_mutex);
2460
2461	return &tpg->se_tpg;
2462}
2463
2464static void vhost_scsi_drop_tpg(struct se_portal_group *se_tpg)
2465{
2466	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2467				struct vhost_scsi_tpg, se_tpg);
2468
2469	mutex_lock(&vhost_scsi_mutex);
2470	list_del(&tpg->tv_tpg_list);
2471	mutex_unlock(&vhost_scsi_mutex);
2472	/*
2473	 * Release the virtual I_T Nexus for this vhost TPG
2474	 */
2475	vhost_scsi_drop_nexus(tpg);
2476	/*
2477	 * Deregister the se_tpg from TCM..
2478	 */
2479	core_tpg_deregister(se_tpg);
2480	kfree(tpg);
2481}
2482
2483static struct se_wwn *
2484vhost_scsi_make_tport(struct target_fabric_configfs *tf,
2485		     struct config_group *group,
2486		     const char *name)
2487{
2488	struct vhost_scsi_tport *tport;
2489	char *ptr;
2490	u64 wwpn = 0;
2491	int off = 0;
2492
2493	/* if (vhost_scsi_parse_wwn(name, &wwpn, 1) < 0)
2494		return ERR_PTR(-EINVAL); */
2495
2496	tport = kzalloc(sizeof(*tport), GFP_KERNEL);
2497	if (!tport) {
2498		pr_err("Unable to allocate struct vhost_scsi_tport");
2499		return ERR_PTR(-ENOMEM);
2500	}
2501	tport->tport_wwpn = wwpn;
2502	/*
2503	 * Determine the emulated Protocol Identifier and Target Port Name
2504	 * based on the incoming configfs directory name.
2505	 */
2506	ptr = strstr(name, "naa.");
2507	if (ptr) {
2508		tport->tport_proto_id = SCSI_PROTOCOL_SAS;
2509		goto check_len;
2510	}
2511	ptr = strstr(name, "fc.");
2512	if (ptr) {
2513		tport->tport_proto_id = SCSI_PROTOCOL_FCP;
2514		off = 3; /* Skip over "fc." */
2515		goto check_len;
2516	}
2517	ptr = strstr(name, "iqn.");
2518	if (ptr) {
2519		tport->tport_proto_id = SCSI_PROTOCOL_ISCSI;
2520		goto check_len;
2521	}
2522
2523	pr_err("Unable to locate prefix for emulated Target Port:"
2524			" %s\n", name);
2525	kfree(tport);
2526	return ERR_PTR(-EINVAL);
2527
2528check_len:
2529	if (strlen(name) >= VHOST_SCSI_NAMELEN) {
2530		pr_err("Emulated %s Address: %s, exceeds"
2531			" max: %d\n", name, vhost_scsi_dump_proto_id(tport),
2532			VHOST_SCSI_NAMELEN);
2533		kfree(tport);
2534		return ERR_PTR(-EINVAL);
2535	}
2536	snprintf(&tport->tport_name[0], VHOST_SCSI_NAMELEN, "%s", &name[off]);
2537
2538	pr_debug("TCM_VHost_ConfigFS: Allocated emulated Target"
2539		" %s Address: %s\n", vhost_scsi_dump_proto_id(tport), name);
2540
2541	return &tport->tport_wwn;
2542}
2543
2544static void vhost_scsi_drop_tport(struct se_wwn *wwn)
2545{
2546	struct vhost_scsi_tport *tport = container_of(wwn,
2547				struct vhost_scsi_tport, tport_wwn);
2548
2549	pr_debug("TCM_VHost_ConfigFS: Deallocating emulated Target"
2550		" %s Address: %s\n", vhost_scsi_dump_proto_id(tport),
2551		tport->tport_name);
2552
2553	kfree(tport);
2554}
2555
2556static ssize_t
2557vhost_scsi_wwn_version_show(struct config_item *item, char *page)
2558{
2559	return sysfs_emit(page, "TCM_VHOST fabric module %s on %s/%s"
2560		"on "UTS_RELEASE"\n", VHOST_SCSI_VERSION, utsname()->sysname,
2561		utsname()->machine);
2562}
2563
2564CONFIGFS_ATTR_RO(vhost_scsi_wwn_, version);
2565
2566static struct configfs_attribute *vhost_scsi_wwn_attrs[] = {
2567	&vhost_scsi_wwn_attr_version,
2568	NULL,
2569};
2570
2571static const struct target_core_fabric_ops vhost_scsi_ops = {
2572	.module				= THIS_MODULE,
2573	.fabric_name			= "vhost",
2574	.max_data_sg_nents		= VHOST_SCSI_PREALLOC_SGLS,
2575	.tpg_get_wwn			= vhost_scsi_get_fabric_wwn,
2576	.tpg_get_tag			= vhost_scsi_get_tpgt,
2577	.tpg_check_demo_mode		= vhost_scsi_check_true,
2578	.tpg_check_demo_mode_cache	= vhost_scsi_check_true,
 
 
2579	.tpg_check_prot_fabric_only	= vhost_scsi_check_prot_fabric_only,
 
2580	.release_cmd			= vhost_scsi_release_cmd,
2581	.check_stop_free		= vhost_scsi_check_stop_free,
 
2582	.sess_get_initiator_sid		= NULL,
2583	.write_pending			= vhost_scsi_write_pending,
 
 
2584	.queue_data_in			= vhost_scsi_queue_data_in,
2585	.queue_status			= vhost_scsi_queue_status,
2586	.queue_tm_rsp			= vhost_scsi_queue_tm_rsp,
2587	.aborted_task			= vhost_scsi_aborted_task,
2588	/*
2589	 * Setup callers for generic logic in target_core_fabric_configfs.c
2590	 */
2591	.fabric_make_wwn		= vhost_scsi_make_tport,
2592	.fabric_drop_wwn		= vhost_scsi_drop_tport,
2593	.fabric_make_tpg		= vhost_scsi_make_tpg,
2594	.fabric_drop_tpg		= vhost_scsi_drop_tpg,
2595	.fabric_post_link		= vhost_scsi_port_link,
2596	.fabric_pre_unlink		= vhost_scsi_port_unlink,
2597
2598	.tfc_wwn_attrs			= vhost_scsi_wwn_attrs,
2599	.tfc_tpg_base_attrs		= vhost_scsi_tpg_attrs,
2600	.tfc_tpg_attrib_attrs		= vhost_scsi_tpg_attrib_attrs,
2601
2602	.default_submit_type		= TARGET_QUEUE_SUBMIT,
2603	.direct_submit_supp		= 1,
2604};
2605
2606static int __init vhost_scsi_init(void)
2607{
2608	int ret = -ENOMEM;
2609
2610	pr_debug("TCM_VHOST fabric module %s on %s/%s"
2611		" on "UTS_RELEASE"\n", VHOST_SCSI_VERSION, utsname()->sysname,
2612		utsname()->machine);
2613
2614	ret = vhost_scsi_register();
2615	if (ret < 0)
2616		goto out;
2617
2618	ret = target_register_template(&vhost_scsi_ops);
2619	if (ret < 0)
2620		goto out_vhost_scsi_deregister;
2621
2622	return 0;
2623
2624out_vhost_scsi_deregister:
2625	vhost_scsi_deregister();
2626out:
2627	return ret;
2628};
2629
2630static void vhost_scsi_exit(void)
2631{
2632	target_unregister_template(&vhost_scsi_ops);
2633	vhost_scsi_deregister();
2634};
2635
2636MODULE_DESCRIPTION("VHOST_SCSI series fabric driver");
2637MODULE_ALIAS("tcm_vhost");
2638MODULE_LICENSE("GPL");
2639module_init(vhost_scsi_init);
2640module_exit(vhost_scsi_exit);
v5.14.15
 
   1/*******************************************************************************
   2 * Vhost kernel TCM fabric driver for virtio SCSI initiators
   3 *
   4 * (C) Copyright 2010-2013 Datera, Inc.
   5 * (C) Copyright 2010-2012 IBM Corp.
   6 *
   7 * Licensed to the Linux Foundation under the General Public License (GPL) version 2.
   8 *
   9 * Authors: Nicholas A. Bellinger <nab@daterainc.com>
  10 *          Stefan Hajnoczi <stefanha@linux.vnet.ibm.com>
  11 *
  12 * This program is free software; you can redistribute it and/or modify
  13 * it under the terms of the GNU General Public License as published by
  14 * the Free Software Foundation; either version 2 of the License, or
  15 * (at your option) any later version.
  16 *
  17 * This program is distributed in the hope that it will be useful,
  18 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  19 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20 * GNU General Public License for more details.
  21 *
  22 ****************************************************************************/
  23
  24#include <linux/module.h>
  25#include <linux/moduleparam.h>
  26#include <generated/utsrelease.h>
  27#include <linux/utsname.h>
  28#include <linux/init.h>
  29#include <linux/slab.h>
  30#include <linux/kthread.h>
  31#include <linux/types.h>
  32#include <linux/string.h>
  33#include <linux/configfs.h>
  34#include <linux/ctype.h>
  35#include <linux/compat.h>
  36#include <linux/eventfd.h>
  37#include <linux/fs.h>
  38#include <linux/vmalloc.h>
  39#include <linux/miscdevice.h>
 
 
  40#include <asm/unaligned.h>
  41#include <scsi/scsi_common.h>
  42#include <scsi/scsi_proto.h>
  43#include <target/target_core_base.h>
  44#include <target/target_core_fabric.h>
  45#include <linux/vhost.h>
  46#include <linux/virtio_scsi.h>
  47#include <linux/llist.h>
  48#include <linux/bitmap.h>
  49
  50#include "vhost.h"
  51
  52#define VHOST_SCSI_VERSION  "v0.1"
  53#define VHOST_SCSI_NAMELEN 256
  54#define VHOST_SCSI_MAX_CDB_SIZE 32
  55#define VHOST_SCSI_PREALLOC_SGLS 2048
  56#define VHOST_SCSI_PREALLOC_UPAGES 2048
  57#define VHOST_SCSI_PREALLOC_PROT_SGLS 2048
  58
  59/* Max number of requests before requeueing the job.
  60 * Using this limit prevents one virtqueue from starving others with
  61 * request.
  62 */
  63#define VHOST_SCSI_WEIGHT 256
  64
  65struct vhost_scsi_inflight {
  66	/* Wait for the flush operation to finish */
  67	struct completion comp;
  68	/* Refcount for the inflight reqs */
  69	struct kref kref;
  70};
  71
  72struct vhost_scsi_cmd {
  73	/* Descriptor from vhost_get_vq_desc() for virt_queue segment */
  74	int tvc_vq_desc;
  75	/* virtio-scsi initiator task attribute */
  76	int tvc_task_attr;
  77	/* virtio-scsi response incoming iovecs */
  78	int tvc_in_iovs;
  79	/* virtio-scsi initiator data direction */
  80	enum dma_data_direction tvc_data_direction;
  81	/* Expected data transfer length from virtio-scsi header */
  82	u32 tvc_exp_data_len;
  83	/* The Tag from include/linux/virtio_scsi.h:struct virtio_scsi_cmd_req */
  84	u64 tvc_tag;
  85	/* The number of scatterlists associated with this cmd */
  86	u32 tvc_sgl_count;
  87	u32 tvc_prot_sgl_count;
  88	/* Saved unpacked SCSI LUN for vhost_scsi_target_queue_cmd() */
  89	u32 tvc_lun;
 
 
 
  90	/* Pointer to the SGL formatted memory from virtio-scsi */
  91	struct scatterlist *tvc_sgl;
  92	struct scatterlist *tvc_prot_sgl;
  93	struct page **tvc_upages;
  94	/* Pointer to response header iovec */
  95	struct iovec tvc_resp_iov;
  96	/* Pointer to vhost_scsi for our device */
  97	struct vhost_scsi *tvc_vhost;
  98	/* Pointer to vhost_virtqueue for the cmd */
  99	struct vhost_virtqueue *tvc_vq;
 100	/* Pointer to vhost nexus memory */
 101	struct vhost_scsi_nexus *tvc_nexus;
 102	/* The TCM I/O descriptor that is accessed via container_of() */
 103	struct se_cmd tvc_se_cmd;
 104	/* Copy of the incoming SCSI command descriptor block (CDB) */
 105	unsigned char tvc_cdb[VHOST_SCSI_MAX_CDB_SIZE];
 106	/* Sense buffer that will be mapped into outgoing status */
 107	unsigned char tvc_sense_buf[TRANSPORT_SENSE_BUFFER];
 108	/* Completed commands list, serviced from vhost worker thread */
 109	struct llist_node tvc_completion_list;
 110	/* Used to track inflight cmd */
 111	struct vhost_scsi_inflight *inflight;
 112};
 113
 114struct vhost_scsi_nexus {
 115	/* Pointer to TCM session for I_T Nexus */
 116	struct se_session *tvn_se_sess;
 117};
 118
 119struct vhost_scsi_tpg {
 120	/* Vhost port target portal group tag for TCM */
 121	u16 tport_tpgt;
 122	/* Used to track number of TPG Port/Lun Links wrt to explict I_T Nexus shutdown */
 123	int tv_tpg_port_count;
 124	/* Used for vhost_scsi device reference to tpg_nexus, protected by tv_tpg_mutex */
 125	int tv_tpg_vhost_count;
 126	/* Used for enabling T10-PI with legacy devices */
 127	int tv_fabric_prot_type;
 128	/* list for vhost_scsi_list */
 129	struct list_head tv_tpg_list;
 130	/* Used to protect access for tpg_nexus */
 131	struct mutex tv_tpg_mutex;
 132	/* Pointer to the TCM VHost I_T Nexus for this TPG endpoint */
 133	struct vhost_scsi_nexus *tpg_nexus;
 134	/* Pointer back to vhost_scsi_tport */
 135	struct vhost_scsi_tport *tport;
 136	/* Returned by vhost_scsi_make_tpg() */
 137	struct se_portal_group se_tpg;
 138	/* Pointer back to vhost_scsi, protected by tv_tpg_mutex */
 139	struct vhost_scsi *vhost_scsi;
 140	struct list_head tmf_queue;
 141};
 142
 143struct vhost_scsi_tport {
 144	/* SCSI protocol the tport is providing */
 145	u8 tport_proto_id;
 146	/* Binary World Wide unique Port Name for Vhost Target port */
 147	u64 tport_wwpn;
 148	/* ASCII formatted WWPN for Vhost Target port */
 149	char tport_name[VHOST_SCSI_NAMELEN];
 150	/* Returned by vhost_scsi_make_tport() */
 151	struct se_wwn tport_wwn;
 152};
 153
 154struct vhost_scsi_evt {
 155	/* event to be sent to guest */
 156	struct virtio_scsi_event event;
 157	/* event list, serviced from vhost worker thread */
 158	struct llist_node list;
 159};
 160
 161enum {
 162	VHOST_SCSI_VQ_CTL = 0,
 163	VHOST_SCSI_VQ_EVT = 1,
 164	VHOST_SCSI_VQ_IO = 2,
 165};
 166
 167/* Note: can't set VIRTIO_F_VERSION_1 yet, since that implies ANY_LAYOUT. */
 168enum {
 169	VHOST_SCSI_FEATURES = VHOST_FEATURES | (1ULL << VIRTIO_SCSI_F_HOTPLUG) |
 170					       (1ULL << VIRTIO_SCSI_F_T10_PI)
 171};
 172
 173#define VHOST_SCSI_MAX_TARGET	256
 174#define VHOST_SCSI_MAX_VQ	128
 175#define VHOST_SCSI_MAX_EVENT	128
 176
 
 
 
 
 177struct vhost_scsi_virtqueue {
 178	struct vhost_virtqueue vq;
 
 179	/*
 180	 * Reference counting for inflight reqs, used for flush operation. At
 181	 * each time, one reference tracks new commands submitted, while we
 182	 * wait for another one to reach 0.
 183	 */
 184	struct vhost_scsi_inflight inflights[2];
 185	/*
 186	 * Indicate current inflight in use, protected by vq->mutex.
 187	 * Writers must also take dev mutex and flush under it.
 188	 */
 189	int inflight_idx;
 190	struct vhost_scsi_cmd *scsi_cmds;
 191	struct sbitmap scsi_tags;
 192	int max_cmds;
 
 
 
 193};
 194
 195struct vhost_scsi {
 196	/* Protected by vhost_scsi->dev.mutex */
 197	struct vhost_scsi_tpg **vs_tpg;
 198	char vs_vhost_wwpn[TRANSPORT_IQN_LEN];
 199
 200	struct vhost_dev dev;
 201	struct vhost_scsi_virtqueue vqs[VHOST_SCSI_MAX_VQ];
 202
 203	struct vhost_work vs_completion_work; /* cmd completion work item */
 204	struct llist_head vs_completion_list; /* cmd completion queue */
 205
 206	struct vhost_work vs_event_work; /* evt injection work item */
 207	struct llist_head vs_event_list; /* evt injection queue */
 208
 209	bool vs_events_missed; /* any missed events, protected by vq->mutex */
 210	int vs_events_nr; /* num of pending events, protected by vq->mutex */
 211};
 212
 213struct vhost_scsi_tmf {
 214	struct vhost_work vwork;
 215	struct vhost_scsi_tpg *tpg;
 216	struct vhost_scsi *vhost;
 217	struct vhost_scsi_virtqueue *svq;
 218	struct list_head queue_entry;
 219
 220	struct se_cmd se_cmd;
 221	u8 scsi_resp;
 222	struct vhost_scsi_inflight *inflight;
 223	struct iovec resp_iov;
 224	int in_iovs;
 225	int vq_desc;
 226};
 227
 228/*
 229 * Context for processing request and control queue operations.
 230 */
 231struct vhost_scsi_ctx {
 232	int head;
 233	unsigned int out, in;
 234	size_t req_size, rsp_size;
 235	size_t out_size, in_size;
 236	u8 *target, *lunp;
 237	void *req;
 238	struct iov_iter out_iter;
 239};
 240
 241/* Global spinlock to protect vhost_scsi TPG list for vhost IOCTL access */
 
 
 
 242static DEFINE_MUTEX(vhost_scsi_mutex);
 243static LIST_HEAD(vhost_scsi_list);
 244
 245static void vhost_scsi_done_inflight(struct kref *kref)
 246{
 247	struct vhost_scsi_inflight *inflight;
 248
 249	inflight = container_of(kref, struct vhost_scsi_inflight, kref);
 250	complete(&inflight->comp);
 251}
 252
 253static void vhost_scsi_init_inflight(struct vhost_scsi *vs,
 254				    struct vhost_scsi_inflight *old_inflight[])
 255{
 256	struct vhost_scsi_inflight *new_inflight;
 257	struct vhost_virtqueue *vq;
 258	int idx, i;
 259
 260	for (i = 0; i < VHOST_SCSI_MAX_VQ; i++) {
 261		vq = &vs->vqs[i].vq;
 262
 263		mutex_lock(&vq->mutex);
 264
 265		/* store old infight */
 266		idx = vs->vqs[i].inflight_idx;
 267		if (old_inflight)
 268			old_inflight[i] = &vs->vqs[i].inflights[idx];
 269
 270		/* setup new infight */
 271		vs->vqs[i].inflight_idx = idx ^ 1;
 272		new_inflight = &vs->vqs[i].inflights[idx ^ 1];
 273		kref_init(&new_inflight->kref);
 274		init_completion(&new_inflight->comp);
 275
 276		mutex_unlock(&vq->mutex);
 277	}
 278}
 279
 280static struct vhost_scsi_inflight *
 281vhost_scsi_get_inflight(struct vhost_virtqueue *vq)
 282{
 283	struct vhost_scsi_inflight *inflight;
 284	struct vhost_scsi_virtqueue *svq;
 285
 286	svq = container_of(vq, struct vhost_scsi_virtqueue, vq);
 287	inflight = &svq->inflights[svq->inflight_idx];
 288	kref_get(&inflight->kref);
 289
 290	return inflight;
 291}
 292
 293static void vhost_scsi_put_inflight(struct vhost_scsi_inflight *inflight)
 294{
 295	kref_put(&inflight->kref, vhost_scsi_done_inflight);
 296}
 297
 298static int vhost_scsi_check_true(struct se_portal_group *se_tpg)
 299{
 300	return 1;
 301}
 302
 303static int vhost_scsi_check_false(struct se_portal_group *se_tpg)
 304{
 305	return 0;
 306}
 307
 308static char *vhost_scsi_get_fabric_wwn(struct se_portal_group *se_tpg)
 309{
 310	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
 311				struct vhost_scsi_tpg, se_tpg);
 312	struct vhost_scsi_tport *tport = tpg->tport;
 313
 314	return &tport->tport_name[0];
 315}
 316
 317static u16 vhost_scsi_get_tpgt(struct se_portal_group *se_tpg)
 318{
 319	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
 320				struct vhost_scsi_tpg, se_tpg);
 321	return tpg->tport_tpgt;
 322}
 323
 324static int vhost_scsi_check_prot_fabric_only(struct se_portal_group *se_tpg)
 325{
 326	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
 327				struct vhost_scsi_tpg, se_tpg);
 328
 329	return tpg->tv_fabric_prot_type;
 330}
 331
 332static u32 vhost_scsi_tpg_get_inst_index(struct se_portal_group *se_tpg)
 333{
 334	return 1;
 335}
 336
 337static void vhost_scsi_release_cmd_res(struct se_cmd *se_cmd)
 338{
 339	struct vhost_scsi_cmd *tv_cmd = container_of(se_cmd,
 340				struct vhost_scsi_cmd, tvc_se_cmd);
 341	struct vhost_scsi_virtqueue *svq = container_of(tv_cmd->tvc_vq,
 342				struct vhost_scsi_virtqueue, vq);
 343	struct vhost_scsi_inflight *inflight = tv_cmd->inflight;
 344	int i;
 345
 346	if (tv_cmd->tvc_sgl_count) {
 347		for (i = 0; i < tv_cmd->tvc_sgl_count; i++)
 348			put_page(sg_page(&tv_cmd->tvc_sgl[i]));
 
 
 
 
 
 349	}
 350	if (tv_cmd->tvc_prot_sgl_count) {
 351		for (i = 0; i < tv_cmd->tvc_prot_sgl_count; i++)
 352			put_page(sg_page(&tv_cmd->tvc_prot_sgl[i]));
 353	}
 354
 355	sbitmap_clear_bit(&svq->scsi_tags, se_cmd->map_tag);
 356	vhost_scsi_put_inflight(inflight);
 357}
 358
 359static void vhost_scsi_release_tmf_res(struct vhost_scsi_tmf *tmf)
 360{
 361	struct vhost_scsi_tpg *tpg = tmf->tpg;
 362	struct vhost_scsi_inflight *inflight = tmf->inflight;
 363
 364	mutex_lock(&tpg->tv_tpg_mutex);
 365	list_add_tail(&tpg->tmf_queue, &tmf->queue_entry);
 366	mutex_unlock(&tpg->tv_tpg_mutex);
 367	vhost_scsi_put_inflight(inflight);
 368}
 369
 370static void vhost_scsi_release_cmd(struct se_cmd *se_cmd)
 371{
 372	if (se_cmd->se_cmd_flags & SCF_SCSI_TMR_CDB) {
 373		struct vhost_scsi_tmf *tmf = container_of(se_cmd,
 374					struct vhost_scsi_tmf, se_cmd);
 
 375
 376		vhost_work_queue(&tmf->vhost->dev, &tmf->vwork);
 377	} else {
 378		struct vhost_scsi_cmd *cmd = container_of(se_cmd,
 379					struct vhost_scsi_cmd, tvc_se_cmd);
 380		struct vhost_scsi *vs = cmd->tvc_vhost;
 
 381
 382		llist_add(&cmd->tvc_completion_list, &vs->vs_completion_list);
 383		vhost_work_queue(&vs->dev, &vs->vs_completion_work);
 384	}
 385}
 386
 387static u32 vhost_scsi_sess_get_index(struct se_session *se_sess)
 388{
 389	return 0;
 390}
 391
 392static int vhost_scsi_write_pending(struct se_cmd *se_cmd)
 393{
 394	/* Go ahead and process the write immediately */
 395	target_execute_cmd(se_cmd);
 396	return 0;
 397}
 398
 399static void vhost_scsi_set_default_node_attrs(struct se_node_acl *nacl)
 400{
 401	return;
 402}
 403
 404static int vhost_scsi_get_cmd_state(struct se_cmd *se_cmd)
 405{
 406	return 0;
 407}
 408
 409static int vhost_scsi_queue_data_in(struct se_cmd *se_cmd)
 410{
 411	transport_generic_free_cmd(se_cmd, 0);
 412	return 0;
 413}
 414
 415static int vhost_scsi_queue_status(struct se_cmd *se_cmd)
 416{
 417	transport_generic_free_cmd(se_cmd, 0);
 418	return 0;
 419}
 420
 421static void vhost_scsi_queue_tm_rsp(struct se_cmd *se_cmd)
 422{
 423	struct vhost_scsi_tmf *tmf = container_of(se_cmd, struct vhost_scsi_tmf,
 424						  se_cmd);
 425
 426	tmf->scsi_resp = se_cmd->se_tmr_req->response;
 427	transport_generic_free_cmd(&tmf->se_cmd, 0);
 428}
 429
 430static void vhost_scsi_aborted_task(struct se_cmd *se_cmd)
 431{
 432	return;
 433}
 434
 435static void vhost_scsi_free_evt(struct vhost_scsi *vs, struct vhost_scsi_evt *evt)
 436{
 437	vs->vs_events_nr--;
 438	kfree(evt);
 439}
 440
 441static struct vhost_scsi_evt *
 442vhost_scsi_allocate_evt(struct vhost_scsi *vs,
 443		       u32 event, u32 reason)
 444{
 445	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
 446	struct vhost_scsi_evt *evt;
 447
 448	if (vs->vs_events_nr > VHOST_SCSI_MAX_EVENT) {
 449		vs->vs_events_missed = true;
 450		return NULL;
 451	}
 452
 453	evt = kzalloc(sizeof(*evt), GFP_KERNEL);
 454	if (!evt) {
 455		vq_err(vq, "Failed to allocate vhost_scsi_evt\n");
 456		vs->vs_events_missed = true;
 457		return NULL;
 458	}
 459
 460	evt->event.event = cpu_to_vhost32(vq, event);
 461	evt->event.reason = cpu_to_vhost32(vq, reason);
 462	vs->vs_events_nr++;
 463
 464	return evt;
 465}
 466
 467static int vhost_scsi_check_stop_free(struct se_cmd *se_cmd)
 468{
 469	return target_put_sess_cmd(se_cmd);
 470}
 471
 472static void
 473vhost_scsi_do_evt_work(struct vhost_scsi *vs, struct vhost_scsi_evt *evt)
 474{
 475	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
 476	struct virtio_scsi_event *event = &evt->event;
 477	struct virtio_scsi_event __user *eventp;
 478	unsigned out, in;
 479	int head, ret;
 480
 481	if (!vhost_vq_get_backend(vq)) {
 482		vs->vs_events_missed = true;
 483		return;
 484	}
 485
 486again:
 487	vhost_disable_notify(&vs->dev, vq);
 488	head = vhost_get_vq_desc(vq, vq->iov,
 489			ARRAY_SIZE(vq->iov), &out, &in,
 490			NULL, NULL);
 491	if (head < 0) {
 492		vs->vs_events_missed = true;
 493		return;
 494	}
 495	if (head == vq->num) {
 496		if (vhost_enable_notify(&vs->dev, vq))
 497			goto again;
 498		vs->vs_events_missed = true;
 499		return;
 500	}
 501
 502	if ((vq->iov[out].iov_len != sizeof(struct virtio_scsi_event))) {
 503		vq_err(vq, "Expecting virtio_scsi_event, got %zu bytes\n",
 504				vq->iov[out].iov_len);
 505		vs->vs_events_missed = true;
 506		return;
 507	}
 508
 509	if (vs->vs_events_missed) {
 510		event->event |= cpu_to_vhost32(vq, VIRTIO_SCSI_T_EVENTS_MISSED);
 511		vs->vs_events_missed = false;
 512	}
 513
 514	eventp = vq->iov[out].iov_base;
 515	ret = __copy_to_user(eventp, event, sizeof(*event));
 516	if (!ret)
 517		vhost_add_used_and_signal(&vs->dev, vq, head, 0);
 518	else
 519		vq_err(vq, "Faulted on vhost_scsi_send_event\n");
 520}
 521
 522static void vhost_scsi_evt_work(struct vhost_work *work)
 523{
 524	struct vhost_scsi *vs = container_of(work, struct vhost_scsi,
 525					vs_event_work);
 526	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
 527	struct vhost_scsi_evt *evt, *t;
 528	struct llist_node *llnode;
 529
 530	mutex_lock(&vq->mutex);
 531	llnode = llist_del_all(&vs->vs_event_list);
 532	llist_for_each_entry_safe(evt, t, llnode, list) {
 533		vhost_scsi_do_evt_work(vs, evt);
 534		vhost_scsi_free_evt(vs, evt);
 535	}
 536	mutex_unlock(&vq->mutex);
 537}
 538
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 539/* Fill in status and signal that we are done processing this command
 540 *
 541 * This is scheduled in the vhost work queue so we are called with the owner
 542 * process mm and can access the vring.
 543 */
 544static void vhost_scsi_complete_cmd_work(struct vhost_work *work)
 545{
 546	struct vhost_scsi *vs = container_of(work, struct vhost_scsi,
 547					vs_completion_work);
 548	DECLARE_BITMAP(signal, VHOST_SCSI_MAX_VQ);
 549	struct virtio_scsi_cmd_resp v_rsp;
 550	struct vhost_scsi_cmd *cmd, *t;
 551	struct llist_node *llnode;
 552	struct se_cmd *se_cmd;
 553	struct iov_iter iov_iter;
 554	int ret, vq;
 
 555
 556	bitmap_zero(signal, VHOST_SCSI_MAX_VQ);
 557	llnode = llist_del_all(&vs->vs_completion_list);
 558	llist_for_each_entry_safe(cmd, t, llnode, tvc_completion_list) {
 559		se_cmd = &cmd->tvc_se_cmd;
 560
 561		pr_debug("%s tv_cmd %p resid %u status %#02x\n", __func__,
 562			cmd, se_cmd->residual_count, se_cmd->scsi_status);
 
 563
 564		memset(&v_rsp, 0, sizeof(v_rsp));
 565		v_rsp.resid = cpu_to_vhost32(cmd->tvc_vq, se_cmd->residual_count);
 566		/* TODO is status_qualifier field needed? */
 567		v_rsp.status = se_cmd->scsi_status;
 568		v_rsp.sense_len = cpu_to_vhost32(cmd->tvc_vq,
 569						 se_cmd->scsi_sense_length);
 570		memcpy(v_rsp.sense, cmd->tvc_sense_buf,
 571		       se_cmd->scsi_sense_length);
 
 
 
 
 572
 573		iov_iter_init(&iov_iter, READ, &cmd->tvc_resp_iov,
 574			      cmd->tvc_in_iovs, sizeof(v_rsp));
 575		ret = copy_to_iter(&v_rsp, sizeof(v_rsp), &iov_iter);
 576		if (likely(ret == sizeof(v_rsp))) {
 577			struct vhost_scsi_virtqueue *q;
 
 578			vhost_add_used(cmd->tvc_vq, cmd->tvc_vq_desc, 0);
 579			q = container_of(cmd->tvc_vq, struct vhost_scsi_virtqueue, vq);
 580			vq = q - vs->vqs;
 581			__set_bit(vq, signal);
 582		} else
 583			pr_err("Faulted on virtio_scsi_cmd_resp\n");
 584
 585		vhost_scsi_release_cmd_res(se_cmd);
 586	}
 587
 588	vq = -1;
 589	while ((vq = find_next_bit(signal, VHOST_SCSI_MAX_VQ, vq + 1))
 590		< VHOST_SCSI_MAX_VQ)
 591		vhost_signal(&vs->dev, &vs->vqs[vq].vq);
 592}
 593
 594static struct vhost_scsi_cmd *
 595vhost_scsi_get_cmd(struct vhost_virtqueue *vq, struct vhost_scsi_tpg *tpg,
 596		   unsigned char *cdb, u64 scsi_tag, u16 lun, u8 task_attr,
 597		   u32 exp_data_len, int data_direction)
 598{
 599	struct vhost_scsi_virtqueue *svq = container_of(vq,
 600					struct vhost_scsi_virtqueue, vq);
 601	struct vhost_scsi_cmd *cmd;
 602	struct vhost_scsi_nexus *tv_nexus;
 603	struct scatterlist *sg, *prot_sg;
 
 604	struct page **pages;
 605	int tag;
 606
 607	tv_nexus = tpg->tpg_nexus;
 608	if (!tv_nexus) {
 609		pr_err("Unable to locate active struct vhost_scsi_nexus\n");
 610		return ERR_PTR(-EIO);
 611	}
 612
 613	tag = sbitmap_get(&svq->scsi_tags);
 614	if (tag < 0) {
 615		pr_err("Unable to obtain tag for vhost_scsi_cmd\n");
 616		return ERR_PTR(-ENOMEM);
 617	}
 618
 619	cmd = &svq->scsi_cmds[tag];
 620	sg = cmd->tvc_sgl;
 621	prot_sg = cmd->tvc_prot_sgl;
 622	pages = cmd->tvc_upages;
 
 623	memset(cmd, 0, sizeof(*cmd));
 624	cmd->tvc_sgl = sg;
 625	cmd->tvc_prot_sgl = prot_sg;
 626	cmd->tvc_upages = pages;
 627	cmd->tvc_se_cmd.map_tag = tag;
 628	cmd->tvc_tag = scsi_tag;
 629	cmd->tvc_lun = lun;
 630	cmd->tvc_task_attr = task_attr;
 631	cmd->tvc_exp_data_len = exp_data_len;
 632	cmd->tvc_data_direction = data_direction;
 633	cmd->tvc_nexus = tv_nexus;
 634	cmd->inflight = vhost_scsi_get_inflight(vq);
 
 635
 636	memcpy(cmd->tvc_cdb, cdb, VHOST_SCSI_MAX_CDB_SIZE);
 637
 638	return cmd;
 639}
 640
 641/*
 642 * Map a user memory range into a scatterlist
 643 *
 644 * Returns the number of scatterlist entries used or -errno on error.
 645 */
 646static int
 647vhost_scsi_map_to_sgl(struct vhost_scsi_cmd *cmd,
 648		      struct iov_iter *iter,
 649		      struct scatterlist *sgl,
 650		      bool write)
 651{
 652	struct page **pages = cmd->tvc_upages;
 653	struct scatterlist *sg = sgl;
 654	ssize_t bytes;
 655	size_t offset;
 656	unsigned int npages = 0;
 657
 658	bytes = iov_iter_get_pages(iter, pages, LONG_MAX,
 659				VHOST_SCSI_PREALLOC_UPAGES, &offset);
 660	/* No pages were pinned */
 661	if (bytes <= 0)
 662		return bytes < 0 ? bytes : -EFAULT;
 663
 664	iov_iter_advance(iter, bytes);
 
 665
 666	while (bytes) {
 667		unsigned n = min_t(unsigned, PAGE_SIZE - offset, bytes);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 668		sg_set_page(sg++, pages[npages++], n, offset);
 669		bytes -= n;
 670		offset = 0;
 671	}
 
 672	return npages;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 673}
 674
 675static int
 676vhost_scsi_calc_sgls(struct iov_iter *iter, size_t bytes, int max_sgls)
 677{
 678	int sgl_count = 0;
 679
 680	if (!iter || !iter->iov) {
 681		pr_err("%s: iter->iov is NULL, but expected bytes: %zu"
 682		       " present\n", __func__, bytes);
 683		return -EINVAL;
 684	}
 685
 686	sgl_count = iov_iter_npages(iter, 0xffff);
 687	if (sgl_count > max_sgls) {
 688		pr_err("%s: requested sgl_count: %d exceeds pre-allocated"
 689		       " max_sgls: %d\n", __func__, sgl_count, max_sgls);
 690		return -EINVAL;
 691	}
 692	return sgl_count;
 693}
 694
 695static int
 696vhost_scsi_iov_to_sgl(struct vhost_scsi_cmd *cmd, bool write,
 697		      struct iov_iter *iter,
 698		      struct scatterlist *sg, int sg_count)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 699{
 700	struct scatterlist *p = sg;
 
 701	int ret;
 702
 703	while (iov_iter_count(iter)) {
 704		ret = vhost_scsi_map_to_sgl(cmd, iter, sg, write);
 705		if (ret < 0) {
 
 
 706			while (p < sg) {
 707				struct page *page = sg_page(p++);
 708				if (page)
 
 709					put_page(page);
 
 
 
 710			}
 
 
 711			return ret;
 712		}
 713		sg += ret;
 714	}
 
 715	return 0;
 716}
 717
 718static int
 719vhost_scsi_mapal(struct vhost_scsi_cmd *cmd,
 720		 size_t prot_bytes, struct iov_iter *prot_iter,
 721		 size_t data_bytes, struct iov_iter *data_iter)
 722{
 723	int sgl_count, ret;
 724	bool write = (cmd->tvc_data_direction == DMA_FROM_DEVICE);
 725
 726	if (prot_bytes) {
 727		sgl_count = vhost_scsi_calc_sgls(prot_iter, prot_bytes,
 728						 VHOST_SCSI_PREALLOC_PROT_SGLS);
 729		if (sgl_count < 0)
 730			return sgl_count;
 731
 732		sg_init_table(cmd->tvc_prot_sgl, sgl_count);
 733		cmd->tvc_prot_sgl_count = sgl_count;
 734		pr_debug("%s prot_sg %p prot_sgl_count %u\n", __func__,
 735			 cmd->tvc_prot_sgl, cmd->tvc_prot_sgl_count);
 736
 737		ret = vhost_scsi_iov_to_sgl(cmd, write, prot_iter,
 738					    cmd->tvc_prot_sgl,
 739					    cmd->tvc_prot_sgl_count);
 740		if (ret < 0) {
 741			cmd->tvc_prot_sgl_count = 0;
 742			return ret;
 743		}
 744	}
 745	sgl_count = vhost_scsi_calc_sgls(data_iter, data_bytes,
 746					 VHOST_SCSI_PREALLOC_SGLS);
 747	if (sgl_count < 0)
 748		return sgl_count;
 749
 750	sg_init_table(cmd->tvc_sgl, sgl_count);
 751	cmd->tvc_sgl_count = sgl_count;
 752	pr_debug("%s data_sg %p data_sgl_count %u\n", __func__,
 753		  cmd->tvc_sgl, cmd->tvc_sgl_count);
 754
 755	ret = vhost_scsi_iov_to_sgl(cmd, write, data_iter,
 756				    cmd->tvc_sgl, cmd->tvc_sgl_count);
 
 
 
 
 
 
 757	if (ret < 0) {
 758		cmd->tvc_sgl_count = 0;
 759		return ret;
 760	}
 761	return 0;
 762}
 763
 764static int vhost_scsi_to_tcm_attr(int attr)
 765{
 766	switch (attr) {
 767	case VIRTIO_SCSI_S_SIMPLE:
 768		return TCM_SIMPLE_TAG;
 769	case VIRTIO_SCSI_S_ORDERED:
 770		return TCM_ORDERED_TAG;
 771	case VIRTIO_SCSI_S_HEAD:
 772		return TCM_HEAD_TAG;
 773	case VIRTIO_SCSI_S_ACA:
 774		return TCM_ACA_TAG;
 775	default:
 776		break;
 777	}
 778	return TCM_SIMPLE_TAG;
 779}
 780
 781static void vhost_scsi_target_queue_cmd(struct vhost_scsi_cmd *cmd)
 782{
 783	struct se_cmd *se_cmd = &cmd->tvc_se_cmd;
 784	struct vhost_scsi_nexus *tv_nexus;
 785	struct scatterlist *sg_ptr, *sg_prot_ptr = NULL;
 786
 787	/* FIXME: BIDI operation */
 788	if (cmd->tvc_sgl_count) {
 789		sg_ptr = cmd->tvc_sgl;
 790
 791		if (cmd->tvc_prot_sgl_count)
 792			sg_prot_ptr = cmd->tvc_prot_sgl;
 793		else
 794			se_cmd->prot_pto = true;
 795	} else {
 796		sg_ptr = NULL;
 797	}
 798	tv_nexus = cmd->tvc_nexus;
 799
 800	se_cmd->tag = 0;
 801	target_init_cmd(se_cmd, tv_nexus->tvn_se_sess, &cmd->tvc_sense_buf[0],
 802			cmd->tvc_lun, cmd->tvc_exp_data_len,
 803			vhost_scsi_to_tcm_attr(cmd->tvc_task_attr),
 804			cmd->tvc_data_direction, TARGET_SCF_ACK_KREF);
 805
 806	if (target_submit_prep(se_cmd, cmd->tvc_cdb, sg_ptr,
 807			       cmd->tvc_sgl_count, NULL, 0, sg_prot_ptr,
 808			       cmd->tvc_prot_sgl_count, GFP_KERNEL))
 809		return;
 810
 811	target_queue_submission(se_cmd);
 812}
 813
 814static void
 815vhost_scsi_send_bad_target(struct vhost_scsi *vs,
 816			   struct vhost_virtqueue *vq,
 817			   int head, unsigned out)
 818{
 819	struct virtio_scsi_cmd_resp __user *resp;
 820	struct virtio_scsi_cmd_resp rsp;
 821	int ret;
 822
 823	memset(&rsp, 0, sizeof(rsp));
 824	rsp.response = VIRTIO_SCSI_S_BAD_TARGET;
 825	resp = vq->iov[out].iov_base;
 826	ret = __copy_to_user(resp, &rsp, sizeof(rsp));
 827	if (!ret)
 828		vhost_add_used_and_signal(&vs->dev, vq, head, 0);
 829	else
 830		pr_err("Faulted on virtio_scsi_cmd_resp\n");
 831}
 832
 833static int
 834vhost_scsi_get_desc(struct vhost_scsi *vs, struct vhost_virtqueue *vq,
 835		    struct vhost_scsi_ctx *vc)
 836{
 837	int ret = -ENXIO;
 838
 839	vc->head = vhost_get_vq_desc(vq, vq->iov,
 840				     ARRAY_SIZE(vq->iov), &vc->out, &vc->in,
 841				     NULL, NULL);
 842
 843	pr_debug("vhost_get_vq_desc: head: %d, out: %u in: %u\n",
 844		 vc->head, vc->out, vc->in);
 845
 846	/* On error, stop handling until the next kick. */
 847	if (unlikely(vc->head < 0))
 848		goto done;
 849
 850	/* Nothing new?  Wait for eventfd to tell us they refilled. */
 851	if (vc->head == vq->num) {
 852		if (unlikely(vhost_enable_notify(&vs->dev, vq))) {
 853			vhost_disable_notify(&vs->dev, vq);
 854			ret = -EAGAIN;
 855		}
 856		goto done;
 857	}
 858
 859	/*
 860	 * Get the size of request and response buffers.
 861	 * FIXME: Not correct for BIDI operation
 862	 */
 863	vc->out_size = iov_length(vq->iov, vc->out);
 864	vc->in_size = iov_length(&vq->iov[vc->out], vc->in);
 865
 866	/*
 867	 * Copy over the virtio-scsi request header, which for a
 868	 * ANY_LAYOUT enabled guest may span multiple iovecs, or a
 869	 * single iovec may contain both the header + outgoing
 870	 * WRITE payloads.
 871	 *
 872	 * copy_from_iter() will advance out_iter, so that it will
 873	 * point at the start of the outgoing WRITE payload, if
 874	 * DMA_TO_DEVICE is set.
 875	 */
 876	iov_iter_init(&vc->out_iter, WRITE, vq->iov, vc->out, vc->out_size);
 877	ret = 0;
 878
 879done:
 880	return ret;
 881}
 882
 883static int
 884vhost_scsi_chk_size(struct vhost_virtqueue *vq, struct vhost_scsi_ctx *vc)
 885{
 886	if (unlikely(vc->in_size < vc->rsp_size)) {
 887		vq_err(vq,
 888		       "Response buf too small, need min %zu bytes got %zu",
 889		       vc->rsp_size, vc->in_size);
 890		return -EINVAL;
 891	} else if (unlikely(vc->out_size < vc->req_size)) {
 892		vq_err(vq,
 893		       "Request buf too small, need min %zu bytes got %zu",
 894		       vc->req_size, vc->out_size);
 895		return -EIO;
 896	}
 897
 898	return 0;
 899}
 900
 901static int
 902vhost_scsi_get_req(struct vhost_virtqueue *vq, struct vhost_scsi_ctx *vc,
 903		   struct vhost_scsi_tpg **tpgp)
 904{
 905	int ret = -EIO;
 906
 907	if (unlikely(!copy_from_iter_full(vc->req, vc->req_size,
 908					  &vc->out_iter))) {
 909		vq_err(vq, "Faulted on copy_from_iter_full\n");
 910	} else if (unlikely(*vc->lunp != 1)) {
 911		/* virtio-scsi spec requires byte 0 of the lun to be 1 */
 912		vq_err(vq, "Illegal virtio-scsi lun: %u\n", *vc->lunp);
 913	} else {
 914		struct vhost_scsi_tpg **vs_tpg, *tpg;
 915
 916		vs_tpg = vhost_vq_get_backend(vq);	/* validated at handler entry */
 917
 918		tpg = READ_ONCE(vs_tpg[*vc->target]);
 919		if (unlikely(!tpg)) {
 920			vq_err(vq, "Target 0x%x does not exist\n", *vc->target);
 921		} else {
 922			if (tpgp)
 923				*tpgp = tpg;
 924			ret = 0;
 925		}
 926	}
 927
 928	return ret;
 929}
 930
 931static u16 vhost_buf_to_lun(u8 *lun_buf)
 932{
 933	return ((lun_buf[2] << 8) | lun_buf[3]) & 0x3FFF;
 934}
 935
 936static void
 937vhost_scsi_handle_vq(struct vhost_scsi *vs, struct vhost_virtqueue *vq)
 938{
 939	struct vhost_scsi_tpg **vs_tpg, *tpg;
 940	struct virtio_scsi_cmd_req v_req;
 941	struct virtio_scsi_cmd_req_pi v_req_pi;
 942	struct vhost_scsi_ctx vc;
 943	struct vhost_scsi_cmd *cmd;
 944	struct iov_iter in_iter, prot_iter, data_iter;
 945	u64 tag;
 946	u32 exp_data_len, data_direction;
 947	int ret, prot_bytes, c = 0;
 948	u16 lun;
 949	u8 task_attr;
 950	bool t10_pi = vhost_has_feature(vq, VIRTIO_SCSI_F_T10_PI);
 951	void *cdb;
 952
 953	mutex_lock(&vq->mutex);
 954	/*
 955	 * We can handle the vq only after the endpoint is setup by calling the
 956	 * VHOST_SCSI_SET_ENDPOINT ioctl.
 957	 */
 958	vs_tpg = vhost_vq_get_backend(vq);
 959	if (!vs_tpg)
 960		goto out;
 961
 962	memset(&vc, 0, sizeof(vc));
 963	vc.rsp_size = sizeof(struct virtio_scsi_cmd_resp);
 964
 965	vhost_disable_notify(&vs->dev, vq);
 966
 967	do {
 968		ret = vhost_scsi_get_desc(vs, vq, &vc);
 969		if (ret)
 970			goto err;
 971
 972		/*
 973		 * Setup pointers and values based upon different virtio-scsi
 974		 * request header if T10_PI is enabled in KVM guest.
 975		 */
 976		if (t10_pi) {
 977			vc.req = &v_req_pi;
 978			vc.req_size = sizeof(v_req_pi);
 979			vc.lunp = &v_req_pi.lun[0];
 980			vc.target = &v_req_pi.lun[1];
 981		} else {
 982			vc.req = &v_req;
 983			vc.req_size = sizeof(v_req);
 984			vc.lunp = &v_req.lun[0];
 985			vc.target = &v_req.lun[1];
 986		}
 987
 988		/*
 989		 * Validate the size of request and response buffers.
 990		 * Check for a sane response buffer so we can report
 991		 * early errors back to the guest.
 992		 */
 993		ret = vhost_scsi_chk_size(vq, &vc);
 994		if (ret)
 995			goto err;
 996
 997		ret = vhost_scsi_get_req(vq, &vc, &tpg);
 998		if (ret)
 999			goto err;
1000
1001		ret = -EIO;	/* bad target on any error from here on */
1002
1003		/*
1004		 * Determine data_direction by calculating the total outgoing
1005		 * iovec sizes + incoming iovec sizes vs. virtio-scsi request +
1006		 * response headers respectively.
1007		 *
1008		 * For DMA_TO_DEVICE this is out_iter, which is already pointing
1009		 * to the right place.
1010		 *
1011		 * For DMA_FROM_DEVICE, the iovec will be just past the end
1012		 * of the virtio-scsi response header in either the same
1013		 * or immediately following iovec.
1014		 *
1015		 * Any associated T10_PI bytes for the outgoing / incoming
1016		 * payloads are included in calculation of exp_data_len here.
1017		 */
1018		prot_bytes = 0;
1019
1020		if (vc.out_size > vc.req_size) {
1021			data_direction = DMA_TO_DEVICE;
1022			exp_data_len = vc.out_size - vc.req_size;
1023			data_iter = vc.out_iter;
1024		} else if (vc.in_size > vc.rsp_size) {
1025			data_direction = DMA_FROM_DEVICE;
1026			exp_data_len = vc.in_size - vc.rsp_size;
1027
1028			iov_iter_init(&in_iter, READ, &vq->iov[vc.out], vc.in,
1029				      vc.rsp_size + exp_data_len);
1030			iov_iter_advance(&in_iter, vc.rsp_size);
1031			data_iter = in_iter;
1032		} else {
1033			data_direction = DMA_NONE;
1034			exp_data_len = 0;
1035		}
1036		/*
1037		 * If T10_PI header + payload is present, setup prot_iter values
1038		 * and recalculate data_iter for vhost_scsi_mapal() mapping to
1039		 * host scatterlists via get_user_pages_fast().
1040		 */
1041		if (t10_pi) {
1042			if (v_req_pi.pi_bytesout) {
1043				if (data_direction != DMA_TO_DEVICE) {
1044					vq_err(vq, "Received non zero pi_bytesout,"
1045						" but wrong data_direction\n");
1046					goto err;
1047				}
1048				prot_bytes = vhost32_to_cpu(vq, v_req_pi.pi_bytesout);
1049			} else if (v_req_pi.pi_bytesin) {
1050				if (data_direction != DMA_FROM_DEVICE) {
1051					vq_err(vq, "Received non zero pi_bytesin,"
1052						" but wrong data_direction\n");
1053					goto err;
1054				}
1055				prot_bytes = vhost32_to_cpu(vq, v_req_pi.pi_bytesin);
1056			}
1057			/*
1058			 * Set prot_iter to data_iter and truncate it to
1059			 * prot_bytes, and advance data_iter past any
1060			 * preceeding prot_bytes that may be present.
1061			 *
1062			 * Also fix up the exp_data_len to reflect only the
1063			 * actual data payload length.
1064			 */
1065			if (prot_bytes) {
1066				exp_data_len -= prot_bytes;
1067				prot_iter = data_iter;
1068				iov_iter_truncate(&prot_iter, prot_bytes);
1069				iov_iter_advance(&data_iter, prot_bytes);
1070			}
1071			tag = vhost64_to_cpu(vq, v_req_pi.tag);
1072			task_attr = v_req_pi.task_attr;
1073			cdb = &v_req_pi.cdb[0];
1074			lun = vhost_buf_to_lun(v_req_pi.lun);
1075		} else {
1076			tag = vhost64_to_cpu(vq, v_req.tag);
1077			task_attr = v_req.task_attr;
1078			cdb = &v_req.cdb[0];
1079			lun = vhost_buf_to_lun(v_req.lun);
1080		}
1081		/*
1082		 * Check that the received CDB size does not exceeded our
1083		 * hardcoded max for vhost-scsi, then get a pre-allocated
1084		 * cmd descriptor for the new virtio-scsi tag.
1085		 *
1086		 * TODO what if cdb was too small for varlen cdb header?
1087		 */
1088		if (unlikely(scsi_command_size(cdb) > VHOST_SCSI_MAX_CDB_SIZE)) {
1089			vq_err(vq, "Received SCSI CDB with command_size: %d that"
1090				" exceeds SCSI_MAX_VARLEN_CDB_SIZE: %d\n",
1091				scsi_command_size(cdb), VHOST_SCSI_MAX_CDB_SIZE);
1092				goto err;
1093		}
1094		cmd = vhost_scsi_get_cmd(vq, tpg, cdb, tag, lun, task_attr,
1095					 exp_data_len + prot_bytes,
1096					 data_direction);
1097		if (IS_ERR(cmd)) {
1098			vq_err(vq, "vhost_scsi_get_cmd failed %ld\n",
1099			       PTR_ERR(cmd));
1100			goto err;
1101		}
1102		cmd->tvc_vhost = vs;
1103		cmd->tvc_vq = vq;
1104		cmd->tvc_resp_iov = vq->iov[vc.out];
 
1105		cmd->tvc_in_iovs = vc.in;
1106
1107		pr_debug("vhost_scsi got command opcode: %#02x, lun: %d\n",
1108			 cmd->tvc_cdb[0], cmd->tvc_lun);
1109		pr_debug("cmd: %p exp_data_len: %d, prot_bytes: %d data_direction:"
1110			 " %d\n", cmd, exp_data_len, prot_bytes, data_direction);
1111
1112		if (data_direction != DMA_NONE) {
1113			if (unlikely(vhost_scsi_mapal(cmd, prot_bytes,
1114						      &prot_iter, exp_data_len,
1115						      &data_iter))) {
1116				vq_err(vq, "Failed to map iov to sgl\n");
1117				vhost_scsi_release_cmd_res(&cmd->tvc_se_cmd);
1118				goto err;
1119			}
1120		}
1121		/*
1122		 * Save the descriptor from vhost_get_vq_desc() to be used to
1123		 * complete the virtio-scsi request in TCM callback context via
1124		 * vhost_scsi_queue_data_in() and vhost_scsi_queue_status()
1125		 */
1126		cmd->tvc_vq_desc = vc.head;
1127		vhost_scsi_target_queue_cmd(cmd);
1128		ret = 0;
1129err:
1130		/*
1131		 * ENXIO:  No more requests, or read error, wait for next kick
1132		 * EINVAL: Invalid response buffer, drop the request
1133		 * EIO:    Respond with bad target
1134		 * EAGAIN: Pending request
1135		 */
1136		if (ret == -ENXIO)
1137			break;
1138		else if (ret == -EIO)
1139			vhost_scsi_send_bad_target(vs, vq, vc.head, vc.out);
1140	} while (likely(!vhost_exceeds_weight(vq, ++c, 0)));
1141out:
1142	mutex_unlock(&vq->mutex);
1143}
1144
1145static void
1146vhost_scsi_send_tmf_resp(struct vhost_scsi *vs, struct vhost_virtqueue *vq,
1147			 int in_iovs, int vq_desc, struct iovec *resp_iov,
1148			 int tmf_resp_code)
1149{
1150	struct virtio_scsi_ctrl_tmf_resp rsp;
1151	struct iov_iter iov_iter;
1152	int ret;
1153
1154	pr_debug("%s\n", __func__);
1155	memset(&rsp, 0, sizeof(rsp));
1156	rsp.response = tmf_resp_code;
1157
1158	iov_iter_init(&iov_iter, READ, resp_iov, in_iovs, sizeof(rsp));
1159
1160	ret = copy_to_iter(&rsp, sizeof(rsp), &iov_iter);
1161	if (likely(ret == sizeof(rsp)))
1162		vhost_add_used_and_signal(&vs->dev, vq, vq_desc, 0);
1163	else
1164		pr_err("Faulted on virtio_scsi_ctrl_tmf_resp\n");
1165}
1166
1167static void vhost_scsi_tmf_resp_work(struct vhost_work *work)
1168{
1169	struct vhost_scsi_tmf *tmf = container_of(work, struct vhost_scsi_tmf,
1170						  vwork);
1171	int resp_code;
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1172
1173	if (tmf->scsi_resp == TMR_FUNCTION_COMPLETE)
1174		resp_code = VIRTIO_SCSI_S_FUNCTION_SUCCEEDED;
1175	else
1176		resp_code = VIRTIO_SCSI_S_FUNCTION_REJECTED;
 
1177
1178	vhost_scsi_send_tmf_resp(tmf->vhost, &tmf->svq->vq, tmf->in_iovs,
1179				 tmf->vq_desc, &tmf->resp_iov, resp_code);
1180	vhost_scsi_release_tmf_res(tmf);
1181}
1182
1183static void
1184vhost_scsi_handle_tmf(struct vhost_scsi *vs, struct vhost_scsi_tpg *tpg,
1185		      struct vhost_virtqueue *vq,
1186		      struct virtio_scsi_ctrl_tmf_req *vtmf,
1187		      struct vhost_scsi_ctx *vc)
1188{
1189	struct vhost_scsi_virtqueue *svq = container_of(vq,
1190					struct vhost_scsi_virtqueue, vq);
1191	struct vhost_scsi_tmf *tmf;
1192
1193	if (vhost32_to_cpu(vq, vtmf->subtype) !=
1194	    VIRTIO_SCSI_T_TMF_LOGICAL_UNIT_RESET)
1195		goto send_reject;
1196
1197	if (!tpg->tpg_nexus || !tpg->tpg_nexus->tvn_se_sess) {
1198		pr_err("Unable to locate active struct vhost_scsi_nexus for LUN RESET.\n");
1199		goto send_reject;
1200	}
1201
1202	mutex_lock(&tpg->tv_tpg_mutex);
1203	if (list_empty(&tpg->tmf_queue)) {
1204		pr_err("Missing reserve TMF. Could not handle LUN RESET.\n");
1205		mutex_unlock(&tpg->tv_tpg_mutex);
1206		goto send_reject;
1207	}
1208
1209	tmf = list_first_entry(&tpg->tmf_queue, struct vhost_scsi_tmf,
1210			       queue_entry);
1211	list_del_init(&tmf->queue_entry);
1212	mutex_unlock(&tpg->tv_tpg_mutex);
1213
1214	tmf->tpg = tpg;
1215	tmf->vhost = vs;
1216	tmf->svq = svq;
1217	tmf->resp_iov = vq->iov[vc->out];
1218	tmf->vq_desc = vc->head;
1219	tmf->in_iovs = vc->in;
1220	tmf->inflight = vhost_scsi_get_inflight(vq);
1221
1222	if (target_submit_tmr(&tmf->se_cmd, tpg->tpg_nexus->tvn_se_sess, NULL,
1223			      vhost_buf_to_lun(vtmf->lun), NULL,
1224			      TMR_LUN_RESET, GFP_KERNEL, 0,
1225			      TARGET_SCF_ACK_KREF) < 0) {
1226		vhost_scsi_release_tmf_res(tmf);
1227		goto send_reject;
1228	}
1229
1230	return;
1231
1232send_reject:
1233	vhost_scsi_send_tmf_resp(vs, vq, vc->in, vc->head, &vq->iov[vc->out],
1234				 VIRTIO_SCSI_S_FUNCTION_REJECTED);
1235}
1236
1237static void
1238vhost_scsi_send_an_resp(struct vhost_scsi *vs,
1239			struct vhost_virtqueue *vq,
1240			struct vhost_scsi_ctx *vc)
1241{
1242	struct virtio_scsi_ctrl_an_resp rsp;
1243	struct iov_iter iov_iter;
1244	int ret;
1245
1246	pr_debug("%s\n", __func__);
1247	memset(&rsp, 0, sizeof(rsp));	/* event_actual = 0 */
1248	rsp.response = VIRTIO_SCSI_S_OK;
1249
1250	iov_iter_init(&iov_iter, READ, &vq->iov[vc->out], vc->in, sizeof(rsp));
1251
1252	ret = copy_to_iter(&rsp, sizeof(rsp), &iov_iter);
1253	if (likely(ret == sizeof(rsp)))
1254		vhost_add_used_and_signal(&vs->dev, vq, vc->head, 0);
1255	else
1256		pr_err("Faulted on virtio_scsi_ctrl_an_resp\n");
1257}
1258
1259static void
1260vhost_scsi_ctl_handle_vq(struct vhost_scsi *vs, struct vhost_virtqueue *vq)
1261{
1262	struct vhost_scsi_tpg *tpg;
1263	union {
1264		__virtio32 type;
1265		struct virtio_scsi_ctrl_an_req an;
1266		struct virtio_scsi_ctrl_tmf_req tmf;
1267	} v_req;
1268	struct vhost_scsi_ctx vc;
1269	size_t typ_size;
1270	int ret, c = 0;
1271
1272	mutex_lock(&vq->mutex);
1273	/*
1274	 * We can handle the vq only after the endpoint is setup by calling the
1275	 * VHOST_SCSI_SET_ENDPOINT ioctl.
1276	 */
1277	if (!vhost_vq_get_backend(vq))
1278		goto out;
1279
1280	memset(&vc, 0, sizeof(vc));
1281
1282	vhost_disable_notify(&vs->dev, vq);
1283
1284	do {
1285		ret = vhost_scsi_get_desc(vs, vq, &vc);
1286		if (ret)
1287			goto err;
1288
1289		/*
1290		 * Get the request type first in order to setup
1291		 * other parameters dependent on the type.
1292		 */
1293		vc.req = &v_req.type;
1294		typ_size = sizeof(v_req.type);
1295
1296		if (unlikely(!copy_from_iter_full(vc.req, typ_size,
1297						  &vc.out_iter))) {
1298			vq_err(vq, "Faulted on copy_from_iter tmf type\n");
1299			/*
1300			 * The size of the response buffer depends on the
1301			 * request type and must be validated against it.
1302			 * Since the request type is not known, don't send
1303			 * a response.
1304			 */
1305			continue;
1306		}
1307
1308		switch (vhost32_to_cpu(vq, v_req.type)) {
1309		case VIRTIO_SCSI_T_TMF:
1310			vc.req = &v_req.tmf;
1311			vc.req_size = sizeof(struct virtio_scsi_ctrl_tmf_req);
1312			vc.rsp_size = sizeof(struct virtio_scsi_ctrl_tmf_resp);
1313			vc.lunp = &v_req.tmf.lun[0];
1314			vc.target = &v_req.tmf.lun[1];
1315			break;
1316		case VIRTIO_SCSI_T_AN_QUERY:
1317		case VIRTIO_SCSI_T_AN_SUBSCRIBE:
1318			vc.req = &v_req.an;
1319			vc.req_size = sizeof(struct virtio_scsi_ctrl_an_req);
1320			vc.rsp_size = sizeof(struct virtio_scsi_ctrl_an_resp);
1321			vc.lunp = &v_req.an.lun[0];
1322			vc.target = NULL;
1323			break;
1324		default:
1325			vq_err(vq, "Unknown control request %d", v_req.type);
1326			continue;
1327		}
1328
1329		/*
1330		 * Validate the size of request and response buffers.
1331		 * Check for a sane response buffer so we can report
1332		 * early errors back to the guest.
1333		 */
1334		ret = vhost_scsi_chk_size(vq, &vc);
1335		if (ret)
1336			goto err;
1337
1338		/*
1339		 * Get the rest of the request now that its size is known.
1340		 */
1341		vc.req += typ_size;
1342		vc.req_size -= typ_size;
1343
1344		ret = vhost_scsi_get_req(vq, &vc, &tpg);
1345		if (ret)
1346			goto err;
1347
1348		if (v_req.type == VIRTIO_SCSI_T_TMF)
1349			vhost_scsi_handle_tmf(vs, tpg, vq, &v_req.tmf, &vc);
1350		else
1351			vhost_scsi_send_an_resp(vs, vq, &vc);
1352err:
1353		/*
1354		 * ENXIO:  No more requests, or read error, wait for next kick
1355		 * EINVAL: Invalid response buffer, drop the request
1356		 * EIO:    Respond with bad target
1357		 * EAGAIN: Pending request
1358		 */
1359		if (ret == -ENXIO)
1360			break;
1361		else if (ret == -EIO)
1362			vhost_scsi_send_bad_target(vs, vq, vc.head, vc.out);
1363	} while (likely(!vhost_exceeds_weight(vq, ++c, 0)));
1364out:
1365	mutex_unlock(&vq->mutex);
1366}
1367
1368static void vhost_scsi_ctl_handle_kick(struct vhost_work *work)
1369{
1370	struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
1371						poll.work);
1372	struct vhost_scsi *vs = container_of(vq->dev, struct vhost_scsi, dev);
1373
1374	pr_debug("%s: The handling func for control queue.\n", __func__);
1375	vhost_scsi_ctl_handle_vq(vs, vq);
1376}
1377
1378static void
1379vhost_scsi_send_evt(struct vhost_scsi *vs,
1380		   struct vhost_scsi_tpg *tpg,
1381		   struct se_lun *lun,
1382		   u32 event,
1383		   u32 reason)
1384{
1385	struct vhost_scsi_evt *evt;
1386
1387	evt = vhost_scsi_allocate_evt(vs, event, reason);
1388	if (!evt)
1389		return;
1390
1391	if (tpg && lun) {
1392		/* TODO: share lun setup code with virtio-scsi.ko */
1393		/*
1394		 * Note: evt->event is zeroed when we allocate it and
1395		 * lun[4-7] need to be zero according to virtio-scsi spec.
1396		 */
1397		evt->event.lun[0] = 0x01;
1398		evt->event.lun[1] = tpg->tport_tpgt;
1399		if (lun->unpacked_lun >= 256)
1400			evt->event.lun[2] = lun->unpacked_lun >> 8 | 0x40 ;
1401		evt->event.lun[3] = lun->unpacked_lun & 0xFF;
1402	}
1403
1404	llist_add(&evt->list, &vs->vs_event_list);
1405	vhost_work_queue(&vs->dev, &vs->vs_event_work);
1406}
1407
1408static void vhost_scsi_evt_handle_kick(struct vhost_work *work)
1409{
1410	struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
1411						poll.work);
1412	struct vhost_scsi *vs = container_of(vq->dev, struct vhost_scsi, dev);
1413
1414	mutex_lock(&vq->mutex);
1415	if (!vhost_vq_get_backend(vq))
1416		goto out;
1417
1418	if (vs->vs_events_missed)
1419		vhost_scsi_send_evt(vs, NULL, NULL, VIRTIO_SCSI_T_NO_EVENT, 0);
 
1420out:
1421	mutex_unlock(&vq->mutex);
1422}
1423
1424static void vhost_scsi_handle_kick(struct vhost_work *work)
1425{
1426	struct vhost_virtqueue *vq = container_of(work, struct vhost_virtqueue,
1427						poll.work);
1428	struct vhost_scsi *vs = container_of(vq->dev, struct vhost_scsi, dev);
1429
1430	vhost_scsi_handle_vq(vs, vq);
1431}
1432
1433/* Callers must hold dev mutex */
1434static void vhost_scsi_flush(struct vhost_scsi *vs)
1435{
1436	struct vhost_scsi_inflight *old_inflight[VHOST_SCSI_MAX_VQ];
1437	int i;
1438
1439	/* Init new inflight and remember the old inflight */
1440	vhost_scsi_init_inflight(vs, old_inflight);
1441
1442	/*
1443	 * The inflight->kref was initialized to 1. We decrement it here to
1444	 * indicate the start of the flush operation so that it will reach 0
1445	 * when all the reqs are finished.
1446	 */
1447	for (i = 0; i < VHOST_SCSI_MAX_VQ; i++)
1448		kref_put(&old_inflight[i]->kref, vhost_scsi_done_inflight);
1449
1450	/* Flush both the vhost poll and vhost work */
1451	vhost_work_dev_flush(&vs->dev);
1452
1453	/* Wait for all reqs issued before the flush to be finished */
1454	for (i = 0; i < VHOST_SCSI_MAX_VQ; i++)
1455		wait_for_completion(&old_inflight[i]->comp);
1456}
1457
1458static void vhost_scsi_destroy_vq_cmds(struct vhost_virtqueue *vq)
1459{
1460	struct vhost_scsi_virtqueue *svq = container_of(vq,
1461					struct vhost_scsi_virtqueue, vq);
1462	struct vhost_scsi_cmd *tv_cmd;
1463	unsigned int i;
1464
1465	if (!svq->scsi_cmds)
1466		return;
1467
1468	for (i = 0; i < svq->max_cmds; i++) {
1469		tv_cmd = &svq->scsi_cmds[i];
1470
1471		kfree(tv_cmd->tvc_sgl);
1472		kfree(tv_cmd->tvc_prot_sgl);
1473		kfree(tv_cmd->tvc_upages);
 
1474	}
1475
1476	sbitmap_free(&svq->scsi_tags);
1477	kfree(svq->scsi_cmds);
1478	svq->scsi_cmds = NULL;
1479}
1480
1481static int vhost_scsi_setup_vq_cmds(struct vhost_virtqueue *vq, int max_cmds)
1482{
1483	struct vhost_scsi_virtqueue *svq = container_of(vq,
1484					struct vhost_scsi_virtqueue, vq);
1485	struct vhost_scsi_cmd *tv_cmd;
1486	unsigned int i;
1487
1488	if (svq->scsi_cmds)
1489		return 0;
1490
1491	if (sbitmap_init_node(&svq->scsi_tags, max_cmds, -1, GFP_KERNEL,
1492			      NUMA_NO_NODE, false, true))
1493		return -ENOMEM;
1494	svq->max_cmds = max_cmds;
1495
1496	svq->scsi_cmds = kcalloc(max_cmds, sizeof(*tv_cmd), GFP_KERNEL);
1497	if (!svq->scsi_cmds) {
1498		sbitmap_free(&svq->scsi_tags);
1499		return -ENOMEM;
1500	}
1501
1502	for (i = 0; i < max_cmds; i++) {
1503		tv_cmd = &svq->scsi_cmds[i];
1504
1505		tv_cmd->tvc_sgl = kcalloc(VHOST_SCSI_PREALLOC_SGLS,
1506					  sizeof(struct scatterlist),
1507					  GFP_KERNEL);
1508		if (!tv_cmd->tvc_sgl) {
1509			pr_err("Unable to allocate tv_cmd->tvc_sgl\n");
1510			goto out;
1511		}
1512
1513		tv_cmd->tvc_upages = kcalloc(VHOST_SCSI_PREALLOC_UPAGES,
1514					     sizeof(struct page *),
1515					     GFP_KERNEL);
1516		if (!tv_cmd->tvc_upages) {
1517			pr_err("Unable to allocate tv_cmd->tvc_upages\n");
1518			goto out;
1519		}
1520
 
 
 
 
 
 
 
 
1521		tv_cmd->tvc_prot_sgl = kcalloc(VHOST_SCSI_PREALLOC_PROT_SGLS,
1522					       sizeof(struct scatterlist),
1523					       GFP_KERNEL);
1524		if (!tv_cmd->tvc_prot_sgl) {
1525			pr_err("Unable to allocate tv_cmd->tvc_prot_sgl\n");
1526			goto out;
1527		}
1528	}
1529	return 0;
1530out:
1531	vhost_scsi_destroy_vq_cmds(vq);
1532	return -ENOMEM;
1533}
1534
1535/*
1536 * Called from vhost_scsi_ioctl() context to walk the list of available
1537 * vhost_scsi_tpg with an active struct vhost_scsi_nexus
1538 *
1539 *  The lock nesting rule is:
1540 *    vhost_scsi_mutex -> vs->dev.mutex -> tpg->tv_tpg_mutex -> vq->mutex
1541 */
1542static int
1543vhost_scsi_set_endpoint(struct vhost_scsi *vs,
1544			struct vhost_scsi_target *t)
1545{
1546	struct se_portal_group *se_tpg;
1547	struct vhost_scsi_tport *tv_tport;
1548	struct vhost_scsi_tpg *tpg;
1549	struct vhost_scsi_tpg **vs_tpg;
1550	struct vhost_virtqueue *vq;
1551	int index, ret, i, len;
1552	bool match = false;
1553
1554	mutex_lock(&vhost_scsi_mutex);
1555	mutex_lock(&vs->dev.mutex);
1556
1557	/* Verify that ring has been setup correctly. */
1558	for (index = 0; index < vs->dev.nvqs; ++index) {
1559		/* Verify that ring has been setup correctly. */
1560		if (!vhost_vq_access_ok(&vs->vqs[index].vq)) {
1561			ret = -EFAULT;
1562			goto out;
1563		}
1564	}
1565
1566	len = sizeof(vs_tpg[0]) * VHOST_SCSI_MAX_TARGET;
1567	vs_tpg = kzalloc(len, GFP_KERNEL);
1568	if (!vs_tpg) {
1569		ret = -ENOMEM;
1570		goto out;
1571	}
1572	if (vs->vs_tpg)
1573		memcpy(vs_tpg, vs->vs_tpg, len);
1574
 
1575	list_for_each_entry(tpg, &vhost_scsi_list, tv_tpg_list) {
1576		mutex_lock(&tpg->tv_tpg_mutex);
1577		if (!tpg->tpg_nexus) {
1578			mutex_unlock(&tpg->tv_tpg_mutex);
1579			continue;
1580		}
1581		if (tpg->tv_tpg_vhost_count != 0) {
1582			mutex_unlock(&tpg->tv_tpg_mutex);
1583			continue;
1584		}
1585		tv_tport = tpg->tport;
1586
1587		if (!strcmp(tv_tport->tport_name, t->vhost_wwpn)) {
1588			if (vs->vs_tpg && vs->vs_tpg[tpg->tport_tpgt]) {
1589				mutex_unlock(&tpg->tv_tpg_mutex);
 
1590				ret = -EEXIST;
1591				goto undepend;
1592			}
1593			/*
1594			 * In order to ensure individual vhost-scsi configfs
1595			 * groups cannot be removed while in use by vhost ioctl,
1596			 * go ahead and take an explicit se_tpg->tpg_group.cg_item
1597			 * dependency now.
1598			 */
1599			se_tpg = &tpg->se_tpg;
1600			ret = target_depend_item(&se_tpg->tpg_group.cg_item);
1601			if (ret) {
1602				pr_warn("target_depend_item() failed: %d\n", ret);
1603				mutex_unlock(&tpg->tv_tpg_mutex);
 
1604				goto undepend;
1605			}
1606			tpg->tv_tpg_vhost_count++;
1607			tpg->vhost_scsi = vs;
1608			vs_tpg[tpg->tport_tpgt] = tpg;
1609			match = true;
1610		}
1611		mutex_unlock(&tpg->tv_tpg_mutex);
1612	}
 
1613
1614	if (match) {
1615		memcpy(vs->vs_vhost_wwpn, t->vhost_wwpn,
1616		       sizeof(vs->vs_vhost_wwpn));
1617
1618		for (i = VHOST_SCSI_VQ_IO; i < VHOST_SCSI_MAX_VQ; i++) {
1619			vq = &vs->vqs[i].vq;
1620			if (!vhost_vq_is_setup(vq))
1621				continue;
1622
1623			ret = vhost_scsi_setup_vq_cmds(vq, vq->num);
1624			if (ret)
1625				goto destroy_vq_cmds;
1626		}
1627
1628		for (i = 0; i < VHOST_SCSI_MAX_VQ; i++) {
1629			vq = &vs->vqs[i].vq;
1630			mutex_lock(&vq->mutex);
1631			vhost_vq_set_backend(vq, vs_tpg);
1632			vhost_vq_init_access(vq);
1633			mutex_unlock(&vq->mutex);
1634		}
1635		ret = 0;
1636	} else {
1637		ret = -EEXIST;
1638	}
1639
1640	/*
1641	 * Act as synchronize_rcu to make sure access to
1642	 * old vs->vs_tpg is finished.
1643	 */
1644	vhost_scsi_flush(vs);
1645	kfree(vs->vs_tpg);
1646	vs->vs_tpg = vs_tpg;
1647	goto out;
1648
1649destroy_vq_cmds:
1650	for (i--; i >= VHOST_SCSI_VQ_IO; i--) {
1651		if (!vhost_vq_get_backend(&vs->vqs[i].vq))
1652			vhost_scsi_destroy_vq_cmds(&vs->vqs[i].vq);
1653	}
1654undepend:
1655	for (i = 0; i < VHOST_SCSI_MAX_TARGET; i++) {
1656		tpg = vs_tpg[i];
1657		if (tpg) {
 
 
1658			tpg->tv_tpg_vhost_count--;
 
1659			target_undepend_item(&tpg->se_tpg.tpg_group.cg_item);
1660		}
1661	}
1662	kfree(vs_tpg);
1663out:
1664	mutex_unlock(&vs->dev.mutex);
1665	mutex_unlock(&vhost_scsi_mutex);
1666	return ret;
1667}
1668
1669static int
1670vhost_scsi_clear_endpoint(struct vhost_scsi *vs,
1671			  struct vhost_scsi_target *t)
1672{
1673	struct se_portal_group *se_tpg;
1674	struct vhost_scsi_tport *tv_tport;
1675	struct vhost_scsi_tpg *tpg;
1676	struct vhost_virtqueue *vq;
1677	bool match = false;
1678	int index, ret, i;
1679	u8 target;
1680
1681	mutex_lock(&vhost_scsi_mutex);
1682	mutex_lock(&vs->dev.mutex);
1683	/* Verify that ring has been setup correctly. */
1684	for (index = 0; index < vs->dev.nvqs; ++index) {
1685		if (!vhost_vq_access_ok(&vs->vqs[index].vq)) {
1686			ret = -EFAULT;
1687			goto err_dev;
1688		}
1689	}
1690
1691	if (!vs->vs_tpg) {
1692		ret = 0;
1693		goto err_dev;
1694	}
1695
1696	for (i = 0; i < VHOST_SCSI_MAX_TARGET; i++) {
1697		target = i;
1698		tpg = vs->vs_tpg[target];
1699		if (!tpg)
1700			continue;
1701
1702		mutex_lock(&tpg->tv_tpg_mutex);
1703		tv_tport = tpg->tport;
1704		if (!tv_tport) {
1705			ret = -ENODEV;
1706			goto err_tpg;
1707		}
1708
1709		if (strcmp(tv_tport->tport_name, t->vhost_wwpn)) {
1710			pr_warn("tv_tport->tport_name: %s, tpg->tport_tpgt: %hu"
1711				" does not match t->vhost_wwpn: %s, t->vhost_tpgt: %hu\n",
1712				tv_tport->tport_name, tpg->tport_tpgt,
1713				t->vhost_wwpn, t->vhost_tpgt);
1714			ret = -EINVAL;
1715			goto err_tpg;
1716		}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1717		tpg->tv_tpg_vhost_count--;
1718		tpg->vhost_scsi = NULL;
1719		vs->vs_tpg[target] = NULL;
1720		match = true;
1721		mutex_unlock(&tpg->tv_tpg_mutex);
1722		/*
1723		 * Release se_tpg->tpg_group.cg_item configfs dependency now
1724		 * to allow vhost-scsi WWPN se_tpg->tpg_group shutdown to occur.
1725		 */
1726		se_tpg = &tpg->se_tpg;
1727		target_undepend_item(&se_tpg->tpg_group.cg_item);
1728	}
1729	if (match) {
1730		for (i = 0; i < VHOST_SCSI_MAX_VQ; i++) {
1731			vq = &vs->vqs[i].vq;
1732			mutex_lock(&vq->mutex);
1733			vhost_vq_set_backend(vq, NULL);
1734			mutex_unlock(&vq->mutex);
1735		}
1736		/* Make sure cmds are not running before tearing them down. */
1737		vhost_scsi_flush(vs);
1738
1739		for (i = 0; i < VHOST_SCSI_MAX_VQ; i++) {
1740			vq = &vs->vqs[i].vq;
1741			vhost_scsi_destroy_vq_cmds(vq);
1742		}
1743	}
1744	/*
1745	 * Act as synchronize_rcu to make sure access to
1746	 * old vs->vs_tpg is finished.
1747	 */
1748	vhost_scsi_flush(vs);
1749	kfree(vs->vs_tpg);
1750	vs->vs_tpg = NULL;
1751	WARN_ON(vs->vs_events_nr);
1752	mutex_unlock(&vs->dev.mutex);
1753	mutex_unlock(&vhost_scsi_mutex);
1754	return 0;
1755
1756err_tpg:
1757	mutex_unlock(&tpg->tv_tpg_mutex);
1758err_dev:
1759	mutex_unlock(&vs->dev.mutex);
1760	mutex_unlock(&vhost_scsi_mutex);
1761	return ret;
1762}
1763
1764static int vhost_scsi_set_features(struct vhost_scsi *vs, u64 features)
1765{
1766	struct vhost_virtqueue *vq;
1767	int i;
1768
1769	if (features & ~VHOST_SCSI_FEATURES)
1770		return -EOPNOTSUPP;
1771
1772	mutex_lock(&vs->dev.mutex);
1773	if ((features & (1 << VHOST_F_LOG_ALL)) &&
1774	    !vhost_log_access_ok(&vs->dev)) {
1775		mutex_unlock(&vs->dev.mutex);
1776		return -EFAULT;
1777	}
1778
1779	for (i = 0; i < VHOST_SCSI_MAX_VQ; i++) {
1780		vq = &vs->vqs[i].vq;
1781		mutex_lock(&vq->mutex);
1782		vq->acked_features = features;
1783		mutex_unlock(&vq->mutex);
1784	}
1785	mutex_unlock(&vs->dev.mutex);
1786	return 0;
1787}
1788
1789static int vhost_scsi_open(struct inode *inode, struct file *f)
1790{
 
1791	struct vhost_scsi *vs;
1792	struct vhost_virtqueue **vqs;
1793	int r = -ENOMEM, i;
1794
1795	vs = kvzalloc(sizeof(*vs), GFP_KERNEL);
1796	if (!vs)
1797		goto err_vs;
1798
1799	vqs = kmalloc_array(VHOST_SCSI_MAX_VQ, sizeof(*vqs), GFP_KERNEL);
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1800	if (!vqs)
1801		goto err_vqs;
1802
1803	vhost_work_init(&vs->vs_completion_work, vhost_scsi_complete_cmd_work);
1804	vhost_work_init(&vs->vs_event_work, vhost_scsi_evt_work);
1805
1806	vs->vs_events_nr = 0;
1807	vs->vs_events_missed = false;
1808
1809	vqs[VHOST_SCSI_VQ_CTL] = &vs->vqs[VHOST_SCSI_VQ_CTL].vq;
1810	vqs[VHOST_SCSI_VQ_EVT] = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
1811	vs->vqs[VHOST_SCSI_VQ_CTL].vq.handle_kick = vhost_scsi_ctl_handle_kick;
1812	vs->vqs[VHOST_SCSI_VQ_EVT].vq.handle_kick = vhost_scsi_evt_handle_kick;
1813	for (i = VHOST_SCSI_VQ_IO; i < VHOST_SCSI_MAX_VQ; i++) {
1814		vqs[i] = &vs->vqs[i].vq;
1815		vs->vqs[i].vq.handle_kick = vhost_scsi_handle_kick;
 
 
 
 
 
 
1816	}
1817	vhost_dev_init(&vs->dev, vqs, VHOST_SCSI_MAX_VQ, UIO_MAXIOV,
1818		       VHOST_SCSI_WEIGHT, 0, true, NULL);
1819
1820	vhost_scsi_init_inflight(vs, NULL);
1821
1822	f->private_data = vs;
1823	return 0;
1824
 
 
1825err_vqs:
 
 
1826	kvfree(vs);
1827err_vs:
1828	return r;
1829}
1830
1831static int vhost_scsi_release(struct inode *inode, struct file *f)
1832{
1833	struct vhost_scsi *vs = f->private_data;
1834	struct vhost_scsi_target t;
1835
1836	mutex_lock(&vs->dev.mutex);
1837	memcpy(t.vhost_wwpn, vs->vs_vhost_wwpn, sizeof(t.vhost_wwpn));
1838	mutex_unlock(&vs->dev.mutex);
1839	vhost_scsi_clear_endpoint(vs, &t);
1840	vhost_dev_stop(&vs->dev);
1841	vhost_dev_cleanup(&vs->dev);
1842	/* Jobs can re-queue themselves in evt kick handler. Do extra flush. */
1843	vhost_scsi_flush(vs);
1844	kfree(vs->dev.vqs);
 
 
1845	kvfree(vs);
1846	return 0;
1847}
1848
1849static long
1850vhost_scsi_ioctl(struct file *f,
1851		 unsigned int ioctl,
1852		 unsigned long arg)
1853{
1854	struct vhost_scsi *vs = f->private_data;
1855	struct vhost_scsi_target backend;
1856	void __user *argp = (void __user *)arg;
1857	u64 __user *featurep = argp;
1858	u32 __user *eventsp = argp;
1859	u32 events_missed;
1860	u64 features;
1861	int r, abi_version = VHOST_SCSI_ABI_VERSION;
1862	struct vhost_virtqueue *vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
1863
1864	switch (ioctl) {
1865	case VHOST_SCSI_SET_ENDPOINT:
1866		if (copy_from_user(&backend, argp, sizeof backend))
1867			return -EFAULT;
1868		if (backend.reserved != 0)
1869			return -EOPNOTSUPP;
1870
1871		return vhost_scsi_set_endpoint(vs, &backend);
1872	case VHOST_SCSI_CLEAR_ENDPOINT:
1873		if (copy_from_user(&backend, argp, sizeof backend))
1874			return -EFAULT;
1875		if (backend.reserved != 0)
1876			return -EOPNOTSUPP;
1877
1878		return vhost_scsi_clear_endpoint(vs, &backend);
1879	case VHOST_SCSI_GET_ABI_VERSION:
1880		if (copy_to_user(argp, &abi_version, sizeof abi_version))
1881			return -EFAULT;
1882		return 0;
1883	case VHOST_SCSI_SET_EVENTS_MISSED:
1884		if (get_user(events_missed, eventsp))
1885			return -EFAULT;
1886		mutex_lock(&vq->mutex);
1887		vs->vs_events_missed = events_missed;
1888		mutex_unlock(&vq->mutex);
1889		return 0;
1890	case VHOST_SCSI_GET_EVENTS_MISSED:
1891		mutex_lock(&vq->mutex);
1892		events_missed = vs->vs_events_missed;
1893		mutex_unlock(&vq->mutex);
1894		if (put_user(events_missed, eventsp))
1895			return -EFAULT;
1896		return 0;
1897	case VHOST_GET_FEATURES:
1898		features = VHOST_SCSI_FEATURES;
1899		if (copy_to_user(featurep, &features, sizeof features))
1900			return -EFAULT;
1901		return 0;
1902	case VHOST_SET_FEATURES:
1903		if (copy_from_user(&features, featurep, sizeof features))
1904			return -EFAULT;
1905		return vhost_scsi_set_features(vs, features);
 
 
 
 
 
 
 
 
1906	default:
1907		mutex_lock(&vs->dev.mutex);
1908		r = vhost_dev_ioctl(&vs->dev, ioctl, argp);
1909		/* TODO: flush backend after dev ioctl. */
1910		if (r == -ENOIOCTLCMD)
1911			r = vhost_vring_ioctl(&vs->dev, ioctl, argp);
1912		mutex_unlock(&vs->dev.mutex);
1913		return r;
1914	}
1915}
1916
1917static const struct file_operations vhost_scsi_fops = {
1918	.owner          = THIS_MODULE,
1919	.release        = vhost_scsi_release,
1920	.unlocked_ioctl = vhost_scsi_ioctl,
1921	.compat_ioctl	= compat_ptr_ioctl,
1922	.open           = vhost_scsi_open,
1923	.llseek		= noop_llseek,
1924};
1925
1926static struct miscdevice vhost_scsi_misc = {
1927	MISC_DYNAMIC_MINOR,
1928	"vhost-scsi",
1929	&vhost_scsi_fops,
1930};
1931
1932static int __init vhost_scsi_register(void)
1933{
1934	return misc_register(&vhost_scsi_misc);
1935}
1936
1937static void vhost_scsi_deregister(void)
1938{
1939	misc_deregister(&vhost_scsi_misc);
1940}
1941
1942static char *vhost_scsi_dump_proto_id(struct vhost_scsi_tport *tport)
1943{
1944	switch (tport->tport_proto_id) {
1945	case SCSI_PROTOCOL_SAS:
1946		return "SAS";
1947	case SCSI_PROTOCOL_FCP:
1948		return "FCP";
1949	case SCSI_PROTOCOL_ISCSI:
1950		return "iSCSI";
1951	default:
1952		break;
1953	}
1954
1955	return "Unknown";
1956}
1957
1958static void
1959vhost_scsi_do_plug(struct vhost_scsi_tpg *tpg,
1960		  struct se_lun *lun, bool plug)
1961{
1962
1963	struct vhost_scsi *vs = tpg->vhost_scsi;
1964	struct vhost_virtqueue *vq;
1965	u32 reason;
1966
1967	if (!vs)
1968		return;
1969
1970	mutex_lock(&vs->dev.mutex);
1971
1972	if (plug)
1973		reason = VIRTIO_SCSI_EVT_RESET_RESCAN;
1974	else
1975		reason = VIRTIO_SCSI_EVT_RESET_REMOVED;
1976
1977	vq = &vs->vqs[VHOST_SCSI_VQ_EVT].vq;
1978	mutex_lock(&vq->mutex);
 
 
 
 
 
 
 
1979	if (vhost_has_feature(vq, VIRTIO_SCSI_F_HOTPLUG))
1980		vhost_scsi_send_evt(vs, tpg, lun,
1981				   VIRTIO_SCSI_T_TRANSPORT_RESET, reason);
 
1982	mutex_unlock(&vq->mutex);
1983	mutex_unlock(&vs->dev.mutex);
1984}
1985
1986static void vhost_scsi_hotplug(struct vhost_scsi_tpg *tpg, struct se_lun *lun)
1987{
1988	vhost_scsi_do_plug(tpg, lun, true);
1989}
1990
1991static void vhost_scsi_hotunplug(struct vhost_scsi_tpg *tpg, struct se_lun *lun)
1992{
1993	vhost_scsi_do_plug(tpg, lun, false);
1994}
1995
1996static int vhost_scsi_port_link(struct se_portal_group *se_tpg,
1997			       struct se_lun *lun)
1998{
1999	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2000				struct vhost_scsi_tpg, se_tpg);
2001	struct vhost_scsi_tmf *tmf;
2002
2003	tmf = kzalloc(sizeof(*tmf), GFP_KERNEL);
2004	if (!tmf)
2005		return -ENOMEM;
2006	INIT_LIST_HEAD(&tmf->queue_entry);
2007	vhost_work_init(&tmf->vwork, vhost_scsi_tmf_resp_work);
2008
2009	mutex_lock(&vhost_scsi_mutex);
2010
2011	mutex_lock(&tpg->tv_tpg_mutex);
2012	tpg->tv_tpg_port_count++;
2013	list_add_tail(&tmf->queue_entry, &tpg->tmf_queue);
2014	mutex_unlock(&tpg->tv_tpg_mutex);
2015
2016	vhost_scsi_hotplug(tpg, lun);
2017
2018	mutex_unlock(&vhost_scsi_mutex);
2019
2020	return 0;
2021}
2022
2023static void vhost_scsi_port_unlink(struct se_portal_group *se_tpg,
2024				  struct se_lun *lun)
2025{
2026	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2027				struct vhost_scsi_tpg, se_tpg);
2028	struct vhost_scsi_tmf *tmf;
2029
2030	mutex_lock(&vhost_scsi_mutex);
2031
2032	mutex_lock(&tpg->tv_tpg_mutex);
2033	tpg->tv_tpg_port_count--;
2034	tmf = list_first_entry(&tpg->tmf_queue, struct vhost_scsi_tmf,
2035			       queue_entry);
2036	list_del(&tmf->queue_entry);
2037	kfree(tmf);
2038	mutex_unlock(&tpg->tv_tpg_mutex);
2039
2040	vhost_scsi_hotunplug(tpg, lun);
2041
2042	mutex_unlock(&vhost_scsi_mutex);
2043}
2044
2045static ssize_t vhost_scsi_tpg_attrib_fabric_prot_type_store(
2046		struct config_item *item, const char *page, size_t count)
2047{
2048	struct se_portal_group *se_tpg = attrib_to_tpg(item);
2049	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2050				struct vhost_scsi_tpg, se_tpg);
2051	unsigned long val;
2052	int ret = kstrtoul(page, 0, &val);
2053
2054	if (ret) {
2055		pr_err("kstrtoul() returned %d for fabric_prot_type\n", ret);
2056		return ret;
2057	}
2058	if (val != 0 && val != 1 && val != 3) {
2059		pr_err("Invalid vhost_scsi fabric_prot_type: %lu\n", val);
2060		return -EINVAL;
2061	}
2062	tpg->tv_fabric_prot_type = val;
2063
2064	return count;
2065}
2066
2067static ssize_t vhost_scsi_tpg_attrib_fabric_prot_type_show(
2068		struct config_item *item, char *page)
2069{
2070	struct se_portal_group *se_tpg = attrib_to_tpg(item);
2071	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2072				struct vhost_scsi_tpg, se_tpg);
2073
2074	return sprintf(page, "%d\n", tpg->tv_fabric_prot_type);
2075}
2076
2077CONFIGFS_ATTR(vhost_scsi_tpg_attrib_, fabric_prot_type);
2078
2079static struct configfs_attribute *vhost_scsi_tpg_attrib_attrs[] = {
2080	&vhost_scsi_tpg_attrib_attr_fabric_prot_type,
2081	NULL,
2082};
2083
2084static int vhost_scsi_make_nexus(struct vhost_scsi_tpg *tpg,
2085				const char *name)
2086{
2087	struct vhost_scsi_nexus *tv_nexus;
2088
2089	mutex_lock(&tpg->tv_tpg_mutex);
2090	if (tpg->tpg_nexus) {
2091		mutex_unlock(&tpg->tv_tpg_mutex);
2092		pr_debug("tpg->tpg_nexus already exists\n");
2093		return -EEXIST;
2094	}
2095
2096	tv_nexus = kzalloc(sizeof(*tv_nexus), GFP_KERNEL);
2097	if (!tv_nexus) {
2098		mutex_unlock(&tpg->tv_tpg_mutex);
2099		pr_err("Unable to allocate struct vhost_scsi_nexus\n");
2100		return -ENOMEM;
2101	}
2102	/*
2103	 * Since we are running in 'demo mode' this call with generate a
2104	 * struct se_node_acl for the vhost_scsi struct se_portal_group with
2105	 * the SCSI Initiator port name of the passed configfs group 'name'.
2106	 */
2107	tv_nexus->tvn_se_sess = target_setup_session(&tpg->se_tpg, 0, 0,
2108					TARGET_PROT_DIN_PASS | TARGET_PROT_DOUT_PASS,
2109					(unsigned char *)name, tv_nexus, NULL);
2110	if (IS_ERR(tv_nexus->tvn_se_sess)) {
2111		mutex_unlock(&tpg->tv_tpg_mutex);
2112		kfree(tv_nexus);
2113		return -ENOMEM;
2114	}
2115	tpg->tpg_nexus = tv_nexus;
2116
2117	mutex_unlock(&tpg->tv_tpg_mutex);
2118	return 0;
2119}
2120
2121static int vhost_scsi_drop_nexus(struct vhost_scsi_tpg *tpg)
2122{
2123	struct se_session *se_sess;
2124	struct vhost_scsi_nexus *tv_nexus;
2125
2126	mutex_lock(&tpg->tv_tpg_mutex);
2127	tv_nexus = tpg->tpg_nexus;
2128	if (!tv_nexus) {
2129		mutex_unlock(&tpg->tv_tpg_mutex);
2130		return -ENODEV;
2131	}
2132
2133	se_sess = tv_nexus->tvn_se_sess;
2134	if (!se_sess) {
2135		mutex_unlock(&tpg->tv_tpg_mutex);
2136		return -ENODEV;
2137	}
2138
2139	if (tpg->tv_tpg_port_count != 0) {
2140		mutex_unlock(&tpg->tv_tpg_mutex);
2141		pr_err("Unable to remove TCM_vhost I_T Nexus with"
2142			" active TPG port count: %d\n",
2143			tpg->tv_tpg_port_count);
2144		return -EBUSY;
2145	}
2146
2147	if (tpg->tv_tpg_vhost_count != 0) {
2148		mutex_unlock(&tpg->tv_tpg_mutex);
2149		pr_err("Unable to remove TCM_vhost I_T Nexus with"
2150			" active TPG vhost count: %d\n",
2151			tpg->tv_tpg_vhost_count);
2152		return -EBUSY;
2153	}
2154
2155	pr_debug("TCM_vhost_ConfigFS: Removing I_T Nexus to emulated"
2156		" %s Initiator Port: %s\n", vhost_scsi_dump_proto_id(tpg->tport),
2157		tv_nexus->tvn_se_sess->se_node_acl->initiatorname);
2158
2159	/*
2160	 * Release the SCSI I_T Nexus to the emulated vhost Target Port
2161	 */
2162	target_remove_session(se_sess);
2163	tpg->tpg_nexus = NULL;
2164	mutex_unlock(&tpg->tv_tpg_mutex);
2165
2166	kfree(tv_nexus);
2167	return 0;
2168}
2169
2170static ssize_t vhost_scsi_tpg_nexus_show(struct config_item *item, char *page)
2171{
2172	struct se_portal_group *se_tpg = to_tpg(item);
2173	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2174				struct vhost_scsi_tpg, se_tpg);
2175	struct vhost_scsi_nexus *tv_nexus;
2176	ssize_t ret;
2177
2178	mutex_lock(&tpg->tv_tpg_mutex);
2179	tv_nexus = tpg->tpg_nexus;
2180	if (!tv_nexus) {
2181		mutex_unlock(&tpg->tv_tpg_mutex);
2182		return -ENODEV;
2183	}
2184	ret = snprintf(page, PAGE_SIZE, "%s\n",
2185			tv_nexus->tvn_se_sess->se_node_acl->initiatorname);
2186	mutex_unlock(&tpg->tv_tpg_mutex);
2187
2188	return ret;
2189}
2190
2191static ssize_t vhost_scsi_tpg_nexus_store(struct config_item *item,
2192		const char *page, size_t count)
2193{
2194	struct se_portal_group *se_tpg = to_tpg(item);
2195	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2196				struct vhost_scsi_tpg, se_tpg);
2197	struct vhost_scsi_tport *tport_wwn = tpg->tport;
2198	unsigned char i_port[VHOST_SCSI_NAMELEN], *ptr, *port_ptr;
2199	int ret;
2200	/*
2201	 * Shutdown the active I_T nexus if 'NULL' is passed..
2202	 */
2203	if (!strncmp(page, "NULL", 4)) {
2204		ret = vhost_scsi_drop_nexus(tpg);
2205		return (!ret) ? count : ret;
2206	}
2207	/*
2208	 * Otherwise make sure the passed virtual Initiator port WWN matches
2209	 * the fabric protocol_id set in vhost_scsi_make_tport(), and call
2210	 * vhost_scsi_make_nexus().
2211	 */
2212	if (strlen(page) >= VHOST_SCSI_NAMELEN) {
2213		pr_err("Emulated NAA Sas Address: %s, exceeds"
2214				" max: %d\n", page, VHOST_SCSI_NAMELEN);
2215		return -EINVAL;
2216	}
2217	snprintf(&i_port[0], VHOST_SCSI_NAMELEN, "%s", page);
2218
2219	ptr = strstr(i_port, "naa.");
2220	if (ptr) {
2221		if (tport_wwn->tport_proto_id != SCSI_PROTOCOL_SAS) {
2222			pr_err("Passed SAS Initiator Port %s does not"
2223				" match target port protoid: %s\n", i_port,
2224				vhost_scsi_dump_proto_id(tport_wwn));
2225			return -EINVAL;
2226		}
2227		port_ptr = &i_port[0];
2228		goto check_newline;
2229	}
2230	ptr = strstr(i_port, "fc.");
2231	if (ptr) {
2232		if (tport_wwn->tport_proto_id != SCSI_PROTOCOL_FCP) {
2233			pr_err("Passed FCP Initiator Port %s does not"
2234				" match target port protoid: %s\n", i_port,
2235				vhost_scsi_dump_proto_id(tport_wwn));
2236			return -EINVAL;
2237		}
2238		port_ptr = &i_port[3]; /* Skip over "fc." */
2239		goto check_newline;
2240	}
2241	ptr = strstr(i_port, "iqn.");
2242	if (ptr) {
2243		if (tport_wwn->tport_proto_id != SCSI_PROTOCOL_ISCSI) {
2244			pr_err("Passed iSCSI Initiator Port %s does not"
2245				" match target port protoid: %s\n", i_port,
2246				vhost_scsi_dump_proto_id(tport_wwn));
2247			return -EINVAL;
2248		}
2249		port_ptr = &i_port[0];
2250		goto check_newline;
2251	}
2252	pr_err("Unable to locate prefix for emulated Initiator Port:"
2253			" %s\n", i_port);
2254	return -EINVAL;
2255	/*
2256	 * Clear any trailing newline for the NAA WWN
2257	 */
2258check_newline:
2259	if (i_port[strlen(i_port)-1] == '\n')
2260		i_port[strlen(i_port)-1] = '\0';
2261
2262	ret = vhost_scsi_make_nexus(tpg, port_ptr);
2263	if (ret < 0)
2264		return ret;
2265
2266	return count;
2267}
2268
2269CONFIGFS_ATTR(vhost_scsi_tpg_, nexus);
2270
2271static struct configfs_attribute *vhost_scsi_tpg_attrs[] = {
2272	&vhost_scsi_tpg_attr_nexus,
2273	NULL,
2274};
2275
2276static struct se_portal_group *
2277vhost_scsi_make_tpg(struct se_wwn *wwn, const char *name)
2278{
2279	struct vhost_scsi_tport *tport = container_of(wwn,
2280			struct vhost_scsi_tport, tport_wwn);
2281
2282	struct vhost_scsi_tpg *tpg;
2283	u16 tpgt;
2284	int ret;
2285
2286	if (strstr(name, "tpgt_") != name)
2287		return ERR_PTR(-EINVAL);
2288	if (kstrtou16(name + 5, 10, &tpgt) || tpgt >= VHOST_SCSI_MAX_TARGET)
2289		return ERR_PTR(-EINVAL);
2290
2291	tpg = kzalloc(sizeof(*tpg), GFP_KERNEL);
2292	if (!tpg) {
2293		pr_err("Unable to allocate struct vhost_scsi_tpg");
2294		return ERR_PTR(-ENOMEM);
2295	}
2296	mutex_init(&tpg->tv_tpg_mutex);
2297	INIT_LIST_HEAD(&tpg->tv_tpg_list);
2298	INIT_LIST_HEAD(&tpg->tmf_queue);
2299	tpg->tport = tport;
2300	tpg->tport_tpgt = tpgt;
2301
2302	ret = core_tpg_register(wwn, &tpg->se_tpg, tport->tport_proto_id);
2303	if (ret < 0) {
2304		kfree(tpg);
2305		return NULL;
2306	}
2307	mutex_lock(&vhost_scsi_mutex);
2308	list_add_tail(&tpg->tv_tpg_list, &vhost_scsi_list);
2309	mutex_unlock(&vhost_scsi_mutex);
2310
2311	return &tpg->se_tpg;
2312}
2313
2314static void vhost_scsi_drop_tpg(struct se_portal_group *se_tpg)
2315{
2316	struct vhost_scsi_tpg *tpg = container_of(se_tpg,
2317				struct vhost_scsi_tpg, se_tpg);
2318
2319	mutex_lock(&vhost_scsi_mutex);
2320	list_del(&tpg->tv_tpg_list);
2321	mutex_unlock(&vhost_scsi_mutex);
2322	/*
2323	 * Release the virtual I_T Nexus for this vhost TPG
2324	 */
2325	vhost_scsi_drop_nexus(tpg);
2326	/*
2327	 * Deregister the se_tpg from TCM..
2328	 */
2329	core_tpg_deregister(se_tpg);
2330	kfree(tpg);
2331}
2332
2333static struct se_wwn *
2334vhost_scsi_make_tport(struct target_fabric_configfs *tf,
2335		     struct config_group *group,
2336		     const char *name)
2337{
2338	struct vhost_scsi_tport *tport;
2339	char *ptr;
2340	u64 wwpn = 0;
2341	int off = 0;
2342
2343	/* if (vhost_scsi_parse_wwn(name, &wwpn, 1) < 0)
2344		return ERR_PTR(-EINVAL); */
2345
2346	tport = kzalloc(sizeof(*tport), GFP_KERNEL);
2347	if (!tport) {
2348		pr_err("Unable to allocate struct vhost_scsi_tport");
2349		return ERR_PTR(-ENOMEM);
2350	}
2351	tport->tport_wwpn = wwpn;
2352	/*
2353	 * Determine the emulated Protocol Identifier and Target Port Name
2354	 * based on the incoming configfs directory name.
2355	 */
2356	ptr = strstr(name, "naa.");
2357	if (ptr) {
2358		tport->tport_proto_id = SCSI_PROTOCOL_SAS;
2359		goto check_len;
2360	}
2361	ptr = strstr(name, "fc.");
2362	if (ptr) {
2363		tport->tport_proto_id = SCSI_PROTOCOL_FCP;
2364		off = 3; /* Skip over "fc." */
2365		goto check_len;
2366	}
2367	ptr = strstr(name, "iqn.");
2368	if (ptr) {
2369		tport->tport_proto_id = SCSI_PROTOCOL_ISCSI;
2370		goto check_len;
2371	}
2372
2373	pr_err("Unable to locate prefix for emulated Target Port:"
2374			" %s\n", name);
2375	kfree(tport);
2376	return ERR_PTR(-EINVAL);
2377
2378check_len:
2379	if (strlen(name) >= VHOST_SCSI_NAMELEN) {
2380		pr_err("Emulated %s Address: %s, exceeds"
2381			" max: %d\n", name, vhost_scsi_dump_proto_id(tport),
2382			VHOST_SCSI_NAMELEN);
2383		kfree(tport);
2384		return ERR_PTR(-EINVAL);
2385	}
2386	snprintf(&tport->tport_name[0], VHOST_SCSI_NAMELEN, "%s", &name[off]);
2387
2388	pr_debug("TCM_VHost_ConfigFS: Allocated emulated Target"
2389		" %s Address: %s\n", vhost_scsi_dump_proto_id(tport), name);
2390
2391	return &tport->tport_wwn;
2392}
2393
2394static void vhost_scsi_drop_tport(struct se_wwn *wwn)
2395{
2396	struct vhost_scsi_tport *tport = container_of(wwn,
2397				struct vhost_scsi_tport, tport_wwn);
2398
2399	pr_debug("TCM_VHost_ConfigFS: Deallocating emulated Target"
2400		" %s Address: %s\n", vhost_scsi_dump_proto_id(tport),
2401		tport->tport_name);
2402
2403	kfree(tport);
2404}
2405
2406static ssize_t
2407vhost_scsi_wwn_version_show(struct config_item *item, char *page)
2408{
2409	return sprintf(page, "TCM_VHOST fabric module %s on %s/%s"
2410		"on "UTS_RELEASE"\n", VHOST_SCSI_VERSION, utsname()->sysname,
2411		utsname()->machine);
2412}
2413
2414CONFIGFS_ATTR_RO(vhost_scsi_wwn_, version);
2415
2416static struct configfs_attribute *vhost_scsi_wwn_attrs[] = {
2417	&vhost_scsi_wwn_attr_version,
2418	NULL,
2419};
2420
2421static const struct target_core_fabric_ops vhost_scsi_ops = {
2422	.module				= THIS_MODULE,
2423	.fabric_name			= "vhost",
2424	.max_data_sg_nents		= VHOST_SCSI_PREALLOC_SGLS,
2425	.tpg_get_wwn			= vhost_scsi_get_fabric_wwn,
2426	.tpg_get_tag			= vhost_scsi_get_tpgt,
2427	.tpg_check_demo_mode		= vhost_scsi_check_true,
2428	.tpg_check_demo_mode_cache	= vhost_scsi_check_true,
2429	.tpg_check_demo_mode_write_protect = vhost_scsi_check_false,
2430	.tpg_check_prod_mode_write_protect = vhost_scsi_check_false,
2431	.tpg_check_prot_fabric_only	= vhost_scsi_check_prot_fabric_only,
2432	.tpg_get_inst_index		= vhost_scsi_tpg_get_inst_index,
2433	.release_cmd			= vhost_scsi_release_cmd,
2434	.check_stop_free		= vhost_scsi_check_stop_free,
2435	.sess_get_index			= vhost_scsi_sess_get_index,
2436	.sess_get_initiator_sid		= NULL,
2437	.write_pending			= vhost_scsi_write_pending,
2438	.set_default_node_attributes	= vhost_scsi_set_default_node_attrs,
2439	.get_cmd_state			= vhost_scsi_get_cmd_state,
2440	.queue_data_in			= vhost_scsi_queue_data_in,
2441	.queue_status			= vhost_scsi_queue_status,
2442	.queue_tm_rsp			= vhost_scsi_queue_tm_rsp,
2443	.aborted_task			= vhost_scsi_aborted_task,
2444	/*
2445	 * Setup callers for generic logic in target_core_fabric_configfs.c
2446	 */
2447	.fabric_make_wwn		= vhost_scsi_make_tport,
2448	.fabric_drop_wwn		= vhost_scsi_drop_tport,
2449	.fabric_make_tpg		= vhost_scsi_make_tpg,
2450	.fabric_drop_tpg		= vhost_scsi_drop_tpg,
2451	.fabric_post_link		= vhost_scsi_port_link,
2452	.fabric_pre_unlink		= vhost_scsi_port_unlink,
2453
2454	.tfc_wwn_attrs			= vhost_scsi_wwn_attrs,
2455	.tfc_tpg_base_attrs		= vhost_scsi_tpg_attrs,
2456	.tfc_tpg_attrib_attrs		= vhost_scsi_tpg_attrib_attrs,
 
 
 
2457};
2458
2459static int __init vhost_scsi_init(void)
2460{
2461	int ret = -ENOMEM;
2462
2463	pr_debug("TCM_VHOST fabric module %s on %s/%s"
2464		" on "UTS_RELEASE"\n", VHOST_SCSI_VERSION, utsname()->sysname,
2465		utsname()->machine);
2466
2467	ret = vhost_scsi_register();
2468	if (ret < 0)
2469		goto out;
2470
2471	ret = target_register_template(&vhost_scsi_ops);
2472	if (ret < 0)
2473		goto out_vhost_scsi_deregister;
2474
2475	return 0;
2476
2477out_vhost_scsi_deregister:
2478	vhost_scsi_deregister();
2479out:
2480	return ret;
2481};
2482
2483static void vhost_scsi_exit(void)
2484{
2485	target_unregister_template(&vhost_scsi_ops);
2486	vhost_scsi_deregister();
2487};
2488
2489MODULE_DESCRIPTION("VHOST_SCSI series fabric driver");
2490MODULE_ALIAS("tcm_vhost");
2491MODULE_LICENSE("GPL");
2492module_init(vhost_scsi_init);
2493module_exit(vhost_scsi_exit);