sqobject.cpp 32 KB

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