X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/674115e8e5e4afdcbb67ef088efb2c88612a8839..47f55bc70a0eef068c8a27ca0618a11c512dbf20:/Console.cpp diff --git a/Console.cpp b/Console.cpp index d91e6d4..f93e090 100644 --- a/Console.cpp +++ b/Console.cpp @@ -616,6 +616,7 @@ int Main(int argc, char const * const argv[], char const * const envp[]) { bool tty(isatty(STDIN_FILENO)); bool compile(false); + bool target(false); CYOptions options; append_history$ = (int (*)(int, const char *)) (dlsym(RTLD_DEFAULT, "append_history")); @@ -631,17 +632,19 @@ int Main(int argc, char const * const argv[], char const * const envp[]) { _aprcall(apr_getopt_init(&state, pool, argc, argv)); for (;;) { - char opt; + int opt; const char *arg; - apr_status_t status(apr_getopt(state, - "cg:n:" + apr_status_t status(apr_getopt_long(state, (apr_getopt_option_t[]) { + {NULL, 'c', false, NULL}, + {NULL, 'g', true, NULL}, + {NULL, 'n', true, NULL}, #ifdef CY_ATTACH - "p:" + {NULL, 'p', true, NULL}, #endif - "r:" - "s" - , &opt, &arg)); + {NULL, 'r', true, NULL}, + {NULL, 's', false, NULL}, + {0, 0, 0, 0}}, &opt, &arg)); switch (status) { case APR_EOF: @@ -662,9 +665,22 @@ int Main(int argc, char const * const argv[], char const * const envp[]) { } switch (opt) { + target: + if (!target) + target = true; + else { + fprintf(stderr, "only one of -[c" +#ifdef CY_ATTACH + "p" +#endif + "r] may be used at a time\n"); + return 1; + } + break; + case 'c': compile = true; - break; + goto target; case 'g': if (false); @@ -736,7 +752,7 @@ int Main(int argc, char const * const argv[], char const * const envp[]) { return 1; } } - } break; + } goto target; #endif case 'r': { @@ -758,7 +774,7 @@ int Main(int argc, char const * const argv[], char const * const envp[]) { host = arg; *colon = '\0'; port = colon + 1; - } break; + } goto target; case 's': strict_ = true; @@ -774,11 +790,6 @@ int Main(int argc, char const * const argv[], char const * const envp[]) { fprintf(stderr, "-p cannot set argv\n"); return 1; } - - if (pid != _not(pid_t) && compile) { - fprintf(stderr, "-p conflicts with -c\n"); - return 1; - } #endif if (ind == argc)