fix(c): modify singleLineMacroDeclaration and macroKeywords #24

- Modify singleLineMacroDeclaration to include Identifier as an option for include - Modify macroKeywords to include 'ifndef' as an option These changes ensure that singleLineMacroDeclaration and macroKeywords are more flexible and can handle different scenarios.
phodal · Jan 31, 2024 · 0eb39a6 · 0eb39a6
1 parent 2d2eecf
commit 0eb39a6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/chapi-ast-c/src/main/antlr/C.g4 b/chapi-ast-c/src/main/antlr/C.g4
@@ -470,16 +470,16 @@ macroStatement
     ;
 
 singleLineMacroDeclaration
-    : '#' include (StringLiteral | ('<' includeIdentifier '>' ))                         #includeDeclaration
+    : '#' include (StringLiteral | Identifier | ('<' includeIdentifier '>' ))            #includeDeclaration
     | '#' 'define' expression* '#' 'undef' identifierList?                               #macroExpansionDeclaration
-    | '#' ('ifdef' | 'endif') Identifier expression* '#' 'endif' Identifier?             #ifdefDeclaration
+    | '#' ('ifdef' | 'ifndef' |'if') Identifier expression* '#' 'endif' Identifier?      #ifdefDeclaration
     | '#' macroKeywords                                                                  #defineDeclaration
     | '#' '#'? Identifier                                                                #macroCastDeclaration
     | '#' macroKeywords expression* '#' macroKeywords identifierList?                    #macroExpansionDeclaration2
     ;
 
 macroKeywords
-    :  'if' | 'undef' | 'else' | 'pragma' | 'endif' | 'ifdef' | 'ifndef' | 'elif' | 'define'
+    :  'if' | 'undef' | 'else' | 'pragma' | 'endif' | 'ifdef' | 'ifndef' | 'elif' | 'define' | 'ifndef'
     ;
 
 labeledStatement

diff --git a/chapi-ast-c/src/test/kotlin/chapi/ast/cast/CFullIdentListenerTest.kt b/chapi-ast-c/src/test/kotlin/chapi/ast/cast/CFullIdentListenerTest.kt
@@ -258,10 +258,11 @@ typedef struct {
             #include "redismodule.h"
             #include <sys/mman.h>
             #include <float.h>
+            #include LUA_USER_H
             """.trimIndent()
 
         val codeFile = CAnalyser().analysis(code, "helloworld.c")
-        assertEquals(codeFile.Imports.size, 6)
+        assertEquals(codeFile.Imports.size, 7)
     }
 
     @Test