"#pragma once" implementation

1 files changed, 3 insertions, 1 deletions

diff --git a/tccpp.c b/tccpp.c
index 4d9430d..b26dbee 100644
--- a/tccpp.c
+++ b/tccpp.c
@@ -1478,6 +1478,8 @@ static void pragma_parse(TCCState *s1)
             }
             next();
             skip(')');
+    } else if (tok == TOK_once) {
+        add_cached_include(s1, file->filename, TOK_once);
     } else {
         tcc_warning("unknown #pragma %s", get_tok_str(tok, &tokc));
     }
@@ -1618,7 +1620,7 @@ ST_FUNC void preprocess(int is_bof)
                     }
 
             e = search_cached_include(s1, buf1);
-            if (e && define_find(e->ifndef_macro)) {
+            if (e && (define_find(e->ifndef_macro) || e->ifndef_macro == TOK_once)) {
                 /* no need to parse the include because the 'ifndef macro'
                    is defined */
 #ifdef INC_DEBUG