Added ideviceprovision tool

2 files changed, 427 insertions, 1 deletions

diff --git a/tools/Makefile.am b/tools/Makefile.am
index c85bf13..6a4822b 100644
--- a/tools/Makefile.am
+++ b/tools/Makefile.am
@@ -3,7 +3,7 @@ AM_CPPFLAGS = -I$(top_srcdir)/include
 AM_CFLAGS = $(GLOBAL_CFLAGS) $(libgnutls_CFLAGS) $(libtasn1_CFLAGS) $(openssl_CFLAGS) $(libplist_CFLAGS) $(LFS_CFLAGS)
 AM_LDFLAGS = $(libgnutls_LIBS) $(libtasn1_LIBS) $(openssl_LIBS) $(libplist_LIBS)
 
-bin_PROGRAMS = idevice_id ideviceinfo idevicepair idevicesyslog ideviceimagemounter idevicescreenshot ideviceenterrecovery idevicedate idevicebackup idevicebackup2
+bin_PROGRAMS = idevice_id ideviceinfo idevicepair idevicesyslog ideviceimagemounter idevicescreenshot ideviceenterrecovery idevicedate idevicebackup idevicebackup2 ideviceprovision
 
 ideviceinfo_SOURCES = ideviceinfo.c
 ideviceinfo_CFLAGS = $(AM_CFLAGS)
@@ -54,3 +54,8 @@ idevicedate_SOURCES = idevicedate.c
 idevicedate_CFLAGS = $(AM_CFLAGS)
 idevicedate_LDFLAGS = $(AM_LDFLAGS)
 idevicedate_LDADD = ../src/libimobiledevice.la
+
+ideviceprovision_SOURCES = ideviceprovision.c
+ideviceprovision_CFLAGS = $(AM_CFLAGS)
+ideviceprovision_LDFLAGS = $(AM_LDFLAGS)
+ideviceprovision_LDADD = ../src/libimobiledevice.la
diff --git a/tools/ideviceprovision.c b/tools/ideviceprovision.c
new file mode 100644
index 0000000..02b4d1e
--- /dev/null
+++ b/tools/ideviceprovision.c
@@ -0,0 +1,421 @@
+/*
+ * ideviceprovision.c
+ * Simple utility to install, get, or remove provisioning profiles
+ *   to/from idevices
+ *
+ * Copyright (c) 2012 Nikias Bassen, All Rights Reserved.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ * 
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA 
+ */
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include <arpa/inet.h>
+
+#include <libimobiledevice/libimobiledevice.h>
+#include <libimobiledevice/lockdown.h>
+#include <libimobiledevice/misagent.h>
+
+static void print_usage(int argc, char **argv)
+{
+	char *name = NULL;
+	
+	name = strrchr(argv[0], '/');
+	printf("Usage: %s [OPTIONS]\n", (name ? name + 1: argv[0]));
+	printf("Display the current date or set it on a device.\n\n");
+	printf("  -d, --debug\t\tenable communication debugging\n");
+	printf("  -u, --udid UDID\ttarget specific device by its 40-digit device UDID\n");
+	printf("  -i, --install FILE\tInstalls the provisioning profile specified by FILE.\n");
+	printf("                    \tA valid .mobileprovision file is expected.\n");
+	printf("  -l, --list\t\tGet a list of all provisioning profiles on the device.\n");
+	printf("  -c, --copy PATH\tRetrieves all provisioning profiles from the device and\n");
+	printf("                 \tstores them into the existing directory specified by PATH.\n");
+	printf("                 \tThe files will be stored as UUID.mobileprovision\n");
+	printf("  -r, --remove UUID\tRemoves the provisioning profile identified by UUID.\n");
+	printf("  -h, --help\t\tprints usage information\n");
+	printf("\n");
+}
+
+enum {
+	OP_INSTALL,
+	OP_LIST,
+	OP_COPY,
+	OP_REMOVE,
+	NUM_OPS
+};
+
+#define ASN1_SEQUENCE 0x30
+#define ASN1_CONTAINER 0xA0
+#define ASN1_OBJECT_IDENTIFIER 0x06
+#define ASN1_OCTET_STRING 0x04
+
+static void asn1_next_item(unsigned char** p)
+{
+	if (*(*p+1) & 0x80) {
+		*p += 4;
+	} else {
+		*p += 3;
+	}
+}
+
+static int asn1_item_get_size(unsigned char* p)
+{
+	int res = 0;
+	if (*(p+1) & 0x80) {
+		uint16_t ws = 0;
+		memcpy(&ws, p+2, 2);
+		ws = ntohs(ws);
+		res = ws;
+	} else {
+		res = (int) *(p+1);
+	}
+	return res;
+}
+
+static void asn1_skip_item(unsigned char** p)
+{
+	int sz = asn1_item_get_size(*p);
+	*p += 2;
+	*p += sz;
+}
+
+static plist_t profile_get_embedded_plist(plist_t profile)
+{
+	if (plist_get_node_type(profile) != PLIST_DATA) {
+		fprintf(stderr, "%s: unexpected plist node type for profile (PLIST_DATA expected)\n", __func__);
+		return NULL;
+	}
+	char* bbuf = NULL;
+	uint64_t blen = 0;
+	plist_get_data_val(profile, &bbuf, &blen);
+	if (!bbuf) {
+		fprintf(stderr, "%s: could not get data value from plist node\n", __func__);
+		return NULL;
+	}
+
+	unsigned char* pp = (unsigned char*)bbuf;
+
+	if (*pp != ASN1_SEQUENCE) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (0)\n", __func__);
+		return NULL;
+	}
+	uint16_t slen = asn1_item_get_size(pp);
+	if (slen+4 != (uint16_t)blen) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (1)\n", __func__);
+		return NULL;
+	}
+	asn1_next_item(&pp);
+
+	if (*pp != ASN1_OBJECT_IDENTIFIER) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (2)\n", __func__);
+		return NULL;
+	}
+	asn1_skip_item(&pp);
+
+	if (*pp != ASN1_CONTAINER) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (3)\n", __func__);
+		return NULL;
+	}
+	asn1_next_item(&pp);
+
+	if (*pp != ASN1_SEQUENCE) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (4)\n", __func__);
+		return NULL;
+	}
+	asn1_next_item(&pp);
+
+	int k = 0;
+	// go to the 3rd element (skip 2)
+	while (k < 2) {
+		asn1_skip_item(&pp);
+		k++;
+	}
+	if (*pp != ASN1_SEQUENCE) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (5)\n", __func__);
+		return NULL;
+	}
+	asn1_next_item(&pp);
+
+	if (*pp != ASN1_OBJECT_IDENTIFIER) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (6)\n", __func__);
+		return NULL;
+	}
+	asn1_skip_item(&pp);
+
+	if (*pp != ASN1_CONTAINER) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (7)\n", __func__);
+		return NULL;
+	}
+	asn1_next_item(&pp);
+
+	if (*pp != ASN1_OCTET_STRING) {
+		free(bbuf);
+		fprintf(stderr, "%s: unexpected profile data (8)\n", __func__);
+		return NULL;
+	}
+	slen = asn1_item_get_size(pp);
+	asn1_next_item(&pp);
+
+	plist_t pl = NULL;
+	plist_from_xml((char*)pp, slen, &pl);
+	free(bbuf);
+
+	return pl;
+}
+
+int main(int argc, char *argv[])
+{
+	lockdownd_client_t client = NULL;
+	idevice_t device = NULL;
+	idevice_error_t ret = IDEVICE_E_UNKNOWN_ERROR;
+	int i;
+	int op = -1;
+	const char* udid = NULL;
+	const char* param = NULL;
+
+	/* parse cmdline args */
+	for (i = 1; i < argc; i++) {
+		if (!strcmp(argv[i], "-d") || !strcmp(argv[i], "--debug")) {
+			idevice_set_debug_level(1);
+			continue;
+		}
+		else if (!strcmp(argv[i], "-u") || !strcmp(argv[i], "--udid")) {
+			i++;
+			if (!argv[i] || (strlen(argv[i]) != 40)) {
+				print_usage(argc, argv);
+				return 0;
+			}
+			udid = argv[i];
+			continue;
+		}
+		else if (!strcmp(argv[i], "-i") || !strcmp(argv[i], "--install")) {
+			i++;
+			if (!argv[i] || (strlen(argv[i]) < 1)) {
+				print_usage(argc, argv);
+				return 0;
+			}
+			param = argv[i];
+			op = OP_INSTALL;
+			continue;
+		}
+		else if (!strcmp(argv[i], "-l") || !strcmp(argv[i], "--list")) {
+			op = OP_LIST;			
+		}
+		else if (!strcmp(argv[i], "-c") || !strcmp(argv[i], "--copy")) {
+			i++;
+			if (!argv[i] || (strlen(argv[i]) < 1)) {
+				print_usage(argc, argv);
+				return 0;
+			}
+			param = argv[i];
+			op = OP_COPY;
+			continue;
+		}
+		else if (!strcmp(argv[i], "-r") || !strcmp(argv[i], "--remove")) {
+			i++;
+			if (!argv[i] || (strlen(argv[i]) < 1)) {
+				print_usage(argc, argv);
+				return 0;
+			}
+			param = argv[i];
+			op = OP_REMOVE;
+			continue;
+		}
+		else if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "--help")) {
+			print_usage(argc, argv);
+			return 0;
+		}
+		else {
+			print_usage(argc, argv);
+			return 0;
+		}
+	}
+
+	if ((op == -1) || (op >= NUM_OPS)) {
+		print_usage(argc, argv);
+		return 0;
+	}
+
+	ret = idevice_new(&device, udid);
+	if (ret != IDEVICE_E_SUCCESS) {
+		if (udid) {
+			printf("No device found with udid %s, is it plugged in?\n", udid);
+		} else {
+			printf("No device found, is it plugged in?\n");
+		}
+		return -1;
+	}
+
+	if (LOCKDOWN_E_SUCCESS != lockdownd_client_new_with_handshake(device, &client, "ideviceprovision")) {
+		idevice_free(device);
+		return -1;
+	}
+
+	uint16_t port = 0;
+	if (LOCKDOWN_E_SUCCESS != lockdownd_start_service(client, "com.apple.misagent", &port)) {
+		fprintf(stderr, "Could not start service \"com.apple.misagent\"\n");
+		lockdownd_client_free(client);
+		idevice_free(device);
+		return -1;
+	}
+	lockdownd_client_free(client);
+	client = NULL;
+
+	misagent_client_t mis = NULL;
+	if (misagent_client_new(device, port, &mis) != MISAGENT_E_SUCCESS) {
+		fprintf(stderr, "Could not connect to \"com.apple.misagent\" on device\n");
+		lockdownd_client_free(client);
+		idevice_free(device);
+		return -1;
+	}
+
+	switch (op) {
+		case OP_INSTALL:
+		{
+			FILE* f = fopen(param, "rb");
+			if (!f) {
+				fprintf(stderr, "Could not open file '%s'\n", param);
+				break;
+			}
+			fseek(f, 0, SEEK_END);
+			long int size = ftell(f);
+			fseek(f, 0, SEEK_SET);
+
+			if (size >= 0x1000000) {
+				fprintf(stderr, "The file '%s' is too large for processing.\n", param);
+				fclose(f);
+				break;
+			}
+
+			char* buf = malloc(size);
+			if (!buf) {
+				fprintf(stderr, "Could not allocate memory...\n");
+				fclose(f);
+				break;
+			}
+
+			long int cur = 0;
+			while (cur < size) {
+				ssize_t r = fread(buf+cur, 1, 512, f);
+				if (r <= 0) {
+					break;
+				}
+				cur += r;
+			}
+			fclose(f);
+
+			if (cur != size) {
+				free(buf);
+				fprintf(stderr, "Could not read in file '%s' (size %ld read %ld)\n", param, size, cur);
+				break;
+			}
+
+			uint64_t psize = size;
+			plist_t pdata = plist_new_data(buf, psize);
+
+			if (misagent_install(mis, pdata) == MISAGENT_E_SUCCESS) {
+				printf("Profile '%s' installed successfully.\n", param);
+			} else {
+				int sc = misagent_get_status_code(mis);
+				fprintf(stderr, "Could not install profile '%s', status code: 0x%x\n", param, sc);
+			}
+			free(buf);
+		}
+			break;
+		case OP_LIST:
+		case OP_COPY:
+		{
+			plist_t profiles = NULL;
+			if (misagent_copy(mis, &profiles) == MISAGENT_E_SUCCESS) {
+				uint32_t num_profiles = plist_array_get_size(profiles);
+				printf("Device has %d provisioning %s installed:\n", num_profiles, (num_profiles == 1) ? "profile" : "profiles");
+				uint32_t j;
+				for (j = 0; j < num_profiles; j++) {
+					char* p_name = NULL;
+					char* p_uuid = NULL;
+					plist_t profile = plist_array_get_item(profiles, j);
+					plist_t pl = profile_get_embedded_plist(profile);
+					if (pl && (plist_get_node_type(pl) == PLIST_DICT)) {
+						plist_t node;
+						node = plist_dict_get_item(pl, "Name");
+						if (node && (plist_get_node_type(node) == PLIST_STRING)) {
+							plist_get_string_val(node, &p_name);
+						}
+						node = plist_dict_get_item(pl, "UUID");
+						if (node && (plist_get_node_type(node) == PLIST_STRING)) {
+							plist_get_string_val(node, &p_uuid);
+						}
+					}
+					printf("%s - %s\n", (p_uuid) ? p_uuid : "(unknown id)", (p_name) ? p_name : "(no name)");
+					if (op == OP_COPY) {
+						char pfname[512];
+						if (p_uuid) {
+							sprintf(pfname, "%s/%s.mobileprovision", param, p_uuid);
+						} else {
+							sprintf(pfname, "%s/profile%d.mobileprovision", param, j);
+						}
+						FILE* f = fopen(pfname, "wb");
+						if (f) {
+							char* dt = NULL;
+							uint64_t ds = 0;
+							plist_get_data_val(profile, &dt, &ds);
+							fwrite(dt, 1, ds, f);
+							fclose(f);
+							printf(" => %s\n", pfname);
+						} else {
+							fprintf(stderr, "Could not open '%s' for writing\n", pfname);
+						}
+					}
+					if (p_uuid) {
+						free(p_uuid);
+					}
+					if (p_name) {
+						free(p_name);
+					}
+				}
+			} else {
+				int sc = misagent_get_status_code(mis);
+				fprintf(stderr, "Could not get installed profiles from device, status code: 0x%x\n", sc);
+			}
+		}
+			break;
+		case OP_REMOVE:
+			if (misagent_remove(mis, param) == MISAGENT_E_SUCCESS) {
+				printf("Profile '%s' removed.\n", param);
+			} else {
+				int sc = misagent_get_status_code(mis);
+				fprintf(stderr, "Could not remove profile '%s', status code 0x%x\n", param, sc);
+			}
+			break;
+		default:
+			break;
+	}
+
+	misagent_client_free(mis);
+
+	idevice_free(device);
+
+	return 0;
+}
+