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