#ifndef __NASAL_VM_H__ #define __NASAL_VM_H__ class nasal_vm { protected: /* registers and constants of nasal_vm */ uint32_t pc; // program counter nasal_ref* localr; // local scope register nasal_ref* memr; // used for mem_call nasal_ref funcr; // function register nasal_ref upvalr; // upvalue register nasal_ref* canary; // avoid stackoverflow nasal_ref* top; // stack top const double* num_table;// const numbers, ref from nasal_codegen const std::string* str_table;// const symbols, ref from nasal_codegen std::vector imm; // immediate number /* garbage collector */ nasal_gc gc; /* main stack */ nasal_ref stack[STACK_DEPTH]; /* values used for debugger */ const std::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(nasal_ref&); void bytecodeinfo(const char*,const uint32_t); void traceback(); void stackinfo(const uint32_t); void register_info(); void global_state(); void local_state(); void upval_state(); void detail(); void opcallsort(const uint64_t*); void die(const std::string&); /* vm calculation functions*/ bool condition(nasal_ref); /* vm operands */ void opr_intg(); void opr_intl(); void opr_loadg(); void opr_loadl(); void opr_loadu(); void opr_pnum(); void opr_pnil(); void opr_pstr(); void opr_newv(); void opr_newh(); void opr_newf(); void opr_happ(); void opr_para(); void opr_deft(); void opr_dyn(); void opr_unot(); void opr_usub(); void opr_add(); void opr_sub(); void opr_mul(); void opr_div(); void opr_lnk(); void opr_addc(); void opr_subc(); void opr_mulc(); void opr_divc(); void opr_lnkc(); void opr_addeq(); void opr_subeq(); void opr_muleq(); void opr_diveq(); void opr_lnkeq(); void opr_addeqc(); void opr_subeqc(); void opr_muleqc(); void opr_diveqc(); void opr_lnkeqc(); void opr_meq(); void opr_eq(); void opr_neq(); void opr_less(); void opr_leq(); void opr_grt(); void opr_geq(); void opr_lessc(); void opr_leqc(); void opr_grtc(); void opr_geqc(); void opr_pop(); void opr_jmp(); void opr_jt(); void opr_jf(); void opr_cnt(); void opr_findex(); void opr_feach(); void opr_callg(); void opr_calll(); void opr_upval(); void opr_callv(); void opr_callvi(); void opr_callh(); void opr_callfv(); void opr_callfh(); void opr_callb(); void opr_slcbeg(); void opr_slcend(); void opr_slc(); void opr_slc2(); void opr_mcallg(); void opr_mcalll(); void opr_mupval(); void opr_mcallv(); void opr_mcallh(); void opr_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, 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(uint32_t i=0;i "< entry:0x"< ["< ["< {"< obj:0x"<<(uint64_t)val.obj().ptr < coroutine";break; default: std::cout<<"| err | <0x"< unknown object";break; } std::cout<<"\n"; } void nasal_vm::bytecodeinfo(const char* header,const uint32_t p) { const opcode& c=bytecode[p]; c.print(header,num_table,str_table,p,true); std::cout<<" ("< ret; for(nasal_ref* i=bottom;i<=t;++i) if(i->type==vm_ret) ret.push(i->ret()); // push pc to ret stack to store the position program crashed ret.push(pc); std::cout<<"trace back:\n"; uint32_t same=0,last=0xffffffff; for(uint32_t point=0;!ret.empty();last=point,ret.pop()) { if((point=ret.top())==last) { ++same; continue; } if(same) std::cout<<" 0x"<, limit "<=bottom;++i,--t) { std::cout<<" 0x"<)\n"<)\n"< upval["< op; std::vector opcall; uint64_t total=0; for(uint32_t i=0;ib.second;} ); std::clog<<"\noperands call info"; for(auto& i:opcall) { uint64_t rate=i.second*100/total; if(rate) std::clog<<"\n "<>16)&0xffff] .upval()[imm[pc]&0xffff]=(top--)[0]; } inline void nasal_vm::opr_pnum() { (++top)[0]={vm_num,num_table[imm[pc]]}; } inline void nasal_vm::opr_pnil() { (++top)[0]=nil; } inline void nasal_vm::opr_pstr() { (++top)[0]=gc.strs[imm[pc]]; } inline void nasal_vm::opr_newv() { nasal_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(uint32_t i=0;isize has 1 place reserved for "me" func.keys[str_table[imm[pc]]]=func.psize; func.local[func.psize++]={vm_none}; } inline void nasal_vm::opr_deft() { nasal_ref val=top[0]; nasal_func& func=(--top)[0].func(); // func->size has 1 place reserved for "me" func.keys[str_table[imm[pc]]]=func.psize; func.local[func.psize++]=val; } inline void nasal_vm::opr_dyn() { top[0].func().dynpara=imm[pc]; } inline void nasal_vm::opr_unot() { nasal_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 double num=str2num(val.str().c_str()); if(std::isnan(num)) top[0]={vm_num,(double)val.str().empty()}; else top[0]=num?zero:one; }break; default:die("unot: incorrect value type");break; } } inline void nasal_vm::opr_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::opr_add(){op_calc(+);} inline void nasal_vm::opr_sub(){op_calc(-);} inline void nasal_vm::opr_mul(){op_calc(*);} inline void nasal_vm::opr_div(){op_calc(/);} inline void nasal_vm::opr_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::opr_addc(){op_calc_const(+);} inline void nasal_vm::opr_subc(){op_calc_const(-);} inline void nasal_vm::opr_mulc(){op_calc_const(*);} inline void nasal_vm::opr_divc(){op_calc_const(/);} inline void nasal_vm::opr_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::opr_addeq(){op_calc_eq(+);} inline void nasal_vm::opr_subeq(){op_calc_eq(-);} inline void nasal_vm::opr_muleq(){op_calc_eq(*);} inline void nasal_vm::opr_diveq(){op_calc_eq(/);} inline void nasal_vm::opr_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::opr_addeqc(){op_calc_eq_const(+);} inline void nasal_vm::opr_subeqc(){op_calc_eq_const(-);} inline void nasal_vm::opr_muleqc(){op_calc_eq_const(*);} inline void nasal_vm::opr_diveqc(){op_calc_eq_const(/);} inline void nasal_vm::opr_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::opr_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::opr_eq() { nasal_ref val2=top[0]; nasal_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::opr_neq() { nasal_ref val2=top[0]; nasal_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::opr_less(){op_cmp(<);} inline void nasal_vm::opr_leq(){op_cmp(<=);} inline void nasal_vm::opr_grt(){op_cmp(>);} inline void nasal_vm::opr_geq(){op_cmp(>=);} #define op_cmp_const(type)\ top[0]=(top[0].tonum() type num_table[imm[pc]])?one:zero; inline void nasal_vm::opr_lessc(){op_cmp_const(<);} inline void nasal_vm::opr_leqc(){op_cmp_const(<=);} inline void nasal_vm::opr_grtc(){op_cmp_const(>);} inline void nasal_vm::opr_geqc(){op_cmp_const(>=);} inline void nasal_vm::opr_pop() { --top; } inline void nasal_vm::opr_jmp() { pc=imm[pc]-1; } inline void nasal_vm::opr_jt() { if(condition(top[0])) pc=imm[pc]-1; } inline void nasal_vm::opr_jf() { if(!condition(top[0])) pc=imm[pc]-1; --top; } inline void nasal_vm::opr_cnt() { (++top)[0]={vm_cnt,(int64_t)-1}; if(top[-1].type!=vm_vec) die("cnt: must use vector in forindex/foreach"); } inline void nasal_vm::opr_findex() { if((size_t)(++top[0].cnt())>=top[-1].vec().size()) { pc=imm[pc]-1; return; } top[1]={vm_num,(double)top[0].cnt()}; ++top; } inline void nasal_vm::opr_feach() { std::vector& ref=top[-1].vec().elems; if((size_t)(++top[0].cnt())>=ref.size()) { pc=imm[pc]-1; return; } top[1]=ref[top[0].cnt()]; ++top; } inline void nasal_vm::opr_callg() { (++top)[0]=stack[imm[pc]]; } inline void nasal_vm::opr_calll() { (++top)[0]=localr[imm[pc]]; } inline void nasal_vm::opr_upval() { (++top)[0]=funcr.func().upvalue[(imm[pc]>>16)&0xffff] .upval()[imm[pc]&0xffff]; } inline void nasal_vm::opr_callv() { nasal_ref val=top[0]; nasal_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) { std::string& str=vec.str(); int num=val.tonum(); int len=str.length(); if(num<-len || num>=len) die("callv: index out of range:"+std::to_string(val.tonum())); top[0]={vm_num,double((uint8_t)str[num>=0? num:num+len])}; } else die("callv: must call a vector/hash/string"); } inline void nasal_vm::opr_callvi() { nasal_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::opr_callh() { nasal_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::opr_callfv() { uint32_t argc=imm[pc]; // arguments counter nasal_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(); nasal_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" uint32_t psize=func.psize-1; if(argc=0)// load dynamic arguments { dynamic=gc.alloc(vm_vec); for(uint32_t 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(uint32_t 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::opr_callfh() { auto& hash=top[0].hash().elems; if(top[-1].type!=vm_func) die("callfh: must call a function"); auto& func=top[-1].func(); nasal_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.dynpara>=0) die("callfh: special call cannot use dynamic argument"); nasal_ref* local=top; top+=func.lsize; for(uint32_t i=0;i& ref=top[-1].vec().elems; std::vector& aim=top[0].vec().elems; uint8_t type1=val1.type,type2=val2.type; int num1=val1.tonum(); int num2=val2.tonum(); int 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(int i=num1;i<=num2;++i) aim.push_back(i>=0?ref[i]:ref[i+size]); } inline void nasal_vm::opr_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::opr_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::opr_mupval() { memr=&(funcr.func().upvalue[(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::opr_mcallv() { nasal_ref val=top[0]; // index nasal_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"); nasal_hash& ref=vec.hash(); std::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::opr_mcallh() { nasal_ref hash=top[0]; // mcall hash, reserved on stack to avoid gc if(hash.type!=vm_hash) die("mcallh: must call a hash"); nasal_hash& ref=hash.hash(); const std::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::opr_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 * +-----------------+ */ nasal_ref ret =top[0]; nasal_ref* local=localr; nasal_ref func =funcr; nasal_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(uint32_t i=0;i& argv, const bool opcnt, const bool detail) { detail_info=detail; init(gen.strs(),gen.nums(),gen.codes(),linker.filelist(),argv); uint64_t count[op_ret+1]={0}; #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::opr_intg, &nasal_vm::opr_intl, &nasal_vm::opr_loadg, &nasal_vm::opr_loadl, &nasal_vm::opr_loadu, &nasal_vm::opr_pnum, &nasal_vm::opr_pnil, &nasal_vm::opr_pstr, &nasal_vm::opr_newv, &nasal_vm::opr_newh, &nasal_vm::opr_newf, &nasal_vm::opr_happ, &nasal_vm::opr_para, &nasal_vm::opr_deft, &nasal_vm::opr_dyn, &nasal_vm::opr_unot, &nasal_vm::opr_usub, &nasal_vm::opr_add, &nasal_vm::opr_sub, &nasal_vm::opr_mul, &nasal_vm::opr_div, &nasal_vm::opr_lnk, &nasal_vm::opr_addc, &nasal_vm::opr_subc, &nasal_vm::opr_mulc, &nasal_vm::opr_divc, &nasal_vm::opr_lnkc, &nasal_vm::opr_addeq, &nasal_vm::opr_subeq, &nasal_vm::opr_muleq, &nasal_vm::opr_diveq, &nasal_vm::opr_lnkeq, &nasal_vm::opr_addeqc, &nasal_vm::opr_subeqc, &nasal_vm::opr_muleqc, &nasal_vm::opr_diveqc, &nasal_vm::opr_lnkeqc, &nasal_vm::opr_meq, &nasal_vm::opr_eq, &nasal_vm::opr_neq, &nasal_vm::opr_less, &nasal_vm::opr_leq, &nasal_vm::opr_grt, &nasal_vm::opr_geq, &nasal_vm::opr_lessc, &nasal_vm::opr_leqc, &nasal_vm::opr_grtc, &nasal_vm::opr_geqc, &nasal_vm::opr_pop, &nasal_vm::opr_jmp, &nasal_vm::opr_jt, &nasal_vm::opr_jf, &nasal_vm::opr_cnt, &nasal_vm::opr_findex, &nasal_vm::opr_feach, &nasal_vm::opr_callg, &nasal_vm::opr_calll, &nasal_vm::opr_upval, &nasal_vm::opr_callv, &nasal_vm::opr_callvi, &nasal_vm::opr_callh, &nasal_vm::opr_callfv, &nasal_vm::opr_callfh, &nasal_vm::opr_callb, &nasal_vm::opr_slcbeg, &nasal_vm::opr_slcend, &nasal_vm::opr_slc, &nasal_vm::opr_slc2, &nasal_vm::opr_mcallg, &nasal_vm::opr_mcalll, &nasal_vm::opr_mupval, &nasal_vm::opr_mcallv, &nasal_vm::opr_mcallh, &nasal_vm::opr_ret }; std::vector code; for(auto& i:gen.codes()) { code.push_back(i.op); imm.push_back(i.num); } while(oprs[code[pc]]){ ++count[code[pc]]; (this->*oprs[code[pc]])(); if(top>=canary) break; ++pc; } #endif vmexit: if(top>=canary) die("stack overflow"); if(opcnt) opcallsort(count); if(detail) gc.info(); gc.clear(); imm.clear(); return; #ifndef _MSC_VER // may cause stackoverflow #define exec_operand(op,num) {\ op();\ ++count[num];\ if(top