drivers/usb/usbip/vudc_sysfs.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (C) 2015 Karol Kosik <[email protected]>
   4  * Copyright (C) 2015-2016 Samsung Electronics
   5  *               Igor Kotrasinski <[email protected]>
   6  *               Krzysztof Opasiak <[email protected]>
   7  */
   8
   9 #include <linux/device.h>
  10 #include <linux/list.h>
  11 #include <linux/usb/gadget.h>
  12 #include <linux/usb/ch9.h>
  13 #include <linux/sysfs.h>
  14 #include <linux/kthread.h>
  15 #include <linux/byteorder/generic.h>
  16
  17 #include "usbip_common.h"
  18 #include "vudc.h"
  19
  20 #include <net/sock.h>
  21
  22 /* called with udc->lock held */
  23 int get_gadget_descs(struct vudc *udc)
  24 {
  25         struct vrequest *usb_req;
  26         struct vep *ep0 = to_vep(udc->gadget.ep0);
  27         struct usb_device_descriptor *ddesc = &udc->dev_desc;
  28         struct usb_ctrlrequest req;
  29         int ret;
  30
  31         if (!udc->driver || !udc->pullup)
  32                 return -EINVAL;
  33
  34         req.bRequestType = USB_DIR_IN | USB_TYPE_STANDARD | USB_RECIP_DEVICE;
  35         req.bRequest = USB_REQ_GET_DESCRIPTOR;
  36         req.wValue = cpu_to_le16(USB_DT_DEVICE << 8);
  37         req.wIndex = cpu_to_le16(0);
  38         req.wLength = cpu_to_le16(sizeof(*ddesc));
  39
  40         spin_unlock(&udc->lock);
  41         ret = udc->driver->setup(&(udc->gadget), &req);
  42         spin_lock(&udc->lock);
  43         if (ret < 0)
  44                 goto out;
  45
  46         /* assuming request queue is empty; request is now on top */
  47         usb_req = list_last_entry(&ep0->req_queue, struct vrequest, req_entry);
  48         list_del(&usb_req->req_entry);
  49
  50         if (usb_req->req.length > sizeof(*ddesc)) {
  51                 ret = -EOVERFLOW;
  52                 goto giveback_req;
  53         }
  54
  55         memcpy(ddesc, usb_req->req.buf, sizeof(*ddesc));
  56         udc->desc_cached = 1;
  57         ret = 0;
  58 giveback_req:
  59         usb_req->req.status = 0;
  60         usb_req->req.actual = usb_req->req.length;
  61         usb_gadget_giveback_request(&(ep0->ep), &(usb_req->req));
  62 out:
  63         return ret;
  64 }
  65
  66 /*
  67  * Exposes device descriptor from the gadget driver.
  68  */
  69 static ssize_t dev_desc_read(struct file *file, struct kobject *kobj,
  70                              struct bin_attribute *attr, char *out,
  71                              loff_t off, size_t count)
  72 {
  73         struct device *dev = kobj_to_dev(kobj);
  74         struct vudc *udc = (struct vudc *)dev_get_drvdata(dev);
  75         char *desc_ptr = (char *) &udc->dev_desc;
  76         unsigned long flags;
  77         int ret;
  78
  79         spin_lock_irqsave(&udc->lock, flags);
  80         if (!udc->desc_cached) {
  81                 ret = -ENODEV;
  82                 goto unlock;
  83         }
  84
  85         memcpy(out, desc_ptr + off, count);
  86         ret = count;
  87 unlock:
  88         spin_unlock_irqrestore(&udc->lock, flags);
  89         return ret;
  90 }
  91 static BIN_ATTR_RO(dev_desc, sizeof(struct usb_device_descriptor));
  92
  93 static ssize_t usbip_sockfd_store(struct device *dev,
  94                                   struct device_attribute *attr,
  95                                   const char *in, size_t count)
  96 {
  97         struct vudc *udc = (struct vudc *) dev_get_drvdata(dev);
  98         int rv;
  99         int sockfd = 0;
 100         int err;
 101         struct socket *socket;
 102         unsigned long flags;
 103         int ret;
 104         struct task_struct *tcp_rx = NULL;
 105         struct task_struct *tcp_tx = NULL;
 106
 107         rv = kstrtoint(in, 0, &sockfd);
 108         if (rv != 0)
 109                 return -EINVAL;
 110
 111         if (!udc) {
 112                 dev_err(dev, "no device");
 113                 return -ENODEV;
 114         }
 115         mutex_lock(&udc->ud.sysfs_lock);
 116         spin_lock_irqsave(&udc->lock, flags);
 117         /* Don't export what we don't have */
 118         if (!udc->driver || !udc->pullup) {
 119                 dev_err(dev, "gadget not bound");
 120                 ret = -ENODEV;
 121                 goto unlock;
 122         }
 123
 124         if (sockfd != -1) {
 125                 if (udc->connected) {
 126                         dev_err(dev, "Device already connected");
 127                         ret = -EBUSY;
 128                         goto unlock;
 129                 }
 130
 131                 spin_lock(&udc->ud.lock);
 132
 133                 if (udc->ud.status != SDEV_ST_AVAILABLE) {
 134                         ret = -EINVAL;
 135                         goto unlock_ud;
 136                 }
 137
 138                 socket = sockfd_lookup(sockfd, &err);
 139                 if (!socket) {
 140                         dev_err(dev, "failed to lookup sock");
 141                         ret = -EINVAL;
 142                         goto unlock_ud;
 143                 }
 144
 145                 if (socket->type != SOCK_STREAM) {
 146                         dev_err(dev, "Expecting SOCK_STREAM - found %d",
 147                                 socket->type);
 148                         ret = -EINVAL;
 149                         goto sock_err;
 150                 }
 151
 152                 /* unlock and create threads and get tasks */
 153                 spin_unlock(&udc->ud.lock);
 154                 spin_unlock_irqrestore(&udc->lock, flags);
 155
 156                 tcp_rx = kthread_create(&v_rx_loop, &udc->ud, "vudc_rx");
 157                 if (IS_ERR(tcp_rx)) {
 158                         sockfd_put(socket);
 159                         mutex_unlock(&udc->ud.sysfs_lock);
 160                         return -EINVAL;
 161                 }
 162                 tcp_tx = kthread_create(&v_tx_loop, &udc->ud, "vudc_tx");
 163                 if (IS_ERR(tcp_tx)) {
 164                         kthread_stop(tcp_rx);
 165                         sockfd_put(socket);
 166                         mutex_unlock(&udc->ud.sysfs_lock);
 167                         return -EINVAL;
 168                 }
 169
 170                 /* get task structs now */
 171                 get_task_struct(tcp_rx);
 172                 get_task_struct(tcp_tx);
 173
 174                 /* lock and update udc->ud state */
 175                 spin_lock_irqsave(&udc->lock, flags);
 176                 spin_lock(&udc->ud.lock);
 177
 178                 udc->ud.tcp_socket = socket;
 179                 udc->ud.tcp_rx = tcp_rx;
 180                 udc->ud.tcp_tx = tcp_tx;
 181                 udc->ud.status = SDEV_ST_USED;
 182
 183                 spin_unlock(&udc->ud.lock);
 184
 185                 ktime_get_ts64(&udc->start_time);
 186                 v_start_timer(udc);
 187                 udc->connected = 1;
 188
 189                 spin_unlock_irqrestore(&udc->lock, flags);
 190
 191                 wake_up_process(udc->ud.tcp_rx);
 192                 wake_up_process(udc->ud.tcp_tx);
 193
 194                 mutex_unlock(&udc->ud.sysfs_lock);
 195                 return count;
 196
 197         } else {
 198                 if (!udc->connected) {
 199                         dev_err(dev, "Device not connected");
 200                         ret = -EINVAL;
 201                         goto unlock;
 202                 }
 203
 204                 spin_lock(&udc->ud.lock);
 205                 if (udc->ud.status != SDEV_ST_USED) {
 206                         ret = -EINVAL;
 207                         goto unlock_ud;
 208                 }
 209                 spin_unlock(&udc->ud.lock);
 210
 211                 usbip_event_add(&udc->ud, VUDC_EVENT_DOWN);
 212         }
 213
 214         spin_unlock_irqrestore(&udc->lock, flags);
 215         mutex_unlock(&udc->ud.sysfs_lock);
 216
 217         return count;
 218
 219 sock_err:
 220         sockfd_put(socket);
 221 unlock_ud:
 222         spin_unlock(&udc->ud.lock);
 223 unlock:
 224         spin_unlock_irqrestore(&udc->lock, flags);
 225         mutex_unlock(&udc->ud.sysfs_lock);
 226
 227         return ret;
 228 }
 229 static DEVICE_ATTR_WO(usbip_sockfd);
 230
 231 static ssize_t usbip_status_show(struct device *dev,
 232                                struct device_attribute *attr, char *out)
 233 {
 234         struct vudc *udc = (struct vudc *) dev_get_drvdata(dev);
 235         int status;
 236
 237         if (!udc) {
 238                 dev_err(dev, "no device");
 239                 return -ENODEV;
 240         }
 241         spin_lock_irq(&udc->ud.lock);
 242         status = udc->ud.status;
 243         spin_unlock_irq(&udc->ud.lock);
 244
 245         return snprintf(out, PAGE_SIZE, "%d\n", status);
 246 }
 247 static DEVICE_ATTR_RO(usbip_status);
 248
 249 static struct attribute *dev_attrs[] = {
 250         &dev_attr_usbip_sockfd.attr,
 251         &dev_attr_usbip_status.attr,
 252         NULL,
 253 };
 254
 255 static struct bin_attribute *dev_bin_attrs[] = {
 256         &bin_attr_dev_desc,
 257         NULL,
 258 };
 259
 260 static const struct attribute_group vudc_attr_group = {
 261         .attrs = dev_attrs,
 262         .bin_attrs = dev_bin_attrs,
 263 };
 264
 265 const struct attribute_group *vudc_groups[] = {
 266         &vudc_attr_group,
 267         NULL,
 268 };