Added the option to filerelaytest to pass sources on the command line.

1 files changed, 16 insertions, 2 deletions

diff --git a/dev/filerelaytest.c b/dev/filerelaytest.c
index 9094130..0f2f117 100644
--- a/dev/filerelaytest.c
+++ b/dev/filerelaytest.c
@@ -19,6 +19,7 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA 
  */
 #include <stdio.h>
+#include <stdlib.h>
 #include <libimobiledevice/libimobiledevice.h>
 #include <libimobiledevice/lockdown.h>
 #include <libimobiledevice/file_relay.h>
@@ -41,7 +42,7 @@ int main(int argc, char **argv)
 		goto leave_cleanup;
 	}
 
-	if (lockdownd_start_service(client, "com.apple.mobile.file_relay", &service) != LOCKDOWN_E_SUCCESS) {
+	if (lockdownd_start_service(client, FILE_RELAY_SERVICE_NAME, &service) != LOCKDOWN_E_SUCCESS) {
 		printf("could not start file_relay service!\n");
 		goto leave_cleanup;
 	}
@@ -62,7 +63,20 @@ int main(int argc, char **argv)
 	}
 
 	idevice_connection_t dump = NULL;
-	const char *sources[] = {"AppleSupport", "Network", "VPN", "WiFi", "UserDatabases", "CrashReporter", "tmp", "SystemConfiguration", NULL};
+	const char **sources;
+	const char *default_sources[] = {"AppleSupport", "Network", "VPN", "WiFi", "UserDatabases", "CrashReporter", "tmp", "SystemConfiguration", NULL};
+
+	if (argc > 1) {
+		sources = calloc(1, argc * sizeof(char *));
+		argc--;
+		argv++;
+		for (int i = 0; i < argc; i++) {
+			sources[i] = argv[i];
+		}
+	}
+	else {
+		sources = default_sources;
+	}
 
 	printf("Requesting");
 	int i = 0;