]>
Commit | Line | Data |
---|---|---|
c8042e10 DM |
1 | \r |
2 | /* Map C struct members to Python object attributes */\r | |
3 | \r | |
4 | #include "Python.h"\r | |
5 | \r | |
6 | #include "structmember.h"\r | |
7 | \r | |
8 | static PyObject *\r | |
9 | listmembers(struct memberlist *mlist)\r | |
10 | {\r | |
11 | int i, n;\r | |
12 | PyObject *v;\r | |
13 | for (n = 0; mlist[n].name != NULL; n++)\r | |
14 | ;\r | |
15 | v = PyList_New(n);\r | |
16 | if (v != NULL) {\r | |
17 | for (i = 0; i < n; i++)\r | |
18 | PyList_SetItem(v, i,\r | |
19 | PyString_FromString(mlist[i].name));\r | |
20 | if (PyErr_Occurred()) {\r | |
21 | Py_DECREF(v);\r | |
22 | v = NULL;\r | |
23 | }\r | |
24 | else {\r | |
25 | PyList_Sort(v);\r | |
26 | }\r | |
27 | }\r | |
28 | return v;\r | |
29 | }\r | |
30 | \r | |
31 | PyObject *\r | |
32 | PyMember_Get(const char *addr, struct memberlist *mlist, const char *name)\r | |
33 | {\r | |
34 | struct memberlist *l;\r | |
35 | \r | |
36 | if (strcmp(name, "__members__") == 0)\r | |
37 | return listmembers(mlist);\r | |
38 | for (l = mlist; l->name != NULL; l++) {\r | |
39 | if (strcmp(l->name, name) == 0) {\r | |
40 | PyMemberDef copy;\r | |
41 | copy.name = l->name;\r | |
42 | copy.type = l->type;\r | |
43 | copy.offset = l->offset;\r | |
44 | copy.flags = l->flags;\r | |
45 | copy.doc = NULL;\r | |
46 | return PyMember_GetOne(addr, ©);\r | |
47 | }\r | |
48 | }\r | |
49 | PyErr_SetString(PyExc_AttributeError, name);\r | |
50 | return NULL;\r | |
51 | }\r | |
52 | \r | |
53 | PyObject *\r | |
54 | PyMember_GetOne(const char *addr, PyMemberDef *l)\r | |
55 | {\r | |
56 | PyObject *v;\r | |
57 | if ((l->flags & READ_RESTRICTED) &&\r | |
58 | PyEval_GetRestricted()) {\r | |
59 | PyErr_SetString(PyExc_RuntimeError, "restricted attribute");\r | |
60 | return NULL;\r | |
61 | }\r | |
62 | addr += l->offset;\r | |
63 | switch (l->type) {\r | |
64 | case T_BOOL:\r | |
65 | v = PyBool_FromLong(*(char*)addr);\r | |
66 | break;\r | |
67 | case T_BYTE:\r | |
68 | v = PyInt_FromLong(*(char*)addr);\r | |
69 | break;\r | |
70 | case T_UBYTE:\r | |
71 | v = PyLong_FromUnsignedLong(*(unsigned char*)addr);\r | |
72 | break;\r | |
73 | case T_SHORT:\r | |
74 | v = PyInt_FromLong(*(short*)addr);\r | |
75 | break;\r | |
76 | case T_USHORT:\r | |
77 | v = PyLong_FromUnsignedLong(*(unsigned short*)addr);\r | |
78 | break;\r | |
79 | case T_INT:\r | |
80 | v = PyInt_FromLong(*(int*)addr);\r | |
81 | break;\r | |
82 | case T_UINT:\r | |
83 | v = PyLong_FromUnsignedLong(*(unsigned int*)addr);\r | |
84 | break;\r | |
85 | case T_LONG:\r | |
86 | v = PyInt_FromLong(*(long*)addr);\r | |
87 | break;\r | |
88 | case T_ULONG:\r | |
89 | v = PyLong_FromUnsignedLong(*(unsigned long*)addr);\r | |
90 | break;\r | |
91 | case T_PYSSIZET:\r | |
92 | v = PyInt_FromSsize_t(*(Py_ssize_t*)addr);\r | |
93 | break;\r | |
94 | case T_FLOAT:\r | |
95 | v = PyFloat_FromDouble((double)*(float*)addr);\r | |
96 | break;\r | |
97 | case T_DOUBLE:\r | |
98 | v = PyFloat_FromDouble(*(double*)addr);\r | |
99 | break;\r | |
100 | case T_STRING:\r | |
101 | if (*(char**)addr == NULL) {\r | |
102 | Py_INCREF(Py_None);\r | |
103 | v = Py_None;\r | |
104 | }\r | |
105 | else\r | |
106 | v = PyString_FromString(*(char**)addr);\r | |
107 | break;\r | |
108 | case T_STRING_INPLACE:\r | |
109 | v = PyString_FromString((char*)addr);\r | |
110 | break;\r | |
111 | case T_CHAR:\r | |
112 | v = PyString_FromStringAndSize((char*)addr, 1);\r | |
113 | break;\r | |
114 | case T_OBJECT:\r | |
115 | v = *(PyObject **)addr;\r | |
116 | if (v == NULL)\r | |
117 | v = Py_None;\r | |
118 | Py_INCREF(v);\r | |
119 | break;\r | |
120 | case T_OBJECT_EX:\r | |
121 | v = *(PyObject **)addr;\r | |
122 | if (v == NULL)\r | |
123 | PyErr_SetString(PyExc_AttributeError, l->name);\r | |
124 | Py_XINCREF(v);\r | |
125 | break;\r | |
126 | #ifdef HAVE_LONG_LONG\r | |
127 | case T_LONGLONG:\r | |
128 | v = PyLong_FromLongLong(*(PY_LONG_LONG *)addr);\r | |
129 | break;\r | |
130 | case T_ULONGLONG:\r | |
131 | v = PyLong_FromUnsignedLongLong(*(unsigned PY_LONG_LONG *)addr);\r | |
132 | break;\r | |
133 | #endif /* HAVE_LONG_LONG */\r | |
134 | default:\r | |
135 | PyErr_SetString(PyExc_SystemError, "bad memberdescr type");\r | |
136 | v = NULL;\r | |
137 | }\r | |
138 | return v;\r | |
139 | }\r | |
140 | \r | |
141 | int\r | |
142 | PyMember_Set(char *addr, struct memberlist *mlist, const char *name, PyObject *v)\r | |
143 | {\r | |
144 | struct memberlist *l;\r | |
145 | \r | |
146 | for (l = mlist; l->name != NULL; l++) {\r | |
147 | if (strcmp(l->name, name) == 0) {\r | |
148 | PyMemberDef copy;\r | |
149 | copy.name = l->name;\r | |
150 | copy.type = l->type;\r | |
151 | copy.offset = l->offset;\r | |
152 | copy.flags = l->flags;\r | |
153 | copy.doc = NULL;\r | |
154 | return PyMember_SetOne(addr, ©, v);\r | |
155 | }\r | |
156 | }\r | |
157 | \r | |
158 | PyErr_SetString(PyExc_AttributeError, name);\r | |
159 | return -1;\r | |
160 | }\r | |
161 | \r | |
162 | #define WARN(msg) \\r | |
163 | do { \\r | |
164 | if (PyErr_Warn(PyExc_RuntimeWarning, msg) < 0) \\r | |
165 | return -1; \\r | |
166 | } while (0)\r | |
167 | \r | |
168 | int\r | |
169 | PyMember_SetOne(char *addr, PyMemberDef *l, PyObject *v)\r | |
170 | {\r | |
171 | PyObject *oldv;\r | |
172 | \r | |
173 | addr += l->offset;\r | |
174 | \r | |
175 | if ((l->flags & READONLY))\r | |
176 | {\r | |
177 | PyErr_SetString(PyExc_TypeError, "readonly attribute");\r | |
178 | return -1;\r | |
179 | }\r | |
180 | if ((l->flags & PY_WRITE_RESTRICTED) && PyEval_GetRestricted()) {\r | |
181 | PyErr_SetString(PyExc_RuntimeError, "restricted attribute");\r | |
182 | return -1;\r | |
183 | }\r | |
184 | if (v == NULL) {\r | |
185 | if (l->type == T_OBJECT_EX) {\r | |
186 | /* Check if the attribute is set. */\r | |
187 | if (*(PyObject **)addr == NULL) {\r | |
188 | PyErr_SetString(PyExc_AttributeError, l->name);\r | |
189 | return -1;\r | |
190 | }\r | |
191 | }\r | |
192 | else if (l->type != T_OBJECT) {\r | |
193 | PyErr_SetString(PyExc_TypeError,\r | |
194 | "can't delete numeric/char attribute");\r | |
195 | return -1;\r | |
196 | }\r | |
197 | }\r | |
198 | switch (l->type) {\r | |
199 | case T_BOOL:{\r | |
200 | if (!PyBool_Check(v)) {\r | |
201 | PyErr_SetString(PyExc_TypeError,\r | |
202 | "attribute value type must be bool");\r | |
203 | return -1;\r | |
204 | }\r | |
205 | if (v == Py_True)\r | |
206 | *(char*)addr = (char) 1;\r | |
207 | else\r | |
208 | *(char*)addr = (char) 0;\r | |
209 | break;\r | |
210 | }\r | |
211 | case T_BYTE:{\r | |
212 | long long_val = PyInt_AsLong(v);\r | |
213 | if ((long_val == -1) && PyErr_Occurred())\r | |
214 | return -1;\r | |
215 | *(char*)addr = (char)long_val;\r | |
216 | /* XXX: For compatibility, only warn about truncations\r | |
217 | for now. */\r | |
218 | if ((long_val > CHAR_MAX) || (long_val < CHAR_MIN))\r | |
219 | WARN("Truncation of value to char");\r | |
220 | break;\r | |
221 | }\r | |
222 | case T_UBYTE:{\r | |
223 | long long_val = PyInt_AsLong(v);\r | |
224 | if ((long_val == -1) && PyErr_Occurred())\r | |
225 | return -1;\r | |
226 | *(unsigned char*)addr = (unsigned char)long_val;\r | |
227 | if ((long_val > UCHAR_MAX) || (long_val < 0))\r | |
228 | WARN("Truncation of value to unsigned char");\r | |
229 | break;\r | |
230 | }\r | |
231 | case T_SHORT:{\r | |
232 | long long_val = PyInt_AsLong(v);\r | |
233 | if ((long_val == -1) && PyErr_Occurred())\r | |
234 | return -1;\r | |
235 | *(short*)addr = (short)long_val;\r | |
236 | if ((long_val > SHRT_MAX) || (long_val < SHRT_MIN))\r | |
237 | WARN("Truncation of value to short");\r | |
238 | break;\r | |
239 | }\r | |
240 | case T_USHORT:{\r | |
241 | long long_val = PyInt_AsLong(v);\r | |
242 | if ((long_val == -1) && PyErr_Occurred())\r | |
243 | return -1;\r | |
244 | *(unsigned short*)addr = (unsigned short)long_val;\r | |
245 | if ((long_val > USHRT_MAX) || (long_val < 0))\r | |
246 | WARN("Truncation of value to unsigned short");\r | |
247 | break;\r | |
248 | }\r | |
249 | case T_INT:{\r | |
250 | long long_val = PyInt_AsLong(v);\r | |
251 | if ((long_val == -1) && PyErr_Occurred())\r | |
252 | return -1;\r | |
253 | *(int *)addr = (int)long_val;\r | |
254 | if ((long_val > INT_MAX) || (long_val < INT_MIN))\r | |
255 | WARN("Truncation of value to int");\r | |
256 | break;\r | |
257 | }\r | |
258 | case T_UINT:{\r | |
259 | unsigned long ulong_val = PyLong_AsUnsignedLong(v);\r | |
260 | if ((ulong_val == (unsigned long)-1) && PyErr_Occurred()) {\r | |
261 | /* XXX: For compatibility, accept negative int values\r | |
262 | as well. */\r | |
263 | PyErr_Clear();\r | |
264 | ulong_val = PyLong_AsLong(v);\r | |
265 | if ((ulong_val == (unsigned long)-1) &&\r | |
266 | PyErr_Occurred())\r | |
267 | return -1;\r | |
268 | *(unsigned int *)addr = (unsigned int)ulong_val;\r | |
269 | WARN("Writing negative value into unsigned field");\r | |
270 | } else\r | |
271 | *(unsigned int *)addr = (unsigned int)ulong_val;\r | |
272 | if (ulong_val > UINT_MAX)\r | |
273 | WARN("Truncation of value to unsigned int");\r | |
274 | break;\r | |
275 | }\r | |
276 | case T_LONG:{\r | |
277 | *(long*)addr = PyLong_AsLong(v);\r | |
278 | if ((*(long*)addr == -1) && PyErr_Occurred())\r | |
279 | return -1;\r | |
280 | break;\r | |
281 | }\r | |
282 | case T_ULONG:{\r | |
283 | *(unsigned long*)addr = PyLong_AsUnsignedLong(v);\r | |
284 | if ((*(unsigned long*)addr == (unsigned long)-1)\r | |
285 | && PyErr_Occurred()) {\r | |
286 | /* XXX: For compatibility, accept negative int values\r | |
287 | as well. */\r | |
288 | PyErr_Clear();\r | |
289 | *(unsigned long*)addr = PyLong_AsLong(v);\r | |
290 | if ((*(unsigned long*)addr == (unsigned long)-1)\r | |
291 | && PyErr_Occurred())\r | |
292 | return -1;\r | |
293 | WARN("Writing negative value into unsigned field");\r | |
294 | }\r | |
295 | break;\r | |
296 | }\r | |
297 | case T_PYSSIZET:{\r | |
298 | *(Py_ssize_t*)addr = PyInt_AsSsize_t(v);\r | |
299 | if ((*(Py_ssize_t*)addr == (Py_ssize_t)-1)\r | |
300 | && PyErr_Occurred())\r | |
301 | return -1;\r | |
302 | break;\r | |
303 | }\r | |
304 | case T_FLOAT:{\r | |
305 | double double_val = PyFloat_AsDouble(v);\r | |
306 | if ((double_val == -1) && PyErr_Occurred())\r | |
307 | return -1;\r | |
308 | *(float*)addr = (float)double_val;\r | |
309 | break;\r | |
310 | }\r | |
311 | case T_DOUBLE:\r | |
312 | *(double*)addr = PyFloat_AsDouble(v);\r | |
313 | if ((*(double*)addr == -1) && PyErr_Occurred())\r | |
314 | return -1;\r | |
315 | break;\r | |
316 | case T_OBJECT:\r | |
317 | case T_OBJECT_EX:\r | |
318 | Py_XINCREF(v);\r | |
319 | oldv = *(PyObject **)addr;\r | |
320 | *(PyObject **)addr = v;\r | |
321 | Py_XDECREF(oldv);\r | |
322 | break;\r | |
323 | case T_CHAR:\r | |
324 | if (PyString_Check(v) && PyString_Size(v) == 1) {\r | |
325 | *(char*)addr = PyString_AsString(v)[0];\r | |
326 | }\r | |
327 | else {\r | |
328 | PyErr_BadArgument();\r | |
329 | return -1;\r | |
330 | }\r | |
331 | break;\r | |
332 | case T_STRING:\r | |
333 | case T_STRING_INPLACE:\r | |
334 | PyErr_SetString(PyExc_TypeError, "readonly attribute");\r | |
335 | return -1;\r | |
336 | #ifdef HAVE_LONG_LONG\r | |
337 | case T_LONGLONG:{\r | |
338 | PY_LONG_LONG value;\r | |
339 | *(PY_LONG_LONG*)addr = value = PyLong_AsLongLong(v);\r | |
340 | if ((value == -1) && PyErr_Occurred())\r | |
341 | return -1;\r | |
342 | break;\r | |
343 | }\r | |
344 | case T_ULONGLONG:{\r | |
345 | unsigned PY_LONG_LONG value;\r | |
346 | /* ??? PyLong_AsLongLong accepts an int, but PyLong_AsUnsignedLongLong\r | |
347 | doesn't ??? */\r | |
348 | if (PyLong_Check(v))\r | |
349 | *(unsigned PY_LONG_LONG*)addr = value = PyLong_AsUnsignedLongLong(v);\r | |
350 | else\r | |
351 | *(unsigned PY_LONG_LONG*)addr = value = PyInt_AsLong(v);\r | |
352 | if ((value == (unsigned PY_LONG_LONG)-1) && PyErr_Occurred())\r | |
353 | return -1;\r | |
354 | break;\r | |
355 | }\r | |
356 | #endif /* HAVE_LONG_LONG */\r | |
357 | default:\r | |
358 | PyErr_Format(PyExc_SystemError,\r | |
359 | "bad memberdescr type for %s", l->name);\r | |
360 | return -1;\r | |
361 | }\r | |
362 | return 0;\r | |
363 | }\r |