drivers/usb/usbip/usbip_common.h

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Copyright (C) 2003-2008 Takahiro Hirofuchi
   4  * Copyright (C) 2015-2016 Samsung Electronics
   5  *               Krzysztof Opasiak <k.opasiak@samsung.com>
   6  */
   7
   8 #ifndef __USBIP_COMMON_H
   9 #define __USBIP_COMMON_H
  10
  11 #include <linux/compiler.h>
  12 #include <linux/device.h>
  13 #include <linux/interrupt.h>
  14 #include <linux/net.h>
  15 #include <linux/printk.h>
  16 #include <linux/spinlock.h>
  17 #include <linux/types.h>
  18 #include <linux/usb.h>
  19 #include <linux/wait.h>
  20 #include <linux/sched/task.h>
  21 #include <linux/kcov.h>
  22 #include <uapi/linux/usbip.h>
  23
  24 #undef pr_fmt
  25
  26 #ifdef DEBUG
  27 #define pr_fmt(fmt)     KBUILD_MODNAME ": %s:%d: " fmt, __func__, __LINE__
  28 #else
  29 #define pr_fmt(fmt)     KBUILD_MODNAME ": " fmt
  30 #endif
  31
  32 enum {
  33         usbip_debug_xmit        = (1 << 0),
  34         usbip_debug_sysfs       = (1 << 1),
  35         usbip_debug_urb         = (1 << 2),
  36         usbip_debug_eh          = (1 << 3),
  37
  38         usbip_debug_stub_cmp    = (1 << 8),
  39         usbip_debug_stub_dev    = (1 << 9),
  40         usbip_debug_stub_rx     = (1 << 10),
  41         usbip_debug_stub_tx     = (1 << 11),
  42
  43         usbip_debug_vhci_rh     = (1 << 8),
  44         usbip_debug_vhci_hc     = (1 << 9),
  45         usbip_debug_vhci_rx     = (1 << 10),
  46         usbip_debug_vhci_tx     = (1 << 11),
  47         usbip_debug_vhci_sysfs  = (1 << 12)
  48 };
  49
  50 #define usbip_dbg_flag_xmit     (usbip_debug_flag & usbip_debug_xmit)
  51 #define usbip_dbg_flag_vhci_rh  (usbip_debug_flag & usbip_debug_vhci_rh)
  52 #define usbip_dbg_flag_vhci_hc  (usbip_debug_flag & usbip_debug_vhci_hc)
  53 #define usbip_dbg_flag_vhci_rx  (usbip_debug_flag & usbip_debug_vhci_rx)
  54 #define usbip_dbg_flag_vhci_tx  (usbip_debug_flag & usbip_debug_vhci_tx)
  55 #define usbip_dbg_flag_stub_rx  (usbip_debug_flag & usbip_debug_stub_rx)
  56 #define usbip_dbg_flag_stub_tx  (usbip_debug_flag & usbip_debug_stub_tx)
  57 #define usbip_dbg_flag_vhci_sysfs  (usbip_debug_flag & usbip_debug_vhci_sysfs)
  58
  59 extern unsigned long usbip_debug_flag;
  60 extern struct device_attribute dev_attr_usbip_debug;
  61
  62 #define usbip_dbg_with_flag(flag, fmt, args...)         \
  63         do {                                            \
  64                 if (flag & usbip_debug_flag)            \
  65                         pr_debug(fmt, ##args);          \
  66         } while (0)
  67
  68 #define usbip_dbg_sysfs(fmt, args...) \
  69         usbip_dbg_with_flag(usbip_debug_sysfs, fmt , ##args)
  70 #define usbip_dbg_xmit(fmt, args...) \
  71         usbip_dbg_with_flag(usbip_debug_xmit, fmt , ##args)
  72 #define usbip_dbg_urb(fmt, args...) \
  73         usbip_dbg_with_flag(usbip_debug_urb, fmt , ##args)
  74 #define usbip_dbg_eh(fmt, args...) \
  75         usbip_dbg_with_flag(usbip_debug_eh, fmt , ##args)
  76
  77 #define usbip_dbg_vhci_rh(fmt, args...) \
  78         usbip_dbg_with_flag(usbip_debug_vhci_rh, fmt , ##args)
  79 #define usbip_dbg_vhci_hc(fmt, args...) \
  80         usbip_dbg_with_flag(usbip_debug_vhci_hc, fmt , ##args)
  81 #define usbip_dbg_vhci_rx(fmt, args...) \
  82         usbip_dbg_with_flag(usbip_debug_vhci_rx, fmt , ##args)
  83 #define usbip_dbg_vhci_tx(fmt, args...) \
  84         usbip_dbg_with_flag(usbip_debug_vhci_tx, fmt , ##args)
  85 #define usbip_dbg_vhci_sysfs(fmt, args...) \
  86         usbip_dbg_with_flag(usbip_debug_vhci_sysfs, fmt , ##args)
  87
  88 #define usbip_dbg_stub_cmp(fmt, args...) \
  89         usbip_dbg_with_flag(usbip_debug_stub_cmp, fmt , ##args)
  90 #define usbip_dbg_stub_rx(fmt, args...) \
  91         usbip_dbg_with_flag(usbip_debug_stub_rx, fmt , ##args)
  92 #define usbip_dbg_stub_tx(fmt, args...) \
  93         usbip_dbg_with_flag(usbip_debug_stub_tx, fmt , ##args)
  94
  95 /*
  96  * USB/IP request headers
  97  *
  98  * Each request is transferred across the network to its counterpart, which
  99  * facilitates the normal USB communication. The values contained in the headers
 100  * are basically the same as in a URB. Currently, four request types are
 101  * defined:
 102  *
 103  *  - USBIP_CMD_SUBMIT: a USB request block, corresponds to usb_submit_urb()
 104  *    (client to server)
 105  *
 106  *  - USBIP_RET_SUBMIT: the result of USBIP_CMD_SUBMIT
 107  *    (server to client)
 108  *
 109  *  - USBIP_CMD_UNLINK: an unlink request of a pending USBIP_CMD_SUBMIT,
 110  *    corresponds to usb_unlink_urb()
 111  *    (client to server)
 112  *
 113  *  - USBIP_RET_UNLINK: the result of USBIP_CMD_UNLINK
 114  *    (server to client)
 115  *
 116  */
 117 #define USBIP_CMD_SUBMIT        0x0001
 118 #define USBIP_CMD_UNLINK        0x0002
 119 #define USBIP_RET_SUBMIT        0x0003
 120 #define USBIP_RET_UNLINK        0x0004
 121
 122 #define USBIP_DIR_OUT   0x00
 123 #define USBIP_DIR_IN    0x01
 124
 125 /*
 126  * Arbitrary limit for the maximum number of isochronous packets in an URB,
 127  * compare for example the uhci_submit_isochronous function in
 128  * drivers/usb/host/uhci-q.c
 129  */
 130 #define USBIP_MAX_ISO_PACKETS 1024
 131
 132 /**
 133  * struct usbip_header_basic - data pertinent to every request
 134  * @command: the usbip request type
 135  * @seqnum: sequential number that identifies requests; incremented per
 136  *          connection
 137  * @devid: specifies a remote USB device uniquely instead of busnum and devnum;
 138  *         in the stub driver, this value is ((busnum << 16) | devnum)
 139  * @direction: direction of the transfer
 140  * @ep: endpoint number
 141  */
 142 struct usbip_header_basic {
 143         __u32 command;
 144         __u32 seqnum;
 145         __u32 devid;
 146         __u32 direction;
 147         __u32 ep;
 148 } __packed;
 149
 150 /**
 151  * struct usbip_header_cmd_submit - USBIP_CMD_SUBMIT packet header
 152  * @transfer_flags: URB flags
 153  * @transfer_buffer_length: the data size for (in) or (out) transfer
 154  * @start_frame: initial frame for isochronous or interrupt transfers
 155  * @number_of_packets: number of isochronous packets
 156  * @interval: maximum time for the request on the server-side host controller
 157  * @setup: setup data for a control request
 158  */
 159 struct usbip_header_cmd_submit {
 160         __u32 transfer_flags;
 161         __s32 transfer_buffer_length;
 162
 163         /* it is difficult for usbip to sync frames (reserved only?) */
 164         __s32 start_frame;
 165         __s32 number_of_packets;
 166         __s32 interval;
 167
 168         unsigned char setup[8];
 169 } __packed;
 170
 171 /**
 172  * struct usbip_header_ret_submit - USBIP_RET_SUBMIT packet header
 173  * @status: return status of a non-iso request
 174  * @actual_length: number of bytes transferred
 175  * @start_frame: initial frame for isochronous or interrupt transfers
 176  * @number_of_packets: number of isochronous packets
 177  * @error_count: number of errors for isochronous transfers
 178  */
 179 struct usbip_header_ret_submit {
 180         __s32 status;
 181         __s32 actual_length;
 182         __s32 start_frame;
 183         __s32 number_of_packets;
 184         __s32 error_count;
 185 } __packed;
 186
 187 /**
 188  * struct usbip_header_cmd_unlink - USBIP_CMD_UNLINK packet header
 189  * @seqnum: the URB seqnum to unlink
 190  */
 191 struct usbip_header_cmd_unlink {
 192         __u32 seqnum;
 193 } __packed;
 194
 195 /**
 196  * struct usbip_header_ret_unlink - USBIP_RET_UNLINK packet header
 197  * @status: return status of the request
 198  */
 199 struct usbip_header_ret_unlink {
 200         __s32 status;
 201 } __packed;
 202
 203 /**
 204  * struct usbip_header - common header for all usbip packets
 205  * @base: the basic header
 206  * @u: packet type dependent header
 207  */
 208 struct usbip_header {
 209         struct usbip_header_basic base;
 210
 211         union {
 212                 struct usbip_header_cmd_submit  cmd_submit;
 213                 struct usbip_header_ret_submit  ret_submit;
 214                 struct usbip_header_cmd_unlink  cmd_unlink;
 215                 struct usbip_header_ret_unlink  ret_unlink;
 216         } u;
 217 } __packed;
 218
 219 /*
 220  * This is the same as usb_iso_packet_descriptor but packed for pdu.
 221  */
 222 struct usbip_iso_packet_descriptor {
 223         __u32 offset;
 224         __u32 length;                   /* expected length */
 225         __u32 actual_length;
 226         __u32 status;
 227 } __packed;
 228
 229 enum usbip_side {
 230         USBIP_VHCI,
 231         USBIP_STUB,
 232         USBIP_VUDC,
 233 };
 234
 235 /* event handler */
 236 #define USBIP_EH_SHUTDOWN       (1 << 0)
 237 #define USBIP_EH_BYE            (1 << 1)
 238 #define USBIP_EH_RESET          (1 << 2)
 239 #define USBIP_EH_UNUSABLE       (1 << 3)
 240
 241 #define SDEV_EVENT_REMOVED      (USBIP_EH_SHUTDOWN | USBIP_EH_BYE)
 242 #define SDEV_EVENT_DOWN         (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 243 #define SDEV_EVENT_ERROR_TCP    (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 244 #define SDEV_EVENT_ERROR_SUBMIT (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 245 #define SDEV_EVENT_ERROR_MALLOC (USBIP_EH_SHUTDOWN | USBIP_EH_UNUSABLE)
 246
 247 #define VUDC_EVENT_REMOVED   (USBIP_EH_SHUTDOWN | USBIP_EH_RESET | USBIP_EH_BYE)
 248 #define VUDC_EVENT_DOWN         (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 249 #define VUDC_EVENT_ERROR_TCP    (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 250 /* catastrophic emulated usb error */
 251 #define VUDC_EVENT_ERROR_USB    (USBIP_EH_SHUTDOWN | USBIP_EH_UNUSABLE)
 252 #define VUDC_EVENT_ERROR_MALLOC (USBIP_EH_SHUTDOWN | USBIP_EH_UNUSABLE)
 253
 254 #define VDEV_EVENT_REMOVED (USBIP_EH_SHUTDOWN | USBIP_EH_RESET | USBIP_EH_BYE)
 255 #define VDEV_EVENT_DOWN         (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 256 #define VDEV_EVENT_ERROR_TCP    (USBIP_EH_SHUTDOWN | USBIP_EH_RESET)
 257 #define VDEV_EVENT_ERROR_MALLOC (USBIP_EH_SHUTDOWN | USBIP_EH_UNUSABLE)
 258
 259 /* a common structure for stub_device and vhci_device */
 260 struct usbip_device {
 261         enum usbip_side side;
 262         enum usbip_device_status status;
 263
 264         /* lock for status */
 265         spinlock_t lock;
 266
 267         /* mutex for synchronizing sysfs store paths */
 268         struct mutex sysfs_lock;
 269
 270         int sockfd;
 271         struct socket *tcp_socket;
 272
 273         struct task_struct *tcp_rx;
 274         struct task_struct *tcp_tx;
 275
 276         unsigned long event;
 277         wait_queue_head_t eh_waitq;
 278
 279         struct eh_ops {
 280                 void (*shutdown)(struct usbip_device *);
 281                 void (*reset)(struct usbip_device *);
 282                 void (*unusable)(struct usbip_device *);
 283         } eh_ops;
 284
 285 #ifdef CONFIG_KCOV
 286         u64 kcov_handle;
 287 #endif
 288 };
 289
 290 #define kthread_get_run(threadfn, data, namefmt, ...)                      \
 291 ({                                                                         \
 292         struct task_struct *__k                                            \
 293                 = kthread_create(threadfn, data, namefmt, ## __VA_ARGS__); \
 294         if (!IS_ERR(__k)) {                                                \
 295                 get_task_struct(__k);                                      \
 296                 wake_up_process(__k);                                      \
 297         }                                                                  \
 298         __k;                                                               \
 299 })
 300
 301 #define kthread_stop_put(k)             \
 302         do {                            \
 303                 kthread_stop(k);        \
 304                 put_task_struct(k);     \
 305         } while (0)
 306
 307 /* usbip_common.c */
 308 void usbip_dump_urb(struct urb *purb);
 309 void usbip_dump_header(struct usbip_header *pdu);
 310
 311 int usbip_recv(struct socket *sock, void *buf, int size);
 312
 313 void usbip_pack_pdu(struct usbip_header *pdu, struct urb *urb, int cmd,
 314                     int pack);
 315 void usbip_header_correct_endian(struct usbip_header *pdu, int send);
 316
 317 struct usbip_iso_packet_descriptor*
 318 usbip_alloc_iso_desc_pdu(struct urb *urb, ssize_t *bufflen);
 319
 320 /* some members of urb must be substituted before. */
 321 int usbip_recv_iso(struct usbip_device *ud, struct urb *urb);
 322 void usbip_pad_iso(struct usbip_device *ud, struct urb *urb);
 323 int usbip_recv_xbuff(struct usbip_device *ud, struct urb *urb);
 324
 325 /* usbip_event.c */
 326 int usbip_init_eh(void);
 327 void usbip_finish_eh(void);
 328 int usbip_start_eh(struct usbip_device *ud);
 329 void usbip_stop_eh(struct usbip_device *ud);
 330 void usbip_event_add(struct usbip_device *ud, unsigned long event);
 331 int usbip_event_happened(struct usbip_device *ud);
 332 int usbip_in_eh(struct task_struct *task);
 333
 334 static inline int interface_to_busnum(struct usb_interface *interface)
 335 {
 336         struct usb_device *udev = interface_to_usbdev(interface);
 337
 338         return udev->bus->busnum;
 339 }
 340
 341 static inline int interface_to_devnum(struct usb_interface *interface)
 342 {
 343         struct usb_device *udev = interface_to_usbdev(interface);
 344
 345         return udev->devnum;
 346 }
 347
 348 #ifdef CONFIG_KCOV
 349
 350 static inline void usbip_kcov_handle_init(struct usbip_device *ud)
 351 {
 352         ud->kcov_handle = kcov_common_handle();
 353 }
 354
 355 static inline void usbip_kcov_remote_start(struct usbip_device *ud)
 356 {
 357         kcov_remote_start_common(ud->kcov_handle);
 358 }
 359
 360 static inline void usbip_kcov_remote_stop(void)
 361 {
 362         kcov_remote_stop();
 363 }
 364
 365 #else /* CONFIG_KCOV */
 366
 367 static inline void usbip_kcov_handle_init(struct usbip_device *ud) { }
 368 static inline void usbip_kcov_remote_start(struct usbip_device *ud) { }
 369 static inline void usbip_kcov_remote_stop(void) { }
 370
 371 #endif /* CONFIG_KCOV */
 372
 373 #endif /* __USBIP_COMMON_H */