From 80aafc63c842ee902cc9e32d692efed8952a1e14 Mon Sep 17 00:00:00 2001
From: Hans de Goede <hdegoede@redhat.com>
Date: Tue, 19 Jul 2011 10:56:19 +0200
Subject: [PATCH 16/18] usb-redir: Add flow control support

Signed-off-by: Hans de Goede <hdegoede@redhat.com>
---
 usb-redir.c |   26 ++++++++++++++++++++++++--
 1 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/usb-redir.c b/usb-redir.c
index e421cff..1289506 100644
--- a/usb-redir.c
+++ b/usb-redir.c
@@ -228,12 +228,22 @@ static int usbredir_read(void *priv, uint8_t *data, int count)
 static int usbredir_write(void *priv, uint8_t *data, int count)
 {
     USBRedirDevice *dev = priv;
+    int r;
 
-    if (!dev->cs->opened) {
+    if (!dev->cs->opened || dev->cs->write_blocked) {
         return 0;
     }
 
-    return qemu_chr_fe_write(dev->cs, data, count);
+    r = qemu_chr_fe_write(dev->cs, data, count);
+
+    if (r < 0) {
+        if (dev->cs->write_blocked) {
+            return 0;
+        }
+        return -1;
+    }
+
+    return r;
 }
 
 /*
@@ -865,10 +875,22 @@ static void usbredir_chardev_event(void *opaque, int event)
     }
 }
 
+static void usbredir_chardev_write_unblocked(void *opaque)
+{
+    USBRedirDevice *dev = opaque;
+
+    if (dev->parser == NULL) {
+        /* usbredir_open_close_bh hasn't handled the open event yet */
+        return;
+    }
+    usbredirparser_do_write(dev->parser);
+}
+
 static const QemuChrHandlers usbredir_chr_handlers = {
     .fd_can_read = usbredir_chardev_can_read,
     .fd_read = usbredir_chardev_read,
     .fd_event = usbredir_chardev_event,
+    .fd_write_unblocked = usbredir_chardev_write_unblocked,
 };
 
 /*
-- 
1.7.5.4

