1 files changed, 24 insertions, 16 deletions

diff --git a/src/iphone.c b/src/iphone.c
index e0e150f..f3b7202 100644
--- a/src/iphone.c
+++ b/src/iphone.c
@@ -34,10 +34,15 @@ extern int debug;
  * @return A structure with data on the first iPhone it finds.  (Or NULL, on
  *         error)
  */
-iPhone_t get_iPhone() {
-	iPhone *phone = (iPhone*)malloc(sizeof(iPhone));
+int  iphone_get_device ( iphone_device_t *device  ){
+	//check we can actually write in device
+	if (!device || (device && *device))
+		return IPHONE_E_INVALID_ARG;
+
 	struct usb_bus *bus, *busses;
 	struct usb_device *dev;
+	iphone_device_t phone = (iphone_device_t)malloc(sizeof(struct iphone_device_int));
+	usbmux_version_header *version = version_header();
 	
 	// Initialize the struct
 	phone->device = NULL;
@@ -74,7 +79,7 @@ iPhone_t get_iPhone() {
 	if (!phone->device || !phone->__device) {
 		free_iPhone(phone);
 		if (debug) fprintf(stderr, "get_iPhone(): iPhone not found\n");
-		return NULL;
+		return IPHONE_E_NO_DEVICE;
 	}
 
 	// Send the version command to the phone
@@ -99,7 +104,7 @@ iPhone_t get_iPhone() {
 		if (debug) fprintf(stderr, "get_iPhone(): Invalid version message -- header too short.\n");
 		if (debug && bytes < 0) fprintf(stderr, "get_iPhone(): libusb error message %d: %s (%s)\n",
 			       			bytes, usb_strerror(), strerror(-bytes));
-		return NULL;
+		return IPHONE_E_NOT_ENOUGH_DATA;
 	}
 
 	// Check for correct version
@@ -107,20 +112,21 @@ iPhone_t get_iPhone() {
 		// We're all ready to roll.
 		fprintf(stderr, "get_iPhone() success\n");
 		free(version);
-		return phone;
+		*device = phone;
+		return IPHONE_E_SUCCESS;
 	} else {
 		// Bad header
 		free_iPhone(phone);
 		free(version);
 		if (debug) fprintf(stderr, "get_iPhone(): Received a bad header/invalid version number.");
-		return NULL;
+		return IPHONE_E_BAD_HEADER;
 	}
 
 	// If it got to this point it's gotta be bad
 	if (debug) fprintf(stderr, "get_iPhone(): Unknown error.\n");
 	free_iPhone(phone);
 	free(version);
-	return NULL; 
+	return IPHONE_E_NO_DEVICE; // if it got to this point it's gotta be bad
 }
 
 /** Cleans up an iPhone structure, then frees the structure itself.  
@@ -129,14 +135,14 @@ iPhone_t get_iPhone() {
  * 
  * @param phone A pointer to an iPhone structure.
  */
-void free_iPhone(iPhone_t phone) {
-	if (phone->buffer) free(phone->buffer);	
-	if (phone->device) {
-		usb_release_interface(phone->device, 1);
-		usb_reset(phone->device);
-		usb_close(phone->device);
+void iphone_free_device ( iphone_device_t device ) {
+	if (device->buffer) free(device->buffer);	
+	if (device->device) {
+		usb_release_interface(device->device, 1);
+		usb_reset(device->device);
+		usb_close(device->device);
 	}
-	free(phone);
+	free(device);
 }
  
 /** Sends data to the phone
@@ -147,7 +153,8 @@ void free_iPhone(iPhone_t phone) {
  * @param datalen The length of the data
  * @return The number of bytes sent, or -1 on error or something.
  */
-int send_to_phone(iPhone *phone, char *data, int datalen) {
+int send_to_phone(iphone_device_t phone, char *data, int datalen) {
+	if (!phone) return -1;
 	int bytes = 0;
 	
 	if (!phone) return -1;
@@ -173,7 +180,8 @@ int send_to_phone(iPhone *phone, char *data, int datalen) {
  * 
  * @return How many bytes were read in, or -1 on error.
  */
-int recv_from_phone(iPhone *phone, char *data, int datalen) {
+int recv_from_phone(iphone_device_t phone, char *data, int datalen) {
+	if (!phone) return -1;
 	int bytes = 0;
 	
 	if (!phone) return -1;