tools: add convenience tool 'idevicename' to get or set device name

2 files changed, 126 insertions, 1 deletions

diff --git a/tools/Makefile.am b/tools/Makefile.am
index 8c29487..c280715 100644
--- a/tools/Makefile.am
+++ b/tools/Makefile.am
@@ -3,13 +3,18 @@ AM_CPPFLAGS = -I$(top_srcdir)/include
 AM_CFLAGS = $(GLOBAL_CFLAGS) $(libgnutls_CFLAGS) $(libtasn1_CFLAGS) $(libgcrypt_CFLAGS) $(openssl_CFLAGS) $(libplist_CFLAGS) $(LFS_CFLAGS)
 AM_LDFLAGS = $(libgnutls_LIBS) $(libtasn1_LIBS) $(libgcrypt_LIBS) $(openssl_LIBS) $(libplist_LIBS)
 
-bin_PROGRAMS = idevice_id ideviceinfo idevicepair idevicesyslog ideviceimagemounter idevicescreenshot ideviceenterrecovery idevicedate idevicebackup idevicebackup2 ideviceprovision idevicedebugserverproxy idevicediagnostics
+bin_PROGRAMS = idevice_id ideviceinfo idevicename idevicepair idevicesyslog ideviceimagemounter idevicescreenshot ideviceenterrecovery idevicedate idevicebackup idevicebackup2 ideviceprovision idevicedebugserverproxy idevicediagnostics
 
 ideviceinfo_SOURCES = ideviceinfo.c
 ideviceinfo_CFLAGS = $(AM_CFLAGS)
 ideviceinfo_LDFLAGS = $(AM_LDFLAGS)
 ideviceinfo_LDADD = ../src/libimobiledevice.la
 
+idevicename_SOURCES = idevicename.c
+idevicename_CFLAGS = $(AM_CFLAGS)
+idevicename_LDFLAGS = $(AM_LDFLAGS)
+idevicename_LDADD = ../src/libimobiledevice.la
+
 idevicepair_SOURCES = idevicepair.c
 idevicepair_CFLAGS = $(AM_CFLAGS)
 idevicepair_LDFLAGS = $(top_srcdir)/common/libinternalcommon.la $(AM_LDFLAGS)
diff --git a/tools/idevicename.c b/tools/idevicename.c
new file mode 100644
index 0000000..a991a8f
--- /dev/null
+++ b/tools/idevicename.c
@@ -0,0 +1,120 @@
+/*
+ * idevicename.c
+ * Simple utility to get or set the device name
+ *
+ * Copyright (c) 2014  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 <string.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <getopt.h>
+
+#include <libimobiledevice/libimobiledevice.h>
+#include <libimobiledevice/lockdown.h>
+
+static void print_usage()
+{
+	printf("\nUsage: idevicename [OPTIONS] [NAME]\n");
+	printf("  --udid|-u UDID  use UDID to target a specific device\n");
+	printf("\n");
+}
+
+int main(int argc, char** argv)
+{
+	int res = -1;
+	char* udid = NULL;
+
+	int c = 0;
+	int optidx = 0;
+	const struct option longopts[] = {
+		{ "udid", required_argument, NULL, 'u' },
+		{ NULL, 0, NULL, 0}
+	};
+
+	signal(SIGPIPE, SIG_IGN);
+
+	while ((c = getopt_long(argc, argv, "u:", longopts, &optidx)) != -1) {
+		switch (c) {
+		case 'u':
+			udid = strdup(optarg);
+			break;
+		default:
+			break;
+		}
+	}
+
+	argc -= optind;
+	argv += optind;
+
+	if (argc > 1) {
+		print_usage();
+		return -1;
+	}
+
+	idevice_t device = NULL;
+	if (idevice_new(&device, udid) != IDEVICE_E_SUCCESS) {
+		fprintf(stderr, "ERROR: Could not connect to device\n");
+		return -1;
+	}
+
+	lockdownd_client_t lockdown = NULL;
+	lockdownd_error_t lerr = lockdownd_client_new_with_handshake(device, &lockdown, "idevicename");
+	if (lerr != LOCKDOWN_E_SUCCESS) {
+		idevice_free(device);
+		fprintf(stderr, "ERROR: lockdown connection failed, lockdown error %d\n", lerr);
+		return -1;
+	}
+
+	plist_t node = NULL;
+
+	if (argc == 0) {
+		// getting device name
+		char* name = NULL;
+		lerr = lockdownd_get_value(lockdown, NULL, "DeviceName", &node);
+		if (node) {
+			plist_get_string_val(node, &name);
+			plist_free(node);
+		}
+		if (name) {
+			printf("%s\n", name);
+			free(name);
+			res = 0;
+		} else {
+			fprintf(stderr, "ERROR: Could not get device name, lockdown error %d\n", lerr);
+		}
+	} else {
+		// setting device name
+		lerr = lockdownd_set_value(lockdown, NULL, "DeviceName", plist_new_string(argv[0]));
+		if (lerr == LOCKDOWN_E_SUCCESS) {
+			printf("device name set to '%s'\n", argv[0]);
+			res = 0;
+		} else {
+			fprintf(stderr, "ERROR: Could not set device name, lockdown error %d\n", lerr);
+		}
+	}
+
+	lockdownd_client_free(lockdown);
+	idevice_free(device);
+
+	if (udid) {
+		free(udid);
+	}
+
+	return res;
+}