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