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