Fix badly placed BEGIN/yylex(). Caused some #ifdefs not to be handled.

1 files changed, 2 insertions, 1 deletions

diff --git a/token.c b/token.c
index aaa51d1..aca9bd1 100644
--- a/token.c
+++ b/token.c
@@ -442,9 +442,11 @@ contr:			while ((ch = input()) == ' ' || ch == '\t')
 #define	CC(s)	if (strcmp(yytext, s) == 0)
 			CC("ifndef") {
 				contr = 0; ifndefstmt();
+				BEGIN 0;
 				goto zagain;
 			} else CC("ifdef") {
 				contr = 0; ifdefstmt();
+				BEGIN 0;
 				goto zagain;
 			} else CC("if") {
 				contr = 0; storepb(); BEGIN IFR;
@@ -907,7 +909,6 @@ ifdefstmt(void)
 		while (input() != '\n')
 			;
 		unput('\n');
-		yylex();
 		flslvl++;
 		return;
 	}