vm: Establish macros PUSH_EXC_BLOCK & POP_EXC_BLOCK to deal with exc stack.

E.g. to handle currently_in_except_block restoring properly.
This commit is contained in:
Paul Sokolovsky 2014-03-29 23:16:27 +02:00
parent d109676ec0
commit a0ad77ba08

19
py/vm.c
View File

@ -45,7 +45,7 @@ typedef enum {
#define TOP() (*sp)
#define SET_TOP(val) *sp = (val)
#define SETUP_BLOCK() \
#define PUSH_EXC_BLOCK() \
DECODE_ULABEL; /* except labels are always forward */ \
++exc_sp; \
exc_sp->opcode = op; \
@ -53,6 +53,10 @@ typedef enum {
exc_sp->val_sp = MP_TAGPTR_MAKE(sp, currently_in_except_block); \
currently_in_except_block = 0; /* in a try block now */
#define POP_EXC_BLOCK() \
currently_in_except_block = MP_TAGPTR_TAG(exc_sp->val_sp); /* restore previous state */ \
exc_sp--; /* pop back to previous exception handler */
mp_vm_return_kind_t mp_execute_byte_code(const byte *code, const mp_obj_t *args, uint n_args, const mp_obj_t *args2, uint n_args2, mp_obj_t *ret) {
const byte *ip = code;
@ -386,7 +390,7 @@ dispatch_loop:
SET_TOP(rt_load_attr(obj1, MP_QSTR___exit__));
rt_load_method(obj1, MP_QSTR___enter__, sp + 1);
obj2 = rt_call_method_n_kw(0, 0, sp + 1);
SETUP_BLOCK();
PUSH_EXC_BLOCK();
PUSH(obj2);
break;
@ -478,7 +482,7 @@ unwind_jump:
// matched against: POP_BLOCK or POP_EXCEPT (anything else?)
case MP_BC_SETUP_EXCEPT:
case MP_BC_SETUP_FINALLY:
SETUP_BLOCK();
PUSH_EXC_BLOCK();
break;
case MP_BC_END_FINALLY:
@ -527,8 +531,7 @@ unwind_jump:
case MP_BC_POP_BLOCK:
// we are exiting an exception handler, so pop the last one of the exception-stack
assert(exc_sp >= exc_stack);
currently_in_except_block = MP_TAGPTR_TAG(exc_sp->val_sp); // restore previous state
exc_sp--; // pop back to previous exception handler
POP_EXC_BLOCK();
break;
// matched against: SETUP_EXCEPT
@ -539,8 +542,7 @@ unwind_jump:
assert(currently_in_except_block);
//sp = (mp_obj_t*)(*exc_sp--);
//exc_sp--; // discard ip
currently_in_except_block = MP_TAGPTR_TAG(exc_sp->val_sp); // restore previous state
exc_sp--; // pop back to previous exception handler
POP_EXC_BLOCK();
//sp -= 3; // pop 3 exception values
break;
@ -830,8 +832,7 @@ yield:
// at the moment we are just raising the very last exception (the one that caused the nested exception)
// move up to previous exception handler
currently_in_except_block = MP_TAGPTR_TAG(exc_sp->val_sp); // restore previous state
exc_sp--; // pop back to previous exception handler
POP_EXC_BLOCK();
}
if (exc_sp >= exc_stack) {