Adding support for offsetof()

2 files changed, 18 insertions, 0 deletions

diff --git a/tests/test_c_lexer.py b/tests/test_c_lexer.py
index 6d0b52d..033963e 100644
--- a/tests/test_c_lexer.py
+++ b/tests/test_c_lexer.py
@@ -84,6 +84,9 @@ class TestCLexerNoErrors(unittest.TestCase):
         # - is MINUS, the rest a constnant
         self.assertTokensTypes('-1', ['MINUS', 'INT_CONST_DEC'])
 
+    def test_special_names(self):
+        self.assertTokensTypes('sizeof offsetof', ['SIZEOF', 'OFFSETOF'])
+
     def test_floating_constants(self):
         self.assertTokensTypes('1.5f', ['FLOAT_CONST'])
         self.assertTokensTypes('01.5', ['FLOAT_CONST'])
diff --git a/tests/test_c_parser.py b/tests/test_c_parser.py
index 035182d..556748c 100755
--- a/tests/test_c_parser.py
+++ b/tests/test_c_parser.py
@@ -475,6 +475,21 @@ class TestCParser_fundamentals(TestCParser_base):
                         ['TypeDecl',
                             ['IdentifierType', ['int']]]]]])
 
+    def test_offsetof(self):
+        e = """
+            void foo() {
+                int a = offsetof(struct S, p);
+                a.b = offsetof(struct sockaddr, sp) + strlen(bar);
+            }
+            """
+        compound = self.parse(e).ext[0].body
+        s1 = compound.block_items[0].init
+        self.assertIsInstance(s1, FuncCall)
+        self.assertIsInstance(s1.name, ID)
+        self.assertEqual(s1.name.name, 'offsetof')
+        self.assertIsInstance(s1.args.exprs[0], Typename)
+        self.assertIsInstance(s1.args.exprs[1], ID)
+
     # The C99 compound literal feature
     #
     def test_compound_literals(self):