2 see copyright notice in squirrel.h
4 #include "sqpcheader.h"
8 #include "sqfuncproto.h"
10 #include "sqclosure.h"
13 #include "squserdata.h"
17 #define TOP() (_stack._vals[_top-1])
19 bool SQVM::BW_OP(SQUnsignedInteger op,SQObjectPtr &trg,const SQObjectPtr &o1,const SQObjectPtr &o2)
22 SQInteger i1 = _integer(o1), i2 = _integer(o2);
23 if((type(o1)==OT_INTEGER) && (type(o2)==OT_INTEGER))
26 case BW_AND: res = i1 & i2; break;
27 case BW_OR: res = i1 | i2; break;
28 case BW_XOR: res = i1 ^ i2; break;
29 case BW_SHIFTL: res = i1 << i2; break;
30 case BW_SHIFTR: res = i1 >> i2; break;
31 case BW_USHIFTR:res = (SQInteger)(*((SQUnsignedInteger*)&i1) >> i2); break;
32 default: { Raise_Error(_SC("internal vm error bitwise op failed")); return false; }
35 else { Raise_Error(_SC("bitwise op between '%s' and '%s'"),GetTypeName(o1),GetTypeName(o2)); return false;}
40 bool SQVM::ARITH_OP(SQUnsignedInteger op,SQObjectPtr &trg,const SQObjectPtr &o1,const SQObjectPtr &o2)
42 if(sq_isnumeric(o1) && sq_isnumeric(o2)) {
43 if((type(o1)==OT_INTEGER) && (type(o2)==OT_INTEGER)) {
44 SQInteger res, i1 = _integer(o1), i2 = _integer(o2);
46 case '+': res = i1 + i2; break;
47 case '-': res = i1 - i2; break;
48 case '/': if(i2 == 0) { Raise_Error(_SC("division by zero")); return false; }
51 case '*': res = i1 * i2; break;
52 case '%': res = i1 % i2; break;
53 default: res = 0xDEADBEEF;
57 SQFloat res, f1 = tofloat(o1), f2 = tofloat(o2);
59 case '+': res = f1 + f2; break;
60 case '-': res = f1 - f2; break;
61 case '/': res = f1 / f2; break;
62 case '*': res = f1 * f2; break;
63 case '%': res = SQFloat(fmod((double)f1,(double)f2)); break;
69 if(op == '+' && (type(o1) == OT_STRING || type(o2) == OT_STRING)){
70 if(!StringCat(o1, o2, trg)) return false;
72 else if(!ArithMetaMethod(op,o1,o2,trg)) {
73 Raise_Error(_SC("arith op %c on between '%s' and '%s'"),op,GetTypeName(o1),GetTypeName(o2)); return false;
79 SQVM::SQVM(SQSharedState *ss)
84 _suspended_root = SQFalse;
89 _errorhandler = _null_;
92 INIT_CHAIN();ADD_TO_CHAIN(&_ss(this)->_gc_chain,this);
99 _errorhandler = _null_;
102 SQInteger size=_stack.size();
103 for(SQInteger i=0;i<size;i++)
110 sq_free(_callsstack,_alloccallsstacksize*sizeof(CallInfo));
111 REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
114 bool SQVM::ArithMetaMethod(SQInteger op,const SQObjectPtr &o1,const SQObjectPtr &o2,SQObjectPtr &dest)
118 case _SC('+'): mm=MT_ADD; break;
119 case _SC('-'): mm=MT_SUB; break;
120 case _SC('/'): mm=MT_DIV; break;
121 case _SC('*'): mm=MT_MUL; break;
122 case _SC('%'): mm=MT_MODULO; break;
123 default: mm = MT_ADD; assert(0); break; //shutup compiler
125 if(is_delegable(o1) && _delegable(o1)->_delegate) {
127 return CallMetaMethod(_delegable(o1),mm,2,dest);
132 bool SQVM::NEG_OP(SQObjectPtr &trg,const SQObjectPtr &o)
145 if(_delegable(o)->_delegate) {
147 if(CallMetaMethod(_delegable(o), MT_UNM, 1, temp_reg)) {
152 default:break; //shutup compiler
154 Raise_Error(_SC("attempt to negate a %s"), GetTypeName(o));
158 #define _RET_SUCCEED(exp) { result = (exp); return true; }
159 bool SQVM::ObjCmp(const SQObjectPtr &o1,const SQObjectPtr &o2,SQInteger &result)
161 if(type(o1)==type(o2)){
162 if(_userpointer(o1)==_userpointer(o2))_RET_SUCCEED(0);
166 _RET_SUCCEED(scstrcmp(_stringval(o1),_stringval(o2)));
168 _RET_SUCCEED(_integer(o1)-_integer(o2));
170 _RET_SUCCEED((_float(o1)<_float(o2))?-1:1);
174 if(_delegable(o1)->_delegate) {
176 if(CallMetaMethod(_delegable(o1),MT_CMP,2,res)) break;
178 //continues through (no break needed)
180 _RET_SUCCEED( _userpointer(o1) < _userpointer(o2)?-1:1 );
182 if(type(res)!=OT_INTEGER) { Raise_CompareError(o1,o2); return false; }
183 _RET_SUCCEED(_integer(res));
187 if(sq_isnumeric(o1) && sq_isnumeric(o2)){
188 if((type(o1)==OT_INTEGER) && (type(o2)==OT_FLOAT)) {
189 if( _integer(o1)==_float(o2) ) { _RET_SUCCEED(0); }
190 else if( _integer(o1)<_float(o2) ) { _RET_SUCCEED(-1); }
194 if( _float(o1)==_integer(o2) ) { _RET_SUCCEED(0); }
195 else if( _float(o1)<_integer(o2) ) { _RET_SUCCEED(-1); }
199 else if(type(o1)==OT_NULL) {_RET_SUCCEED(-1);}
200 else if(type(o2)==OT_NULL) {_RET_SUCCEED(1);}
201 else { Raise_CompareError(o1,o2); return false; }
205 _RET_SUCCEED(0); //cannot happen
208 bool SQVM::CMP_OP(CmpOP op, const SQObjectPtr &o1,const SQObjectPtr &o2,SQObjectPtr &res)
211 if(ObjCmp(o1,o2,r)) {
213 case CMP_G: res = (r > 0)?_true_:_false_; return true;
214 case CMP_GE: res = (r >= 0)?_true_:_false_; return true;
215 case CMP_L: res = (r < 0)?_true_:_false_; return true;
216 case CMP_LE: res = (r <= 0)?_true_:_false_; return true;
224 void SQVM::ToString(const SQObjectPtr &o,SQObjectPtr &res)
231 scsprintf(_sp(rsl(NUMBER_MAX_CHAR+1)),_SC("%g"),_float(o));
234 scsprintf(_sp(rsl(NUMBER_MAX_CHAR+1)),_SC("%d"),_integer(o));
237 scsprintf(_sp(rsl(6)),_integer(o)?_SC("true"):_SC("false"));
242 if(_delegable(o)->_delegate) {
244 if(CallMetaMethod(_delegable(o),MT_TOSTRING,1,res)) {
245 if(type(res) == OT_STRING)
247 //else keeps going to the default
251 scsprintf(_sp(rsl(sizeof(void*)+20)),_SC("(%s : 0x%p)"),GetTypeName(o),(void*)_rawval(o));
253 res = SQString::Create(_ss(this),_spval);
257 bool SQVM::StringCat(const SQObjectPtr &str,const SQObjectPtr &obj,SQObjectPtr &dest)
262 SQInteger l = _string(a)->_len , ol = _string(b)->_len;
263 SQChar *s = _sp(rsl(l + ol + 1));
264 memcpy(s, _stringval(a), rsl(l));
265 memcpy(s + l, _stringval(b), rsl(ol));
266 dest = SQString::Create(_ss(this), _spval, l + ol);
270 const SQChar *IdType2Name(SQObjectType type)
272 switch(_RAW_TYPE(type))
274 case _RT_NULL:return _SC("null");
275 case _RT_INTEGER:return _SC("integer");
276 case _RT_FLOAT:return _SC("float");
277 case _RT_BOOL:return _SC("bool");
278 case _RT_STRING:return _SC("string");
279 case _RT_TABLE:return _SC("table");
280 case _RT_ARRAY:return _SC("array");
281 case _RT_GENERATOR:return _SC("generator");
283 case _RT_NATIVECLOSURE:
284 return _SC("function");
286 case _RT_USERPOINTER:
287 return _SC("userdata");
288 case _RT_THREAD: return _SC("thread");
289 case _RT_FUNCPROTO: return _SC("function");
290 case _RT_CLASS: return _SC("class");
291 case _RT_INSTANCE: return _SC("instance");
292 case _RT_WEAKREF: return _SC("weakref");
298 const SQChar *GetTypeName(const SQObjectPtr &obj1)
300 return IdType2Name(type(obj1));
303 void SQVM::TypeOf(const SQObjectPtr &obj1,SQObjectPtr &dest)
305 if(is_delegable(obj1) && _delegable(obj1)->_delegate) {
307 if(CallMetaMethod(_delegable(obj1),MT_TYPEOF,1,dest))
310 dest = SQString::Create(_ss(this),GetTypeName(obj1));
313 bool SQVM::Init(SQVM *friendvm, SQInteger stacksize)
315 _stack.resize(stacksize);
316 //_callsstack.reserve(4);
317 _alloccallsstacksize = 4;
319 _callsstack = (CallInfo*)sq_malloc(_alloccallsstacksize*sizeof(CallInfo));
323 _roottable = SQTable::Create(_ss(this), 0);
325 _roottable = friendvm->_roottable;
326 _errorhandler = friendvm->_errorhandler;
327 _debughook = friendvm->_debughook;
330 sq_base_register(this);
334 extern SQInstructionDesc g_InstrDesc[];
336 bool SQVM::StartCall(SQClosure *closure,SQInteger target,SQInteger nargs,SQInteger stackbase,bool tailcall)
338 SQFunctionProto *func = _funcproto(closure->_function);
340 const SQInteger paramssize = func->_nparameters;
341 const SQInteger newtop = stackbase + func->_stacksize;
344 if (paramssize != nargs) {
347 if (nargs < paramssize) {
348 Raise_Error(_SC("wrong number of parameters"));
351 for(SQInteger n = 0; n < nargs - paramssize; n++) {
352 _vargsstack.push_back(_stack._vals[stackbase+paramssize+n]);
353 _stack._vals[stackbase+paramssize+n] = _null_;
357 Raise_Error(_SC("wrong number of parameters"));
362 if(type(closure->_env) == OT_WEAKREF) {
363 _stack._vals[stackbase] = _weakref(closure->_env)->_obj;
369 lc._prevstkbase = (SQInt32) ( stackbase - _stackbase );
370 lc._target = (SQInt32) target;
371 lc._prevtop = (SQInt32) (_top - _stackbase);
374 PUSH_CALLINFO(this, lc);
379 ci->_vargs.size = (SQInt32)(nargs - paramssize);
380 ci->_vargs.base = (SQInt32) (_vargsstack.size()-(ci->_vargs.size));
381 ci->_closure._unVal.pClosure = closure;
382 ci->_closure._type = OT_CLOSURE;
383 ci->_literals = func->_literals;
384 ci->_ip = func->_instructions;
385 //grows the stack if needed
386 if (((SQUnsignedInteger)newtop + (func->_stacksize<<1)) > _stack.size()) {
387 _stack.resize(_stack.size() + (func->_stacksize<<1));
391 _stackbase = stackbase;
395 bool SQVM::Return(SQInteger _arg0, SQInteger _arg1, SQObjectPtr &retval)
397 if (type(_debughook) != OT_NULL && _rawval(_debughook) != _rawval(ci->_closure))
398 for(SQInteger i=0;i<ci->_ncalls;i++)
399 CallDebugHook(_SC('r'));
401 SQBool broot = ci->_root;
402 SQInteger last_top = _top;
403 SQInteger target = ci->_target;
404 SQInteger oldstackbase = _stackbase;
405 _stackbase -= ci->_prevstkbase;
406 _top = _stackbase + ci->_prevtop;
407 if(ci->_vargs.size) PopVarArgs(ci->_vargs);
410 if (_arg0 != MAX_FUNC_STACKSIZE) retval = _stack._vals[oldstackbase+_arg1];
411 else retval = _null_;
414 if(target != -1) { //-1 is when a class contructor ret value has to be ignored
415 if (_arg0 != MAX_FUNC_STACKSIZE)
416 STK(target) = _stack._vals[oldstackbase+_arg1];
418 STK(target) = _null_;
422 while (last_top >= _top) _stack._vals[last_top--].Null();
423 assert(oldstackbase >= _stackbase);
424 return broot?true:false;
427 #define _RET_ON_FAIL(exp) { if(!exp) return false; }
429 bool SQVM::LOCAL_INC(SQInteger op,SQObjectPtr &target, SQObjectPtr &a, SQObjectPtr &incr)
431 _RET_ON_FAIL(ARITH_OP( op , target, a, incr));
436 bool SQVM::PLOCAL_INC(SQInteger op,SQObjectPtr &target, SQObjectPtr &a, SQObjectPtr &incr)
439 _RET_ON_FAIL(ARITH_OP( op , trg, a, incr));
445 bool SQVM::DerefInc(SQInteger op,SQObjectPtr &target, SQObjectPtr &self, SQObjectPtr &key, SQObjectPtr &incr, bool postfix)
447 SQObjectPtr tmp, tself = self, tkey = key;
448 if (!Get(tself, tkey, tmp, false, true)) { Raise_IdxError(tkey); return false; }
449 _RET_ON_FAIL(ARITH_OP( op , target, tmp, incr))
450 Set(tself, tkey, target,true);
451 if (postfix) target = tmp;
455 #define arg0 (_i_._arg0)
456 #define arg1 (_i_._arg1)
457 #define sarg1 (*((SQInt32 *)&_i_._arg1))
458 #define arg2 (_i_._arg2)
459 #define arg3 (_i_._arg3)
460 #define sarg3 ((SQInteger)*((signed char *)&_i_._arg3))
462 SQRESULT SQVM::Suspend()
465 return sq_throwerror(this, _SC("cannot suspend an already suspended vm"));
466 if (_nnativecalls!=2)
467 return sq_throwerror(this, _SC("cannot suspend through native calls/metamethods"));
468 return SQ_SUSPEND_FLAG;
471 void SQVM::PopVarArgs(VarArgs &vargs)
473 for(SQInteger n = 0; n< vargs.size; n++)
474 _vargsstack.pop_back();
477 #define _FINISH(howmuchtojump) {jump = howmuchtojump; return true; }
478 bool SQVM::FOREACH_OP(SQObjectPtr &o1,SQObjectPtr &o2,SQObjectPtr
479 &o3,SQObjectPtr &o4,SQInteger arg_2,int exitpos,int &jump)
484 if((nrefidx = _table(o1)->Next(false,o4, o2, o3)) == -1) _FINISH(exitpos);
485 o4 = (SQInteger)nrefidx; _FINISH(1);
487 if((nrefidx = _array(o1)->Next(o4, o2, o3)) == -1) _FINISH(exitpos);
488 o4 = (SQInteger) nrefidx; _FINISH(1);
490 if((nrefidx = _string(o1)->Next(o4, o2, o3)) == -1)_FINISH(exitpos);
491 o4 = (SQInteger)nrefidx; _FINISH(1);
493 if((nrefidx = _class(o1)->Next(o4, o2, o3)) == -1)_FINISH(exitpos);
494 o4 = (SQInteger)nrefidx; _FINISH(1);
497 if(_delegable(o1)->_delegate) {
501 if(CallMetaMethod(_delegable(o1), MT_NEXTI, 2, itr)){
503 if(type(itr) == OT_NULL) _FINISH(exitpos);
504 if(!Get(o1, itr, o3, false,false)) {
505 Raise_Error(_SC("_nexti returned an invalid idx"));
510 Raise_Error(_SC("_nexti failed"));
515 if(_generator(o1)->_state == SQGenerator::eDead) _FINISH(exitpos);
516 if(_generator(o1)->_state == SQGenerator::eSuspended) {
518 if(type(o4) == OT_INTEGER) {
519 idx = _integer(o4) + 1;
523 _generator(o1)->Resume(this, arg_2+1);
527 Raise_Error(_SC("cannot iterate %s"), GetTypeName(o1));
529 return false; //cannot be hit(just to avoid warnings)
532 bool SQVM::DELEGATE_OP(SQObjectPtr &trg,SQObjectPtr &o1,SQObjectPtr &o2)
534 if(type(o1) != OT_TABLE) { Raise_Error(_SC("delegating a '%s'"), GetTypeName(o1)); return false; }
537 if(!_table(o1)->SetDelegate(_table(o2))){
538 Raise_Error(_SC("delegate cycle detected"));
543 _table(o1)->SetDelegate(NULL);
546 Raise_Error(_SC("using '%s' as delegate"), GetTypeName(o2));
553 #define COND_LITERAL (arg3!=0?ci->_literals[arg1]:STK(arg1))
555 #define _GUARD(exp) { if(!exp) { Raise_Error(_lasterror); SQ_THROW();} }
557 #define SQ_THROW() { goto exception_trap; }
559 bool SQVM::CLOSURE_OP(SQObjectPtr &target, SQFunctionProto *func)
562 SQClosure *closure = SQClosure::Create(_ss(this), func);
563 if((nouters = func->_noutervalues)) {
564 closure->_outervalues.reserve(nouters);
565 for(SQInteger i = 0; i<nouters; i++) {
566 SQOuterVar &v = func->_outervalues[i];
569 closure->_outervalues.push_back(_null_);
570 if(!Get(_stack._vals[_stackbase]/*STK(0)*/, v._src, closure->_outervalues.top(), false,true))
571 {Raise_IdxError(v._src); return false; }
574 closure->_outervalues.push_back(_stack._vals[_stackbase+_integer(v._src)]);
577 closure->_outervalues.push_back(_closure(ci->_closure)->_outervalues[_integer(v._src)]);
587 bool SQVM::GETVARGV_OP(SQObjectPtr &target,SQObjectPtr &index,CallInfo *ci)
589 if(ci->_vargs.size == 0) {
590 Raise_Error(_SC("the function doesn't have var args"));
593 if(!sq_isnumeric(index)){
594 Raise_Error(_SC("indexing 'vargv' with %s"),GetTypeName(index));
597 SQInteger idx = tointeger(index);
598 if(idx < 0 || idx >= ci->_vargs.size){ Raise_Error(_SC("vargv index out of range")); return false; }
599 target = _vargsstack[ci->_vargs.base+idx];
603 bool SQVM::CLASS_OP(SQObjectPtr &target,SQInteger baseclass,SQInteger attributes)
605 SQClass *base = NULL;
607 if(baseclass != -1) {
608 if(type(_stack._vals[_stackbase+baseclass]) != OT_CLASS) { Raise_Error(_SC("trying to inherit from a %s"),GetTypeName(_stack._vals[_stackbase+baseclass])); return false; }
609 base = _class(_stack._vals[_stackbase + baseclass]);
611 if(attributes != MAX_FUNC_STACKSIZE) {
612 attrs = _stack._vals[_stackbase+attributes];
614 target = SQClass::Create(_ss(this),base);
615 if(type(_class(target)->_metamethods[MT_INHERITED]) != OT_NULL) {
618 Push(target); Push(attrs);
619 Call(_class(target)->_metamethods[MT_INHERITED],nparams,_top - nparams, ret, false);
622 _class(target)->_attributes = attrs;
628 bool SQVM::IsEqual(SQObjectPtr &o1,SQObjectPtr &o2,bool &res)
630 if(type(o1) == type(o2)) {
631 res = ((_userpointer(o1) == _userpointer(o2)?true:false));
634 if(sq_isnumeric(o1) && sq_isnumeric(o2)) {
636 if(!ObjCmp(o1, o2,cmpres)) return false;
646 bool SQVM::IsFalse(SQObjectPtr &o)
648 if((type(o) & SQOBJECT_CANBEFALSE) && ( (type(o) == OT_FLOAT) && (_float(o) == SQFloat(0.0)) )
649 || (_integer(o) == 0) ) { //OT_NULL|OT_INTEGER|OT_BOOL
655 bool SQVM::GETPARENT_OP(SQObjectPtr &o,SQObjectPtr &target)
658 case OT_TABLE: target = _table(o)->_delegate?SQObjectPtr(_table(o)->_delegate):_null_;
660 case OT_CLASS: target = _class(o)->_base?_class(o)->_base:_null_;
663 Raise_Error(_SC("the %s type doesn't have a parent slot"), GetTypeName(o));
669 bool SQVM::Execute(SQObjectPtr &closure, SQInteger target, SQInteger nargs, SQInteger stackbase,SQObjectPtr &outres, SQBool raiseerror,ExecutionType et)
671 if ((_nnativecalls + 1) > MAX_NATIVE_CALLS) { Raise_Error(_SC("Native stack overflow")); return false; }
673 AutoDec ad(&_nnativecalls);
675 //temp_reg vars for OP_CALL
677 SQInteger ct_stackbase;
682 if(!StartCall(_closure(closure), _top - nargs, nargs, stackbase, false)) {
683 //call the handler if there are no calls in the stack, if not relies on the previous node
684 if(ci == NULL) CallErrorHandler(_lasterror);
689 case ET_RESUME_GENERATOR: _generator(closure)->Resume(this, target); ci->_root = SQTrue; traps += ci->_etraps; break;
691 traps = _suspended_traps;
692 ci->_root = _suspended_root;
693 ci->_vargs = _suspend_varargs;
694 _suspended = SQFalse;
703 const SQInstruction &_i_ = *ci->_ip++;
704 //dumpstack(_stackbase);
705 //scprintf("\n[%d] %s %d %d %d %d\n",ci->_ip-ci->_iv->_vals,g_InstrDesc[_i_.op].name,arg0,arg1,arg2,arg3);
709 if(type(_debughook) != OT_NULL && _rawval(_debughook) != _rawval(ci->_closure))
710 CallDebugHook(_SC('l'),arg1);
712 case _OP_LOAD: TARGET = ci->_literals[arg1]; continue;
713 case _OP_LOADINT: TARGET = (SQInteger)arg1; continue;
714 case _OP_LOADFLOAT: TARGET = *((SQFloat *)&arg1); continue;
715 case _OP_DLOAD: TARGET = ci->_literals[arg1]; STK(arg2) = ci->_literals[arg3];continue;
717 temp_reg = STK(arg1);
718 if (type(temp_reg) == OT_CLOSURE){
720 if(ci->_vargs.size) PopVarArgs(ci->_vargs);
721 for (SQInteger i = 0; i < arg3; i++) STK(i) = STK(arg2 + i);
722 ct_target = ci->_target;
723 ct_stackbase = _stackbase;
729 temp_reg = STK(arg1);
730 ct_stackbase = _stackbase+arg2;
733 SQInteger last_top = _top;
734 switch (type(temp_reg)) {
736 _GUARD(StartCall(_closure(temp_reg), ct_target, arg3, ct_stackbase, ct_tailcall));
737 if (_funcproto(_closure(temp_reg)->_function)->_bgenerator) {
738 SQGenerator *gen = SQGenerator::Create(_ss(this), _closure(temp_reg));
739 _GUARD(gen->Yield(this));
740 Return(1, ct_target, temp_reg);
745 STK(ct_target) = gen;
746 while (last_top >= _top) _stack._vals[last_top--].Null();
749 if (type(_debughook) != OT_NULL && _rawval(_debughook) != _rawval(ci->_closure))
750 CallDebugHook(_SC('c'));
753 case OT_NATIVECLOSURE: {
755 _GUARD(CallNative(_nativeclosure(temp_reg), arg3, ct_stackbase, temp_reg,suspend));
758 _suspended_target = ct_target;
759 _suspended_root = ci->_root;
760 _suspended_traps = traps;
761 _suspend_varargs = ci->_vargs;
765 if(ct_target != -1) { //skip return value for constructors
766 STK(ct_target) = temp_reg;
772 _GUARD(CreateClassInstance(_class(temp_reg),inst,temp_reg));
773 STK(ct_target) = inst;
774 ct_target = -1; //fakes return value target so that is not overwritten by the constructor
775 if(type(temp_reg) != OT_NULL) {
776 _stack._vals[ct_stackbase] = inst;
777 goto common_call; //hard core spaghetti code(reissues the OP_CALL to invoke the constructor)
786 for (SQInteger i = 0; i < arg3; i++) Push(STK(arg2 + i));
787 if (_delegable(temp_reg) && CallMetaMethod(_delegable(temp_reg), MT_CALL, arg3+1, temp_reg)){
788 STK(ct_target) = temp_reg;
791 Raise_Error(_SC("attempt to call '%s'"), GetTypeName(temp_reg));
795 Raise_Error(_SC("attempt to call '%s'"), GetTypeName(temp_reg));
803 SQObjectPtr &key = _i_.op == _OP_PREPCALLK?(ci->_literals)[arg1]:STK(arg1);
804 SQObjectPtr &o = STK(arg2);
805 if (!Get(o, key, temp_reg,false,true)) {
806 if(type(o) == OT_CLASS) { //hack?
807 if(_class_ddel->Get(key,temp_reg)) {
813 { Raise_IdxError(key); SQ_THROW();}
816 STK(arg3) = type(o) == OT_CLASS?STK(0):o;
821 if (!Get(STK(arg2), ci->_literals[arg1], temp_reg, false,true)) { Raise_IdxError(ci->_literals[arg1]); SQ_THROW();}
824 case _OP_MOVE: TARGET = STK(arg1); continue;
826 _GUARD(NewSlot(STK(arg1), STK(arg2), STK(arg3),false));
827 if(arg0 != arg3) TARGET = STK(arg3);
829 case _OP_DELETE: _GUARD(DeleteSlot(STK(arg1), STK(arg2), TARGET)); continue;
831 if (!Set(STK(arg1), STK(arg2), STK(arg3),true)) { Raise_IdxError(STK(arg2)); SQ_THROW(); }
832 if (arg0 != arg3) TARGET = STK(arg3);
835 if (!Get(STK(arg1), STK(arg2), temp_reg, false,true)) { Raise_IdxError(STK(arg2)); SQ_THROW(); }
840 if(!IsEqual(STK(arg2),COND_LITERAL,res)) { SQ_THROW(); }
841 TARGET = res?_true_:_false_;
845 if(!IsEqual(STK(arg2),COND_LITERAL,res)) { SQ_THROW(); }
846 TARGET = (!res)?_true_:_false_;
848 case _OP_ARITH: _GUARD(ARITH_OP( arg3 , temp_reg, STK(arg2), STK(arg1))); TARGET = temp_reg; continue;
849 case _OP_BITW: _GUARD(BW_OP( arg3,TARGET,STK(arg2),STK(arg1))); continue;
851 if(type((ci)->_generator) == OT_GENERATOR) {
852 _generator((ci)->_generator)->Kill();
854 if(Return(arg0, arg1, temp_reg)){
860 case _OP_LOADNULLS:{ for(SQInt32 n=0; n < arg1; n++) STK(arg0+n) = _null_; }continue;
861 case _OP_LOADROOTTABLE: TARGET = _roottable; continue;
862 case _OP_LOADBOOL: TARGET = arg1?_true_:_false_; continue;
863 case _OP_DMOVE: STK(arg0) = STK(arg1); STK(arg2) = STK(arg3); continue;
864 case _OP_JMP: ci->_ip += (sarg1); continue;
865 case _OP_JNZ: if(!IsFalse(STK(arg0))) ci->_ip+=(sarg1); continue;
866 case _OP_JZ: if(IsFalse(STK(arg0))) ci->_ip+=(sarg1); continue;
867 case _OP_LOADFREEVAR: TARGET = _closure(ci->_closure)->_outervalues[arg1]; continue;
868 case _OP_VARGC: TARGET = SQInteger(ci->_vargs.size); continue;
870 if(!GETVARGV_OP(TARGET,STK(arg1),ci)) { SQ_THROW(); }
872 case _OP_NEWTABLE: TARGET = SQTable::Create(_ss(this), arg1); continue;
873 case _OP_NEWARRAY: TARGET = SQArray::Create(_ss(this), 0); _array(TARGET)->Reserve(arg1); continue;
874 case _OP_APPENDARRAY: _array(STK(arg0))->Append(COND_LITERAL); continue;
875 case _OP_GETPARENT: _GUARD(GETPARENT_OP(STK(arg1),TARGET)); continue;
876 case _OP_COMPARITH: _GUARD(DerefInc(arg3, TARGET, STK((((SQUnsignedInteger)arg1&0xFFFF0000)>>16)), STK(arg2), STK(arg1&0x0000FFFF), false)); continue;
877 case _OP_COMPARITHL: _GUARD(LOCAL_INC(arg3, TARGET, STK(arg1), STK(arg2))); continue;
878 case _OP_INC: {SQObjectPtr o(sarg3); _GUARD(DerefInc('+',TARGET, STK(arg1), STK(arg2), o, false));} continue;
879 case _OP_INCL: {SQObjectPtr o(sarg3); _GUARD(LOCAL_INC('+',TARGET, STK(arg1), o));} continue;
880 case _OP_PINC: {SQObjectPtr o(sarg3); _GUARD(DerefInc('+',TARGET, STK(arg1), STK(arg2), o, true));} continue;
881 case _OP_PINCL: {SQObjectPtr o(sarg3); _GUARD(PLOCAL_INC('+',TARGET, STK(arg1), o));} continue;
882 case _OP_CMP: _GUARD(CMP_OP((CmpOP)arg3,STK(arg2),STK(arg1),TARGET)) continue;
883 case _OP_EXISTS: TARGET = Get(STK(arg1), STK(arg2), temp_reg, true,false)?_true_:_false_;continue;
885 if(type(STK(arg1)) != OT_CLASS || type(STK(arg2)) != OT_INSTANCE)
886 {Raise_Error(_SC("cannot apply instanceof between a %s and a %s"),GetTypeName(STK(arg1)),GetTypeName(STK(arg2))); SQ_THROW();}
887 TARGET = _instance(STK(arg2))->InstanceOf(_class(STK(arg1)))?_true_:_false_;
890 if(IsFalse(STK(arg2))) {
896 if(!IsFalse(STK(arg2))) {
901 case _OP_NEG: _GUARD(NEG_OP(TARGET,STK(arg1))); continue;
902 case _OP_NOT: TARGET = (IsFalse(STK(arg1))?_true_:_false_); continue;
904 if(type(STK(arg1)) == OT_INTEGER) {
905 SQInteger t = _integer(STK(arg1));
906 TARGET = SQInteger(~t);
909 Raise_Error(_SC("attempt to perform a bitwise op on a %s"), GetTypeName(STK(arg1)));
912 SQClosure *c = ci->_closure._unVal.pClosure;
913 SQFunctionProto *fp = c->_function._unVal.pFunctionProto;
914 if(!CLOSURE_OP(TARGET,fp->_functions[arg1]._unVal.pFunctionProto)) { SQ_THROW(); }
918 if(type(ci->_generator) == OT_GENERATOR) {
919 if(sarg1 != MAX_FUNC_STACKSIZE) temp_reg = STK(arg1);
920 _GUARD(_generator(ci->_generator)->Yield(this));
921 traps -= ci->_etraps;
922 if(sarg1 != MAX_FUNC_STACKSIZE) STK(arg1) = temp_reg;
924 else { Raise_Error(_SC("trying to yield a '%s',only genenerator can be yielded"), GetTypeName(ci->_generator)); SQ_THROW();}
925 if(Return(arg0, arg1, temp_reg)){
934 if(type(STK(arg1)) != OT_GENERATOR){ Raise_Error(_SC("trying to resume a '%s',only genenerator can be resumed"), GetTypeName(STK(arg1))); SQ_THROW();}
935 _GUARD(_generator(STK(arg1))->Resume(this, arg0));
936 traps += ci->_etraps;
938 case _OP_FOREACH:{ int tojump;
939 _GUARD(FOREACH_OP(STK(arg0),STK(arg2),STK(arg2+1),STK(arg2+2),arg2,sarg1,tojump));
942 case _OP_POSTFOREACH:
943 assert(type(STK(arg0)) == OT_GENERATOR);
944 if(_generator(STK(arg0))->_state == SQGenerator::eDead)
945 ci->_ip += (sarg1 - 1);
947 case _OP_DELEGATE: _GUARD(DELEGATE_OP(TARGET,STK(arg1),STK(arg2))); continue;
949 if(!Clone(STK(arg1), TARGET))
950 { Raise_Error(_SC("cloning a %s"), GetTypeName(STK(arg1))); SQ_THROW();}
952 case _OP_TYPEOF: TypeOf(STK(arg1), TARGET); continue;
954 SQInstruction *_iv = _funcproto(_closure(ci->_closure)->_function)->_instructions;
955 _etraps.push_back(SQExceptionTrap(_top,_stackbase, &_iv[(ci->_ip-_iv)+arg1], arg0)); traps++;
960 for(SQInteger i = 0; i < arg0; i++) {
961 _etraps.pop_back(); traps--;
966 case _OP_THROW: Raise_Error(TARGET); SQ_THROW(); continue;
967 case _OP_CLASS: _GUARD(CLASS_OP(TARGET,arg1,arg2)); continue;
969 bool bstatic = (arg0&NEW_SLOT_STATIC_FLAG)?true:false;
970 if(type(STK(arg1)) == OT_CLASS) {
971 if(type(_class(STK(arg1))->_metamethods[MT_NEWMEMBER]) != OT_NULL ) {
972 Push(STK(arg1)); Push(STK(arg2)); Push(STK(arg3));
973 Push((arg0&NEW_SLOT_ATTRIBUTES_FLAG) ? STK(arg2-1) : _null_);
975 if(Call(_class(STK(arg1))->_metamethods[MT_NEWMEMBER], nparams, _top - nparams, temp_reg,SQFalse)) {
981 _GUARD(NewSlot(STK(arg1), STK(arg2), STK(arg3),bstatic));
982 if((arg0&NEW_SLOT_ATTRIBUTES_FLAG)) {
983 _class(STK(arg1))->SetAttributes(STK(arg2),STK(arg2-1));
992 SQObjectPtr currerror = _lasterror;
993 // dumpstack(_stackbase);
995 SQInteger last_top = _top;
997 if(_ss(this)->_notifyallexceptions) CallErrorHandler(currerror);
1001 if(ci->_etraps > 0) {
1002 SQExceptionTrap &et = _etraps.top();
1004 _top = et._stacksize;
1005 _stackbase = et._stackbase;
1006 _stack._vals[_stackbase+et._extarget] = currerror;
1007 _etraps.pop_back(); traps--; ci->_etraps--;
1008 while(last_top >= _top) _stack._vals[last_top--].Null();
1009 goto exception_restore;
1011 //if is a native closure
1012 if(type(ci->_closure) != OT_CLOSURE && n)
1014 if(type(ci->_generator) == OT_GENERATOR) _generator(ci->_generator)->Kill();
1015 PopVarArgs(ci->_vargs);
1018 } while(_callsstacksize);
1022 if(raiseerror && !_ss(this)->_notifyallexceptions)
1023 CallErrorHandler(currerror);
1025 //remove call stack until a C function is found or the cstack is empty
1027 SQBool exitafterthisone = ci->_root;
1028 if(type(ci->_generator) == OT_GENERATOR) _generator(ci->_generator)->Kill();
1029 _stackbase -= ci->_prevstkbase;
1030 _top = _stackbase + ci->_prevtop;
1031 PopVarArgs(ci->_vargs);
1033 if( (ci && type(ci->_closure) != OT_CLOSURE) || exitafterthisone) break;
1034 } while(_callsstacksize);
1036 while(last_top >= _top) _stack._vals[last_top--].Null();
1038 _lasterror = currerror;
1044 bool SQVM::CreateClassInstance(SQClass *theclass, SQObjectPtr &inst, SQObjectPtr &constructor)
1046 inst = theclass->CreateInstance();
1047 if(!theclass->Get(_ss(this)->_constructoridx,constructor)) {
1048 //if(!Call(constr,nargs,stackbase,constr,false))
1050 constructor = _null_;
1055 void SQVM::CallErrorHandler(SQObjectPtr &error)
1057 if(type(_errorhandler) != OT_NULL) {
1059 Push(_roottable); Push(error);
1060 Call(_errorhandler, 2, _top-2, out,SQFalse);
1065 void SQVM::CallDebugHook(SQInteger type,SQInteger forcedline)
1067 SQObjectPtr temp_reg;
1068 SQInteger nparams=5;
1069 SQFunctionProto *func=_funcproto(_closure(ci->_closure)->_function);
1070 Push(_roottable); Push(type); Push(func->_sourcename); Push(forcedline?forcedline:func->GetLine(ci->_ip)); Push(func->_name);
1071 Call(_debughook,nparams,_top-nparams,temp_reg,SQFalse);
1075 bool SQVM::CallNative(SQNativeClosure *nclosure,SQInteger nargs,SQInteger stackbase,SQObjectPtr &retval,bool &suspend)
1077 if (_nnativecalls + 1 > MAX_NATIVE_CALLS) { Raise_Error(_SC("Native stack overflow")); return false; }
1078 SQInteger nparamscheck = nclosure->_nparamscheck;
1079 if(((nparamscheck > 0) && (nparamscheck != nargs))
1080 || ((nparamscheck < 0) && (nargs < (-nparamscheck)))) {
1081 Raise_Error(_SC("wrong number of parameters"));
1086 if((tcs = nclosure->_typecheck.size())) {
1087 for(SQInteger i = 0; i < nargs && i < tcs; i++)
1088 if((nclosure->_typecheck._vals[i] != -1) && !(type(_stack._vals[stackbase+i]) & nclosure->_typecheck[i])) {
1089 Raise_ParamTypeError(i,nclosure->_typecheck._vals[i],type(_stack._vals[stackbase+i]));
1094 if ((_top + MIN_STACK_OVERHEAD) > (SQInteger)_stack.size()) {
1095 _stack.resize(_stack.size() + (MIN_STACK_OVERHEAD<<1));
1097 SQInteger oldtop = _top;
1098 SQInteger oldstackbase = _stackbase;
1099 _top = stackbase + nargs;
1102 lci._closure._unVal.pNativeClosure = nclosure;
1103 lci._closure._type = OT_NATIVECLOSURE;
1104 lci._prevstkbase = (SQInt32) (stackbase - _stackbase);
1106 lci._prevtop = (SQInt32) (oldtop - oldstackbase);
1107 PUSH_CALLINFO(this, lci);
1108 _stackbase = stackbase;
1109 //push free variables
1110 SQInteger outers = nclosure->_outervalues.size();
1111 for (SQInteger i = 0; i < outers; i++) {
1112 Push(nclosure->_outervalues[i]);
1115 if(type(nclosure->_env) == OT_WEAKREF) {
1116 _stack[stackbase] = _weakref(nclosure->_env)->_obj;
1120 SQInteger ret = (nclosure->_function)(this);
1123 if( ret == SQ_SUSPEND_FLAG) suspend = true;
1125 _stackbase = oldstackbase;
1128 Raise_Error(_lasterror);
1132 if (ret != 0){ retval = TOP(); }
1133 else { retval = _null_; }
1134 _stackbase = oldstackbase;
1140 bool SQVM::Get(const SQObjectPtr &self,const SQObjectPtr &key,SQObjectPtr &dest,bool raw, bool fetchroot)
1144 if(_table(self)->Get(key,dest))return true;
1147 if(sq_isnumeric(key)){
1148 return _array(self)->Get(tointeger(key),dest);
1152 if(_instance(self)->Get(key,dest)) return true;
1154 default:break; //shut up compiler
1156 if(FallBackGet(self,key,dest,raw)) return true;
1159 if(_rawval(STK(0)) == _rawval(self) &&
1160 type(STK(0)) == type(self)) {
1161 return _table(_roottable)->Get(key,dest);
1167 bool SQVM::FallBackGet(const SQObjectPtr &self,const SQObjectPtr &key,SQObjectPtr &dest,bool raw)
1171 return _class(self)->Get(key,dest);
1176 if(_delegable(self)->_delegate) {
1177 if(Get(SQObjectPtr(_delegable(self)->_delegate),key,dest,raw,false))
1179 if(raw)return false;
1180 Push(self);Push(key);
1181 if(CallMetaMethod(_delegable(self),MT_GET,2,dest))
1184 if(type(self) == OT_TABLE) {
1185 if(raw) return false;
1186 return _table_ddel->Get(key,dest);
1191 if(raw)return false;
1192 return _array_ddel->Get(key,dest);
1194 if(sq_isnumeric(key)){
1195 SQInteger n=tointeger(key);
1196 if(abs((int)n)<_string(self)->_len){
1197 if(n<0)n=_string(self)->_len-n;
1198 dest=SQInteger(_stringval(self)[n]);
1204 if(raw)return false;
1205 return _string_ddel->Get(key,dest);
1209 if(raw)return false;
1210 Push(self);Push(key);
1211 if(!CallMetaMethod(_delegable(self),MT_GET,2,dest)) {
1212 return _instance_ddel->Get(key,dest);
1215 case OT_INTEGER:case OT_FLOAT:case OT_BOOL:
1216 if(raw)return false;
1217 return _number_ddel->Get(key,dest);
1219 if(raw)return false;
1220 return _generator_ddel->Get(key,dest);
1221 case OT_CLOSURE: case OT_NATIVECLOSURE:
1222 if(raw)return false;
1223 return _closure_ddel->Get(key,dest);
1225 if(raw)return false;
1226 return _thread_ddel->Get(key,dest);
1228 if(raw)return false;
1229 return _weakref_ddel->Get(key,dest);
1230 default:return false;
1235 bool SQVM::Set(const SQObjectPtr &self,const SQObjectPtr &key,const SQObjectPtr &val,bool fetchroot)
1239 if(_table(self)->Set(key,val))
1241 if(_table(self)->_delegate) {
1242 if(Set(_table(self)->_delegate,key,val,false)) {
1248 if(_delegable(self)->_delegate) {
1250 Push(self);Push(key);Push(val);
1251 if(CallMetaMethod(_delegable(self),MT_SET,3,t)) return true;
1255 if(_instance(self)->Set(key,val))
1258 Push(self);Push(key);Push(val);
1259 if(CallMetaMethod(_delegable(self),MT_SET,3,t)) return true;
1263 if(!sq_isnumeric(key)) {Raise_Error(_SC("indexing %s with %s"),GetTypeName(self),GetTypeName(key)); return false; }
1264 return _array(self)->Set(tointeger(key),val);
1266 Raise_Error(_SC("trying to set '%s'"),GetTypeName(self));
1270 if(_rawval(STK(0)) == _rawval(self) &&
1271 type(STK(0)) == type(self)) {
1272 return _table(_roottable)->Set(key,val);
1278 bool SQVM::Clone(const SQObjectPtr &self,SQObjectPtr &target)
1280 SQObjectPtr temp_reg;
1284 newobj = _table(self)->Clone();
1287 newobj = _instance(self)->Clone(_ss(this));
1289 if(_delegable(newobj)->_delegate){
1292 CallMetaMethod(_delegable(newobj),MT_CLONED,2,temp_reg);
1297 target = _array(self)->Clone();
1299 default: return false;
1303 bool SQVM::NewSlot(const SQObjectPtr &self,const SQObjectPtr &key,const SQObjectPtr &val,bool bstatic)
1305 if(type(key) == OT_NULL) { Raise_Error(_SC("null cannot be used as index")); return false; }
1306 switch(type(self)) {
1308 bool rawcall = true;
1309 if(_table(self)->_delegate) {
1311 if(!_table(self)->Get(key,res)) {
1312 Push(self);Push(key);Push(val);
1313 rawcall = !CallMetaMethod(_table(self),MT_NEWSLOT,3,res);
1316 if(rawcall) _table(self)->NewSlot(key,val); //cannot fail
1320 if(!_class(self)->NewSlot(_ss(this),key,val,bstatic)) {
1321 if(_class(self)->_locked) {
1322 Raise_Error(_SC("trying to modify a class that has already been instantiated"));
1326 SQObjectPtr oval = PrintObjVal(key);
1327 Raise_Error(_SC("the property '%s' already exists"),_stringval(oval));
1333 Raise_Error(_SC("indexing %s with %s"),GetTypeName(self),GetTypeName(key));
1340 bool SQVM::DeleteSlot(const SQObjectPtr &self,const SQObjectPtr &key,SQObjectPtr &res)
1342 switch(type(self)) {
1347 bool handled = false;
1348 if(_delegable(self)->_delegate) {
1349 Push(self);Push(key);
1350 handled = CallMetaMethod(_delegable(self),MT_DELSLOT,2,t);
1354 if(type(self) == OT_TABLE) {
1355 if(_table(self)->Get(key,t)) {
1356 _table(self)->Remove(key);
1359 Raise_IdxError((SQObject &)key);
1364 Raise_Error(_SC("cannot delete a slot from %s"),GetTypeName(self));
1372 Raise_Error(_SC("attempt to delete a slot from a %s"),GetTypeName(self));
1378 bool SQVM::Call(SQObjectPtr &closure,SQInteger nparams,SQInteger stackbase,SQObjectPtr &outres,SQBool raiseerror)
1381 SQInteger prevstackbase = _stackbase;
1383 switch(type(closure)) {
1385 return Execute(closure, _top - nparams, nparams, stackbase,outres,raiseerror);
1387 case OT_NATIVECLOSURE:{
1389 return CallNative(_nativeclosure(closure), nparams, stackbase, outres,suspend);
1396 CreateClassInstance(_class(closure),outres,constr);
1397 if(type(constr) != OT_NULL) {
1398 _stack[stackbase] = outres;
1399 return Call(constr,nparams,stackbase,temp,raiseerror);
1409 assert(_stackbase == prevstackbase);
1415 bool SQVM::CallMetaMethod(SQDelegable *del,SQMetaMethod mm,SQInteger nparams,SQObjectPtr &outres)
1417 SQObjectPtr closure;
1418 if(del->GetMetaMethod(this, mm, closure)) {
1419 if(Call(closure, nparams, _top - nparams, outres, SQFalse)) {
1428 void SQVM::Remove(SQInteger n) {
1429 n = (n >= 0)?n + _stackbase - 1:_top + n;
1430 for(SQInteger i = n; i < _top; i++){
1431 _stack[i] = _stack[i+1];
1433 _stack[_top] = _null_;
1438 _stack[--_top] = _null_;
1441 void SQVM::Pop(SQInteger n) {
1442 for(SQInteger i = 0; i < n; i++){
1443 _stack[--_top] = _null_;
1447 void SQVM::Push(const SQObjectPtr &o) { _stack[_top++] = o; }
1448 SQObjectPtr &SQVM::Top() { return _stack[_top-1]; }
1449 SQObjectPtr &SQVM::PopGet() { return _stack[--_top]; }
1450 SQObjectPtr &SQVM::GetUp(SQInteger n) { return _stack[_top+n]; }
1451 SQObjectPtr &SQVM::GetAt(SQInteger n) { return _stack[n]; }
1454 void SQVM::dumpstack(SQInteger stackbase,bool dumpall)
1456 SQInteger size=dumpall?_stack.size():_top;
1458 scprintf(_SC("\n>>>>stack dump<<<<\n"));
1459 CallInfo &ci=_callsstack[_callsstacksize-1];
1460 scprintf(_SC("IP: %p\n"),ci._ip);
1461 scprintf(_SC("prev stack base: %d\n"),ci._prevstkbase);
1462 scprintf(_SC("prev top: %d\n"),ci._prevtop);
1463 for(SQInteger i=0;i<size;i++){
1464 SQObjectPtr &obj=_stack[i];
1465 if(stackbase==i)scprintf(_SC(">"));else scprintf(_SC(" "));
1466 scprintf(_SC("[%d]:"),n);
1468 case OT_FLOAT: scprintf(_SC("FLOAT %.3f"),_float(obj));break;
1469 case OT_INTEGER: scprintf(_SC("INTEGER %d"),_integer(obj));break;
1470 case OT_BOOL: scprintf(_SC("BOOL %s"),_integer(obj)?"true":"false");break;
1471 case OT_STRING: scprintf(_SC("STRING %s"),_stringval(obj));break;
1472 case OT_NULL: scprintf(_SC("NULL")); break;
1473 case OT_TABLE: scprintf(_SC("TABLE %p[%p]"),_table(obj),_table(obj)->_delegate);break;
1474 case OT_ARRAY: scprintf(_SC("ARRAY %p"),_array(obj));break;
1475 case OT_CLOSURE: scprintf(_SC("CLOSURE [%p]"),_closure(obj));break;
1476 case OT_NATIVECLOSURE: scprintf(_SC("NATIVECLOSURE"));break;
1477 case OT_USERDATA: scprintf(_SC("USERDATA %p[%p]"),_userdataval(obj),_userdata(obj)->_delegate);break;
1478 case OT_GENERATOR: scprintf(_SC("GENERATOR"));break;
1479 case OT_THREAD: scprintf(_SC("THREAD [%p]"),_thread(obj));break;
1480 case OT_USERPOINTER: scprintf(_SC("USERPOINTER %p"),_userpointer(obj));break;
1481 case OT_CLASS: scprintf(_SC("CLASS %p"),_class(obj));break;
1482 case OT_INSTANCE: scprintf(_SC("INSTANCE %p"),_instance(obj));break;
1483 case OT_WEAKREF: scprintf(_SC("WEAKERF %p"),_weakref(obj));break;
1488 scprintf(_SC("\n"));