diff options
-rw-r--r-- | src/satr.c | 3 | ||||
-rw-r--r-- | src/satrm.c | 3 |
2 files changed, 4 insertions, 2 deletions
@@ -21,6 +21,7 @@ */ #include <stdio.h> #include <stdlib.h> +#include <string.h> #include <errno.h> #include "client.h" @@ -70,7 +71,7 @@ main(int argc, char *argv[]) if (!strcmp(argv[1], "--")) argv++, argc--; for (i = 1; i < argc; i++) - if (argv[i][0] == '-') + if (strchr("-", argv[i][0])) usage(); n = measure_array(argv + 1); diff --git a/src/satrm.c b/src/satrm.c index 8097b18..28bea0b 100644 --- a/src/satrm.c +++ b/src/satrm.c @@ -21,6 +21,7 @@ */ #include <stdio.h> #include <stdlib.h> +#include <string.h> #include <errno.h> #include "client.h" @@ -69,7 +70,7 @@ main(int argc, char *argv[]) if (!strcmp(argv[1], "--") argv++, argc--; for (i = 1; i < argc; i++) - if (argv[i][0] == '-') + if (strchr("-", argv[i][0])) usage(); if (!(msg = malloc(n = measure_array(argv + 1)))) |