From eec2f63b3f6117256e0d8f328e40a63bcab84bb4 Mon Sep 17 00:00:00 2001 From: Mark Whitley Date: Tue, 6 Jun 2000 18:11:46 +0000 Subject: [PATCH] Fixed segfaults for "chown -R" and "chgrp -R". Also added a message for "too few arguments". --- chmod_chown_chgrp.c | 22 +++++++++++++--------- messages.c | 3 +++ 2 files changed, 16 insertions(+), 9 deletions(-) diff --git a/chmod_chown_chgrp.c b/chmod_chown_chgrp.c index c0e380a8e..a3af4fbb3 100644 --- a/chmod_chown_chgrp.c +++ b/chmod_chown_chgrp.c @@ -25,6 +25,7 @@ #include "internal.h" #define BB_DECLARE_EXTERN #define bb_need_invalid_option +#define bb_need_too_few_args #include "messages.c" #include @@ -114,24 +115,27 @@ int chmod_chown_chgrp_main(int argc, char **argv) if (argc < 2) usage(appUsage); invocationName = *argv; - argc--; argv++; /* Parse options */ - while (argc && (**argv == '-')) { + while (--argc >= 0 && *argv && (**argv == '-')) { while (*++(*argv)) switch (**argv) { - case 'R': - recursiveFlag = TRUE; - break; - default: - fprintf(stderr, invalid_option, invocationName, **argv); - usage(appUsage); + case 'R': + recursiveFlag = TRUE; + break; + default: + fprintf(stderr, invalid_option, invocationName, **argv); + usage(appUsage); } - argc--; argv++; } + if (argc == 0 || *argv == NULL) { + fprintf(stderr, too_few_args, invocationName); + usage(appUsage); + } + if (whichApp == CHMOD_APP) { theMode = *argv; } else { diff --git a/messages.c b/messages.c index 2f8aab515..cb8de4077 100644 --- a/messages.c +++ b/messages.c @@ -74,6 +74,9 @@ #if defined bb_need_write_error || ! defined BB_DECLARE_EXTERN BB_DEF_MESSAGE(write_error, "Write Error\n") #endif +#if defined bb_need_too_few_args || ! defined BB_DECLARE_EXTERN + BB_DEF_MESSAGE(too_few_args, "%s: too few arguments\n") +#endif