hopefully fixed race condition on connection setup

1 files changed, 13 insertions, 0 deletions

diff --git a/src/usbmux.c b/src/usbmux.c
index 87cdede..927928e 100644
--- a/src/usbmux.c
+++ b/src/usbmux.c
@@ -100,6 +100,8 @@ struct usbmux_client_int {
         int error;
 
         int cleanup;
+
+        int connected;
 };
 
 
@@ -734,6 +736,7 @@ int usbmux_new_client(usbmux_device_t device, uint16_t src_port,
                 add_connection(new_connection);
                 new_connection->error = 0;
                 new_connection->cleanup = 0;
+                new_connection->connected = 0;
                 hton_header(new_connection->header);
                 log_debug_msg("%s: send_to_device (%d --> %d)\n", __func__,
                                           ntohs(new_connection->header->sport),
@@ -931,6 +934,8 @@ uint32_t append_receive_buffer(usbmux_client_t client, char *packet)
                                  sizeof(usbmux_tcp_header)) <= 0) {
                                 log_debug_msg("%s: error when pushing to usb...\n",
                                                           __func__);
+                        } else {
+                                client->connected = 1;
                         }
                         // need to revert some of the fields back to host notation.
                         ntoh_header(client->header);
@@ -1253,3 +1258,11 @@ int usbmux_recv_timeout(usbmux_client_t client, char *data,
 
         return 0;
 }
+
+int usbmux_is_connected(usbmux_client_t client)
+{
+        if (!client) {
+                return 0;
+        }
+        return client->connected;
+}