sqobject.cpp 35 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052
  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)SQTranslateIndex(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 SQTranslateIndex(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) || !defined(SQUSEDOUBLE) && defined(_SQ64)
  82. _weakref->_obj._unVal.raw = 0; //without this on 32bits using double the comparison operator do not work
  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--; //I've changed this one
  238. line = _lineinfos[mid]._line;
  239. return line;
  240. }
  241. SQClosure::~SQClosure()
  242. {
  243. __ObjRelease(_env);
  244. __ObjRelease(_base);
  245. REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
  246. }
  247. #define _CHECK_IO(exp) { if(!exp)return false; }
  248. static bool SafeWrite(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUserPointer dest,SQInteger size)
  249. {
  250. if(write(up,dest,size) != size) {
  251. v->Raise_Error(_SC("io error (write function failure)"));
  252. return false;
  253. }
  254. return true;
  255. }
  256. static bool SafeWriteFmt(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up, const SQChar *fmt, ...)
  257. {
  258. if(fmt){
  259. SQChar str[8192];
  260. va_list vl;
  261. va_start(vl, fmt);
  262. SQInteger len = scvsprintf(str, sizeof(str), fmt, vl);
  263. va_end(vl);
  264. return SafeWrite(v, write, up, str, len);
  265. }
  266. return false;
  267. }
  268. static bool SafeRead(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUserPointer dest,SQInteger size)
  269. {
  270. if(size && read(up,dest,size) != size) {
  271. v->Raise_Error(_SC("io error, read function failure, the origin stream could be corrupted/trucated"));
  272. return false;
  273. }
  274. return true;
  275. }
  276. static bool WriteTag(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUnsignedInteger32 tag)
  277. {
  278. return SafeWrite(v,write,up,&tag,sizeof(tag));
  279. }
  280. static bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUnsignedInteger32 tag)
  281. {
  282. SQUnsignedInteger32 t;
  283. _CHECK_IO(SafeRead(v,read,up,&t,sizeof(t)));
  284. if(t != tag){
  285. v->Raise_Error(_SC("invalid or corrupted closure stream"));
  286. return false;
  287. }
  288. return true;
  289. }
  290. static bool WriteObjectAsCode(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o, bool withQuotes=true)
  291. {
  292. SQChar buf[32];
  293. SQInteger sz;
  294. switch(type(o)){
  295. case OT_STRING:{
  296. SQInteger str_size = _string(o)->_len;
  297. if(str_size){
  298. const SQChar *str = _stringval(o);
  299. bool needQuote = false;
  300. for(int i=0; i < str_size; ++i){
  301. if(str[i] == '"' || str[i] == '\n'){
  302. needQuote = true;
  303. break;
  304. }
  305. }
  306. if(!needQuote) withQuotes = true;
  307. const SQChar *d1 = withQuotes ? _SC("\"") : _SC("[==[");
  308. const SQChar *d2 = withQuotes ? _SC("\"") : _SC("]==]");
  309. _CHECK_IO(SafeWrite(v,write,up, (void*)d1, scstrlen(d1)));
  310. _CHECK_IO(SafeWrite(v,write,up, (void*)str,rsl(_string(o)->_len)));
  311. _CHECK_IO(SafeWrite(v,write,up, (void*)d2, scstrlen(d2)));
  312. }
  313. else
  314. {
  315. const SQChar *str = _SC("null");
  316. _CHECK_IO(SafeWrite(v,write,up, (void*)str, scstrlen(str)));
  317. }
  318. }
  319. break;
  320. case OT_INTEGER:
  321. sz = scsprintf(buf, sizeof(buf), _PRINT_INT_FMT, _integer(o));
  322. _CHECK_IO(SafeWrite(v,write,up,buf,sz));break;
  323. case OT_FLOAT:
  324. sz = scsprintf(buf, sizeof(buf), "%f", _float(o));
  325. _CHECK_IO(SafeWrite(v,write,up,buf,sz));break;
  326. case OT_NULL:{
  327. const SQChar str[] = _SC("null");
  328. _CHECK_IO(SafeWrite(v,write,up, (void*)str, scstrlen(str)));
  329. }
  330. break;
  331. default:
  332. v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
  333. return false;
  334. }
  335. return true;
  336. }
  337. static bool WriteObject(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o)
  338. {
  339. SQUnsignedInteger32 _type = (SQUnsignedInteger32)type(o);
  340. _CHECK_IO(SafeWrite(v,write,up,&_type,sizeof(_type)));
  341. switch(type(o)){
  342. case OT_STRING:
  343. _CHECK_IO(SafeWrite(v,write,up,&_string(o)->_len,sizeof(SQInteger)));
  344. _CHECK_IO(SafeWrite(v,write,up,_stringval(o),rsl(_string(o)->_len)));
  345. break;
  346. case OT_BOOL:
  347. case OT_INTEGER:
  348. _CHECK_IO(SafeWrite(v,write,up,&_integer(o),sizeof(SQInteger)));break;
  349. case OT_FLOAT:
  350. _CHECK_IO(SafeWrite(v,write,up,&_float(o),sizeof(SQFloat)));break;
  351. case OT_NULL:
  352. break;
  353. default:
  354. v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
  355. return false;
  356. }
  357. return true;
  358. }
  359. static bool ReadObject(HSQUIRRELVM v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &o)
  360. {
  361. SQUnsignedInteger32 _type;
  362. _CHECK_IO(SafeRead(v,read,up,&_type,sizeof(_type)));
  363. SQObjectType t = (SQObjectType)_type;
  364. switch(t){
  365. case OT_STRING:{
  366. SQInteger len;
  367. _CHECK_IO(SafeRead(v,read,up,&len,sizeof(SQInteger)));
  368. _CHECK_IO(SafeRead(v,read,up,_ss(v)->GetScratchPad(sq_rsl(len)),sq_rsl(len)));
  369. o=SQString::Create(_ss(v),_ss(v)->GetScratchPad(-1),len);
  370. }
  371. break;
  372. case OT_INTEGER:{
  373. SQInteger i;
  374. _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o = i; break;
  375. }
  376. case OT_BOOL:{
  377. SQInteger i;
  378. _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o._type = OT_BOOL; o._unVal.nInteger = i; break;
  379. }
  380. case OT_FLOAT:{
  381. SQFloat f;
  382. _CHECK_IO(SafeRead(v,read,up,&f,sizeof(SQFloat))); o = f; break;
  383. }
  384. case OT_NULL:
  385. o.Null();
  386. break;
  387. default:
  388. v->Raise_Error(_SC("cannot serialize a %s"),IdType2Name(t));
  389. return false;
  390. }
  391. return true;
  392. }
  393. bool SQClosure::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  394. {
  395. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_HEAD));
  396. _CHECK_IO(WriteTag(v,write,up,sizeof(SQChar)));
  397. _CHECK_IO(WriteTag(v,write,up,sizeof(SQInteger)));
  398. _CHECK_IO(WriteTag(v,write,up,sizeof(SQFloat)));
  399. _CHECK_IO(_function->Save(v,up,write));
  400. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_TAIL));
  401. return true;
  402. }
  403. bool SQClosure::SaveAsSource(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  404. {
  405. _CHECK_IO(_function->SaveAsSource(v,up,write));
  406. return true;
  407. }
  408. bool SQClosure::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
  409. {
  410. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_HEAD));
  411. _CHECK_IO(CheckTag(v,read,up,sizeof(SQChar)));
  412. _CHECK_IO(CheckTag(v,read,up,sizeof(SQInteger)));
  413. _CHECK_IO(CheckTag(v,read,up,sizeof(SQFloat)));
  414. SQObjectPtr func;
  415. _CHECK_IO(SQFunctionProto::Load(v,up,read,func));
  416. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_TAIL));
  417. ret = SQClosure::Create(_ss(v),_funcproto(func),_table(v->_roottable)->GetWeakRef(OT_TABLE));
  418. //FIXME: load an root for this closure
  419. return true;
  420. }
  421. SQFunctionProto::SQFunctionProto(SQSharedState *ss)
  422. {
  423. _stacksize=0;
  424. _bgenerator=false;
  425. INIT_CHAIN();ADD_TO_CHAIN(&_ss(this)->_gc_chain,this);
  426. }
  427. SQFunctionProto::~SQFunctionProto()
  428. {
  429. REMOVE_FROM_CHAIN(&_ss(this)->_gc_chain,this);
  430. }
  431. bool SQFunctionProto::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  432. {
  433. SQInteger i,nliterals = _nliterals,nparameters = _nparameters;
  434. SQInteger noutervalues = _noutervalues,nlocalvarinfos = _nlocalvarinfos;
  435. SQInteger nlineinfos=_nlineinfos,ninstructions = _ninstructions,nfunctions=_nfunctions;
  436. SQInteger ndefaultparams = _ndefaultparams;
  437. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  438. _CHECK_IO(WriteObject(v,up,write,_sourcename));
  439. _CHECK_IO(WriteObject(v,up,write,_name));
  440. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  441. _CHECK_IO(SafeWrite(v,write,up,&nliterals,sizeof(nliterals)));
  442. _CHECK_IO(SafeWrite(v,write,up,&nparameters,sizeof(nparameters)));
  443. _CHECK_IO(SafeWrite(v,write,up,&noutervalues,sizeof(noutervalues)));
  444. _CHECK_IO(SafeWrite(v,write,up,&nlocalvarinfos,sizeof(nlocalvarinfos)));
  445. _CHECK_IO(SafeWrite(v,write,up,&nlineinfos,sizeof(nlineinfos)));
  446. _CHECK_IO(SafeWrite(v,write,up,&ndefaultparams,sizeof(ndefaultparams)));
  447. _CHECK_IO(SafeWrite(v,write,up,&ninstructions,sizeof(ninstructions)));
  448. _CHECK_IO(SafeWrite(v,write,up,&nfunctions,sizeof(nfunctions)));
  449. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  450. for(i=0;i<nliterals;i++){
  451. _CHECK_IO(WriteObject(v,up,write,_literals[i]));
  452. }
  453. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  454. for(i=0;i<nparameters;i++){
  455. _CHECK_IO(WriteObject(v,up,write,_parameters[i]));
  456. }
  457. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  458. for(i=0;i<noutervalues;i++){
  459. _CHECK_IO(SafeWrite(v,write,up,&_outervalues[i]._type,sizeof(SQUnsignedInteger)));
  460. _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._src));
  461. _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._name));
  462. }
  463. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  464. for(i=0;i<nlocalvarinfos;i++){
  465. SQLocalVarInfo &lvi=_localvarinfos[i];
  466. _CHECK_IO(WriteObject(v,up,write,lvi._name));
  467. _CHECK_IO(SafeWrite(v,write,up,&lvi._pos,sizeof(SQUnsignedInteger)));
  468. _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(SQUnsignedInteger)));
  469. _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(SQUnsignedInteger)));
  470. }
  471. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  472. _CHECK_IO(SafeWrite(v,write,up,_lineinfos,sizeof(SQLineInfo)*nlineinfos));
  473. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  474. _CHECK_IO(SafeWrite(v,write,up,_defaultparams,sizeof(SQInteger)*ndefaultparams));
  475. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  476. _CHECK_IO(SafeWrite(v,write,up,_instructions,sizeof(SQInstruction)*ninstructions));
  477. _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
  478. for(i=0;i<nfunctions;i++){
  479. _CHECK_IO(_funcproto(_functions[i])->Save(v,up,write));
  480. }
  481. _CHECK_IO(SafeWrite(v,write,up,&_stacksize,sizeof(_stacksize)));
  482. _CHECK_IO(SafeWrite(v,write,up,&_bgenerator,sizeof(_bgenerator)));
  483. _CHECK_IO(SafeWrite(v,write,up,&_varparams,sizeof(_varparams)));
  484. return true;
  485. }
  486. const SQChar *SQGetBitwiseOpName(int it){
  487. #define SCASE(x) case x: return _SC(#x); break;
  488. switch(it){
  489. SCASE(BW_AND)
  490. SCASE(BW_OR)
  491. SCASE(BW_XOR)
  492. SCASE(BW_SHIFTL)
  493. SCASE(BW_SHIFTR)
  494. SCASE(BW_USHIFTR)
  495. default: return _SC("?");
  496. }
  497. #undef SCASE
  498. }
  499. const SQChar *SQGetCmpOpName(int it){
  500. #define SCASE(x) case x: return _SC(#x); break;
  501. switch(it){
  502. SCASE(CMP_G)
  503. SCASE(CMP_GE)
  504. SCASE(CMP_L)
  505. SCASE(CMP_LE)
  506. SCASE(CMP_3W)
  507. default: return _SC("?");
  508. }
  509. #undef SCASE
  510. }
  511. const SQChar *SQGetArrayAppendTypeName(int it){
  512. #define SCASE(x) case x: return _SC(#x); break;
  513. switch(it){
  514. SCASE(AAT_STACK)
  515. SCASE(AAT_LITERAL)
  516. SCASE(AAT_INT)
  517. SCASE(AAT_FLOAT)
  518. SCASE(AAT_BOOL)
  519. default: return _SC("?");
  520. }
  521. #undef SCASE
  522. }
  523. const SQChar *SQGetNewObjTypeName(int it){
  524. #define SCASE(x) case x: return _SC(#x); break;
  525. switch(it){
  526. SCASE(NOT_TABLE)
  527. SCASE(NOT_ARRAY)
  528. SCASE(NOT_CLASS)
  529. default: return _SC("?");
  530. }
  531. #undef SCASE
  532. }
  533. const SQChar *SQGetArithOpName(int it){
  534. #define SCASE(x, z) case x: return _SC(#z); break;
  535. switch(it){
  536. SCASE(_OP_ADD, +)
  537. SCASE(_OP_SUB, -)
  538. SCASE(_OP_MUL, *)
  539. SCASE(_OP_DIV, /)
  540. SCASE(_OP_MOD, %)
  541. default: return _SC("?");
  542. }
  543. #undef SCASE
  544. }
  545. const SQChar *SQGetOpName(int op_code) {
  546. const SQChar *str_op;
  547. switch(op_code){
  548. #define ENUM_OP(a,b) case a: str_op = _SC(#a); break;
  549. SQ_OP_CODE_LIST()
  550. #undef ENUM_OP
  551. default:
  552. str_op = _SC("???");
  553. }
  554. return str_op;
  555. }
  556. bool SQFunctionProto::SaveAsSource(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
  557. {
  558. SQInteger i,nliterals = _nliterals,nparameters = _nparameters;
  559. SQInteger noutervalues = _noutervalues,nlocalvarinfos = _nlocalvarinfos;
  560. SQInteger nlineinfos=_nlineinfos,ninstructions = _ninstructions,nfunctions=_nfunctions;
  561. SQInteger ndefaultparams = _ndefaultparams;
  562. SafeWriteFmt(v,write,up,"{\n");
  563. SafeWriteFmt(v,write,up,"\tsource_name = ");
  564. _CHECK_IO(WriteObjectAsCode(v,up,write,_sourcename));
  565. SafeWriteFmt(v,write,up,",\n\tfunction_name = ");
  566. _CHECK_IO(WriteObjectAsCode(v,up,write,_name));
  567. SafeWriteFmt(v,write,up,"\n");
  568. SafeWriteFmt(v,write,up,"\tliterals = [\n");
  569. for(i=0;i<nliterals;i++){
  570. SafeWriteFmt(v,write,up,"\t\t/*%d*/ ", i);
  571. _CHECK_IO(WriteObjectAsCode(v,up,write,_literals[i], false));
  572. SafeWriteFmt(v,write,up,",\n");
  573. }
  574. SafeWriteFmt(v,write,up,"\t],\n");
  575. SafeWriteFmt(v,write,up,"\tparameters = [\n");
  576. for(i=0;i<nparameters;i++){
  577. SafeWriteFmt(v,write,up,"\t\t/*%d*/ ", i);
  578. _CHECK_IO(WriteObjectAsCode(v,up,write,_parameters[i]));
  579. SafeWriteFmt(v,write,up,",\n");
  580. }
  581. SafeWriteFmt(v,write,up,"\t],\n");
  582. SafeWriteFmt(v,write,up,"\toutervalues = [\n\t\t//[type, src, name],\n");
  583. for(i=0;i<noutervalues;i++){
  584. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [%d, ", i, _outervalues[i]._type);
  585. _CHECK_IO(WriteObjectAsCode(v,up,write,_outervalues[i]._src));
  586. SafeWriteFmt(v,write,up,", ");
  587. _CHECK_IO(WriteObjectAsCode(v,up,write,_outervalues[i]._name));
  588. SafeWriteFmt(v,write,up,"],\n");
  589. }
  590. SafeWriteFmt(v,write,up,"\t],\n");
  591. SafeWriteFmt(v,write,up,"\tlocalvarinfos = [\n\t\t//[pos, name, start_op, end_op, scope, type],\n");
  592. for(i=0;i<nlocalvarinfos;i++){
  593. SQLocalVarInfo &lvi=_localvarinfos[i];
  594. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [", i);
  595. SafeWriteFmt(v,write,up,"%d, ", lvi._pos);
  596. _CHECK_IO(WriteObjectAsCode(v,up,write,lvi._name));
  597. SafeWriteFmt(v,write,up,", %d", lvi._start_op);
  598. SafeWriteFmt(v,write,up,", %d", lvi._end_op);
  599. SafeWriteFmt(v,write,up,", %d", lvi._scope);
  600. SafeWriteFmt(v,write,up,", %d],\n", lvi._type);
  601. }
  602. SafeWriteFmt(v,write,up,"\t],\n");
  603. SafeWriteFmt(v,write,up,"\tlineinfos = [\n\t\t//[op, line],\n");
  604. for(i=0;i<nlineinfos;i++){
  605. SQLineInfo &li=_lineinfos[i];
  606. SafeWriteFmt(v,write,up,"\t\t/*%d*/ [%d, %d], /*%s*/\n", i, li._op, li._line, SQGetOpName(li._op));
  607. }
  608. SafeWriteFmt(v,write,up,"\t],\n");
  609. SafeWriteFmt(v,write,up,"\tdefaultparams = [");
  610. for(i=0;i<ndefaultparams;i++){
  611. SafeWriteFmt(v,write,up,"%d,", _defaultparams[i]);
  612. }
  613. SafeWriteFmt(v,write,up,"],\n");
  614. SafeWriteFmt(v,write,up,"\tinstructions = [ /*stack*/\n");
  615. SafeWriteFmt(v,write,up,"\t\t//[op_str, op, arg0, arg1, arg2, arg3],\n");
  616. const SQChar *str_op;
  617. for(i=0;i<ninstructions;i++){
  618. SQInstruction &inst = _instructions[i];
  619. str_op = SQGetOpName(inst.op);
  620. 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);
  621. switch(inst.op){
  622. case _OP_LOAD:
  623. case _OP_DLOAD:
  624. case _OP_GETK:{
  625. SQInteger lidx = inst._arg1;
  626. if(lidx >= 0xFFFFFFFF) SafeWriteFmt(v,write,up,"\t\t/* stk[%d] <- null */", inst._arg0);
  627. else
  628. {
  629. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] <- literals[%d] */", inst._arg0, lidx);
  630. //_CHECK_IO(WriteObjectAsCode(v,up,write,_literals[lidx], false));
  631. //SafeWriteFmt(v,write,up," */");
  632. }
  633. if(inst.op == _OP_DLOAD) {
  634. lidx = inst._arg3;
  635. if(lidx >= 0xFFFFFFFF) SafeWriteFmt(v,write,up," /* stk[%d] <- null */", inst._arg2);
  636. else {
  637. SafeWriteFmt(v,write,up," /* stk[%d] <- literals[%d] ", inst._arg2, lidx);
  638. _CHECK_IO(WriteObjectAsCode(v,up,write,_literals[lidx], false));
  639. SafeWriteFmt(v,write,up," */");
  640. }
  641. }
  642. }
  643. break;
  644. case _OP_GET:
  645. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d] = stk_at_arg1[%d].get(stk_at_arg2(%d)) */",
  646. inst._arg0, inst._arg1, inst._arg2);
  647. break;
  648. case _OP_SET:
  649. SafeWriteFmt(v,write,up,"\t/* stk_at_arg1[%d].set(stk_at_arg2(%d), stk_at_arg3(%d)) */",
  650. inst._arg1, inst._arg2, inst._arg3);
  651. break;
  652. case _OP_PREPCALLK:
  653. case _OP_PREPCALL:
  654. SafeWriteFmt(v,write,up,"\t/* closure_at_stk[%d], stk[%d].get(%s[%d]) -> stk[%d] */",
  655. inst._arg0, inst._arg2, inst.op == _OP_PREPCALLK ? "literals" : "stk", inst._arg1, inst._arg3);
  656. break;
  657. case _OP_LOADFLOAT:
  658. SafeWriteFmt(v,write,up,"\t/* %f */", *((SQFloat*)&inst._arg1));
  659. break;
  660. case _OP_GETOUTER:
  661. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- outervalues[%d] == (%s) */",
  662. inst._arg0, inst._arg1, _stringval(_outervalues[inst._arg1]._name));
  663. break;
  664. case _OP_CALL:
  665. SafeWriteFmt(v,write,up,"\t\t/* target[%d], closure_at_stk[%d], stackbase(%d), nargs(%d) */",
  666. inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  667. break;
  668. case _OP_MOVE:
  669. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] <- stk[%d] */", inst._arg0, inst._arg1);
  670. break;
  671. case _OP_DMOVE:
  672. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- stk[%d], stk[%d] <- stk[%d] */",
  673. inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  674. break;
  675. case _OP_LOADINT:
  676. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- arg1(%d) */", inst._arg0, inst._arg1);
  677. break;
  678. case _OP_EQ:
  679. 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]) */",
  680. inst._arg0, inst._arg2, inst._arg3, inst._arg1, inst._arg1);
  681. break;
  682. case _OP_NOT:
  683. SafeWriteFmt(v,write,up,"\t\t/* stk_at_arg0[%d] = IsFalse(stk_at_arg1[%d]) */",
  684. inst._arg0, inst._arg1);
  685. break;
  686. case _OP_AND:
  687. case _OP_OR:
  688. SafeWriteFmt(v,write,up,"\t\t/* %s(stk_at_arg2[%d]) {stk_at_arg0[%d] = stk[%d]; goto[%d]} */",
  689. _OP_OR ? "IsTrue" : "IsFalse",
  690. inst._arg2, inst._arg0, inst._arg2, i + inst._arg1 + 1);
  691. break;
  692. case _OP_JZ:
  693. SafeWriteFmt(v,write,up,"\t\t/* IsFalse(STK(%d) (ci->_ip+=(%d) -> goto[%d]) */",
  694. inst._arg0, inst._arg1, i + inst._arg1 + 1);
  695. break;
  696. case _OP_JCMP:
  697. SafeWriteFmt(v,write,up,"\t\t/* IsFalse(STK(%d) %s STK(%d)) (ci->_ip+=(%d) -> goto[%d]) */",
  698. inst._arg2, SQGetCmpOpName(inst._arg3), inst._arg0, inst._arg1, i + inst._arg1 + 1);
  699. break;
  700. case _OP_JMP:
  701. SafeWriteFmt(v,write,up,"\t\t/* (ci->_ip+=(%d)) -> goto[%d] */",
  702. inst._arg1, i + inst._arg1 + 1);
  703. break;
  704. case _OP_RETURN:
  705. SafeWriteFmt(v,write,up,"\t/* _arg0(%d) != 255 ? stk[%d] : null */",
  706. inst._arg0, inst._arg1);
  707. break;
  708. case _OP_NEWOBJ:
  709. SafeWriteFmt(v,write,up,"\t/* stk[%d], len(%d), %s(%d) */",
  710. inst._arg0, inst._arg1, SQGetNewObjTypeName(inst._arg3), inst._arg3);
  711. break;
  712. case _OP_APPENDARRAY:
  713. SafeWriteFmt(v,write,up,"\t/* array_at_stk(%d), %s(%d), type(%d) */",
  714. inst._arg0, SQGetArrayAppendTypeName(inst._arg2), inst._arg1, inst._arg2);
  715. break;
  716. case _OP_NEWSLOT:
  717. case _OP_NEWSLOTA:
  718. SafeWriteFmt(v,write,up,"\t/* flags(%d), table_at_stk(%d), key_at_stk(%d), val_at_stk(%d) */",
  719. inst._arg0, inst._arg1, inst._arg2, inst._arg3);
  720. break;
  721. case _OP_LOADBOOL:
  722. SafeWriteFmt(v,write,up,"\t/* stk[%d] <- bool(%d) */",
  723. inst._arg0, inst._arg2);
  724. break;
  725. case _OP_LOADNULLS:
  726. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d .. (arg0 + arg1(%d))] = null */",
  727. inst._arg0, inst._arg1);
  728. break;
  729. case _OP_LOADROOT:
  730. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d] <- roottable */",
  731. inst._arg0);
  732. break;
  733. case _OP_CLOSURE:
  734. SafeWriteFmt(v,write,up,"\t/* stk_at_arg0[%d] <- functions[%d] == (",
  735. inst._arg0, inst._arg1);
  736. _CHECK_IO(WriteObjectAsCode(v,up,write,_funcproto(_functions[inst._arg1])->_name));
  737. SafeWriteFmt(v,write,up,"), isLanbda(%d) */", inst._arg2);
  738. break;
  739. case _OP_ADD:
  740. case _OP_SUB:
  741. case _OP_DIV:
  742. case _OP_MUL:
  743. case _OP_MOD:
  744. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] = stk[%d] %s stk[%d] */",
  745. inst._arg0, inst._arg1, SQGetArithOpName(inst.op), inst._arg2);
  746. break;
  747. case _OP_INCL:
  748. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] = stk[%d] + sarg3(%d) */",
  749. inst._arg1, inst._arg1, ((char)inst._arg3));
  750. break;
  751. case _OP_PINCL:
  752. SafeWriteFmt(v,write,up,"\t\t/* target = stk(%d); stk[%d] = stk[%d] + sarg3(%d) */",
  753. inst._arg1, inst._arg1, inst._arg1, ((char)inst._arg3));
  754. break;
  755. case _OP_BITW:
  756. SafeWriteFmt(v,write,up,"\t\t/* stk[%d] = stk[%d] %s stk[%d] */",
  757. inst._arg0, inst._arg1, SQGetBitwiseOpName(inst._arg3), inst._arg2);
  758. break;
  759. //default:
  760. }
  761. SafeWriteFmt(v,write,up,"\n");
  762. }
  763. SafeWriteFmt(v,write,up,"\t],\n");
  764. SafeWriteFmt(v,write,up,"\tfunctions = [\n");
  765. for(i=0;i<nfunctions;i++){
  766. SafeWriteFmt(v,write,up,"/*function %d*/", i);
  767. _CHECK_IO(_funcproto(_functions[i])->SaveAsSource(v,up,write));
  768. SafeWriteFmt(v,write,up,",\n");
  769. }
  770. SafeWriteFmt(v,write,up,"],\n");
  771. SafeWriteFmt(v,write,up,"\tstacksize = %d,\n", _stacksize);
  772. SafeWriteFmt(v,write,up,"\tbgenerator = %d,\n", _bgenerator);
  773. SafeWriteFmt(v,write,up,"\tvarparams = %d,\n", _varparams);
  774. SafeWriteFmt(v,write,up,"}");
  775. return true;
  776. }
  777. bool SQFunctionProto::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
  778. {
  779. SQInteger i, nliterals,nparameters;
  780. SQInteger noutervalues ,nlocalvarinfos ;
  781. SQInteger nlineinfos,ninstructions ,nfunctions,ndefaultparams ;
  782. SQObjectPtr sourcename, name;
  783. SQObjectPtr o;
  784. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  785. _CHECK_IO(ReadObject(v, up, read, sourcename));
  786. _CHECK_IO(ReadObject(v, up, read, name));
  787. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  788. _CHECK_IO(SafeRead(v,read,up, &nliterals, sizeof(nliterals)));
  789. _CHECK_IO(SafeRead(v,read,up, &nparameters, sizeof(nparameters)));
  790. _CHECK_IO(SafeRead(v,read,up, &noutervalues, sizeof(noutervalues)));
  791. _CHECK_IO(SafeRead(v,read,up, &nlocalvarinfos, sizeof(nlocalvarinfos)));
  792. _CHECK_IO(SafeRead(v,read,up, &nlineinfos, sizeof(nlineinfos)));
  793. _CHECK_IO(SafeRead(v,read,up, &ndefaultparams, sizeof(ndefaultparams)));
  794. _CHECK_IO(SafeRead(v,read,up, &ninstructions, sizeof(ninstructions)));
  795. _CHECK_IO(SafeRead(v,read,up, &nfunctions, sizeof(nfunctions)));
  796. SQFunctionProto *f = SQFunctionProto::Create(_opt_ss(v),ninstructions,nliterals,nparameters,
  797. nfunctions,noutervalues,nlineinfos,nlocalvarinfos,ndefaultparams);
  798. SQObjectPtr proto = f; //gets a ref in case of failure
  799. f->_sourcename = sourcename;
  800. f->_name = name;
  801. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  802. for(i = 0;i < nliterals; i++){
  803. _CHECK_IO(ReadObject(v, up, read, o));
  804. f->_literals[i] = o;
  805. }
  806. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  807. for(i = 0; i < nparameters; i++){
  808. _CHECK_IO(ReadObject(v, up, read, o));
  809. f->_parameters[i] = o;
  810. }
  811. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  812. for(i = 0; i < noutervalues; i++){
  813. SQUnsignedInteger type;
  814. SQObjectPtr name;
  815. _CHECK_IO(SafeRead(v,read,up, &type, sizeof(SQUnsignedInteger)));
  816. _CHECK_IO(ReadObject(v, up, read, o));
  817. _CHECK_IO(ReadObject(v, up, read, name));
  818. f->_outervalues[i] = SQOuterVar(name,o, (SQOuterType)type);
  819. }
  820. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  821. for(i = 0; i < nlocalvarinfos; i++){
  822. SQLocalVarInfo lvi;
  823. _CHECK_IO(ReadObject(v, up, read, lvi._name));
  824. _CHECK_IO(SafeRead(v,read,up, &lvi._pos, sizeof(SQUnsignedInteger)));
  825. _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(SQUnsignedInteger)));
  826. _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(SQUnsignedInteger)));
  827. f->_localvarinfos[i] = lvi;
  828. }
  829. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  830. _CHECK_IO(SafeRead(v,read,up, f->_lineinfos, sizeof(SQLineInfo)*nlineinfos));
  831. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  832. _CHECK_IO(SafeRead(v,read,up, f->_defaultparams, sizeof(SQInteger)*ndefaultparams));
  833. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  834. _CHECK_IO(SafeRead(v,read,up, f->_instructions, sizeof(SQInstruction)*ninstructions));
  835. _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
  836. for(i = 0; i < nfunctions; i++){
  837. _CHECK_IO(_funcproto(o)->Load(v, up, read, o));
  838. f->_functions[i] = o;
  839. }
  840. _CHECK_IO(SafeRead(v,read,up, &f->_stacksize, sizeof(f->_stacksize)));
  841. _CHECK_IO(SafeRead(v,read,up, &f->_bgenerator, sizeof(f->_bgenerator)));
  842. _CHECK_IO(SafeRead(v,read,up, &f->_varparams, sizeof(f->_varparams)));
  843. ret = f;
  844. return true;
  845. }
  846. #ifndef NO_GARBAGE_COLLECTOR
  847. #define START_MARK() if(!(_uiRef&MARK_FLAG)){ \
  848. _uiRef|=MARK_FLAG;
  849. #define END_MARK() RemoveFromChain(&_sharedstate->_gc_chain, this); \
  850. AddToChain(chain, this); }
  851. void SQVM::Mark(SQCollectable **chain)
  852. {
  853. START_MARK()
  854. SQSharedState::MarkObject(_lasterror,chain);
  855. SQSharedState::MarkObject(_errorhandler,chain);
  856. SQSharedState::MarkObject(_debughook_closure,chain);
  857. SQSharedState::MarkObject(_roottable, chain);
  858. SQSharedState::MarkObject(temp_reg, chain);
  859. for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
  860. for(SQInteger k = 0; k < _callsstacksize; k++) SQSharedState::MarkObject(_callsstack[k]._closure, chain);
  861. END_MARK()
  862. }
  863. void SQArray::Mark(SQCollectable **chain)
  864. {
  865. START_MARK()
  866. SQInteger len = _values.size();
  867. for(SQInteger i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);
  868. END_MARK()
  869. }
  870. void SQTable::Mark(SQCollectable **chain)
  871. {
  872. START_MARK()
  873. if(_delegate) _delegate->Mark(chain);
  874. SQInteger len = _numofnodes;
  875. for(SQInteger i = 0; i < len; i++){
  876. SQSharedState::MarkObject(_nodes[i].key, chain);
  877. SQSharedState::MarkObject(_nodes[i].val, chain);
  878. }
  879. END_MARK()
  880. }
  881. void SQClass::Mark(SQCollectable **chain)
  882. {
  883. START_MARK()
  884. _members->Mark(chain);
  885. if(_base) _base->Mark(chain);
  886. SQSharedState::MarkObject(_attributes, chain);
  887. for(SQUnsignedInteger i =0; i< _defaultvalues.size(); i++) {
  888. SQSharedState::MarkObject(_defaultvalues[i].val, chain);
  889. SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);
  890. }
  891. for(SQUnsignedInteger j =0; j< _methods.size(); j++) {
  892. SQSharedState::MarkObject(_methods[j].val, chain);
  893. SQSharedState::MarkObject(_methods[j].attrs, chain);
  894. }
  895. for(SQUnsignedInteger k =0; k< MT_LAST; k++) {
  896. SQSharedState::MarkObject(_metamethods[k], chain);
  897. }
  898. END_MARK()
  899. }
  900. void SQInstance::Mark(SQCollectable **chain)
  901. {
  902. START_MARK()
  903. _class->Mark(chain);
  904. SQUnsignedInteger nvalues = _class->_defaultvalues.size();
  905. for(SQUnsignedInteger i =0; i< nvalues; i++) {
  906. SQSharedState::MarkObject(_values[i], chain);
  907. }
  908. END_MARK()
  909. }
  910. void SQGenerator::Mark(SQCollectable **chain)
  911. {
  912. START_MARK()
  913. for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
  914. SQSharedState::MarkObject(_closure, chain);
  915. END_MARK()
  916. }
  917. void SQFunctionProto::Mark(SQCollectable **chain)
  918. {
  919. START_MARK()
  920. for(SQInteger i = 0; i < _nliterals; i++) SQSharedState::MarkObject(_literals[i], chain);
  921. for(SQInteger k = 0; k < _nfunctions; k++) SQSharedState::MarkObject(_functions[k], chain);
  922. END_MARK()
  923. }
  924. void SQClosure::Mark(SQCollectable **chain)
  925. {
  926. START_MARK()
  927. if(_base) _base->Mark(chain);
  928. SQFunctionProto *fp = _function;
  929. fp->Mark(chain);
  930. for(SQInteger i = 0; i < fp->_noutervalues; i++) SQSharedState::MarkObject(_outervalues[i], chain);
  931. for(SQInteger k = 0; k < fp->_ndefaultparams; k++) SQSharedState::MarkObject(_defaultparams[k], chain);
  932. END_MARK()
  933. }
  934. void SQNativeClosure::Mark(SQCollectable **chain)
  935. {
  936. START_MARK()
  937. for(SQUnsignedInteger i = 0; i < _noutervalues; i++) SQSharedState::MarkObject(_outervalues[i], chain);
  938. END_MARK()
  939. }
  940. void SQOuter::Mark(SQCollectable **chain)
  941. {
  942. START_MARK()
  943. /* If the valptr points to a closed value, that value is alive */
  944. if(_valptr == &_value) {
  945. SQSharedState::MarkObject(_value, chain);
  946. }
  947. END_MARK()
  948. }
  949. void SQUserData::Mark(SQCollectable **chain){
  950. START_MARK()
  951. if(_delegate) _delegate->Mark(chain);
  952. END_MARK()
  953. }
  954. void SQCollectable::UnMark() { _uiRef&=~MARK_FLAG; }
  955. #endif