lexer.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <stdarg.h>
   5
   6 #include "gmqcc.h"
   7 #include "lexer.h"
   8
   9 char* *lex_filenames;
  10
  11 void lexerror(lex_file *lex, const char *fmt, ...)
  12 {
  13         va_list ap;
  14
  15         va_start(ap, fmt);
  16         if (lex)
  17         con_vprintmsg(LVL_ERROR, lex->name, lex->sline, "parse error", fmt, ap);
  18     else
  19         con_vprintmsg(LVL_ERROR, "", 0, "parse error", fmt, ap);
  20         va_end(ap);
  21 }
  22
  23 bool lexwarn(lex_file *lex, int warntype, const char *fmt, ...)
  24 {
  25         va_list ap;
  26         int lvl = LVL_WARNING;
  27
  28     if (!OPTS_WARN(warntype))
  29         return false;
  30
  31     if (opts_werror)
  32             lvl = LVL_ERROR;
  33
  34         va_start(ap, fmt);
  35     con_vprintmsg(lvl, lex->name, lex->sline, "warning", fmt, ap);
  36         va_end(ap);
  37
  38         return opts_werror;
  39 }
  40
  41
  42 #if 0
  43 token* token_new()
  44 {
  45     token *tok = (token*)mem_a(sizeof(token));
  46     if (!tok)
  47         return NULL;
  48     memset(tok, 0, sizeof(*tok));
  49     return tok;
  50 }
  51
  52 void token_delete(token *self)
  53 {
  54     if (self->next && self->next->prev == self)
  55         self->next->prev = self->prev;
  56     if (self->prev && self->prev->next == self)
  57         self->prev->next = self->next;
  58     MEM_VECTOR_CLEAR(self, value);
  59     mem_d(self);
  60 }
  61
  62 token* token_copy(const token *cp)
  63 {
  64     token* self = token_new();
  65     if (!self)
  66         return NULL;
  67     /* copy the value */
  68     self->value_alloc = cp->value_count + 1;
  69     self->value_count = cp->value_count;
  70     self->value = (char*)mem_a(self->value_alloc);
  71     if (!self->value) {
  72         mem_d(self);
  73         return NULL;
  74     }
  75     memcpy(self->value, cp->value, cp->value_count);
  76     self->value[self->value_alloc-1] = 0;
  77
  78     /* rest */
  79     self->ctx = cp->ctx;
  80     self->ttype = cp->ttype;
  81     memcpy(&self->constval, &cp->constval, sizeof(self->constval));
  82     return self;
  83 }
  84
  85 void token_delete_all(token *t)
  86 {
  87     token *n;
  88
  89     do {
  90         n = t->next;
  91         token_delete(t);
  92         t = n;
  93     } while(t);
  94 }
  95
  96 token* token_copy_all(const token *cp)
  97 {
  98     token *cur;
  99     token *out;
 100
 101     out = cur = token_copy(cp);
 102     if (!out)
 103         return NULL;
 104
 105     while (cp->next) {
 106         cp = cp->next;
 107         cur->next = token_copy(cp);
 108         if (!cur->next) {
 109             token_delete_all(out);
 110             return NULL;
 111         }
 112         cur->next->prev = cur;
 113         cur = cur->next;
 114     }
 115
 116     return out;
 117 }
 118 #else
 119 static void lex_token_new(lex_file *lex)
 120 {
 121 #if 0
 122     if (lex->tok)
 123         token_delete(lex->tok);
 124     lex->tok = token_new();
 125 #else
 126     if (lex->tok.value)
 127         vec_shrinkto(lex->tok.value, 0);
 128     lex->tok.constval.t  = 0;
 129     lex->tok.ctx.line = lex->sline;
 130     lex->tok.ctx.file = lex->name;
 131 #endif
 132 }
 133 #endif
 134
 135 lex_file* lex_open(const char *file)
 136 {
 137     lex_file *lex;
 138     FILE *in = util_fopen(file, "rb");
 139
 140     if (!in) {
 141         lexerror(NULL, "open failed: '%s'\n", file);
 142         return NULL;
 143     }
 144
 145     lex = (lex_file*)mem_a(sizeof(*lex));
 146     if (!lex) {
 147         fclose(in);
 148         lexerror(NULL, "out of memory\n");
 149         return NULL;
 150     }
 151
 152     memset(lex, 0, sizeof(*lex));
 153
 154     lex->file = in;
 155     lex->name = util_strdup(file);
 156     lex->line = 1; /* we start counting at 1 */
 157
 158     lex->peekpos = 0;
 159     lex->eof = false;
 160
 161     vec_push(lex_filenames, lex->name);
 162     return lex;
 163 }
 164
 165 lex_file* lex_open_string(const char *str, size_t len, const char *name)
 166 {
 167     lex_file *lex;
 168
 169     lex = (lex_file*)mem_a(sizeof(*lex));
 170     if (!lex) {
 171         lexerror(NULL, "out of memory\n");
 172         return NULL;
 173     }
 174
 175     memset(lex, 0, sizeof(*lex));
 176
 177     lex->file = NULL;
 178     lex->open_string        = str;
 179     lex->open_string_length = len;
 180     lex->open_string_pos    = 0;
 181
 182     lex->name = util_strdup(name ? name : "<string-source>");
 183     lex->line = 1; /* we start counting at 1 */
 184
 185     lex->peekpos = 0;
 186     lex->eof = false;
 187
 188     vec_push(lex_filenames, lex->name);
 189
 190     return lex;
 191 }
 192
 193 void lex_cleanup(void)
 194 {
 195     size_t i;
 196     for (i = 0; i < vec_size(lex_filenames); ++i)
 197         mem_d(lex_filenames[i]);
 198     vec_free(lex_filenames);
 199 }
 200
 201 void lex_close(lex_file *lex)
 202 {
 203     size_t i;
 204     for (i = 0; i < vec_size(lex->frames); ++i)
 205         mem_d(lex->frames[i].name);
 206     vec_free(lex->frames);
 207
 208     if (lex->modelname)
 209         vec_free(lex->modelname);
 210
 211     if (lex->file)
 212         fclose(lex->file);
 213 #if 0
 214     if (lex->tok)
 215         token_delete(lex->tok);
 216 #else
 217     vec_free(lex->tok.value);
 218 #endif
 219     /* mem_d(lex->name); collected in lex_filenames */
 220     mem_d(lex);
 221 }
 222
 223 static int lex_fgetc(lex_file *lex)
 224 {
 225     if (lex->file)
 226         return fgetc(lex->file);
 227     if (lex->open_string) {
 228         if (lex->open_string_pos >= lex->open_string_length)
 229             return EOF;
 230         return lex->open_string[lex->open_string_pos++];
 231     }
 232     return EOF;
 233 }
 234
 235 /* Get or put-back data
 236  * The following to functions do NOT understand what kind of data they
 237  * are working on.
 238  * The are merely wrapping get/put in order to count line numbers.
 239  */
 240 static void lex_ungetch(lex_file *lex, int ch);
 241 static int lex_try_trigraph(lex_file *lex, int old)
 242 {
 243     int c2, c3;
 244     c2 = lex_fgetc(lex);
 245     if (c2 != '?') {
 246         lex_ungetch(lex, c2);
 247         return old;
 248     }
 249
 250     c3 = lex_fgetc(lex);
 251     switch (c3) {
 252         case '=': return '#';
 253         case '/': return '\\';
 254         case '\'': return '^';
 255         case '(': return '[';
 256         case ')': return ']';
 257         case '!': return '|';
 258         case '<': return '{';
 259         case '>': return '}';
 260         case '-': return '~';
 261         default:
 262             lex_ungetch(lex, c3);
 263             lex_ungetch(lex, c2);
 264             return old;
 265     }
 266 }
 267
 268 static int lex_try_digraph(lex_file *lex, int ch)
 269 {
 270     int c2;
 271     c2 = lex_fgetc(lex);
 272     if      (ch == '<' && c2 == ':')
 273         return '[';
 274     else if (ch == ':' && c2 == '>')
 275         return ']';
 276     else if (ch == '<' && c2 == '%')
 277         return '{';
 278     else if (ch == '%' && c2 == '>')
 279         return '}';
 280     else if (ch == '%' && c2 == ':')
 281         return '#';
 282     lex_ungetch(lex, c2);
 283     return ch;
 284 }
 285
 286 static int lex_getch(lex_file *lex)
 287 {
 288     int ch;
 289
 290     if (lex->peekpos) {
 291         lex->peekpos--;
 292         if (!lex->push_line && lex->peek[lex->peekpos] == '\n')
 293             lex->line++;
 294         return lex->peek[lex->peekpos];
 295     }
 296
 297     ch = lex_fgetc(lex);
 298     if (!lex->push_line && ch == '\n')
 299         lex->line++;
 300     else if (ch == '?')
 301         return lex_try_trigraph(lex, ch);
 302     else if (!lex->flags.nodigraphs && (ch == '<' || ch == ':' || ch == '%'))
 303         return lex_try_digraph(lex, ch);
 304     return ch;
 305 }
 306
 307 static void lex_ungetch(lex_file *lex, int ch)
 308 {
 309     lex->peek[lex->peekpos++] = ch;
 310     if (!lex->push_line && ch == '\n')
 311         lex->line--;
 312 }
 313
 314 /* classify characters
 315  * some additions to the is*() functions of ctype.h
 316  */
 317
 318 /* Idents are alphanumberic, but they start with alpha or _ */
 319 static bool isident_start(int ch)
 320 {
 321     return isalpha(ch) || ch == '_';
 322 }
 323
 324 static bool isident(int ch)
 325 {
 326     return isident_start(ch) || isdigit(ch);
 327 }
 328
 329 /* isxdigit_only is used when we already know it's not a digit
 330  * and want to see if it's a hex digit anyway.
 331  */
 332 static bool isxdigit_only(int ch)
 333 {
 334     return (ch >= 'a' && ch <= 'f') || (ch >= 'A' && ch <= 'F');
 335 }
 336
 337 /* Append a character to the token buffer */
 338 static void lex_tokench(lex_file *lex, int ch)
 339 {
 340     vec_push(lex->tok.value, ch);
 341 }
 342
 343 /* Append a trailing null-byte */
 344 static void lex_endtoken(lex_file *lex)
 345 {
 346     vec_push(lex->tok.value, 0);
 347     vec_shrinkby(lex->tok.value, 1);
 348 }
 349
 350 static bool lex_try_pragma(lex_file *lex)
 351 {
 352     int ch;
 353     char *pragma  = NULL;
 354     char *command = NULL;
 355     char *param   = NULL;
 356     size_t line;
 357
 358     if (lex->flags.preprocessing)
 359         return false;
 360
 361     line = lex->line;
 362
 363     ch = lex_getch(lex);
 364     if (ch != '#') {
 365         lex_ungetch(lex, ch);
 366         return false;
 367     }
 368
 369     for (ch = lex_getch(lex); vec_size(pragma) < 8 && ch >= 'a' && ch <= 'z'; ch = lex_getch(lex))
 370         vec_push(pragma, ch);
 371     vec_push(pragma, 0);
 372
 373     if (ch != ' ' || strcmp(pragma, "pragma")) {
 374         lex_ungetch(lex, ch);
 375         goto unroll;
 376     }
 377
 378     for (ch = lex_getch(lex); vec_size(command) < 32 && ch >= 'a' && ch <= 'z'; ch = lex_getch(lex))
 379         vec_push(command, ch);
 380     vec_push(command, 0);
 381
 382     if (ch != '(') {
 383         lex_ungetch(lex, ch);
 384         goto unroll;
 385     }
 386
 387     for (ch = lex_getch(lex); vec_size(param) < 32 && ch != ')' && ch != '\n'; ch = lex_getch(lex))
 388         vec_push(param, ch);
 389     vec_push(param, 0);
 390
 391     if (ch != ')') {
 392         lex_ungetch(lex, ch);
 393         goto unroll;
 394     }
 395
 396     if (!strcmp(command, "push")) {
 397         if (!strcmp(param, "line")) {
 398             lex->push_line++;
 399             --line;
 400         }
 401         else
 402             goto unroll;
 403     }
 404     else if (!strcmp(command, "pop")) {
 405         if (!strcmp(param, "line")) {
 406             if (lex->push_line)
 407                 lex->push_line--;
 408             --line;
 409         }
 410         else
 411             goto unroll;
 412     }
 413     else if (!strcmp(command, "file")) {
 414         lex->name = util_strdup(param);
 415         vec_push(lex_filenames, lex->name);
 416     }
 417     else if (!strcmp(command, "line")) {
 418         line = strtol(param, NULL, 0)-1;
 419     }
 420     else
 421         goto unroll;
 422
 423     lex->line = line;
 424     while (ch != '\n' && ch != EOF)
 425         ch = lex_getch(lex);
 426     return true;
 427
 428 unroll:
 429     if (command) {
 430         vec_pop(command);
 431         while (vec_size(command)) {
 432             lex_ungetch(lex, vec_last(command));
 433             vec_pop(command);
 434         }
 435         vec_free(command);
 436     }
 437     if (command) {
 438         vec_pop(command);
 439         while (vec_size(command)) {
 440             lex_ungetch(lex, vec_last(command));
 441             vec_pop(command);
 442         }
 443         vec_free(command);
 444     }
 445     if (pragma) {
 446         vec_pop(pragma);
 447         while (vec_size(pragma)) {
 448             lex_ungetch(lex, vec_last(pragma));
 449             vec_pop(pragma);
 450         }
 451         vec_free(pragma);
 452     }
 453     lex_ungetch(lex, '#');
 454
 455     lex->line = line;
 456     return false;
 457 }
 458
 459 /* Skip whitespace and comments and return the first
 460  * non-white character.
 461  * As this makes use of the above getch() ungetch() functions,
 462  * we don't need to care at all about line numbering anymore.
 463  *
 464  * In theory, this function should only be used at the beginning
 465  * of lexing, or when we *know* the next character is part of the token.
 466  * Otherwise, if the parser throws an error, the linenumber may not be
 467  * the line of the error, but the line of the next token AFTER the error.
 468  *
 469  * This is currently only problematic when using c-like string-continuation,
 470  * since comments and whitespaces are allowed between 2 such strings.
 471  * Example:
 472 printf(   "line one\n"
 473 // A comment
 474           "A continuation of the previous string"
 475 // This line is skipped
 476       , foo);
 477
 478  * In this case, if the parse decides it didn't actually want a string,
 479  * and uses lex->line to print an error, it will show the ', foo);' line's
 480  * linenumber.
 481  *
 482  * On the other hand, the parser is supposed to remember the line of the next
 483  * token's beginning. In this case we would want skipwhite() to be called
 484  * AFTER reading a token, so that the parser, before reading the NEXT token,
 485  * doesn't store teh *comment's* linenumber, but the actual token's linenumber.
 486  *
 487  * THIS SOLUTION
 488  *    here is to store the line of the first character after skipping
 489  *    the initial whitespace in lex->sline, this happens in lex_do.
 490  */
 491 static int lex_skipwhite(lex_file *lex)
 492 {
 493     int ch = 0;
 494     bool haswhite = false;
 495
 496     do
 497     {
 498         ch = lex_getch(lex);
 499         while (ch != EOF && isspace(ch)) {
 500             if (ch == '\n') {
 501                 if (lex_try_pragma(lex))
 502                     continue;
 503             }
 504             if (lex->flags.preprocessing) {
 505                 if (ch == '\n') {
 506                     /* end-of-line */
 507                     /* see if there was whitespace first */
 508                     if (haswhite) { /* (vec_size(lex->tok.value)) { */
 509                         lex_ungetch(lex, ch);
 510                         lex_endtoken(lex);
 511                         return TOKEN_WHITE;
 512                     }
 513                     /* otherwise return EOL */
 514                     return TOKEN_EOL;
 515                 }
 516                 haswhite = true;
 517                 lex_tokench(lex, ch);
 518             }
 519             ch = lex_getch(lex);
 520         }
 521
 522         if (ch == '/') {
 523             ch = lex_getch(lex);
 524             if (ch == '/')
 525             {
 526                 /* one line comment */
 527                 ch = lex_getch(lex);
 528
 529                 if (lex->flags.preprocessing) {
 530                     haswhite = true;
 531                     /*
 532                     lex_tokench(lex, '/');
 533                     lex_tokench(lex, '/');
 534                     */
 535                     lex_tokench(lex, ' ');
 536                     lex_tokench(lex, ' ');
 537                 }
 538
 539                 while (ch != EOF && ch != '\n') {
 540                     if (lex->flags.preprocessing)
 541                         lex_tokench(lex, ' '); /* ch); */
 542                     ch = lex_getch(lex);
 543                 }
 544                 if (lex->flags.preprocessing) {
 545                     lex_ungetch(lex, '\n');
 546                     lex_endtoken(lex);
 547                     return TOKEN_WHITE;
 548                 }
 549                 continue;
 550             }
 551             if (ch == '*')
 552             {
 553                 /* multiline comment */
 554                 if (lex->flags.preprocessing) {
 555                     haswhite = true;
 556                     /*
 557                     lex_tokench(lex, '/');
 558                     lex_tokench(lex, '*');
 559                     */
 560                     lex_tokench(lex, ' ');
 561                     lex_tokench(lex, ' ');
 562                 }
 563
 564                 while (ch != EOF)
 565                 {
 566                     ch = lex_getch(lex);
 567                     if (ch == '*') {
 568                         ch = lex_getch(lex);
 569                         if (ch == '/') {
 570                             if (lex->flags.preprocessing) {
 571                                 /*
 572                                 lex_tokench(lex, '*');
 573                                 lex_tokench(lex, '/');
 574                                 */
 575                                 lex_tokench(lex, ' ');
 576                                 lex_tokench(lex, ' ');
 577                             }
 578                             break;
 579                         }
 580                     }
 581                     if (lex->flags.preprocessing) {
 582                         lex_tokench(lex, ' '); /* ch); */
 583                     }
 584                 }
 585                 ch = ' '; /* cause TRUE in the isspace check */
 586                 continue;
 587             }
 588             /* Otherwise roll back to the slash and break out of the loop */
 589             lex_ungetch(lex, ch);
 590             ch = '/';
 591             break;
 592         }
 593     } while (ch != EOF && isspace(ch));
 594
 595     if (haswhite) {
 596         lex_endtoken(lex);
 597         lex_ungetch(lex, ch);
 598         return TOKEN_WHITE;
 599     }
 600     return ch;
 601 }
 602
 603 /* Get a token */
 604 static bool GMQCC_WARN lex_finish_ident(lex_file *lex)
 605 {
 606     int ch;
 607
 608     ch = lex_getch(lex);
 609     while (ch != EOF && isident(ch))
 610     {
 611         lex_tokench(lex, ch);
 612         ch = lex_getch(lex);
 613     }
 614
 615     /* last ch was not an ident ch: */
 616     lex_ungetch(lex, ch);
 617
 618     return true;
 619 }
 620
 621 /* read one ident for the frame list */
 622 static int lex_parse_frame(lex_file *lex)
 623 {
 624     int ch;
 625
 626     lex_token_new(lex);
 627
 628     ch = lex_getch(lex);
 629     while (ch != EOF && ch != '\n' && isspace(ch))
 630         ch = lex_getch(lex);
 631
 632     if (ch == '\n')
 633         return 1;
 634
 635     if (!isident_start(ch)) {
 636         lexerror(lex, "invalid framename, must start with one of a-z or _, got %c", ch);
 637         return -1;
 638     }
 639
 640     lex_tokench(lex, ch);
 641     if (!lex_finish_ident(lex))
 642         return -1;
 643     lex_endtoken(lex);
 644     return 0;
 645 }
 646
 647 /* read a list of $frames */
 648 static bool lex_finish_frames(lex_file *lex)
 649 {
 650     do {
 651         size_t i;
 652         int    rc;
 653         frame_macro m;
 654
 655         rc = lex_parse_frame(lex);
 656         if (rc > 0) /* end of line */
 657             return true;
 658         if (rc < 0) /* error */
 659             return false;
 660
 661         for (i = 0; i < vec_size(lex->frames); ++i) {
 662             if (!strcmp(lex->tok.value, lex->frames[i].name)) {
 663                 lex->frames[i].value = lex->framevalue++;
 664                 if (lexwarn(lex, WARN_FRAME_MACROS, "duplicate frame macro defined: `%s`", lex->tok.value))
 665                     return false;
 666                 break;
 667             }
 668         }
 669         if (i < vec_size(lex->frames))
 670             continue;
 671
 672         m.value = lex->framevalue++;
 673         m.name = util_strdup(lex->tok.value);
 674         vec_shrinkto(lex->tok.value, 0);
 675         vec_push(lex->frames, m);
 676     } while (true);
 677 }
 678
 679 static int GMQCC_WARN lex_finish_string(lex_file *lex, int quote)
 680 {
 681     int ch = 0;
 682
 683     while (ch != EOF)
 684     {
 685         ch = lex_getch(lex);
 686         if (ch == quote)
 687             return TOKEN_STRINGCONST;
 688
 689         if (lex->flags.preprocessing && ch == '\\') {
 690             lex_tokench(lex, ch);
 691             ch = lex_getch(lex);
 692             if (ch == EOF) {
 693                 lexerror(lex, "unexpected end of file");
 694                 lex_ungetch(lex, EOF); /* next token to be TOKEN_EOF */
 695                 return (lex->tok.ttype = TOKEN_ERROR);
 696             }
 697             lex_tokench(lex, ch);
 698         }
 699         else if (ch == '\\') {
 700             ch = lex_getch(lex);
 701             if (ch == EOF) {
 702                 lexerror(lex, "unexpected end of file");
 703                 lex_ungetch(lex, EOF); /* next token to be TOKEN_EOF */
 704                 return (lex->tok.ttype = TOKEN_ERROR);
 705             }
 706
 707             switch (ch) {
 708             case '\\': break;
 709             case '\'': break;
 710             case '"':  break;
 711             case 'a':  ch = '\a'; break;
 712             case 'b':  ch = '\b'; break;
 713             case 'r':  ch = '\r'; break;
 714             case 'n':  ch = '\n'; break;
 715             case 't':  ch = '\t'; break;
 716             case 'f':  ch = '\f'; break;
 717             case 'v':  ch = '\v'; break;
 718             default:
 719                 lexwarn(lex, WARN_UNKNOWN_CONTROL_SEQUENCE, "unrecognized control sequence: \\%c", ch);
 720                 /* so we just add the character plus backslash no matter what it actually is */
 721                 lex_tokench(lex, '\\');
 722             }
 723             /* add the character finally */
 724             lex_tokench(lex, ch);
 725         }
 726         else
 727             lex_tokench(lex, ch);
 728     }
 729     lexerror(lex, "unexpected end of file within string constant");
 730     lex_ungetch(lex, EOF); /* next token to be TOKEN_EOF */
 731     return (lex->tok.ttype = TOKEN_ERROR);
 732 }
 733
 734 static int GMQCC_WARN lex_finish_digit(lex_file *lex, int lastch)
 735 {
 736     bool ishex = false;
 737
 738     int  ch = lastch;
 739
 740     /* parse a number... */
 741     lex->tok.ttype = TOKEN_INTCONST;
 742
 743     lex_tokench(lex, ch);
 744
 745     ch = lex_getch(lex);
 746     if (ch != '.' && !isdigit(ch))
 747     {
 748         if (lastch != '0' || ch != 'x')
 749         {
 750             /* end of the number or EOF */
 751             lex_ungetch(lex, ch);
 752             lex_endtoken(lex);
 753
 754             lex->tok.constval.i = lastch - '0';
 755             return lex->tok.ttype;
 756         }
 757
 758         ishex = true;
 759     }
 760
 761     /* EOF would have been caught above */
 762
 763     if (ch != '.')
 764     {
 765         lex_tokench(lex, ch);
 766         ch = lex_getch(lex);
 767         while (isdigit(ch) || (ishex && isxdigit_only(ch)))
 768         {
 769             lex_tokench(lex, ch);
 770             ch = lex_getch(lex);
 771         }
 772     }
 773     /* NOT else, '.' can come from above as well */
 774     if (ch == '.' && !ishex)
 775     {
 776         /* Allow floating comma in non-hex mode */
 777         lex->tok.ttype = TOKEN_FLOATCONST;
 778         lex_tokench(lex, ch);
 779
 780         /* continue digits-only */
 781         ch = lex_getch(lex);
 782         while (isdigit(ch))
 783         {
 784             lex_tokench(lex, ch);
 785             ch = lex_getch(lex);
 786         }
 787     }
 788     /* put back the last character */
 789     /* but do not put back the trailing 'f' or a float */
 790     if (lex->tok.ttype == TOKEN_FLOATCONST && ch == 'f')
 791         ch = lex_getch(lex);
 792
 793     /* generally we don't want words to follow numbers: */
 794     if (isident(ch)) {
 795         lexerror(lex, "unexpected trailing characters after number");
 796         return (lex->tok.ttype = TOKEN_ERROR);
 797     }
 798     lex_ungetch(lex, ch);
 799
 800     lex_endtoken(lex);
 801     if (lex->tok.ttype == TOKEN_FLOATCONST)
 802         lex->tok.constval.f = strtod(lex->tok.value, NULL);
 803     else
 804         lex->tok.constval.i = strtol(lex->tok.value, NULL, 0);
 805     return lex->tok.ttype;
 806 }
 807
 808 int lex_do(lex_file *lex)
 809 {
 810     int ch, nextch;
 811
 812     lex_token_new(lex);
 813 #if 0
 814     if (!lex->tok)
 815         return TOKEN_FATAL;
 816 #endif
 817
 818     while (true) {
 819         ch = lex_skipwhite(lex);
 820         if (!lex->flags.mergelines || ch != '\\')
 821             break;
 822         ch = lex_getch(lex);
 823         if (ch != '\n') {
 824             lex_ungetch(lex, ch);
 825             ch = '\\';
 826             break;
 827         }
 828         /* we reached a linemerge */
 829         lex_tokench(lex, '\n');
 830         continue;
 831     }
 832
 833     lex->sline = lex->line;
 834     lex->tok.ctx.line = lex->sline;
 835     lex->tok.ctx.file = lex->name;
 836
 837     if (lex->flags.preprocessing && (ch == TOKEN_WHITE || ch == TOKEN_EOL || ch == TOKEN_FATAL)) {
 838         return (lex->tok.ttype = ch);
 839     }
 840
 841     if (lex->eof)
 842         return (lex->tok.ttype = TOKEN_FATAL);
 843
 844     if (ch == EOF) {
 845         lex->eof = true;
 846         return (lex->tok.ttype = TOKEN_EOF);
 847     }
 848
 849     /* modelgen / spiritgen commands */
 850     if (ch == '$') {
 851         const char *v;
 852         size_t frame;
 853
 854         ch = lex_getch(lex);
 855         if (!isident_start(ch)) {
 856             lexerror(lex, "hanging '$' modelgen/spritegen command line");
 857             return lex_do(lex);
 858         }
 859         lex_tokench(lex, ch);
 860         if (!lex_finish_ident(lex))
 861             return (lex->tok.ttype = TOKEN_ERROR);
 862         lex_endtoken(lex);
 863         /* skip the known commands */
 864         v = lex->tok.value;
 865
 866         if (!strcmp(v, "frame") || !strcmp(v, "framesave"))
 867         {
 868             /* frame/framesave command works like an enum
 869              * similar to fteqcc we handle this in the lexer.
 870              * The reason for this is that it is sensitive to newlines,
 871              * which the parser is unaware of
 872              */
 873             if (!lex_finish_frames(lex))
 874                  return (lex->tok.ttype = TOKEN_ERROR);
 875             return lex_do(lex);
 876         }
 877
 878         if (!strcmp(v, "framevalue"))
 879         {
 880             ch = lex_getch(lex);
 881             while (ch != EOF && isspace(ch) && ch != '\n')
 882                 ch = lex_getch(lex);
 883
 884             if (!isdigit(ch)) {
 885                 lexerror(lex, "$framevalue requires an integer parameter");
 886                 return lex_do(lex);
 887             }
 888
 889             lex_token_new(lex);
 890             lex->tok.ttype = lex_finish_digit(lex, ch);
 891             lex_endtoken(lex);
 892             if (lex->tok.ttype != TOKEN_INTCONST) {
 893                 lexerror(lex, "$framevalue requires an integer parameter");
 894                 return lex_do(lex);
 895             }
 896             lex->framevalue = lex->tok.constval.i;
 897             return lex_do(lex);
 898         }
 899
 900         if (!strcmp(v, "framerestore"))
 901         {
 902             int rc;
 903
 904             lex_token_new(lex);
 905
 906             rc = lex_parse_frame(lex);
 907
 908             if (rc > 0) {
 909                 lexerror(lex, "$framerestore requires a framename parameter");
 910                 return lex_do(lex);
 911             }
 912             if (rc < 0)
 913                 return (lex->tok.ttype = TOKEN_FATAL);
 914
 915             v = lex->tok.value;
 916             for (frame = 0; frame < vec_size(lex->frames); ++frame) {
 917                 if (!strcmp(v, lex->frames[frame].name)) {
 918                     lex->framevalue = lex->frames[frame].value;
 919                     return lex_do(lex);
 920                 }
 921             }
 922             lexerror(lex, "unknown framename `%s`", v);
 923             return lex_do(lex);
 924         }
 925
 926         if (!strcmp(v, "modelname"))
 927         {
 928             int rc;
 929
 930             lex_token_new(lex);
 931
 932             rc = lex_parse_frame(lex);
 933
 934             if (rc > 0) {
 935                 lexerror(lex, "$modelname requires a parameter");
 936                 return lex_do(lex);
 937             }
 938             if (rc < 0)
 939                 return (lex->tok.ttype = TOKEN_FATAL);
 940
 941             v = lex->tok.value;
 942             if (lex->modelname) {
 943                 frame_macro m;
 944                 m.value = lex->framevalue;
 945                 m.name = lex->modelname;
 946                 lex->modelname = NULL;
 947                 vec_push(lex->frames, m);
 948             }
 949             lex->modelname = lex->tok.value;
 950             lex->tok.value = NULL;
 951             return lex_do(lex);
 952         }
 953
 954         if (!strcmp(v, "flush"))
 955         {
 956             size_t frame;
 957             for (frame = 0; frame < vec_size(lex->frames); ++frame)
 958                 mem_d(lex->frames[frame].name);
 959             vec_free(lex->frames);
 960             /* skip line (fteqcc does it too) */
 961             ch = lex_getch(lex);
 962             while (ch != EOF && ch != '\n')
 963                 ch = lex_getch(lex);
 964             return lex_do(lex);
 965         }
 966
 967         if (!strcmp(v, "cd") ||
 968             !strcmp(v, "origin") ||
 969             !strcmp(v, "base") ||
 970             !strcmp(v, "flags") ||
 971             !strcmp(v, "scale") ||
 972             !strcmp(v, "skin"))
 973         {
 974             /* skip line */
 975             ch = lex_getch(lex);
 976             while (ch != EOF && ch != '\n')
 977                 ch = lex_getch(lex);
 978             return lex_do(lex);
 979         }
 980
 981         for (frame = 0; frame < vec_size(lex->frames); ++frame) {
 982             if (!strcmp(v, lex->frames[frame].name)) {
 983                 lex->tok.constval.i = lex->frames[frame].value;
 984                 return (lex->tok.ttype = TOKEN_INTCONST);
 985             }
 986         }
 987
 988         lexerror(lex, "invalid frame macro");
 989         return lex_do(lex);
 990     }
 991
 992     /* single-character tokens */
 993     switch (ch)
 994     {
 995         case '[':
 996         case '(':
 997             lex_tokench(lex, ch);
 998             lex_endtoken(lex);
 999             if (lex->flags.noops)
1000                 return (lex->tok.ttype = ch);
1001             else
1002                 return (lex->tok.ttype = TOKEN_OPERATOR);
1003         case ')':
1004         case ';':
1005         case ':':
1006         case '{':
1007         case '}':
1008         case ']':
1009
1010         case '#':
1011             lex_tokench(lex, ch);
1012             lex_endtoken(lex);
1013             return (lex->tok.ttype = ch);
1014         default:
1015             break;
1016     }
1017
1018     if (lex->flags.noops)
1019     {
1020         /* Detect characters early which are normally
1021          * operators OR PART of an operator.
1022          */
1023         switch (ch)
1024         {
1025             case '+':
1026             case '-':
1027             case '*':
1028             case '/':
1029             case '<':
1030             case '>':
1031             case '=':
1032             case '&':
1033             case '|':
1034             case '^':
1035             case '~':
1036             case ',':
1037             case '!':
1038                 lex_tokench(lex, ch);
1039                 lex_endtoken(lex);
1040                 return (lex->tok.ttype = ch);
1041             default:
1042                 break;
1043         }
1044
1045         if (ch == '.')
1046         {
1047             lex_tokench(lex, ch);
1048             /* peak ahead once */
1049             nextch = lex_getch(lex);
1050             if (nextch != '.') {
1051                 lex_ungetch(lex, nextch);
1052                 lex_endtoken(lex);
1053                 return (lex->tok.ttype = ch);
1054             }
1055             /* peak ahead again */
1056             nextch = lex_getch(lex);
1057             if (nextch != '.') {
1058                 lex_ungetch(lex, nextch);
1059                 lex_ungetch(lex, nextch);
1060                 lex_endtoken(lex);
1061                 return (lex->tok.ttype = ch);
1062             }
1063             /* fill the token to be "..." */
1064             lex_tokench(lex, ch);
1065             lex_tokench(lex, ch);
1066             lex_endtoken(lex);
1067             return (lex->tok.ttype = TOKEN_DOTS);
1068         }
1069     }
1070
1071     if (ch == ',' || ch == '.') {
1072         lex_tokench(lex, ch);
1073         lex_endtoken(lex);
1074         return (lex->tok.ttype = TOKEN_OPERATOR);
1075     }
1076
1077     if (ch == '+' || ch == '-' || /* ++, --, +=, -=  and -> as well! */
1078         ch == '>' || ch == '<' || /* <<, >>, <=, >= */
1079         ch == '=' || ch == '!' || /* ==, != */
1080         ch == '&' || ch == '|')   /* &&, ||, &=, |= */
1081     {
1082         lex_tokench(lex, ch);
1083
1084         nextch = lex_getch(lex);
1085         if (nextch == ch || nextch == '=') {
1086             lex_tokench(lex, nextch);
1087         } else if (ch == '-' && nextch == '>') {
1088             lex_tokench(lex, nextch);
1089         } else
1090             lex_ungetch(lex, nextch);
1091
1092         lex_endtoken(lex);
1093         return (lex->tok.ttype = TOKEN_OPERATOR);
1094     }
1095
1096     /*
1097     if (ch == '^' || ch == '~' || ch == '!')
1098     {
1099         lex_tokench(lex, ch);
1100         lex_endtoken(lex);
1101         return (lex->tok.ttype = TOKEN_OPERATOR);
1102     }
1103     */
1104
1105     if (ch == '*' || ch == '/') /* *=, /= */
1106     {
1107         lex_tokench(lex, ch);
1108
1109         nextch = lex_getch(lex);
1110         if (nextch == '=') {
1111             lex_tokench(lex, nextch);
1112         } else
1113             lex_ungetch(lex, nextch);
1114
1115         lex_endtoken(lex);
1116         return (lex->tok.ttype = TOKEN_OPERATOR);
1117     }
1118
1119     if (isident_start(ch))
1120     {
1121         const char *v;
1122
1123         lex_tokench(lex, ch);
1124         if (!lex_finish_ident(lex)) {
1125             /* error? */
1126             return (lex->tok.ttype = TOKEN_ERROR);
1127         }
1128         lex_endtoken(lex);
1129         lex->tok.ttype = TOKEN_IDENT;
1130
1131         v = lex->tok.value;
1132         if (!strcmp(v, "void")) {
1133             lex->tok.ttype = TOKEN_TYPENAME;
1134             lex->tok.constval.t = TYPE_VOID;
1135         } else if (!strcmp(v, "int")) {
1136             lex->tok.ttype = TOKEN_TYPENAME;
1137             lex->tok.constval.t = TYPE_INTEGER;
1138         } else if (!strcmp(v, "float")) {
1139             lex->tok.ttype = TOKEN_TYPENAME;
1140             lex->tok.constval.t = TYPE_FLOAT;
1141         } else if (!strcmp(v, "string")) {
1142             lex->tok.ttype = TOKEN_TYPENAME;
1143             lex->tok.constval.t = TYPE_STRING;
1144         } else if (!strcmp(v, "entity")) {
1145             lex->tok.ttype = TOKEN_TYPENAME;
1146             lex->tok.constval.t = TYPE_ENTITY;
1147         } else if (!strcmp(v, "vector")) {
1148             lex->tok.ttype = TOKEN_TYPENAME;
1149             lex->tok.constval.t = TYPE_VECTOR;
1150         } else if (!strcmp(v, "for")  ||
1151                  !strcmp(v, "while")  ||
1152                  !strcmp(v, "do")     ||
1153                  !strcmp(v, "if")     ||
1154                  !strcmp(v, "else")   ||
1155                  !strcmp(v, "local")  ||
1156                  !strcmp(v, "return") ||
1157                  !strcmp(v, "not")    ||
1158                  !strcmp(v, "const"))
1159         {
1160             lex->tok.ttype = TOKEN_KEYWORD;
1161         }
1162         else if (opts_standard != COMPILER_QCC)
1163         {
1164             /* other standards reserve these keywords */
1165             if (!strcmp(v, "switch") ||
1166                 !strcmp(v, "struct") ||
1167                 !strcmp(v, "union")  ||
1168                 !strcmp(v, "break")  ||
1169                 !strcmp(v, "continue") ||
1170                 !strcmp(v, "var"))
1171             {
1172                 lex->tok.ttype = TOKEN_KEYWORD;
1173             }
1174         }
1175
1176         return lex->tok.ttype;
1177     }
1178
1179     if (ch == '"')
1180     {
1181         lex->flags.nodigraphs = true;
1182         if (lex->flags.preprocessing)
1183             lex_tokench(lex, ch);
1184         lex->tok.ttype = lex_finish_string(lex, '"');
1185         if (lex->flags.preprocessing)
1186             lex_tokench(lex, ch);
1187         while (!lex->flags.preprocessing && lex->tok.ttype == TOKEN_STRINGCONST)
1188         {
1189             /* Allow c style "string" "continuation" */
1190             ch = lex_skipwhite(lex);
1191             if (ch != '"') {
1192                 lex_ungetch(lex, ch);
1193                 break;
1194             }
1195
1196             lex->tok.ttype = lex_finish_string(lex, '"');
1197         }
1198         lex->flags.nodigraphs = false;
1199         lex_endtoken(lex);
1200         return lex->tok.ttype;
1201     }
1202
1203     if (ch == '\'')
1204     {
1205         /* we parse character constants like string,
1206          * but return TOKEN_CHARCONST, or a vector type if it fits...
1207          * Likewise actual unescaping has to be done by the parser.
1208          * The difference is we don't allow 'char' 'continuation'.
1209          */
1210         if (lex->flags.preprocessing)
1211             lex_tokench(lex, ch);
1212         lex->tok.ttype = lex_finish_string(lex, '\'');
1213         if (lex->flags.preprocessing)
1214             lex_tokench(lex, ch);
1215         lex_endtoken(lex);
1216
1217          /* It's a vector if we can successfully scan 3 floats */
1218 #ifdef WIN32
1219         if (sscanf_s(lex->tok.value, " %f %f %f ",
1220                    &lex->tok.constval.v.x, &lex->tok.constval.v.y, &lex->tok.constval.v.z) == 3)
1221 #else
1222         if (sscanf(lex->tok.value, " %f %f %f ",
1223                    &lex->tok.constval.v.x, &lex->tok.constval.v.y, &lex->tok.constval.v.z) == 3)
1224 #endif
1225
1226         {
1227              lex->tok.ttype = TOKEN_VECTORCONST;
1228         }
1229
1230         return lex->tok.ttype;
1231     }
1232
1233     if (isdigit(ch))
1234     {
1235         lex->tok.ttype = lex_finish_digit(lex, ch);
1236         lex_endtoken(lex);
1237         return lex->tok.ttype;
1238     }
1239
1240     lexerror(lex, "unknown token");
1241     return (lex->tok.ttype = TOKEN_ERROR);
1242 }