(yywrap): Close yyin, and clean up the buffer

(get_token): New function. Interface to yylex.

1 files changed, 14 insertions, 0 deletions

diff --git a/src/c.l b/src/c.l
index eb4f503..43d6066 100644
--- a/src/c.l
+++ b/src/c.l
@@ -312,13 +312,27 @@ pp_close(FILE *fp)
 int
 yywrap()
 {
+     if (!yyin)
+	  return 1;
      if (preprocess_option)
 	  pp_close(yyin);
+     else
+	  fclose(yyin);
+     yyin = NULL;
+#ifdef FLEX_SCANNER
+     yy_delete_buffer(yy_current_buffer);
+#endif
      delete_statics();
      return 1;
 }
 
 int
+get_token()
+{
+     return yyin ? yylex() : 0;
+}
+
+int
 source(char *name)
 {
      FILE *fp;