test: Enable inp_include.c test + feat: ifndef, fixes #60

keyvank · Nov 28, 2024 · 200dda4 · 200dda4
1 parent d849085
commit 200dda4
Show file tree

Hide file tree

Showing 4 changed files with 111 additions and 1 deletion.
diff --git a/preprocess/macro_ifndef.c b/preprocess/macro_ifndef.c
@@ -0,0 +1,60 @@
+#include <stdlib.h>
+#include <stdio.h>
+#include "macro_ifndef.h"
+#include "preprocess.h"
+#include <string.h>
+
+seg_ifndef *parse_ifndef(prep_ctx *ctx, typed_token **tkns_ptr)
+{
+    typed_token *tkn = *tkns_ptr;
+    if (tkn->type_id == TKN_DIRECTIVE)
+    {
+        typed_token *inner_tkn = (typed_token *)tkn->data;
+        tkn = tkn->next;
+        if (inner_tkn->type_id == TKN_ID)
+        {
+            if (strcmp((char *)inner_tkn->data, "ifndef") == 0)
+            {
+                inner_tkn = inner_tkn->next;
+                if (inner_tkn->type_id == TKN_ID)
+                {
+                    char *def = (char *)inner_tkn->data;
+                    inner_tkn = inner_tkn->next;
+
+                    if (inner_tkn->type_id == TKN_EOF)
+                    {
+                        *tkns_ptr = tkn;
+                        seg_ifndef *ret = malloc(sizeof(seg_ifndef));
+                        ret->def = def;
+                        return ret;
+                    }
+                    else
+                    {
+                        return NULL;
+                    }
+                }
+            }
+        }
+    }
+    return NULL;
+}
+
+int is_endif(typed_token *tkn)
+{
+    if (tkn->type_id == TKN_DIRECTIVE)
+    {
+        typed_token *inner_tkn = (typed_token *)tkn->data;
+        if (inner_tkn->type_id == TKN_ID)
+        {
+            if (strcmp((char *)inner_tkn->data, "endif") == 0)
+            {
+                inner_tkn = inner_tkn->next;
+                if (inner_tkn->type_id == TKN_EOF)
+                {
+                    return 1;
+                }
+            }
+        }
+    }
+    return 0;
+}
diff --git a/preprocess/macro_ifndef.h b/preprocess/macro_ifndef.h
@@ -0,0 +1,15 @@
+#ifndef MACRO_IFNDEF_H
+#define MACRO_IFNDEF_H
+
+#include "../linked_list.h"
+#include "preprocess.h"
+
+typedef struct
+{
+    char *def;
+} seg_ifndef;
+
+seg_ifndef *parse_ifndef(prep_ctx *ctx, typed_token **tkns_ptr);
+int is_endif(typed_token *tkn);
+
+#endif
diff --git a/preprocess/preprocess.c b/preprocess/preprocess.c
@@ -9,6 +9,7 @@
 #include "macro_define.h"
 #include "macro_call.h"
 #include "macro_include.h"
+#include "macro_ifndef.h"
 #include "token.h"
 
 typed_token *chain_tokens(linked_list *tkns)
@@ -37,6 +38,40 @@ typed_token *preprocess(prep_ctx *ctx, char *path)
 
     while (tkn->type_id != TKN_EOF)
     {
+        if (is_endif(tkn))
+        {
+            tkn = tkn->next;
+            continue;
+        }
+
+        seg_ifndef *ifndef = parse_ifndef(ctx, &tkn);
+        if (ifndef)
+        {
+            if (find_define(ctx, ifndef->def))
+            {
+                int endif_found = 0;
+                while (tkn->type_id != TKN_EOF)
+                {
+                    if (is_endif(tkn))
+                    {
+                        tkn = tkn->next;
+                        endif_found = 1;
+                        break;
+                    }
+                    tkn = tkn->next;
+                }
+                if (!endif_found)
+                {
+                    fprintf(stderr, "Couldn't find #endif!\n");
+                    exit(1);
+                }
+                else
+                {
+                    continue;
+                }
+            }
+        }
+
         seg *s = NULL;
         if (!s)
             s = parse_token(ctx, &tkn);

diff --git a/scripts/test.py b/scripts/test.py
@@ -15,7 +15,7 @@
     "./examples/inp_linked_list.c": [],
     "./examples/inp_unary_op.c": [],
     "./examples/lib_usage.c": [],
-    #"./examples/inp_include.c": [],
+    "./examples/inp_include.c": [],
     "./examples/inp_func_ptrs.c": [],
     "./examples/inp_goto.c": [],
     "./examples/inp_break.c": [],