#ifndef __NASAL_VM_H__ #define __NASAL_VM_H__ class nasal_vm { protected: /* registers and constants of nasal_vm */ u32 pc; // program counter nas_ref* localr; // local scope register nas_ref* memr; // used for mem_call nas_ref funcr; // function register nas_ref upvalr; // upvalue register nas_ref* canary; // avoid stackoverflow nas_ref* top; // stack top const f64* num_table;// const numbers, ref from nasal_codegen const string* str_table;// const symbols, ref from nasal_codegen std::vector imm; // immediate number /* garbage collector */ nasal_gc gc; /* main stack */ nas_ref stack[STACK_DEPTH]; /* values used for debugger */ const string* files; // ref from nasal_import const opcode* bytecode; // ref from nasal_codegen void init( const std::vector&, const std::vector&, const std::vector&, const std::vector&, const std::vector&); /* debug functions */ bool detail_info; void valinfo(nas_ref&); void bytecodeinfo(const char*,const u32); void traceback(); void stackinfo(const u32); void reginfo(); void gstate(); void lstate(); void ustate(); void detail(); [[noreturn]] void die(const string&); /* vm calculation functions*/ bool condition(nas_ref); /* vm operands */ void o_intg(); void o_intl(); void o_loadg(); void o_loadl(); void o_loadu(); void o_pnum(); void o_pnil(); void o_pstr(); void o_newv(); void o_newh(); void o_newf(); void o_happ(); void o_para(); void o_deft(); void o_dyn(); void o_unot(); void o_usub(); void o_add(); void o_sub(); void o_mul(); void o_div(); void o_lnk(); void o_addc(); void o_subc(); void o_mulc(); void o_divc(); void o_lnkc(); void o_addeq(); void o_subeq(); void o_muleq(); void o_diveq(); void o_lnkeq(); void o_addeqc(); void o_subeqc(); void o_muleqc(); void o_diveqc(); void o_lnkeqc(); void o_meq(); void o_eq(); void o_neq(); void o_less(); void o_leq(); void o_grt(); void o_geq(); void o_lessc(); void o_leqc(); void o_grtc(); void o_geqc(); void o_pop(); void o_jmp(); void o_jt(); void o_jf(); void o_cnt(); void o_findex(); void o_feach(); void o_callg(); void o_calll(); void o_upval(); void o_callv(); void o_callvi(); void o_callh(); void o_callfv(); void o_callfh(); void o_callb(); void o_slcbeg(); void o_slcend(); void o_slc(); void o_slc2(); void o_mcallg(); void o_mcalll(); void o_mupval(); void o_mcallv(); void o_mcallh(); void o_ret(); public: nasal_vm():pc(0),localr(nullptr),memr(nullptr),funcr(nil), upvalr(nil),canary(nullptr),top(stack), num_table(nullptr),str_table(nullptr), gc(pc,localr,memr,funcr,upvalr,canary,top,stack), files(nullptr),bytecode(nullptr),detail_info(false){} void run( const nasal_codegen&, const nasal_import&, const std::vector&, const bool); }; void nasal_vm::init( const std::vector& strs, const std::vector& nums, const std::vector& code, const std::vector& filenames, const std::vector& argv) { gc.init(strs,argv); num_table=nums.data(); str_table=strs.data(); bytecode=code.data(); files=filenames.data(); /* set canary and program counter */ pc=0; localr=memr=nullptr; funcr=upvalr=nil; canary=stack+STACK_DEPTH-1; // stack[STACK_DEPTH-1] top=stack; /* clear main stack */ for(u32 i=0;i "< entry:0x"< ["< ["< {"< obj:0x"<<(u64)val.obj().ptr < coroutine";break; default: std::cout<<"| err | <0x"< unknown object";break; } std::cout<<"\n"; } void nasal_vm::bytecodeinfo(const char* header,const u32 p) { const opcode& c=bytecode[p]; c.print(header,num_table,str_table,p,true); std::cout<<" ("< ret; for(nas_ref* i=bottom;i<=top;++i) if(i->type==vm_ret) ret.push(i->ret()); ret.push(pc); // store the position program crashed std::cout<<"trace back:\n"; for(u32 p=0,same=0,prev=0xffffffff;!ret.empty();prev=p,ret.pop()) { if((p=ret.top())==prev) { ++same; continue; } if(same) std::cout <<" 0x"<, limit "<=bottom;++i,--t) { std::cout<<" 0x"<)\n"<)\n"< upval["<>16)&0xffff] .upval()[imm[pc]&0xffff]=(top--)[0]; } inline void nasal_vm::o_pnum() { (++top)[0]={vm_num,num_table[imm[pc]]}; } inline void nasal_vm::o_pnil() { (++top)[0]=nil; } inline void nasal_vm::o_pstr() { (++top)[0]=gc.strs[imm[pc]]; } inline void nasal_vm::o_newv() { nas_ref newv=gc.alloc(vm_vec); auto& vec=newv.vec().elems; vec.resize(imm[pc]); // use top-=imm[pc]-1 here will cause error if imm[pc] is 0 top=top-imm[pc]+1; for(u32 i=0;isize has 1 place reserved for "me" func.keys[imm[pc]]=func.psize; func.local[func.psize++]={vm_none}; } inline void nasal_vm::o_deft() { nas_ref val=top[0]; nas_func& func=(--top)[0].func(); // func->size has 1 place reserved for "me" func.keys[imm[pc]]=func.psize; func.local[func.psize++]=val; } inline void nasal_vm::o_dyn() { top[0].func().dpara=imm[pc]; } inline void nasal_vm::o_unot() { nas_ref val=top[0]; switch(val.type) { case vm_nil:top[0]=one;break; case vm_num:top[0]=val.num()?zero:one;break; case vm_str: { const f64 num=str2num(val.str().c_str()); if(std::isnan(num)) top[0]={vm_num,(f64)val.str().empty()}; else top[0]=num?zero:one; }break; default:die("unot: incorrect value type");break; } } inline void nasal_vm::o_usub() { top[0]={vm_num,-top[0].tonum()}; } #define op_calc(type)\ top[-1]={vm_num,top[-1].tonum() type top[0].tonum()};\ --top; inline void nasal_vm::o_add(){op_calc(+);} inline void nasal_vm::o_sub(){op_calc(-);} inline void nasal_vm::o_mul(){op_calc(*);} inline void nasal_vm::o_div(){op_calc(/);} inline void nasal_vm::o_lnk() { top[-1]=gc.newstr(top[-1].tostr()+top[0].tostr()); --top; } #define op_calc_const(type)\ top[0]={vm_num,top[0].tonum() type num_table[imm[pc]]}; inline void nasal_vm::o_addc(){op_calc_const(+);} inline void nasal_vm::o_subc(){op_calc_const(-);} inline void nasal_vm::o_mulc(){op_calc_const(*);} inline void nasal_vm::o_divc(){op_calc_const(/);} inline void nasal_vm::o_lnkc() { top[0]=gc.newstr(top[0].tostr()+str_table[imm[pc]]); } #define op_calc_eq(type)\ top[-1]=memr[0]={vm_num,memr[0].tonum() type top[-1].tonum()};\ memr=nullptr;\ top-=imm[pc]+1; inline void nasal_vm::o_addeq(){op_calc_eq(+);} inline void nasal_vm::o_subeq(){op_calc_eq(-);} inline void nasal_vm::o_muleq(){op_calc_eq(*);} inline void nasal_vm::o_diveq(){op_calc_eq(/);} inline void nasal_vm::o_lnkeq() { top[-1]=memr[0]=gc.newstr(memr[0].tostr()+top[-1].tostr()); memr=nullptr; top-=imm[pc]+1; } #define op_calc_eq_const(type)\ top[0]=memr[0]={vm_num,memr[0].tonum() type num_table[imm[pc]&0x7fffffff]};\ memr=nullptr;\ top-=(imm[pc]>>31); inline void nasal_vm::o_addeqc(){op_calc_eq_const(+);} inline void nasal_vm::o_subeqc(){op_calc_eq_const(-);} inline void nasal_vm::o_muleqc(){op_calc_eq_const(*);} inline void nasal_vm::o_diveqc(){op_calc_eq_const(/);} inline void nasal_vm::o_lnkeqc() { top[0]=memr[0]=gc.newstr(memr[0].tostr()+str_table[imm[pc]&0x7fffffff]); memr=nullptr; top-=(imm[pc]>>31); } inline void nasal_vm::o_meq() { // pop old memr[0] and replace it // the reason why we should get memr and push the old value on stack // is that when lnkeq/lnkeqc is called, there will be // a new gc object vm_str which is returned by gc::alloc // this may cause gc, so we should temporarily put it on stack memr[0]=top[-1]; memr=nullptr; top-=imm[pc]+1; } inline void nasal_vm::o_eq() { nas_ref val2=top[0]; nas_ref val1=(--top)[0]; if(val1.type==vm_nil && val2.type==vm_nil) top[0]=one; else if(val1.type==vm_str && val2.type==vm_str) top[0]=(val1.str()==val2.str())?one:zero; else if((val1.type==vm_num || val2.type==vm_num) && val1.type!=vm_nil && val2.type!=vm_nil) top[0]=(val1.tonum()==val2.tonum())?one:zero; else top[0]=(val1==val2)?one:zero; } inline void nasal_vm::o_neq() { nas_ref val2=top[0]; nas_ref val1=(--top)[0]; if(val1.type==vm_nil && val2.type==vm_nil) top[0]=zero; else if(val1.type==vm_str && val2.type==vm_str) top[0]=(val1.str()!=val2.str())?one:zero; else if((val1.type==vm_num || val2.type==vm_num) && val1.type!=vm_nil && val2.type!=vm_nil) top[0]=(val1.tonum()!=val2.tonum())?one:zero; else top[0]=(val1!=val2)?one:zero; } #define op_cmp(type)\ --top;\ top[0]=(top[0].tonum() type top[1].tonum())?one:zero; inline void nasal_vm::o_less(){op_cmp(<);} inline void nasal_vm::o_leq(){op_cmp(<=);} inline void nasal_vm::o_grt(){op_cmp(>);} inline void nasal_vm::o_geq(){op_cmp(>=);} #define op_cmp_const(type)\ top[0]=(top[0].tonum() type num_table[imm[pc]])?one:zero; inline void nasal_vm::o_lessc(){op_cmp_const(<);} inline void nasal_vm::o_leqc(){op_cmp_const(<=);} inline void nasal_vm::o_grtc(){op_cmp_const(>);} inline void nasal_vm::o_geqc(){op_cmp_const(>=);} inline void nasal_vm::o_pop() { --top; } inline void nasal_vm::o_jmp() { pc=imm[pc]-1; } inline void nasal_vm::o_jt() { if(condition(top[0])) pc=imm[pc]-1; } inline void nasal_vm::o_jf() { if(!condition(top[0])) pc=imm[pc]-1; --top; } inline void nasal_vm::o_cnt() { if(top[0].type!=vm_vec) die("cnt: must use vector in forindex/foreach"); (++top)[0]={vm_cnt,(i64)-1}; } inline void nasal_vm::o_findex() { if((usize)(++top[0].cnt())>=top[-1].vec().size()) { pc=imm[pc]-1; return; } top[1]={vm_num,(f64)top[0].cnt()}; ++top; } inline void nasal_vm::o_feach() { std::vector& ref=top[-1].vec().elems; if((usize)(++top[0].cnt())>=ref.size()) { pc=imm[pc]-1; return; } top[1]=ref[top[0].cnt()]; ++top; } inline void nasal_vm::o_callg() { (++top)[0]=stack[imm[pc]]; } inline void nasal_vm::o_calll() { (++top)[0]=localr[imm[pc]]; } inline void nasal_vm::o_upval() { (++top)[0]=funcr.func().upval[(imm[pc]>>16)&0xffff] .upval()[imm[pc]&0xffff]; } inline void nasal_vm::o_callv() { nas_ref val=top[0]; nas_ref vec=(--top)[0]; if(vec.type==vm_vec) { top[0]=vec.vec().get_val(val.tonum()); if(top[0].type==vm_none) die("callv: index out of range:"+std::to_string(val.tonum())); } else if(vec.type==vm_hash) { if(val.type!=vm_str) die("callv: must use string as the key"); top[0]=vec.hash().get_val(val.str()); if(top[0].type==vm_none) die("callv: cannot find member \""+val.str()+"\""); if(top[0].type==vm_func) top[0].func().local[0]=val;// 'me' } else if(vec.type==vm_str) { string& str=vec.str(); i32 num=val.tonum(); i32 len=str.length(); if(num<-len || num>=len) die("callv: index out of range:"+std::to_string(val.tonum())); top[0]={vm_num,f64((u8)str[num>=0? num:num+len])}; } else die("callv: must call a vector/hash/string"); } inline void nasal_vm::o_callvi() { nas_ref val=top[0]; if(val.type!=vm_vec) die("callvi: must use a vector"); // cannot use operator[],because this may cause overflow (++top)[0]=val.vec().get_val(imm[pc]); if(top[0].type==vm_none) die("callvi: index out of range:"+std::to_string(imm[pc])); } inline void nasal_vm::o_callh() { nas_ref val=top[0]; if(val.type!=vm_hash) die("callh: must call a hash"); top[0]=val.hash().get_val(str_table[imm[pc]]); if(top[0].type==vm_none) die("callh: member \""+str_table[imm[pc]]+"\" does not exist"); if(top[0].type==vm_func) top[0].func().local[0]=val;// 'me' } inline void nasal_vm::o_callfv() { u32 argc=imm[pc]; // arguments counter nas_ref* local=top-argc+1; // arguments begin address if(local[-1].type!=vm_func) die("callfv: must call a function"); auto& func=local[-1].func(); nas_ref tmp=local[-1]; local[-1]=funcr; funcr=tmp; // top-argc+lsize(local) +1(old pc) +1(old localr) +1(old upvalr) if(top-argc+func.lsize+3>=canary) die("stack overflow"); // parameter size is func->psize-1, 1 is reserved for "me" u32 psize=func.psize-1; if(argc=0)// load dynamic arguments { dynamic=gc.alloc(vm_vec); for(u32 i=psize;i=1;--i)// load arguments local[i]=local[i-1]; local[0]=func.local[0];// load "me" // load local scope & default arguments for(u32 i=min_size+1;i=0) local[psize+1]=dynamic; top[0]=upvalr; (++top)[0]={vm_addr,localr}; (++top)[0]={vm_ret,pc}; pc=func.entry-1; localr=local; upvalr=nil; } inline void nasal_vm::o_callfh() { auto& hash=top[0].hash().elems; if(top[-1].type!=vm_func) die("callfh: must call a function"); auto& func=top[-1].func(); nas_ref tmp=top[-1]; top[-1]=funcr; funcr=tmp; // top -1(hash) +lsize(local) +1(old pc) +1(old localr) +1(old upvalr) if(top+func.lsize+2>=canary) die("stack overflow"); if(func.dpara>=0) die("callfh: special call cannot use dynamic argument"); nas_ref* local=top; top+=func.lsize; for(u32 i=0;i& ref=top[-1].vec().elems; std::vector& aim=top[0].vec().elems; u8 type1=val1.type,type2=val2.type; i32 num1=val1.tonum(); i32 num2=val2.tonum(); i32 size=ref.size(); if(type1==vm_nil && type2==vm_nil) { num1=0; num2=size-1; } else if(type1==vm_nil && type2!=vm_nil) num1=num2<0? -size:0; else if(type1!=vm_nil && type2==vm_nil) num2=num1<0? -1:size-1; if(num1>num2) die("slc2: begin index must be less than or equal to end index"); else if(num1<-size || num1>=size) die("slc2: begin index out of range: "+std::to_string(num1)); else if(num2<-size || num2>=size) die("slc2: end index out of range: "+std::to_string(num2)); else for(i32 i=num1;i<=num2;++i) aim.push_back(i>=0?ref[i]:ref[i+size]); } inline void nasal_vm::o_mcallg() { memr=stack+imm[pc]; (++top)[0]=memr[0]; // push value in this memory space on stack // to avoid being garbage collected } inline void nasal_vm::o_mcalll() { memr=localr+imm[pc]; (++top)[0]=memr[0]; // push value in this memory space on stack // to avoid being garbage collected } inline void nasal_vm::o_mupval() { memr=&(funcr.func().upval[(imm[pc]>>16)&0xffff].upval()[imm[pc]&0xffff]); (++top)[0]=memr[0]; // push value in this memory space on stack // to avoid being garbage collected } inline void nasal_vm::o_mcallv() { nas_ref val=top[0]; // index nas_ref vec=(--top)[0]; // mcall vector, reserved on stack to avoid gc if(vec.type==vm_vec) { memr=vec.vec().get_mem(val.tonum()); if(!memr) die("mcallv: index out of range:"+std::to_string(val.tonum())); } else if(vec.type==vm_hash) // do mcallh but use the mcallv way { if(val.type!=vm_str) die("mcallv: must use string as the key"); nas_hash& ref=vec.hash(); string& str=val.str(); memr=ref.get_mem(str); if(!memr) { ref.elems[str]=nil; memr=ref.get_mem(str); } } else die("mcallv: cannot get memory space in other types"); } inline void nasal_vm::o_mcallh() { nas_ref hash=top[0]; // mcall hash, reserved on stack to avoid gc if(hash.type!=vm_hash) die("mcallh: must call a hash"); nas_hash& ref=hash.hash(); const string& str=str_table[imm[pc]]; memr=ref.get_mem(str); if(!memr) // create a new key { ref.elems[str]=nil; memr=ref.get_mem(str); } } inline void nasal_vm::o_ret() { /* +-------------+ * | return value| <- top[0] * +-------------+ * | old pc | <- top[-1] * +-------------+ * | old localr | <- top[-2] * +-------------+ * | old upvalr | <- top[-3] * +-------------+ * | local scope | * | ... | * +-------------+ <- local pointer stored in localr * | old funcr | <- old function stored in funcr * +-------------+ */ nas_ref ret =top[0]; nas_ref* local=localr; nas_ref func =funcr; nas_ref up =upvalr; pc =top[-1].ret(); localr=top[-2].addr(); upvalr=top[-3]; top=local-1; funcr=top[0]; top[0]=ret; // rewrite func with returned value if(up.type==vm_upval) // synchronize upvalue { auto& upval=up.upval(); auto size=func.func().lsize; upval.onstk=false; for(u32 i=0;i& argv, const bool detail) { detail_info=detail; init(gen.strs(),gen.nums(),gen.codes(),linker.filelist(),argv); #ifndef _MSC_VER const void* oprs[]= { &&vmexit, &&intg, &&intl, &&loadg, &&loadl, &&loadu, &&pnum, &&pnil, &&pstr, &&newv, &&newh, &&newf, &&happ, &¶, &&deft, &&dyn, &&unot, &&usub, &&add, &&sub, &&mul, &&div, &&lnk, &&addc, &&subc, &&mulc, &&divc, &&lnkc, &&addeq, &&subeq, &&muleq, &&diveq, &&lnkeq, &&addeqc, &&subeqc, &&muleqc, &&diveqc, &&lnkeqc, &&meq, &&eq, &&neq, &&less, &&leq, &&grt, &&geq, &&lessc, &&leqc, &&grtc, &&geqc, &&pop, &&jmp, &&jt, &&jf, &&cnt, &&findex, &&feach, &&callg, &&calll, &&upval, &&callv, &&callvi, &&callh, &&callfv, &&callfh, &&callb, &&slcbeg, &&slcend, &&slc, &&slc2, &&mcallg, &&mcalll, &&mupval, &&mcallv, &&mcallh, &&ret }; std::vector code; for(auto& i:gen.codes()) { code.push_back(oprs[i.op]); imm.push_back(i.num); } // goto the first operand goto *code[pc]; #else typedef void (nasal_vm::*nafunc)(); const nafunc oprs[]= { nullptr, &nasal_vm::o_intg, &nasal_vm::o_intl, &nasal_vm::o_loadg, &nasal_vm::o_loadl, &nasal_vm::o_loadu, &nasal_vm::o_pnum, &nasal_vm::o_pnil, &nasal_vm::o_pstr, &nasal_vm::o_newv, &nasal_vm::o_newh, &nasal_vm::o_newf, &nasal_vm::o_happ, &nasal_vm::o_para, &nasal_vm::o_deft, &nasal_vm::o_dyn, &nasal_vm::o_unot, &nasal_vm::o_usub, &nasal_vm::o_add, &nasal_vm::o_sub, &nasal_vm::o_mul, &nasal_vm::o_div, &nasal_vm::o_lnk, &nasal_vm::o_addc, &nasal_vm::o_subc, &nasal_vm::o_mulc, &nasal_vm::o_divc, &nasal_vm::o_lnkc, &nasal_vm::o_addeq, &nasal_vm::o_subeq, &nasal_vm::o_muleq, &nasal_vm::o_diveq, &nasal_vm::o_lnkeq, &nasal_vm::o_addeqc, &nasal_vm::o_subeqc, &nasal_vm::o_muleqc, &nasal_vm::o_diveqc, &nasal_vm::o_lnkeqc, &nasal_vm::o_meq, &nasal_vm::o_eq, &nasal_vm::o_neq, &nasal_vm::o_less, &nasal_vm::o_leq, &nasal_vm::o_grt, &nasal_vm::o_geq, &nasal_vm::o_lessc, &nasal_vm::o_leqc, &nasal_vm::o_grtc, &nasal_vm::o_geqc, &nasal_vm::o_pop, &nasal_vm::o_jmp, &nasal_vm::o_jt, &nasal_vm::o_jf, &nasal_vm::o_cnt, &nasal_vm::o_findex, &nasal_vm::o_feach, &nasal_vm::o_callg, &nasal_vm::o_calll, &nasal_vm::o_upval, &nasal_vm::o_callv, &nasal_vm::o_callvi, &nasal_vm::o_callh, &nasal_vm::o_callfv, &nasal_vm::o_callfh, &nasal_vm::o_callb, &nasal_vm::o_slcbeg, &nasal_vm::o_slcend, &nasal_vm::o_slc, &nasal_vm::o_slc2, &nasal_vm::o_mcallg, &nasal_vm::o_mcalll, &nasal_vm::o_mupval, &nasal_vm::o_mcallv, &nasal_vm::o_mcallh, &nasal_vm::o_ret }; std::vector code; for(auto& i:gen.codes()) { code.push_back(oprs[i.op]); imm.push_back(i.num); } while(code[pc]){ (this->*code[pc])(); if(top>=canary) break; ++pc; } #endif vmexit: if(top>=canary) die("stack overflow"); if(detail) gc.info(); gc.clear(); imm.clear(); return; #ifndef _MSC_VER // may cause stackoverflow #define exec_check(op) {\ op();\ if(top