[linux-2.6-block.git] / drivers / usb / usbip / vhci_tx.c

// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2003-2008 Takahiro Hirofuchi
 */

#include <linux/kthread.h>
#include <linux/slab.h>

#include "usbip_common.h"
#include "vhci.h"

static void setup_cmd_submit_pdu(struct usbip_header *pdup,  struct urb *urb)
{
	struct vhci_priv *priv = ((struct vhci_priv *)urb->hcpriv);
	struct vhci_device *vdev = priv->vdev;

	usbip_dbg_vhci_tx("URB, local devnum %u, remote devid %u\n",
			  usb_pipedevice(urb->pipe), vdev->devid);

	pdup->base.command   = USBIP_CMD_SUBMIT;
	pdup->base.seqnum    = priv->seqnum;
	pdup->base.devid     = vdev->devid;
	pdup->base.direction = usb_pipein(urb->pipe) ?
		USBIP_DIR_IN : USBIP_DIR_OUT;
	pdup->base.ep	     = usb_pipeendpoint(urb->pipe);

	usbip_pack_pdu(pdup, urb, USBIP_CMD_SUBMIT, 1);

	if (urb->setup_packet)
		memcpy(pdup->u.cmd_submit.setup, urb->setup_packet, 8);
}

static struct vhci_priv *dequeue_from_priv_tx(struct vhci_device *vdev)
{
	struct vhci_priv *priv, *tmp;
	unsigned long flags;

	spin_lock_irqsave(&vdev->priv_lock, flags);

	list_for_each_entry_safe(priv, tmp, &vdev->priv_tx, list) {
		list_move_tail(&priv->list, &vdev->priv_rx);
		spin_unlock_irqrestore(&vdev->priv_lock, flags);
		return priv;
	}

	spin_unlock_irqrestore(&vdev->priv_lock, flags);

	return NULL;
}

static int vhci_send_cmd_submit(struct vhci_device *vdev)
{
	struct vhci_priv *priv = NULL;

	struct msghdr msg;
	struct kvec iov[3];
	size_t txsize;

	size_t total_size = 0;

	while ((priv = dequeue_from_priv_tx(vdev)) != NULL) {
		int ret;
		struct urb *urb = priv->urb;
		struct usbip_header pdu_header;
		struct usbip_iso_packet_descriptor *iso_buffer = NULL;

		txsize = 0;
		memset(&pdu_header, 0, sizeof(pdu_header));
		memset(&msg, 0, sizeof(msg));
		memset(&iov, 0, sizeof(iov));

		usbip_dbg_vhci_tx("setup txdata urb seqnum %lu\n",
				  priv->seqnum);

		/* 1. setup usbip_header */
		setup_cmd_submit_pdu(&pdu_header, urb);
		usbip_header_correct_endian(&pdu_header, 1);

		iov[0].iov_base = &pdu_header;
		iov[0].iov_len  = sizeof(pdu_header);
		txsize += sizeof(pdu_header);

		/* 2. setup transfer buffer */
		if (!usb_pipein(urb->pipe) && urb->transfer_buffer_length > 0) {
			iov[1].iov_base = urb->transfer_buffer;
			iov[1].iov_len  = urb->transfer_buffer_length;
			txsize += urb->transfer_buffer_length;
		}

		/* 3. setup iso_packet_descriptor */
		if (usb_pipetype(urb->pipe) == PIPE_ISOCHRONOUS) {
			ssize_t len = 0;

			iso_buffer = usbip_alloc_iso_desc_pdu(urb, &len);
			if (!iso_buffer) {
				usbip_event_add(&vdev->ud,
						SDEV_EVENT_ERROR_MALLOC);
				return -1;
			}

			iov[2].iov_base = iso_buffer;
			iov[2].iov_len  = len;
			txsize += len;
		}

		ret = kernel_sendmsg(vdev->ud.tcp_socket, &msg, iov, 3, txsize);
		if (ret != txsize) {
			pr_err("sendmsg failed!, ret=%d for %zd\n", ret,
			       txsize);
			kfree(iso_buffer);
			usbip_event_add(&vdev->ud, VDEV_EVENT_ERROR_TCP);
			return -1;
		}

		kfree(iso_buffer);
		usbip_dbg_vhci_tx("send txdata\n");

		total_size += txsize;
	}

	return total_size;
}

static struct vhci_unlink *dequeue_from_unlink_tx(struct vhci_device *vdev)
{
	struct vhci_unlink *unlink, *tmp;
	unsigned long flags;

	spin_lock_irqsave(&vdev->priv_lock, flags);

	list_for_each_entry_safe(unlink, tmp, &vdev->unlink_tx, list) {
		list_move_tail(&unlink->list, &vdev->unlink_rx);
		spin_unlock_irqrestore(&vdev->priv_lock, flags);
		return unlink;
	}

	spin_unlock_irqrestore(&vdev->priv_lock, flags);

	return NULL;
}

static int vhci_send_cmd_unlink(struct vhci_device *vdev)
{
	struct vhci_unlink *unlink = NULL;

	struct msghdr msg;
	struct kvec iov[3];
	size_t txsize;

	size_t total_size = 0;

	while ((unlink = dequeue_from_unlink_tx(vdev)) != NULL) {
		int ret;
		struct usbip_header pdu_header;

		txsize = 0;
		memset(&pdu_header, 0, sizeof(pdu_header));
		memset(&msg, 0, sizeof(msg));
		memset(&iov, 0, sizeof(iov));

		usbip_dbg_vhci_tx("setup cmd unlink, %lu\n", unlink->seqnum);

		/* 1. setup usbip_header */
		pdu_header.base.command = USBIP_CMD_UNLINK;
		pdu_header.base.seqnum  = unlink->seqnum;
		pdu_header.base.devid	= vdev->devid;
		pdu_header.base.ep	= 0;
		pdu_header.u.cmd_unlink.seqnum = unlink->unlink_seqnum;

		usbip_header_correct_endian(&pdu_header, 1);

		iov[0].iov_base = &pdu_header;
		iov[0].iov_len  = sizeof(pdu_header);
		txsize += sizeof(pdu_header);

		ret = kernel_sendmsg(vdev->ud.tcp_socket, &msg, iov, 1, txsize);
		if (ret != txsize) {
			pr_err("sendmsg failed!, ret=%d for %zd\n", ret,
			       txsize);
			usbip_event_add(&vdev->ud, VDEV_EVENT_ERROR_TCP);
			return -1;
		}

		usbip_dbg_vhci_tx("send txdata\n");

		total_size += txsize;
	}

	return total_size;
}

int vhci_tx_loop(void *data)
{
	struct usbip_device *ud = data;
	struct vhci_device *vdev = container_of(ud, struct vhci_device, ud);

	while (!kthread_should_stop()) {
		if (vhci_send_cmd_submit(vdev) < 0)
			break;

		if (vhci_send_cmd_unlink(vdev) < 0)
			break;

		wait_event_interruptible(vdev->waitq_tx,
					 (!list_empty(&vdev->priv_tx) ||
					  !list_empty(&vdev->unlink_tx) ||
					  kthread_should_stop()));

		usbip_dbg_vhci_tx("pending urbs ?, now wake up\n");
	}

	return 0;
}
Commit	Line	Data
5fd54ace	1	// SPDX-License-Identifier: GPL-2.0+
04679b34 TH	2	/*
04679b34 TH	3	* Copyright (C) 2003-2008 Takahiro Hirofuchi
04679b34 TH	4	*/
04679b34 TH	5
9720b4bc	6	#include <linux/kthread.h>
7aaacb43	7	#include <linux/slab.h>
5a0e3ad6	8
04679b34 TH	9	#include "usbip_common.h"
	10	#include "vhci.h"
	11
04679b34 TH	12	static void setup_cmd_submit_pdu(struct usbip_header pdup, struct urb urb)
	13	{
	14	struct vhci_priv priv = ((struct vhci_priv )urb->hcpriv);
	15	struct vhci_device *vdev = priv->vdev;
	16
b8868e45	17	usbip_dbg_vhci_tx("URB, local devnum %u, remote devid %u\n",
5c6499d8	18	usb_pipedevice(urb->pipe), vdev->devid);
04679b34	19
5c6499d8	20	pdup->base.command = USBIP_CMD_SUBMIT;
	21	pdup->base.seqnum = priv->seqnum;
	22	pdup->base.devid = vdev->devid;
	23	pdup->base.direction = usb_pipein(urb->pipe) ?
	24	USBIP_DIR_IN : USBIP_DIR_OUT;
	25	pdup->base.ep = usb_pipeendpoint(urb->pipe);
04679b34 TH	26
	27	usbip_pack_pdu(pdup, urb, USBIP_CMD_SUBMIT, 1);
	28
	29	if (urb->setup_packet)
	30	memcpy(pdup->u.cmd_submit.setup, urb->setup_packet, 8);
	31	}
	32
	33	static struct vhci_priv dequeue_from_priv_tx(struct vhci_device vdev)
	34	{
04679b34	35	struct vhci_priv priv, tmp;
21619792	36	unsigned long flags;
04679b34	37
21619792	38	spin_lock_irqsave(&vdev->priv_lock, flags);
04679b34 TH	39
	40	list_for_each_entry_safe(priv, tmp, &vdev->priv_tx, list) {
	41	list_move_tail(&priv->list, &vdev->priv_rx);
21619792	42	spin_unlock_irqrestore(&vdev->priv_lock, flags);
04679b34 TH	43	return priv;
	44	}
	45
21619792	46	spin_unlock_irqrestore(&vdev->priv_lock, flags);
04679b34 TH	47
	48	return NULL;
	49	}
	50
04679b34 TH	51	static int vhci_send_cmd_submit(struct vhci_device *vdev)
	52	{
	53	struct vhci_priv *priv = NULL;
	54
	55	struct msghdr msg;
	56	struct kvec iov[3];
	57	size_t txsize;
	58
	59	size_t total_size = 0;
	60
	61	while ((priv = dequeue_from_priv_tx(vdev)) != NULL) {
	62	int ret;
	63	struct urb *urb = priv->urb;
	64	struct usbip_header pdu_header;
36ac9b05	65	struct usbip_iso_packet_descriptor *iso_buffer = NULL;
04679b34 TH	66
	67	txsize = 0;
	68	memset(&pdu_header, 0, sizeof(pdu_header));
	69	memset(&msg, 0, sizeof(msg));
	70	memset(&iov, 0, sizeof(iov));
	71
8272d099 SK	72	usbip_dbg_vhci_tx("setup txdata urb seqnum %lu\n",
8272d099 SK	73	priv->seqnum);
04679b34	74
04679b34 TH	75	/* 1. setup usbip_header */
	76	setup_cmd_submit_pdu(&pdu_header, urb);
	77	usbip_header_correct_endian(&pdu_header, 1);
	78
	79	iov[0].iov_base = &pdu_header;
	80	iov[0].iov_len = sizeof(pdu_header);
	81	txsize += sizeof(pdu_header);
	82
	83	/* 2. setup transfer buffer */
	84	if (!usb_pipein(urb->pipe) && urb->transfer_buffer_length > 0) {
	85	iov[1].iov_base = urb->transfer_buffer;
	86	iov[1].iov_len = urb->transfer_buffer_length;
	87	txsize += urb->transfer_buffer_length;
	88	}
	89
	90	/* 3. setup iso_packet_descriptor */
	91	if (usb_pipetype(urb->pipe) == PIPE_ISOCHRONOUS) {
	92	ssize_t len = 0;
	93
	94	iso_buffer = usbip_alloc_iso_desc_pdu(urb, &len);
	95	if (!iso_buffer) {
	96	usbip_event_add(&vdev->ud,
	97	SDEV_EVENT_ERROR_MALLOC);
	98	return -1;
	99	}
	100
	101	iov[2].iov_base = iso_buffer;
	102	iov[2].iov_len = len;
	103	txsize += len;
	104	}
	105
	106	ret = kernel_sendmsg(vdev->ud.tcp_socket, &msg, iov, 3, txsize);
	107	if (ret != txsize) {
1a4b6f66	108	pr_err("sendmsg failed!, ret=%d for %zd\n", ret,
1a4b6f66	109	txsize);
04679b34 TH	110	kfree(iso_buffer);
	111	usbip_event_add(&vdev->ud, VDEV_EVENT_ERROR_TCP);
	112	return -1;
	113	}
	114
	115	kfree(iso_buffer);
b8868e45	116	usbip_dbg_vhci_tx("send txdata\n");
04679b34 TH	117
	118	total_size += txsize;
	119	}
	120
	121	return total_size;
	122	}
	123
04679b34 TH	124	static struct vhci_unlink dequeue_from_unlink_tx(struct vhci_device vdev)
04679b34 TH	125	{
04679b34	126	struct vhci_unlink unlink, tmp;
21619792	127	unsigned long flags;
04679b34	128
21619792	129	spin_lock_irqsave(&vdev->priv_lock, flags);
04679b34 TH	130
	131	list_for_each_entry_safe(unlink, tmp, &vdev->unlink_tx, list) {
	132	list_move_tail(&unlink->list, &vdev->unlink_rx);
21619792	133	spin_unlock_irqrestore(&vdev->priv_lock, flags);
04679b34 TH	134	return unlink;
	135	}
	136
21619792	137	spin_unlock_irqrestore(&vdev->priv_lock, flags);
04679b34 TH	138
	139	return NULL;
	140	}
	141
	142	static int vhci_send_cmd_unlink(struct vhci_device *vdev)
	143	{
	144	struct vhci_unlink *unlink = NULL;
	145
	146	struct msghdr msg;
	147	struct kvec iov[3];
	148	size_t txsize;
	149
	150	size_t total_size = 0;
	151
	152	while ((unlink = dequeue_from_unlink_tx(vdev)) != NULL) {
	153	int ret;
	154	struct usbip_header pdu_header;
	155
	156	txsize = 0;
	157	memset(&pdu_header, 0, sizeof(pdu_header));
	158	memset(&msg, 0, sizeof(msg));
	159	memset(&iov, 0, sizeof(iov));
	160
acc4e416	161	usbip_dbg_vhci_tx("setup cmd unlink, %lu\n", unlink->seqnum);
04679b34	162
04679b34 TH	163	/* 1. setup usbip_header */
	164	pdu_header.base.command = USBIP_CMD_UNLINK;
	165	pdu_header.base.seqnum = unlink->seqnum;
	166	pdu_header.base.devid = vdev->devid;
	167	pdu_header.base.ep = 0;
	168	pdu_header.u.cmd_unlink.seqnum = unlink->unlink_seqnum;
	169
	170	usbip_header_correct_endian(&pdu_header, 1);
	171
	172	iov[0].iov_base = &pdu_header;
	173	iov[0].iov_len = sizeof(pdu_header);
	174	txsize += sizeof(pdu_header);
	175
	176	ret = kernel_sendmsg(vdev->ud.tcp_socket, &msg, iov, 1, txsize);
	177	if (ret != txsize) {
1a4b6f66	178	pr_err("sendmsg failed!, ret=%d for %zd\n", ret,
1a4b6f66	179	txsize);
04679b34 TH	180	usbip_event_add(&vdev->ud, VDEV_EVENT_ERROR_TCP);
	181	return -1;
	182	}
	183
b8868e45	184	usbip_dbg_vhci_tx("send txdata\n");
04679b34 TH	185
	186	total_size += txsize;
	187	}
	188
	189	return total_size;
	190	}
	191
9720b4bc	192	int vhci_tx_loop(void *data)
04679b34	193	{
9720b4bc	194	struct usbip_device *ud = data;
04679b34 TH	195	struct vhci_device *vdev = container_of(ud, struct vhci_device, ud);
04679b34 TH	196
9720b4bc	197	while (!kthread_should_stop()) {
04679b34 TH	198	if (vhci_send_cmd_submit(vdev) < 0)
	199	break;
	200
	201	if (vhci_send_cmd_unlink(vdev) < 0)
	202	break;
	203
	204	wait_event_interruptible(vdev->waitq_tx,
5c6499d8	205	(!list_empty(&vdev->priv_tx) \|\|
	206	!list_empty(&vdev->unlink_tx) \|\|
	207	kthread_should_stop()));
04679b34	208
b8868e45	209	usbip_dbg_vhci_tx("pending urbs ?, now wake up\n");
04679b34	210	}
9720b4bc AB	211
9720b4bc AB	212	return 0;
04679b34	213	}