Flush input buffer for a client connection when calling connection_teardown().

Signed-off-by: Martin Szulecki <m.szulecki@libimobiledevice.org>

1 files changed, 16 insertions, 0 deletions

diff --git a/src/device.c b/src/device.c
index 9f45ea9..ec0c9f2 100644
--- a/src/device.c
+++ b/src/device.c
@@ -300,6 +300,7 @@ static int send_tcp(struct mux_connection *conn, uint8_t flags, const unsigned c
 static void connection_teardown(struct mux_connection *conn)
 {
         int res;
+        int size;
         if(conn->state == CONN_DEAD)
                 return;
         usbmuxd_log(LL_DEBUG, "connection_teardown dev %d sport %d dport %d", conn->dev->id, conn->sport, conn->dport);
@@ -313,6 +314,21 @@ static void connection_teardown(struct mux_connection *conn)
                         client_notify_connect(conn->client, RESULT_CONNREFUSED);
                 } else {
                         conn->state = CONN_DEAD;
+                        if((conn->events & POLLOUT) && conn->ib_size > 0){
+                                while(1){
+                                        size = client_write(conn->client, conn->ib_buf, conn->ib_size);
+                                        if(size <= 0) {
+                                                break;
+                                        }
+                                        if(size == (int)conn->ib_size) {
+                                                conn->ib_size = 0;
+                                                break;
+                                        } else {
+                                                conn->ib_size -= size;
+                                                memmove(conn->ib_buf, conn->ib_buf + size, conn->ib_size);
+                                        }
+                                }
+                        }
                         client_close(conn->client);
                 }
         }