sqobject.cpp 43 KB

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