redirect.c 77 KB

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