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