nsz Git - cparser/blob - ast.c

   1 #include <config.h>
   2
   3 #include "ast_t.h"
   4 #include "type_t.h"
   5
   6 #include <assert.h>
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <ctype.h>
  10
  11 #include "adt/error.h"
  12
  13 struct obstack ast_obstack;
  14
  15 static FILE *out;
  16 static int   indent;
  17
  18 /** If set, implicit casts are printed. */
  19 bool print_implicit_casts = false;
  20
  21 /** If set parenthesis are printed to indicate operator precedence. */
  22 bool print_parenthesis = false;
  23
  24 static void print_statement(const statement_t *statement);
  25 static void print_expression_prec(const expression_t *expression, unsigned prec);
  26
  27 void change_indent(int delta)
  28 {
  29         indent += delta;
  30         assert(indent >= 0);
  31 }
  32
  33 void print_indent(void)
  34 {
  35         for(int i = 0; i < indent; ++i)
  36                 fprintf(out, "\t");
  37 }
  38
  39 enum precedence_t {
  40         PREC_BOTTOM  =  0,
  41         PREC_COMMA   =  2, /* ,                                    left to right */
  42         PREC_ASSIGN  =  4, /* = += -= *= /= %= <<= >>= &= ^= |=    right to left */
  43         PREC_COND    =  6, /* ?:                                   right to left */
  44         PREC_LOG_OR  =  8, /* ||                                   left to right */
  45         PREC_LOG_AND = 10, /* &&                                   left to right */
  46         PREC_BIT_OR  = 12, /* |                                    left to right */
  47         PREC_BIT_XOR = 14, /* ^                                    left to right */
  48         PREC_BIT_AND = 16, /* &                                    left to right */
  49         PREC_EQ      = 18, /* == !=                                left to right */
  50         PREC_CMP     = 20, /* < <= > >=                            left to right */
  51         PREC_SHF     = 22, /* << >>                                left to right */
  52         PREC_PLUS    = 24, /* + -                                  left to right */
  53         PREC_MUL     = 26, /* * / %                                left to right */
  54         PREC_UNARY   = 28, /* ! ~ ++ -- + - (type) * & sizeof      right to left */
  55         PREC_ACCESS  = 30, /* () [] -> .                           left to right */
  56         PREC_PRIM    = 32, /* primary */
  57         PREC_TOP     = 34
  58 };
  59
  60 /**
  61  * Returns 1 if a given precedence level has right-to-left
  62  * associativity, else -1.
  63  */
  64 static int right_to_left(unsigned precedence) {
  65         return (precedence == PREC_ASSIGN || precedence == PREC_COND ||
  66                 precedence == PREC_UNARY) ? 1 : -1;
  67 }
  68
  69 /**
  70  * Return the precedence of an expression.
  71  */
  72 static unsigned get_expression_precedence(expression_kind_t kind)
  73 {
  74         static const unsigned prec[] = {
  75                 [EXPR_UNKNOWN]                   = PREC_PRIM,
  76                 [EXPR_INVALID]                   = PREC_PRIM,
  77                 [EXPR_REFERENCE]                 = PREC_PRIM,
  78                 [EXPR_CONST]                     = PREC_PRIM,
  79                 [EXPR_STRING_LITERAL]            = PREC_PRIM,
  80                 [EXPR_WIDE_STRING_LITERAL]       = PREC_PRIM,
  81                 [EXPR_CALL]                      = PREC_PRIM,
  82                 [EXPR_CONDITIONAL]               = PREC_COND,
  83                 [EXPR_SELECT]                    = PREC_ACCESS,
  84                 [EXPR_ARRAY_ACCESS]              = PREC_ACCESS,
  85                 [EXPR_SIZEOF]                    = PREC_UNARY,
  86                 [EXPR_CLASSIFY_TYPE]             = PREC_UNARY,
  87                 [EXPR_ALIGNOF]                   = PREC_UNARY,
  88
  89                 [EXPR_FUNCTION]                  = PREC_PRIM,
  90                 [EXPR_PRETTY_FUNCTION]           = PREC_PRIM,
  91                 [EXPR_BUILTIN_SYMBOL]            = PREC_PRIM,
  92                 [EXPR_BUILTIN_CONSTANT_P]        = PREC_PRIM,
  93                 [EXPR_BUILTIN_PREFETCH]          = PREC_PRIM,
  94                 [EXPR_OFFSETOF]                  = PREC_PRIM,
  95                 [EXPR_VA_START]                  = PREC_PRIM,
  96                 [EXPR_VA_ARG]                    = PREC_PRIM,
  97                 [EXPR_STATEMENT]                 = PREC_ACCESS,
  98
  99                 [EXPR_UNARY_NEGATE]              = PREC_UNARY,
 100                 [EXPR_UNARY_PLUS]                = PREC_UNARY,
 101                 [EXPR_UNARY_BITWISE_NEGATE]      = PREC_UNARY,
 102                 [EXPR_UNARY_NOT]                 = PREC_UNARY,
 103                 [EXPR_UNARY_DEREFERENCE]         = PREC_UNARY,
 104                 [EXPR_UNARY_TAKE_ADDRESS]        = PREC_UNARY,
 105                 [EXPR_UNARY_POSTFIX_INCREMENT]   = PREC_UNARY,
 106                 [EXPR_UNARY_POSTFIX_DECREMENT]   = PREC_UNARY,
 107                 [EXPR_UNARY_PREFIX_INCREMENT]    = PREC_UNARY,
 108                 [EXPR_UNARY_PREFIX_DECREMENT]    = PREC_UNARY,
 109                 [EXPR_UNARY_CAST]                = PREC_UNARY,
 110                 [EXPR_UNARY_CAST_IMPLICIT]       = PREC_UNARY,
 111                 [EXPR_UNARY_ASSUME]              = PREC_PRIM,
 112                 [EXPR_UNARY_BITFIELD_EXTRACT]    = PREC_ACCESS,
 113
 114                 [EXPR_BINARY_ADD]                = PREC_PLUS,
 115                 [EXPR_BINARY_SUB]                = PREC_PLUS,
 116                 [EXPR_BINARY_MUL]                = PREC_MUL,
 117                 [EXPR_BINARY_DIV]                = PREC_MUL,
 118                 [EXPR_BINARY_MOD]                = PREC_MUL,
 119                 [EXPR_BINARY_EQUAL]              = PREC_EQ,
 120                 [EXPR_BINARY_NOTEQUAL]           = PREC_EQ,
 121                 [EXPR_BINARY_LESS]               = PREC_CMP,
 122                 [EXPR_BINARY_LESSEQUAL]          = PREC_CMP,
 123                 [EXPR_BINARY_GREATER]            = PREC_CMP,
 124                 [EXPR_BINARY_GREATEREQUAL]       = PREC_CMP,
 125                 [EXPR_BINARY_BITWISE_AND]        = PREC_BIT_AND,
 126                 [EXPR_BINARY_BITWISE_OR]         = PREC_BIT_OR,
 127                 [EXPR_BINARY_BITWISE_XOR]        = PREC_BIT_XOR,
 128                 [EXPR_BINARY_LOGICAL_AND]        = PREC_LOG_AND,
 129                 [EXPR_BINARY_LOGICAL_OR]         = PREC_LOG_OR,
 130                 [EXPR_BINARY_SHIFTLEFT]          = PREC_SHF,
 131                 [EXPR_BINARY_SHIFTRIGHT]         = PREC_SHF,
 132                 [EXPR_BINARY_ASSIGN]             = PREC_ASSIGN,
 133                 [EXPR_BINARY_MUL_ASSIGN]         = PREC_ASSIGN,
 134                 [EXPR_BINARY_DIV_ASSIGN]         = PREC_ASSIGN,
 135                 [EXPR_BINARY_MOD_ASSIGN]         = PREC_ASSIGN,
 136                 [EXPR_BINARY_ADD_ASSIGN]         = PREC_ASSIGN,
 137                 [EXPR_BINARY_SUB_ASSIGN]         = PREC_ASSIGN,
 138                 [EXPR_BINARY_SHIFTLEFT_ASSIGN]   = PREC_ASSIGN,
 139                 [EXPR_BINARY_SHIFTRIGHT_ASSIGN]  = PREC_ASSIGN,
 140                 [EXPR_BINARY_BITWISE_AND_ASSIGN] = PREC_ASSIGN,
 141                 [EXPR_BINARY_BITWISE_XOR_ASSIGN] = PREC_ASSIGN,
 142                 [EXPR_BINARY_BITWISE_OR_ASSIGN]  = PREC_ASSIGN,
 143                 [EXPR_BINARY_COMMA]              = PREC_COMMA,
 144
 145                 [EXPR_BINARY_BUILTIN_EXPECT]     = PREC_PRIM,
 146                 [EXPR_BINARY_ISGREATER]          = PREC_PRIM,
 147                 [EXPR_BINARY_ISGREATEREQUAL]     = PREC_PRIM,
 148                 [EXPR_BINARY_ISLESS]             = PREC_PRIM,
 149                 [EXPR_BINARY_ISLESSEQUAL]        = PREC_PRIM,
 150                 [EXPR_BINARY_ISLESSGREATER]      = PREC_PRIM,
 151                 [EXPR_BINARY_ISUNORDERED]        = PREC_PRIM
 152         };
 153 #ifndef NDEBUG
 154         if ((unsigned)kind >= (sizeof(prec)/sizeof(prec[0]))) {
 155                 panic("wrong expression kind");
 156         }
 157         unsigned res = prec[kind];
 158         if (res == PREC_BOTTOM) {
 159                 panic("expression kind not defined in get_expression_precedence()");
 160         }
 161 #endif
 162         return res;
 163 }
 164
 165 /**
 166  * Print a constant expression.
 167  */
 168 static void print_const(const const_expression_t *cnst)
 169 {
 170         if(cnst->base.type == NULL)
 171                 return;
 172
 173         if(is_type_integer(cnst->base.type)) {
 174                 fprintf(out, "%lld", cnst->v.int_value);
 175         } else if(is_type_float(cnst->base.type)) {
 176                 fprintf(out, "%Lf", cnst->v.float_value);
 177         }
 178 }
 179
 180 /**
 181  * Print a quoted string constant.
 182  */
 183 static void print_quoted_string(const string_t *const string)
 184 {
 185         fputc('"', out);
 186         const char *end = string->begin + string->size;
 187         for (const char *c = string->begin; c != end; ++c) {
 188                 switch(*c) {
 189                 case '\"':  fputs("\\\"", out); break;
 190                 case '\\':  fputs("\\\\", out); break;
 191                 case '\a':  fputs("\\a", out); break;
 192                 case '\b':  fputs("\\b", out); break;
 193                 case '\f':  fputs("\\f", out); break;
 194                 case '\n':  fputs("\\n", out); break;
 195                 case '\r':  fputs("\\r", out); break;
 196                 case '\t':  fputs("\\t", out); break;
 197                 case '\v':  fputs("\\v", out); break;
 198                 case '\?':  fputs("\\?", out); break;
 199                 default:
 200                         if(!isprint(*c)) {
 201                                 fprintf(out, "\\%03o", *c);
 202                                 break;
 203                         }
 204                         fputc(*c, out);
 205                         break;
 206                 }
 207         }
 208         fputc('"', out);
 209 }
 210
 211 /**
 212  * Prints a string literal expression.
 213  */
 214 static void print_string_literal(
 215                 const string_literal_expression_t *string_literal)
 216 {
 217         print_quoted_string(&string_literal->value);
 218 }
 219
 220 /**
 221  * Prints a wide string literal expression.
 222  */
 223 static void print_wide_string_literal(
 224         const wide_string_literal_expression_t *const wstr)
 225 {
 226         fputs("L\"", out);
 227         for (const wchar_rep_t *c   = wstr->value.begin,
 228                                *end = c + wstr->value.size;
 229              c != end; ++c) {
 230                 switch (*c) {
 231                         case L'\"':  fputs("\\\"", out); break;
 232                         case L'\\':  fputs("\\\\", out); break;
 233                         case L'\a':  fputs("\\a",  out); break;
 234                         case L'\b':  fputs("\\b",  out); break;
 235                         case L'\f':  fputs("\\f",  out); break;
 236                         case L'\n':  fputs("\\n",  out); break;
 237                         case L'\r':  fputs("\\r",  out); break;
 238                         case L'\t':  fputs("\\t",  out); break;
 239                         case L'\v':  fputs("\\v",  out); break;
 240                         case L'\?':  fputs("\\?",  out); break;
 241                         default: {
 242                                 const unsigned tc = *c;
 243                                 if (tc < 0x80U) {
 244                                         if (!isprint(*c))  {
 245                                                 fprintf(out, "\\%03o", (char)*c);
 246                                         } else {
 247                                                 fputc(*c, out);
 248                                         }
 249                                 } else if (tc < 0x800) {
 250                                         fputc(0xC0 | (tc >> 6),   out);
 251                                         fputc(0x80 | (tc & 0x3F), out);
 252                                 } else if (tc < 0x10000) {
 253                                         fputc(0xE0 | ( tc >> 12),         out);
 254                                         fputc(0x80 | ((tc >>  6) & 0x3F), out);
 255                                         fputc(0x80 | ( tc        & 0x3F), out);
 256                                 } else {
 257                                         fputc(0xF0 | ( tc >> 18),         out);
 258                                         fputc(0x80 | ((tc >> 12) & 0x3F), out);
 259                                         fputc(0x80 | ((tc >>  6) & 0x3F), out);
 260                                         fputc(0x80 | ( tc        & 0x3F), out);
 261                                 }
 262                         }
 263                 }
 264         }
 265         fputc('"', out);
 266 }
 267
 268 /**
 269  * Prints a call expression.
 270  */
 271 static void print_call_expression(const call_expression_t *call)
 272 {
 273         unsigned prec = get_expression_precedence(call->base.kind);
 274         print_expression_prec(call->function, prec);
 275         fprintf(out, "(");
 276         call_argument_t *argument = call->arguments;
 277         int              first    = 1;
 278         while(argument != NULL) {
 279                 if(!first) {
 280                         fprintf(out, ", ");
 281                 } else {
 282                         first = 0;
 283                 }
 284                 print_expression_prec(argument->expression, PREC_COMMA + 1);
 285
 286                 argument = argument->next;
 287         }
 288         fprintf(out, ")");
 289 }
 290
 291 /**
 292  * Prints a binary expression.
 293  */
 294 static void print_binary_expression(const binary_expression_t *binexpr)
 295 {
 296         unsigned prec = get_expression_precedence(binexpr->base.kind);
 297         int      r2l  = right_to_left(prec);
 298
 299         if(binexpr->base.kind == EXPR_BINARY_BUILTIN_EXPECT) {
 300                 fputs("__builtin_expect(", out);
 301                 print_expression_prec(binexpr->left, prec);
 302                 fputs(", ", out);
 303                 print_expression_prec(binexpr->right, prec);
 304                 fputc(')', out);
 305                 return;
 306         }
 307
 308         print_expression_prec(binexpr->left, prec + r2l);
 309         if (binexpr->base.kind != EXPR_BINARY_COMMA) {
 310                 fputc(' ', out);
 311         }
 312         switch (binexpr->base.kind) {
 313         case EXPR_BINARY_COMMA:              fputs(",", out);     break;
 314         case EXPR_BINARY_ASSIGN:             fputs("=", out);     break;
 315         case EXPR_BINARY_ADD:                fputs("+", out);     break;
 316         case EXPR_BINARY_SUB:                fputs("-", out);     break;
 317         case EXPR_BINARY_MUL:                fputs("*", out);     break;
 318         case EXPR_BINARY_MOD:                fputs("%", out);     break;
 319         case EXPR_BINARY_DIV:                fputs("/", out);     break;
 320         case EXPR_BINARY_BITWISE_OR:         fputs("|", out);     break;
 321         case EXPR_BINARY_BITWISE_AND:        fputs("&", out);     break;
 322         case EXPR_BINARY_BITWISE_XOR:        fputs("^", out);     break;
 323         case EXPR_BINARY_LOGICAL_OR:         fputs("||", out);    break;
 324         case EXPR_BINARY_LOGICAL_AND:        fputs("&&", out);    break;
 325         case EXPR_BINARY_NOTEQUAL:           fputs("!=", out);    break;
 326         case EXPR_BINARY_EQUAL:              fputs("==", out);    break;
 327         case EXPR_BINARY_LESS:               fputs("<", out);     break;
 328         case EXPR_BINARY_LESSEQUAL:          fputs("<=", out);    break;
 329         case EXPR_BINARY_GREATER:            fputs(">", out);     break;
 330         case EXPR_BINARY_GREATEREQUAL:       fputs(">=", out);    break;
 331         case EXPR_BINARY_SHIFTLEFT:          fputs("<<", out);    break;
 332         case EXPR_BINARY_SHIFTRIGHT:         fputs(">>", out);    break;
 333
 334         case EXPR_BINARY_ADD_ASSIGN:         fputs("+=", out);    break;
 335         case EXPR_BINARY_SUB_ASSIGN:         fputs("-=", out);    break;
 336         case EXPR_BINARY_MUL_ASSIGN:         fputs("*=", out);    break;
 337         case EXPR_BINARY_MOD_ASSIGN:         fputs("%=", out);    break;
 338         case EXPR_BINARY_DIV_ASSIGN:         fputs("/=", out);    break;
 339         case EXPR_BINARY_BITWISE_OR_ASSIGN:  fputs("|=", out);    break;
 340         case EXPR_BINARY_BITWISE_AND_ASSIGN: fputs("&=", out);    break;
 341         case EXPR_BINARY_BITWISE_XOR_ASSIGN: fputs("^=", out);    break;
 342         case EXPR_BINARY_SHIFTLEFT_ASSIGN:   fputs("<<=", out);   break;
 343         case EXPR_BINARY_SHIFTRIGHT_ASSIGN:  fputs(">>=", out);   break;
 344         default: panic("invalid binexpression found");
 345         }
 346         fputc(' ', out);
 347         print_expression_prec(binexpr->right, prec - r2l);
 348 }
 349
 350 /**
 351  * Prints an unary expression.
 352  */
 353 static void print_unary_expression(const unary_expression_t *unexpr)
 354 {
 355         unsigned prec = get_expression_precedence(unexpr->base.kind);
 356         switch(unexpr->base.kind) {
 357         case EXPR_UNARY_NEGATE:           fputs("-", out);  break;
 358         case EXPR_UNARY_PLUS:             fputs("+", out);  break;
 359         case EXPR_UNARY_NOT:              fputs("!", out);  break;
 360         case EXPR_UNARY_BITWISE_NEGATE:   fputs("~", out);  break;
 361         case EXPR_UNARY_PREFIX_INCREMENT: fputs("++", out); break;
 362         case EXPR_UNARY_PREFIX_DECREMENT: fputs("--", out); break;
 363         case EXPR_UNARY_DEREFERENCE:      fputs("*", out);  break;
 364         case EXPR_UNARY_TAKE_ADDRESS:     fputs("&", out);  break;
 365
 366         case EXPR_UNARY_BITFIELD_EXTRACT:
 367                 print_expression_prec(unexpr->value, prec);
 368                 return;
 369
 370         case EXPR_UNARY_POSTFIX_INCREMENT:
 371                 print_expression_prec(unexpr->value, prec);
 372                 fputs("++", out);
 373                 return;
 374         case EXPR_UNARY_POSTFIX_DECREMENT:
 375                 print_expression_prec(unexpr->value, prec);
 376                 fputs("--", out);
 377                 return;
 378         case EXPR_UNARY_CAST_IMPLICIT:
 379                 if(!print_implicit_casts) {
 380                         print_expression_prec(unexpr->value, prec);
 381                         return;
 382                 }
 383                 /* fallthrough */
 384         case EXPR_UNARY_CAST:
 385                 fputc('(', out);
 386                 print_type(unexpr->base.type);
 387                 fputc(')', out);
 388                 break;
 389         case EXPR_UNARY_ASSUME:
 390                 fputs("__assume(", out);
 391                 print_expression_prec(unexpr->value, PREC_COMMA + 1);
 392                 fputc(')', out);
 393                 return;
 394         default:
 395                 panic("invalid unary expression found");
 396         }
 397         print_expression_prec(unexpr->value, prec);
 398 }
 399
 400 /**
 401  * Prints a reference expression.
 402  */
 403 static void print_reference_expression(const reference_expression_t *ref)
 404 {
 405         fprintf(out, "%s", ref->declaration->symbol->string);
 406 }
 407
 408 /**
 409  * Prints an array expression.
 410  */
 411 static void print_array_expression(const array_access_expression_t *expression)
 412 {
 413         unsigned prec = get_expression_precedence(expression->base.kind);
 414         if(!expression->flipped) {
 415                 print_expression_prec(expression->array_ref, prec);
 416                 fputc('[', out);
 417                 print_expression_prec(expression->index, prec);
 418                 fputc(']', out);
 419         } else {
 420                 print_expression_prec(expression->index, prec);
 421                 fputc('[', out);
 422                 print_expression_prec(expression->array_ref, prec);
 423                 fputc(']', out);
 424         }
 425 }
 426
 427 /**
 428  * Prints a typeproperty expression.
 429  */
 430 static void print_typeprop_expression(const typeprop_expression_t *expression)
 431 {
 432         if (expression->base.kind == EXPR_SIZEOF) {
 433                 fputs("sizeof", out);
 434         } else {
 435                 assert(expression->base.kind == EXPR_ALIGNOF);
 436                 fputs("__alignof__", out);
 437         }
 438         if(expression->tp_expression != NULL) {
 439                 /* always print the '()' here, sizeof x is right but unusual */
 440                 fputc('(', out);
 441                 print_expression_prec(expression->tp_expression, PREC_ACCESS);
 442                 fputc(')', out);
 443         } else {
 444                 fputc('(', out);
 445                 print_type(expression->type);
 446                 fputc(')', out);
 447         }
 448 }
 449
 450 /**
 451  * Prints an builtin symbol
 452  */
 453 static void print_builtin_symbol(const builtin_symbol_expression_t *expression)
 454 {
 455         fputs(expression->symbol->string, out);
 456 }
 457
 458 /**
 459  * Prints a builtin constant expression.
 460  */
 461 static void print_builtin_constant(const builtin_constant_expression_t *expression)
 462 {
 463         fputs("__builtin_constant_p(", out);
 464         print_expression_prec(expression->value, PREC_COMMA + 1);
 465         fputc(')', out);
 466 }
 467
 468 /**
 469  * Prints a builtin prefetch expression.
 470  */
 471 static void print_builtin_prefetch(const builtin_prefetch_expression_t *expression)
 472 {
 473         fputs("__builtin_prefetch(", out);
 474         print_expression_prec(expression->adr, PREC_COMMA + 1);
 475         if (expression->rw) {
 476                 fputc(',', out);
 477                 print_expression_prec(expression->rw, PREC_COMMA + 1);
 478         }
 479         if (expression->locality) {
 480                 fputc(',', out);
 481                 print_expression_prec(expression->locality, PREC_COMMA + 1);
 482         }
 483         fputc(')', out);
 484 }
 485
 486 /**
 487  * Prints a conditional expression.
 488  */
 489 static void print_conditional(const conditional_expression_t *expression)
 490 {
 491         unsigned prec = get_expression_precedence(expression->base.kind);
 492         fputs("(", out);
 493         print_expression_prec(expression->condition, prec);
 494         fputs(" ? ", out);
 495         print_expression_prec(expression->true_expression, prec);
 496         fputs(" : ", out);
 497         print_expression_prec(expression->false_expression, prec);
 498         fputs(")", out);
 499 }
 500
 501 /**
 502  * Prints a va_start expression.
 503  */
 504 static void print_va_start(const va_start_expression_t *const expression)
 505 {
 506         fputs("__builtin_va_start(", out);
 507         print_expression_prec(expression->ap, PREC_COMMA + 1);
 508         fputs(", ", out);
 509         fputs(expression->parameter->symbol->string, out);
 510         fputs(")", out);
 511 }
 512
 513 /**
 514  * Prints a va_arg expression.
 515  */
 516 static void print_va_arg(const va_arg_expression_t *expression)
 517 {
 518         fputs("__builtin_va_arg(", out);
 519         print_expression_prec(expression->ap, PREC_COMMA + 1);
 520         fputs(", ", out);
 521         print_type(expression->base.type);
 522         fputs(")", out);
 523 }
 524
 525 /**
 526  * Prints a select expression.
 527  */
 528 static void print_select(const select_expression_t *expression)
 529 {
 530         unsigned prec = get_expression_precedence(expression->base.kind);
 531         print_expression_prec(expression->compound, prec);
 532         if(expression->compound->base.type == NULL ||
 533                         expression->compound->base.type->kind == TYPE_POINTER) {
 534                 fputs("->", out);
 535         } else {
 536                 fputc('.', out);
 537         }
 538         fputs(expression->symbol->string, out);
 539 }
 540
 541 /**
 542  * Prints a type classify expression.
 543  */
 544 static void print_classify_type_expression(
 545         const classify_type_expression_t *const expr)
 546 {
 547         fputs("__builtin_classify_type(", out);
 548         print_expression_prec(expr->type_expression, PREC_COMMA + 1);
 549         fputc(')', out);
 550 }
 551
 552 static void print_designator(const designator_t *designator)
 553 {
 554         fputs(designator->symbol->string, out);
 555         for (designator = designator->next; designator != NULL; designator = designator->next) {
 556                 if (designator->array_access) {
 557                         fputc('[', out);
 558                         print_expression_prec(designator->array_access, PREC_ACCESS);
 559                         fputc(']', out);
 560                 } else {
 561                         fputc('.', out);
 562                         fputs(designator->symbol->string, out);
 563                 }
 564         }
 565 }
 566
 567 /**
 568  * Prints an offsetof classify expression.
 569  */
 570 static void print_offsetof_expression(const offsetof_expression_t *expression)
 571 {
 572         fputs("__builtin_offsetof", out);
 573         fputc('(', out);
 574         print_type(expression->type);
 575         fputc(',', out);
 576         print_designator(expression->designator);
 577         fputc(')', out);
 578 }
 579
 580 /**
 581  * Prints a statement expression.
 582  */
 583 static void print_statement_expression(const statement_expression_t *expression)
 584 {
 585         fputc('(', out);
 586         print_statement(expression->statement);
 587         fputc(')', out);
 588 }
 589
 590 static void print_expression_prec(const expression_t *expression, unsigned top_prec)
 591 {
 592         unsigned prec = get_expression_precedence(expression->base.kind);
 593         if (print_parenthesis && top_prec != PREC_BOTTOM)
 594                 top_prec = PREC_TOP;
 595         if (top_prec > prec)
 596                 fputc('(', out);
 597         switch(expression->kind) {
 598         case EXPR_UNKNOWN:
 599         case EXPR_INVALID:
 600                 fprintf(out, "*invalid expression*");
 601                 break;
 602         case EXPR_CONST:
 603                 print_const(&expression->conste);
 604                 break;
 605         case EXPR_FUNCTION:
 606         case EXPR_PRETTY_FUNCTION:
 607         case EXPR_STRING_LITERAL:
 608                 print_string_literal(&expression->string);
 609                 break;
 610         case EXPR_WIDE_STRING_LITERAL:
 611                 print_wide_string_literal(&expression->wide_string);
 612                 break;
 613         case EXPR_CALL:
 614                 print_call_expression(&expression->call);
 615                 break;
 616         EXPR_BINARY_CASES
 617                 print_binary_expression(&expression->binary);
 618                 break;
 619         case EXPR_REFERENCE:
 620                 print_reference_expression(&expression->reference);
 621                 break;
 622         case EXPR_ARRAY_ACCESS:
 623                 print_array_expression(&expression->array_access);
 624                 break;
 625         EXPR_UNARY_CASES
 626                 print_unary_expression(&expression->unary);
 627                 break;
 628         case EXPR_SIZEOF:
 629         case EXPR_ALIGNOF:
 630                 print_typeprop_expression(&expression->typeprop);
 631                 break;
 632         case EXPR_BUILTIN_SYMBOL:
 633                 print_builtin_symbol(&expression->builtin_symbol);
 634                 break;
 635         case EXPR_BUILTIN_CONSTANT_P:
 636                 print_builtin_constant(&expression->builtin_constant);
 637                 break;
 638         case EXPR_BUILTIN_PREFETCH:
 639                 print_builtin_prefetch(&expression->builtin_prefetch);
 640                 break;
 641         case EXPR_CONDITIONAL:
 642                 print_conditional(&expression->conditional);
 643                 break;
 644         case EXPR_VA_START:
 645                 print_va_start(&expression->va_starte);
 646                 break;
 647         case EXPR_VA_ARG:
 648                 print_va_arg(&expression->va_arge);
 649                 break;
 650         case EXPR_SELECT:
 651                 print_select(&expression->select);
 652                 break;
 653         case EXPR_CLASSIFY_TYPE:
 654                 print_classify_type_expression(&expression->classify_type);
 655                 break;
 656         case EXPR_OFFSETOF:
 657                 print_offsetof_expression(&expression->offsetofe);
 658                 break;
 659         case EXPR_STATEMENT:
 660                 print_statement_expression(&expression->statement);
 661                 break;
 662
 663         default:
 664                 /* TODO */
 665                 fprintf(out, "some expression of type %d", (int) expression->kind);
 666                 break;
 667         }
 668         if (top_prec > prec)
 669                 fputc(')', out);
 670 }
 671
 672 static void print_compound_statement(const compound_statement_t *block)
 673 {
 674         fputs("{\n", out);
 675         ++indent;
 676
 677         statement_t *statement = block->statements;
 678         while(statement != NULL) {
 679                 if (statement->base.kind == STATEMENT_CASE_LABEL)
 680                         --indent;
 681                 print_indent();
 682                 print_statement(statement);
 683
 684                 statement = statement->base.next;
 685         }
 686         --indent;
 687         print_indent();
 688         fputs("}\n", out);
 689 }
 690
 691 static void print_return_statement(const return_statement_t *statement)
 692 {
 693         fprintf(out, "return ");
 694         if(statement->value != NULL)
 695                 print_expression_prec(statement->value, PREC_BOTTOM);
 696         fputs(";\n", out);
 697 }
 698
 699 static void print_expression_statement(const expression_statement_t *statement)
 700 {
 701         print_expression(statement->expression);
 702         fputs(";\n", out);
 703 }
 704
 705 static void print_goto_statement(const goto_statement_t *statement)
 706 {
 707         fprintf(out, "goto ");
 708         fputs(statement->label->symbol->string, out);
 709         fprintf(stderr, "(%p)", (void*) statement->label);
 710         fputs(";\n", out);
 711 }
 712
 713 static void print_label_statement(const label_statement_t *statement)
 714 {
 715         fprintf(stderr, "(%p)", (void*) statement->label);
 716         fprintf(out, "%s:\n", statement->label->symbol->string);
 717         if(statement->statement != NULL) {
 718                 print_statement(statement->statement);
 719         }
 720 }
 721
 722 static void print_if_statement(const if_statement_t *statement)
 723 {
 724         fputs("if(", out);
 725         print_expression(statement->condition);
 726         fputs(") ", out);
 727         if(statement->true_statement != NULL) {
 728                 print_statement(statement->true_statement);
 729         }
 730
 731         if(statement->false_statement != NULL) {
 732                 print_indent();
 733                 fputs("else ", out);
 734                 print_statement(statement->false_statement);
 735         }
 736 }
 737
 738 static void print_switch_statement(const switch_statement_t *statement)
 739 {
 740         fputs("switch(", out);
 741         print_expression(statement->expression);
 742         fputs(") ", out);
 743         print_statement(statement->body);
 744 }
 745
 746 static void print_case_label(const case_label_statement_t *statement)
 747 {
 748         if(statement->expression == NULL) {
 749                 fputs("default:\n", out);
 750         } else {
 751                 fputs("case ", out);
 752                 print_expression(statement->expression);
 753                 fputs(":\n", out);
 754         }
 755         ++indent;
 756         if(statement->statement != NULL) {
 757                 if (statement->statement->base.kind == STATEMENT_CASE_LABEL) {
 758                         --indent;
 759                 }
 760                 print_indent();
 761                 print_statement(statement->statement);
 762         }
 763 }
 764
 765 static void print_declaration_statement(
 766                 const declaration_statement_t *statement)
 767 {
 768         int first = 1;
 769         declaration_t *declaration = statement->declarations_begin;
 770         for( ; declaration != statement->declarations_end->next;
 771                declaration = declaration->next) {
 772                 if(!first) {
 773                         print_indent();
 774                 } else {
 775                         first = 0;
 776                 }
 777                 print_declaration(declaration);
 778                 fputc('\n', out);
 779         }
 780 }
 781
 782 static void print_while_statement(const while_statement_t *statement)
 783 {
 784         fputs("while(", out);
 785         print_expression(statement->condition);
 786         fputs(") ", out);
 787         print_statement(statement->body);
 788 }
 789
 790 static void print_do_while_statement(const do_while_statement_t *statement)
 791 {
 792         fputs("do ", out);
 793         print_statement(statement->body);
 794         print_indent();
 795         fputs("while(", out);
 796         print_expression(statement->condition);
 797         fputs(");\n", out);
 798 }
 799
 800 static void print_for_statement(const for_statement_t *statement)
 801 {
 802         fputs("for(", out);
 803         if(statement->scope.declarations != NULL) {
 804                 assert(statement->initialisation == NULL);
 805                 print_declaration(statement->scope.declarations);
 806                 if(statement->scope.declarations->next != NULL) {
 807                         panic("multiple declarations in for statement not supported yet");
 808                 }
 809                 fputc(' ', out);
 810         } else {
 811                 if(statement->initialisation) {
 812                         print_expression(statement->initialisation);
 813                 }
 814                 fputs("; ", out);
 815         }
 816         if(statement->condition != NULL) {
 817                 print_expression(statement->condition);
 818         }
 819         fputs("; ", out);
 820         if(statement->step != NULL) {
 821                 print_expression(statement->step);
 822         }
 823         fputs(")", out);
 824         print_statement(statement->body);
 825 }
 826
 827 static void print_asm_constraints(asm_constraint_t *constraints)
 828 {
 829         asm_constraint_t *constraint = constraints;
 830         for( ; constraint != NULL; constraint = constraint->next) {
 831                 if(constraint != constraints)
 832                         fputs(", ", out);
 833
 834                 if(constraint->symbol) {
 835                         fprintf(out, "[%s] ", constraint->symbol->string);
 836                 }
 837                 print_quoted_string(&constraint->constraints);
 838                 fputs(" (", out);
 839                 print_expression(constraint->expression);
 840                 fputs(")", out);
 841         }
 842 }
 843
 844 static void print_asm_clobbers(asm_clobber_t *clobbers)
 845 {
 846         asm_clobber_t *clobber = clobbers;
 847         for( ; clobber != NULL; clobber = clobber->next) {
 848                 if(clobber != clobbers)
 849                         fputs(", ", out);
 850
 851                 print_quoted_string(&clobber->clobber);
 852         }
 853 }
 854
 855 static void print_asm_statement(const asm_statement_t *statement)
 856 {
 857         fputs("asm ", out);
 858         if(statement->is_volatile) {
 859                 fputs("volatile ", out);
 860         }
 861         fputs("(", out);
 862         print_quoted_string(&statement->asm_text);
 863         if(statement->inputs == NULL && statement->outputs == NULL
 864                         && statement->clobbers == NULL)
 865                 goto end_of_print_asm_statement;
 866
 867         fputs(" : ", out);
 868         print_asm_constraints(statement->inputs);
 869         if(statement->outputs == NULL && statement->clobbers == NULL)
 870                 goto end_of_print_asm_statement;
 871
 872         fputs(" : ", out);
 873         print_asm_constraints(statement->outputs);
 874         if(statement->clobbers == NULL)
 875                 goto end_of_print_asm_statement;
 876
 877         fputs(" : ", out);
 878         print_asm_clobbers(statement->clobbers);
 879
 880 end_of_print_asm_statement:
 881         fputs(");\n", out);
 882 }
 883
 884 void print_statement(const statement_t *statement)
 885 {
 886         switch(statement->kind) {
 887         case STATEMENT_COMPOUND:
 888                 print_compound_statement(&statement->compound);
 889                 break;
 890         case STATEMENT_RETURN:
 891                 print_return_statement(&statement->returns);
 892                 break;
 893         case STATEMENT_EXPRESSION:
 894                 print_expression_statement(&statement->expression);
 895                 break;
 896         case STATEMENT_LABEL:
 897                 print_label_statement(&statement->label);
 898                 break;
 899         case STATEMENT_GOTO:
 900                 print_goto_statement(&statement->gotos);
 901                 break;
 902         case STATEMENT_CONTINUE:
 903                 fputs("continue;\n", out);
 904                 break;
 905         case STATEMENT_BREAK:
 906                 fputs("break;\n", out);
 907                 break;
 908         case STATEMENT_IF:
 909                 print_if_statement(&statement->ifs);
 910                 break;
 911         case STATEMENT_SWITCH:
 912                 print_switch_statement(&statement->switchs);
 913                 break;
 914         case STATEMENT_CASE_LABEL:
 915                 print_case_label(&statement->case_label);
 916                 break;
 917         case STATEMENT_DECLARATION:
 918                 print_declaration_statement(&statement->declaration);
 919                 break;
 920         case STATEMENT_WHILE:
 921                 print_while_statement(&statement->whiles);
 922                 break;
 923         case STATEMENT_DO_WHILE:
 924                 print_do_while_statement(&statement->do_while);
 925                 break;
 926         case STATEMENT_FOR:
 927                 print_for_statement(&statement->fors);
 928                 break;
 929         case STATEMENT_ASM:
 930                 print_asm_statement(&statement->asms);
 931                 break;
 932         case STATEMENT_INVALID:
 933                 fprintf(out, "*invalid statement*");
 934                 break;
 935         }
 936 }
 937
 938 static void print_storage_class(unsigned storage_class)
 939 {
 940         switch((storage_class_tag_t) storage_class) {
 941         case STORAGE_CLASS_ENUM_ENTRY:
 942         case STORAGE_CLASS_NONE:
 943                 break;
 944         case STORAGE_CLASS_TYPEDEF:       fputs("typedef ",        out); break;
 945         case STORAGE_CLASS_EXTERN:        fputs("extern ",         out); break;
 946         case STORAGE_CLASS_STATIC:        fputs("static ",         out); break;
 947         case STORAGE_CLASS_AUTO:          fputs("auto ",           out); break;
 948         case STORAGE_CLASS_REGISTER:      fputs("register ",       out); break;
 949         case STORAGE_CLASS_THREAD:        fputs("__thread",        out); break;
 950         case STORAGE_CLASS_THREAD_EXTERN: fputs("extern __thread", out); break;
 951         case STORAGE_CLASS_THREAD_STATIC: fputs("static __thread", out); break;
 952         }
 953 }
 954
 955 void print_initializer(const initializer_t *initializer)
 956 {
 957         if(initializer->kind == INITIALIZER_VALUE) {
 958                 const initializer_value_t *value = &initializer->value;
 959                 print_expression(value->value);
 960                 return;
 961         }
 962
 963         assert(initializer->kind == INITIALIZER_LIST);
 964         fputs("{ ", out);
 965         const initializer_list_t *list = &initializer->list;
 966
 967         for(size_t i = 0 ; i < list->len; ++i) {
 968                 if(i > 0) {
 969                         fputs(", ", out);
 970                 }
 971                 print_initializer(list->initializers[i]);
 972         }
 973         fputs("}", out);
 974 }
 975
 976 static void print_normal_declaration(const declaration_t *declaration)
 977 {
 978         print_storage_class(declaration->storage_class);
 979         if(declaration->is_inline) {
 980                 if (declaration->modifiers & DM_FORCEINLINE)
 981                         fputs("__forceinline ", out);
 982                 else
 983                         fputs("inline ", out);
 984         }
 985         print_type_ext(declaration->type, declaration->symbol,
 986                        &declaration->scope);
 987
 988         if(declaration->type->kind == TYPE_FUNCTION) {
 989                 if(declaration->init.statement != NULL) {
 990                         fputs("\n", out);
 991                         print_statement(declaration->init.statement);
 992                         return;
 993                 }
 994         } else if(declaration->init.initializer != NULL) {
 995                 fputs(" = ", out);
 996                 print_initializer(declaration->init.initializer);
 997         }
 998         fputc(';', out);
 999 }
1000
1001 /**
1002  * Prints an expression.
1003  */
1004 void print_expression(const expression_t *expression) {
1005         print_expression_prec(expression, PREC_BOTTOM);
1006 }
1007
1008 void print_declaration(const declaration_t *declaration)
1009 {
1010         if(declaration->namespc != NAMESPACE_NORMAL &&
1011                         declaration->symbol == NULL)
1012                 return;
1013
1014         switch(declaration->namespc) {
1015         case NAMESPACE_NORMAL:
1016                 print_normal_declaration(declaration);
1017                 break;
1018         case NAMESPACE_STRUCT:
1019                 fputs("struct ", out);
1020                 fputs(declaration->symbol->string, out);
1021                 fputc(' ', out);
1022                 print_compound_definition(declaration);
1023                 fputc(';', out);
1024                 break;
1025         case NAMESPACE_UNION:
1026                 fputs("union ", out);
1027                 fputs(declaration->symbol->string, out);
1028                 fputc(' ', out);
1029                 print_compound_definition(declaration);
1030                 fputc(';', out);
1031                 break;
1032         case NAMESPACE_ENUM:
1033                 fputs("enum ", out);
1034                 fputs(declaration->symbol->string, out);
1035                 fputc(' ', out);
1036                 print_enum_definition(declaration);
1037                 fputc(';', out);
1038                 break;
1039         }
1040 }
1041
1042 void print_ast(const translation_unit_t *unit)
1043 {
1044         inc_type_visited();
1045
1046         declaration_t *declaration = unit->scope.declarations;
1047         for( ; declaration != NULL; declaration = declaration->next) {
1048                 if(declaration->storage_class == STORAGE_CLASS_ENUM_ENTRY)
1049                         continue;
1050                 if(declaration->namespc != NAMESPACE_NORMAL &&
1051                                 declaration->symbol == NULL)
1052                         continue;
1053
1054                 print_indent();
1055                 print_declaration(declaration);
1056                 fputc('\n', out);
1057         }
1058 }
1059
1060 bool is_constant_expression(const expression_t *expression)
1061 {
1062         switch(expression->kind) {
1063
1064         case EXPR_CONST:
1065         case EXPR_STRING_LITERAL:
1066         case EXPR_WIDE_STRING_LITERAL:
1067         case EXPR_SIZEOF:
1068         case EXPR_CLASSIFY_TYPE:
1069         case EXPR_FUNCTION:
1070         case EXPR_PRETTY_FUNCTION:
1071         case EXPR_OFFSETOF:
1072         case EXPR_ALIGNOF:
1073         case EXPR_BUILTIN_CONSTANT_P:
1074                 return true;
1075
1076         case EXPR_BUILTIN_SYMBOL:
1077         case EXPR_BUILTIN_PREFETCH:
1078         case EXPR_CALL:
1079         case EXPR_SELECT:
1080         case EXPR_VA_START:
1081         case EXPR_VA_ARG:
1082         case EXPR_STATEMENT:
1083         case EXPR_UNARY_POSTFIX_INCREMENT:
1084         case EXPR_UNARY_POSTFIX_DECREMENT:
1085         case EXPR_UNARY_PREFIX_INCREMENT:
1086         case EXPR_UNARY_PREFIX_DECREMENT:
1087         case EXPR_UNARY_BITFIELD_EXTRACT:
1088         case EXPR_UNARY_ASSUME: /* has VOID type */
1089         case EXPR_BINARY_ASSIGN:
1090         case EXPR_BINARY_MUL_ASSIGN:
1091         case EXPR_BINARY_DIV_ASSIGN:
1092         case EXPR_BINARY_MOD_ASSIGN:
1093         case EXPR_BINARY_ADD_ASSIGN:
1094         case EXPR_BINARY_SUB_ASSIGN:
1095         case EXPR_BINARY_SHIFTLEFT_ASSIGN:
1096         case EXPR_BINARY_SHIFTRIGHT_ASSIGN:
1097         case EXPR_BINARY_BITWISE_AND_ASSIGN:
1098         case EXPR_BINARY_BITWISE_XOR_ASSIGN:
1099         case EXPR_BINARY_BITWISE_OR_ASSIGN:
1100         case EXPR_BINARY_COMMA:
1101                 return false;
1102
1103         case EXPR_UNARY_NEGATE:
1104         case EXPR_UNARY_PLUS:
1105         case EXPR_UNARY_BITWISE_NEGATE:
1106         case EXPR_UNARY_NOT:
1107         case EXPR_UNARY_DEREFERENCE:
1108         case EXPR_UNARY_TAKE_ADDRESS:
1109         case EXPR_UNARY_CAST:
1110         case EXPR_UNARY_CAST_IMPLICIT:
1111                 return is_constant_expression(expression->unary.value);
1112
1113         case EXPR_BINARY_ADD:
1114         case EXPR_BINARY_SUB:
1115         case EXPR_BINARY_MUL:
1116         case EXPR_BINARY_DIV:
1117         case EXPR_BINARY_MOD:
1118         case EXPR_BINARY_EQUAL:
1119         case EXPR_BINARY_NOTEQUAL:
1120         case EXPR_BINARY_LESS:
1121         case EXPR_BINARY_LESSEQUAL:
1122         case EXPR_BINARY_GREATER:
1123         case EXPR_BINARY_GREATEREQUAL:
1124         case EXPR_BINARY_BITWISE_AND:
1125         case EXPR_BINARY_BITWISE_OR:
1126         case EXPR_BINARY_BITWISE_XOR:
1127         case EXPR_BINARY_LOGICAL_AND:
1128         case EXPR_BINARY_LOGICAL_OR:
1129         case EXPR_BINARY_SHIFTLEFT:
1130         case EXPR_BINARY_SHIFTRIGHT:
1131         case EXPR_BINARY_BUILTIN_EXPECT:
1132         case EXPR_BINARY_ISGREATER:
1133         case EXPR_BINARY_ISGREATEREQUAL:
1134         case EXPR_BINARY_ISLESS:
1135         case EXPR_BINARY_ISLESSEQUAL:
1136         case EXPR_BINARY_ISLESSGREATER:
1137         case EXPR_BINARY_ISUNORDERED:
1138                 return is_constant_expression(expression->binary.left)
1139                         && is_constant_expression(expression->binary.right);
1140
1141         case EXPR_CONDITIONAL:
1142                 /* TODO: not correct, we only have to test expressions which are
1143                  * evaluated, which means either the true or false part might be not
1144                  * constant */
1145                 return is_constant_expression(expression->conditional.condition)
1146                         && is_constant_expression(expression->conditional.true_expression)
1147                         && is_constant_expression(expression->conditional.false_expression);
1148
1149         case EXPR_ARRAY_ACCESS:
1150                 return is_constant_expression(expression->array_access.array_ref)
1151                         && is_constant_expression(expression->array_access.index);
1152
1153         case EXPR_REFERENCE: {
1154                 declaration_t *declaration = expression->reference.declaration;
1155                 if(declaration->storage_class == STORAGE_CLASS_ENUM_ENTRY)
1156                         return true;
1157
1158                 return false;
1159         }
1160
1161         case EXPR_UNKNOWN:
1162         case EXPR_INVALID:
1163                 break;
1164         }
1165         panic("invalid expression found (is constant expression)");
1166 }
1167
1168
1169 void init_ast(void)
1170 {
1171         obstack_init(&ast_obstack);
1172 }
1173
1174 void exit_ast(void)
1175 {
1176         obstack_free(&ast_obstack, NULL);
1177 }
1178
1179 void ast_set_output(FILE *stream)
1180 {
1181         out = stream;
1182         type_set_output(stream);
1183 }
1184
1185 void* (allocate_ast) (size_t size)
1186 {
1187         return _allocate_ast(size);
1188 }