1 files changed, 38 insertions, 13 deletions

diff --git a/usr.bin/calendar/io.c b/usr.bin/calendar/io.c
index 3c9024b1d103..be5d232bcd62 100644
--- a/usr.bin/calendar/io.c
+++ b/usr.bin/calendar/io.c
@@ -89,22 +89,29 @@ static StringList *definitions = NULL;
 static struct event *events[MAXCOUNT];
 static char *extradata[MAXCOUNT];
 
-static void
+static char *
 trimlr(char **buf)
 {
 	char *walk = *buf;
+	char *sep;
 	char *last;
 
 	while (isspace(*walk))
 		walk++;
-	if (*walk != '\0') {
-		last = walk + strlen(walk) - 1;
+	*buf = walk;
+
+	sep = walk;
+	while (*sep != '\0' && !isspace(*sep))
+		sep++;
+
+	if (*sep != '\0') {
+		last = sep + strlen(sep) - 1;
 		while (last > walk && isspace(*last))
 			last--;
 		*(last+1) = 0;
 	}
 
-	*buf = walk;
+	return (sep);
 }
 
 static FILE *
@@ -167,7 +174,7 @@ cal_path(void)
 static int
 token(char *line, FILE *out, int *skip, int *unskip)
 {
-	char *walk, c, a;
+	char *walk, *sep, a, c;
 	const char *this_cal_home;
 	const char *this_cal_dir;
 	const char *this_cal_file;
@@ -188,14 +195,20 @@ token(char *line, FILE *out, int *skip, int *unskip)
 
 	if (strncmp(line, "ifdef", 5) == 0) {
 		walk = line + 5;
-		trimlr(&walk);
+		sep = trimlr(&walk);
 
 		if (*walk == '\0') {
 			WARN0("Expecting arguments after #ifdef");
 			return (T_ERR);
 		}
+		if (*sep != '\0') {
+			WARN1("Expecting a single word after #ifdef "
+			    "but got \"%s\"", walk);
+			return (T_ERR);
+		}
 
-		if (*skip != 0 || definitions == NULL || sl_find(definitions, walk) == NULL)
+		if (*skip != 0 ||
+		    definitions == NULL || sl_find(definitions, walk) == NULL)
 			++*skip;
 		else
 			++*unskip;
@@ -205,14 +218,20 @@ token(char *line, FILE *out, int *skip, int *unskip)
 
 	if (strncmp(line, "ifndef", 6) == 0) {
 		walk = line + 6;
-		trimlr(&walk);
+		sep = trimlr(&walk);
 
 		if (*walk == '\0') {
 			WARN0("Expecting arguments after #ifndef");
 			return (T_ERR);
 		}
+		if (*sep != '\0') {
+			WARN1("Expecting a single word after #ifndef "
+			    "but got \"%s\"", walk);
+			return (T_ERR);
+		}
 
-		if (*skip != 0 || (definitions != NULL && sl_find(definitions, walk) != NULL))
+		if (*skip != 0 ||
+		    (definitions != NULL && sl_find(definitions, walk) != NULL))
 			++*skip;
 		else
 			++*unskip;
@@ -222,7 +241,7 @@ token(char *line, FILE *out, int *skip, int *unskip)
 
 	if (strncmp(line, "else", 4) == 0) {
 		walk = line + 4;
-		trimlr(&walk);
+		(void)trimlr(&walk);
 
 		if (*walk != '\0') {
 			WARN0("Expecting no arguments after #else");
@@ -251,7 +270,7 @@ token(char *line, FILE *out, int *skip, int *unskip)
 	if (strncmp(line, "include", 7) == 0) {
 		walk = line + 7;
 
-		trimlr(&walk);
+		(void)trimlr(&walk);
 
 		if (*walk == '\0') {
 			WARN0("Expecting arguments after #include");
@@ -291,7 +310,8 @@ token(char *line, FILE *out, int *skip, int *unskip)
 		if (definitions == NULL)
 			definitions = sl_init();
 		walk = line + 6;
-		trimlr(&walk);
+		sep = trimlr(&walk);
+		*sep = '\0';
 
 		if (*walk == '\0') {
 			WARN0("Expecting arguments after #define");
@@ -306,12 +326,17 @@ token(char *line, FILE *out, int *skip, int *unskip)
 	if (strncmp(line, "undef", 5) == 0) {
 		if (definitions != NULL) {
 			walk = line + 5;
-			trimlr(&walk);
+			sep = trimlr(&walk);
 
 			if (*walk == '\0') {
 				WARN0("Expecting arguments after #undef");
 				return (T_ERR);
 			}
+			if (*sep != '\0') {
+				WARN1("Expecting a single word after #undef "
+				    "but got \"%s\"", walk);
+				return (T_ERR);
+			}
 
 			walk = sl_find(definitions, walk);
 			if (walk != NULL)