sqobject.cpp 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677
  1. /*
  2. see copyright notice in squirrel.h
  3. */
  4. #include "sqpcheader.h"
  5. #include "sqvm.h"
  6. #include "sqstring.h"
  7. #include "sqarray.h"
  8. #include "sqtable.h"
  9. #include "squserdata.h"
  10. #include "sqfuncproto.h"
  11. #include "sqclass.h"
  12. #include "sqclosure.h"
  13. const SQChar *IdType2Name(SQObjectType type)
  14. {
  15. switch(_RAW_TYPE(type))
  16. {
  17. case _RT_NULL:return _SC("null");
  18. case _RT_INTEGER:return _SC("integer");
  19. case _RT_FLOAT:return _SC("float");
  20. case _RT_BOOL:return _SC("bool");
  21. case _RT_STRING:return _SC("string");
  22. case _RT_TABLE:return _SC("table");
  23. case _RT_ARRAY:return _SC("array");
  24. case _RT_GENERATOR:return _SC("generator");
  25. case _RT_CLOSURE:
  26. case _RT_NATIVECLOSURE:
  27. return _SC("function");
  28. case _RT_USERDATA:
  29. case _RT_USERPOINTER:
  30. return _SC("userdata");
  31. case _RT_THREAD: return _SC("thread");
  32. case _RT_FUNCPROTO: return _SC("function");
  33. case _RT_CLASS: return _SC("class");
  34. case _RT_INSTANCE: return _SC("instance");
  35. case _RT_WEAKREF: return _SC("weakref");
  36. case _RT_OUTER: return _SC("outer");
  37. default:
  38. return NULL;
  39. }
  40. }
  41. const SQChar *GetTypeName(const SQObjectPtr &obj1)
  42. {
  43. return IdType2Name(type(obj1));
  44. }
  45. SQString *SQString::Create(SQSharedState *ss,const SQChar *s,SQInteger len)
  46. {
  47. SQString *str=ADD_STRING(ss,s,len);
  48. return str;
  49. }
  50. void SQString::Release()
  51. {
  52. REMOVE_STRING(_sharedstate,this);
  53. }
  54. SQInteger SQString::Next(const SQObjectPtr &refpos, SQObjectPtr &outkey, SQObjectPtr &outval)
  55. {
  56. SQInteger idx = (SQInteger)TranslateIndex(refpos);
  57. while(idx < _len){
  58. outkey = (SQInteger)idx;
  59. outval = (SQInteger)((SQUnsignedInteger)_val[idx]);
  60. //return idx for the next iteration
  61. return ++idx;
  62. }
  63. //nothing to iterate anymore
  64. return -1;
  65. }
  66. SQUnsignedInteger TranslateIndex(const SQObjectPtr &idx)
  67. {
  68. switch(type(idx)){
  69. case OT_NULL:
  70. return 0;
  71. case OT_INTEGER:
  72. return (SQUnsignedInteger)_integer(idx);
  73. default: assert(0); break;
  74. }
  75. return 0;
  76. }
  77. SQWeakRef *SQRefCounted::GetWeakRef(SQObjectType type)
  78. {
  79. if(!_weakref) {
  80. sq_new(_weakref,SQWeakRef);
  81. #if defined(SQUSEDOUBLE) && !defined(_SQ64)
  82. _weakref->_obj._unVal.raw = 0; //clean the whole union on 32 bits with double
  83. #endif
  84. _weakref->_obj._type = type;
  85. _weakref->_obj._unVal.pRefCounted = this;
  86. }
  87. return _weakref;
  88. }
  89. SQRefCounted::~SQRefCounted()
  90. {
  91. if(_weakref) {
  92. _weakref->_obj._type = OT_NULL;
  93. _weakref->_obj._unVal.pRefCounted = NULL;
  94. }
  95. }
  96. void SQWeakRef::Release() {
  97. if(ISREFCOUNTED(_obj._type)) {
  98. _obj._unVal.pRefCounted->_weakref = NULL;
  99. }
  100. sq_delete(this,SQWeakRef);
  101. }
  102. bool SQDelegable::GetMetaMethod(SQVM *v,SQMetaMethod mm,SQObjectPtr &res) {
  103. if(_delegate) {
  104. return _delegate->Get((*_ss(v)->_metamethods)[mm],res);
  105. }
  106. return false;
  107. }
  108. bool SQDelegable::SetDelegate(SQTable *mt)
  109. {
  110. SQTable *temp = mt;
  111. if(temp == this) return false;
  112. while (temp) {
  113. if (temp->_delegate == this) return false; //cycle detected
  114. temp = temp->_delegate;
  115. }
  116. if (mt) __ObjAddRef(mt);
  117. __ObjRelease(_delegate);
  118. _delegate = mt;
  119. return true;
  120. }
  121. bool SQGenerator::Yield(SQVM *v,SQInteger target)
  122. {
  123. if(_state==eSuspended) { v->Raise_Error(_SC("internal vm error, yielding dead generator")); return false;}
  124. if(_state==eDead) { v->Raise_Error(_SC("internal vm error, yielding a dead generator")); return false; }
  125. SQInteger size = v->_top-v->_stackbase;
  126. _stack.resize(size);
  127. SQObject _this = v->_stack[v->_stackbase];
  128. _stack._vals[0] = ISREFCOUNTED(type(_this)) ? SQObjectPtr(_refcounted(_this)->GetWeakRef(type(_this))) : _this;
  129. for(SQInteger n =1; n<target; n++) {
  130. _stack._vals[n] = v->_stack[v->_stackbase+n];
  131. }
  132. for(SQInteger j =0; j < size; j++)
  133. {
  134. v->_stack[v->_stackbase+j].Null();
  135. }
  136. _ci = *v->ci;
  137. _ci._generator=NULL;
  138. for(SQInteger i=0;i<_ci._etraps;i++) {
  139. _etraps.push_back(v->_etraps.top());
  140. v->_etraps.pop_back();
  141. // store relative stack base and size in case of resume to other _top
  142. SQExceptionTrap &et = _etraps.back();
  143. et._stackbase -= v->_stackbase;
  144. et._stacksize -= v->_stackbase;
  145. }
  146. _state=eSuspended;
  147. return true;
  148. }
  149. bool SQGenerator::Resume(SQVM *v,SQObjectPtr &dest)
  150. {
  151. if(_state==eDead){ v->Raise_Error(_SC("resuming dead generator")); return false; }
  152. if(_state==eRunning){ v->Raise_Error(_SC("resuming active generator")); return false; }
  153. SQInteger size = _stack.size();
  154. SQInteger target = &dest - &(v->_stack._vals[v->_stackbase]);
  155. assert(target>=0 && target<=255);
  156. SQInteger newbase = v->_top;
  157. if(!v->EnterFrame(v->_top, v->_top + size, false))
  158. return false;
  159. v->ci->_generator = this;
  160. v->ci->_target = (SQInt32)target;
  161. v->ci->_closure = _ci._closure;
  162. v->ci->_ip = _ci._ip;
  163. v->ci->_literals = _ci._literals;
  164. v->ci->_ncalls = _ci._ncalls;
  165. v->ci->_etraps = _ci._etraps;
  166. v->ci->_root = _ci._root;
  167. for(SQInteger i=0;i<_ci._etraps;i++) {
  168. v->_etraps.push_back(_etraps.top());
  169. _etraps.pop_back();
  170. SQExceptionTrap &et = v->_etraps.back();
  171. // restore absolute stack base and size
  172. et._stackbase += newbase;
  173. et._stacksize += newbase;
  174. }
  175. SQObject _this = _stack._vals[0];
  176. v->_stack[v->_stackbase] = type(_this) == OT_WEAKREF ? _weakref(_this)->_obj : _this;
  177. for(SQInteger n = 1; n<size; n++) {
  178. v->_stack[v->_stackbase+n] = _stack._vals[n];
  179. _stack._vals[n].Null();
  180. }
  181. _state=eRunning;
  182. if (v->_debughook)
  183. v->CallDebugHook(_SC('c'));
  184. return true;
  185. }
  186. void SQArray::Extend(const SQArray *a){
  187. SQInteger xlen;
  188. if((xlen=a->Size()))
  189. for(SQInteger i=0;i<xlen;i++)
  190. Append(a->_values[i]);
  191. }
  192. const SQChar* SQFunctionProto::GetLocal(SQVM *vm,SQUnsignedInteger stackbase,SQUnsignedInteger nseq,SQUnsignedInteger nop)
  193. {
  194. SQUnsignedInteger nvars=_nlocalvarinfos;
  195. const SQChar *res=NULL;
  196. if(nvars>=nseq){
  197. for(SQUnsignedInteger i=0;i<nvars;i++){
  198. if(_localvarinfos[i]._start_op<=nop && _localvarinfos[i]._end_op>=nop)
  199. {
  200. if(nseq==0){
  201. vm->Push(vm->_stack[stackbase+_localvarinfos[i]._pos]);
  202. res=_stringval(_localvarinfos[i]._name);
  203. break;
  204. }
  205. nseq--;
  206. }
  207. }
  208. }
  209. return res;
  210. }
  211. SQInteger SQFunctionProto::GetLine(SQInstruction *curr)
  212. {
  213. SQInteger op = (SQInteger)(curr-_instructions);
  214. SQInteger line=_lineinfos[0]._line;
  215. SQInteger low = 0;
  216. SQInteger high = _nlineinfos - 1;
  217. SQInteger mid = 0;
  218. while(low <= high)
  219. {
  220. mid = low + ((high - low) >> 1);
  221. SQInteger curop = _lineinfos[mid]._op;
  222. if(curop > op)
  223. {
  224. high = mid - 1;
  225. }
  226. else if(curop < op) {
  227. if(mid < (_nlineinfos - 1)
  228. && _lineinfos[mid + 1]._op >= op) {
  229. break;
  230. }
  231. low = mid + 1;
  232. }
  233. else { //equal
  234. break;
  235. }
  236. }
  237. while(mid > 0 && _lineinfos[mid]._op >= op) mid--;
  238. line = _lineinfos[mid]._line;
  239. return line;
  240. }
  241. SQClosure::~SQClosure()
  242. {
  243. __ObjRelease(_root);
  244. __ObjRelease(_env);
  245. __ObjRelease(_base);
  246. REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
  247. }
  248. #define _CHECK_IO(exp) { if(!exp)return false; }
  249. bool SafeWrite(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUserPointer dest,SQInteger size)
  250. {
  251. if(write(up,dest,size) != size) {
  252. v->Raise_Error(_SC("io error (write function failure)"));
  253. return false;
  254. }
  255. return true;
  256. }
  257. bool SafeRead(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUserPointer dest,SQInteger size)
  258. {
  259. if(size && read(up,dest,size) != size) {
  260. v->Raise_Error(_SC("io error, read function failure, the origin stream could be corrupted/trucated"));
  261. return false;
  262. }
  263. return true;
  264. }
  265. bool WriteTag(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUnsignedInteger32 tag)
  266. {
  267. return SafeWrite(v,write,up,&tag,sizeof(tag));
  268. }
  269. bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUnsignedInteger32 tag)
  270. {
  271. SQUnsignedInteger32 t;
  272. _CHECK_IO(SafeRead(v,read,up,&t,sizeof(t)));
  273. if(t != tag){
  274. v->Raise_Error(_SC("invalid or corrupted closure stream"));
  275. return false;
  276. }
  277. return true;
  278. }
  279. bool WriteObject(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o)
  280. {
  281. SQUnsignedInteger32 _type = (SQUnsignedInteger32)type(o);
  282. _CHECK_IO(SafeWrite(v,write,up,&_type,sizeof(_type)));
  283. switch(type(o)){
  284. case OT_STRING:
  285. _CHECK_IO(SafeWrite(v,write,up,&_string(o)->_len,sizeof(SQInteger)));
  286. _CHECK_IO(SafeWrite(v,write,up,_stringval(o),sq_rsl(_string(o)->_len)));
  287. break;
  288. case OT_BOOL:
  289. case OT_INTEGER:
  290. _CHECK_IO(SafeWrite(v,write,up,&_integer(o),sizeof(SQInteger)));break;
  291. case OT_FLOAT:
  292. _CHECK_IO(SafeWrite(v,write,up,&_float(o),sizeof(SQFloat)));break;
  293. case OT_NULL:
  294. break;
  295. default:
  296. v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
  297. return false;
  298. }
  299. return true;
  300. }
  301. bool ReadObject(HSQUIRRELVM v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &o)
  302. {
  303. SQUnsignedInteger32 _type;
  304. _CHECK_IO(SafeRead(v,read,up,&_type,sizeof(_type)));
  305. SQObjectType t = (SQObjectType)_type;
  306. switch(t){
  307. case OT_STRING:{
  308. SQInteger len;
  309. _CHECK_IO(SafeRead(v,read,up,&len,sizeof(SQInteger)));
  310. _CHECK_IO(SafeRead(v,read,up,_ss(v)->GetScratchPad(sq_rsl(len)),sq_rsl(len)));
  311. o=SQString::Create(_ss(v),_ss(v)->GetScratchPad(-1),len);
  312. }
  313. break;
  314. case OT_INTEGER:{
  315. SQInteger i;
  316. _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o = i; break;
  317. }
  318. case OT_BOOL:{
  319. SQInteger i;
  320. _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o._type = OT_BOOL; o._unVal.nInteger = i; break;
  321. }
  322. case OT_FLOAT:{
  323. SQFloat f;
  324. _CHECK_IO(SafeRead(v,read,up,&f,sizeof(SQFloat))); o = f; break;
  325. }
  326. case OT_NULL:
  327. o.Null();
  328. break;
  329. default:
  330. v->Raise_Error(_SC("cannot serialize a %s"),IdType2Name(t));
  331. return false;
  332. }
  333. return true;
  334. }
  335. bool SQClosure::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  336. {
  337. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_HEAD));
  338. _CHECK_IO(WriteTag(v,write,up,sizeof(SQChar)));
  339. _CHECK_IO(WriteTag(v,write,up,sizeof(SQInteger)));
  340. _CHECK_IO(WriteTag(v,write,up,sizeof(SQFloat)));
  341. _CHECK_IO(_function->Save(v,up,write));
  342. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_TAIL));
  343. return true;
  344. }
  345. bool SQClosure::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
  346. {
  347. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_HEAD));
  348. _CHECK_IO(CheckTag(v,read,up,sizeof(SQChar)));
  349. _CHECK_IO(CheckTag(v,read,up,sizeof(SQInteger)));
  350. _CHECK_IO(CheckTag(v,read,up,sizeof(SQFloat)));
  351. SQObjectPtr func;
  352. _CHECK_IO(SQFunctionProto::Load(v,up,read,func));
  353. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_TAIL));
  354. ret = SQClosure::Create(_ss(v),_funcproto(func),_table(v->_roottable)->GetWeakRef(OT_TABLE));
  355. //FIXME: load an root for this closure
  356. return true;
  357. }
  358. SQFunctionProto::SQFunctionProto(SQSharedState *ss)
  359. {
  360. _stacksize=0;
  361. _bgenerator=false;
  362. INIT_CHAIN();ADD_TO_CHAIN(&_ss(this)->_gc_chain,this);
  363. }
  364. SQFunctionProto::~SQFunctionProto()
  365. {
  366. REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
  367. }
  368. bool SQFunctionProto::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  369. {
  370. SQInteger i,nliterals = _nliterals,nparameters = _nparameters;
  371. SQInteger noutervalues = _noutervalues,nlocalvarinfos = _nlocalvarinfos;
  372. SQInteger nlineinfos=_nlineinfos,ninstructions = _ninstructions,nfunctions=_nfunctions;
  373. SQInteger ndefaultparams = _ndefaultparams;
  374. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  375. _CHECK_IO(WriteObject(v,up,write,_sourcename));
  376. _CHECK_IO(WriteObject(v,up,write,_name));
  377. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  378. _CHECK_IO(SafeWrite(v,write,up,&nliterals,sizeof(nliterals)));
  379. _CHECK_IO(SafeWrite(v,write,up,&nparameters,sizeof(nparameters)));
  380. _CHECK_IO(SafeWrite(v,write,up,&noutervalues,sizeof(noutervalues)));
  381. _CHECK_IO(SafeWrite(v,write,up,&nlocalvarinfos,sizeof(nlocalvarinfos)));
  382. _CHECK_IO(SafeWrite(v,write,up,&nlineinfos,sizeof(nlineinfos)));
  383. _CHECK_IO(SafeWrite(v,write,up,&ndefaultparams,sizeof(ndefaultparams)));
  384. _CHECK_IO(SafeWrite(v,write,up,&ninstructions,sizeof(ninstructions)));
  385. _CHECK_IO(SafeWrite(v,write,up,&nfunctions,sizeof(nfunctions)));
  386. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  387. for(i=0;i<nliterals;i++){
  388. _CHECK_IO(WriteObject(v,up,write,_literals[i]));
  389. }
  390. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  391. for(i=0;i<nparameters;i++){
  392. _CHECK_IO(WriteObject(v,up,write,_parameters[i]));
  393. }
  394. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  395. for(i=0;i<noutervalues;i++){
  396. _CHECK_IO(SafeWrite(v,write,up,&_outervalues[i]._type,sizeof(SQUnsignedInteger)));
  397. _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._src));
  398. _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._name));
  399. }
  400. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  401. for(i=0;i<nlocalvarinfos;i++){
  402. SQLocalVarInfo &lvi=_localvarinfos[i];
  403. _CHECK_IO(WriteObject(v,up,write,lvi._name));
  404. _CHECK_IO(SafeWrite(v,write,up,&lvi._pos,sizeof(SQUnsignedInteger)));
  405. _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(SQUnsignedInteger)));
  406. _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(SQUnsignedInteger)));
  407. }
  408. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  409. _CHECK_IO(SafeWrite(v,write,up,_lineinfos,sizeof(SQLineInfo)*nlineinfos));
  410. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  411. _CHECK_IO(SafeWrite(v,write,up,_defaultparams,sizeof(SQInteger)*ndefaultparams));
  412. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  413. _CHECK_IO(SafeWrite(v,write,up,_instructions,sizeof(SQInstruction)*ninstructions));
  414. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  415. for(i=0;i<nfunctions;i++){
  416. _CHECK_IO(_funcproto(_functions[i])->Save(v,up,write));
  417. }
  418. _CHECK_IO(SafeWrite(v,write,up,&_stacksize,sizeof(_stacksize)));
  419. _CHECK_IO(SafeWrite(v,write,up,&_bgenerator,sizeof(_bgenerator)));
  420. _CHECK_IO(SafeWrite(v,write,up,&_varparams,sizeof(_varparams)));
  421. return true;
  422. }
  423. bool SQFunctionProto::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
  424. {
  425. SQInteger i, nliterals,nparameters;
  426. SQInteger noutervalues ,nlocalvarinfos ;
  427. SQInteger nlineinfos,ninstructions ,nfunctions,ndefaultparams ;
  428. SQObjectPtr sourcename, name;
  429. SQObjectPtr o;
  430. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  431. _CHECK_IO(ReadObject(v, up, read, sourcename));
  432. _CHECK_IO(ReadObject(v, up, read, name));
  433. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  434. _CHECK_IO(SafeRead(v,read,up, &nliterals, sizeof(nliterals)));
  435. _CHECK_IO(SafeRead(v,read,up, &nparameters, sizeof(nparameters)));
  436. _CHECK_IO(SafeRead(v,read,up, &noutervalues, sizeof(noutervalues)));
  437. _CHECK_IO(SafeRead(v,read,up, &nlocalvarinfos, sizeof(nlocalvarinfos)));
  438. _CHECK_IO(SafeRead(v,read,up, &nlineinfos, sizeof(nlineinfos)));
  439. _CHECK_IO(SafeRead(v,read,up, &ndefaultparams, sizeof(ndefaultparams)));
  440. _CHECK_IO(SafeRead(v,read,up, &ninstructions, sizeof(ninstructions)));
  441. _CHECK_IO(SafeRead(v,read,up, &nfunctions, sizeof(nfunctions)));
  442. SQFunctionProto *f = SQFunctionProto::Create(_opt_ss(v),ninstructions,nliterals,nparameters,
  443. nfunctions,noutervalues,nlineinfos,nlocalvarinfos,ndefaultparams);
  444. SQObjectPtr proto = f; //gets a ref in case of failure
  445. f->_sourcename = sourcename;
  446. f->_name = name;
  447. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  448. for(i = 0;i < nliterals; i++){
  449. _CHECK_IO(ReadObject(v, up, read, o));
  450. f->_literals[i] = o;
  451. }
  452. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  453. for(i = 0; i < nparameters; i++){
  454. _CHECK_IO(ReadObject(v, up, read, o));
  455. f->_parameters[i] = o;
  456. }
  457. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  458. for(i = 0; i < noutervalues; i++){
  459. SQUnsignedInteger type;
  460. SQObjectPtr name;
  461. _CHECK_IO(SafeRead(v,read,up, &type, sizeof(SQUnsignedInteger)));
  462. _CHECK_IO(ReadObject(v, up, read, o));
  463. _CHECK_IO(ReadObject(v, up, read, name));
  464. f->_outervalues[i] = SQOuterVar(name,o, (SQOuterType)type);
  465. }
  466. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  467. for(i = 0; i < nlocalvarinfos; i++){
  468. SQLocalVarInfo lvi;
  469. _CHECK_IO(ReadObject(v, up, read, lvi._name));
  470. _CHECK_IO(SafeRead(v,read,up, &lvi._pos, sizeof(SQUnsignedInteger)));
  471. _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(SQUnsignedInteger)));
  472. _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(SQUnsignedInteger)));
  473. f->_localvarinfos[i] = lvi;
  474. }
  475. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  476. _CHECK_IO(SafeRead(v,read,up, f->_lineinfos, sizeof(SQLineInfo)*nlineinfos));
  477. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  478. _CHECK_IO(SafeRead(v,read,up, f->_defaultparams, sizeof(SQInteger)*ndefaultparams));
  479. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  480. _CHECK_IO(SafeRead(v,read,up, f->_instructions, sizeof(SQInstruction)*ninstructions));
  481. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  482. for(i = 0; i < nfunctions; i++){
  483. _CHECK_IO(_funcproto(o)->Load(v, up, read, o));
  484. f->_functions[i] = o;
  485. }
  486. _CHECK_IO(SafeRead(v,read,up, &f->_stacksize, sizeof(f->_stacksize)));
  487. _CHECK_IO(SafeRead(v,read,up, &f->_bgenerator, sizeof(f->_bgenerator)));
  488. _CHECK_IO(SafeRead(v,read,up, &f->_varparams, sizeof(f->_varparams)));
  489. ret = f;
  490. return true;
  491. }
  492. #ifndef NO_GARBAGE_COLLECTOR
  493. #define START_MARK() if(!(_uiRef&MARK_FLAG)){ \
  494. _uiRef|=MARK_FLAG;
  495. #define END_MARK() RemoveFromChain(&_sharedstate->_gc_chain, this); \
  496. AddToChain(chain, this); }
  497. void SQVM::Mark(SQCollectable **chain)
  498. {
  499. START_MARK()
  500. SQSharedState::MarkObject(_lasterror,chain);
  501. SQSharedState::MarkObject(_errorhandler,chain);
  502. SQSharedState::MarkObject(_debughook_closure,chain);
  503. SQSharedState::MarkObject(_roottable, chain);
  504. SQSharedState::MarkObject(temp_reg, chain);
  505. for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
  506. for(SQInteger k = 0; k < _callsstacksize; k++) SQSharedState::MarkObject(_callsstack[k]._closure, chain);
  507. END_MARK()
  508. }
  509. void SQArray::Mark(SQCollectable **chain)
  510. {
  511. START_MARK()
  512. SQInteger len = _values.size();
  513. for(SQInteger i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);
  514. END_MARK()
  515. }
  516. void SQTable::Mark(SQCollectable **chain)
  517. {
  518. START_MARK()
  519. if(_delegate) _delegate->Mark(chain);
  520. SQInteger len = _numofnodes;
  521. for(SQInteger i = 0; i < len; i++){
  522. SQSharedState::MarkObject(_nodes[i].key, chain);
  523. SQSharedState::MarkObject(_nodes[i].val, chain);
  524. }
  525. END_MARK()
  526. }
  527. void SQClass::Mark(SQCollectable **chain)
  528. {
  529. START_MARK()
  530. _members->Mark(chain);
  531. if(_base) _base->Mark(chain);
  532. SQSharedState::MarkObject(_attributes, chain);
  533. for(SQUnsignedInteger i =0; i< _defaultvalues.size(); i++) {
  534. SQSharedState::MarkObject(_defaultvalues[i].val, chain);
  535. SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);
  536. }
  537. for(SQUnsignedInteger j =0; j< _methods.size(); j++) {
  538. SQSharedState::MarkObject(_methods[j].val, chain);
  539. SQSharedState::MarkObject(_methods[j].attrs, chain);
  540. }
  541. for(SQUnsignedInteger k =0; k< MT_LAST; k++) {
  542. SQSharedState::MarkObject(_metamethods[k], chain);
  543. }
  544. END_MARK()
  545. }
  546. void SQInstance::Mark(SQCollectable **chain)
  547. {
  548. START_MARK()
  549. _class->Mark(chain);
  550. SQUnsignedInteger nvalues = _class->_defaultvalues.size();
  551. for(SQUnsignedInteger i =0; i< nvalues; i++) {
  552. SQSharedState::MarkObject(_values[i], chain);
  553. }
  554. END_MARK()
  555. }
  556. void SQGenerator::Mark(SQCollectable **chain)
  557. {
  558. START_MARK()
  559. for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
  560. SQSharedState::MarkObject(_closure, chain);
  561. END_MARK()
  562. }
  563. void SQFunctionProto::Mark(SQCollectable **chain)
  564. {
  565. START_MARK()
  566. for(SQInteger i = 0; i < _nliterals; i++) SQSharedState::MarkObject(_literals[i], chain);
  567. for(SQInteger k = 0; k < _nfunctions; k++) SQSharedState::MarkObject(_functions[k], chain);
  568. END_MARK()
  569. }
  570. void SQClosure::Mark(SQCollectable **chain)
  571. {
  572. START_MARK()
  573. if(_base) _base->Mark(chain);
  574. SQFunctionProto *fp = _function;
  575. fp->Mark(chain);
  576. for(SQInteger i = 0; i < fp->_noutervalues; i++) SQSharedState::MarkObject(_outervalues[i], chain);
  577. for(SQInteger k = 0; k < fp->_ndefaultparams; k++) SQSharedState::MarkObject(_defaultparams[k], chain);
  578. END_MARK()
  579. }
  580. void SQNativeClosure::Mark(SQCollectable **chain)
  581. {
  582. START_MARK()
  583. for(SQUnsignedInteger i = 0; i < _noutervalues; i++) SQSharedState::MarkObject(_outervalues[i], chain);
  584. END_MARK()
  585. }
  586. void SQOuter::Mark(SQCollectable **chain)
  587. {
  588. START_MARK()
  589. /* If the valptr points to a closed value, that value is alive */
  590. if(_valptr == &_value) {
  591. SQSharedState::MarkObject(_value, chain);
  592. }
  593. END_MARK()
  594. }
  595. void SQUserData::Mark(SQCollectable **chain){
  596. START_MARK()
  597. if(_delegate) _delegate->Mark(chain);
  598. END_MARK()
  599. }
  600. void SQCollectable::UnMark() { _uiRef&=~MARK_FLAG; }
  601. #endif