add default definition of YYSTYPE in .tab.h

1 files changed, 6 insertions, 3 deletions

diff --git a/myacc.c b/myacc.c
index 990015f..13cae7e 100644
--- a/myacc.c
+++ b/myacc.c
@@ -692,9 +692,12 @@ tblout()
 	}
 	aout("yytrns", o, m);
 	if (fhdr) {
-		fprintf(fhdr, "int yyparse(void);\n");
-		fprintf(fhdr, "extern YYSTYPE yylval;\n");
-		fprintf(fhdr, "#endif\n");
+		fputs("int yyparse(void);\n", fhdr);
+		fputs("#ifndef YYSTYPE\n", fhdr);
+		fputs("#define YYSTYPE int\n", fhdr);
+		fputs("#endif\n", fhdr);
+		fputs("extern YYSTYPE yylval;\n", fhdr);
+		fputs("#endif\n", fhdr);
 	}
 	free(o);
 }