OLD | NEW |
1 # Python bindings for Yasm: Pyrex input file for symrec.h | 1 # Python bindings for Yasm: Pyrex input file for symrec.h |
2 # | 2 # |
3 # Copyright (C) 2006 Michael Urman, Peter Johnson | 3 # Copyright (C) 2006 Michael Urman, Peter Johnson |
4 # | 4 # |
5 # Redistribution and use in source and binary forms, with or without | 5 # Redistribution and use in source and binary forms, with or without |
6 # modification, are permitted provided that the following conditions | 6 # modification, are permitted provided that the following conditions |
7 # are met: | 7 # are met: |
8 # 1. Redistributions of source code must retain the above copyright | 8 # 1. Redistributions of source code must retain the above copyright |
9 # notice, this list of conditions and the following disclaimer. | 9 # notice, this list of conditions and the following disclaimer. |
10 # 2. Redistributions in binary form must reproduce the above copyright | 10 # 2. Redistributions in binary form must reproduce the above copyright |
11 # notice, this list of conditions and the following disclaimer in the | 11 # notice, this list of conditions and the following disclaimer in the |
12 # documentation and/or other materials provided with the distribution. | 12 # documentation and/or other materials provided with the distribution. |
13 # | 13 # |
14 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND OTHER CONTRIBUTORS ``AS IS'' | 14 # THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND OTHER CONTRIBUTORS ``AS IS'' |
15 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | 15 # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
16 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | 16 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
17 # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR OTHER CONTRIBUTORS BE | 17 # ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR OTHER CONTRIBUTORS BE |
18 # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR | 18 # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
19 # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF | 19 # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
20 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | 20 # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
21 # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN | 21 # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
22 # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) | 22 # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
23 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE | 23 # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
24 # POSSIBILITY OF SUCH DAMAGE. | 24 # POSSIBILITY OF SUCH DAMAGE. |
25 | 25 |
26 cdef class Symbol: | 26 cdef class Symbol: |
27 cdef yasm_symrec *sym | 27 cdef yasm_symrec *sym |
28 | 28 |
29 def __new__(self, symrec): | 29 def __cinit__(self, symrec): |
30 self.sym = NULL | 30 self.sym = NULL |
31 if PyCObject_Check(symrec): | 31 if PyCObject_Check(symrec): |
32 self.sym = <yasm_symrec *>__get_voidp(symrec, Symbol) | 32 self.sym = <yasm_symrec *>__get_voidp(symrec, Symbol) |
33 else: | 33 else: |
34 raise NotImplementedError | 34 raise NotImplementedError |
35 | 35 |
36 # no deref or destroy necessary | 36 # no deref or destroy necessary |
37 | 37 |
38 property name: | 38 property name: |
39 def __get__(self): return yasm_symrec_get_name(self.sym) | 39 def __get__(self): return yasm_symrec_get_name(self.sym) |
40 | 40 |
41 property status: | 41 property status: |
42 def __get__(self): | 42 def __get__(self): |
43 cdef yasm_sym_status status | 43 cdef yasm_sym_status status |
44 s = set() | 44 s = set() |
45 status = yasm_symrec_get_status(self.sym) | 45 status = yasm_symrec_get_status(self.sym) |
46 if <int>status & <int>SYM_USED: s.add('used') | 46 if <int>status & <int>YASM_SYM_USED: s.add('used') |
47 if <int>status & <int>SYM_DEFINED: s.add('defined') | 47 if <int>status & <int>YASM_SYM_DEFINED: s.add('defined') |
48 if <int>status & <int>SYM_VALUED: s.add('valued') | 48 if <int>status & <int>YASM_SYM_VALUED: s.add('valued') |
49 return s | 49 return s |
50 | 50 |
51 property in_table: | 51 property in_table: |
52 def __get__(self): | 52 def __get__(self): |
53 return bool(<int>yasm_symrec_get_status(self.sym) & | 53 return bool(<int>yasm_symrec_get_status(self.sym) & |
54 <int>SYM_NOTINTABLE) | 54 <int>YASM_SYM_NOTINTABLE) |
55 | 55 |
56 property visibility: | 56 property visibility: |
57 def __get__(self): | 57 def __get__(self): |
58 cdef yasm_sym_vis vis | 58 cdef yasm_sym_vis vis |
59 s = set() | 59 s = set() |
60 vis = yasm_symrec_get_visibility(self.sym) | 60 vis = yasm_symrec_get_visibility(self.sym) |
61 if <int>vis & <int>YASM_SYM_GLOBAL: s.add('global') | 61 if <int>vis & <int>YASM_SYM_GLOBAL: s.add('global') |
62 if <int>vis & <int>YASM_SYM_COMMON: s.add('common') | 62 if <int>vis & <int>YASM_SYM_COMMON: s.add('common') |
63 if <int>vis & <int>YASM_SYM_EXTERN: s.add('extern') | 63 if <int>vis & <int>YASM_SYM_EXTERN: s.add('extern') |
64 if <int>vis & <int>YASM_SYM_DLOCAL: s.add('dlocal') | 64 if <int>vis & <int>YASM_SYM_DLOCAL: s.add('dlocal') |
(...skipping 52 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
117 <void *>symbol) | 117 <void *>symbol) |
118 Py_INCREF(symbol) # We're keeping a reference on the C side! | 118 Py_INCREF(symbol) # We're keeping a reference on the C side! |
119 return symbol | 119 return symbol |
120 | 120 |
121 cdef class Bytecode | 121 cdef class Bytecode |
122 cdef class SymbolTable | 122 cdef class SymbolTable |
123 | 123 |
124 cdef class SymbolTableKeyIterator: | 124 cdef class SymbolTableKeyIterator: |
125 cdef yasm_symtab_iter *iter | 125 cdef yasm_symtab_iter *iter |
126 | 126 |
127 def __new__(self, symtab): | 127 def __cinit__(self, symtab): |
128 if not isinstance(symtab, SymbolTable): | 128 if not isinstance(symtab, SymbolTable): |
129 raise TypeError | 129 raise TypeError |
130 self.iter = yasm_symtab_first((<SymbolTable>symtab).symtab) | 130 self.iter = yasm_symtab_first((<SymbolTable>symtab).symtab) |
131 | 131 |
132 def __iter__(self): | 132 def __iter__(self): |
133 return self | 133 return self |
134 | 134 |
135 def __next__(self): | 135 def __next__(self): |
136 if self.iter == NULL: | 136 if self.iter == NULL: |
137 raise StopIteration | 137 raise StopIteration |
138 rv = yasm_symrec_get_name(yasm_symtab_iter_value(self.iter)) | 138 rv = yasm_symrec_get_name(yasm_symtab_iter_value(self.iter)) |
139 self.iter = yasm_symtab_next(self.iter) | 139 self.iter = yasm_symtab_next(self.iter) |
140 return rv | 140 return rv |
141 | 141 |
142 cdef class SymbolTableValueIterator: | 142 cdef class SymbolTableValueIterator: |
143 cdef yasm_symtab_iter *iter | 143 cdef yasm_symtab_iter *iter |
144 | 144 |
145 def __new__(self, symtab): | 145 def __cinit__(self, symtab): |
146 if not isinstance(symtab, SymbolTable): | 146 if not isinstance(symtab, SymbolTable): |
147 raise TypeError | 147 raise TypeError |
148 self.iter = yasm_symtab_first((<SymbolTable>symtab).symtab) | 148 self.iter = yasm_symtab_first((<SymbolTable>symtab).symtab) |
149 | 149 |
150 def __iter__(self): | 150 def __iter__(self): |
151 return self | 151 return self |
152 | 152 |
153 def __next__(self): | 153 def __next__(self): |
154 if self.iter == NULL: | 154 if self.iter == NULL: |
155 raise StopIteration | 155 raise StopIteration |
156 rv = __make_symbol(yasm_symtab_iter_value(self.iter)) | 156 rv = __make_symbol(yasm_symtab_iter_value(self.iter)) |
157 self.iter = yasm_symtab_next(self.iter) | 157 self.iter = yasm_symtab_next(self.iter) |
158 return rv | 158 return rv |
159 | 159 |
160 cdef class SymbolTableItemIterator: | 160 cdef class SymbolTableItemIterator: |
161 cdef yasm_symtab_iter *iter | 161 cdef yasm_symtab_iter *iter |
162 | 162 |
163 def __new__(self, symtab): | 163 def __cinit__(self, symtab): |
164 if not isinstance(symtab, SymbolTable): | 164 if not isinstance(symtab, SymbolTable): |
165 raise TypeError | 165 raise TypeError |
166 self.iter = yasm_symtab_first((<SymbolTable>symtab).symtab) | 166 self.iter = yasm_symtab_first((<SymbolTable>symtab).symtab) |
167 | 167 |
168 def __iter__(self): | 168 def __iter__(self): |
169 return self | 169 return self |
170 | 170 |
171 def __next__(self): | 171 def __next__(self): |
172 cdef yasm_symrec *sym | 172 cdef yasm_symrec *sym |
173 if self.iter == NULL: | 173 if self.iter == NULL: |
174 raise StopIteration | 174 raise StopIteration |
175 sym = yasm_symtab_iter_value(self.iter) | 175 sym = yasm_symtab_iter_value(self.iter) |
176 rv = (yasm_symrec_get_name(sym), __make_symbol(sym)) | 176 rv = (yasm_symrec_get_name(sym), __make_symbol(sym)) |
177 self.iter = yasm_symtab_next(self.iter) | 177 self.iter = yasm_symtab_next(self.iter) |
178 return rv | 178 return rv |
179 | 179 |
180 cdef int __parse_vis(vis) except -1: | 180 cdef int __parse_vis(vis) except -1: |
181 if not vis or vis == 'local': return YASM_SYM_LOCAL | 181 if not vis or vis == 'local': return YASM_SYM_LOCAL |
182 if vis == 'global': return YASM_SYM_GLOBAL | 182 if vis == 'global': return YASM_SYM_GLOBAL |
183 if vis == 'common': return YASM_SYM_COMMON | 183 if vis == 'common': return YASM_SYM_COMMON |
184 if vis == 'extern': return YASM_SYM_EXTERN | 184 if vis == 'extern': return YASM_SYM_EXTERN |
185 if vis == 'dlocal': return YASM_SYM_DLOCAL | 185 if vis == 'dlocal': return YASM_SYM_DLOCAL |
186 msg = "bad visibility value %r" % vis | 186 msg = "bad visibility value %r" % vis |
187 PyErr_SetString(ValueError, msg) | 187 PyErr_SetString(ValueError, msg) |
188 return -1 | 188 return -1 |
189 | 189 |
190 cdef class SymbolTable: | 190 cdef class SymbolTable: |
191 cdef yasm_symtab *symtab | 191 cdef yasm_symtab *symtab |
192 | 192 |
193 def __new__(self): | 193 def __cinit__(self): |
194 self.symtab = yasm_symtab_create() | 194 self.symtab = yasm_symtab_create() |
195 | 195 |
196 def __dealloc__(self): | 196 def __dealloc__(self): |
197 if self.symtab != NULL: yasm_symtab_destroy(self.symtab) | 197 if self.symtab != NULL: yasm_symtab_destroy(self.symtab) |
198 | 198 |
199 def use(self, name, line): | 199 def use(self, name, line): |
200 return __make_symbol(yasm_symtab_use(self.symtab, name, line)) | 200 return __make_symbol(yasm_symtab_use(self.symtab, name, line)) |
201 | 201 |
202 def define_equ(self, name, expr, line): | 202 def define_equ(self, name, expr, line): |
203 if not isinstance(expr, Expression): | 203 if not isinstance(expr, Expression): |
(...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
276 symrec = yasm_symtab_get(self.symtab, key) | 276 symrec = yasm_symtab_get(self.symtab, key) |
277 if symrec == NULL: | 277 if symrec == NULL: |
278 return x | 278 return x |
279 return __make_symbol(symrec) | 279 return __make_symbol(symrec) |
280 | 280 |
281 def iterkeys(self): return SymbolTableKeyIterator(self) | 281 def iterkeys(self): return SymbolTableKeyIterator(self) |
282 def itervalues(self): return SymbolTableValueIterator(self) | 282 def itervalues(self): return SymbolTableValueIterator(self) |
283 def iteritems(self): return SymbolTableItemIterator(self) | 283 def iteritems(self): return SymbolTableItemIterator(self) |
284 def __iter__(self): return SymbolTableKeyIterator(self) | 284 def __iter__(self): return SymbolTableKeyIterator(self) |
285 | 285 |
OLD | NEW |