1/* 2 * Copyright (C) 2003-2008 Takahiro Hirofuchi 3 * 4 * This is free software; you can redistribute it and/or modify 5 * it under the terms of the GNU General Public License as published by 6 * the Free Software Foundation; either version 2 of the License, or 7 * (at your option) any later version. 8 * 9 * This is distributed in the hope that it will be useful, 10 * but WITHOUT ANY WARRANTY; without even the implied warranty of 11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 12 * GNU General Public License for more details. 13 * 14 * You should have received a copy of the GNU General Public License 15 * along with this program; if not, write to the Free Software 16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, 17 * USA. 18 */ 19 20#include <linux/slab.h> 21 22#include "usbip_common.h" 23#include "vhci.h" 24 25 26static void setup_cmd_submit_pdu(struct usbip_header *pdup, struct urb *urb) 27{ 28 struct vhci_priv *priv = ((struct vhci_priv *)urb->hcpriv); 29 struct vhci_device *vdev = priv->vdev; 30 31 usbip_dbg_vhci_tx("URB, local devnum %u, remote devid %u\n", 32 usb_pipedevice(urb->pipe), vdev->devid); 33 34 pdup->base.command = USBIP_CMD_SUBMIT; 35 pdup->base.seqnum = priv->seqnum; 36 pdup->base.devid = vdev->devid; 37 if (usb_pipein(urb->pipe)) 38 pdup->base.direction = USBIP_DIR_IN; 39 else 40 pdup->base.direction = USBIP_DIR_OUT; 41 pdup->base.ep = usb_pipeendpoint(urb->pipe); 42 43 usbip_pack_pdu(pdup, urb, USBIP_CMD_SUBMIT, 1); 44 45 if (urb->setup_packet) 46 memcpy(pdup->u.cmd_submit.setup, urb->setup_packet, 8); 47} 48 49static struct vhci_priv *dequeue_from_priv_tx(struct vhci_device *vdev) 50{ 51 unsigned long flags; 52 struct vhci_priv *priv, *tmp; 53 54 spin_lock_irqsave(&vdev->priv_lock, flags); 55 56 list_for_each_entry_safe(priv, tmp, &vdev->priv_tx, list) { 57 list_move_tail(&priv->list, &vdev->priv_rx); 58 spin_unlock_irqrestore(&vdev->priv_lock, flags); 59 return priv; 60 } 61 62 spin_unlock_irqrestore(&vdev->priv_lock, flags); 63 64 return NULL; 65} 66 67 68 69static int vhci_send_cmd_submit(struct vhci_device *vdev) 70{ 71 struct vhci_priv *priv = NULL; 72 73 struct msghdr msg; 74 struct kvec iov[3]; 75 size_t txsize; 76 77 size_t total_size = 0; 78 79 while ((priv = dequeue_from_priv_tx(vdev)) != NULL) { 80 int ret; 81 struct urb *urb = priv->urb; 82 struct usbip_header pdu_header; 83 void *iso_buffer = NULL; 84 85 txsize = 0; 86 memset(&pdu_header, 0, sizeof(pdu_header)); 87 memset(&msg, 0, sizeof(msg)); 88 memset(&iov, 0, sizeof(iov)); 89 90 usbip_dbg_vhci_tx("setup txdata urb %p\n", urb); 91 92 93 /* 1. setup usbip_header */ 94 setup_cmd_submit_pdu(&pdu_header, urb); 95 usbip_header_correct_endian(&pdu_header, 1); 96 97 iov[0].iov_base = &pdu_header; 98 iov[0].iov_len = sizeof(pdu_header); 99 txsize += sizeof(pdu_header); 100 101 /* 2. setup transfer buffer */ 102 if (!usb_pipein(urb->pipe) && urb->transfer_buffer_length > 0) { 103 iov[1].iov_base = urb->transfer_buffer; 104 iov[1].iov_len = urb->transfer_buffer_length; 105 txsize += urb->transfer_buffer_length; 106 } 107 108 /* 3. setup iso_packet_descriptor */ 109 if (usb_pipetype(urb->pipe) == PIPE_ISOCHRONOUS) { 110 ssize_t len = 0; 111 112 iso_buffer = usbip_alloc_iso_desc_pdu(urb, &len); 113 if (!iso_buffer) { 114 usbip_event_add(&vdev->ud, 115 SDEV_EVENT_ERROR_MALLOC); 116 return -1; 117 } 118 119 iov[2].iov_base = iso_buffer; 120 iov[2].iov_len = len; 121 txsize += len; 122 } 123 124 ret = kernel_sendmsg(vdev->ud.tcp_socket, &msg, iov, 3, txsize); 125 if (ret != txsize) { 126 usbip_uerr("sendmsg failed!, retval %d for %zd\n", ret, 127 txsize); 128 kfree(iso_buffer); 129 usbip_event_add(&vdev->ud, VDEV_EVENT_ERROR_TCP); 130 return -1; 131 } 132 133 kfree(iso_buffer); 134 usbip_dbg_vhci_tx("send txdata\n"); 135 136 total_size += txsize; 137 } 138 139 return total_size; 140} 141 142 143/*-------------------------------------------------------------------------*/ 144 145static struct vhci_unlink *dequeue_from_unlink_tx(struct vhci_device *vdev) 146{ 147 unsigned long flags; 148 struct vhci_unlink *unlink, *tmp; 149 150 spin_lock_irqsave(&vdev->priv_lock, flags); 151 152 list_for_each_entry_safe(unlink, tmp, &vdev->unlink_tx, list) { 153 list_move_tail(&unlink->list, &vdev->unlink_rx); 154 spin_unlock_irqrestore(&vdev->priv_lock, flags); 155 return unlink; 156 } 157 158 spin_unlock_irqrestore(&vdev->priv_lock, flags); 159 160 return NULL; 161} 162 163static int vhci_send_cmd_unlink(struct vhci_device *vdev) 164{ 165 struct vhci_unlink *unlink = NULL; 166 167 struct msghdr msg; 168 struct kvec iov[3]; 169 size_t txsize; 170 171 size_t total_size = 0; 172 173 while ((unlink = dequeue_from_unlink_tx(vdev)) != NULL) { 174 int ret; 175 struct usbip_header pdu_header; 176 177 txsize = 0; 178 memset(&pdu_header, 0, sizeof(pdu_header)); 179 memset(&msg, 0, sizeof(msg)); 180 memset(&iov, 0, sizeof(iov)); 181 182 usbip_dbg_vhci_tx("setup cmd unlink, %lu\n", unlink->seqnum); 183 184 185 /* 1. setup usbip_header */ 186 pdu_header.base.command = USBIP_CMD_UNLINK; 187 pdu_header.base.seqnum = unlink->seqnum; 188 pdu_header.base.devid = vdev->devid; 189 pdu_header.base.ep = 0; 190 pdu_header.u.cmd_unlink.seqnum = unlink->unlink_seqnum; 191 192 usbip_header_correct_endian(&pdu_header, 1); 193 194 iov[0].iov_base = &pdu_header; 195 iov[0].iov_len = sizeof(pdu_header); 196 txsize += sizeof(pdu_header); 197 198 ret = kernel_sendmsg(vdev->ud.tcp_socket, &msg, iov, 1, txsize); 199 if (ret != txsize) { 200 usbip_uerr("sendmsg failed!, retval %d for %zd\n", ret, 201 txsize); 202 usbip_event_add(&vdev->ud, VDEV_EVENT_ERROR_TCP); 203 return -1; 204 } 205 206 207 usbip_dbg_vhci_tx("send txdata\n"); 208 209 total_size += txsize; 210 } 211 212 return total_size; 213} 214 215 216/*-------------------------------------------------------------------------*/ 217 218void vhci_tx_loop(struct usbip_task *ut) 219{ 220 struct usbip_device *ud = container_of(ut, struct usbip_device, tcp_tx); 221 struct vhci_device *vdev = container_of(ud, struct vhci_device, ud); 222 223 while (1) { 224 if (signal_pending(current)) { 225 usbip_uinfo("vhci_tx signal catched\n"); 226 break; 227 } 228 229 if (vhci_send_cmd_submit(vdev) < 0) 230 break; 231 232 if (vhci_send_cmd_unlink(vdev) < 0) 233 break; 234 235 wait_event_interruptible(vdev->waitq_tx, 236 (!list_empty(&vdev->priv_tx) || 237 !list_empty(&vdev->unlink_tx))); 238 239 usbip_dbg_vhci_tx("pending urbs ?, now wake up\n"); 240 } 241} 242