sqlexer.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870
  1. /*
  2. see copyright notice in squirrel.h
  3. */
  4. #include "sqpcheader.h"
  5. #include <ctype.h>
  6. #include <stdlib.h>
  7. #include <limits.h>
  8. #include "sqtable.h"
  9. #include "sqstring.h"
  10. #include "sqcompiler.h"
  11. #include "sqlexer.h"
  12. #define CUR_CHAR (data->currdata)
  13. #define RETURN_TOKEN(t) { data->prevtoken = data->curtoken; data->curtoken = t; return t;}
  14. #define IS_EOB() (CUR_CHAR <= SQUIRREL_EOB)
  15. //#define NEXT() {SQInteger rc = Next(); if(rc < 0) return rc; data->currentcolumn++;}
  16. #define NEXT() {if(Next()) return -1;}
  17. #define INIT_TEMP_STRING() { data->longstr.resize(0);}
  18. #define APPEND_CHAR(c) { data->longstr.push_back(c);}
  19. #define TERMINATE_BUFFER() {data->longstr.push_back(_SC('\0'));}
  20. #define ADD_KEYWORD(key,id) tbl->NewSlot( SQString::Create(_sharedstate, _SC(#key)) ,SQInteger(id))
  21. SQLexer::SQLexer(){_keywords=0;}
  22. SQLexer::~SQLexer()
  23. {
  24. _keywords->Release();
  25. }
  26. SQInteger SQLexer::Init(SQSharedState *ss, SQLEXREADFUNC rg,
  27. SQUserPointer up,CompilerErrorFunc efunc,void *ed, SQBool want_comments)
  28. {
  29. _want_comments = want_comments;
  30. data = &_data;
  31. _data_lookahead.currentline = -1;
  32. _errfunc = efunc;
  33. _errtarget = ed;
  34. _sharedstate = ss;
  35. if(_keywords) _keywords->Release();
  36. _keywords = GetKeywords();
  37. return ResetReader(rg, up, 1);
  38. }
  39. SQInteger SQLexer::ResetReader(SQLEXREADFUNC rg, SQUserPointer up, SQInteger line)
  40. {
  41. _readf = rg;
  42. _up = up;
  43. data->lasttokenline = data->currentline = line;
  44. data->lasttokencolumn = 0;
  45. data->currentcolumn = 0;
  46. data->prevtoken = -1;
  47. data->reached_eof = SQFalse;
  48. return Next();
  49. }
  50. SQTable * SQLexer::GetKeywords()
  51. {
  52. SQTable *tbl = SQTable::Create(_sharedstate, (TK_LAST_ENUM_TOKEN - TK_FIRST_ENUM_TOKEN - 1) /*26*/);
  53. ADD_KEYWORD(any_t, TK_LOCAL_ANY_T);
  54. ADD_KEYWORD(array_t, TK_LOCAL_ARRAY_T);
  55. ADD_KEYWORD(as, TK_AS);
  56. ADD_KEYWORD(auto, TK_LOCAL);
  57. ADD_KEYWORD(base, TK_BASE);
  58. ADD_KEYWORD(bool_t, TK_LOCAL_BOOL_T);
  59. ADD_KEYWORD(break, TK_BREAK);
  60. ADD_KEYWORD(case, TK_CASE);
  61. ADD_KEYWORD(catch, TK_CATCH);
  62. ADD_KEYWORD(char_t, TK_LOCAL_CHAR_T);
  63. ADD_KEYWORD(class,TK_CLASS);
  64. ADD_KEYWORD(clone, TK_CLONE);
  65. ADD_KEYWORD(constructor,TK_CONSTRUCTOR);
  66. ADD_KEYWORD(const,TK_CONST);
  67. ADD_KEYWORD(constexpr,TK_CONSTEXPR);
  68. ADD_KEYWORD(continue, TK_CONTINUE);
  69. ADD_KEYWORD(declare, TK_DECLARE);
  70. ADD_KEYWORD(default, TK_DEFAULT);
  71. ADD_KEYWORD(delete, TK_DELETE);
  72. ADD_KEYWORD(destructor,TK_DESTRUCTOR);
  73. ADD_KEYWORD(do, TK_DO);
  74. ADD_KEYWORD(double_t, TK_LOCAL_DOUBLE_T);
  75. ADD_KEYWORD(else, TK_ELSE);
  76. ADD_KEYWORD(enum,TK_ENUM);
  77. ADD_KEYWORD(extends,TK_EXTENDS);
  78. ADD_KEYWORD(extern,TK_EXTERN);
  79. ADD_KEYWORD(false,TK_FALSE);
  80. ADD_KEYWORD(__FILE__,TK___FILE__);
  81. ADD_KEYWORD(float_t, TK_LOCAL_FLOAT_T);
  82. ADD_KEYWORD(foreach, TK_FOREACH);
  83. ADD_KEYWORD(for, TK_FOR);
  84. ADD_KEYWORD(friend, TK_FRIEND);
  85. ADD_KEYWORD(function, TK_FUNCTION);
  86. ADD_KEYWORD(__FUNCTION__,TK___FUNCTION__);
  87. ADD_KEYWORD(goto, TK_GOTO);
  88. ADD_KEYWORD(if, TK_IF);
  89. ADD_KEYWORD(instanceof,TK_INSTANCEOF);
  90. ADD_KEYWORD(int16_t, TK_LOCAL_INT16_T);
  91. ADD_KEYWORD(int32_t, TK_LOCAL_INT32_T);
  92. ADD_KEYWORD(int64_t, TK_LOCAL_INT64_T);
  93. ADD_KEYWORD(int8_t, TK_LOCAL_INT8_T);
  94. ADD_KEYWORD(in, TK_IN);
  95. ADD_KEYWORD(inline, TK_INLINE);
  96. ADD_KEYWORD(int_t, TK_LOCAL_INT_T);
  97. ADD_KEYWORD(let, TK_LOCAL);
  98. ADD_KEYWORD(__LINE__,TK___LINE__);
  99. ADD_KEYWORD(local, TK_LOCAL);
  100. ADD_KEYWORD(long_double_t, TK_LOCAL_LONG_DOUBLE_T);
  101. ADD_KEYWORD(new,TK_IGNORE);
  102. ADD_KEYWORD(noexcept,TK_NOEXCEPT);
  103. ADD_KEYWORD(number_t, TK_LOCAL_NUMBER_T);
  104. ADD_KEYWORD(null, TK_NULL);
  105. ADD_KEYWORD(NULL, TK_NULL);
  106. ADD_KEYWORD(private,TK_PRIVATE);
  107. ADD_KEYWORD(public,TK_PUBLIC);
  108. ADD_KEYWORD(rawcall, TK_RAWCALL);
  109. ADD_KEYWORD(resume, TK_RESUME);
  110. ADD_KEYWORD(return, TK_RETURN);
  111. ADD_KEYWORD(size_t,TK_LOCAL_SIZE_T);
  112. ADD_KEYWORD(ssize_t,TK_LOCAL_SSIZE_T);
  113. ADD_KEYWORD(static,TK_STATIC);
  114. ADD_KEYWORD(string_t, TK_LOCAL_STRING_T);
  115. ADD_KEYWORD(struct,TK_STRUCT);
  116. ADD_KEYWORD(switch, TK_SWITCH);
  117. ADD_KEYWORD(table_t, TK_LOCAL_TABLE_T);
  118. ADD_KEYWORD(template, TK_TEMPLATE);
  119. ADD_KEYWORD(this, TK_THIS);
  120. ADD_KEYWORD(throw, TK_THROW);
  121. ADD_KEYWORD(typedef, TK_TYPEDEF);
  122. ADD_KEYWORD(true,TK_TRUE);
  123. ADD_KEYWORD(try, TK_TRY);
  124. ADD_KEYWORD(typeof, TK_TYPEOF);
  125. ADD_KEYWORD(uint16_t, TK_LOCAL_UINT16_T);
  126. ADD_KEYWORD(uint32_t, TK_LOCAL_UINT32_T);
  127. ADD_KEYWORD(uint64_t, TK_LOCAL_UINT64_T);
  128. ADD_KEYWORD(uint8_t, TK_LOCAL_UINT8_T);
  129. ADD_KEYWORD(uint_t, TK_LOCAL_UINT_T);
  130. ADD_KEYWORD(unsafe, TK_UNSAFE);
  131. ADD_KEYWORD(using, TK_USING);
  132. ADD_KEYWORD(var, TK_LOCAL);
  133. ADD_KEYWORD(virtual, TK_VIRTUAL);
  134. ADD_KEYWORD(void_ptr_t, TK_LOCAL_VOIDPTR_T);
  135. ADD_KEYWORD(void, TK_VOID);
  136. ADD_KEYWORD(volatile, TK_VOLATILE);
  137. ADD_KEYWORD(wchar_t, TK_LOCAL_WCHAR_T);
  138. ADD_KEYWORD(weakref_t, TK_LOCAL_WEAKREF_T);
  139. ADD_KEYWORD(while, TK_WHILE);
  140. ADD_KEYWORD(yield, TK_YIELD);
  141. return tbl;
  142. }
  143. SQInteger SQLexer::Error(const SQChar *fmt, ...)
  144. {
  145. va_list vl;
  146. va_start(vl, fmt);
  147. scvsprintf(data->lasterror, sizeof(data->lasterror), fmt, vl);
  148. va_end(vl);
  149. if(_errfunc) _errfunc(_errtarget,data->lasterror);
  150. return -1;
  151. }
  152. SQInteger SQLexer::Next()
  153. {
  154. SQInteger t = _readf(_up);
  155. if(t > MAX_CHAR) return Error(_SC("Invalid character"));
  156. if(t != 0) {
  157. data->currdata = (LexChar)t;
  158. ++data->currentcolumn;
  159. return 0;
  160. }
  161. data->currdata = SQUIRREL_EOB;
  162. data->reached_eof = SQTrue;
  163. return 0;
  164. }
  165. const SQChar *SQLexer::Tok2Str(SQInteger tok)
  166. {
  167. SQObjectPtr itr, key, val;
  168. SQInteger nitr;
  169. while((nitr = _keywords->Next(false,itr, key, val)) != -1) {
  170. itr = (SQInteger)nitr;
  171. if(((SQInteger)_integer(val)) == tok)
  172. return _stringval(key);
  173. }
  174. return NULL;
  175. }
  176. const SQChar *SQLexer::GetTokenName(int tk_code) {
  177. const SQChar *str_tk;
  178. switch(tk_code){
  179. #define ENUM_TK(a) case TK_##a: str_tk = _SC("TK_" #a); break;
  180. SQ_KEYWORDS_LIST()
  181. #undef ENUM_TK
  182. default:
  183. str_tk = _SC("()");
  184. }
  185. return str_tk;
  186. }
  187. SQInteger SQLexer::LexBlockComment()
  188. {
  189. /*
  190. if(CUR_CHAR == _SC('*'))
  191. {
  192. NEXT();
  193. if(CUR_CHAR != _SC('*')){ //document comment
  194. printf("Doument comment found at line %d\n", data->currentline);
  195. }
  196. }
  197. */
  198. bool done = false;
  199. if(_want_comments) INIT_TEMP_STRING();
  200. NEXT(); //remove the comment token '*'
  201. while(!done) {
  202. switch(CUR_CHAR) {
  203. case _SC('*'): {
  204. NEXT();
  205. if(CUR_CHAR == _SC('/')) { done = true; NEXT(); continue;}
  206. if(_want_comments) APPEND_CHAR(CUR_CHAR);
  207. continue;
  208. };
  209. break;
  210. case _SC('\n'): data->currentline++; break;
  211. case SQUIRREL_EOB: return Error(_SC("missing \"*/\" in comment"));
  212. }
  213. if(_want_comments) APPEND_CHAR(CUR_CHAR);
  214. NEXT();
  215. }
  216. if(_want_comments)
  217. {
  218. TERMINATE_BUFFER();
  219. if(data->longstr.size() > 0) data->longstr.pop_back(); //remove the last '*'
  220. data->svalue = &data->longstr[0];
  221. }
  222. return 0;
  223. }
  224. SQInteger SQLexer::LexLineComment()
  225. {
  226. if(_want_comments) INIT_TEMP_STRING();
  227. NEXT(); //remove the comment token
  228. while (CUR_CHAR != _SC('\n') && (!IS_EOB())) {if(_want_comments) APPEND_CHAR(CUR_CHAR); NEXT();}
  229. if(_want_comments)
  230. {
  231. TERMINATE_BUFFER();
  232. data->svalue = &data->longstr[0];
  233. }
  234. return 0;
  235. }
  236. SQInteger SQLexer::LookaheadLex()
  237. {
  238. if(CUR_CHAR == SQUIRREL_EOB) return 0;
  239. if(_data_lookahead.currentline >= 0)
  240. {
  241. return Error(_SC("lex lookahead already done"));
  242. }
  243. _data_lookahead.copy(&_data);
  244. data = &_data_lookahead;
  245. Lex();
  246. data = &_data;
  247. return _data_lookahead.curtoken;
  248. }
  249. SQInteger SQLexer::Lex()
  250. {
  251. if(_data_lookahead.currentline >= 0 && data != &_data_lookahead)
  252. {
  253. //we did a lookahead before, reuse it now
  254. _data.copy(&_data_lookahead);
  255. _data_lookahead.currentline = -1;
  256. return _data.curtoken;
  257. }
  258. data->lasttokenline = data->currentline;
  259. data->lasttokencolumn = data->currentcolumn;
  260. while(CUR_CHAR != SQUIRREL_EOB) {
  261. switch(CUR_CHAR){
  262. case _SC('\t'): case _SC('\r'): case _SC(' '): NEXT(); continue;
  263. case _SC('\n'):
  264. data->currentline++;
  265. data->prevtoken=data->curtoken;
  266. data->curtoken=_SC('\n');
  267. NEXT();
  268. data->currentcolumn=1;
  269. continue;
  270. case _SC('#'):
  271. NEXT();
  272. if(CUR_CHAR == '!') //shell shebang
  273. {
  274. if(LexLineComment()) return -1;
  275. if(_want_comments) RETURN_TOKEN(TK_COMMENT_LINE)
  276. continue;
  277. }
  278. RETURN_TOKEN(TK_PRAGMA);
  279. continue;
  280. case _SC('/'):
  281. NEXT();
  282. switch(CUR_CHAR){
  283. case _SC('*'):
  284. if(LexBlockComment()) return -1;
  285. if(_want_comments) RETURN_TOKEN(TK_COMMENT_BLOCK)
  286. continue;
  287. case _SC('/'):
  288. if(LexLineComment()) return -1;
  289. if(_want_comments) RETURN_TOKEN(TK_COMMENT_LINE)
  290. continue;
  291. case _SC('='):
  292. NEXT();
  293. RETURN_TOKEN(TK_DIVEQ);
  294. continue;
  295. case _SC('>'):
  296. NEXT();
  297. RETURN_TOKEN(TK_ATTR_CLOSE);
  298. continue;
  299. default:
  300. RETURN_TOKEN('/');
  301. }
  302. case _SC('='):
  303. NEXT();
  304. if (CUR_CHAR != _SC('=')){ RETURN_TOKEN('=') }
  305. else {
  306. NEXT();
  307. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_EQ_IDENTITY) }
  308. else { RETURN_TOKEN(TK_EQ); }
  309. }
  310. case _SC('<'):
  311. NEXT();
  312. switch(CUR_CHAR) {
  313. case _SC('='):
  314. NEXT();
  315. if(CUR_CHAR == _SC('>')) {
  316. NEXT();
  317. RETURN_TOKEN(TK_3WAYSCMP);
  318. }
  319. RETURN_TOKEN(TK_LE)
  320. break;
  321. case _SC('-'): NEXT(); RETURN_TOKEN(TK_NEWSLOT); break;
  322. case _SC('<'): NEXT(); RETURN_TOKEN(TK_SHIFTL); break;
  323. case _SC('/'): NEXT(); RETURN_TOKEN(TK_ATTR_OPEN); break;
  324. }
  325. RETURN_TOKEN('<');
  326. case _SC('>'):
  327. NEXT();
  328. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_GE);}
  329. else if(CUR_CHAR == _SC('>')){
  330. NEXT();
  331. if(CUR_CHAR == _SC('>')){
  332. NEXT();
  333. RETURN_TOKEN(TK_USHIFTR);
  334. }
  335. RETURN_TOKEN(TK_SHIFTR);
  336. }
  337. else { RETURN_TOKEN('>') }
  338. case _SC('!'):
  339. NEXT();
  340. if (CUR_CHAR != _SC('=')){ RETURN_TOKEN('!')}
  341. else {
  342. NEXT();
  343. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_NE_IDENTITY)}
  344. else { RETURN_TOKEN(TK_NE); }
  345. }
  346. case _SC('@'): {
  347. SQInteger stype;
  348. NEXT();
  349. if(CUR_CHAR != _SC('"')) {
  350. RETURN_TOKEN('@');
  351. }
  352. if((stype=ReadString('"',true))!=-1) {
  353. RETURN_TOKEN(stype);
  354. }
  355. return Error(_SC("error parsing the string"));
  356. }
  357. case _SC('"'):
  358. case _SC('\''): {
  359. SQInteger stype;
  360. if((stype=ReadString(CUR_CHAR,false))!=-1){
  361. RETURN_TOKEN(stype);
  362. }
  363. return Error(_SC("error parsing the string"));
  364. }
  365. case _SC('{'): case _SC('}'): case _SC('('): case _SC(')'): case _SC('['): case _SC(']'):
  366. case _SC(';'): case _SC(','): case _SC('?'): case _SC('~'):
  367. {
  368. SQInteger ret = CUR_CHAR;
  369. NEXT();
  370. if((ret == _SC('[') || ret == _SC('{') || ret == _SC('(')) && CUR_CHAR == _SC('=')){
  371. //lets try lua literal delimiters
  372. SQInteger stype;
  373. if((stype=ReadString(ret,true))!=-1){
  374. RETURN_TOKEN(stype);
  375. }
  376. return Error(_SC("error parsing the string"));
  377. }
  378. else RETURN_TOKEN(ret);
  379. }
  380. case _SC('.'):
  381. NEXT();
  382. if (CUR_CHAR != _SC('.')){ RETURN_TOKEN('.') }
  383. NEXT();
  384. if (CUR_CHAR != _SC('.')){ return Error(_SC("invalid token '..'")); }
  385. NEXT();
  386. RETURN_TOKEN(TK_VARPARAMS);
  387. case _SC('^'):
  388. NEXT();
  389. //if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_BIT_XOR_EQ);}
  390. RETURN_TOKEN('^');
  391. case _SC('&'):
  392. NEXT();
  393. //if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_BIT_AND_EQ);}
  394. if (CUR_CHAR != _SC('&')){ RETURN_TOKEN('&') }
  395. else { NEXT(); RETURN_TOKEN(TK_AND); }
  396. case _SC('|'):
  397. NEXT();
  398. //if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_BIT_OR_EQ);}
  399. if (CUR_CHAR != _SC('|')){ RETURN_TOKEN('|') }
  400. else { NEXT(); RETURN_TOKEN(TK_OR); }
  401. case _SC(':'):
  402. NEXT();
  403. if (CUR_CHAR != _SC(':')){ RETURN_TOKEN(':') }
  404. else { NEXT(); RETURN_TOKEN(TK_DOUBLE_COLON); }
  405. case _SC('*'):
  406. NEXT();
  407. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_MULEQ);}
  408. else RETURN_TOKEN('*');
  409. case _SC('%'):
  410. NEXT();
  411. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_MODEQ);}
  412. else RETURN_TOKEN('%');
  413. case _SC('-'):
  414. NEXT();
  415. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_MINUSEQ);}
  416. else if (CUR_CHAR == _SC('-')){ NEXT(); RETURN_TOKEN(TK_MINUSMINUS);}
  417. else if (CUR_CHAR == _SC('>')){ NEXT(); RETURN_TOKEN('.');} //accept C/C++ like pointers
  418. else RETURN_TOKEN('-');
  419. case _SC('+'):
  420. NEXT();
  421. if (CUR_CHAR == _SC('=')){ NEXT(); RETURN_TOKEN(TK_PLUSEQ);}
  422. else if (CUR_CHAR == _SC('+')){ NEXT(); RETURN_TOKEN(TK_PLUSPLUS);}
  423. else RETURN_TOKEN('+');
  424. case SQUIRREL_EOB:
  425. return 0;
  426. default:{
  427. if (scisdigit(CUR_CHAR)) {
  428. SQInteger ret = ReadNumber();
  429. if(ret < 0) return -1;
  430. RETURN_TOKEN(ret);
  431. }
  432. else if (scisalpha(CUR_CHAR) || CUR_CHAR == _SC('_')) {
  433. SQInteger t = ReadID();
  434. if(t < 0) return -1;
  435. RETURN_TOKEN(t);
  436. }
  437. else {
  438. SQInteger c = CUR_CHAR;
  439. if (sciscntrl((int)c)) return Error(_SC("unexpected character(control)"));
  440. NEXT();
  441. RETURN_TOKEN(c);
  442. }
  443. RETURN_TOKEN(0);
  444. }
  445. }
  446. }
  447. return 0;
  448. }
  449. SQInteger SQLexer::GetIDType(const SQChar *s,SQInteger len)
  450. {
  451. SQObjectPtr t;
  452. if(_keywords->GetStr(s,len, t)) {
  453. return SQInteger(_integer(t));
  454. }
  455. return TK_IDENTIFIER;
  456. }
  457. #ifdef SQUNICODE
  458. #if WCHAR_SIZE == 2
  459. SQInteger SQLexer::AddUTF16(SQUnsignedInteger ch)
  460. {
  461. if (ch >= 0x10000)
  462. {
  463. SQUnsignedInteger code = (ch - 0x10000);
  464. APPEND_CHAR((SQChar)(0xD800 | (code >> 10)));
  465. APPEND_CHAR((SQChar)(0xDC00 | (code & 0x3FF)));
  466. return 2;
  467. }
  468. else {
  469. APPEND_CHAR((SQChar)ch);
  470. return 1;
  471. }
  472. }
  473. #endif
  474. #else
  475. SQInteger SQLexer::AddUTF8(SQUnsignedInteger ch)
  476. {
  477. if (ch < 0x80) {
  478. APPEND_CHAR((char)ch);
  479. return 1;
  480. }
  481. if (ch < 0x800) {
  482. APPEND_CHAR((SQChar)((ch >> 6) | 0xC0));
  483. APPEND_CHAR((SQChar)((ch & 0x3F) | 0x80));
  484. return 2;
  485. }
  486. if (ch < 0x10000) {
  487. APPEND_CHAR((SQChar)((ch >> 12) | 0xE0));
  488. APPEND_CHAR((SQChar)(((ch >> 6) & 0x3F) | 0x80));
  489. APPEND_CHAR((SQChar)((ch & 0x3F) | 0x80));
  490. return 3;
  491. }
  492. if (ch < 0x110000) {
  493. APPEND_CHAR((SQChar)((ch >> 18) | 0xF0));
  494. APPEND_CHAR((SQChar)(((ch >> 12) & 0x3F) | 0x80));
  495. APPEND_CHAR((SQChar)(((ch >> 6) & 0x3F) | 0x80));
  496. APPEND_CHAR((SQChar)((ch & 0x3F) | 0x80));
  497. return 4;
  498. }
  499. return 0;
  500. }
  501. #endif
  502. SQInteger SQLexer::ProcessStringHexEscape(SQChar *dest, SQInteger maxdigits)
  503. {
  504. NEXT();
  505. if (!isxdigit(CUR_CHAR)) return Error(_SC("hexadecimal number expected"));
  506. SQInteger n = 0;
  507. while (isxdigit(CUR_CHAR) && n < maxdigits) {
  508. dest[n] = CUR_CHAR;
  509. n++;
  510. NEXT();
  511. }
  512. dest[n] = 0;
  513. return n;
  514. }
  515. SQInteger SQLexer::ReadString(SQInteger ndelim,bool verbatim)
  516. {
  517. INIT_TEMP_STRING();
  518. SQInteger start_equals = 0;
  519. SQChar cpp_delimin[32], cdelim1, cdelim2, saved_ndelim = ndelim;
  520. if(ndelim == _SC('{')){
  521. cdelim1 = _SC('{');
  522. cdelim2 = _SC('}');
  523. }
  524. else if(ndelim == _SC('(')){
  525. cdelim1 = _SC('(');
  526. cdelim2 = _SC(')');
  527. }
  528. else if(ndelim == _SC('[')){
  529. cdelim1 = _SC('[');
  530. cdelim2 = _SC(']');
  531. }
  532. else if(ndelim == _SC('R')){
  533. cdelim1 = _SC('(');
  534. cdelim2 = _SC(')');
  535. ndelim = cdelim2;
  536. if(CUR_CHAR != _SC('"'))
  537. {
  538. return Error(_SC("expect '\"' on literal delimiter"));
  539. }
  540. NEXT();
  541. size_t i=0;
  542. for(;(i < sizeof(cpp_delimin)-1) && (CUR_CHAR != _SC('(')); ++i)
  543. {
  544. cpp_delimin[i] = CUR_CHAR;
  545. NEXT();
  546. }
  547. cpp_delimin[i] = _SC('\0');
  548. }
  549. else
  550. {
  551. cdelim1 = cdelim2 = _SC('\0');
  552. }
  553. if((cdelim1 == saved_ndelim) && (CUR_CHAR == _SC('='))){
  554. //lua like literal
  555. while(!IS_EOB() && CUR_CHAR == _SC('=')) {
  556. ++start_equals;
  557. NEXT();
  558. }
  559. if(CUR_CHAR != cdelim1){
  560. //it's not a lua literal delimiter
  561. return Error(_SC("expect '%c' on literal delimiter"), cdelim1);
  562. }
  563. ndelim = cdelim2;
  564. }
  565. NEXT();
  566. if(IS_EOB()) return -1;
  567. if(start_equals) {
  568. int cr_nl = CUR_CHAR == _SC('\r');
  569. if(cr_nl) NEXT();
  570. cr_nl = CUR_CHAR == _SC('\n');
  571. if(cr_nl) NEXT();
  572. if(cr_nl) {//if a new line follows the start of delimiter drop it
  573. ++data->currentline;
  574. if(IS_EOB())
  575. {
  576. return Error(_SC("unfinished string"));
  577. }
  578. }
  579. }
  580. for(;;) {
  581. try_again:
  582. while(CUR_CHAR != ndelim) {
  583. SQInteger x = CUR_CHAR;
  584. switch(x) {
  585. case SQUIRREL_EOB:
  586. return Error(_SC("unfinished string"));
  587. case _SC('\n'):
  588. if(!verbatim) return Error(_SC("newline in a constant"));
  589. APPEND_CHAR(CUR_CHAR); NEXT();
  590. data->currentline++;
  591. break;
  592. case _SC('\\'):
  593. if(verbatim) {
  594. APPEND_CHAR('\\'); NEXT();
  595. }
  596. else {
  597. NEXT();
  598. switch(CUR_CHAR) {
  599. case _SC('x'): {
  600. const SQInteger maxdigits = sizeof(SQChar) * 2;
  601. SQChar temp[maxdigits + 1];
  602. if(ProcessStringHexEscape(temp, maxdigits) < 0) return -1;
  603. SQChar *stemp;
  604. APPEND_CHAR((SQChar)scstrtoul(temp, &stemp, 16));
  605. }
  606. break;
  607. case _SC('U'):
  608. case _SC('u'): {
  609. const SQInteger maxdigits = x == 'u' ? 4 : 8;
  610. SQChar temp[8 + 1];
  611. if(ProcessStringHexEscape(temp, maxdigits) < 0) return -1;
  612. SQChar *stemp;
  613. #ifdef SQUNICODE
  614. #if WCHAR_SIZE == 2
  615. AddUTF16(scstrtoul(temp, &stemp, 16));
  616. #else
  617. ADD_CHAR((SQChar)scstrtoul(temp, &stemp, 16));
  618. #endif
  619. #else
  620. AddUTF8(scstrtoul(temp, &stemp, 16));
  621. #endif
  622. }
  623. break;
  624. case _SC('t'): APPEND_CHAR(_SC('\t')); NEXT(); break;
  625. case _SC('a'): APPEND_CHAR(_SC('\a')); NEXT(); break;
  626. case _SC('b'): APPEND_CHAR(_SC('\b')); NEXT(); break;
  627. case _SC('n'): APPEND_CHAR(_SC('\n')); NEXT(); break;
  628. case _SC('r'): APPEND_CHAR(_SC('\r')); NEXT(); break;
  629. case _SC('v'): APPEND_CHAR(_SC('\v')); NEXT(); break;
  630. case _SC('f'): APPEND_CHAR(_SC('\f')); NEXT(); break;
  631. case _SC('0'): APPEND_CHAR(_SC('\0')); NEXT(); break;
  632. case _SC('\\'): APPEND_CHAR(_SC('\\')); NEXT(); break;
  633. case _SC('"'): APPEND_CHAR(_SC('"')); NEXT(); break;
  634. case _SC('\''): APPEND_CHAR(_SC('\'')); NEXT(); break;
  635. default:
  636. return Error(_SC("unrecognised escaper char"));
  637. break;
  638. }
  639. }
  640. break;
  641. default:
  642. APPEND_CHAR(CUR_CHAR);
  643. NEXT();
  644. }
  645. }
  646. NEXT();
  647. if(start_equals){
  648. bool lastBraceAdded = false;
  649. if(CUR_CHAR == _SC('=')){
  650. SQInteger end_equals = start_equals;
  651. NEXT();
  652. if(CUR_CHAR == _SC('=') || CUR_CHAR == cdelim2){
  653. --end_equals;
  654. while(!IS_EOB() && CUR_CHAR == _SC('=')) {
  655. --end_equals;
  656. NEXT();
  657. }
  658. if(end_equals) return Error(_SC("expect same number of '=' on literal delimiter"));
  659. if(CUR_CHAR != cdelim2) return Error(_SC("expect '%c' to close literal delimiter"), cdelim2);
  660. NEXT();
  661. break;
  662. }
  663. APPEND_CHAR(cdelim2); //the first NEXT() after break the while loop
  664. APPEND_CHAR(_SC('='));
  665. lastBraceAdded = true;
  666. }
  667. if(!lastBraceAdded) APPEND_CHAR(cdelim2); //the first NEXT() after break the while loop
  668. APPEND_CHAR(CUR_CHAR);
  669. NEXT();
  670. }
  671. else if(saved_ndelim == _SC('R')) {
  672. if(CUR_CHAR == ndelim)
  673. {
  674. APPEND_CHAR(ndelim);
  675. goto try_again;
  676. }
  677. size_t i = 0;
  678. for(;(i < sizeof(cpp_delimin)-1) && (CUR_CHAR != _SC('"')) && cpp_delimin[i]; ++i)
  679. {
  680. if(CUR_CHAR != cpp_delimin[i])
  681. {
  682. //false positive append all chars till here and continue
  683. APPEND_CHAR(ndelim);
  684. for(int j=0; j < i; ++j) APPEND_CHAR(cpp_delimin[j]); //recover already eaten chars from buffer
  685. APPEND_CHAR(CUR_CHAR); //append the last one that mismatch
  686. if(CUR_CHAR == _SC('\n')) data->currentline++;
  687. NEXT();
  688. goto try_again;
  689. //return Error(_SC("expect \"%s\" to close literal delimiter"), cpp_delimin);
  690. }
  691. NEXT();
  692. }
  693. if(CUR_CHAR != _SC('"')) return Error(_SC("expect '\"' to close literal delimiter"));
  694. NEXT(); //eat last '"'
  695. break;
  696. }
  697. else if(verbatim && CUR_CHAR == _SC('"')) { //double quotation
  698. APPEND_CHAR(CUR_CHAR);
  699. NEXT();
  700. }
  701. else {
  702. break;
  703. }
  704. }
  705. TERMINATE_BUFFER();
  706. SQInteger len = data->longstr.size()-1;
  707. if(ndelim == _SC('\'')) {
  708. if(len == 0) return Error(_SC("empty constant"));
  709. if(len > 1) return Error(_SC("constant too long"));
  710. data->nvalue = data->longstr[0];
  711. return TK_INTEGER;
  712. }
  713. data->svalue = &data->longstr[0];
  714. return TK_STRING_LITERAL;
  715. }
  716. void LexHexadecimal(const SQChar *s,SQUnsignedInteger *res)
  717. {
  718. *res = 0;
  719. while(*s != 0)
  720. {
  721. if(scisdigit(*s)) *res = (*res)*16+((*s++)-'0');
  722. else if(scisxdigit(*s)) *res = (*res)*16+(toupper(*s++)-'A'+10);
  723. else { assert(0); }
  724. }
  725. }
  726. void LexInteger(const SQChar *s,SQUnsignedInteger *res)
  727. {
  728. *res = 0;
  729. while(*s != 0)
  730. {
  731. *res = (*res)*10+((*s++)-'0');
  732. }
  733. }
  734. SQInteger scisodigit(SQInteger c) { return c >= _SC('0') && c <= _SC('7'); }
  735. void LexOctal(const SQChar *s,SQUnsignedInteger *res)
  736. {
  737. *res = 0;
  738. while(*s != 0)
  739. {
  740. if(scisodigit(*s)) *res = (*res)*8+((*s++)-'0');
  741. else { assert(0); }
  742. }
  743. }
  744. SQInteger isexponent(SQInteger c) { return c == 'e' || c=='E'; }
  745. #define MAX_HEX_DIGITS (sizeof(SQInteger)*2)
  746. SQInteger SQLexer::ReadNumber()
  747. {
  748. #define TINT 1
  749. #define TFLOAT 2
  750. #define THEX 3
  751. #define TSCIENTIFIC 4
  752. #define TOCTAL 5
  753. SQInteger type = TINT, firstchar = CUR_CHAR;
  754. SQUnsignedInteger itmp=0;
  755. SQChar *sTemp;
  756. INIT_TEMP_STRING();
  757. NEXT();
  758. if(firstchar == _SC('0') && (toupper(CUR_CHAR) == _SC('X') || scisodigit(CUR_CHAR)) ) {
  759. if(scisodigit(CUR_CHAR)) {
  760. type = TOCTAL;
  761. while(scisodigit(CUR_CHAR)) {
  762. APPEND_CHAR(CUR_CHAR);
  763. NEXT();
  764. }
  765. if(scisdigit(CUR_CHAR)) return Error(_SC("invalid octal number"));
  766. }
  767. else {
  768. NEXT();
  769. type = THEX;
  770. while(isxdigit(CUR_CHAR)) {
  771. APPEND_CHAR(CUR_CHAR);
  772. NEXT();
  773. }
  774. if(data->longstr.size() > MAX_HEX_DIGITS) return Error(_SC("too many digits for an Hex number"));
  775. }
  776. }
  777. else {
  778. APPEND_CHAR((int)firstchar);
  779. while (CUR_CHAR == _SC('.') || scisdigit(CUR_CHAR) || isexponent(CUR_CHAR)) {
  780. if(CUR_CHAR == _SC('.') || isexponent(CUR_CHAR)) type = TFLOAT;
  781. if(isexponent(CUR_CHAR)) {
  782. if(type != TFLOAT) return Error(_SC("invalid numeric format"));
  783. type = TSCIENTIFIC;
  784. APPEND_CHAR(CUR_CHAR);
  785. NEXT();
  786. if(CUR_CHAR == '+' || CUR_CHAR == '-'){
  787. APPEND_CHAR(CUR_CHAR);
  788. NEXT();
  789. }
  790. if(!scisdigit(CUR_CHAR)) return Error(_SC("exponent expected"));
  791. }
  792. APPEND_CHAR(CUR_CHAR);
  793. NEXT();
  794. }
  795. }
  796. TERMINATE_BUFFER();
  797. switch(type) {
  798. case TSCIENTIFIC:
  799. case TFLOAT:
  800. data->fvalue = (SQFloat)scstrtod(&data->longstr[0],&sTemp);
  801. return TK_FLOAT;
  802. case TINT:
  803. LexInteger(&data->longstr[0],&itmp);
  804. break;
  805. case THEX:
  806. LexHexadecimal(&data->longstr[0],&itmp);
  807. break;
  808. case TOCTAL:
  809. LexOctal(&data->longstr[0],&itmp);
  810. break;
  811. }
  812. switch(type) {
  813. case TINT:
  814. case THEX:
  815. case TOCTAL:
  816. //to allow 64 bits integers comment bellow
  817. //if(itmp > INT_MAX) return Error(_SC("integer overflow %ulld %d"));
  818. data->nvalue = (SQInteger) itmp;
  819. return TK_INTEGER;
  820. }
  821. return 0;
  822. }
  823. SQInteger SQLexer::ReadID()
  824. {
  825. SQInteger res;
  826. INIT_TEMP_STRING();
  827. do {
  828. APPEND_CHAR(CUR_CHAR);
  829. NEXT();
  830. } while(scisalnum(CUR_CHAR) || CUR_CHAR == _SC('_'));
  831. TERMINATE_BUFFER();
  832. if((CUR_CHAR == _SC('"')) && (data->longstr[0] == _SC('R')) && (data->longstr.size() == 2))
  833. {
  834. //C++ multiline string
  835. return ReadString(_SC('R'),true);
  836. }
  837. res = GetIDType(&data->longstr[0],data->longstr.size() - 1);
  838. if(res == TK_IDENTIFIER || res == TK_CONSTRUCTOR || res == TK_DESTRUCTOR) {
  839. data->svalue = &data->longstr[0];
  840. }
  841. return res;
  842. }