1 #include <squirrel.h>
\r
3 #include <sqstdblob.h>
\r
5 #include "sqdbgserver.h"
\r
9 #define scstrcpy strcpy
\r
11 #define scstrcpy wcscpy
\r
18 #define SQDBG_DEBUG_HOOK _SC("_sqdbg_debug_hook_")
\r
19 #define SQDBG_ERROR_HANDLER _SC("_sqdbg_error_handler_")
\r
21 XMLEscape g_escapes[]={
\r
22 {_SC('<'),_SC("<")},{'>',_SC(">")},{_SC('&'),_SC("&")},{_SC('\''),_SC("'")},{_SC('\"'),_SC(""")},{_SC('\n'),_SC(""n")},{_SC('\r'),_SC(""r")},{NULL,NULL}
\r
25 const SQChar *IntToString(SQInteger n)
\r
27 static SQChar temp[256];
\r
28 scsprintf(temp,_SC("%d"),n);
\r
32 const SQChar *PtrToString(void *p)
\r
34 static SQChar temp[256];
\r
35 scsprintf(temp,_SC("%p"),p);
\r
39 SQInteger debug_hook(HSQUIRRELVM v);
\r
40 SQInteger error_handler(HSQUIRRELVM v);
\r
42 SQInteger beginelement(HSQUIRRELVM v)
\r
46 sq_getuserpointer(v,-1,&up);
\r
47 SQDbgServer *self = (SQDbgServer*)up;
\r
48 sq_getuserpointer(v,-1,&up);
\r
49 sq_getstring(v,2,&name);
\r
50 self->BeginElement(name);
\r
54 SQInteger endelement(HSQUIRRELVM v)
\r
58 sq_getuserpointer(v,-1,&up);
\r
59 SQDbgServer *self = (SQDbgServer*)up;
\r
60 sq_getuserpointer(v,-1,&up);
\r
61 sq_getstring(v,2,&name);
\r
62 self->EndElement(name);
\r
66 SQInteger attribute(HSQUIRRELVM v)
\r
69 const SQChar *name,*value;
\r
70 sq_getuserpointer(v,-1,&up);
\r
71 SQDbgServer *self = (SQDbgServer*)up;
\r
72 sq_getuserpointer(v,-1,&up);
\r
73 sq_getstring(v,2,&name);
\r
74 sq_getstring(v,3,&value);
\r
75 self->Attribute(name,value);
\r
79 SQDbgServer::SQDbgServer(HSQUIRRELVM v)
\r
83 _autoupdate = false;
\r
85 _state = eDBG_Running;
\r
86 _accept = INVALID_SOCKET;
\r
87 _endpoint = INVALID_SOCKET;
\r
88 //_maxrecursion = 10;
\r
89 sq_resetobject(&_debugroot);
\r
92 SQDbgServer::~SQDbgServer()
\r
94 VMStateMap::iterator itr = _vmstate.begin();
\r
95 while(itr != _vmstate.end()) {
\r
96 VMState *vs = itr->second;
\r
101 sq_pushobject(_v,_debugroot);
\r
103 sq_release(_v,&_debugroot);
\r
104 if(_accept != INVALID_SOCKET)
\r
105 sqdbg_closesocket(_accept);
\r
106 if(_endpoint != INVALID_SOCKET)
\r
107 sqdbg_closesocket(_endpoint);
\r
110 bool SQDbgServer::Init()
\r
112 //creates an environment table for the debugger
\r
115 sq_getstackobj(_v,-1,&_debugroot);
\r
116 sq_addref(_v,&_debugroot);
\r
118 //creates a emptyslot to store the watches
\r
119 sq_pushstring(_v,_SC("watches"),-1);
\r
121 sq_newslot(_v,-3, SQFalse);
\r
123 sq_pushstring(_v,_SC("beginelement"),-1);
\r
124 sq_pushuserpointer(_v,this);
\r
125 sq_newclosure(_v,beginelement,1);
\r
126 sq_setparamscheck(_v,2,_SC(".s"));
\r
127 sq_newslot(_v,-3, SQFalse);
\r
129 sq_pushstring(_v,_SC("endelement"),-1);
\r
130 sq_pushuserpointer(_v,this);
\r
131 sq_newclosure(_v,endelement,1);
\r
132 sq_setparamscheck(_v,2,_SC(".s"));
\r
133 sq_newslot(_v,-3, SQFalse);
\r
135 sq_pushstring(_v,_SC("attribute"),-1);
\r
136 sq_pushuserpointer(_v,this);
\r
137 sq_newclosure(_v,attribute,1);
\r
138 sq_setparamscheck(_v,3,_SC(".ss"));
\r
139 sq_newslot(_v,-3, SQFalse);
\r
143 //stores debug hook and error handler in the registry
\r
144 sq_pushregistrytable(_v);
\r
146 sq_pushstring(_v,SQDBG_DEBUG_HOOK,-1);
\r
147 sq_pushuserpointer(_v,this);
\r
148 sq_newclosure(_v,debug_hook,1);
\r
149 sq_newslot(_v,-3, SQFalse);
\r
151 sq_pushstring(_v,SQDBG_ERROR_HANDLER,-1);
\r
152 sq_pushuserpointer(_v,this);
\r
153 sq_newclosure(_v,error_handler,1);
\r
154 sq_newslot(_v,-3, SQFalse);
\r
159 //sets the error handlers
\r
160 SetErrorHandlers(_v);
\r
164 bool SQDbgServer::ReadMsg()
\r
169 void SQDbgServer::BusyWait()
\r
171 while( !ReadMsg() )
\r
177 void SQDbgServer::SendChunk(const SQChar *chunk)
\r
182 buf_len=(int)scstrlen(chunk)+1;
\r
183 buf=(char *)sq_getscratchpad(_v,(buf_len)*3);
\r
184 //wcstombs((char *)buf,chunk,buf_len*3);
\r
185 WideCharToMultiByte(CP_UTF8,0,chunk,-1,buf,buf_len*3,NULL,NULL);
\r
187 buf_len=(int)scstrlen(chunk);
\r
190 send(_endpoint,(const char*)buf,(int)strlen((const char *)buf),0);
\r
194 void SQDbgServer::Terminated()
\r
196 BeginElement(_SC("terminated"));
\r
197 EndElement(_SC("terminated"));
\r
201 VMState *SQDbgServer::GetVMState(HSQUIRRELVM v)
\r
203 VMState *ret = NULL;
\r
204 VMStateMap::iterator itr = _vmstate.find(v);
\r
205 if(itr == _vmstate.end()) {
\r
206 ret = new VMState();
\r
207 _vmstate.insert(VMStateMap::value_type(v,ret));
\r
215 void SQDbgServer::Hook(HSQUIRRELVM v,SQInteger type,SQInteger line,const SQChar *src,const SQChar *func)
\r
218 VMState *vs = GetVMState(v);
\r
221 if(type==_SC('l') && _breakpoints.size()) {
\r
222 BreakPointSetItor itr = _breakpoints.find(BreakPoint(line,src));
\r
223 if(itr != _breakpoints.end()) {
\r
224 Break(v,line,src,_SC("breakpoint"));
\r
229 case eDBG_Suspended:
\r
230 vs->_nestedcalls=0;
\r
231 case eDBG_StepOver:
\r
234 if(vs->_nestedcalls==0) {
\r
235 Break(v,line,src,_SC("step"));
\r
240 vs->_nestedcalls++;
\r
243 if(vs->_nestedcalls==0){
\r
244 vs->_nestedcalls=0;
\r
247 vs->_nestedcalls--;
\r
252 case eDBG_StepInto:
\r
255 vs->_nestedcalls=0;
\r
256 Break(v,line,src,_SC("step"));
\r
262 case eDBG_StepReturn:
\r
267 vs->_nestedcalls++;
\r
270 if(vs->_nestedcalls==0){
\r
271 vs->_nestedcalls=0;
\r
272 _state=eDBG_StepOver;
\r
274 vs->_nestedcalls--;
\r
280 case eDBG_Disabled:
\r
286 #define MSG_ID(x,y) ((y<<8)|x)
\r
287 //ab Add Breakpoint
\r
288 //rb Remove Breakpoint
\r
290 void SQDbgServer::ParseMsg(const char *msg)
\r
293 switch(*((unsigned short *)msg)){
\r
294 case MSG_ID('a','b'): {
\r
296 if(ParseBreakpoint(msg+3,bp)){
\r
298 scprintf(_SC("added bp %d %s\n"),bp._line,bp._src.c_str());
\r
301 scprintf(_SC("error parsing add breakpoint"));
\r
304 case MSG_ID('r','b'): {
\r
306 if(ParseBreakpoint(msg+3,bp)){
\r
307 RemoveBreakpoint(bp);
\r
308 scprintf(_SC("removed bp %d %s\n"),bp._line,bp._src.c_str());
\r
310 scprintf(_SC("error parsing remove breakpoint"));
\r
313 case MSG_ID('g','o'):
\r
314 if(_state!=eDBG_Running){
\r
315 _state=eDBG_Running;
\r
317 BeginElement(_SC("resumed"));
\r
318 EndElement(_SC("resumed"));
\r
320 // Send(_SC("<resumed/>\r\n"));
\r
321 scprintf(_SC("go (execution resumed)\n"));
\r
324 case MSG_ID('s','p'):
\r
325 if(_state!=eDBG_Suspended){
\r
326 _state=eDBG_Suspended;
\r
327 scprintf(_SC("suspend\n"));
\r
330 case MSG_ID('s','o'):
\r
331 if(_state==eDBG_Suspended){
\r
332 _state=eDBG_StepOver;
\r
335 case MSG_ID('s','i'):
\r
336 if(_state==eDBG_Suspended){
\r
337 _state=eDBG_StepInto;
\r
338 scprintf(_SC("step into\n"));
\r
341 case MSG_ID('s','r'):
\r
342 if(_state==eDBG_Suspended){
\r
343 _state=eDBG_StepReturn;
\r
344 scprintf(_SC("step return\n"));
\r
347 case MSG_ID('d','i'):
\r
348 if(_state!=eDBG_Disabled){
\r
349 _state=eDBG_Disabled;
\r
350 scprintf(_SC("disabled\n"));
\r
353 case MSG_ID('a','w'): {
\r
355 if(ParseWatch(msg+3,w))
\r
358 scprintf(_SC("added watch %d %s\n"),w._id,w._exp.c_str());
\r
359 /*if(_state == eDBG_Suspended) {
\r
360 Break(_line,_src.c_str(),_break_type.c_str());
\r
364 scprintf(_SC("error parsing add watch"));
\r
367 case MSG_ID('r','w'): {
\r
369 if(ParseRemoveWatch(msg+3,id))
\r
372 scprintf(_SC("added watch %d\n"),id);
\r
375 scprintf(_SC("error parsing remove watch"));
\r
378 case MSG_ID('t','r'):
\r
379 scprintf(_SC("terminate from user\n"));
\r
381 case MSG_ID('r','d'):
\r
382 scprintf(_SC("ready\n"));
\r
386 scprintf(_SC("unknown packet"));
\r
391 bool SQDbgServer::ParseBreakpoint(const char *msg,BreakPoint &out)
\r
393 static char stemp[MAX_BP_PATH];
\r
394 static SQChar desttemp[MAX_BP_PATH];
\r
396 out._line=strtoul(msg,&ep,16);
\r
397 if(ep==msg || (*ep)!=':')return false;
\r
401 while((*ep)!='\n' && (*ep)!='\0')
\r
403 *dest=tolower(*ep);
\r
410 int len=(int)strlen(stemp);
\r
411 SQChar *p = desttemp;
\r
412 size_t destlen = mbstowcs(p,stemp,len);
\r
413 p[destlen]=_SC('\0');
\r
421 bool SQDbgServer::ParseWatch(const char *msg,Watch &out)
\r
424 out._id=strtoul(msg,&ep,16);
\r
425 if(ep==msg || (*ep)!=':')return false;
\r
427 //char *dest=out._src;
\r
429 while((*ep)!='\n' && (*ep)!='\0')
\r
431 out._exp.append(1,*ep);
\r
437 bool SQDbgServer::ParseRemoveWatch(const char *msg,SQInteger &id)
\r
440 id=strtoul(msg,&ep,16);
\r
441 if(ep==msg)return false;
\r
446 void SQDbgServer::BreakExecution()
\r
448 _state=eDBG_Suspended;
\r
449 while(_state==eDBG_Suspended){
\r
450 if(SQ_FAILED(sq_rdbg_update(this)))
\r
457 void SQDbgServer::AddBreakpoint(BreakPoint &bp)
\r
459 _breakpoints.insert(bp);
\r
461 BeginElement(_SC("addbreakpoint"));
\r
462 Attribute(_SC("line"),IntToString(bp._line));
\r
463 Attribute(_SC("src"),bp._src.c_str());
\r
464 EndElement(_SC("addbreakpoint"));
\r
468 void SQDbgServer::AddWatch(Watch &w)
\r
470 _watches.insert(w);
\r
473 void SQDbgServer::RemoveWatch(SQInteger id)
\r
475 WatchSetItor itor=_watches.find(Watch(id,_SC("")));
\r
476 if(itor==_watches.end()){
\r
478 BeginElement(_SC("error"));
\r
479 Attribute(_SC("desc"),_SC("the watch does not exists"));
\r
480 EndElement(_SC("error"));
\r
484 _watches.erase(itor);
\r
485 scprintf(_SC("removed watch %d\n"),id);
\r
489 void SQDbgServer::RemoveBreakpoint(BreakPoint &bp)
\r
491 BreakPointSetItor itor=_breakpoints.find(bp);
\r
492 if(itor==_breakpoints.end()){
\r
494 BeginElement(_SC("break"));
\r
495 Attribute(_SC("desc"),_SC("the breakpoint doesn't exists"));
\r
496 EndElement(_SC("break"));
\r
501 BeginElement(_SC("removebreakpoint"));
\r
502 Attribute(_SC("line"),IntToString(bp._line));
\r
503 Attribute(_SC("src"),bp._src.c_str());
\r
504 EndElement(_SC("removebreakpoint"));
\r
506 _breakpoints.erase(itor);
\r
510 void SQDbgServer::Break(HSQUIRRELVM v,SQInteger line,const SQChar *src,const SQChar *type,const SQChar *error)
\r
517 BeginElement(_SC("break"));
\r
518 Attribute(_SC("thread"),PtrToString(v));
\r
519 Attribute(_SC("line"),IntToString(line));
\r
520 Attribute(_SC("src"),src);
\r
521 Attribute(_SC("type"),type);
\r
523 EndElement(_SC("break"));
\r
527 BeginElement(_SC("break"));
\r
528 Attribute(_SC("thread"),PtrToString(v));
\r
529 Attribute(_SC("line"),IntToString(line));
\r
530 Attribute(_SC("src"),src);
\r
531 Attribute(_SC("type"),type);
\r
532 Attribute(_SC("error"),error);
\r
534 EndElement(_SC("break"));
\r
539 void SQDbgServer::SerializeState(HSQUIRRELVM v)
\r
542 sq_setdebughook(v);
\r
544 sq_seterrorhandler(v);
\r
545 sq_pushobject(v,_serializefunc);
\r
546 sq_pushobject(v,_debugroot);
\r
547 sq_pushstring(v,_SC("watches"),-1);
\r
549 for(WatchSetItor i=_watches.begin(); i!=_watches.end(); ++i)
\r
551 sq_pushinteger(v,i->_id);
\r
552 sq_pushstring(v,i->_exp.c_str(),(SQInteger)i->_exp.length());
\r
553 sq_createslot(v,-3);
\r
556 if(SQ_SUCCEEDED(sq_call(v,1,SQTrue,SQFalse))){
\r
557 //if(SQ_SUCCEEDED(sqstd_getblob(v,-1,(SQUserPointer*)&sz)))
\r
562 SetErrorHandlers(v);
\r
566 void SQDbgServer::SetErrorHandlers(HSQUIRRELVM v)
\r
568 sq_pushregistrytable(v);
\r
569 sq_pushstring(v,SQDBG_DEBUG_HOOK,-1);
\r
571 sq_setdebughook(v);
\r
572 sq_pushstring(v,SQDBG_ERROR_HANDLER,-1);
\r
574 sq_seterrorhandler(v);
\r
578 void SQDbgServer::BeginDocument()
\r
580 _xmlcurrentement = -1;
\r
581 SendChunk(_SC("<?xml version='1.0' encoding='utf-8'?>"));
\r
584 void SQDbgServer::BeginElement(const SQChar *name)
\r
586 _xmlcurrentement++;
\r
587 XMLElementState *self = &xmlstate[_xmlcurrentement];
\r
588 scstrcpy(self->name,name);
\r
589 self->haschildren = false;
\r
590 if(_xmlcurrentement > 0) {
\r
591 XMLElementState *parent = &xmlstate[_xmlcurrentement-1];
\r
592 if(!parent->haschildren) {
\r
593 SendChunk(_SC(">")); // closes the parent tag
\r
594 parent->haschildren = true;
\r
597 _scratchstring.resize(2+scstrlen(name));
\r
598 scsprintf(&_scratchstring[0],_SC("<%s"),name);
\r
599 SendChunk(&_scratchstring[0]);
\r
602 void SQDbgServer::Attribute(const SQChar *name,const SQChar *value)
\r
604 XMLElementState *self = &xmlstate[_xmlcurrentement];
\r
605 assert(!self->haschildren); //cannot have attributes if already has children
\r
606 const SQChar *escval = escape_xml(value);
\r
607 _scratchstring.resize(10+scstrlen(name)+scstrlen(escval));
\r
608 scsprintf(&_scratchstring[0],_SC(" %s=\"%s\""),name,escval);
\r
609 SendChunk(&_scratchstring[0]);
\r
612 void SQDbgServer::EndElement(const SQChar *name)
\r
614 XMLElementState *self = &xmlstate[_xmlcurrentement];
\r
615 assert(scstrcmp(self->name,name) == 0);
\r
616 if(self->haschildren) {
\r
617 _scratchstring.resize(10+scstrlen(name));
\r
618 scsprintf(&_scratchstring[0],_SC("</%s>"),name);
\r
619 SendChunk(&_scratchstring[0]);
\r
623 SendChunk(_SC("/>"));
\r
625 _xmlcurrentement--;
\r
628 void SQDbgServer::EndDocument()
\r
630 SendChunk(_SC("\r\n"));
\r
633 //this can be done much better/faster(do we need that?)
\r
634 const SQChar *SQDbgServer::escape_xml(const SQChar *s)
\r
636 SQChar *temp=sq_getscratchpad(_v,((SQInteger)scstrlen(s)*6) + sizeof SQChar);
\r
638 while(*s!=_SC('\0')){
\r
640 const SQChar *escape = NULL;
\r
642 case _SC('<'): escape = _SC("<"); break;
\r
643 case _SC('>'): escape = _SC(">"); break;
\r
644 case _SC('&'): escape = _SC("&"); break;
\r
645 case _SC('\''): escape = _SC("'"); break;
\r
646 case _SC('\"'): escape = _SC("""); break;
\r
647 case _SC('\n'): escape = _SC("\\n"); break;
\r
648 case _SC('\r'): escape = _SC("\\r"); break;
\r
651 scstrcpy(dest,escape);
\r
652 dest += scstrlen(escape);
\r