From 291032f873225858b9d2d62a65a329da6dfa0d30 Mon Sep 17 00:00:00 2001 From: Calvin Morrison Date: Fri, 17 Jan 2014 15:07:49 -0500 Subject: check argc --- src/strstreamone.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/strstreamone.c') diff --git a/src/strstreamone.c b/src/strstreamone.c index 68da309..000fa64 100644 --- a/src/strstreamone.c +++ b/src/strstreamone.c @@ -16,7 +16,11 @@ int main(int argc, char **argv){ unsigned long long cpy_size = 0; // get max argument length - int save_size = strlen(argv[1]); + if(argc < 2) { + printf("please supply a string to search for in stdin\n"); + exit("EXIT_FAILURE"); + } + unsigned int save_size = strlen(argv[1]); cpy = save_size - 1; cpy_size = BUFSIZ - cpy; -- cgit v1.2.3