2 files changed, 7 insertions, 0 deletions

diff --git a/src/idevice.c b/src/idevice.c
index f0d89df..c605da3 100644
--- a/src/idevice.c
+++ b/src/idevice.c
@@ -238,6 +238,7 @@ idevice_error_t idevice_connect(idevice_t device, uint16_t port, idevice_connect
                 new_connection->type = CONNECTION_USBMUXD;
                 new_connection->data = (void*)(long)sfd;
                 new_connection->ssl_data = NULL;
+                idevice_get_udid(device, &new_connection->udid);
                 *connection = new_connection;
                 return IDEVICE_E_SUCCESS;
         } else {
@@ -270,7 +271,12 @@ idevice_error_t idevice_disconnect(idevice_connection_t connection)
         } else {
                 debug_info("Unknown connection type %d", connection->type);
         }
+
+        if (connection->udid)
+                free(connection->udid);
+
         free(connection);
+
         return result;
 }
 
diff --git a/src/idevice.h b/src/idevice.h
index 4aeaf33..01b13df 100644
--- a/src/idevice.h
+++ b/src/idevice.h
@@ -58,6 +58,7 @@ struct ssl_data_private {
 typedef struct ssl_data_private *ssl_data_t;
 
 struct idevice_connection_private {
+        char *udid;
         enum connection_type type;
         void *data;
         ssl_data_t ssl_data;