tools/ideviceimagemounter.c - third_party/libimobiledevice - Git at Google

 /*
  * ideviceimagemounter.c
  * Mount developer/debug disk images on the device
  *
  * Copyright (C) 2010 Nikias Bassen <nikias@gmx.li>
  *
  * 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
  */

 #ifdef HAVE_CONFIG_H
 #include <config.h>
 #endif

 #define TOOL_NAME "ideviceimagemounter"

 #include <stdlib.h>
 #define _GNU_SOURCE 1
 #define __USE_GNU 1
 #include <stdio.h>
 #include <string.h>
 #include <getopt.h>
 #include <errno.h>
 #include <libgen.h>
 #include <time.h>
 #include <sys/time.h>
 #include <inttypes.h>
 #ifndef WIN32
 #include <signal.h>
 #endif

 #include <libimobiledevice/libimobiledevice.h>
 #include <libimobiledevice/lockdown.h>
 #include <libimobiledevice/afc.h>
 #include <libimobiledevice/notification_proxy.h>
 #include <libimobiledevice/mobile_image_mounter.h>
 #include <asprintf.h>
 #include <plist/plist.h>

 static int list_mode = 0;
 static int use_network = 0;
 static int xml_mode = 0;
 static const char *udid = NULL;
 static const char *imagetype = NULL;

 static const char PKG_PATH[] = "PublicStaging";
 static const char PATH_PREFIX[] = "/private/var/mobile/Media";

 typedef enum {
 	DISK_IMAGE_UPLOAD_TYPE_AFC,
 	DISK_IMAGE_UPLOAD_TYPE_UPLOAD_IMAGE
 } disk_image_upload_type_t;

 static void print_usage(int argc, char **argv, int is_error)
 {
 	char *name = strrchr(argv[0], '/');
 	fprintf(is_error ? stderr : stdout, "Usage: %s [OPTIONS] IMAGE_FILE IMAGE_SIGNATURE_FILE\n", (name ? name + 1: argv[0]));
 	fprintf(is_error ? stderr : stdout,
 		"\n"
 		"Mounts the specified disk image on the device.\n"
 		"\n"
 		"OPTIONS:\n"
 		"  -u, --udid UDID       target specific device by UDID\n"
 		"  -n, --network         connect to network device\n"
 		"  -l, --list            List mount information\n"
 		"  -t, --imagetype TYPE  Image type to use, default is 'Developer'\n"
 		"  -x, --xml             Use XML output\n"
 		"  -d, --debug           enable communication debugging\n"
 		"  -h, --help            prints usage information\n"
 		"  -v, --version         prints version information\n"
 		"\n"
 		"Homepage:    <" PACKAGE_URL ">\n"
 		"Bug Reports: <" PACKAGE_BUGREPORT ">\n"
 	);
 }

 static void parse_opts(int argc, char **argv)
 {
 	static struct option longopts[] = {
 		{ "help",      no_argument,       NULL, 'h' },
 		{ "udid",      required_argument, NULL, 'u' },
 		{ "network",   no_argument,       NULL, 'n' },
 		{ "list",      no_argument,       NULL, 'l' },
 		{ "imagetype", required_argument, NULL, 't' },
 		{ "xml",       no_argument,       NULL, 'x' },
 		{ "debug",     no_argument,       NULL, 'd' },
 		{ "version",   no_argument,       NULL, 'v' },
 		{ NULL, 0, NULL, 0 }
 	};
 	int c;

 	while (1) {
 		c = getopt_long(argc, argv, "hu:lt:xdnv", longopts, NULL);
 		if (c == -1) {
 			break;
 		}

 		switch (c) {
 		case 'h':
 			print_usage(argc, argv, 0);
 			exit(0);
 		case 'u':
 			if (!*optarg) {
 				fprintf(stderr, "ERROR: UDID must not be empty!\n");
 				print_usage(argc, argv, 1);
 				exit(2);
 			}
 			udid = optarg;
 			break;
 		case 'n':
 			use_network = 1;
 			break;
 		case 'l':
 			list_mode = 1;
 			break;
 		case 't':
 			imagetype = optarg;
 			break;
 		case 'x':
 			xml_mode = 1;
 			break;
 		case 'd':
 			idevice_set_debug_level(1);
 			break;
 		case 'v':
 			printf("%s %s\n", TOOL_NAME, PACKAGE_VERSION);
 			exit(0);
 		default:
 			print_usage(argc, argv, 1);
 			exit(2);
 		}
 	}
 }

 static ssize_t mim_upload_cb(void* buf, size_t size, void* userdata)
 {
 	return fread(buf, 1, size, (FILE*)userdata);
 }

 int main(int argc, char **argv)
 {
 	idevice_t device = NULL;
 	lockdownd_client_t lckd = NULL;
 	lockdownd_error_t ldret = LOCKDOWN_E_UNKNOWN_ERROR;
 	mobile_image_mounter_client_t mim = NULL;
 	afc_client_t afc = NULL;
 	lockdownd_service_descriptor_t service = NULL;
 	int res = -1;
 	char *image_path = NULL;
 	size_t image_size = 0;
 	char *image_sig_path = NULL;

 #ifndef WIN32
 	signal(SIGPIPE, SIG_IGN);
 #endif
 	parse_opts(argc, argv);

 	argc -= optind;
 	argv += optind;

 	if (!list_mode) {
 		if (argc < 1) {
 			printf("ERROR: No IMAGE_FILE has been given!\n");
 			return -1;
 		}
 		image_path = strdup(argv[0]);
 		if (argc >= 2) {
 			image_sig_path = strdup(argv[1]);
 		} else {
 			if (asprintf(&image_sig_path, "%s.signature", image_path) < 0) {
 				printf("Out of memory?!\n");
 				return -1;
 			}
 		}
 	}

 	if (IDEVICE_E_SUCCESS != idevice_new_with_options(&device, udid, (use_network) ? IDEVICE_LOOKUP_NETWORK : IDEVICE_LOOKUP_USBMUX)) {
 		if (udid) {
 			printf("No device found with udid %s.\n", udid);
 		} else {
 			printf("No device found.\n");
 		}
 		return -1;
 	}

 	if (LOCKDOWN_E_SUCCESS != (ldret = lockdownd_client_new_with_handshake(device, &lckd, TOOL_NAME))) {
 		printf("ERROR: Could not connect to lockdown, error code %d.\n", ldret);
 		goto leave;
 	}

 	plist_t pver = NULL;
 	char *product_version = NULL;
 	lockdownd_get_value(lckd, NULL, "ProductVersion", &pver);
 	if (pver && plist_get_node_type(pver) == PLIST_STRING) {
 		plist_get_string_val(pver, &product_version);
 	}
 	disk_image_upload_type_t disk_image_upload_type = DISK_IMAGE_UPLOAD_TYPE_AFC;
 	int product_version_major = 0;
 	int product_version_minor = 0;
 	if (product_version) {
 		if (sscanf(product_version, "%d.%d.%*d", &product_version_major, &product_version_minor) == 2) {
 			if (product_version_major >= 7)
 				disk_image_upload_type = DISK_IMAGE_UPLOAD_TYPE_UPLOAD_IMAGE;
 		}
 	}

 	if (product_version_major == 16) {
 		uint8_t dev_mode_status = 0;
 		plist_t val = NULL;
 		ldret = lockdownd_get_value(lckd, "com.apple.security.mac.amfi", "DeveloperModeStatus", &val);
 		if (ldret == LOCKDOWN_E_SUCCESS) {
 			plist_get_bool_val(val, &dev_mode_status);
 			plist_free(val);
 		}
 		if (!dev_mode_status) {
 			printf("ERROR: You have to enable Developer Mode on the given device in order to allowing mounting a developer disk image.\n");
 			goto leave;
 		}
 	}

 	lockdownd_start_service(lckd, "com.apple.mobile.mobile_image_mounter", &service);

 	if (!service || service->port == 0) {
 		printf("ERROR: Could not start mobile_image_mounter service!\n");
 		goto leave;
 	}

 	if (mobile_image_mounter_new(device, service, &mim) != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
 		printf("ERROR: Could not connect to mobile_image_mounter!\n");
 		goto leave;
 	}

 	if (service) {
 		lockdownd_service_descriptor_free(service);
 		service = NULL;
 	}

 	if (!list_mode) {
 		struct stat fst;
 		if (disk_image_upload_type == DISK_IMAGE_UPLOAD_TYPE_AFC) {
 			if ((lockdownd_start_service(lckd, "com.apple.afc", &service) !=
 				 LOCKDOWN_E_SUCCESS) || !service || !service->port) {
 				fprintf(stderr, "Could not start com.apple.afc!\n");
 				goto leave;
 			}
 			if (afc_client_new(device, service, &afc) != AFC_E_SUCCESS) {
 				fprintf(stderr, "Could not connect to AFC!\n");
 				goto leave;
 			}
 			if (service) {
 				lockdownd_service_descriptor_free(service);
 				service = NULL;
 			}
 		}
 		if (stat(image_path, &fst) != 0) {
 			fprintf(stderr, "ERROR: stat: %s: %s\n", image_path, strerror(errno));
 			goto leave;
 		}
 		image_size = fst.st_size;
 		if (stat(image_sig_path, &fst) != 0) {
 			fprintf(stderr, "ERROR: stat: %s: %s\n", image_sig_path, strerror(errno));
 			goto leave;
 		}
 	}

 	lockdownd_client_free(lckd);
 	lckd = NULL;

 	mobile_image_mounter_error_t err = MOBILE_IMAGE_MOUNTER_E_UNKNOWN_ERROR;
 	plist_t result = NULL;

 	if (list_mode) {
 		/* list mounts mode */
 		if (!imagetype) {
 			imagetype = "Developer";
 		}
 		err = mobile_image_mounter_lookup_image(mim, imagetype, &result);
 		if (err == MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
 			res = 0;
 			plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
 		} else {
 			printf("Error: lookup_image returned %d\n", err);
 		}
 	} else {
 		char sig[8192];
 		size_t sig_length = 0;
 		FILE *f = fopen(image_sig_path, "rb");
 		if (!f) {
 			fprintf(stderr, "Error opening signature file '%s': %s\n", image_sig_path, strerror(errno));
 			goto leave;
 		}
 		sig_length = fread(sig, 1, sizeof(sig), f);
 		fclose(f);
 		if (sig_length == 0) {
 			fprintf(stderr, "Could not read signature from file '%s'\n", image_sig_path);
 			goto leave;
 		}

 		f = fopen(image_path, "rb");
 		if (!f) {
 			fprintf(stderr, "Error opening image file '%s': %s\n", image_path, strerror(errno));
 			goto leave;
 		}

 		char *targetname = NULL;
 		if (asprintf(&targetname, "%s/%s", PKG_PATH, "staging.dimage") < 0) {
 			fprintf(stderr, "Out of memory!?\n");
 			goto leave;
 		}
 		char *mountname = NULL;
 		if (asprintf(&mountname, "%s/%s", PATH_PREFIX, targetname) < 0) {
 			fprintf(stderr, "Out of memory!?\n");
 			goto leave;
 		}


 		if (!imagetype) {
 			imagetype = "Developer";
 		}

 		switch(disk_image_upload_type) {
 			case DISK_IMAGE_UPLOAD_TYPE_UPLOAD_IMAGE:
 				printf("Uploading %s\n", image_path);
 				err = mobile_image_mounter_upload_image(mim, imagetype, image_size, sig, sig_length, mim_upload_cb, f);
 				break;
 			case DISK_IMAGE_UPLOAD_TYPE_AFC:
 			default:
 				printf("Uploading %s --> afc:///%s\n", image_path, targetname);
 				char **strs = NULL;
 				if (afc_get_file_info(afc, PKG_PATH, &strs) != AFC_E_SUCCESS) {
 					if (afc_make_directory(afc, PKG_PATH) != AFC_E_SUCCESS) {
 						fprintf(stderr, "WARNING: Could not create directory '%s' on device!\n", PKG_PATH);
 					}
 				}
 				if (strs) {
 					int i = 0;
 					while (strs[i]) {
 						free(strs[i]);
 						i++;
 					}
 					free(strs);
 				}

 				uint64_t af = 0;
 				if ((afc_file_open(afc, targetname, AFC_FOPEN_WRONLY, &af) !=
 					 AFC_E_SUCCESS) || !af) {
 					fclose(f);
 					fprintf(stderr, "afc_file_open on '%s' failed!\n", targetname);
 					goto leave;
 				}

 				char buf[8192];
 				size_t amount = 0;
 				do {
 					amount = fread(buf, 1, sizeof(buf), f);
 					if (amount > 0) {
 						uint32_t written, total = 0;
 						while (total < amount) {
 							written = 0;
 							if (afc_file_write(afc, af, buf + total, amount - total, &written) !=
 								AFC_E_SUCCESS) {
 								fprintf(stderr, "AFC Write error!\n");
 								break;
 							}
 							total += written;
 						}
 						if (total != amount) {
 							fprintf(stderr, "Error: wrote only %d of %d\n", total,
 									(unsigned int)amount);
 							afc_file_close(afc, af);
 							fclose(f);
 							goto leave;
 						}
 					}
 				}
 				while (amount > 0);

 				afc_file_close(afc, af);
 				break;
 		}

 		fclose(f);

 		if (err != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
 			if (err == MOBILE_IMAGE_MOUNTER_E_DEVICE_LOCKED) {
 				printf("ERROR: Device is locked, can't mount. Unlock device and try again.\n");
 			} else {
 				printf("ERROR: Unknown error occurred, can't mount.\n");
 			}
 			goto error_out;
 		}
 		printf("done.\n");

 		printf("Mounting...\n");
 		err = mobile_image_mounter_mount_image(mim, mountname, sig, sig_length, imagetype, &result);
 		if (err == MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
 			if (result) {
 				plist_t node = plist_dict_get_item(result, "Status");
 				if (node) {
 					char *status = NULL;
 					plist_get_string_val(node, &status);
 					if (status) {
 						if (!strcmp(status, "Complete")) {
 							printf("Done.\n");
 							res = 0;
 						} else {
 							printf("unexpected status value:\n");
 							plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
 						}
 						free(status);
 					} else {
 						printf("unexpected result:\n");
 						plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
 					}
 				}
 				node = plist_dict_get_item(result, "Error");
 				if (node) {
 					char *error = NULL;
 					plist_get_string_val(node, &error);
 					if (error) {
 						printf("Error: %s\n", error);
 						free(error);
 					} else {
 						printf("unexpected result:\n");
 						plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
 					}

 				} else {
 					plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
 				}
 			}
 		} else {
 			printf("Error: mount_image returned %d\n", err);

 		}
 	}

 	if (result) {
 		plist_free(result);
 	}

 error_out:
 	/* perform hangup command */
 	mobile_image_mounter_hangup(mim);
 	/* free client */
 	mobile_image_mounter_free(mim);

 leave:
 	if (afc) {
 		afc_client_free(afc);
 	}
 	if (lckd) {
 		lockdownd_client_free(lckd);
 	}
 	idevice_free(device);

 	if (image_path)
 			free(image_path);
 	if (image_sig_path)
 		free(image_sig_path);

 	return res;
 }
	/*
	* ideviceimagemounter.c
	* Mount developer/debug disk images on the device
	*
	* Copyright (C) 2010 Nikias Bassen <nikias@gmx.li>
	*
	* 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
	*/

	#ifdef HAVE_CONFIG_H
	#include <config.h>
	#endif

	#define TOOL_NAME "ideviceimagemounter"

	#include <stdlib.h>
	#define _GNU_SOURCE 1
	#define __USE_GNU 1
	#include <stdio.h>
	#include <string.h>
	#include <getopt.h>
	#include <errno.h>
	#include <libgen.h>
	#include <time.h>
	#include <sys/time.h>
	#include <inttypes.h>
	#ifndef WIN32
	#include <signal.h>
	#endif

	#include <libimobiledevice/libimobiledevice.h>
	#include <libimobiledevice/lockdown.h>
	#include <libimobiledevice/afc.h>
	#include <libimobiledevice/notification_proxy.h>
	#include <libimobiledevice/mobile_image_mounter.h>
	#include <asprintf.h>
	#include <plist/plist.h>

	static int list_mode = 0;
	static int use_network = 0;
	static int xml_mode = 0;
	static const char *udid = NULL;
	static const char *imagetype = NULL;

	static const char PKG_PATH[] = "PublicStaging";
	static const char PATH_PREFIX[] = "/private/var/mobile/Media";

	typedef enum {
	DISK_IMAGE_UPLOAD_TYPE_AFC,
	DISK_IMAGE_UPLOAD_TYPE_UPLOAD_IMAGE
	} disk_image_upload_type_t;

	static void print_usage(int argc, char **argv, int is_error)
	{
	char *name = strrchr(argv[0], '/');
	fprintf(is_error ? stderr : stdout, "Usage: %s [OPTIONS] IMAGE_FILE IMAGE_SIGNATURE_FILE\n", (name ? name + 1: argv[0]));
	fprintf(is_error ? stderr : stdout,
	"\n"
	"Mounts the specified disk image on the device.\n"
	"\n"
	"OPTIONS:\n"
	" -u, --udid UDID target specific device by UDID\n"
	" -n, --network connect to network device\n"
	" -l, --list List mount information\n"
	" -t, --imagetype TYPE Image type to use, default is 'Developer'\n"
	" -x, --xml Use XML output\n"
	" -d, --debug enable communication debugging\n"
	" -h, --help prints usage information\n"
	" -v, --version prints version information\n"
	"\n"
	"Homepage: <" PACKAGE_URL ">\n"
	"Bug Reports: <" PACKAGE_BUGREPORT ">\n"
	);
	}

	static void parse_opts(int argc, char **argv)
	{
	static struct option longopts[] = {
	{ "help", no_argument, NULL, 'h' },
	{ "udid", required_argument, NULL, 'u' },
	{ "network", no_argument, NULL, 'n' },
	{ "list", no_argument, NULL, 'l' },
	{ "imagetype", required_argument, NULL, 't' },
	{ "xml", no_argument, NULL, 'x' },
	{ "debug", no_argument, NULL, 'd' },
	{ "version", no_argument, NULL, 'v' },
	{ NULL, 0, NULL, 0 }
	};
	int c;

	while (1) {
	c = getopt_long(argc, argv, "hu:lt:xdnv", longopts, NULL);
	if (c == -1) {
	break;
	}

	switch (c) {
	case 'h':
	print_usage(argc, argv, 0);
	exit(0);
	case 'u':
	if (!*optarg) {
	fprintf(stderr, "ERROR: UDID must not be empty!\n");
	print_usage(argc, argv, 1);
	exit(2);
	}
	udid = optarg;
	break;
	case 'n':
	use_network = 1;
	break;
	case 'l':
	list_mode = 1;
	break;
	case 't':
	imagetype = optarg;
	break;
	case 'x':
	xml_mode = 1;
	break;
	case 'd':
	idevice_set_debug_level(1);
	break;
	case 'v':
	printf("%s %s\n", TOOL_NAME, PACKAGE_VERSION);
	exit(0);
	default:
	print_usage(argc, argv, 1);
	exit(2);
	}
	}
	}

	static ssize_t mim_upload_cb(void* buf, size_t size, void* userdata)
	{
	return fread(buf, 1, size, (FILE*)userdata);
	}

	int main(int argc, char **argv)
	{
	idevice_t device = NULL;
	lockdownd_client_t lckd = NULL;
	lockdownd_error_t ldret = LOCKDOWN_E_UNKNOWN_ERROR;
	mobile_image_mounter_client_t mim = NULL;
	afc_client_t afc = NULL;
	lockdownd_service_descriptor_t service = NULL;
	int res = -1;
	char *image_path = NULL;
	size_t image_size = 0;
	char *image_sig_path = NULL;

	#ifndef WIN32
	signal(SIGPIPE, SIG_IGN);
	#endif
	parse_opts(argc, argv);

	argc -= optind;
	argv += optind;

	if (!list_mode) {
	if (argc < 1) {
	printf("ERROR: No IMAGE_FILE has been given!\n");
	return -1;
	}
	image_path = strdup(argv[0]);
	if (argc >= 2) {
	image_sig_path = strdup(argv[1]);
	} else {
	if (asprintf(&image_sig_path, "%s.signature", image_path) < 0) {
	printf("Out of memory?!\n");
	return -1;
	}
	}
	}

	if (IDEVICE_E_SUCCESS != idevice_new_with_options(&device, udid, (use_network) ? IDEVICE_LOOKUP_NETWORK : IDEVICE_LOOKUP_USBMUX)) {
	if (udid) {
	printf("No device found with udid %s.\n", udid);
	} else {
	printf("No device found.\n");
	}
	return -1;
	}

	if (LOCKDOWN_E_SUCCESS != (ldret = lockdownd_client_new_with_handshake(device, &lckd, TOOL_NAME))) {
	printf("ERROR: Could not connect to lockdown, error code %d.\n", ldret);
	goto leave;
	}

	plist_t pver = NULL;
	char *product_version = NULL;
	lockdownd_get_value(lckd, NULL, "ProductVersion", &pver);
	if (pver && plist_get_node_type(pver) == PLIST_STRING) {
	plist_get_string_val(pver, &product_version);
	}
	disk_image_upload_type_t disk_image_upload_type = DISK_IMAGE_UPLOAD_TYPE_AFC;
	int product_version_major = 0;
	int product_version_minor = 0;
	if (product_version) {
	if (sscanf(product_version, "%d.%d.%*d", &product_version_major, &product_version_minor) == 2) {
	if (product_version_major >= 7)
	disk_image_upload_type = DISK_IMAGE_UPLOAD_TYPE_UPLOAD_IMAGE;
	}
	}

	if (product_version_major == 16) {
	uint8_t dev_mode_status = 0;
	plist_t val = NULL;
	ldret = lockdownd_get_value(lckd, "com.apple.security.mac.amfi", "DeveloperModeStatus", &val);
	if (ldret == LOCKDOWN_E_SUCCESS) {
	plist_get_bool_val(val, &dev_mode_status);
	plist_free(val);
	}
	if (!dev_mode_status) {
	printf("ERROR: You have to enable Developer Mode on the given device in order to allowing mounting a developer disk image.\n");
	goto leave;
	}
	}

	lockdownd_start_service(lckd, "com.apple.mobile.mobile_image_mounter", &service);

	if (!service \|\| service->port == 0) {
	printf("ERROR: Could not start mobile_image_mounter service!\n");
	goto leave;
	}

	if (mobile_image_mounter_new(device, service, &mim) != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
	printf("ERROR: Could not connect to mobile_image_mounter!\n");
	goto leave;
	}

	if (service) {
	lockdownd_service_descriptor_free(service);
	service = NULL;
	}

	if (!list_mode) {
	struct stat fst;
	if (disk_image_upload_type == DISK_IMAGE_UPLOAD_TYPE_AFC) {
	if ((lockdownd_start_service(lckd, "com.apple.afc", &service) !=
	LOCKDOWN_E_SUCCESS) \|\| !service \|\| !service->port) {
	fprintf(stderr, "Could not start com.apple.afc!\n");
	goto leave;
	}
	if (afc_client_new(device, service, &afc) != AFC_E_SUCCESS) {
	fprintf(stderr, "Could not connect to AFC!\n");
	goto leave;
	}
	if (service) {
	lockdownd_service_descriptor_free(service);
	service = NULL;
	}
	}
	if (stat(image_path, &fst) != 0) {
	fprintf(stderr, "ERROR: stat: %s: %s\n", image_path, strerror(errno));
	goto leave;
	}
	image_size = fst.st_size;
	if (stat(image_sig_path, &fst) != 0) {
	fprintf(stderr, "ERROR: stat: %s: %s\n", image_sig_path, strerror(errno));
	goto leave;
	}
	}

	lockdownd_client_free(lckd);
	lckd = NULL;

	mobile_image_mounter_error_t err = MOBILE_IMAGE_MOUNTER_E_UNKNOWN_ERROR;
	plist_t result = NULL;

	if (list_mode) {
	/* list mounts mode */
	if (!imagetype) {
	imagetype = "Developer";
	}
	err = mobile_image_mounter_lookup_image(mim, imagetype, &result);
	if (err == MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
	res = 0;
	plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
	} else {
	printf("Error: lookup_image returned %d\n", err);
	}
	} else {
	char sig[8192];
	size_t sig_length = 0;
	FILE *f = fopen(image_sig_path, "rb");
	if (!f) {
	fprintf(stderr, "Error opening signature file '%s': %s\n", image_sig_path, strerror(errno));
	goto leave;
	}
	sig_length = fread(sig, 1, sizeof(sig), f);
	fclose(f);
	if (sig_length == 0) {
	fprintf(stderr, "Could not read signature from file '%s'\n", image_sig_path);
	goto leave;
	}

	f = fopen(image_path, "rb");
	if (!f) {
	fprintf(stderr, "Error opening image file '%s': %s\n", image_path, strerror(errno));
	goto leave;
	}

	char *targetname = NULL;
	if (asprintf(&targetname, "%s/%s", PKG_PATH, "staging.dimage") < 0) {
	fprintf(stderr, "Out of memory!?\n");
	goto leave;
	}
	char *mountname = NULL;
	if (asprintf(&mountname, "%s/%s", PATH_PREFIX, targetname) < 0) {
	fprintf(stderr, "Out of memory!?\n");
	goto leave;
	}


	if (!imagetype) {
	imagetype = "Developer";
	}

	switch(disk_image_upload_type) {
	case DISK_IMAGE_UPLOAD_TYPE_UPLOAD_IMAGE:
	printf("Uploading %s\n", image_path);
	err = mobile_image_mounter_upload_image(mim, imagetype, image_size, sig, sig_length, mim_upload_cb, f);
	break;
	case DISK_IMAGE_UPLOAD_TYPE_AFC:
	default:
	printf("Uploading %s --> afc:///%s\n", image_path, targetname);
	char **strs = NULL;
	if (afc_get_file_info(afc, PKG_PATH, &strs) != AFC_E_SUCCESS) {
	if (afc_make_directory(afc, PKG_PATH) != AFC_E_SUCCESS) {
	fprintf(stderr, "WARNING: Could not create directory '%s' on device!\n", PKG_PATH);
	}
	}
	if (strs) {
	int i = 0;
	while (strs[i]) {
	free(strs[i]);
	i++;
	}
	free(strs);
	}

	uint64_t af = 0;
	if ((afc_file_open(afc, targetname, AFC_FOPEN_WRONLY, &af) !=
	AFC_E_SUCCESS) \|\| !af) {
	fclose(f);
	fprintf(stderr, "afc_file_open on '%s' failed!\n", targetname);
	goto leave;
	}

	char buf[8192];
	size_t amount = 0;
	do {
	amount = fread(buf, 1, sizeof(buf), f);
	if (amount > 0) {
	uint32_t written, total = 0;
	while (total < amount) {
	written = 0;
	if (afc_file_write(afc, af, buf + total, amount - total, &written) !=
	AFC_E_SUCCESS) {
	fprintf(stderr, "AFC Write error!\n");
	break;
	}
	total += written;
	}
	if (total != amount) {
	fprintf(stderr, "Error: wrote only %d of %d\n", total,
	(unsigned int)amount);
	afc_file_close(afc, af);
	fclose(f);
	goto leave;
	}
	}
	}
	while (amount > 0);

	afc_file_close(afc, af);
	break;
	}

	fclose(f);

	if (err != MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
	if (err == MOBILE_IMAGE_MOUNTER_E_DEVICE_LOCKED) {
	printf("ERROR: Device is locked, can't mount. Unlock device and try again.\n");
	} else {
	printf("ERROR: Unknown error occurred, can't mount.\n");
	}
	goto error_out;
	}
	printf("done.\n");

	printf("Mounting...\n");
	err = mobile_image_mounter_mount_image(mim, mountname, sig, sig_length, imagetype, &result);
	if (err == MOBILE_IMAGE_MOUNTER_E_SUCCESS) {
	if (result) {
	plist_t node = plist_dict_get_item(result, "Status");
	if (node) {
	char *status = NULL;
	plist_get_string_val(node, &status);
	if (status) {
	if (!strcmp(status, "Complete")) {
	printf("Done.\n");
	res = 0;
	} else {
	printf("unexpected status value:\n");
	plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
	}
	free(status);
	} else {
	printf("unexpected result:\n");
	plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
	}
	}
	node = plist_dict_get_item(result, "Error");
	if (node) {
	char *error = NULL;
	plist_get_string_val(node, &error);
	if (error) {
	printf("Error: %s\n", error);
	free(error);
	} else {
	printf("unexpected result:\n");
	plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
	}

	} else {
	plist_write_to_stream(result, stdout, (xml_mode) ? PLIST_FORMAT_XML : PLIST_FORMAT_LIMD, 0);
	}
	}
	} else {
	printf("Error: mount_image returned %d\n", err);

	}
	}

	if (result) {
	plist_free(result);
	}

	error_out:
	/* perform hangup command */
	mobile_image_mounter_hangup(mim);
	/* free client */
	mobile_image_mounter_free(mim);

	leave:
	if (afc) {
	afc_client_free(afc);
	}
	if (lckd) {
	lockdownd_client_free(lckd);
	}
	idevice_free(device);

	if (image_path)
	free(image_path);
	if (image_sig_path)
	free(image_sig_path);

	return res;
	}