-/*\r
- see copyright notice in squirrel.h\r
-*/\r
-#include "sqpcheader.h"\r
-#include "sqvm.h"\r
-#include "sqstring.h"\r
-#include "sqarray.h"\r
-#include "sqtable.h"\r
-#include "squserdata.h"\r
-#include "sqfuncproto.h"\r
-#include "sqclass.h"\r
-#include "sqclosure.h"\r
-\r
-SQString *SQString::Create(SQSharedState *ss,const SQChar *s,SQInteger len)\r
-{\r
- SQString *str=ADD_STRING(ss,s,len);\r
- str->_sharedstate=ss;\r
- return str;\r
-}\r
-\r
-void SQString::Release()\r
-{\r
- REMOVE_STRING(_sharedstate,this);\r
-}\r
-\r
-SQInteger SQString::Next(const SQObjectPtr &refpos, SQObjectPtr &outkey, SQObjectPtr &outval)\r
-{\r
- SQInteger idx = (SQInteger)TranslateIndex(refpos);\r
- while(idx < _len){\r
- outkey = (SQInteger)idx;\r
- outval = SQInteger(_val[idx]);\r
- //return idx for the next iteration\r
- return ++idx;\r
- }\r
- //nothing to iterate anymore\r
- return -1;\r
-}\r
-\r
-SQUnsignedInteger TranslateIndex(const SQObjectPtr &idx)\r
-{\r
- switch(type(idx)){\r
- case OT_NULL:\r
- return 0;\r
- case OT_INTEGER:\r
- return (SQUnsignedInteger)_integer(idx);\r
- default: assert(0); break;\r
- }\r
- return 0;\r
-}\r
-\r
-SQWeakRef *SQRefCounted::GetWeakRef(SQObjectType type)\r
-{\r
- if(!_weakref) {\r
- sq_new(_weakref,SQWeakRef);\r
- _weakref->_obj._type = type;\r
- _weakref->_obj._unVal.pRefCounted = this;\r
- }\r
- return _weakref;\r
-}\r
-\r
-SQRefCounted::~SQRefCounted()\r
-{\r
- if(_weakref) {\r
- _weakref->_obj._type = OT_NULL;\r
- _weakref->_obj._unVal.pRefCounted = NULL;\r
- }\r
-}\r
-\r
-void SQWeakRef::Release() { \r
- if(ISREFCOUNTED(_obj._type)) { \r
- _obj._unVal.pRefCounted->_weakref = NULL;\r
- } \r
- sq_delete(this,SQWeakRef);\r
-}\r
-\r
-bool SQDelegable::GetMetaMethod(SQVM *v,SQMetaMethod mm,SQObjectPtr &res) {\r
- if(_delegate) {\r
- return _delegate->Get((*_ss(v)->_metamethods)[mm],res);\r
- }\r
- return false;\r
-}\r
-\r
-bool SQDelegable::SetDelegate(SQTable *mt)\r
-{\r
- SQTable *temp = mt;\r
- while (temp) {\r
- if (temp->_delegate == this) return false; //cycle detected\r
- temp = temp->_delegate;\r
- }\r
- if (mt) __ObjAddRef(mt);\r
- __ObjRelease(_delegate);\r
- _delegate = mt;\r
- return true;\r
-}\r
-\r
-bool SQGenerator::Yield(SQVM *v)\r
-{\r
- if(_state==eSuspended) { v->Raise_Error(_SC("internal vm error, yielding dead generator")); return false;}\r
- if(_state==eDead) { v->Raise_Error(_SC("internal vm error, yielding a dead generator")); return false; }\r
- SQInteger size = v->_top-v->_stackbase;\r
- _ci=*v->ci;\r
- _stack.resize(size);\r
- for(SQInteger n =0; n<size; n++) {\r
- _stack._vals[n] = v->_stack[v->_stackbase+n];\r
- v->_stack[v->_stackbase+n] = _null_;\r
- }\r
- SQInteger nvargs = v->ci->_vargs.size;\r
- SQInteger vargsbase = v->ci->_vargs.base;\r
- for(SQInteger j = nvargs - 1; j >= 0; j--) {\r
- _vargsstack.push_back(v->_vargsstack[vargsbase+j]);\r
- }\r
- _ci._generator=_null_;\r
- for(SQInteger i=0;i<_ci._etraps;i++) {\r
- _etraps.push_back(v->_etraps.top());\r
- v->_etraps.pop_back();\r
- }\r
- _state=eSuspended;\r
- return true;\r
-}\r
-\r
-bool SQGenerator::Resume(SQVM *v,SQInteger target)\r
-{\r
- SQInteger size=_stack.size();\r
- if(_state==eDead){ v->Raise_Error(_SC("resuming dead generator")); return false; }\r
- if(_state==eRunning){ v->Raise_Error(_SC("resuming active generator")); return false; }\r
- SQInteger prevtop=v->_top-v->_stackbase;\r
- PUSH_CALLINFO(v,_ci);\r
- SQInteger oldstackbase=v->_stackbase;\r
- v->_stackbase=v->_top;\r
- v->ci->_target=target;\r
- v->ci->_generator=SQObjectPtr(this);\r
- v->ci->_vargs.size = _vargsstack.size();\r
- \r
- for(SQInteger i=0;i<_ci._etraps;i++) {\r
- v->_etraps.push_back(_etraps.top());\r
- _etraps.pop_back();\r
- }\r
- for(SQInteger n =0; n<size; n++) {\r
- v->_stack[v->_stackbase+n] = _stack._vals[n];\r
- _stack._vals[0] = _null_;\r
- }\r
- while(_vargsstack.size()) {\r
- v->_vargsstack.push_back(_vargsstack.back());\r
- _vargsstack.pop_back();\r
- }\r
- v->ci->_vargs.base = v->_vargsstack.size() - v->ci->_vargs.size;\r
- v->_top=v->_stackbase+size;\r
- v->ci->_prevtop=prevtop;\r
- v->ci->_prevstkbase=v->_stackbase-oldstackbase;\r
- _state=eRunning;\r
- return true;\r
-}\r
-\r
-void SQArray::Extend(const SQArray *a){\r
- SQInteger xlen;\r
- if((xlen=a->Size()))\r
- for(SQInteger i=0;i<xlen;i++)\r
- Append(a->_values[i]);\r
-}\r
-\r
-const SQChar* SQFunctionProto::GetLocal(SQVM *vm,SQUnsignedInteger stackbase,SQUnsignedInteger nseq,SQUnsignedInteger nop)\r
-{\r
- SQUnsignedInteger nvars=_localvarinfos.size();\r
- const SQChar *res=NULL; \r
- if(nvars>=nseq){\r
- for(SQUnsignedInteger i=0;i<nvars;i++){\r
- if(_localvarinfos[i]._start_op<=nop && _localvarinfos[i]._end_op>=nop)\r
- {\r
- if(nseq==0){\r
- vm->Push(vm->_stack[stackbase+_localvarinfos[i]._pos]);\r
- res=_stringval(_localvarinfos[i]._name);\r
- break;\r
- }\r
- nseq--;\r
- }\r
- }\r
- }\r
- return res;\r
-}\r
-\r
-SQInteger SQFunctionProto::GetLine(SQInstruction *curr)\r
-{\r
- SQInteger op = (SQInteger)(curr-_instructions._vals);\r
- SQInteger line=_lineinfos[0]._line;\r
- for(SQUnsignedInteger i=1;i<_lineinfos.size();i++){\r
- if(_lineinfos[i]._op>=op)\r
- return line;\r
- line=_lineinfos[i]._line;\r
- }\r
- return line;\r
-}\r
-\r
-//#define _ERROR_TRAP() error_trap:\r
-#define _CHECK_IO(exp) { if(!exp)return false; }\r
-bool SafeWrite(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUserPointer dest,SQInteger size)\r
-{\r
- if(write(up,dest,size) != size) {\r
- v->Raise_Error(_SC("io error (write function failure)"));\r
- return false;\r
- }\r
- return true;\r
-}\r
-\r
-bool SafeRead(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUserPointer dest,SQInteger size)\r
-{\r
- if(size && read(up,dest,size) != size) {\r
- v->Raise_Error(_SC("io error, read function failure, the origin stream could be corrupted/trucated"));\r
- return false;\r
- }\r
- return true;\r
-}\r
-\r
-bool WriteTag(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQInteger tag)\r
-{\r
- return SafeWrite(v,write,up,&tag,sizeof(tag));\r
-}\r
-\r
-bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQInteger tag)\r
-{\r
- SQInteger t;\r
- _CHECK_IO(SafeRead(v,read,up,&t,sizeof(t)));\r
- if(t != tag){\r
- v->Raise_Error(_SC("invalid or corrupted closure stream"));\r
- return false;\r
- }\r
- return true;\r
-}\r
-\r
-bool WriteObject(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o)\r
-{\r
- _CHECK_IO(SafeWrite(v,write,up,&type(o),sizeof(SQObjectType)));\r
- switch(type(o)){\r
- case OT_STRING:\r
- _CHECK_IO(SafeWrite(v,write,up,&_string(o)->_len,sizeof(SQInteger)));\r
- _CHECK_IO(SafeWrite(v,write,up,_stringval(o),rsl(_string(o)->_len)));\r
- break;\r
- case OT_INTEGER:\r
- _CHECK_IO(SafeWrite(v,write,up,&_integer(o),sizeof(SQInteger)));break;\r
- case OT_FLOAT:\r
- _CHECK_IO(SafeWrite(v,write,up,&_float(o),sizeof(SQFloat)));break;\r
- case OT_NULL:\r
- break;\r
- default:\r
- v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));\r
- return false;\r
- }\r
- return true;\r
-}\r
-\r
-bool ReadObject(HSQUIRRELVM v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &o)\r
-{\r
- SQObjectType t;\r
- _CHECK_IO(SafeRead(v,read,up,&t,sizeof(SQObjectType)));\r
- switch(t){\r
- case OT_STRING:{\r
- SQInteger len;\r
- _CHECK_IO(SafeRead(v,read,up,&len,sizeof(SQInteger)));\r
- _CHECK_IO(SafeRead(v,read,up,_ss(v)->GetScratchPad(rsl(len)),rsl(len)));\r
- o=SQString::Create(_ss(v),_ss(v)->GetScratchPad(-1),len);\r
- }\r
- break;\r
- case OT_INTEGER:{\r
- SQInteger i;\r
- _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o = i; break;\r
- }\r
- case OT_FLOAT:{\r
- SQFloat f;\r
- _CHECK_IO(SafeRead(v,read,up,&f,sizeof(SQFloat))); o = f; break;\r
- }\r
- case OT_NULL:\r
- o=_null_;\r
- break;\r
- default:\r
- v->Raise_Error(_SC("cannot serialize a %s"),IdType2Name(t));\r
- return false;\r
- }\r
- return true;\r
-}\r
-\r
-bool SQClosure::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)\r
-{\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_HEAD));\r
- _CHECK_IO(WriteTag(v,write,up,sizeof(SQChar)));\r
- _CHECK_IO(_funcproto(_function)->Save(v,up,write));\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_TAIL));\r
- return true;\r
-}\r
-\r
-bool SQClosure::Load(SQVM *v,SQUserPointer up,SQREADFUNC read)\r
-{\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_HEAD));\r
- _CHECK_IO(CheckTag(v,read,up,sizeof(SQChar)));\r
- _CHECK_IO(_funcproto(_function)->Load(v,up,read));\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_TAIL));\r
- return true;\r
-}\r
-\r
-bool SQFunctionProto::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)\r
-{\r
- SQInteger i,nsize=_literals.size();\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(WriteObject(v,up,write,_sourcename));\r
- _CHECK_IO(WriteObject(v,up,write,_name));\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- for(i=0;i<nsize;i++){\r
- _CHECK_IO(WriteObject(v,up,write,_literals[i]));\r
- }\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- nsize=_parameters.size();\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- for(i=0;i<nsize;i++){\r
- _CHECK_IO(WriteObject(v,up,write,_parameters[i]));\r
- }\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- nsize=_outervalues.size();\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- for(i=0;i<nsize;i++){\r
- _CHECK_IO(SafeWrite(v,write,up,&_outervalues[i]._type,sizeof(SQUnsignedInteger)));\r
- _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._src));\r
- _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._name));\r
- }\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- nsize=_localvarinfos.size();\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- for(i=0;i<nsize;i++){\r
- SQLocalVarInfo &lvi=_localvarinfos[i];\r
- _CHECK_IO(WriteObject(v,up,write,lvi._name));\r
- _CHECK_IO(SafeWrite(v,write,up,&lvi._pos,sizeof(SQUnsignedInteger)));\r
- _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(SQUnsignedInteger)));\r
- _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(SQUnsignedInteger)));\r
- }\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- nsize=_lineinfos.size();\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- _CHECK_IO(SafeWrite(v,write,up,&_lineinfos[0],sizeof(SQLineInfo)*nsize));\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- nsize=_instructions.size();\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- _CHECK_IO(SafeWrite(v,write,up,&_instructions[0],sizeof(SQInstruction)*nsize));\r
- _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));\r
- nsize=_functions.size();\r
- _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));\r
- for(i=0;i<nsize;i++){\r
- _CHECK_IO(_funcproto(_functions[i])->Save(v,up,write));\r
- }\r
- _CHECK_IO(SafeWrite(v,write,up,&_stacksize,sizeof(_stacksize)));\r
- _CHECK_IO(SafeWrite(v,write,up,&_bgenerator,sizeof(_bgenerator)));\r
- _CHECK_IO(SafeWrite(v,write,up,&_varparams,sizeof(_varparams)));\r
- return true;\r
-}\r
-\r
-bool SQFunctionProto::Load(SQVM *v,SQUserPointer up,SQREADFUNC read)\r
-{\r
- SQInteger i, nsize = _literals.size();\r
- SQObjectPtr o;\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(ReadObject(v, up, read, _sourcename));\r
- _CHECK_IO(ReadObject(v, up, read, _name));\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));\r
- for(i = 0;i < nsize; i++){\r
- _CHECK_IO(ReadObject(v, up, read, o));\r
- _literals.push_back(o);\r
- }\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));\r
- for(i = 0; i < nsize; i++){\r
- _CHECK_IO(ReadObject(v, up, read, o));\r
- _parameters.push_back(o);\r
- }\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up,&nsize,sizeof(nsize)));\r
- for(i = 0; i < nsize; i++){\r
- SQUnsignedInteger type;\r
- SQObjectPtr name;\r
- _CHECK_IO(SafeRead(v,read,up, &type, sizeof(SQUnsignedInteger)));\r
- _CHECK_IO(ReadObject(v, up, read, o));\r
- _CHECK_IO(ReadObject(v, up, read, name));\r
- _outervalues.push_back(SQOuterVar(name,o, (SQOuterType)type));\r
- }\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up,&nsize, sizeof(nsize)));\r
- for(i = 0; i < nsize; i++){\r
- SQLocalVarInfo lvi;\r
- _CHECK_IO(ReadObject(v, up, read, lvi._name));\r
- _CHECK_IO(SafeRead(v,read,up, &lvi._pos, sizeof(SQUnsignedInteger)));\r
- _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(SQUnsignedInteger)));\r
- _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(SQUnsignedInteger)));\r
- _localvarinfos.push_back(lvi);\r
- }\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up, &nsize,sizeof(nsize)));\r
- _lineinfos.resize(nsize);\r
- _CHECK_IO(SafeRead(v,read,up, &_lineinfos[0], sizeof(SQLineInfo)*nsize));\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));\r
- _instructions.resize(nsize);\r
- _CHECK_IO(SafeRead(v,read,up, &_instructions[0], sizeof(SQInstruction)*nsize));\r
- _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));\r
- _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));\r
- for(i = 0; i < nsize; i++){\r
- o = SQFunctionProto::Create();\r
- _CHECK_IO(_funcproto(o)->Load(v, up, read));\r
- _functions.push_back(o);\r
- }\r
- _CHECK_IO(SafeRead(v,read,up, &_stacksize, sizeof(_stacksize)));\r
- _CHECK_IO(SafeRead(v,read,up, &_bgenerator, sizeof(_bgenerator)));\r
- _CHECK_IO(SafeRead(v,read,up, &_varparams, sizeof(_varparams)));\r
- return true;\r
-}\r
-\r
-#ifndef NO_GARBAGE_COLLECTOR\r
-\r
-#define START_MARK() if(!(_uiRef&MARK_FLAG)){ \\r
- _uiRef|=MARK_FLAG;\r
-\r
-#define END_MARK() RemoveFromChain(&_sharedstate->_gc_chain, this); \\r
- AddToChain(chain, this); }\r
-\r
-void SQVM::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- SQSharedState::MarkObject(_lasterror,chain);\r
- SQSharedState::MarkObject(_errorhandler,chain);\r
- SQSharedState::MarkObject(_debughook,chain);\r
- SQSharedState::MarkObject(_roottable, chain);\r
- SQSharedState::MarkObject(temp_reg, chain);\r
- for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);\r
- for(SQUnsignedInteger j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);\r
- END_MARK()\r
-}\r
-\r
-void SQArray::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- SQInteger len = _values.size();\r
- for(SQInteger i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);\r
- END_MARK()\r
-}\r
-void SQTable::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- if(_delegate) _delegate->Mark(chain);\r
- SQInteger len = _numofnodes;\r
- for(SQInteger i = 0; i < len; i++){\r
- SQSharedState::MarkObject(_nodes[i].key, chain);\r
- SQSharedState::MarkObject(_nodes[i].val, chain);\r
- }\r
- END_MARK()\r
-}\r
-\r
-void SQClass::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- _members->Mark(chain);\r
- if(_base) _base->Mark(chain);\r
- SQSharedState::MarkObject(_attributes, chain);\r
- for(SQUnsignedInteger i =0; i< _defaultvalues.size(); i++) {\r
- SQSharedState::MarkObject(_defaultvalues[i].val, chain);\r
- SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);\r
- }\r
- for(SQUnsignedInteger j =0; j< _methods.size(); j++) {\r
- SQSharedState::MarkObject(_methods[j].val, chain);\r
- SQSharedState::MarkObject(_methods[j].attrs, chain);\r
- }\r
- for(SQUnsignedInteger k =0; k< _metamethods.size(); k++) {\r
- SQSharedState::MarkObject(_metamethods[k], chain);\r
- }\r
- END_MARK()\r
-}\r
-\r
-void SQInstance::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- _class->Mark(chain);\r
- for(SQUnsignedInteger i =0; i< _nvalues; i++) {\r
- SQSharedState::MarkObject(_values[i], chain);\r
- }\r
- END_MARK()\r
-}\r
-\r
-void SQGenerator::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);\r
- for(SQUnsignedInteger j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);\r
- SQSharedState::MarkObject(_closure, chain);\r
- END_MARK()\r
-}\r
-\r
-void SQClosure::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- for(SQUnsignedInteger i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);\r
- END_MARK()\r
-}\r
-\r
-void SQNativeClosure::Mark(SQCollectable **chain)\r
-{\r
- START_MARK()\r
- for(SQUnsignedInteger i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);\r
- END_MARK()\r
-}\r
-\r
-void SQUserData::Mark(SQCollectable **chain){\r
- START_MARK()\r
- if(_delegate) _delegate->Mark(chain);\r
- END_MARK()\r
-}\r
-\r
-void SQCollectable::UnMark() { _uiRef&=~MARK_FLAG; }\r
-\r
-#endif\r
-\r
+/*
+ see copyright notice in squirrel.h
+*/
+#include "sqpcheader.h"
+#include "sqvm.h"
+#include "sqstring.h"
+#include "sqarray.h"
+#include "sqtable.h"
+#include "squserdata.h"
+#include "sqfuncproto.h"
+#include "sqclass.h"
+#include "sqclosure.h"
+
+SQString *SQString::Create(SQSharedState *ss,const SQChar *s,SQInteger len)
+{
+ SQString *str=ADD_STRING(ss,s,len);
+ str->_sharedstate=ss;
+ return str;
+}
+
+void SQString::Release()
+{
+ REMOVE_STRING(_sharedstate,this);
+}
+
+SQInteger SQString::Next(const SQObjectPtr &refpos, SQObjectPtr &outkey, SQObjectPtr &outval)
+{
+ SQInteger idx = (SQInteger)TranslateIndex(refpos);
+ while(idx < _len){
+ outkey = (SQInteger)idx;
+ outval = SQInteger(_val[idx]);
+ //return idx for the next iteration
+ return ++idx;
+ }
+ //nothing to iterate anymore
+ return -1;
+}
+
+SQUnsignedInteger TranslateIndex(const SQObjectPtr &idx)
+{
+ switch(type(idx)){
+ case OT_NULL:
+ return 0;
+ case OT_INTEGER:
+ return (SQUnsignedInteger)_integer(idx);
+ default: assert(0); break;
+ }
+ return 0;
+}
+
+SQWeakRef *SQRefCounted::GetWeakRef(SQObjectType type)
+{
+ if(!_weakref) {
+ sq_new(_weakref,SQWeakRef);
+ _weakref->_obj._type = type;
+ _weakref->_obj._unVal.pRefCounted = this;
+ }
+ return _weakref;
+}
+
+SQRefCounted::~SQRefCounted()
+{
+ if(_weakref) {
+ _weakref->_obj._type = OT_NULL;
+ _weakref->_obj._unVal.pRefCounted = NULL;
+ }
+}
+
+void SQWeakRef::Release() {
+ if(ISREFCOUNTED(_obj._type)) {
+ _obj._unVal.pRefCounted->_weakref = NULL;
+ }
+ sq_delete(this,SQWeakRef);
+}
+
+bool SQDelegable::GetMetaMethod(SQVM *v,SQMetaMethod mm,SQObjectPtr &res) {
+ if(_delegate) {
+ return _delegate->Get((*_ss(v)->_metamethods)[mm],res);
+ }
+ return false;
+}
+
+bool SQDelegable::SetDelegate(SQTable *mt)
+{
+ SQTable *temp = mt;
+ while (temp) {
+ if (temp->_delegate == this) return false; //cycle detected
+ temp = temp->_delegate;
+ }
+ if (mt) __ObjAddRef(mt);
+ __ObjRelease(_delegate);
+ _delegate = mt;
+ return true;
+}
+
+bool SQGenerator::Yield(SQVM *v)
+{
+ if(_state==eSuspended) { v->Raise_Error(_SC("internal vm error, yielding dead generator")); return false;}
+ if(_state==eDead) { v->Raise_Error(_SC("internal vm error, yielding a dead generator")); return false; }
+ SQInteger size = v->_top-v->_stackbase;
+ _ci=*v->ci;
+ _stack.resize(size);
+ for(SQInteger n =0; n<size; n++) {
+ _stack._vals[n] = v->_stack[v->_stackbase+n];
+ v->_stack[v->_stackbase+n] = _null_;
+ }
+ SQInteger nvargs = v->ci->_vargs.size;
+ SQInteger vargsbase = v->ci->_vargs.base;
+ for(SQInteger j = nvargs - 1; j >= 0; j--) {
+ _vargsstack.push_back(v->_vargsstack[vargsbase+j]);
+ }
+ _ci._generator=_null_;
+ for(SQInteger i=0;i<_ci._etraps;i++) {
+ _etraps.push_back(v->_etraps.top());
+ v->_etraps.pop_back();
+ }
+ _state=eSuspended;
+ return true;
+}
+
+bool SQGenerator::Resume(SQVM *v,SQInteger target)
+{
+ SQInteger size=_stack.size();
+ if(_state==eDead){ v->Raise_Error(_SC("resuming dead generator")); return false; }
+ if(_state==eRunning){ v->Raise_Error(_SC("resuming active generator")); return false; }
+ SQInteger prevtop=v->_top-v->_stackbase;
+ PUSH_CALLINFO(v,_ci);
+ SQInteger oldstackbase=v->_stackbase;
+ v->_stackbase=v->_top;
+ v->ci->_target=target;
+ v->ci->_generator=SQObjectPtr(this);
+ v->ci->_vargs.size = _vargsstack.size();
+
+ for(SQInteger i=0;i<_ci._etraps;i++) {
+ v->_etraps.push_back(_etraps.top());
+ _etraps.pop_back();
+ }
+ for(SQInteger n =0; n<size; n++) {
+ v->_stack[v->_stackbase+n] = _stack._vals[n];
+ _stack._vals[0] = _null_;
+ }
+ while(_vargsstack.size()) {
+ v->_vargsstack.push_back(_vargsstack.back());
+ _vargsstack.pop_back();
+ }
+ v->ci->_vargs.base = v->_vargsstack.size() - v->ci->_vargs.size;
+ v->_top=v->_stackbase+size;
+ v->ci->_prevtop=prevtop;
+ v->ci->_prevstkbase=v->_stackbase-oldstackbase;
+ _state=eRunning;
+ return true;
+}
+
+void SQArray::Extend(const SQArray *a){
+ SQInteger xlen;
+ if((xlen=a->Size()))
+ for(SQInteger i=0;i<xlen;i++)
+ Append(a->_values[i]);
+}
+
+const SQChar* SQFunctionProto::GetLocal(SQVM *vm,SQUnsignedInteger stackbase,SQUnsignedInteger nseq,SQUnsignedInteger nop)
+{
+ SQUnsignedInteger nvars=_localvarinfos.size();
+ const SQChar *res=NULL;
+ if(nvars>=nseq){
+ for(SQUnsignedInteger i=0;i<nvars;i++){
+ if(_localvarinfos[i]._start_op<=nop && _localvarinfos[i]._end_op>=nop)
+ {
+ if(nseq==0){
+ vm->Push(vm->_stack[stackbase+_localvarinfos[i]._pos]);
+ res=_stringval(_localvarinfos[i]._name);
+ break;
+ }
+ nseq--;
+ }
+ }
+ }
+ return res;
+}
+
+SQInteger SQFunctionProto::GetLine(SQInstruction *curr)
+{
+ SQInteger op = (SQInteger)(curr-_instructions._vals);
+ SQInteger line=_lineinfos[0]._line;
+ for(SQUnsignedInteger i=1;i<_lineinfos.size();i++){
+ if(_lineinfos[i]._op>=op)
+ return line;
+ line=_lineinfos[i]._line;
+ }
+ return line;
+}
+
+//#define _ERROR_TRAP() error_trap:
+#define _CHECK_IO(exp) { if(!exp)return false; }
+bool SafeWrite(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUserPointer dest,SQInteger size)
+{
+ if(write(up,dest,size) != size) {
+ v->Raise_Error(_SC("io error (write function failure)"));
+ return false;
+ }
+ return true;
+}
+
+bool SafeRead(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUserPointer dest,SQInteger size)
+{
+ if(size && read(up,dest,size) != size) {
+ v->Raise_Error(_SC("io error, read function failure, the origin stream could be corrupted/trucated"));
+ return false;
+ }
+ return true;
+}
+
+bool WriteTag(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQInteger tag)
+{
+ return SafeWrite(v,write,up,&tag,sizeof(tag));
+}
+
+bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQInteger tag)
+{
+ SQInteger t;
+ _CHECK_IO(SafeRead(v,read,up,&t,sizeof(t)));
+ if(t != tag){
+ v->Raise_Error(_SC("invalid or corrupted closure stream"));
+ return false;
+ }
+ return true;
+}
+
+bool WriteObject(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o)
+{
+ _CHECK_IO(SafeWrite(v,write,up,&type(o),sizeof(SQObjectType)));
+ switch(type(o)){
+ case OT_STRING:
+ _CHECK_IO(SafeWrite(v,write,up,&_string(o)->_len,sizeof(SQInteger)));
+ _CHECK_IO(SafeWrite(v,write,up,_stringval(o),rsl(_string(o)->_len)));
+ break;
+ case OT_INTEGER:
+ _CHECK_IO(SafeWrite(v,write,up,&_integer(o),sizeof(SQInteger)));break;
+ case OT_FLOAT:
+ _CHECK_IO(SafeWrite(v,write,up,&_float(o),sizeof(SQFloat)));break;
+ case OT_NULL:
+ break;
+ default:
+ v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
+ return false;
+ }
+ return true;
+}
+
+bool ReadObject(HSQUIRRELVM v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &o)
+{
+ SQObjectType t;
+ _CHECK_IO(SafeRead(v,read,up,&t,sizeof(SQObjectType)));
+ switch(t){
+ case OT_STRING:{
+ SQInteger len;
+ _CHECK_IO(SafeRead(v,read,up,&len,sizeof(SQInteger)));
+ _CHECK_IO(SafeRead(v,read,up,_ss(v)->GetScratchPad(rsl(len)),rsl(len)));
+ o=SQString::Create(_ss(v),_ss(v)->GetScratchPad(-1),len);
+ }
+ break;
+ case OT_INTEGER:{
+ SQInteger i;
+ _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o = i; break;
+ }
+ case OT_FLOAT:{
+ SQFloat f;
+ _CHECK_IO(SafeRead(v,read,up,&f,sizeof(SQFloat))); o = f; break;
+ }
+ case OT_NULL:
+ o=_null_;
+ break;
+ default:
+ v->Raise_Error(_SC("cannot serialize a %s"),IdType2Name(t));
+ return false;
+ }
+ return true;
+}
+
+bool SQClosure::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
+{
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_HEAD));
+ _CHECK_IO(WriteTag(v,write,up,sizeof(SQChar)));
+ _CHECK_IO(_funcproto(_function)->Save(v,up,write));
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_TAIL));
+ return true;
+}
+
+bool SQClosure::Load(SQVM *v,SQUserPointer up,SQREADFUNC read)
+{
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_HEAD));
+ _CHECK_IO(CheckTag(v,read,up,sizeof(SQChar)));
+ _CHECK_IO(_funcproto(_function)->Load(v,up,read));
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_TAIL));
+ return true;
+}
+
+bool SQFunctionProto::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
+{
+ SQInteger i,nsize=_literals.size();
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(WriteObject(v,up,write,_sourcename));
+ _CHECK_IO(WriteObject(v,up,write,_name));
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ for(i=0;i<nsize;i++){
+ _CHECK_IO(WriteObject(v,up,write,_literals[i]));
+ }
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ nsize=_parameters.size();
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ for(i=0;i<nsize;i++){
+ _CHECK_IO(WriteObject(v,up,write,_parameters[i]));
+ }
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ nsize=_outervalues.size();
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ for(i=0;i<nsize;i++){
+ _CHECK_IO(SafeWrite(v,write,up,&_outervalues[i]._type,sizeof(SQUnsignedInteger)));
+ _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._src));
+ _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._name));
+ }
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ nsize=_localvarinfos.size();
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ for(i=0;i<nsize;i++){
+ SQLocalVarInfo &lvi=_localvarinfos[i];
+ _CHECK_IO(WriteObject(v,up,write,lvi._name));
+ _CHECK_IO(SafeWrite(v,write,up,&lvi._pos,sizeof(SQUnsignedInteger)));
+ _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(SQUnsignedInteger)));
+ _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(SQUnsignedInteger)));
+ }
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ nsize=_lineinfos.size();
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ _CHECK_IO(SafeWrite(v,write,up,&_lineinfos[0],sizeof(SQLineInfo)*nsize));
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ nsize=_instructions.size();
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ _CHECK_IO(SafeWrite(v,write,up,&_instructions[0],sizeof(SQInstruction)*nsize));
+ _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
+ nsize=_functions.size();
+ _CHECK_IO(SafeWrite(v,write,up,&nsize,sizeof(nsize)));
+ for(i=0;i<nsize;i++){
+ _CHECK_IO(_funcproto(_functions[i])->Save(v,up,write));
+ }
+ _CHECK_IO(SafeWrite(v,write,up,&_stacksize,sizeof(_stacksize)));
+ _CHECK_IO(SafeWrite(v,write,up,&_bgenerator,sizeof(_bgenerator)));
+ _CHECK_IO(SafeWrite(v,write,up,&_varparams,sizeof(_varparams)));
+ return true;
+}
+
+bool SQFunctionProto::Load(SQVM *v,SQUserPointer up,SQREADFUNC read)
+{
+ SQInteger i, nsize = _literals.size();
+ SQObjectPtr o;
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(ReadObject(v, up, read, _sourcename));
+ _CHECK_IO(ReadObject(v, up, read, _name));
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));
+ for(i = 0;i < nsize; i++){
+ _CHECK_IO(ReadObject(v, up, read, o));
+ _literals.push_back(o);
+ }
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));
+ for(i = 0; i < nsize; i++){
+ _CHECK_IO(ReadObject(v, up, read, o));
+ _parameters.push_back(o);
+ }
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up,&nsize,sizeof(nsize)));
+ for(i = 0; i < nsize; i++){
+ SQUnsignedInteger type;
+ SQObjectPtr name;
+ _CHECK_IO(SafeRead(v,read,up, &type, sizeof(SQUnsignedInteger)));
+ _CHECK_IO(ReadObject(v, up, read, o));
+ _CHECK_IO(ReadObject(v, up, read, name));
+ _outervalues.push_back(SQOuterVar(name,o, (SQOuterType)type));
+ }
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up,&nsize, sizeof(nsize)));
+ for(i = 0; i < nsize; i++){
+ SQLocalVarInfo lvi;
+ _CHECK_IO(ReadObject(v, up, read, lvi._name));
+ _CHECK_IO(SafeRead(v,read,up, &lvi._pos, sizeof(SQUnsignedInteger)));
+ _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(SQUnsignedInteger)));
+ _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(SQUnsignedInteger)));
+ _localvarinfos.push_back(lvi);
+ }
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up, &nsize,sizeof(nsize)));
+ _lineinfos.resize(nsize);
+ _CHECK_IO(SafeRead(v,read,up, &_lineinfos[0], sizeof(SQLineInfo)*nsize));
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));
+ _instructions.resize(nsize);
+ _CHECK_IO(SafeRead(v,read,up, &_instructions[0], sizeof(SQInstruction)*nsize));
+ _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
+ _CHECK_IO(SafeRead(v,read,up, &nsize, sizeof(nsize)));
+ for(i = 0; i < nsize; i++){
+ o = SQFunctionProto::Create();
+ _CHECK_IO(_funcproto(o)->Load(v, up, read));
+ _functions.push_back(o);
+ }
+ _CHECK_IO(SafeRead(v,read,up, &_stacksize, sizeof(_stacksize)));
+ _CHECK_IO(SafeRead(v,read,up, &_bgenerator, sizeof(_bgenerator)));
+ _CHECK_IO(SafeRead(v,read,up, &_varparams, sizeof(_varparams)));
+ return true;
+}
+
+#ifndef NO_GARBAGE_COLLECTOR
+
+#define START_MARK() if(!(_uiRef&MARK_FLAG)){ \
+ _uiRef|=MARK_FLAG;
+
+#define END_MARK() RemoveFromChain(&_sharedstate->_gc_chain, this); \
+ AddToChain(chain, this); }
+
+void SQVM::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ SQSharedState::MarkObject(_lasterror,chain);
+ SQSharedState::MarkObject(_errorhandler,chain);
+ SQSharedState::MarkObject(_debughook,chain);
+ SQSharedState::MarkObject(_roottable, chain);
+ SQSharedState::MarkObject(temp_reg, chain);
+ for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
+ for(SQUnsignedInteger j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);
+ END_MARK()
+}
+
+void SQArray::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ SQInteger len = _values.size();
+ for(SQInteger i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);
+ END_MARK()
+}
+void SQTable::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ if(_delegate) _delegate->Mark(chain);
+ SQInteger len = _numofnodes;
+ for(SQInteger i = 0; i < len; i++){
+ SQSharedState::MarkObject(_nodes[i].key, chain);
+ SQSharedState::MarkObject(_nodes[i].val, chain);
+ }
+ END_MARK()
+}
+
+void SQClass::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ _members->Mark(chain);
+ if(_base) _base->Mark(chain);
+ SQSharedState::MarkObject(_attributes, chain);
+ for(SQUnsignedInteger i =0; i< _defaultvalues.size(); i++) {
+ SQSharedState::MarkObject(_defaultvalues[i].val, chain);
+ SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);
+ }
+ for(SQUnsignedInteger j =0; j< _methods.size(); j++) {
+ SQSharedState::MarkObject(_methods[j].val, chain);
+ SQSharedState::MarkObject(_methods[j].attrs, chain);
+ }
+ for(SQUnsignedInteger k =0; k< _metamethods.size(); k++) {
+ SQSharedState::MarkObject(_metamethods[k], chain);
+ }
+ END_MARK()
+}
+
+void SQInstance::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ _class->Mark(chain);
+ for(SQUnsignedInteger i =0; i< _nvalues; i++) {
+ SQSharedState::MarkObject(_values[i], chain);
+ }
+ END_MARK()
+}
+
+void SQGenerator::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
+ for(SQUnsignedInteger j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);
+ SQSharedState::MarkObject(_closure, chain);
+ END_MARK()
+}
+
+void SQClosure::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ for(SQUnsignedInteger i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);
+ END_MARK()
+}
+
+void SQNativeClosure::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ for(SQUnsignedInteger i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);
+ END_MARK()
+}
+
+void SQUserData::Mark(SQCollectable **chain){
+ START_MARK()
+ if(_delegate) _delegate->Mark(chain);
+ END_MARK()
+}
+
+void SQCollectable::UnMark() { _uiRef&=~MARK_FLAG; }
+
+#endif
projects / supertux.git / blobdiff