2 see copyright notice in squirrel.h
4 #include "sqpcheader.h"
9 #include "squserdata.h"
10 #include "sqfuncproto.h"
12 #include "sqclosure.h"
15 const SQChar *IdType2Name(SQObjectType type)
17 switch(_RAW_TYPE(type))
19 case _RT_NULL:return _SC("null");
20 case _RT_INTEGER:return _SC("integer");
21 case _RT_FLOAT:return _SC("float");
22 case _RT_BOOL:return _SC("bool");
23 case _RT_STRING:return _SC("string");
24 case _RT_TABLE:return _SC("table");
25 case _RT_ARRAY:return _SC("array");
26 case _RT_GENERATOR:return _SC("generator");
28 case _RT_NATIVECLOSURE:
29 return _SC("function");
32 return _SC("userdata");
33 case _RT_THREAD: return _SC("thread");
34 case _RT_FUNCPROTO: return _SC("function");
35 case _RT_CLASS: return _SC("class");
36 case _RT_INSTANCE: return _SC("instance");
37 case _RT_WEAKREF: return _SC("weakref");
43 const SQChar *GetTypeName(const SQObjectPtr &obj1)
45 return IdType2Name(type(obj1));
48 SQString *SQString::Create(SQSharedState *ss,const SQChar *s,SQInteger len)
50 SQString *str=ADD_STRING(ss,s,len);
55 void SQString::Release()
57 REMOVE_STRING(_sharedstate,this);
60 SQInteger SQString::Next(const SQObjectPtr &refpos, SQObjectPtr &outkey, SQObjectPtr &outval)
62 SQInteger idx = (SQInteger)TranslateIndex(refpos);
64 outkey = (SQInteger)idx;
65 outval = SQInteger(_val[idx]);
66 //return idx for the next iteration
69 //nothing to iterate anymore
73 SQUnsignedInteger TranslateIndex(const SQObjectPtr &idx)
79 return (SQUnsignedInteger)_integer(idx);
80 default: assert(0); break;
85 SQWeakRef *SQRefCounted::GetWeakRef(SQObjectType type)
88 sq_new(_weakref,SQWeakRef);
89 _weakref->_obj._type = type;
90 _weakref->_obj._unVal.pRefCounted = this;
95 SQRefCounted::~SQRefCounted()
98 _weakref->_obj._type = OT_NULL;
99 _weakref->_obj._unVal.pRefCounted = NULL;
103 void SQWeakRef::Release() {
104 if(ISREFCOUNTED(_obj._type)) {
105 _obj._unVal.pRefCounted->_weakref = NULL;
107 sq_delete(this,SQWeakRef);
110 bool SQDelegable::GetMetaMethod(SQVM *v,SQMetaMethod mm,SQObjectPtr &res) {
112 return _delegate->Get((*_ss(v)->_metamethods)[mm],res);
117 bool SQDelegable::SetDelegate(SQTable *mt)
120 if(temp == this) return false;
122 if (temp->_delegate == this) return false; //cycle detected
123 temp = temp->_delegate;
125 if (mt) __ObjAddRef(mt);
126 __ObjRelease(_delegate);
131 bool SQGenerator::Yield(SQVM *v)
133 if(_state==eSuspended) { v->Raise_Error(_SC("internal vm error, yielding dead generator")); return false;}
134 if(_state==eDead) { v->Raise_Error(_SC("internal vm error, yielding a dead generator")); return false; }
135 SQInteger size = v->_top-v->_stackbase;
138 for(SQInteger n =0; n<size; n++) {
139 _stack._vals[n] = v->_stack[v->_stackbase+n];
140 v->_stack[v->_stackbase+n] = _null_;
142 SQInteger nvargs = v->ci->_vargs.size;
143 SQInteger vargsbase = v->ci->_vargs.base;
144 for(SQInteger j = nvargs - 1; j >= 0; j--) {
145 _vargsstack.push_back(v->_vargsstack[vargsbase+j]);
148 for(SQInteger i=0;i<_ci._etraps;i++) {
149 _etraps.push_back(v->_etraps.top());
150 v->_etraps.pop_back();
156 bool SQGenerator::Resume(SQVM *v,SQInteger target)
158 SQInteger size=_stack.size();
159 if(_state==eDead){ v->Raise_Error(_SC("resuming dead generator")); return false; }
160 if(_state==eRunning){ v->Raise_Error(_SC("resuming active generator")); return false; }
161 SQInteger prevtop=v->_top-v->_stackbase;
162 PUSH_CALLINFO(v,_ci);
163 SQInteger oldstackbase=v->_stackbase;
164 v->_stackbase = v->_top;
165 v->ci->_target = (SQInt32)target;
166 v->ci->_generator = this;
167 v->ci->_vargs.size = (unsigned short)_vargsstack.size();
169 for(SQInteger i=0;i<_ci._etraps;i++) {
170 v->_etraps.push_back(_etraps.top());
173 for(SQInteger n =0; n<size; n++) {
174 v->_stack[v->_stackbase+n] = _stack._vals[n];
175 _stack._vals[0] = _null_;
177 while(_vargsstack.size()) {
178 v->_vargsstack.push_back(_vargsstack.back());
179 _vargsstack.pop_back();
181 v->ci->_vargs.base = (unsigned short)(v->_vargsstack.size() - v->ci->_vargs.size);
182 v->_top=v->_stackbase+size;
183 v->ci->_prevtop = (SQInt32)prevtop;
184 v->ci->_prevstkbase = (SQInt32)(v->_stackbase - oldstackbase);
186 if (type(v->_debughook) != OT_NULL && _rawval(v->_debughook) != _rawval(v->ci->_closure))
187 v->CallDebugHook(_SC('c'));
192 void SQArray::Extend(const SQArray *a){
195 for(SQInteger i=0;i<xlen;i++)
196 Append(a->_values[i]);
199 const SQChar* SQFunctionProto::GetLocal(SQVM *vm,SQUnsignedInteger stackbase,SQUnsignedInteger nseq,SQUnsignedInteger nop)
201 SQUnsignedInteger nvars=_nlocalvarinfos;
202 const SQChar *res=NULL;
204 for(SQUnsignedInteger i=0;i<nvars;i++){
205 if(_localvarinfos[i]._start_op<=nop && _localvarinfos[i]._end_op>=nop)
208 vm->Push(vm->_stack[stackbase+_localvarinfos[i]._pos]);
209 res=_stringval(_localvarinfos[i]._name);
219 SQInteger SQFunctionProto::GetLine(SQInstruction *curr)
221 SQInteger op = (SQInteger)(curr-_instructions);
222 SQInteger line=_lineinfos[0]._line;
223 for(SQInteger i=1;i<_nlineinfos;i++){
224 if(_lineinfos[i]._op>=op)
226 line=_lineinfos[i]._line;
231 #define _CHECK_IO(exp) { if(!exp)return false; }
232 bool SafeWrite(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQUserPointer dest,SQInteger size)
234 if(write(up,dest,size) != size) {
235 v->Raise_Error(_SC("io error (write function failure)"));
241 bool SafeRead(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQUserPointer dest,SQInteger size)
243 if(size && read(up,dest,size) != size) {
244 v->Raise_Error(_SC("io error, read function failure, the origin stream could be corrupted/trucated"));
250 bool WriteTag(HSQUIRRELVM v,SQWRITEFUNC write,SQUserPointer up,SQInteger tag)
252 return SafeWrite(v,write,up,&tag,sizeof(tag));
255 bool CheckTag(HSQUIRRELVM v,SQWRITEFUNC read,SQUserPointer up,SQInteger tag)
258 _CHECK_IO(SafeRead(v,read,up,&t,sizeof(t)));
260 v->Raise_Error(_SC("invalid or corrupted closure stream"));
266 bool WriteObject(HSQUIRRELVM v,SQUserPointer up,SQWRITEFUNC write,SQObjectPtr &o)
268 _CHECK_IO(SafeWrite(v,write,up,&type(o),sizeof(SQObjectType)));
271 _CHECK_IO(SafeWrite(v,write,up,&_string(o)->_len,sizeof(SQInteger)));
272 _CHECK_IO(SafeWrite(v,write,up,_stringval(o),rsl(_string(o)->_len)));
275 _CHECK_IO(SafeWrite(v,write,up,&_integer(o),sizeof(SQInteger)));break;
277 _CHECK_IO(SafeWrite(v,write,up,&_float(o),sizeof(SQFloat)));break;
281 v->Raise_Error(_SC("cannot serialize a %s"),GetTypeName(o));
287 bool ReadObject(HSQUIRRELVM v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &o)
290 _CHECK_IO(SafeRead(v,read,up,&t,sizeof(SQObjectType)));
294 _CHECK_IO(SafeRead(v,read,up,&len,sizeof(SQInteger)));
295 _CHECK_IO(SafeRead(v,read,up,_ss(v)->GetScratchPad(rsl(len)),rsl(len)));
296 o=SQString::Create(_ss(v),_ss(v)->GetScratchPad(-1),len);
301 _CHECK_IO(SafeRead(v,read,up,&i,sizeof(SQInteger))); o = i; break;
305 _CHECK_IO(SafeRead(v,read,up,&f,sizeof(SQFloat))); o = f; break;
311 v->Raise_Error(_SC("cannot serialize a %s"),IdType2Name(t));
317 bool SQClosure::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
319 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_HEAD));
320 _CHECK_IO(WriteTag(v,write,up,sizeof(SQChar)));
321 _CHECK_IO(_funcproto(_function)->Save(v,up,write));
322 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_TAIL));
326 bool SQClosure::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
328 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_HEAD));
329 _CHECK_IO(CheckTag(v,read,up,sizeof(SQChar)));
331 _CHECK_IO(SQFunctionProto::Load(v,up,read,func));
332 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_TAIL));
333 ret = SQClosure::Create(_ss(v),_funcproto(func));
337 bool SQFunctionProto::Save(SQVM *v,SQUserPointer up,SQWRITEFUNC write)
339 SQInteger i,nliterals = _nliterals,nparameters = _nparameters;
340 SQInteger noutervalues = _noutervalues,nlocalvarinfos = _nlocalvarinfos;
341 SQInteger nlineinfos=_nlineinfos,ninstructions = _ninstructions,nfunctions=_nfunctions;
342 SQInteger ndefaultparams = _ndefaultparams;
343 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
344 _CHECK_IO(WriteObject(v,up,write,_sourcename));
345 _CHECK_IO(WriteObject(v,up,write,_name));
346 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
347 _CHECK_IO(SafeWrite(v,write,up,&nliterals,sizeof(nliterals)));
348 _CHECK_IO(SafeWrite(v,write,up,&nparameters,sizeof(nparameters)));
349 _CHECK_IO(SafeWrite(v,write,up,&noutervalues,sizeof(noutervalues)));
350 _CHECK_IO(SafeWrite(v,write,up,&nlocalvarinfos,sizeof(nlocalvarinfos)));
351 _CHECK_IO(SafeWrite(v,write,up,&nlineinfos,sizeof(nlineinfos)));
352 _CHECK_IO(SafeWrite(v,write,up,&ndefaultparams,sizeof(ndefaultparams)));
353 _CHECK_IO(SafeWrite(v,write,up,&ninstructions,sizeof(ninstructions)));
354 _CHECK_IO(SafeWrite(v,write,up,&nfunctions,sizeof(nfunctions)));
355 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
356 for(i=0;i<nliterals;i++){
357 _CHECK_IO(WriteObject(v,up,write,_literals[i]));
360 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
361 for(i=0;i<nparameters;i++){
362 _CHECK_IO(WriteObject(v,up,write,_parameters[i]));
365 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
366 for(i=0;i<noutervalues;i++){
367 _CHECK_IO(SafeWrite(v,write,up,&_outervalues[i]._type,sizeof(SQUnsignedInteger)));
368 _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._src));
369 _CHECK_IO(WriteObject(v,up,write,_outervalues[i]._name));
372 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
373 for(i=0;i<nlocalvarinfos;i++){
374 SQLocalVarInfo &lvi=_localvarinfos[i];
375 _CHECK_IO(WriteObject(v,up,write,lvi._name));
376 _CHECK_IO(SafeWrite(v,write,up,&lvi._pos,sizeof(SQUnsignedInteger)));
377 _CHECK_IO(SafeWrite(v,write,up,&lvi._start_op,sizeof(SQUnsignedInteger)));
378 _CHECK_IO(SafeWrite(v,write,up,&lvi._end_op,sizeof(SQUnsignedInteger)));
381 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
382 _CHECK_IO(SafeWrite(v,write,up,_lineinfos,sizeof(SQLineInfo)*nlineinfos));
384 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
385 _CHECK_IO(SafeWrite(v,write,up,_defaultparams,sizeof(SQInteger)*ndefaultparams));
387 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
388 _CHECK_IO(SafeWrite(v,write,up,_instructions,sizeof(SQInstruction)*ninstructions));
390 _CHECK_IO(WriteTag(v,write,up,SQ_CLOSURESTREAM_PART));
391 for(i=0;i<nfunctions;i++){
392 _CHECK_IO(_funcproto(_functions[i])->Save(v,up,write));
394 _CHECK_IO(SafeWrite(v,write,up,&_stacksize,sizeof(_stacksize)));
395 _CHECK_IO(SafeWrite(v,write,up,&_bgenerator,sizeof(_bgenerator)));
396 _CHECK_IO(SafeWrite(v,write,up,&_varparams,sizeof(_varparams)));
400 bool SQFunctionProto::Load(SQVM *v,SQUserPointer up,SQREADFUNC read,SQObjectPtr &ret)
402 SQInteger i, nliterals,nparameters;
403 SQInteger noutervalues ,nlocalvarinfos ;
404 SQInteger nlineinfos,ninstructions ,nfunctions,ndefaultparams ;
405 SQObjectPtr sourcename, name;
407 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
408 _CHECK_IO(ReadObject(v, up, read, sourcename));
409 _CHECK_IO(ReadObject(v, up, read, name));
411 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
412 _CHECK_IO(SafeRead(v,read,up, &nliterals, sizeof(nliterals)));
413 _CHECK_IO(SafeRead(v,read,up, &nparameters, sizeof(nparameters)));
414 _CHECK_IO(SafeRead(v,read,up, &noutervalues, sizeof(noutervalues)));
415 _CHECK_IO(SafeRead(v,read,up, &nlocalvarinfos, sizeof(nlocalvarinfos)));
416 _CHECK_IO(SafeRead(v,read,up, &nlineinfos, sizeof(nlineinfos)));
417 _CHECK_IO(SafeRead(v,read,up, &ndefaultparams, sizeof(ndefaultparams)));
418 _CHECK_IO(SafeRead(v,read,up, &ninstructions, sizeof(ninstructions)));
419 _CHECK_IO(SafeRead(v,read,up, &nfunctions, sizeof(nfunctions)));
422 SQFunctionProto *f = SQFunctionProto::Create(ninstructions,nliterals,nparameters,
423 nfunctions,noutervalues,nlineinfos,nlocalvarinfos,ndefaultparams);
424 SQObjectPtr proto = f; //gets a ref in case of failure
425 f->_sourcename = sourcename;
428 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
430 for(i = 0;i < nliterals; i++){
431 _CHECK_IO(ReadObject(v, up, read, o));
434 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
436 for(i = 0; i < nparameters; i++){
437 _CHECK_IO(ReadObject(v, up, read, o));
438 f->_parameters[i] = o;
440 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
442 for(i = 0; i < noutervalues; i++){
443 SQUnsignedInteger type;
445 _CHECK_IO(SafeRead(v,read,up, &type, sizeof(SQUnsignedInteger)));
446 _CHECK_IO(ReadObject(v, up, read, o));
447 _CHECK_IO(ReadObject(v, up, read, name));
448 f->_outervalues[i] = SQOuterVar(name,o, (SQOuterType)type);
450 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
452 for(i = 0; i < nlocalvarinfos; i++){
454 _CHECK_IO(ReadObject(v, up, read, lvi._name));
455 _CHECK_IO(SafeRead(v,read,up, &lvi._pos, sizeof(SQUnsignedInteger)));
456 _CHECK_IO(SafeRead(v,read,up, &lvi._start_op, sizeof(SQUnsignedInteger)));
457 _CHECK_IO(SafeRead(v,read,up, &lvi._end_op, sizeof(SQUnsignedInteger)));
458 f->_localvarinfos[i] = lvi;
460 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
461 _CHECK_IO(SafeRead(v,read,up, f->_lineinfos, sizeof(SQLineInfo)*nlineinfos));
463 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
464 _CHECK_IO(SafeRead(v,read,up, f->_defaultparams, sizeof(SQInteger)*ndefaultparams));
466 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
467 _CHECK_IO(SafeRead(v,read,up, f->_instructions, sizeof(SQInstruction)*ninstructions));
469 _CHECK_IO(CheckTag(v,read,up,SQ_CLOSURESTREAM_PART));
470 for(i = 0; i < nfunctions; i++){
471 _CHECK_IO(_funcproto(o)->Load(v, up, read, o));
472 f->_functions[i] = o;
474 _CHECK_IO(SafeRead(v,read,up, &f->_stacksize, sizeof(f->_stacksize)));
475 _CHECK_IO(SafeRead(v,read,up, &f->_bgenerator, sizeof(f->_bgenerator)));
476 _CHECK_IO(SafeRead(v,read,up, &f->_varparams, sizeof(f->_varparams)));
482 #ifndef NO_GARBAGE_COLLECTOR
484 #define START_MARK() if(!(_uiRef&MARK_FLAG)){ \
487 #define END_MARK() RemoveFromChain(&_sharedstate->_gc_chain, this); \
488 AddToChain(chain, this); }
490 void SQVM::Mark(SQCollectable **chain)
493 SQSharedState::MarkObject(_lasterror,chain);
494 SQSharedState::MarkObject(_errorhandler,chain);
495 SQSharedState::MarkObject(_debughook,chain);
496 SQSharedState::MarkObject(_roottable, chain);
497 SQSharedState::MarkObject(temp_reg, chain);
498 for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
499 for(SQUnsignedInteger j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);
500 for(SQInteger k = 0; k < _callsstacksize; k++) SQSharedState::MarkObject(_callsstack[k]._closure, chain);
504 void SQArray::Mark(SQCollectable **chain)
507 SQInteger len = _values.size();
508 for(SQInteger i = 0;i < len; i++) SQSharedState::MarkObject(_values[i], chain);
511 void SQTable::Mark(SQCollectable **chain)
514 if(_delegate) _delegate->Mark(chain);
515 SQInteger len = _numofnodes;
516 for(SQInteger i = 0; i < len; i++){
517 SQSharedState::MarkObject(_nodes[i].key, chain);
518 SQSharedState::MarkObject(_nodes[i].val, chain);
523 void SQClass::Mark(SQCollectable **chain)
526 _members->Mark(chain);
527 if(_base) _base->Mark(chain);
528 SQSharedState::MarkObject(_attributes, chain);
529 for(SQUnsignedInteger i =0; i< _defaultvalues.size(); i++) {
530 SQSharedState::MarkObject(_defaultvalues[i].val, chain);
531 SQSharedState::MarkObject(_defaultvalues[i].attrs, chain);
533 for(SQUnsignedInteger j =0; j< _methods.size(); j++) {
534 SQSharedState::MarkObject(_methods[j].val, chain);
535 SQSharedState::MarkObject(_methods[j].attrs, chain);
537 for(SQUnsignedInteger k =0; k< _metamethods.size(); k++) {
538 SQSharedState::MarkObject(_metamethods[k], chain);
543 void SQInstance::Mark(SQCollectable **chain)
547 SQUnsignedInteger nvalues = _class->_defaultvalues.size();
548 for(SQUnsignedInteger i =0; i< nvalues; i++) {
549 SQSharedState::MarkObject(_values[i], chain);
554 void SQGenerator::Mark(SQCollectable **chain)
557 for(SQUnsignedInteger i = 0; i < _stack.size(); i++) SQSharedState::MarkObject(_stack[i], chain);
558 for(SQUnsignedInteger j = 0; j < _vargsstack.size(); j++) SQSharedState::MarkObject(_vargsstack[j], chain);
559 SQSharedState::MarkObject(_closure, chain);
563 void SQClosure::Mark(SQCollectable **chain)
566 for(SQUnsignedInteger i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);
567 for(SQUnsignedInteger i = 0; i < _defaultparams.size(); i++) SQSharedState::MarkObject(_defaultparams[i], chain);
571 void SQNativeClosure::Mark(SQCollectable **chain)
574 for(SQUnsignedInteger i = 0; i < _outervalues.size(); i++) SQSharedState::MarkObject(_outervalues[i], chain);
578 void SQUserData::Mark(SQCollectable **chain){
580 if(_delegate) _delegate->Mark(chain);
584 void SQCollectable::UnMark() { _uiRef&=~MARK_FLAG; }