Add support for wireless pairing

1 files changed, 179 insertions, 6 deletions

diff --git a/tools/idevicepair.c b/tools/idevicepair.c
index 0dcd45f..a2dc944 100644
--- a/tools/idevicepair.c
+++ b/tools/idevicepair.c
@@ -2,7 +2,7 @@
  * idevicepair.c
  * Manage pairings with devices and this host
  *
- * Copyright (c) 2010-2019 Nikias Bassen, All Rights Reserved.
+ * Copyright (c) 2010-2021 Nikias Bassen, All Rights Reserved.
  * Copyright (c) 2014 Martin Szulecki, All Rights Reserved.
  *
  * This library is free software; you can redistribute it and/or
@@ -30,16 +30,88 @@
 #include <string.h>
 #include <stdlib.h>
 #include <getopt.h>
-#ifndef WIN32
+#include <ctype.h>
+#include <unistd.h>
+#ifdef WIN32
+#include <windows.h>
+#include <conio.h>
+#else
+#include <termios.h>
 #include <signal.h>
 #endif
+
 #include "common/userpref.h"
+#include <libimobiledevice-glue/utils.h>
 
 #include <libimobiledevice/libimobiledevice.h>
 #include <libimobiledevice/lockdown.h>
+#include <plist/plist.h>
 
 static char *udid = NULL;
 
+#ifdef HAVE_WIRELESS_PAIRING
+
+#ifdef WIN32
+#define BS_CC '\b'
+#define my_getch getch
+#else
+#define BS_CC 0x7f
+static int my_getch(void)
+{
+	struct termios oldt, newt;
+	int ch;
+	tcgetattr(STDIN_FILENO, &oldt);
+	newt = oldt;
+	newt.c_lflag &= ~(ICANON | ECHO);
+	tcsetattr(STDIN_FILENO, TCSANOW, &newt);
+	ch = getchar();
+	tcsetattr(STDIN_FILENO, TCSANOW, &oldt);
+	return ch;
+}
+#endif
+
+static int get_hidden_input(char *buf, int maxlen)
+{
+	int pwlen = 0;
+	int c;
+
+	while ((c = my_getch())) {
+		if ((c == '\r') || (c == '\n')) {
+			break;
+		} else if (isprint(c)) {
+			if (pwlen < maxlen-1)
+				buf[pwlen++] = c;
+			fputc('*', stderr);
+		} else if (c == BS_CC) {
+			if (pwlen > 0) {
+				fputs("\b \b", stderr);
+				pwlen--;
+			}
+		}
+	}
+	buf[pwlen] = 0;
+	return pwlen;
+}
+
+static void pairing_cb(lockdownd_cu_pairing_cb_type_t cb_type, void *user_data, void* data_ptr, unsigned int* data_size)
+{
+	if (cb_type == LOCKDOWN_CU_PAIRING_PIN_REQUESTED) {
+		printf("Enter PIN: ");
+		fflush(stdout);
+
+		*data_size = get_hidden_input((char*)data_ptr, *data_size);
+
+		printf("\n");
+	} else if (cb_type == LOCKDOWN_CU_PAIRING_DEVICE_INFO) {
+		printf("Device info:\n");
+		plist_print_to_stream_with_indentation((plist_t)data_ptr, stdout, 2);
+	} else if (cb_type == LOCKDOWN_CU_PAIRING_ERROR) {
+		printf("ERROR: %s\n", (data_ptr) ? (char*)data_ptr : "(unknown)");
+	}
+}
+
+#endif /* HAVE_WIRELESS_PAIRING */
+
 static void print_error_message(lockdownd_error_t err)
 {
 	switch (err) {
@@ -56,6 +128,16 @@ static void print_error_message(lockdownd_error_t err)
 		case LOCKDOWN_E_USER_DENIED_PAIRING:
 			printf("ERROR: Device %s said that the user denied the trust dialog.\n", udid);
 			break;
+		case LOCKDOWN_E_PAIRING_FAILED:
+			printf("ERROR: Pairing with device %s failed.\n", udid);
+			break;
+		case LOCKDOWN_E_GET_PROHIBITED:
+		case LOCKDOWN_E_PAIRING_PROHIBITED_OVER_THIS_CONNECTION:
+			printf("ERROR: Pairing is not possible over this connection.\n");
+#ifdef HAVE_WIRELESS_PAIRING
+			printf("To perform a wireless pairing use the -w command line switch. See usage or man page for details.\n");
+#endif
+			break;
 		default:
 			printf("ERROR: Device %s returned unhandled error code %d\n", udid, err);
 			break;
@@ -81,9 +163,20 @@ static void print_usage(int argc, char **argv)
 	printf("\n");
 	printf("The following OPTIONS are accepted:\n");
 	printf("  -u, --udid UDID  target specific device by UDID\n");
+#ifdef HAVE_WIRELESS_PAIRING
+	printf("  -w, --wireless   perform wireless pairing (see NOTE)\n");
+	printf("  -n, --network    connect to network device (see NOTE)\n");
+#endif
 	printf("  -d, --debug      enable communication debugging\n");
 	printf("  -h, --help       prints usage information\n");
 	printf("  -v, --version    prints version information\n");
+#ifdef HAVE_WIRELESS_PAIRING
+	printf("\n");
+	printf("NOTE: Pairing over network (wireless pairing) is only supported by Apple TV\n");
+	printf("devices. To perform a wireless pairing, you need to use the -w command line\n");
+	printf("switch. Make sure to put the device into pairing mode first by opening\n");
+	printf("Settings > Remotes and Devices > Remote App and Devices.\n");
+#endif
 	printf("\n");
 	printf("Homepage:    <" PACKAGE_URL ">\n");
 	printf("Bug Reports: <" PACKAGE_BUGREPORT ">\n");
@@ -95,10 +188,20 @@ int main(int argc, char **argv)
 	static struct option longopts[] = {
 		{ "help",    no_argument,       NULL, 'h' },
 		{ "udid",    required_argument, NULL, 'u' },
+#ifdef HAVE_WIRELESS_PAIRING
+		{ "wireless", no_argument,      NULL, 'w' },
+		{ "network", no_argument,       NULL, 'n' },
+		{ "hostinfo", required_argument, NULL,  1 },
+#endif
 		{ "debug",   no_argument,       NULL, 'd' },
 		{ "version", no_argument,       NULL, 'v' },
 		{ NULL, 0, NULL, 0}
 	};
+#ifdef HAVE_WIRELESS_PAIRING
+#define SHORT_OPTIONS "hu:wndv"
+#else
+#define SHORT_OPTIONS "hu:dv"
+#endif
 	lockdownd_client_t client = NULL;
 	idevice_t device = NULL;
 	idevice_error_t ret = IDEVICE_E_UNKNOWN_ERROR;
@@ -106,13 +209,18 @@ int main(int argc, char **argv)
 	int result;
 
 	char *type = NULL;
+	int use_network = 0;
+	int wireless_pairing = 0;
+#ifdef HAVE_WIRELESS_PAIRING
+	plist_t host_info_plist = NULL;
+#endif
 	char *cmd;
 	typedef enum {
 		OP_NONE = 0, OP_PAIR, OP_VALIDATE, OP_UNPAIR, OP_LIST, OP_HOSTID, OP_SYSTEMBUID
 	} op_t;
 	op_t op = OP_NONE;
 
-	while ((c = getopt_long(argc, argv, "hu:dv", longopts, NULL)) != -1) {
+	while ((c = getopt_long(argc, argv, SHORT_OPTIONS, longopts, NULL)) != -1) {
 		switch (c) {
 		case 'h':
 			print_usage(argc, argv);
@@ -127,6 +235,43 @@ int main(int argc, char **argv)
 			free(udid);
 			udid = strdup(optarg);
 			break;
+#ifdef HAVE_WIRELESS_PAIRING
+		case 'w':
+			wireless_pairing = 1;
+			break;
+		case 'n':
+			use_network = 1;
+			break;
+		case 1:
+			if (!*optarg) {
+				fprintf(stderr, "ERROR: --hostinfo argument must not be empty!\n");
+				result = EXIT_FAILURE;
+				goto leave;
+			}
+			if (*optarg == '@') {
+				plist_read_from_filename(&host_info_plist, optarg+1);
+				if (!host_info_plist) {
+					fprintf(stderr, "ERROR: Could not read from file '%s'\n", optarg+1);
+					result = EXIT_FAILURE;
+					goto leave;
+				}
+			}
+#ifdef HAVE_PLIST_JSON
+			else if (*optarg == '{') {
+				if (plist_from_json(optarg, strlen(optarg), &host_info_plist) != PLIST_ERR_SUCCESS) {
+					fprintf(stderr, "ERROR: --hostinfo argument not valid. Make sure it is a JSON dictionary.\n");
+					result = EXIT_FAILURE;
+					goto leave;
+				}
+			}
+#endif
+			else {
+				fprintf(stderr, "ERROR: --hostinfo argument not valid. To specify a path prefix with '@'\n");
+				result = EXIT_FAILURE;
+				goto leave;
+			}
+			break;
+#endif
 		case 'd':
 			idevice_set_debug_level(1);
 			break;
@@ -152,6 +297,13 @@ int main(int argc, char **argv)
 		goto leave;
 	}
 
+	if (wireless_pairing && use_network) {
+		printf("ERROR: You cannot use -w and -n together.\n");
+		print_usage(argc, argv);
+		result = EXIT_FAILURE;
+		goto leave;
+	}
+
 	cmd = (argv+optind)[0];
 
 	if (!strcmp(cmd, "pair")) {
@@ -169,7 +321,18 @@ int main(int argc, char **argv)
 	} else {
 		printf("ERROR: Invalid command '%s' specified\n", cmd);
 		print_usage(argc, argv);
-		exit(EXIT_FAILURE);
+		result = EXIT_FAILURE;
+		goto leave;
+	}
+
+	if (wireless_pairing) {
+		if (op == OP_VALIDATE || op == OP_UNPAIR) {
+			printf("ERROR: Command '%s' is not supported with -w\n", cmd);
+			print_usage(argc, argv);
+			result = EXIT_FAILURE;
+			goto leave;
+		}
+		use_network = 1;
 	}
 
 	if (op == OP_SYSTEMBUID) {
@@ -198,7 +361,7 @@ int main(int argc, char **argv)
 		goto leave;
 	}
 
-	ret = idevice_new(&device, udid);
+	ret = idevice_new_with_options(&device, udid, (use_network) ? IDEVICE_LOOKUP_NETWORK : IDEVICE_LOOKUP_USBMUX);
 	if (ret != IDEVICE_E_SUCCESS) {
 		if (udid) {
 			printf("No device found with udid %s.\n", udid);
@@ -257,7 +420,17 @@ int main(int argc, char **argv)
 	switch(op) {
 		default:
 		case OP_PAIR:
-		lerr = lockdownd_pair(client, NULL);
+#ifdef HAVE_WIRELESS_PAIRING
+		if (wireless_pairing) {
+			lerr = lockdownd_cu_pairing_create(client, pairing_cb, NULL, host_info_plist, NULL);
+			if (lerr == LOCKDOWN_E_SUCCESS) {
+				lerr = lockdownd_pair_cu(client);
+			}
+		} else
+#endif
+		{
+			lerr = lockdownd_pair(client, NULL);
+		}
 		if (lerr == LOCKDOWN_E_SUCCESS) {
 			printf("SUCCESS: Paired with device %s\n", udid);
 		} else {