Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1 | #!/usr/bin/env python |
| 2 | |
| 3 | import pprint |
| 4 | import re |
Eli Bendersky | d097378 | 2012-01-19 08:09:33 +0200 | [diff] [blame] | 5 | import os, sys |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 6 | import unittest |
| 7 | |
| 8 | sys.path.insert(0, '..') |
| 9 | |
| 10 | from pycparser import c_parser |
| 11 | from pycparser.c_ast import * |
| 12 | from pycparser.c_parser import CParser, Coord, ParseError |
| 13 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 14 | _c_parser = c_parser.CParser( |
| 15 | lex_optimize=False, |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 16 | yacc_debug=True, |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 17 | yacc_optimize=False, |
| 18 | yacctab='yacctab') |
| 19 | |
| 20 | |
| 21 | def expand_decl(decl): |
| 22 | """ Converts the declaration into a nested list. |
| 23 | """ |
| 24 | typ = type(decl) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 25 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 26 | if typ == TypeDecl: |
| 27 | return ['TypeDecl', expand_decl(decl.type)] |
| 28 | elif typ == IdentifierType: |
| 29 | return ['IdentifierType', decl.names] |
| 30 | elif typ == ID: |
| 31 | return ['ID', decl.name] |
| 32 | elif typ in [Struct, Union]: |
| 33 | decls = [expand_decl(d) for d in decl.decls or []] |
| 34 | return [typ.__name__, decl.name, decls] |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 35 | else: |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 36 | nested = expand_decl(decl.type) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 37 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 38 | if typ == Decl: |
| 39 | if decl.quals: |
| 40 | return ['Decl', decl.quals, decl.name, nested] |
| 41 | else: |
| 42 | return ['Decl', decl.name, nested] |
| 43 | elif typ == Typename: # for function parameters |
| 44 | if decl.quals: |
| 45 | return ['Typename', decl.quals, nested] |
| 46 | else: |
| 47 | return ['Typename', nested] |
| 48 | elif typ == ArrayDecl: |
| 49 | dimval = decl.dim.value if decl.dim else '' |
| 50 | return ['ArrayDecl', dimval, nested] |
| 51 | elif typ == PtrDecl: |
| 52 | return ['PtrDecl', nested] |
| 53 | elif typ == Typedef: |
| 54 | return ['Typedef', decl.name, nested] |
| 55 | elif typ == FuncDecl: |
| 56 | if decl.args: |
| 57 | params = [expand_decl(param) for param in decl.args.params] |
| 58 | else: |
| 59 | params = [] |
| 60 | return ['FuncDecl', params, nested] |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 61 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 62 | |
| 63 | def expand_init(init): |
| 64 | """ Converts an initialization into a nested list |
| 65 | """ |
| 66 | typ = type(init) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 67 | |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 68 | if typ == NamedInitializer: |
| 69 | des = [expand_init(dp) for dp in init.name] |
| 70 | return (des, expand_init(init.expr)) |
Eli Bendersky | 293ea91 | 2012-12-25 14:52:48 -0800 | [diff] [blame] | 71 | elif typ in (InitList, ExprList): |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 72 | return [expand_init(expr) for expr in init.exprs] |
| 73 | elif typ == Constant: |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 74 | return ['Constant', init.type, init.value] |
| 75 | elif typ == ID: |
| 76 | return ['ID', init.name] |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 77 | |
| 78 | |
eli.bendersky | 85d2e73 | 2011-05-20 19:47:26 +0300 | [diff] [blame] | 79 | class TestCParser_base(unittest.TestCase): |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 80 | def parse(self, txt, filename=''): |
| 81 | return self.cparser.parse(txt, filename) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 82 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 83 | def setUp(self): |
| 84 | self.cparser = _c_parser |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 85 | |
eli.bendersky | 85d2e73 | 2011-05-20 19:47:26 +0300 | [diff] [blame] | 86 | |
| 87 | class TestCParser_fundamentals(TestCParser_base): |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 88 | def get_decl(self, txt, index=0): |
| 89 | """ Given a source and an index returns the expanded |
| 90 | declaration at that index. |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 91 | |
| 92 | FileAST holds a list of 'external declarations'. |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 93 | index is the offset of the desired declaration in that |
| 94 | list. |
| 95 | """ |
| 96 | t = self.parse(txt).ext[index] |
| 97 | return expand_decl(t) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 98 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 99 | def get_decl_init(self, txt, index=0): |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 100 | """ Returns the expanded initializer of the declaration |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 101 | at index. |
| 102 | """ |
| 103 | t = self.parse(txt).ext[index] |
| 104 | return expand_init(t.init) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 105 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 106 | def test_FileAST(self): |
| 107 | t = self.parse('int a; char c;') |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 108 | self.assertTrue(isinstance(t, FileAST)) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 109 | self.assertEqual(len(t.ext), 2) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 110 | |
eli.bendersky | 43cf0b2 | 2011-10-19 05:56:15 +0200 | [diff] [blame] | 111 | # empty file |
| 112 | t2 = self.parse('') |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 113 | self.assertTrue(isinstance(t2, FileAST)) |
eli.bendersky | 43cf0b2 | 2011-10-19 05:56:15 +0200 | [diff] [blame] | 114 | self.assertEqual(len(t2.ext), 0) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 115 | |
eli.bendersky | 38165b7 | 2011-02-04 08:13:39 +0200 | [diff] [blame] | 116 | def test_empty_toplevel_decl(self): |
| 117 | code = 'int foo;;' |
| 118 | t = self.parse(code) |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 119 | self.assertTrue(isinstance(t, FileAST)) |
eli.bendersky | 38165b7 | 2011-02-04 08:13:39 +0200 | [diff] [blame] | 120 | self.assertEqual(len(t.ext), 1) |
| 121 | self.assertEqual(self.get_decl(code), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 122 | ['Decl', 'foo', |
eli.bendersky | 38165b7 | 2011-02-04 08:13:39 +0200 | [diff] [blame] | 123 | ['TypeDecl', ['IdentifierType', ['int']]]]) |
| 124 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 125 | def assert_coord(self, node, line, file=None): |
| 126 | self.assertEqual(node.coord.line, line) |
| 127 | if file: |
| 128 | self.assertEqual(node.coord.file, file) |
| 129 | |
| 130 | def test_coords(self): |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 131 | """ Tests the "coordinates" of parsed elements - file |
eli.bendersky | 38165b7 | 2011-02-04 08:13:39 +0200 | [diff] [blame] | 132 | name and line numbers, with modification insterted by |
| 133 | #line directives. |
| 134 | """ |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 135 | self.assert_coord(self.parse('int a;').ext[0], 1) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 136 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 137 | t1 = """ |
| 138 | int a; |
| 139 | int b;\n\n |
| 140 | int c; |
| 141 | """ |
| 142 | f1 = self.parse(t1, filename='test.c') |
| 143 | self.assert_coord(f1.ext[0], 2, 'test.c') |
| 144 | self.assert_coord(f1.ext[1], 3, 'test.c') |
| 145 | self.assert_coord(f1.ext[2], 6, 'test.c') |
| 146 | |
| 147 | t1_1 = ''' |
| 148 | int main() { |
| 149 | k = p; |
| 150 | printf("%d", b); |
| 151 | return 0; |
| 152 | }''' |
| 153 | f1_1 = self.parse(t1_1, filename='test.c') |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 154 | self.assert_coord(f1_1.ext[0].body.block_items[0], 3, 'test.c') |
| 155 | self.assert_coord(f1_1.ext[0].body.block_items[1], 4, 'test.c') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 156 | |
eli.bendersky | fc96e5e | 2011-03-04 09:51:23 +0200 | [diff] [blame] | 157 | t1_2 = ''' |
| 158 | int main () { |
| 159 | int p = (int) k; |
| 160 | }''' |
| 161 | f1_2 = self.parse(t1_2, filename='test.c') |
| 162 | # make sure that the Cast has a coord (issue 23) |
| 163 | self.assert_coord(f1_2.ext[0].body.block_items[0].init, 3, 'test.c') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 164 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 165 | t2 = """ |
| 166 | #line 99 |
| 167 | int c; |
| 168 | """ |
| 169 | self.assert_coord(self.parse(t2).ext[0], 99) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 170 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 171 | t3 = """ |
| 172 | int dsf; |
| 173 | char p; |
| 174 | #line 3000 "in.h" |
| 175 | char d; |
| 176 | """ |
| 177 | f3 = self.parse(t3, filename='test.c') |
| 178 | self.assert_coord(f3.ext[0], 2, 'test.c') |
| 179 | self.assert_coord(f3.ext[1], 3, 'test.c') |
| 180 | self.assert_coord(f3.ext[2], 3000, 'in.h') |
| 181 | |
| 182 | t4 = """ |
| 183 | #line 20 "restore.h" |
| 184 | int maydler(char); |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 185 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 186 | #line 30 "includes/daween.ph" |
| 187 | long j, k; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 188 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 189 | #line 50000 |
| 190 | char* ro; |
| 191 | """ |
| 192 | f4 = self.parse(t4, filename='myb.c') |
| 193 | self.assert_coord(f4.ext[0], 20, 'restore.h') |
| 194 | self.assert_coord(f4.ext[1], 30, 'includes/daween.ph') |
| 195 | self.assert_coord(f4.ext[2], 30, 'includes/daween.ph') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 196 | self.assert_coord(f4.ext[3], 50000, 'includes/daween.ph') |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 197 | |
| 198 | t5 = """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 199 | int |
| 200 | #line 99 |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 201 | c; |
| 202 | """ |
| 203 | self.assert_coord(self.parse(t5).ext[0], 99) |
| 204 | |
Eli Bendersky | 203b967 | 2012-06-15 10:11:24 +0300 | [diff] [blame] | 205 | # coord for ellipsis |
| 206 | t6 = """ |
| 207 | int foo(int j, |
| 208 | ...) { |
| 209 | }""" |
| 210 | f6 = self.parse(t6, filename='z.c') |
| 211 | self.assert_coord(self.parse(t6).ext[0].decl.type.args.params[1], 3) |
| 212 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 213 | def test_simple_decls(self): |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 214 | self.assertEqual(self.get_decl('int a;'), |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 215 | ['Decl', 'a', ['TypeDecl', ['IdentifierType', ['int']]]]) |
| 216 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 217 | self.assertEqual(self.get_decl('unsigned int a;'), |
Eli Bendersky | 68497c2 | 2012-01-19 06:04:58 +0200 | [diff] [blame] | 218 | ['Decl', 'a', ['TypeDecl', ['IdentifierType', ['unsigned', 'int']]]]) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 219 | |
eli.bendersky | affe032 | 2011-09-24 06:16:55 +0300 | [diff] [blame] | 220 | self.assertEqual(self.get_decl('_Bool a;'), |
| 221 | ['Decl', 'a', ['TypeDecl', ['IdentifierType', ['_Bool']]]]) |
| 222 | |
Eli Bendersky | f4d7346 | 2012-01-19 05:56:27 +0200 | [diff] [blame] | 223 | self.assertEqual(self.get_decl('float _Complex fcc;'), |
Eli Bendersky | 68497c2 | 2012-01-19 06:04:58 +0200 | [diff] [blame] | 224 | ['Decl', 'fcc', ['TypeDecl', ['IdentifierType', ['float', '_Complex']]]]) |
Eli Bendersky | f4d7346 | 2012-01-19 05:56:27 +0200 | [diff] [blame] | 225 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 226 | self.assertEqual(self.get_decl('char* string;'), |
| 227 | ['Decl', 'string', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 228 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]]) |
| 229 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 230 | self.assertEqual(self.get_decl('long ar[15];'), |
| 231 | ['Decl', 'ar', |
| 232 | ['ArrayDecl', '15', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 233 | ['TypeDecl', ['IdentifierType', ['long']]]]]) |
eli.bendersky | 98f4537 | 2010-10-30 09:46:29 +0200 | [diff] [blame] | 234 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 235 | self.assertEqual(self.get_decl('long long ar[15];'), |
| 236 | ['Decl', 'ar', |
| 237 | ['ArrayDecl', '15', |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 238 | ['TypeDecl', ['IdentifierType', ['long', 'long']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 239 | |
| 240 | self.assertEqual(self.get_decl('unsigned ar[];'), |
| 241 | ['Decl', 'ar', |
| 242 | ['ArrayDecl', '', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 243 | ['TypeDecl', ['IdentifierType', ['unsigned']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 244 | |
| 245 | self.assertEqual(self.get_decl('int strlen(char* s);'), |
| 246 | ['Decl', 'strlen', |
| 247 | ['FuncDecl', |
| 248 | [['Decl', 's', |
| 249 | ['PtrDecl', |
| 250 | ['TypeDecl', ['IdentifierType', ['char']]]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 251 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 252 | |
| 253 | self.assertEqual(self.get_decl('int strcmp(char* s1, char* s2);'), |
| 254 | ['Decl', 'strcmp', |
| 255 | ['FuncDecl', |
| 256 | [ ['Decl', 's1', |
| 257 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]], |
| 258 | ['Decl', 's2', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 259 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]] |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 260 | ], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 261 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
| 262 | |
Sye van der Veen | 51a46ec | 2013-06-10 12:56:15 -0400 | [diff] [blame] | 263 | # function return values and parameters may not have type information |
Eli Bendersky | 09e22a6 | 2013-07-02 06:00:36 -0700 | [diff] [blame] | 264 | self.assertEqual(self.get_decl('extern foobar(foo, bar);'), |
| 265 | ['Decl', 'foobar', |
| 266 | ['FuncDecl', |
Sye van der Veen | 51a46ec | 2013-06-10 12:56:15 -0400 | [diff] [blame] | 267 | [ ['ID', 'foo'], |
| 268 | ['ID', 'bar'] |
Eli Bendersky | 09e22a6 | 2013-07-02 06:00:36 -0700 | [diff] [blame] | 269 | ], |
Sye van der Veen | 51a46ec | 2013-06-10 12:56:15 -0400 | [diff] [blame] | 270 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
| 271 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 272 | def test_nested_decls(self): # the fun begins |
| 273 | self.assertEqual(self.get_decl('char** ar2D;'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 274 | ['Decl', 'ar2D', |
| 275 | ['PtrDecl', ['PtrDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 276 | ['TypeDecl', ['IdentifierType', ['char']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 277 | |
| 278 | self.assertEqual(self.get_decl('int (*a)[1][2];'), |
| 279 | ['Decl', 'a', |
| 280 | ['PtrDecl', |
| 281 | ['ArrayDecl', '1', |
| 282 | ['ArrayDecl', '2', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 283 | ['TypeDecl', ['IdentifierType', ['int']]]]]]]) |
| 284 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 285 | self.assertEqual(self.get_decl('int *a[1][2];'), |
| 286 | ['Decl', 'a', |
| 287 | ['ArrayDecl', '1', |
| 288 | ['ArrayDecl', '2', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 289 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['int']]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 290 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 291 | self.assertEqual(self.get_decl('char ***ar3D[40];'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 292 | ['Decl', 'ar3D', |
| 293 | ['ArrayDecl', '40', |
| 294 | ['PtrDecl', ['PtrDecl', ['PtrDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 295 | ['TypeDecl', ['IdentifierType', ['char']]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 296 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 297 | self.assertEqual(self.get_decl('char (***ar3D)[40];'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 298 | ['Decl', 'ar3D', |
| 299 | ['PtrDecl', ['PtrDecl', ['PtrDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 300 | ['ArrayDecl', '40', ['TypeDecl', ['IdentifierType', ['char']]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 301 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 302 | self.assertEqual(self.get_decl('int (*x[4])(char, int);'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 303 | ['Decl', 'x', |
| 304 | ['ArrayDecl', '4', |
| 305 | ['PtrDecl', |
| 306 | ['FuncDecl', |
| 307 | [ ['Typename', ['TypeDecl', ['IdentifierType', ['char']]]], |
| 308 | ['Typename', ['TypeDecl', ['IdentifierType', ['int']]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 309 | ['TypeDecl', ['IdentifierType', ['int']]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 310 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 311 | self.assertEqual(self.get_decl('char *(*(**foo [][8])())[];'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 312 | ['Decl', 'foo', |
| 313 | ['ArrayDecl', '', |
| 314 | ['ArrayDecl', '8', |
| 315 | ['PtrDecl', ['PtrDecl', |
| 316 | ['FuncDecl', |
| 317 | [], |
| 318 | ['PtrDecl', |
| 319 | ['ArrayDecl', '', |
| 320 | ['PtrDecl', |
| 321 | ['TypeDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 322 | ['IdentifierType', ['char']]]]]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 323 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 324 | # explore named and unnamed function pointer parameters, |
| 325 | # with and without qualifiers |
| 326 | # |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 327 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 328 | # unnamed w/o quals |
| 329 | self.assertEqual(self.get_decl('int (*k)(int);'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 330 | ['Decl', 'k', |
| 331 | ['PtrDecl', |
| 332 | ['FuncDecl', |
| 333 | [['Typename', ['TypeDecl', ['IdentifierType', ['int']]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 334 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 335 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 336 | # unnamed w/ quals |
| 337 | self.assertEqual(self.get_decl('int (*k)(const int);'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 338 | ['Decl', 'k', |
| 339 | ['PtrDecl', |
| 340 | ['FuncDecl', |
| 341 | [['Typename', ['const'], ['TypeDecl', ['IdentifierType', ['int']]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 342 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 343 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 344 | # named w/o quals |
| 345 | self.assertEqual(self.get_decl('int (*k)(int q);'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 346 | ['Decl', 'k', |
| 347 | ['PtrDecl', |
| 348 | ['FuncDecl', |
| 349 | [['Decl', 'q', ['TypeDecl', ['IdentifierType', ['int']]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 350 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 351 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 352 | # named w/ quals |
| 353 | self.assertEqual(self.get_decl('int (*k)(const volatile int q);'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 354 | ['Decl', 'k', |
| 355 | ['PtrDecl', |
| 356 | ['FuncDecl', |
| 357 | [['Decl', ['const', 'volatile'], 'q', |
| 358 | ['TypeDecl', ['IdentifierType', ['int']]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 359 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 360 | |
eli.bendersky | 79d5cf6 | 2010-10-29 13:33:52 +0200 | [diff] [blame] | 361 | # restrict qualifier |
| 362 | self.assertEqual(self.get_decl('int (*k)(restrict int* q);'), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 363 | ['Decl', 'k', |
| 364 | ['PtrDecl', |
| 365 | ['FuncDecl', |
| 366 | [['Decl', ['restrict'], 'q', |
eli.bendersky | 79d5cf6 | 2010-10-29 13:33:52 +0200 | [diff] [blame] | 367 | ['PtrDecl', |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 368 | ['TypeDecl', ['IdentifierType', ['int']]]]]], |
eli.bendersky | 79d5cf6 | 2010-10-29 13:33:52 +0200 | [diff] [blame] | 369 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 370 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 371 | def test_qualifiers_storage_specifiers(self): |
| 372 | def assert_qs(txt, index, quals, storage): |
| 373 | d = self.parse(txt).ext[index] |
| 374 | self.assertEqual(d.quals, quals) |
| 375 | self.assertEqual(d.storage, storage) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 376 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 377 | assert_qs("extern int p;", 0, [], ['extern']) |
| 378 | assert_qs("const long p = 6;", 0, ['const'], []) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 379 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 380 | d1 = "static const int p, q, r;" |
| 381 | for i in range(3): |
| 382 | assert_qs(d1, i, ['const'], ['static']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 383 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 384 | d2 = "static char * const p;" |
| 385 | assert_qs(d2, 0, [], ['static']) |
| 386 | pdecl = self.parse(d2).ext[0].type |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 387 | self.assertTrue(isinstance(pdecl, PtrDecl)) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 388 | self.assertEqual(pdecl.quals, ['const']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 389 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 390 | def test_sizeof(self): |
| 391 | e = """ |
| 392 | void foo() |
| 393 | { |
| 394 | int a = sizeof k; |
| 395 | int b = sizeof(int); |
eli.bendersky | c51e1d3 | 2011-02-04 08:52:33 +0200 | [diff] [blame] | 396 | int c = sizeof(int**);; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 397 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 398 | char* p = "just to make sure this parses w/o error..."; |
| 399 | int d = sizeof(int()); |
| 400 | } |
| 401 | """ |
| 402 | compound = self.parse(e).ext[0].body |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 403 | |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 404 | s1 = compound.block_items[0].init |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 405 | self.assertTrue(isinstance(s1, UnaryOp)) |
| 406 | self.assertEqual(s1.op, 'sizeof') |
| 407 | self.assertTrue(isinstance(s1.expr, ID)) |
| 408 | self.assertEqual(s1.expr.name, 'k') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 409 | |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 410 | s2 = compound.block_items[1].init |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 411 | self.assertEqual(expand_decl(s2.expr), |
| 412 | ['Typename', ['TypeDecl', ['IdentifierType', ['int']]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 413 | |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 414 | s3 = compound.block_items[2].init |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 415 | self.assertEqual(expand_decl(s3.expr), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 416 | ['Typename', |
| 417 | ['PtrDecl', |
| 418 | ['PtrDecl', |
| 419 | ['TypeDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 420 | ['IdentifierType', ['int']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 421 | |
eli.bendersky | 9f48156 | 2010-10-30 15:50:47 +0200 | [diff] [blame] | 422 | # The C99 compound literal feature |
| 423 | # |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 424 | def test_compound_literals(self): |
eli.bendersky | 9f48156 | 2010-10-30 15:50:47 +0200 | [diff] [blame] | 425 | ps1 = self.parse(r''' |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 426 | void foo() { |
eli.bendersky | 9f48156 | 2010-10-30 15:50:47 +0200 | [diff] [blame] | 427 | p = (long long){k}; |
| 428 | tc = (struct jk){.a = {1, 2}, .b[0] = t}; |
| 429 | }''') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 430 | |
eli.bendersky | 9f48156 | 2010-10-30 15:50:47 +0200 | [diff] [blame] | 431 | compound = ps1.ext[0].body.block_items[0].rvalue |
| 432 | self.assertEqual(expand_decl(compound.type), |
| 433 | ['Typename', ['TypeDecl', ['IdentifierType', ['long', 'long']]]]) |
| 434 | self.assertEqual(expand_init(compound.init), |
| 435 | [['ID', 'k']]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 436 | |
| 437 | compound = ps1.ext[0].body.block_items[1].rvalue |
eli.bendersky | 9f48156 | 2010-10-30 15:50:47 +0200 | [diff] [blame] | 438 | self.assertEqual(expand_decl(compound.type), |
| 439 | ['Typename', ['TypeDecl', ['Struct', 'jk', []]]]) |
| 440 | self.assertEqual(expand_init(compound.init), |
| 441 | [ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 442 | ([['ID', 'a']], [['Constant', 'int', '1'], ['Constant', 'int', '2']]), |
eli.bendersky | 9f48156 | 2010-10-30 15:50:47 +0200 | [diff] [blame] | 443 | ([['ID', 'b'], ['Constant', 'int', '0']], ['ID', 't'])]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 444 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 445 | def test_enums(self): |
| 446 | e1 = "enum mycolor op;" |
| 447 | e1_type = self.parse(e1).ext[0].type.type |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 448 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 449 | self.assertTrue(isinstance(e1_type, Enum)) |
| 450 | self.assertEqual(e1_type.name, 'mycolor') |
| 451 | self.assertEqual(e1_type.values, None) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 452 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 453 | e2 = "enum mysize {large=20, small, medium} shoes;" |
| 454 | e2_type = self.parse(e2).ext[0].type.type |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 455 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 456 | self.assertTrue(isinstance(e2_type, Enum)) |
| 457 | self.assertEqual(e2_type.name, 'mysize') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 458 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 459 | e2_elist = e2_type.values |
| 460 | self.assertTrue(isinstance(e2_elist, EnumeratorList)) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 461 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 462 | for e2_eval in e2_elist.enumerators: |
| 463 | self.assertTrue(isinstance(e2_eval, Enumerator)) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 464 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 465 | self.assertEqual(e2_elist.enumerators[0].name, 'large') |
| 466 | self.assertEqual(e2_elist.enumerators[0].value.value, '20') |
| 467 | self.assertEqual(e2_elist.enumerators[2].name, 'medium') |
| 468 | self.assertEqual(e2_elist.enumerators[2].value, None) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 469 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 470 | # enum with trailing comma (C99 feature) |
| 471 | e3 = """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 472 | enum |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 473 | { |
| 474 | red, |
| 475 | blue, |
| 476 | green, |
| 477 | } color; |
| 478 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 479 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 480 | e3_type = self.parse(e3).ext[0].type.type |
| 481 | self.assertTrue(isinstance(e3_type, Enum)) |
| 482 | e3_elist = e3_type.values |
| 483 | self.assertTrue(isinstance(e3_elist, EnumeratorList)) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 484 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 485 | for e3_eval in e3_elist.enumerators: |
| 486 | self.assertTrue(isinstance(e3_eval, Enumerator)) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 487 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 488 | self.assertEqual(e3_elist.enumerators[0].name, 'red') |
| 489 | self.assertEqual(e3_elist.enumerators[0].value, None) |
| 490 | self.assertEqual(e3_elist.enumerators[1].name, 'blue') |
| 491 | self.assertEqual(e3_elist.enumerators[2].name, 'green') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 492 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 493 | def test_typedef(self): |
| 494 | # without typedef, error |
| 495 | s1 = """ |
| 496 | node k; |
| 497 | """ |
| 498 | self.assertRaises(ParseError, self.parse, s1) |
| 499 | |
| 500 | # now with typedef, works |
| 501 | s2 = """ |
| 502 | typedef void* node; |
| 503 | node k; |
| 504 | """ |
| 505 | ps2 = self.parse(s2) |
| 506 | self.assertEqual(expand_decl(ps2.ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 507 | ['Typedef', 'node', |
| 508 | ['PtrDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 509 | ['TypeDecl', ['IdentifierType', ['void']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 510 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 511 | self.assertEqual(expand_decl(ps2.ext[1]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 512 | ['Decl', 'k', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 513 | ['TypeDecl', ['IdentifierType', ['node']]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 514 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 515 | s3 = """ |
| 516 | typedef int T; |
| 517 | typedef T *pT; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 518 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 519 | pT aa, bb; |
| 520 | """ |
| 521 | ps3 = self.parse(s3) |
| 522 | self.assertEqual(expand_decl(ps3.ext[3]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 523 | ['Decl', 'bb', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 524 | ['TypeDecl', ['IdentifierType', ['pT']]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 525 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 526 | s4 = ''' |
| 527 | typedef char* __builtin_va_list; |
| 528 | typedef __builtin_va_list __gnuc_va_list; |
| 529 | ''' |
| 530 | ps4 = self.parse(s4) |
| 531 | self.assertEqual(expand_decl(ps4.ext[1]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 532 | ['Typedef', '__gnuc_va_list', |
| 533 | ['TypeDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 534 | ['IdentifierType', ['__builtin_va_list']]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 535 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 536 | s5 = '''typedef struct tagHash Hash;''' |
| 537 | ps5 = self.parse(s5) |
| 538 | self.assertEqual(expand_decl(ps5.ext[0]), |
| 539 | ['Typedef', 'Hash', ['TypeDecl', ['Struct', 'tagHash', []]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 540 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 541 | def test_struct_union(self): |
| 542 | s1 = """ |
| 543 | struct { |
| 544 | int id; |
| 545 | char* name; |
| 546 | } joe; |
| 547 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 548 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 549 | self.assertEqual(expand_decl(self.parse(s1).ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 550 | ['Decl', 'joe', |
| 551 | ['TypeDecl', ['Struct', None, |
| 552 | [ ['Decl', 'id', |
| 553 | ['TypeDecl', |
| 554 | ['IdentifierType', ['int']]]], |
| 555 | ['Decl', 'name', |
| 556 | ['PtrDecl', |
| 557 | ['TypeDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 558 | ['IdentifierType', ['char']]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 559 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 560 | s2 = """ |
| 561 | struct node p; |
| 562 | """ |
| 563 | self.assertEqual(expand_decl(self.parse(s2).ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 564 | ['Decl', 'p', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 565 | ['TypeDecl', ['Struct', 'node', []]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 566 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 567 | s21 = """ |
| 568 | union pri ra; |
| 569 | """ |
| 570 | self.assertEqual(expand_decl(self.parse(s21).ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 571 | ['Decl', 'ra', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 572 | ['TypeDecl', ['Union', 'pri', []]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 573 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 574 | s3 = """ |
| 575 | struct node* p; |
| 576 | """ |
| 577 | self.assertEqual(expand_decl(self.parse(s3).ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 578 | ['Decl', 'p', |
| 579 | ['PtrDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 580 | ['TypeDecl', ['Struct', 'node', []]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 581 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 582 | s4 = """ |
| 583 | struct node; |
| 584 | """ |
| 585 | self.assertEqual(expand_decl(self.parse(s4).ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 586 | ['Decl', None, |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 587 | ['Struct', 'node', []]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 588 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 589 | s5 = """ |
| 590 | union |
| 591 | { |
| 592 | struct |
| 593 | { |
| 594 | int type; |
| 595 | } n; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 596 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 597 | struct |
| 598 | { |
| 599 | int type; |
| 600 | int intnode; |
| 601 | } ni; |
| 602 | } u; |
| 603 | """ |
| 604 | self.assertEqual(expand_decl(self.parse(s5).ext[0]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 605 | ['Decl', 'u', |
| 606 | ['TypeDecl', |
| 607 | ['Union', None, |
| 608 | [['Decl', 'n', |
| 609 | ['TypeDecl', |
| 610 | ['Struct', None, |
| 611 | [['Decl', 'type', |
| 612 | ['TypeDecl', ['IdentifierType', ['int']]]]]]]], |
| 613 | ['Decl', 'ni', |
| 614 | ['TypeDecl', |
| 615 | ['Struct', None, |
| 616 | [['Decl', 'type', |
| 617 | ['TypeDecl', ['IdentifierType', ['int']]]], |
| 618 | ['Decl', 'intnode', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 619 | ['TypeDecl', ['IdentifierType', ['int']]]]]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 620 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 621 | s6 = """ |
| 622 | typedef struct foo_tag |
| 623 | { |
| 624 | void* data; |
| 625 | } foo, *pfoo; |
| 626 | """ |
| 627 | s6_ast = self.parse(s6) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 628 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 629 | self.assertEqual(expand_decl(s6_ast.ext[0]), |
| 630 | ['Typedef', 'foo', |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 631 | ['TypeDecl', |
| 632 | ['Struct', 'foo_tag', |
| 633 | [['Decl', 'data', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 634 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['void']]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 635 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 636 | self.assertEqual(expand_decl(s6_ast.ext[1]), |
| 637 | ['Typedef', 'pfoo', |
| 638 | ['PtrDecl', |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 639 | ['TypeDecl', |
| 640 | ['Struct', 'foo_tag', |
| 641 | [['Decl', 'data', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 642 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['void']]]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 643 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 644 | s7 = r""" |
| 645 | struct _on_exit_args { |
| 646 | void * _fnargs[32]; |
| 647 | void * _dso_handle[32]; |
| 648 | |
| 649 | long _fntypes; |
| 650 | #line 77 "D:\eli\cpp_stuff\libc_include/sys/reent.h" |
| 651 | |
| 652 | long _is_cxa; |
| 653 | }; |
| 654 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 655 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 656 | s7_ast = self.parse(s7, filename='test.c') |
| 657 | self.assert_coord(s7_ast.ext[0].type.decls[2], 6, 'test.c') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 658 | self.assert_coord(s7_ast.ext[0].type.decls[3], 78, |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 659 | r'D:\eli\cpp_stuff\libc_include/sys/reent.h') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 660 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 661 | s8 = """ |
| 662 | typedef enum tagReturnCode {SUCCESS, FAIL} ReturnCode; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 663 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 664 | typedef struct tagEntry |
| 665 | { |
| 666 | char* key; |
| 667 | char* value; |
| 668 | } Entry; |
| 669 | |
| 670 | |
| 671 | typedef struct tagNode |
| 672 | { |
| 673 | Entry* entry; |
| 674 | |
| 675 | struct tagNode* next; |
| 676 | } Node; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 677 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 678 | typedef struct tagHash |
| 679 | { |
| 680 | unsigned int table_size; |
| 681 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 682 | Node** heads; |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 683 | |
| 684 | } Hash; |
| 685 | """ |
| 686 | s8_ast = self.parse(s8) |
| 687 | self.assertEqual(expand_decl(s8_ast.ext[3]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 688 | ['Typedef', 'Hash', |
| 689 | ['TypeDecl', ['Struct', 'tagHash', |
| 690 | [['Decl', 'table_size', |
| 691 | ['TypeDecl', ['IdentifierType', ['unsigned', 'int']]]], |
| 692 | ['Decl', 'heads', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 693 | ['PtrDecl', ['PtrDecl', ['TypeDecl', ['IdentifierType', ['Node']]]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 694 | |
eli.bendersky | 697ecc5 | 2011-02-10 07:05:13 +0200 | [diff] [blame] | 695 | def test_anonymous_struct_union(self): |
| 696 | s1 = """ |
| 697 | union |
| 698 | { |
| 699 | union |
| 700 | { |
| 701 | int i; |
| 702 | long l; |
| 703 | }; |
| 704 | |
| 705 | struct |
| 706 | { |
| 707 | int type; |
| 708 | int intnode; |
| 709 | }; |
| 710 | } u; |
| 711 | """ |
| 712 | |
| 713 | self.assertEqual(expand_decl(self.parse(s1).ext[0]), |
| 714 | ['Decl', 'u', |
| 715 | ['TypeDecl', |
| 716 | ['Union', None, |
| 717 | [['Decl', None, |
| 718 | ['Union', None, |
| 719 | [['Decl', 'i', |
| 720 | ['TypeDecl', |
| 721 | ['IdentifierType', ['int']]]], |
| 722 | ['Decl', 'l', |
| 723 | ['TypeDecl', |
| 724 | ['IdentifierType', ['long']]]]]]], |
| 725 | ['Decl', None, |
| 726 | ['Struct', None, |
| 727 | [['Decl', 'type', |
| 728 | ['TypeDecl', |
| 729 | ['IdentifierType', ['int']]]], |
| 730 | ['Decl', 'intnode', |
| 731 | ['TypeDecl', |
| 732 | ['IdentifierType', ['int']]]]]]]]]]]) |
| 733 | |
| 734 | s2 = """ |
| 735 | struct |
| 736 | { |
| 737 | int i; |
| 738 | union |
| 739 | { |
| 740 | int id; |
| 741 | char* name; |
| 742 | }; |
| 743 | float f; |
| 744 | } joe; |
| 745 | """ |
| 746 | |
| 747 | self.assertEqual(expand_decl(self.parse(s2).ext[0]), |
| 748 | ['Decl', 'joe', |
| 749 | ['TypeDecl', |
| 750 | ['Struct', None, |
| 751 | [['Decl', 'i', |
| 752 | ['TypeDecl', |
| 753 | ['IdentifierType', ['int']]]], |
| 754 | ['Decl', None, |
| 755 | ['Union', None, |
| 756 | [['Decl', 'id', |
| 757 | ['TypeDecl', |
| 758 | ['IdentifierType', ['int']]]], |
| 759 | ['Decl', 'name', |
| 760 | ['PtrDecl', |
| 761 | ['TypeDecl', |
| 762 | ['IdentifierType', ['char']]]]]]]], |
| 763 | ['Decl', 'f', |
| 764 | ['TypeDecl', |
| 765 | ['IdentifierType', ['float']]]]]]]]) |
| 766 | |
| 767 | # ISO/IEC 9899:201x Commitee Draft 2010-11-16, N1539 |
| 768 | # section 6.7.2.1, par. 19, example 1 |
| 769 | s3 = """ |
| 770 | struct v { |
| 771 | union { |
| 772 | struct { int i, j; }; |
| 773 | struct { long k, l; } w; |
| 774 | }; |
| 775 | int m; |
| 776 | } v1; |
| 777 | """ |
| 778 | |
| 779 | self.assertEqual(expand_decl(self.parse(s3).ext[0]), |
| 780 | ['Decl', 'v1', |
| 781 | ['TypeDecl', |
| 782 | ['Struct', 'v', |
| 783 | [['Decl', None, |
| 784 | ['Union', None, |
| 785 | [['Decl', None, |
| 786 | ['Struct', None, |
| 787 | [['Decl', 'i', |
| 788 | ['TypeDecl', |
| 789 | ['IdentifierType', ['int']]]], |
| 790 | ['Decl', 'j', |
| 791 | ['TypeDecl', |
| 792 | ['IdentifierType', ['int']]]]]]], |
| 793 | ['Decl', 'w', |
| 794 | ['TypeDecl', |
| 795 | ['Struct', None, |
| 796 | [['Decl', 'k', |
| 797 | ['TypeDecl', |
| 798 | ['IdentifierType', ['long']]]], |
| 799 | ['Decl', 'l', |
| 800 | ['TypeDecl', |
| 801 | ['IdentifierType', ['long']]]]]]]]]]], |
| 802 | ['Decl', 'm', |
| 803 | ['TypeDecl', |
| 804 | ['IdentifierType', ['int']]]]]]]]) |
| 805 | |
eli.bendersky | dce29a0 | 2011-02-10 07:55:00 +0200 | [diff] [blame] | 806 | s4 = """ |
| 807 | struct v { |
| 808 | int i; |
| 809 | float; |
| 810 | } v2;""" |
eli.bendersky | 6b01179 | 2011-06-22 17:50:56 +0300 | [diff] [blame] | 811 | # just make sure this doesn't raise ParseError |
| 812 | self.parse(s4) |
eli.bendersky | dce29a0 | 2011-02-10 07:55:00 +0200 | [diff] [blame] | 813 | |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 814 | def test_struct_members_namespace(self): |
Eli Bendersky | 4d969b3 | 2013-07-13 06:07:07 -0700 | [diff] [blame] | 815 | """ Tests that structure/union member names reside in a separate |
| 816 | namespace and can be named after existing types. |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 817 | """ |
| 818 | s1 = """ |
| 819 | typedef int Name; |
| 820 | typedef Name NameArray[10]; |
| 821 | |
| 822 | struct { |
| 823 | Name Name; |
| 824 | Name NameArray[3]; |
| 825 | } sye; |
| 826 | |
| 827 | void main(void) |
| 828 | { |
| 829 | sye.Name = 1; |
| 830 | } |
| 831 | """ |
| 832 | |
| 833 | s1_ast = self.parse(s1) |
| 834 | self.assertEqual(expand_decl(s1_ast.ext[2]), |
| 835 | ['Decl', 'sye', |
| 836 | ['TypeDecl', ['Struct', None, |
| 837 | [ ['Decl', 'Name', |
| 838 | ['TypeDecl', |
| 839 | ['IdentifierType', ['Name']]]], |
| 840 | ['Decl', 'NameArray', |
| 841 | ['ArrayDecl', '3', |
| 842 | ['TypeDecl', ['IdentifierType', ['Name']]]]]]]]]) |
| 843 | self.assertEqual(s1_ast.ext[3].body.block_items[0].lvalue.field.name, 'Name') |
| 844 | |
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 845 | def test_struct_bitfields(self): |
eli.bendersky | 38ed9a9 | 2010-10-09 09:29:59 +0200 | [diff] [blame] | 846 | # a struct with two bitfields, one unnamed |
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 847 | s1 = """ |
| 848 | struct { |
| 849 | int k:6; |
| 850 | int :2; |
| 851 | } joe; |
| 852 | """ |
| 853 | |
| 854 | parsed_struct = self.parse(s1).ext[0] |
| 855 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 856 | # We can see here the name of the decl for the unnamed bitfield is |
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 857 | # None, but expand_decl doesn't show bitfield widths |
| 858 | # ... |
| 859 | self.assertEqual(expand_decl(parsed_struct), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 860 | ['Decl', 'joe', |
| 861 | ['TypeDecl', ['Struct', None, |
| 862 | [ ['Decl', 'k', |
| 863 | ['TypeDecl', |
| 864 | ['IdentifierType', ['int']]]], |
| 865 | ['Decl', None, |
| 866 | ['TypeDecl', |
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 867 | ['IdentifierType', ['int']]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 868 | |
eli.bendersky | 0e0a71f | 2010-10-09 08:32:00 +0200 | [diff] [blame] | 869 | # ... |
| 870 | # so we test them manually |
| 871 | self.assertEqual(parsed_struct.type.type.decls[0].bitsize.value, '6') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 872 | self.assertEqual(parsed_struct.type.type.decls[1].bitsize.value, '2') |
| 873 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 874 | def test_tags_namespace(self): |
| 875 | """ Tests that the tags of structs/unions/enums reside in a separate namespace and |
| 876 | can be named after existing types. |
| 877 | """ |
| 878 | s1 = """ |
| 879 | typedef int tagEntry; |
| 880 | |
| 881 | struct tagEntry |
| 882 | { |
| 883 | char* key; |
| 884 | char* value; |
| 885 | } Entry; |
| 886 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 887 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 888 | s1_ast = self.parse(s1) |
| 889 | self.assertEqual(expand_decl(s1_ast.ext[1]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 890 | ['Decl', 'Entry', |
| 891 | ['TypeDecl', ['Struct', 'tagEntry', |
| 892 | [['Decl', 'key', |
| 893 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]], |
| 894 | ['Decl', 'value', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 895 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 896 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 897 | s2 = """ |
| 898 | struct tagEntry; |
| 899 | |
| 900 | typedef struct tagEntry tagEntry; |
| 901 | |
| 902 | struct tagEntry |
| 903 | { |
| 904 | char* key; |
| 905 | char* value; |
| 906 | } Entry; |
| 907 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 908 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 909 | s2_ast = self.parse(s2) |
| 910 | self.assertEqual(expand_decl(s2_ast.ext[2]), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 911 | ['Decl', 'Entry', |
| 912 | ['TypeDecl', ['Struct', 'tagEntry', |
| 913 | [['Decl', 'key', |
| 914 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]], |
| 915 | ['Decl', 'value', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 916 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 917 | |
| 918 | s3 = """ |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 919 | typedef int mytag; |
| 920 | |
| 921 | enum mytag {ABC, CDE}; |
| 922 | enum mytag joe; |
| 923 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 924 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 925 | s3_type = self.parse(s3).ext[1].type |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 926 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 927 | self.assertTrue(isinstance(s3_type, Enum)) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 928 | self.assertEqual(s3_type.name, 'mytag') |
| 929 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 930 | def test_multi_decls(self): |
| 931 | d1 = 'int a, b;' |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 932 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 933 | self.assertEqual(self.get_decl(d1, 0), |
| 934 | ['Decl', 'a', ['TypeDecl', ['IdentifierType', ['int']]]]) |
| 935 | self.assertEqual(self.get_decl(d1, 1), |
| 936 | ['Decl', 'b', ['TypeDecl', ['IdentifierType', ['int']]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 937 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 938 | d2 = 'char* p, notp, ar[4];' |
| 939 | self.assertEqual(self.get_decl(d2, 0), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 940 | ['Decl', 'p', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 941 | ['PtrDecl', |
| 942 | ['TypeDecl', ['IdentifierType', ['char']]]]]) |
| 943 | self.assertEqual(self.get_decl(d2, 1), |
| 944 | ['Decl', 'notp', ['TypeDecl', ['IdentifierType', ['char']]]]) |
| 945 | self.assertEqual(self.get_decl(d2, 2), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 946 | ['Decl', 'ar', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 947 | ['ArrayDecl', '4', |
| 948 | ['TypeDecl', ['IdentifierType', ['char']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 949 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 950 | def test_invalid_multiple_types_error(self): |
| 951 | bad = [ |
| 952 | 'int enum {ab, cd} fubr;', |
| 953 | 'enum kid char brbr;'] |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 954 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 955 | for b in bad: |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 956 | self.assertRaises(ParseError, self.parse, b) |
Eli Bendersky | 9dadc3f | 2012-07-06 15:53:53 +0300 | [diff] [blame] | 957 | |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 958 | def test_duplicate_typedef(self): |
| 959 | """ Tests that redeclarations of existing types are parsed correctly. |
| 960 | This is non-standard, but allowed by many compilers. |
| 961 | """ |
| 962 | d1 = ''' |
Eli Bendersky | 9dadc3f | 2012-07-06 15:53:53 +0300 | [diff] [blame] | 963 | typedef int numbertype; |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 964 | typedef int numbertype; |
Eli Bendersky | 9dadc3f | 2012-07-06 15:53:53 +0300 | [diff] [blame] | 965 | ''' |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 966 | |
| 967 | self.assertEqual(self.get_decl(d1, 0), |
| 968 | ['Typedef', 'numbertype', |
| 969 | ['TypeDecl', ['IdentifierType', ['int']]]]) |
| 970 | self.assertEqual(self.get_decl(d1, 1), |
| 971 | ['Typedef', 'numbertype', |
| 972 | ['TypeDecl', ['IdentifierType', ['int']]]]) |
| 973 | |
| 974 | d2 = ''' |
| 975 | typedef int (*funcptr)(int x); |
| 976 | typedef int (*funcptr)(int x); |
| 977 | ''' |
| 978 | self.assertEqual(self.get_decl(d2, 0), |
| 979 | ['Typedef', 'funcptr', |
| 980 | ['PtrDecl', ['FuncDecl', |
| 981 | [['Decl', 'x', ['TypeDecl', ['IdentifierType', ['int']]]]], |
| 982 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
| 983 | self.assertEqual(self.get_decl(d2, 1), |
| 984 | ['Typedef', 'funcptr', |
| 985 | ['PtrDecl', ['FuncDecl', |
| 986 | [['Decl', 'x', ['TypeDecl', ['IdentifierType', ['int']]]]], |
| 987 | ['TypeDecl', ['IdentifierType', ['int']]]]]]) |
| 988 | |
| 989 | d3 = ''' |
| 990 | typedef int numberarray[5]; |
| 991 | typedef int numberarray[5]; |
| 992 | ''' |
| 993 | self.assertEqual(self.get_decl(d3, 0), |
| 994 | ['Typedef', 'numberarray', |
| 995 | ['ArrayDecl', '5', |
| 996 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
| 997 | self.assertEqual(self.get_decl(d3, 1), |
| 998 | ['Typedef', 'numberarray', |
| 999 | ['ArrayDecl', '5', |
| 1000 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1001 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1002 | def test_decl_inits(self): |
| 1003 | d1 = 'int a = 16;' |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 1004 | #~ self.parse(d1).show() |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1005 | self.assertEqual(self.get_decl(d1), |
| 1006 | ['Decl', 'a', ['TypeDecl', ['IdentifierType', ['int']]]]) |
| 1007 | self.assertEqual(self.get_decl_init(d1), |
| 1008 | ['Constant', 'int', '16']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1009 | |
Eli Bendersky | 3b1b08d | 2012-06-15 12:37:54 +0300 | [diff] [blame] | 1010 | d1_1 = 'float f = 0xEF.56p1;' |
| 1011 | self.assertEqual(self.get_decl_init(d1_1), |
| 1012 | ['Constant', 'float', '0xEF.56p1']) |
| 1013 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1014 | d2 = 'long ar[] = {7, 8, 9};' |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 1015 | #~ self.parse(d2).show() |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1016 | self.assertEqual(self.get_decl(d2), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1017 | ['Decl', 'ar', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1018 | ['ArrayDecl', '', |
| 1019 | ['TypeDecl', ['IdentifierType', ['long']]]]]) |
| 1020 | self.assertEqual(self.get_decl_init(d2), |
| 1021 | [ ['Constant', 'int', '7'], |
| 1022 | ['Constant', 'int', '8'], |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 1023 | ['Constant', 'int', '9']]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1024 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1025 | d3 = 'char p = j;' |
| 1026 | self.assertEqual(self.get_decl(d3), |
| 1027 | ['Decl', 'p', ['TypeDecl', ['IdentifierType', ['char']]]]) |
| 1028 | self.assertEqual(self.get_decl_init(d3), |
| 1029 | ['ID', 'j']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1030 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1031 | d4 = "char x = 'c', *p = {0, 1, 2, {4, 5}, 6};" |
| 1032 | self.assertEqual(self.get_decl(d4, 0), |
| 1033 | ['Decl', 'x', ['TypeDecl', ['IdentifierType', ['char']]]]) |
| 1034 | self.assertEqual(self.get_decl_init(d4, 0), |
| 1035 | ['Constant', 'char', "'c'"]) |
| 1036 | self.assertEqual(self.get_decl(d4, 1), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1037 | ['Decl', 'p', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1038 | ['PtrDecl', |
| 1039 | ['TypeDecl', ['IdentifierType', ['char']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1040 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1041 | self.assertEqual(self.get_decl_init(d4, 1), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1042 | [ ['Constant', 'int', '0'], |
| 1043 | ['Constant', 'int', '1'], |
| 1044 | ['Constant', 'int', '2'], |
| 1045 | [['Constant', 'int', '4'], |
| 1046 | ['Constant', 'int', '5']], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1047 | ['Constant', 'int', '6']]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1048 | |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 1049 | def test_decl_named_inits(self): |
| 1050 | d1 = 'int a = {.k = 16};' |
| 1051 | self.assertEqual(self.get_decl_init(d1), |
| 1052 | [( [['ID', 'k']], |
| 1053 | ['Constant', 'int', '16'])]) |
| 1054 | |
| 1055 | d2 = 'int a = { [0].a = {1}, [1].a[0] = 2 };' |
| 1056 | self.assertEqual(self.get_decl_init(d2), |
| 1057 | [ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1058 | ([['Constant', 'int', '0'], ['ID', 'a']], |
| 1059 | [['Constant', 'int', '1']]), |
| 1060 | ([['Constant', 'int', '1'], ['ID', 'a'], ['Constant', 'int', '0']], |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 1061 | ['Constant', 'int', '2'])]) |
| 1062 | |
| 1063 | d3 = 'int a = { .a = 1, .c = 3, 4, .b = 5};' |
| 1064 | self.assertEqual(self.get_decl_init(d3), |
| 1065 | [ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1066 | ([['ID', 'a']], ['Constant', 'int', '1']), |
| 1067 | ([['ID', 'c']], ['Constant', 'int', '3']), |
| 1068 | ['Constant', 'int', '4'], |
eli.bendersky | f890a86 | 2010-10-30 12:13:23 +0200 | [diff] [blame] | 1069 | ([['ID', 'b']], ['Constant', 'int', '5'])]) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1070 | |
| 1071 | def test_function_definitions(self): |
| 1072 | def parse_fdef(str): |
| 1073 | return self.parse(str).ext[0] |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1074 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1075 | def fdef_decl(fdef): |
| 1076 | return expand_decl(fdef.decl) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1077 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1078 | f1 = parse_fdef(''' |
| 1079 | int factorial(int p) |
| 1080 | { |
| 1081 | return 3; |
| 1082 | } |
| 1083 | ''') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1084 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1085 | self.assertEqual(fdef_decl(f1), |
| 1086 | ['Decl', 'factorial', |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1087 | ['FuncDecl', |
| 1088 | [['Decl', 'p', ['TypeDecl', ['IdentifierType', ['int']]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1089 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1090 | |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 1091 | self.assertEqual(type(f1.body.block_items[0]), Return) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1092 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1093 | f2 = parse_fdef(''' |
| 1094 | char* zzz(int p, char* c) |
| 1095 | { |
| 1096 | int a; |
| 1097 | char b; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1098 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1099 | a = b + 2; |
| 1100 | return 3; |
| 1101 | } |
| 1102 | ''') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1103 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1104 | self.assertEqual(fdef_decl(f2), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1105 | ['Decl', 'zzz', |
| 1106 | ['FuncDecl', |
| 1107 | [ ['Decl', 'p', ['TypeDecl', ['IdentifierType', ['int']]]], |
| 1108 | ['Decl', 'c', ['PtrDecl', |
| 1109 | ['TypeDecl', ['IdentifierType', ['char']]]]]], |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1110 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1111 | |
| 1112 | self.assertEqual(list(map(type, f2.body.block_items)), |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 1113 | [Decl, Decl, Assignment, Return]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1114 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1115 | f3 = parse_fdef(''' |
| 1116 | char* zzz(p, c) |
| 1117 | long p, *c; |
| 1118 | { |
| 1119 | int a; |
| 1120 | char b; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1121 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1122 | a = b + 2; |
| 1123 | return 3; |
| 1124 | } |
| 1125 | ''') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1126 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1127 | self.assertEqual(fdef_decl(f3), |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1128 | ['Decl', 'zzz', |
| 1129 | ['FuncDecl', |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1130 | [ ['ID', 'p'], |
| 1131 | ['ID', 'c']], |
| 1132 | ['PtrDecl', ['TypeDecl', ['IdentifierType', ['char']]]]]]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1133 | |
| 1134 | self.assertEqual(list(map(type, f3.body.block_items)), |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 1135 | [Decl, Decl, Assignment, Return]) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1136 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1137 | self.assertEqual(expand_decl(f3.param_decls[0]), |
| 1138 | ['Decl', 'p', ['TypeDecl', ['IdentifierType', ['long']]]]) |
| 1139 | self.assertEqual(expand_decl(f3.param_decls[1]), |
| 1140 | ['Decl', 'c', ['PtrDecl', ['TypeDecl', ['IdentifierType', ['long']]]]]) |
| 1141 | |
Sye van der Veen | 51a46ec | 2013-06-10 12:56:15 -0400 | [diff] [blame] | 1142 | # function return values and parameters may not have type information |
| 1143 | f4 = parse_fdef(''' |
| 1144 | que(p) |
| 1145 | { |
| 1146 | return 3; |
| 1147 | } |
| 1148 | ''') |
| 1149 | |
| 1150 | self.assertEqual(fdef_decl(f4), |
| 1151 | ['Decl', 'que', |
| 1152 | ['FuncDecl', |
| 1153 | [['ID', 'p']], |
| 1154 | ['TypeDecl', ['IdentifierType', ['int']]]]]) |
| 1155 | |
eli.bendersky | 7154066 | 2010-07-03 12:58:52 +0200 | [diff] [blame] | 1156 | def test_unified_string_literals(self): |
| 1157 | # simple string, for reference |
| 1158 | d1 = self.get_decl_init('char* s = "hello";') |
| 1159 | self.assertEqual(d1, ['Constant', 'string', '"hello"']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1160 | |
eli.bendersky | 7154066 | 2010-07-03 12:58:52 +0200 | [diff] [blame] | 1161 | d2 = self.get_decl_init('char* s = "hello" " world";') |
| 1162 | self.assertEqual(d2, ['Constant', 'string', '"hello world"']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1163 | |
eli.bendersky | 7154066 | 2010-07-03 12:58:52 +0200 | [diff] [blame] | 1164 | # the test case from issue 6 |
| 1165 | d3 = self.parse(r''' |
| 1166 | int main() { |
| 1167 | fprintf(stderr, |
| 1168 | "Wrong Params?\n" |
| 1169 | "Usage:\n" |
| 1170 | "%s <binary_file_path>\n", |
| 1171 | argv[0] |
| 1172 | ); |
| 1173 | } |
| 1174 | ''') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1175 | |
eli.bendersky | 7154066 | 2010-07-03 12:58:52 +0200 | [diff] [blame] | 1176 | self.assertEqual( |
eli.bendersky | ef29ff9 | 2010-10-29 16:25:43 +0200 | [diff] [blame] | 1177 | d3.ext[0].body.block_items[0].args.exprs[1].value, |
eli.bendersky | 7154066 | 2010-07-03 12:58:52 +0200 | [diff] [blame] | 1178 | r'"Wrong Params?\nUsage:\n%s <binary_file_path>\n"') |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1179 | |
eli.bendersky | 4a89f11 | 2010-07-05 06:02:03 +0200 | [diff] [blame] | 1180 | d4 = self.get_decl_init('char* s = "" "foobar";') |
| 1181 | self.assertEqual(d4, ['Constant', 'string', '"foobar"']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1182 | |
eli.bendersky | 4a89f11 | 2010-07-05 06:02:03 +0200 | [diff] [blame] | 1183 | d5 = self.get_decl_init(r'char* s = "foo\"" "bar";') |
| 1184 | self.assertEqual(d5, ['Constant', 'string', r'"foo\"bar"']) |
eli.bendersky | 7154066 | 2010-07-03 12:58:52 +0200 | [diff] [blame] | 1185 | |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1186 | def test_inline_specifier(self): |
eli.bendersky | 79d5cf6 | 2010-10-29 13:33:52 +0200 | [diff] [blame] | 1187 | ps2 = self.parse('static inline void inlinefoo(void);') |
| 1188 | self.assertEqual(ps2.ext[0].funcspec, ['inline']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1189 | |
eli.bendersky | 2e907fa | 2010-10-29 15:51:07 +0200 | [diff] [blame] | 1190 | # variable length array |
| 1191 | def test_vla(self): |
| 1192 | ps2 = self.parse(r''' |
| 1193 | int main() { |
| 1194 | int size; |
| 1195 | int var[size = 5]; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1196 | |
eli.bendersky | 2e907fa | 2010-10-29 15:51:07 +0200 | [diff] [blame] | 1197 | int var2[*]; |
| 1198 | } |
| 1199 | ''') |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1200 | self.assertTrue(isinstance(ps2.ext[0].body.block_items[1].type.dim, Assignment)) |
| 1201 | self.assertTrue(isinstance(ps2.ext[0].body.block_items[2].type.dim, ID)) |
eli.bendersky | 79d5cf6 | 2010-10-29 13:33:52 +0200 | [diff] [blame] | 1202 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1203 | |
eli.bendersky | 85d2e73 | 2011-05-20 19:47:26 +0300 | [diff] [blame] | 1204 | class TestCParser_whole_code(TestCParser_base): |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1205 | """ Testing of parsing whole chunks of code. |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1206 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1207 | Since I don't want to rely on the structure of ASTs too |
| 1208 | much, most of these tests are implemented with visitors. |
| 1209 | """ |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1210 | # A simple helper visitor that lists the values of all the |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1211 | # Constant nodes it sees. |
| 1212 | # |
| 1213 | class ConstantVisitor(NodeVisitor): |
| 1214 | def __init__(self): |
| 1215 | self.values = [] |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1216 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1217 | def visit_Constant(self, node): |
| 1218 | self.values.append(node.value) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1219 | |
| 1220 | # This visitor counts the amount of references to the ID |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1221 | # with the name provided to it in the constructor. |
| 1222 | # |
| 1223 | class IDNameCounter(NodeVisitor): |
| 1224 | def __init__(self, name): |
| 1225 | self.name = name |
| 1226 | self.nrefs = 0 |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1227 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1228 | def visit_ID(self, node): |
| 1229 | if node.name == self.name: |
| 1230 | self.nrefs += 1 |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1231 | |
| 1232 | # Counts the amount of nodes of a given class |
| 1233 | # |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1234 | class NodeKlassCounter(NodeVisitor): |
| 1235 | def __init__(self, node_klass): |
| 1236 | self.klass = node_klass |
| 1237 | self.n = 0 |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1238 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1239 | def generic_visit(self, node): |
| 1240 | if node.__class__ == self.klass: |
| 1241 | self.n += 1 |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1242 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1243 | NodeVisitor.generic_visit(self, node) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1244 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1245 | def assert_all_Constants(self, code, constants): |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1246 | """ Asserts that the list of all Constant values (by |
| 1247 | 'preorder' appearance) in the chunk of code is as |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1248 | given. |
| 1249 | """ |
eli.bendersky | ed89049 | 2010-06-25 08:25:55 +0300 | [diff] [blame] | 1250 | if isinstance(code, str): |
| 1251 | parsed = self.parse(code) |
| 1252 | else: |
| 1253 | parsed = code |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1254 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1255 | cv = self.ConstantVisitor() |
| 1256 | cv.visit(parsed) |
| 1257 | self.assertEqual(cv.values, constants) |
| 1258 | |
| 1259 | def assert_num_ID_refs(self, code, name, num): |
| 1260 | """ Asserts the number of references to the ID with |
| 1261 | the given name. |
| 1262 | """ |
| 1263 | if isinstance(code, str): |
| 1264 | parsed = self.parse(code) |
| 1265 | else: |
| 1266 | parsed = code |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1267 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1268 | iv = self.IDNameCounter(name) |
| 1269 | iv.visit(parsed) |
| 1270 | self.assertEqual(iv.nrefs, num) |
| 1271 | |
| 1272 | def assert_num_klass_nodes(self, code, klass, num): |
| 1273 | """ Asserts the amount of klass nodes in the code. |
| 1274 | """ |
| 1275 | if isinstance(code, str): |
| 1276 | parsed = self.parse(code) |
| 1277 | else: |
| 1278 | parsed = code |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1279 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1280 | cv = self.NodeKlassCounter(klass) |
| 1281 | cv.visit(parsed) |
| 1282 | self.assertEqual(cv.n, num) |
| 1283 | |
| 1284 | def test_expressions(self): |
| 1285 | e1 = '''int k = (r + 10.0) >> 6 + 8 << (3 & 0x14);''' |
| 1286 | self.assert_all_Constants(e1, ['10.0', '6', '8', '3', '0x14']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1287 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1288 | e2 = r'''char n = '\n', *prefix = "st_";''' |
| 1289 | self.assert_all_Constants(e2, [r"'\n'", '"st_"']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1290 | |
Eli Bendersky | 09e22a6 | 2013-07-02 06:00:36 -0700 | [diff] [blame] | 1291 | s1 = r'''int main() { |
| 1292 | int i = 5, j = 6, k = 1; |
| 1293 | if ((i=j && k == 1) || k > j) |
| 1294 | printf("Hello, world\n"); |
| 1295 | return 0; |
| 1296 | }''' |
| 1297 | ps1 = self.parse(s1) |
| 1298 | self.assert_all_Constants(ps1, |
| 1299 | ['5', '6', '1', '1', '"Hello, world\\n"', '0']) |
| 1300 | self.assert_num_ID_refs(ps1, 'i', 1) |
| 1301 | self.assert_num_ID_refs(ps1, 'j', 2) |
| 1302 | |
| 1303 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1304 | def test_statements(self): |
| 1305 | s1 = r''' |
| 1306 | void foo(){ |
| 1307 | if (sp == 1) |
| 1308 | if (optind >= argc || |
| 1309 | argv[optind][0] != '-' || argv[optind][1] == '\0') |
| 1310 | return -1; |
| 1311 | else if (strcmp(argv[optind], "--") == 0) { |
| 1312 | optind++; |
| 1313 | return -1; |
| 1314 | } |
| 1315 | } |
| 1316 | ''' |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1317 | |
| 1318 | self.assert_all_Constants(s1, |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1319 | ['1', '0', r"'-'", '1', r"'\0'", '1', r'"--"', '0', '1']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1320 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1321 | ps1 = self.parse(s1) |
| 1322 | self.assert_num_ID_refs(ps1, 'argv', 3) |
| 1323 | self.assert_num_ID_refs(ps1, 'optind', 5) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1324 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1325 | self.assert_num_klass_nodes(ps1, If, 3) |
| 1326 | self.assert_num_klass_nodes(ps1, Return, 2) |
| 1327 | self.assert_num_klass_nodes(ps1, FuncCall, 1) # strcmp |
| 1328 | self.assert_num_klass_nodes(ps1, BinaryOp, 7) |
| 1329 | |
| 1330 | # In the following code, Hash and Node were defined as |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1331 | # int to pacify the parser that sees they're used as |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1332 | # types |
| 1333 | # |
| 1334 | s2 = r''' |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1335 | typedef int Hash, Node; |
| 1336 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1337 | void HashDestroy(Hash* hash) |
| 1338 | { |
| 1339 | unsigned int i; |
| 1340 | |
| 1341 | if (hash == NULL) |
| 1342 | return; |
| 1343 | |
| 1344 | for (i = 0; i < hash->table_size; ++i) |
| 1345 | { |
| 1346 | Node* temp = hash->heads[i]; |
| 1347 | |
| 1348 | while (temp != NULL) |
| 1349 | { |
| 1350 | Node* temp2 = temp; |
| 1351 | |
| 1352 | free(temp->entry->key); |
| 1353 | free(temp->entry->value); |
| 1354 | free(temp->entry); |
| 1355 | |
| 1356 | temp = temp->next; |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1357 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1358 | free(temp2); |
| 1359 | } |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1360 | } |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1361 | |
| 1362 | free(hash->heads); |
| 1363 | hash->heads = NULL; |
| 1364 | |
| 1365 | free(hash); |
| 1366 | } |
| 1367 | ''' |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1368 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1369 | ps2 = self.parse(s2) |
| 1370 | self.assert_num_klass_nodes(ps2, FuncCall, 6) |
| 1371 | self.assert_num_klass_nodes(ps2, FuncDef, 1) |
| 1372 | self.assert_num_klass_nodes(ps2, For, 1) |
| 1373 | self.assert_num_klass_nodes(ps2, While, 1) |
| 1374 | self.assert_num_klass_nodes(ps2, StructRef, 10) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1375 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1376 | # declarations don't count |
| 1377 | self.assert_num_ID_refs(ps2, 'hash', 6) |
| 1378 | self.assert_num_ID_refs(ps2, 'i', 4) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1379 | |
eli.bendersky | ed89049 | 2010-06-25 08:25:55 +0300 | [diff] [blame] | 1380 | s3 = r''' |
| 1381 | void x(void) { |
| 1382 | int a, b; |
| 1383 | if (a < b) |
| 1384 | do { |
| 1385 | a = 0; |
| 1386 | } while (0); |
| 1387 | else if (a == b) { |
| 1388 | a = 1; |
| 1389 | } |
| 1390 | } |
| 1391 | ''' |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1392 | |
eli.bendersky | ed89049 | 2010-06-25 08:25:55 +0300 | [diff] [blame] | 1393 | ps3 = self.parse(s3) |
| 1394 | self.assert_num_klass_nodes(ps3, DoWhile, 1) |
| 1395 | self.assert_num_ID_refs(ps3, 'a', 4) |
| 1396 | self.assert_all_Constants(ps3, ['0', '0', '1']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1397 | |
eli.bendersky | 91c0aa3 | 2011-10-16 05:50:43 +0200 | [diff] [blame] | 1398 | def test_empty_statement(self): |
| 1399 | s1 = r''' |
| 1400 | void foo(void){ |
| 1401 | ; |
| 1402 | return; |
| 1403 | } |
| 1404 | ''' |
| 1405 | ps1 = self.parse(s1) |
| 1406 | self.assert_num_klass_nodes(ps1, EmptyStatement, 1) |
| 1407 | self.assert_num_klass_nodes(ps1, Return, 1) |
eli.bendersky | 145890d | 2010-10-29 12:02:32 +0200 | [diff] [blame] | 1408 | |
Ben | 5cd3fd6 | 2012-02-03 06:02:40 +0200 | [diff] [blame] | 1409 | def test_switch_statement(self): |
Eli Bendersky | 12f0c9d | 2012-02-03 11:22:25 +0200 | [diff] [blame] | 1410 | def assert_case_node(node, const_value): |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1411 | self.assertTrue(isinstance(node, Case)) |
| 1412 | self.assertTrue(isinstance(node.expr, Constant)) |
Eli Bendersky | 12f0c9d | 2012-02-03 11:22:25 +0200 | [diff] [blame] | 1413 | self.assertEqual(node.expr.value, const_value) |
| 1414 | |
| 1415 | def assert_default_node(node): |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1416 | self.assertTrue(isinstance(node, Default)) |
Eli Bendersky | 12f0c9d | 2012-02-03 11:22:25 +0200 | [diff] [blame] | 1417 | |
Ben | 5cd3fd6 | 2012-02-03 06:02:40 +0200 | [diff] [blame] | 1418 | s1 = r''' |
| 1419 | int foo(void) { |
| 1420 | switch (myvar) { |
| 1421 | case 10: |
| 1422 | k = 10; |
| 1423 | p = k + 1; |
| 1424 | return 10; |
| 1425 | case 20: |
| 1426 | case 30: |
| 1427 | return 20; |
| 1428 | default: |
| 1429 | break; |
| 1430 | } |
| 1431 | return 0; |
| 1432 | } |
| 1433 | ''' |
| 1434 | ps1 = self.parse(s1) |
Eli Bendersky | 12f0c9d | 2012-02-03 11:22:25 +0200 | [diff] [blame] | 1435 | switch = ps1.ext[0].body.block_items[0] |
| 1436 | |
| 1437 | block = switch.stmt.block_items |
| 1438 | assert_case_node(block[0], '10') |
| 1439 | self.assertEqual(len(block[0].stmts), 3) |
| 1440 | assert_case_node(block[1], '20') |
| 1441 | self.assertEqual(len(block[1].stmts), 0) |
| 1442 | assert_case_node(block[2], '30') |
| 1443 | self.assertEqual(len(block[2].stmts), 1) |
| 1444 | assert_default_node(block[3]) |
| 1445 | |
| 1446 | s2 = r''' |
| 1447 | int foo(void) { |
| 1448 | switch (myvar) { |
| 1449 | default: |
| 1450 | joe = moe; |
| 1451 | return 10; |
| 1452 | case 10: |
| 1453 | case 20: |
| 1454 | case 30: |
| 1455 | case 40: |
| 1456 | break; |
| 1457 | } |
| 1458 | return 0; |
| 1459 | } |
| 1460 | ''' |
| 1461 | ps2 = self.parse(s2) |
| 1462 | switch = ps2.ext[0].body.block_items[0] |
| 1463 | |
| 1464 | block = switch.stmt.block_items |
| 1465 | assert_default_node(block[0]) |
| 1466 | self.assertEqual(len(block[0].stmts), 2) |
| 1467 | assert_case_node(block[1], '10') |
| 1468 | self.assertEqual(len(block[1].stmts), 0) |
| 1469 | assert_case_node(block[2], '20') |
| 1470 | self.assertEqual(len(block[1].stmts), 0) |
| 1471 | assert_case_node(block[3], '30') |
| 1472 | self.assertEqual(len(block[1].stmts), 0) |
| 1473 | assert_case_node(block[4], '40') |
| 1474 | self.assertEqual(len(block[4].stmts), 1) |
Ben | 5cd3fd6 | 2012-02-03 06:02:40 +0200 | [diff] [blame] | 1475 | |
eli.bendersky | 145890d | 2010-10-29 12:02:32 +0200 | [diff] [blame] | 1476 | def test_for_statement(self): |
| 1477 | s2 = r''' |
| 1478 | void x(void) |
| 1479 | { |
| 1480 | int i; |
| 1481 | for (i = 0; i < 5; ++i) { |
| 1482 | x = 50; |
| 1483 | } |
| 1484 | } |
| 1485 | ''' |
| 1486 | ps2 = self.parse(s2) |
| 1487 | self.assert_num_klass_nodes(ps2, For, 1) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1488 | # here there are 3 refs to 'i' since the declaration doesn't count as |
eli.bendersky | 145890d | 2010-10-29 12:02:32 +0200 | [diff] [blame] | 1489 | # a ref in the visitor |
| 1490 | # |
| 1491 | self.assert_num_ID_refs(ps2, 'i', 3) |
Eli Bendersky | 4476d09 | 2012-12-25 14:07:57 -0800 | [diff] [blame] | 1492 | |
eli.bendersky | 145890d | 2010-10-29 12:02:32 +0200 | [diff] [blame] | 1493 | s3 = r''' |
| 1494 | void x(void) |
| 1495 | { |
| 1496 | for (int i = 0; i < 5; ++i) { |
| 1497 | x = 50; |
| 1498 | } |
| 1499 | } |
| 1500 | ''' |
| 1501 | ps3 = self.parse(s3) |
eli.bendersky | 145890d | 2010-10-29 12:02:32 +0200 | [diff] [blame] | 1502 | self.assert_num_klass_nodes(ps3, For, 1) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1503 | # here there are 2 refs to 'i' since the declaration doesn't count as |
eli.bendersky | 145890d | 2010-10-29 12:02:32 +0200 | [diff] [blame] | 1504 | # a ref in the visitor |
| 1505 | # |
| 1506 | self.assert_num_ID_refs(ps3, 'i', 2) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1507 | |
Eli Bendersky | 4476d09 | 2012-12-25 14:07:57 -0800 | [diff] [blame] | 1508 | s4 = r''' |
| 1509 | void x(void) { |
| 1510 | for (int i = 0;;) |
| 1511 | i; |
| 1512 | } |
| 1513 | ''' |
| 1514 | ps4 = self.parse(s4) |
| 1515 | self.assert_num_ID_refs(ps4, 'i', 1) |
| 1516 | |
Eli Bendersky | d097378 | 2012-01-19 08:09:33 +0200 | [diff] [blame] | 1517 | def _open_c_file(self, name): |
| 1518 | """ Find a c file by name, taking into account the current dir can be |
| 1519 | in a couple of typical places |
| 1520 | """ |
| 1521 | fullnames = [ |
| 1522 | os.path.join('c_files', name), |
| 1523 | os.path.join('tests', 'c_files', name)] |
| 1524 | for fullname in fullnames: |
| 1525 | if os.path.exists(fullname): |
| 1526 | return open(fullname, 'rU') |
| 1527 | assert False, "Unreachable" |
| 1528 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1529 | def test_whole_file(self): |
| 1530 | # See how pycparser handles a whole, real C file. |
| 1531 | # |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1532 | with self._open_c_file('memmgr_with_h.c') as f: |
| 1533 | code = f.read() |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1534 | p = self.parse(code) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1535 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1536 | self.assert_num_klass_nodes(p, FuncDef, 5) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1537 | |
| 1538 | # each FuncDef also has a FuncDecl. 4 declarations |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1539 | # + 5 definitions, overall 9 |
| 1540 | self.assert_num_klass_nodes(p, FuncDecl, 9) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1541 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1542 | self.assert_num_klass_nodes(p, Typedef, 4) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1543 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1544 | self.assertEqual(p.ext[4].coord.line, 88) |
| 1545 | self.assertEqual(p.ext[4].coord.file, "./memmgr.h") |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1546 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1547 | self.assertEqual(p.ext[6].coord.line, 10) |
| 1548 | self.assertEqual(p.ext[6].coord.file, "memmgr.c") |
| 1549 | |
| 1550 | def test_whole_file_with_stdio(self): |
| 1551 | # Parse a whole file with stdio.h included by cpp |
| 1552 | # |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1553 | with self._open_c_file('cppd_with_stdio_h.c') as f: |
| 1554 | code = f.read() |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1555 | p = self.parse(code) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1556 | |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1557 | self.assertTrue(isinstance(p.ext[0], Typedef)) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1558 | self.assertEqual(p.ext[0].coord.line, 213) |
| 1559 | self.assertEqual(p.ext[0].coord.file, "D:\eli\cpp_stuff\libc_include/stddef.h") |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1560 | |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1561 | self.assertTrue(isinstance(p.ext[-1], FuncDef)) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1562 | self.assertEqual(p.ext[-1].coord.line, 15) |
| 1563 | self.assertEqual(p.ext[-1].coord.file, "example_c_file.c") |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1564 | |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1565 | self.assertTrue(isinstance(p.ext[-8], Typedef)) |
| 1566 | self.assertTrue(isinstance(p.ext[-8].type, TypeDecl)) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1567 | self.assertEqual(p.ext[-8].name, 'cookie_io_functions_t') |
eli.bendersky | 85d2e73 | 2011-05-20 19:47:26 +0300 | [diff] [blame] | 1568 | |
| 1569 | |
| 1570 | class TestCParser_typenames(TestCParser_base): |
| 1571 | """ Test issues related to the typedef-name problem. |
| 1572 | """ |
| 1573 | def test_innerscope_typedef(self): |
| 1574 | # should fail since TT is not a type in bar |
| 1575 | s1 = r''' |
| 1576 | void foo() { |
| 1577 | typedef char TT; |
| 1578 | TT x; |
| 1579 | } |
| 1580 | void bar() { |
| 1581 | TT y; |
| 1582 | } |
| 1583 | ''' |
| 1584 | self.assertRaises(ParseError, self.parse, s1) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1585 | |
eli.bendersky | 85d2e73 | 2011-05-20 19:47:26 +0300 | [diff] [blame] | 1586 | # should succeed since TT is not a type in bar |
| 1587 | s2 = r''' |
| 1588 | void foo() { |
| 1589 | typedef char TT; |
| 1590 | TT x; |
| 1591 | } |
| 1592 | void bar() { |
| 1593 | unsigned TT; |
| 1594 | } |
| 1595 | ''' |
Eli Bendersky | a1da7fd | 2012-07-06 15:48:57 +0300 | [diff] [blame] | 1596 | self.assertTrue(isinstance(self.parse(s2), FileAST)) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1597 | |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 1598 | def test_innerscope_reuse_typedef_name(self): |
| 1599 | # identifiers can be reused in inner scopes; the original should be |
| 1600 | # restored at the end of the block |
| 1601 | s1 = r''' |
| 1602 | typedef char TT; |
| 1603 | void foo(void) { |
| 1604 | unsigned TT; |
| 1605 | TT = 10; |
| 1606 | } |
| 1607 | TT x = 5; |
| 1608 | ''' |
| 1609 | s1_ast = self.parse(s1) |
| 1610 | self.assertEqual(expand_decl(s1_ast.ext[1].body.block_items[0]), |
| 1611 | ['Decl', 'TT', ['TypeDecl', ['IdentifierType', ['unsigned']]]]) |
eli.bendersky | 85d2e73 | 2011-05-20 19:47:26 +0300 | [diff] [blame] | 1612 | |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 1613 | # this should be recognized even with an initializer |
| 1614 | s2 = r''' |
| 1615 | typedef char TT; |
| 1616 | void foo(void) { |
| 1617 | unsigned TT = 10; |
| 1618 | } |
| 1619 | ''' |
| 1620 | s2_ast = self.parse(s2) |
| 1621 | self.assertEqual(expand_decl(s2_ast.ext[1].body.block_items[0]), |
| 1622 | ['Decl', 'TT', ['TypeDecl', ['IdentifierType', ['unsigned']]]]) |
| 1623 | |
| 1624 | # before the second local variable, TT is a type; after, it's a |
| 1625 | # variable |
| 1626 | s3 = r''' |
| 1627 | typedef char TT; |
| 1628 | void foo(void) { |
| 1629 | TT tt = sizeof(TT); |
| 1630 | unsigned TT = 10; |
| 1631 | } |
| 1632 | ''' |
| 1633 | s3_ast = self.parse(s3) |
| 1634 | self.assertEqual(expand_decl(s3_ast.ext[1].body.block_items[0]), |
| 1635 | ['Decl', 'tt', ['TypeDecl', ['IdentifierType', ['TT']]]]) |
| 1636 | self.assertEqual(expand_decl(s3_ast.ext[1].body.block_items[1]), |
| 1637 | ['Decl', 'TT', ['TypeDecl', ['IdentifierType', ['unsigned']]]]) |
| 1638 | |
| 1639 | # a variable and its type can even share the same name |
| 1640 | s4 = r''' |
| 1641 | typedef char TT; |
| 1642 | void foo(void) { |
| 1643 | TT TT = sizeof(TT); |
| 1644 | unsigned uu = TT * 2; |
| 1645 | } |
| 1646 | ''' |
| 1647 | s4_ast = self.parse(s4) |
| 1648 | self.assertEqual(expand_decl(s4_ast.ext[1].body.block_items[0]), |
| 1649 | ['Decl', 'TT', ['TypeDecl', ['IdentifierType', ['TT']]]]) |
| 1650 | self.assertEqual(expand_decl(s4_ast.ext[1].body.block_items[1]), |
| 1651 | ['Decl', 'uu', ['TypeDecl', ['IdentifierType', ['unsigned']]]]) |
| 1652 | |
| 1653 | # ensure an error is raised if a type, redeclared as a variable, is |
| 1654 | # used as a type |
| 1655 | s5 = r''' |
| 1656 | typedef char TT; |
| 1657 | void foo(void) { |
| 1658 | unsigned TT = 10; |
| 1659 | TT erroneous = 20; |
| 1660 | } |
| 1661 | ''' |
| 1662 | self.assertRaises(ParseError, self.parse, s5) |
| 1663 | |
| 1664 | def test_parameter_reuse_typedef_name(self): |
| 1665 | # identifiers can be reused as parameter names; parameter name scope |
| 1666 | # begins and ends with the function body; it's important that TT is |
Eli Bendersky | 4d969b3 | 2013-07-13 06:07:07 -0700 | [diff] [blame] | 1667 | # used immediately before the LBRACE or after the RBRACE, to test |
| 1668 | # a corner case |
Sye van der Veen | 9ec6c42 | 2013-07-11 09:10:38 -0400 | [diff] [blame] | 1669 | s1 = r''' |
| 1670 | typedef char TT; |
| 1671 | void foo(unsigned TT, TT bar) { |
| 1672 | TT = 10; |
| 1673 | } |
| 1674 | TT x = 5; |
| 1675 | ''' |
| 1676 | s1_ast = self.parse(s1) |
| 1677 | self.assertEqual(expand_decl(s1_ast.ext[1].decl), |
| 1678 | ['Decl', 'foo', |
| 1679 | ['FuncDecl', |
| 1680 | [ ['Decl', 'TT', ['TypeDecl', ['IdentifierType', ['unsigned']]]], |
| 1681 | ['Decl', 'bar', ['TypeDecl', ['IdentifierType', ['TT']]]]], |
| 1682 | ['TypeDecl', ['IdentifierType', ['void']]]]]) |
| 1683 | |
| 1684 | # the scope of a parameter name in a function declaration ends at the |
| 1685 | # end of the declaration...so it is effectively never used; it's |
| 1686 | # important that TT is used immediately after the declaration, to |
| 1687 | # test a corner case |
| 1688 | s2 = r''' |
| 1689 | typedef char TT; |
| 1690 | void foo(unsigned TT, TT bar); |
| 1691 | TT x = 5; |
| 1692 | ''' |
| 1693 | s2_ast = self.parse(s2) |
| 1694 | self.assertEqual(expand_decl(s2_ast.ext[1]), |
| 1695 | ['Decl', 'foo', |
| 1696 | ['FuncDecl', |
| 1697 | [ ['Decl', 'TT', ['TypeDecl', ['IdentifierType', ['unsigned']]]], |
| 1698 | ['Decl', 'bar', ['TypeDecl', ['IdentifierType', ['TT']]]]], |
| 1699 | ['TypeDecl', ['IdentifierType', ['void']]]]]) |
| 1700 | |
| 1701 | # ensure an error is raised if a type, redeclared as a parameter, is |
| 1702 | # used as a type |
| 1703 | s3 = r''' |
| 1704 | typedef char TT; |
| 1705 | void foo(unsigned TT, TT bar) { |
| 1706 | TT erroneous = 20; |
| 1707 | } |
| 1708 | ''' |
| 1709 | self.assertRaises(ParseError, self.parse, s3) |
| 1710 | |
| 1711 | def test_nested_function_decls(self): |
| 1712 | # parameter names of nested function declarations must not escape into |
| 1713 | # the top-level function _definition's_ scope; the following must |
| 1714 | # succeed because TT is still a typedef inside foo's body |
| 1715 | s1 = r''' |
| 1716 | typedef char TT; |
| 1717 | void foo(unsigned bar(int TT)) { |
| 1718 | TT x = 10; |
| 1719 | } |
| 1720 | ''' |
| 1721 | self.assertTrue(isinstance(self.parse(s1), FileAST)) |
| 1722 | |
| 1723 | def test_samescope_reuse_name(self): |
| 1724 | # a typedef name cannot be reused as an object name in the same scope |
| 1725 | s1 = r''' |
| 1726 | typedef char TT; |
| 1727 | char TT = 5; |
| 1728 | ''' |
| 1729 | self.assertRaises(ParseError, self.parse, s1) |
| 1730 | |
| 1731 | # ...and vice-versa |
| 1732 | s2 = r''' |
| 1733 | char TT = 5; |
| 1734 | typedef char TT; |
| 1735 | ''' |
| 1736 | self.assertRaises(ParseError, self.parse, s2) |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1737 | |
| 1738 | if __name__ == '__main__': |
| 1739 | #~ suite = unittest.TestLoader().loadTestsFromNames( |
| 1740 | #~ ['test_c_parser.TestCParser_fundamentals.test_typedef']) |
Eli Bendersky | 86f2eee | 2013-01-18 06:04:01 -0800 | [diff] [blame] | 1741 | |
Eli Bendersky | 3921e8e | 2010-05-21 09:05:39 +0300 | [diff] [blame] | 1742 | #~ unittest.TextTestRunner(verbosity=2).run(suite) |
| 1743 | unittest.main() |