/* see copyright notice in squirrel.h */ #include "sqpcheader.h" #include #include #include "sqopcodes.h" #include "sqfuncproto.h" #include "sqvm.h" #include "sqclosure.h" #include "sqstring.h" #include "sqtable.h" #include "squserdata.h" #include "sqarray.h" #include "sqclass.h" #define TOP() (_stack[_top-1]) bool SQVM::BW_OP(SQUnsignedInteger op,SQObjectPtr &trg,const SQObjectPtr &o1,const SQObjectPtr &o2) { SQInteger res; SQInteger i1 = _integer(o1), i2 = _integer(o2); if((type(o1)==OT_INTEGER) && (type(o2)==OT_INTEGER)) { switch(op) { case BW_AND: res = i1 & i2; break; case BW_OR: res = i1 | i2; break; case BW_XOR: res = i1 ^ i2; break; case BW_SHIFTL: res = i1 << i2; break; case BW_SHIFTR: res = i1 >> i2; break; case BW_USHIFTR:res = (SQInteger)(*((SQUnsignedInteger*)&i1) >> i2); break; default: { Raise_Error(_SC("internal vm error bitwise op failed")); return false; } } } else { Raise_Error(_SC("bitwise op between '%s' and '%s'"),GetTypeName(o1),GetTypeName(o2)); return false;} trg = res; return true; } bool SQVM::ARITH_OP(SQUnsignedInteger op,SQObjectPtr &trg,const SQObjectPtr &o1,const SQObjectPtr &o2) { if(sq_isnumeric(o1) && sq_isnumeric(o2)) { if((type(o1)==OT_INTEGER) && (type(o2)==OT_INTEGER)) { switch(op) { case '+': trg = _integer(o1) + _integer(o2); break; case '-': trg = _integer(o1) - _integer(o2); break; case '/': if(_integer(o2) == 0) { Raise_Error(_SC("division by zero")); return false; } trg = _integer(o1) / _integer(o2); break; case '*': trg = _integer(o1) * _integer(o2); break; case '%': trg = _integer(o1) % _integer(o2); break; } }else{ switch(op) { case '+': trg = tofloat(o1) + tofloat(o2); break; case '-': trg = tofloat(o1) - tofloat(o2); break; case '/': trg = tofloat(o1) / tofloat(o2); break; case '*': trg = tofloat(o1) * tofloat(o2); break; case '%': trg = SQFloat(fmod((double)tofloat(o1),(double)tofloat(o2))); break; } } } else { if(op == '+' && (type(o1) == OT_STRING || type(o2) == OT_STRING)){ if(!StringCat(o1, o2, trg)) return false; } else if(!ArithMetaMethod(op,o1,o2,trg)) { Raise_Error(_SC("arith op %c on between '%s' and '%s'"),op,GetTypeName(o1),GetTypeName(o2)); return false; } } return true; } SQObjectPtr &stack_get(HSQUIRRELVM v,SQInteger idx){return ((idx>=0)?(v->GetAt(idx+v->_stackbase-1)):(v->GetUp(idx)));} SQVM::SQVM(SQSharedState *ss) { _sharedstate=ss; _suspended = SQFalse; _suspended_target=-1; _suspended_root = SQFalse; _suspended_traps=-1; _foreignptr=NULL; _nnativecalls=0; _lasterror = _null_; _errorhandler = _null_; _debughook = _null_; INIT_CHAIN();ADD_TO_CHAIN(&_ss(this)->_gc_chain,this); } void SQVM::Finalize() { _roottable = _null_; _lasterror = _null_; _errorhandler = _null_; _debughook = _null_; temp_reg = _null_; SQInteger size=_stack.size(); for(SQInteger i=0;i_gc_chain,this); } bool SQVM::ArithMetaMethod(SQInteger op,const SQObjectPtr &o1,const SQObjectPtr &o2,SQObjectPtr &dest) { SQMetaMethod mm; switch(op){ case _SC('+'): mm=MT_ADD; break; case _SC('-'): mm=MT_SUB; break; case _SC('/'): mm=MT_DIV; break; case _SC('*'): mm=MT_MUL; break; case _SC('%'): mm=MT_MODULO; break; } if(is_delegable(o1) && _delegable(o1)->_delegate) { Push(o1);Push(o2); return CallMetaMethod(_delegable(o1),mm,2,dest); } return false; } bool SQVM::NEG_OP(SQObjectPtr &trg,const SQObjectPtr &o) { switch(type(o)) { case OT_INTEGER: trg = -_integer(o); return true; case OT_FLOAT: trg = -_float(o); return true; case OT_TABLE: case OT_USERDATA: case OT_INSTANCE: if(_delegable(o)->_delegate) { Push(o); if(CallMetaMethod(_delegable(o), MT_UNM, 1, temp_reg)) { trg = temp_reg; return true; } } return true; } Raise_Error(_SC("attempt to negate a %s"), GetTypeName(o)); return false; } #define _RET_SUCCEED(exp) { result = (exp); return true; } bool SQVM::ObjCmp(const SQObjectPtr &o1,const SQObjectPtr &o2,SQInteger &result) { if(type(o1)==type(o2)){ if(_userpointer(o1)==_userpointer(o2))_RET_SUCCEED(0); SQObjectPtr res; switch(type(o1)){ case OT_STRING: _RET_SUCCEED(scstrcmp(_stringval(o1),_stringval(o2))); case OT_INTEGER: _RET_SUCCEED(_integer(o1)-_integer(o2)); case OT_FLOAT: _RET_SUCCEED((_float(o1)<_float(o2))?-1:1); case OT_TABLE: case OT_USERDATA: case OT_INSTANCE: Push(o1);Push(o2); if(_delegable(o1)->_delegate)CallMetaMethod(_delegable(o1),MT_CMP,2,res); break; } if(type(res)!=OT_INTEGER) { Raise_CompareError(o1,o2); return false; } _RET_SUCCEED(_integer(res)); } else{ if(sq_isnumeric(o1) && sq_isnumeric(o2)){ if((type(o1)==OT_INTEGER) && (type(o2)==OT_FLOAT)) { if( _integer(o1)==_float(o2) ) { _RET_SUCCEED(0); } else if( _integer(o1)<_float(o2) ) { _RET_SUCCEED(-1); } _RET_SUCCEED(1); } else{ if( _float(o1)==_integer(o2) ) { _RET_SUCCEED(0); } else if( _float(o1)<_integer(o2) ) { _RET_SUCCEED(-1); } _RET_SUCCEED(1); } } else if(type(o1)==OT_NULL) {_RET_SUCCEED(-1);} else if(type(o2)==OT_NULL) {_RET_SUCCEED(1);} else { Raise_CompareError(o1,o2); return false; } } assert(0); _RET_SUCCEED(0); //cannot happen } bool SQVM::CMP_OP(CmpOP op, const SQObjectPtr &o1,const SQObjectPtr &o2,SQObjectPtr &res) { SQInteger r; if(ObjCmp(o1,o2,r)) { switch(op) { case CMP_G: res = (r > 0)?_true_:_false_; return true; case CMP_GE: res = (r >= 0)?_true_:_false_; return true; case CMP_L: res = (r < 0)?_true_:_false_; return true; case CMP_LE: res = (r <= 0)?_true_:_false_; return true; } assert(0); } return false; } bool SQVM::StringCat(const SQObjectPtr &str,const SQObjectPtr &obj,SQObjectPtr &dest) { switch(type(obj)) { case OT_STRING: switch(type(str)){ case OT_STRING: { SQInteger l=_string(str)->_len,ol=_string(obj)->_len; SQChar *s=_sp(rsl(l+ol+1)); memcpy(s,_stringval(str),rsl(l));memcpy(s+l,_stringval(obj),rsl(ol));s[l+ol]=_SC('\0'); break; } case OT_FLOAT: scsprintf(_sp(rsl(NUMBER_MAX_CHAR+_string(obj)->_len+1)),_SC("%g%s"),_float(str),_stringval(obj)); break; case OT_INTEGER: scsprintf(_sp(rsl(NUMBER_MAX_CHAR+_string(obj)->_len+1)),_SC("%d%s"),_integer(str),_stringval(obj)); break; default: Raise_Error(_SC("string concatenation between '%s' and '%s'"),GetTypeName(str),GetTypeName(obj)); return false; } dest=SQString::Create(_ss(this),_spval); break; case OT_FLOAT: scsprintf(_sp(rsl(NUMBER_MAX_CHAR+_string(str)->_len+1)),_SC("%s%g"),_stringval(str),_float(obj)); dest=SQString::Create(_ss(this),_spval); break; case OT_INTEGER: scsprintf(_sp(rsl(NUMBER_MAX_CHAR+_string(str)->_len+1)),_SC("%s%d"),_stringval(str),_integer(obj)); dest=SQString::Create(_ss(this),_spval); break; default: Raise_Error(_SC("string concatenation between '%s' and '%s'"),GetTypeName(str),GetTypeName(obj)); return false; } return true; } const SQChar *IdType2Name(SQObjectType type) { switch(_RAW_TYPE(type)) { case _RT_NULL:return _SC("null"); case _RT_INTEGER:return _SC("integer"); case _RT_FLOAT:return _SC("float"); case _RT_BOOL:return _SC("bool"); case _RT_STRING:return _SC("string"); case _RT_TABLE:return _SC("table"); case _RT_ARRAY:return _SC("array"); case _RT_GENERATOR:return _SC("generator"); case _RT_CLOSURE: case _RT_NATIVECLOSURE: return _SC("function"); case _RT_USERDATA: case _RT_USERPOINTER: return _SC("userdata"); case _RT_THREAD: return _SC("thread"); case _RT_FUNCPROTO: return _SC("function"); case _RT_CLASS: return _SC("class"); case _RT_INSTANCE: return _SC("instance"); case _RT_WEAKREF: return _SC("weakref"); default: return NULL; } } const SQChar *GetTypeName(const SQObjectPtr &obj1) { return IdType2Name(type(obj1)); } void SQVM::TypeOf(const SQObjectPtr &obj1,SQObjectPtr &dest) { if(is_delegable(obj1) && _delegable(obj1)->_delegate) { Push(obj1); if(CallMetaMethod(_delegable(obj1),MT_TYPEOF,1,dest)) return; } dest = SQString::Create(_ss(this),GetTypeName(obj1)); } bool SQVM::Init(SQVM *friendvm, SQInteger stacksize) { _stack.resize(stacksize); _callsstack.reserve(4); _stackbase = 0; _top = 0; if(!friendvm) _roottable = SQTable::Create(_ss(this), 0); else { _roottable = friendvm->_roottable; _errorhandler = friendvm->_errorhandler; _debughook = friendvm->_debughook; } sq_base_register(this); return true; } extern SQInstructionDesc g_InstrDesc[]; bool SQVM::StartCall(SQClosure *closure,SQInteger target,SQInteger nargs,SQInteger stackbase,bool tailcall) { SQFunctionProto *func = _funcproto(closure->_function); //const SQInteger outerssize = func->_outervalues.size(); const SQInteger paramssize = func->_parameters.size(); const SQInteger oldtop = _top; const SQInteger newtop = stackbase + func->_stacksize; if(func->_varparams) { if (nargs < paramssize) { Raise_Error(_SC("wrong number of parameters")); return false; } for(SQInteger n = 0; n < nargs - paramssize; n++) { _vargsstack.push_back(_stack[stackbase+paramssize+n]); _stack[stackbase+paramssize+n] = _null_; } } else { if (paramssize != nargs) { Raise_Error(_SC("wrong number of parameters")); return false; } } if (!tailcall) { PUSH_CALLINFO(this, CallInfo()); ci->_etraps = 0; ci->_prevstkbase = stackbase - _stackbase; ci->_target = target; ci->_prevtop = _top - _stackbase; ci->_ncalls = 1; ci->_root = SQFalse; } else { ci->_ncalls++; } ci->_vargs.size = (nargs - paramssize); ci->_vargs.base = _vargsstack.size()-(nargs - paramssize); ci->_closure._unVal.pClosure = closure; ci->_closure._type = OT_CLOSURE; ci->_iv = &func->_instructions; ci->_literals = &func->_literals; //grows the stack if needed if (((SQUnsignedInteger)newtop + (func->_stacksize<<1)) > _stack.size()) { _stack.resize(_stack.size() + (func->_stacksize<<1)); } _top = newtop; _stackbase = stackbase; ci->_ip = ci->_iv->_vals; return true; } bool SQVM::Return(SQInteger _arg0, SQInteger _arg1, SQObjectPtr &retval) { if (type(_debughook) != OT_NULL && _rawval(_debughook) != _rawval(ci->_closure)) for(SQInteger i=0;i_ncalls;i++) CallDebugHook(_SC('r')); SQBool broot = ci->_root; SQInteger last_top = _top; SQInteger target = ci->_target; SQInteger oldstackbase = _stackbase; _stackbase -= ci->_prevstkbase; _top = _stackbase + ci->_prevtop; if(ci->_vargs.size) PopVarArgs(ci->_vargs); POP_CALLINFO(this); if (broot) { if (_arg0 != MAX_FUNC_STACKSIZE) retval = _stack[oldstackbase+_arg1]; else retval = _null_; } else { if (_arg0 != MAX_FUNC_STACKSIZE) STK(target) = _stack[oldstackbase+_arg1]; else STK(target) = _null_; } while (last_top >= _top) _stack[last_top--].Null(); assert(oldstackbase >= _stackbase); return broot?true:false; } #define _RET_ON_FAIL(exp) { if(!exp) return false; } bool SQVM::LOCAL_INC(SQInteger op,SQObjectPtr &target, SQObjectPtr &a, SQObjectPtr &incr) { _RET_ON_FAIL(ARITH_OP( op , target, a, incr)); a = target; return true; } bool SQVM::PLOCAL_INC(SQInteger op,SQObjectPtr &target, SQObjectPtr &a, SQObjectPtr &incr) { SQObjectPtr trg; _RET_ON_FAIL(ARITH_OP( op , trg, a, incr)); target = a; a = trg; return true; } bool SQVM::DerefInc(SQInteger op,SQObjectPtr &target, SQObjectPtr &self, SQObjectPtr &key, SQObjectPtr &incr, bool postfix) { SQObjectPtr tmp, tself = self, tkey = key; if (!Get(tself, tkey, tmp, false, true)) { Raise_IdxError(tkey); return false; } _RET_ON_FAIL(ARITH_OP( op , target, tmp, incr)) Set(tself, tkey, target,true); if (postfix) target = tmp; return true; } #define arg0 (_i_._arg0) #define arg1 (_i_._arg1) #define sarg1 (*((SQInteger *)&_i_._arg1)) #define arg2 (_i_._arg2) #define arg3 (_i_._arg3) #define sarg3 (*((char *)&_i_._arg3)) SQRESULT SQVM::Suspend() { if (_suspended) return sq_throwerror(this, _SC("cannot suspend an already suspended vm")); if (_nnativecalls!=2) return sq_throwerror(this, _SC("cannot suspend through native calls/metamethods")); return SQ_SUSPEND_FLAG; } void SQVM::PopVarArgs(VarArgs &vargs) { for(SQInteger n = 0; n< vargs.size; n++) _vargsstack.pop_back(); } #define _FINISH(stoploop) {finished = stoploop; return true; } bool SQVM::FOREACH_OP(SQObjectPtr &o1,SQObjectPtr &o2,SQObjectPtr &o3,SQObjectPtr &o4,SQInteger arg_2,bool &finished) { SQInteger nrefidx; switch(type(o1)) { case OT_TABLE: if((nrefidx = _table(o1)->Next(false,o4, o2, o3)) == -1) _FINISH(true); o4 = (SQInteger)nrefidx; _FINISH(false); case OT_ARRAY: if((nrefidx = _array(o1)->Next(o4, o2, o3)) == -1) _FINISH(true); o4 = (SQInteger) nrefidx; _FINISH(false); case OT_STRING: if((nrefidx = _string(o1)->Next(o4, o2, o3)) == -1)_FINISH(true); o4 = (SQInteger)nrefidx; _FINISH(false); case OT_CLASS: if((nrefidx = _class(o1)->Next(o4, o2, o3)) == -1)_FINISH(true); o4 = (SQInteger)nrefidx; _FINISH(false); case OT_USERDATA: case OT_INSTANCE: if(_delegable(o1)->_delegate) { SQObjectPtr itr; Push(o1); Push(o4); if(CallMetaMethod(_delegable(o1), MT_NEXTI, 2, itr)){ o4 = o2 = itr; if(type(itr) == OT_NULL) _FINISH(true); if(!Get(o1, itr, o3, false,false)) { Raise_Error(_SC("_nexti returned an invalid idx")); return false; } _FINISH(false); } Raise_Error(_SC("_nexti failed")); return false; } break; case OT_GENERATOR: if(_generator(o1)->_state == SQGenerator::eDead) _FINISH(true); if(_generator(o1)->_state == SQGenerator::eSuspended) { SQInteger idx = 0; if(type(o4) == OT_INTEGER) { idx = _integer(o4) + 1; } o2 = idx; o4 = idx; _generator(o1)->Resume(this, arg_2+1); _FINISH(false); } } Raise_Error(_SC("cannot iterate %s"), GetTypeName(o1)); return false; //cannot be hit(just to avoid warnings) } bool SQVM::DELEGATE_OP(SQObjectPtr &trg,SQObjectPtr &o1,SQObjectPtr &o2) { if(type(o1) != OT_TABLE) { Raise_Error(_SC("delegating a '%s'"), GetTypeName(o1)); return false; } switch(type(o2)) { case OT_TABLE: if(!_table(o1)->SetDelegate(_table(o2))){ Raise_Error(_SC("delegate cycle detected")); return false; } break; case OT_NULL: _table(o1)->SetDelegate(NULL); break; default: Raise_Error(_SC("using '%s' as delegate"), GetTypeName(o2)); return false; break; } trg = o1; return true; } #define COND_LITERAL (arg3!=0?(*ci->_literals)[arg1]:STK(arg1)) #define _GUARD(exp) { if(!exp) { Raise_Error(_lasterror); SQ_THROW();} } #define SQ_THROW() { goto exception_trap; } bool SQVM::CLOSURE_OP(SQObjectPtr &target, SQFunctionProto *func) { SQInteger nouters; SQClosure *closure = SQClosure::Create(_ss(this), func); if(nouters = func->_outervalues.size()) { closure->_outervalues.reserve(nouters); for(SQInteger i = 0; i_outervalues[i]; switch(v._type){ case otSYMBOL: closure->_outervalues.push_back(_null_); if(!Get(_stack._vals[_stackbase]/*STK(0)*/, v._src, closure->_outervalues.top(), false,true)) {Raise_IdxError(v._src); return false; } break; case otLOCAL: closure->_outervalues.push_back(_stack._vals[_stackbase+_integer(v._src)]); break; case otOUTER: closure->_outervalues.push_back(_closure(ci->_closure)->_outervalues[_integer(v._src)]); break; } } } target = closure; return true; } bool SQVM::GETVARGV_OP(SQObjectPtr &target,SQObjectPtr &index,CallInfo *ci) { if(ci->_vargs.size == 0) { Raise_Error(_SC("the function doesn't have var args")); return false; } if(!sq_isnumeric(index)){ Raise_Error(_SC("indexing 'vargv' with %s"),GetTypeName(index)); return false; } SQInteger idx = tointeger(index); if(idx < 0 || idx >= ci->_vargs.size){ Raise_Error(_SC("vargv index out of range")); return false; } target = _vargsstack[ci->_vargs.base+idx]; return true; } bool SQVM::CLASS_OP(SQObjectPtr &target,SQInteger baseclass,SQInteger attributes) { SQClass *base = NULL; SQObjectPtr attrs; if(baseclass != MAX_LITERALS) { if(type(_stack._vals[_stackbase+baseclass]) != OT_CLASS) { Raise_Error(_SC("trying to inherit from a %s"),GetTypeName(_stack._vals[_stackbase+baseclass])); return false; } base = _class(_stack._vals[_stackbase + baseclass]); } if(attributes != MAX_FUNC_STACKSIZE) { attrs = _stack._vals[_stackbase+attributes]; } target = SQClass::Create(_ss(this),base); _class(target)->_attributes = attrs; return true; } bool SQVM::IsFalse(SQObjectPtr &o) { SQObjectType t = type(o); if((t & SQOBJECT_CANBEFALSE) && ((t == OT_NULL) || ((t == OT_INTEGER || t == OT_BOOL) && _integer(o) == 0) || (t == OT_FLOAT && _float(o) == SQFloat(0.0)))) { return true; } return false; } bool SQVM::IsEqual(SQObjectPtr &o1,SQObjectPtr &o2,bool &res) { if(type(o1) == type(o2)) { res = ((_userpointer(o1) == _userpointer(o2)?true:false)); } else { if(sq_isnumeric(o1) && sq_isnumeric(o2)) { SQInteger cmpres; if(!ObjCmp(o1, o2,cmpres)) return false; res = (cmpres == 0); } else { res = false; } } return true; } bool SQVM::Execute(SQObjectPtr &closure, SQInteger target, SQInteger nargs, SQInteger stackbase,SQObjectPtr &outres, ExecutionType et) { if ((_nnativecalls + 1) > MAX_NATIVE_CALLS) { Raise_Error(_SC("Native stack overflow")); return false; } _nnativecalls++; AutoDec ad(&_nnativecalls); SQInteger traps = 0; //temp_reg vars for OP_CALL SQInteger ct_target; bool ct_tailcall; switch(et) { case ET_CALL: if(!StartCall(_closure(closure), _top - nargs, nargs, stackbase, false)) { //call the handler if there are no calls in the stack, if not relies on the previous node if(ci == NULL) CallErrorHandler(_lasterror); return false; } ci->_root = SQTrue; break; case ET_RESUME_GENERATOR: _generator(closure)->Resume(this, target); ci->_root = SQTrue; traps += ci->_etraps; break; case ET_RESUME_VM: traps = _suspended_traps; ci->_root = _suspended_root; _suspended = SQFalse; break; } exception_restore: // { for(;;) { const SQInstruction &_i_ = *ci->_ip++; //dumpstack(_stackbase); //scprintf("\n[%d] %s %d %d %d %d\n",ci->_ip-ci->_iv->_vals,g_InstrDesc[_i_.op].name,arg0,arg1,arg2,arg3); switch(_i_.op) { case _OP_LINE: if(type(_debughook) != OT_NULL && _rawval(_debughook) != _rawval(ci->_closure)) CallDebugHook(_SC('l'),arg1); continue; case _OP_LOAD: TARGET = (*ci->_literals)[arg1]; continue; case _OP_TAILCALL: temp_reg = STK(arg1); if (type(temp_reg) == OT_CLOSURE){ ct_tailcall = true; if(ci->_vargs.size) PopVarArgs(ci->_vargs); for (SQInteger i = 0; i < arg3; i++) STK(i) = STK(arg2 + i); ct_target = ci->_target; goto common_call; } case _OP_CALL: { ct_tailcall = false; ct_target = arg0; temp_reg = STK(arg1); common_call: SQInteger last_top = _top; switch (type(temp_reg)) { case OT_CLOSURE:{ _GUARD(StartCall(_closure(temp_reg), ct_target, arg3, ct_tailcall?_stackbase:_stackbase+arg2, ct_tailcall)); if (_funcproto(_closure(temp_reg)->_function)->_bgenerator) { SQGenerator *gen = SQGenerator::Create(_ss(this), _closure(temp_reg)); _GUARD(gen->Yield(this)); Return(1, ct_target, temp_reg); STK(ct_target) = gen; while (last_top >= _top) _stack[last_top--].Null(); continue; } if (type(_debughook) != OT_NULL && _rawval(_debughook) != _rawval(ci->_closure)) CallDebugHook(_SC('c')); } break; case OT_NATIVECLOSURE: { bool suspend; _GUARD(CallNative(_nativeclosure(temp_reg), arg3, _stackbase+arg2, ct_tailcall, temp_reg,suspend)); if(suspend){ _suspended = SQTrue; _suspended_target = ct_target; _suspended_root = ci->_root; _suspended_traps = traps; outres = temp_reg; return true; } STK(ct_target) = temp_reg; } break; case OT_CLASS:{ _GUARD(CreateClassInstance(_class(temp_reg),arg3,_stackbase+arg2,STK(ct_target))); } break; case OT_TABLE: case OT_USERDATA: case OT_INSTANCE: { Push(temp_reg); for (SQInteger i = 0; i < arg3; i++) Push(STK(arg2 + i)); if (_delegable(temp_reg) && CallMetaMethod(_delegable(temp_reg), MT_CALL, arg3+1, temp_reg)){ STK(ct_target) = temp_reg; break; } Raise_Error(_SC("attempt to call '%s'"), GetTypeName(temp_reg)); SQ_THROW(); } default: Raise_Error(_SC("attempt to call '%s'"), GetTypeName(temp_reg)); SQ_THROW(); } } continue; case _OP_PREPCALL: if (!Get(STK(arg2), STK(arg1), temp_reg, false,true)) { Raise_IdxError(STK(arg1)); SQ_THROW(); } goto common_prepcall; case _OP_PREPCALLK: if (!Get(STK(arg2), (*ci->_literals)[arg1], temp_reg,false,true)) { if(type(STK(arg2)) == OT_CLASS) { //hack? if(_class_ddel->Get((*ci->_literals)[arg1],temp_reg)) { STK(arg3) = STK(arg2); TARGET = temp_reg; continue; } } { Raise_IdxError((*ci->_literals)[arg1]); SQ_THROW();} } common_prepcall: if(type(STK(arg2)) == OT_CLASS) { STK(arg3) = STK(0); // this } else { STK(arg3) = STK(arg2); } TARGET = temp_reg; continue; case _OP_GETK: if (!Get(STK(arg2), (*ci->_literals)[arg1], temp_reg, false,true)) { Raise_IdxError((*ci->_literals)[arg1]); SQ_THROW();} TARGET = temp_reg; continue; case _OP_MOVE: TARGET = STK(arg1); continue; case _OP_NEWSLOT: _GUARD(NewSlot(STK(arg1), STK(arg2), STK(arg3))); if(arg0 != arg3) TARGET = STK(arg3); continue; case _OP_DELETE: _GUARD(DeleteSlot(STK(arg1), STK(arg2), TARGET)); continue; case _OP_SET: if (!Set(STK(arg1), STK(arg2), STK(arg3),true)) { Raise_IdxError(STK(arg2)); SQ_THROW(); } if (arg0 != arg3) TARGET = STK(arg3); continue; case _OP_GET: if (!Get(STK(arg1), STK(arg2), temp_reg, false,true)) { Raise_IdxError(STK(arg2)); SQ_THROW(); } TARGET = temp_reg; continue; case _OP_EQ:{ bool res; if(!IsEqual(STK(arg2),COND_LITERAL,res)) { SQ_THROW(); } TARGET = res?_true_:_false_; }continue; case _OP_NE:{ bool res; if(!IsEqual(STK(arg2),COND_LITERAL,res)) { SQ_THROW(); } TARGET = (!res)?_true_:_false_; } continue; case _OP_ARITH: _GUARD(ARITH_OP( arg3 , temp_reg, STK(arg2), STK(arg1))); TARGET = temp_reg; continue; case _OP_BITW: _GUARD(BW_OP( arg3,TARGET,STK(arg2),STK(arg1))); continue; case _OP_RETURN: if(type((ci)->_generator) == OT_GENERATOR) { _generator((ci)->_generator)->Kill(); } if(Return(arg0, arg1, temp_reg)){ assert(traps==0); outres = temp_reg; return true; } continue; case _OP_LOADNULLS:{ for(SQInt32 n=0; n < arg1; n++) STK(arg0+n) = _null_; }continue; case _OP_LOADROOTTABLE: TARGET = _roottable; continue; case _OP_LOADBOOL: TARGET = arg1?_true_:_false_; continue; case _OP_DMOVE: STK(arg0) = STK(arg1); STK(arg2) = STK(arg3); continue; case _OP_JMP: ci->_ip += (sarg1); continue; case _OP_JNZ: if(!IsFalse(STK(arg0))) ci->_ip+=(sarg1); continue; case _OP_JZ: if(IsFalse(STK(arg0))) ci->_ip+=(sarg1); continue; case _OP_LOADFREEVAR: TARGET = _closure(ci->_closure)->_outervalues[arg1]; continue; case _OP_VARGC: TARGET = SQInteger(ci->_vargs.size); continue; case _OP_GETVARGV: if(!GETVARGV_OP(TARGET,STK(arg1),ci)) { SQ_THROW(); } continue; case _OP_NEWTABLE: TARGET = SQTable::Create(_ss(this), arg1); continue; case _OP_NEWARRAY: TARGET = SQArray::Create(_ss(this), 0); _array(TARGET)->Reserve(arg1); continue; case _OP_APPENDARRAY: _array(STK(arg0))->Append(COND_LITERAL); continue; case _OP_GETPARENT: switch(type(STK(arg1))) { case OT_TABLE: TARGET = _table(STK(arg1))->_delegate?SQObjectPtr(_table(STK(arg1))->_delegate):_null_; continue; case OT_CLASS: TARGET = _class(STK(arg1))->_base?_class(STK(arg1))->_base:_null_; continue; } Raise_Error(_SC("the %s type doesn't have a parent slot"), GetTypeName(STK(arg1))); SQ_THROW(); continue; case _OP_COMPARITH: _GUARD(DerefInc(arg3, TARGET, STK((((SQUnsignedInteger)arg1&0xFFFF0000)>>16)), STK(arg2), STK(arg1&0x0000FFFF), false)); continue; case _OP_COMPARITHL: _GUARD(LOCAL_INC(arg3, TARGET, STK(arg1), STK(arg2))); continue; case _OP_INC: {SQObjectPtr o(sarg3); _GUARD(DerefInc('+',TARGET, STK(arg1), STK(arg2), o, false));} continue; case _OP_INCL: {SQObjectPtr o(sarg3); _GUARD(LOCAL_INC('+',TARGET, STK(arg1), o));} continue; case _OP_PINC: {SQObjectPtr o(sarg3); _GUARD(DerefInc('+',TARGET, STK(arg1), STK(arg2), o, true));} continue; case _OP_PINCL: {SQObjectPtr o(sarg3); _GUARD(PLOCAL_INC('+',TARGET, STK(arg1), o));} continue; case _OP_CMP: _GUARD(CMP_OP((CmpOP)arg3,STK(arg2),STK(arg1),TARGET)) continue; case _OP_EXISTS: TARGET = Get(STK(arg1), STK(arg2), temp_reg, true,false)?_true_:_false_;continue; case _OP_INSTANCEOF: if(type(STK(arg1)) != OT_CLASS || type(STK(arg2)) != OT_INSTANCE) {Raise_Error(_SC("cannot apply instanceof between a %s and a %s"),GetTypeName(STK(arg1)),GetTypeName(STK(arg2))); SQ_THROW();} TARGET = _instance(STK(arg2))->InstanceOf(_class(STK(arg1)))?_true_:_false_; continue; case _OP_AND: if(IsFalse(STK(arg2))) { TARGET = STK(arg2); ci->_ip += (sarg1); } continue; case _OP_OR: if(!IsFalse(STK(arg2))) { TARGET = STK(arg2); ci->_ip += (sarg1); } continue; case _OP_NEG: _GUARD(NEG_OP(TARGET,STK(arg1))); continue; case _OP_NOT: TARGET = (IsFalse(STK(arg1))?_true_:_false_); continue; case _OP_BWNOT: if(type(STK(arg1)) == OT_INTEGER) { TARGET = SQInteger(~_integer(STK(arg1))); continue; } Raise_Error(_SC("attempt to perform a bitwise op on a %s"), GetTypeName(STK(arg1))); SQ_THROW(); case _OP_CLOSURE: { SQClosure *c = ci->_closure._unVal.pClosure; SQFunctionProto *fp = c->_function._unVal.pFunctionProto; if(!CLOSURE_OP(TARGET,fp->_functions[arg1]._unVal.pFunctionProto)) { SQ_THROW(); } continue; } case _OP_YIELD:{ if(type(ci->_generator) == OT_GENERATOR) { if(sarg1 != MAX_FUNC_STACKSIZE) temp_reg = STK(arg1); _GUARD(_generator(ci->_generator)->Yield(this)); traps -= ci->_etraps; if(sarg1 != MAX_FUNC_STACKSIZE) STK(arg1) = temp_reg; } else { Raise_Error(_SC("trying to yield a '%s',only genenerator can be yielded"), GetTypeName(ci->_generator)); SQ_THROW();} if(Return(arg0, arg1, temp_reg)){ assert(traps==0); outres = temp_reg; return true; } } continue; case _OP_RESUME: if(type(STK(arg1)) != OT_GENERATOR){ Raise_Error(_SC("trying to resume a '%s',only genenerator can be resumed"), GetTypeName(STK(arg1))); SQ_THROW();} _GUARD(_generator(STK(arg1))->Resume(this, arg0)); traps += ci->_etraps; continue; case _OP_FOREACH:{ bool finished; _GUARD(FOREACH_OP(STK(arg0),STK(arg2),STK(arg2+1),STK(arg2+2),arg2,finished)); if(finished) ci->_ip += sarg1; } continue; case _OP_DELEGATE: _GUARD(DELEGATE_OP(TARGET,STK(arg1),STK(arg2))); continue; case _OP_CLONE: if(!Clone(STK(arg1), TARGET)) { Raise_Error(_SC("cloning a %s"), GetTypeName(STK(arg1))); SQ_THROW();} continue; case _OP_TYPEOF: TypeOf(STK(arg1), TARGET); continue; case _OP_PUSHTRAP: _etraps.push_back(SQExceptionTrap(_top,_stackbase, &ci->_iv->_vals[(ci->_ip-ci->_iv->_vals)+arg1], arg0)); traps++; ci->_etraps++; continue; case _OP_POPTRAP:{ for(SQInteger i=0; i_etraps--; }} continue; case _OP_THROW: Raise_Error(TARGET); SQ_THROW(); continue; case _OP_CLASS: _GUARD(CLASS_OP(TARGET,arg1,arg2)); continue; case _OP_NEWSLOTA: _GUARD(NewSlot(STK(arg1), STK(arg2), STK(arg3))); _class(STK(arg1))->SetAttributes(STK(arg2),STK(arg2-1)); if(arg0 != arg3) TARGET = STK(arg3); continue; } } } exception_trap: { SQObjectPtr currerror = _lasterror; // dumpstack(_stackbase); SQInteger n = 0; SQInteger last_top = _top; if(ci) { if(traps) { do { if(ci->_etraps > 0) { SQExceptionTrap &et = _etraps.top(); ci->_ip = et._ip; _top = et._stacksize; _stackbase = et._stackbase; _stack[_stackbase+et._extarget] = currerror; _etraps.pop_back(); traps--; ci->_etraps--; while(last_top >= _top) _stack[last_top--].Null(); goto exception_restore; } //if is a native closure if(type(ci->_closure) != OT_CLOSURE && n) break; if(type(ci->_generator) == OT_GENERATOR) _generator(ci->_generator)->Kill(); PopVarArgs(ci->_vargs); POP_CALLINFO(this); n++; }while(_callsstack.size()); } //call the hook CallErrorHandler(currerror); //remove call stack until a C function is found or the cstack is empty if(ci) do{ SQBool exitafterthisone = ci->_root; if(type(ci->_generator) == OT_GENERATOR) _generator(ci->_generator)->Kill(); _stackbase -= ci->_prevstkbase; _top = _stackbase + ci->_prevtop; PopVarArgs(ci->_vargs); POP_CALLINFO(this); if( (ci && type(ci->_closure) != OT_CLOSURE) || exitafterthisone) break; }while(_callsstack.size()); while(last_top >= _top) _stack[last_top--].Null(); } _lasterror = currerror; return false; } assert(0); } bool SQVM::CreateClassInstance(SQClass *theclass, SQInteger nargs, SQInteger stackbase, SQObjectPtr &retval) { SQObjectPtr constr; SQObjectPtr inst = theclass->CreateInstance(); _stack[stackbase] = inst; if(theclass->Get(_ss(this)->_constructoridx,constr)) { if(!Call(constr,nargs,stackbase,constr)) return false; } retval = inst; return true; } void SQVM::CallErrorHandler(SQObjectPtr &error) { if(type(_errorhandler) != OT_NULL) { SQObjectPtr out; Push(_roottable); Push(error); Call(_errorhandler, 2, _top-2, out); Pop(2); } } void SQVM::CallDebugHook(SQInteger type,SQInteger forcedline) { SQObjectPtr temp_reg; SQInteger nparams=5; SQFunctionProto *func=_funcproto(_closure(ci->_closure)->_function); Push(_roottable); Push(type); Push(func->_sourcename); Push(forcedline?forcedline:func->GetLine(ci->_ip)); Push(func->_name); Call(_debughook,nparams,_top-nparams,temp_reg); Pop(nparams); } bool SQVM::CallNative(SQNativeClosure *nclosure,SQInteger nargs,SQInteger stackbase,bool tailcall,SQObjectPtr &retval,bool &suspend) { if (_nnativecalls + 1 > MAX_NATIVE_CALLS) { Raise_Error(_SC("Native stack overflow")); return false; } SQInteger nparamscheck = nclosure->_nparamscheck; if(((nparamscheck > 0) && (nparamscheck != nargs)) || ((nparamscheck < 0) && (nargs < (-nparamscheck)))) { Raise_Error(_SC("wrong number of parameters")); return false; } SQInteger tcs; if(tcs = nclosure->_typecheck.size()) { for(SQInteger i = 0; i < nargs && i < tcs; i++) if((nclosure->_typecheck[i] != -1) && !(type(_stack[stackbase+i]) & nclosure->_typecheck[i])) { Raise_ParamTypeError(i,nclosure->_typecheck[i],type(_stack[stackbase+i])); return false; } } _nnativecalls++; if ((_top + MIN_STACK_OVERHEAD) > (SQInteger)_stack.size()) { _stack.resize(_stack.size() + (MIN_STACK_OVERHEAD<<1)); } SQInteger oldtop = _top; SQInteger oldstackbase = _stackbase; _top = stackbase + nargs; PUSH_CALLINFO(this, CallInfo()); ci->_etraps = 0; ci->_closure._unVal.pNativeClosure = nclosure; ci->_closure._type = OT_NATIVECLOSURE; ci->_prevstkbase = stackbase - _stackbase; ci->_ncalls = 1; _stackbase = stackbase; //push free variables SQInteger outers = nclosure->_outervalues.size(); for (SQInteger i = 0; i < outers; i++) { Push(nclosure->_outervalues[i]); } ci->_prevtop = (oldtop - oldstackbase); SQInteger ret = (nclosure->_function)(this); _nnativecalls--; suspend = false; if( ret == SQ_SUSPEND_FLAG) suspend = true; else if (ret < 0) { _stackbase = oldstackbase; _top = oldtop; POP_CALLINFO(this); Raise_Error(_lasterror); return false; } if (ret != 0){ retval = TOP(); } else { retval = _null_; } _stackbase = oldstackbase; _top = oldtop; POP_CALLINFO(this); return true; } bool SQVM::Get(const SQObjectPtr &self,const SQObjectPtr &key,SQObjectPtr &dest,bool raw, bool fetchroot) { switch(type(self)){ case OT_TABLE: if(_table(self)->Get(key,dest))return true; break; case OT_ARRAY: if(sq_isnumeric(key)){ return _array(self)->Get(tointeger(key),dest); } break; case OT_INSTANCE: if(_instance(self)->Get(key,dest)) return true; break; } if(FallBackGet(self,key,dest,raw)) return true; if(fetchroot) { if(_rawval(STK(0)) == _rawval(self) && type(STK(0)) == type(self)) { return _table(_roottable)->Get(key,dest); } } return false; } bool SQVM::FallBackGet(const SQObjectPtr &self,const SQObjectPtr &key,SQObjectPtr &dest,bool raw) { switch(type(self)){ case OT_CLASS: return _class(self)->Get(key,dest); break; case OT_TABLE: case OT_USERDATA: //delegation if(_delegable(self)->_delegate) { if(Get(SQObjectPtr(_delegable(self)->_delegate),key,dest,raw,false)) return true; if(raw)return false; Push(self);Push(key); if(CallMetaMethod(_delegable(self),MT_GET,2,dest)) return true; } if(type(self) == OT_TABLE) { if(raw) return false; return _table_ddel->Get(key,dest); } return false; break; case OT_ARRAY: if(raw)return false; return _array_ddel->Get(key,dest); case OT_STRING: if(sq_isnumeric(key)){ SQInteger n=tointeger(key); if(abs(n)<_string(self)->_len){ if(n<0)n=_string(self)->_len-n; dest=SQInteger(_stringval(self)[n]); return true; } return false; } else { if(raw)return false; return _string_ddel->Get(key,dest); } break; case OT_INSTANCE: if(raw)return false; Push(self);Push(key); if(!CallMetaMethod(_delegable(self),MT_GET,2,dest)) { return _instance_ddel->Get(key,dest); } return true; case OT_INTEGER:case OT_FLOAT:case OT_BOOL: if(raw)return false; return _number_ddel->Get(key,dest); case OT_GENERATOR: if(raw)return false; return _generator_ddel->Get(key,dest); case OT_CLOSURE: case OT_NATIVECLOSURE: if(raw)return false; return _closure_ddel->Get(key,dest); case OT_THREAD: if(raw)return false; return _thread_ddel->Get(key,dest); case OT_WEAKREF: if(raw)return false; return _weakref_ddel->Get(key,dest); default:return false; } return false; } bool SQVM::Set(const SQObjectPtr &self,const SQObjectPtr &key,const SQObjectPtr &val,bool fetchroot) { switch(type(self)){ case OT_TABLE: if(_table(self)->Set(key,val)) return true; if(_table(self)->_delegate) { if(Set(_table(self)->_delegate,key,val,false)) { return true; } } //keeps going case OT_USERDATA: if(_delegable(self)->_delegate) { SQObjectPtr t; Push(self);Push(key);Push(val); if(CallMetaMethod(_delegable(self),MT_SET,3,t)) return true; } break; case OT_INSTANCE:{ if(_instance(self)->Set(key,val)) return true; SQObjectPtr t; Push(self);Push(key);Push(val); if(CallMetaMethod(_delegable(self),MT_SET,3,t)) return true; } break; case OT_ARRAY: if(!sq_isnumeric(key)) {Raise_Error(_SC("indexing %s with %s"),GetTypeName(self),GetTypeName(key)); return false; } return _array(self)->Set(tointeger(key),val); default: Raise_Error(_SC("trying to set '%s'"),GetTypeName(self)); return false; } if(fetchroot) { if(_rawval(STK(0)) == _rawval(self) && type(STK(0)) == type(self)) { return _table(_roottable)->Set(key,val); } } return false; } bool SQVM::Clone(const SQObjectPtr &self,SQObjectPtr &target) { SQObjectPtr temp_reg; switch(type(self)){ case OT_TABLE: target = _table(self)->Clone(); goto cloned_mt; case OT_INSTANCE: target = _instance(self)->Clone(_ss(this)); cloned_mt: if(_delegable(target)->_delegate){ Push(target); Push(self); CallMetaMethod(_delegable(target),MT_CLONED,2,temp_reg); } return true; case OT_ARRAY: target=_array(self)->Clone(); return true; default: return false; } } bool SQVM::NewSlot(const SQObjectPtr &self,const SQObjectPtr &key,const SQObjectPtr &val) { if(type(key) == OT_NULL) { Raise_Error(_SC("null cannot be used as index")); return false; } switch(type(self)) { case OT_TABLE: { bool rawcall = true; if(_table(self)->_delegate) { SQObjectPtr res; if(!_table(self)->Get(key,res)) { Push(self);Push(key);Push(val); rawcall = !CallMetaMethod(_table(self),MT_NEWSLOT,3,res); } } if(rawcall) _table(self)->NewSlot(key,val); //cannot fail break;} case OT_CLASS: if(!_class(self)->NewSlot(key,val)) { if(_class(self)->_locked) { Raise_Error(_SC("trying to modify a class that has already been instantiated")); return false; } else { SQObjectPtr oval = PrintObjVal(key); Raise_Error(_SC("the property '%s' already exists"),_stringval(oval)); return false; } } break; default: Raise_Error(_SC("indexing %s with %s"),GetTypeName(self),GetTypeName(key)); return false; break; } return true; } bool SQVM::DeleteSlot(const SQObjectPtr &self,const SQObjectPtr &key,SQObjectPtr &res) { switch(type(self)) { case OT_TABLE: case OT_INSTANCE: case OT_USERDATA: { SQObjectPtr t; bool handled = false; if(_delegable(self)->_delegate) { Push(self);Push(key); handled = CallMetaMethod(_delegable(self),MT_DELSLOT,2,t); } if(!handled) { if(type(self) == OT_TABLE) { if(_table(self)->Get(key,t)) { _table(self)->Remove(key); } else { Raise_IdxError((SQObject &)key); return false; } } else { Raise_Error(_SC("cannot delete a slot from %s"),GetTypeName(self)); return false; } } res = t; } break; default: Raise_Error(_SC("attempt to delete a slot from a %s"),GetTypeName(self)); return false; } return true; } bool SQVM::Call(SQObjectPtr &closure,SQInteger nparams,SQInteger stackbase,SQObjectPtr &outres) { #ifdef _DEBUG SQInteger prevstackbase = _stackbase; #endif switch(type(closure)) { case OT_CLOSURE: return Execute(closure, _top - nparams, nparams, stackbase,outres); break; case OT_NATIVECLOSURE:{ bool suspend; return CallNative(_nativeclosure(closure), nparams, stackbase, false, outres,suspend); } break; case OT_CLASS: return CreateClassInstance(_class(closure),nparams,stackbase,outres); break; default: return false; } #ifdef _DEBUG if(!_suspended) { assert(_stackbase == prevstackbase); } #endif return true; } bool SQVM::CallMetaMethod(SQDelegable *del,SQMetaMethod mm,SQInteger nparams,SQObjectPtr &outres) { SQObjectPtr closure; if(del->GetMetaMethod(mm, closure)) { if(Call(closure, nparams, _top - nparams, outres)) { Pop(nparams); return true; } } Pop(nparams); return false; } void SQVM::Remove(SQInteger n) { n = (n >= 0)?n + _stackbase - 1:_top + n; for(SQInteger i = n; i < _top; i++){ _stack[i] = _stack[i+1]; } _stack[_top] = _null_; _top--; } #ifdef _DEBUG_DUMP void SQVM::dumpstack(SQInteger stackbase,bool dumpall) { SQInteger size=dumpall?_stack.size():_top; SQInteger n=0; scprintf(_SC("\n>>>>stack dump<<<<\n")); CallInfo &ci=_callsstack.back(); scprintf(_SC("IP: %d\n"),ci._ip); scprintf(_SC("prev stack base: %d\n"),ci._prevstkbase); scprintf(_SC("prev top: %d\n"),ci._prevtop); for(SQInteger i=0;i"));else scprintf(_SC(" ")); scprintf(_SC("[%d]:"),n); switch(type(obj)){ case OT_FLOAT: scprintf(_SC("FLOAT %.3f"),_float(obj));break; case OT_INTEGER: scprintf(_SC("INTEGER %d"),_integer(obj));break; case OT_BOOL: scprintf(_SC("BOOL %s"),_integer(obj)?"true":"false");break; case OT_STRING: scprintf(_SC("STRING %s"),_stringval(obj));break; case OT_NULL: scprintf(_SC("NULL")); break; case OT_TABLE: scprintf(_SC("TABLE %p[%p]"),_table(obj),_table(obj)->_delegate);break; case OT_ARRAY: scprintf(_SC("ARRAY %p"),_array(obj));break; case OT_CLOSURE: scprintf(_SC("CLOSURE [%p]"),_closure(obj));break; case OT_NATIVECLOSURE: scprintf(_SC("NATIVECLOSURE"));break; case OT_USERDATA: scprintf(_SC("USERDATA %p[%p]"),_userdataval(obj),_userdata(obj)->_delegate);break; case OT_GENERATOR: scprintf(_SC("GENERATOR"));break; case OT_THREAD: scprintf(_SC("THREAD [%p]"),_thread(obj));break; case OT_USERPOINTER: scprintf(_SC("USERPOINTER %p"),_userpointer(obj));break; case OT_CLASS: scprintf(_SC("CLASS %p"),_class(obj));break; case OT_INSTANCE: scprintf(_SC("INSTANCE %p"),_instance(obj));break; case OT_WEAKREF: scprintf(_SC("WEAKERF %p"),_weakref(obj));break; default: assert(0); break; }; scprintf(_SC("\n")); ++n; } } #endif