2013-10-04 14:53:11 -04:00
# include <stdint.h>
# include <stdlib.h>
# include <stdio.h>
# include <string.h>
# include <assert.h>
2013-10-15 17:25:17 -04:00
# include "nlr.h"
2013-10-04 14:53:11 -04:00
# include "misc.h"
2013-12-21 13:17:45 -05:00
# include "mpconfig.h"
2014-01-21 16:40:13 -05:00
# include "qstr.h"
2013-12-21 13:17:45 -05:00
# include "obj.h"
2013-10-04 14:53:11 -04:00
# include "runtime.h"
2013-12-21 13:17:45 -05:00
# include "bc0.h"
2013-10-10 17:06:54 -04:00
# include "bc.h"
2013-10-04 14:53:11 -04:00
2014-01-31 12:45:15 -05:00
// Value stack grows up (this makes it incompatible with native C stack, but
// makes sure that arguments to functions are in natural order arg1..argN
// (Python semantics mandates left-to-right evaluation order, including for
// function arguments). Stack pointer is pre-incremented and points at the
// top element.
// Exception stack also grows up, top element is also pointed at.
2014-01-31 12:40:09 -05:00
// Exception stack entry
typedef struct _mp_exc_stack {
const byte * handler ;
// bit 0 is saved currently_in_except_block value
machine_uint_t val_sp ;
// We might only have 2 interesting cases here: SETUP_EXCEPT & SETUP_FINALLY,
// consider storing it in bit 1 of val_sp. TODO: SETUP_WITH?
byte opcode ;
} mp_exc_stack ;
2014-01-31 17:55:05 -05:00
// Exception stack unwind reasons (WHY_* in CPython-speak)
typedef enum {
UNWIND_RETURN = 1 ,
UNWIND_BREAK ,
UNWIND_CONTINUE ,
} mp_unwind_reason_t ;
2013-10-04 14:53:11 -04:00
# define DECODE_UINT do { unum = *ip++; if (unum > 127) { unum = ((unum & 0x3f) << 8) | (*ip++); } } while (0)
2013-11-05 17:06:08 -05:00
# define DECODE_ULABEL do { unum = (ip[0] | (ip[1] << 8)); ip += 2; } while (0)
# define DECODE_SLABEL do { unum = (ip[0] | (ip[1] << 8)) - 0x8000; ip += 2; } while (0)
2014-01-19 06:48:48 -05:00
# define DECODE_QSTR do { qst = *ip++; if (qst > 127) { qst = ((qst & 0x3f) << 8) | (*ip++); } } while (0)
2014-01-18 09:10:48 -05:00
# define PUSH(val) *++sp = (val)
# define POP() (*sp--)
2013-12-10 12:27:24 -05:00
# define TOP() (*sp)
# define SET_TOP(val) *sp = (val)
2013-10-04 14:53:11 -04:00
2014-02-01 13:29:40 -05:00
mp_obj_t mp_execute_byte_code ( const byte * code , const mp_obj_t * args , uint n_args , const mp_obj_t * args2 , uint n_args2 , uint n_state ) {
2014-01-18 09:10:48 -05:00
// allocate state for locals and stack
mp_obj_t temp_state [ 10 ] ;
2013-12-21 13:17:45 -05:00
mp_obj_t * state = & temp_state [ 0 ] ;
2013-11-05 17:16:22 -05:00
if ( n_state > 10 ) {
2013-12-21 13:17:45 -05:00
state = m_new ( mp_obj_t , n_state ) ;
2013-11-05 17:16:22 -05:00
}
2014-01-18 09:10:48 -05:00
mp_obj_t * sp = & state [ 0 ] - 1 ;
2013-10-16 15:39:12 -04:00
// init args
2014-02-01 13:29:40 -05:00
for ( uint i = 0 ; i < n_args ; i + + ) {
2014-01-18 09:10:48 -05:00
state [ n_state - 1 - i ] = args [ i ] ;
2013-10-16 15:39:12 -04:00
}
2014-02-01 13:29:40 -05:00
for ( uint i = 0 ; i < n_args2 ; i + + ) {
state [ n_state - 1 - n_args - i ] = args2 [ i ] ;
}
2014-01-18 18:24:36 -05:00
2013-10-16 15:39:12 -04:00
const byte * ip = code ;
2013-12-30 17:32:17 -05:00
2014-01-18 18:24:36 -05:00
// get code info size
machine_uint_t code_info_size = ip [ 0 ] | ( ip [ 1 ] < < 8 ) | ( ip [ 2 ] < < 16 ) | ( ip [ 3 ] < < 24 ) ;
ip + = code_info_size ;
2013-12-30 17:32:17 -05:00
// execute prelude to make any cells (closed over variables)
{
for ( uint n_local = * ip + + ; n_local > 0 ; n_local - - ) {
uint local_num = * ip + + ;
2014-02-01 13:29:40 -05:00
if ( local_num < n_args + n_args2 ) {
2014-01-18 09:10:48 -05:00
state [ n_state - 1 - local_num ] = mp_obj_new_cell ( state [ n_state - 1 - local_num ] ) ;
2013-12-30 17:32:17 -05:00
} else {
2014-01-18 09:10:48 -05:00
state [ n_state - 1 - local_num ] = mp_obj_new_cell ( MP_OBJ_NULL ) ;
2013-12-30 17:32:17 -05:00
}
}
}
// execute the byte code
2014-01-18 18:24:36 -05:00
if ( mp_execute_byte_code_2 ( code , & ip , & state [ n_state - 1 ] , & sp ) ) {
2013-10-16 15:39:12 -04:00
// it shouldn't yield
assert ( 0 ) ;
}
2013-12-30 17:32:17 -05:00
2013-11-02 10:33:10 -04:00
// TODO check fails if, eg, return from within for loop
//assert(sp == &state[17]);
2013-10-16 15:39:12 -04:00
return * sp ;
}
2014-01-18 09:10:48 -05:00
// fastn has items in reverse order (fastn[0] is local[0], fastn[-1] is local[1], etc)
// sp points to bottom of stack which grows up
2014-01-26 13:50:11 -05:00
// returns true if bytecode yielded
2014-01-18 18:24:36 -05:00
bool mp_execute_byte_code_2 ( const byte * code_info , const byte * * ip_in_out , mp_obj_t * fastn , mp_obj_t * * sp_in_out ) {
2013-10-15 18:46:01 -04:00
// careful: be sure to declare volatile any variables read in the exception handler (written is ok, I think)
2013-10-16 15:39:12 -04:00
const byte * ip = * ip_in_out ;
2013-12-21 13:17:45 -05:00
mp_obj_t * sp = * sp_in_out ;
2013-10-04 14:53:11 -04:00
machine_uint_t unum ;
2014-01-19 06:48:48 -05:00
qstr qst ;
2013-12-21 13:17:45 -05:00
mp_obj_t obj1 , obj2 ;
2013-10-15 17:25:17 -04:00
nlr_buf_t nlr ;
2013-10-04 14:53:11 -04:00
2013-12-29 11:54:59 -05:00
volatile machine_uint_t currently_in_except_block = 0 ; // 0 or 1, to detect nested exceptions
2014-01-31 12:40:09 -05:00
mp_exc_stack exc_stack [ 4 ] ;
mp_exc_stack * volatile exc_sp = & exc_stack [ 0 ] - 1 ; // stack grows up, exc_sp points to top of stack
2014-01-18 18:24:36 -05:00
const byte * volatile save_ip = ip ; // this is so we can access ip in the exception handler without making ip volatile (which means the compiler can't keep it in a register in the main loop)
2013-10-15 18:46:01 -04:00
2013-10-15 17:25:17 -04:00
// outer exception handling loop
2013-10-04 14:53:11 -04:00
for ( ; ; ) {
2013-10-15 17:25:17 -04:00
if ( nlr_push ( & nlr ) = = 0 ) {
// loop to execute byte code
for ( ; ; ) {
2014-01-31 17:55:05 -05:00
dispatch_loop :
2014-01-18 18:24:36 -05:00
save_ip = ip ;
2013-10-15 17:25:17 -04:00
int op = * ip + + ;
switch ( op ) {
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_FALSE :
PUSH ( mp_const_false ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_NONE :
PUSH ( mp_const_none ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_TRUE :
PUSH ( mp_const_true ) ;
2013-10-15 17:25:17 -04:00
break ;
2014-01-04 13:44:46 -05:00
case MP_BC_LOAD_CONST_ELLIPSIS :
PUSH ( mp_const_ellipsis ) ;
break ;
2013-10-15 17:25:17 -04:00
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_SMALL_INT :
2013-11-05 17:06:08 -05:00
unum = ( ip [ 0 ] | ( ip [ 1 ] < < 8 ) | ( ip [ 2 ] < < 16 ) ) - 0x800000 ;
2013-11-04 18:04:50 -05:00
ip + = 3 ;
2014-01-06 10:49:46 -05:00
PUSH ( MP_OBJ_NEW_SMALL_INT ( unum ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2014-01-17 12:51:46 -05:00
case MP_BC_LOAD_CONST_INT :
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
PUSH ( mp_obj_new_int_from_long_str ( qstr_str ( qst ) ) ) ;
2014-01-17 12:51:46 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_DEC :
2013-11-02 15:47:57 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
PUSH ( rt_load_const_dec ( qst ) ) ;
2013-11-02 15:47:57 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_ID :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
PUSH ( rt_load_const_str ( qst ) ) ; // TODO
2013-10-15 17:25:17 -04:00
break ;
2014-01-02 11:46:27 -05:00
case MP_BC_LOAD_CONST_BYTES :
DECODE_QSTR ;
2014-01-24 15:50:40 -05:00
PUSH ( rt_load_const_bytes ( qst ) ) ;
2014-01-02 11:46:27 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_CONST_STRING :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
PUSH ( rt_load_const_str ( qst ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_FAST_0 :
2014-01-29 15:30:52 -05:00
PUSH ( fastn [ 0 ] ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_FAST_1 :
2014-01-29 15:30:52 -05:00
PUSH ( fastn [ - 1 ] ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_FAST_2 :
2014-01-29 15:30:52 -05:00
PUSH ( fastn [ - 2 ] ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_FAST_N :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
PUSH ( fastn [ - unum ] ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_DEREF :
2013-12-10 19:41:43 -05:00
DECODE_UINT ;
2014-01-29 15:30:52 -05:00
PUSH ( rt_get_cell ( fastn [ - unum ] ) ) ;
2013-12-10 19:41:43 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_NAME :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
PUSH ( rt_load_name ( qst ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_GLOBAL :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
PUSH ( rt_load_global ( qst ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_ATTR :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
SET_TOP ( rt_load_attr ( TOP ( ) , qst ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_METHOD :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
rt_load_method ( * sp , qst , sp ) ;
2014-01-18 09:10:48 -05:00
sp + = 1 ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LOAD_BUILD_CLASS :
2013-10-15 17:25:17 -04:00
PUSH ( rt_load_build_class ( ) ) ;
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_FAST_0 :
2014-01-29 15:30:52 -05:00
fastn [ 0 ] = POP ( ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_FAST_1 :
2014-01-29 15:30:52 -05:00
fastn [ - 1 ] = POP ( ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_FAST_2 :
2014-01-29 15:30:52 -05:00
fastn [ - 2 ] = POP ( ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_FAST_N :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
fastn [ - unum ] = POP ( ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_DEREF :
2013-12-10 19:41:43 -05:00
DECODE_UINT ;
2014-01-29 15:30:52 -05:00
rt_set_cell ( fastn [ - unum ] , POP ( ) ) ;
2013-12-10 19:41:43 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_NAME :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
rt_store_name ( qst , POP ( ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_GLOBAL :
2013-11-04 18:04:50 -05:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
rt_store_global ( qst , POP ( ) ) ;
2013-11-04 18:04:50 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_ATTR :
2013-10-15 17:25:17 -04:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
rt_store_attr ( sp [ 0 ] , qst , sp [ - 1 ] ) ;
2014-01-18 09:10:48 -05:00
sp - = 2 ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_SUBSCR :
2014-01-18 09:10:48 -05:00
rt_store_subscr ( sp [ - 1 ] , sp [ 0 ] , sp [ - 2 ] ) ;
sp - = 3 ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_DUP_TOP :
2013-12-10 12:27:24 -05:00
obj1 = TOP ( ) ;
2013-10-15 17:25:17 -04:00
PUSH ( obj1 ) ;
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_DUP_TOP_TWO :
2014-01-18 09:10:48 -05:00
sp + = 2 ;
sp [ 0 ] = sp [ - 2 ] ;
sp [ - 1 ] = sp [ - 3 ] ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_POP_TOP :
2014-01-18 09:10:48 -05:00
sp - = 1 ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_ROT_TWO :
2013-11-02 10:33:10 -04:00
obj1 = sp [ 0 ] ;
2014-01-18 09:10:48 -05:00
sp [ 0 ] = sp [ - 1 ] ;
sp [ - 1 ] = obj1 ;
2013-11-02 10:33:10 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_ROT_THREE :
2013-10-15 17:25:17 -04:00
obj1 = sp [ 0 ] ;
2014-01-18 09:10:48 -05:00
sp [ 0 ] = sp [ - 1 ] ;
sp [ - 1 ] = sp [ - 2 ] ;
sp [ - 2 ] = obj1 ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_JUMP :
2013-11-05 17:06:08 -05:00
DECODE_SLABEL ;
ip + = unum ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_POP_JUMP_IF_TRUE :
2013-11-05 17:06:08 -05:00
DECODE_SLABEL ;
2013-10-15 17:25:17 -04:00
if ( rt_is_true ( POP ( ) ) ) {
2013-11-05 17:06:08 -05:00
ip + = unum ;
2013-10-15 17:25:17 -04:00
}
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_POP_JUMP_IF_FALSE :
2013-11-05 17:06:08 -05:00
DECODE_SLABEL ;
2013-10-15 17:25:17 -04:00
if ( ! rt_is_true ( POP ( ) ) ) {
2013-11-05 17:06:08 -05:00
ip + = unum ;
2013-10-15 17:25:17 -04:00
}
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_JUMP_IF_TRUE_OR_POP :
2013-11-09 15:12:32 -05:00
DECODE_SLABEL ;
2013-12-10 12:27:24 -05:00
if ( rt_is_true ( TOP ( ) ) ) {
2013-11-09 15:12:32 -05:00
ip + = unum ;
} else {
2014-01-18 09:10:48 -05:00
sp - - ;
2013-11-09 15:12:32 -05:00
}
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_JUMP_IF_FALSE_OR_POP :
2013-11-09 15:12:32 -05:00
DECODE_SLABEL ;
2013-12-10 12:27:24 -05:00
if ( rt_is_true ( TOP ( ) ) ) {
2014-01-18 09:10:48 -05:00
sp - - ;
2013-11-09 15:12:32 -05:00
} else {
ip + = unum ;
}
break ;
2013-10-15 17:25:17 -04:00
/* we are trying to get away without using this opcode
2013-12-21 13:17:45 -05:00
case MP_BC_SETUP_LOOP :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2013-12-21 13:17:45 -05:00
// push_block(MP_BC_SETUP_LOOP, ip + unum, sp)
2013-10-15 17:25:17 -04:00
break ;
*/
2014-01-21 18:48:04 -05:00
// TODO this might need more sophisticated handling when breaking from within an except
case MP_BC_BREAK_LOOP :
DECODE_ULABEL ;
ip + = unum ;
break ;
// TODO this might need more sophisticated handling when breaking from within an except
case MP_BC_CONTINUE_LOOP :
DECODE_ULABEL ;
ip + = unum ;
break ;
2013-12-29 13:48:37 -05:00
// matched against: POP_BLOCK or POP_EXCEPT (anything else?)
2013-12-21 13:17:45 -05:00
case MP_BC_SETUP_EXCEPT :
2014-01-31 12:40:09 -05:00
case MP_BC_SETUP_FINALLY :
2013-11-05 17:06:08 -05:00
DECODE_ULABEL ; // except labels are always forward
2014-01-31 12:40:09 -05:00
+ + exc_sp ;
exc_sp - > opcode = op ;
exc_sp - > handler = ip + unum ;
exc_sp - > val_sp = ( ( ( machine_uint_t ) sp ) | currently_in_except_block ) ;
2013-12-29 11:54:59 -05:00
currently_in_except_block = 0 ; // in a try block now
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_END_FINALLY :
2014-01-30 06:49:18 -05:00
// not fully implemented
2013-10-15 17:25:17 -04:00
// if TOS is an exception, reraises the exception (3 values on TOS)
// if TOS is None, just pops it and continues
2014-01-30 06:49:18 -05:00
// if TOS is an integer, does something else
2013-10-15 17:25:17 -04:00
// else error
2014-01-30 06:49:18 -05:00
if ( MP_OBJ_IS_TYPE ( TOP ( ) , & exception_type ) ) {
nlr_jump ( TOP ( ) ) ;
}
if ( TOP ( ) = = mp_const_none ) {
sp - - ;
2014-01-31 17:55:05 -05:00
} else if ( MP_OBJ_IS_SMALL_INT ( TOP ( ) ) ) {
// We finished "finally" coroutine and now dispatch back
// to our caller, based on TOS value
mp_unwind_reason_t reason = MP_OBJ_SMALL_INT_VALUE ( POP ( ) ) ;
switch ( reason ) {
case UNWIND_RETURN :
goto unwind_return ;
// TODO
case UNWIND_BREAK :
case UNWIND_CONTINUE :
;
}
assert ( 0 ) ;
2014-01-30 06:49:18 -05:00
} else {
assert ( 0 ) ;
}
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_GET_ITER :
2013-12-10 12:27:24 -05:00
SET_TOP ( rt_getiter ( TOP ( ) ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_FOR_ITER :
2013-11-05 17:06:08 -05:00
DECODE_ULABEL ; // the jump offset if iteration finishes; for labels are always forward
2013-12-10 12:27:24 -05:00
obj1 = rt_iternext ( TOP ( ) ) ;
2013-12-21 13:17:45 -05:00
if ( obj1 = = mp_const_stop_iteration ) {
2014-01-18 09:10:48 -05:00
- - sp ; // pop the exhausted iterator
2013-11-05 17:06:08 -05:00
ip + = unum ; // jump to after for-block
2013-10-15 17:25:17 -04:00
} else {
PUSH ( obj1 ) ; // push the next iteration value
}
break ;
2013-12-29 13:48:37 -05:00
// matched against: SETUP_EXCEPT, SETUP_FINALLY, SETUP_WITH
2013-12-21 13:17:45 -05:00
case MP_BC_POP_BLOCK :
2013-12-29 13:48:37 -05:00
// we are exiting an exception handler, so pop the last one of the exception-stack
assert ( exc_sp > = & exc_stack [ 0 ] ) ;
2014-01-31 12:40:09 -05:00
currently_in_except_block = ( exc_sp - > val_sp & 1 ) ; // restore previous state
exc_sp - - ; // pop back to previous exception handler
2013-10-15 17:25:17 -04:00
break ;
2014-01-21 18:48:04 -05:00
// matched against: SETUP_EXCEPT
2013-12-21 13:17:45 -05:00
case MP_BC_POP_EXCEPT :
2013-10-15 18:46:01 -04:00
// TODO need to work out how blocks work etc
2013-10-15 17:25:17 -04:00
// pops block, checks it's an exception block, and restores the stack, saving the 3 exception values to local threadstate
2013-10-15 18:46:01 -04:00
assert ( exc_sp > = & exc_stack [ 0 ] ) ;
2014-01-30 06:49:18 -05:00
assert ( currently_in_except_block ) ;
2013-12-21 13:17:45 -05:00
//sp = (mp_obj_t*)(*exc_sp--);
2013-10-15 18:46:01 -04:00
//exc_sp--; // discard ip
2014-01-31 12:40:09 -05:00
currently_in_except_block = ( exc_sp - > val_sp & 1 ) ; // restore previous state
exc_sp - - ; // pop back to previous exception handler
2014-01-18 09:10:48 -05:00
//sp -= 3; // pop 3 exception values
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_UNARY_OP :
2013-11-02 15:47:57 -04:00
unum = * ip + + ;
2013-12-10 12:27:24 -05:00
SET_TOP ( rt_unary_op ( unum , TOP ( ) ) ) ;
2013-11-02 15:47:57 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_BINARY_OP :
2013-10-15 17:25:17 -04:00
unum = * ip + + ;
obj2 = POP ( ) ;
2013-12-10 12:27:24 -05:00
obj1 = TOP ( ) ;
SET_TOP ( rt_binary_op ( unum , obj1 , obj2 ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_BUILD_TUPLE :
2013-10-16 11:12:52 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
sp - = unum - 1 ;
SET_TOP ( rt_build_tuple ( unum , sp ) ) ;
2013-10-16 11:12:52 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_BUILD_LIST :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
sp - = unum - 1 ;
SET_TOP ( rt_build_list ( unum , sp ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_LIST_APPEND :
2013-10-16 11:12:52 -04:00
DECODE_UINT ;
// I think it's guaranteed by the compiler that sp[unum] is a list
2014-01-18 09:10:48 -05:00
rt_list_append ( sp [ - unum ] , sp [ 0 ] ) ;
sp - - ;
2013-10-16 11:12:52 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_BUILD_MAP :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
PUSH ( rt_build_map ( unum ) ) ;
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_STORE_MAP :
2014-01-18 09:10:48 -05:00
sp - = 2 ;
rt_store_map ( sp [ 0 ] , sp [ 2 ] , sp [ 1 ] ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_MAP_ADD :
2013-10-16 15:54:01 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
// I think it's guaranteed by the compiler that sp[-unum - 1] is a map
rt_store_map ( sp [ - unum - 1 ] , sp [ 0 ] , sp [ - 1 ] ) ;
sp - = 2 ;
2013-10-16 15:54:01 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_BUILD_SET :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
sp - = unum - 1 ;
SET_TOP ( rt_build_set ( unum , sp ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_SET_ADD :
2013-10-16 15:57:49 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
// I think it's guaranteed by the compiler that sp[-unum] is a set
rt_store_set ( sp [ - unum ] , sp [ 0 ] ) ;
sp - - ;
2013-10-16 15:57:49 -04:00
break ;
2014-01-03 18:34:23 -05:00
# if MICROPY_ENABLE_SLICE
2014-01-02 19:48:56 -05:00
case MP_BC_BUILD_SLICE :
DECODE_UINT ;
if ( unum = = 2 ) {
obj2 = POP ( ) ;
obj1 = TOP ( ) ;
SET_TOP ( mp_obj_new_slice ( obj1 , obj2 , NULL ) ) ;
} else {
printf ( " 3-argument slice is not supported \n " ) ;
assert ( 0 ) ;
}
break ;
2014-01-03 18:34:23 -05:00
# endif
2014-01-02 19:48:56 -05:00
2013-12-21 13:17:45 -05:00
case MP_BC_UNPACK_SEQUENCE :
2013-11-26 10:15:50 -05:00
DECODE_UINT ;
2014-01-18 18:42:49 -05:00
rt_unpack_sequence ( sp [ 0 ] , unum , sp ) ;
2014-01-18 09:10:48 -05:00
sp + = unum - 1 ;
2013-11-26 10:15:50 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_MAKE_FUNCTION :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-02-01 08:05:04 -05:00
PUSH ( rt_make_function_from_id ( unum , MP_OBJ_NULL ) ) ;
break ;
case MP_BC_MAKE_FUNCTION_DEFARGS :
DECODE_UINT ;
SET_TOP ( rt_make_function_from_id ( unum , TOP ( ) ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_MAKE_CLOSURE :
2013-12-10 19:41:43 -05:00
DECODE_UINT ;
2014-01-29 13:58:52 -05:00
SET_TOP ( rt_make_closure_from_id ( unum , TOP ( ) ) ) ;
2013-12-10 19:41:43 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_CALL_FUNCTION :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
// unum & 0xff == n_positional
// (unum >> 8) & 0xff == n_keyword
sp - = ( unum & 0xff ) + ( ( unum > > 7 ) & 0x1fe ) ;
SET_TOP ( rt_call_function_n_kw ( * sp , unum & 0xff , ( unum > > 8 ) & 0xff , sp + 1 ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_CALL_METHOD :
2013-10-15 17:25:17 -04:00
DECODE_UINT ;
2014-01-18 09:10:48 -05:00
// unum & 0xff == n_positional
// (unum >> 8) & 0xff == n_keyword
sp - = ( unum & 0xff ) + ( ( unum > > 7 ) & 0x1fe ) + 1 ;
SET_TOP ( rt_call_method_n_kw ( unum & 0xff , ( unum > > 8 ) & 0xff , sp ) ) ;
2013-10-15 17:25:17 -04:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_RETURN_VALUE :
2014-01-31 17:55:05 -05:00
unwind_return :
while ( exc_sp > = exc_stack ) {
if ( exc_sp - > opcode = = MP_BC_SETUP_FINALLY ) {
// We're going to run "finally" code as a coroutine
// (not calling it recursively). Set up a sentinel
// on a stack so it can return back to us when it is
// done (when END_FINALLY reached).
PUSH ( MP_OBJ_NEW_SMALL_INT ( UNWIND_RETURN ) ) ;
ip = exc_sp - > handler ;
// We don't need to do anything with sp, finally is just
// syntactic sugar for sequential execution??
// sp =
exc_sp - - ;
goto dispatch_loop ;
}
exc_sp - - ;
}
2013-10-15 17:25:17 -04:00
nlr_pop ( ) ;
2013-10-16 15:39:12 -04:00
* sp_in_out = sp ;
2014-01-02 13:20:41 -05:00
assert ( exc_sp = = & exc_stack [ 0 ] - 1 ) ;
2013-10-16 15:39:12 -04:00
return false ;
2014-01-10 09:09:55 -05:00
case MP_BC_RAISE_VARARGS :
unum = * ip + + ;
assert ( unum = = 1 ) ;
obj1 = POP ( ) ;
nlr_jump ( obj1 ) ;
2013-12-21 13:17:45 -05:00
case MP_BC_YIELD_VALUE :
2013-10-16 15:39:12 -04:00
nlr_pop ( ) ;
* ip_in_out = ip ;
* sp_in_out = sp ;
return true ;
2013-10-15 17:25:17 -04:00
2013-12-21 13:17:45 -05:00
case MP_BC_IMPORT_NAME :
2013-12-10 12:27:24 -05:00
DECODE_QSTR ;
obj1 = POP ( ) ;
2014-01-19 06:48:48 -05:00
SET_TOP ( rt_import_name ( qst , obj1 , TOP ( ) ) ) ;
2013-12-10 12:27:24 -05:00
break ;
2013-12-21 13:17:45 -05:00
case MP_BC_IMPORT_FROM :
2013-12-10 12:27:24 -05:00
DECODE_QSTR ;
2014-01-19 06:48:48 -05:00
obj1 = rt_import_from ( TOP ( ) , qst ) ;
2013-12-10 12:27:24 -05:00
PUSH ( obj1 ) ;
break ;
2013-10-15 17:25:17 -04:00
default :
2013-11-05 17:06:08 -05:00
printf ( " code %p, byte code 0x%02x not implemented \n " , ip , op ) ;
2013-10-15 17:25:17 -04:00
assert ( 0 ) ;
nlr_pop ( ) ;
2013-10-16 15:39:12 -04:00
return false ;
2013-10-04 14:53:11 -04:00
}
2013-10-15 17:25:17 -04:00
}
} else {
// exception occurred
2014-01-18 18:24:36 -05:00
// set file and line number that the exception occurred at
2014-01-30 06:49:18 -05:00
// TODO: don't set traceback for exceptions re-raised by END_FINALLY.
// But consider how to handle nested exceptions.
2014-01-18 18:24:36 -05:00
if ( MP_OBJ_IS_TYPE ( nlr . ret_val , & exception_type ) ) {
machine_uint_t code_info_size = code_info [ 0 ] | ( code_info [ 1 ] < < 8 ) | ( code_info [ 2 ] < < 16 ) | ( code_info [ 3 ] < < 24 ) ;
2014-01-19 06:48:48 -05:00
qstr source_file = code_info [ 4 ] | ( code_info [ 5 ] < < 8 ) | ( code_info [ 6 ] < < 16 ) | ( code_info [ 7 ] < < 24 ) ;
qstr block_name = code_info [ 8 ] | ( code_info [ 9 ] < < 8 ) | ( code_info [ 10 ] < < 16 ) | ( code_info [ 11 ] < < 24 ) ;
2014-01-18 18:24:36 -05:00
machine_uint_t source_line = 1 ;
machine_uint_t bc = save_ip - code_info - code_info_size ;
2014-01-19 06:48:48 -05:00
//printf("find %lu %d %d\n", bc, code_info[12], code_info[13]);
2014-01-25 06:43:20 -05:00
for ( const byte * ci = code_info + 12 ; * ci & & bc > = ( ( * ci ) & 31 ) ; ci + + ) {
bc - = * ci & 31 ;
source_line + = * ci > > 5 ;
2014-01-18 18:24:36 -05:00
}
2014-01-19 07:38:49 -05:00
mp_obj_exception_add_traceback ( nlr . ret_val , source_file , source_line , block_name ) ;
2014-01-18 18:24:36 -05:00
}
2013-12-29 11:54:59 -05:00
while ( currently_in_except_block ) {
// nested exception
assert ( exc_sp > = & exc_stack [ 0 ] ) ;
// TODO make a proper message for nested exception
// at the moment we are just raising the very last exception (the one that caused the nested exception)
// move up to previous exception handler
2014-01-31 12:40:09 -05:00
currently_in_except_block = ( exc_sp - > val_sp & 1 ) ; // restore previous state
exc_sp - - ; // pop back to previous exception handler
2013-12-29 11:54:59 -05:00
}
2013-10-15 18:46:01 -04:00
if ( exc_sp > = & exc_stack [ 0 ] ) {
2013-12-29 11:54:59 -05:00
// set flag to indicate that we are now handling an exception
currently_in_except_block = 1 ;
2013-10-15 17:25:17 -04:00
// catch exception and pass to byte code
2014-01-31 12:40:09 -05:00
sp = ( mp_obj_t * ) ( exc_sp - > val_sp & ( ~ ( ( machine_uint_t ) 1 ) ) ) ;
ip = exc_sp - > handler ;
2013-10-15 18:46:01 -04:00
// push(traceback, exc-val, exc-type)
2013-12-21 13:17:45 -05:00
PUSH ( mp_const_none ) ;
2013-10-15 18:46:01 -04:00
PUSH ( nlr . ret_val ) ;
2013-12-29 12:17:43 -05:00
PUSH ( nlr . ret_val ) ; // TODO should be type(nlr.ret_val), I think...
2013-12-29 11:54:59 -05:00
2013-10-15 17:25:17 -04:00
} else {
2013-12-29 11:54:59 -05:00
// re-raise exception to higher level
2013-10-16 15:39:12 -04:00
// TODO what to do if this is a generator??
2013-10-15 17:25:17 -04:00
nlr_jump ( nlr . ret_val ) ;
}
2013-10-04 14:53:11 -04:00
}
}
}