|
6
|
1 /* Module support implementation */
|
|
|
2
|
|
|
3 #include <stdio.h>
|
|
|
4
|
|
|
5 #include "PROTO.h"
|
|
|
6 #include "object.h"
|
|
|
7 #include "intobject.h"
|
|
|
8 #include "stringobject.h"
|
|
|
9 #include "tupleobject.h"
|
|
|
10 #include "listobject.h"
|
|
|
11 #include "methodobject.h"
|
|
|
12 #include "moduleobject.h"
|
|
|
13 #include "modsupport.h"
|
|
|
14 #include "import.h"
|
|
|
15 #include "errors.h"
|
|
|
16
|
|
|
17
|
|
|
18 /* Find a method in a module's method table.
|
|
|
19 Usually called from a module's getattr method. */
|
|
|
20
|
|
|
21 object *
|
|
|
22 findmethod(ml, op, name)
|
|
|
23 struct methodlist *ml;
|
|
|
24 object *op;
|
|
|
25 char *name;
|
|
|
26 {
|
|
|
27 for (; ml->ml_name != NULL; ml++) {
|
|
|
28 if (strcmp(name, ml->ml_name) == 0)
|
|
|
29 return newmethodobject(ml->ml_name, ml->ml_meth, op);
|
|
|
30 }
|
|
|
31 err_setstr(NameError, name);
|
|
|
32 return NULL;
|
|
|
33 }
|
|
|
34
|
|
|
35
|
|
|
36 object *
|
|
|
37 initmodule(name, methods)
|
|
|
38 char *name;
|
|
|
39 struct methodlist *methods;
|
|
|
40 {
|
|
|
41 object *m, *d, *v;
|
|
|
42 struct methodlist *ml;
|
|
|
43 if ((m = new_module(name)) == NULL) {
|
|
|
44 fprintf(stderr, "initializing module: %s\n", name);
|
|
|
45 fatal("can't create a module");
|
|
|
46 }
|
|
|
47 d = getmoduledict(m);
|
|
|
48 for (ml = methods; ml->ml_name != NULL; ml++) {
|
|
|
49 v = newmethodobject(ml->ml_name, ml->ml_meth, (object *)NULL);
|
|
|
50 if (v == NULL || dictinsert(d, ml->ml_name, v) != 0) {
|
|
|
51 fprintf(stderr, "initializing module: %s\n", name);
|
|
|
52 fatal("can't initialize module");
|
|
|
53 }
|
|
|
54 DECREF(v);
|
|
|
55 }
|
|
|
56 DECREF(m);
|
|
|
57 return m; /* Yes, it still exists, in sys.modules... */
|
|
|
58 }
|
|
|
59
|
|
|
60
|
|
|
61 /* Convenience functions to set a type error exception and return 0 */
|
|
|
62
|
|
|
63 int
|
|
|
64 err_badarg()
|
|
|
65 {
|
|
|
66 err_setstr(TypeError, "illegal argument type for built-in function");
|
|
|
67 return 0;
|
|
|
68 }
|
|
|
69
|
|
|
70 object *
|
|
|
71 err_nomem()
|
|
|
72 {
|
|
|
73 err_setstr(MemoryError, "in built-in function");
|
|
|
74 return NULL;
|
|
|
75 }
|
|
|
76
|
|
|
77 /* Argument list handling tools.
|
|
|
78 All return 1 for success, or call err_set*() and return 0 for failure */
|
|
|
79
|
|
|
80 int
|
|
|
81 getnoarg(v)
|
|
|
82 object *v;
|
|
|
83 {
|
|
|
84 if (v != NULL) {
|
|
|
85 return err_badarg();
|
|
|
86 }
|
|
|
87 return 1;
|
|
|
88 }
|
|
|
89
|
|
|
90 int
|
|
|
91 getintarg(v, a)
|
|
|
92 object *v;
|
|
|
93 int *a;
|
|
|
94 {
|
|
|
95 if (v == NULL || !is_intobject(v)) {
|
|
|
96 return err_badarg();
|
|
|
97 }
|
|
|
98 *a = getintvalue(v);
|
|
|
99 return 1;
|
|
|
100 }
|
|
|
101
|
|
|
102 int
|
|
|
103 getintintarg(v, a, b)
|
|
|
104 object *v;
|
|
|
105 int *a;
|
|
|
106 int *b;
|
|
|
107 {
|
|
|
108 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
109 return err_badarg();
|
|
|
110 }
|
|
|
111 return getintarg(gettupleitem(v, 0), a) &&
|
|
|
112 getintarg(gettupleitem(v, 1), b);
|
|
|
113 }
|
|
|
114
|
|
|
115 int
|
|
|
116 getlongarg(v, a)
|
|
|
117 object *v;
|
|
|
118 long *a;
|
|
|
119 {
|
|
|
120 if (v == NULL || !is_intobject(v)) {
|
|
|
121 return err_badarg();
|
|
|
122 }
|
|
|
123 *a = getintvalue(v);
|
|
|
124 return 1;
|
|
|
125 }
|
|
|
126
|
|
|
127 int
|
|
|
128 getlonglongargs(v, a, b)
|
|
|
129 object *v;
|
|
|
130 long *a, *b;
|
|
|
131 {
|
|
|
132 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
133 return err_badarg();
|
|
|
134 }
|
|
|
135 return getlongarg(gettupleitem(v, 0), a) &&
|
|
|
136 getlongarg(gettupleitem(v, 1), b);
|
|
|
137 }
|
|
|
138
|
|
|
139 int
|
|
|
140 getlonglongobjectargs(v, a, b, c)
|
|
|
141 object *v;
|
|
|
142 long *a, *b;
|
|
|
143 object **c;
|
|
|
144 {
|
|
|
145 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
146 return err_badarg();
|
|
|
147 }
|
|
|
148 if (getlongarg(gettupleitem(v, 0), a) &&
|
|
|
149 getlongarg(gettupleitem(v, 1), b)) {
|
|
|
150 *c = gettupleitem(v, 2);
|
|
|
151 return 1;
|
|
|
152 }
|
|
|
153 else {
|
|
|
154 return err_badarg();
|
|
|
155 }
|
|
|
156 }
|
|
|
157
|
|
|
158 int
|
|
|
159 getstrarg(v, a)
|
|
|
160 object *v;
|
|
|
161 object **a;
|
|
|
162 {
|
|
|
163 if (v == NULL || !is_stringobject(v)) {
|
|
|
164 return err_badarg();
|
|
|
165 }
|
|
|
166 *a = v;
|
|
|
167 return 1;
|
|
|
168 }
|
|
|
169
|
|
|
170 int
|
|
|
171 getstrstrarg(v, a, b)
|
|
|
172 object *v;
|
|
|
173 object **a;
|
|
|
174 object **b;
|
|
|
175 {
|
|
|
176 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
177 return err_badarg();
|
|
|
178 }
|
|
|
179 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
180 getstrarg(gettupleitem(v, 1), b);
|
|
|
181 }
|
|
|
182
|
|
|
183 int
|
|
|
184 getstrstrintarg(v, a, b, c)
|
|
|
185 object *v;
|
|
|
186 object **a;
|
|
|
187 object **b;
|
|
|
188 int *c;
|
|
|
189 {
|
|
|
190 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
191 return err_badarg();
|
|
|
192 }
|
|
|
193 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
194 getstrarg(gettupleitem(v, 1), b) &&
|
|
|
195 getintarg(gettupleitem(v, 2), c);
|
|
|
196 }
|
|
|
197
|
|
|
198 int
|
|
|
199 getstrintarg(v, a, b)
|
|
|
200 object *v;
|
|
|
201 object **a;
|
|
|
202 int *b;
|
|
|
203 {
|
|
|
204 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
205 return err_badarg();
|
|
|
206 }
|
|
|
207 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
208 getintarg(gettupleitem(v, 1), b);
|
|
|
209 }
|
|
|
210
|
|
|
211 int
|
|
|
212 getintstrarg(v, a, b)
|
|
|
213 object *v;
|
|
|
214 int *a;
|
|
|
215 object **b;
|
|
|
216 {
|
|
|
217 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
218 return err_badarg();
|
|
|
219 }
|
|
|
220 return getintarg(gettupleitem(v, 0), a) &&
|
|
|
221 getstrarg(gettupleitem(v, 1), b);
|
|
|
222 }
|
|
|
223
|
|
|
224 int
|
|
|
225 getpointarg(v, a)
|
|
|
226 object *v;
|
|
|
227 int *a; /* [2] */
|
|
|
228 {
|
|
|
229 return getintintarg(v, a, a+1);
|
|
|
230 }
|
|
|
231
|
|
|
232 int
|
|
|
233 get3pointarg(v, a)
|
|
|
234 object *v;
|
|
|
235 int *a; /* [6] */
|
|
|
236 {
|
|
|
237 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
238 return err_badarg();
|
|
|
239 }
|
|
|
240 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
241 getpointarg(gettupleitem(v, 1), a+2) &&
|
|
|
242 getpointarg(gettupleitem(v, 2), a+4);
|
|
|
243 }
|
|
|
244
|
|
|
245 int
|
|
|
246 getrectarg(v, a)
|
|
|
247 object *v;
|
|
|
248 int *a; /* [2+2] */
|
|
|
249 {
|
|
|
250 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
251 return err_badarg();
|
|
|
252 }
|
|
|
253 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
254 getpointarg(gettupleitem(v, 1), a+2);
|
|
|
255 }
|
|
|
256
|
|
|
257 int
|
|
|
258 getrectintarg(v, a)
|
|
|
259 object *v;
|
|
|
260 int *a; /* [4+1] */
|
|
|
261 {
|
|
|
262 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
263 return err_badarg();
|
|
|
264 }
|
|
|
265 return getrectarg(gettupleitem(v, 0), a) &&
|
|
|
266 getintarg(gettupleitem(v, 1), a+4);
|
|
|
267 }
|
|
|
268
|
|
|
269 int
|
|
|
270 getpointintarg(v, a)
|
|
|
271 object *v;
|
|
|
272 int *a; /* [2+1] */
|
|
|
273 {
|
|
|
274 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
275 return err_badarg();
|
|
|
276 }
|
|
|
277 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
278 getintarg(gettupleitem(v, 1), a+2);
|
|
|
279 }
|
|
|
280
|
|
|
281 int
|
|
|
282 getpointstrarg(v, a, b)
|
|
|
283 object *v;
|
|
|
284 int *a; /* [2] */
|
|
|
285 object **b;
|
|
|
286 {
|
|
|
287 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
288 return err_badarg();
|
|
|
289 }
|
|
|
290 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
291 getstrarg(gettupleitem(v, 1), b);
|
|
|
292 }
|
|
|
293
|
|
|
294 int
|
|
|
295 getstrintintarg(v, a, b, c)
|
|
|
296 object *v;
|
|
|
297 object *a;
|
|
|
298 int *b, *c;
|
|
|
299 {
|
|
|
300 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
301 return err_badarg();
|
|
|
302 }
|
|
|
303 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
304 getintarg(gettupleitem(v, 1), b) &&
|
|
|
305 getintarg(gettupleitem(v, 2), c);
|
|
|
306 }
|
|
|
307
|
|
|
308 int
|
|
|
309 getrectpointarg(v, a)
|
|
|
310 object *v;
|
|
|
311 int *a; /* [4+2] */
|
|
|
312 {
|
|
|
313 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
314 return err_badarg();
|
|
|
315 }
|
|
|
316 return getrectarg(gettupleitem(v, 0), a) &&
|
|
|
317 getpointarg(gettupleitem(v, 1), a+4);
|
|
|
318 }
|
|
|
319
|
|
|
320 int
|
|
|
321 getlongtuplearg(args, a, n)
|
|
|
322 object *args;
|
|
|
323 long *a; /* [n] */
|
|
|
324 int n;
|
|
|
325 {
|
|
|
326 int i;
|
|
|
327 if (!is_tupleobject(args) || gettuplesize(args) != n) {
|
|
|
328 return err_badarg();
|
|
|
329 }
|
|
|
330 for (i = 0; i < n; i++) {
|
|
|
331 object *v = gettupleitem(args, i);
|
|
|
332 if (!is_intobject(v)) {
|
|
|
333 return err_badarg();
|
|
|
334 }
|
|
|
335 a[i] = getintvalue(v);
|
|
|
336 }
|
|
|
337 return 1;
|
|
|
338 }
|
|
|
339
|
|
|
340 int
|
|
|
341 getshorttuplearg(args, a, n)
|
|
|
342 object *args;
|
|
|
343 short *a; /* [n] */
|
|
|
344 int n;
|
|
|
345 {
|
|
|
346 int i;
|
|
|
347 if (!is_tupleobject(args) || gettuplesize(args) != n) {
|
|
|
348 return err_badarg();
|
|
|
349 }
|
|
|
350 for (i = 0; i < n; i++) {
|
|
|
351 object *v = gettupleitem(args, i);
|
|
|
352 if (!is_intobject(v)) {
|
|
|
353 return err_badarg();
|
|
|
354 }
|
|
|
355 a[i] = getintvalue(v);
|
|
|
356 }
|
|
|
357 return 1;
|
|
|
358 }
|
|
|
359
|
|
|
360 int
|
|
|
361 getlonglistarg(args, a, n)
|
|
|
362 object *args;
|
|
|
363 long *a; /* [n] */
|
|
|
364 int n;
|
|
|
365 {
|
|
|
366 int i;
|
|
|
367 if (!is_listobject(args) || getlistsize(args) != n) {
|
|
|
368 return err_badarg();
|
|
|
369 }
|
|
|
370 for (i = 0; i < n; i++) {
|
|
|
371 object *v = getlistitem(args, i);
|
|
|
372 if (!is_intobject(v)) {
|
|
|
373 return err_badarg();
|
|
|
374 }
|
|
|
375 a[i] = getintvalue(v);
|
|
|
376 }
|
|
|
377 return 1;
|
|
|
378 }
|
|
|
379
|
|
|
380 int
|
|
|
381 getshortlistarg(args, a, n)
|
|
|
382 object *args;
|
|
|
383 short *a; /* [n] */
|
|
|
384 int n;
|
|
|
385 {
|
|
|
386 int i;
|
|
|
387 if (!is_listobject(args) || getlistsize(args) != n) {
|
|
|
388 return err_badarg();
|
|
|
389 }
|
|
|
390 for (i = 0; i < n; i++) {
|
|
|
391 object *v = getlistitem(args, i);
|
|
|
392 if (!is_intobject(v)) {
|
|
|
393 return err_badarg();
|
|
|
394 }
|
|
|
395 a[i] = getintvalue(v);
|
|
|
396 }
|
|
|
397 return 1;
|
|
|
398 }
|
