From 6f511706f3fc6dc1ab301cb63ad1853f489ab3d3 Mon Sep 17 00:00:00 2001 From: David Phillips Date: Thu, 27 Apr 2017 18:56:46 +1200 Subject: Correct command line argument checking --- sand-leek.c | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/sand-leek.c b/sand-leek.c index 4327f4d..5074adb 100644 --- a/sand-leek.c +++ b/sand-leek.c @@ -137,6 +137,16 @@ STOP: return NULL; } +void +die_usage(const char *argv0) { + fprintf(stderr, + "usage: %s [-t threads] -s search\n" + "searches for keys for onion addresses beginning with `search`\n", + argv0 + ); + exit(1); +} + int main(int argc, char **argv) { char opt = '\0'; @@ -158,7 +168,11 @@ main(int argc, char **argv) { } if (thread_count <= 0) { - fprintf(stderr, "thread count must be > 0\n"); + die_usage(argv[0]); + } + + if (search == NULL || strlen(search) <= 0) { + die_usage(argv[0]); } search_len = strlen(search); -- cgit v1.1