2
0

redirect.c 85 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631
  1. /*
  2. * USB redirector usb-guest
  3. *
  4. * Copyright (c) 2011-2012 Red Hat, Inc.
  5. *
  6. * Red Hat Authors:
  7. * Hans de Goede <hdegoede@redhat.com>
  8. *
  9. * Permission is hereby granted, free of charge, to any person obtaining a copy
  10. * of this software and associated documentation files (the "Software"), to deal
  11. * in the Software without restriction, including without limitation the rights
  12. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  13. * copies of the Software, and to permit persons to whom the Software is
  14. * furnished to do so, subject to the following conditions:
  15. *
  16. * The above copyright notice and this permission notice shall be included in
  17. * all copies or substantial portions of the Software.
  18. *
  19. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  22. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  24. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  25. * THE SOFTWARE.
  26. */
  27. #include "qemu/osdep.h"
  28. #include "qemu/cutils.h"
  29. #include "qemu/units.h"
  30. #include "qapi/error.h"
  31. #include "qemu/timer.h"
  32. #include "sysemu/runstate.h"
  33. #include "sysemu/sysemu.h"
  34. #include "qapi/qmp/qerror.h"
  35. #include "qemu/error-report.h"
  36. #include "qemu/iov.h"
  37. #include "qemu/module.h"
  38. #include "chardev/char-fe.h"
  39. #include <usbredirparser.h>
  40. #include <usbredirfilter.h>
  41. #include "hw/qdev-properties.h"
  42. #include "hw/qdev-properties-system.h"
  43. #include "hw/usb.h"
  44. #include "migration/qemu-file-types.h"
  45. #include "migration/vmstate.h"
  46. #include "qom/object.h"
  47. /* ERROR is defined below. Remove any previous definition. */
  48. #undef ERROR
  49. #define MAX_ENDPOINTS 32
  50. #define NO_INTERFACE_INFO 255 /* Valid interface_count always <= 32 */
  51. #define EP2I(ep_address) (((ep_address & 0x80) >> 3) | (ep_address & 0x0f))
  52. #define I2EP(i) (((i & 0x10) << 3) | (i & 0x0f))
  53. #define USBEP2I(usb_ep) (((usb_ep)->pid == USB_TOKEN_IN) ? \
  54. ((usb_ep)->nr | 0x10) : ((usb_ep)->nr))
  55. #define I2USBEP(d, i) (usb_ep_get(&(d)->dev, \
  56. ((i) & 0x10) ? USB_TOKEN_IN : USB_TOKEN_OUT, \
  57. (i) & 0x0f))
  58. #ifndef USBREDIR_VERSION /* This is not defined in older usbredir versions */
  59. #define USBREDIR_VERSION 0
  60. #endif
  61. typedef struct USBRedirDevice USBRedirDevice;
  62. /* Struct to hold buffered packets */
  63. struct buf_packet {
  64. uint8_t *data;
  65. void *free_on_destroy;
  66. uint16_t len;
  67. uint16_t offset;
  68. uint8_t status;
  69. QTAILQ_ENTRY(buf_packet)next;
  70. };
  71. struct endp_data {
  72. USBRedirDevice *dev;
  73. uint8_t type;
  74. uint8_t interval;
  75. uint8_t interface; /* bInterfaceNumber this ep belongs to */
  76. uint16_t max_packet_size; /* In bytes, not wMaxPacketSize format !! */
  77. uint32_t max_streams;
  78. uint8_t iso_started;
  79. uint8_t iso_error; /* For reporting iso errors to the HC */
  80. uint8_t interrupt_started;
  81. uint8_t interrupt_error;
  82. uint8_t bulk_receiving_enabled;
  83. uint8_t bulk_receiving_started;
  84. uint8_t bufpq_prefilled;
  85. uint8_t bufpq_dropping_packets;
  86. QTAILQ_HEAD(, buf_packet) bufpq;
  87. int32_t bufpq_size;
  88. int32_t bufpq_target_size;
  89. USBPacket *pending_async_packet;
  90. };
  91. struct PacketIdQueueEntry {
  92. uint64_t id;
  93. QTAILQ_ENTRY(PacketIdQueueEntry)next;
  94. };
  95. struct PacketIdQueue {
  96. USBRedirDevice *dev;
  97. const char *name;
  98. QTAILQ_HEAD(, PacketIdQueueEntry) head;
  99. int size;
  100. };
  101. struct USBRedirDevice {
  102. USBDevice dev;
  103. /* Properties */
  104. CharBackend cs;
  105. bool enable_streams;
  106. bool suppress_remote_wake;
  107. bool in_write;
  108. uint8_t debug;
  109. int32_t bootindex;
  110. char *filter_str;
  111. /* Data passed from chardev the fd_read cb to the usbredirparser read cb */
  112. const uint8_t *read_buf;
  113. int read_buf_size;
  114. /* Active chardev-watch-tag */
  115. guint watch;
  116. /* For async handling of close / reject */
  117. QEMUBH *chardev_close_bh;
  118. QEMUBH *device_reject_bh;
  119. /* To delay the usb attach in case of quick chardev close + open */
  120. QEMUTimer *attach_timer;
  121. int64_t next_attach_time;
  122. struct usbredirparser *parser;
  123. struct endp_data endpoint[MAX_ENDPOINTS];
  124. struct PacketIdQueue cancelled;
  125. struct PacketIdQueue already_in_flight;
  126. void (*buffered_bulk_in_complete)(USBRedirDevice *, USBPacket *, uint8_t);
  127. /* Data for device filtering */
  128. struct usb_redir_device_connect_header device_info;
  129. struct usb_redir_interface_info_header interface_info;
  130. struct usbredirfilter_rule *filter_rules;
  131. int filter_rules_count;
  132. int compatible_speedmask;
  133. VMChangeStateEntry *vmstate;
  134. };
  135. #define TYPE_USB_REDIR "usb-redir"
  136. DECLARE_INSTANCE_CHECKER(USBRedirDevice, USB_REDIRECT,
  137. TYPE_USB_REDIR)
  138. static void usbredir_hello(void *priv, struct usb_redir_hello_header *h);
  139. static void usbredir_device_connect(void *priv,
  140. struct usb_redir_device_connect_header *device_connect);
  141. static void usbredir_device_disconnect(void *priv);
  142. static void usbredir_interface_info(void *priv,
  143. struct usb_redir_interface_info_header *interface_info);
  144. static void usbredir_ep_info(void *priv,
  145. struct usb_redir_ep_info_header *ep_info);
  146. static void usbredir_configuration_status(void *priv, uint64_t id,
  147. struct usb_redir_configuration_status_header *configuration_status);
  148. static void usbredir_alt_setting_status(void *priv, uint64_t id,
  149. struct usb_redir_alt_setting_status_header *alt_setting_status);
  150. static void usbredir_iso_stream_status(void *priv, uint64_t id,
  151. struct usb_redir_iso_stream_status_header *iso_stream_status);
  152. static void usbredir_interrupt_receiving_status(void *priv, uint64_t id,
  153. struct usb_redir_interrupt_receiving_status_header
  154. *interrupt_receiving_status);
  155. static void usbredir_bulk_streams_status(void *priv, uint64_t id,
  156. struct usb_redir_bulk_streams_status_header *bulk_streams_status);
  157. static void usbredir_bulk_receiving_status(void *priv, uint64_t id,
  158. struct usb_redir_bulk_receiving_status_header *bulk_receiving_status);
  159. static void usbredir_control_packet(void *priv, uint64_t id,
  160. struct usb_redir_control_packet_header *control_packet,
  161. uint8_t *data, int data_len);
  162. static void usbredir_bulk_packet(void *priv, uint64_t id,
  163. struct usb_redir_bulk_packet_header *bulk_packet,
  164. uint8_t *data, int data_len);
  165. static void usbredir_iso_packet(void *priv, uint64_t id,
  166. struct usb_redir_iso_packet_header *iso_packet,
  167. uint8_t *data, int data_len);
  168. static void usbredir_interrupt_packet(void *priv, uint64_t id,
  169. struct usb_redir_interrupt_packet_header *interrupt_header,
  170. uint8_t *data, int data_len);
  171. static void usbredir_buffered_bulk_packet(void *priv, uint64_t id,
  172. struct usb_redir_buffered_bulk_packet_header *buffered_bulk_packet,
  173. uint8_t *data, int data_len);
  174. static void usbredir_handle_status(USBRedirDevice *dev, USBPacket *p,
  175. int status);
  176. #define VERSION "qemu usb-redir guest " QEMU_VERSION
  177. /*
  178. * Logging stuff
  179. */
  180. #define ERROR(...) \
  181. do { \
  182. if (dev->debug >= usbredirparser_error) { \
  183. error_report("usb-redir error: " __VA_ARGS__); \
  184. } \
  185. } while (0)
  186. #define WARNING(...) \
  187. do { \
  188. if (dev->debug >= usbredirparser_warning) { \
  189. warn_report("" __VA_ARGS__); \
  190. } \
  191. } while (0)
  192. #define INFO(...) \
  193. do { \
  194. if (dev->debug >= usbredirparser_info) { \
  195. error_report("usb-redir: " __VA_ARGS__); \
  196. } \
  197. } while (0)
  198. #define DPRINTF(...) \
  199. do { \
  200. if (dev->debug >= usbredirparser_debug) { \
  201. error_report("usb-redir: " __VA_ARGS__); \
  202. } \
  203. } while (0)
  204. #define DPRINTF2(...) \
  205. do { \
  206. if (dev->debug >= usbredirparser_debug_data) { \
  207. error_report("usb-redir: " __VA_ARGS__); \
  208. } \
  209. } while (0)
  210. static void usbredir_log(void *priv, int level, const char *msg)
  211. {
  212. USBRedirDevice *dev = priv;
  213. if (dev->debug < level) {
  214. return;
  215. }
  216. error_report("%s", msg);
  217. }
  218. static void usbredir_log_data(USBRedirDevice *dev, const char *desc,
  219. const uint8_t *data, int len)
  220. {
  221. if (dev->debug < usbredirparser_debug_data) {
  222. return;
  223. }
  224. qemu_hexdump(stderr, desc, data, len);
  225. }
  226. /*
  227. * usbredirparser io functions
  228. */
  229. static int usbredir_read(void *priv, uint8_t *data, int count)
  230. {
  231. USBRedirDevice *dev = priv;
  232. if (dev->read_buf_size < count) {
  233. count = dev->read_buf_size;
  234. }
  235. memcpy(data, dev->read_buf, count);
  236. dev->read_buf_size -= count;
  237. if (dev->read_buf_size) {
  238. dev->read_buf += count;
  239. } else {
  240. dev->read_buf = NULL;
  241. }
  242. return count;
  243. }
  244. static gboolean usbredir_write_unblocked(void *do_not_use, GIOCondition cond,
  245. void *opaque)
  246. {
  247. USBRedirDevice *dev = opaque;
  248. dev->watch = 0;
  249. usbredirparser_do_write(dev->parser);
  250. return FALSE;
  251. }
  252. static int usbredir_write(void *priv, uint8_t *data, int count)
  253. {
  254. USBRedirDevice *dev = priv;
  255. int r;
  256. if (!qemu_chr_fe_backend_open(&dev->cs)) {
  257. return 0;
  258. }
  259. /* Don't send new data to the chardev until our state is fully synced */
  260. if (!runstate_check(RUN_STATE_RUNNING)) {
  261. return 0;
  262. }
  263. /* Recursion check */
  264. if (dev->in_write) {
  265. DPRINTF("usbredir_write recursion\n");
  266. return 0;
  267. }
  268. dev->in_write = true;
  269. r = qemu_chr_fe_write(&dev->cs, data, count);
  270. if (r < count) {
  271. if (!dev->watch) {
  272. dev->watch = qemu_chr_fe_add_watch(&dev->cs, G_IO_OUT | G_IO_HUP,
  273. usbredir_write_unblocked, dev);
  274. }
  275. if (r < 0) {
  276. r = 0;
  277. }
  278. }
  279. dev->in_write = false;
  280. return r;
  281. }
  282. /*
  283. * Cancelled and buffered packets helpers
  284. */
  285. static void packet_id_queue_init(struct PacketIdQueue *q,
  286. USBRedirDevice *dev, const char *name)
  287. {
  288. q->dev = dev;
  289. q->name = name;
  290. QTAILQ_INIT(&q->head);
  291. q->size = 0;
  292. }
  293. static void packet_id_queue_add(struct PacketIdQueue *q, uint64_t id)
  294. {
  295. USBRedirDevice *dev = q->dev;
  296. struct PacketIdQueueEntry *e;
  297. DPRINTF("adding packet id %"PRIu64" to %s queue\n", id, q->name);
  298. e = g_new0(struct PacketIdQueueEntry, 1);
  299. e->id = id;
  300. QTAILQ_INSERT_TAIL(&q->head, e, next);
  301. q->size++;
  302. }
  303. static int packet_id_queue_remove(struct PacketIdQueue *q, uint64_t id)
  304. {
  305. USBRedirDevice *dev = q->dev;
  306. struct PacketIdQueueEntry *e;
  307. QTAILQ_FOREACH(e, &q->head, next) {
  308. if (e->id == id) {
  309. DPRINTF("removing packet id %"PRIu64" from %s queue\n",
  310. id, q->name);
  311. QTAILQ_REMOVE(&q->head, e, next);
  312. q->size--;
  313. g_free(e);
  314. return 1;
  315. }
  316. }
  317. return 0;
  318. }
  319. static void packet_id_queue_empty(struct PacketIdQueue *q)
  320. {
  321. USBRedirDevice *dev = q->dev;
  322. struct PacketIdQueueEntry *e, *next_e;
  323. DPRINTF("removing %d packet-ids from %s queue\n", q->size, q->name);
  324. QTAILQ_FOREACH_SAFE(e, &q->head, next, next_e) {
  325. QTAILQ_REMOVE(&q->head, e, next);
  326. g_free(e);
  327. }
  328. q->size = 0;
  329. }
  330. static void usbredir_cancel_packet(USBDevice *udev, USBPacket *p)
  331. {
  332. USBRedirDevice *dev = USB_REDIRECT(udev);
  333. int i = USBEP2I(p->ep);
  334. if (p->combined) {
  335. usb_combined_packet_cancel(udev, p);
  336. return;
  337. }
  338. if (dev->endpoint[i].pending_async_packet) {
  339. assert(dev->endpoint[i].pending_async_packet == p);
  340. dev->endpoint[i].pending_async_packet = NULL;
  341. return;
  342. }
  343. packet_id_queue_add(&dev->cancelled, p->id);
  344. usbredirparser_send_cancel_data_packet(dev->parser, p->id);
  345. usbredirparser_do_write(dev->parser);
  346. }
  347. static int usbredir_is_cancelled(USBRedirDevice *dev, uint64_t id)
  348. {
  349. if (!dev->dev.attached) {
  350. return 1; /* Treat everything as cancelled after a disconnect */
  351. }
  352. return packet_id_queue_remove(&dev->cancelled, id);
  353. }
  354. static void usbredir_fill_already_in_flight_from_ep(USBRedirDevice *dev,
  355. struct USBEndpoint *ep)
  356. {
  357. static USBPacket *p;
  358. /* async handled packets for bulk receiving eps do not count as inflight */
  359. if (dev->endpoint[USBEP2I(ep)].bulk_receiving_started) {
  360. return;
  361. }
  362. QTAILQ_FOREACH(p, &ep->queue, queue) {
  363. /* Skip combined packets, except for the first */
  364. if (p->combined && p != p->combined->first) {
  365. continue;
  366. }
  367. if (p->state == USB_PACKET_ASYNC) {
  368. packet_id_queue_add(&dev->already_in_flight, p->id);
  369. }
  370. }
  371. }
  372. static void usbredir_fill_already_in_flight(USBRedirDevice *dev)
  373. {
  374. int ep;
  375. struct USBDevice *udev = &dev->dev;
  376. usbredir_fill_already_in_flight_from_ep(dev, &udev->ep_ctl);
  377. for (ep = 0; ep < USB_MAX_ENDPOINTS; ep++) {
  378. usbredir_fill_already_in_flight_from_ep(dev, &udev->ep_in[ep]);
  379. usbredir_fill_already_in_flight_from_ep(dev, &udev->ep_out[ep]);
  380. }
  381. }
  382. static int usbredir_already_in_flight(USBRedirDevice *dev, uint64_t id)
  383. {
  384. return packet_id_queue_remove(&dev->already_in_flight, id);
  385. }
  386. static USBPacket *usbredir_find_packet_by_id(USBRedirDevice *dev,
  387. uint8_t ep, uint64_t id)
  388. {
  389. USBPacket *p;
  390. if (usbredir_is_cancelled(dev, id)) {
  391. return NULL;
  392. }
  393. p = usb_ep_find_packet_by_id(&dev->dev,
  394. (ep & USB_DIR_IN) ? USB_TOKEN_IN : USB_TOKEN_OUT,
  395. ep & 0x0f, id);
  396. if (p == NULL) {
  397. ERROR("could not find packet with id %"PRIu64"\n", id);
  398. }
  399. return p;
  400. }
  401. static int bufp_alloc(USBRedirDevice *dev, uint8_t *data, uint16_t len,
  402. uint8_t status, uint8_t ep, void *free_on_destroy)
  403. {
  404. struct buf_packet *bufp;
  405. if (!dev->endpoint[EP2I(ep)].bufpq_dropping_packets &&
  406. dev->endpoint[EP2I(ep)].bufpq_size >
  407. 2 * dev->endpoint[EP2I(ep)].bufpq_target_size) {
  408. DPRINTF("bufpq overflow, dropping packets ep %02X\n", ep);
  409. dev->endpoint[EP2I(ep)].bufpq_dropping_packets = 1;
  410. }
  411. /* Since we're interupting the stream anyways, drop enough packets to get
  412. back to our target buffer size */
  413. if (dev->endpoint[EP2I(ep)].bufpq_dropping_packets) {
  414. if (dev->endpoint[EP2I(ep)].bufpq_size >
  415. dev->endpoint[EP2I(ep)].bufpq_target_size) {
  416. free(free_on_destroy);
  417. return -1;
  418. }
  419. dev->endpoint[EP2I(ep)].bufpq_dropping_packets = 0;
  420. }
  421. bufp = g_new(struct buf_packet, 1);
  422. bufp->data = data;
  423. bufp->len = len;
  424. bufp->offset = 0;
  425. bufp->status = status;
  426. bufp->free_on_destroy = free_on_destroy;
  427. QTAILQ_INSERT_TAIL(&dev->endpoint[EP2I(ep)].bufpq, bufp, next);
  428. dev->endpoint[EP2I(ep)].bufpq_size++;
  429. return 0;
  430. }
  431. static void bufp_free(USBRedirDevice *dev, struct buf_packet *bufp,
  432. uint8_t ep)
  433. {
  434. QTAILQ_REMOVE(&dev->endpoint[EP2I(ep)].bufpq, bufp, next);
  435. dev->endpoint[EP2I(ep)].bufpq_size--;
  436. free(bufp->free_on_destroy);
  437. g_free(bufp);
  438. }
  439. static void usbredir_free_bufpq(USBRedirDevice *dev, uint8_t ep)
  440. {
  441. struct buf_packet *buf, *buf_next;
  442. QTAILQ_FOREACH_SAFE(buf, &dev->endpoint[EP2I(ep)].bufpq, next, buf_next) {
  443. bufp_free(dev, buf, ep);
  444. }
  445. }
  446. /*
  447. * USBDevice callbacks
  448. */
  449. static void usbredir_handle_reset(USBDevice *udev)
  450. {
  451. USBRedirDevice *dev = USB_REDIRECT(udev);
  452. DPRINTF("reset device\n");
  453. usbredirparser_send_reset(dev->parser);
  454. usbredirparser_do_write(dev->parser);
  455. }
  456. static void usbredir_handle_iso_data(USBRedirDevice *dev, USBPacket *p,
  457. uint8_t ep)
  458. {
  459. int status, len;
  460. if (!dev->endpoint[EP2I(ep)].iso_started &&
  461. !dev->endpoint[EP2I(ep)].iso_error) {
  462. struct usb_redir_start_iso_stream_header start_iso = {
  463. .endpoint = ep,
  464. };
  465. int pkts_per_sec;
  466. if (dev->dev.speed == USB_SPEED_HIGH) {
  467. pkts_per_sec = 8000 / dev->endpoint[EP2I(ep)].interval;
  468. } else {
  469. pkts_per_sec = 1000 / dev->endpoint[EP2I(ep)].interval;
  470. }
  471. /* Testing has shown that we need circa 60 ms buffer */
  472. dev->endpoint[EP2I(ep)].bufpq_target_size = (pkts_per_sec * 60) / 1000;
  473. /* Aim for approx 100 interrupts / second on the client to
  474. balance latency and interrupt load */
  475. start_iso.pkts_per_urb = pkts_per_sec / 100;
  476. if (start_iso.pkts_per_urb < 1) {
  477. start_iso.pkts_per_urb = 1;
  478. } else if (start_iso.pkts_per_urb > 32) {
  479. start_iso.pkts_per_urb = 32;
  480. }
  481. start_iso.no_urbs = DIV_ROUND_UP(
  482. dev->endpoint[EP2I(ep)].bufpq_target_size,
  483. start_iso.pkts_per_urb);
  484. /* Output endpoints pre-fill only 1/2 of the packets, keeping the rest
  485. as overflow buffer. Also see the usbredir protocol documentation */
  486. if (!(ep & USB_DIR_IN)) {
  487. start_iso.no_urbs *= 2;
  488. }
  489. if (start_iso.no_urbs > 16) {
  490. start_iso.no_urbs = 16;
  491. }
  492. /* No id, we look at the ep when receiving a status back */
  493. usbredirparser_send_start_iso_stream(dev->parser, 0, &start_iso);
  494. usbredirparser_do_write(dev->parser);
  495. DPRINTF("iso stream started pkts/sec %d pkts/urb %d urbs %d ep %02X\n",
  496. pkts_per_sec, start_iso.pkts_per_urb, start_iso.no_urbs, ep);
  497. dev->endpoint[EP2I(ep)].iso_started = 1;
  498. dev->endpoint[EP2I(ep)].bufpq_prefilled = 0;
  499. dev->endpoint[EP2I(ep)].bufpq_dropping_packets = 0;
  500. }
  501. if (ep & USB_DIR_IN) {
  502. struct buf_packet *isop;
  503. if (dev->endpoint[EP2I(ep)].iso_started &&
  504. !dev->endpoint[EP2I(ep)].bufpq_prefilled) {
  505. if (dev->endpoint[EP2I(ep)].bufpq_size <
  506. dev->endpoint[EP2I(ep)].bufpq_target_size) {
  507. return;
  508. }
  509. dev->endpoint[EP2I(ep)].bufpq_prefilled = 1;
  510. }
  511. isop = QTAILQ_FIRST(&dev->endpoint[EP2I(ep)].bufpq);
  512. if (isop == NULL) {
  513. DPRINTF("iso-token-in ep %02X, no isop, iso_error: %d\n",
  514. ep, dev->endpoint[EP2I(ep)].iso_error);
  515. /* Re-fill the buffer */
  516. dev->endpoint[EP2I(ep)].bufpq_prefilled = 0;
  517. /* Check iso_error for stream errors, otherwise its an underrun */
  518. status = dev->endpoint[EP2I(ep)].iso_error;
  519. dev->endpoint[EP2I(ep)].iso_error = 0;
  520. p->status = status ? USB_RET_IOERROR : USB_RET_SUCCESS;
  521. return;
  522. }
  523. DPRINTF2("iso-token-in ep %02X status %d len %d queue-size: %d\n", ep,
  524. isop->status, isop->len, dev->endpoint[EP2I(ep)].bufpq_size);
  525. status = isop->status;
  526. len = isop->len;
  527. if (len > p->iov.size) {
  528. ERROR("received iso data is larger then packet ep %02X (%d > %d)\n",
  529. ep, len, (int)p->iov.size);
  530. len = p->iov.size;
  531. status = usb_redir_babble;
  532. }
  533. usb_packet_copy(p, isop->data, len);
  534. bufp_free(dev, isop, ep);
  535. usbredir_handle_status(dev, p, status);
  536. } else {
  537. /* If the stream was not started because of a pending error don't
  538. send the packet to the usb-host */
  539. if (dev->endpoint[EP2I(ep)].iso_started) {
  540. struct usb_redir_iso_packet_header iso_packet = {
  541. .endpoint = ep,
  542. .length = p->iov.size
  543. };
  544. g_autofree uint8_t *buf = g_malloc(p->iov.size);
  545. /* No id, we look at the ep when receiving a status back */
  546. usb_packet_copy(p, buf, p->iov.size);
  547. usbredirparser_send_iso_packet(dev->parser, 0, &iso_packet,
  548. buf, p->iov.size);
  549. usbredirparser_do_write(dev->parser);
  550. }
  551. status = dev->endpoint[EP2I(ep)].iso_error;
  552. dev->endpoint[EP2I(ep)].iso_error = 0;
  553. DPRINTF2("iso-token-out ep %02X status %d len %zd\n", ep, status,
  554. p->iov.size);
  555. usbredir_handle_status(dev, p, status);
  556. }
  557. }
  558. static void usbredir_stop_iso_stream(USBRedirDevice *dev, uint8_t ep)
  559. {
  560. struct usb_redir_stop_iso_stream_header stop_iso_stream = {
  561. .endpoint = ep
  562. };
  563. if (dev->endpoint[EP2I(ep)].iso_started) {
  564. usbredirparser_send_stop_iso_stream(dev->parser, 0, &stop_iso_stream);
  565. DPRINTF("iso stream stopped ep %02X\n", ep);
  566. dev->endpoint[EP2I(ep)].iso_started = 0;
  567. }
  568. dev->endpoint[EP2I(ep)].iso_error = 0;
  569. usbredir_free_bufpq(dev, ep);
  570. }
  571. /*
  572. * The usb-host may poll the endpoint faster then our guest, resulting in lots
  573. * of smaller bulkp-s. The below buffered_bulk_in_complete* functions combine
  574. * data from multiple bulkp-s into a single packet, avoiding bufpq overflows.
  575. */
  576. static void usbredir_buffered_bulk_add_data_to_packet(USBRedirDevice *dev,
  577. struct buf_packet *bulkp, int count, USBPacket *p, uint8_t ep)
  578. {
  579. usb_packet_copy(p, bulkp->data + bulkp->offset, count);
  580. bulkp->offset += count;
  581. if (bulkp->offset == bulkp->len) {
  582. /* Store status in the last packet with data from this bulkp */
  583. usbredir_handle_status(dev, p, bulkp->status);
  584. bufp_free(dev, bulkp, ep);
  585. }
  586. }
  587. static void usbredir_buffered_bulk_in_complete_raw(USBRedirDevice *dev,
  588. USBPacket *p, uint8_t ep)
  589. {
  590. struct buf_packet *bulkp;
  591. int count;
  592. while ((bulkp = QTAILQ_FIRST(&dev->endpoint[EP2I(ep)].bufpq)) &&
  593. p->actual_length < p->iov.size && p->status == USB_RET_SUCCESS) {
  594. count = bulkp->len - bulkp->offset;
  595. if (count > (p->iov.size - p->actual_length)) {
  596. count = p->iov.size - p->actual_length;
  597. }
  598. usbredir_buffered_bulk_add_data_to_packet(dev, bulkp, count, p, ep);
  599. }
  600. }
  601. static void usbredir_buffered_bulk_in_complete_ftdi(USBRedirDevice *dev,
  602. USBPacket *p, uint8_t ep)
  603. {
  604. const int maxp = dev->endpoint[EP2I(ep)].max_packet_size;
  605. uint8_t header[2] = { 0, 0 };
  606. struct buf_packet *bulkp;
  607. int count;
  608. while ((bulkp = QTAILQ_FIRST(&dev->endpoint[EP2I(ep)].bufpq)) &&
  609. p->actual_length < p->iov.size && p->status == USB_RET_SUCCESS) {
  610. if (bulkp->len < 2) {
  611. WARNING("malformed ftdi bulk in packet\n");
  612. bufp_free(dev, bulkp, ep);
  613. continue;
  614. }
  615. if ((p->actual_length % maxp) == 0) {
  616. usb_packet_copy(p, bulkp->data, 2);
  617. memcpy(header, bulkp->data, 2);
  618. } else {
  619. if (bulkp->data[0] != header[0] || bulkp->data[1] != header[1]) {
  620. break; /* Different header, add to next packet */
  621. }
  622. }
  623. if (bulkp->offset == 0) {
  624. bulkp->offset = 2; /* Skip header */
  625. }
  626. count = bulkp->len - bulkp->offset;
  627. /* Must repeat the header at maxp interval */
  628. if (count > (maxp - (p->actual_length % maxp))) {
  629. count = maxp - (p->actual_length % maxp);
  630. }
  631. usbredir_buffered_bulk_add_data_to_packet(dev, bulkp, count, p, ep);
  632. }
  633. }
  634. static void usbredir_buffered_bulk_in_complete(USBRedirDevice *dev,
  635. USBPacket *p, uint8_t ep)
  636. {
  637. p->status = USB_RET_SUCCESS; /* Clear previous ASYNC status */
  638. dev->buffered_bulk_in_complete(dev, p, ep);
  639. DPRINTF("bulk-token-in ep %02X status %d len %d id %"PRIu64"\n",
  640. ep, p->status, p->actual_length, p->id);
  641. }
  642. static void usbredir_handle_buffered_bulk_in_data(USBRedirDevice *dev,
  643. USBPacket *p, uint8_t ep)
  644. {
  645. /* Input bulk endpoint, buffered packet input */
  646. if (!dev->endpoint[EP2I(ep)].bulk_receiving_started) {
  647. int bpt;
  648. struct usb_redir_start_bulk_receiving_header start = {
  649. .endpoint = ep,
  650. .stream_id = 0,
  651. .no_transfers = 5,
  652. };
  653. /* Round bytes_per_transfer up to a multiple of max_packet_size */
  654. bpt = 512 + dev->endpoint[EP2I(ep)].max_packet_size - 1;
  655. bpt /= dev->endpoint[EP2I(ep)].max_packet_size;
  656. bpt *= dev->endpoint[EP2I(ep)].max_packet_size;
  657. start.bytes_per_transfer = bpt;
  658. /* No id, we look at the ep when receiving a status back */
  659. usbredirparser_send_start_bulk_receiving(dev->parser, 0, &start);
  660. usbredirparser_do_write(dev->parser);
  661. DPRINTF("bulk receiving started bytes/transfer %u count %d ep %02X\n",
  662. start.bytes_per_transfer, start.no_transfers, ep);
  663. dev->endpoint[EP2I(ep)].bulk_receiving_started = 1;
  664. /* We don't really want to drop bulk packets ever, but
  665. having some upper limit to how much we buffer is good. */
  666. dev->endpoint[EP2I(ep)].bufpq_target_size = 5000;
  667. dev->endpoint[EP2I(ep)].bufpq_dropping_packets = 0;
  668. }
  669. if (QTAILQ_EMPTY(&dev->endpoint[EP2I(ep)].bufpq)) {
  670. DPRINTF("bulk-token-in ep %02X, no bulkp\n", ep);
  671. assert(dev->endpoint[EP2I(ep)].pending_async_packet == NULL);
  672. dev->endpoint[EP2I(ep)].pending_async_packet = p;
  673. p->status = USB_RET_ASYNC;
  674. return;
  675. }
  676. usbredir_buffered_bulk_in_complete(dev, p, ep);
  677. }
  678. static void usbredir_stop_bulk_receiving(USBRedirDevice *dev, uint8_t ep)
  679. {
  680. struct usb_redir_stop_bulk_receiving_header stop_bulk = {
  681. .endpoint = ep,
  682. .stream_id = 0,
  683. };
  684. if (dev->endpoint[EP2I(ep)].bulk_receiving_started) {
  685. usbredirparser_send_stop_bulk_receiving(dev->parser, 0, &stop_bulk);
  686. DPRINTF("bulk receiving stopped ep %02X\n", ep);
  687. dev->endpoint[EP2I(ep)].bulk_receiving_started = 0;
  688. }
  689. usbredir_free_bufpq(dev, ep);
  690. }
  691. static void usbredir_handle_bulk_data(USBRedirDevice *dev, USBPacket *p,
  692. uint8_t ep)
  693. {
  694. struct usb_redir_bulk_packet_header bulk_packet;
  695. size_t size = usb_packet_size(p);
  696. const int maxp = dev->endpoint[EP2I(ep)].max_packet_size;
  697. if (usbredir_already_in_flight(dev, p->id)) {
  698. p->status = USB_RET_ASYNC;
  699. return;
  700. }
  701. if (dev->endpoint[EP2I(ep)].bulk_receiving_enabled) {
  702. if (size != 0 && (size % maxp) == 0) {
  703. usbredir_handle_buffered_bulk_in_data(dev, p, ep);
  704. return;
  705. }
  706. WARNING("bulk recv invalid size %zd ep %02x, disabling\n", size, ep);
  707. assert(dev->endpoint[EP2I(ep)].pending_async_packet == NULL);
  708. usbredir_stop_bulk_receiving(dev, ep);
  709. dev->endpoint[EP2I(ep)].bulk_receiving_enabled = 0;
  710. }
  711. DPRINTF("bulk-out ep %02X stream %u len %zd id %"PRIu64"\n",
  712. ep, p->stream, size, p->id);
  713. bulk_packet.endpoint = ep;
  714. bulk_packet.length = size;
  715. bulk_packet.stream_id = p->stream;
  716. bulk_packet.length_high = size >> 16;
  717. assert(bulk_packet.length_high == 0 ||
  718. usbredirparser_peer_has_cap(dev->parser,
  719. usb_redir_cap_32bits_bulk_length));
  720. if (ep & USB_DIR_IN || size == 0) {
  721. usbredirparser_send_bulk_packet(dev->parser, p->id,
  722. &bulk_packet, NULL, 0);
  723. } else {
  724. g_autofree uint8_t *buf = g_malloc(size);
  725. usb_packet_copy(p, buf, size);
  726. usbredir_log_data(dev, "bulk data out:", buf, size);
  727. usbredirparser_send_bulk_packet(dev->parser, p->id,
  728. &bulk_packet, buf, size);
  729. }
  730. usbredirparser_do_write(dev->parser);
  731. p->status = USB_RET_ASYNC;
  732. }
  733. static void usbredir_handle_interrupt_in_data(USBRedirDevice *dev,
  734. USBPacket *p, uint8_t ep)
  735. {
  736. /* Input interrupt endpoint, buffered packet input */
  737. struct buf_packet *intp, *intp_to_free;
  738. int status, len, sum;
  739. if (!dev->endpoint[EP2I(ep)].interrupt_started &&
  740. !dev->endpoint[EP2I(ep)].interrupt_error) {
  741. struct usb_redir_start_interrupt_receiving_header start_int = {
  742. .endpoint = ep,
  743. };
  744. /* No id, we look at the ep when receiving a status back */
  745. usbredirparser_send_start_interrupt_receiving(dev->parser, 0,
  746. &start_int);
  747. usbredirparser_do_write(dev->parser);
  748. DPRINTF("interrupt recv started ep %02X\n", ep);
  749. dev->endpoint[EP2I(ep)].interrupt_started = 1;
  750. /* We don't really want to drop interrupt packets ever, but
  751. having some upper limit to how much we buffer is good. */
  752. dev->endpoint[EP2I(ep)].bufpq_target_size = 1000;
  753. dev->endpoint[EP2I(ep)].bufpq_dropping_packets = 0;
  754. }
  755. /* check for completed interrupt message (with all fragments) */
  756. sum = 0;
  757. QTAILQ_FOREACH(intp, &dev->endpoint[EP2I(ep)].bufpq, next) {
  758. sum += intp->len;
  759. if (intp->len < dev->endpoint[EP2I(ep)].max_packet_size ||
  760. sum >= p->iov.size)
  761. break;
  762. }
  763. if (intp == NULL) {
  764. DPRINTF2("interrupt-token-in ep %02X, no intp, buffered %d\n", ep, sum);
  765. /* Check interrupt_error for stream errors */
  766. status = dev->endpoint[EP2I(ep)].interrupt_error;
  767. dev->endpoint[EP2I(ep)].interrupt_error = 0;
  768. if (status) {
  769. usbredir_handle_status(dev, p, status);
  770. } else {
  771. p->status = USB_RET_NAK;
  772. }
  773. return;
  774. }
  775. /* copy of completed interrupt message */
  776. sum = 0;
  777. status = usb_redir_success;
  778. intp_to_free = NULL;
  779. QTAILQ_FOREACH(intp, &dev->endpoint[EP2I(ep)].bufpq, next) {
  780. if (intp_to_free) {
  781. bufp_free(dev, intp_to_free, ep);
  782. }
  783. DPRINTF("interrupt-token-in ep %02X fragment status %d len %d\n", ep,
  784. intp->status, intp->len);
  785. sum += intp->len;
  786. len = intp->len;
  787. if (status == usb_redir_success) {
  788. status = intp->status;
  789. }
  790. if (sum > p->iov.size) {
  791. ERROR("received int data is larger then packet ep %02X\n", ep);
  792. len -= (sum - p->iov.size);
  793. sum = p->iov.size;
  794. status = usb_redir_babble;
  795. }
  796. usb_packet_copy(p, intp->data, len);
  797. intp_to_free = intp;
  798. if (intp->len < dev->endpoint[EP2I(ep)].max_packet_size ||
  799. sum >= p->iov.size)
  800. break;
  801. }
  802. if (intp_to_free) {
  803. bufp_free(dev, intp_to_free, ep);
  804. }
  805. DPRINTF("interrupt-token-in ep %02X summary status %d len %d\n", ep,
  806. status, sum);
  807. usbredir_handle_status(dev, p, status);
  808. }
  809. /*
  810. * Handle interrupt out data, the usbredir protocol expects us to do this
  811. * async, so that it can report back a completion status. But guests will
  812. * expect immediate completion for an interrupt endpoint, and handling this
  813. * async causes migration issues. So we report success directly, counting
  814. * on the fact that output interrupt packets normally always succeed.
  815. */
  816. static void usbredir_handle_interrupt_out_data(USBRedirDevice *dev,
  817. USBPacket *p, uint8_t ep)
  818. {
  819. struct usb_redir_interrupt_packet_header interrupt_packet;
  820. g_autofree uint8_t *buf = g_malloc(p->iov.size);
  821. DPRINTF("interrupt-out ep %02X len %zd id %"PRIu64"\n", ep,
  822. p->iov.size, p->id);
  823. interrupt_packet.endpoint = ep;
  824. interrupt_packet.length = p->iov.size;
  825. usb_packet_copy(p, buf, p->iov.size);
  826. usbredir_log_data(dev, "interrupt data out:", buf, p->iov.size);
  827. usbredirparser_send_interrupt_packet(dev->parser, p->id,
  828. &interrupt_packet, buf, p->iov.size);
  829. usbredirparser_do_write(dev->parser);
  830. }
  831. static void usbredir_stop_interrupt_receiving(USBRedirDevice *dev,
  832. uint8_t ep)
  833. {
  834. struct usb_redir_stop_interrupt_receiving_header stop_interrupt_recv = {
  835. .endpoint = ep
  836. };
  837. if (dev->endpoint[EP2I(ep)].interrupt_started) {
  838. usbredirparser_send_stop_interrupt_receiving(dev->parser, 0,
  839. &stop_interrupt_recv);
  840. DPRINTF("interrupt recv stopped ep %02X\n", ep);
  841. dev->endpoint[EP2I(ep)].interrupt_started = 0;
  842. }
  843. dev->endpoint[EP2I(ep)].interrupt_error = 0;
  844. usbredir_free_bufpq(dev, ep);
  845. }
  846. static void usbredir_handle_data(USBDevice *udev, USBPacket *p)
  847. {
  848. USBRedirDevice *dev = USB_REDIRECT(udev);
  849. uint8_t ep;
  850. ep = p->ep->nr;
  851. if (p->pid == USB_TOKEN_IN) {
  852. ep |= USB_DIR_IN;
  853. }
  854. switch (dev->endpoint[EP2I(ep)].type) {
  855. case USB_ENDPOINT_XFER_CONTROL:
  856. ERROR("handle_data called for control transfer on ep %02X\n", ep);
  857. p->status = USB_RET_NAK;
  858. break;
  859. case USB_ENDPOINT_XFER_BULK:
  860. if (p->state == USB_PACKET_SETUP && p->pid == USB_TOKEN_IN &&
  861. p->ep->pipeline) {
  862. p->status = USB_RET_ADD_TO_QUEUE;
  863. break;
  864. }
  865. usbredir_handle_bulk_data(dev, p, ep);
  866. break;
  867. case USB_ENDPOINT_XFER_ISOC:
  868. usbredir_handle_iso_data(dev, p, ep);
  869. break;
  870. case USB_ENDPOINT_XFER_INT:
  871. if (ep & USB_DIR_IN) {
  872. usbredir_handle_interrupt_in_data(dev, p, ep);
  873. } else {
  874. usbredir_handle_interrupt_out_data(dev, p, ep);
  875. }
  876. break;
  877. default:
  878. ERROR("handle_data ep %02X has unknown type %d\n", ep,
  879. dev->endpoint[EP2I(ep)].type);
  880. p->status = USB_RET_NAK;
  881. }
  882. }
  883. static void usbredir_flush_ep_queue(USBDevice *dev, USBEndpoint *ep)
  884. {
  885. if (ep->pid == USB_TOKEN_IN && ep->pipeline) {
  886. usb_ep_combine_input_packets(ep);
  887. }
  888. }
  889. static void usbredir_stop_ep(USBRedirDevice *dev, int i)
  890. {
  891. uint8_t ep = I2EP(i);
  892. switch (dev->endpoint[i].type) {
  893. case USB_ENDPOINT_XFER_BULK:
  894. if (ep & USB_DIR_IN) {
  895. usbredir_stop_bulk_receiving(dev, ep);
  896. }
  897. break;
  898. case USB_ENDPOINT_XFER_ISOC:
  899. usbredir_stop_iso_stream(dev, ep);
  900. break;
  901. case USB_ENDPOINT_XFER_INT:
  902. if (ep & USB_DIR_IN) {
  903. usbredir_stop_interrupt_receiving(dev, ep);
  904. }
  905. break;
  906. }
  907. usbredir_free_bufpq(dev, ep);
  908. }
  909. static void usbredir_ep_stopped(USBDevice *udev, USBEndpoint *uep)
  910. {
  911. USBRedirDevice *dev = USB_REDIRECT(udev);
  912. usbredir_stop_ep(dev, USBEP2I(uep));
  913. usbredirparser_do_write(dev->parser);
  914. }
  915. static void usbredir_set_config(USBRedirDevice *dev, USBPacket *p,
  916. int config)
  917. {
  918. struct usb_redir_set_configuration_header set_config;
  919. int i;
  920. DPRINTF("set config %d id %"PRIu64"\n", config, p->id);
  921. for (i = 0; i < MAX_ENDPOINTS; i++) {
  922. usbredir_stop_ep(dev, i);
  923. }
  924. set_config.configuration = config;
  925. usbredirparser_send_set_configuration(dev->parser, p->id, &set_config);
  926. usbredirparser_do_write(dev->parser);
  927. p->status = USB_RET_ASYNC;
  928. }
  929. static void usbredir_get_config(USBRedirDevice *dev, USBPacket *p)
  930. {
  931. DPRINTF("get config id %"PRIu64"\n", p->id);
  932. usbredirparser_send_get_configuration(dev->parser, p->id);
  933. usbredirparser_do_write(dev->parser);
  934. p->status = USB_RET_ASYNC;
  935. }
  936. static void usbredir_set_interface(USBRedirDevice *dev, USBPacket *p,
  937. int interface, int alt)
  938. {
  939. struct usb_redir_set_alt_setting_header set_alt;
  940. int i;
  941. DPRINTF("set interface %d alt %d id %"PRIu64"\n", interface, alt, p->id);
  942. for (i = 0; i < MAX_ENDPOINTS; i++) {
  943. if (dev->endpoint[i].interface == interface) {
  944. usbredir_stop_ep(dev, i);
  945. }
  946. }
  947. set_alt.interface = interface;
  948. set_alt.alt = alt;
  949. usbredirparser_send_set_alt_setting(dev->parser, p->id, &set_alt);
  950. usbredirparser_do_write(dev->parser);
  951. p->status = USB_RET_ASYNC;
  952. }
  953. static void usbredir_get_interface(USBRedirDevice *dev, USBPacket *p,
  954. int interface)
  955. {
  956. struct usb_redir_get_alt_setting_header get_alt;
  957. DPRINTF("get interface %d id %"PRIu64"\n", interface, p->id);
  958. get_alt.interface = interface;
  959. usbredirparser_send_get_alt_setting(dev->parser, p->id, &get_alt);
  960. usbredirparser_do_write(dev->parser);
  961. p->status = USB_RET_ASYNC;
  962. }
  963. static void usbredir_handle_control(USBDevice *udev, USBPacket *p,
  964. int request, int value, int index, int length, uint8_t *data)
  965. {
  966. USBRedirDevice *dev = USB_REDIRECT(udev);
  967. struct usb_redir_control_packet_header control_packet;
  968. if (usbredir_already_in_flight(dev, p->id)) {
  969. p->status = USB_RET_ASYNC;
  970. return;
  971. }
  972. /* Special cases for certain standard device requests */
  973. switch (request) {
  974. case DeviceOutRequest | USB_REQ_SET_ADDRESS:
  975. DPRINTF("set address %d\n", value);
  976. dev->dev.addr = value;
  977. return;
  978. case DeviceOutRequest | USB_REQ_SET_CONFIGURATION:
  979. usbredir_set_config(dev, p, value & 0xff);
  980. return;
  981. case DeviceRequest | USB_REQ_GET_CONFIGURATION:
  982. usbredir_get_config(dev, p);
  983. return;
  984. case InterfaceOutRequest | USB_REQ_SET_INTERFACE:
  985. usbredir_set_interface(dev, p, index, value);
  986. return;
  987. case InterfaceRequest | USB_REQ_GET_INTERFACE:
  988. usbredir_get_interface(dev, p, index);
  989. return;
  990. }
  991. /* Normal ctrl requests, note request is (bRequestType << 8) | bRequest */
  992. DPRINTF(
  993. "ctrl-out type 0x%x req 0x%x val 0x%x index %d len %d id %"PRIu64"\n",
  994. request >> 8, request & 0xff, value, index, length, p->id);
  995. control_packet.request = request & 0xFF;
  996. control_packet.requesttype = request >> 8;
  997. control_packet.endpoint = control_packet.requesttype & USB_DIR_IN;
  998. control_packet.value = value;
  999. control_packet.index = index;
  1000. control_packet.length = length;
  1001. if (control_packet.requesttype & USB_DIR_IN) {
  1002. usbredirparser_send_control_packet(dev->parser, p->id,
  1003. &control_packet, NULL, 0);
  1004. } else {
  1005. usbredir_log_data(dev, "ctrl data out:", data, length);
  1006. usbredirparser_send_control_packet(dev->parser, p->id,
  1007. &control_packet, data, length);
  1008. }
  1009. usbredirparser_do_write(dev->parser);
  1010. p->status = USB_RET_ASYNC;
  1011. }
  1012. static int usbredir_alloc_streams(USBDevice *udev, USBEndpoint **eps,
  1013. int nr_eps, int streams)
  1014. {
  1015. USBRedirDevice *dev = USB_REDIRECT(udev);
  1016. #if USBREDIR_VERSION >= 0x000700
  1017. struct usb_redir_alloc_bulk_streams_header alloc_streams;
  1018. int i;
  1019. if (!usbredirparser_peer_has_cap(dev->parser,
  1020. usb_redir_cap_bulk_streams)) {
  1021. ERROR("peer does not support streams\n");
  1022. goto reject;
  1023. }
  1024. if (streams == 0) {
  1025. ERROR("request to allocate 0 streams\n");
  1026. return -1;
  1027. }
  1028. alloc_streams.no_streams = streams;
  1029. alloc_streams.endpoints = 0;
  1030. for (i = 0; i < nr_eps; i++) {
  1031. alloc_streams.endpoints |= 1 << USBEP2I(eps[i]);
  1032. }
  1033. usbredirparser_send_alloc_bulk_streams(dev->parser, 0, &alloc_streams);
  1034. usbredirparser_do_write(dev->parser);
  1035. return 0;
  1036. #else
  1037. ERROR("usbredir_alloc_streams not implemented\n");
  1038. goto reject;
  1039. #endif
  1040. reject:
  1041. ERROR("streams are not available, disconnecting\n");
  1042. qemu_bh_schedule(dev->device_reject_bh);
  1043. return -1;
  1044. }
  1045. static void usbredir_free_streams(USBDevice *udev, USBEndpoint **eps,
  1046. int nr_eps)
  1047. {
  1048. #if USBREDIR_VERSION >= 0x000700
  1049. USBRedirDevice *dev = USB_REDIRECT(udev);
  1050. struct usb_redir_free_bulk_streams_header free_streams;
  1051. int i;
  1052. if (!usbredirparser_peer_has_cap(dev->parser,
  1053. usb_redir_cap_bulk_streams)) {
  1054. return;
  1055. }
  1056. free_streams.endpoints = 0;
  1057. for (i = 0; i < nr_eps; i++) {
  1058. free_streams.endpoints |= 1 << USBEP2I(eps[i]);
  1059. }
  1060. usbredirparser_send_free_bulk_streams(dev->parser, 0, &free_streams);
  1061. usbredirparser_do_write(dev->parser);
  1062. #endif
  1063. }
  1064. /*
  1065. * Close events can be triggered by usbredirparser_do_write which gets called
  1066. * from within the USBDevice data / control packet callbacks and doing a
  1067. * usb_detach from within these callbacks is not a good idea.
  1068. *
  1069. * So we use a bh handler to take care of close events.
  1070. */
  1071. static void usbredir_chardev_close_bh(void *opaque)
  1072. {
  1073. USBRedirDevice *dev = opaque;
  1074. qemu_bh_cancel(dev->device_reject_bh);
  1075. usbredir_device_disconnect(dev);
  1076. if (dev->parser) {
  1077. DPRINTF("destroying usbredirparser\n");
  1078. usbredirparser_destroy(dev->parser);
  1079. dev->parser = NULL;
  1080. }
  1081. if (dev->watch) {
  1082. g_source_remove(dev->watch);
  1083. dev->watch = 0;
  1084. }
  1085. }
  1086. static void usbredir_create_parser(USBRedirDevice *dev)
  1087. {
  1088. uint32_t caps[USB_REDIR_CAPS_SIZE] = { 0, };
  1089. int flags = 0;
  1090. DPRINTF("creating usbredirparser\n");
  1091. dev->parser = usbredirparser_create();
  1092. if (!dev->parser) {
  1093. error_report("usbredirparser_create() failed");
  1094. exit(1);
  1095. }
  1096. dev->parser->priv = dev;
  1097. dev->parser->log_func = usbredir_log;
  1098. dev->parser->read_func = usbredir_read;
  1099. dev->parser->write_func = usbredir_write;
  1100. dev->parser->hello_func = usbredir_hello;
  1101. dev->parser->device_connect_func = usbredir_device_connect;
  1102. dev->parser->device_disconnect_func = usbredir_device_disconnect;
  1103. dev->parser->interface_info_func = usbredir_interface_info;
  1104. dev->parser->ep_info_func = usbredir_ep_info;
  1105. dev->parser->configuration_status_func = usbredir_configuration_status;
  1106. dev->parser->alt_setting_status_func = usbredir_alt_setting_status;
  1107. dev->parser->iso_stream_status_func = usbredir_iso_stream_status;
  1108. dev->parser->interrupt_receiving_status_func =
  1109. usbredir_interrupt_receiving_status;
  1110. dev->parser->bulk_streams_status_func = usbredir_bulk_streams_status;
  1111. dev->parser->bulk_receiving_status_func = usbredir_bulk_receiving_status;
  1112. dev->parser->control_packet_func = usbredir_control_packet;
  1113. dev->parser->bulk_packet_func = usbredir_bulk_packet;
  1114. dev->parser->iso_packet_func = usbredir_iso_packet;
  1115. dev->parser->interrupt_packet_func = usbredir_interrupt_packet;
  1116. dev->parser->buffered_bulk_packet_func = usbredir_buffered_bulk_packet;
  1117. dev->read_buf = NULL;
  1118. dev->read_buf_size = 0;
  1119. usbredirparser_caps_set_cap(caps, usb_redir_cap_connect_device_version);
  1120. usbredirparser_caps_set_cap(caps, usb_redir_cap_filter);
  1121. usbredirparser_caps_set_cap(caps, usb_redir_cap_ep_info_max_packet_size);
  1122. usbredirparser_caps_set_cap(caps, usb_redir_cap_64bits_ids);
  1123. usbredirparser_caps_set_cap(caps, usb_redir_cap_32bits_bulk_length);
  1124. usbredirparser_caps_set_cap(caps, usb_redir_cap_bulk_receiving);
  1125. #if USBREDIR_VERSION >= 0x000700
  1126. if (dev->enable_streams) {
  1127. usbredirparser_caps_set_cap(caps, usb_redir_cap_bulk_streams);
  1128. }
  1129. #endif
  1130. if (runstate_check(RUN_STATE_INMIGRATE) ||
  1131. runstate_check(RUN_STATE_RESTORE_VM)) {
  1132. flags |= usbredirparser_fl_no_hello;
  1133. }
  1134. usbredirparser_init(dev->parser, VERSION, caps, USB_REDIR_CAPS_SIZE,
  1135. flags);
  1136. usbredirparser_do_write(dev->parser);
  1137. }
  1138. static void usbredir_reject_device(USBRedirDevice *dev)
  1139. {
  1140. usbredir_device_disconnect(dev);
  1141. if (usbredirparser_peer_has_cap(dev->parser, usb_redir_cap_filter)) {
  1142. usbredirparser_send_filter_reject(dev->parser);
  1143. usbredirparser_do_write(dev->parser);
  1144. }
  1145. }
  1146. /*
  1147. * We may need to reject the device when the hcd calls alloc_streams, doing
  1148. * an usb_detach from within a hcd call is not a good idea, hence this bh.
  1149. */
  1150. static void usbredir_device_reject_bh(void *opaque)
  1151. {
  1152. USBRedirDevice *dev = opaque;
  1153. usbredir_reject_device(dev);
  1154. }
  1155. static void usbredir_do_attach(void *opaque)
  1156. {
  1157. USBRedirDevice *dev = opaque;
  1158. Error *local_err = NULL;
  1159. /* In order to work properly with XHCI controllers we need these caps */
  1160. if ((dev->dev.port->speedmask & USB_SPEED_MASK_SUPER) && !(
  1161. usbredirparser_peer_has_cap(dev->parser,
  1162. usb_redir_cap_ep_info_max_packet_size) &&
  1163. usbredirparser_peer_has_cap(dev->parser,
  1164. usb_redir_cap_32bits_bulk_length) &&
  1165. usbredirparser_peer_has_cap(dev->parser,
  1166. usb_redir_cap_64bits_ids))) {
  1167. ERROR("usb-redir-host lacks capabilities needed for use with XHCI\n");
  1168. usbredir_reject_device(dev);
  1169. return;
  1170. }
  1171. usb_device_attach(&dev->dev, &local_err);
  1172. if (local_err) {
  1173. error_report_err(local_err);
  1174. WARNING("rejecting device due to speed mismatch\n");
  1175. usbredir_reject_device(dev);
  1176. }
  1177. }
  1178. /*
  1179. * chardev callbacks
  1180. */
  1181. static int usbredir_chardev_can_read(void *opaque)
  1182. {
  1183. USBRedirDevice *dev = opaque;
  1184. if (!dev->parser) {
  1185. WARNING("chardev_can_read called on non open chardev!\n");
  1186. return 0;
  1187. }
  1188. /* Don't read new data from the chardev until our state is fully synced */
  1189. if (!runstate_check(RUN_STATE_RUNNING)) {
  1190. return 0;
  1191. }
  1192. /* usbredir_parser_do_read will consume *all* data we give it */
  1193. return 1 * MiB;
  1194. }
  1195. static void usbredir_chardev_read(void *opaque, const uint8_t *buf, int size)
  1196. {
  1197. USBRedirDevice *dev = opaque;
  1198. /* No recursion allowed! */
  1199. assert(dev->read_buf == NULL);
  1200. dev->read_buf = buf;
  1201. dev->read_buf_size = size;
  1202. usbredirparser_do_read(dev->parser);
  1203. /* Send any acks, etc. which may be queued now */
  1204. usbredirparser_do_write(dev->parser);
  1205. }
  1206. static void usbredir_chardev_event(void *opaque, QEMUChrEvent event)
  1207. {
  1208. USBRedirDevice *dev = opaque;
  1209. switch (event) {
  1210. case CHR_EVENT_OPENED:
  1211. DPRINTF("chardev open\n");
  1212. /* Make sure any pending closes are handled (no-op if none pending) */
  1213. usbredir_chardev_close_bh(dev);
  1214. qemu_bh_cancel(dev->chardev_close_bh);
  1215. usbredir_create_parser(dev);
  1216. break;
  1217. case CHR_EVENT_CLOSED:
  1218. DPRINTF("chardev close\n");
  1219. qemu_bh_schedule(dev->chardev_close_bh);
  1220. break;
  1221. case CHR_EVENT_BREAK:
  1222. case CHR_EVENT_MUX_IN:
  1223. case CHR_EVENT_MUX_OUT:
  1224. /* Ignore */
  1225. break;
  1226. }
  1227. }
  1228. /*
  1229. * init + destroy
  1230. */
  1231. static void usbredir_vm_state_change(void *priv, bool running, RunState state)
  1232. {
  1233. USBRedirDevice *dev = priv;
  1234. if (state == RUN_STATE_RUNNING && dev->parser != NULL) {
  1235. usbredirparser_do_write(dev->parser); /* Flush any pending writes */
  1236. }
  1237. }
  1238. static void usbredir_init_endpoints(USBRedirDevice *dev)
  1239. {
  1240. int i;
  1241. usb_ep_init(&dev->dev);
  1242. memset(dev->endpoint, 0, sizeof(dev->endpoint));
  1243. for (i = 0; i < MAX_ENDPOINTS; i++) {
  1244. dev->endpoint[i].dev = dev;
  1245. QTAILQ_INIT(&dev->endpoint[i].bufpq);
  1246. }
  1247. }
  1248. static void usbredir_realize(USBDevice *udev, Error **errp)
  1249. {
  1250. USBRedirDevice *dev = USB_REDIRECT(udev);
  1251. int i;
  1252. if (!qemu_chr_fe_backend_connected(&dev->cs)) {
  1253. error_setg(errp, QERR_MISSING_PARAMETER, "chardev");
  1254. return;
  1255. }
  1256. if (dev->filter_str) {
  1257. i = usbredirfilter_string_to_rules(dev->filter_str, ":", "|",
  1258. &dev->filter_rules,
  1259. &dev->filter_rules_count);
  1260. if (i) {
  1261. error_setg(errp, QERR_INVALID_PARAMETER_VALUE, "filter",
  1262. "a usb device filter string");
  1263. return;
  1264. }
  1265. }
  1266. dev->chardev_close_bh = qemu_bh_new(usbredir_chardev_close_bh, dev);
  1267. dev->device_reject_bh = qemu_bh_new(usbredir_device_reject_bh, dev);
  1268. dev->attach_timer = timer_new_ms(QEMU_CLOCK_VIRTUAL, usbredir_do_attach, dev);
  1269. packet_id_queue_init(&dev->cancelled, dev, "cancelled");
  1270. packet_id_queue_init(&dev->already_in_flight, dev, "already-in-flight");
  1271. usbredir_init_endpoints(dev);
  1272. /* We'll do the attach once we receive the speed from the usb-host */
  1273. udev->auto_attach = 0;
  1274. /* Will be cleared during setup when we find conflicts */
  1275. dev->compatible_speedmask = USB_SPEED_MASK_FULL | USB_SPEED_MASK_HIGH;
  1276. /* Let the backend know we are ready */
  1277. qemu_chr_fe_set_handlers(&dev->cs, usbredir_chardev_can_read,
  1278. usbredir_chardev_read, usbredir_chardev_event,
  1279. NULL, dev, NULL, true);
  1280. dev->vmstate =
  1281. qemu_add_vm_change_state_handler(usbredir_vm_state_change, dev);
  1282. }
  1283. static void usbredir_cleanup_device_queues(USBRedirDevice *dev)
  1284. {
  1285. int i;
  1286. packet_id_queue_empty(&dev->cancelled);
  1287. packet_id_queue_empty(&dev->already_in_flight);
  1288. for (i = 0; i < MAX_ENDPOINTS; i++) {
  1289. usbredir_free_bufpq(dev, I2EP(i));
  1290. }
  1291. }
  1292. static void usbredir_unrealize(USBDevice *udev)
  1293. {
  1294. USBRedirDevice *dev = USB_REDIRECT(udev);
  1295. qemu_chr_fe_deinit(&dev->cs, true);
  1296. /* Note must be done after qemu_chr_close, as that causes a close event */
  1297. qemu_bh_delete(dev->chardev_close_bh);
  1298. qemu_bh_delete(dev->device_reject_bh);
  1299. timer_free(dev->attach_timer);
  1300. usbredir_cleanup_device_queues(dev);
  1301. if (dev->parser) {
  1302. usbredirparser_destroy(dev->parser);
  1303. }
  1304. if (dev->watch) {
  1305. g_source_remove(dev->watch);
  1306. }
  1307. free(dev->filter_rules);
  1308. qemu_del_vm_change_state_handler(dev->vmstate);
  1309. }
  1310. static int usbredir_check_filter(USBRedirDevice *dev)
  1311. {
  1312. if (dev->interface_info.interface_count == NO_INTERFACE_INFO) {
  1313. ERROR("No interface info for device\n");
  1314. goto error;
  1315. }
  1316. if (dev->filter_rules) {
  1317. if (!usbredirparser_peer_has_cap(dev->parser,
  1318. usb_redir_cap_connect_device_version)) {
  1319. ERROR("Device filter specified and peer does not have the "
  1320. "connect_device_version capability\n");
  1321. goto error;
  1322. }
  1323. if (usbredirfilter_check(
  1324. dev->filter_rules,
  1325. dev->filter_rules_count,
  1326. dev->device_info.device_class,
  1327. dev->device_info.device_subclass,
  1328. dev->device_info.device_protocol,
  1329. dev->interface_info.interface_class,
  1330. dev->interface_info.interface_subclass,
  1331. dev->interface_info.interface_protocol,
  1332. dev->interface_info.interface_count,
  1333. dev->device_info.vendor_id,
  1334. dev->device_info.product_id,
  1335. dev->device_info.device_version_bcd,
  1336. 0) != 0) {
  1337. goto error;
  1338. }
  1339. }
  1340. return 0;
  1341. error:
  1342. usbredir_reject_device(dev);
  1343. return -1;
  1344. }
  1345. static void usbredir_check_bulk_receiving(USBRedirDevice *dev)
  1346. {
  1347. int i, j, quirks;
  1348. if (!usbredirparser_peer_has_cap(dev->parser,
  1349. usb_redir_cap_bulk_receiving)) {
  1350. return;
  1351. }
  1352. for (i = EP2I(USB_DIR_IN); i < MAX_ENDPOINTS; i++) {
  1353. dev->endpoint[i].bulk_receiving_enabled = 0;
  1354. }
  1355. if (dev->interface_info.interface_count == NO_INTERFACE_INFO) {
  1356. return;
  1357. }
  1358. for (i = 0; i < dev->interface_info.interface_count; i++) {
  1359. quirks = usb_get_quirks(dev->device_info.vendor_id,
  1360. dev->device_info.product_id,
  1361. dev->interface_info.interface_class[i],
  1362. dev->interface_info.interface_subclass[i],
  1363. dev->interface_info.interface_protocol[i]);
  1364. if (!(quirks & USB_QUIRK_BUFFER_BULK_IN)) {
  1365. continue;
  1366. }
  1367. if (quirks & USB_QUIRK_IS_FTDI) {
  1368. dev->buffered_bulk_in_complete =
  1369. usbredir_buffered_bulk_in_complete_ftdi;
  1370. } else {
  1371. dev->buffered_bulk_in_complete =
  1372. usbredir_buffered_bulk_in_complete_raw;
  1373. }
  1374. for (j = EP2I(USB_DIR_IN); j < MAX_ENDPOINTS; j++) {
  1375. if (dev->endpoint[j].interface ==
  1376. dev->interface_info.interface[i] &&
  1377. dev->endpoint[j].type == USB_ENDPOINT_XFER_BULK &&
  1378. dev->endpoint[j].max_packet_size != 0) {
  1379. dev->endpoint[j].bulk_receiving_enabled = 1;
  1380. /*
  1381. * With buffering pipelining is not necessary. Also packet
  1382. * combining and bulk in buffering don't play nice together!
  1383. */
  1384. I2USBEP(dev, j)->pipeline = false;
  1385. break; /* Only buffer for the first ep of each intf */
  1386. }
  1387. }
  1388. }
  1389. }
  1390. /*
  1391. * usbredirparser packet complete callbacks
  1392. */
  1393. static void usbredir_handle_status(USBRedirDevice *dev, USBPacket *p,
  1394. int status)
  1395. {
  1396. switch (status) {
  1397. case usb_redir_success:
  1398. p->status = USB_RET_SUCCESS; /* Clear previous ASYNC status */
  1399. break;
  1400. case usb_redir_stall:
  1401. p->status = USB_RET_STALL;
  1402. break;
  1403. case usb_redir_cancelled:
  1404. /*
  1405. * When the usbredir-host unredirects a device, it will report a status
  1406. * of cancelled for all pending packets, followed by a disconnect msg.
  1407. */
  1408. p->status = USB_RET_IOERROR;
  1409. break;
  1410. case usb_redir_inval:
  1411. WARNING("got invalid param error from usb-host?\n");
  1412. p->status = USB_RET_IOERROR;
  1413. break;
  1414. case usb_redir_babble:
  1415. p->status = USB_RET_BABBLE;
  1416. break;
  1417. case usb_redir_ioerror:
  1418. case usb_redir_timeout:
  1419. default:
  1420. p->status = USB_RET_IOERROR;
  1421. }
  1422. }
  1423. static void usbredir_hello(void *priv, struct usb_redir_hello_header *h)
  1424. {
  1425. USBRedirDevice *dev = priv;
  1426. /* Try to send the filter info now that we've the usb-host's caps */
  1427. if (usbredirparser_peer_has_cap(dev->parser, usb_redir_cap_filter) &&
  1428. dev->filter_rules) {
  1429. usbredirparser_send_filter_filter(dev->parser, dev->filter_rules,
  1430. dev->filter_rules_count);
  1431. usbredirparser_do_write(dev->parser);
  1432. }
  1433. }
  1434. static void usbredir_device_connect(void *priv,
  1435. struct usb_redir_device_connect_header *device_connect)
  1436. {
  1437. USBRedirDevice *dev = priv;
  1438. const char *speed;
  1439. if (timer_pending(dev->attach_timer) || dev->dev.attached) {
  1440. ERROR("Received device connect while already connected\n");
  1441. return;
  1442. }
  1443. switch (device_connect->speed) {
  1444. case usb_redir_speed_low:
  1445. speed = "low speed";
  1446. dev->dev.speed = USB_SPEED_LOW;
  1447. dev->compatible_speedmask &= ~USB_SPEED_MASK_FULL;
  1448. dev->compatible_speedmask &= ~USB_SPEED_MASK_HIGH;
  1449. break;
  1450. case usb_redir_speed_full:
  1451. speed = "full speed";
  1452. dev->dev.speed = USB_SPEED_FULL;
  1453. dev->compatible_speedmask &= ~USB_SPEED_MASK_HIGH;
  1454. break;
  1455. case usb_redir_speed_high:
  1456. speed = "high speed";
  1457. dev->dev.speed = USB_SPEED_HIGH;
  1458. break;
  1459. case usb_redir_speed_super:
  1460. speed = "super speed";
  1461. dev->dev.speed = USB_SPEED_SUPER;
  1462. break;
  1463. default:
  1464. speed = "unknown speed";
  1465. dev->dev.speed = USB_SPEED_FULL;
  1466. }
  1467. if (usbredirparser_peer_has_cap(dev->parser,
  1468. usb_redir_cap_connect_device_version)) {
  1469. INFO("attaching %s device %04x:%04x version %d.%d class %02x\n",
  1470. speed, device_connect->vendor_id, device_connect->product_id,
  1471. ((device_connect->device_version_bcd & 0xf000) >> 12) * 10 +
  1472. ((device_connect->device_version_bcd & 0x0f00) >> 8),
  1473. ((device_connect->device_version_bcd & 0x00f0) >> 4) * 10 +
  1474. ((device_connect->device_version_bcd & 0x000f) >> 0),
  1475. device_connect->device_class);
  1476. } else {
  1477. INFO("attaching %s device %04x:%04x class %02x\n", speed,
  1478. device_connect->vendor_id, device_connect->product_id,
  1479. device_connect->device_class);
  1480. }
  1481. dev->dev.speedmask = (1 << dev->dev.speed) | dev->compatible_speedmask;
  1482. dev->device_info = *device_connect;
  1483. if (usbredir_check_filter(dev)) {
  1484. WARNING("Device %04x:%04x rejected by device filter, not attaching\n",
  1485. device_connect->vendor_id, device_connect->product_id);
  1486. return;
  1487. }
  1488. usbredir_check_bulk_receiving(dev);
  1489. timer_mod(dev->attach_timer, dev->next_attach_time);
  1490. }
  1491. static void usbredir_device_disconnect(void *priv)
  1492. {
  1493. USBRedirDevice *dev = priv;
  1494. /* Stop any pending attaches */
  1495. timer_del(dev->attach_timer);
  1496. if (dev->dev.attached) {
  1497. DPRINTF("detaching device\n");
  1498. usb_device_detach(&dev->dev);
  1499. /*
  1500. * Delay next usb device attach to give the guest a chance to see
  1501. * see the detach / attach in case of quick close / open succession
  1502. */
  1503. dev->next_attach_time = qemu_clock_get_ms(QEMU_CLOCK_VIRTUAL) + 200;
  1504. }
  1505. /* Reset state so that the next dev connected starts with a clean slate */
  1506. usbredir_cleanup_device_queues(dev);
  1507. usbredir_init_endpoints(dev);
  1508. dev->interface_info.interface_count = NO_INTERFACE_INFO;
  1509. dev->dev.addr = 0;
  1510. dev->dev.speed = 0;
  1511. dev->compatible_speedmask = USB_SPEED_MASK_FULL | USB_SPEED_MASK_HIGH;
  1512. }
  1513. static void usbredir_interface_info(void *priv,
  1514. struct usb_redir_interface_info_header *interface_info)
  1515. {
  1516. USBRedirDevice *dev = priv;
  1517. dev->interface_info = *interface_info;
  1518. /*
  1519. * If we receive interface info after the device has already been
  1520. * connected (ie on a set_config), re-check interface dependent things.
  1521. */
  1522. if (timer_pending(dev->attach_timer) || dev->dev.attached) {
  1523. usbredir_check_bulk_receiving(dev);
  1524. if (usbredir_check_filter(dev)) {
  1525. ERROR("Device no longer matches filter after interface info "
  1526. "change, disconnecting!\n");
  1527. }
  1528. }
  1529. }
  1530. static void usbredir_mark_speed_incompatible(USBRedirDevice *dev, int speed)
  1531. {
  1532. dev->compatible_speedmask &= ~(1 << speed);
  1533. dev->dev.speedmask = (1 << dev->dev.speed) | dev->compatible_speedmask;
  1534. }
  1535. static void usbredir_set_pipeline(USBRedirDevice *dev, struct USBEndpoint *uep)
  1536. {
  1537. if (uep->type != USB_ENDPOINT_XFER_BULK) {
  1538. return;
  1539. }
  1540. if (uep->pid == USB_TOKEN_OUT) {
  1541. uep->pipeline = true;
  1542. }
  1543. if (uep->pid == USB_TOKEN_IN && uep->max_packet_size != 0 &&
  1544. usbredirparser_peer_has_cap(dev->parser,
  1545. usb_redir_cap_32bits_bulk_length)) {
  1546. uep->pipeline = true;
  1547. }
  1548. }
  1549. static void usbredir_setup_usb_eps(USBRedirDevice *dev)
  1550. {
  1551. struct USBEndpoint *usb_ep;
  1552. int i;
  1553. for (i = 0; i < MAX_ENDPOINTS; i++) {
  1554. usb_ep = I2USBEP(dev, i);
  1555. usb_ep->type = dev->endpoint[i].type;
  1556. usb_ep->ifnum = dev->endpoint[i].interface;
  1557. usb_ep->max_packet_size = dev->endpoint[i].max_packet_size;
  1558. usb_ep->max_streams = dev->endpoint[i].max_streams;
  1559. usbredir_set_pipeline(dev, usb_ep);
  1560. }
  1561. }
  1562. static void usbredir_ep_info(void *priv,
  1563. struct usb_redir_ep_info_header *ep_info)
  1564. {
  1565. USBRedirDevice *dev = priv;
  1566. int i;
  1567. assert(dev != NULL);
  1568. for (i = 0; i < MAX_ENDPOINTS; i++) {
  1569. dev->endpoint[i].type = ep_info->type[i];
  1570. dev->endpoint[i].interval = ep_info->interval[i];
  1571. dev->endpoint[i].interface = ep_info->interface[i];
  1572. if (usbredirparser_peer_has_cap(dev->parser,
  1573. usb_redir_cap_ep_info_max_packet_size)) {
  1574. dev->endpoint[i].max_packet_size = ep_info->max_packet_size[i];
  1575. }
  1576. #if USBREDIR_VERSION >= 0x000700
  1577. if (usbredirparser_peer_has_cap(dev->parser,
  1578. usb_redir_cap_bulk_streams)) {
  1579. dev->endpoint[i].max_streams = ep_info->max_streams[i];
  1580. }
  1581. #endif
  1582. switch (dev->endpoint[i].type) {
  1583. case usb_redir_type_invalid:
  1584. break;
  1585. case usb_redir_type_iso:
  1586. usbredir_mark_speed_incompatible(dev, USB_SPEED_FULL);
  1587. usbredir_mark_speed_incompatible(dev, USB_SPEED_HIGH);
  1588. /* Fall through */
  1589. case usb_redir_type_interrupt:
  1590. if (!usbredirparser_peer_has_cap(dev->parser,
  1591. usb_redir_cap_ep_info_max_packet_size) ||
  1592. ep_info->max_packet_size[i] > 64) {
  1593. usbredir_mark_speed_incompatible(dev, USB_SPEED_FULL);
  1594. }
  1595. if (!usbredirparser_peer_has_cap(dev->parser,
  1596. usb_redir_cap_ep_info_max_packet_size) ||
  1597. ep_info->max_packet_size[i] > 1024) {
  1598. usbredir_mark_speed_incompatible(dev, USB_SPEED_HIGH);
  1599. }
  1600. if (dev->endpoint[i].interval == 0) {
  1601. ERROR("Received 0 interval for isoc or irq endpoint\n");
  1602. usbredir_reject_device(dev);
  1603. return;
  1604. }
  1605. /* Fall through */
  1606. case usb_redir_type_control:
  1607. case usb_redir_type_bulk:
  1608. DPRINTF("ep: %02X type: %d interface: %d\n", I2EP(i),
  1609. dev->endpoint[i].type, dev->endpoint[i].interface);
  1610. break;
  1611. default:
  1612. ERROR("Received invalid endpoint type\n");
  1613. usbredir_reject_device(dev);
  1614. return;
  1615. }
  1616. }
  1617. /* The new ep info may have caused a speed incompatibility, recheck */
  1618. if (dev->dev.attached &&
  1619. !(dev->dev.port->speedmask & dev->dev.speedmask)) {
  1620. ERROR("Device no longer matches speed after endpoint info change, "
  1621. "disconnecting!\n");
  1622. usbredir_reject_device(dev);
  1623. return;
  1624. }
  1625. usbredir_setup_usb_eps(dev);
  1626. usbredir_check_bulk_receiving(dev);
  1627. }
  1628. static void usbredir_configuration_status(void *priv, uint64_t id,
  1629. struct usb_redir_configuration_status_header *config_status)
  1630. {
  1631. USBRedirDevice *dev = priv;
  1632. USBPacket *p;
  1633. DPRINTF("set config status %d config %d id %"PRIu64"\n",
  1634. config_status->status, config_status->configuration, id);
  1635. p = usbredir_find_packet_by_id(dev, 0, id);
  1636. if (p) {
  1637. if (dev->dev.setup_buf[0] & USB_DIR_IN) {
  1638. dev->dev.data_buf[0] = config_status->configuration;
  1639. p->actual_length = 1;
  1640. }
  1641. usbredir_handle_status(dev, p, config_status->status);
  1642. usb_generic_async_ctrl_complete(&dev->dev, p);
  1643. }
  1644. }
  1645. static void usbredir_alt_setting_status(void *priv, uint64_t id,
  1646. struct usb_redir_alt_setting_status_header *alt_setting_status)
  1647. {
  1648. USBRedirDevice *dev = priv;
  1649. USBPacket *p;
  1650. DPRINTF("alt status %d intf %d alt %d id: %"PRIu64"\n",
  1651. alt_setting_status->status, alt_setting_status->interface,
  1652. alt_setting_status->alt, id);
  1653. p = usbredir_find_packet_by_id(dev, 0, id);
  1654. if (p) {
  1655. if (dev->dev.setup_buf[0] & USB_DIR_IN) {
  1656. dev->dev.data_buf[0] = alt_setting_status->alt;
  1657. p->actual_length = 1;
  1658. }
  1659. usbredir_handle_status(dev, p, alt_setting_status->status);
  1660. usb_generic_async_ctrl_complete(&dev->dev, p);
  1661. }
  1662. }
  1663. static void usbredir_iso_stream_status(void *priv, uint64_t id,
  1664. struct usb_redir_iso_stream_status_header *iso_stream_status)
  1665. {
  1666. USBRedirDevice *dev = priv;
  1667. uint8_t ep = iso_stream_status->endpoint;
  1668. DPRINTF("iso status %d ep %02X id %"PRIu64"\n", iso_stream_status->status,
  1669. ep, id);
  1670. if (!dev->dev.attached || !dev->endpoint[EP2I(ep)].iso_started) {
  1671. return;
  1672. }
  1673. dev->endpoint[EP2I(ep)].iso_error = iso_stream_status->status;
  1674. if (iso_stream_status->status == usb_redir_stall) {
  1675. DPRINTF("iso stream stopped by peer ep %02X\n", ep);
  1676. dev->endpoint[EP2I(ep)].iso_started = 0;
  1677. }
  1678. }
  1679. static void usbredir_interrupt_receiving_status(void *priv, uint64_t id,
  1680. struct usb_redir_interrupt_receiving_status_header
  1681. *interrupt_receiving_status)
  1682. {
  1683. USBRedirDevice *dev = priv;
  1684. uint8_t ep = interrupt_receiving_status->endpoint;
  1685. DPRINTF("interrupt recv status %d ep %02X id %"PRIu64"\n",
  1686. interrupt_receiving_status->status, ep, id);
  1687. if (!dev->dev.attached || !dev->endpoint[EP2I(ep)].interrupt_started) {
  1688. return;
  1689. }
  1690. dev->endpoint[EP2I(ep)].interrupt_error =
  1691. interrupt_receiving_status->status;
  1692. if (interrupt_receiving_status->status == usb_redir_stall) {
  1693. DPRINTF("interrupt receiving stopped by peer ep %02X\n", ep);
  1694. dev->endpoint[EP2I(ep)].interrupt_started = 0;
  1695. }
  1696. }
  1697. static void usbredir_bulk_streams_status(void *priv, uint64_t id,
  1698. struct usb_redir_bulk_streams_status_header *bulk_streams_status)
  1699. {
  1700. #if USBREDIR_VERSION >= 0x000700
  1701. USBRedirDevice *dev = priv;
  1702. if (bulk_streams_status->status == usb_redir_success) {
  1703. DPRINTF("bulk streams status %d eps %08x\n",
  1704. bulk_streams_status->status, bulk_streams_status->endpoints);
  1705. } else {
  1706. ERROR("bulk streams %s failed status %d eps %08x\n",
  1707. (bulk_streams_status->no_streams == 0) ? "free" : "alloc",
  1708. bulk_streams_status->status, bulk_streams_status->endpoints);
  1709. ERROR("usb-redir-host does not provide streams, disconnecting\n");
  1710. usbredir_reject_device(dev);
  1711. }
  1712. #endif
  1713. }
  1714. static void usbredir_bulk_receiving_status(void *priv, uint64_t id,
  1715. struct usb_redir_bulk_receiving_status_header *bulk_receiving_status)
  1716. {
  1717. USBRedirDevice *dev = priv;
  1718. uint8_t ep = bulk_receiving_status->endpoint;
  1719. DPRINTF("bulk recv status %d ep %02X id %"PRIu64"\n",
  1720. bulk_receiving_status->status, ep, id);
  1721. if (!dev->dev.attached || !dev->endpoint[EP2I(ep)].bulk_receiving_started) {
  1722. return;
  1723. }
  1724. if (bulk_receiving_status->status == usb_redir_stall) {
  1725. DPRINTF("bulk receiving stopped by peer ep %02X\n", ep);
  1726. dev->endpoint[EP2I(ep)].bulk_receiving_started = 0;
  1727. }
  1728. }
  1729. static void usbredir_control_packet(void *priv, uint64_t id,
  1730. struct usb_redir_control_packet_header *control_packet,
  1731. uint8_t *data, int data_len)
  1732. {
  1733. USBRedirDevice *dev = priv;
  1734. USBPacket *p;
  1735. int len = control_packet->length;
  1736. DPRINTF("ctrl-in status %d len %d id %"PRIu64"\n", control_packet->status,
  1737. len, id);
  1738. /* Fix up USB-3 ep0 maxpacket size to allow superspeed connected devices
  1739. * to work redirected to a not superspeed capable hcd */
  1740. if (dev->dev.speed == USB_SPEED_SUPER &&
  1741. !((dev->dev.port->speedmask & USB_SPEED_MASK_SUPER)) &&
  1742. control_packet->requesttype == 0x80 &&
  1743. control_packet->request == 6 &&
  1744. control_packet->value == 0x100 && control_packet->index == 0 &&
  1745. data_len >= 18 && data[7] == 9) {
  1746. data[7] = 64;
  1747. }
  1748. p = usbredir_find_packet_by_id(dev, 0, id);
  1749. if (p) {
  1750. usbredir_handle_status(dev, p, control_packet->status);
  1751. if (data_len > 0) {
  1752. usbredir_log_data(dev, "ctrl data in:", data, data_len);
  1753. if (data_len > sizeof(dev->dev.data_buf)) {
  1754. ERROR("ctrl buffer too small (%d > %zu)\n",
  1755. data_len, sizeof(dev->dev.data_buf));
  1756. p->status = USB_RET_STALL;
  1757. data_len = len = sizeof(dev->dev.data_buf);
  1758. }
  1759. memcpy(dev->dev.data_buf, data, data_len);
  1760. }
  1761. p->actual_length = len;
  1762. /*
  1763. * If this is GET_DESCRIPTOR request for configuration descriptor,
  1764. * remove 'remote wakeup' flag from it to prevent idle power down
  1765. * in Windows guest
  1766. */
  1767. if (dev->suppress_remote_wake &&
  1768. control_packet->requesttype == USB_DIR_IN &&
  1769. control_packet->request == USB_REQ_GET_DESCRIPTOR &&
  1770. control_packet->value == (USB_DT_CONFIG << 8) &&
  1771. control_packet->index == 0 &&
  1772. /* bmAttributes field of config descriptor */
  1773. len > 7 && (dev->dev.data_buf[7] & USB_CFG_ATT_WAKEUP)) {
  1774. DPRINTF("Removed remote wake %04X:%04X\n",
  1775. dev->device_info.vendor_id,
  1776. dev->device_info.product_id);
  1777. dev->dev.data_buf[7] &= ~USB_CFG_ATT_WAKEUP;
  1778. }
  1779. usb_generic_async_ctrl_complete(&dev->dev, p);
  1780. }
  1781. free(data);
  1782. }
  1783. static void usbredir_bulk_packet(void *priv, uint64_t id,
  1784. struct usb_redir_bulk_packet_header *bulk_packet,
  1785. uint8_t *data, int data_len)
  1786. {
  1787. USBRedirDevice *dev = priv;
  1788. uint8_t ep = bulk_packet->endpoint;
  1789. int len = (bulk_packet->length_high << 16) | bulk_packet->length;
  1790. USBPacket *p;
  1791. DPRINTF("bulk-in status %d ep %02X stream %u len %d id %"PRIu64"\n",
  1792. bulk_packet->status, ep, bulk_packet->stream_id, len, id);
  1793. p = usbredir_find_packet_by_id(dev, ep, id);
  1794. if (p) {
  1795. size_t size = usb_packet_size(p);
  1796. usbredir_handle_status(dev, p, bulk_packet->status);
  1797. if (data_len > 0) {
  1798. usbredir_log_data(dev, "bulk data in:", data, data_len);
  1799. if (data_len > size) {
  1800. ERROR("bulk got more data then requested (%d > %zd)\n",
  1801. data_len, p->iov.size);
  1802. p->status = USB_RET_BABBLE;
  1803. data_len = len = size;
  1804. }
  1805. usb_packet_copy(p, data, data_len);
  1806. }
  1807. p->actual_length = len;
  1808. if (p->pid == USB_TOKEN_IN && p->ep->pipeline) {
  1809. usb_combined_input_packet_complete(&dev->dev, p);
  1810. } else {
  1811. usb_packet_complete(&dev->dev, p);
  1812. }
  1813. }
  1814. free(data);
  1815. }
  1816. static void usbredir_iso_packet(void *priv, uint64_t id,
  1817. struct usb_redir_iso_packet_header *iso_packet,
  1818. uint8_t *data, int data_len)
  1819. {
  1820. USBRedirDevice *dev = priv;
  1821. uint8_t ep = iso_packet->endpoint;
  1822. DPRINTF2("iso-in status %d ep %02X len %d id %"PRIu64"\n",
  1823. iso_packet->status, ep, data_len, id);
  1824. if (dev->endpoint[EP2I(ep)].type != USB_ENDPOINT_XFER_ISOC) {
  1825. ERROR("received iso packet for non iso endpoint %02X\n", ep);
  1826. free(data);
  1827. return;
  1828. }
  1829. if (dev->endpoint[EP2I(ep)].iso_started == 0) {
  1830. DPRINTF("received iso packet for non started stream ep %02X\n", ep);
  1831. free(data);
  1832. return;
  1833. }
  1834. /* bufp_alloc also adds the packet to the ep queue */
  1835. bufp_alloc(dev, data, data_len, iso_packet->status, ep, data);
  1836. }
  1837. static void usbredir_interrupt_packet(void *priv, uint64_t id,
  1838. struct usb_redir_interrupt_packet_header *interrupt_packet,
  1839. uint8_t *data, int data_len)
  1840. {
  1841. USBRedirDevice *dev = priv;
  1842. uint8_t ep = interrupt_packet->endpoint;
  1843. DPRINTF("interrupt-in status %d ep %02X len %d id %"PRIu64"\n",
  1844. interrupt_packet->status, ep, data_len, id);
  1845. if (dev->endpoint[EP2I(ep)].type != USB_ENDPOINT_XFER_INT) {
  1846. ERROR("received int packet for non interrupt endpoint %02X\n", ep);
  1847. free(data);
  1848. return;
  1849. }
  1850. if (ep & USB_DIR_IN) {
  1851. if (dev->endpoint[EP2I(ep)].interrupt_started == 0) {
  1852. DPRINTF("received int packet while not started ep %02X\n", ep);
  1853. free(data);
  1854. return;
  1855. }
  1856. /* bufp_alloc also adds the packet to the ep queue */
  1857. bufp_alloc(dev, data, data_len, interrupt_packet->status, ep, data);
  1858. /* insufficient data solved with USB_RET_NAK */
  1859. usb_wakeup(usb_ep_get(&dev->dev, USB_TOKEN_IN, ep & 0x0f), 0);
  1860. } else {
  1861. /*
  1862. * We report output interrupt packets as completed directly upon
  1863. * submission, so all we can do here if one failed is warn.
  1864. */
  1865. if (interrupt_packet->status) {
  1866. WARNING("interrupt output failed status %d ep %02X id %"PRIu64"\n",
  1867. interrupt_packet->status, ep, id);
  1868. }
  1869. }
  1870. }
  1871. static void usbredir_buffered_bulk_packet(void *priv, uint64_t id,
  1872. struct usb_redir_buffered_bulk_packet_header *buffered_bulk_packet,
  1873. uint8_t *data, int data_len)
  1874. {
  1875. USBRedirDevice *dev = priv;
  1876. uint8_t status, ep = buffered_bulk_packet->endpoint;
  1877. void *free_on_destroy;
  1878. int i, len;
  1879. DPRINTF("buffered-bulk-in status %d ep %02X len %d id %"PRIu64"\n",
  1880. buffered_bulk_packet->status, ep, data_len, id);
  1881. if (dev->endpoint[EP2I(ep)].type != USB_ENDPOINT_XFER_BULK) {
  1882. ERROR("received buffered-bulk packet for non bulk ep %02X\n", ep);
  1883. free(data);
  1884. return;
  1885. }
  1886. if (dev->endpoint[EP2I(ep)].bulk_receiving_started == 0) {
  1887. DPRINTF("received buffered-bulk packet on not started ep %02X\n", ep);
  1888. free(data);
  1889. return;
  1890. }
  1891. /* Data must be in maxp chunks for buffered_bulk_add_*_data_to_packet */
  1892. len = dev->endpoint[EP2I(ep)].max_packet_size;
  1893. status = usb_redir_success;
  1894. free_on_destroy = NULL;
  1895. for (i = 0; i < data_len; i += len) {
  1896. int r;
  1897. if (len >= (data_len - i)) {
  1898. len = data_len - i;
  1899. status = buffered_bulk_packet->status;
  1900. free_on_destroy = data;
  1901. }
  1902. /* bufp_alloc also adds the packet to the ep queue */
  1903. r = bufp_alloc(dev, data + i, len, status, ep, free_on_destroy);
  1904. if (r) {
  1905. break;
  1906. }
  1907. }
  1908. if (dev->endpoint[EP2I(ep)].pending_async_packet) {
  1909. USBPacket *p = dev->endpoint[EP2I(ep)].pending_async_packet;
  1910. dev->endpoint[EP2I(ep)].pending_async_packet = NULL;
  1911. usbredir_buffered_bulk_in_complete(dev, p, ep);
  1912. usb_packet_complete(&dev->dev, p);
  1913. }
  1914. }
  1915. /*
  1916. * Migration code
  1917. */
  1918. static int usbredir_pre_save(void *priv)
  1919. {
  1920. USBRedirDevice *dev = priv;
  1921. usbredir_fill_already_in_flight(dev);
  1922. return 0;
  1923. }
  1924. static int usbredir_post_load(void *priv, int version_id)
  1925. {
  1926. USBRedirDevice *dev = priv;
  1927. if (dev == NULL || dev->parser == NULL) {
  1928. return 0;
  1929. }
  1930. switch (dev->device_info.speed) {
  1931. case usb_redir_speed_low:
  1932. dev->dev.speed = USB_SPEED_LOW;
  1933. break;
  1934. case usb_redir_speed_full:
  1935. dev->dev.speed = USB_SPEED_FULL;
  1936. break;
  1937. case usb_redir_speed_high:
  1938. dev->dev.speed = USB_SPEED_HIGH;
  1939. break;
  1940. case usb_redir_speed_super:
  1941. dev->dev.speed = USB_SPEED_SUPER;
  1942. break;
  1943. default:
  1944. dev->dev.speed = USB_SPEED_FULL;
  1945. }
  1946. dev->dev.speedmask = (1 << dev->dev.speed);
  1947. usbredir_setup_usb_eps(dev);
  1948. usbredir_check_bulk_receiving(dev);
  1949. return 0;
  1950. }
  1951. /* For usbredirparser migration */
  1952. static int usbredir_put_parser(QEMUFile *f, void *priv, size_t unused,
  1953. const VMStateField *field, JSONWriter *vmdesc)
  1954. {
  1955. USBRedirDevice *dev = priv;
  1956. uint8_t *data;
  1957. int len;
  1958. if (dev->parser == NULL) {
  1959. qemu_put_be32(f, 0);
  1960. return 0;
  1961. }
  1962. usbredirparser_serialize(dev->parser, &data, &len);
  1963. if (!data) {
  1964. error_report("usbredirparser_serialize failed");
  1965. exit(1);
  1966. }
  1967. qemu_put_be32(f, len);
  1968. qemu_put_buffer(f, data, len);
  1969. free(data);
  1970. return 0;
  1971. }
  1972. static int usbredir_get_parser(QEMUFile *f, void *priv, size_t unused,
  1973. const VMStateField *field)
  1974. {
  1975. USBRedirDevice *dev = priv;
  1976. uint8_t *data;
  1977. int len, ret;
  1978. len = qemu_get_be32(f);
  1979. if (len == 0) {
  1980. return 0;
  1981. }
  1982. /*
  1983. * If our chardev is not open already at this point the usbredir connection
  1984. * has been broken (non seamless migration, or restore from disk).
  1985. *
  1986. * In this case create a temporary parser to receive the migration data,
  1987. * and schedule the close_bh to report the device as disconnected to the
  1988. * guest and to destroy the parser again.
  1989. */
  1990. if (dev->parser == NULL) {
  1991. WARNING("usb-redir connection broken during migration\n");
  1992. usbredir_create_parser(dev);
  1993. qemu_bh_schedule(dev->chardev_close_bh);
  1994. }
  1995. data = g_malloc(len);
  1996. qemu_get_buffer(f, data, len);
  1997. ret = usbredirparser_unserialize(dev->parser, data, len);
  1998. g_free(data);
  1999. return ret;
  2000. }
  2001. static const VMStateInfo usbredir_parser_vmstate_info = {
  2002. .name = "usb-redir-parser",
  2003. .put = usbredir_put_parser,
  2004. .get = usbredir_get_parser,
  2005. };
  2006. /* For buffered packets (iso/irq) queue migration */
  2007. static int usbredir_put_bufpq(QEMUFile *f, void *priv, size_t unused,
  2008. const VMStateField *field, JSONWriter *vmdesc)
  2009. {
  2010. struct endp_data *endp = priv;
  2011. USBRedirDevice *dev = endp->dev;
  2012. struct buf_packet *bufp;
  2013. int len, i = 0;
  2014. qemu_put_be32(f, endp->bufpq_size);
  2015. QTAILQ_FOREACH(bufp, &endp->bufpq, next) {
  2016. len = bufp->len - bufp->offset;
  2017. DPRINTF("put_bufpq %d/%d len %d status %d\n", i + 1, endp->bufpq_size,
  2018. len, bufp->status);
  2019. qemu_put_be32(f, len);
  2020. qemu_put_be32(f, bufp->status);
  2021. qemu_put_buffer(f, bufp->data + bufp->offset, len);
  2022. i++;
  2023. }
  2024. assert(i == endp->bufpq_size);
  2025. return 0;
  2026. }
  2027. static int usbredir_get_bufpq(QEMUFile *f, void *priv, size_t unused,
  2028. const VMStateField *field)
  2029. {
  2030. struct endp_data *endp = priv;
  2031. USBRedirDevice *dev = endp->dev;
  2032. struct buf_packet *bufp;
  2033. int i;
  2034. endp->bufpq_size = qemu_get_be32(f);
  2035. for (i = 0; i < endp->bufpq_size; i++) {
  2036. bufp = g_new(struct buf_packet, 1);
  2037. bufp->len = qemu_get_be32(f);
  2038. bufp->status = qemu_get_be32(f);
  2039. bufp->offset = 0;
  2040. bufp->data = malloc(bufp->len); /* regular malloc! */
  2041. if (!bufp->data) {
  2042. error_report("usbredir_get_bufpq: out of memory");
  2043. exit(1);
  2044. }
  2045. bufp->free_on_destroy = bufp->data;
  2046. qemu_get_buffer(f, bufp->data, bufp->len);
  2047. QTAILQ_INSERT_TAIL(&endp->bufpq, bufp, next);
  2048. DPRINTF("get_bufpq %d/%d len %d status %d\n", i + 1, endp->bufpq_size,
  2049. bufp->len, bufp->status);
  2050. }
  2051. return 0;
  2052. }
  2053. static const VMStateInfo usbredir_ep_bufpq_vmstate_info = {
  2054. .name = "usb-redir-bufpq",
  2055. .put = usbredir_put_bufpq,
  2056. .get = usbredir_get_bufpq,
  2057. };
  2058. /* For endp_data migration */
  2059. static bool usbredir_bulk_receiving_needed(void *priv)
  2060. {
  2061. struct endp_data *endp = priv;
  2062. return endp->bulk_receiving_started;
  2063. }
  2064. static const VMStateDescription usbredir_bulk_receiving_vmstate = {
  2065. .name = "usb-redir-ep/bulk-receiving",
  2066. .version_id = 1,
  2067. .minimum_version_id = 1,
  2068. .needed = usbredir_bulk_receiving_needed,
  2069. .fields = (VMStateField[]) {
  2070. VMSTATE_UINT8(bulk_receiving_started, struct endp_data),
  2071. VMSTATE_END_OF_LIST()
  2072. }
  2073. };
  2074. static bool usbredir_stream_needed(void *priv)
  2075. {
  2076. struct endp_data *endp = priv;
  2077. return endp->max_streams;
  2078. }
  2079. static const VMStateDescription usbredir_stream_vmstate = {
  2080. .name = "usb-redir-ep/stream-state",
  2081. .version_id = 1,
  2082. .minimum_version_id = 1,
  2083. .needed = usbredir_stream_needed,
  2084. .fields = (VMStateField[]) {
  2085. VMSTATE_UINT32(max_streams, struct endp_data),
  2086. VMSTATE_END_OF_LIST()
  2087. }
  2088. };
  2089. static const VMStateDescription usbredir_ep_vmstate = {
  2090. .name = "usb-redir-ep",
  2091. .version_id = 1,
  2092. .minimum_version_id = 1,
  2093. .fields = (VMStateField[]) {
  2094. VMSTATE_UINT8(type, struct endp_data),
  2095. VMSTATE_UINT8(interval, struct endp_data),
  2096. VMSTATE_UINT8(interface, struct endp_data),
  2097. VMSTATE_UINT16(max_packet_size, struct endp_data),
  2098. VMSTATE_UINT8(iso_started, struct endp_data),
  2099. VMSTATE_UINT8(iso_error, struct endp_data),
  2100. VMSTATE_UINT8(interrupt_started, struct endp_data),
  2101. VMSTATE_UINT8(interrupt_error, struct endp_data),
  2102. VMSTATE_UINT8(bufpq_prefilled, struct endp_data),
  2103. VMSTATE_UINT8(bufpq_dropping_packets, struct endp_data),
  2104. {
  2105. .name = "bufpq",
  2106. .version_id = 0,
  2107. .field_exists = NULL,
  2108. .size = 0,
  2109. .info = &usbredir_ep_bufpq_vmstate_info,
  2110. .flags = VMS_SINGLE,
  2111. .offset = 0,
  2112. },
  2113. VMSTATE_INT32(bufpq_target_size, struct endp_data),
  2114. VMSTATE_END_OF_LIST()
  2115. },
  2116. .subsections = (const VMStateDescription*[]) {
  2117. &usbredir_bulk_receiving_vmstate,
  2118. &usbredir_stream_vmstate,
  2119. NULL
  2120. }
  2121. };
  2122. /* For PacketIdQueue migration */
  2123. static int usbredir_put_packet_id_q(QEMUFile *f, void *priv, size_t unused,
  2124. const VMStateField *field,
  2125. JSONWriter *vmdesc)
  2126. {
  2127. struct PacketIdQueue *q = priv;
  2128. USBRedirDevice *dev = q->dev;
  2129. struct PacketIdQueueEntry *e;
  2130. int remain = q->size;
  2131. DPRINTF("put_packet_id_q %s size %d\n", q->name, q->size);
  2132. qemu_put_be32(f, q->size);
  2133. QTAILQ_FOREACH(e, &q->head, next) {
  2134. qemu_put_be64(f, e->id);
  2135. remain--;
  2136. }
  2137. assert(remain == 0);
  2138. return 0;
  2139. }
  2140. static int usbredir_get_packet_id_q(QEMUFile *f, void *priv, size_t unused,
  2141. const VMStateField *field)
  2142. {
  2143. struct PacketIdQueue *q = priv;
  2144. USBRedirDevice *dev = q->dev;
  2145. int i, size;
  2146. uint64_t id;
  2147. size = qemu_get_be32(f);
  2148. DPRINTF("get_packet_id_q %s size %d\n", q->name, size);
  2149. for (i = 0; i < size; i++) {
  2150. id = qemu_get_be64(f);
  2151. packet_id_queue_add(q, id);
  2152. }
  2153. assert(q->size == size);
  2154. return 0;
  2155. }
  2156. static const VMStateInfo usbredir_ep_packet_id_q_vmstate_info = {
  2157. .name = "usb-redir-packet-id-q",
  2158. .put = usbredir_put_packet_id_q,
  2159. .get = usbredir_get_packet_id_q,
  2160. };
  2161. static const VMStateDescription usbredir_ep_packet_id_queue_vmstate = {
  2162. .name = "usb-redir-packet-id-queue",
  2163. .version_id = 1,
  2164. .minimum_version_id = 1,
  2165. .fields = (VMStateField[]) {
  2166. {
  2167. .name = "queue",
  2168. .version_id = 0,
  2169. .field_exists = NULL,
  2170. .size = 0,
  2171. .info = &usbredir_ep_packet_id_q_vmstate_info,
  2172. .flags = VMS_SINGLE,
  2173. .offset = 0,
  2174. },
  2175. VMSTATE_END_OF_LIST()
  2176. }
  2177. };
  2178. /* For usb_redir_device_connect_header migration */
  2179. static const VMStateDescription usbredir_device_info_vmstate = {
  2180. .name = "usb-redir-device-info",
  2181. .version_id = 1,
  2182. .minimum_version_id = 1,
  2183. .fields = (VMStateField[]) {
  2184. VMSTATE_UINT8(speed, struct usb_redir_device_connect_header),
  2185. VMSTATE_UINT8(device_class, struct usb_redir_device_connect_header),
  2186. VMSTATE_UINT8(device_subclass, struct usb_redir_device_connect_header),
  2187. VMSTATE_UINT8(device_protocol, struct usb_redir_device_connect_header),
  2188. VMSTATE_UINT16(vendor_id, struct usb_redir_device_connect_header),
  2189. VMSTATE_UINT16(product_id, struct usb_redir_device_connect_header),
  2190. VMSTATE_UINT16(device_version_bcd,
  2191. struct usb_redir_device_connect_header),
  2192. VMSTATE_END_OF_LIST()
  2193. }
  2194. };
  2195. /* For usb_redir_interface_info_header migration */
  2196. static const VMStateDescription usbredir_interface_info_vmstate = {
  2197. .name = "usb-redir-interface-info",
  2198. .version_id = 1,
  2199. .minimum_version_id = 1,
  2200. .fields = (VMStateField[]) {
  2201. VMSTATE_UINT32(interface_count,
  2202. struct usb_redir_interface_info_header),
  2203. VMSTATE_UINT8_ARRAY(interface,
  2204. struct usb_redir_interface_info_header, 32),
  2205. VMSTATE_UINT8_ARRAY(interface_class,
  2206. struct usb_redir_interface_info_header, 32),
  2207. VMSTATE_UINT8_ARRAY(interface_subclass,
  2208. struct usb_redir_interface_info_header, 32),
  2209. VMSTATE_UINT8_ARRAY(interface_protocol,
  2210. struct usb_redir_interface_info_header, 32),
  2211. VMSTATE_END_OF_LIST()
  2212. }
  2213. };
  2214. /* And finally the USBRedirDevice vmstate itself */
  2215. static const VMStateDescription usbredir_vmstate = {
  2216. .name = "usb-redir",
  2217. .version_id = 1,
  2218. .minimum_version_id = 1,
  2219. .pre_save = usbredir_pre_save,
  2220. .post_load = usbredir_post_load,
  2221. .fields = (VMStateField[]) {
  2222. VMSTATE_USB_DEVICE(dev, USBRedirDevice),
  2223. VMSTATE_TIMER_PTR(attach_timer, USBRedirDevice),
  2224. {
  2225. .name = "parser",
  2226. .version_id = 0,
  2227. .field_exists = NULL,
  2228. .size = 0,
  2229. .info = &usbredir_parser_vmstate_info,
  2230. .flags = VMS_SINGLE,
  2231. .offset = 0,
  2232. },
  2233. VMSTATE_STRUCT_ARRAY(endpoint, USBRedirDevice, MAX_ENDPOINTS, 1,
  2234. usbredir_ep_vmstate, struct endp_data),
  2235. VMSTATE_STRUCT(cancelled, USBRedirDevice, 1,
  2236. usbredir_ep_packet_id_queue_vmstate,
  2237. struct PacketIdQueue),
  2238. VMSTATE_STRUCT(already_in_flight, USBRedirDevice, 1,
  2239. usbredir_ep_packet_id_queue_vmstate,
  2240. struct PacketIdQueue),
  2241. VMSTATE_STRUCT(device_info, USBRedirDevice, 1,
  2242. usbredir_device_info_vmstate,
  2243. struct usb_redir_device_connect_header),
  2244. VMSTATE_STRUCT(interface_info, USBRedirDevice, 1,
  2245. usbredir_interface_info_vmstate,
  2246. struct usb_redir_interface_info_header),
  2247. VMSTATE_END_OF_LIST()
  2248. }
  2249. };
  2250. static Property usbredir_properties[] = {
  2251. DEFINE_PROP_CHR("chardev", USBRedirDevice, cs),
  2252. DEFINE_PROP_UINT8("debug", USBRedirDevice, debug, usbredirparser_warning),
  2253. DEFINE_PROP_STRING("filter", USBRedirDevice, filter_str),
  2254. DEFINE_PROP_BOOL("streams", USBRedirDevice, enable_streams, true),
  2255. DEFINE_PROP_BOOL("suppress-remote-wake", USBRedirDevice,
  2256. suppress_remote_wake, true),
  2257. DEFINE_PROP_END_OF_LIST(),
  2258. };
  2259. static void usbredir_class_initfn(ObjectClass *klass, void *data)
  2260. {
  2261. USBDeviceClass *uc = USB_DEVICE_CLASS(klass);
  2262. DeviceClass *dc = DEVICE_CLASS(klass);
  2263. uc->realize = usbredir_realize;
  2264. uc->product_desc = "USB Redirection Device";
  2265. uc->unrealize = usbredir_unrealize;
  2266. uc->cancel_packet = usbredir_cancel_packet;
  2267. uc->handle_reset = usbredir_handle_reset;
  2268. uc->handle_data = usbredir_handle_data;
  2269. uc->handle_control = usbredir_handle_control;
  2270. uc->flush_ep_queue = usbredir_flush_ep_queue;
  2271. uc->ep_stopped = usbredir_ep_stopped;
  2272. uc->alloc_streams = usbredir_alloc_streams;
  2273. uc->free_streams = usbredir_free_streams;
  2274. dc->vmsd = &usbredir_vmstate;
  2275. device_class_set_props(dc, usbredir_properties);
  2276. set_bit(DEVICE_CATEGORY_MISC, dc->categories);
  2277. }
  2278. static void usbredir_instance_init(Object *obj)
  2279. {
  2280. USBDevice *udev = USB_DEVICE(obj);
  2281. USBRedirDevice *dev = USB_REDIRECT(udev);
  2282. device_add_bootindex_property(obj, &dev->bootindex,
  2283. "bootindex", NULL,
  2284. &udev->qdev);
  2285. }
  2286. static const TypeInfo usbredir_dev_info = {
  2287. .name = TYPE_USB_REDIR,
  2288. .parent = TYPE_USB_DEVICE,
  2289. .instance_size = sizeof(USBRedirDevice),
  2290. .class_init = usbredir_class_initfn,
  2291. .instance_init = usbredir_instance_init,
  2292. };
  2293. module_obj(TYPE_USB_REDIR);
  2294. module_kconfig(USB);
  2295. static void usbredir_register_types(void)
  2296. {
  2297. type_register_static(&usbredir_dev_info);
  2298. }
  2299. type_init(usbredir_register_types)