aboutsummaryrefslogtreecommitdiff
blob: 50b20ed739eb81ba18b3eaa74a63ae0957d2ca42 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
import glob
from ply import lex
from ply import yacc

#lex stuff begins here

def scanincludes(string,inclst,curdir,incpaths):
    """Scan ctype files for #includes

    Adds and returns new includes to the supplied include list
    input:
    string with the file contents to scan,
    a include list
    string with the current working dir
    """
    tokens = (
            "GINCLUDE",
            "LINCLUDE",
            #"BUNDLEINC",
            "IFDEF",
            "ENDIF",
            )

    states = (
            ("com","exclusive"), #comment
            ("ifdef","inclusive"),
            )

    t_ANY_ignore = " \t"

    def t_begin_com(t):
        r"/\*"
        t.lexer.push_state("com")

    def t_com_end(t):
        r"\*/"
        t.lexer.pop_state()
        pass

    def t_line_com(t):
        r"//.*"
        pass

    def t_ANY_begin_if0(t):
        r"\#if[ \t]+0"
        t.lexer.push_state("com")

    def t_com_endif(t):
        r"\#endif"
        t.lexer.pop_state()
        pass

    def t_com_ifdef(t):
        r"\#ifdef"
        t.lexer.push_state("com")

    def t_IFDEF(t):
        r"\#ifdef[ \t]+[a-zA-Z_][a-zA-Z0-9_]*"
        t.value = t.value[6:].strip() #return the ifdef name
        t.lexer.push_state("ifdef")
        return t

    def t_ifdef_ENDIF(t):
        r"\#endif"
        t.lexer.pop_state()
        return t

    def t_GINCLUDE(t):
        r"\#[ \t]*[Ii][Nn][Cc][Ll][Uu][Dd][Ee][ \t]+<.*\.h>"
        t.value = t.value[t.value.find("<"):].strip().strip("<>")
        return t

    def t_LINCLUDE(t):
        r"\#[ \t]*[Ii][Nn][Cc][Ll][Uu][Dd][Ee][ \t]+\".*\.h\""
        t.value = t.value[t.value.find('"'):].strip().strip('""')
        return t

    def t_BUNDLEINC(t):
        r"\#[ \t]*[Ii][Nn][Cc][Ll][Uu][Dd][Ee][ \t]+<.*>"
        pass

    def t_ANY_error(t):
        #print("Illegal character '%s'" % t.value[0])
        t.lexer.skip(1)

    lexer = lex.lex()

    #lexer.input(string)
    #
    #for tok in lexer:
    #    print(tok)
    #
    #YACC stuff here

    def p_includes2(p):
        """
        includes : includes ginc
        """
        if islocalinc(p[2],curdir,incpaths):
            p[1][1].add(p[2])
        else:
            p[1][0].add(p[2])
        p[0] = p[1]

    def p_lincludes(p):
        """
        includes : includes linc
        """
        locincpaths = incpaths + [curdir + "/"]
        if islocalinc(p[2],curdir,locincpaths):
            p[1][1].add(p[2])
        else:
            p[1][0].add(p[2])
        p[0] = p[1]

    def p_ifdef(p):
        """
        includes : includes IFDEF includes ENDIF
                 | IFDEF includes ENDIF
        """
        if len(p) == 5:
            p[1][2] = addnewifdefs(p[1][2],{p[2] : p[3]})
            p[0] = p[1]
        else:
            ifdef = {}
            ifdef[p[1]] = p[2]
            p[0] = [set(),set(),ifdef]

    def p_ifdefempty(p):
        """
        includes : includes IFDEF ENDIF
                 | IFDEF ENDIF
        """
        if len(p) == 4:
            p[0] = p[1]
        else:
            p[0] = [set(),set(),{}]

    def p_ginc(p):
        "includes : ginc"
        globinc = set()
        globinc.add(p[1])
        if islocalinc(p[1], curdir, incpaths):
            p[0] = [set(),globinc,{}]
        else:
            p[0] = [globinc,set(),{}]

    def p_linc(p):
        "includes : linc"
        locinc = set()
        locinc.add(p[1])
        locincpaths = incpaths + [curdir + "/"]
        if islocalinc(p[1], curdir, locincpaths):
            p[0] = [set(),locinc,{}]
        else:
            p[0] = [locinc,set(),{}]

    def p_ginclude(p):
        "ginc : GINCLUDE"
        p[0] = p[1]

    def p_linclude(p):
        "linc : LINCLUDE"
        p[0] = p[1]

    def p_error(p):
        print("syntax error at '%s'" % p.type)
        pass

    yacc.yacc()

    newinclst = yacc.parse(string)
    if newinclst == None:
        #Check if the file didn't have any includes
        return(inclst)
    newinclst = addnewincludes(newinclst,inclst)
    return(newinclst)

def islocalinc(inc, curdir, incpaths):
    """Checks if this is a local include

    Checks if the file can be found with the path that is supplied.
    If not this is probably a global include and thus return False
    """

    for incpath in incpaths:
        #check if the path for a local inc is correct.
        #The work dir is in /tmp.
        if incpath[:4] == "/tmp":
            if not glob.glob(incpath + inc) == []:
                return True

    return False

def addnewincludes(inclist1,inclist2):
    """Adds new includes to the first inclist and return it

    Does a deeper scan for ifdef includes
    """
    #come up with better names!!
    inclist1[0] = inclist1[0] | inclist2[0]
    inclist1[1] = inclist1[1] | inclist2[1]
    inclist1[2] = addnewifdefs(inclist1[2],inclist2[2])
    return(inclist1)

def addnewifdefs(dict1,dict2):
    """Merges the ifdef section of the inclst

    Returns a new list with all of the ifdefs
    """

    if dict1 == {} and dict2 == {}:
        #we are done here
        return(dict())
    dups = dict1.keys() & dict2.keys()
    if dups == set():
        #no duplicates, empty set()
        for name in dict2:
            dict1[name] = dict2[name]
        return(dict1)

    for name in dups:
        dict1[name][0] = dict1[name][0] | dict2[name][0]
        dict1[name][1] = dict1[name][1] | dict2[name][1]
        dict1[name][2] = addnewifdefs(dict1[name][2],dict2[name][2])
        dict2.pop(name)
    for name in dict2:
        dict1[name] = dict2[name]
    return(dict1)