blob: 975b78881bd0aab8d05335fb55599a61e9d59b29 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001/***********************************************************
2Copyright 1991 by Stichting Mathematisch Centrum, Amsterdam, The
3Netherlands.
4
5 All Rights Reserved
6
7Permission to use, copy, modify, and distribute this software and its
8documentation for any purpose and without fee is hereby granted,
9provided that the above copyright notice appear in all copies and that
10both that copyright notice and this permission notice appear in
11supporting documentation, and that the names of Stichting Mathematisch
12Centrum or CWI not be used in advertising or publicity pertaining to
13distribution of the software without specific, written prior permission.
14
15STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO
16THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
17FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE
18FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
19WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
20ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
21OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
22
23******************************************************************/
24
Guido van Rossum3f5da241990-12-20 15:06:42 +000025/* Execute compiled code */
Guido van Rossum10dc2e81990-11-18 17:27:39 +000026
Guido van Rossum3f5da241990-12-20 15:06:42 +000027#include "allobjects.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000028
Guido van Rossum10dc2e81990-11-18 17:27:39 +000029#include "import.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000030#include "sysmodule.h"
31#include "compile.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000032#include "frameobject.h"
33#include "ceval.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000034#include "opcode.h"
Guido van Rossum86cd6e61991-01-21 15:12:35 +000035#include "bltinmodule.h"
Guido van Rossum3f5da241990-12-20 15:06:42 +000036#include "traceback.h"
Guido van Rossum10dc2e81990-11-18 17:27:39 +000037
Guido van Rossum3f5da241990-12-20 15:06:42 +000038#ifndef NDEBUG
39#define TRACE
Guido van Rossum10dc2e81990-11-18 17:27:39 +000040#endif
41
Guido van Rossum374a9221991-04-04 10:40:29 +000042/* Forward declarations */
43/* XXX Need prototypes */
44
45void flushline();
46static object *add();
47static object *sub();
48static object *mul();
49static object *divide();
50static object *rem();
51static object *neg();
52static object *pos();
53static object *not();
54static object *call_builtin();
55static object *call_function();
56static object *apply_subscript();
57static object *loop_subscript();
58static object *apply_slice();
59static object *cmp_outcome();
60static object *build_class();
61static int testbool();
62static int assign_subscript PROTO((object *, object *, object *));
63static int assign_slice PROTO((object *, object *, object *, object *));
64static int import_from PROTO((object *, object *, char *));
65
66
67static frameobject *current_frame;
68
69
70/* Status code for main loop (reason for stack unwind) */
71
72enum why_code {
73 WHY_NOT, /* No error */
74 WHY_EXCEPTION, /* Exception occurred */
75 WHY_RERAISE, /* Exception re-raised by 'finally' */
76 WHY_RETURN, /* 'return' statement */
77 WHY_BREAK /* 'break' statement */
78};
79
80
81/* Interpreter main loop */
82
83object *
84eval_code(co, globals, locals, arg)
85 codeobject *co;
86 object *globals;
87 object *locals;
88 object *arg;
89{
90 register unsigned char *next_instr;
91 register int opcode; /* Current opcode */
92 register int oparg; /* Current opcode argument, if any */
93 register object **stack_pointer;
94 register enum why_code why; /* Reason for block stack unwind */
95 register int err; /* Error status -- nonzero if error */
96 register object *x; /* Result object -- NULL if error */
97 register object *v; /* Temporary objects popped off stack */
98 register object *w;
99 register object *u;
100 register object *t;
101 register frameobject *f; /* Current frame */
102 int lineno; /* Current line number */
103 object *retval; /* Return value iff why == WHY_RETURN */
104 char *name; /* Name used by some instructions */
105 FILE *fp; /* Used by print operations */
106#ifdef TRACE
107 int trace = dictlookup(globals, "__trace__") != NULL;
108#endif
109
110/* Code access macros */
111
112#define GETCONST(i) Getconst(f, i)
113#define GETNAME(i) Getname(f, i)
114#define GETNAMEV(i) Getnamev(f, i)
115#define FIRST_INSTR() (GETUSTRINGVALUE(f->f_code->co_code))
116#define INSTR_OFFSET() (next_instr - FIRST_INSTR())
117#define NEXTOP() (*next_instr++)
118#define NEXTARG() (next_instr += 2, (next_instr[-1]<<8) + next_instr[-2])
119#define JUMPTO(x) (next_instr = FIRST_INSTR() + (x))
120#define JUMPBY(x) (next_instr += (x))
121
122/* Stack manipulation macros */
123
124#define STACK_LEVEL() (stack_pointer - f->f_valuestack)
125#define EMPTY() (STACK_LEVEL() == 0)
126#define TOP() (stack_pointer[-1])
127#define BASIC_PUSH(v) (*stack_pointer++ = (v))
128#define BASIC_POP() (*--stack_pointer)
129
130#ifdef TRACE
131#define PUSH(v) (BASIC_PUSH(v), trace && prtrace(TOP(), "push"))
132#define POP() (trace && prtrace(TOP(), "pop"), BASIC_POP())
133#else
134#define PUSH(v) BASIC_PUSH(v)
135#define POP() BASIC_POP()
136#endif
137
138 f = newframeobject(
139 current_frame, /*back*/
140 co, /*code*/
141 globals, /*globals*/
142 locals, /*locals*/
143 50, /*nvalues*/
144 20); /*nblocks*/
145 if (f == NULL)
146 return NULL;
147
148 current_frame = f;
149
150 next_instr = GETUSTRINGVALUE(f->f_code->co_code);
151
152 stack_pointer = f->f_valuestack;
153
154 if (arg != NULL) {
155 INCREF(arg);
156 PUSH(arg);
157 }
158
159 why = WHY_NOT;
160 err = 0;
161 x = None; /* Not a reference, just anything non-NULL */
162 lineno = -1;
163
164 for (;;) {
Guido van Rossum4965bc81991-05-14 11:51:49 +0000165 static int ticker;
Guido van Rossum374a9221991-04-04 10:40:29 +0000166
167 /* Do periodic things */
168
169 if (--ticker < 0) {
170 ticker = 100;
171 if (intrcheck()) {
172 err_set(KeyboardInterrupt);
173 why = WHY_EXCEPTION;
174 tb_here(f, INSTR_OFFSET(), lineno);
175 goto on_error;
176 }
177 }
178
179 /* Extract opcode and argument */
180
181 opcode = NEXTOP();
182 if (HAS_ARG(opcode))
183 oparg = NEXTARG();
184
185#ifdef TRACE
186 /* Instruction tracing */
187
188 if (trace) {
189 if (HAS_ARG(opcode)) {
190 printf("%d: %d, %d\n",
191 (int) (INSTR_OFFSET() - 3),
192 opcode, oparg);
193 }
194 else {
195 printf("%d: %d\n",
196 (int) (INSTR_OFFSET() - 1), opcode);
197 }
198 }
199#endif
200
201 /* Main switch on opcode */
202
203 switch (opcode) {
204
205 /* BEWARE!
206 It is essential that any operation that fails sets either
207 x to NULL, err to nonzero, or why to anything but WHY_NOT,
208 and that no operation that succeeds does this! */
209
210 /* case STOP_CODE: this is an error! */
211
212 case POP_TOP:
213 v = POP();
214 DECREF(v);
215 break;
216
217 case ROT_TWO:
218 v = POP();
219 w = POP();
220 PUSH(v);
221 PUSH(w);
222 break;
223
224 case ROT_THREE:
225 v = POP();
226 w = POP();
227 x = POP();
228 PUSH(v);
229 PUSH(x);
230 PUSH(w);
231 break;
232
233 case DUP_TOP:
234 v = TOP();
235 INCREF(v);
236 PUSH(v);
237 break;
238
239 case UNARY_POSITIVE:
240 v = POP();
241 x = pos(v);
242 DECREF(v);
243 PUSH(x);
244 break;
245
246 case UNARY_NEGATIVE:
247 v = POP();
248 x = neg(v);
249 DECREF(v);
250 PUSH(x);
251 break;
252
253 case UNARY_NOT:
254 v = POP();
255 x = not(v);
256 DECREF(v);
257 PUSH(x);
258 break;
259
260 case UNARY_CONVERT:
261 v = POP();
262 x = reprobject(v);
263 DECREF(v);
264 PUSH(x);
265 break;
266
267 case UNARY_CALL:
268 v = POP();
Guido van Rossume8122f11991-05-05 20:03:07 +0000269 if (is_instancemethodobject(v) || is_funcobject(v))
Guido van Rossum374a9221991-04-04 10:40:29 +0000270 x = call_function(v, (object *)NULL);
271 else
272 x = call_builtin(v, (object *)NULL);
273 DECREF(v);
274 PUSH(x);
275 break;
276
277 case BINARY_MULTIPLY:
278 w = POP();
279 v = POP();
280 x = mul(v, w);
281 DECREF(v);
282 DECREF(w);
283 PUSH(x);
284 break;
285
286 case BINARY_DIVIDE:
287 w = POP();
288 v = POP();
289 x = divide(v, w);
290 DECREF(v);
291 DECREF(w);
292 PUSH(x);
293 break;
294
295 case BINARY_MODULO:
296 w = POP();
297 v = POP();
298 x = rem(v, w);
299 DECREF(v);
300 DECREF(w);
301 PUSH(x);
302 break;
303
304 case BINARY_ADD:
305 w = POP();
306 v = POP();
307 x = add(v, w);
308 DECREF(v);
309 DECREF(w);
310 PUSH(x);
311 break;
312
313 case BINARY_SUBTRACT:
314 w = POP();
315 v = POP();
316 x = sub(v, w);
317 DECREF(v);
318 DECREF(w);
319 PUSH(x);
320 break;
321
322 case BINARY_SUBSCR:
323 w = POP();
324 v = POP();
325 x = apply_subscript(v, w);
326 DECREF(v);
327 DECREF(w);
328 PUSH(x);
329 break;
330
331 case BINARY_CALL:
332 w = POP();
333 v = POP();
Guido van Rossume8122f11991-05-05 20:03:07 +0000334 if (is_instancemethodobject(v) || is_funcobject(v))
Guido van Rossum374a9221991-04-04 10:40:29 +0000335 x = call_function(v, w);
336 else
337 x = call_builtin(v, w);
338 DECREF(v);
339 DECREF(w);
340 PUSH(x);
341 break;
342
343 case SLICE+0:
344 case SLICE+1:
345 case SLICE+2:
346 case SLICE+3:
347 if ((opcode-SLICE) & 2)
348 w = POP();
349 else
350 w = NULL;
351 if ((opcode-SLICE) & 1)
352 v = POP();
353 else
354 v = NULL;
355 u = POP();
356 x = apply_slice(u, v, w);
357 DECREF(u);
358 XDECREF(v);
359 XDECREF(w);
360 PUSH(x);
361 break;
362
363 case STORE_SLICE+0:
364 case STORE_SLICE+1:
365 case STORE_SLICE+2:
366 case STORE_SLICE+3:
367 if ((opcode-STORE_SLICE) & 2)
368 w = POP();
369 else
370 w = NULL;
371 if ((opcode-STORE_SLICE) & 1)
372 v = POP();
373 else
374 v = NULL;
375 u = POP();
376 t = POP();
377 err = assign_slice(u, v, w, t); /* u[v:w] = t */
378 DECREF(t);
379 DECREF(u);
380 XDECREF(v);
381 XDECREF(w);
382 break;
383
384 case DELETE_SLICE+0:
385 case DELETE_SLICE+1:
386 case DELETE_SLICE+2:
387 case DELETE_SLICE+3:
388 if ((opcode-DELETE_SLICE) & 2)
389 w = POP();
390 else
391 w = NULL;
392 if ((opcode-DELETE_SLICE) & 1)
393 v = POP();
394 else
395 v = NULL;
396 u = POP();
397 err = assign_slice(u, v, w, (object *)NULL);
398 /* del u[v:w] */
399 DECREF(u);
400 XDECREF(v);
401 XDECREF(w);
402 break;
403
404 case STORE_SUBSCR:
405 w = POP();
406 v = POP();
407 u = POP();
408 /* v[w] = u */
409 err = assign_subscript(v, w, u);
410 DECREF(u);
411 DECREF(v);
412 DECREF(w);
413 break;
414
415 case DELETE_SUBSCR:
416 w = POP();
417 v = POP();
418 /* del v[w] */
419 err = assign_subscript(v, w, (object *)NULL);
420 DECREF(v);
421 DECREF(w);
422 break;
423
424 case PRINT_EXPR:
425 v = POP();
426 fp = sysgetfile("stdout", stdout);
427 /* Print value except if procedure result */
428 if (v != None) {
429 flushline();
430 softspace(sysget("stdout"), 1);
Guido van Rossum90933611991-06-07 16:10:43 +0000431 err = printobject(v, fp, 0);
Guido van Rossum374a9221991-04-04 10:40:29 +0000432 flushline();
433 }
434 DECREF(v);
435 break;
436
437 case PRINT_ITEM:
438 v = POP();
439 fp = sysgetfile("stdout", stdout);
440 if (softspace(sysget("stdout"), 1))
441 fprintf(fp, " ");
442 if (is_stringobject(v)) {
443 char *s = getstringvalue(v);
444 int len = getstringsize(v);
445 fwrite(s, 1, len, fp);
446 if (len > 0 && s[len-1] == '\n')
447 softspace(sysget("stdout"), 0);
448 }
449 else {
Guido van Rossum90933611991-06-07 16:10:43 +0000450 err = printobject(v, fp, 0);
Guido van Rossum374a9221991-04-04 10:40:29 +0000451 }
452 DECREF(v);
453 break;
454
455 case PRINT_NEWLINE:
456 fp = sysgetfile("stdout", stdout);
457 fprintf(fp, "\n");
458 softspace(sysget("stdout"), 0);
459 break;
460
461 case BREAK_LOOP:
462 why = WHY_BREAK;
463 break;
464
465 case RAISE_EXCEPTION:
466 v = POP();
467 w = POP();
468 if (!is_stringobject(w))
469 err_setstr(TypeError,
470 "exceptions must be strings");
471 else
472 err_setval(w, v);
473 DECREF(v);
474 DECREF(w);
475 why = WHY_EXCEPTION;
476 break;
477
478 case LOAD_LOCALS:
479 v = f->f_locals;
480 INCREF(v);
481 PUSH(v);
482 break;
483
484 case RETURN_VALUE:
485 retval = POP();
486 why = WHY_RETURN;
487 break;
488
489 case REQUIRE_ARGS:
490 if (EMPTY()) {
491 err_setstr(TypeError,
492 "function expects argument(s)");
493 why = WHY_EXCEPTION;
494 }
495 break;
496
497 case REFUSE_ARGS:
498 if (!EMPTY()) {
499 err_setstr(TypeError,
500 "function expects no argument(s)");
501 why = WHY_EXCEPTION;
502 }
503 break;
504
505 case BUILD_FUNCTION:
506 v = POP();
507 x = newfuncobject(v, f->f_globals);
508 DECREF(v);
509 PUSH(x);
510 break;
511
512 case POP_BLOCK:
513 {
514 block *b = pop_block(f);
515 while (STACK_LEVEL() > b->b_level) {
516 v = POP();
517 DECREF(v);
518 }
519 }
520 break;
521
522 case END_FINALLY:
523 v = POP();
524 if (is_intobject(v)) {
525 why = (enum why_code) getintvalue(v);
526 if (why == WHY_RETURN)
527 retval = POP();
528 }
529 else if (is_stringobject(v)) {
530 w = POP();
531 err_setval(v, w);
532 DECREF(w);
533 w = POP();
534 tb_store(w);
535 DECREF(w);
536 why = WHY_RERAISE;
537 }
538 else if (v != None) {
539 err_setstr(SystemError,
540 "'finally' pops bad exception");
541 why = WHY_EXCEPTION;
542 }
543 DECREF(v);
544 break;
545
546 case BUILD_CLASS:
547 w = POP();
548 v = POP();
549 x = build_class(v, w);
550 PUSH(x);
551 DECREF(v);
552 DECREF(w);
553 break;
554
555 case STORE_NAME:
556 w = GETNAMEV(oparg);
557 v = POP();
558 err = dict2insert(f->f_locals, w, v);
559 DECREF(v);
560 break;
561
562 case DELETE_NAME:
563 name = GETNAME(oparg);
564 if ((err = dictremove(f->f_locals, name)) != 0)
565 err_setstr(NameError, name);
566 break;
567
568 case UNPACK_TUPLE:
569 v = POP();
570 if (!is_tupleobject(v)) {
571 err_setstr(TypeError, "unpack non-tuple");
572 why = WHY_EXCEPTION;
573 }
574 else if (gettuplesize(v) != oparg) {
575 err_setstr(RuntimeError,
576 "unpack tuple of wrong size");
577 why = WHY_EXCEPTION;
578 }
579 else {
580 for (; --oparg >= 0; ) {
581 w = gettupleitem(v, oparg);
582 INCREF(w);
583 PUSH(w);
584 }
585 }
586 DECREF(v);
587 break;
588
589 case UNPACK_LIST:
590 v = POP();
591 if (!is_listobject(v)) {
592 err_setstr(TypeError, "unpack non-list");
593 why = WHY_EXCEPTION;
594 }
595 else if (getlistsize(v) != oparg) {
596 err_setstr(RuntimeError,
597 "unpack list of wrong size");
598 why = WHY_EXCEPTION;
599 }
600 else {
601 for (; --oparg >= 0; ) {
602 w = getlistitem(v, oparg);
603 INCREF(w);
604 PUSH(w);
605 }
606 }
607 DECREF(v);
608 break;
609
610 case STORE_ATTR:
611 name = GETNAME(oparg);
612 v = POP();
613 u = POP();
614 err = setattr(v, name, u); /* v.name = u */
615 DECREF(v);
616 DECREF(u);
617 break;
618
619 case DELETE_ATTR:
620 name = GETNAME(oparg);
621 v = POP();
622 err = setattr(v, name, (object *)NULL);
623 /* del v.name */
624 DECREF(v);
625 break;
626
627 case LOAD_CONST:
628 x = GETCONST(oparg);
629 INCREF(x);
630 PUSH(x);
631 break;
632
633 case LOAD_NAME:
634 name = GETNAME(oparg);
635 x = dictlookup(f->f_locals, name);
636 if (x == NULL) {
637 x = dictlookup(f->f_globals, name);
638 if (x == NULL) {
639 x = getbuiltin(name);
640 if (x == NULL) {
641 err_setstr(NameError, name);
642 break;
643 }
644 }
645 }
646 INCREF(x);
647 PUSH(x);
648 break;
649
650 case LOAD_GLOBAL:
651 name = GETNAME(oparg);
652 x = dictlookup(f->f_globals, name);
653 if (x == NULL) {
654 x = getbuiltin(name);
655 if (x == NULL) {
656 err_setstr(NameError, name);
657 break;
658 }
659 }
660 INCREF(x);
661 PUSH(x);
662 break;
663
664 case LOAD_LOCAL:
665 name = GETNAME(oparg);
666 x = dictlookup(f->f_locals, name);
667 if (x == NULL) {
668 err_setstr(NameError, name);
669 break;
670 }
671 INCREF(x);
672 PUSH(x);
673 break;
674
675 case BUILD_TUPLE:
676 x = newtupleobject(oparg);
677 if (x != NULL) {
678 for (; --oparg >= 0;) {
679 w = POP();
680 err = settupleitem(x, oparg, w);
681 if (err != 0)
682 break;
683 }
684 PUSH(x);
685 }
686 break;
687
688 case BUILD_LIST:
689 x = newlistobject(oparg);
690 if (x != NULL) {
691 for (; --oparg >= 0;) {
692 w = POP();
693 err = setlistitem(x, oparg, w);
694 if (err != 0)
695 break;
696 }
697 PUSH(x);
698 }
699 break;
700
701 case BUILD_MAP:
702 x = newdictobject();
703 PUSH(x);
704 break;
705
706 case LOAD_ATTR:
707 name = GETNAME(oparg);
708 v = POP();
709 x = getattr(v, name);
710 DECREF(v);
711 PUSH(x);
712 break;
713
714 case COMPARE_OP:
715 w = POP();
716 v = POP();
717 x = cmp_outcome((enum cmp_op)oparg, v, w);
718 DECREF(v);
719 DECREF(w);
720 PUSH(x);
721 break;
722
723 case IMPORT_NAME:
724 name = GETNAME(oparg);
725 x = import_module(name);
726 XINCREF(x);
727 PUSH(x);
728 break;
729
730 case IMPORT_FROM:
731 name = GETNAME(oparg);
732 v = TOP();
733 err = import_from(f->f_locals, v, name);
734 break;
735
736 case JUMP_FORWARD:
737 JUMPBY(oparg);
738 break;
739
740 case JUMP_IF_FALSE:
741 if (!testbool(TOP()))
742 JUMPBY(oparg);
743 break;
744
745 case JUMP_IF_TRUE:
746 if (testbool(TOP()))
747 JUMPBY(oparg);
748 break;
749
750 case JUMP_ABSOLUTE:
751 JUMPTO(oparg);
752 break;
753
754 case FOR_LOOP:
755 /* for v in s: ...
756 On entry: stack contains s, i.
757 On exit: stack contains s, i+1, s[i];
758 but if loop exhausted:
759 s, i are popped, and we jump */
760 w = POP(); /* Loop index */
761 v = POP(); /* Sequence object */
762 u = loop_subscript(v, w);
763 if (u != NULL) {
764 PUSH(v);
765 x = newintobject(getintvalue(w)+1);
766 PUSH(x);
767 DECREF(w);
768 PUSH(u);
769 }
770 else {
771 DECREF(v);
772 DECREF(w);
773 /* A NULL can mean "s exhausted"
774 but also an error: */
775 if (err_occurred())
776 why = WHY_EXCEPTION;
777 else
778 JUMPBY(oparg);
779 }
780 break;
781
782 case SETUP_LOOP:
783 case SETUP_EXCEPT:
784 case SETUP_FINALLY:
785 setup_block(f, opcode, INSTR_OFFSET() + oparg,
786 STACK_LEVEL());
787 break;
788
789 case SET_LINENO:
790#ifdef TRACE
791 if (trace)
792 printf("--- Line %d ---\n", oparg);
793#endif
794 lineno = oparg;
795 break;
796
797 default:
798 fprintf(stderr,
799 "XXX lineno: %d, opcode: %d\n",
800 lineno, opcode);
801 err_setstr(SystemError, "eval_code: unknown opcode");
802 why = WHY_EXCEPTION;
803 break;
804
805 } /* switch */
806
807 on_error:
808
809 /* Quickly continue if no error occurred */
810
811 if (why == WHY_NOT) {
812 if (err == 0 && x != NULL)
813 continue; /* Normal, fast path */
814 why = WHY_EXCEPTION;
815 x = None;
816 err = 0;
817 }
818
819#ifndef NDEBUG
820 /* Double-check exception status */
821
822 if (why == WHY_EXCEPTION || why == WHY_RERAISE) {
823 if (!err_occurred()) {
824 fprintf(stderr, "XXX ghost error\n");
825 err_setstr(SystemError, "ghost error");
826 why = WHY_EXCEPTION;
827 }
828 }
829 else {
830 if (err_occurred()) {
831 fprintf(stderr, "XXX undetected error\n");
832 why = WHY_EXCEPTION;
833 }
834 }
835#endif
836
837 /* Log traceback info if this is a real exception */
838
839 if (why == WHY_EXCEPTION) {
840 int lasti = INSTR_OFFSET() - 1;
841 if (HAS_ARG(opcode))
842 lasti -= 2;
843 tb_here(f, lasti, lineno);
844 }
845
846 /* For the rest, treat WHY_RERAISE as WHY_EXCEPTION */
847
848 if (why == WHY_RERAISE)
849 why = WHY_EXCEPTION;
850
851 /* Unwind stacks if a (pseudo) exception occurred */
852
853 while (why != WHY_NOT && f->f_iblock > 0) {
854 block *b = pop_block(f);
855 while (STACK_LEVEL() > b->b_level) {
856 v = POP();
857 XDECREF(v);
858 }
859 if (b->b_type == SETUP_LOOP && why == WHY_BREAK) {
860 why = WHY_NOT;
861 JUMPTO(b->b_handler);
862 break;
863 }
864 if (b->b_type == SETUP_FINALLY ||
865 b->b_type == SETUP_EXCEPT &&
866 why == WHY_EXCEPTION) {
867 if (why == WHY_EXCEPTION) {
868 object *exc, *val;
869 err_get(&exc, &val);
870 if (val == NULL) {
871 val = None;
872 INCREF(val);
873 }
874 v = tb_fetch();
875 /* Make the raw exception data
876 available to the handler,
877 so a program can emulate the
878 Python main loop. Don't do
879 this for 'finally'. */
880 if (b->b_type == SETUP_EXCEPT) {
881#if 0 /* Oops, this breaks too many things */
882 sysset("exc_traceback", v);
883#endif
884 sysset("exc_value", val);
885 sysset("exc_type", exc);
886 err_clear();
887 }
888 PUSH(v);
889 PUSH(val);
890 PUSH(exc);
891 }
892 else {
893 if (why == WHY_RETURN)
894 PUSH(retval);
895 v = newintobject((long)why);
896 PUSH(v);
897 }
898 why = WHY_NOT;
899 JUMPTO(b->b_handler);
900 break;
901 }
902 } /* unwind stack */
903
904 /* End the loop if we still have an error (or return) */
905
906 if (why != WHY_NOT)
907 break;
908
909 } /* main loop */
910
911 /* Pop remaining stack entries */
912
913 while (!EMPTY()) {
914 v = POP();
915 XDECREF(v);
916 }
917
918 /* Restore previous frame and release the current one */
919
920 current_frame = f->f_back;
921 DECREF(f);
922
923 if (why == WHY_RETURN)
924 return retval;
925 else
926 return NULL;
927}
928
Guido van Rossum3f5da241990-12-20 15:06:42 +0000929#ifdef TRACE
930static int
931prtrace(v, str)
932 object *v;
933 char *str;
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000934{
Guido van Rossum3f5da241990-12-20 15:06:42 +0000935 printf("%s ", str);
Guido van Rossum90933611991-06-07 16:10:43 +0000936 if (printobject(v, stdout, 0) != 0)
937 err_clear(); /* Don't know what else to do */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000938 printf("\n");
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000939}
Guido van Rossum3f5da241990-12-20 15:06:42 +0000940#endif
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000941
Guido van Rossum3f5da241990-12-20 15:06:42 +0000942object *
943getlocals()
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000944{
Guido van Rossum3f5da241990-12-20 15:06:42 +0000945 if (current_frame == NULL)
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000946 return NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000947 else
948 return current_frame->f_locals;
949}
950
951object *
952getglobals()
953{
954 if (current_frame == NULL)
955 return NULL;
956 else
957 return current_frame->f_globals;
958}
959
960void
961printtraceback(fp)
962 FILE *fp;
963{
964 object *v = tb_fetch();
965 if (v != NULL) {
966 fprintf(fp, "Stack backtrace (innermost last):\n");
967 tb_print(v, fp);
968 DECREF(v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000969 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000970}
971
972
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000973void
974flushline()
975{
Guido van Rossum374a9221991-04-04 10:40:29 +0000976 if (softspace(sysget("stdout"), 0))
977 fprintf(sysgetfile("stdout", stdout), "\n");
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000978}
979
Guido van Rossum3f5da241990-12-20 15:06:42 +0000980
981/* Test a value used as condition, e.g., in a for or if statement */
982
983static int
984testbool(v)
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000985 object *v;
986{
Guido van Rossum4965bc81991-05-14 11:51:49 +0000987 if (v == None)
988 return 0;
989 if (v->ob_type->tp_as_number != NULL)
990 return (*v->ob_type->tp_as_number->nb_nonzero)(v);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000991 if (v->ob_type->tp_as_sequence != NULL)
992 return (*v->ob_type->tp_as_sequence->sq_length)(v) != 0;
993 if (v->ob_type->tp_as_mapping != NULL)
994 return (*v->ob_type->tp_as_mapping->mp_length)(v) != 0;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000995 /* All other objects are 'true' */
996 return 1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +0000997}
998
999static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001000add(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001001 object *v, *w;
1002{
1003 if (v->ob_type->tp_as_number != NULL)
1004 v = (*v->ob_type->tp_as_number->nb_add)(v, w);
1005 else if (v->ob_type->tp_as_sequence != NULL)
1006 v = (*v->ob_type->tp_as_sequence->sq_concat)(v, w);
1007 else {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001008 err_setstr(TypeError, "+ not supported by operands");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001009 return NULL;
1010 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001011 return v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001012}
1013
1014static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001015sub(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001016 object *v, *w;
1017{
1018 if (v->ob_type->tp_as_number != NULL)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001019 return (*v->ob_type->tp_as_number->nb_subtract)(v, w);
1020 err_setstr(TypeError, "bad operand type(s) for -");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001021 return NULL;
1022}
1023
1024static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001025mul(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001026 object *v, *w;
1027{
1028 typeobject *tp;
1029 if (is_intobject(v) && w->ob_type->tp_as_sequence != NULL) {
1030 /* int*sequence -- swap v and w */
1031 object *tmp = v;
1032 v = w;
1033 w = tmp;
1034 }
1035 tp = v->ob_type;
1036 if (tp->tp_as_number != NULL)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001037 return (*tp->tp_as_number->nb_multiply)(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001038 if (tp->tp_as_sequence != NULL) {
1039 if (!is_intobject(w)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001040 err_setstr(TypeError,
1041 "can't multiply sequence with non-int");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001042 return NULL;
1043 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001044 return (*tp->tp_as_sequence->sq_repeat)
1045 (v, (int)getintvalue(w));
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001046 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001047 err_setstr(TypeError, "bad operand type(s) for *");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001048 return NULL;
1049}
1050
1051static object *
Guido van Rossum40d0b7e1990-12-20 23:03:11 +00001052divide(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001053 object *v, *w;
1054{
1055 if (v->ob_type->tp_as_number != NULL)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001056 return (*v->ob_type->tp_as_number->nb_divide)(v, w);
1057 err_setstr(TypeError, "bad operand type(s) for /");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001058 return NULL;
1059}
1060
1061static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001062rem(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001063 object *v, *w;
1064{
1065 if (v->ob_type->tp_as_number != NULL)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001066 return (*v->ob_type->tp_as_number->nb_remainder)(v, w);
1067 err_setstr(TypeError, "bad operand type(s) for %");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001068 return NULL;
1069}
1070
1071static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001072neg(v)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001073 object *v;
1074{
1075 if (v->ob_type->tp_as_number != NULL)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001076 return (*v->ob_type->tp_as_number->nb_negative)(v);
1077 err_setstr(TypeError, "bad operand type(s) for unary -");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001078 return NULL;
1079}
1080
1081static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001082pos(v)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001083 object *v;
1084{
1085 if (v->ob_type->tp_as_number != NULL)
Guido van Rossum3f5da241990-12-20 15:06:42 +00001086 return (*v->ob_type->tp_as_number->nb_positive)(v);
1087 err_setstr(TypeError, "bad operand type(s) for unary +");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001088 return NULL;
1089}
1090
1091static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001092not(v)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001093 object *v;
1094{
Guido van Rossum3f5da241990-12-20 15:06:42 +00001095 int outcome = testbool(v);
1096 object *w = outcome == 0 ? True : False;
1097 INCREF(w);
1098 return w;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001099}
1100
1101static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001102call_builtin(func, arg)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001103 object *func;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001104 object *arg;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001105{
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001106 if (is_methodobject(func)) {
1107 method meth = getmethod(func);
1108 object *self = getself(func);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001109 return (*meth)(self, arg);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001110 }
1111 if (is_classobject(func)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001112 if (arg != NULL) {
1113 err_setstr(TypeError,
1114 "classobject() allows no arguments");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001115 return NULL;
1116 }
Guido van Rossum374a9221991-04-04 10:40:29 +00001117 return newinstanceobject(func);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001118 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001119 err_setstr(TypeError, "call of non-function");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001120 return NULL;
1121}
1122
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001123static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001124call_function(func, arg)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001125 object *func;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001126 object *arg;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001127{
Guido van Rossum3f5da241990-12-20 15:06:42 +00001128 object *newarg = NULL;
1129 object *newlocals, *newglobals;
1130 object *co, *v;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001131
Guido van Rossume8122f11991-05-05 20:03:07 +00001132 if (is_instancemethodobject(func)) {
1133 object *self = instancemethodgetself(func);
1134 func = instancemethodgetfunc(func);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001135 if (arg == NULL) {
1136 arg = self;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001137 }
1138 else {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001139 newarg = newtupleobject(2);
1140 if (newarg == NULL)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001141 return NULL;
1142 INCREF(self);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001143 INCREF(arg);
1144 settupleitem(newarg, 0, self);
1145 settupleitem(newarg, 1, arg);
1146 arg = newarg;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001147 }
1148 }
1149 else {
1150 if (!is_funcobject(func)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001151 err_setstr(TypeError, "call of non-function");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001152 return NULL;
1153 }
1154 }
1155
Guido van Rossum3f5da241990-12-20 15:06:42 +00001156 co = getfunccode(func);
1157 if (co == NULL) {
1158 XDECREF(newarg);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001159 return NULL;
1160 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001161 if (!is_codeobject(co)) {
1162 fprintf(stderr, "XXX Bad code\n");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001163 abort();
1164 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001165 newlocals = newdictobject();
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001166 if (newlocals == NULL) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001167 XDECREF(newarg);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001168 return NULL;
1169 }
1170
Guido van Rossum3f5da241990-12-20 15:06:42 +00001171 newglobals = getfuncglobals(func);
1172 INCREF(newglobals);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001173
Guido van Rossum3f5da241990-12-20 15:06:42 +00001174 v = eval_code((codeobject *)co, newglobals, newlocals, arg);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001175
Guido van Rossum3f5da241990-12-20 15:06:42 +00001176 DECREF(newlocals);
1177 DECREF(newglobals);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001178
Guido van Rossum3f5da241990-12-20 15:06:42 +00001179 XDECREF(newarg);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001180
1181 return v;
1182}
1183
1184static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001185apply_subscript(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001186 object *v, *w;
1187{
1188 typeobject *tp = v->ob_type;
1189 if (tp->tp_as_sequence == NULL && tp->tp_as_mapping == NULL) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001190 err_setstr(TypeError, "unsubscriptable object");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001191 return NULL;
1192 }
1193 if (tp->tp_as_sequence != NULL) {
1194 int i;
1195 if (!is_intobject(w)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001196 err_setstr(TypeError, "sequence subscript not int");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001197 return NULL;
1198 }
1199 i = getintvalue(w);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001200 return (*tp->tp_as_sequence->sq_item)(v, i);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001201 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001202 return (*tp->tp_as_mapping->mp_subscript)(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001203}
1204
1205static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001206loop_subscript(v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001207 object *v, *w;
1208{
1209 sequence_methods *sq = v->ob_type->tp_as_sequence;
1210 int i, n;
1211 if (sq == NULL) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001212 err_setstr(TypeError, "loop over non-sequence");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001213 return NULL;
1214 }
1215 i = getintvalue(w);
1216 n = (*sq->sq_length)(v);
1217 if (i >= n)
1218 return NULL; /* End of loop */
Guido van Rossum3f5da241990-12-20 15:06:42 +00001219 return (*sq->sq_item)(v, i);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001220}
1221
1222static int
Guido van Rossum3f5da241990-12-20 15:06:42 +00001223slice_index(v, isize, pi)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001224 object *v;
1225 int isize;
1226 int *pi;
1227{
1228 if (v != NULL) {
1229 if (!is_intobject(v)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001230 err_setstr(TypeError, "slice index must be int");
1231 return -1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001232 }
1233 *pi = getintvalue(v);
1234 if (*pi < 0)
1235 *pi += isize;
1236 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001237 return 0;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001238}
1239
1240static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001241apply_slice(u, v, w) /* return u[v:w] */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001242 object *u, *v, *w;
1243{
1244 typeobject *tp = u->ob_type;
1245 int ilow, ihigh, isize;
1246 if (tp->tp_as_sequence == NULL) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001247 err_setstr(TypeError, "only sequences can be sliced");
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001248 return NULL;
1249 }
1250 ilow = 0;
1251 isize = ihigh = (*tp->tp_as_sequence->sq_length)(u);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001252 if (slice_index(v, isize, &ilow) != 0)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001253 return NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001254 if (slice_index(w, isize, &ihigh) != 0)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001255 return NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001256 return (*tp->tp_as_sequence->sq_slice)(u, ilow, ihigh);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001257}
Guido van Rossum3f5da241990-12-20 15:06:42 +00001258
1259static int
1260assign_subscript(w, key, v) /* w[key] = v */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001261 object *w;
1262 object *key;
1263 object *v;
1264{
1265 typeobject *tp = w->ob_type;
1266 sequence_methods *sq;
1267 mapping_methods *mp;
1268 int (*func)();
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001269 if ((sq = tp->tp_as_sequence) != NULL &&
1270 (func = sq->sq_ass_item) != NULL) {
1271 if (!is_intobject(key)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001272 err_setstr(TypeError,
1273 "sequence subscript must be integer");
1274 return -1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001275 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001276 else
1277 return (*func)(w, (int)getintvalue(key), v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001278 }
1279 else if ((mp = tp->tp_as_mapping) != NULL &&
1280 (func = mp->mp_ass_subscript) != NULL) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001281 return (*func)(w, key, v);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001282 }
1283 else {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001284 err_setstr(TypeError,
1285 "can't assign to this subscripted object");
1286 return -1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001287 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001288}
1289
Guido van Rossum3f5da241990-12-20 15:06:42 +00001290static int
1291assign_slice(u, v, w, x) /* u[v:w] = x */
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001292 object *u, *v, *w, *x;
1293{
Guido van Rossum3f5da241990-12-20 15:06:42 +00001294 sequence_methods *sq = u->ob_type->tp_as_sequence;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001295 int ilow, ihigh, isize;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001296 if (sq == NULL) {
1297 err_setstr(TypeError, "assign to slice of non-sequence");
1298 return -1;
1299 }
1300 if (sq == NULL || sq->sq_ass_slice == NULL) {
1301 err_setstr(TypeError, "unassignable slice");
1302 return -1;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001303 }
1304 ilow = 0;
Guido van Rossum3f5da241990-12-20 15:06:42 +00001305 isize = ihigh = (*sq->sq_length)(u);
1306 if (slice_index(v, isize, &ilow) != 0)
1307 return -1;
1308 if (slice_index(w, isize, &ihigh) != 0)
1309 return -1;
1310 return (*sq->sq_ass_slice)(u, ilow, ihigh, x);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001311}
1312
1313static int
1314cmp_exception(err, v)
1315 object *err, *v;
1316{
1317 if (is_tupleobject(v)) {
1318 int i, n;
1319 n = gettuplesize(v);
1320 for (i = 0; i < n; i++) {
1321 if (err == gettupleitem(v, i))
1322 return 1;
1323 }
1324 return 0;
1325 }
1326 return err == v;
1327}
1328
Guido van Rossum3f5da241990-12-20 15:06:42 +00001329static int
1330cmp_member(v, w)
1331 object *v, *w;
1332{
1333 int i, n, cmp;
1334 object *x;
1335 sequence_methods *sq;
1336 /* Special case for char in string */
1337 if (is_stringobject(w)) {
1338 register char *s, *end;
1339 register char c;
1340 if (!is_stringobject(v) || getstringsize(v) != 1) {
1341 err_setstr(TypeError,
1342 "string member test needs char left operand");
1343 return -1;
1344 }
1345 c = getstringvalue(v)[0];
1346 s = getstringvalue(w);
1347 end = s + getstringsize(w);
1348 while (s < end) {
1349 if (c == *s++)
1350 return 1;
1351 }
1352 return 0;
1353 }
1354 sq = w->ob_type->tp_as_sequence;
1355 if (sq == NULL) {
1356 err_setstr(TypeError,
1357 "'in' or 'not in' needs sequence right argument");
1358 return -1;
1359 }
1360 n = (*sq->sq_length)(w);
1361 for (i = 0; i < n; i++) {
1362 x = (*sq->sq_item)(w, i);
1363 cmp = cmpobject(v, x);
1364 XDECREF(x);
1365 if (cmp == 0)
1366 return 1;
1367 }
1368 return 0;
1369}
1370
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001371static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001372cmp_outcome(op, v, w)
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001373 enum cmp_op op;
1374 register object *v;
1375 register object *w;
1376{
1377 register int cmp;
1378 register int res = 0;
1379 switch (op) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001380 case IS:
1381 case IS_NOT:
1382 res = (v == w);
1383 if (op == IS_NOT)
1384 res = !res;
1385 break;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001386 case IN:
1387 case NOT_IN:
Guido van Rossum3f5da241990-12-20 15:06:42 +00001388 res = cmp_member(v, w);
1389 if (res < 0)
1390 return NULL;
1391 if (op == NOT_IN)
1392 res = !res;
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001393 break;
1394 case EXC_MATCH:
Guido van Rossum3f5da241990-12-20 15:06:42 +00001395 res = cmp_exception(v, w);
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001396 break;
1397 default:
Guido van Rossum3f5da241990-12-20 15:06:42 +00001398 cmp = cmpobject(v, w);
1399 switch (op) {
1400 case LT: res = cmp < 0; break;
1401 case LE: res = cmp <= 0; break;
1402 case EQ: res = cmp == 0; break;
1403 case NE: res = cmp != 0; break;
1404 case GT: res = cmp > 0; break;
1405 case GE: res = cmp >= 0; break;
1406 /* XXX no default? (res is initialized to 0 though) */
1407 }
Guido van Rossum10dc2e81990-11-18 17:27:39 +00001408 }
1409 v = res ? True : False;
1410 INCREF(v);
1411 return v;
1412}
1413
Guido van Rossum374a9221991-04-04 10:40:29 +00001414/* XXX This function should use dict2 variants (change interface!) */
1415
Guido van Rossum3f5da241990-12-20 15:06:42 +00001416static int
1417import_from(locals, v, name)
1418 object *locals;
Guido van Rossume9736fc1990-11-18 17:33:06 +00001419 object *v;
1420 char *name;
1421{
1422 object *w, *x;
1423 w = getmoduledict(v);
1424 if (name[0] == '*') {
1425 int i;
1426 int n = getdictsize(w);
1427 for (i = 0; i < n; i++) {
1428 name = getdictkey(w, i);
1429 if (name == NULL || name[0] == '_')
1430 continue;
1431 x = dictlookup(w, name);
1432 if (x == NULL) {
1433 /* XXX can't happen? */
Guido van Rossum3f5da241990-12-20 15:06:42 +00001434 err_setstr(NameError, name);
1435 return -1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00001436 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001437 if (dictinsert(locals, name, x) != 0)
1438 return -1;
Guido van Rossume9736fc1990-11-18 17:33:06 +00001439 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001440 return 0;
Guido van Rossume9736fc1990-11-18 17:33:06 +00001441 }
1442 else {
1443 x = dictlookup(w, name);
Guido van Rossum3f5da241990-12-20 15:06:42 +00001444 if (x == NULL) {
1445 err_setstr(NameError, name);
1446 return -1;
1447 }
1448 else
1449 return dictinsert(locals, name, x);
Guido van Rossume9736fc1990-11-18 17:33:06 +00001450 }
1451}
1452
1453static object *
Guido van Rossum3f5da241990-12-20 15:06:42 +00001454build_class(v, w)
Guido van Rossume9736fc1990-11-18 17:33:06 +00001455 object *v; /* None or tuple containing base classes */
1456 object *w; /* dictionary */
1457{
1458 if (is_tupleobject(v)) {
1459 int i;
1460 for (i = gettuplesize(v); --i >= 0; ) {
1461 object *x = gettupleitem(v, i);
1462 if (!is_classobject(x)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001463 err_setstr(TypeError,
1464 "base is not a class object");
Guido van Rossume9736fc1990-11-18 17:33:06 +00001465 return NULL;
1466 }
1467 }
1468 }
1469 else {
1470 v = NULL;
1471 }
1472 if (!is_dictobject(w)) {
Guido van Rossum3f5da241990-12-20 15:06:42 +00001473 err_setstr(SystemError, "build_class with non-dictionary");
Guido van Rossume9736fc1990-11-18 17:33:06 +00001474 return NULL;
1475 }
Guido van Rossum3f5da241990-12-20 15:06:42 +00001476 return newclassobject(v, w);
Guido van Rossume9736fc1990-11-18 17:33:06 +00001477}