From 4f6ff00bd402a74d794214f3c58c19b98eafa756 Mon Sep 17 00:00:00 2001 From: Steven Schubiger Date: Sat, 16 Nov 2013 23:19:06 +0100 Subject: Validate color provided to exclude-random switch --- colorize.c | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) (limited to 'colorize.c') diff --git a/colorize.c b/colorize.c index 319c799..4303eb7 100644 --- a/colorize.c +++ b/colorize.c @@ -262,13 +262,21 @@ main (int argc, char **argv) clean_all = true; break; case OPT_EXCLUDE_RANDOM: { - char *p; + bool valid = false; + unsigned int i; exclude = xstrdup (optarg); STACK_VAR (exclude); - for (p = exclude; *p; p++) - *p = tolower (*p); - if (streq (exclude, "random")) - vfprintf_fail (formats[FMT_GENERIC], "--exclude-random switch must be provided a color"); + for (i = 1; i < tables[FOREGROUND].count - 1; i++) /* skip color none and default */ + { + const struct color *entry = &tables[FOREGROUND].entries[i]; + if (streq (exclude, entry->name)) + { + valid = true; + break; + } + } + if (!valid) + vfprintf_fail (formats[FMT_GENERIC], "--exclude-random switch must be provided a plain color"); break; } case OPT_HELP: -- cgit v1.2.3