sqobject.cpp 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072
  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(sq_type(obj1));
  44. }
  45. SQObjectPtr SQObjectPtr::operator[](SQInteger nidx) {
  46. SQObjectPtr val;
  47. if(sq_type(*this) == OT_ARRAY) _array(*this)->Get(nidx, val);
  48. return val;
  49. }
  50. SQObjectPtr SQObjectPtr::operator[](const SQChar *key) {
  51. SQObjectPtr val;
  52. switch(sq_type(*this))
  53. {
  54. case OT_TABLE: _table(*this)->Get(key, val); break;
  55. case OT_CLASS: _class(*this)->Get(key, val); break;
  56. case OT_INSTANCE: _instance(*this)->Get(key, val); break;
  57. default:
  58. break;
  59. }
  60. return val;
  61. }
  62. SQString *SQString::Create(SQSharedState *ss,const SQChar *s,SQInteger len)
  63. {
  64. SQString *str=ADD_STRING(ss,s,len);
  65. return str;
  66. }
  67. void SQString::Release()
  68. {
  69. REMOVE_STRING(_sharedstate,this);
  70. }
  71. SQInteger SQString::Next(const SQObjectPtr &refpos, SQObjectPtr &outkey, SQObjectPtr &outval)
  72. {
  73. SQInteger idx = (SQInteger)SQTranslateIndex(refpos);
  74. while(idx < _len){
  75. outkey = (SQInteger)idx;
  76. outval = (SQInteger)((SQUnsignedInteger)_val[idx]);
  77. //return idx for the next iteration
  78. return ++idx;
  79. }
  80. //nothing to iterate anymore
  81. return -1;
  82. }
  83. SQUnsignedInteger SQTranslateIndex(const SQObjectPtr &idx)
  84. {
  85. switch(sq_type(idx)){
  86. case OT_NULL:
  87. return 0;
  88. case OT_INTEGER:
  89. return (SQUnsignedInteger)_integer(idx);
  90. default: assert(0); break;
  91. }
  92. return 0;
  93. }
  94. SQWeakRef *SQRefCounted::GetWeakRef(SQObjectType type)
  95. {
  96. if(!_weakref) {
  97. sq_new(_weakref,SQWeakRef);
  98. #if defined(SQUSEDOUBLE) && !defined(_SQ64)
  99. _weakref->_obj._unVal.raw = 0; //clean the whole union on 32 bits with double
  100. #endif
  101. _weakref->_obj._type = type;
  102. _weakref->_obj._unVal.pRefCounted = this;
  103. }
  104. return _weakref;
  105. }
  106. SQRefCounted::~SQRefCounted()
  107. {
  108. if(_weakref) {
  109. _weakref->_obj._type = OT_NULL;
  110. _weakref->_obj._unVal.pRefCounted = NULL;
  111. }
  112. }
  113. void SQWeakRef::Release() {
  114. if(ISREFCOUNTED(_obj._type)) {
  115. _obj._unVal.pRefCounted->_weakref = NULL;
  116. }
  117. sq_delete(this,SQWeakRef);
  118. }
  119. bool SQDelegable::GetMetaMethod(SQVM *v,SQMetaMethod mm,SQObjectPtr &res) {
  120. if(_delegate) {
  121. return _delegate->Get((*_ss(v)->_metamethods)[mm],res);
  122. }
  123. return false;
  124. }
  125. bool SQDelegable::SetDelegate(SQTable *mt)
  126. {
  127. SQTable *temp = mt;
  128. if(temp == this) return false;
  129. while (temp) {
  130. if (temp->_delegate == this) return false; //cycle detected
  131. temp = temp->_delegate;
  132. }
  133. if (mt) __ObjAddRef(mt);
  134. __ObjRelease(_delegate);
  135. _delegate = mt;
  136. return true;
  137. }
  138. bool SQGenerator::Yield(SQVM *v,SQInteger target)
  139. {
  140. if(_state==eSuspended) { v->Raise_Error(_SC("internal vm error, yielding dead generator")); return false;}
  141. if(_state==eDead) { v->Raise_Error(_SC("internal vm error, yielding a dead generator")); return false; }
  142. SQInteger size = v->_top-v->_stackbase;
  143. _stack.resize(size);
  144. SQObject _this = v->_stack[v->_stackbase];
  145. _stack._vals[0] = ISREFCOUNTED(sq_type(_this)) ? SQObjectPtr(_refcounted(_this)->GetWeakRef(sq_type(_this))) : _this;
  146. for(SQInteger n =1; n<target; n++) {
  147. _stack._vals[n] = v->_stack[v->_stackbase+n];
  148. }
  149. for(SQInteger j =0; j < size; j++)
  150. {
  151. v->_stack[v->_stackbase+j].Null();
  152. }
  153. _ci = *v->ci;
  154. _ci._generator=NULL;
  155. for(SQInteger i=0;i<_ci._etraps;i++) {
  156. _etraps.push_back(v->_etraps.top());
  157. v->_etraps.pop_back();
  158. // store relative stack base and size in case of resume to other _top
  159. SQExceptionTrap &et = _etraps.back();
  160. et._stackbase -= v->_stackbase;
  161. et._stacksize -= v->_stackbase;
  162. }
  163. _state=eSuspended;
  164. return true;
  165. }
  166. bool SQGenerator::Resume(SQVM *v,SQObjectPtr &dest)
  167. {
  168. if(_state==eDead){ v->Raise_Error(_SC("resuming dead generator")); return false; }
  169. if(_state==eRunning){ v->Raise_Error(_SC("resuming active generator")); return false; }
  170. SQInteger size = _stack.size();
  171. SQInteger target = &dest - &(v->_stack._vals[v->_stackbase]);
  172. assert(target>=0 && target<=255);
  173. SQInteger newbase = v->_top;
  174. if(!v->EnterFrame(v->_top, v->_top + size, false))
  175. return false;
  176. v->ci->_generator = this;
  177. v->ci->_target = (SQInt32)target;
  178. v->ci->_closure = _ci._closure;
  179. v->ci->_ip = _ci._ip;
  180. v->ci->_literals = _ci._literals;
  181. v->ci->_ncalls = _ci._ncalls;
  182. v->ci->_etraps = _ci._etraps;
  183. v->ci->_root = _ci._root;
  184. for(SQInteger i=0;i<_ci._etraps;i++) {
  185. v->_etraps.push_back(_etraps.top());
  186. _etraps.pop_back();
  187. SQExceptionTrap &et = v->_etraps.back();
  188. // restore absolute stack base and size
  189. et._stackbase += newbase;
  190. et._stacksize += newbase;
  191. }
  192. SQObject _this = _stack._vals[0];
  193. v->_stack[v->_stackbase] = sq_type(_this) == OT_WEAKREF ? _weakref(_this)->_obj : _this;
  194. for(SQInteger n = 1; n<size; n++) {
  195. v->_stack[v->_stackbase+n] = _stack._vals[n];
  196. _stack._vals[n].Null();
  197. }
  198. _state=eRunning;
  199. if (v->_debughook)
  200. v->CallDebugHook(_SC('c'));
  201. return true;
  202. }
  203. void SQArray::Extend(const SQArray *a){
  204. SQInteger xlen;
  205. if((xlen=a->Size()))
  206. for(SQInteger i=0;i<xlen;i++)
  207. Append(a->_values[i]);
  208. }
  209. const SQChar* SQFunctionProto::GetLocal(SQVM *vm,SQUnsignedInteger stackbase,SQUnsignedInteger nseq,SQUnsignedInteger nop)
  210. {
  211. SQUnsignedInteger nvars=_nlocalvarinfos;
  212. const SQChar *res=NULL;
  213. if(nvars>=nseq){
  214. for(SQUnsignedInteger i=0;i<nvars;i++){
  215. if(_localvarinfos[i]._start_op<=nop && _localvarinfos[i]._end_op>=nop)
  216. {
  217. if(nseq==0){
  218. vm->Push(vm->_stack[stackbase+_localvarinfos[i]._pos]);
  219. res=_stringval(_localvarinfos[i]._name);
  220. break;
  221. }
  222. nseq--;
  223. }
  224. }
  225. }
  226. return res;
  227. }
  228. SQInteger SQFunctionProto::GetLine(SQInstruction *curr)
  229. {
  230. SQInteger op = (SQInteger)(curr-_instructions);
  231. SQInteger line=_lineinfos[0]._line;
  232. SQInteger low = 0;
  233. SQInteger high = _nlineinfos - 1;
  234. SQInteger mid = 0;
  235. while(low <= high)
  236. {
  237. mid = low + ((high - low) >> 1);
  238. SQInteger curop = _lineinfos[mid]._op;
  239. if(curop > op)
  240. {
  241. high = mid - 1;
  242. }
  243. else if(curop < op) {
  244. if(mid < (_nlineinfos - 1)
  245. && _lineinfos[mid + 1]._op >= op) {
  246. break;
  247. }
  248. low = mid + 1;
  249. }
  250. else { //equal
  251. break;
  252. }
  253. }
  254. while(mid > 0 && _lineinfos[mid]._op >= op) mid--; //I've changed this one
  255. line = _lineinfos[mid]._line;
  256. return line;
  257. }
  258. SQClosure::~SQClosure()
  259. {
  260. __ObjRelease(_root);
  261. __ObjRelease(_env);
  262. __ObjRelease(_base);
  263. REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
  264. }
  265. #define _CHECK_IO(exp) { if(!exp)return false; }
  266. static bool SafeWrite(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUserPointer dest,SQInteger size)
  267. {
  268. if(write(up,dest,size) != size) {
  269. v->Raise_Error(_SC("io error (write function failure)"));
  270. return false;
  271. }
  272. return true;
  273. }
  274. static bool SafeWriteFmt(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up, const SQChar *fmt, ...)
  275. {
  276. if(fmt){
  277. SQChar str[8192];
  278. va_list vl;
  279. va_start(vl, fmt);
  280. SQInteger len = scvsprintf(str, sizeof(str), fmt, vl);
  281. va_end(vl);
  282. return SafeWrite(v, write, up, str, len);
  283. }
  284. return false;
  285. }
  286. static bool SafeRead(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUserPointer dest,SQInteger size)
  287. {
  288. if(size && read(up,dest,size) != size) {
  289. v->Raise_Error(_SC("io error, read function failure, the origin stream could be corrupted/trucated"));
  290. return false;
  291. }
  292. return true;
  293. }
  294. static bool WriteTag(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUnsignedInteger32 tag)
  295. {
  296. return SafeWrite(v,write,up,&tag,sizeof(tag));
  297. }
  298. static bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUnsignedInteger32 tag)
  299. {
  300. SQUnsignedInteger32 t;
  301. _CHECK_IO(SafeRead(v,read,up,&t,sizeof(t)));
  302. if(t != tag){
  303. v->Raise_Error(_SC("invalid or corrupted closure stream"));
  304. return false;
  305. }
  306. return true;
  307. }
  308. static bool WriteObjectAsCode(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o, bool withQuotes=true)
  309. {
  310. SQChar buf[32];
  311. SQInteger sz;
  312. switch(sq_type(o)){
  313. case OT_STRING:{
  314. SQInteger str_size = _string(o)->_len;
  315. if(str_size){
  316. const SQChar *str = _stringval(o);
  317. bool needQuote = false;
  318. for(int i=0; i < str_size; ++i){
  319. if(str[i] == '"' || str[i] == '\n'){
  320. needQuote = true;
  321. break;
  322. }
  323. }
  324. if(!needQuote) withQuotes = true;
  325. const SQChar *d1 = withQuotes ? _SC("\"") : _SC("[==[");
  326. const SQChar *d2 = withQuotes ? _SC("\"") : _SC("]==]");
  327. _CHECK_IO(SafeWrite(v,write,up, (void*)d1, scstrlen(d1)));
  328. _CHECK_IO(SafeWrite(v,write,up, (void*)str,rsl(_string(o)->_len)));
  329. _CHECK_IO(SafeWrite(v,write,up, (void*)d2, scstrlen(d2)));
  330. }
  331. else
  332. {
  333. const SQChar *str = _SC("null");
  334. _CHECK_IO(SafeWrite(v,write,up, (void*)str, scstrlen(str)));
  335. }
  336. }
  337. break;
  338. case OT_INTEGER:
  339. sz = scsprintf(buf, sizeof(buf), _PRINT_INT_FMT, _integer(o));
  340. _CHECK_IO(SafeWrite(v,write,up,buf,sz));break;
  341. case OT_FLOAT:
  342. sz = scsprintf(buf, sizeof(buf), "%f", _float(o));
  343. _CHECK_IO(SafeWrite(v,write,up,buf,sz));break;
  344. case OT_NULL:{
  345. const SQChar str[] = _SC("null");
  346. _CHECK_IO(SafeWrite(v,write,up, (void*)str, scstrlen(str)));
  347. }
  348. break;
  349. default:
  350. v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
  351. return false;
  352. }
  353. return true;
  354. }
  355. static bool WriteObject(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o)
  356. {
  357. SQUnsignedInteger32 _type = (SQUnsignedInteger32)sq_type(o);
  358. _CHECK_IO(SafeWrite(v,write,up,&_type,sizeof(_type)));
  359. switch(sq_type(o)){
  360. case OT_STRING:
  361. _CHECK_IO(SafeWrite(v,write,up,&_string(o)->_len,sizeof(SQInteger)));
  362. _CHECK_IO(SafeWrite(v,write,up,_stringval(o),rsl(_string(o)->_len)));
  363. break;
  364. case OT_BOOL:
  365. case OT_INTEGER:
  366. _CHECK_IO(SafeWrite(v,write,up,&_integer(o),sizeof(SQInteger)));break;
  367. case OT_FLOAT:
  368. _CHECK_IO(SafeWrite(v,write,up,&_float(o),sizeof(SQFloat)));break;
  369. case OT_NULL:
  370. break;
  371. default:
  372. v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
  373. return false;
  374. }
  375. return true;
  376. }
  377. static bool ReadObject(HSQUIRRELVM v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &o)
  378. {
  379. SQUnsignedInteger32 _type;
  380. _CHECK_IO(SafeRead(v,read,up,&_type,sizeof(_type)));
  381. SQObjectType t = (SQObjectType)_type;
  382. switch(t){
  383. case OT_STRING:{
  384. SQInteger len;
  385. _CHECK_IO(SafeRead(v,read,up,&len,sizeof(SQInteger)));
  386. _CHECK_IO(SafeRead(v,read,up,_ss(v)->GetScratchPad(sq_rsl(len)),sq_rsl(len)));
  387. o=SQString::Create(_ss(v),_ss(v)->GetScratchPad(-1),len);
  388. }
  389. break;
  390. case OT_INTEGER:{
  391. SQInteger i;
  392. _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o = i; break;
  393. }
  394. case OT_BOOL:{
  395. SQInteger i;
  396. _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o._type = OT_BOOL; o._unVal.nInteger = i; break;
  397. }
  398. case OT_FLOAT:{
  399. SQFloat f;
  400. _CHECK_IO(SafeRead(v,read,up,&f,sizeof(SQFloat))); o = f; break;
  401. }
  402. case OT_NULL:
  403. o.Null();
  404. break;
  405. default:
  406. v->Raise_Error(_SC("cannot serialize a %s"),IdType2Name(t));
  407. return false;
  408. }
  409. return true;
  410. }
  411. bool SQClosure::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  412. {
  413. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_HEAD));
  414. _CHECK_IO(WriteTag(v,write,up,sizeof(SQChar)));
  415. _CHECK_IO(WriteTag(v,write,up,sizeof(SQInteger)));
  416. _CHECK_IO(WriteTag(v,write,up,sizeof(SQFloat)));
  417. _CHECK_IO(_function->Save(v,up,write));
  418. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_TAIL));
  419. return true;
  420. }
  421. bool SQClosure::SaveAsSource(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  422. {
  423. _CHECK_IO(_function->SaveAsSource(v,up,write));
  424. return true;
  425. }
  426. bool SQClosure::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
  427. {
  428. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_HEAD));
  429. _CHECK_IO(CheckTag(v,read,up,sizeof(SQChar)));
  430. _CHECK_IO(CheckTag(v,read,up,sizeof(SQInteger)));
  431. _CHECK_IO(CheckTag(v,read,up,sizeof(SQFloat)));
  432. SQObjectPtr func;
  433. _CHECK_IO(SQFunctionProto::Load(v,up,read,func));
  434. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_TAIL));
  435. ret = SQClosure::Create(_ss(v),_funcproto(func),_table(v->_roottable)->GetWeakRef(OT_TABLE));
  436. //FIXME: load an root for this closure
  437. return true;
  438. }
  439. SQFunctionProto::SQFunctionProto(SQSharedState *ss)
  440. {
  441. _stacksize=0;
  442. _bgenerator=false;
  443. INIT_CHAIN();ADD_TO_CHAIN(&_ss(this)->_gc_chain,this);
  444. }
  445. SQFunctionProto::~SQFunctionProto()
  446. {
  447. REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
  448. }
  449. bool SQFunctionProto::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  450. {
  451. SQInteger i,nliterals = _nliterals,nparameters = _nparameters;
  452. SQInteger noutervalues = _noutervalues,nlocalvarinfos = _nlocalvarinfos;
  453. SQInteger nlineinfos=_nlineinfos,ninstructions = _ninstructions,nfunctions=_nfunctions;
  454. SQInteger ndefaultparams = _ndefaultparams;
  455. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  456. _CHECK_IO(WriteObject(v,up,write,_sourcename));
  457. _CHECK_IO(WriteObject(v,up,write,_name));
  458. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  459. _CHECK_IO(SafeWrite(v,write,up,&nliterals,sizeof(nliterals)));
  460. _CHECK_IO(SafeWrite(v,write,up,&nparameters,sizeof(nparameters)));
  461. _CHECK_IO(SafeWrite(v,write,up,&noutervalues,sizeof(noutervalues)));
  462. _CHECK_IO(SafeWrite(v,write,up,&nlocalvarinfos,sizeof(nlocalvarinfos)));
  463. _CHECK_IO(SafeWrite(v,write,up,&nlineinfos,sizeof(nlineinfos)));
  464. _CHECK_IO(SafeWrite(v,write,up,&ndefaultparams,sizeof(ndefaultparams)));
  465. _CHECK_IO(SafeWrite(v,write,up,&ninstructions,sizeof(ninstructions)));
  466. _CHECK_IO(SafeWrite(v,write,up,&nfunctions,sizeof(nfunctions)));
  467. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  468. for(i=0;i<nliterals;i++){
  469. _CHECK_IO(WriteObject(v,up,write,_literals[i]));
  470. }
  471. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  472. for(i=0;i<nparameters;i++){
  473. _CHECK_IO(WriteObject(v,up,write,_parameters[i]));
  474. }
  475. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  476. for(i=0;i<noutervalues;i++){
  477. _CHECK_IO(SafeWrite(v,write,up,&_outervalues[i]._type,sizeof(SQUnsignedInteger)));
  478. _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._src));
  479. _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._name));
  480. }
  481. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  482. for(i=0;i<nlocalvarinfos;i++){
  483. SQLocalVarInfo &lvi=_localvarinfos[i];
  484. _CHECK_IO(WriteObject(v,up,write,lvi._name));
  485. _CHECK_IO(SafeWrite(v,write,up,&lvi._pos,sizeof(SQUnsignedInteger)));
  486. _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(SQUnsignedInteger)));
  487. _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(SQUnsignedInteger)));
  488. }
  489. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  490. _CHECK_IO(SafeWrite(v,write,up,_lineinfos,sizeof(SQLineInfo)*nlineinfos));
  491. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  492. _CHECK_IO(SafeWrite(v,write,up,_defaultparams,sizeof(SQInteger)*ndefaultparams));
  493. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  494. _CHECK_IO(SafeWrite(v,write,up,_instructions,sizeof(SQInstruction)*ninstructions));
  495. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  496. for(i=0;i<nfunctions;i++){
  497. _CHECK_IO(_funcproto(_functions[i])->Save(v,up,write));
  498. }
  499. _CHECK_IO(SafeWrite(v,write,up,&_stacksize,sizeof(_stacksize)));
  500. _CHECK_IO(SafeWrite(v,write,up,&_bgenerator,sizeof(_bgenerator)));
  501. _CHECK_IO(SafeWrite(v,write,up,&_varparams,sizeof(_varparams)));
  502. return true;
  503. }
  504. const SQChar *SQGetBitwiseOpName(int it){
  505. #define SCASE(x) case x: return _SC(#x); break;
  506. switch(it){
  507. SCASE(BW_AND)
  508. SCASE(BW_OR)
  509. SCASE(BW_XOR)
  510. SCASE(BW_SHIFTL)
  511. SCASE(BW_SHIFTR)
  512. SCASE(BW_USHIFTR)
  513. default: return _SC("?");
  514. }
  515. #undef SCASE
  516. }
  517. const SQChar *SQGetCmpOpName(int it){
  518. #define SCASE(x) case x: return _SC(#x); break;
  519. switch(it){
  520. SCASE(CMP_G)
  521. SCASE(CMP_GE)
  522. SCASE(CMP_L)
  523. SCASE(CMP_LE)
  524. SCASE(CMP_3W)
  525. default: return _SC("?");
  526. }
  527. #undef SCASE
  528. }
  529. const SQChar *SQGetArrayAppendTypeName(int it){
  530. #define SCASE(x) case x: return _SC(#x); break;
  531. switch(it){
  532. SCASE(AAT_STACK)
  533. SCASE(AAT_LITERAL)
  534. SCASE(AAT_INT)
  535. SCASE(AAT_FLOAT)
  536. SCASE(AAT_BOOL)
  537. default: return _SC("?");
  538. }
  539. #undef SCASE
  540. }
  541. const SQChar *SQGetNewObjTypeName(int it){
  542. #define SCASE(x) case x: return _SC(#x); break;
  543. switch(it){
  544. SCASE(NOT_TABLE)
  545. SCASE(NOT_ARRAY)
  546. SCASE(NOT_CLASS)
  547. default: return _SC("?");
  548. }
  549. #undef SCASE
  550. }
  551. const SQChar *SQGetArithOpName(int it){
  552. #define SCASE(x, z) case x: return _SC(#z); break;
  553. switch(it){
  554. SCASE(_OP_ADD, +)
  555. SCASE(_OP_SUB, -)
  556. SCASE(_OP_MUL, *)
  557. SCASE(_OP_DIV, /)
  558. SCASE(_OP_MOD, %)
  559. default: return _SC("?");
  560. }
  561. #undef SCASE
  562. }
  563. const SQChar *SQGetOpName(int op_code) {
  564. const SQChar *str_op;
  565. switch(op_code){
  566. #define ENUM_OP(a,b) case a: str_op = _SC(#a); break;
  567. SQ_OP_CODE_LIST()
  568. #undef ENUM_OP
  569. default:
  570. str_op = _SC("???");
  571. }
  572. return str_op;
  573. }
  574. bool SQFunctionProto::SaveAsSource(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  575. {
  576. SQInteger i,nliterals = _nliterals,nparameters = _nparameters;
  577. SQInteger noutervalues = _noutervalues,nlocalvarinfos = _nlocalvarinfos;
  578. SQInteger nlineinfos=_nlineinfos,ninstructions = _ninstructions,nfunctions=_nfunctions;
  579. SQInteger ndefaultparams = _ndefaultparams;
  580. SafeWriteFmt(v,write,up,"{\n");
  581. SafeWriteFmt(v,write,up,"\tsource_name = ");
  582. _CHECK_IO(WriteObjectAsCode(v,up,write,_sourcename));
  583. SafeWriteFmt(v,write,up,",\n\tfunction_name = ");
  584. _CHECK_IO(WriteObjectAsCode(v,up,write,_name));
  585. SafeWriteFmt(v,write,up,"\n");
  586. SafeWriteFmt(v,write,up,"\tliterals = [\n");
  587. for(i=0;i<nliterals;i++){
  588. SafeWriteFmt(v,write,up,"\t\t/*%d*/ ", i);
  589. _CHECK_IO(WriteObjectAsCode(v,up,write,_literals[i], false));
  590. SafeWriteFmt(v,write,up,",\n");
  591. }
  592. SafeWriteFmt(v,write,up,"\t],\n");
  593. SafeWriteFmt(v,write,up,"\tparameters = [\n");
  594. for(i=0;i<nparameters;i++){
  595. SafeWriteFmt(v,write,up,"\t\t/*%d*/ ", i);
  596. _CHECK_IO(WriteObjectAsCode(v,up,write,_parameters[i]));
  597. SafeWriteFmt(v,write,up,",\n");
  598. }
  599. SafeWriteFmt(v,write,up,"\t],\n");
  600. SafeWriteFmt(v,write,up,"\toutervalues = [\n\t\t//[type, src, name],\n");
  601. for(i=0;i<noutervalues;i++){
  602. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [%d, ", i, _outervalues[i]._type);
  603. _CHECK_IO(WriteObjectAsCode(v,up,write,_outervalues[i]._src));
  604. SafeWriteFmt(v,write,up,", ");
  605. _CHECK_IO(WriteObjectAsCode(v,up,write,_outervalues[i]._name));
  606. SafeWriteFmt(v,write,up,"],\n");
  607. }
  608. SafeWriteFmt(v,write,up,"\t],\n");
  609. SafeWriteFmt(v,write,up,"\tlocalvarinfos = [\n\t\t//[pos, name, start_op, end_op, scope, type],\n");
  610. for(i=0;i<nlocalvarinfos;i++){
  611. SQLocalVarInfo &lvi=_localvarinfos[i];
  612. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [", i);
  613. SafeWriteFmt(v,write,up,"%d, ", lvi._pos);
  614. _CHECK_IO(WriteObjectAsCode(v,up,write,lvi._name));
  615. SafeWriteFmt(v,write,up,", %d", lvi._start_op);
  616. SafeWriteFmt(v,write,up,", %d", lvi._end_op);
  617. SafeWriteFmt(v,write,up,", %d", lvi._scope);
  618. SafeWriteFmt(v,write,up,", %d],\n", lvi._type);
  619. }
  620. SafeWriteFmt(v,write,up,"\t],\n");
  621. SafeWriteFmt(v,write,up,"\tlineinfos = [\n\t\t//[op, line],\n");
  622. for(i=0;i<nlineinfos;i++){
  623. SQLineInfo &li=_lineinfos[i];
  624. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [%d, %d], /*%s*/\n", i, li._op, li._line, SQGetOpName(li._op));
  625. }
  626. SafeWriteFmt(v,write,up,"\t],\n");
  627. SafeWriteFmt(v,write,up,"\tdefaultparams = [");
  628. for(i=0;i<ndefaultparams;i++){
  629. SafeWriteFmt(v,write,up,"%d,", _defaultparams[i]);
  630. }
  631. SafeWriteFmt(v,write,up,"],\n");
  632. SafeWriteFmt(v,write,up,"\tinstructions = [ /*stack*/\n");
  633. SafeWriteFmt(v,write,up,"\t\t//[op_str, op, arg0, arg1, arg2, arg3],\n");
  634. const SQChar *str_op;
  635. for(i=0;i<ninstructions;i++){
  636. SQInstruction &inst = _instructions[i];
  637. str_op = SQGetOpName(inst.op);
  638. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [\"%s\", %d, %d, %d, %d, %d],", i, str_op, inst.op, inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  639. switch(inst.op){
  640. case _OP_LOAD:
  641. case _OP_DLOAD:
  642. case _OP_GETK:{
  643. SQInteger lidx = inst._arg1;
  644. if(lidx >= 0xFFFFFFFF) SafeWriteFmt(v,write,up,"\t\t/* stk[%d] <- null */", inst._arg0);
  645. else
  646. {
  647. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] <- literals[%d] */", inst._arg0, lidx);
  648. //_CHECK_IO(WriteObjectAsCode(v,up,write,_literals[lidx], false));
  649. //SafeWriteFmt(v,write,up," */");
  650. }
  651. if(inst.op == _OP_DLOAD) {
  652. lidx = inst._arg3;
  653. if(lidx >= 0xFFFFFFFF) SafeWriteFmt(v,write,up," /* stk[%d] <- null */", inst._arg2);
  654. else {
  655. SafeWriteFmt(v,write,up," /* stk[%d] <- literals[%d] ", inst._arg2, lidx);
  656. _CHECK_IO(WriteObjectAsCode(v,up,write,_literals[lidx], false));
  657. SafeWriteFmt(v,write,up," */");
  658. }
  659. }
  660. }
  661. break;
  662. case _OP_GET:
  663. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d] = stk_at_arg1[%d].get(stk_at_arg2(%d)) */",
  664. inst._arg0, inst._arg1, inst._arg2);
  665. break;
  666. case _OP_SET:
  667. SafeWriteFmt(v,write,up,"\t/* stk_at_arg1[%d].set(stk_at_arg2(%d), stk_at_arg3(%d)) */",
  668. inst._arg1, inst._arg2, inst._arg3);
  669. break;
  670. case _OP_PREPCALLK:
  671. case _OP_PREPCALL:
  672. SafeWriteFmt(v,write,up,"\t/* closure_at_stk[%d], stk[%d].get(%s[%d]) -> stk[%d] */",
  673. inst._arg0, inst._arg2, inst.op == _OP_PREPCALLK ? "literals" : "stk", inst._arg1, inst._arg3);
  674. break;
  675. case _OP_LOADFLOAT:
  676. SafeWriteFmt(v,write,up,"\t/* %f */", *((SQFloat*)&inst._arg1));
  677. break;
  678. case _OP_GETOUTER:
  679. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- outervalues[%d] == (%s) */",
  680. inst._arg0, inst._arg1, _stringval(_outervalues[inst._arg1]._name));
  681. break;
  682. case _OP_CALL:
  683. SafeWriteFmt(v,write,up,"\t\t/* target[%d], closure_at_stk[%d], stackbase(%d), nargs(%d) */",
  684. inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  685. break;
  686. case _OP_MOVE:
  687. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] <- stk[%d] */", inst._arg0, inst._arg1);
  688. break;
  689. case _OP_DMOVE:
  690. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- stk[%d], stk[%d] <- stk[%d] */",
  691. inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  692. break;
  693. case _OP_LOADINT:
  694. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- arg1(%d) */", inst._arg0, inst._arg1);
  695. break;
  696. case _OP_EQ:
  697. SafeWriteFmt(v,write,up,"\t\t/* stk_at_arg0[%d] = stk_at_arg2[%d] == (arg3(%d) !=0 ? literals_at_arg1[%d] : stk_at_arg1[%d]) */",
  698. inst._arg0, inst._arg2, inst._arg3, inst._arg1, inst._arg1);
  699. break;
  700. case _OP_NOT:
  701. SafeWriteFmt(v,write,up,"\t\t/* stk_at_arg0[%d] = IsFalse(stk_at_arg1[%d]) */",
  702. inst._arg0, inst._arg1);
  703. break;
  704. case _OP_AND:
  705. case _OP_OR:
  706. SafeWriteFmt(v,write,up,"\t\t/* %s(stk_at_arg2[%d]) {stk_at_arg0[%d] = stk[%d]; goto[%d]} */",
  707. _OP_OR ? "IsTrue" : "IsFalse",
  708. inst._arg2, inst._arg0, inst._arg2, i + inst._arg1 + 1);
  709. break;
  710. case _OP_JZ:
  711. SafeWriteFmt(v,write,up,"\t\t/* IsFalse(STK(%d) (ci->_ip+=(%d) -> goto[%d]) */",
  712. inst._arg0, inst._arg1, i + inst._arg1 + 1);
  713. break;
  714. case _OP_JCMP:
  715. SafeWriteFmt(v,write,up,"\t\t/* IsFalse(STK(%d) %s STK(%d)) (ci->_ip+=(%d) -> goto[%d]) */",
  716. inst._arg2, SQGetCmpOpName(inst._arg3), inst._arg0, inst._arg1, i + inst._arg1 + 1);
  717. break;
  718. case _OP_JMP:
  719. SafeWriteFmt(v,write,up,"\t\t/* (ci->_ip+=(%d)) -> goto[%d] */",
  720. inst._arg1, i + inst._arg1 + 1);
  721. break;
  722. case _OP_RETURN:
  723. SafeWriteFmt(v,write,up,"\t/* _arg0(%d) != 255 ? stk[%d] : null */",
  724. inst._arg0, inst._arg1);
  725. break;
  726. case _OP_NEWOBJ:
  727. SafeWriteFmt(v,write,up,"\t/* stk[%d], len(%d), %s(%d) */",
  728. inst._arg0, inst._arg1, SQGetNewObjTypeName(inst._arg3), inst._arg3);
  729. break;
  730. case _OP_APPENDARRAY:
  731. SafeWriteFmt(v,write,up,"\t/* array_at_stk(%d), %s(%d), sq_type(%d) */",
  732. inst._arg0, SQGetArrayAppendTypeName(inst._arg2), inst._arg1, inst._arg2);
  733. break;
  734. case _OP_NEWSLOT:
  735. case _OP_NEWSLOTA:
  736. SafeWriteFmt(v,write,up,"\t/* flags(%d), table_at_stk(%d), key_at_stk(%d), val_at_stk(%d) */",
  737. inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  738. break;
  739. case _OP_LOADBOOL:
  740. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- bool(%d) */",
  741. inst._arg0, inst._arg2);
  742. break;
  743. case _OP_LOADNULLS:
  744. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d .. (arg0 + arg1(%d))] = null */",
  745. inst._arg0, inst._arg1);
  746. break;
  747. case _OP_LOADROOT:
  748. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d] <- roottable */",
  749. inst._arg0);
  750. break;
  751. case _OP_CLOSURE:
  752. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d] <- functions[%d] == (",
  753. inst._arg0, inst._arg1);
  754. _CHECK_IO(WriteObjectAsCode(v,up,write,_funcproto(_functions[inst._arg1])->_name));
  755. SafeWriteFmt(v,write,up,"), isLanbda(%d) */", inst._arg2);
  756. break;
  757. case _OP_ADD:
  758. case _OP_SUB:
  759. case _OP_DIV:
  760. case _OP_MUL:
  761. case _OP_MOD:
  762. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] = stk[%d] %s stk[%d] */",
  763. inst._arg0, inst._arg1, SQGetArithOpName(inst.op), inst._arg2);
  764. break;
  765. case _OP_INCL:
  766. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] = stk[%d] + sarg3(%d) */",
  767. inst._arg1, inst._arg1, ((char)inst._arg3));
  768. break;
  769. case _OP_PINCL:
  770. SafeWriteFmt(v,write,up,"\t\t/* target = stk(%d); stk[%d] = stk[%d] + sarg3(%d) */",
  771. inst._arg1, inst._arg1, inst._arg1, ((char)inst._arg3));
  772. break;
  773. case _OP_BITW:
  774. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] = stk[%d] %s stk[%d] */",
  775. inst._arg0, inst._arg1, SQGetBitwiseOpName(inst._arg3), inst._arg2);
  776. break;
  777. //default:
  778. }
  779. SafeWriteFmt(v,write,up,"\n");
  780. }
  781. SafeWriteFmt(v,write,up,"\t],\n");
  782. SafeWriteFmt(v,write,up,"\tfunctions = [\n");
  783. for(i=0;i<nfunctions;i++){
  784. SafeWriteFmt(v,write,up,"/*function %d*/", i);
  785. _CHECK_IO(_funcproto(_functions[i])->SaveAsSource(v,up,write));
  786. SafeWriteFmt(v,write,up,",\n");
  787. }
  788. SafeWriteFmt(v,write,up,"],\n");
  789. SafeWriteFmt(v,write,up,"\tstacksize = %d,\n", _stacksize);
  790. SafeWriteFmt(v,write,up,"\tbgenerator = %d,\n", _bgenerator);
  791. SafeWriteFmt(v,write,up,"\tvarparams = %d,\n", _varparams);
  792. SafeWriteFmt(v,write,up,"}");
  793. return true;
  794. }
  795. bool SQFunctionProto::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
  796. {
  797. SQInteger i, nliterals,nparameters;
  798. SQInteger noutervalues ,nlocalvarinfos ;
  799. SQInteger nlineinfos,ninstructions ,nfunctions,ndefaultparams ;
  800. SQObjectPtr sourcename, name;
  801. SQObjectPtr o;
  802. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  803. _CHECK_IO(ReadObject(v, up, read, sourcename));
  804. _CHECK_IO(ReadObject(v, up, read, name));
  805. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  806. _CHECK_IO(SafeRead(v,read,up, &nliterals, sizeof(nliterals)));
  807. _CHECK_IO(SafeRead(v,read,up, &nparameters, sizeof(nparameters)));
  808. _CHECK_IO(SafeRead(v,read,up, &noutervalues, sizeof(noutervalues)));
  809. _CHECK_IO(SafeRead(v,read,up, &nlocalvarinfos, sizeof(nlocalvarinfos)));
  810. _CHECK_IO(SafeRead(v,read,up, &nlineinfos, sizeof(nlineinfos)));
  811. _CHECK_IO(SafeRead(v,read,up, &ndefaultparams, sizeof(ndefaultparams)));
  812. _CHECK_IO(SafeRead(v,read,up, &ninstructions, sizeof(ninstructions)));
  813. _CHECK_IO(SafeRead(v,read,up, &nfunctions, sizeof(nfunctions)));
  814. SQFunctionProto *f = SQFunctionProto::Create(_opt_ss(v),ninstructions,nliterals,nparameters,
  815. nfunctions,noutervalues,nlineinfos,nlocalvarinfos,ndefaultparams);
  816. SQObjectPtr proto = f; //gets a ref in case of failure
  817. f->_sourcename = sourcename;
  818. f->_name = name;
  819. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  820. for(i = 0;i < nliterals; i++){
  821. _CHECK_IO(ReadObject(v, up, read, o));
  822. f->_literals[i] = o;
  823. }
  824. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  825. for(i = 0; i < nparameters; i++){
  826. _CHECK_IO(ReadObject(v, up, read, o));
  827. f->_parameters[i] = o;
  828. }
  829. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  830. for(i = 0; i < noutervalues; i++){
  831. SQUnsignedInteger type;
  832. SQObjectPtr name;
  833. _CHECK_IO(SafeRead(v,read,up, &type, sizeof(SQUnsignedInteger)));
  834. _CHECK_IO(ReadObject(v, up, read, o));
  835. _CHECK_IO(ReadObject(v, up, read, name));
  836. f->_outervalues[i] = SQOuterVar(name,o, (SQOuterType)type);
  837. }
  838. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  839. for(i = 0; i < nlocalvarinfos; i++){
  840. SQLocalVarInfo lvi;
  841. _CHECK_IO(ReadObject(v, up, read, lvi._name));
  842. _CHECK_IO(SafeRead(v,read,up, &lvi._pos, sizeof(SQUnsignedInteger)));
  843. _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(SQUnsignedInteger)));
  844. _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(SQUnsignedInteger)));
  845. f->_localvarinfos[i] = lvi;
  846. }
  847. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  848. _CHECK_IO(SafeRead(v,read,up, f->_lineinfos, sizeof(SQLineInfo)*nlineinfos));
  849. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  850. _CHECK_IO(SafeRead(v,read,up, f->_defaultparams, sizeof(SQInteger)*ndefaultparams));
  851. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  852. _CHECK_IO(SafeRead(v,read,up, f->_instructions, sizeof(SQInstruction)*ninstructions));
  853. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  854. for(i = 0; i < nfunctions; i++){
  855. _CHECK_IO(_funcproto(o)->Load(v, up, read, o));
  856. f->_functions[i] = o;
  857. }
  858. _CHECK_IO(SafeRead(v,read,up, &f->_stacksize, sizeof(f->_stacksize)));
  859. _CHECK_IO(SafeRead(v,read,up, &f->_bgenerator, sizeof(f->_bgenerator)));
  860. _CHECK_IO(SafeRead(v,read,up, &f->_varparams, sizeof(f->_varparams)));
  861. ret = f;
  862. return true;
  863. }
  864. #ifndef NO_GARBAGE_COLLECTOR
  865. #define START_MARK() if(!(_uiRef&MARK_FLAG)){ \
  866. _uiRef|=MARK_FLAG;
  867. #define END_MARK() RemoveFromChain(&_sharedstate->_gc_chain, this); \
  868. AddToChain(chain, this); }
  869. void SQVM::Mark(SQCollectable **chain)
  870. {
  871. START_MARK()
  872. SQSharedState::MarkObject(_lasterror,chain);
  873. SQSharedState::MarkObject(_errorhandler,chain);
  874. SQSharedState::MarkObject(_debughook_closure,chain);
  875. SQSharedState::MarkObject(_roottable, chain);
  876. SQSharedState::MarkObject(temp_reg, chain);
  877. for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
  878. for(SQInteger k = 0; k < _callsstacksize; k++) SQSharedState::MarkObject(_callsstack[k]._closure, chain);
  879. END_MARK()
  880. }
  881. void SQArray::Mark(SQCollectable **chain)
  882. {
  883. START_MARK()
  884. SQInteger len = _values.size();
  885. for(SQInteger i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);
  886. END_MARK()
  887. }
  888. void SQTable::Mark(SQCollectable **chain)
  889. {
  890. START_MARK()
  891. if(_delegate) _delegate->Mark(chain);
  892. SQInteger len = _numofnodes;
  893. for(SQInteger i = 0; i < len; i++){
  894. SQSharedState::MarkObject(_nodes[i].key, chain);
  895. SQSharedState::MarkObject(_nodes[i].val, chain);
  896. }
  897. END_MARK()
  898. }
  899. void SQClass::Mark(SQCollectable **chain)
  900. {
  901. START_MARK()
  902. _members->Mark(chain);
  903. if(_base) _base->Mark(chain);
  904. SQSharedState::MarkObject(_attributes, chain);
  905. for(SQUnsignedInteger i =0; i< _defaultvalues.size(); i++) {
  906. SQSharedState::MarkObject(_defaultvalues[i].val, chain);
  907. SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);
  908. }
  909. for(SQUnsignedInteger j =0; j< _methods.size(); j++) {
  910. SQSharedState::MarkObject(_methods[j].val, chain);
  911. SQSharedState::MarkObject(_methods[j].attrs, chain);
  912. }
  913. for(SQUnsignedInteger k =0; k< MT_LAST; k++) {
  914. SQSharedState::MarkObject(_metamethods[k], chain);
  915. }
  916. END_MARK()
  917. }
  918. void SQInstance::Mark(SQCollectable **chain)
  919. {
  920. START_MARK()
  921. _class->Mark(chain);
  922. SQUnsignedInteger nvalues = _class->_defaultvalues.size();
  923. for(SQUnsignedInteger i =0; i< nvalues; i++) {
  924. SQSharedState::MarkObject(_values[i], chain);
  925. }
  926. END_MARK()
  927. }
  928. void SQGenerator::Mark(SQCollectable **chain)
  929. {
  930. START_MARK()
  931. for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
  932. SQSharedState::MarkObject(_closure, chain);
  933. END_MARK()
  934. }
  935. void SQFunctionProto::Mark(SQCollectable **chain)
  936. {
  937. START_MARK()
  938. for(SQInteger i = 0; i < _nliterals; i++) SQSharedState::MarkObject(_literals[i], chain);
  939. for(SQInteger k = 0; k < _nfunctions; k++) SQSharedState::MarkObject(_functions[k], chain);
  940. END_MARK()
  941. }
  942. void SQClosure::Mark(SQCollectable **chain)
  943. {
  944. START_MARK()
  945. if(_base) _base->Mark(chain);
  946. SQFunctionProto *fp = _function;
  947. fp->Mark(chain);
  948. for(SQInteger i = 0; i < fp->_noutervalues; i++) SQSharedState::MarkObject(_outervalues[i], chain);
  949. for(SQInteger k = 0; k < fp->_ndefaultparams; k++) SQSharedState::MarkObject(_defaultparams[k], chain);
  950. END_MARK()
  951. }
  952. void SQNativeClosure::Mark(SQCollectable **chain)
  953. {
  954. START_MARK()
  955. for(SQUnsignedInteger i = 0; i < _noutervalues; i++) SQSharedState::MarkObject(_outervalues[i], chain);
  956. END_MARK()
  957. }
  958. void SQOuter::Mark(SQCollectable **chain)
  959. {
  960. START_MARK()
  961. /* If the valptr points to a closed value, that value is alive */
  962. if(_valptr == &_value) {
  963. SQSharedState::MarkObject(_value, chain);
  964. }
  965. END_MARK()
  966. }
  967. void SQUserData::Mark(SQCollectable **chain){
  968. START_MARK()
  969. if(_delegate) _delegate->Mark(chain);
  970. END_MARK()
  971. }
  972. void SQCollectable::UnMark() { _uiRef&=~MARK_FLAG; }
  973. #endif