1 files changed, 69 insertions, 40 deletions

diff --git a/swig/iphone.i b/swig/iphone.i
index a0fe340..e970e89 100644
--- a/swig/iphone.i
+++ b/swig/iphone.i
@@ -10,13 +10,21 @@
  } iPhone;
 
  typedef struct {
-	iphone_device_t dev;
+	iPhone* dev;
 	iphone_lckd_client_t client;
  } Lockdownd;
 
  typedef struct {
+	Lockdownd* lckd;
 	iphone_msync_client_t client;
  } MobileSync;
+
+//now declare funtions to handle creation and deletion of objects
+void my_delete_iPhone(iPhone* dev);
+Lockdownd* my_new_Lockdownd(iPhone* phone);
+void my_delete_Lockdownd(Lockdownd* lckd);
+MobileSync* my_new_MobileSync(Lockdownd* lckd);
+
  %}
 /* Parse the header file to generate wrappers */
 %include "stdint.i"
@@ -27,14 +35,64 @@ typedef struct {
 } iPhone;
 
 typedef struct {
-	iphone_device_t dev;
+	iPhone* dev;
 	iphone_lckd_client_t client;
 } Lockdownd;
 
 typedef struct {
+	Lockdownd* lckd;
 	iphone_msync_client_t client;
 } MobileSync;
 
+%inline %{
+//now define funtions to handle creation and deletion of objects
+
+
+void my_delete_iPhone(iPhone* dev) {
+	if (dev) {
+		iphone_free_device ( dev->dev );
+		free(dev);
+	}
+}
+
+Lockdownd* my_new_Lockdownd(iPhone* phone) {
+	if (!phone) return NULL;
+	Lockdownd* client = (Lockdownd*) malloc(sizeof(Lockdownd));
+	client->dev = phone;
+	client->client = NULL;
+	if (IPHONE_E_SUCCESS == iphone_lckd_new_client ( phone->dev , &(client->client))) {
+		return client;
+	}
+	else {
+		free(client);
+		return NULL;
+	}
+}
+
+void my_delete_Lockdownd(Lockdownd* lckd) {
+	if (lckd) {
+		my_delete_iPhone(lckd->dev);
+		iphone_lckd_free_client ( lckd->client );
+		free(lckd);
+	}
+}
+
+MobileSync* my_new_MobileSync(Lockdownd* lckd) {
+	if (!lckd || !lckd->dev) return NULL;
+	MobileSync* client = NULL;
+	int port = 0;
+	if (IPHONE_E_SUCCESS == iphone_lckd_start_service ( lckd->client, "com.apple.mobilesync", &port )) {
+		client = (MobileSync*) malloc(sizeof(MobileSync));
+		client->lckd = lckd;
+		client->client = NULL;
+		iphone_msync_new_client ( lckd->dev->dev, 3432, port, &(client->client));
+	}
+	return client;
+}
+
+%}
+
+
 %extend iPhone {             // Attach these functions to struct iPhone
 	iPhone() {
 		iPhone* phone = (iPhone*) malloc(sizeof(iPhone));
@@ -44,8 +102,7 @@ typedef struct {
 	}
 
 	~iPhone() {
-		iphone_free_device ( $self->dev );
-		free($self);
+		my_delete_iPhone($self);
 	}
 
 	int InitDevice() {
@@ -55,60 +112,32 @@ typedef struct {
 	}
 
 	Lockdownd* GetLockdownClient() {
-		Lockdownd* client = (Lockdownd*) malloc(sizeof(Lockdownd));
-		client->client = NULL;
-		client->dev = NULL;
-		if (IPHONE_E_SUCCESS == iphone_lckd_new_client ( $self->dev , &(client->client)) ) {
-			client->dev = $self->dev;
-			return client;
-		}
-		free(client);
-		return NULL;
+		return my_new_Lockdownd($self);
 	}
 };
 
+
 %extend Lockdownd {             // Attach these functions to struct Lockdownd
 	Lockdownd(iPhone* phone) {
-		if (!phone) return NULL;
-		Lockdownd* client = (Lockdownd*) malloc(sizeof(Lockdownd));
-		client->client = NULL;
-		if (IPHONE_E_SUCCESS == iphone_lckd_new_client ( phone->dev , &(client->client))) {
-			client->dev = phone->dev;
-			return client;
-		}
-		else {
-			free(client);
-			return NULL;
-		}
+		return my_new_Lockdownd(phone);
 	}
 
 	~Lockdownd() {
-		iphone_lckd_free_client ( $self->client );
-		free($self);
+		my_delete_Lockdownd($self);
 	}
 
 	MobileSync* GetMobileSyncClient() {
-		int port = 0;
-		if (IPHONE_E_SUCCESS == iphone_lckd_start_service ( $self->client, "com.apple.mobilesync", &port )) {
-			MobileSync* client = (MobileSync*) malloc(sizeof(MobileSync));
-			client->client = NULL;
-			if (IPHONE_E_SUCCESS == iphone_msync_new_client ( $self->dev, 3432, port, &(client->client)))
-				return client;
-		}
-		return NULL;
+		return my_new_MobileSync($self);
 	}
 };
 
 %extend MobileSync {             // Attach these functions to struct MobileSync
-	MobileSync(iPhone* phone, int src_port, int dst_port) {
-		if (!phone) return NULL;
-		MobileSync* client = (MobileSync*) malloc(sizeof(MobileSync));
-		client->client = NULL;
-		iphone_msync_new_client ( phone->dev, src_port, dst_port, &(client->client));
-		return client;
+	MobileSync(Lockdownd* lckd) {
+		return my_new_MobileSync(lckd);
 	}
 
 	~MobileSync() {
+		my_delete_Lockdownd($self->lckd);
 		iphone_msync_free_client ( $self->client );
 		free($self);
 	}