From b08ce880822875d4bef103719179501d6ba38730 Mon Sep 17 00:00:00 2001
From: seyko <seyko2@gmail.com>
Date: Tue, 21 Apr 2015 15:46:29 +0300
Subject: "#pragma once" implementation

---
 tccpp.c | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'tccpp.c')

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
-- 
cgit v1.3.1