sqobject.cpp 35 KB

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