-/*\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,int 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
-unsigned int TranslateIndex(const SQObjectPtr &idx)\r
-{\r
- switch(type(idx)){\r
- case OT_NULL:\r
- return 0;\r
- case OT_INTEGER:\r
- return (unsigned int)_integer(idx);\r
- }\r
- assert(0);\r
- return 0;\r
-}\r
-\r
-bool SQDelegable::GetMetaMethod(SQMetaMethod mm,SQObjectPtr &res) {\r
- if(_delegate) {\r
- return _delegate->Get((*_ss(this)->_metamethods)[mm],res);\r
- }\r
- return false;\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
- int size = v->_top-v->_stackbase;\r
- _ci=*v->ci;\r
- _stack.resize(size);\r
- for(int n =0; n<size; n++) {\r
- _stack._vals[n] = v->_stack[v->_stackbase+n];\r
- v->_stack[v->_stackbase+n] = _null_;\r
- }\r
- int nvargs = v->ci->_vargs.size;\r
- int vargsbase = v->ci->_vargs.base;\r
- for(int j = nvargs - 1; j >= 0; j--) {\r
- _vargsstack.push_back(v->_vargsstack[vargsbase+j]);\r
- }\r
- _ci._generator=_null_;\r
- for(int 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,int target)\r
-{\r
- int 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
- int prevtop=v->_top-v->_stackbase;\r
- PUSH_CALLINFO(v,_ci);\r
- int 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(int i=0;i<_ci._etraps;i++) {\r
- v->_etraps.push_back(_etraps.top());\r
- _etraps.pop_back();\r
- }\r
- for(int 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
- int xlen;\r
- if((xlen=a->Size()))\r
- for(int i=0;i<xlen;i++)\r
- Append(a->_values[i]);\r
-}\r
-\r
-const SQChar* SQFunctionProto::GetLocal(SQVM *vm,unsigned int stackbase,unsigned int nseq,unsigned int nop)\r
-{\r
- unsigned int nvars=_localvarinfos.size();\r
- const SQChar *res=NULL; \r
- if(nvars>=nseq){\r
- for(unsigned int 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
-int SQFunctionProto::GetLine(SQInstruction *curr)\r
-{\r
- int op=(curr-_instructions._vals);\r
- int line=_lineinfos[0]._line;\r
- for(unsigned int 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,int 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,int 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,int tag)\r
-{\r
- return SafeWrite(v,write,up,&tag,sizeof(tag));\r
-}\r
-\r
-bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,int tag)\r
-{\r
- int 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
- int 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
- int 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(unsigned int)));\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(unsigned int)));\r
- _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(unsigned int)));\r
- _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(unsigned int)));\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
- int 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
- unsigned int type;\r
- SQObjectPtr name;\r
- _CHECK_IO(SafeRead(v,read,up, &type, sizeof(unsigned int)));\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(unsigned int)));\r
- _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(unsigned int)));\r
- _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(unsigned int)));\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(unsigned int i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);\r
- for(unsigned int 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
- int len = _values.size();\r
- for(int 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
- int len = _numofnodes;\r
- for(int 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(unsigned int i =0; i< _defaultvalues.size(); i++) {\r
- SQSharedState::MarkObject(_defaultvalues[i].val, chain);\r
- SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);\r
- }\r
- for(unsigned int j =0; j< _methods.size(); j++) {\r
- SQSharedState::MarkObject(_methods[j].val, chain);\r
- SQSharedState::MarkObject(_methods[j].attrs, chain);\r
- }\r
- for(unsigned int 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(unsigned int i =0; i< _values.size(); 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(unsigned int i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);\r
- for(unsigned int 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(unsigned int 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(unsigned int 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,int len)
+{
+ SQString *str=ADD_STRING(ss,s,len);
+ str->_sharedstate=ss;
+ return str;
+}
+
+void SQString::Release()
+{
+ REMOVE_STRING(_sharedstate,this);
+}
+
+unsigned int TranslateIndex(const SQObjectPtr &idx)
+{
+ switch(type(idx)){
+ case OT_NULL:
+ return 0;
+ case OT_INTEGER:
+ return (unsigned int)_integer(idx);
+ default:
+ break;
+ }
+ assert(0);
+ return 0;
+}
+
+bool SQDelegable::GetMetaMethod(SQMetaMethod mm,SQObjectPtr &res) {
+ if(_delegate) {
+ return _delegate->Get((*_ss(this)->_metamethods)[mm],res);
+ }
+ return false;
+}
+
+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; }
+ int size = v->_top-v->_stackbase;
+ _ci=*v->ci;
+ _stack.resize(size);
+ for(int n =0; n<size; n++) {
+ _stack._vals[n] = v->_stack[v->_stackbase+n];
+ v->_stack[v->_stackbase+n] = _null_;
+ }
+ int nvargs = v->ci->_vargs.size;
+ int vargsbase = v->ci->_vargs.base;
+ for(int j = nvargs - 1; j >= 0; j--) {
+ _vargsstack.push_back(v->_vargsstack[vargsbase+j]);
+ }
+ _ci._generator=_null_;
+ for(int 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,int target)
+{
+ int 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; }
+ int prevtop=v->_top-v->_stackbase;
+ PUSH_CALLINFO(v,_ci);
+ int oldstackbase=v->_stackbase;
+ v->_stackbase=v->_top;
+ v->ci->_target=target;
+ v->ci->_generator=SQObjectPtr(this);
+ v->ci->_vargs.size = _vargsstack.size();
+
+ for(int i=0;i<_ci._etraps;i++) {
+ v->_etraps.push_back(_etraps.top());
+ _etraps.pop_back();
+ }
+ for(int 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){
+ int xlen;
+ if((xlen=a->Size()))
+ for(int i=0;i<xlen;i++)
+ Append(a->_values[i]);
+}
+
+const SQChar* SQFunctionProto::GetLocal(SQVM *vm,unsigned int stackbase,unsigned int nseq,unsigned int nop)
+{
+ unsigned int nvars=_localvarinfos.size();
+ const SQChar *res=NULL;
+ if(nvars>=nseq){
+ for(unsigned int 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;
+}
+
+int SQFunctionProto::GetLine(SQInstruction *curr)
+{
+ int op=(curr-_instructions._vals);
+ int line=_lineinfos[0]._line;
+ for(unsigned int 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,int 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,int 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,int tag)
+{
+ return SafeWrite(v,write,up,&tag,sizeof(tag));
+}
+
+bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,int tag)
+{
+ int 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:{
+ int 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)
+{
+ int 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(unsigned int)));
+ _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(unsigned int)));
+ _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(unsigned int)));
+ _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(unsigned int)));
+ }
+ _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)
+{
+ int 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++){
+ unsigned int type;
+ SQObjectPtr name;
+ _CHECK_IO(SafeRead(v,read,up, &type, sizeof(unsigned int)));
+ _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(unsigned int)));
+ _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(unsigned int)));
+ _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(unsigned int)));
+ _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(unsigned int i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
+ for(unsigned int j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);
+ END_MARK()
+}
+
+void SQArray::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ int len = _values.size();
+ for(int i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);
+ END_MARK()
+}
+void SQTable::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ if(_delegate) _delegate->Mark(chain);
+ int len = _numofnodes;
+ for(int 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(unsigned int i =0; i< _defaultvalues.size(); i++) {
+ SQSharedState::MarkObject(_defaultvalues[i].val, chain);
+ SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);
+ }
+ for(unsigned int j =0; j< _methods.size(); j++) {
+ SQSharedState::MarkObject(_methods[j].val, chain);
+ SQSharedState::MarkObject(_methods[j].attrs, chain);
+ }
+ for(unsigned int k =0; k< _metamethods.size(); k++) {
+ SQSharedState::MarkObject(_metamethods[k], chain);
+ }
+ END_MARK()
+}
+
+void SQInstance::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ _class->Mark(chain);
+ for(unsigned int i =0; i< _values.size(); i++) {
+ SQSharedState::MarkObject(_values[i], chain);
+ }
+ END_MARK()
+}
+
+void SQGenerator::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ for(unsigned int i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
+ for(unsigned int 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(unsigned int i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);
+ END_MARK()
+}
+
+void SQNativeClosure::Mark(SQCollectable **chain)
+{
+ START_MARK()
+ for(unsigned int 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
+