|
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 /* Argument list handling tools.
|
|
|
62 All return 1 for success, or call err_set*() and return 0 for failure */
|
|
|
63
|
|
|
64 int
|
|
|
65 getnoarg(v)
|
|
|
66 object *v;
|
|
|
67 {
|
|
|
68 if (v != NULL) {
|
|
|
69 return err_badarg();
|
|
|
70 }
|
|
|
71 return 1;
|
|
|
72 }
|
|
|
73
|
|
|
74 int
|
|
|
75 getintarg(v, a)
|
|
|
76 object *v;
|
|
|
77 int *a;
|
|
|
78 {
|
|
|
79 if (v == NULL || !is_intobject(v)) {
|
|
|
80 return err_badarg();
|
|
|
81 }
|
|
|
82 *a = getintvalue(v);
|
|
|
83 return 1;
|
|
|
84 }
|
|
|
85
|
|
|
86 int
|
|
|
87 getintintarg(v, a, b)
|
|
|
88 object *v;
|
|
|
89 int *a;
|
|
|
90 int *b;
|
|
|
91 {
|
|
|
92 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
93 return err_badarg();
|
|
|
94 }
|
|
|
95 return getintarg(gettupleitem(v, 0), a) &&
|
|
|
96 getintarg(gettupleitem(v, 1), b);
|
|
|
97 }
|
|
|
98
|
|
|
99 int
|
|
|
100 getlongarg(v, a)
|
|
|
101 object *v;
|
|
|
102 long *a;
|
|
|
103 {
|
|
|
104 if (v == NULL || !is_intobject(v)) {
|
|
|
105 return err_badarg();
|
|
|
106 }
|
|
|
107 *a = getintvalue(v);
|
|
|
108 return 1;
|
|
|
109 }
|
|
|
110
|
|
|
111 int
|
|
|
112 getlonglongargs(v, a, b)
|
|
|
113 object *v;
|
|
|
114 long *a, *b;
|
|
|
115 {
|
|
|
116 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
117 return err_badarg();
|
|
|
118 }
|
|
|
119 return getlongarg(gettupleitem(v, 0), a) &&
|
|
|
120 getlongarg(gettupleitem(v, 1), b);
|
|
|
121 }
|
|
|
122
|
|
|
123 int
|
|
|
124 getlonglongobjectargs(v, a, b, c)
|
|
|
125 object *v;
|
|
|
126 long *a, *b;
|
|
|
127 object **c;
|
|
|
128 {
|
|
|
129 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
130 return err_badarg();
|
|
|
131 }
|
|
|
132 if (getlongarg(gettupleitem(v, 0), a) &&
|
|
|
133 getlongarg(gettupleitem(v, 1), b)) {
|
|
|
134 *c = gettupleitem(v, 2);
|
|
|
135 return 1;
|
|
|
136 }
|
|
|
137 else {
|
|
|
138 return err_badarg();
|
|
|
139 }
|
|
|
140 }
|
|
|
141
|
|
|
142 int
|
|
|
143 getstrarg(v, a)
|
|
|
144 object *v;
|
|
|
145 object **a;
|
|
|
146 {
|
|
|
147 if (v == NULL || !is_stringobject(v)) {
|
|
|
148 return err_badarg();
|
|
|
149 }
|
|
|
150 *a = v;
|
|
|
151 return 1;
|
|
|
152 }
|
|
|
153
|
|
|
154 int
|
|
|
155 getstrstrarg(v, a, b)
|
|
|
156 object *v;
|
|
|
157 object **a;
|
|
|
158 object **b;
|
|
|
159 {
|
|
|
160 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
161 return err_badarg();
|
|
|
162 }
|
|
|
163 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
164 getstrarg(gettupleitem(v, 1), b);
|
|
|
165 }
|
|
|
166
|
|
|
167 int
|
|
|
168 getstrstrintarg(v, a, b, c)
|
|
|
169 object *v;
|
|
|
170 object **a;
|
|
|
171 object **b;
|
|
|
172 int *c;
|
|
|
173 {
|
|
|
174 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
175 return err_badarg();
|
|
|
176 }
|
|
|
177 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
178 getstrarg(gettupleitem(v, 1), b) &&
|
|
|
179 getintarg(gettupleitem(v, 2), c);
|
|
|
180 }
|
|
|
181
|
|
|
182 int
|
|
|
183 getstrintarg(v, a, b)
|
|
|
184 object *v;
|
|
|
185 object **a;
|
|
|
186 int *b;
|
|
|
187 {
|
|
|
188 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
189 return err_badarg();
|
|
|
190 }
|
|
|
191 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
192 getintarg(gettupleitem(v, 1), b);
|
|
|
193 }
|
|
|
194
|
|
|
195 int
|
|
|
196 getintstrarg(v, a, b)
|
|
|
197 object *v;
|
|
|
198 int *a;
|
|
|
199 object **b;
|
|
|
200 {
|
|
|
201 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
202 return err_badarg();
|
|
|
203 }
|
|
|
204 return getintarg(gettupleitem(v, 0), a) &&
|
|
|
205 getstrarg(gettupleitem(v, 1), b);
|
|
|
206 }
|
|
|
207
|
|
|
208 int
|
|
|
209 getpointarg(v, a)
|
|
|
210 object *v;
|
|
|
211 int *a; /* [2] */
|
|
|
212 {
|
|
|
213 return getintintarg(v, a, a+1);
|
|
|
214 }
|
|
|
215
|
|
|
216 int
|
|
|
217 get3pointarg(v, a)
|
|
|
218 object *v;
|
|
|
219 int *a; /* [6] */
|
|
|
220 {
|
|
|
221 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
222 return err_badarg();
|
|
|
223 }
|
|
|
224 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
225 getpointarg(gettupleitem(v, 1), a+2) &&
|
|
|
226 getpointarg(gettupleitem(v, 2), a+4);
|
|
|
227 }
|
|
|
228
|
|
|
229 int
|
|
|
230 getrectarg(v, a)
|
|
|
231 object *v;
|
|
|
232 int *a; /* [2+2] */
|
|
|
233 {
|
|
|
234 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
235 return err_badarg();
|
|
|
236 }
|
|
|
237 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
238 getpointarg(gettupleitem(v, 1), a+2);
|
|
|
239 }
|
|
|
240
|
|
|
241 int
|
|
|
242 getrectintarg(v, a)
|
|
|
243 object *v;
|
|
|
244 int *a; /* [4+1] */
|
|
|
245 {
|
|
|
246 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
247 return err_badarg();
|
|
|
248 }
|
|
|
249 return getrectarg(gettupleitem(v, 0), a) &&
|
|
|
250 getintarg(gettupleitem(v, 1), a+4);
|
|
|
251 }
|
|
|
252
|
|
|
253 int
|
|
|
254 getpointintarg(v, a)
|
|
|
255 object *v;
|
|
|
256 int *a; /* [2+1] */
|
|
|
257 {
|
|
|
258 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
259 return err_badarg();
|
|
|
260 }
|
|
|
261 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
262 getintarg(gettupleitem(v, 1), a+2);
|
|
|
263 }
|
|
|
264
|
|
|
265 int
|
|
|
266 getpointstrarg(v, a, b)
|
|
|
267 object *v;
|
|
|
268 int *a; /* [2] */
|
|
|
269 object **b;
|
|
|
270 {
|
|
|
271 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
272 return err_badarg();
|
|
|
273 }
|
|
|
274 return getpointarg(gettupleitem(v, 0), a) &&
|
|
|
275 getstrarg(gettupleitem(v, 1), b);
|
|
|
276 }
|
|
|
277
|
|
|
278 int
|
|
|
279 getstrintintarg(v, a, b, c)
|
|
|
280 object *v;
|
|
|
281 object *a;
|
|
|
282 int *b, *c;
|
|
|
283 {
|
|
|
284 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 3) {
|
|
|
285 return err_badarg();
|
|
|
286 }
|
|
|
287 return getstrarg(gettupleitem(v, 0), a) &&
|
|
|
288 getintarg(gettupleitem(v, 1), b) &&
|
|
|
289 getintarg(gettupleitem(v, 2), c);
|
|
|
290 }
|
|
|
291
|
|
|
292 int
|
|
|
293 getrectpointarg(v, a)
|
|
|
294 object *v;
|
|
|
295 int *a; /* [4+2] */
|
|
|
296 {
|
|
|
297 if (v == NULL || !is_tupleobject(v) || gettuplesize(v) != 2) {
|
|
|
298 return err_badarg();
|
|
|
299 }
|
|
|
300 return getrectarg(gettupleitem(v, 0), a) &&
|
|
|
301 getpointarg(gettupleitem(v, 1), a+4);
|
|
|
302 }
|
|
|
303
|
|
|
304 int
|
|
|
305 getlongtuplearg(args, a, n)
|
|
|
306 object *args;
|
|
|
307 long *a; /* [n] */
|
|
|
308 int n;
|
|
|
309 {
|
|
|
310 int i;
|
|
|
311 if (!is_tupleobject(args) || gettuplesize(args) != n) {
|
|
|
312 return err_badarg();
|
|
|
313 }
|
|
|
314 for (i = 0; i < n; i++) {
|
|
|
315 object *v = gettupleitem(args, i);
|
|
|
316 if (!is_intobject(v)) {
|
|
|
317 return err_badarg();
|
|
|
318 }
|
|
|
319 a[i] = getintvalue(v);
|
|
|
320 }
|
|
|
321 return 1;
|
|
|
322 }
|
|
|
323
|
|
|
324 int
|
|
|
325 getshorttuplearg(args, a, n)
|
|
|
326 object *args;
|
|
|
327 short *a; /* [n] */
|
|
|
328 int n;
|
|
|
329 {
|
|
|
330 int i;
|
|
|
331 if (!is_tupleobject(args) || gettuplesize(args) != n) {
|
|
|
332 return err_badarg();
|
|
|
333 }
|
|
|
334 for (i = 0; i < n; i++) {
|
|
|
335 object *v = gettupleitem(args, i);
|
|
|
336 if (!is_intobject(v)) {
|
|
|
337 return err_badarg();
|
|
|
338 }
|
|
|
339 a[i] = getintvalue(v);
|
|
|
340 }
|
|
|
341 return 1;
|
|
|
342 }
|
|
|
343
|
|
|
344 int
|
|
|
345 getlonglistarg(args, a, n)
|
|
|
346 object *args;
|
|
|
347 long *a; /* [n] */
|
|
|
348 int n;
|
|
|
349 {
|
|
|
350 int i;
|
|
|
351 if (!is_listobject(args) || getlistsize(args) != n) {
|
|
|
352 return err_badarg();
|
|
|
353 }
|
|
|
354 for (i = 0; i < n; i++) {
|
|
|
355 object *v = getlistitem(args, i);
|
|
|
356 if (!is_intobject(v)) {
|
|
|
357 return err_badarg();
|
|
|
358 }
|
|
|
359 a[i] = getintvalue(v);
|
|
|
360 }
|
|
|
361 return 1;
|
|
|
362 }
|
|
|
363
|
|
|
364 int
|
|
|
365 getshortlistarg(args, a, n)
|
|
|
366 object *args;
|
|
|
367 short *a; /* [n] */
|
|
|
368 int n;
|
|
|
369 {
|
|
|
370 int i;
|
|
|
371 if (!is_listobject(args) || getlistsize(args) != n) {
|
|
|
372 return err_badarg();
|
|
|
373 }
|
|
|
374 for (i = 0; i < n; i++) {
|
|
|
375 object *v = getlistitem(args, i);
|
|
|
376 if (!is_intobject(v)) {
|
|
|
377 return err_badarg();
|
|
|
378 }
|
|
|
379 a[i] = getintvalue(v);
|
|
|
380 }
|
|
|
381 return 1;
|
|
|
382 }
|
