1 files changed, 46 insertions, 71 deletions

diff --git a/tools/idevicedebugserverproxy.c b/tools/idevicedebugserverproxy.c
index 475749f..2209abc 100644
--- a/tools/idevicedebugserverproxy.c
+++ b/tools/idevicedebugserverproxy.c
@@ -26,7 +26,7 @@
 #include <signal.h>
 
 #include <libimobiledevice/libimobiledevice.h>
-#include <libimobiledevice/lockdown.h>
+#include <libimobiledevice/debugserver.h>
 
 #include "common/socket.h"
 #include "common/thread.h"
@@ -41,16 +41,30 @@ typedef struct {
 	int server_fd;
 	int client_fd;
 	uint16_t local_port;
-	uint16_t remote_port;
-	idevice_connection_t device_connection;
+	idevice_t device;
+	debugserver_client_t debugserver_client;
 	volatile int stop_ctod;
 	volatile int stop_dtoc;
 } socket_info_t;
 
+static socket_info_t global_socket_info;
+
 static void clean_exit(int sig)
 {
-	fprintf(stderr, "Exiting...\n");
+	if (quit_flag == 0)
+		fprintf(stderr, "Setting quit status. Cancel again to quit server.\n");
+	else
+		fprintf(stderr, "Exiting...\n");
+
 	quit_flag++;
+
+	/* shutdown server socket if we have to terminate to unblock the server loop */
+	if (quit_flag > 1) {
+		if (global_socket_info.server_fd > 0) {
+			socket_shutdown(global_socket_info.server_fd, SHUT_RDWR);
+			socket_close(global_socket_info.server_fd);
+		}
+	}
 }
 
 static void print_usage(int argc, char **argv)
@@ -83,7 +97,7 @@ static void *thread_device_to_client(void *data)
 	while (!quit_flag && !socket_info->stop_dtoc && socket_info->client_fd > 0 && socket_info->server_fd > 0) {
 		debug("%s: receiving data from device...\n", __func__);
 
-		res = idevice_connection_receive_timeout(socket_info->device_connection, buffer, sizeof(buffer), (uint32_t*)&recv_len, 5000);
+		res = debugserver_client_receive_with_timeout(socket_info->debugserver_client, buffer, sizeof(buffer), (uint32_t*)&recv_len, 5000);
 
 		if (recv_len <= 0) {
 			if (recv_len == 0 && res == IDEVICE_E_SUCCESS) {
@@ -161,7 +175,7 @@ static void *thread_client_to_device(void *data)
 		} else {
 			/* forward data to device */
 			debug("%s: sending data to device...\n", __func__);
-			res = idevice_connection_send(socket_info->device_connection, buffer, recv_len, (uint32_t*)&sent);
+			res = debugserver_client_send(socket_info->debugserver_client, buffer, recv_len, (uint32_t*)&sent);
 
 			if (sent < recv_len || res != IDEVICE_E_SUCCESS) {
 				if (sent <= 0) {
@@ -194,11 +208,18 @@ static void *thread_client_to_device(void *data)
 
 static void* connection_handler(void* data)
 {
+	debugserver_error_t derr = DEBUGSERVER_E_SUCCESS;
 	socket_info_t* socket_info = (socket_info_t*)data;
 	thread_t ctod;
 
 	debug("%s: client_fd = %d\n", __func__, socket_info->client_fd);
 
+	derr = debugserver_client_start_service(socket_info->device, &socket_info->debugserver_client, "idevicedebugserverproxy");
+	if (derr != DEBUGSERVER_E_SUCCESS) {
+		fprintf(stderr, "Could not start debugserver on device!\nPlease make sure to mount a developer disk image first.\n");
+		return NULL;
+	}
+
 	/* spawn client to device thread */
 	socket_info->stop_ctod = 0;
 	if (thread_new(&ctod, thread_client_to_device, data) != 0) {
@@ -209,29 +230,23 @@ static void* connection_handler(void* data)
 	thread_join(ctod);
 	thread_free(ctod);
 
+	debug("%s: shutting down...\n", __func__);
+
+	debugserver_client_free(socket_info->debugserver_client);
+	socket_info->debugserver_client = NULL;
+
 	/* shutdown client socket */
 	socket_shutdown(socket_info->client_fd, SHUT_RDWR);
 	socket_close(socket_info->client_fd);
 
-	/* shutdown server socket if we have to terminate to unblock the server loop */
-	if (quit_flag) {
-		socket_shutdown(socket_info->server_fd, SHUT_RDWR);
-		socket_close(socket_info->server_fd);
-	}
-
 	return NULL;
 }
 
 int main(int argc, char *argv[])
 {
-	lockdownd_client_t lockdown = NULL;
-	lockdownd_error_t ldret = LOCKDOWN_E_UNKNOWN_ERROR;
-	idevice_t device = NULL;
-	idevice_connection_t connection = NULL;
 	idevice_error_t ret = IDEVICE_E_UNKNOWN_ERROR;
 	thread_t th;
 	const char* udid = NULL;
-	lockdownd_service_descriptor_t service = NULL;
 	uint16_t local_port = 0;
 	int result = EXIT_SUCCESS;
 	int i;
@@ -282,8 +297,11 @@ int main(int argc, char *argv[])
 		goto leave_cleanup;
 	}
 
+	/* setup and create socket endpoint */
+	global_socket_info.local_port = local_port;
+
 	/* start services and connect to device */
-	ret = idevice_new(&device, udid);
+	ret = idevice_new(&global_socket_info.device, udid);
 	if (ret != IDEVICE_E_SUCCESS) {
 		if (udid) {
 			fprintf(stderr, "No device found with udid %s, is it plugged in?\n", udid);
@@ -294,66 +312,29 @@ int main(int argc, char *argv[])
 		goto leave_cleanup;
 	}
 
-	if (LOCKDOWN_E_SUCCESS != (ldret = lockdownd_client_new_with_handshake(device, &lockdown, "idevicedebugserverproxy"))) {
-		fprintf(stderr, "ERROR: Could not connect to lockdownd, error code %d\n", ldret);
-		result = EXIT_FAILURE;
-		goto leave_cleanup;
-	}
-
-	if ((lockdownd_start_service(lockdown, "com.apple.debugserver", &service) != LOCKDOWN_E_SUCCESS) || !service || !service->port) {
-		fprintf(stderr, "Could not start com.apple.debugserver!\nPlease make sure to mount the developer disk image first.\n");
-		result = EXIT_FAILURE;
-		goto leave_cleanup;
-	}
-
-	if (idevice_connect(device, service->port, &connection) != IDEVICE_E_SUCCESS) {
-		fprintf(stderr, "Connection to debugserver port %d failed!\n", (int)service->port);
-		result = EXIT_FAILURE;
-		goto leave_cleanup;
-	}
-
-	/* free lockdown connection if running as it is not needed anymore */
-	if (lockdown) {
-		lockdownd_client_free(lockdown);
-		lockdown = NULL;
-	}
-
-	/* setup and create socket endpoint */
-	socket_info_t socket_info;
-
-	socket_info.device_connection = connection;
-	socket_info.local_port = local_port;
-	socket_info.remote_port = service->port;
-
-	if (service) {
-		lockdownd_service_descriptor_free(service);
-		service = NULL;
-	}
-
 	/* create local socket */
-	socket_info.server_fd = socket_create(socket_info.local_port);
-	if (socket_info.server_fd < 0) {
+	global_socket_info.server_fd = socket_create(global_socket_info.local_port);
+	if (global_socket_info.server_fd < 0) {
 		fprintf(stderr, "Could not create socket\n");
 		result = EXIT_FAILURE;
 		goto leave_cleanup;
 	}
 
 	while (!quit_flag) {
-		debug("%s: Waiting for connection on local port %d\n", __func__, socket_info.local_port);
+		debug("%s: Waiting for connection on local port %d\n", __func__, global_socket_info.local_port);
 
 		/* wait for client */
-		socket_info.client_fd = socket_accept(socket_info.server_fd, socket_info.local_port);
-		if (socket_info.client_fd < 0) {
-			debug("%s: Continuing...\n", __func__);
+		global_socket_info.client_fd = socket_accept(global_socket_info.server_fd, global_socket_info.local_port);
+		if (global_socket_info.client_fd < 0) {
 			continue;
 		}
 
 		debug("%s: Handling new client connection...\n", __func__);
 
-		if (thread_new(&th, connection_handler, (void*)&socket_info) != 0) {
+		if (thread_new(&th, connection_handler, (void*)&global_socket_info) != 0) {
 			fprintf(stderr, "Could not start connection handler.\n");
-			socket_shutdown(socket_info.server_fd, SHUT_RDWR);
-			socket_close(socket_info.server_fd);
+			socket_shutdown(global_socket_info.server_fd, SHUT_RDWR);
+			socket_close(global_socket_info.server_fd);
 			continue;
 		}
 
@@ -364,14 +345,8 @@ int main(int argc, char *argv[])
 	debug("%s: Shutting down debugserver proxy...\n", __func__);
 
 leave_cleanup:
-	if (connection) {
-		idevice_disconnect(connection);
-	}
-	if (lockdown) {
-		lockdownd_client_free(lockdown);
-	}
-	if (device) {
-		idevice_free(device);
+	if (global_socket_info.device) {
+		idevice_free(global_socket_info.device);
 	}
 
 	return result;