nsz Git - cparser/blob - lexer.c

   1 /*
   2  * This file is part of cparser.
   3  * Copyright (C) 2007-2009 Matthias Braun <matze@braunis.de>
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License
   7  * as published by the Free Software Foundation; either version 2
   8  * of the License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  18  * 02111-1307, USA.
  19  */
  20 #include <config.h>
  21
  22 #include "input.h"
  23 #include "diagnostic.h"
  24 #include "lexer.h"
  25 #include "symbol_t.h"
  26 #include "token_t.h"
  27 #include "symbol_table_t.h"
  28 #include "adt/error.h"
  29 #include "adt/strset.h"
  30 #include "adt/util.h"
  31 #include "types.h"
  32 #include "type_t.h"
  33 #include "target_architecture.h"
  34 #include "parser.h"
  35 #include "warning.h"
  36 #include "lang_features.h"
  37
  38 #include <assert.h>
  39 #include <errno.h>
  40 #include <string.h>
  41 #include <stdbool.h>
  42 #include <ctype.h>
  43
  44 #ifndef _WIN32
  45 #include <strings.h>
  46 #endif
  47
  48 #define MAX_PUTBACK 16    // 3 would be enough, but 16 gives a nicer alignment
  49 #define BUF_SIZE    1024
  50
  51 static input_t           *input;
  52 static utf32              input_buf[BUF_SIZE + MAX_PUTBACK];
  53 static const utf32       *bufpos;
  54 static const utf32       *bufend;
  55 static utf32              c;
  56 static source_position_t  lexer_pos;
  57 token_t                   lexer_token;
  58 static symbol_t          *symbol_L;
  59 static strset_t           stringset;
  60 static char              *encoding;
  61 bool                      allow_dollar_in_symbol = true;
  62
  63 /**
  64  * Prints a parse error message at the current token.
  65  *
  66  * @param msg   the error message
  67  */
  68 static void parse_error(const char *msg)
  69 {
  70         errorf(&lexer_pos, "%s", msg);
  71 }
  72
  73 /**
  74  * Prints an internal error message at the current token.
  75  *
  76  * @param msg   the error message
  77  */
  78 static NORETURN internal_error(const char *msg)
  79 {
  80         internal_errorf(&lexer_pos, "%s", msg);
  81 }
  82
  83 static inline void next_real_char(void)
  84 {
  85         assert(bufpos <= bufend);
  86         if (bufpos >= bufend) {
  87                 size_t n = decode(input, input_buf+MAX_PUTBACK, BUF_SIZE);
  88                 if (n == 0) {
  89                         c = EOF;
  90                         return;
  91                 }
  92                 bufpos = input_buf + MAX_PUTBACK;
  93                 bufend = bufpos + n;
  94         }
  95         c = *bufpos++;
  96         ++lexer_pos.colno;
  97 }
  98
  99 /**
 100  * Put a character back into the buffer.
 101  *
 102  * @param pc  the character to put back
 103  */
 104 static inline void put_back(utf32 const pc)
 105 {
 106         *(--bufpos - input_buf + input_buf) = pc;
 107         --lexer_pos.colno;
 108 }
 109
 110 static inline void next_char(void);
 111
 112 #define MATCH_NEWLINE(code)  \
 113         case '\r':               \
 114                 next_char();         \
 115                 if (c == '\n') {     \
 116         case '\n':               \
 117                         next_char();     \
 118                 }                    \
 119                 lexer_pos.lineno++;  \
 120                 lexer_pos.colno = 1; \
 121                 code
 122
 123 #define eat(c_type) (assert(c == c_type), next_char())
 124
 125 static void maybe_concat_lines(void)
 126 {
 127         eat('\\');
 128
 129         switch (c) {
 130         MATCH_NEWLINE(return;)
 131
 132         default:
 133                 break;
 134         }
 135
 136         put_back(c);
 137         c = '\\';
 138 }
 139
 140 /**
 141  * Set c to the next input character, ie.
 142  * after expanding trigraphs.
 143  */
 144 static inline void next_char(void)
 145 {
 146         next_real_char();
 147
 148         /* filter trigraphs */
 149         if (UNLIKELY(c == '\\')) {
 150                 maybe_concat_lines();
 151                 return;
 152         }
 153
 154         if (LIKELY(c != '?'))
 155                 return;
 156
 157         next_real_char();
 158         if (LIKELY(c != '?')) {
 159                 put_back(c);
 160                 c = '?';
 161                 return;
 162         }
 163
 164         next_real_char();
 165         switch (c) {
 166         case '=': c = '#'; break;
 167         case '(': c = '['; break;
 168         case '/': c = '\\'; maybe_concat_lines(); break;
 169         case ')': c = ']'; break;
 170         case '\'': c = '^'; break;
 171         case '<': c = '{'; break;
 172         case '!': c = '|'; break;
 173         case '>': c = '}'; break;
 174         case '-': c = '~'; break;
 175         default:
 176                 put_back(c);
 177                 put_back('?');
 178                 c = '?';
 179                 break;
 180         }
 181 }
 182
 183 #define SYMBOL_CHARS  \
 184         case '$': if (!allow_dollar_in_symbol) goto dollar_sign; \
 185         case 'a':         \
 186         case 'b':         \
 187         case 'c':         \
 188         case 'd':         \
 189         case 'e':         \
 190         case 'f':         \
 191         case 'g':         \
 192         case 'h':         \
 193         case 'i':         \
 194         case 'j':         \
 195         case 'k':         \
 196         case 'l':         \
 197         case 'm':         \
 198         case 'n':         \
 199         case 'o':         \
 200         case 'p':         \
 201         case 'q':         \
 202         case 'r':         \
 203         case 's':         \
 204         case 't':         \
 205         case 'u':         \
 206         case 'v':         \
 207         case 'w':         \
 208         case 'x':         \
 209         case 'y':         \
 210         case 'z':         \
 211         case 'A':         \
 212         case 'B':         \
 213         case 'C':         \
 214         case 'D':         \
 215         case 'E':         \
 216         case 'F':         \
 217         case 'G':         \
 218         case 'H':         \
 219         case 'I':         \
 220         case 'J':         \
 221         case 'K':         \
 222         case 'L':         \
 223         case 'M':         \
 224         case 'N':         \
 225         case 'O':         \
 226         case 'P':         \
 227         case 'Q':         \
 228         case 'R':         \
 229         case 'S':         \
 230         case 'T':         \
 231         case 'U':         \
 232         case 'V':         \
 233         case 'W':         \
 234         case 'X':         \
 235         case 'Y':         \
 236         case 'Z':         \
 237         case '_':
 238
 239 #define DIGITS        \
 240         case '0':         \
 241         case '1':         \
 242         case '2':         \
 243         case '3':         \
 244         case '4':         \
 245         case '5':         \
 246         case '6':         \
 247         case '7':         \
 248         case '8':         \
 249         case '9':
 250
 251 /**
 252  * Read a symbol from the input and build
 253  * the lexer_token.
 254  */
 255 static void parse_symbol(void)
 256 {
 257         obstack_1grow(&symbol_obstack, (char) c);
 258         next_char();
 259
 260         while (true) {
 261                 switch (c) {
 262                 DIGITS
 263                 SYMBOL_CHARS
 264                         obstack_1grow(&symbol_obstack, (char) c);
 265                         next_char();
 266                         break;
 267
 268                 default:
 269 dollar_sign:
 270                         goto end_symbol;
 271                 }
 272         }
 273
 274 end_symbol:
 275         obstack_1grow(&symbol_obstack, '\0');
 276
 277         char     *string = obstack_finish(&symbol_obstack);
 278         symbol_t *symbol = symbol_table_insert(string);
 279
 280         lexer_token.kind              = symbol->ID;
 281         lexer_token.identifier.symbol = symbol;
 282
 283         if (symbol->string != string) {
 284                 obstack_free(&symbol_obstack, string);
 285         }
 286 }
 287
 288 static string_t identify_string(char *string, size_t len)
 289 {
 290         /* TODO hash */
 291 #if 0
 292         const char *result = strset_insert(&stringset, concat);
 293         if (result != concat) {
 294                 obstack_free(&symbol_obstack, concat);
 295         }
 296 #else
 297         const char *result = string;
 298 #endif
 299         return (string_t) {result, len};
 300 }
 301
 302 /**
 303  * parse suffixes like 'LU' or 'f' after numbers
 304  */
 305 static void parse_number_suffix(void)
 306 {
 307         assert(obstack_object_size(&symbol_obstack) == 0);
 308         while (true) {
 309                 switch (c) {
 310                 SYMBOL_CHARS
 311                         obstack_1grow(&symbol_obstack, (char) c);
 312                         next_char();
 313                         break;
 314                 default:
 315                 dollar_sign:
 316                         goto finish_suffix;
 317                 }
 318         }
 319 finish_suffix:
 320         if (obstack_object_size(&symbol_obstack) == 0) {
 321                 lexer_token.number.suffix.begin = NULL;
 322                 lexer_token.number.suffix.size  = 0;
 323                 return;
 324         }
 325
 326         obstack_1grow(&symbol_obstack, '\0');
 327         size_t    size   = obstack_object_size(&symbol_obstack);
 328         char     *string = obstack_finish(&symbol_obstack);
 329
 330         lexer_token.number.suffix = identify_string(string, size);
 331 }
 332
 333 /**
 334  * Parses a hex number including hex floats and set the
 335  * lexer_token.
 336  */
 337 static void parse_number_hex(void)
 338 {
 339         bool is_float   = false;
 340         bool has_digits = false;
 341
 342         assert(obstack_object_size(&symbol_obstack) == 0);
 343         while (isxdigit(c)) {
 344                 has_digits = true;
 345                 obstack_1grow(&symbol_obstack, (char) c);
 346                 next_char();
 347         }
 348
 349         if (c == '.') {
 350                 is_float = true;
 351                 obstack_1grow(&symbol_obstack, (char) c);
 352                 next_char();
 353
 354                 while (isxdigit(c)) {
 355                         has_digits = true;
 356                         obstack_1grow(&symbol_obstack, (char) c);
 357                         next_char();
 358                 }
 359         }
 360         if (c == 'p' || c == 'P') {
 361                 is_float = true;
 362                 obstack_1grow(&symbol_obstack, (char) c);
 363                 next_char();
 364
 365                 if (c == '-' || c == '+') {
 366                         obstack_1grow(&symbol_obstack, (char) c);
 367                         next_char();
 368                 }
 369
 370                 while (isxdigit(c)) {
 371                         obstack_1grow(&symbol_obstack, (char) c);
 372                         next_char();
 373                 }
 374         } else if (is_float) {
 375                 errorf(&lexer_token.base.source_position,
 376                        "hexadecimal floatingpoint constant requires an exponent");
 377         }
 378         obstack_1grow(&symbol_obstack, '\0');
 379
 380         size_t  size   = obstack_object_size(&symbol_obstack) - 1;
 381         char   *string = obstack_finish(&symbol_obstack);
 382         lexer_token.number.number = identify_string(string, size);
 383
 384         lexer_token.kind    =
 385                 is_float ? T_FLOATINGPOINT_HEXADECIMAL : T_INTEGER_HEXADECIMAL;
 386
 387         if (!has_digits) {
 388                 errorf(&lexer_token.base.source_position,
 389                        "invalid number literal '0x%S'", &lexer_token.number.number);
 390                 lexer_token.number.number.begin = "0";
 391                 lexer_token.number.number.size  = 1;
 392         }
 393
 394         parse_number_suffix();
 395 }
 396
 397 /**
 398  * Returns true if the given char is a octal digit.
 399  *
 400  * @param char  the character to check
 401  */
 402 static bool is_octal_digit(utf32 chr)
 403 {
 404         return '0' <= chr && chr <= '7';
 405 }
 406
 407 /**
 408  * Parses a number and sets the lexer_token.
 409  */
 410 static void parse_number(void)
 411 {
 412         bool is_float   = false;
 413         bool has_digits = false;
 414
 415         assert(obstack_object_size(&symbol_obstack) == 0);
 416         if (c == '0') {
 417                 next_char();
 418                 if (c == 'x' || c == 'X') {
 419                         next_char();
 420                         parse_number_hex();
 421                         return;
 422                 } else {
 423                         has_digits = true;
 424                 }
 425                 obstack_1grow(&symbol_obstack, '0');
 426         }
 427
 428         while (isdigit(c)) {
 429                 has_digits = true;
 430                 obstack_1grow(&symbol_obstack, (char) c);
 431                 next_char();
 432         }
 433
 434         if (c == '.') {
 435                 is_float = true;
 436                 obstack_1grow(&symbol_obstack, '.');
 437                 next_char();
 438
 439                 while (isdigit(c)) {
 440                         has_digits = true;
 441                         obstack_1grow(&symbol_obstack, (char) c);
 442                         next_char();
 443                 }
 444         }
 445         if (c == 'e' || c == 'E') {
 446                 is_float = true;
 447                 obstack_1grow(&symbol_obstack, 'e');
 448                 next_char();
 449
 450                 if (c == '-' || c == '+') {
 451                         obstack_1grow(&symbol_obstack, (char) c);
 452                         next_char();
 453                 }
 454
 455                 while (isdigit(c)) {
 456                         obstack_1grow(&symbol_obstack, (char) c);
 457                         next_char();
 458                 }
 459         }
 460
 461         obstack_1grow(&symbol_obstack, '\0');
 462         size_t  size   = obstack_object_size(&symbol_obstack) - 1;
 463         char   *string = obstack_finish(&symbol_obstack);
 464         lexer_token.number.number = identify_string(string, size);
 465
 466         /* is it an octal number? */
 467         if (is_float) {
 468                 lexer_token.kind = T_FLOATINGPOINT;
 469         } else if (string[0] == '0') {
 470                 lexer_token.kind = T_INTEGER_OCTAL;
 471
 472                 /* check for invalid octal digits */
 473                 for (size_t i= 0; i < size; ++i) {
 474                         char t = string[i];
 475                         if (t >= '8')
 476                                 errorf(&lexer_token.base.source_position,
 477                                        "invalid digit '%c' in octal number", t);
 478                 }
 479         } else {
 480                 lexer_token.kind = T_INTEGER;
 481         }
 482
 483         if (!has_digits) {
 484                 errorf(&lexer_token.base.source_position, "invalid number literal '%S'",
 485                        &lexer_token.number.number);
 486         }
 487
 488         parse_number_suffix();
 489 }
 490
 491 /**
 492  * Returns the value of a digit.
 493  * The only portable way to do it ...
 494  */
 495 static int digit_value(utf32 const digit)
 496 {
 497         switch (digit) {
 498         case '0': return 0;
 499         case '1': return 1;
 500         case '2': return 2;
 501         case '3': return 3;
 502         case '4': return 4;
 503         case '5': return 5;
 504         case '6': return 6;
 505         case '7': return 7;
 506         case '8': return 8;
 507         case '9': return 9;
 508         case 'a':
 509         case 'A': return 10;
 510         case 'b':
 511         case 'B': return 11;
 512         case 'c':
 513         case 'C': return 12;
 514         case 'd':
 515         case 'D': return 13;
 516         case 'e':
 517         case 'E': return 14;
 518         case 'f':
 519         case 'F': return 15;
 520         default:
 521                 internal_error("wrong character given");
 522         }
 523 }
 524
 525 /**
 526  * Parses an octal character sequence.
 527  *
 528  * @param first_digit  the already read first digit
 529  */
 530 static utf32 parse_octal_sequence(utf32 const first_digit)
 531 {
 532         assert(is_octal_digit(first_digit));
 533         utf32 value = digit_value(first_digit);
 534         if (!is_octal_digit(c)) return value;
 535         value = 8 * value + digit_value(c);
 536         next_char();
 537         if (!is_octal_digit(c)) return value;
 538         value = 8 * value + digit_value(c);
 539         next_char();
 540         return value;
 541 }
 542
 543 /**
 544  * Parses a hex character sequence.
 545  */
 546 static utf32 parse_hex_sequence(void)
 547 {
 548         utf32 value = 0;
 549         while (isxdigit(c)) {
 550                 value = 16 * value + digit_value(c);
 551                 next_char();
 552         }
 553         return value;
 554 }
 555
 556 /**
 557  * Parse an escape sequence.
 558  */
 559 static utf32 parse_escape_sequence(void)
 560 {
 561         eat('\\');
 562
 563         utf32 const ec = c;
 564         next_char();
 565
 566         switch (ec) {
 567         case '"':  return '"';
 568         case '\'': return '\'';
 569         case '\\': return '\\';
 570         case '?': return '\?';
 571         case 'a': return '\a';
 572         case 'b': return '\b';
 573         case 'f': return '\f';
 574         case 'n': return '\n';
 575         case 'r': return '\r';
 576         case 't': return '\t';
 577         case 'v': return '\v';
 578         case 'x':
 579                 return parse_hex_sequence();
 580         case '0':
 581         case '1':
 582         case '2':
 583         case '3':
 584         case '4':
 585         case '5':
 586         case '6':
 587         case '7':
 588                 return parse_octal_sequence(ec);
 589         case EOF:
 590                 parse_error("reached end of file while parsing escape sequence");
 591                 return EOF;
 592         /* \E is not documented, but handled, by GCC.  It is acceptable according
 593          * to §6.11.4, whereas \e is not. */
 594         case 'E':
 595         case 'e':
 596                 if (c_mode & _GNUC)
 597                         return 27;   /* hopefully 27 is ALWAYS the code for ESCAPE */
 598                 break;
 599         case 'u':
 600         case 'U':
 601                 parse_error("universal character parsing not implemented yet");
 602                 return EOF;
 603         default:
 604                 break;
 605         }
 606         /* §6.4.4.4:8 footnote 64 */
 607         parse_error("unknown escape sequence");
 608         return EOF;
 609 }
 610
 611 /**
 612  * Concatenate two strings.
 613  */
 614 string_t concat_strings(const string_t *const s1, const string_t *const s2)
 615 {
 616         const size_t len1 = s1->size - 1;
 617         const size_t len2 = s2->size - 1;
 618
 619         char *const concat = obstack_alloc(&symbol_obstack, len1 + len2 + 1);
 620         memcpy(concat, s1->begin, len1);
 621         memcpy(concat + len1, s2->begin, len2 + 1);
 622
 623         return identify_string(concat, len1 + len2 + 1);
 624 }
 625
 626 string_t make_string(const char *string)
 627 {
 628         size_t      len   = strlen(string) + 1;
 629         char *const space = obstack_alloc(&symbol_obstack, len);
 630         memcpy(space, string, len);
 631
 632         return identify_string(space, len);
 633 }
 634
 635 /**
 636  * Parse a string literal and set lexer_token.
 637  */
 638 static void parse_string_literal(void)
 639 {
 640         eat('"');
 641
 642         while (true) {
 643                 switch (c) {
 644                 case '\\': {
 645                         utf32 const tc = parse_escape_sequence();
 646                         if (tc >= 0x100) {
 647                                 warningf(WARN_OTHER, &lexer_pos, "escape sequence out of range");
 648                         }
 649                         obstack_1grow(&symbol_obstack, tc);
 650                         break;
 651                 }
 652
 653                 case EOF: {
 654                         errorf(&lexer_token.base.source_position, "string has no end");
 655                         lexer_token.kind = T_ERROR;
 656                         return;
 657                 }
 658
 659                 case '"':
 660                         next_char();
 661                         goto end_of_string;
 662
 663                 default:
 664                         obstack_grow_symbol(&symbol_obstack, c);
 665                         next_char();
 666                         break;
 667                 }
 668         }
 669
 670 end_of_string:
 671
 672         /* TODO: concatenate multiple strings separated by whitespace... */
 673
 674         /* add finishing 0 to the string */
 675         obstack_1grow(&symbol_obstack, '\0');
 676         const size_t  size   = (size_t)obstack_object_size(&symbol_obstack);
 677         char         *string = obstack_finish(&symbol_obstack);
 678
 679         lexer_token.kind          = T_STRING_LITERAL;
 680         lexer_token.string.string = identify_string(string, size);
 681 }
 682
 683 /**
 684  * Parse a wide character constant and set lexer_token.
 685  */
 686 static void parse_wide_character_constant(void)
 687 {
 688         eat('\'');
 689
 690         while (true) {
 691                 switch (c) {
 692                 case '\\': {
 693                         const utf32 tc = parse_escape_sequence();
 694                         obstack_grow_symbol(&symbol_obstack, tc);
 695                         break;
 696                 }
 697
 698                 MATCH_NEWLINE(
 699                         parse_error("newline while parsing character constant");
 700                         break;
 701                 )
 702
 703                 case '\'':
 704                         next_char();
 705                         goto end_of_wide_char_constant;
 706
 707                 case EOF: {
 708                         errorf(&lexer_token.base.source_position,
 709                                "EOF while parsing character constant");
 710                         lexer_token.kind = T_ERROR;
 711                         return;
 712                 }
 713
 714                 default:
 715                         obstack_grow_symbol(&symbol_obstack, c);
 716                         next_char();
 717                         break;
 718                 }
 719         }
 720
 721 end_of_wide_char_constant:;
 722         obstack_1grow(&symbol_obstack, '\0');
 723         size_t  size   = (size_t) obstack_object_size(&symbol_obstack) - 1;
 724         char   *string = obstack_finish(&symbol_obstack);
 725
 726         lexer_token.kind          = T_WIDE_CHARACTER_CONSTANT;
 727         lexer_token.string.string = identify_string(string, size);
 728
 729         if (size == 0) {
 730                 errorf(&lexer_token.base.source_position, "empty character constant");
 731         }
 732 }
 733
 734 /**
 735  * Parse a wide string literal and set lexer_token.
 736  */
 737 static void parse_wide_string_literal(void)
 738 {
 739         parse_string_literal();
 740         if (lexer_token.kind == T_STRING_LITERAL)
 741                 lexer_token.kind = T_WIDE_STRING_LITERAL;
 742 }
 743
 744 /**
 745  * Parse a character constant and set lexer_token.
 746  */
 747 static void parse_character_constant(void)
 748 {
 749         eat('\'');
 750
 751         while (true) {
 752                 switch (c) {
 753                 case '\\': {
 754                         utf32 const tc = parse_escape_sequence();
 755                         if (tc >= 0x100) {
 756                                 warningf(WARN_OTHER, &lexer_pos, "escape sequence out of range");
 757                         }
 758                         obstack_1grow(&symbol_obstack, tc);
 759                         break;
 760                 }
 761
 762                 MATCH_NEWLINE(
 763                         parse_error("newline while parsing character constant");
 764                         break;
 765                 )
 766
 767                 case '\'':
 768                         next_char();
 769                         goto end_of_char_constant;
 770
 771                 case EOF: {
 772                         errorf(&lexer_token.base.source_position,
 773                                "EOF while parsing character constant");
 774                         lexer_token.kind = T_ERROR;
 775                         return;
 776                 }
 777
 778                 default:
 779                         obstack_grow_symbol(&symbol_obstack, c);
 780                         next_char();
 781                         break;
 782
 783                 }
 784         }
 785
 786 end_of_char_constant:;
 787         obstack_1grow(&symbol_obstack, '\0');
 788         const size_t        size   = (size_t)obstack_object_size(&symbol_obstack)-1;
 789         char         *const string = obstack_finish(&symbol_obstack);
 790
 791         lexer_token.kind          = T_CHARACTER_CONSTANT;
 792         lexer_token.string.string = identify_string(string, size);
 793
 794         if (size == 0) {
 795                 errorf(&lexer_token.base.source_position, "empty character constant");
 796         }
 797 }
 798
 799 /**
 800  * Skip a multiline comment.
 801  */
 802 static void skip_multiline_comment(void)
 803 {
 804         while (true) {
 805                 switch (c) {
 806                 case '/':
 807                         next_char();
 808                         if (c == '*') {
 809                                 /* nested comment, warn here */
 810                                 warningf(WARN_COMMENT, &lexer_pos, "'/*' within comment");
 811                         }
 812                         break;
 813                 case '*':
 814                         next_char();
 815                         if (c == '/') {
 816                                 next_char();
 817                                 return;
 818                         }
 819                         break;
 820
 821                 MATCH_NEWLINE(break;)
 822
 823                 case EOF: {
 824                         errorf(&lexer_token.base.source_position,
 825                                "at end of file while looking for comment end");
 826                         return;
 827                 }
 828
 829                 default:
 830                         next_char();
 831                         break;
 832                 }
 833         }
 834 }
 835
 836 /**
 837  * Skip a single line comment.
 838  */
 839 static void skip_line_comment(void)
 840 {
 841         while (true) {
 842                 switch (c) {
 843                 case EOF:
 844                         return;
 845
 846                 case '\n':
 847                 case '\r':
 848                         return;
 849
 850                 case '\\':
 851                         next_char();
 852                         if (c == '\n' || c == '\r') {
 853                                 warningf(WARN_COMMENT, &lexer_pos, "multi-line comment");
 854                                 return;
 855                         }
 856                         break;
 857
 858                 default:
 859                         next_char();
 860                         break;
 861                 }
 862         }
 863 }
 864
 865 /** The current preprocessor token. */
 866 static token_t pp_token;
 867
 868 /**
 869  * Read the next preprocessor token.
 870  */
 871 static inline void next_pp_token(void)
 872 {
 873         lexer_next_preprocessing_token();
 874         pp_token = lexer_token;
 875 }
 876
 877 /**
 878  * Eat all preprocessor tokens until newline.
 879  */
 880 static void eat_until_newline(void)
 881 {
 882         while (pp_token.kind != '\n' && pp_token.kind != T_EOF) {
 883                 next_pp_token();
 884         }
 885 }
 886
 887 /**
 888  * Handle the define directive.
 889  */
 890 static void define_directive(void)
 891 {
 892         lexer_next_preprocessing_token();
 893         if (lexer_token.kind != T_IDENTIFIER) {
 894                 parse_error("expected identifier after #define\n");
 895                 eat_until_newline();
 896         }
 897 }
 898
 899 /**
 900  * Handle the ifdef directive.
 901  */
 902 static void ifdef_directive(int is_ifndef)
 903 {
 904         (void) is_ifndef;
 905         lexer_next_preprocessing_token();
 906         //expect_identifier();
 907         //extect_newline();
 908 }
 909
 910 /**
 911  * Handle the endif directive.
 912  */
 913 static void endif_directive(void)
 914 {
 915         //expect_newline();
 916 }
 917
 918 /**
 919  * Parse the line directive.
 920  */
 921 static void parse_line_directive(void)
 922 {
 923         if (pp_token.kind != T_INTEGER) {
 924                 parse_error("expected integer");
 925         } else {
 926                 /* use offset -1 as this is about the next line */
 927                 lexer_pos.lineno = atoi(pp_token.number.number.begin) - 1;
 928                 next_pp_token();
 929         }
 930         if (pp_token.kind == T_STRING_LITERAL) {
 931                 lexer_pos.input_name = pp_token.string.string.begin;
 932                 next_pp_token();
 933         }
 934
 935         eat_until_newline();
 936 }
 937
 938 /**
 939  * STDC pragmas.
 940  */
 941 typedef enum stdc_pragma_kind_t {
 942         STDC_UNKNOWN,
 943         STDC_FP_CONTRACT,
 944         STDC_FENV_ACCESS,
 945         STDC_CX_LIMITED_RANGE
 946 } stdc_pragma_kind_t;
 947
 948 /**
 949  * STDC pragma values.
 950  */
 951 typedef enum stdc_pragma_value_kind_t {
 952         STDC_VALUE_UNKNOWN,
 953         STDC_VALUE_ON,
 954         STDC_VALUE_OFF,
 955         STDC_VALUE_DEFAULT
 956 } stdc_pragma_value_kind_t;
 957
 958 /**
 959  * Parse a pragma directive.
 960  */
 961 static void parse_pragma(void)
 962 {
 963         bool unknown_pragma = true;
 964
 965         next_pp_token();
 966         if (pp_token.kind != T_IDENTIFIER) {
 967                 warningf(WARN_UNKNOWN_PRAGMAS, &pp_token.base.source_position,
 968                          "expected identifier after #pragma");
 969                 eat_until_newline();
 970                 return;
 971         }
 972
 973         symbol_t *symbol = pp_token.identifier.symbol;
 974         if (symbol->pp_ID == TP_STDC) {
 975                 stdc_pragma_kind_t kind = STDC_UNKNOWN;
 976                 /* a STDC pragma */
 977                 if (c_mode & _C99) {
 978                         next_pp_token();
 979
 980                         switch (pp_token.identifier.symbol->pp_ID) {
 981                         case TP_FP_CONTRACT:
 982                                 kind = STDC_FP_CONTRACT;
 983                                 break;
 984                         case TP_FENV_ACCESS:
 985                                 kind = STDC_FENV_ACCESS;
 986                                 break;
 987                         case TP_CX_LIMITED_RANGE:
 988                                 kind = STDC_CX_LIMITED_RANGE;
 989                                 break;
 990                         default:
 991                                 break;
 992                         }
 993                         if (kind != STDC_UNKNOWN) {
 994                                 stdc_pragma_value_kind_t value = STDC_VALUE_UNKNOWN;
 995                                 next_pp_token();
 996                                 switch (pp_token.identifier.symbol->pp_ID) {
 997                                 case TP_ON:
 998                                         value = STDC_VALUE_ON;
 999                                         break;
1000                                 case TP_OFF:
1001                                         value = STDC_VALUE_OFF;
1002                                         break;
1003                                 case TP_DEFAULT:
1004                                         value = STDC_VALUE_DEFAULT;
1005                                         break;
1006                                 default:
1007                                         break;
1008                                 }
1009                                 if (value != STDC_VALUE_UNKNOWN) {
1010                                         unknown_pragma = false;
1011                                 } else {
1012                                         errorf(&pp_token.base.source_position,
1013                                                "bad STDC pragma argument");
1014                                 }
1015                         }
1016                 }
1017         } else {
1018                 unknown_pragma = true;
1019         }
1020         eat_until_newline();
1021         if (unknown_pragma) {
1022                 warningf(WARN_UNKNOWN_PRAGMAS, &pp_token.base.source_position,
1023                          "encountered unknown #pragma");
1024         }
1025 }
1026
1027 /**
1028  * Parse a preprocessor non-null directive.
1029  */
1030 static void parse_preprocessor_identifier(void)
1031 {
1032         assert(pp_token.kind == T_IDENTIFIER);
1033         symbol_t *symbol = pp_token.identifier.symbol;
1034
1035         switch (symbol->pp_ID) {
1036         case TP_include:
1037                 printf("include - enable header name parsing!\n");
1038                 break;
1039         case TP_define:
1040                 define_directive();
1041                 break;
1042         case TP_ifdef:
1043                 ifdef_directive(0);
1044                 break;
1045         case TP_ifndef:
1046                 ifdef_directive(1);
1047                 break;
1048         case TP_endif:
1049                 endif_directive();
1050                 break;
1051         case TP_line:
1052                 next_pp_token();
1053                 parse_line_directive();
1054                 break;
1055         case TP_if:
1056         case TP_else:
1057         case TP_elif:
1058         case TP_undef:
1059         case TP_error:
1060                 /* TODO; output the rest of the line */
1061                 parse_error("#error directive: ");
1062                 break;
1063         case TP_pragma:
1064                 parse_pragma();
1065                 break;
1066         }
1067 }
1068
1069 /**
1070  * Parse a preprocessor directive.
1071  */
1072 static void parse_preprocessor_directive(void)
1073 {
1074         next_pp_token();
1075
1076         switch (pp_token.kind) {
1077         case T_IDENTIFIER:
1078                 parse_preprocessor_identifier();
1079                 break;
1080         case T_INTEGER:
1081                 parse_line_directive();
1082                 break;
1083         case '\n':
1084                 /* NULL directive, see §6.10.7 */
1085                 break;
1086         default:
1087                 parse_error("invalid preprocessor directive");
1088                 eat_until_newline();
1089                 break;
1090         }
1091 }
1092
1093 #define MAYBE_PROLOG                                       \
1094                         next_char();                                   \
1095                         while (true) {                                 \
1096                                 switch (c) {
1097
1098 #define MAYBE(ch, set_type)                                \
1099                                 case ch:                                   \
1100                                         next_char();                           \
1101                                         lexer_token.kind = set_type;           \
1102                                         return;
1103
1104 /* must use this as last thing */
1105 #define MAYBE_MODE(ch, set_type, mode)                     \
1106                                 case ch:                                   \
1107                                         if (c_mode & mode) {                   \
1108                                                 next_char();                       \
1109                                                 lexer_token.kind = set_type;       \
1110                                                 return;                            \
1111                                         }                                      \
1112                                         /* fallthrough */
1113
1114 #define ELSE_CODE(code)                                    \
1115                                 default:                                   \
1116                                         code                                   \
1117                                         return;                                \
1118                                 }                                          \
1119                         } /* end of while (true) */                    \
1120
1121 #define ELSE(set_type)                                     \
1122                 ELSE_CODE(                                         \
1123                         lexer_token.kind = set_type;                   \
1124                 )
1125
1126 void lexer_next_preprocessing_token(void)
1127 {
1128         while (true) {
1129                 lexer_token.base.source_position = lexer_pos;
1130
1131                 switch (c) {
1132                 case ' ':
1133                 case '\t':
1134                         next_char();
1135                         break;
1136
1137                 MATCH_NEWLINE(
1138                         lexer_token.kind = '\n';
1139                         return;
1140                 )
1141
1142                 SYMBOL_CHARS
1143                         parse_symbol();
1144                         /* might be a wide string ( L"string" ) */
1145                         if (lexer_token.identifier.symbol == symbol_L) {
1146                                 switch (c) {
1147                                         case '"':  parse_wide_string_literal();     break;
1148                                         case '\'': parse_wide_character_constant(); break;
1149                                 }
1150                         }
1151                         return;
1152
1153                 DIGITS
1154                         parse_number();
1155                         return;
1156
1157                 case '"':
1158                         parse_string_literal();
1159                         return;
1160
1161                 case '\'':
1162                         parse_character_constant();
1163                         return;
1164
1165                 case '.':
1166                         MAYBE_PROLOG
1167                                 DIGITS
1168                                         put_back(c);
1169                                         c = '.';
1170                                         parse_number();
1171                                         return;
1172
1173                                 case '.':
1174                                         MAYBE_PROLOG
1175                                         MAYBE('.', T_DOTDOTDOT)
1176                                         ELSE_CODE(
1177                                                 put_back(c);
1178                                                 c = '.';
1179                                                 lexer_token.kind = '.';
1180                                         )
1181                         ELSE('.')
1182                 case '&':
1183                         MAYBE_PROLOG
1184                         MAYBE('&', T_ANDAND)
1185                         MAYBE('=', T_ANDEQUAL)
1186                         ELSE('&')
1187                 case '*':
1188                         MAYBE_PROLOG
1189                         MAYBE('=', T_ASTERISKEQUAL)
1190                         ELSE('*')
1191                 case '+':
1192                         MAYBE_PROLOG
1193                         MAYBE('+', T_PLUSPLUS)
1194                         MAYBE('=', T_PLUSEQUAL)
1195                         ELSE('+')
1196                 case '-':
1197                         MAYBE_PROLOG
1198                         MAYBE('>', T_MINUSGREATER)
1199                         MAYBE('-', T_MINUSMINUS)
1200                         MAYBE('=', T_MINUSEQUAL)
1201                         ELSE('-')
1202                 case '!':
1203                         MAYBE_PROLOG
1204                         MAYBE('=', T_EXCLAMATIONMARKEQUAL)
1205                         ELSE('!')
1206                 case '/':
1207                         MAYBE_PROLOG
1208                         MAYBE('=', T_SLASHEQUAL)
1209                                 case '*':
1210                                         next_char();
1211                                         skip_multiline_comment();
1212                                         lexer_next_preprocessing_token();
1213                                         return;
1214                                 case '/':
1215                                         next_char();
1216                                         skip_line_comment();
1217                                         lexer_next_preprocessing_token();
1218                                         return;
1219                         ELSE('/')
1220                 case '%':
1221                         MAYBE_PROLOG
1222                         MAYBE('>', '}')
1223                         MAYBE('=', T_PERCENTEQUAL)
1224                                 case ':':
1225                                         MAYBE_PROLOG
1226                                                 case '%':
1227                                                         MAYBE_PROLOG
1228                                                         MAYBE(':', T_HASHHASH)
1229                                                         ELSE_CODE(
1230                                                                 put_back(c);
1231                                                                 c = '%';
1232                                                                 lexer_token.kind = '#';
1233                                                         )
1234                                         ELSE('#')
1235                         ELSE('%')
1236                 case '<':
1237                         MAYBE_PROLOG
1238                         MAYBE(':', '[')
1239                         MAYBE('%', '{')
1240                         MAYBE('=', T_LESSEQUAL)
1241                                 case '<':
1242                                         MAYBE_PROLOG
1243                                         MAYBE('=', T_LESSLESSEQUAL)
1244                                         ELSE(T_LESSLESS)
1245                         ELSE('<')
1246                 case '>':
1247                         MAYBE_PROLOG
1248                         MAYBE('=', T_GREATEREQUAL)
1249                                 case '>':
1250                                         MAYBE_PROLOG
1251                                         MAYBE('=', T_GREATERGREATEREQUAL)
1252                                         ELSE(T_GREATERGREATER)
1253                         ELSE('>')
1254                 case '^':
1255                         MAYBE_PROLOG
1256                         MAYBE('=', T_CARETEQUAL)
1257                         ELSE('^')
1258                 case '|':
1259                         MAYBE_PROLOG
1260                         MAYBE('=', T_PIPEEQUAL)
1261                         MAYBE('|', T_PIPEPIPE)
1262                         ELSE('|')
1263                 case ':':
1264                         MAYBE_PROLOG
1265                         MAYBE('>', ']')
1266                         MAYBE_MODE(':', T_COLONCOLON, _CXX)
1267                         ELSE(':')
1268                 case '=':
1269                         MAYBE_PROLOG
1270                         MAYBE('=', T_EQUALEQUAL)
1271                         ELSE('=')
1272                 case '#':
1273                         MAYBE_PROLOG
1274                         MAYBE('#', T_HASHHASH)
1275                         ELSE('#')
1276
1277                 case '?':
1278                 case '[':
1279                 case ']':
1280                 case '(':
1281                 case ')':
1282                 case '{':
1283                 case '}':
1284                 case '~':
1285                 case ';':
1286                 case ',':
1287                 case '\\':
1288                         lexer_token.kind = c;
1289                         next_char();
1290                         return;
1291
1292                 case EOF:
1293                         lexer_token.kind = T_EOF;
1294                         return;
1295
1296                 default:
1297 dollar_sign:
1298                         errorf(&lexer_pos, "unknown character '%c' found", c);
1299                         next_char();
1300                         lexer_token.kind = T_ERROR;
1301                         return;
1302                 }
1303         }
1304 }
1305
1306 void lexer_next_token(void)
1307 {
1308         lexer_next_preprocessing_token();
1309
1310         while (lexer_token.kind == '\n') {
1311 newline_found:
1312                 lexer_next_preprocessing_token();
1313         }
1314
1315         if (lexer_token.kind == '#') {
1316                 parse_preprocessor_directive();
1317                 goto newline_found;
1318         }
1319 }
1320
1321 void init_lexer(void)
1322 {
1323         strset_init(&stringset);
1324         symbol_L = symbol_table_insert("L");
1325 }
1326
1327 static void input_error(unsigned delta_lines, unsigned delta_cols,
1328                         const char *message)
1329 {
1330         lexer_pos.lineno += delta_lines;
1331         lexer_pos.colno  += delta_cols;
1332         errorf(&lexer_pos, "%s", message);
1333 }
1334
1335 void select_input_encoding(char const* new_encoding)
1336 {
1337         if (encoding != NULL)
1338                 xfree(encoding);
1339         encoding = xstrdup(new_encoding);
1340 }
1341
1342 void lexer_open_stream(FILE *stream, const char *input_name)
1343 {
1344         if (input != NULL) {
1345                 input_free(input);
1346                 input = NULL;
1347         }
1348
1349         lexer_pos.lineno     = 0;
1350         lexer_pos.colno      = 0;
1351         lexer_pos.input_name = input_name;
1352
1353         set_input_error_callback(input_error);
1354         input  = input_from_stream(stream, encoding);
1355         bufpos = NULL;
1356         bufend = NULL;
1357
1358         /* place a virtual \n at the beginning so the lexer knows that we're
1359          * at the beginning of a line */
1360         c = '\n';
1361 }
1362
1363 void exit_lexer(void)
1364 {
1365         if (input != NULL) {
1366                 input_free(input);
1367                 input = NULL;
1368         }
1369         strset_destroy(&stringset);
1370 }
1371
1372 static __attribute__((unused))
1373 void dbg_pos(const source_position_t source_position)
1374 {
1375         fprintf(stdout, "%s:%u:%u\n", source_position.input_name,
1376                 source_position.lineno, source_position.colno);
1377         fflush(stdout);
1378 }