| /* |
| * idevicenotificationproxy.c |
| * Simple client for the notification_proxy service |
| * |
| * Copyright (c) 2009-2015 Martin Szulecki 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 |
| */ |
| |
| #ifdef HAVE_CONFIG_H |
| #include <config.h> |
| #endif |
| |
| #define TOOL_NAME "idevicenotificationproxy" |
| |
| #include <stdio.h> |
| #include <string.h> |
| #include <errno.h> |
| #include <signal.h> |
| #include <stdlib.h> |
| |
| #ifdef WIN32 |
| #include <windows.h> |
| #define sleep(x) Sleep(x*1000) |
| #else |
| #include <unistd.h> |
| #endif |
| |
| #include <libimobiledevice/libimobiledevice.h> |
| #include <libimobiledevice/lockdown.h> |
| #include <libimobiledevice/notification_proxy.h> |
| |
| enum cmd_mode { |
| CMD_NONE = 0, |
| CMD_OBSERVE, |
| CMD_POST |
| }; |
| |
| static int quit_flag = 0; |
| |
| /** |
| * signal handler function for cleaning up properly |
| */ |
| static void clean_exit(int sig) |
| { |
| fprintf(stderr, "Exiting...\n"); |
| quit_flag++; |
| } |
| |
| static void print_usage(int argc, char **argv) |
| { |
| char *name = NULL; |
| |
| name = strrchr(argv[0], '/'); |
| printf("Usage: %s [OPTIONS] COMMAND\n", (name ? name + 1: argv[0])); |
| printf("\n"); |
| printf("Post or observe notifications on a device.\n"); |
| printf("\n"); |
| printf("Where COMMAND is one of:\n"); |
| printf(" post ID [...]\t\tpost notification IDs to device and exit\n"); |
| printf(" observe ID [...]\tobserve notification IDs in the foreground until CTRL+C or signal is received\n"); |
| printf("\n"); |
| printf("The following OPTIONS are accepted:\n"); |
| printf(" -u, --udid UDID\ttarget specific device by UDID\n"); |
| printf(" -n, --network\t\tconnect to network device\n"); |
| printf(" -d, --debug\t\tenable communication debugging\n"); |
| printf(" -h, --help\t\tprints usage information\n"); |
| printf(" -v, --version\t\tprints version information\n"); |
| printf("\n"); |
| printf("Homepage: <" PACKAGE_URL ">\n"); |
| printf("Bug Reports: <" PACKAGE_BUGREPORT ">\n"); |
| } |
| |
| static void notify_cb(const char *notification, void *user_data) |
| { |
| printf("> %s\n", notification); |
| } |
| |
| int main(int argc, char *argv[]) |
| { |
| lockdownd_error_t ret = LOCKDOWN_E_UNKNOWN_ERROR; |
| lockdownd_service_descriptor_t service = NULL; |
| lockdownd_client_t client = NULL; |
| idevice_t device = NULL; |
| np_client_t gnp = NULL; |
| |
| int result = -1; |
| int i; |
| const char* udid = NULL; |
| int use_network = 0; |
| int cmd = CMD_NONE; |
| char* cmd_arg = NULL; |
| |
| int count = 0; |
| char **nspec = NULL; |
| char **nspectmp = NULL; |
| |
| signal(SIGINT, clean_exit); |
| signal(SIGTERM, clean_exit); |
| #ifndef WIN32 |
| signal(SIGQUIT, clean_exit); |
| signal(SIGPIPE, SIG_IGN); |
| #endif |
| |
| /* 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] || !*argv[i]) { |
| print_usage(argc, argv); |
| result = 0; |
| goto cleanup; |
| } |
| udid = argv[i]; |
| continue; |
| } |
| else if (!strcmp(argv[i], "-h") || !strcmp(argv[i], "--help")) { |
| print_usage(argc, argv); |
| result = 0; |
| goto cleanup; |
| } |
| else if (!strcmp(argv[i], "-n") || !strcmp(argv[i], "--network")) { |
| use_network = 1; |
| continue; |
| } |
| else if (!strcmp(argv[i], "-v") || !strcmp(argv[i], "--version")) { |
| printf("%s %s\n", TOOL_NAME, PACKAGE_VERSION); |
| result = 0; |
| goto cleanup; |
| } |
| else if (!strcmp(argv[i], "post") || !strcmp(argv[i], "observe")) { |
| cmd = CMD_POST; |
| if (!strcmp(argv[i], "observe")) { |
| cmd = CMD_OBSERVE; |
| } |
| |
| i++; |
| |
| if (!argv[i] || argv[i] == NULL || (!strncmp(argv[i], "-", 1))) { |
| printf("Please supply a valid notification identifier.\n"); |
| print_usage(argc, argv); |
| goto cleanup; |
| } |
| |
| count = 0; |
| nspec = malloc(sizeof(char*) * (count+1)); |
| |
| while(1) { |
| if (argv[i] && (strlen(argv[i]) >= 2) && (strncmp(argv[i], "-", 1) != 0)) { |
| nspectmp = realloc(nspec, sizeof(char*) * (count+1)); |
| nspectmp[count] = strdup(argv[i]); |
| nspec = nspectmp; |
| count = count+1; |
| i++; |
| } else { |
| i--; |
| break; |
| } |
| } |
| |
| nspectmp = realloc(nspec, sizeof(char*) * (count+1)); |
| nspectmp[count] = NULL; |
| nspec = nspectmp; |
| continue; |
| } |
| else { |
| print_usage(argc, argv); |
| return 0; |
| } |
| } |
| |
| /* verify options */ |
| if (cmd == CMD_NONE) { |
| print_usage(argc, argv); |
| goto cleanup; |
| } |
| |
| 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"); |
| } |
| goto cleanup; |
| } |
| |
| if (LOCKDOWN_E_SUCCESS != (ret = lockdownd_client_new_with_handshake(device, &client, TOOL_NAME))) { |
| fprintf(stderr, "ERROR: Could not connect to lockdownd, error code %d\n", ret); |
| goto cleanup; |
| } |
| |
| ret = lockdownd_start_service(client, NP_SERVICE_NAME, &service); |
| |
| lockdownd_client_free(client); |
| |
| if ((ret == LOCKDOWN_E_SUCCESS) && (service->port > 0)) { |
| if (np_client_new(device, service, &gnp) != NP_E_SUCCESS) { |
| printf("Could not connect to notification_proxy!\n"); |
| result = -1; |
| } else { |
| np_set_notify_callback(gnp, notify_cb, NULL); |
| |
| switch (cmd) { |
| case CMD_POST: |
| i = 0; |
| while(nspec[i] != NULL && i < (count+1)) { |
| printf("< posting \"%s\"\n", nspec[i]); |
| np_post_notification(gnp, nspec[i]); |
| i++; |
| } |
| break; |
| case CMD_OBSERVE: |
| default: |
| i = 0; |
| while(nspec[i] != NULL && i < (count+1)) { |
| printf("! observing \"%s\"\n", nspec[i]); |
| np_observe_notification(gnp, nspec[i]); |
| i++; |
| } |
| |
| /* just sleep and wait for notifications */ |
| while (!quit_flag) { |
| sleep(1); |
| } |
| |
| break; |
| } |
| |
| result = EXIT_SUCCESS; |
| |
| if (gnp) { |
| np_client_free(gnp); |
| gnp = NULL; |
| } |
| } |
| } else { |
| printf("ERROR: Could not start service %s: %s\n", NP_SERVICE_NAME, lockdownd_strerror(ret)); |
| } |
| |
| if (service) { |
| lockdownd_service_descriptor_free(service); |
| service = NULL; |
| } |
| |
| cleanup: |
| if (nspec) { |
| i = 0; |
| while(nspec[i] != NULL && i < (count+1)) { |
| free(nspec[i]); |
| i++; |
| } |
| free(nspec); |
| } |
| |
| if (cmd_arg) { |
| free(cmd_arg); |
| } |
| |
| if (device) |
| idevice_free(device); |
| |
| return result; |
| } |