tcc.c

   1 /*
   2  *  TCC - Tiny C Compiler
   3  *
   4  *  Copyright (c) 2001-2004 Fabrice Bellard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20 #define _GNU_SOURCE
  21 #include "config.h"
  22
  23 #ifdef CONFIG_TCCBOOT
  24
  25 #include "tccboot.h"
  26 #define CONFIG_TCC_STATIC
  27
  28 #else
  29
  30 #include <stdlib.h>
  31 #include <stdio.h>
  32 #include <stdarg.h>
  33 #include <string.h>
  34 #include <errno.h>
  35 #include <math.h>
  36 #include <unistd.h>
  37 #include <signal.h>
  38 #include <fcntl.h>
  39 #include <setjmp.h>
  40 #include <time.h>
  41 #ifdef _WIN32
  42 #include <sys/timeb.h>
  43 #include <windows.h>
  44 #endif
  45 #ifndef _WIN32
  46 #include <sys/time.h>
  47 #include <sys/ucontext.h>
  48 #include <sys/mman.h>
  49 #endif
  50
  51 #endif /* !CONFIG_TCCBOOT */
  52
  53 #ifndef PAGESIZE
  54 #define PAGESIZE 4096
  55 #endif
  56
  57 #include "elf.h"
  58 #include "stab.h"
  59
  60 #ifndef O_BINARY
  61 #define O_BINARY 0
  62 #endif
  63
  64 #include "libtcc.h"
  65
  66 /* parser debug */
  67 //#define PARSE_DEBUG
  68 /* preprocessor debug */
  69 //#define PP_DEBUG
  70 /* include file debug */
  71 //#define INC_DEBUG
  72
  73 //#define MEM_DEBUG
  74
  75 /* assembler debug */
  76 //#define ASM_DEBUG
  77
  78 /* target selection */
  79 //#define TCC_TARGET_I386   /* i386 code generator */
  80 //#define TCC_TARGET_ARM    /* ARMv4 code generator */
  81 //#define TCC_TARGET_C67    /* TMS320C67xx code generator */
  82
  83 /* default target is I386 */
  84 #if !defined(TCC_TARGET_I386) && !defined(TCC_TARGET_ARM) && \
  85     !defined(TCC_TARGET_C67)
  86 #define TCC_TARGET_I386
  87 #endif
  88
  89 #if !defined(_WIN32) && !defined(TCC_UCLIBC) && !defined(TCC_TARGET_ARM) && \
  90     !defined(TCC_TARGET_C67)
  91 #define CONFIG_TCC_BCHECK /* enable bound checking code */
  92 #endif
  93
  94 #if defined(_WIN32) && !defined(TCC_TARGET_PE)
  95 #define CONFIG_TCC_STATIC
  96 #endif
  97
  98 /* define it to include assembler support */
  99 #if !defined(TCC_TARGET_ARM) && !defined(TCC_TARGET_C67)
 100 #define CONFIG_TCC_ASM
 101 #endif
 102
 103 /* object format selection */
 104 #if defined(TCC_TARGET_C67)
 105 #define TCC_TARGET_COFF
 106 #endif
 107
 108 #define FALSE 0
 109 #define false 0
 110 #define TRUE 1
 111 #define true 1
 112 typedef int BOOL;
 113
 114 /* path to find crt1.o, crti.o and crtn.o. Only needed when generating
 115    executables or dlls */
 116 #define CONFIG_TCC_CRT_PREFIX CONFIG_SYSROOT "/usr/lib"
 117
 118 #define INCLUDE_STACK_SIZE  32
 119 #define IFDEF_STACK_SIZE    64
 120 #define VSTACK_SIZE         256
 121 #define STRING_MAX_SIZE     1024
 122 #define PACK_STACK_SIZE     8
 123
 124 #define TOK_HASH_SIZE       8192 /* must be a power of two */
 125 #define TOK_ALLOC_INCR      512  /* must be a power of two */
 126 #define TOK_MAX_SIZE        4 /* token max size in int unit when stored in string */
 127
 128 /* token symbol management */
 129 typedef struct TokenSym {
 130     struct TokenSym *hash_next;
 131     struct Sym *sym_define; /* direct pointer to define */
 132     struct Sym *sym_label; /* direct pointer to label */
 133     struct Sym *sym_struct; /* direct pointer to structure */
 134     struct Sym *sym_identifier; /* direct pointer to identifier */
 135     int tok; /* token number */
 136     int len;
 137     char str[1];
 138 } TokenSym;
 139
 140 #ifdef TCC_TARGET_PE
 141 typedef unsigned short nwchar_t;
 142 #else
 143 typedef int nwchar_t;
 144 #endif
 145
 146 typedef struct CString {
 147     int size; /* size in bytes */
 148     void *data; /* either 'char *' or 'nwchar_t *' */
 149     int size_allocated;
 150     void *data_allocated; /* if non NULL, data has been malloced */
 151 } CString;
 152
 153 /* type definition */
 154 typedef struct CType {
 155     int t;
 156     struct Sym *ref;
 157 } CType;
 158
 159 /* constant value */
 160 typedef union CValue {
 161     long double ld;
 162     double d;
 163     float f;
 164     int i;
 165     unsigned int ui;
 166     unsigned int ul; /* address (should be unsigned long on 64 bit cpu) */
 167     long long ll;
 168     unsigned long long ull;
 169     struct CString *cstr;
 170     void *ptr;
 171     int tab[1];
 172 } CValue;
 173
 174 /* value on stack */
 175 typedef struct SValue {
 176     CType type;      /* type */
 177     unsigned short r;      /* register + flags */
 178     unsigned short r2;     /* second register, used for 'long long'
 179                               type. If not used, set to VT_CONST */
 180     CValue c;              /* constant, if VT_CONST */
 181     struct Sym *sym;       /* symbol, if (VT_SYM | VT_CONST) */
 182 } SValue;
 183
 184 /* symbol management */
 185 typedef struct Sym {
 186     int v;    /* symbol token */
 187     int r;    /* associated register */
 188     int c;    /* associated number */
 189     CType type;    /* associated type */
 190     struct Sym *next; /* next related symbol */
 191     struct Sym *prev; /* prev symbol in stack */
 192     struct Sym *prev_tok; /* previous symbol for this token */
 193 } Sym;
 194
 195 /* section definition */
 196 /* XXX: use directly ELF structure for parameters ? */
 197 /* special flag to indicate that the section should not be linked to
 198    the other ones */
 199 #define SHF_PRIVATE 0x80000000
 200
 201 typedef struct Section {
 202     unsigned long data_offset; /* current data offset */
 203     unsigned char *data;       /* section data */
 204     unsigned long data_allocated; /* used for realloc() handling */
 205     int sh_name;             /* elf section name (only used during output) */
 206     int sh_num;              /* elf section number */
 207     int sh_type;             /* elf section type */
 208     int sh_flags;            /* elf section flags */
 209     int sh_info;             /* elf section info */
 210     int sh_addralign;        /* elf section alignment */
 211     int sh_entsize;          /* elf entry size */
 212     unsigned long sh_size;   /* section size (only used during output) */
 213     unsigned long sh_addr;      /* address at which the section is relocated */
 214     unsigned long sh_offset;    /* file offset */
 215     int nb_hashed_syms;      /* used to resize the hash table */
 216     struct Section *link;    /* link to another section */
 217     struct Section *reloc;   /* corresponding section for relocation, if any */
 218     struct Section *hash;     /* hash table for symbols */
 219     struct Section *next;
 220     char name[1];           /* section name */
 221 } Section;
 222
 223 typedef struct DLLReference {
 224     int level;
 225     void *handle;
 226     char name[1];
 227 } DLLReference;
 228
 229 /* GNUC attribute definition */
 230 typedef struct AttributeDef {
 231     int aligned;
 232     int packed;
 233     Section *section;
 234     int func_attr; /* calling convention, exports, ... */
 235 } AttributeDef;
 236
 237 /* -------------------------------------------------- */
 238 /* gr: wrappers for casting sym->r for other purposes */
 239 typedef struct {
 240     unsigned
 241       func_call : 8,
 242       func_args : 8,
 243       func_export : 1;
 244 } func_attr_t;
 245
 246 #define FUNC_CALL(r) (((func_attr_t*)&(r))->func_call)
 247 #define FUNC_EXPORT(r) (((func_attr_t*)&(r))->func_export)
 248 #define FUNC_ARGS(r) (((func_attr_t*)&(r))->func_args)
 249 #define INLINE_DEF(r) (*(int **)&(r))
 250 /* -------------------------------------------------- */
 251
 252 #define SYM_STRUCT     0x40000000 /* struct/union/enum symbol space */
 253 #define SYM_FIELD      0x20000000 /* struct/union field symbol space */
 254 #define SYM_FIRST_ANOM 0x10000000 /* first anonymous sym */
 255
 256 /* stored in 'Sym.c' field */
 257 #define FUNC_NEW       1 /* ansi function prototype */
 258 #define FUNC_OLD       2 /* old function prototype */
 259 #define FUNC_ELLIPSIS  3 /* ansi function prototype with ... */
 260
 261 /* stored in 'Sym.r' field */
 262 #define FUNC_CDECL     0 /* standard c call */
 263 #define FUNC_STDCALL   1 /* pascal c call */
 264 #define FUNC_FASTCALL1 2 /* first param in %eax */
 265 #define FUNC_FASTCALL2 3 /* first parameters in %eax, %edx */
 266 #define FUNC_FASTCALL3 4 /* first parameter in %eax, %edx, %ecx */
 267 #define FUNC_FASTCALLW 5 /* first parameter in %ecx, %edx */
 268
 269 /* field 'Sym.t' for macros */
 270 #define MACRO_OBJ      0 /* object like macro */
 271 #define MACRO_FUNC     1 /* function like macro */
 272
 273 /* field 'Sym.r' for C labels */
 274 #define LABEL_DEFINED  0 /* label is defined */
 275 #define LABEL_FORWARD  1 /* label is forward defined */
 276 #define LABEL_DECLARED 2 /* label is declared but never used */
 277
 278 /* type_decl() types */
 279 #define TYPE_ABSTRACT  1 /* type without variable */
 280 #define TYPE_DIRECT    2 /* type with variable */
 281
 282 #define IO_BUF_SIZE 8192
 283
 284 typedef struct BufferedFile {
 285     uint8_t *buf_ptr;
 286     uint8_t *buf_end;
 287     int fd;
 288     int line_num;    /* current line number - here to simplify code */
 289     int ifndef_macro;  /* #ifndef macro / #endif search */
 290     int ifndef_macro_saved; /* saved ifndef_macro */
 291     int *ifdef_stack_ptr; /* ifdef_stack value at the start of the file */
 292     char inc_type;          /* type of include */
 293     char inc_filename[512]; /* filename specified by the user */
 294     char filename[1024];    /* current filename - here to simplify code */
 295     unsigned char buffer[IO_BUF_SIZE + 1]; /* extra size for CH_EOB char */
 296 } BufferedFile;
 297
 298 #define CH_EOB   '\\'       /* end of buffer or '\0' char in file */
 299 #define CH_EOF   (-1)   /* end of file */
 300
 301 /* parsing state (used to save parser state to reparse part of the
 302    source several times) */
 303 typedef struct ParseState {
 304     int *macro_ptr;
 305     int line_num;
 306     int tok;
 307     CValue tokc;
 308 } ParseState;
 309
 310 /* used to record tokens */
 311 typedef struct TokenString {
 312     int *str;
 313     int len;
 314     int allocated_len;
 315     int last_line_num;
 316 } TokenString;
 317
 318 /* include file cache, used to find files faster and also to eliminate
 319    inclusion if the include file is protected by #ifndef ... #endif */
 320 typedef struct CachedInclude {
 321     int ifndef_macro;
 322     int hash_next; /* -1 if none */
 323     char type; /* '"' or '>' to give include type */
 324     char filename[1]; /* path specified in #include */
 325 } CachedInclude;
 326
 327 #define CACHED_INCLUDES_HASH_SIZE 512
 328
 329 /* parser */
 330 static struct BufferedFile *file;
 331 static int ch, tok;
 332 static CValue tokc;
 333 static CString tokcstr; /* current parsed string, if any */
 334 /* additional informations about token */
 335 static int tok_flags;
 336 #define TOK_FLAG_BOL   0x0001 /* beginning of line before */
 337 #define TOK_FLAG_BOF   0x0002 /* beginning of file before */
 338 #define TOK_FLAG_ENDIF 0x0004 /* a endif was found matching starting #ifdef */
 339 #define TOK_FLAG_EOF   0x0008 /* end of file */
 340
 341 static int *macro_ptr, *macro_ptr_allocated;
 342 static int *unget_saved_macro_ptr;
 343 static int unget_saved_buffer[TOK_MAX_SIZE + 1];
 344 static int unget_buffer_enabled;
 345 static int parse_flags;
 346 #define PARSE_FLAG_PREPROCESS 0x0001 /* activate preprocessing */
 347 #define PARSE_FLAG_TOK_NUM    0x0002 /* return numbers instead of TOK_PPNUM */
 348 #define PARSE_FLAG_LINEFEED   0x0004 /* line feed is returned as a
 349                                         token. line feed is also
 350                                         returned at eof */
 351 #define PARSE_FLAG_ASM_COMMENTS 0x0008 /* '#' can be used for line comment */
 352
 353 static Section *text_section, *data_section, *bss_section; /* predefined sections */
 354 static Section *cur_text_section; /* current section where function code is
 355                               generated */
 356 #ifdef CONFIG_TCC_ASM
 357 static Section *last_text_section; /* to handle .previous asm directive */
 358 #endif
 359 /* bound check related sections */
 360 static Section *bounds_section; /* contains global data bound description */
 361 static Section *lbounds_section; /* contains local data bound description */
 362 /* symbol sections */
 363 static Section *symtab_section, *strtab_section;
 364
 365 /* debug sections */
 366 static Section *stab_section, *stabstr_section;
 367
 368 /* loc : local variable index
 369    ind : output code index
 370    rsym: return symbol
 371    anon_sym: anonymous symbol index
 372 */
 373 static int rsym, anon_sym, ind, loc;
 374 /* expression generation modifiers */
 375 static int const_wanted; /* true if constant wanted */
 376 static int nocode_wanted; /* true if no code generation wanted for an expression */
 377 static int global_expr;  /* true if compound literals must be allocated
 378                             globally (used during initializers parsing */
 379 static CType func_vt; /* current function return type (used by return
 380                          instruction) */
 381 static int func_vc;
 382 static int last_line_num, last_ind, func_ind; /* debug last line number and pc */
 383 static int tok_ident;
 384 static TokenSym **table_ident;
 385 static TokenSym *hash_ident[TOK_HASH_SIZE];
 386 static char token_buf[STRING_MAX_SIZE + 1];
 387 static char *funcname;
 388 static Sym *global_stack, *local_stack;
 389 static Sym *define_stack;
 390 static Sym *global_label_stack, *local_label_stack;
 391 /* symbol allocator */
 392 #define SYM_POOL_NB (8192 / sizeof(Sym))
 393 static Sym *sym_free_first;
 394 static void **sym_pools;
 395 static int nb_sym_pools;
 396
 397 static SValue vstack[VSTACK_SIZE], *vtop;
 398 /* some predefined types */
 399 static CType char_pointer_type, func_old_type, int_type;
 400 /* true if isid(c) || isnum(c) */
 401 static unsigned char isidnum_table[256-CH_EOF];
 402
 403 /* display some information during compilation */
 404 static int verbose = 0;
 405
 406 /* compile with debug symbol (and use them if error during execution) */
 407 static int do_debug = 0;
 408
 409 /* compile with built-in memory and bounds checker */
 410 static int do_bounds_check = 0;
 411
 412 /* display benchmark infos */
 413 #if !defined(LIBTCC)
 414 static int do_bench = 0;
 415 #endif
 416 static int total_lines;
 417 static int total_bytes;
 418
 419 /* use GNU C extensions */
 420 static int gnu_ext = 1;
 421
 422 /* use Tiny C extensions */
 423 static int tcc_ext = 1;
 424
 425 /* max number of callers shown if error */
 426 static int num_callers = 6;
 427 static const char **rt_bound_error_msg;
 428
 429 /* XXX: get rid of this ASAP */
 430 static struct TCCState *tcc_state;
 431
 432 /* give the path of the tcc libraries */
 433 static const char *tcc_lib_path = CONFIG_TCCDIR;
 434
 435 struct TCCState {
 436     int output_type;
 437
 438     BufferedFile **include_stack_ptr;
 439     int *ifdef_stack_ptr;
 440
 441     /* include file handling */
 442     char **include_paths;
 443     int nb_include_paths;
 444     char **sysinclude_paths;
 445     int nb_sysinclude_paths;
 446     CachedInclude **cached_includes;
 447     int nb_cached_includes;
 448
 449     char **library_paths;
 450     int nb_library_paths;
 451
 452     /* array of all loaded dlls (including those referenced by loaded
 453        dlls) */
 454     DLLReference **loaded_dlls;
 455     int nb_loaded_dlls;
 456
 457     /* sections */
 458     Section **sections;
 459     int nb_sections; /* number of sections, including first dummy section */
 460
 461     /* got handling */
 462     Section *got;
 463     Section *plt;
 464     unsigned long *got_offsets;
 465     int nb_got_offsets;
 466     /* give the correspondance from symtab indexes to dynsym indexes */
 467     int *symtab_to_dynsym;
 468
 469     /* temporary dynamic symbol sections (for dll loading) */
 470     Section *dynsymtab_section;
 471     /* exported dynamic symbol section */
 472     Section *dynsym;
 473
 474     int nostdinc; /* if true, no standard headers are added */
 475     int nostdlib; /* if true, no standard libraries are added */
 476
 477     int nocommon; /* if true, do not use common symbols for .bss data */
 478
 479     /* if true, static linking is performed */
 480     int static_link;
 481
 482     /* soname as specified on the command line (-soname) */
 483     const char *soname;
 484
 485     /* if true, all symbols are exported */
 486     int rdynamic;
 487
 488     /* if true, only link in referenced objects from archive */
 489     int alacarte_link;
 490
 491     /* address of text section */
 492     unsigned long text_addr;
 493     int has_text_addr;
 494
 495     /* output format, see TCC_OUTPUT_FORMAT_xxx */
 496     int output_format;
 497
 498     /* C language options */
 499     int char_is_unsigned;
 500     int leading_underscore;
 501
 502     /* warning switches */
 503     int warn_write_strings;
 504     int warn_unsupported;
 505     int warn_error;
 506     int warn_none;
 507     int warn_implicit_function_declaration;
 508
 509     /* error handling */
 510     void *error_opaque;
 511     void (*error_func)(void *opaque, const char *msg);
 512     int error_set_jmp_enabled;
 513     jmp_buf error_jmp_buf;
 514     int nb_errors;
 515
 516     /* tiny assembler state */
 517     Sym *asm_labels;
 518
 519     /* see include_stack_ptr */
 520     BufferedFile *include_stack[INCLUDE_STACK_SIZE];
 521
 522     /* see ifdef_stack_ptr */
 523     int ifdef_stack[IFDEF_STACK_SIZE];
 524
 525     /* see cached_includes */
 526     int cached_includes_hash[CACHED_INCLUDES_HASH_SIZE];
 527
 528     /* pack stack */
 529     int pack_stack[PACK_STACK_SIZE];
 530     int *pack_stack_ptr;
 531
 532     /* output file for preprocessing */
 533     FILE *outfile;
 534 };
 535
 536 /* The current value can be: */
 537 #define VT_VALMASK   0x00ff
 538 #define VT_CONST     0x00f0  /* constant in vc
 539                               (must be first non register value) */
 540 #define VT_LLOCAL    0x00f1  /* lvalue, offset on stack */
 541 #define VT_LOCAL     0x00f2  /* offset on stack */
 542 #define VT_CMP       0x00f3  /* the value is stored in processor flags (in vc) */
 543 #define VT_JMP       0x00f4  /* value is the consequence of jmp true (even) */
 544 #define VT_JMPI      0x00f5  /* value is the consequence of jmp false (odd) */
 545 #define VT_LVAL      0x0100  /* var is an lvalue */
 546 #define VT_SYM       0x0200  /* a symbol value is added */
 547 #define VT_MUSTCAST  0x0400  /* value must be casted to be correct (used for
 548                                 char/short stored in integer registers) */
 549 #define VT_MUSTBOUND 0x0800  /* bound checking must be done before
 550                                 dereferencing value */
 551 #define VT_BOUNDED   0x8000  /* value is bounded. The address of the
 552                                 bounding function call point is in vc */
 553 #define VT_LVAL_BYTE     0x1000  /* lvalue is a byte */
 554 #define VT_LVAL_SHORT    0x2000  /* lvalue is a short */
 555 #define VT_LVAL_UNSIGNED 0x4000  /* lvalue is unsigned */
 556 #define VT_LVAL_TYPE     (VT_LVAL_BYTE | VT_LVAL_SHORT | VT_LVAL_UNSIGNED)
 557
 558 /* types */
 559 #define VT_INT        0  /* integer type */
 560 #define VT_BYTE       1  /* signed byte type */
 561 #define VT_SHORT      2  /* short type */
 562 #define VT_VOID       3  /* void type */
 563 #define VT_PTR        4  /* pointer */
 564 #define VT_ENUM       5  /* enum definition */
 565 #define VT_FUNC       6  /* function type */
 566 #define VT_STRUCT     7  /* struct/union definition */
 567 #define VT_FLOAT      8  /* IEEE float */
 568 #define VT_DOUBLE     9  /* IEEE double */
 569 #define VT_LDOUBLE   10  /* IEEE long double */
 570 #define VT_BOOL      11  /* ISOC99 boolean type */
 571 #define VT_LLONG     12  /* 64 bit integer */
 572 #define VT_LONG      13  /* long integer (NEVER USED as type, only
 573                             during parsing) */
 574 #define VT_BTYPE      0x000f /* mask for basic type */
 575 #define VT_UNSIGNED   0x0010  /* unsigned type */
 576 #define VT_ARRAY      0x0020  /* array type (also has VT_PTR) */
 577 #define VT_BITFIELD   0x0040  /* bitfield modifier */
 578 #define VT_CONSTANT   0x0800  /* const modifier */
 579 #define VT_VOLATILE   0x1000  /* volatile modifier */
 580 #define VT_SIGNED     0x2000  /* signed type */
 581
 582 /* storage */
 583 #define VT_EXTERN  0x00000080  /* extern definition */
 584 #define VT_STATIC  0x00000100  /* static variable */
 585 #define VT_TYPEDEF 0x00000200  /* typedef definition */
 586 #define VT_INLINE  0x00000400  /* inline definition */
 587
 588 #define VT_STRUCT_SHIFT 16   /* shift for bitfield shift values */
 589
 590 /* type mask (except storage) */
 591 #define VT_STORAGE (VT_EXTERN | VT_STATIC | VT_TYPEDEF | VT_INLINE)
 592 #define VT_TYPE    (~(VT_STORAGE))
 593
 594 /* token values */
 595
 596 /* warning: the following compare tokens depend on i386 asm code */
 597 #define TOK_ULT 0x92
 598 #define TOK_UGE 0x93
 599 #define TOK_EQ  0x94
 600 #define TOK_NE  0x95
 601 #define TOK_ULE 0x96
 602 #define TOK_UGT 0x97
 603 #define TOK_Nset 0x98
 604 #define TOK_Nclear 0x99
 605 #define TOK_LT  0x9c
 606 #define TOK_GE  0x9d
 607 #define TOK_LE  0x9e
 608 #define TOK_GT  0x9f
 609
 610 #define TOK_LAND  0xa0
 611 #define TOK_LOR   0xa1
 612
 613 #define TOK_DEC   0xa2
 614 #define TOK_MID   0xa3 /* inc/dec, to void constant */
 615 #define TOK_INC   0xa4
 616 #define TOK_UDIV  0xb0 /* unsigned division */
 617 #define TOK_UMOD  0xb1 /* unsigned modulo */
 618 #define TOK_PDIV  0xb2 /* fast division with undefined rounding for pointers */
 619 #define TOK_CINT   0xb3 /* number in tokc */
 620 #define TOK_CCHAR 0xb4 /* char constant in tokc */
 621 #define TOK_STR   0xb5 /* pointer to string in tokc */
 622 #define TOK_TWOSHARPS 0xb6 /* ## preprocessing token */
 623 #define TOK_LCHAR    0xb7
 624 #define TOK_LSTR     0xb8
 625 #define TOK_CFLOAT   0xb9 /* float constant */
 626 #define TOK_LINENUM  0xba /* line number info */
 627 #define TOK_CDOUBLE  0xc0 /* double constant */
 628 #define TOK_CLDOUBLE 0xc1 /* long double constant */
 629 #define TOK_UMULL    0xc2 /* unsigned 32x32 -> 64 mul */
 630 #define TOK_ADDC1    0xc3 /* add with carry generation */
 631 #define TOK_ADDC2    0xc4 /* add with carry use */
 632 #define TOK_SUBC1    0xc5 /* add with carry generation */
 633 #define TOK_SUBC2    0xc6 /* add with carry use */
 634 #define TOK_CUINT    0xc8 /* unsigned int constant */
 635 #define TOK_CLLONG   0xc9 /* long long constant */
 636 #define TOK_CULLONG  0xca /* unsigned long long constant */
 637 #define TOK_ARROW    0xcb
 638 #define TOK_DOTS     0xcc /* three dots */
 639 #define TOK_SHR      0xcd /* unsigned shift right */
 640 #define TOK_PPNUM    0xce /* preprocessor number */
 641
 642 #define TOK_SHL   0x01 /* shift left */
 643 #define TOK_SAR   0x02 /* signed shift right */
 644
 645 /* assignement operators : normal operator or 0x80 */
 646 #define TOK_A_MOD 0xa5
 647 #define TOK_A_AND 0xa6
 648 #define TOK_A_MUL 0xaa
 649 #define TOK_A_ADD 0xab
 650 #define TOK_A_SUB 0xad
 651 #define TOK_A_DIV 0xaf
 652 #define TOK_A_XOR 0xde
 653 #define TOK_A_OR  0xfc
 654 #define TOK_A_SHL 0x81
 655 #define TOK_A_SAR 0x82
 656
 657 #ifndef offsetof
 658 #define offsetof(type, field) ((size_t) &((type *)0)->field)
 659 #endif
 660
 661 #ifndef countof
 662 #define countof(tab) (sizeof(tab) / sizeof((tab)[0]))
 663 #endif
 664
 665 /* WARNING: the content of this string encodes token numbers */
 666 static char tok_two_chars[] = "<=\236>=\235!=\225&&\240||\241++\244--\242==\224<<\1>>\2+=\253-=\255*=\252/=\257%=\245&=\246^=\336|=\374->\313..\250##\266";
 667
 668 #define TOK_EOF       (-1)  /* end of file */
 669 #define TOK_LINEFEED  10    /* line feed */
 670
 671 /* all identificators and strings have token above that */
 672 #define TOK_IDENT 256
 673
 674 /* only used for i386 asm opcodes definitions */
 675 #define DEF_ASM(x) DEF(TOK_ASM_ ## x, #x)
 676
 677 #define DEF_BWL(x) \
 678  DEF(TOK_ASM_ ## x ## b, #x "b") \
 679  DEF(TOK_ASM_ ## x ## w, #x "w") \
 680  DEF(TOK_ASM_ ## x ## l, #x "l") \
 681  DEF(TOK_ASM_ ## x, #x)
 682
 683 #define DEF_WL(x) \
 684  DEF(TOK_ASM_ ## x ## w, #x "w") \
 685  DEF(TOK_ASM_ ## x ## l, #x "l") \
 686  DEF(TOK_ASM_ ## x, #x)
 687
 688 #define DEF_FP1(x) \
 689  DEF(TOK_ASM_ ## f ## x ## s, "f" #x "s") \
 690  DEF(TOK_ASM_ ## fi ## x ## l, "fi" #x "l") \
 691  DEF(TOK_ASM_ ## f ## x ## l, "f" #x "l") \
 692  DEF(TOK_ASM_ ## fi ## x ## s, "fi" #x "s")
 693
 694 #define DEF_FP(x) \
 695  DEF(TOK_ASM_ ## f ## x, "f" #x ) \
 696  DEF(TOK_ASM_ ## f ## x ## p, "f" #x "p") \
 697  DEF_FP1(x)
 698
 699 #define DEF_ASMTEST(x) \
 700  DEF_ASM(x ## o) \
 701  DEF_ASM(x ## no) \
 702  DEF_ASM(x ## b) \
 703  DEF_ASM(x ## c) \
 704  DEF_ASM(x ## nae) \
 705  DEF_ASM(x ## nb) \
 706  DEF_ASM(x ## nc) \
 707  DEF_ASM(x ## ae) \
 708  DEF_ASM(x ## e) \
 709  DEF_ASM(x ## z) \
 710  DEF_ASM(x ## ne) \
 711  DEF_ASM(x ## nz) \
 712  DEF_ASM(x ## be) \
 713  DEF_ASM(x ## na) \
 714  DEF_ASM(x ## nbe) \
 715  DEF_ASM(x ## a) \
 716  DEF_ASM(x ## s) \
 717  DEF_ASM(x ## ns) \
 718  DEF_ASM(x ## p) \
 719  DEF_ASM(x ## pe) \
 720  DEF_ASM(x ## np) \
 721  DEF_ASM(x ## po) \
 722  DEF_ASM(x ## l) \
 723  DEF_ASM(x ## nge) \
 724  DEF_ASM(x ## nl) \
 725  DEF_ASM(x ## ge) \
 726  DEF_ASM(x ## le) \
 727  DEF_ASM(x ## ng) \
 728  DEF_ASM(x ## nle) \
 729  DEF_ASM(x ## g)
 730
 731 #define TOK_ASM_int TOK_INT
 732
 733 enum tcc_token {
 734     TOK_LAST = TOK_IDENT - 1,
 735 #define DEF(id, str) id,
 736 #include "tcctok.h"
 737 #undef DEF
 738 };
 739
 740 static const char tcc_keywords[] =
 741 #define DEF(id, str) str "\0"
 742 #include "tcctok.h"
 743 #undef DEF
 744 ;
 745
 746 #define TOK_UIDENT TOK_DEFINE
 747
 748 #ifdef _WIN32
 749 #define snprintf _snprintf
 750 #define vsnprintf _vsnprintf
 751 #ifndef __GNUC__
 752   #define strtold (long double)strtod
 753   #define strtof (float)strtod
 754   #define strtoll (long long)strtol
 755 #endif
 756 #elif defined(TCC_UCLIBC) || defined(__FreeBSD__) || defined(__DragonFly__) \
 757     || defined(__OpenBSD__)
 758 /* currently incorrect */
 759 long double strtold(const char *nptr, char **endptr)
 760 {
 761     return (long double)strtod(nptr, endptr);
 762 }
 763 float strtof(const char *nptr, char **endptr)
 764 {
 765     return (float)strtod(nptr, endptr);
 766 }
 767 #else
 768 /* XXX: need to define this to use them in non ISOC99 context */
 769 extern float strtof (const char *__nptr, char **__endptr);
 770 extern long double strtold (const char *__nptr, char **__endptr);
 771 #endif
 772
 773 static char *pstrcpy(char *buf, int buf_size, const char *s);
 774 static char *pstrcat(char *buf, int buf_size, const char *s);
 775 static char *tcc_basename(const char *name);
 776 static char *tcc_fileextension (const char *p);
 777
 778 static void next(void);
 779 static void next_nomacro(void);
 780 static void parse_expr_type(CType *type);
 781 static void expr_type(CType *type);
 782 static void unary_type(CType *type);
 783 static void block(int *bsym, int *csym, int *case_sym, int *def_sym,
 784                   int case_reg, int is_expr);
 785 static int expr_const(void);
 786 static void expr_eq(void);
 787 static void gexpr(void);
 788 static void gen_inline_functions(void);
 789 static void decl(int l);
 790 static void decl_initializer(CType *type, Section *sec, unsigned long c,
 791                              int first, int size_only);
 792 static void decl_initializer_alloc(CType *type, AttributeDef *ad, int r,
 793                                    int has_init, int v, int scope);
 794 int gv(int rc);
 795 void gv2(int rc1, int rc2);
 796 void move_reg(int r, int s);
 797 void save_regs(int n);
 798 void save_reg(int r);
 799 void vpop(void);
 800 void vswap(void);
 801 void vdup(void);
 802 int get_reg(int rc);
 803 int get_reg_ex(int rc,int rc2);
 804
 805 struct macro_level {
 806     struct macro_level *prev;
 807     int *p;
 808 };
 809
 810 static void macro_subst(TokenString *tok_str, Sym **nested_list,
 811                         const int *macro_str, struct macro_level **can_read_stream);
 812 void gen_op(int op);
 813 void force_charshort_cast(int t);
 814 static void gen_cast(CType *type);
 815 void vstore(void);
 816 static Sym *sym_find(int v);
 817 static Sym *sym_push(int v, CType *type, int r, int c);
 818
 819 /* type handling */
 820 static int type_size(CType *type, int *a);
 821 static inline CType *pointed_type(CType *type);
 822 static int pointed_size(CType *type);
 823 static int lvalue_type(int t);
 824 static int parse_btype(CType *type, AttributeDef *ad);
 825 static void type_decl(CType *type, AttributeDef *ad, int *v, int td);
 826 static int compare_types(CType *type1, CType *type2, int unqualified);
 827 static int is_compatible_types(CType *type1, CType *type2);
 828 static int is_compatible_parameter_types(CType *type1, CType *type2);
 829
 830 int ieee_finite(double d);
 831 void error(const char *fmt, ...);
 832 void vpushi(int v);
 833 void vrott(int n);
 834 void vnrott(int n);
 835 void lexpand_nr(void);
 836 static void vpush_global_sym(CType *type, int v);
 837 void vset(CType *type, int r, int v);
 838 void type_to_str(char *buf, int buf_size,
 839                  CType *type, const char *varstr);
 840 char *get_tok_str(int v, CValue *cv);
 841 static Sym *get_sym_ref(CType *type, Section *sec,
 842                         unsigned long offset, unsigned long size);
 843 static Sym *external_global_sym(int v, CType *type, int r);
 844
 845 /* section generation */
 846 static void section_realloc(Section *sec, unsigned long new_size);
 847 static void *section_ptr_add(Section *sec, unsigned long size);
 848 static void put_extern_sym(Sym *sym, Section *section,
 849                            unsigned long value, unsigned long size);
 850 static void greloc(Section *s, Sym *sym, unsigned long addr, int type);
 851 static int put_elf_str(Section *s, const char *sym);
 852 static int put_elf_sym(Section *s,
 853                        unsigned long value, unsigned long size,
 854                        int info, int other, int shndx, const char *name);
 855 static int add_elf_sym(Section *s, unsigned long value, unsigned long size,
 856                        int info, int other, int sh_num, const char *name);
 857 static void put_elf_reloc(Section *symtab, Section *s, unsigned long offset,
 858                           int type, int symbol);
 859 static void put_stabs(const char *str, int type, int other, int desc,
 860                       unsigned long value);
 861 static void put_stabs_r(const char *str, int type, int other, int desc,
 862                         unsigned long value, Section *sec, int sym_index);
 863 static void put_stabn(int type, int other, int desc, int value);
 864 static void put_stabd(int type, int other, int desc);
 865 static int tcc_add_dll(TCCState *s, const char *filename, int flags);
 866
 867 #define AFF_PRINT_ERROR     0x0001 /* print error if file not found */
 868 #define AFF_REFERENCED_DLL  0x0002 /* load a referenced dll from another dll */
 869 #define AFF_PREPROCESS      0x0004 /* preprocess file */
 870 static int tcc_add_file_internal(TCCState *s, const char *filename, int flags);
 871
 872 /* tcccoff.c */
 873 int tcc_output_coff(TCCState *s1, FILE *f);
 874
 875 /* tccpe.c */
 876 void *resolve_sym(TCCState *s1, const char *sym, int type);
 877 int pe_load_def_file(struct TCCState *s1, int fd);
 878 int pe_test_res_file(void *v, int size);
 879 int pe_load_res_file(struct TCCState *s1, int fd);
 880 void pe_add_runtime(struct TCCState *s1);
 881 void pe_guess_outfile(char *objfilename, int output_type);
 882 int pe_output_file(struct TCCState *s1, const char *filename);
 883
 884 /* tccasm.c */
 885
 886 #ifdef CONFIG_TCC_ASM
 887
 888 typedef struct ExprValue {
 889     uint32_t v;
 890     Sym *sym;
 891 } ExprValue;
 892
 893 #define MAX_ASM_OPERANDS 30
 894
 895 typedef struct ASMOperand {
 896     int id; /* GCC 3 optionnal identifier (0 if number only supported */
 897     char *constraint;
 898     char asm_str[16]; /* computed asm string for operand */
 899     SValue *vt; /* C value of the expression */
 900     int ref_index; /* if >= 0, gives reference to a output constraint */
 901     int input_index; /* if >= 0, gives reference to an input constraint */
 902     int priority; /* priority, used to assign registers */
 903     int reg; /* if >= 0, register number used for this operand */
 904     int is_llong; /* true if double register value */
 905     int is_memory; /* true if memory operand */
 906     int is_rw;     /* for '+' modifier */
 907 } ASMOperand;
 908
 909 static void asm_expr(TCCState *s1, ExprValue *pe);
 910 static int asm_int_expr(TCCState *s1);
 911 static int find_constraint(ASMOperand *operands, int nb_operands,
 912                            const char *name, const char **pp);
 913
 914 static int tcc_assemble(TCCState *s1, int do_preprocess);
 915
 916 #endif
 917
 918 static void asm_instr(void);
 919 static void asm_global_instr(void);
 920
 921 /* true if float/double/long double type */
 922 static inline int is_float(int t)
 923 {
 924     int bt;
 925     bt = t & VT_BTYPE;
 926     return bt == VT_LDOUBLE || bt == VT_DOUBLE || bt == VT_FLOAT;
 927 }
 928
 929 #ifdef TCC_TARGET_I386
 930 #include "i386-gen.c"
 931 #endif
 932
 933 #ifdef TCC_TARGET_ARM
 934 #include "arm-gen.c"
 935 #endif
 936
 937 #ifdef TCC_TARGET_C67
 938 #include "c67-gen.c"
 939 #endif
 940
 941 #ifdef CONFIG_TCC_STATIC
 942
 943 #define RTLD_LAZY       0x001
 944 #define RTLD_NOW        0x002
 945 #define RTLD_GLOBAL     0x100
 946 #define RTLD_DEFAULT    NULL
 947
 948 /* dummy function for profiling */
 949 void *dlopen(const char *filename, int flag)
 950 {
 951     return NULL;
 952 }
 953
 954 const char *dlerror(void)
 955 {
 956     return "error";
 957 }
 958
 959 typedef struct TCCSyms {
 960     char *str;
 961     void *ptr;
 962 } TCCSyms;
 963
 964 #define TCCSYM(a) { #a, &a, },
 965
 966 /* add the symbol you want here if no dynamic linking is done */
 967 static TCCSyms tcc_syms[] = {
 968 #if !defined(CONFIG_TCCBOOT)
 969     TCCSYM(printf)
 970     TCCSYM(fprintf)
 971     TCCSYM(fopen)
 972     TCCSYM(fclose)
 973 #endif
 974     { NULL, NULL },
 975 };
 976
 977 void *resolve_sym(TCCState *s1, const char *symbol, int type)
 978 {
 979     TCCSyms *p;
 980     p = tcc_syms;
 981     while (p->str != NULL) {
 982         if (!strcmp(p->str, symbol))
 983             return p->ptr;
 984         p++;
 985     }
 986     return NULL;
 987 }
 988
 989 #elif !defined(_WIN32)
 990
 991 #include <dlfcn.h>
 992
 993 void *resolve_sym(TCCState *s1, const char *sym, int type)
 994 {
 995     return dlsym(RTLD_DEFAULT, sym);
 996 }
 997
 998 #endif
 999
1000 /********************************************************/
1001
1002 /* we use our own 'finite' function to avoid potential problems with
1003    non standard math libs */
1004 /* XXX: endianness dependent */
1005 int ieee_finite(double d)
1006 {
1007     int *p = (int *)&d;
1008     return ((unsigned)((p[1] | 0x800fffff) + 1)) >> 31;
1009 }
1010
1011 /* copy a string and truncate it. */
1012 static char *pstrcpy(char *buf, int buf_size, const char *s)
1013 {
1014     char *q, *q_end;
1015     int c;
1016
1017     if (buf_size > 0) {
1018         q = buf;
1019         q_end = buf + buf_size - 1;
1020         while (q < q_end) {
1021             c = *s++;
1022             if (c == '\0')
1023                 break;
1024             *q++ = c;
1025         }
1026         *q = '\0';
1027     }
1028     return buf;
1029 }
1030
1031 /* strcat and truncate. */
1032 static char *pstrcat(char *buf, int buf_size, const char *s)
1033 {
1034     int len;
1035     len = strlen(buf);
1036     if (len < buf_size)
1037         pstrcpy(buf + len, buf_size - len, s);
1038     return buf;
1039 }
1040
1041 #ifndef LIBTCC
1042 static int strstart(const char *str, const char *val, const char **ptr)
1043 {
1044     const char *p, *q;
1045     p = str;
1046     q = val;
1047     while (*q != '\0') {
1048         if (*p != *q)
1049             return 0;
1050         p++;
1051         q++;
1052     }
1053     if (ptr)
1054         *ptr = p;
1055     return 1;
1056 }
1057 #endif
1058
1059 /* extract the basename of a file */
1060 static char *tcc_basename(const char *name)
1061 {
1062     char *p = strchr(name, 0);
1063     while (p > name
1064         && p[-1] != '/'
1065 #ifdef _WIN32
1066         && p[-1] != '\\'
1067 #endif
1068         )
1069         --p;
1070     return p;
1071 }
1072
1073 static char *tcc_fileextension (const char *name)
1074 {
1075     char *b = tcc_basename(name);
1076     char *e = strrchr(b, '.');
1077     return e ? e : strchr(b, 0);
1078 }
1079
1080 #ifdef _WIN32
1081 char *normalize_slashes(char *path)
1082 {
1083     char *p;
1084     for (p = path; *p; ++p)
1085         if (*p == '\\')
1086             *p = '/';
1087     return path;
1088 }
1089
1090 char *w32_tcc_lib_path(void)
1091 {
1092     /* on win32, we suppose the lib and includes are at the location
1093        of 'tcc.exe' */
1094     char path[1024], *p;
1095     GetModuleFileNameA(NULL, path, sizeof path);
1096     p = tcc_basename(normalize_slashes(strlwr(path)));
1097     if (p - 5 > path && 0 == strncmp(p - 5, "/bin/", 5))
1098         p -= 5;
1099     else if (p > path)
1100         p--;
1101     *p = 0;
1102     return strdup(path);
1103 }
1104 #endif
1105
1106 void set_pages_executable(void *ptr, unsigned long length)
1107 {
1108 #ifdef _WIN32
1109     unsigned long old_protect;
1110     VirtualProtect(ptr, length, PAGE_EXECUTE_READWRITE, &old_protect);
1111 #else
1112     unsigned long start, end;
1113     start = (unsigned long)ptr & ~(PAGESIZE - 1);
1114     end = (unsigned long)ptr + length;
1115     end = (end + PAGESIZE - 1) & ~(PAGESIZE - 1);
1116     mprotect((void *)start, end - start, PROT_READ | PROT_WRITE | PROT_EXEC);
1117 #endif
1118 }
1119
1120 /* memory management */
1121 #ifdef MEM_DEBUG
1122 int mem_cur_size;
1123 int mem_max_size;
1124 unsigned malloc_usable_size(void*);
1125 #endif
1126
1127 static inline void tcc_free(void *ptr)
1128 {
1129 #ifdef MEM_DEBUG
1130     mem_cur_size -= malloc_usable_size(ptr);
1131 #endif
1132     free(ptr);
1133 }
1134
1135 static void *tcc_malloc(unsigned long size)
1136 {
1137     void *ptr;
1138     ptr = malloc(size);
1139     if (!ptr && size)
1140         error("memory full");
1141 #ifdef MEM_DEBUG
1142     mem_cur_size += malloc_usable_size(ptr);
1143     if (mem_cur_size > mem_max_size)
1144         mem_max_size = mem_cur_size;
1145 #endif
1146     return ptr;
1147 }
1148
1149 static void *tcc_mallocz(unsigned long size)
1150 {
1151     void *ptr;
1152     ptr = tcc_malloc(size);
1153     memset(ptr, 0, size);
1154     return ptr;
1155 }
1156
1157 static inline void *tcc_realloc(void *ptr, unsigned long size)
1158 {
1159     void *ptr1;
1160 #ifdef MEM_DEBUG
1161     mem_cur_size -= malloc_usable_size(ptr);
1162 #endif
1163     ptr1 = realloc(ptr, size);
1164 #ifdef MEM_DEBUG
1165     /* NOTE: count not correct if alloc error, but not critical */
1166     mem_cur_size += malloc_usable_size(ptr1);
1167     if (mem_cur_size > mem_max_size)
1168         mem_max_size = mem_cur_size;
1169 #endif
1170     return ptr1;
1171 }
1172
1173 static char *tcc_strdup(const char *str)
1174 {
1175     char *ptr;
1176     ptr = tcc_malloc(strlen(str) + 1);
1177     strcpy(ptr, str);
1178     return ptr;
1179 }
1180
1181 #define free(p) use_tcc_free(p)
1182 #define malloc(s) use_tcc_malloc(s)
1183 #define realloc(p, s) use_tcc_realloc(p, s)
1184
1185 static void dynarray_add(void ***ptab, int *nb_ptr, void *data)
1186 {
1187     int nb, nb_alloc;
1188     void **pp;
1189
1190     nb = *nb_ptr;
1191     pp = *ptab;
1192     /* every power of two we double array size */
1193     if ((nb & (nb - 1)) == 0) {
1194         if (!nb)
1195             nb_alloc = 1;
1196         else
1197             nb_alloc = nb * 2;
1198         pp = tcc_realloc(pp, nb_alloc * sizeof(void *));
1199         if (!pp)
1200             error("memory full");
1201         *ptab = pp;
1202     }
1203     pp[nb++] = data;
1204     *nb_ptr = nb;
1205 }
1206
1207 static void dynarray_reset(void *pp, int *n)
1208 {
1209     void **p;
1210     for (p = *(void***)pp; *n; ++p, --*n)
1211         if (*p)
1212             tcc_free(*p);
1213     tcc_free(*(void**)pp);
1214     *(void**)pp = NULL;
1215 }
1216
1217 /* symbol allocator */
1218 static Sym *__sym_malloc(void)
1219 {
1220     Sym *sym_pool, *sym, *last_sym;
1221     int i;
1222
1223     sym_pool = tcc_malloc(SYM_POOL_NB * sizeof(Sym));
1224     dynarray_add(&sym_pools, &nb_sym_pools, sym_pool);
1225
1226     last_sym = sym_free_first;
1227     sym = sym_pool;
1228     for(i = 0; i < SYM_POOL_NB; i++) {
1229         sym->next = last_sym;
1230         last_sym = sym;
1231         sym++;
1232     }
1233     sym_free_first = last_sym;
1234     return last_sym;
1235 }
1236
1237 static inline Sym *sym_malloc(void)
1238 {
1239     Sym *sym;
1240     sym = sym_free_first;
1241     if (!sym)
1242         sym = __sym_malloc();
1243     sym_free_first = sym->next;
1244     return sym;
1245 }
1246
1247 static inline void sym_free(Sym *sym)
1248 {
1249     sym->next = sym_free_first;
1250     sym_free_first = sym;
1251 }
1252
1253 Section *new_section(TCCState *s1, const char *name, int sh_type, int sh_flags)
1254 {
1255     Section *sec;
1256
1257     sec = tcc_mallocz(sizeof(Section) + strlen(name));
1258     strcpy(sec->name, name);
1259     sec->sh_type = sh_type;
1260     sec->sh_flags = sh_flags;
1261     switch(sh_type) {
1262     case SHT_HASH:
1263     case SHT_REL:
1264     case SHT_DYNSYM:
1265     case SHT_SYMTAB:
1266     case SHT_DYNAMIC:
1267         sec->sh_addralign = 4;
1268         break;
1269     case SHT_STRTAB:
1270         sec->sh_addralign = 1;
1271         break;
1272     default:
1273         sec->sh_addralign = 32; /* default conservative alignment */
1274         break;
1275     }
1276
1277     /* only add section if not private */
1278     if (!(sh_flags & SHF_PRIVATE)) {
1279         sec->sh_num = s1->nb_sections;
1280         dynarray_add((void ***)&s1->sections, &s1->nb_sections, sec);
1281     }
1282     return sec;
1283 }
1284
1285 static void free_section(Section *s)
1286 {
1287     if (s->link && (s->link->sh_flags & SHF_PRIVATE))
1288         free_section(s->link);
1289     if (s->hash && (s->hash->sh_flags & SHF_PRIVATE))
1290         s->hash->link = NULL, free_section(s->hash);
1291     tcc_free(s->data);
1292     tcc_free(s);
1293 }
1294
1295 /* realloc section and set its content to zero */
1296 static void section_realloc(Section *sec, unsigned long new_size)
1297 {
1298     unsigned long size;
1299     unsigned char *data;
1300
1301     size = sec->data_allocated;
1302     if (size == 0)
1303         size = 1;
1304     while (size < new_size)
1305         size = size * 2;
1306     data = tcc_realloc(sec->data, size);
1307     if (!data)
1308         error("memory full");
1309     memset(data + sec->data_allocated, 0, size - sec->data_allocated);
1310     sec->data = data;
1311     sec->data_allocated = size;
1312 }
1313
1314 /* reserve at least 'size' bytes in section 'sec' from
1315    sec->data_offset. */
1316 static void *section_ptr_add(Section *sec, unsigned long size)
1317 {
1318     unsigned long offset, offset1;
1319
1320     offset = sec->data_offset;
1321     offset1 = offset + size;
1322     if (offset1 > sec->data_allocated)
1323         section_realloc(sec, offset1);
1324     sec->data_offset = offset1;
1325     return sec->data + offset;
1326 }
1327
1328 /* return a reference to a section, and create it if it does not
1329    exists */
1330 Section *find_section(TCCState *s1, const char *name)
1331 {
1332     Section *sec;
1333     int i;
1334     for(i = 1; i < s1->nb_sections; i++) {
1335         sec = s1->sections[i];
1336         if (!strcmp(name, sec->name))
1337             return sec;
1338     }
1339     /* sections are created as PROGBITS */
1340     return new_section(s1, name, SHT_PROGBITS, SHF_ALLOC);
1341 }
1342
1343 #define SECTION_ABS ((void *)1)
1344
1345 /* update sym->c so that it points to an external symbol in section
1346    'section' with value 'value' */
1347 static void put_extern_sym2(Sym *sym, Section *section,
1348                             unsigned long value, unsigned long size,
1349                             int can_add_underscore)
1350 {
1351     int sym_type, sym_bind, sh_num, info, other, attr;
1352     Elf32_Sym *esym;
1353     const char *name;
1354     char buf1[256];
1355
1356     if (section == NULL)
1357         sh_num = SHN_UNDEF;
1358     else if (section == SECTION_ABS)
1359         sh_num = SHN_ABS;
1360     else
1361         sh_num = section->sh_num;
1362
1363     other = attr = 0;
1364
1365     if ((sym->type.t & VT_BTYPE) == VT_FUNC) {
1366         sym_type = STT_FUNC;
1367 #ifdef TCC_TARGET_PE
1368         if (sym->type.ref)
1369             attr = sym->type.ref->r;
1370         if (FUNC_EXPORT(attr))
1371             other |= 1;
1372         if (FUNC_CALL(attr) == FUNC_STDCALL)
1373             other |= 2;
1374 #endif
1375     } else {
1376         sym_type = STT_OBJECT;
1377     }
1378
1379     if (sym->type.t & VT_STATIC)
1380         sym_bind = STB_LOCAL;
1381     else
1382         sym_bind = STB_GLOBAL;
1383
1384     if (!sym->c) {
1385         name = get_tok_str(sym->v, NULL);
1386 #ifdef CONFIG_TCC_BCHECK
1387         if (do_bounds_check) {
1388             char buf[32];
1389
1390             /* XXX: avoid doing that for statics ? */
1391             /* if bound checking is activated, we change some function
1392                names by adding the "__bound" prefix */
1393             switch(sym->v) {
1394 #if 0
1395             /* XXX: we rely only on malloc hooks */
1396             case TOK_malloc:
1397             case TOK_free:
1398             case TOK_realloc:
1399             case TOK_memalign:
1400             case TOK_calloc:
1401 #endif
1402             case TOK_memcpy:
1403             case TOK_memmove:
1404             case TOK_memset:
1405             case TOK_strlen:
1406             case TOK_strcpy:
1407             case TOK__alloca:
1408                 strcpy(buf, "__bound_");
1409                 strcat(buf, name);
1410                 name = buf;
1411                 break;
1412             }
1413         }
1414 #endif
1415
1416 #ifdef TCC_TARGET_PE
1417         if ((other & 2) && can_add_underscore) {
1418             sprintf(buf1, "_%s@%d", name, FUNC_ARGS(attr));
1419             name = buf1;
1420         } else
1421 #endif
1422         if (tcc_state->leading_underscore && can_add_underscore) {
1423             buf1[0] = '_';
1424             pstrcpy(buf1 + 1, sizeof(buf1) - 1, name);
1425             name = buf1;
1426         }
1427         info = ELF32_ST_INFO(sym_bind, sym_type);
1428         sym->c = add_elf_sym(symtab_section, value, size, info, other, sh_num, name);
1429     } else {
1430         esym = &((Elf32_Sym *)symtab_section->data)[sym->c];
1431         esym->st_value = value;
1432         esym->st_size = size;
1433         esym->st_shndx = sh_num;
1434         esym->st_other |= other;
1435     }
1436 }
1437
1438 static void put_extern_sym(Sym *sym, Section *section,
1439                            unsigned long value, unsigned long size)
1440 {
1441     put_extern_sym2(sym, section, value, size, 1);
1442 }
1443
1444 /* add a new relocation entry to symbol 'sym' in section 's' */
1445 static void greloc(Section *s, Sym *sym, unsigned long offset, int type)
1446 {
1447     if (!sym->c)
1448         put_extern_sym(sym, NULL, 0, 0);
1449     /* now we can add ELF relocation info */
1450     put_elf_reloc(symtab_section, s, offset, type, sym->c);
1451 }
1452
1453 static inline int isid(int c)
1454 {
1455     return (c >= 'a' && c <= 'z') ||
1456         (c >= 'A' && c <= 'Z') ||
1457         c == '_';
1458 }
1459
1460 static inline int isnum(int c)
1461 {
1462     return c >= '0' && c <= '9';
1463 }
1464
1465 static inline int isoct(int c)
1466 {
1467     return c >= '0' && c <= '7';
1468 }
1469
1470 static inline int toup(int c)
1471 {
1472     if (c >= 'a' && c <= 'z')
1473         return c - 'a' + 'A';
1474     else
1475         return c;
1476 }
1477
1478 static void strcat_vprintf(char *buf, int buf_size, const char *fmt, va_list ap)
1479 {
1480     int len;
1481     len = strlen(buf);
1482     vsnprintf(buf + len, buf_size - len, fmt, ap);
1483 }
1484
1485 static void strcat_printf(char *buf, int buf_size, const char *fmt, ...)
1486 {
1487     va_list ap;
1488     va_start(ap, fmt);
1489     strcat_vprintf(buf, buf_size, fmt, ap);
1490     va_end(ap);
1491 }
1492
1493 void error1(TCCState *s1, int is_warning, const char *fmt, va_list ap)
1494 {
1495     char buf[2048];
1496     BufferedFile **f;
1497
1498     buf[0] = '\0';
1499     if (file) {
1500         for(f = s1->include_stack; f < s1->include_stack_ptr; f++)
1501             strcat_printf(buf, sizeof(buf), "In file included from %s:%d:\n",
1502                           (*f)->filename, (*f)->line_num);
1503         if (file->line_num > 0) {
1504             strcat_printf(buf, sizeof(buf),
1505                           "%s:%d: ", file->filename, file->line_num);
1506         } else {
1507             strcat_printf(buf, sizeof(buf),
1508                           "%s: ", file->filename);
1509         }
1510     } else {
1511         strcat_printf(buf, sizeof(buf),
1512                       "tcc: ");
1513     }
1514     if (is_warning)
1515         strcat_printf(buf, sizeof(buf), "warning: ");
1516     strcat_vprintf(buf, sizeof(buf), fmt, ap);
1517
1518     if (!s1->error_func) {
1519         /* default case: stderr */
1520         fprintf(stderr, "%s\n", buf);
1521     } else {
1522         s1->error_func(s1->error_opaque, buf);
1523     }
1524     if (!is_warning || s1->warn_error)
1525         s1->nb_errors++;
1526 }
1527
1528 #ifdef LIBTCC
1529 void tcc_set_error_func(TCCState *s, void *error_opaque,
1530                         void (*error_func)(void *opaque, const char *msg))
1531 {
1532     s->error_opaque = error_opaque;
1533     s->error_func = error_func;
1534 }
1535 #endif
1536
1537 /* error without aborting current compilation */
1538 void error_noabort(const char *fmt, ...)
1539 {
1540     TCCState *s1 = tcc_state;
1541     va_list ap;
1542
1543     va_start(ap, fmt);
1544     error1(s1, 0, fmt, ap);
1545     va_end(ap);
1546 }
1547
1548 void error(const char *fmt, ...)
1549 {
1550     TCCState *s1 = tcc_state;
1551     va_list ap;
1552
1553     va_start(ap, fmt);
1554     error1(s1, 0, fmt, ap);
1555     va_end(ap);
1556     /* better than nothing: in some cases, we accept to handle errors */
1557     if (s1->error_set_jmp_enabled) {
1558         longjmp(s1->error_jmp_buf, 1);
1559     } else {
1560         /* XXX: eliminate this someday */
1561         exit(1);
1562     }
1563 }
1564
1565 void expect(const char *msg)
1566 {
1567     error("%s expected", msg);
1568 }
1569
1570 void warning(const char *fmt, ...)
1571 {
1572     TCCState *s1 = tcc_state;
1573     va_list ap;
1574
1575     if (s1->warn_none)
1576         return;
1577
1578     va_start(ap, fmt);
1579     error1(s1, 1, fmt, ap);
1580     va_end(ap);
1581 }
1582
1583 void skip(int c)
1584 {
1585     if (tok != c)
1586         error("'%c' expected", c);
1587     next();
1588 }
1589
1590 static void test_lvalue(void)
1591 {
1592     if (!(vtop->r & VT_LVAL))
1593         expect("lvalue");
1594 }
1595
1596 /* allocate a new token */
1597 static TokenSym *tok_alloc_new(TokenSym **pts, const char *str, int len)
1598 {
1599     TokenSym *ts, **ptable;
1600     int i;
1601
1602     if (tok_ident >= SYM_FIRST_ANOM)
1603         error("memory full");
1604
1605     /* expand token table if needed */
1606     i = tok_ident - TOK_IDENT;
1607     if ((i % TOK_ALLOC_INCR) == 0) {
1608         ptable = tcc_realloc(table_ident, (i + TOK_ALLOC_INCR) * sizeof(TokenSym *));
1609         if (!ptable)
1610             error("memory full");
1611         table_ident = ptable;
1612     }
1613
1614     ts = tcc_malloc(sizeof(TokenSym) + len);
1615     table_ident[i] = ts;
1616     ts->tok = tok_ident++;
1617     ts->sym_define = NULL;
1618     ts->sym_label = NULL;
1619     ts->sym_struct = NULL;
1620     ts->sym_identifier = NULL;
1621     ts->len = len;
1622     ts->hash_next = NULL;
1623     memcpy(ts->str, str, len);
1624     ts->str[len] = '\0';
1625     *pts = ts;
1626     return ts;
1627 }
1628
1629 #define TOK_HASH_INIT 1
1630 #define TOK_HASH_FUNC(h, c) ((h) * 263 + (c))
1631
1632 /* find a token and add it if not found */
1633 static TokenSym *tok_alloc(const char *str, int len)
1634 {
1635     TokenSym *ts, **pts;
1636     int i;
1637     unsigned int h;
1638
1639     h = TOK_HASH_INIT;
1640     for(i=0;i<len;i++)
1641         h = TOK_HASH_FUNC(h, ((unsigned char *)str)[i]);
1642     h &= (TOK_HASH_SIZE - 1);
1643
1644     pts = &hash_ident[h];
1645     for(;;) {
1646         ts = *pts;
1647         if (!ts)
1648             break;
1649         if (ts->len == len && !memcmp(ts->str, str, len))
1650             return ts;
1651         pts = &(ts->hash_next);
1652     }
1653     return tok_alloc_new(pts, str, len);
1654 }
1655
1656 /* CString handling */
1657
1658 static void cstr_realloc(CString *cstr, int new_size)
1659 {
1660     int size;
1661     void *data;
1662
1663     size = cstr->size_allocated;
1664     if (size == 0)
1665         size = 8; /* no need to allocate a too small first string */
1666     while (size < new_size)
1667         size = size * 2;
1668     data = tcc_realloc(cstr->data_allocated, size);
1669     if (!data)
1670         error("memory full");
1671     cstr->data_allocated = data;
1672     cstr->size_allocated = size;
1673     cstr->data = data;
1674 }
1675
1676 /* add a byte */
1677 static inline void cstr_ccat(CString *cstr, int ch)
1678 {
1679     int size;
1680     size = cstr->size + 1;
1681     if (size > cstr->size_allocated)
1682         cstr_realloc(cstr, size);
1683     ((unsigned char *)cstr->data)[size - 1] = ch;
1684     cstr->size = size;
1685 }
1686
1687 static void cstr_cat(CString *cstr, const char *str)
1688 {
1689     int c;
1690     for(;;) {
1691         c = *str;
1692         if (c == '\0')
1693             break;
1694         cstr_ccat(cstr, c);
1695         str++;
1696     }
1697 }
1698
1699 /* add a wide char */
1700 static void cstr_wccat(CString *cstr, int ch)
1701 {
1702     int size;
1703     size = cstr->size + sizeof(nwchar_t);
1704     if (size > cstr->size_allocated)
1705         cstr_realloc(cstr, size);
1706     *(nwchar_t *)(((unsigned char *)cstr->data) + size - sizeof(nwchar_t)) = ch;
1707     cstr->size = size;
1708 }
1709
1710 static void cstr_new(CString *cstr)
1711 {
1712     memset(cstr, 0, sizeof(CString));
1713 }
1714
1715 /* free string and reset it to NULL */
1716 static void cstr_free(CString *cstr)
1717 {
1718     tcc_free(cstr->data_allocated);
1719     cstr_new(cstr);
1720 }
1721
1722 #define cstr_reset(cstr) cstr_free(cstr)
1723
1724 /* XXX: unicode ? */
1725 static void add_char(CString *cstr, int c)
1726 {
1727     if (c == '\'' || c == '\"' || c == '\\') {
1728         /* XXX: could be more precise if char or string */
1729         cstr_ccat(cstr, '\\');
1730     }
1731     if (c >= 32 && c <= 126) {
1732         cstr_ccat(cstr, c);
1733     } else {
1734         cstr_ccat(cstr, '\\');
1735         if (c == '\n') {
1736             cstr_ccat(cstr, 'n');
1737         } else {
1738             cstr_ccat(cstr, '0' + ((c >> 6) & 7));
1739             cstr_ccat(cstr, '0' + ((c >> 3) & 7));
1740             cstr_ccat(cstr, '0' + (c & 7));
1741         }
1742     }
1743 }
1744
1745 /* XXX: buffer overflow */
1746 /* XXX: float tokens */
1747 char *get_tok_str(int v, CValue *cv)
1748 {
1749     static char buf[STRING_MAX_SIZE + 1];
1750     static CString cstr_buf;
1751     CString *cstr;
1752     unsigned char *q;
1753     char *p;
1754     int i, len;
1755
1756     /* NOTE: to go faster, we give a fixed buffer for small strings */
1757     cstr_reset(&cstr_buf);
1758     cstr_buf.data = buf;
1759     cstr_buf.size_allocated = sizeof(buf);
1760     p = buf;
1761
1762     switch(v) {
1763     case TOK_CINT:
1764     case TOK_CUINT:
1765         /* XXX: not quite exact, but only useful for testing */
1766         sprintf(p, "%u", cv->ui);
1767         break;
1768     case TOK_CLLONG:
1769     case TOK_CULLONG:
1770         /* XXX: not quite exact, but only useful for testing  */
1771         sprintf(p, "%Lu", cv->ull);
1772         break;
1773     case TOK_CCHAR:
1774     case TOK_LCHAR:
1775         cstr_ccat(&cstr_buf, '\'');
1776         add_char(&cstr_buf, cv->i);
1777         cstr_ccat(&cstr_buf, '\'');
1778         cstr_ccat(&cstr_buf, '\0');
1779         break;
1780     case TOK_PPNUM:
1781         cstr = cv->cstr;
1782         len = cstr->size - 1;
1783         for(i=0;i<len;i++)
1784             add_char(&cstr_buf, ((unsigned char *)cstr->data)[i]);
1785         cstr_ccat(&cstr_buf, '\0');
1786         break;
1787     case TOK_STR:
1788     case TOK_LSTR:
1789         cstr = cv->cstr;
1790         cstr_ccat(&cstr_buf, '\"');
1791         if (v == TOK_STR) {
1792             len = cstr->size - 1;
1793             for(i=0;i<len;i++)
1794                 add_char(&cstr_buf, ((unsigned char *)cstr->data)[i]);
1795         } else {
1796             len = (cstr->size / sizeof(nwchar_t)) - 1;
1797             for(i=0;i<len;i++)
1798                 add_char(&cstr_buf, ((nwchar_t *)cstr->data)[i]);
1799         }
1800         cstr_ccat(&cstr_buf, '\"');
1801         cstr_ccat(&cstr_buf, '\0');
1802         break;
1803     case TOK_LT:
1804         v = '<';
1805         goto addv;
1806     case TOK_GT:
1807         v = '>';
1808         goto addv;
1809     case TOK_DOTS:
1810         return strcpy(p, "...");
1811     case TOK_A_SHL:
1812         return strcpy(p, "<<=");
1813     case TOK_A_SAR:
1814         return strcpy(p, ">>=");
1815     default:
1816         if (v < TOK_IDENT) {
1817             /* search in two bytes table */
1818             q = tok_two_chars;
1819             while (*q) {
1820                 if (q[2] == v) {
1821                     *p++ = q[0];
1822                     *p++ = q[1];
1823                     *p = '\0';
1824                     return buf;
1825                 }
1826                 q += 3;
1827             }
1828         addv:
1829             *p++ = v;
1830             *p = '\0';
1831         } else if (v < tok_ident) {
1832             return table_ident[v - TOK_IDENT]->str;
1833         } else if (v >= SYM_FIRST_ANOM) {
1834             /* special name for anonymous symbol */
1835             sprintf(p, "L.%u", v - SYM_FIRST_ANOM);
1836         } else {
1837             /* should never happen */
1838             return NULL;
1839         }
1840         break;
1841     }
1842     return cstr_buf.data;
1843 }
1844
1845 /* push, without hashing */
1846 static Sym *sym_push2(Sym **ps, int v, int t, int c)
1847 {
1848     Sym *s;
1849     s = sym_malloc();
1850     s->v = v;
1851     s->type.t = t;
1852     s->c = c;
1853     s->next = NULL;
1854     /* add in stack */
1855     s->prev = *ps;
1856     *ps = s;
1857     return s;
1858 }
1859
1860 /* find a symbol and return its associated structure. 's' is the top
1861    of the symbol stack */
1862 static Sym *sym_find2(Sym *s, int v)
1863 {
1864     while (s) {
1865         if (s->v == v)
1866             return s;
1867         s = s->prev;
1868     }
1869     return NULL;
1870 }
1871
1872 /* structure lookup */
1873 static inline Sym *struct_find(int v)
1874 {
1875     v -= TOK_IDENT;
1876     if ((unsigned)v >= (unsigned)(tok_ident - TOK_IDENT))
1877         return NULL;
1878     return table_ident[v]->sym_struct;
1879 }
1880
1881 /* find an identifier */
1882 static inline Sym *sym_find(int v)
1883 {
1884     v -= TOK_IDENT;
1885     if ((unsigned)v >= (unsigned)(tok_ident - TOK_IDENT))
1886         return NULL;
1887     return table_ident[v]->sym_identifier;
1888 }
1889
1890 /* push a given symbol on the symbol stack */
1891 static Sym *sym_push(int v, CType *type, int r, int c)
1892 {
1893     Sym *s, **ps;
1894     TokenSym *ts;
1895
1896     if (local_stack)
1897         ps = &local_stack;
1898     else
1899         ps = &global_stack;
1900     s = sym_push2(ps, v, type->t, c);
1901     s->type.ref = type->ref;
1902     s->r = r;
1903     /* don't record fields or anonymous symbols */
1904     /* XXX: simplify */
1905     if (!(v & SYM_FIELD) && (v & ~SYM_STRUCT) < SYM_FIRST_ANOM) {
1906         /* record symbol in token array */
1907         ts = table_ident[(v & ~SYM_STRUCT) - TOK_IDENT];
1908         if (v & SYM_STRUCT)
1909             ps = &ts->sym_struct;
1910         else
1911             ps = &ts->sym_identifier;
1912         s->prev_tok = *ps;
1913         *ps = s;
1914     }
1915     return s;
1916 }
1917
1918 /* push a global identifier */
1919 static Sym *global_identifier_push(int v, int t, int c)
1920 {
1921     Sym *s, **ps;
1922     s = sym_push2(&global_stack, v, t, c);
1923     /* don't record anonymous symbol */
1924     if (v < SYM_FIRST_ANOM) {
1925         ps = &table_ident[v - TOK_IDENT]->sym_identifier;
1926         /* modify the top most local identifier, so that
1927            sym_identifier will point to 's' when popped */
1928         while (*ps != NULL)
1929             ps = &(*ps)->prev_tok;
1930         s->prev_tok = NULL;
1931         *ps = s;
1932     }
1933     return s;
1934 }
1935
1936 /* pop symbols until top reaches 'b' */
1937 static void sym_pop(Sym **ptop, Sym *b)
1938 {
1939     Sym *s, *ss, **ps;
1940     TokenSym *ts;
1941     int v;
1942
1943     s = *ptop;
1944     while(s != b) {
1945         ss = s->prev;
1946         v = s->v;
1947         /* remove symbol in token array */
1948         /* XXX: simplify */
1949         if (!(v & SYM_FIELD) && (v & ~SYM_STRUCT) < SYM_FIRST_ANOM) {
1950             ts = table_ident[(v & ~SYM_STRUCT) - TOK_IDENT];
1951             if (v & SYM_STRUCT)
1952                 ps = &ts->sym_struct;
1953             else
1954                 ps = &ts->sym_identifier;
1955             *ps = s->prev_tok;
1956         }
1957         sym_free(s);
1958         s = ss;
1959     }
1960     *ptop = b;
1961 }
1962
1963 /* I/O layer */
1964
1965 BufferedFile *tcc_open(TCCState *s1, const char *filename)
1966 {
1967     int fd;
1968     BufferedFile *bf;
1969
1970     if (strcmp(filename, "-") == 0)
1971         fd = 0, filename = "stdin";
1972     else
1973         fd = open(filename, O_RDONLY | O_BINARY);
1974     if ((verbose == 2 && fd >= 0) || verbose == 3)
1975         printf("%s %*s%s\n", fd < 0 ? "nf":"->",
1976                (s1->include_stack_ptr - s1->include_stack), "", filename);
1977     if (fd < 0)
1978         return NULL;
1979     bf = tcc_malloc(sizeof(BufferedFile));
1980     bf->fd = fd;
1981     bf->buf_ptr = bf->buffer;
1982     bf->buf_end = bf->buffer;
1983     bf->buffer[0] = CH_EOB; /* put eob symbol */
1984     pstrcpy(bf->filename, sizeof(bf->filename), filename);
1985 #ifdef _WIN32
1986     normalize_slashes(bf->filename);
1987 #endif
1988     bf->line_num = 1;
1989     bf->ifndef_macro = 0;
1990     bf->ifdef_stack_ptr = s1->ifdef_stack_ptr;
1991     //    printf("opening '%s'\n", filename);
1992     return bf;
1993 }
1994
1995 void tcc_close(BufferedFile *bf)
1996 {
1997     total_lines += bf->line_num;
1998     close(bf->fd);
1999     tcc_free(bf);
2000 }
2001
2002 /* fill input buffer and peek next char */
2003 static int tcc_peekc_slow(BufferedFile *bf)
2004 {
2005     int len;
2006     /* only tries to read if really end of buffer */
2007     if (bf->buf_ptr >= bf->buf_end) {
2008         if (bf->fd != -1) {
2009 #if defined(PARSE_DEBUG)
2010             len = 8;
2011 #else
2012             len = IO_BUF_SIZE;
2013 #endif
2014             len = read(bf->fd, bf->buffer, len);
2015             if (len < 0)
2016                 len = 0;
2017         } else {
2018             len = 0;
2019         }
2020         total_bytes += len;
2021         bf->buf_ptr = bf->buffer;
2022         bf->buf_end = bf->buffer + len;
2023         *bf->buf_end = CH_EOB;
2024     }
2025     if (bf->buf_ptr < bf->buf_end) {
2026         return bf->buf_ptr[0];
2027     } else {
2028         bf->buf_ptr = bf->buf_end;
2029         return CH_EOF;
2030     }
2031 }
2032
2033 /* return the current character, handling end of block if necessary
2034    (but not stray) */
2035 static int handle_eob(void)
2036 {
2037     return tcc_peekc_slow(file);
2038 }
2039
2040 /* read next char from current input file and handle end of input buffer */
2041 static inline void inp(void)
2042 {
2043     ch = *(++(file->buf_ptr));
2044     /* end of buffer/file handling */
2045     if (ch == CH_EOB)
2046         ch = handle_eob();
2047 }
2048
2049 /* handle '\[\r]\n' */
2050 static int handle_stray_noerror(void)
2051 {
2052     while (ch == '\\') {
2053         inp();
2054         if (ch == '\n') {
2055             file->line_num++;
2056             inp();
2057         } else if (ch == '\r') {
2058             inp();
2059             if (ch != '\n')
2060                 goto fail;
2061             file->line_num++;
2062             inp();
2063         } else {
2064         fail:
2065             return 1;
2066         }
2067     }
2068     return 0;
2069 }
2070
2071 static void handle_stray(void)
2072 {
2073     if (handle_stray_noerror())
2074         error("stray '\\' in program");
2075 }
2076
2077 /* skip the stray and handle the \\n case. Output an error if
2078    incorrect char after the stray */
2079 static int handle_stray1(uint8_t *p)
2080 {
2081     int c;
2082
2083     if (p >= file->buf_end) {
2084         file->buf_ptr = p;
2085         c = handle_eob();
2086         p = file->buf_ptr;
2087         if (c == '\\')
2088             goto parse_stray;
2089     } else {
2090     parse_stray:
2091         file->buf_ptr = p;
2092         ch = *p;
2093         handle_stray();
2094         p = file->buf_ptr;
2095         c = *p;
2096     }
2097     return c;
2098 }
2099
2100 /* handle just the EOB case, but not stray */
2101 #define PEEKC_EOB(c, p)\
2102 {\
2103     p++;\
2104     c = *p;\
2105     if (c == '\\') {\
2106         file->buf_ptr = p;\
2107         c = handle_eob();\
2108         p = file->buf_ptr;\
2109     }\
2110 }
2111
2112 /* handle the complicated stray case */
2113 #define PEEKC(c, p)\
2114 {\
2115     p++;\
2116     c = *p;\
2117     if (c == '\\') {\
2118         c = handle_stray1(p);\
2119         p = file->buf_ptr;\
2120     }\
2121 }
2122
2123 /* input with '\[\r]\n' handling. Note that this function cannot
2124    handle other characters after '\', so you cannot call it inside
2125    strings or comments */
2126 static void minp(void)
2127 {
2128     inp();
2129     if (ch == '\\')
2130         handle_stray();
2131 }
2132
2133
2134 /* single line C++ comments */
2135 static uint8_t *parse_line_comment(uint8_t *p)
2136 {
2137     int c;
2138
2139     p++;
2140     for(;;) {
2141         c = *p;
2142     redo:
2143         if (c == '\n' || c == CH_EOF) {
2144             break;
2145         } else if (c == '\\') {
2146             file->buf_ptr = p;
2147             c = handle_eob();
2148             p = file->buf_ptr;
2149             if (c == '\\') {
2150                 PEEKC_EOB(c, p);
2151                 if (c == '\n') {
2152                     file->line_num++;
2153                     PEEKC_EOB(c, p);
2154                 } else if (c == '\r') {
2155                     PEEKC_EOB(c, p);
2156                     if (c == '\n') {
2157                         file->line_num++;
2158                         PEEKC_EOB(c, p);
2159                     }
2160                 }
2161             } else {
2162                 goto redo;
2163             }
2164         } else {
2165             p++;
2166         }
2167     }
2168     return p;
2169 }
2170
2171 /* C comments */
2172 static uint8_t *parse_comment(uint8_t *p)
2173 {
2174     int c;
2175
2176     p++;
2177     for(;;) {
2178         /* fast skip loop */
2179         for(;;) {
2180             c = *p;
2181             if (c == '\n' || c == '*' || c == '\\')
2182                 break;
2183             p++;
2184             c = *p;
2185             if (c == '\n' || c == '*' || c == '\\')
2186                 break;
2187             p++;
2188         }
2189         /* now we can handle all the cases */
2190         if (c == '\n') {
2191             file->line_num++;
2192             p++;
2193         } else if (c == '*') {
2194             p++;
2195             for(;;) {
2196                 c = *p;
2197                 if (c == '*') {
2198                     p++;
2199                 } else if (c == '/') {
2200                     goto end_of_comment;
2201                 } else if (c == '\\') {
2202                     file->buf_ptr = p;
2203                     c = handle_eob();
2204                     p = file->buf_ptr;
2205                     if (c == '\\') {
2206                         /* skip '\[\r]\n', otherwise just skip the stray */
2207                         while (c == '\\') {
2208                             PEEKC_EOB(c, p);
2209                             if (c == '\n') {
2210                                 file->line_num++;
2211                                 PEEKC_EOB(c, p);
2212                             } else if (c == '\r') {
2213                                 PEEKC_EOB(c, p);
2214                                 if (c == '\n') {
2215                                     file->line_num++;
2216                                     PEEKC_EOB(c, p);
2217                                 }
2218                             } else {
2219                                 goto after_star;
2220                             }
2221                         }
2222                     }
2223                 } else {
2224                     break;
2225                 }
2226             }
2227         after_star: ;
2228         } else {
2229             /* stray, eob or eof */
2230             file->buf_ptr = p;
2231             c = handle_eob();
2232             p = file->buf_ptr;
2233             if (c == CH_EOF) {
2234                 error("unexpected end of file in comment");
2235             } else if (c == '\\') {
2236                 p++;
2237             }
2238         }
2239     }
2240  end_of_comment:
2241     p++;
2242     return p;
2243 }
2244
2245 #define cinp minp
2246
2247 /* space exlcuding newline */
2248 static inline int is_space(int ch)
2249 {
2250     return ch == ' ' || ch == '\t' || ch == '\v' || ch == '\f' || ch == '\r';
2251 }
2252
2253 static inline void skip_spaces(void)
2254 {
2255     while (is_space(ch))
2256         cinp();
2257 }
2258
2259 /* parse a string without interpreting escapes */
2260 static uint8_t *parse_pp_string(uint8_t *p,
2261                                 int sep, CString *str)
2262 {
2263     int c;
2264     p++;
2265     for(;;) {
2266         c = *p;
2267         if (c == sep) {
2268             break;
2269         } else if (c == '\\') {
2270             file->buf_ptr = p;
2271             c = handle_eob();
2272             p = file->buf_ptr;
2273             if (c == CH_EOF) {
2274             unterminated_string:
2275                 /* XXX: indicate line number of start of string */
2276                 error("missing terminating %c character", sep);
2277             } else if (c == '\\') {
2278                 /* escape : just skip \[\r]\n */
2279                 PEEKC_EOB(c, p);
2280                 if (c == '\n') {
2281                     file->line_num++;
2282                     p++;
2283                 } else if (c == '\r') {
2284                     PEEKC_EOB(c, p);
2285                     if (c != '\n')
2286                         expect("'\n' after '\r'");
2287                     file->line_num++;
2288                     p++;
2289                 } else if (c == CH_EOF) {
2290                     goto unterminated_string;
2291                 } else {
2292                     if (str) {
2293                         cstr_ccat(str, '\\');
2294                         cstr_ccat(str, c);
2295                     }
2296                     p++;
2297                 }
2298             }
2299         } else if (c == '\n') {
2300             file->line_num++;
2301             goto add_char;
2302         } else if (c == '\r') {
2303             PEEKC_EOB(c, p);
2304             if (c != '\n') {
2305                 if (str)
2306                     cstr_ccat(str, '\r');
2307             } else {
2308                 file->line_num++;
2309                 goto add_char;
2310             }
2311         } else {
2312         add_char:
2313             if (str)
2314                 cstr_ccat(str, c);
2315             p++;
2316         }
2317     }
2318     p++;
2319     return p;
2320 }
2321
2322 /* skip block of text until #else, #elif or #endif. skip also pairs of
2323    #if/#endif */
2324 void preprocess_skip(void)
2325 {
2326     int a, start_of_line, c, in_warn_or_error;
2327     uint8_t *p;
2328
2329     p = file->buf_ptr;
2330     a = 0;
2331 redo_start:
2332     start_of_line = 1;
2333     in_warn_or_error = 0;
2334     for(;;) {
2335     redo_no_start:
2336         c = *p;
2337         switch(c) {
2338         case ' ':
2339         case '\t':
2340         case '\f':
2341         case '\v':
2342         case '\r':
2343             p++;
2344             goto redo_no_start;
2345         case '\n':
2346             file->line_num++;
2347             p++;
2348             goto redo_start;
2349         case '\\':
2350             file->buf_ptr = p;
2351             c = handle_eob();
2352             if (c == CH_EOF) {
2353                 expect("#endif");
2354             } else if (c == '\\') {
2355                 ch = file->buf_ptr[0];
2356                 handle_stray_noerror();
2357             }
2358             p = file->buf_ptr;
2359             goto redo_no_start;
2360         /* skip strings */
2361         case '\"':
2362         case '\'':
2363             if (in_warn_or_error)
2364                 goto _default;
2365             p = parse_pp_string(p, c, NULL);
2366             break;
2367         /* skip comments */
2368         case '/':
2369             if (in_warn_or_error)
2370                 goto _default;
2371             file->buf_ptr = p;
2372             ch = *p;
2373             minp();
2374             p = file->buf_ptr;
2375             if (ch == '*') {
2376                 p = parse_comment(p);
2377             } else if (ch == '/') {
2378                 p = parse_line_comment(p);
2379             }
2380             break;
2381         case '#':
2382             p++;
2383             if (start_of_line) {
2384                 file->buf_ptr = p;
2385                 next_nomacro();
2386                 p = file->buf_ptr;
2387                 if (a == 0 &&
2388                     (tok == TOK_ELSE || tok == TOK_ELIF || tok == TOK_ENDIF))
2389                     goto the_end;
2390                 if (tok == TOK_IF || tok == TOK_IFDEF || tok == TOK_IFNDEF)
2391                     a++;
2392                 else if (tok == TOK_ENDIF)
2393                     a--;
2394                 else if( tok == TOK_ERROR || tok == TOK_WARNING)
2395                     in_warn_or_error = 1;
2396             }
2397             break;
2398 _default:
2399         default:
2400             p++;
2401             break;
2402         }
2403         start_of_line = 0;
2404     }
2405  the_end: ;
2406     file->buf_ptr = p;
2407 }
2408
2409 /* ParseState handling */
2410
2411 /* XXX: currently, no include file info is stored. Thus, we cannot display
2412    accurate messages if the function or data definition spans multiple
2413    files */
2414
2415 /* save current parse state in 's' */
2416 void save_parse_state(ParseState *s)
2417 {
2418     s->line_num = file->line_num;
2419     s->macro_ptr = macro_ptr;
2420     s->tok = tok;
2421     s->tokc = tokc;
2422 }
2423
2424 /* restore parse state from 's' */
2425 void restore_parse_state(ParseState *s)
2426 {
2427     file->line_num = s->line_num;
2428     macro_ptr = s->macro_ptr;
2429     tok = s->tok;
2430     tokc = s->tokc;
2431 }
2432
2433 /* return the number of additional 'ints' necessary to store the
2434    token */
2435 static inline int tok_ext_size(int t)
2436 {
2437     switch(t) {
2438         /* 4 bytes */
2439     case TOK_CINT:
2440     case TOK_CUINT:
2441     case TOK_CCHAR:
2442     case TOK_LCHAR:
2443     case TOK_CFLOAT:
2444     case TOK_LINENUM:
2445         return 1;
2446     case TOK_STR:
2447     case TOK_LSTR:
2448     case TOK_PPNUM:
2449         error("unsupported token");
2450         return 1;
2451     case TOK_CDOUBLE:
2452     case TOK_CLLONG:
2453     case TOK_CULLONG:
2454         return 2;
2455     case TOK_CLDOUBLE:
2456         return LDOUBLE_SIZE / 4;
2457     default:
2458         return 0;
2459     }
2460 }
2461
2462 /* token string handling */
2463
2464 static inline void tok_str_new(TokenString *s)
2465 {
2466     s->str = NULL;
2467     s->len = 0;
2468     s->allocated_len = 0;
2469     s->last_line_num = -1;
2470 }
2471
2472 static void tok_str_free(int *str)
2473 {
2474     tcc_free(str);
2475 }
2476
2477 static int *tok_str_realloc(TokenString *s)
2478 {
2479     int *str, len;
2480
2481     if (s->allocated_len == 0) {
2482         len = 8;
2483     } else {
2484         len = s->allocated_len * 2;
2485     }
2486     str = tcc_realloc(s->str, len * sizeof(int));
2487     if (!str)
2488         error("memory full");
2489     s->allocated_len = len;
2490     s->str = str;
2491     return str;
2492 }
2493
2494 static void tok_str_add(TokenString *s, int t)
2495 {
2496     int len, *str;
2497
2498     len = s->len;
2499     str = s->str;
2500     if (len >= s->allocated_len)
2501         str = tok_str_realloc(s);
2502     str[len++] = t;
2503     s->len = len;
2504 }
2505
2506 static void tok_str_add2(TokenString *s, int t, CValue *cv)
2507 {
2508     int len, *str;
2509
2510     len = s->len;
2511     str = s->str;
2512
2513     /* allocate space for worst case */
2514     if (len + TOK_MAX_SIZE > s->allocated_len)
2515         str = tok_str_realloc(s);
2516     str[len++] = t;
2517     switch(t) {
2518     case TOK_CINT:
2519     case TOK_CUINT:
2520     case TOK_CCHAR:
2521     case TOK_LCHAR:
2522     case TOK_CFLOAT:
2523     case TOK_LINENUM:
2524         str[len++] = cv->tab[0];
2525         break;
2526     case TOK_PPNUM:
2527     case TOK_STR:
2528     case TOK_LSTR:
2529         {
2530             int nb_words;
2531             CString *cstr;
2532
2533             nb_words = (sizeof(CString) + cv->cstr->size + 3) >> 2;
2534             while ((len + nb_words) > s->allocated_len)
2535                 str = tok_str_realloc(s);
2536             cstr = (CString *)(str + len);
2537             cstr->data = NULL;
2538             cstr->size = cv->cstr->size;
2539             cstr->data_allocated = NULL;
2540             cstr->size_allocated = cstr->size;
2541             memcpy((char *)cstr + sizeof(CString),
2542                    cv->cstr->data, cstr->size);
2543             len += nb_words;
2544         }
2545         break;
2546     case TOK_CDOUBLE:
2547     case TOK_CLLONG:
2548     case TOK_CULLONG:
2549 #if LDOUBLE_SIZE == 8
2550     case TOK_CLDOUBLE:
2551 #endif
2552         str[len++] = cv->tab[0];
2553         str[len++] = cv->tab[1];
2554         break;
2555 #if LDOUBLE_SIZE == 12
2556     case TOK_CLDOUBLE:
2557         str[len++] = cv->tab[0];
2558         str[len++] = cv->tab[1];
2559         str[len++] = cv->tab[2];
2560 #elif LDOUBLE_SIZE != 8
2561 #error add long double size support
2562 #endif
2563         break;
2564     default:
2565         break;
2566     }
2567     s->len = len;
2568 }
2569
2570 /* add the current parse token in token string 's' */
2571 static void tok_str_add_tok(TokenString *s)
2572 {
2573     CValue cval;
2574
2575     /* save line number info */
2576     if (file->line_num != s->last_line_num) {
2577         s->last_line_num = file->line_num;
2578         cval.i = s->last_line_num;
2579         tok_str_add2(s, TOK_LINENUM, &cval);
2580     }
2581     tok_str_add2(s, tok, &tokc);
2582 }
2583
2584 #if LDOUBLE_SIZE == 12
2585 #define LDOUBLE_GET(p, cv)                      \
2586         cv.tab[0] = p[0];                       \
2587         cv.tab[1] = p[1];                       \
2588         cv.tab[2] = p[2];
2589 #elif LDOUBLE_SIZE == 8
2590 #define LDOUBLE_GET(p, cv)                      \
2591         cv.tab[0] = p[0];                       \
2592         cv.tab[1] = p[1];
2593 #else
2594 #error add long double size support
2595 #endif
2596
2597
2598 /* get a token from an integer array and increment pointer
2599    accordingly. we code it as a macro to avoid pointer aliasing. */
2600 #define TOK_GET(t, p, cv)                       \
2601 {                                               \
2602     t = *p++;                                   \
2603     switch(t) {                                 \
2604     case TOK_CINT:                              \
2605     case TOK_CUINT:                             \
2606     case TOK_CCHAR:                             \
2607     case TOK_LCHAR:                             \
2608     case TOK_CFLOAT:                            \
2609     case TOK_LINENUM:                           \
2610         cv.tab[0] = *p++;                       \
2611         break;                                  \
2612     case TOK_STR:                               \
2613     case TOK_LSTR:                              \
2614     case TOK_PPNUM:                             \
2615         cv.cstr = (CString *)p;                 \
2616         cv.cstr->data = (char *)p + sizeof(CString);\
2617         p += (sizeof(CString) + cv.cstr->size + 3) >> 2;\
2618         break;                                  \
2619     case TOK_CDOUBLE:                           \
2620     case TOK_CLLONG:                            \
2621     case TOK_CULLONG:                           \
2622         cv.tab[0] = p[0];                       \
2623         cv.tab[1] = p[1];                       \
2624         p += 2;                                 \
2625         break;                                  \
2626     case TOK_CLDOUBLE:                          \
2627         LDOUBLE_GET(p, cv);                     \
2628         p += LDOUBLE_SIZE / 4;                  \
2629         break;                                  \
2630     default:                                    \
2631         break;                                  \
2632     }                                           \
2633 }
2634
2635 /* defines handling */
2636 static inline void define_push(int v, int macro_type, int *str, Sym *first_arg)
2637 {
2638     Sym *s;
2639
2640     s = sym_push2(&define_stack, v, macro_type, (int)str);
2641     s->next = first_arg;
2642     table_ident[v - TOK_IDENT]->sym_define = s;
2643 }
2644
2645 /* undefined a define symbol. Its name is just set to zero */
2646 static void define_undef(Sym *s)
2647 {
2648     int v;
2649     v = s->v;
2650     if (v >= TOK_IDENT && v < tok_ident)
2651         table_ident[v - TOK_IDENT]->sym_define = NULL;
2652     s->v = 0;
2653 }
2654
2655 static inline Sym *define_find(int v)
2656 {
2657     v -= TOK_IDENT;
2658     if ((unsigned)v >= (unsigned)(tok_ident - TOK_IDENT))
2659         return NULL;
2660     return table_ident[v]->sym_define;
2661 }
2662
2663 /* free define stack until top reaches 'b' */
2664 static void free_defines(Sym *b)
2665 {
2666     Sym *top, *top1;
2667     int v;
2668
2669     top = define_stack;
2670     while (top != b) {
2671         top1 = top->prev;
2672         /* do not free args or predefined defines */
2673         if (top->c)
2674             tok_str_free((int *)top->c);
2675         v = top->v;
2676         if (v >= TOK_IDENT && v < tok_ident)
2677             table_ident[v - TOK_IDENT]->sym_define = NULL;
2678         sym_free(top);
2679         top = top1;
2680     }
2681     define_stack = b;
2682 }
2683
2684 /* label lookup */
2685 static Sym *label_find(int v)
2686 {
2687     v -= TOK_IDENT;
2688     if ((unsigned)v >= (unsigned)(tok_ident - TOK_IDENT))
2689         return NULL;
2690     return table_ident[v]->sym_label;
2691 }
2692
2693 static Sym *label_push(Sym **ptop, int v, int flags)
2694 {
2695     Sym *s, **ps;
2696     s = sym_push2(ptop, v, 0, 0);
2697     s->r = flags;
2698     ps = &table_ident[v - TOK_IDENT]->sym_label;
2699     if (ptop == &global_label_stack) {
2700         /* modify the top most local identifier, so that
2701            sym_identifier will point to 's' when popped */
2702         while (*ps != NULL)
2703             ps = &(*ps)->prev_tok;
2704     }
2705     s->prev_tok = *ps;
2706     *ps = s;
2707     return s;
2708 }
2709
2710 /* pop labels until element last is reached. Look if any labels are
2711    undefined. Define symbols if '&&label' was used. */
2712 static void label_pop(Sym **ptop, Sym *slast)
2713 {
2714     Sym *s, *s1;
2715     for(s = *ptop; s != slast; s = s1) {
2716         s1 = s->prev;
2717         if (s->r == LABEL_DECLARED) {
2718             warning("label '%s' declared but not used", get_tok_str(s->v, NULL));
2719         } else if (s->r == LABEL_FORWARD) {
2720                 error("label '%s' used but not defined",
2721                       get_tok_str(s->v, NULL));
2722         } else {
2723             if (s->c) {
2724                 /* define corresponding symbol. A size of
2725                    1 is put. */
2726                 put_extern_sym(s, cur_text_section, (long)s->next, 1);
2727             }
2728         }
2729         /* remove label */
2730         table_ident[s->v - TOK_IDENT]->sym_label = s->prev_tok;
2731         sym_free(s);
2732     }
2733     *ptop = slast;
2734 }
2735
2736 /* eval an expression for #if/#elif */
2737 static int expr_preprocess(void)
2738 {
2739     int c, t;
2740     TokenString str;
2741
2742     tok_str_new(&str);
2743     while (tok != TOK_LINEFEED && tok != TOK_EOF) {
2744         next(); /* do macro subst */
2745         if (tok == TOK_DEFINED) {
2746             next_nomacro();
2747             t = tok;
2748             if (t == '(')
2749                 next_nomacro();
2750             c = define_find(tok) != 0;
2751             if (t == '(')
2752                 next_nomacro();
2753             tok = TOK_CINT;
2754             tokc.i = c;
2755         } else if (tok >= TOK_IDENT) {
2756             /* if undefined macro */
2757             tok = TOK_CINT;
2758             tokc.i = 0;
2759         }
2760         tok_str_add_tok(&str);
2761     }
2762     tok_str_add(&str, -1); /* simulate end of file */
2763     tok_str_add(&str, 0);
2764     /* now evaluate C constant expression */
2765     macro_ptr = str.str;
2766     next();
2767     c = expr_const();
2768     macro_ptr = NULL;
2769     tok_str_free(str.str);
2770     return c != 0;
2771 }
2772
2773 #if defined(PARSE_DEBUG) || defined(PP_DEBUG)
2774 static void tok_print(int *str)
2775 {
2776     int t;
2777     CValue cval;
2778
2779     while (1) {
2780         TOK_GET(t, str, cval);
2781         if (!t)
2782             break;
2783         printf(" %s", get_tok_str(t, &cval));
2784     }
2785     printf("\n");
2786 }
2787 #endif
2788
2789 /* parse after #define */
2790 static void parse_define(void)
2791 {
2792     Sym *s, *first, **ps;
2793     int v, t, varg, is_vaargs, c;
2794     TokenString str;
2795
2796     v = tok;
2797     if (v < TOK_IDENT)
2798         error("invalid macro name '%s'", get_tok_str(tok, &tokc));
2799     /* XXX: should check if same macro (ANSI) */
2800     first = NULL;
2801     t = MACRO_OBJ;
2802     /* '(' must be just after macro definition for MACRO_FUNC */
2803     c = file->buf_ptr[0];
2804     if (c == '\\')
2805         c = handle_stray1(file->buf_ptr);
2806     if (c == '(') {
2807         next_nomacro();
2808         next_nomacro();
2809         ps = &first;
2810         while (tok != ')') {
2811             varg = tok;
2812             next_nomacro();
2813             is_vaargs = 0;
2814             if (varg == TOK_DOTS) {
2815                 varg = TOK___VA_ARGS__;
2816                 is_vaargs = 1;
2817             } else if (tok == TOK_DOTS && gnu_ext) {
2818                 is_vaargs = 1;
2819                 next_nomacro();
2820             }
2821             if (varg < TOK_IDENT)
2822                 error("badly punctuated parameter list");
2823             s = sym_push2(&define_stack, varg | SYM_FIELD, is_vaargs, 0);
2824             *ps = s;
2825             ps = &s->next;
2826             if (tok != ',')
2827                 break;
2828             next_nomacro();
2829         }
2830         t = MACRO_FUNC;
2831     }
2832     tok_str_new(&str);
2833     next_nomacro();
2834     /* EOF testing necessary for '-D' handling */
2835     while (tok != TOK_LINEFEED && tok != TOK_EOF) {
2836         tok_str_add2(&str, tok, &tokc);
2837         next_nomacro();
2838     }
2839     tok_str_add(&str, 0);
2840 #ifdef PP_DEBUG
2841     printf("define %s %d: ", get_tok_str(v, NULL), t);
2842     tok_print(str.str);
2843 #endif
2844     define_push(v, t, str.str, first);
2845 }
2846
2847 static inline int hash_cached_include(int type, const char *filename)
2848 {
2849     const unsigned char *s;
2850     unsigned int h;
2851
2852     h = TOK_HASH_INIT;
2853     h = TOK_HASH_FUNC(h, type);
2854     s = filename;
2855     while (*s) {
2856         h = TOK_HASH_FUNC(h, *s);
2857         s++;
2858     }
2859     h &= (CACHED_INCLUDES_HASH_SIZE - 1);
2860     return h;
2861 }
2862
2863 /* XXX: use a token or a hash table to accelerate matching ? */
2864 static CachedInclude *search_cached_include(TCCState *s1,
2865                                             int type, const char *filename)
2866 {
2867     CachedInclude *e;
2868     int i, h;
2869     h = hash_cached_include(type, filename);
2870     i = s1->cached_includes_hash[h];
2871     for(;;) {
2872         if (i == 0)
2873             break;
2874         e = s1->cached_includes[i - 1];
2875         if (e->type == type && !strcmp(e->filename, filename))
2876             return e;
2877         i = e->hash_next;
2878     }
2879     return NULL;
2880 }
2881
2882 static inline void add_cached_include(TCCState *s1, int type,
2883                                       const char *filename, int ifndef_macro)
2884 {
2885     CachedInclude *e;
2886     int h;
2887
2888     if (search_cached_include(s1, type, filename))
2889         return;
2890 #ifdef INC_DEBUG
2891     printf("adding cached '%s' %s\n", filename, get_tok_str(ifndef_macro, NULL));
2892 #endif
2893     e = tcc_malloc(sizeof(CachedInclude) + strlen(filename));
2894     if (!e)
2895         return;
2896     e->type = type;
2897     strcpy(e->filename, filename);
2898     e->ifndef_macro = ifndef_macro;
2899     dynarray_add((void ***)&s1->cached_includes, &s1->nb_cached_includes, e);
2900     /* add in hash table */
2901     h = hash_cached_include(type, filename);
2902     e->hash_next = s1->cached_includes_hash[h];
2903     s1->cached_includes_hash[h] = s1->nb_cached_includes;
2904 }
2905
2906 static void pragma_parse(TCCState *s1)
2907 {
2908     int val;
2909
2910     next();
2911     if (tok == TOK_pack) {
2912         /*
2913           This may be:
2914           #pragma pack(1) // set
2915           #pragma pack() // reset to default
2916           #pragma pack(push,1) // push & set
2917           #pragma pack(pop) // restore previous
2918         */
2919         next();
2920         skip('(');
2921         if (tok == TOK_ASM_pop) {
2922             next();
2923             if (s1->pack_stack_ptr <= s1->pack_stack) {
2924             stk_error:
2925                 error("out of pack stack");
2926             }
2927             s1->pack_stack_ptr--;
2928         } else {
2929             val = 0;
2930             if (tok != ')') {
2931                 if (tok == TOK_ASM_push) {
2932                     next();
2933                     if (s1->pack_stack_ptr >= s1->pack_stack + PACK_STACK_SIZE - 1)
2934                         goto stk_error;
2935                     s1->pack_stack_ptr++;
2936                     skip(',');
2937                 }
2938                 if (tok != TOK_CINT) {
2939                 pack_error:
2940                     error("invalid pack pragma");
2941                 }
2942                 val = tokc.i;
2943                 if (val < 1 || val > 16 || (val & (val - 1)) != 0)
2944                     goto pack_error;
2945                 next();
2946             }
2947             *s1->pack_stack_ptr = val;
2948             skip(')');
2949         }
2950     }
2951 }
2952
2953 /* is_bof is true if first non space token at beginning of file */
2954 static void preprocess(int is_bof)
2955 {
2956     TCCState *s1 = tcc_state;
2957     int size, i, c, n, saved_parse_flags;
2958     char buf[1024], *q;
2959     char buf1[1024];
2960     BufferedFile *f;
2961     Sym *s;
2962     CachedInclude *e;
2963
2964     saved_parse_flags = parse_flags;
2965     parse_flags = PARSE_FLAG_PREPROCESS | PARSE_FLAG_TOK_NUM |
2966         PARSE_FLAG_LINEFEED;
2967     next_nomacro();
2968  redo:
2969     switch(tok) {
2970     case TOK_DEFINE:
2971         next_nomacro();
2972         parse_define();
2973         break;
2974     case TOK_UNDEF:
2975         next_nomacro();
2976         s = define_find(tok);
2977         /* undefine symbol by putting an invalid name */
2978         if (s)
2979             define_undef(s);
2980         break;
2981     case TOK_INCLUDE:
2982     case TOK_INCLUDE_NEXT:
2983         ch = file->buf_ptr[0];
2984         /* XXX: incorrect if comments : use next_nomacro with a special mode */
2985         skip_spaces();
2986         if (ch == '<') {
2987             c = '>';
2988             goto read_name;
2989         } else if (ch == '\"') {
2990             c = ch;
2991         read_name:
2992             inp();
2993             q = buf;
2994             while (ch != c && ch != '\n' && ch != CH_EOF) {
2995                 if ((q - buf) < sizeof(buf) - 1)
2996                     *q++ = ch;
2997                 if (ch == '\\') {
2998                     if (handle_stray_noerror() == 0)
2999                         --q;
3000                 } else
3001                     inp();
3002             }
3003             *q = '\0';
3004             minp();
3005 #if 0
3006             /* eat all spaces and comments after include */
3007             /* XXX: slightly incorrect */
3008             while (ch1 != '\n' && ch1 != CH_EOF)
3009                 inp();
3010 #endif
3011         } else {
3012             /* computed #include : either we have only strings or
3013                we have anything enclosed in '<>' */
3014             next();
3015             buf[0] = '\0';
3016             if (tok == TOK_STR) {
3017                 while (tok != TOK_LINEFEED) {
3018                     if (tok != TOK_STR) {
3019                     include_syntax:
3020                         error("'#include' expects \"FILENAME\" or <FILENAME>");
3021                     }
3022                     pstrcat(buf, sizeof(buf), (char *)tokc.cstr->data);
3023                     next();
3024                 }
3025                 c = '\"';
3026             } else {
3027                 int len;
3028                 while (tok != TOK_LINEFEED) {
3029                     pstrcat(buf, sizeof(buf), get_tok_str(tok, &tokc));
3030                     next();
3031                 }
3032                 len = strlen(buf);
3033                 /* check syntax and remove '<>' */
3034                 if (len < 2 || buf[0] != '<' || buf[len - 1] != '>')
3035                     goto include_syntax;
3036                 memmove(buf, buf + 1, len - 2);
3037                 buf[len - 2] = '\0';
3038                 c = '>';
3039             }
3040         }
3041
3042         e = search_cached_include(s1, c, buf);
3043         if (e && define_find(e->ifndef_macro)) {
3044             /* no need to parse the include because the 'ifndef macro'
3045                is defined */
3046 #ifdef INC_DEBUG
3047             printf("%s: skipping %s\n", file->filename, buf);
3048 #endif
3049         } else {
3050             if (s1->include_stack_ptr >= s1->include_stack + INCLUDE_STACK_SIZE)
3051                 error("#include recursion too deep");
3052             /* push current file in stack */
3053             /* XXX: fix current line init */
3054             *s1->include_stack_ptr++ = file;
3055             if (c == '\"') {
3056                 /* first search in current dir if "header.h" */
3057                 size = tcc_basename(file->filename) - file->filename;
3058                 if (size > sizeof(buf1) - 1)
3059                     size = sizeof(buf1) - 1;
3060                 memcpy(buf1, file->filename, size);
3061                 buf1[size] = '\0';
3062                 pstrcat(buf1, sizeof(buf1), buf);
3063                 f = tcc_open(s1, buf1);
3064                 if (f) {
3065                     if (tok == TOK_INCLUDE_NEXT)
3066                         tok = TOK_INCLUDE;
3067                     else
3068                         goto found;
3069                 }
3070             }
3071             /* now search in all the include paths */
3072             n = s1->nb_include_paths + s1->nb_sysinclude_paths;
3073             for(i = 0; i < n; i++) {
3074                 const char *path;
3075                 if (i < s1->nb_include_paths)
3076                     path = s1->include_paths[i];
3077                 else
3078                     path = s1->sysinclude_paths[i - s1->nb_include_paths];
3079                 pstrcpy(buf1, sizeof(buf1), path);
3080                 pstrcat(buf1, sizeof(buf1), "/");
3081                 pstrcat(buf1, sizeof(buf1), buf);
3082                 f = tcc_open(s1, buf1);
3083                 if (f) {
3084                     if (tok == TOK_INCLUDE_NEXT)
3085                         tok = TOK_INCLUDE;
3086                     else
3087                         goto found;
3088                 }
3089             }
3090             --s1->include_stack_ptr;
3091             error("include file '%s' not found", buf);
3092             break;
3093         found:
3094 #ifdef INC_DEBUG
3095             printf("%s: including %s\n", file->filename, buf1);
3096 #endif
3097             f->inc_type = c;
3098             pstrcpy(f->inc_filename, sizeof(f->inc_filename), buf);
3099             file = f;
3100             /* add include file debug info */
3101             if (do_debug) {
3102                 put_stabs(file->filename, N_BINCL, 0, 0, 0);
3103             }
3104             tok_flags |= TOK_FLAG_BOF | TOK_FLAG_BOL;
3105             ch = file->buf_ptr[0];
3106             goto the_end;
3107         }
3108         break;
3109     case TOK_IFNDEF:
3110         c = 1;
3111         goto do_ifdef;
3112     case TOK_IF:
3113         c = expr_preprocess();
3114         goto do_if;
3115     case TOK_IFDEF:
3116         c = 0;
3117     do_ifdef:
3118         next_nomacro();
3119         if (tok < TOK_IDENT)
3120             error("invalid argument for '#if%sdef'", c ? "n" : "");
3121         if (is_bof) {
3122             if (c) {
3123 #ifdef INC_DEBUG
3124                 printf("#ifndef %s\n", get_tok_str(tok, NULL));
3125 #endif
3126                 file->ifndef_macro = tok;
3127             }
3128         }
3129         c = (define_find(tok) != 0) ^ c;
3130     do_if:
3131         if (s1->ifdef_stack_ptr >= s1->ifdef_stack + IFDEF_STACK_SIZE)
3132             error("memory full");
3133         *s1->ifdef_stack_ptr++ = c;
3134         goto test_skip;
3135     case TOK_ELSE:
3136         if (s1->ifdef_stack_ptr == s1->ifdef_stack)
3137             error("#else without matching #if");
3138         if (s1->ifdef_stack_ptr[-1] & 2)
3139             error("#else after #else");
3140         c = (s1->ifdef_stack_ptr[-1] ^= 3);
3141         goto test_skip;
3142     case TOK_ELIF:
3143         if (s1->ifdef_stack_ptr == s1->ifdef_stack)
3144             error("#elif without matching #if");
3145         c = s1->ifdef_stack_ptr[-1];
3146         if (c > 1)
3147             error("#elif after #else");
3148         /* last #if/#elif expression was true: we skip */
3149         if (c == 1)
3150             goto skip;
3151         c = expr_preprocess();
3152         s1->ifdef_stack_ptr[-1] = c;
3153     test_skip:
3154         if (!(c & 1)) {
3155         skip:
3156             preprocess_skip();
3157             is_bof = 0;
3158             goto redo;
3159         }
3160         break;
3161     case TOK_ENDIF:
3162         if (s1->ifdef_stack_ptr <= file->ifdef_stack_ptr)
3163             error("#endif without matching #if");
3164         s1->ifdef_stack_ptr--;
3165         /* '#ifndef macro' was at the start of file. Now we check if
3166            an '#endif' is exactly at the end of file */
3167         if (file->ifndef_macro &&
3168             s1->ifdef_stack_ptr == file->ifdef_stack_ptr) {
3169             file->ifndef_macro_saved = file->ifndef_macro;
3170             /* need to set to zero to avoid false matches if another
3171                #ifndef at middle of file */
3172             file->ifndef_macro = 0;
3173             while (tok != TOK_LINEFEED)
3174                 next_nomacro();
3175             tok_flags |= TOK_FLAG_ENDIF;
3176             goto the_end;
3177         }
3178         break;
3179     case TOK_LINE:
3180         next();
3181         if (tok != TOK_CINT)
3182             error("#line");
3183         file->line_num = tokc.i - 1; /* the line number will be incremented after */
3184         next();
3185         if (tok != TOK_LINEFEED) {
3186             if (tok != TOK_STR)
3187                 error("#line");
3188             pstrcpy(file->filename, sizeof(file->filename),
3189                     (char *)tokc.cstr->data);
3190         }
3191         break;
3192     case TOK_ERROR:
3193     case TOK_WARNING:
3194         c = tok;
3195         ch = file->buf_ptr[0];
3196         skip_spaces();
3197         q = buf;
3198         while (ch != '\n' && ch != CH_EOF) {
3199             if ((q - buf) < sizeof(buf) - 1)
3200                 *q++ = ch;
3201             if (ch == '\\') {
3202                 if (handle_stray_noerror() == 0)
3203                     --q;
3204             } else
3205                 inp();
3206         }
3207         *q = '\0';
3208         if (c == TOK_ERROR)
3209             error("#error %s", buf);
3210         else
3211             warning("#warning %s", buf);
3212         break;
3213     case TOK_PRAGMA:
3214         pragma_parse(s1);
3215         break;
3216     default:
3217         if (tok == TOK_LINEFEED || tok == '!' || tok == TOK_CINT) {
3218             /* '!' is ignored to allow C scripts. numbers are ignored
3219                to emulate cpp behaviour */
3220         } else {
3221             if (!(saved_parse_flags & PARSE_FLAG_ASM_COMMENTS))
3222                 warning("Ignoring unknown preprocessing directive #%s", get_tok_str(tok, &tokc));
3223         }
3224         break;
3225     }
3226     /* ignore other preprocess commands or #! for C scripts */
3227     while (tok != TOK_LINEFEED)
3228         next_nomacro();
3229  the_end:
3230     parse_flags = saved_parse_flags;
3231 }
3232
3233 /* evaluate escape codes in a string. */
3234 static void parse_escape_string(CString *outstr, const uint8_t *buf, int is_long)
3235 {
3236     int c, n;
3237     const uint8_t *p;
3238
3239     p = buf;
3240     for(;;) {
3241         c = *p;
3242         if (c == '\0')
3243             break;
3244         if (c == '\\') {
3245             p++;
3246             /* escape */
3247             c = *p;
3248             switch(c) {
3249             case '0': case '1': case '2': case '3':
3250             case '4': case '5': case '6': case '7':
3251                 /* at most three octal digits */
3252                 n = c - '0';
3253                 p++;
3254                 c = *p;
3255                 if (isoct(c)) {
3256                     n = n * 8 + c - '0';
3257                     p++;
3258                     c = *p;
3259                     if (isoct(c)) {
3260                         n = n * 8 + c - '0';
3261                         p++;
3262                     }
3263                 }
3264                 c = n;
3265                 goto add_char_nonext;
3266             case 'x':
3267             case 'u':
3268             case 'U':
3269                 p++;
3270                 n = 0;
3271                 for(;;) {
3272                     c = *p;
3273                     if (c >= 'a' && c <= 'f')
3274                         c = c - 'a' + 10;
3275                     else if (c >= 'A' && c <= 'F')
3276                         c = c - 'A' + 10;
3277                     else if (isnum(c))
3278                         c = c - '0';
3279                     else
3280                         break;
3281                     n = n * 16 + c;
3282                     p++;
3283                 }
3284                 c = n;
3285                 goto add_char_nonext;
3286             case 'a':
3287                 c = '\a';
3288                 break;
3289             case 'b':
3290                 c = '\b';
3291                 break;
3292             case 'f':
3293                 c = '\f';
3294                 break;
3295             case 'n':
3296                 c = '\n';
3297                 break;
3298             case 'r':
3299                 c = '\r';
3300                 break;
3301             case 't':
3302                 c = '\t';
3303                 break;
3304             case 'v':
3305                 c = '\v';
3306                 break;
3307             case 'e':
3308                 if (!gnu_ext)
3309                     goto invalid_escape;
3310                 c = 27;
3311                 break;
3312             case '\'':
3313             case '\"':
3314             case '\\':
3315             case '?':
3316                 break;
3317             default:
3318             invalid_escape:
3319                 if (c >= '!' && c <= '~')
3320                     warning("unknown escape sequence: \'\\%c\'", c);
3321                 else
3322                     warning("unknown escape sequence: \'\\x%x\'", c);
3323                 break;
3324             }
3325         }
3326         p++;
3327     add_char_nonext:
3328         if (!is_long)
3329             cstr_ccat(outstr, c);
3330         else
3331             cstr_wccat(outstr, c);
3332     }
3333     /* add a trailing '\0' */
3334     if (!is_long)
3335         cstr_ccat(outstr, '\0');
3336     else
3337         cstr_wccat(outstr, '\0');
3338 }
3339
3340 /* we use 64 bit numbers */
3341 #define BN_SIZE 2
3342
3343 /* bn = (bn << shift) | or_val */
3344 void bn_lshift(unsigned int *bn, int shift, int or_val)
3345 {
3346     int i;
3347     unsigned int v;
3348     for(i=0;i<BN_SIZE;i++) {
3349         v = bn[i];
3350         bn[i] = (v << shift) | or_val;
3351         or_val = v >> (32 - shift);
3352     }
3353 }
3354
3355 void bn_zero(unsigned int *bn)
3356 {
3357     int i;
3358     for(i=0;i<BN_SIZE;i++) {
3359         bn[i] = 0;
3360     }
3361 }
3362
3363 /* parse number in null terminated string 'p' and return it in the
3364    current token */
3365 void parse_number(const char *p)
3366 {
3367     int b, t, shift, frac_bits, s, exp_val, ch;
3368     char *q;
3369     unsigned int bn[BN_SIZE];
3370     double d;
3371
3372     /* number */
3373     q = token_buf;
3374     ch = *p++;
3375     t = ch;
3376     ch = *p++;
3377     *q++ = t;
3378     b = 10;
3379     if (t == '.') {
3380         goto float_frac_parse;
3381     } else if (t == '0') {
3382         if (ch == 'x' || ch == 'X') {
3383             q--;
3384             ch = *p++;
3385             b = 16;
3386         } else if (tcc_ext && (ch == 'b' || ch == 'B')) {
3387             q--;
3388             ch = *p++;
3389             b = 2;
3390         }
3391     }
3392     /* parse all digits. cannot check octal numbers at this stage
3393        because of floating point constants */
3394     while (1) {
3395         if (ch >= 'a' && ch <= 'f')
3396             t = ch - 'a' + 10;
3397         else if (ch >= 'A' && ch <= 'F')
3398             t = ch - 'A' + 10;
3399         else if (isnum(ch))
3400             t = ch - '0';
3401         else
3402             break;
3403         if (t >= b)
3404             break;
3405         if (q >= token_buf + STRING_MAX_SIZE) {
3406         num_too_long:
3407             error("number too long");
3408         }
3409         *q++ = ch;
3410         ch = *p++;
3411     }
3412     if (ch == '.' ||
3413         ((ch == 'e' || ch == 'E') && b == 10) ||
3414         ((ch == 'p' || ch == 'P') && (b == 16 || b == 2))) {
3415         if (b != 10) {
3416             /* NOTE: strtox should support that for hexa numbers, but
3417                non ISOC99 libcs do not support it, so we prefer to do
3418                it by hand */
3419             /* hexadecimal or binary floats */
3420             /* XXX: handle overflows */
3421             *q = '\0';
3422             if (b == 16)
3423                 shift = 4;
3424             else
3425                 shift = 2;
3426             bn_zero(bn);
3427             q = token_buf;
3428             while (1) {
3429                 t = *q++;
3430                 if (t == '\0') {
3431                     break;
3432                 } else if (t >= 'a') {
3433                     t = t - 'a' + 10;
3434                 } else if (t >= 'A') {
3435                     t = t - 'A' + 10;
3436                 } else {
3437                     t = t - '0';
3438                 }
3439                 bn_lshift(bn, shift, t);
3440             }
3441             frac_bits = 0;
3442             if (ch == '.') {
3443                 ch = *p++;
3444                 while (1) {
3445                     t = ch;
3446                     if (t >= 'a' && t <= 'f') {
3447                         t = t - 'a' + 10;
3448                     } else if (t >= 'A' && t <= 'F') {
3449                         t = t - 'A' + 10;
3450                     } else if (t >= '0' && t <= '9') {
3451                         t = t - '0';
3452                     } else {
3453                         break;
3454                     }
3455                     if (t >= b)
3456                         error("invalid digit");
3457                     bn_lshift(bn, shift, t);
3458                     frac_bits += shift;
3459                     ch = *p++;
3460                 }
3461             }
3462             if (ch != 'p' && ch != 'P')
3463                 expect("exponent");
3464             ch = *p++;
3465             s = 1;
3466             exp_val = 0;
3467             if (ch == '+') {
3468                 ch = *p++;
3469             } else if (ch == '-') {
3470                 s = -1;
3471                 ch = *p++;
3472             }
3473             if (ch < '0' || ch > '9')
3474                 expect("exponent digits");
3475             while (ch >= '0' && ch <= '9') {
3476                 exp_val = exp_val * 10 + ch - '0';
3477                 ch = *p++;
3478             }
3479             exp_val = exp_val * s;
3480
3481             /* now we can generate the number */
3482             /* XXX: should patch directly float number */
3483             d = (double)bn[1] * 4294967296.0 + (double)bn[0];
3484             d = ldexp(d, exp_val - frac_bits);
3485             t = toup(ch);
3486             if (t == 'F') {
3487                 ch = *p++;
3488                 tok = TOK_CFLOAT;
3489                 /* float : should handle overflow */
3490                 tokc.f = (float)d;
3491             } else if (t == 'L') {
3492                 ch = *p++;
3493                 tok = TOK_CLDOUBLE;
3494                 /* XXX: not large enough */
3495                 tokc.ld = (long double)d;
3496             } else {
3497                 tok = TOK_CDOUBLE;
3498                 tokc.d = d;
3499             }
3500         } else {
3501             /* decimal floats */
3502             if (ch == '.') {
3503                 if (q >= token_buf + STRING_MAX_SIZE)
3504                     goto num_too_long;
3505                 *q++ = ch;
3506                 ch = *p++;
3507             float_frac_parse:
3508                 while (ch >= '0' && ch <= '9') {
3509                     if (q >= token_buf + STRING_MAX_SIZE)
3510                         goto num_too_long;
3511                     *q++ = ch;
3512                     ch = *p++;
3513                 }
3514             }
3515             if (ch == 'e' || ch == 'E') {
3516                 if (q >= token_buf + STRING_MAX_SIZE)
3517                     goto num_too_long;
3518                 *q++ = ch;
3519                 ch = *p++;
3520                 if (ch == '-' || ch == '+') {
3521                     if (q >= token_buf + STRING_MAX_SIZE)
3522                         goto num_too_long;
3523                     *q++ = ch;
3524                     ch = *p++;
3525                 }
3526                 if (ch < '0' || ch > '9')
3527                     expect("exponent digits");
3528                 while (ch >= '0' && ch <= '9') {
3529                     if (q >= token_buf + STRING_MAX_SIZE)
3530                         goto num_too_long;
3531                     *q++ = ch;
3532                     ch = *p++;
3533                 }
3534             }
3535             *q = '\0';
3536             t = toup(ch);
3537             errno = 0;
3538             if (t == 'F') {
3539                 ch = *p++;
3540                 tok = TOK_CFLOAT;
3541                 tokc.f = strtof(token_buf, NULL);
3542             } else if (t == 'L') {
3543                 ch = *p++;
3544                 tok = TOK_CLDOUBLE;
3545                 tokc.ld = strtold(token_buf, NULL);
3546             } else {
3547                 tok = TOK_CDOUBLE;
3548                 tokc.d = strtod(token_buf, NULL);
3549             }
3550         }
3551     } else {
3552         unsigned long long n, n1;
3553         int lcount, ucount;
3554
3555         /* integer number */
3556         *q = '\0';
3557         q = token_buf;
3558         if (b == 10 && *q == '0') {
3559             b = 8;
3560             q++;
3561         }
3562         n = 0;
3563         while(1) {
3564             t = *q++;
3565             /* no need for checks except for base 10 / 8 errors */
3566             if (t == '\0') {
3567                 break;
3568             } else if (t >= 'a') {
3569                 t = t - 'a' + 10;
3570             } else if (t >= 'A') {
3571                 t = t - 'A' + 10;
3572             } else {
3573                 t = t - '0';
3574                 if (t >= b)
3575                     error("invalid digit");
3576             }
3577             n1 = n;
3578             n = n * b + t;
3579             /* detect overflow */
3580             /* XXX: this test is not reliable */
3581             if (n < n1)
3582                 error("integer constant overflow");
3583         }
3584
3585         /* XXX: not exactly ANSI compliant */
3586         if ((n & 0xffffffff00000000LL) != 0) {
3587             if ((n >> 63) != 0)
3588                 tok = TOK_CULLONG;
3589             else
3590                 tok = TOK_CLLONG;
3591         } else if (n > 0x7fffffff) {
3592             tok = TOK_CUINT;
3593         } else {
3594             tok = TOK_CINT;
3595         }
3596         lcount = 0;
3597         ucount = 0;
3598         for(;;) {
3599             t = toup(ch);
3600             if (t == 'L') {
3601                 if (lcount >= 2)
3602                     error("three 'l's in integer constant");
3603                 lcount++;
3604                 if (lcount == 2) {
3605                     if (tok == TOK_CINT)
3606                         tok = TOK_CLLONG;
3607                     else if (tok == TOK_CUINT)
3608                         tok = TOK_CULLONG;
3609                 }
3610                 ch = *p++;
3611             } else if (t == 'U') {
3612                 if (ucount >= 1)
3613                     error("two 'u's in integer constant");
3614                 ucount++;
3615                 if (tok == TOK_CINT)
3616                     tok = TOK_CUINT;
3617                 else if (tok == TOK_CLLONG)
3618                     tok = TOK_CULLONG;
3619                 ch = *p++;
3620             } else {
3621                 break;
3622             }
3623         }
3624         if (tok == TOK_CINT || tok == TOK_CUINT)
3625             tokc.ui = n;
3626         else
3627             tokc.ull = n;
3628     }
3629 }
3630
3631
3632 #define PARSE2(c1, tok1, c2, tok2)              \
3633     case c1:                                    \
3634         PEEKC(c, p);                            \
3635         if (c == c2) {                          \
3636             p++;                                \
3637             tok = tok2;                         \
3638         } else {                                \
3639             tok = tok1;                         \
3640         }                                       \
3641         break;
3642
3643 /* return next token without macro substitution */
3644 static inline void next_nomacro1(void)
3645 {
3646     int t, c, is_long;
3647     TokenSym *ts;
3648     uint8_t *p, *p1;
3649     unsigned int h;
3650
3651     p = file->buf_ptr;
3652  redo_no_start:
3653     c = *p;
3654     switch(c) {
3655     case ' ':
3656     case '\t':
3657     case '\f':
3658     case '\v':
3659     case '\r':
3660         p++;
3661         goto redo_no_start;
3662
3663     case '\\':
3664         /* first look if it is in fact an end of buffer */
3665         if (p >= file->buf_end) {
3666             file->buf_ptr = p;
3667             handle_eob();
3668             p = file->buf_ptr;
3669             if (p >= file->buf_end)
3670                 goto parse_eof;
3671             else
3672                 goto redo_no_start;
3673         } else {
3674             file->buf_ptr = p;
3675             ch = *p;
3676             handle_stray();
3677             p = file->buf_ptr;
3678             goto redo_no_start;
3679         }
3680     parse_eof:
3681         {
3682             TCCState *s1 = tcc_state;
3683             if ((parse_flags & PARSE_FLAG_LINEFEED)
3684                 && !(tok_flags & TOK_FLAG_EOF)) {
3685                 tok_flags |= TOK_FLAG_EOF;
3686                 tok = TOK_LINEFEED;
3687                 goto keep_tok_flags;
3688             } else if (s1->include_stack_ptr == s1->include_stack ||
3689                        !(parse_flags & PARSE_FLAG_PREPROCESS)) {
3690                 /* no include left : end of file. */
3691                 tok = TOK_EOF;
3692             } else {
3693                 tok_flags &= ~TOK_FLAG_EOF;
3694                 /* pop include file */
3695
3696                 /* test if previous '#endif' was after a #ifdef at
3697                    start of file */
3698                 if (tok_flags & TOK_FLAG_ENDIF) {
3699 #ifdef INC_DEBUG
3700                     printf("#endif %s\n", get_tok_str(file->ifndef_macro_saved, NULL));
3701 #endif
3702                     add_cached_include(s1, file->inc_type, file->inc_filename,
3703                                        file->ifndef_macro_saved);
3704                 }
3705
3706                 /* add end of include file debug info */
3707                 if (do_debug) {
3708                     put_stabd(N_EINCL, 0, 0);
3709                 }
3710                 /* pop include stack */
3711                 tcc_close(file);
3712                 s1->include_stack_ptr--;
3713                 file = *s1->include_stack_ptr;
3714                 p = file->buf_ptr;
3715                 goto redo_no_start;
3716             }
3717         }
3718         break;
3719
3720     case '\n':
3721         file->line_num++;
3722         tok_flags |= TOK_FLAG_BOL;
3723         p++;
3724         if (0 == (parse_flags & PARSE_FLAG_LINEFEED))
3725             goto redo_no_start;
3726         tok = TOK_LINEFEED;
3727         goto keep_tok_flags;
3728
3729     case '#':
3730         /* XXX: simplify */
3731         PEEKC(c, p);
3732         if ((tok_flags & TOK_FLAG_BOL) &&
3733             (parse_flags & PARSE_FLAG_PREPROCESS)) {
3734             file->buf_ptr = p;
3735             preprocess(tok_flags & TOK_FLAG_BOF);
3736             p = file->buf_ptr;
3737             goto redo_no_start;
3738         } else {
3739             if (c == '#') {
3740                 p++;
3741                 tok = TOK_TWOSHARPS;
3742             } else {
3743                 if (parse_flags & PARSE_FLAG_ASM_COMMENTS) {
3744                     p = parse_line_comment(p - 1);
3745                     goto redo_no_start;
3746                 } else {
3747                     tok = '#';
3748                 }
3749             }
3750         }
3751         break;
3752
3753     case 'a': case 'b': case 'c': case 'd':
3754     case 'e': case 'f': case 'g': case 'h':
3755     case 'i': case 'j': case 'k': case 'l':
3756     case 'm': case 'n': case 'o': case 'p':
3757     case 'q': case 'r': case 's': case 't':
3758     case 'u': case 'v': case 'w': case 'x':
3759     case 'y': case 'z':
3760     case 'A': case 'B': case 'C': case 'D':
3761     case 'E': case 'F': case 'G': case 'H':
3762     case 'I': case 'J': case 'K':
3763     case 'M': case 'N': case 'O': case 'P':
3764     case 'Q': case 'R': case 'S': case 'T':
3765     case 'U': case 'V': case 'W': case 'X':
3766     case 'Y': case 'Z':
3767     case '_':
3768     parse_ident_fast:
3769         p1 = p;
3770         h = TOK_HASH_INIT;
3771         h = TOK_HASH_FUNC(h, c);
3772         p++;
3773         for(;;) {
3774             c = *p;
3775             if (!isidnum_table[c-CH_EOF])
3776                 break;
3777             h = TOK_HASH_FUNC(h, c);
3778             p++;
3779         }
3780         if (c != '\\') {
3781             TokenSym **pts;
3782             int len;
3783
3784             /* fast case : no stray found, so we have the full token
3785                and we have already hashed it */
3786             len = p - p1;
3787             h &= (TOK_HASH_SIZE - 1);
3788             pts = &hash_ident[h];
3789             for(;;) {
3790                 ts = *pts;
3791                 if (!ts)
3792                     break;
3793                 if (ts->len == len && !memcmp(ts->str, p1, len))
3794                     goto token_found;
3795                 pts = &(ts->hash_next);
3796             }
3797             ts = tok_alloc_new(pts, p1, len);
3798         token_found: ;
3799         } else {
3800             /* slower case */
3801             cstr_reset(&tokcstr);
3802
3803             while (p1 < p) {
3804                 cstr_ccat(&tokcstr, *p1);
3805                 p1++;
3806             }
3807             p--;
3808             PEEKC(c, p);
3809         parse_ident_slow:
3810             while (isidnum_table[c-CH_EOF]) {
3811                 cstr_ccat(&tokcstr, c);
3812                 PEEKC(c, p);
3813             }
3814             ts = tok_alloc(tokcstr.data, tokcstr.size);
3815         }
3816         tok = ts->tok;
3817         break;
3818     case 'L':
3819         t = p[1];
3820         if (t != '\\' && t != '\'' && t != '\"') {
3821             /* fast case */
3822             goto parse_ident_fast;
3823         } else {
3824             PEEKC(c, p);
3825             if (c == '\'' || c == '\"') {
3826                 is_long = 1;
3827                 goto str_const;
3828             } else {
3829                 cstr_reset(&tokcstr);
3830                 cstr_ccat(&tokcstr, 'L');
3831                 goto parse_ident_slow;
3832             }
3833         }
3834         break;
3835     case '0': case '1': case '2': case '3':
3836     case '4': case '5': case '6': case '7':
3837     case '8': case '9':
3838
3839         cstr_reset(&tokcstr);
3840         /* after the first digit, accept digits, alpha, '.' or sign if
3841            prefixed by 'eEpP' */
3842     parse_num:
3843         for(;;) {
3844             t = c;
3845             cstr_ccat(&tokcstr, c);
3846             PEEKC(c, p);
3847             if (!(isnum(c) || isid(c) || c == '.' ||
3848                   ((c == '+' || c == '-') &&
3849                    (t == 'e' || t == 'E' || t == 'p' || t == 'P'))))
3850                 break;
3851         }
3852         /* We add a trailing '\0' to ease parsing */
3853         cstr_ccat(&tokcstr, '\0');
3854         tokc.cstr = &tokcstr;
3855         tok = TOK_PPNUM;
3856         break;
3857     case '.':
3858         /* special dot handling because it can also start a number */
3859         PEEKC(c, p);
3860         if (isnum(c)) {
3861             cstr_reset(&tokcstr);
3862             cstr_ccat(&tokcstr, '.');
3863             goto parse_num;
3864         } else if (c == '.') {
3865             PEEKC(c, p);
3866             if (c != '.')
3867                 expect("'.'");
3868             PEEKC(c, p);
3869             tok = TOK_DOTS;
3870         } else {
3871             tok = '.';
3872         }
3873         break;
3874     case '\'':
3875     case '\"':
3876         is_long = 0;
3877     str_const:
3878         {
3879             CString str;
3880             int sep;
3881
3882             sep = c;
3883
3884             /* parse the string */
3885             cstr_new(&str);
3886             p = parse_pp_string(p, sep, &str);
3887             cstr_ccat(&str, '\0');
3888
3889             /* eval the escape (should be done as TOK_PPNUM) */
3890             cstr_reset(&tokcstr);
3891             parse_escape_string(&tokcstr, str.data, is_long);
3892             cstr_free(&str);
3893
3894             if (sep == '\'') {
3895                 int char_size;
3896                 /* XXX: make it portable */
3897                 if (!is_long)
3898                     char_size = 1;
3899                 else
3900                     char_size = sizeof(nwchar_t);
3901                 if (tokcstr.size <= char_size)
3902                     error("empty character constant");
3903                 if (tokcstr.size > 2 * char_size)
3904                     warning("multi-character character constant");
3905                 if (!is_long) {
3906                     tokc.i = *(int8_t *)tokcstr.data;
3907                     tok = TOK_CCHAR;
3908                 } else {
3909                     tokc.i = *(nwchar_t *)tokcstr.data;
3910                     tok = TOK_LCHAR;
3911                 }
3912             } else {
3913                 tokc.cstr = &tokcstr;
3914                 if (!is_long)
3915                     tok = TOK_STR;
3916                 else
3917                     tok = TOK_LSTR;
3918             }
3919         }
3920         break;
3921
3922     case '<':
3923         PEEKC(c, p);
3924         if (c == '=') {
3925             p++;
3926             tok = TOK_LE;
3927         } else if (c == '<') {
3928             PEEKC(c, p);
3929             if (c == '=') {
3930                 p++;
3931                 tok = TOK_A_SHL;
3932             } else {
3933                 tok = TOK_SHL;
3934             }
3935         } else {
3936             tok = TOK_LT;
3937         }
3938         break;
3939
3940     case '>':
3941         PEEKC(c, p);
3942         if (c == '=') {
3943             p++;
3944             tok = TOK_GE;
3945         } else if (c == '>') {
3946             PEEKC(c, p);
3947             if (c == '=') {
3948                 p++;
3949                 tok = TOK_A_SAR;
3950             } else {
3951                 tok = TOK_SAR;
3952             }
3953         } else {
3954             tok = TOK_GT;
3955         }
3956         break;
3957
3958     case '&':
3959         PEEKC(c, p);
3960         if (c == '&') {
3961             p++;
3962             tok = TOK_LAND;
3963         } else if (c == '=') {
3964             p++;
3965             tok = TOK_A_AND;
3966         } else {
3967             tok = '&';
3968         }
3969         break;
3970
3971     case '|':
3972         PEEKC(c, p);
3973         if (c == '|') {
3974             p++;
3975             tok = TOK_LOR;
3976         } else if (c == '=') {
3977             p++;
3978             tok = TOK_A_OR;
3979         } else {
3980             tok = '|';
3981         }
3982         break;
3983
3984     case '+':
3985         PEEKC(c, p);
3986         if (c == '+') {
3987             p++;
3988             tok = TOK_INC;
3989         } else if (c == '=') {
3990             p++;
3991             tok = TOK_A_ADD;
3992         } else {
3993             tok = '+';
3994         }
3995         break;
3996
3997     case '-':
3998         PEEKC(c, p);
3999         if (c == '-') {
4000             p++;
4001             tok = TOK_DEC;
4002         } else if (c == '=') {
4003             p++;
4004             tok = TOK_A_SUB;
4005         } else if (c == '>') {
4006             p++;
4007             tok = TOK_ARROW;
4008         } else {
4009             tok = '-';
4010         }
4011         break;
4012
4013     PARSE2('!', '!', '=', TOK_NE)
4014     PARSE2('=', '=', '=', TOK_EQ)
4015     PARSE2('*', '*', '=', TOK_A_MUL)
4016     PARSE2('%', '%', '=', TOK_A_MOD)
4017     PARSE2('^', '^', '=', TOK_A_XOR)
4018
4019         /* comments or operator */
4020     case '/':
4021         PEEKC(c, p);
4022         if (c == '*') {
4023             p = parse_comment(p);
4024             goto redo_no_start;
4025         } else if (c == '/') {
4026             p = parse_line_comment(p);
4027             goto redo_no_start;
4028         } else if (c == '=') {
4029             p++;
4030             tok = TOK_A_DIV;
4031         } else {
4032             tok = '/';
4033         }
4034         break;
4035
4036         /* simple tokens */
4037     case '(':
4038     case ')':
4039     case '[':
4040     case ']':
4041     case '{':
4042     case '}':
4043     case ',':
4044     case ';':
4045     case ':':
4046     case '?':
4047     case '~':
4048     case '$': /* only used in assembler */
4049     case '@': /* dito */
4050         tok = c;
4051         p++;
4052         break;
4053     default:
4054         error("unrecognized character \\x%02x", c);
4055         break;
4056     }
4057     tok_flags = 0;
4058 keep_tok_flags:
4059     file->buf_ptr = p;
4060 #if defined(PARSE_DEBUG)
4061     printf("token = %s\n", get_tok_str(tok, &tokc));
4062 #endif
4063 }
4064
4065 /* return next token without macro substitution. Can read input from
4066    macro_ptr buffer */
4067 static void next_nomacro(void)
4068 {
4069     if (macro_ptr) {
4070     redo:
4071         tok = *macro_ptr;
4072         if (tok) {
4073             TOK_GET(tok, macro_ptr, tokc);
4074             if (tok == TOK_LINENUM) {
4075                 file->line_num = tokc.i;
4076                 goto redo;
4077             }
4078         }
4079     } else {
4080         next_nomacro1();
4081     }
4082 }
4083
4084 /* substitute args in macro_str and return allocated string */
4085 static int *macro_arg_subst(Sym **nested_list, int *macro_str, Sym *args)
4086 {
4087     int *st, last_tok, t, notfirst;
4088     Sym *s;
4089     CValue cval;
4090     TokenString str;
4091     CString cstr;
4092
4093     tok_str_new(&str);
4094     last_tok = 0;
4095     while(1) {
4096         TOK_GET(t, macro_str, cval);
4097         if (!t)
4098             break;
4099         if (t == '#') {
4100             /* stringize */
4101             TOK_GET(t, macro_str, cval);
4102             if (!t)
4103                 break;
4104             s = sym_find2(args, t);
4105             if (s) {
4106                 cstr_new(&cstr);
4107                 st = (int *)s->c;
4108                 notfirst = 0;
4109                 while (*st) {
4110                     if (notfirst)
4111                         cstr_ccat(&cstr, ' ');
4112                     TOK_GET(t, st, cval);
4113                     cstr_cat(&cstr, get_tok_str(t, &cval));
4114 #ifndef PP_NOSPACES
4115                     notfirst = 1;
4116 #endif
4117                 }
4118                 cstr_ccat(&cstr, '\0');
4119 #ifdef PP_DEBUG
4120                 printf("stringize: %s\n", (char *)cstr.data);
4121 #endif
4122                 /* add string */
4123                 cval.cstr = &cstr;
4124                 tok_str_add2(&str, TOK_STR, &cval);
4125                 cstr_free(&cstr);
4126             } else {
4127                 tok_str_add2(&str, t, &cval);
4128             }
4129         } else if (t >= TOK_IDENT) {
4130             s = sym_find2(args, t);
4131             if (s) {
4132                 st = (int *)s->c;
4133                 /* if '##' is present before or after, no arg substitution */
4134                 if (*macro_str == TOK_TWOSHARPS || last_tok == TOK_TWOSHARPS) {
4135                     /* special case for var arg macros : ## eats the
4136                        ',' if empty VA_ARGS variable. */
4137                     /* XXX: test of the ',' is not 100%
4138                        reliable. should fix it to avoid security
4139                        problems */
4140                     if (gnu_ext && s->type.t &&
4141                         last_tok == TOK_TWOSHARPS &&
4142                         str.len >= 2 && str.str[str.len - 2] == ',') {
4143                         if (*st == 0) {
4144                             /* suppress ',' '##' */
4145                             str.len -= 2;
4146                         } else {
4147                             /* suppress '##' and add variable */
4148                             str.len--;
4149                             goto add_var;
4150                         }
4151                     } else {
4152                         int t1;
4153                     add_var:
4154                         for(;;) {
4155                             TOK_GET(t1, st, cval);
4156                             if (!t1)
4157                                 break;
4158                             tok_str_add2(&str, t1, &cval);
4159                         }
4160                     }
4161                 } else {
4162                     /* NOTE: the stream cannot be read when macro
4163                        substituing an argument */
4164                     macro_subst(&str, nested_list, st, NULL);
4165                 }
4166             } else {
4167                 tok_str_add(&str, t);
4168             }
4169         } else {
4170             tok_str_add2(&str, t, &cval);
4171         }
4172         last_tok = t;
4173     }
4174     tok_str_add(&str, 0);
4175     return str.str;
4176 }
4177
4178 static char const ab_month_name[12][4] =
4179 {
4180     "Jan", "Feb", "Mar", "Apr", "May", "Jun",
4181     "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
4182 };
4183
4184 /* do macro substitution of current token with macro 's' and add
4185    result to (tok_str,tok_len). 'nested_list' is the list of all
4186    macros we got inside to avoid recursing. Return non zero if no
4187    substitution needs to be done */
4188 static int macro_subst_tok(TokenString *tok_str,
4189                            Sym **nested_list, Sym *s, struct macro_level **can_read_stream)
4190 {
4191     Sym *args, *sa, *sa1;
4192     int mstr_allocated, parlevel, *mstr, t, t1;
4193     TokenString str;
4194     char *cstrval;
4195     CValue cval;
4196     CString cstr;
4197     char buf[32];
4198
4199     /* if symbol is a macro, prepare substitution */
4200     /* special macros */
4201     if (tok == TOK___LINE__) {
4202         snprintf(buf, sizeof(buf), "%d", file->line_num);
4203         cstrval = buf;
4204         t1 = TOK_PPNUM;
4205         goto add_cstr1;
4206     } else if (tok == TOK___FILE__) {
4207         cstrval = file->filename;
4208         goto add_cstr;
4209     } else if (tok == TOK___DATE__ || tok == TOK___TIME__) {
4210         time_t ti;
4211         struct tm *tm;
4212
4213         time(&ti);
4214         tm = localtime(&ti);
4215         if (tok == TOK___DATE__) {
4216             snprintf(buf, sizeof(buf), "%s %2d %d",
4217                      ab_month_name[tm->tm_mon], tm->tm_mday, tm->tm_year + 1900);
4218         } else {
4219             snprintf(buf, sizeof(buf), "%02d:%02d:%02d",
4220                      tm->tm_hour, tm->tm_min, tm->tm_sec);
4221         }
4222         cstrval = buf;
4223     add_cstr:
4224         t1 = TOK_STR;
4225     add_cstr1:
4226         cstr_new(&cstr);
4227         cstr_cat(&cstr, cstrval);
4228         cstr_ccat(&cstr, '\0');
4229         cval.cstr = &cstr;
4230         tok_str_add2(tok_str, t1, &cval);
4231         cstr_free(&cstr);
4232     } else {
4233         mstr = (int *)s->c;
4234         mstr_allocated = 0;
4235         if (s->type.t == MACRO_FUNC) {
4236             /* NOTE: we do not use next_nomacro to avoid eating the
4237                next token. XXX: find better solution */
4238         redo:
4239             if (macro_ptr) {
4240                 t = *macro_ptr;
4241                 if (t == 0 && can_read_stream) {
4242                     /* end of macro stream: we must look at the token
4243                        after in the file */
4244                     struct macro_level *ml = *can_read_stream;
4245                     macro_ptr = NULL;
4246                     if (ml)
4247                     {
4248                         macro_ptr = ml->p;
4249                         ml->p = NULL;
4250                         *can_read_stream = ml -> prev;
4251                     }
4252                     goto redo;
4253                 }
4254             } else {
4255                 /* XXX: incorrect with comments */
4256                 ch = file->buf_ptr[0];
4257                 while (is_space(ch) || ch == '\n')
4258                     cinp();
4259                 t = ch;
4260             }
4261             if (t != '(') /* no macro subst */
4262                 return -1;
4263
4264             /* argument macro */
4265             next_nomacro();
4266             next_nomacro();
4267             args = NULL;
4268             sa = s->next;
4269             /* NOTE: empty args are allowed, except if no args */
4270             for(;;) {
4271                 /* handle '()' case */
4272                 if (!args && !sa && tok == ')')
4273                     break;
4274                 if (!sa)
4275                     error("macro '%s' used with too many args",
4276                           get_tok_str(s->v, 0));
4277                 tok_str_new(&str);
4278                 parlevel = 0;
4279                 /* NOTE: non zero sa->t indicates VA_ARGS */
4280                 while ((parlevel > 0 ||
4281                         (tok != ')' &&
4282                          (tok != ',' || sa->type.t))) &&
4283                        tok != -1) {
4284                     if (tok == '(')
4285                         parlevel++;
4286                     else if (tok == ')')
4287                         parlevel--;
4288                     if (tok != TOK_LINEFEED)
4289                         tok_str_add2(&str, tok, &tokc);
4290                     next_nomacro();
4291                 }
4292                 tok_str_add(&str, 0);
4293                 sym_push2(&args, sa->v & ~SYM_FIELD, sa->type.t, (int)str.str);
4294                 sa = sa->next;
4295                 if (tok == ')') {
4296                     /* special case for gcc var args: add an empty
4297                        var arg argument if it is omitted */
4298                     if (sa && sa->type.t && gnu_ext)
4299                         continue;
4300                     else
4301                         break;
4302                 }
4303                 if (tok != ',')
4304                     expect(",");
4305                 next_nomacro();
4306             }
4307             if (sa) {
4308                 error("macro '%s' used with too few args",
4309                       get_tok_str(s->v, 0));
4310             }
4311
4312             /* now subst each arg */
4313             mstr = macro_arg_subst(nested_list, mstr, args);
4314             /* free memory */
4315             sa = args;
4316             while (sa) {
4317                 sa1 = sa->prev;
4318                 tok_str_free((int *)sa->c);
4319                 sym_free(sa);
4320                 sa = sa1;
4321             }
4322             mstr_allocated = 1;
4323         }
4324         sym_push2(nested_list, s->v, 0, 0);
4325         macro_subst(tok_str, nested_list, mstr, can_read_stream);
4326         /* pop nested defined symbol */
4327         sa1 = *nested_list;
4328         *nested_list = sa1->prev;
4329         sym_free(sa1);
4330         if (mstr_allocated)
4331             tok_str_free(mstr);
4332     }
4333     return 0;
4334 }
4335
4336 /* handle the '##' operator. Return NULL if no '##' seen. Otherwise
4337    return the resulting string (which must be freed). */
4338 static inline int *macro_twosharps(const int *macro_str)
4339 {
4340     TokenSym *ts;
4341     const int *macro_ptr1, *start_macro_ptr, *ptr, *saved_macro_ptr;
4342     int t;
4343     const char *p1, *p2;
4344     CValue cval;
4345     TokenString macro_str1;
4346     CString cstr;
4347
4348     start_macro_ptr = macro_str;
4349     /* we search the first '##' */
4350     for(;;) {
4351         macro_ptr1 = macro_str;
4352         TOK_GET(t, macro_str, cval);
4353         /* nothing more to do if end of string */
4354         if (t == 0)
4355             return NULL;
4356         if (*macro_str == TOK_TWOSHARPS)
4357             break;
4358     }
4359
4360     /* we saw '##', so we need more processing to handle it */
4361     cstr_new(&cstr);
4362     tok_str_new(&macro_str1);
4363     tok = t;
4364     tokc = cval;
4365
4366     /* add all tokens seen so far */
4367     for(ptr = start_macro_ptr; ptr < macro_ptr1;) {
4368         TOK_GET(t, ptr, cval);
4369         tok_str_add2(&macro_str1, t, &cval);
4370     }
4371     saved_macro_ptr = macro_ptr;
4372     /* XXX: get rid of the use of macro_ptr here */
4373     macro_ptr = (int *)macro_str;
4374     for(;;) {
4375         while (*macro_ptr == TOK_TWOSHARPS) {
4376             macro_ptr++;
4377             macro_ptr1 = macro_ptr;
4378             t = *macro_ptr;
4379             if (t) {
4380                 TOK_GET(t, macro_ptr, cval);
4381                 /* We concatenate the two tokens if we have an
4382                    identifier or a preprocessing number */
4383                 cstr_reset(&cstr);
4384                 p1 = get_tok_str(tok, &tokc);
4385                 cstr_cat(&cstr, p1);
4386                 p2 = get_tok_str(t, &cval);
4387                 cstr_cat(&cstr, p2);
4388                 cstr_ccat(&cstr, '\0');
4389
4390                 if ((tok >= TOK_IDENT || tok == TOK_PPNUM) &&
4391                     (t >= TOK_IDENT || t == TOK_PPNUM)) {
4392                     if (tok == TOK_PPNUM) {
4393                         /* if number, then create a number token */
4394                         /* NOTE: no need to allocate because
4395                            tok_str_add2() does it */
4396                         cstr_reset(&tokcstr);
4397                         tokcstr = cstr;
4398                         cstr_new(&cstr);
4399                         tokc.cstr = &tokcstr;
4400                     } else {
4401                         /* if identifier, we must do a test to
4402                            validate we have a correct identifier */
4403                         if (t == TOK_PPNUM) {
4404                             const char *p;
4405                             int c;
4406
4407                             p = p2;
4408                             for(;;) {
4409                                 c = *p;
4410                                 if (c == '\0')
4411                                     break;
4412                                 p++;
4413                                 if (!isnum(c) && !isid(c))
4414                                     goto error_pasting;
4415                             }
4416                         }
4417                         ts = tok_alloc(cstr.data, strlen(cstr.data));
4418                         tok = ts->tok; /* modify current token */
4419                     }
4420                 } else {
4421                     const char *str = cstr.data;
4422                     const unsigned char *q;
4423
4424                     /* we look for a valid token */
4425                     /* XXX: do more extensive checks */
4426                     if (!strcmp(str, ">>=")) {
4427                         tok = TOK_A_SAR;
4428                     } else if (!strcmp(str, "<<=")) {
4429                         tok = TOK_A_SHL;
4430                     } else if (strlen(str) == 2) {
4431                         /* search in two bytes table */
4432                         q = tok_two_chars;
4433                         for(;;) {
4434                             if (!*q)
4435                                 goto error_pasting;
4436                             if (q[0] == str[0] && q[1] == str[1])
4437                                 break;
4438                             q += 3;
4439                         }
4440                         tok = q[2];
4441                     } else {
4442                     error_pasting:
4443                         /* NOTE: because get_tok_str use a static buffer,
4444                            we must save it */
4445                         cstr_reset(&cstr);
4446                         p1 = get_tok_str(tok, &tokc);
4447                         cstr_cat(&cstr, p1);
4448                         cstr_ccat(&cstr, '\0');
4449                         p2 = get_tok_str(t, &cval);
4450                         warning("pasting \"%s\" and \"%s\" does not give a valid preprocessing token", cstr.data, p2);
4451                         /* cannot merge tokens: just add them separately */
4452                         tok_str_add2(&macro_str1, tok, &tokc);
4453                         /* XXX: free associated memory ? */
4454                         tok = t;
4455                         tokc = cval;
4456                     }
4457                 }
4458             }
4459         }
4460         tok_str_add2(&macro_str1, tok, &tokc);
4461         next_nomacro();
4462         if (tok == 0)
4463             break;
4464     }
4465     macro_ptr = (int *)saved_macro_ptr;
4466     cstr_free(&cstr);
4467     tok_str_add(&macro_str1, 0);
4468     return macro_str1.str;
4469 }
4470
4471
4472 /* do macro substitution of macro_str and add result to
4473    (tok_str,tok_len). 'nested_list' is the list of all macros we got
4474    inside to avoid recursing. */
4475 static void macro_subst(TokenString *tok_str, Sym **nested_list,
4476                         const int *macro_str, struct macro_level ** can_read_stream)
4477 {
4478     Sym *s;
4479     int *macro_str1;
4480     const int *ptr;
4481     int t, ret;
4482     CValue cval;
4483     struct macro_level ml;
4484
4485     /* first scan for '##' operator handling */
4486     ptr = macro_str;
4487     macro_str1 = macro_twosharps(ptr);
4488     if (macro_str1)
4489         ptr = macro_str1;
4490     while (1) {
4491         /* NOTE: ptr == NULL can only happen if tokens are read from
4492            file stream due to a macro function call */
4493         if (ptr == NULL)
4494             break;
4495         TOK_GET(t, ptr, cval);
4496         if (t == 0)
4497             break;
4498         s = define_find(t);
4499         if (s != NULL) {
4500             /* if nested substitution, do nothing */
4501             if (sym_find2(*nested_list, t))
4502                 goto no_subst;
4503             ml.p = macro_ptr;
4504             if (can_read_stream)
4505                 ml.prev = *can_read_stream, *can_read_stream = &ml;
4506             macro_ptr = (int *)ptr;
4507             tok = t;
4508             ret = macro_subst_tok(tok_str, nested_list, s, can_read_stream);
4509             ptr = (int *)macro_ptr;
4510             macro_ptr = ml.p;
4511             if (can_read_stream && *can_read_stream == &ml)
4512                 *can_read_stream = ml.prev;
4513             if (ret != 0)
4514                 goto no_subst;
4515         } else {
4516         no_subst:
4517             tok_str_add2(tok_str, t, &cval);
4518         }
4519     }
4520     if (macro_str1)
4521         tok_str_free(macro_str1);
4522 }
4523
4524 /* return next token with macro substitution */
4525 static void next(void)
4526 {
4527     Sym *nested_list, *s;
4528     TokenString str;
4529     struct macro_level *ml;
4530
4531  redo:
4532     next_nomacro();
4533     if (!macro_ptr) {
4534         /* if not reading from macro substituted string, then try
4535            to substitute macros */
4536         if (tok >= TOK_IDENT &&
4537             (parse_flags & PARSE_FLAG_PREPROCESS)) {
4538             s = define_find(tok);
4539             if (s) {
4540                 /* we have a macro: we try to substitute */
4541                 tok_str_new(&str);
4542                 nested_list = NULL;
4543                 ml = NULL;
4544                 if (macro_subst_tok(&str, &nested_list, s, &ml) == 0) {
4545                     /* substitution done, NOTE: maybe empty */
4546                     tok_str_add(&str, 0);
4547                     macro_ptr = str.str;
4548                     macro_ptr_allocated = str.str;
4549                     goto redo;
4550                 }
4551             }
4552         }
4553     } else {
4554         if (tok == 0) {
4555             /* end of macro or end of unget buffer */
4556             if (unget_buffer_enabled) {
4557                 macro_ptr = unget_saved_macro_ptr;
4558                 unget_buffer_enabled = 0;
4559             } else {
4560                 /* end of macro string: free it */
4561                 tok_str_free(macro_ptr_allocated);
4562                 macro_ptr = NULL;
4563             }
4564             goto redo;
4565         }
4566     }
4567
4568     /* convert preprocessor tokens into C tokens */
4569     if (tok == TOK_PPNUM &&
4570         (parse_flags & PARSE_FLAG_TOK_NUM)) {
4571         parse_number((char *)tokc.cstr->data);
4572     }
4573 }
4574
4575 /* push back current token and set current token to 'last_tok'. Only
4576    identifier case handled for labels. */
4577 static inline void unget_tok(int last_tok)
4578 {
4579     int i, n;
4580     int *q;
4581     unget_saved_macro_ptr = macro_ptr;
4582     unget_buffer_enabled = 1;
4583     q = unget_saved_buffer;
4584     macro_ptr = q;
4585     *q++ = tok;
4586     n = tok_ext_size(tok) - 1;
4587     for(i=0;i<n;i++)
4588         *q++ = tokc.tab[i];
4589     *q = 0; /* end of token string */
4590     tok = last_tok;
4591 }
4592
4593
4594 void swap(int *p, int *q)
4595 {
4596     int t;
4597     t = *p;
4598     *p = *q;
4599     *q = t;
4600 }
4601
4602 void vsetc(CType *type, int r, CValue *vc)
4603 {
4604     int v;
4605
4606     if (vtop >= vstack + (VSTACK_SIZE - 1))
4607         error("memory full");
4608     /* cannot let cpu flags if other instruction are generated. Also
4609        avoid leaving VT_JMP anywhere except on the top of the stack
4610        because it would complicate the code generator. */
4611     if (vtop >= vstack) {
4612         v = vtop->r & VT_VALMASK;
4613         if (v == VT_CMP || (v & ~1) == VT_JMP)
4614             gv(RC_INT);
4615     }
4616     vtop++;
4617     vtop->type = *type;
4618     vtop->r = r;
4619     vtop->r2 = VT_CONST;
4620     vtop->c = *vc;
4621 }
4622
4623 /* push integer constant */
4624 void vpushi(int v)
4625 {
4626     CValue cval;
4627     cval.i = v;
4628     vsetc(&int_type, VT_CONST, &cval);
4629 }
4630
4631 /* Return a static symbol pointing to a section */
4632 static Sym *get_sym_ref(CType *type, Section *sec,
4633                         unsigned long offset, unsigned long size)
4634 {
4635     int v;
4636     Sym *sym;
4637
4638     v = anon_sym++;
4639     sym = global_identifier_push(v, type->t | VT_STATIC, 0);
4640     sym->type.ref = type->ref;
4641     sym->r = VT_CONST | VT_SYM;
4642     put_extern_sym(sym, sec, offset, size);
4643     return sym;
4644 }
4645
4646 /* push a reference to a section offset by adding a dummy symbol */
4647 static void vpush_ref(CType *type, Section *sec, unsigned long offset, unsigned long size)
4648 {
4649     CValue cval;
4650
4651     cval.ul = 0;
4652     vsetc(type, VT_CONST | VT_SYM, &cval);
4653     vtop->sym = get_sym_ref(type, sec, offset, size);
4654 }
4655
4656 /* define a new external reference to a symbol 'v' of type 'u' */
4657 static Sym *external_global_sym(int v, CType *type, int r)
4658 {
4659     Sym *s;
4660
4661     s = sym_find(v);
4662     if (!s) {
4663         /* push forward reference */
4664         s = global_identifier_push(v, type->t | VT_EXTERN, 0);
4665         s->type.ref = type->ref;
4666         s->r = r | VT_CONST | VT_SYM;
4667     }
4668     return s;
4669 }
4670
4671 /* define a new external reference to a symbol 'v' of type 'u' */
4672 static Sym *external_sym(int v, CType *type, int r)
4673 {
4674     Sym *s;
4675
4676     s = sym_find(v);
4677     if (!s) {
4678         /* push forward reference */
4679         s = sym_push(v, type, r | VT_CONST | VT_SYM, 0);
4680         s->type.t |= VT_EXTERN;
4681     } else {
4682         if (!is_compatible_types(&s->type, type))
4683             error("incompatible types for redefinition of '%s'",
4684                   get_tok_str(v, NULL));
4685     }
4686     return s;
4687 }
4688
4689 /* push a reference to global symbol v */
4690 static void vpush_global_sym(CType *type, int v)
4691 {
4692     Sym *sym;
4693     CValue cval;
4694
4695     sym = external_global_sym(v, type, 0);
4696     cval.ul = 0;
4697     vsetc(type, VT_CONST | VT_SYM, &cval);
4698     vtop->sym = sym;
4699 }
4700
4701 void vset(CType *type, int r, int v)
4702 {
4703     CValue cval;
4704
4705     cval.i = v;
4706     vsetc(type, r, &cval);
4707 }
4708
4709 void vseti(int r, int v)
4710 {
4711     CType type;
4712     type.t = VT_INT;
4713     vset(&type, r, v);
4714 }
4715
4716 void vswap(void)
4717 {
4718     SValue tmp;
4719
4720     tmp = vtop[0];
4721     vtop[0] = vtop[-1];
4722     vtop[-1] = tmp;
4723 }
4724
4725 void vpushv(SValue *v)
4726 {
4727     if (vtop >= vstack + (VSTACK_SIZE - 1))
4728         error("memory full");
4729     vtop++;
4730     *vtop = *v;
4731 }
4732
4733 void vdup(void)
4734 {
4735     vpushv(vtop);
4736 }
4737
4738 /* save r to the memory stack, and mark it as being free */
4739 void save_reg(int r)
4740 {
4741     int l, saved, size, align;
4742     SValue *p, sv;
4743     CType *type;
4744
4745     /* modify all stack values */
4746     saved = 0;
4747     l = 0;
4748     for(p=vstack;p<=vtop;p++) {
4749         if ((p->r & VT_VALMASK) == r ||
4750             ((p->type.t & VT_BTYPE) == VT_LLONG && (p->r2 & VT_VALMASK) == r)) {
4751             /* must save value on stack if not already done */
4752             if (!saved) {
4753                 /* NOTE: must reload 'r' because r might be equal to r2 */
4754                 r = p->r & VT_VALMASK;
4755                 /* store register in the stack */
4756                 type = &p->type;
4757                 if ((p->r & VT_LVAL) ||
4758                     (!is_float(type->t) && (type->t & VT_BTYPE) != VT_LLONG))
4759                     type = &int_type;
4760                 size = type_size(type, &align);
4761                 loc = (loc - size) & -align;
4762                 sv.type.t = type->t;
4763                 sv.r = VT_LOCAL | VT_LVAL;
4764                 sv.c.ul = loc;
4765                 store(r, &sv);
4766 #ifdef TCC_TARGET_I386
4767                 /* x86 specific: need to pop fp register ST0 if saved */
4768                 if (r == TREG_ST0) {
4769                     o(0xd9dd); /* fstp %st(1) */
4770                 }
4771 #endif
4772                 /* special long long case */
4773                 if ((type->t & VT_BTYPE) == VT_LLONG) {
4774                     sv.c.ul += 4;
4775                     store(p->r2, &sv);
4776                 }
4777                 l = loc;
4778                 saved = 1;
4779             }
4780             /* mark that stack entry as being saved on the stack */
4781             if (p->r & VT_LVAL) {
4782                 /* also clear the bounded flag because the
4783                    relocation address of the function was stored in
4784                    p->c.ul */
4785                 p->r = (p->r & ~(VT_VALMASK | VT_BOUNDED)) | VT_LLOCAL;
4786             } else {
4787                 p->r = lvalue_type(p->type.t) | VT_LOCAL;
4788             }
4789             p->r2 = VT_CONST;
4790             p->c.ul = l;
4791         }
4792     }
4793 }
4794
4795 /* find a register of class 'rc2' with at most one reference on stack.
4796  * If none, call get_reg(rc) */
4797 int get_reg_ex(int rc, int rc2)
4798 {
4799     int r;
4800     SValue *p;
4801
4802     for(r=0;r<NB_REGS;r++) {
4803         if (reg_classes[r] & rc2) {
4804             int n;
4805             n=0;
4806             for(p = vstack; p <= vtop; p++) {
4807                 if ((p->r & VT_VALMASK) == r ||
4808                     (p->r2 & VT_VALMASK) == r)
4809                     n++;
4810             }
4811             if (n <= 1)
4812                 return r;
4813         }
4814     }
4815     return get_reg(rc);
4816 }
4817
4818 /* find a free register of class 'rc'. If none, save one register */
4819 int get_reg(int rc)
4820 {
4821     int r;
4822     SValue *p;
4823
4824     /* find a free register */
4825     for(r=0;r<NB_REGS;r++) {
4826         if (reg_classes[r] & rc) {
4827             for(p=vstack;p<=vtop;p++) {
4828                 if ((p->r & VT_VALMASK) == r ||
4829                     (p->r2 & VT_VALMASK) == r)
4830                     goto notfound;
4831             }
4832             return r;
4833         }
4834     notfound: ;
4835     }
4836
4837     /* no register left : free the first one on the stack (VERY
4838        IMPORTANT to start from the bottom to ensure that we don't
4839        spill registers used in gen_opi()) */
4840     for(p=vstack;p<=vtop;p++) {
4841         r = p->r & VT_VALMASK;
4842         if (r < VT_CONST && (reg_classes[r] & rc))
4843             goto save_found;
4844         /* also look at second register (if long long) */
4845         r = p->r2 & VT_VALMASK;
4846         if (r < VT_CONST && (reg_classes[r] & rc)) {
4847         save_found:
4848             save_reg(r);
4849             return r;
4850         }
4851     }
4852     /* Should never comes here */
4853     return -1;
4854 }
4855
4856 /* save registers up to (vtop - n) stack entry */
4857 void save_regs(int n)
4858 {
4859     int r;
4860     SValue *p, *p1;
4861     p1 = vtop - n;
4862     for(p = vstack;p <= p1; p++) {
4863         r = p->r & VT_VALMASK;
4864         if (r < VT_CONST) {
4865             save_reg(r);
4866         }
4867     }
4868 }
4869
4870 /* move register 's' to 'r', and flush previous value of r to memory
4871    if needed */
4872 void move_reg(int r, int s)
4873 {
4874     SValue sv;
4875
4876     if (r != s) {
4877         save_reg(r);
4878         sv.type.t = VT_INT;
4879         sv.r = s;
4880         sv.c.ul = 0;
4881         load(r, &sv);
4882     }
4883 }
4884
4885 /* get address of vtop (vtop MUST BE an lvalue) */
4886 void gaddrof(void)
4887 {
4888     vtop->r &= ~VT_LVAL;
4889     /* tricky: if saved lvalue, then we can go back to lvalue */
4890     if ((vtop->r & VT_VALMASK) == VT_LLOCAL)
4891         vtop->r = (vtop->r & ~(VT_VALMASK | VT_LVAL_TYPE)) | VT_LOCAL | VT_LVAL;
4892 }
4893
4894 #ifdef CONFIG_TCC_BCHECK
4895 /* generate lvalue bound code */
4896 void gbound(void)
4897 {
4898     int lval_type;
4899     CType type1;
4900
4901     vtop->r &= ~VT_MUSTBOUND;
4902     /* if lvalue, then use checking code before dereferencing */
4903     if (vtop->r & VT_LVAL) {
4904         /* if not VT_BOUNDED value, then make one */
4905         if (!(vtop->r & VT_BOUNDED)) {
4906             lval_type = vtop->r & (VT_LVAL_TYPE | VT_LVAL);
4907             /* must save type because we must set it to int to get pointer */
4908             type1 = vtop->type;
4909             vtop->type.t = VT_INT;
4910             gaddrof();
4911             vpushi(0);
4912             gen_bounded_ptr_add();
4913             vtop->r |= lval_type;
4914             vtop->type = type1;
4915         }
4916         /* then check for dereferencing */
4917         gen_bounded_ptr_deref();
4918     }
4919 }
4920 #endif
4921
4922 /* store vtop a register belonging to class 'rc'. lvalues are
4923    converted to values. Cannot be used if cannot be converted to
4924    register value (such as structures). */
4925 int gv(int rc)
4926 {
4927     int r, r2, rc2, bit_pos, bit_size, size, align, i;
4928     unsigned long long ll;
4929
4930     /* NOTE: get_reg can modify vstack[] */
4931     if (vtop->type.t & VT_BITFIELD) {
4932         CType type;
4933         bit_pos = (vtop->type.t >> VT_STRUCT_SHIFT) & 0x3f;
4934         bit_size = (vtop->type.t >> (VT_STRUCT_SHIFT + 6)) & 0x3f;
4935         /* remove bit field info to avoid loops */
4936         vtop->type.t &= ~(VT_BITFIELD | (-1 << VT_STRUCT_SHIFT));
4937         /* cast to int to propagate signedness in following ops */
4938         type.t = VT_INT;
4939         if((vtop->type.t & VT_UNSIGNED) ||
4940            (vtop->type.t & VT_BTYPE) == VT_BOOL)
4941             type.t |= VT_UNSIGNED;
4942         gen_cast(&type);
4943         /* generate shifts */
4944         vpushi(32 - (bit_pos + bit_size));
4945         gen_op(TOK_SHL);
4946         vpushi(32 - bit_size);
4947         /* NOTE: transformed to SHR if unsigned */
4948         gen_op(TOK_SAR);
4949         r = gv(rc);
4950     } else {
4951         if (is_float(vtop->type.t) &&
4952             (vtop->r & (VT_VALMASK | VT_LVAL)) == VT_CONST) {
4953             Sym *sym;
4954             int *ptr;
4955             unsigned long offset;
4956 #if defined(TCC_TARGET_ARM) && !defined(TCC_ARM_VFP)
4957             CValue check;
4958 #endif
4959
4960             /* XXX: unify with initializers handling ? */
4961             /* CPUs usually cannot use float constants, so we store them
4962                generically in data segment */
4963             size = type_size(&vtop->type, &align);
4964             offset = (data_section->data_offset + align - 1) & -align;
4965             data_section->data_offset = offset;
4966             /* XXX: not portable yet */
4967 #ifdef __i386__
4968             /* Zero pad x87 tenbyte long doubles */
4969             if (size == 12)
4970                 vtop->c.tab[2] &= 0xffff;
4971 #endif
4972             ptr = section_ptr_add(data_section, size);
4973             size = size >> 2;
4974 #if defined(TCC_TARGET_ARM) && !defined(TCC_ARM_VFP)
4975             check.d = 1;
4976             if(check.tab[0])
4977                 for(i=0;i<size;i++)
4978                     ptr[i] = vtop->c.tab[size-1-i];
4979             else
4980 #endif
4981             for(i=0;i<size;i++)
4982                 ptr[i] = vtop->c.tab[i];
4983             sym = get_sym_ref(&vtop->type, data_section, offset, size << 2);
4984             vtop->r |= VT_LVAL | VT_SYM;
4985             vtop->sym = sym;
4986             vtop->c.ul = 0;
4987         }
4988 #ifdef CONFIG_TCC_BCHECK
4989         if (vtop->r & VT_MUSTBOUND)
4990             gbound();
4991 #endif
4992
4993         r = vtop->r & VT_VALMASK;
4994         rc2 = RC_INT;
4995         if (rc == RC_IRET)
4996             rc2 = RC_LRET;
4997         /* need to reload if:
4998            - constant
4999            - lvalue (need to dereference pointer)
5000            - already a register, but not in the right class */
5001         if (r >= VT_CONST ||
5002             (vtop->r & VT_LVAL) ||
5003             !(reg_classes[r] & rc) ||
5004             ((vtop->type.t & VT_BTYPE) == VT_LLONG &&
5005              !(reg_classes[vtop->r2] & rc2))) {
5006             r = get_reg(rc);
5007             if ((vtop->type.t & VT_BTYPE) == VT_LLONG) {
5008                 /* two register type load : expand to two words
5009                    temporarily */
5010                 if ((vtop->r & (VT_VALMASK | VT_LVAL)) == VT_CONST) {
5011                     /* load constant */
5012                     ll = vtop->c.ull;
5013                     vtop->c.ui = ll; /* first word */
5014                     load(r, vtop);
5015                     vtop->r = r; /* save register value */
5016                     vpushi(ll >> 32); /* second word */
5017                 } else if (r >= VT_CONST || /* XXX: test to VT_CONST incorrect ? */
5018                            (vtop->r & VT_LVAL)) {
5019                     /* We do not want to modifier the long long
5020                        pointer here, so the safest (and less
5021                        efficient) is to save all the other registers
5022                        in the stack. XXX: totally inefficient. */
5023                     save_regs(1);
5024                     /* load from memory */
5025                     load(r, vtop);
5026                     vdup();
5027                     vtop[-1].r = r; /* save register value */
5028                     /* increment pointer to get second word */
5029                     vtop->type.t = VT_INT;
5030                     gaddrof();
5031                     vpushi(4);
5032                     gen_op('+');
5033                     vtop->r |= VT_LVAL;
5034                 } else {
5035                     /* move registers */
5036                     load(r, vtop);
5037                     vdup();
5038                     vtop[-1].r = r; /* save register value */
5039                     vtop->r = vtop[-1].r2;
5040                 }
5041                 /* allocate second register */
5042                 r2 = get_reg(rc2);
5043                 load(r2, vtop);
5044                 vpop();
5045                 /* write second register */
5046                 vtop->r2 = r2;
5047             } else if ((vtop->r & VT_LVAL) && !is_float(vtop->type.t)) {
5048                 int t1, t;
5049                 /* lvalue of scalar type : need to use lvalue type
5050                    because of possible cast */
5051                 t = vtop->type.t;
5052                 t1 = t;
5053                 /* compute memory access type */
5054                 if (vtop->r & VT_LVAL_BYTE)
5055                     t = VT_BYTE;
5056                 else if (vtop->r & VT_LVAL_SHORT)
5057                     t = VT_SHORT;
5058                 if (vtop->r & VT_LVAL_UNSIGNED)
5059                     t |= VT_UNSIGNED;
5060                 vtop->type.t = t;
5061                 load(r, vtop);
5062                 /* restore wanted type */
5063                 vtop->type.t = t1;
5064             } else {
5065                 /* one register type load */
5066                 load(r, vtop);
5067             }
5068         }
5069         vtop->r = r;
5070 #ifdef TCC_TARGET_C67
5071         /* uses register pairs for doubles */
5072         if ((vtop->type.t & VT_BTYPE) == VT_DOUBLE)
5073             vtop->r2 = r+1;
5074 #endif
5075     }
5076     return r;
5077 }
5078
5079 /* generate vtop[-1] and vtop[0] in resp. classes rc1 and rc2 */
5080 void gv2(int rc1, int rc2)
5081 {
5082     int v;
5083
5084     /* generate more generic register first. But VT_JMP or VT_CMP
5085        values must be generated first in all cases to avoid possible
5086        reload errors */
5087     v = vtop[0].r & VT_VALMASK;
5088     if (v != VT_CMP && (v & ~1) != VT_JMP && rc1 <= rc2) {
5089         vswap();
5090         gv(rc1);
5091         vswap();
5092         gv(rc2);
5093         /* test if reload is needed for first register */
5094         if ((vtop[-1].r & VT_VALMASK) >= VT_CONST) {
5095             vswap();
5096             gv(rc1);
5097             vswap();
5098         }
5099     } else {
5100         gv(rc2);
5101         vswap();
5102         gv(rc1);
5103         vswap();
5104         /* test if reload is needed for first register */
5105         if ((vtop[0].r & VT_VALMASK) >= VT_CONST) {
5106             gv(rc2);
5107         }
5108     }
5109 }
5110
5111 /* expand long long on stack in two int registers */
5112 void lexpand(void)
5113 {
5114     int u;
5115
5116     u = vtop->type.t & VT_UNSIGNED;
5117     gv(RC_INT);
5118     vdup();
5119     vtop[0].r = vtop[-1].r2;
5120     vtop[0].r2 = VT_CONST;
5121     vtop[-1].r2 = VT_CONST;
5122     vtop[0].type.t = VT_INT | u;
5123     vtop[-1].type.t = VT_INT | u;
5124 }
5125
5126 #ifdef TCC_TARGET_ARM
5127 /* expand long long on stack */
5128 void lexpand_nr(void)
5129 {
5130     int u,v;
5131
5132     u = vtop->type.t & VT_UNSIGNED;
5133     vdup();
5134     vtop->r2 = VT_CONST;
5135     vtop->type.t = VT_INT | u;
5136     v=vtop[-1].r & (VT_VALMASK | VT_LVAL);
5137     if (v == VT_CONST) {
5138       vtop[-1].c.ui = vtop->c.ull;
5139       vtop->c.ui = vtop->c.ull >> 32;
5140       vtop->r = VT_CONST;
5141     } else if (v == (VT_LVAL|VT_CONST) || v == (VT_LVAL|VT_LOCAL)) {
5142       vtop->c.ui += 4;
5143       vtop->r = vtop[-1].r;
5144     } else if (v > VT_CONST) {
5145       vtop--;
5146       lexpand();
5147     } else
5148       vtop->r = vtop[-1].r2;
5149     vtop[-1].r2 = VT_CONST;
5150     vtop[-1].type.t = VT_INT | u;
5151 }
5152 #endif
5153
5154 /* build a long long from two ints */
5155 void lbuild(int t)
5156 {
5157     gv2(RC_INT, RC_INT);
5158     vtop[-1].r2 = vtop[0].r;
5159     vtop[-1].type.t = t;
5160     vpop();
5161 }
5162
5163 /* rotate n first stack elements to the bottom
5164    I1 ... In -> I2 ... In I1 [top is right]
5165 */
5166 void vrotb(int n)
5167 {
5168     int i;
5169     SValue tmp;
5170
5171     tmp = vtop[-n + 1];
5172     for(i=-n+1;i!=0;i++)
5173         vtop[i] = vtop[i+1];
5174     vtop[0] = tmp;
5175 }
5176
5177 /* rotate n first stack elements to the top
5178    I1 ... In -> In I1 ... I(n-1)  [top is right]
5179  */
5180 void vrott(int n)
5181 {
5182     int i;
5183     SValue tmp;
5184
5185     tmp = vtop[0];
5186     for(i = 0;i < n - 1; i++)
5187         vtop[-i] = vtop[-i - 1];
5188     vtop[-n + 1] = tmp;
5189 }
5190
5191 #ifdef TCC_TARGET_ARM
5192 /* like vrott but in other direction
5193    In ... I1 -> I(n-1) ... I1 In  [top is right]
5194  */
5195 void vnrott(int n)
5196 {
5197     int i;
5198     SValue tmp;
5199
5200     tmp = vtop[-n + 1];
5201     for(i = n - 1; i > 0; i--)
5202         vtop[-i] = vtop[-i + 1];
5203     vtop[0] = tmp;
5204 }
5205 #endif
5206
5207 /* pop stack value */
5208 void vpop(void)
5209 {
5210     int v;
5211     v = vtop->r & VT_VALMASK;
5212 #ifdef TCC_TARGET_I386
5213     /* for x86, we need to pop the FP stack */
5214     if (v == TREG_ST0 && !nocode_wanted) {
5215         o(0xd9dd); /* fstp %st(1) */
5216     } else
5217 #endif
5218     if (v == VT_JMP || v == VT_JMPI) {
5219         /* need to put correct jump if && or || without test */
5220         gsym(vtop->c.ul);
5221     }
5222     vtop--;
5223 }
5224
5225 /* convert stack entry to register and duplicate its value in another
5226    register */
5227 void gv_dup(void)
5228 {
5229     int rc, t, r, r1;
5230     SValue sv;
5231
5232     t = vtop->type.t;
5233     if ((t & VT_BTYPE) == VT_LLONG) {
5234         lexpand();
5235         gv_dup();
5236         vswap();
5237         vrotb(3);
5238         gv_dup();
5239         vrotb(4);
5240         /* stack: H L L1 H1 */
5241         lbuild(t);
5242         vrotb(3);
5243         vrotb(3);
5244         vswap();
5245         lbuild(t);
5246         vswap();
5247     } else {
5248         /* duplicate value */
5249         rc = RC_INT;
5250         sv.type.t = VT_INT;
5251         if (is_float(t)) {
5252             rc = RC_FLOAT;
5253             sv.type.t = t;
5254         }
5255         r = gv(rc);
5256         r1 = get_reg(rc);
5257         sv.r = r;
5258         sv.c.ul = 0;
5259         load(r1, &sv); /* move r to r1 */
5260         vdup();
5261         /* duplicates value */
5262         vtop->r = r1;
5263     }
5264 }
5265
5266 /* generate CPU independent (unsigned) long long operations */
5267 void gen_opl(int op)
5268 {
5269     int t, a, b, op1, c, i;
5270     int func;
5271     unsigned short reg_iret = REG_IRET;
5272     unsigned short reg_lret = REG_LRET;
5273     SValue tmp;
5274
5275     switch(op) {
5276     case '/':
5277     case TOK_PDIV:
5278         func = TOK___divdi3;
5279         goto gen_func;
5280     case TOK_UDIV:
5281         func = TOK___udivdi3;
5282         goto gen_func;
5283     case '%':
5284         func = TOK___moddi3;
5285         goto gen_mod_func;
5286     case TOK_UMOD:
5287         func = TOK___umoddi3;
5288     gen_mod_func:
5289 #ifdef TCC_ARM_EABI
5290         reg_iret = TREG_R2;
5291         reg_lret = TREG_R3;
5292 #endif
5293     gen_func:
5294         /* call generic long long function */
5295         vpush_global_sym(&func_old_type, func);
5296         vrott(3);
5297         gfunc_call(2);
5298         vpushi(0);
5299         vtop->r = reg_iret;
5300         vtop->r2 = reg_lret;
5301         break;
5302     case '^':
5303     case '&':
5304     case '|':
5305     case '*':
5306     case '+':
5307     case '-':
5308         t = vtop->type.t;
5309         vswap();
5310         lexpand();
5311         vrotb(3);
5312         lexpand();
5313         /* stack: L1 H1 L2 H2 */
5314         tmp = vtop[0];
5315         vtop[0] = vtop[-3];
5316         vtop[-3] = tmp;
5317         tmp = vtop[-2];
5318         vtop[-2] = vtop[-3];
5319         vtop[-3] = tmp;
5320         vswap();
5321         /* stack: H1 H2 L1 L2 */
5322         if (op == '*') {
5323             vpushv(vtop - 1);
5324             vpushv(vtop - 1);
5325             gen_op(TOK_UMULL);
5326             lexpand();
5327             /* stack: H1 H2 L1 L2 ML MH */
5328             for(i=0;i<4;i++)
5329                 vrotb(6);
5330             /* stack: ML MH H1 H2 L1 L2 */
5331             tmp = vtop[0];
5332             vtop[0] = vtop[-2];
5333             vtop[-2] = tmp;
5334             /* stack: ML MH H1 L2 H2 L1 */
5335             gen_op('*');
5336             vrotb(3);
5337             vrotb(3);
5338             gen_op('*');
5339             /* stack: ML MH M1 M2 */
5340             gen_op('+');
5341             gen_op('+');
5342         } else if (op == '+' || op == '-') {
5343             /* XXX: add non carry method too (for MIPS or alpha) */
5344             if (op == '+')
5345                 op1 = TOK_ADDC1;
5346             else
5347                 op1 = TOK_SUBC1;
5348             gen_op(op1);
5349             /* stack: H1 H2 (L1 op L2) */
5350             vrotb(3);
5351             vrotb(3);
5352             gen_op(op1 + 1); /* TOK_xxxC2 */
5353         } else {
5354             gen_op(op);
5355             /* stack: H1 H2 (L1 op L2) */
5356             vrotb(3);
5357             vrotb(3);
5358             /* stack: (L1 op L2) H1 H2 */
5359             gen_op(op);
5360             /* stack: (L1 op L2) (H1 op H2) */
5361         }
5362         /* stack: L H */
5363         lbuild(t);
5364         break;
5365     case TOK_SAR:
5366     case TOK_SHR:
5367     case TOK_SHL:
5368         if ((vtop->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST) {
5369             t = vtop[-1].type.t;
5370             vswap();
5371             lexpand();
5372             vrotb(3);
5373             /* stack: L H shift */
5374             c = (int)vtop->c.i;
5375             /* constant: simpler */
5376             /* NOTE: all comments are for SHL. the other cases are
5377                done by swaping words */
5378             vpop();
5379             if (op != TOK_SHL)
5380                 vswap();
5381             if (c >= 32) {
5382                 /* stack: L H */
5383                 vpop();
5384                 if (c > 32) {
5385                     vpushi(c - 32);
5386                     gen_op(op);
5387                 }
5388                 if (op != TOK_SAR) {
5389                     vpushi(0);
5390                 } else {
5391                     gv_dup();
5392                     vpushi(31);
5393                     gen_op(TOK_SAR);
5394                 }
5395                 vswap();
5396             } else {
5397                 vswap();
5398                 gv_dup();
5399                 /* stack: H L L */
5400                 vpushi(c);
5401                 gen_op(op);
5402                 vswap();
5403                 vpushi(32 - c);
5404                 if (op == TOK_SHL)
5405                     gen_op(TOK_SHR);
5406                 else
5407                     gen_op(TOK_SHL);
5408                 vrotb(3);
5409                 /* stack: L L H */
5410                 vpushi(c);
5411                 if (op == TOK_SHL)
5412                     gen_op(TOK_SHL);
5413                 else
5414                     gen_op(TOK_SHR);
5415                 gen_op('|');
5416             }
5417             if (op != TOK_SHL)
5418                 vswap();
5419             lbuild(t);
5420         } else {
5421             /* XXX: should provide a faster fallback on x86 ? */
5422             switch(op) {
5423             case TOK_SAR:
5424                 func = TOK___ashrdi3;
5425                 goto gen_func;
5426             case TOK_SHR:
5427                 func = TOK___lshrdi3;
5428                 goto gen_func;
5429             case TOK_SHL:
5430                 func = TOK___ashldi3;
5431                 goto gen_func;
5432             }
5433         }
5434         break;
5435     default:
5436         /* compare operations */
5437         t = vtop->type.t;
5438         vswap();
5439         lexpand();
5440         vrotb(3);
5441         lexpand();
5442         /* stack: L1 H1 L2 H2 */
5443         tmp = vtop[-1];
5444         vtop[-1] = vtop[-2];
5445         vtop[-2] = tmp;
5446         /* stack: L1 L2 H1 H2 */
5447         /* compare high */
5448         op1 = op;
5449         /* when values are equal, we need to compare low words. since
5450            the jump is inverted, we invert the test too. */
5451         if (op1 == TOK_LT)
5452             op1 = TOK_LE;
5453         else if (op1 == TOK_GT)
5454             op1 = TOK_GE;
5455         else if (op1 == TOK_ULT)
5456             op1 = TOK_ULE;
5457         else if (op1 == TOK_UGT)
5458             op1 = TOK_UGE;
5459         a = 0;
5460         b = 0;
5461         gen_op(op1);
5462         if (op1 != TOK_NE) {
5463             a = gtst(1, 0);
5464         }
5465         if (op != TOK_EQ) {
5466             /* generate non equal test */
5467             /* XXX: NOT PORTABLE yet */
5468             if (a == 0) {
5469                 b = gtst(0, 0);
5470             } else {
5471 #if defined(TCC_TARGET_I386)
5472                 b = psym(0x850f, 0);
5473 #elif defined(TCC_TARGET_ARM)
5474                 b = ind;
5475                 o(0x1A000000 | encbranch(ind, 0, 1));
5476 #elif defined(TCC_TARGET_C67)
5477                 error("not implemented");
5478 #else
5479 #error not supported
5480 #endif
5481             }
5482         }
5483         /* compare low. Always unsigned */
5484         op1 = op;
5485         if (op1 == TOK_LT)
5486             op1 = TOK_ULT;
5487         else if (op1 == TOK_LE)
5488             op1 = TOK_ULE;
5489         else if (op1 == TOK_GT)
5490             op1 = TOK_UGT;
5491         else if (op1 == TOK_GE)
5492             op1 = TOK_UGE;
5493         gen_op(op1);
5494         a = gtst(1, a);
5495         gsym(b);
5496         vseti(VT_JMPI, a);
5497         break;
5498     }
5499 }
5500
5501 /* handle integer constant optimizations and various machine
5502    independent opt */
5503 void gen_opic(int op)
5504 {
5505     int c1, c2, t1, t2, n;
5506     SValue *v1, *v2;
5507     long long l1, l2;
5508     typedef unsigned long long U;
5509
5510     v1 = vtop - 1;
5511     v2 = vtop;
5512     t1 = v1->type.t & VT_BTYPE;
5513     t2 = v2->type.t & VT_BTYPE;
5514     l1 = (t1 == VT_LLONG) ? v1->c.ll : v1->c.i;
5515     l2 = (t2 == VT_LLONG) ? v2->c.ll : v2->c.i;
5516
5517     /* currently, we cannot do computations with forward symbols */
5518     c1 = (v1->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST;
5519     c2 = (v2->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST;
5520     if (c1 && c2) {
5521         switch(op) {
5522         case '+': l1 += l2; break;
5523         case '-': l1 -= l2; break;
5524         case '&': l1 &= l2; break;
5525         case '^': l1 ^= l2; break;
5526         case '|': l1 |= l2; break;
5527         case '*': l1 *= l2; break;
5528
5529         case TOK_PDIV:
5530         case '/':
5531         case '%':
5532         case TOK_UDIV:
5533         case TOK_UMOD:
5534             /* if division by zero, generate explicit division */
5535             if (l2 == 0) {
5536                 if (const_wanted)
5537                     error("division by zero in constant");
5538                 goto general_case;
5539             }
5540             switch(op) {
5541             default: l1 /= l2; break;
5542             case '%': l1 %= l2; break;
5543             case TOK_UDIV: l1 = (U)l1 / l2; break;
5544             case TOK_UMOD: l1 = (U)l1 % l2; break;
5545             }
5546             break;
5547         case TOK_SHL: l1 <<= l2; break;
5548         case TOK_SHR: l1 = (U)l1 >> l2; break;
5549         case TOK_SAR: l1 >>= l2; break;
5550             /* tests */
5551         case TOK_ULT: l1 = (U)l1 < (U)l2; break;
5552         case TOK_UGE: l1 = (U)l1 >= (U)l2; break;
5553         case TOK_EQ: l1 = l1 == l2; break;
5554         case TOK_NE: l1 = l1 != l2; break;
5555         case TOK_ULE: l1 = (U)l1 <= (U)l2; break;
5556         case TOK_UGT: l1 = (U)l1 > (U)l2; break;
5557         case TOK_LT: l1 = l1 < l2; break;
5558         case TOK_GE: l1 = l1 >= l2; break;
5559         case TOK_LE: l1 = l1 <= l2; break;
5560         case TOK_GT: l1 = l1 > l2; break;
5561             /* logical */
5562         case TOK_LAND: l1 = l1 && l2; break;
5563         case TOK_LOR: l1 = l1 || l2; break;
5564         default:
5565             goto general_case;
5566         }
5567         v1->c.ll = l1;
5568         vtop--;
5569     } else {
5570         /* if commutative ops, put c2 as constant */
5571         if (c1 && (op == '+' || op == '&' || op == '^' ||
5572                    op == '|' || op == '*')) {
5573             vswap();
5574             c2 = c1; //c = c1, c1 = c2, c2 = c;
5575             l2 = l1; //l = l1, l1 = l2, l2 = l;
5576         }
5577         /* Filter out NOP operations like x*1, x-0, x&-1... */
5578         if (c2 && (((op == '*' || op == '/' || op == TOK_UDIV ||
5579                      op == TOK_PDIV) &&
5580                     l2 == 1) ||
5581                    ((op == '+' || op == '-' || op == '|' || op == '^' ||
5582                      op == TOK_SHL || op == TOK_SHR || op == TOK_SAR) &&
5583                     l2 == 0) ||
5584                    (op == '&' &&
5585                     l2 == -1))) {
5586             /* nothing to do */
5587             vtop--;
5588         } else if (c2 && (op == '*' || op == TOK_PDIV || op == TOK_UDIV)) {
5589             /* try to use shifts instead of muls or divs */
5590             if (l2 > 0 && (l2 & (l2 - 1)) == 0) {
5591                 n = -1;
5592                 while (l2) {
5593                     l2 >>= 1;
5594                     n++;
5595                 }
5596                 vtop->c.ll = n;
5597                 if (op == '*')
5598                     op = TOK_SHL;
5599                 else if (op == TOK_PDIV)
5600                     op = TOK_SAR;
5601                 else
5602                     op = TOK_SHR;
5603             }
5604             goto general_case;
5605         } else if (c2 && (op == '+' || op == '-') &&
5606                    ((vtop[-1].r & (VT_VALMASK | VT_LVAL | VT_SYM)) ==
5607                    (VT_CONST | VT_SYM) ||
5608                    (vtop[-1].r & (VT_VALMASK | VT_LVAL)) == VT_LOCAL)) {
5609             /* symbol + constant case */
5610             if (op == '-')
5611                 l2 = -l2;
5612             vtop--;
5613             vtop->c.ll += l2;
5614         } else {
5615         general_case:
5616             if (!nocode_wanted) {
5617                 /* call low level op generator */
5618                 if (t1 == VT_LLONG || t2 == VT_LLONG)
5619                     gen_opl(op);
5620                 else
5621                     gen_opi(op);
5622             } else {
5623                 vtop--;
5624             }
5625         }
5626     }
5627 }
5628
5629 /* generate a floating point operation with constant propagation */
5630 void gen_opif(int op)
5631 {
5632     int c1, c2;
5633     SValue *v1, *v2;
5634     long double f1, f2;
5635
5636     v1 = vtop - 1;
5637     v2 = vtop;
5638     /* currently, we cannot do computations with forward symbols */
5639     c1 = (v1->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST;
5640     c2 = (v2->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST;
5641     if (c1 && c2) {
5642         if (v1->type.t == VT_FLOAT) {
5643             f1 = v1->c.f;
5644             f2 = v2->c.f;
5645         } else if (v1->type.t == VT_DOUBLE) {
5646             f1 = v1->c.d;
5647             f2 = v2->c.d;
5648         } else {
5649             f1 = v1->c.ld;
5650             f2 = v2->c.ld;
5651         }
5652
5653         /* NOTE: we only do constant propagation if finite number (not
5654            NaN or infinity) (ANSI spec) */
5655         if (!ieee_finite(f1) || !ieee_finite(f2))
5656             goto general_case;
5657
5658         switch(op) {
5659         case '+': f1 += f2; break;
5660         case '-': f1 -= f2; break;
5661         case '*': f1 *= f2; break;
5662         case '/':
5663             if (f2 == 0.0) {
5664                 if (const_wanted)
5665                     error("division by zero in constant");
5666                 goto general_case;
5667             }
5668             f1 /= f2;
5669             break;
5670             /* XXX: also handles tests ? */
5671         default:
5672             goto general_case;
5673         }
5674         /* XXX: overflow test ? */
5675         if (v1->type.t == VT_FLOAT) {
5676             v1->c.f = f1;
5677         } else if (v1->type.t == VT_DOUBLE) {
5678             v1->c.d = f1;
5679         } else {
5680             v1->c.ld = f1;
5681         }
5682         vtop--;
5683     } else {
5684     general_case:
5685         if (!nocode_wanted) {
5686             gen_opf(op);
5687         } else {
5688             vtop--;
5689         }
5690     }
5691 }
5692
5693 static int pointed_size(CType *type)
5694 {
5695     int align;
5696     return type_size(pointed_type(type), &align);
5697 }
5698
5699 static inline int is_null_pointer(SValue *p)
5700 {
5701     if ((p->r & (VT_VALMASK | VT_LVAL | VT_SYM)) != VT_CONST)
5702         return 0;
5703     return ((p->type.t & VT_BTYPE) == VT_INT && p->c.i == 0) ||
5704         ((p->type.t & VT_BTYPE) == VT_LLONG && p->c.ll == 0);
5705 }
5706
5707 static inline int is_integer_btype(int bt)
5708 {
5709     return (bt == VT_BYTE || bt == VT_SHORT ||
5710             bt == VT_INT || bt == VT_LLONG);
5711 }
5712
5713 /* check types for comparison or substraction of pointers */
5714 static void check_comparison_pointer_types(SValue *p1, SValue *p2, int op)
5715 {
5716     CType *type1, *type2, tmp_type1, tmp_type2;
5717     int bt1, bt2;
5718
5719     /* null pointers are accepted for all comparisons as gcc */
5720     if (is_null_pointer(p1) || is_null_pointer(p2))
5721         return;
5722     type1 = &p1->type;
5723     type2 = &p2->type;
5724     bt1 = type1->t & VT_BTYPE;
5725     bt2 = type2->t & VT_BTYPE;
5726     /* accept comparison between pointer and integer with a warning */
5727     if ((is_integer_btype(bt1) || is_integer_btype(bt2)) && op != '-') {
5728         if (op != TOK_LOR && op != TOK_LAND )
5729             warning("comparison between pointer and integer");
5730         return;
5731     }
5732
5733     /* both must be pointers or implicit function pointers */
5734     if (bt1 == VT_PTR) {
5735         type1 = pointed_type(type1);
5736     } else if (bt1 != VT_FUNC)
5737         goto invalid_operands;
5738
5739     if (bt2 == VT_PTR) {
5740         type2 = pointed_type(type2);
5741     } else if (bt2 != VT_FUNC) {
5742     invalid_operands:
5743         error("invalid operands to binary %s", get_tok_str(op, NULL));
5744     }
5745     if ((type1->t & VT_BTYPE) == VT_VOID ||
5746         (type2->t & VT_BTYPE) == VT_VOID)
5747         return;
5748     tmp_type1 = *type1;
5749     tmp_type2 = *type2;
5750     tmp_type1.t &= ~(VT_UNSIGNED | VT_CONSTANT | VT_VOLATILE);
5751     tmp_type2.t &= ~(VT_UNSIGNED | VT_CONSTANT | VT_VOLATILE);
5752     if (!is_compatible_types(&tmp_type1, &tmp_type2)) {
5753         /* gcc-like error if '-' is used */
5754         if (op == '-')
5755             goto invalid_operands;
5756         else
5757             warning("comparison of distinct pointer types lacks a cast");
5758     }
5759 }
5760
5761 /* generic gen_op: handles types problems */
5762 void gen_op(int op)
5763 {
5764     int u, t1, t2, bt1, bt2, t;
5765     CType type1;
5766
5767     t1 = vtop[-1].type.t;
5768     t2 = vtop[0].type.t;
5769     bt1 = t1 & VT_BTYPE;
5770     bt2 = t2 & VT_BTYPE;
5771
5772     if (bt1 == VT_PTR || bt2 == VT_PTR) {
5773         /* at least one operand is a pointer */
5774         /* relationnal op: must be both pointers */
5775         if (op >= TOK_ULT && op <= TOK_LOR) {
5776             check_comparison_pointer_types(vtop - 1, vtop, op);
5777             /* pointers are handled are unsigned */
5778             t = VT_INT | VT_UNSIGNED;
5779             goto std_op;
5780         }
5781         /* if both pointers, then it must be the '-' op */
5782         if (bt1 == VT_PTR && bt2 == VT_PTR) {
5783             if (op != '-')
5784                 error("cannot use pointers here");
5785             check_comparison_pointer_types(vtop - 1, vtop, op);
5786             /* XXX: check that types are compatible */
5787             u = pointed_size(&vtop[-1].type);
5788             gen_opic(op);
5789             /* set to integer type */
5790             vtop->type.t = VT_INT;
5791             vpushi(u);
5792             gen_op(TOK_PDIV);
5793         } else {
5794             /* exactly one pointer : must be '+' or '-'. */
5795             if (op != '-' && op != '+')
5796                 error("cannot use pointers here");
5797             /* Put pointer as first operand */
5798             if (bt2 == VT_PTR) {
5799                 vswap();
5800                 swap(&t1, &t2);
5801             }
5802             type1 = vtop[-1].type;
5803             /* XXX: cast to int ? (long long case) */
5804             vpushi(pointed_size(&vtop[-1].type));
5805             gen_op('*');
5806 #ifdef CONFIG_TCC_BCHECK
5807             /* if evaluating constant expression, no code should be
5808                generated, so no bound check */
5809             if (do_bounds_check && !const_wanted) {
5810                 /* if bounded pointers, we generate a special code to
5811                    test bounds */
5812                 if (op == '-') {
5813                     vpushi(0);
5814                     vswap();
5815                     gen_op('-');
5816                 }
5817                 gen_bounded_ptr_add();
5818             } else
5819 #endif
5820             {
5821                 gen_opic(op);
5822             }
5823             /* put again type if gen_opic() swaped operands */
5824             vtop->type = type1;
5825         }
5826     } else if (is_float(bt1) || is_float(bt2)) {
5827         /* compute bigger type and do implicit casts */
5828         if (bt1 == VT_LDOUBLE || bt2 == VT_LDOUBLE) {
5829             t = VT_LDOUBLE;
5830         } else if (bt1 == VT_DOUBLE || bt2 == VT_DOUBLE) {
5831             t = VT_DOUBLE;
5832         } else {
5833             t = VT_FLOAT;
5834         }
5835         /* floats can only be used for a few operations */
5836         if (op != '+' && op != '-' && op != '*' && op != '/' &&
5837             (op < TOK_ULT || op > TOK_GT))
5838             error("invalid operands for binary operation");
5839         goto std_op;
5840     } else if (bt1 == VT_LLONG || bt2 == VT_LLONG) {
5841         /* cast to biggest op */
5842         t = VT_LLONG;
5843         /* convert to unsigned if it does not fit in a long long */
5844         if ((t1 & (VT_BTYPE | VT_UNSIGNED)) == (VT_LLONG | VT_UNSIGNED) ||
5845             (t2 & (VT_BTYPE | VT_UNSIGNED)) == (VT_LLONG | VT_UNSIGNED))
5846             t |= VT_UNSIGNED;
5847         goto std_op;
5848     } else {
5849         /* integer operations */
5850         t = VT_INT;
5851         /* convert to unsigned if it does not fit in an integer */
5852         if ((t1 & (VT_BTYPE | VT_UNSIGNED)) == (VT_INT | VT_UNSIGNED) ||
5853             (t2 & (VT_BTYPE | VT_UNSIGNED)) == (VT_INT | VT_UNSIGNED))
5854             t |= VT_UNSIGNED;
5855     std_op:
5856         /* XXX: currently, some unsigned operations are explicit, so
5857            we modify them here */
5858         if (t & VT_UNSIGNED) {
5859             if (op == TOK_SAR)
5860                 op = TOK_SHR;
5861             else if (op == '/')
5862                 op = TOK_UDIV;
5863             else if (op == '%')
5864                 op = TOK_UMOD;
5865             else if (op == TOK_LT)
5866                 op = TOK_ULT;
5867             else if (op == TOK_GT)
5868                 op = TOK_UGT;
5869             else if (op == TOK_LE)
5870                 op = TOK_ULE;
5871             else if (op == TOK_GE)
5872                 op = TOK_UGE;
5873         }
5874         vswap();
5875         type1.t = t;
5876         gen_cast(&type1);
5877         vswap();
5878         /* special case for shifts and long long: we keep the shift as
5879            an integer */
5880         if (op == TOK_SHR || op == TOK_SAR || op == TOK_SHL)
5881             type1.t = VT_INT;
5882         gen_cast(&type1);
5883         if (is_float(t))
5884             gen_opif(op);
5885         else
5886             gen_opic(op);
5887         if (op >= TOK_ULT && op <= TOK_GT) {
5888             /* relationnal op: the result is an int */
5889             vtop->type.t = VT_INT;
5890         } else {
5891             vtop->type.t = t;
5892         }
5893     }
5894 }
5895
5896 #ifndef TCC_TARGET_ARM
5897 /* generic itof for unsigned long long case */
5898 void gen_cvt_itof1(int t)
5899 {
5900     if ((vtop->type.t & (VT_BTYPE | VT_UNSIGNED)) ==
5901         (VT_LLONG | VT_UNSIGNED)) {
5902
5903         if (t == VT_FLOAT)
5904             vpush_global_sym(&func_old_type, TOK___floatundisf);
5905 #if LDOUBLE_SIZE != 8
5906         else if (t == VT_LDOUBLE)
5907             vpush_global_sym(&func_old_type, TOK___floatundixf);
5908 #endif
5909         else
5910             vpush_global_sym(&func_old_type, TOK___floatundidf);
5911         vrott(2);
5912         gfunc_call(1);
5913         vpushi(0);
5914         vtop->r = REG_FRET;
5915     } else {
5916         gen_cvt_itof(t);
5917     }
5918 }
5919 #endif
5920
5921 /* generic ftoi for unsigned long long case */
5922 void gen_cvt_ftoi1(int t)
5923 {
5924     int st;
5925
5926     if (t == (VT_LLONG | VT_UNSIGNED)) {
5927         /* not handled natively */
5928         st = vtop->type.t & VT_BTYPE;
5929         if (st == VT_FLOAT)
5930             vpush_global_sym(&func_old_type, TOK___fixunssfdi);
5931 #if LDOUBLE_SIZE != 8
5932         else if (st == VT_LDOUBLE)
5933             vpush_global_sym(&func_old_type, TOK___fixunsxfdi);
5934 #endif
5935         else
5936             vpush_global_sym(&func_old_type, TOK___fixunsdfdi);
5937         vrott(2);
5938         gfunc_call(1);
5939         vpushi(0);
5940         vtop->r = REG_IRET;
5941         vtop->r2 = REG_LRET;
5942     } else {
5943         gen_cvt_ftoi(t);
5944     }
5945 }
5946
5947 /* force char or short cast */
5948 void force_charshort_cast(int t)
5949 {
5950     int bits, dbt;
5951     dbt = t & VT_BTYPE;
5952     /* XXX: add optimization if lvalue : just change type and offset */
5953     if (dbt == VT_BYTE)
5954         bits = 8;
5955     else
5956         bits = 16;
5957     if (t & VT_UNSIGNED) {
5958         vpushi((1 << bits) - 1);
5959         gen_op('&');
5960     } else {
5961         bits = 32 - bits;
5962         vpushi(bits);
5963         gen_op(TOK_SHL);
5964         /* result must be signed or the SAR is converted to an SHL
5965            This was not the case when "t" was a signed short
5966            and the last value on the stack was an unsigned int */
5967         vtop->type.t &= ~VT_UNSIGNED;
5968         vpushi(bits);
5969         gen_op(TOK_SAR);
5970     }
5971 }
5972
5973 /* cast 'vtop' to 'type'. Casting to bitfields is forbidden. */
5974 static void gen_cast(CType *type)
5975 {
5976     int sbt, dbt, sf, df, c;
5977
5978     /* special delayed cast for char/short */
5979     /* XXX: in some cases (multiple cascaded casts), it may still
5980        be incorrect */
5981     if (vtop->r & VT_MUSTCAST) {
5982         vtop->r &= ~VT_MUSTCAST;
5983         force_charshort_cast(vtop->type.t);
5984     }
5985
5986     /* bitfields first get cast to ints */
5987     if (vtop->type.t & VT_BITFIELD) {
5988         gv(RC_INT);
5989     }
5990
5991     dbt = type->t & (VT_BTYPE | VT_UNSIGNED);
5992     sbt = vtop->type.t & (VT_BTYPE | VT_UNSIGNED);
5993
5994     if (sbt != dbt) {
5995         sf = is_float(sbt);
5996         df = is_float(dbt);
5997         c = (vtop->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST;
5998         if (c) {
5999             /* constant case: we can do it now */
6000             /* XXX: in ISOC, cannot do it if error in convert */
6001             if (sbt == VT_FLOAT)
6002                 vtop->c.ld = vtop->c.f;
6003             else if (sbt == VT_DOUBLE)
6004                 vtop->c.ld = vtop->c.d;
6005
6006             if (df) {
6007                 if ((sbt & VT_BTYPE) == VT_LLONG) {
6008                     if (sbt & VT_UNSIGNED)
6009                         vtop->c.ld = vtop->c.ull;
6010                     else
6011                         vtop->c.ld = vtop->c.ll;
6012                 } else if(!sf) {
6013                     if (sbt & VT_UNSIGNED)
6014                         vtop->c.ld = vtop->c.ui;
6015                     else
6016                         vtop->c.ld = vtop->c.i;
6017                 }
6018
6019                 if (dbt == VT_FLOAT)
6020                     vtop->c.f = (float)vtop->c.ld;
6021                 else if (dbt == VT_DOUBLE)
6022                     vtop->c.d = (double)vtop->c.ld;
6023             } else if (sf && dbt == (VT_LLONG|VT_UNSIGNED)) {
6024                 vtop->c.ull = (unsigned long long)vtop->c.ld;
6025             } else if (sf && dbt == VT_BOOL) {
6026                 vtop->c.i = (vtop->c.ld != 0);
6027             } else {
6028                 if(sf)
6029                     vtop->c.ll = (long long)vtop->c.ld;
6030                 else if (sbt == (VT_LLONG|VT_UNSIGNED))
6031                     vtop->c.ll = vtop->c.ull;
6032                 else if (sbt & VT_UNSIGNED)
6033                     vtop->c.ll = vtop->c.ui;
6034                 else if (sbt != VT_LLONG)
6035                     vtop->c.ll = vtop->c.i;
6036
6037                 if (dbt == (VT_LLONG|VT_UNSIGNED))
6038                     vtop->c.ull = vtop->c.ll;
6039                 else if (dbt == VT_BOOL)
6040                     vtop->c.i = (vtop->c.ll != 0);
6041                 else if (dbt != VT_LLONG) {
6042                     int s = 0;
6043                     if ((dbt & VT_BTYPE) == VT_BYTE)
6044                         s = 24;
6045                     else if ((dbt & VT_BTYPE) == VT_SHORT)
6046                         s = 16;
6047
6048                     if(dbt & VT_UNSIGNED)
6049                         vtop->c.ui = ((unsigned int)vtop->c.ll << s) >> s;
6050                     else
6051                         vtop->c.i = ((int)vtop->c.ll << s) >> s;
6052                 }
6053             }
6054         } else if (!nocode_wanted) {
6055             /* non constant case: generate code */
6056             if (sf && df) {
6057                 /* convert from fp to fp */
6058                 gen_cvt_ftof(dbt);
6059             } else if (df) {
6060                 /* convert int to fp */
6061                 gen_cvt_itof1(dbt);
6062             } else if (sf) {
6063                 /* convert fp to int */
6064                 if (dbt == VT_BOOL) {
6065                      vpushi(0);
6066                      gen_op(TOK_NE);
6067                 } else {
6068                     /* we handle char/short/etc... with generic code */
6069                     if (dbt != (VT_INT | VT_UNSIGNED) &&
6070                         dbt != (VT_LLONG | VT_UNSIGNED) &&
6071                         dbt != VT_LLONG)
6072                         dbt = VT_INT;
6073                     gen_cvt_ftoi1(dbt);
6074                     if (dbt == VT_INT && (type->t & (VT_BTYPE | VT_UNSIGNED)) != dbt) {
6075                         /* additional cast for char/short... */
6076                         vtop->type.t = dbt;
6077                         gen_cast(type);
6078                     }
6079                 }
6080             } else if ((dbt & VT_BTYPE) == VT_LLONG) {
6081                 if ((sbt & VT_BTYPE) != VT_LLONG) {
6082                     /* scalar to long long */
6083                     /* machine independent conversion */
6084                     gv(RC_INT);
6085                     /* generate high word */
6086                     if (sbt == (VT_INT | VT_UNSIGNED)) {
6087                         vpushi(0);
6088                         gv(RC_INT);
6089                     } else {
6090                         gv_dup();
6091                         vpushi(31);
6092                         gen_op(TOK_SAR);
6093                     }
6094                     /* patch second register */
6095                     vtop[-1].r2 = vtop->r;
6096                     vpop();
6097                 }
6098             } else if (dbt == VT_BOOL) {
6099                 /* scalar to bool */
6100                 vpushi(0);
6101                 gen_op(TOK_NE);
6102             } else if ((dbt & VT_BTYPE) == VT_BYTE ||
6103                        (dbt & VT_BTYPE) == VT_SHORT) {
6104                 if (sbt == VT_PTR) {
6105                     vtop->type.t = VT_INT;
6106                     warning("nonportable conversion from pointer to char/short");
6107                 }
6108                 force_charshort_cast(dbt);
6109             } else if ((dbt & VT_BTYPE) == VT_INT) {
6110                 /* scalar to int */
6111                 if (sbt == VT_LLONG) {
6112                     /* from long long: just take low order word */
6113                     lexpand();
6114                     vpop();
6115                 }
6116                 /* if lvalue and single word type, nothing to do because
6117                    the lvalue already contains the real type size (see
6118                    VT_LVAL_xxx constants) */
6119             }
6120         } else
6121             expect("constant expression");
6122     } else if ((dbt & VT_BTYPE) == VT_PTR && !(vtop->r & VT_LVAL)) {
6123         /* if we are casting between pointer types,
6124            we must update the VT_LVAL_xxx size */
6125         vtop->r = (vtop->r & ~VT_LVAL_TYPE)
6126                   | (lvalue_type(type->ref->type.t) & VT_LVAL_TYPE);
6127     }
6128     vtop->type = *type;
6129 }
6130
6131 /* return type size. Put alignment at 'a' */
6132 static int type_size(CType *type, int *a)
6133 {
6134     Sym *s;
6135     int bt;
6136
6137     bt = type->t & VT_BTYPE;
6138     if (bt == VT_STRUCT) {
6139         /* struct/union */
6140         s = type->ref;
6141         *a = s->r;
6142         return s->c;
6143     } else if (bt == VT_PTR) {
6144         if (type->t & VT_ARRAY) {
6145             s = type->ref;
6146             return type_size(&s->type, a) * s->c;
6147         } else {
6148             *a = PTR_SIZE;
6149             return PTR_SIZE;
6150         }
6151     } else if (bt == VT_LDOUBLE) {
6152         *a = LDOUBLE_ALIGN;
6153         return LDOUBLE_SIZE;
6154     } else if (bt == VT_DOUBLE || bt == VT_LLONG) {
6155 #ifdef TCC_TARGET_I386
6156         *a = 4;
6157 #elif defined(TCC_TARGET_ARM)
6158 #ifdef TCC_ARM_EABI
6159         *a = 8;
6160 #else
6161         *a = 4;
6162 #endif
6163 #else
6164         *a = 8;
6165 #endif
6166         return 8;
6167     } else if (bt == VT_INT || bt == VT_ENUM || bt == VT_FLOAT) {
6168         *a = 4;
6169         return 4;
6170     } else if (bt == VT_SHORT) {
6171         *a = 2;
6172         return 2;
6173     } else {
6174         /* char, void, function, _Bool */
6175         *a = 1;
6176         return 1;
6177     }
6178 }
6179
6180 /* return the pointed type of t */
6181 static inline CType *pointed_type(CType *type)
6182 {
6183     return &type->ref->type;
6184 }
6185
6186 /* modify type so that its it is a pointer to type. */
6187 static void mk_pointer(CType *type)
6188 {
6189     Sym *s;
6190     s = sym_push(SYM_FIELD, type, 0, -1);
6191     type->t = VT_PTR | (type->t & ~VT_TYPE);
6192     type->ref = s;
6193 }
6194
6195 /* compare function types. OLD functions match any new functions */
6196 static int is_compatible_func(CType *type1, CType *type2)
6197 {
6198     Sym *s1, *s2;
6199
6200     s1 = type1->ref;
6201     s2 = type2->ref;
6202     if (!is_compatible_types(&s1->type, &s2->type))
6203         return 0;
6204     /* check func_call */
6205     if (FUNC_CALL(s1->r) != FUNC_CALL(s2->r))
6206         return 0;
6207     /* XXX: not complete */
6208     if (s1->c == FUNC_OLD || s2->c == FUNC_OLD)
6209         return 1;
6210     if (s1->c != s2->c)
6211         return 0;
6212     while (s1 != NULL) {
6213         if (s2 == NULL)
6214             return 0;
6215         if (!is_compatible_parameter_types(&s1->type, &s2->type))
6216             return 0;
6217         s1 = s1->next;
6218         s2 = s2->next;
6219     }
6220     if (s2)
6221         return 0;
6222     return 1;
6223 }
6224
6225 /* return true if type1 and type2 are the same.  If unqualified is
6226    true, qualifiers on the types are ignored.
6227
6228    - enums are not checked as gcc __builtin_types_compatible_p ()
6229  */
6230 static int compare_types(CType *type1, CType *type2, int unqualified)
6231 {
6232     int bt1, t1, t2;
6233
6234     t1 = type1->t & VT_TYPE;
6235     t2 = type2->t & VT_TYPE;
6236     if (unqualified) {
6237         /* strip qualifiers before comparing */
6238         t1 &= ~(VT_CONSTANT | VT_VOLATILE);
6239         t2 &= ~(VT_CONSTANT | VT_VOLATILE);
6240     }
6241     /* XXX: bitfields ? */
6242     if (t1 != t2)
6243         return 0;
6244     /* test more complicated cases */
6245     bt1 = t1 & VT_BTYPE;
6246     if (bt1 == VT_PTR) {
6247         type1 = pointed_type(type1);
6248         type2 = pointed_type(type2);
6249         return is_compatible_types(type1, type2);
6250     } else if (bt1 == VT_STRUCT) {
6251         return (type1->ref == type2->ref);
6252     } else if (bt1 == VT_FUNC) {
6253         return is_compatible_func(type1, type2);
6254     } else {
6255         return 1;
6256     }
6257 }
6258
6259 /* return true if type1 and type2 are exactly the same (including
6260    qualifiers).
6261 */
6262 static int is_compatible_types(CType *type1, CType *type2)
6263 {
6264     return compare_types(type1,type2,0);
6265 }
6266
6267 /* return true if type1 and type2 are the same (ignoring qualifiers).
6268 */
6269 static int is_compatible_parameter_types(CType *type1, CType *type2)
6270 {
6271     return compare_types(type1,type2,1);
6272 }
6273
6274 /* print a type. If 'varstr' is not NULL, then the variable is also
6275    printed in the type */
6276 /* XXX: union */
6277 /* XXX: add array and function pointers */
6278 void type_to_str(char *buf, int buf_size,
6279                  CType *type, const char *varstr)
6280 {
6281     int bt, v, t;
6282     Sym *s, *sa;
6283     char buf1[256];
6284     const char *tstr;
6285
6286     t = type->t & VT_TYPE;
6287     bt = t & VT_BTYPE;
6288     buf[0] = '\0';
6289     if (t & VT_CONSTANT)
6290         pstrcat(buf, buf_size, "const ");
6291     if (t & VT_VOLATILE)
6292         pstrcat(buf, buf_size, "volatile ");
6293     if (t & VT_UNSIGNED)
6294         pstrcat(buf, buf_size, "unsigned ");
6295     switch(bt) {
6296     case VT_VOID:
6297         tstr = "void";
6298         goto add_tstr;
6299     case VT_BOOL:
6300         tstr = "_Bool";
6301         goto add_tstr;
6302     case VT_BYTE:
6303         tstr = "char";
6304         goto add_tstr;
6305     case VT_SHORT:
6306         tstr = "short";
6307         goto add_tstr;
6308     case VT_INT:
6309         tstr = "int";
6310         goto add_tstr;
6311     case VT_LONG:
6312         tstr = "long";
6313         goto add_tstr;
6314     case VT_LLONG:
6315         tstr = "long long";
6316         goto add_tstr;
6317     case VT_FLOAT:
6318         tstr = "float";
6319         goto add_tstr;
6320     case VT_DOUBLE:
6321         tstr = "double";
6322         goto add_tstr;
6323     case VT_LDOUBLE:
6324         tstr = "long double";
6325     add_tstr:
6326         pstrcat(buf, buf_size, tstr);
6327         break;
6328     case VT_ENUM:
6329     case VT_STRUCT:
6330         if (bt == VT_STRUCT)
6331             tstr = "struct ";
6332         else
6333             tstr = "enum ";
6334         pstrcat(buf, buf_size, tstr);
6335         v = type->ref->v & ~SYM_STRUCT;
6336         if (v >= SYM_FIRST_ANOM)
6337             pstrcat(buf, buf_size, "<anonymous>");
6338         else
6339             pstrcat(buf, buf_size, get_tok_str(v, NULL));
6340         break;
6341     case VT_FUNC:
6342         s = type->ref;
6343         type_to_str(buf, buf_size, &s->type, varstr);
6344         pstrcat(buf, buf_size, "(");
6345         sa = s->next;
6346         while (sa != NULL) {
6347             type_to_str(buf1, sizeof(buf1), &sa->type, NULL);
6348             pstrcat(buf, buf_size, buf1);
6349             sa = sa->next;
6350             if (sa)
6351                 pstrcat(buf, buf_size, ", ");
6352         }
6353         pstrcat(buf, buf_size, ")");
6354         goto no_var;
6355     case VT_PTR:
6356         s = type->ref;
6357         pstrcpy(buf1, sizeof(buf1), "*");
6358         if (varstr)
6359             pstrcat(buf1, sizeof(buf1), varstr);
6360         type_to_str(buf, buf_size, &s->type, buf1);
6361         goto no_var;
6362     }
6363     if (varstr) {
6364         pstrcat(buf, buf_size, " ");
6365         pstrcat(buf, buf_size, varstr);
6366     }
6367  no_var: ;
6368 }
6369
6370 /* verify type compatibility to store vtop in 'dt' type, and generate
6371    casts if needed. */
6372 static void gen_assign_cast(CType *dt)
6373 {
6374     CType *st, *type1, *type2, tmp_type1, tmp_type2;
6375     char buf1[256], buf2[256];
6376     int dbt, sbt;
6377
6378     st = &vtop->type; /* source type */
6379     dbt = dt->t & VT_BTYPE;
6380     sbt = st->t & VT_BTYPE;
6381     if (dt->t & VT_CONSTANT)
6382         warning("assignment of read-only location");
6383     switch(dbt) {
6384     case VT_PTR:
6385         /* special cases for pointers */
6386         /* '0' can also be a pointer */
6387         if (is_null_pointer(vtop))
6388             goto type_ok;
6389         /* accept implicit pointer to integer cast with warning */
6390         if (is_integer_btype(sbt)) {
6391             warning("assignment makes pointer from integer without a cast");
6392             goto type_ok;
6393         }
6394         type1 = pointed_type(dt);
6395         /* a function is implicitely a function pointer */
6396         if (sbt == VT_FUNC) {
6397             if ((type1->t & VT_BTYPE) != VT_VOID &&
6398                 !is_compatible_types(pointed_type(dt), st))
6399                 goto error;
6400             else
6401                 goto type_ok;
6402         }
6403         if (sbt != VT_PTR)
6404             goto error;
6405         type2 = pointed_type(st);
6406         if ((type1->t & VT_BTYPE) == VT_VOID ||
6407             (type2->t & VT_BTYPE) == VT_VOID) {
6408             /* void * can match anything */
6409         } else {
6410             /* exact type match, except for unsigned */
6411             tmp_type1 = *type1;
6412             tmp_type2 = *type2;
6413             tmp_type1.t &= ~(VT_UNSIGNED | VT_CONSTANT | VT_VOLATILE);
6414             tmp_type2.t &= ~(VT_UNSIGNED | VT_CONSTANT | VT_VOLATILE);
6415             if (!is_compatible_types(&tmp_type1, &tmp_type2))
6416                 warning("assignment from incompatible pointer type");
6417         }
6418         /* check const and volatile */
6419         if ((!(type1->t & VT_CONSTANT) && (type2->t & VT_CONSTANT)) ||
6420             (!(type1->t & VT_VOLATILE) && (type2->t & VT_VOLATILE)))
6421             warning("assignment discards qualifiers from pointer target type");
6422         break;
6423     case VT_BYTE:
6424     case VT_SHORT:
6425     case VT_INT:
6426     case VT_LLONG:
6427         if (sbt == VT_PTR || sbt == VT_FUNC) {
6428             warning("assignment makes integer from pointer without a cast");
6429         }
6430         /* XXX: more tests */
6431         break;
6432     case VT_STRUCT:
6433         tmp_type1 = *dt;
6434         tmp_type2 = *st;
6435         tmp_type1.t &= ~(VT_CONSTANT | VT_VOLATILE);
6436         tmp_type2.t &= ~(VT_CONSTANT | VT_VOLATILE);
6437         if (!is_compatible_types(&tmp_type1, &tmp_type2)) {
6438         error:
6439             type_to_str(buf1, sizeof(buf1), st, NULL);
6440             type_to_str(buf2, sizeof(buf2), dt, NULL);
6441             error("cannot cast '%s' to '%s'", buf1, buf2);
6442         }
6443         break;
6444     }
6445  type_ok:
6446     gen_cast(dt);
6447 }
6448
6449 /* store vtop in lvalue pushed on stack */
6450 void vstore(void)
6451 {
6452     int sbt, dbt, ft, r, t, size, align, bit_size, bit_pos, rc, delayed_cast;
6453
6454     ft = vtop[-1].type.t;
6455     sbt = vtop->type.t & VT_BTYPE;
6456     dbt = ft & VT_BTYPE;
6457     if (((sbt == VT_INT || sbt == VT_SHORT) && dbt == VT_BYTE) ||
6458         (sbt == VT_INT && dbt == VT_SHORT)) {
6459         /* optimize char/short casts */
6460         delayed_cast = VT_MUSTCAST;
6461         vtop->type.t = ft & (VT_TYPE & ~(VT_BITFIELD | (-1 << VT_STRUCT_SHIFT)));
6462         /* XXX: factorize */
6463         if (ft & VT_CONSTANT)
6464             warning("assignment of read-only location");
6465     } else {
6466         delayed_cast = 0;
6467         if (!(ft & VT_BITFIELD))
6468             gen_assign_cast(&vtop[-1].type);
6469     }
6470
6471     if (sbt == VT_STRUCT) {
6472         /* if structure, only generate pointer */
6473         /* structure assignment : generate memcpy */
6474         /* XXX: optimize if small size */
6475         if (!nocode_wanted) {
6476             size = type_size(&vtop->type, &align);
6477
6478 #ifdef TCC_ARM_EABI
6479             if(!(align & 7))
6480                 vpush_global_sym(&func_old_type, TOK_memcpy8);
6481             else if(!(align & 3))
6482                 vpush_global_sym(&func_old_type, TOK_memcpy4);
6483             else
6484 #endif
6485             vpush_global_sym(&func_old_type, TOK_memcpy);
6486
6487             /* destination */
6488             vpushv(vtop - 2);
6489             vtop->type.t = VT_INT;
6490             gaddrof();
6491             /* source */
6492             vpushv(vtop - 2);
6493             vtop->type.t = VT_INT;
6494             gaddrof();
6495             /* type size */
6496             vpushi(size);
6497             gfunc_call(3);
6498
6499             vswap();
6500             vpop();
6501         } else {
6502             vswap();
6503             vpop();
6504         }
6505         /* leave source on stack */
6506     } else if (ft & VT_BITFIELD) {
6507         /* bitfield store handling */
6508         bit_pos = (ft >> VT_STRUCT_SHIFT) & 0x3f;
6509         bit_size = (ft >> (VT_STRUCT_SHIFT + 6)) & 0x3f;
6510         /* remove bit field info to avoid loops */
6511         vtop[-1].type.t = ft & ~(VT_BITFIELD | (-1 << VT_STRUCT_SHIFT));
6512
6513         /* duplicate source into other register */
6514         gv_dup();
6515         vswap();
6516         vrott(3);
6517
6518         if((ft & VT_BTYPE) == VT_BOOL) {
6519             gen_cast(&vtop[-1].type);
6520             vtop[-1].type.t = (vtop[-1].type.t & ~VT_BTYPE) | (VT_BYTE | VT_UNSIGNED);
6521         }
6522
6523         /* duplicate destination */
6524         vdup();
6525         vtop[-1] = vtop[-2];
6526
6527         /* mask and shift source */
6528         if((ft & VT_BTYPE) != VT_BOOL) {
6529             vpushi((1 << bit_size) - 1);
6530             gen_op('&');
6531         }
6532         vpushi(bit_pos);
6533         gen_op(TOK_SHL);
6534         /* load destination, mask and or with source */
6535         vswap();
6536         vpushi(~(((1 << bit_size) - 1) << bit_pos));
6537         gen_op('&');
6538         gen_op('|');
6539         /* store result */
6540         vstore();
6541
6542         /* pop off shifted source from "duplicate source..." above */
6543         vpop();
6544
6545     } else {
6546 #ifdef CONFIG_TCC_BCHECK
6547         /* bound check case */
6548         if (vtop[-1].r & VT_MUSTBOUND) {
6549             vswap();
6550             gbound();
6551             vswap();
6552         }
6553 #endif
6554         if (!nocode_wanted) {
6555             rc = RC_INT;
6556             if (is_float(ft))
6557                 rc = RC_FLOAT;
6558             r = gv(rc);  /* generate value */
6559             /* if lvalue was saved on stack, must read it */
6560             if ((vtop[-1].r & VT_VALMASK) == VT_LLOCAL) {
6561                 SValue sv;
6562                 t = get_reg(RC_INT);
6563                 sv.type.t = VT_INT;
6564                 sv.r = VT_LOCAL | VT_LVAL;
6565                 sv.c.ul = vtop[-1].c.ul;
6566                 load(t, &sv);
6567                 vtop[-1].r = t | VT_LVAL;
6568             }
6569             store(r, vtop - 1);
6570             /* two word case handling : store second register at word + 4 */
6571             if ((ft & VT_BTYPE) == VT_LLONG) {
6572                 vswap();
6573                 /* convert to int to increment easily */
6574                 vtop->type.t = VT_INT;
6575                 gaddrof();
6576                 vpushi(4);
6577                 gen_op('+');
6578                 vtop->r |= VT_LVAL;
6579                 vswap();
6580                 /* XXX: it works because r2 is spilled last ! */
6581                 store(vtop->r2, vtop - 1);
6582             }
6583         }
6584         vswap();
6585         vtop--; /* NOT vpop() because on x86 it would flush the fp stack */
6586         vtop->r |= delayed_cast;
6587     }
6588 }
6589
6590 /* post defines POST/PRE add. c is the token ++ or -- */
6591 void inc(int post, int c)
6592 {
6593     test_lvalue();
6594     vdup(); /* save lvalue */
6595     if (post) {
6596         gv_dup(); /* duplicate value */
6597         vrotb(3);
6598         vrotb(3);
6599     }
6600     /* add constant */
6601     vpushi(c - TOK_MID);
6602     gen_op('+');
6603     vstore(); /* store value */
6604     if (post)
6605         vpop(); /* if post op, return saved value */
6606 }
6607
6608 /* Parse GNUC __attribute__ extension. Currently, the following
6609    extensions are recognized:
6610    - aligned(n) : set data/function alignment.
6611    - packed : force data alignment to 1
6612    - section(x) : generate data/code in this section.
6613    - unused : currently ignored, but may be used someday.
6614    - regparm(n) : pass function parameters in registers (i386 only)
6615  */
6616 static void parse_attribute(AttributeDef *ad)
6617 {
6618     int t, n;
6619
6620     while (tok == TOK_ATTRIBUTE1 || tok == TOK_ATTRIBUTE2) {
6621     next();
6622     skip('(');
6623     skip('(');
6624     while (tok != ')') {
6625         if (tok < TOK_IDENT)
6626             expect("attribute name");
6627         t = tok;
6628         next();
6629         switch(t) {
6630         case TOK_SECTION1:
6631         case TOK_SECTION2:
6632             skip('(');
6633             if (tok != TOK_STR)
6634                 expect("section name");
6635             ad->section = find_section(tcc_state, (char *)tokc.cstr->data);
6636             next();
6637             skip(')');
6638             break;
6639         case TOK_ALIGNED1:
6640         case TOK_ALIGNED2:
6641             if (tok == '(') {
6642                 next();
6643                 n = expr_const();
6644                 if (n <= 0 || (n & (n - 1)) != 0)
6645                     error("alignment must be a positive power of two");
6646                 skip(')');
6647             } else {
6648                 n = MAX_ALIGN;
6649             }
6650             ad->aligned = n;
6651             break;
6652         case TOK_PACKED1:
6653         case TOK_PACKED2:
6654             ad->packed = 1;
6655             break;
6656         case TOK_UNUSED1:
6657         case TOK_UNUSED2:
6658             /* currently, no need to handle it because tcc does not
6659                track unused objects */
6660             break;
6661         case TOK_NORETURN1:
6662         case TOK_NORETURN2:
6663             /* currently, no need to handle it because tcc does not
6664                track unused objects */
6665             break;
6666         case TOK_CDECL1:
6667         case TOK_CDECL2:
6668         case TOK_CDECL3:
6669             FUNC_CALL(ad->func_attr) = FUNC_CDECL;
6670             break;
6671         case TOK_STDCALL1:
6672         case TOK_STDCALL2:
6673         case TOK_STDCALL3:
6674             FUNC_CALL(ad->func_attr) = FUNC_STDCALL;
6675             break;
6676 #ifdef TCC_TARGET_I386
6677         case TOK_REGPARM1:
6678         case TOK_REGPARM2:
6679             skip('(');
6680             n = expr_const();
6681             if (n > 3)
6682                 n = 3;
6683             else if (n < 0)
6684                 n = 0;
6685             if (n > 0)
6686                 FUNC_CALL(ad->func_attr) = FUNC_FASTCALL1 + n - 1;
6687             skip(')');
6688             break;
6689         case TOK_FASTCALL1:
6690         case TOK_FASTCALL2:
6691         case TOK_FASTCALL3:
6692             FUNC_CALL(ad->func_attr) = FUNC_FASTCALLW;
6693             break;
6694 #endif
6695         case TOK_DLLEXPORT:
6696             FUNC_EXPORT(ad->func_attr) = 1;
6697             break;
6698         default:
6699             if (tcc_state->warn_unsupported)
6700                 warning("'%s' attribute ignored", get_tok_str(t, NULL));
6701             /* skip parameters */
6702             if (tok == '(') {
6703                 int parenthesis = 0;
6704                 do {
6705                     if (tok == '(')
6706                         parenthesis++;
6707                     else if (tok == ')')
6708                         parenthesis--;
6709                     next();
6710                 } while (parenthesis && tok != -1);
6711             }
6712             break;
6713         }
6714         if (tok != ',')
6715             break;
6716         next();
6717     }
6718     skip(')');
6719     skip(')');
6720     }
6721 }
6722
6723 /* enum/struct/union declaration. u is either VT_ENUM or VT_STRUCT */
6724 static void struct_decl(CType *type, int u)
6725 {
6726     int a, v, size, align, maxalign, c, offset;
6727     int bit_size, bit_pos, bsize, bt, lbit_pos, prevbt;
6728     Sym *s, *ss, *ass, **ps;
6729     AttributeDef ad;
6730     CType type1, btype;
6731
6732     a = tok; /* save decl type */
6733     next();
6734     if (tok != '{') {
6735         v = tok;
6736         next();
6737         /* struct already defined ? return it */
6738         if (v < TOK_IDENT)
6739             expect("struct/union/enum name");
6740         s = struct_find(v);
6741         if (s) {
6742             if (s->type.t != a)
6743                 error("invalid type");
6744             goto do_decl;
6745         }
6746     } else {
6747         v = anon_sym++;
6748     }
6749     type1.t = a;
6750     /* we put an undefined size for struct/union */
6751     s = sym_push(v | SYM_STRUCT, &type1, 0, -1);
6752     s->r = 0; /* default alignment is zero as gcc */
6753     /* put struct/union/enum name in type */
6754  do_decl:
6755     type->t = u;
6756     type->ref = s;
6757
6758     if (tok == '{') {
6759         next();
6760         if (s->c != -1)
6761             error("struct/union/enum already defined");
6762         /* cannot be empty */
6763         c = 0;
6764         /* non empty enums are not allowed */
6765         if (a == TOK_ENUM) {
6766             for(;;) {
6767                 v = tok;
6768                 if (v < TOK_UIDENT)
6769                     expect("identifier");
6770                 next();
6771                 if (tok == '=') {
6772                     next();
6773                     c = expr_const();
6774                 }
6775                 /* enum symbols have static storage */
6776                 ss = sym_push(v, &int_type, VT_CONST, c);
6777                 ss->type.t |= VT_STATIC;
6778                 if (tok != ',')
6779                     break;
6780                 next();
6781                 c++;
6782                 /* NOTE: we accept a trailing comma */
6783                 if (tok == '}')
6784                     break;
6785             }
6786             skip('}');
6787         } else {
6788             maxalign = 1;
6789             ps = &s->next;
6790             prevbt = VT_INT;
6791             bit_pos = 0;
6792             offset = 0;
6793             while (tok != '}') {
6794                 parse_btype(&btype, &ad);
6795                 while (1) {
6796                     bit_size = -1;
6797                     v = 0;
6798                     type1 = btype;
6799                     if (tok != ':') {
6800                         type_decl(&type1, &ad, &v, TYPE_DIRECT | TYPE_ABSTRACT);
6801                         if (v == 0 && (type1.t & VT_BTYPE) != VT_STRUCT)
6802                             expect("identifier");
6803                         if ((type1.t & VT_BTYPE) == VT_FUNC ||
6804                             (type1.t & (VT_TYPEDEF | VT_STATIC | VT_EXTERN | VT_INLINE)))
6805                             error("invalid type for '%s'",
6806                                   get_tok_str(v, NULL));
6807                     }
6808                     if (tok == ':') {
6809                         next();
6810                         bit_size = expr_const();
6811                         /* XXX: handle v = 0 case for messages */
6812                         if (bit_size < 0)
6813                             error("negative width in bit-field '%s'",
6814                                   get_tok_str(v, NULL));
6815                         if (v && bit_size == 0)
6816                             error("zero width for bit-field '%s'",
6817                                   get_tok_str(v, NULL));
6818                     }
6819                     size = type_size(&type1, &align);
6820                     if (ad.aligned) {
6821                         if (align < ad.aligned)
6822                             align = ad.aligned;
6823                     } else if (ad.packed) {
6824                         align = 1;
6825                     } else if (*tcc_state->pack_stack_ptr) {
6826                         if (align > *tcc_state->pack_stack_ptr)
6827                             align = *tcc_state->pack_stack_ptr;
6828                     }
6829                     lbit_pos = 0;
6830                     if (bit_size >= 0) {
6831                         bt = type1.t & VT_BTYPE;
6832                         if (bt != VT_INT &&
6833                             bt != VT_BYTE &&
6834                             bt != VT_SHORT &&
6835                             bt != VT_BOOL &&
6836                             bt != VT_ENUM)
6837                             error("bitfields must have scalar type");
6838                         bsize = size * 8;
6839                         if (bit_size > bsize) {
6840                             error("width of '%s' exceeds its type",
6841                                   get_tok_str(v, NULL));
6842                         } else if (bit_size == bsize) {
6843                             /* no need for bit fields */
6844                             bit_pos = 0;
6845                         } else if (bit_size == 0) {
6846                             /* XXX: what to do if only padding in a
6847                                structure ? */
6848                             /* zero size: means to pad */
6849                             bit_pos = 0;
6850                         } else {
6851                             /* we do not have enough room ?
6852                                did the type change?
6853                                is it a union? */
6854                             if ((bit_pos + bit_size) > bsize ||
6855                                 bt != prevbt || a == TOK_UNION)
6856                                 bit_pos = 0;
6857                             lbit_pos = bit_pos;
6858                             /* XXX: handle LSB first */
6859                             type1.t |= VT_BITFIELD |
6860                                 (bit_pos << VT_STRUCT_SHIFT) |
6861                                 (bit_size << (VT_STRUCT_SHIFT + 6));
6862                             bit_pos += bit_size;
6863                         }
6864                         prevbt = bt;
6865                     } else {
6866                         bit_pos = 0;
6867                     }
6868                     if (v != 0 || (type1.t & VT_BTYPE) == VT_STRUCT) {
6869                         /* add new memory data only if starting
6870                            bit field */
6871                         if (lbit_pos == 0) {
6872                             if (a == TOK_STRUCT) {
6873                                 c = (c + align - 1) & -align;
6874                                 offset = c;
6875                                 if (size > 0)
6876                                     c += size;
6877                             } else {
6878                                 offset = 0;
6879                                 if (size > c)
6880                                     c = size;
6881                             }
6882                             if (align > maxalign)
6883                                 maxalign = align;
6884                         }
6885 #if 0
6886                         printf("add field %s offset=%d",
6887                                get_tok_str(v, NULL), offset);
6888                         if (type1.t & VT_BITFIELD) {
6889                             printf(" pos=%d size=%d",
6890                                    (type1.t >> VT_STRUCT_SHIFT) & 0x3f,
6891                                    (type1.t >> (VT_STRUCT_SHIFT + 6)) & 0x3f);
6892                         }
6893                         printf("\n");
6894 #endif
6895                     }
6896                     if (v == 0 && (type1.t & VT_BTYPE) == VT_STRUCT) {
6897                         ass = type1.ref;
6898                         while ((ass = ass->next) != NULL) {
6899                            ss = sym_push(ass->v, &ass->type, 0, offset + ass->c);
6900                            *ps = ss;
6901                            ps = &ss->next;
6902                         }
6903                     } else if (v) {
6904                         ss = sym_push(v | SYM_FIELD, &type1, 0, offset);
6905                         *ps = ss;
6906                         ps = &ss->next;
6907                     }
6908                     if (tok == ';' || tok == TOK_EOF)
6909                         break;
6910                     skip(',');
6911                 }
6912                 skip(';');
6913             }
6914             skip('}');
6915             /* store size and alignment */
6916             s->c = (c + maxalign - 1) & -maxalign;
6917             s->r = maxalign;
6918         }
6919     }
6920 }
6921
6922 /* return 0 if no type declaration. otherwise, return the basic type
6923    and skip it.
6924  */
6925 static int parse_btype(CType *type, AttributeDef *ad)
6926 {
6927     int t, u, type_found, typespec_found, typedef_found;
6928     Sym *s;
6929     CType type1;
6930
6931     memset(ad, 0, sizeof(AttributeDef));
6932     type_found = 0;
6933     typespec_found = 0;
6934     typedef_found = 0;
6935     t = 0;
6936     while(1) {
6937         switch(tok) {
6938         case TOK_EXTENSION:
6939             /* currently, we really ignore extension */
6940             next();
6941             continue;
6942
6943             /* basic types */
6944         case TOK_CHAR:
6945             u = VT_BYTE;
6946         basic_type:
6947             next();
6948         basic_type1:
6949             if ((t & VT_BTYPE) != 0)
6950                 error("too many basic types");
6951             t |= u;
6952             typespec_found = 1;
6953             break;
6954         case TOK_VOID:
6955             u = VT_VOID;
6956             goto basic_type;
6957         case TOK_SHORT:
6958             u = VT_SHORT;
6959             goto basic_type;
6960         case TOK_INT:
6961             next();
6962             typespec_found = 1;
6963             break;
6964         case TOK_LONG:
6965             next();
6966             if ((t & VT_BTYPE) == VT_DOUBLE) {
6967                 t = (t & ~VT_BTYPE) | VT_LDOUBLE;
6968             } else if ((t & VT_BTYPE) == VT_LONG) {
6969                 t = (t & ~VT_BTYPE) | VT_LLONG;
6970             } else {
6971                 u = VT_LONG;
6972                 goto basic_type1;
6973             }
6974             break;
6975         case TOK_BOOL:
6976             u = VT_BOOL;
6977             goto basic_type;
6978         case TOK_FLOAT:
6979             u = VT_FLOAT;
6980             goto basic_type;
6981         case TOK_DOUBLE:
6982             next();
6983             if ((t & VT_BTYPE) == VT_LONG) {
6984                 t = (t & ~VT_BTYPE) | VT_LDOUBLE;
6985             } else {
6986                 u = VT_DOUBLE;
6987                 goto basic_type1;
6988             }
6989             break;
6990         case TOK_ENUM:
6991             struct_decl(&type1, VT_ENUM);
6992         basic_type2:
6993             u = type1.t;
6994             type->ref = type1.ref;
6995             goto basic_type1;
6996         case TOK_STRUCT:
6997         case TOK_UNION:
6998             struct_decl(&type1, VT_STRUCT);
6999             goto basic_type2;
7000
7001             /* type modifiers */
7002         case TOK_CONST1:
7003         case TOK_CONST2:
7004         case TOK_CONST3:
7005             t |= VT_CONSTANT;
7006             next();
7007             break;
7008         case TOK_VOLATILE1:
7009         case TOK_VOLATILE2:
7010         case TOK_VOLATILE3:
7011             t |= VT_VOLATILE;
7012             next();
7013             break;
7014         case TOK_SIGNED1:
7015         case TOK_SIGNED2:
7016         case TOK_SIGNED3:
7017             typespec_found = 1;
7018             t |= VT_SIGNED;
7019             next();
7020             break;
7021         case TOK_REGISTER:
7022         case TOK_AUTO:
7023         case TOK_RESTRICT1:
7024         case TOK_RESTRICT2:
7025         case TOK_RESTRICT3:
7026             next();
7027             break;
7028         case TOK_UNSIGNED:
7029             t |= VT_UNSIGNED;
7030             next();
7031             typespec_found = 1;
7032             break;
7033
7034             /* storage */
7035         case TOK_EXTERN:
7036             t |= VT_EXTERN;
7037             next();
7038             break;
7039         case TOK_STATIC:
7040             t |= VT_STATIC;
7041             next();
7042             break;
7043         case TOK_TYPEDEF:
7044             t |= VT_TYPEDEF;
7045             next();
7046             break;
7047         case TOK_INLINE1:
7048         case TOK_INLINE2:
7049         case TOK_INLINE3:
7050             t |= VT_INLINE;
7051             next();
7052             break;
7053
7054             /* GNUC attribute */
7055         case TOK_ATTRIBUTE1:
7056         case TOK_ATTRIBUTE2:
7057             parse_attribute(ad);
7058             break;
7059             /* GNUC typeof */
7060         case TOK_TYPEOF1:
7061         case TOK_TYPEOF2:
7062         case TOK_TYPEOF3:
7063             next();
7064             parse_expr_type(&type1);
7065             goto basic_type2;
7066         default:
7067             if (typespec_found || typedef_found)
7068                 goto the_end;
7069             s = sym_find(tok);
7070             if (!s || !(s->type.t & VT_TYPEDEF))
7071                 goto the_end;
7072             typedef_found = 1;
7073             t |= (s->type.t & ~VT_TYPEDEF);
7074             type->ref = s->type.ref;
7075             next();
7076             typespec_found = 1;
7077             break;
7078         }
7079         type_found = 1;
7080     }
7081 the_end:
7082     if ((t & (VT_SIGNED|VT_UNSIGNED)) == (VT_SIGNED|VT_UNSIGNED))
7083         error("signed and unsigned modifier");
7084     if (tcc_state->char_is_unsigned) {
7085         if ((t & (VT_SIGNED|VT_UNSIGNED|VT_BTYPE)) == VT_BYTE)
7086             t |= VT_UNSIGNED;
7087     }
7088     t &= ~VT_SIGNED;
7089
7090     /* long is never used as type */
7091     if ((t & VT_BTYPE) == VT_LONG)
7092         t = (t & ~VT_BTYPE) | VT_INT;
7093     type->t = t;
7094     return type_found;
7095 }
7096
7097 /* convert a function parameter type (array to pointer and function to
7098    function pointer) */
7099 static inline void convert_parameter_type(CType *pt)
7100 {
7101     /* remove const and volatile qualifiers (XXX: const could be used
7102        to indicate a const function parameter */
7103     pt->t &= ~(VT_CONSTANT | VT_VOLATILE);
7104     /* array must be transformed to pointer according to ANSI C */
7105     pt->t &= ~VT_ARRAY;
7106     if ((pt->t & VT_BTYPE) == VT_FUNC) {
7107         mk_pointer(pt);
7108     }
7109 }
7110
7111 static void post_type(CType *type, AttributeDef *ad)
7112 {
7113     int n, l, t1, arg_size, align;
7114     Sym **plast, *s, *first;
7115     AttributeDef ad1;
7116     CType pt;
7117
7118     if (tok == '(') {
7119         /* function declaration */
7120         next();
7121         l = 0;
7122         first = NULL;
7123         plast = &first;
7124         arg_size = 0;
7125         if (tok != ')') {
7126             for(;;) {
7127                 /* read param name and compute offset */
7128                 if (l != FUNC_OLD) {
7129                     if (!parse_btype(&pt, &ad1)) {
7130                         if (l) {
7131                             error("invalid type");
7132                         } else {
7133                             l = FUNC_OLD;
7134                             goto old_proto;
7135                         }
7136                     }
7137                     l = FUNC_NEW;
7138                     if ((pt.t & VT_BTYPE) == VT_VOID && tok == ')')
7139                         break;
7140                     type_decl(&pt, &ad1, &n, TYPE_DIRECT | TYPE_ABSTRACT);
7141                     if ((pt.t & VT_BTYPE) == VT_VOID)
7142                         error("parameter declared as void");
7143                     arg_size += (type_size(&pt, &align) + 3) & ~3;
7144                 } else {
7145                 old_proto:
7146                     n = tok;
7147                     if (n < TOK_UIDENT)
7148                         expect("identifier");
7149                     pt.t = VT_INT;
7150                     next();
7151                 }
7152                 convert_parameter_type(&pt);
7153                 s = sym_push(n | SYM_FIELD, &pt, 0, 0);
7154                 *plast = s;
7155                 plast = &s->next;
7156                 if (tok == ')')
7157                     break;
7158                 skip(',');
7159                 if (l == FUNC_NEW && tok == TOK_DOTS) {
7160                     l = FUNC_ELLIPSIS;
7161                     next();
7162                     break;
7163                 }
7164             }
7165         }
7166         /* if no parameters, then old type prototype */
7167         if (l == 0)
7168             l = FUNC_OLD;
7169         skip(')');
7170         t1 = type->t & VT_STORAGE;
7171         /* NOTE: const is ignored in returned type as it has a special
7172            meaning in gcc / C++ */
7173         type->t &= ~(VT_STORAGE | VT_CONSTANT);
7174         post_type(type, ad);
7175         /* we push a anonymous symbol which will contain the function prototype */
7176         FUNC_ARGS(ad->func_attr) = arg_size;
7177         s = sym_push(SYM_FIELD, type, ad->func_attr, l);
7178         s->next = first;
7179         type->t = t1 | VT_FUNC;
7180         type->ref = s;
7181     } else if (tok == '[') {
7182         /* array definition */
7183         next();
7184         n = -1;
7185         if (tok != ']') {
7186             n = expr_const();
7187             if (n < 0)
7188                 error("invalid array size");
7189         }
7190         skip(']');
7191         /* parse next post type */
7192         t1 = type->t & VT_STORAGE;
7193         type->t &= ~VT_STORAGE;
7194         post_type(type, ad);
7195
7196         /* we push a anonymous symbol which will contain the array
7197            element type */
7198         s = sym_push(SYM_FIELD, type, 0, n);
7199         type->t = t1 | VT_ARRAY | VT_PTR;
7200         type->ref = s;
7201     }
7202 }
7203
7204 /* Parse a type declaration (except basic type), and return the type
7205    in 'type'. 'td' is a bitmask indicating which kind of type decl is
7206    expected. 'type' should contain the basic type. 'ad' is the
7207    attribute definition of the basic type. It can be modified by
7208    type_decl().
7209  */
7210 static void type_decl(CType *type, AttributeDef *ad, int *v, int td)
7211 {
7212     Sym *s;
7213     CType type1, *type2;
7214     int qualifiers;
7215
7216     while (tok == '*') {
7217         qualifiers = 0;
7218     redo:
7219         next();
7220         switch(tok) {
7221         case TOK_CONST1:
7222         case TOK_CONST2:
7223         case TOK_CONST3:
7224             qualifiers |= VT_CONSTANT;
7225             goto redo;
7226         case TOK_VOLATILE1:
7227         case TOK_VOLATILE2:
7228         case TOK_VOLATILE3:
7229             qualifiers |= VT_VOLATILE;
7230             goto redo;
7231         case TOK_RESTRICT1:
7232         case TOK_RESTRICT2:
7233         case TOK_RESTRICT3:
7234             goto redo;
7235         }
7236         mk_pointer(type);
7237         type->t |= qualifiers;
7238     }
7239
7240     /* XXX: clarify attribute handling */
7241     if (tok == TOK_ATTRIBUTE1 || tok == TOK_ATTRIBUTE2)
7242         parse_attribute(ad);
7243
7244     /* recursive type */
7245     /* XXX: incorrect if abstract type for functions (e.g. 'int ()') */
7246     type1.t = 0; /* XXX: same as int */
7247     if (tok == '(') {
7248         next();
7249         /* XXX: this is not correct to modify 'ad' at this point, but
7250            the syntax is not clear */
7251         if (tok == TOK_ATTRIBUTE1 || tok == TOK_ATTRIBUTE2)
7252             parse_attribute(ad);
7253         type_decl(&type1, ad, v, td);
7254         skip(')');
7255     } else {
7256         /* type identifier */
7257         if (tok >= TOK_IDENT && (td & TYPE_DIRECT)) {
7258             *v = tok;
7259             next();
7260         } else {
7261             if (!(td & TYPE_ABSTRACT))
7262                 expect("identifier");
7263             *v = 0;
7264         }
7265     }
7266     post_type(type, ad);
7267     if (tok == TOK_ATTRIBUTE1 || tok == TOK_ATTRIBUTE2)
7268         parse_attribute(ad);
7269     if (!type1.t)
7270         return;
7271     /* append type at the end of type1 */
7272     type2 = &type1;
7273     for(;;) {
7274         s = type2->ref;
7275         type2 = &s->type;
7276         if (!type2->t) {
7277             *type2 = *type;
7278             break;
7279         }
7280     }
7281     *type = type1;
7282 }
7283
7284 /* compute the lvalue VT_LVAL_xxx needed to match type t. */
7285 static int lvalue_type(int t)
7286 {
7287     int bt, r;
7288     r = VT_LVAL;
7289     bt = t & VT_BTYPE;
7290     if (bt == VT_BYTE || bt == VT_BOOL)
7291         r |= VT_LVAL_BYTE;
7292     else if (bt == VT_SHORT)
7293         r |= VT_LVAL_SHORT;
7294     else
7295         return r;
7296     if (t & VT_UNSIGNED)
7297         r |= VT_LVAL_UNSIGNED;
7298     return r;
7299 }
7300
7301 /* indirection with full error checking and bound check */
7302 static void indir(void)
7303 {
7304     if ((vtop->type.t & VT_BTYPE) != VT_PTR) {
7305         if ((vtop->type.t & VT_BTYPE) == VT_FUNC)
7306             return;
7307         expect("pointer");
7308     }
7309     if ((vtop->r & VT_LVAL) && !nocode_wanted)
7310         gv(RC_INT);
7311     vtop->type = *pointed_type(&vtop->type);
7312     /* Arrays and functions are never lvalues */
7313     if (!(vtop->type.t & VT_ARRAY)
7314         && (vtop->type.t & VT_BTYPE) != VT_FUNC) {
7315         vtop->r |= lvalue_type(vtop->type.t);
7316         /* if bound checking, the referenced pointer must be checked */
7317         if (do_bounds_check)
7318             vtop->r |= VT_MUSTBOUND;
7319     }
7320 }
7321
7322 /* pass a parameter to a function and do type checking and casting */
7323 static void gfunc_param_typed(Sym *func, Sym *arg)
7324 {
7325     int func_type;
7326     CType type;
7327
7328     func_type = func->c;
7329     if (func_type == FUNC_OLD ||
7330         (func_type == FUNC_ELLIPSIS && arg == NULL)) {
7331         /* default casting : only need to convert float to double */
7332         if ((vtop->type.t & VT_BTYPE) == VT_FLOAT) {
7333             type.t = VT_DOUBLE;
7334             gen_cast(&type);
7335         }
7336     } else if (arg == NULL) {
7337         error("too many arguments to function");
7338     } else {
7339         type = arg->type;
7340         type.t &= ~VT_CONSTANT; /* need to do that to avoid false warning */
7341         gen_assign_cast(&type);
7342     }
7343 }
7344
7345 /* parse an expression of the form '(type)' or '(expr)' and return its
7346    type */
7347 static void parse_expr_type(CType *type)
7348 {
7349     int n;
7350     AttributeDef ad;
7351
7352     skip('(');
7353     if (parse_btype(type, &ad)) {
7354         type_decl(type, &ad, &n, TYPE_ABSTRACT);
7355     } else {
7356         expr_type(type);
7357     }
7358     skip(')');
7359 }
7360
7361 static void parse_type(CType *type)
7362 {
7363     AttributeDef ad;
7364     int n;
7365
7366     if (!parse_btype(type, &ad)) {
7367         expect("type");
7368     }
7369     type_decl(type, &ad, &n, TYPE_ABSTRACT);
7370 }
7371
7372 static void vpush_tokc(int t)
7373 {
7374     CType type;
7375     type.t = t;
7376     vsetc(&type, VT_CONST, &tokc);
7377 }
7378
7379 static void unary(void)
7380 {
7381     int n, t, align, size, r;
7382     CType type;
7383     Sym *s;
7384     AttributeDef ad;
7385
7386     /* XXX: GCC 2.95.3 does not generate a table although it should be
7387        better here */
7388  tok_next:
7389     switch(tok) {
7390     case TOK_EXTENSION:
7391         next();
7392         goto tok_next;
7393     case TOK_CINT:
7394     case TOK_CCHAR:
7395     case TOK_LCHAR:
7396         vpushi(tokc.i);
7397         next();
7398         break;
7399     case TOK_CUINT:
7400         vpush_tokc(VT_INT | VT_UNSIGNED);
7401         next();
7402         break;
7403     case TOK_CLLONG:
7404         vpush_tokc(VT_LLONG);
7405         next();
7406         break;
7407     case TOK_CULLONG:
7408         vpush_tokc(VT_LLONG | VT_UNSIGNED);
7409         next();
7410         break;
7411     case TOK_CFLOAT:
7412         vpush_tokc(VT_FLOAT);
7413         next();
7414         break;
7415     case TOK_CDOUBLE:
7416         vpush_tokc(VT_DOUBLE);
7417         next();
7418         break;
7419     case TOK_CLDOUBLE:
7420         vpush_tokc(VT_LDOUBLE);
7421         next();
7422         break;
7423     case TOK___FUNCTION__:
7424         if (!gnu_ext)
7425             goto tok_identifier;
7426         /* fall thru */
7427     case TOK___FUNC__:
7428         {
7429             void *ptr;
7430             int len;
7431             /* special function name identifier */
7432             len = strlen(funcname) + 1;
7433             /* generate char[len] type */
7434             type.t = VT_BYTE;
7435             mk_pointer(&type);
7436             type.t |= VT_ARRAY;
7437             type.ref->c = len;
7438             vpush_ref(&type, data_section, data_section->data_offset, len);
7439             ptr = section_ptr_add(data_section, len);
7440             memcpy(ptr, funcname, len);
7441             next();
7442         }
7443         break;
7444     case TOK_LSTR:
7445 #ifdef TCC_TARGET_PE
7446         t = VT_SHORT | VT_UNSIGNED;
7447 #else
7448         t = VT_INT;
7449 #endif
7450         goto str_init;
7451     case TOK_STR:
7452         /* string parsing */
7453         t = VT_BYTE;
7454     str_init:
7455         if (tcc_state->warn_write_strings)
7456             t |= VT_CONSTANT;
7457         type.t = t;
7458         mk_pointer(&type);
7459         type.t |= VT_ARRAY;
7460         memset(&ad, 0, sizeof(AttributeDef));
7461         decl_initializer_alloc(&type, &ad, VT_CONST, 2, 0, 0);
7462         break;
7463     case '(':
7464         next();
7465         /* cast ? */
7466         if (parse_btype(&type, &ad)) {
7467             type_decl(&type, &ad, &n, TYPE_ABSTRACT);
7468             skip(')');
7469             /* check ISOC99 compound literal */
7470             if (tok == '{') {
7471                     /* data is allocated locally by default */
7472                 if (global_expr)
7473                     r = VT_CONST;
7474                 else
7475                     r = VT_LOCAL;
7476                 /* all except arrays are lvalues */
7477                 if (!(type.t & VT_ARRAY))
7478                     r |= lvalue_type(type.t);
7479                 memset(&ad, 0, sizeof(AttributeDef));
7480                 decl_initializer_alloc(&type, &ad, r, 1, 0, 0);
7481             } else {
7482                 unary();
7483                 gen_cast(&type);
7484             }
7485         } else if (tok == '{') {
7486             /* save all registers */
7487             save_regs(0);
7488             /* statement expression : we do not accept break/continue
7489                inside as GCC does */
7490             block(NULL, NULL, NULL, NULL, 0, 1);
7491             skip(')');
7492         } else {
7493             gexpr();
7494             skip(')');
7495         }
7496         break;
7497     case '*':
7498         next();
7499         unary();
7500         indir();
7501         break;
7502     case '&':
7503         next();
7504         unary();
7505         /* functions names must be treated as function pointers,
7506            except for unary '&' and sizeof. Since we consider that
7507            functions are not lvalues, we only have to handle it
7508            there and in function calls. */
7509         /* arrays can also be used although they are not lvalues */
7510         if ((vtop->type.t & VT_BTYPE) != VT_FUNC &&
7511             !(vtop->type.t & VT_ARRAY) && !(vtop->type.t & VT_LLOCAL))
7512             test_lvalue();
7513         mk_pointer(&vtop->type);
7514         gaddrof();
7515         break;
7516     case '!':
7517         next();
7518         unary();
7519         if ((vtop->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST) {
7520             CType boolean;
7521             boolean.t = VT_BOOL;
7522             gen_cast(&boolean);
7523             vtop->c.i = !vtop->c.i;
7524         } else if ((vtop->r & VT_VALMASK) == VT_CMP)
7525             vtop->c.i = vtop->c.i ^ 1;
7526         else {
7527             save_regs(1);
7528             vseti(VT_JMP, gtst(1, 0));
7529         }
7530         break;
7531     case '~':
7532         next();
7533         unary();
7534         vpushi(-1);
7535         gen_op('^');
7536         break;
7537     case '+':
7538         next();
7539         /* in order to force cast, we add zero */
7540         unary();
7541         if ((vtop->type.t & VT_BTYPE) == VT_PTR)
7542             error("pointer not accepted for unary plus");
7543         vpushi(0);
7544         gen_op('+');
7545         break;
7546     case TOK_SIZEOF:
7547     case TOK_ALIGNOF1:
7548     case TOK_ALIGNOF2:
7549         t = tok;
7550         next();
7551         if (tok == '(') {
7552             parse_expr_type(&type);
7553         } else {
7554             unary_type(&type);
7555         }
7556         size = type_size(&type, &align);
7557         if (t == TOK_SIZEOF) {
7558             if (size < 0)
7559                 error("sizeof applied to an incomplete type");
7560             vpushi(size);
7561         } else {
7562             vpushi(align);
7563         }
7564         vtop->type.t |= VT_UNSIGNED;
7565         break;
7566
7567     case TOK_builtin_types_compatible_p:
7568         {
7569             CType type1, type2;
7570             next();
7571             skip('(');
7572             parse_type(&type1);
7573             skip(',');
7574             parse_type(&type2);
7575             skip(')');
7576             type1.t &= ~(VT_CONSTANT | VT_VOLATILE);
7577             type2.t &= ~(VT_CONSTANT | VT_VOLATILE);
7578             vpushi(is_compatible_types(&type1, &type2));
7579         }
7580         break;
7581     case TOK_builtin_constant_p:
7582         {
7583             int saved_nocode_wanted, res;
7584             next();
7585             skip('(');
7586             saved_nocode_wanted = nocode_wanted;
7587             nocode_wanted = 1;
7588             gexpr();
7589             res = (vtop->r & (VT_VALMASK | VT_LVAL | VT_SYM)) == VT_CONST;
7590             vpop();
7591             nocode_wanted = saved_nocode_wanted;
7592             skip(')');
7593             vpushi(res);
7594         }
7595         break;
7596     case TOK_INC:
7597     case TOK_DEC:
7598         t = tok;
7599         next();
7600         unary();
7601         inc(0, t);
7602         break;
7603     case '-':
7604         next();
7605         vpushi(0);
7606         unary();
7607         gen_op('-');
7608         break;
7609     case TOK_LAND:
7610         if (!gnu_ext)
7611             goto tok_identifier;
7612         next();
7613         /* allow to take the address of a label */
7614         if (tok < TOK_UIDENT)
7615             expect("label identifier");
7616         s = label_find(tok);
7617         if (!s) {
7618             s = label_push(&global_label_stack, tok, LABEL_FORWARD);
7619         } else {
7620             if (s->r == LABEL_DECLARED)
7621                 s->r = LABEL_FORWARD;
7622         }
7623         if (!s->type.t) {
7624             s->type.t = VT_VOID;
7625             mk_pointer(&s->type);
7626             s->type.t |= VT_STATIC;
7627         }
7628         vset(&s->type, VT_CONST | VT_SYM, 0);
7629         vtop->sym = s;
7630         next();
7631         break;
7632     default:
7633     tok_identifier:
7634         t = tok;
7635         next();
7636         if (t < TOK_UIDENT)
7637             expect("identifier");
7638         s = sym_find(t);
7639         if (!s) {
7640             if (tok != '(')
7641                 error("'%s' undeclared", get_tok_str(t, NULL));
7642             /* for simple function calls, we tolerate undeclared
7643                external reference to int() function */
7644             if (tcc_state->warn_implicit_function_declaration)
7645                 warning("implicit declaration of function '%s'",
7646                         get_tok_str(t, NULL));
7647             s = external_global_sym(t, &func_old_type, 0);
7648         }
7649         if ((s->type.t & (VT_STATIC | VT_INLINE | VT_BTYPE)) ==
7650             (VT_STATIC | VT_INLINE | VT_FUNC)) {
7651             /* if referencing an inline function, then we generate a
7652                symbol to it if not already done. It will have the
7653                effect to generate code for it at the end of the
7654                compilation unit. Inline function as always
7655                generated in the text section. */
7656             if (!s->c)
7657                 put_extern_sym(s, text_section, 0, 0);
7658             r = VT_SYM | VT_CONST;
7659         } else {
7660             r = s->r;
7661         }
7662         vset(&s->type, r, s->c);
7663         /* if forward reference, we must point to s */
7664         if (vtop->r & VT_SYM) {
7665             vtop->sym = s;
7666             vtop->c.ul = 0;
7667         }
7668         break;
7669     }
7670
7671     /* post operations */
7672     while (1) {
7673         if (tok == TOK_INC || tok == TOK_DEC) {
7674             inc(1, tok);
7675             next();
7676         } else if (tok == '.' || tok == TOK_ARROW) {
7677             /* field */
7678             if (tok == TOK_ARROW)
7679                 indir();
7680             test_lvalue();
7681             gaddrof();
7682             next();
7683             /* expect pointer on structure */
7684             if ((vtop->type.t & VT_BTYPE) != VT_STRUCT)
7685                 expect("struct or union");
7686             s = vtop->type.ref;
7687             /* find field */
7688             tok |= SYM_FIELD;
7689             while ((s = s->next) != NULL) {
7690                 if (s->v == tok)
7691                     break;
7692             }
7693             if (!s)
7694                 error("field not found: %s",  get_tok_str(tok & ~SYM_FIELD, NULL));
7695             /* add field offset to pointer */
7696             vtop->type = char_pointer_type; /* change type to 'char *' */
7697             vpushi(s->c);
7698             gen_op('+');
7699             /* change type to field type, and set to lvalue */
7700             vtop->type = s->type;
7701             /* an array is never an lvalue */
7702             if (!(vtop->type.t & VT_ARRAY)) {
7703                 vtop->r |= lvalue_type(vtop->type.t);
7704                 /* if bound checking, the referenced pointer must be checked */
7705                 if (do_bounds_check)
7706                     vtop->r |= VT_MUSTBOUND;
7707             }
7708             next();
7709         } else if (tok == '[') {
7710             next();
7711             gexpr();
7712             gen_op('+');
7713             indir();
7714             skip(']');
7715         } else if (tok == '(') {
7716             SValue ret;
7717             Sym *sa;
7718             int nb_args;
7719
7720             /* function call  */
7721             if ((vtop->type.t & VT_BTYPE) != VT_FUNC) {
7722                 /* pointer test (no array accepted) */
7723                 if ((vtop->type.t & (VT_BTYPE | VT_ARRAY)) == VT_PTR) {
7724                     vtop->type = *pointed_type(&vtop->type);
7725                     if ((vtop->type.t & VT_BTYPE) != VT_FUNC)
7726                         goto error_func;
7727                 } else {
7728                 error_func:
7729                     expect("function pointer");
7730                 }
7731             } else {
7732                 vtop->r &= ~VT_LVAL; /* no lvalue */
7733             }
7734             /* get return type */
7735             s = vtop->type.ref;
7736             next();
7737             sa = s->next; /* first parameter */
7738             nb_args = 0;
7739             ret.r2 = VT_CONST;
7740             /* compute first implicit argument if a structure is returned */
7741             if ((s->type.t & VT_BTYPE) == VT_STRUCT) {
7742                 /* get some space for the returned structure */
7743                 size = type_size(&s->type, &align);
7744                 loc = (loc - size) & -align;
7745                 ret.type = s->type;
7746                 ret.r = VT_LOCAL | VT_LVAL;
7747                 /* pass it as 'int' to avoid structure arg passing
7748                    problems */
7749                 vseti(VT_LOCAL, loc);
7750                 ret.c = vtop->c;
7751                 nb_args++;
7752             } else {
7753                 ret.type = s->type;
7754                 /* return in register */
7755                 if (is_float(ret.type.t)) {
7756                     ret.r = REG_FRET;
7757                 } else {
7758                     if ((ret.type.t & VT_BTYPE) == VT_LLONG)
7759                         ret.r2 = REG_LRET;
7760                     ret.r = REG_IRET;
7761                 }
7762                 ret.c.i = 0;
7763             }
7764             if (tok != ')') {
7765                 for(;;) {
7766                     expr_eq();
7767                     gfunc_param_typed(s, sa);
7768                     nb_args++;
7769                     if (sa)
7770                         sa = sa->next;
7771                     if (tok == ')')
7772                         break;
7773                     skip(',');
7774                 }
7775             }
7776             if (sa)
7777                 error("too few arguments to function");
7778             skip(')');
7779             if (!nocode_wanted) {
7780                 gfunc_call(nb_args);
7781             } else {
7782                 vtop -= (nb_args + 1);
7783             }
7784             /* return value */
7785             vsetc(&ret.type, ret.r, &ret.c);
7786             vtop->r2 = ret.r2;
7787         } else {
7788             break;
7789         }
7790     }
7791 }
7792
7793 static void uneq(void)
7794 {
7795     int t;
7796
7797     unary();
7798     if (tok == '=' ||
7799         (tok >= TOK_A_MOD && tok <= TOK_A_DIV) ||
7800         tok == TOK_A_XOR || tok == TOK_A_OR ||
7801         tok == TOK_A_SHL || tok == TOK_A_SAR) {
7802         test_lvalue();
7803         t = tok;
7804         next();
7805         if (t == '=') {
7806             expr_eq();
7807         } else {
7808             vdup();
7809             expr_eq();
7810             gen_op(t & 0x7f);
7811         }
7812         vstore();
7813     }
7814 }
7815
7816 static void expr_prod(void)
7817 {
7818     int t;
7819
7820     uneq();
7821     while (tok == '*' || tok == '/' || tok == '%') {
7822         t = tok;
7823         next();
7824         uneq();
7825         gen_op(t);
7826     }
7827 }
7828
7829 static void expr_sum(void)
7830 {
7831     int t;
7832
7833     expr_prod();
7834     while (tok == '+' || tok == '-') {
7835         t = tok;
7836         next();
7837         expr_prod();
7838         gen_op(t);
7839     }
7840 }
7841
7842 static void expr_shift(void)
7843 {
7844     int t;
7845
7846     expr_sum();
7847     while (tok == TOK_SHL || tok == TOK_SAR) {
7848         t = tok;
7849         next();
7850         expr_sum();
7851         gen_op(t);
7852     }
7853 }
7854
7855 static void expr_cmp(void)
7856 {
7857     int t;
7858
7859     expr_shift();
7860     while ((tok >= TOK_ULE && tok <= TOK_GT) ||
7861            tok == TOK_ULT || tok == TOK_UGE) {
7862         t = tok;
7863         next();
7864         expr_shift();
7865         gen_op(t);
7866     }
7867 }
7868
7869 static void expr_cmpeq(void)
7870 {
7871     int t;
7872
7873     expr_cmp();
7874     while (tok == TOK_EQ || tok == TOK_NE) {
7875         t = tok;
7876         next();
7877         expr_cmp();
7878         gen_op(t);
7879     }
7880 }
7881
7882 static void expr_and(void)
7883 {
7884     expr_cmpeq();
7885     while (tok == '&') {
7886         next();
7887         expr_cmpeq();
7888         gen_op('&');
7889     }
7890 }
7891
7892 static void expr_xor(void)
7893 {
7894     expr_and();
7895     while (tok == '^') {
7896         next();
7897         expr_and();
7898         gen_op('^');
7899     }
7900 }
7901
7902 static void expr_or(void)
7903 {
7904     expr_xor();
7905     while (tok == '|') {
7906         next();
7907         expr_xor();
7908         gen_op('|');
7909     }
7910 }
7911
7912 /* XXX: fix this mess */
7913 static void expr_land_const(void)
7914 {
7915     expr_or();
7916     while (tok == TOK_LAND) {
7917         next();
7918         expr_or();
7919         gen_op(TOK_LAND);
7920     }
7921 }
7922
7923 /* XXX: fix this mess */
7924 static void expr_lor_const(void)
7925 {
7926     expr_land_const();
7927     while (tok == TOK_LOR) {
7928         next();
7929         expr_land_const();
7930         gen_op(TOK_LOR);
7931     }
7932 }
7933
7934 /* only used if non constant */
7935 static void expr_land(void)
7936 {
7937     int t;
7938
7939     expr_or();
7940     if (tok == TOK_LAND) {
7941         t = 0;
7942         save_regs(1);
7943         for(;;) {
7944             t = gtst(1, t);
7945             if (tok != TOK_LAND) {
7946                 vseti(VT_JMPI, t);
7947                 break;
7948             }
7949             next();
7950             expr_or();
7951         }
7952     }
7953 }
7954
7955 static void expr_lor(void)
7956 {
7957     int t;
7958
7959     expr_land();
7960     if (tok == TOK_LOR) {
7961         t = 0;
7962         save_regs(1);
7963         for(;;) {
7964             t = gtst(0, t);
7965             if (tok != TOK_LOR) {
7966                 vseti(VT_JMP, t);
7967                 break;
7968             }
7969             next();
7970             expr_land();
7971         }
7972     }
7973 }
7974
7975 /* XXX: better constant handling */
7976 static void expr_eq(void)
7977 {
7978     int tt, u, r1, r2, rc, t1, t2, bt1, bt2;
7979     SValue sv;
7980     CType type, type1, type2;
7981
7982     if (const_wanted) {
7983         int c1, c;
7984         expr_lor_const();
7985         if (tok == '?') {
7986             c = vtop->c.i;
7987             vpop();
7988             next();
7989             if (tok == ':' && gnu_ext) {
7990                 c1 = c;
7991             } else {
7992                 gexpr();
7993                 c1 = vtop->c.i;
7994                 vpop();
7995             }
7996             skip(':');
7997             expr_eq();
7998             if (c)
7999                 vtop->c.i = c1;
8000         }
8001     } else {
8002         expr_lor();
8003         if (tok == '?') {
8004             next();
8005             if (vtop != vstack) {
8006                 /* needed to avoid having different registers saved in
8007                    each branch */
8008                 if (is_float(vtop->type.t))
8009                     rc = RC_FLOAT;
8010                 else
8011                     rc = RC_INT;
8012                     gv(rc);
8013                     save_regs(1);
8014             }
8015             if (tok == ':' && gnu_ext) {
8016                 gv_dup();
8017                 tt = gtst(1, 0);
8018             } else {
8019                 tt = gtst(1, 0);
8020                 gexpr();
8021             }
8022             type1 = vtop->type;
8023             sv = *vtop; /* save value to handle it later */
8024             vtop--; /* no vpop so that FP stack is not flushed */
8025             skip(':');
8026             u = gjmp(0);
8027             gsym(tt);
8028             expr_eq();
8029             type2 = vtop->type;
8030
8031             t1 = type1.t;
8032             bt1 = t1 & VT_BTYPE;
8033             t2 = type2.t;
8034             bt2 = t2 & VT_BTYPE;
8035             /* cast operands to correct type according to ISOC rules */
8036             if (is_float(bt1) || is_float(bt2)) {
8037                 if (bt1 == VT_LDOUBLE || bt2 == VT_LDOUBLE) {
8038                     type.t = VT_LDOUBLE;
8039                 } else if (bt1 == VT_DOUBLE || bt2 == VT_DOUBLE) {
8040                     type.t = VT_DOUBLE;
8041                 } else {
8042                     type.t = VT_FLOAT;
8043                 }
8044             } else if (bt1 == VT_LLONG || bt2 == VT_LLONG) {
8045                 /* cast to biggest op */
8046                 type.t = VT_LLONG;
8047                 /* convert to unsigned if it does not fit in a long long */
8048                 if ((t1 & (VT_BTYPE | VT_UNSIGNED)) == (VT_LLONG | VT_UNSIGNED) ||
8049                     (t2 & (VT_BTYPE | VT_UNSIGNED)) == (VT_LLONG | VT_UNSIGNED))
8050                     type.t |= VT_UNSIGNED;
8051             } else if (bt1 == VT_PTR || bt2 == VT_PTR) {
8052                 /* XXX: test pointer compatibility */
8053                 type = type1;
8054             } else if (bt1 == VT_FUNC || bt2 == VT_FUNC) {
8055                 /* XXX: test function pointer compatibility */
8056                 type = type1;
8057             } else if (bt1 == VT_STRUCT || bt2 == VT_STRUCT) {
8058                 /* XXX: test structure compatibility */
8059                 type = type1;
8060             } else if (bt1 == VT_VOID || bt2 == VT_VOID) {
8061                 /* NOTE: as an extension, we accept void on only one side */
8062                 type.t = VT_VOID;
8063             } else {
8064                 /* integer operations */
8065                 type.t = VT_INT;
8066                 /* convert to unsigned if it does not fit in an integer */
8067                 if ((t1 & (VT_BTYPE | VT_UNSIGNED)) == (VT_INT | VT_UNSIGNED) ||
8068                     (t2 & (VT_BTYPE | VT_UNSIGNED)) == (VT_INT | VT_UNSIGNED))
8069                     type.t |= VT_UNSIGNED;
8070             }
8071
8072             /* now we convert second operand */
8073             gen_cast(&type);
8074             if (VT_STRUCT == (vtop->type.t & VT_BTYPE))
8075                 gaddrof();
8076             rc = RC_INT;
8077             if (is_float(type.t)) {
8078                 rc = RC_FLOAT;
8079             } else if ((type.t & VT_BTYPE) == VT_LLONG) {
8080                 /* for long longs, we use fixed registers to avoid having
8081                    to handle a complicated move */
8082                 rc = RC_IRET;
8083             }
8084
8085             r2 = gv(rc);
8086             /* this is horrible, but we must also convert first
8087                operand */
8088             tt = gjmp(0);
8089             gsym(u);
8090             /* put again first value and cast it */
8091             *vtop = sv;
8092             gen_cast(&type);
8093             if (VT_STRUCT == (vtop->type.t & VT_BTYPE))
8094                 gaddrof();
8095             r1 = gv(rc);
8096             move_reg(r2, r1);
8097             vtop->r = r2;
8098             gsym(tt);
8099         }
8100     }
8101 }
8102
8103 static void gexpr(void)
8104 {
8105     while (1) {
8106         expr_eq();
8107         if (tok != ',')
8108             break;
8109         vpop();
8110         next();
8111     }
8112 }
8113
8114 /* parse an expression and return its type without any side effect. */
8115 static void expr_type(CType *type)
8116 {
8117     int saved_nocode_wanted;
8118
8119     saved_nocode_wanted = nocode_wanted;
8120     nocode_wanted = 1;
8121     gexpr();
8122     *type = vtop->type;
8123     vpop();
8124     nocode_wanted = saved_nocode_wanted;
8125 }
8126
8127 /* parse a unary expression and return its type without any side
8128    effect. */
8129 static void unary_type(CType *type)
8130 {
8131     int a;
8132
8133     a = nocode_wanted;
8134     nocode_wanted = 1;
8135     unary();
8136     *type = vtop->type;
8137     vpop();
8138     nocode_wanted = a;
8139 }
8140
8141 /* parse a constant expression and return value in vtop.  */
8142 static void expr_const1(void)
8143 {
8144     int a;
8145     a = const_wanted;
8146     const_wanted = 1;
8147     expr_eq();
8148     const_wanted = a;
8149 }
8150
8151 /* parse an integer constant and return its value. */
8152 static int expr_const(void)
8153 {
8154     int c;
8155     expr_const1();
8156     if ((vtop->r & (VT_VALMASK | VT_LVAL | VT_SYM)) != VT_CONST)
8157         expect("constant expression");
8158     c = vtop->c.i;
8159     vpop();
8160     return c;
8161 }
8162
8163 /* return the label token if current token is a label, otherwise
8164    return zero */
8165 static int is_label(void)
8166 {
8167     int last_tok;
8168
8169     /* fast test first */
8170     if (tok < TOK_UIDENT)
8171         return 0;
8172     /* no need to save tokc because tok is an identifier */
8173     last_tok = tok;
8174     next();
8175     if (tok == ':') {
8176         next();
8177         return last_tok;
8178     } else {
8179         unget_tok(last_tok);
8180         return 0;
8181     }
8182 }
8183
8184 static void block(int *bsym, int *csym, int *case_sym, int *def_sym,
8185                   int case_reg, int is_expr)
8186 {
8187     int a, b, c, d;
8188     Sym *s;
8189
8190     /* generate line number info */
8191     if (do_debug &&
8192         (last_line_num != file->line_num || last_ind != ind)) {
8193         put_stabn(N_SLINE, 0, file->line_num, ind - func_ind);
8194         last_ind = ind;
8195         last_line_num = file->line_num;
8196     }
8197
8198     if (is_expr) {
8199         /* default return value is (void) */
8200         vpushi(0);
8201         vtop->type.t = VT_VOID;
8202     }
8203
8204     if (tok == TOK_IF) {
8205         /* if test */
8206         next();
8207         skip('(');
8208         gexpr();
8209         skip(')');
8210         a = gtst(1, 0);
8211         block(bsym, csym, case_sym, def_sym, case_reg, 0);
8212         c = tok;
8213         if (c == TOK_ELSE) {
8214             next();
8215             d = gjmp(0);
8216             gsym(a);
8217             block(bsym, csym, case_sym, def_sym, case_reg, 0);
8218             gsym(d); /* patch else jmp */
8219         } else
8220             gsym(a);
8221     } else if (tok == TOK_WHILE) {
8222         next();
8223         d = ind;
8224         skip('(');
8225         gexpr();
8226         skip(')');
8227         a = gtst(1, 0);
8228         b = 0;
8229         block(&a, &b, case_sym, def_sym, case_reg, 0);
8230         gjmp_addr(d);
8231         gsym(a);
8232         gsym_addr(b, d);
8233     } else if (tok == '{') {
8234         Sym *llabel;
8235
8236         next();
8237         /* record local declaration stack position */
8238         s = local_stack;
8239         llabel = local_label_stack;
8240         /* handle local labels declarations */
8241         if (tok == TOK_LABEL) {
8242             next();
8243             for(;;) {
8244                 if (tok < TOK_UIDENT)
8245                     expect("label identifier");
8246                 label_push(&local_label_stack, tok, LABEL_DECLARED);
8247                 next();
8248                 if (tok == ',') {
8249                     next();
8250                 } else {
8251                     skip(';');
8252                     break;
8253                 }
8254             }
8255         }
8256         while (tok != '}') {
8257             decl(VT_LOCAL);
8258             if (tok != '}') {
8259                 if (is_expr)
8260                     vpop();
8261                 block(bsym, csym, case_sym, def_sym, case_reg, is_expr);
8262             }
8263         }
8264         /* pop locally defined labels */
8265         label_pop(&local_label_stack, llabel);
8266         /* pop locally defined symbols */
8267         if(is_expr) {
8268             /* XXX: this solution makes only valgrind happy...
8269                triggered by gcc.c-torture/execute/20000917-1.c */
8270             Sym *p;
8271             switch(vtop->type.t & VT_BTYPE) {
8272             case VT_PTR:
8273             case VT_STRUCT:
8274             case VT_ENUM:
8275             case VT_FUNC:
8276                 for(p=vtop->type.ref;p;p=p->prev)
8277                     if(p->prev==s)
8278                         error("unsupported expression type");
8279             }
8280         }
8281         sym_pop(&local_stack, s);
8282         next();
8283     } else if (tok == TOK_RETURN) {
8284         next();
8285         if (tok != ';') {
8286             gexpr();
8287             gen_assign_cast(&func_vt);
8288             if ((func_vt.t & VT_BTYPE) == VT_STRUCT) {
8289                 CType type;
8290                 /* if returning structure, must copy it to implicit
8291                    first pointer arg location */
8292 #ifdef TCC_ARM_EABI
8293                 int align, size;
8294                 size = type_size(&func_vt,&align);
8295                 if(size <= 4)
8296                 {
8297                     if((vtop->r != (VT_LOCAL | VT_LVAL) || (vtop->c.i & 3))
8298                        && (align & 3))
8299                     {
8300                         int addr;
8301                         loc = (loc - size) & -4;
8302                         addr = loc;
8303                         type = func_vt;
8304                         vset(&type, VT_LOCAL | VT_LVAL, addr);
8305                         vswap();
8306                         vstore();
8307                         vset(&int_type, VT_LOCAL | VT_LVAL, addr);
8308                     }
8309                     vtop->type = int_type;
8310                     gv(RC_IRET);
8311                 } else {
8312 #endif
8313                 type = func_vt;
8314                 mk_pointer(&type);
8315                 vset(&type, VT_LOCAL | VT_LVAL, func_vc);
8316                 indir();
8317                 vswap();
8318                 /* copy structure value to pointer */
8319                 vstore();
8320 #ifdef TCC_ARM_EABI
8321                 }
8322 #endif
8323             } else if (is_float(func_vt.t)) {
8324                 gv(RC_FRET);
8325             } else {
8326                 gv(RC_IRET);
8327             }
8328             vtop--; /* NOT vpop() because on x86 it would flush the fp stack */
8329         }
8330         skip(';');
8331         rsym = gjmp(rsym); /* jmp */
8332     } else if (tok == TOK_BREAK) {
8333         /* compute jump */
8334         if (!bsym)
8335             error("cannot break");
8336         *bsym = gjmp(*bsym);
8337         next();
8338         skip(';');
8339     } else if (tok == TOK_CONTINUE) {
8340         /* compute jump */
8341         if (!csym)
8342             error("cannot continue");
8343         *csym = gjmp(*csym);
8344         next();
8345         skip(';');
8346     } else if (tok == TOK_FOR) {
8347         int e;
8348         next();
8349         skip('(');
8350         if (tok != ';') {
8351             gexpr();
8352             vpop();
8353         }
8354         skip(';');
8355         d = ind;
8356         c = ind;
8357         a = 0;
8358         b = 0;
8359         if (tok != ';') {
8360             gexpr();
8361             a = gtst(1, 0);
8362         }
8363         skip(';');
8364         if (tok != ')') {
8365             e = gjmp(0);
8366             c = ind;
8367             gexpr();
8368             vpop();
8369             gjmp_addr(d);
8370             gsym(e);
8371         }
8372         skip(')');
8373         block(&a, &b, case_sym, def_sym, case_reg, 0);
8374         gjmp_addr(c);
8375         gsym(a);
8376         gsym_addr(b, c);
8377     } else
8378     if (tok == TOK_DO) {
8379         next();
8380         a = 0;
8381         b = 0;
8382         d = ind;
8383         block(&a, &b, case_sym, def_sym, case_reg, 0);
8384         skip(TOK_WHILE);
8385         skip('(');
8386         gsym(b);
8387         gexpr();
8388         c = gtst(0, 0);
8389         gsym_addr(c, d);
8390         skip(')');
8391         gsym(a);
8392         skip(';');
8393     } else
8394     if (tok == TOK_SWITCH) {
8395         next();
8396         skip('(');
8397         gexpr();
8398         /* XXX: other types than integer */
8399         case_reg = gv(RC_INT);
8400         vpop();
8401         skip(')');
8402         a = 0;
8403         b = gjmp(0); /* jump to first case */
8404         c = 0;
8405         block(&a, csym, &b, &c, case_reg, 0);
8406         /* if no default, jmp after switch */
8407         if (c == 0)
8408             c = ind;
8409         /* default label */
8410         gsym_addr(b, c);
8411         /* break label */
8412         gsym(a);
8413     } else
8414     if (tok == TOK_CASE) {
8415         int v1, v2;
8416         if (!case_sym)
8417             expect("switch");
8418         next();
8419         v1 = expr_const();
8420         v2 = v1;
8421         if (gnu_ext && tok == TOK_DOTS) {
8422             next();
8423             v2 = expr_const();
8424             if (v2 < v1)
8425                 warning("empty case range");
8426         }
8427         /* since a case is like a label, we must skip it with a jmp */
8428         b = gjmp(0);
8429         gsym(*case_sym);
8430         vseti(case_reg, 0);
8431         vpushi(v1);
8432         if (v1 == v2) {
8433             gen_op(TOK_EQ);
8434             *case_sym = gtst(1, 0);
8435         } else {
8436             gen_op(TOK_GE);
8437             *case_sym = gtst(1, 0);
8438             vseti(case_reg, 0);
8439             vpushi(v2);
8440             gen_op(TOK_LE);
8441             *case_sym = gtst(1, *case_sym);
8442         }
8443         gsym(b);
8444         skip(':');
8445         is_expr = 0;
8446         goto block_after_label;
8447     } else
8448     if (tok == TOK_DEFAULT) {
8449         next();
8450         skip(':');
8451         if (!def_sym)
8452             expect("switch");
8453         if (*def_sym)
8454             error("too many 'default'");
8455         *def_sym = ind;
8456         is_expr = 0;
8457         goto block_after_label;
8458     } else
8459     if (tok == TOK_GOTO) {
8460         next();
8461         if (tok == '*' && gnu_ext) {
8462             /* computed goto */
8463             next();
8464             gexpr();
8465             if ((vtop->type.t & VT_BTYPE) != VT_PTR)
8466                 expect("pointer");
8467             ggoto();
8468         } else if (tok >= TOK_UIDENT) {
8469             s = label_find(tok);
8470             /* put forward definition if needed */
8471             if (!s) {
8472                 s = label_push(&global_label_stack, tok, LABEL_FORWARD);
8473             } else {
8474                 if (s->r == LABEL_DECLARED)
8475                     s->r = LABEL_FORWARD;
8476             }
8477             /* label already defined */
8478             if (s->r & LABEL_FORWARD)
8479                 s->next = (void *)gjmp((long)s->next);
8480             else
8481                 gjmp_addr((long)s->next);
8482             next();
8483         } else {
8484             expect("label identifier");
8485         }
8486         skip(';');
8487     } else if (tok == TOK_ASM1 || tok == TOK_ASM2 || tok == TOK_ASM3) {
8488         asm_instr();
8489     } else {
8490         b = is_label();
8491         if (b) {
8492             /* label case */
8493             s = label_find(b);
8494             if (s) {
8495                 if (s->r == LABEL_DEFINED)
8496                     error("duplicate label '%s'", get_tok_str(s->v, NULL));
8497                 gsym((long)s->next);
8498                 s->r = LABEL_DEFINED;
8499             } else {
8500                 s = label_push(&global_label_stack, b, LABEL_DEFINED);
8501             }
8502             s->next = (void *)ind;
8503             /* we accept this, but it is a mistake */
8504         block_after_label:
8505             if (tok == '}') {
8506                 warning("deprecated use of label at end of compound statement");
8507             } else {
8508                 if (is_expr)
8509                     vpop();
8510                 block(bsym, csym, case_sym, def_sym, case_reg, is_expr);
8511             }
8512         } else {
8513             /* expression case */
8514             if (tok != ';') {
8515                 if (is_expr) {
8516                     vpop();
8517                     gexpr();
8518                 } else {
8519                     gexpr();
8520                     vpop();
8521                 }
8522             }
8523             skip(';');
8524         }
8525     }
8526 }
8527
8528 /* t is the array or struct type. c is the array or struct
8529    address. cur_index/cur_field is the pointer to the current
8530    value. 'size_only' is true if only size info is needed (only used
8531    in arrays) */
8532 static void decl_designator(CType *type, Section *sec, unsigned long c,
8533                             int *cur_index, Sym **cur_field,
8534                             int size_only)
8535 {
8536     Sym *s, *f;
8537     int notfirst, index, index_last, align, l, nb_elems, elem_size;
8538     CType type1;
8539
8540     notfirst = 0;
8541     elem_size = 0;
8542     nb_elems = 1;
8543     if (gnu_ext && (l = is_label()) != 0)
8544         goto struct_field;
8545     while (tok == '[' || tok == '.') {
8546         if (tok == '[') {
8547             if (!(type->t & VT_ARRAY))
8548                 expect("array type");
8549             s = type->ref;
8550             next();
8551             index = expr_const();
8552             if (index < 0 || (s->c >= 0 && index >= s->c))
8553                 expect("invalid index");
8554             if (tok == TOK_DOTS && gnu_ext) {
8555                 next();
8556                 index_last = expr_const();
8557                 if (index_last < 0 ||
8558                     (s->c >= 0 && index_last >= s->c) ||
8559                     index_last < index)
8560                     expect("invalid index");
8561             } else {
8562                 index_last = index;
8563             }
8564             skip(']');
8565             if (!notfirst)
8566                 *cur_index = index_last;
8567             type = pointed_type(type);
8568             elem_size = type_size(type, &align);
8569             c += index * elem_size;
8570             /* NOTE: we only support ranges for last designator */
8571             nb_elems = index_last - index + 1;
8572             if (nb_elems != 1) {
8573                 notfirst = 1;
8574                 break;
8575             }
8576         } else {
8577             next();
8578             l = tok;
8579             next();
8580         struct_field:
8581             if ((type->t & VT_BTYPE) != VT_STRUCT)
8582                 expect("struct/union type");
8583             s = type->ref;
8584             l |= SYM_FIELD;
8585             f = s->next;
8586             while (f) {
8587                 if (f->v == l)
8588                     break;
8589                 f = f->next;
8590             }
8591             if (!f)
8592                 expect("field");
8593             if (!notfirst)
8594                 *cur_field = f;
8595             /* XXX: fix this mess by using explicit storage field */
8596             type1 = f->type;
8597             type1.t |= (type->t & ~VT_TYPE);
8598             type = &type1;
8599             c += f->c;
8600         }
8601         notfirst = 1;
8602     }
8603     if (notfirst) {
8604         if (tok == '=') {
8605             next();
8606         } else {
8607             if (!gnu_ext)
8608                 expect("=");
8609         }
8610     } else {
8611         if (type->t & VT_ARRAY) {
8612             index = *cur_index;
8613             type = pointed_type(type);
8614             c += index * type_size(type, &align);
8615         } else {
8616             f = *cur_field;
8617             if (!f)
8618                 error("too many field init");
8619             /* XXX: fix this mess by using explicit storage field */
8620             type1 = f->type;
8621             type1.t |= (type->t & ~VT_TYPE);
8622             type = &type1;
8623             c += f->c;
8624         }
8625     }
8626     decl_initializer(type, sec, c, 0, size_only);
8627
8628     /* XXX: make it more general */
8629     if (!size_only && nb_elems > 1) {
8630         unsigned long c_end;
8631         uint8_t *src, *dst;
8632         int i;
8633
8634         if (!sec)
8635             error("range init not supported yet for dynamic storage");
8636         c_end = c + nb_elems * elem_size;
8637         if (c_end > sec->data_allocated)
8638             section_realloc(sec, c_end);
8639         src = sec->data + c;
8640         dst = src;
8641         for(i = 1; i < nb_elems; i++) {
8642             dst += elem_size;
8643             memcpy(dst, src, elem_size);
8644         }
8645     }
8646 }
8647
8648 #define EXPR_VAL   0
8649 #define EXPR_CONST 1
8650 #define EXPR_ANY   2
8651
8652 /* store a value or an expression directly in global data or in local array */
8653 static void init_putv(CType *type, Section *sec, unsigned long c,
8654                       int v, int expr_type)
8655 {
8656     int saved_global_expr, bt, bit_pos, bit_size;
8657     void *ptr;
8658     unsigned long long bit_mask;
8659     CType dtype;
8660
8661     switch(expr_type) {
8662     case EXPR_VAL:
8663         vpushi(v);
8664         break;
8665     case EXPR_CONST:
8666         /* compound literals must be allocated globally in this case */
8667         saved_global_expr = global_expr;
8668         global_expr = 1;
8669         expr_const1();
8670         global_expr = saved_global_expr;
8671         /* NOTE: symbols are accepted */
8672         if ((vtop->r & (VT_VALMASK | VT_LVAL)) != VT_CONST)
8673             error("initializer element is not constant");
8674         break;
8675     case EXPR_ANY:
8676         expr_eq();
8677         break;
8678     }
8679
8680     dtype = *type;
8681     dtype.t &= ~VT_CONSTANT; /* need to do that to avoid false warning */
8682
8683     if (sec) {
8684         /* XXX: not portable */
8685         /* XXX: generate error if incorrect relocation */
8686         gen_assign_cast(&dtype);
8687         bt = type->t & VT_BTYPE;
8688         ptr = sec->data + c;
8689         /* XXX: make code faster ? */
8690         if (!(type->t & VT_BITFIELD)) {
8691             bit_pos = 0;
8692             bit_size = 32;
8693             bit_mask = -1LL;
8694         } else {
8695             bit_pos = (vtop->type.t >> VT_STRUCT_SHIFT) & 0x3f;
8696             bit_size = (vtop->type.t >> (VT_STRUCT_SHIFT + 6)) & 0x3f;
8697             bit_mask = (1LL << bit_size) - 1;
8698         }
8699         if ((vtop->r & VT_SYM) &&
8700             (bt == VT_BYTE ||
8701              bt == VT_SHORT ||
8702              bt == VT_DOUBLE ||
8703              bt == VT_LDOUBLE ||
8704              bt == VT_LLONG ||
8705              (bt == VT_INT && bit_size != 32)))
8706             error("initializer element is not computable at load time");
8707         switch(bt) {
8708         case VT_BOOL:
8709             vtop->c.i = (vtop->c.i != 0);
8710         case VT_BYTE:
8711             *(char *)ptr |= (vtop->c.i & bit_mask) << bit_pos;
8712             break;
8713         case VT_SHORT:
8714             *(short *)ptr |= (vtop->c.i & bit_mask) << bit_pos;
8715             break;
8716         case VT_DOUBLE:
8717             *(double *)ptr = vtop->c.d;
8718             break;
8719         case VT_LDOUBLE:
8720             *(long double *)ptr = vtop->c.ld;
8721             break;
8722         case VT_LLONG:
8723             *(long long *)ptr |= (vtop->c.ll & bit_mask) << bit_pos;
8724             break;
8725         default:
8726             if (vtop->r & VT_SYM) {
8727                 greloc(sec, vtop->sym, c, R_DATA_32);
8728             }
8729             *(int *)ptr |= (vtop->c.i & bit_mask) << bit_pos;
8730             break;
8731         }
8732         vtop--;
8733     } else {
8734         vset(&dtype, VT_LOCAL|VT_LVAL, c);
8735         vswap();
8736         vstore();
8737         vpop();
8738     }
8739 }
8740
8741 /* put zeros for variable based init */
8742 static void init_putz(CType *t, Section *sec, unsigned long c, int size)
8743 {
8744     if (sec) {
8745         /* nothing to do because globals are already set to zero */
8746     } else {
8747         vpush_global_sym(&func_old_type, TOK_memset);
8748         vseti(VT_LOCAL, c);
8749         vpushi(0);
8750         vpushi(size);
8751         gfunc_call(3);
8752     }
8753 }
8754
8755 /* 't' contains the type and storage info. 'c' is the offset of the
8756    object in section 'sec'. If 'sec' is NULL, it means stack based
8757    allocation. 'first' is true if array '{' must be read (multi
8758    dimension implicit array init handling). 'size_only' is true if
8759    size only evaluation is wanted (only for arrays). */
8760 static void decl_initializer(CType *type, Section *sec, unsigned long c,
8761                              int first, int size_only)
8762 {
8763     int index, array_length, n, no_oblock, nb, parlevel, i;
8764     int size1, align1, expr_type;
8765     Sym *s, *f;
8766     CType *t1;
8767
8768     if (type->t & VT_ARRAY) {
8769         s = type->ref;
8770         n = s->c;
8771         array_length = 0;
8772         t1 = pointed_type(type);
8773         size1 = type_size(t1, &align1);
8774
8775         no_oblock = 1;
8776         if ((first && tok != TOK_LSTR && tok != TOK_STR) ||
8777             tok == '{') {
8778             skip('{');
8779             no_oblock = 0;
8780         }
8781
8782         /* only parse strings here if correct type (otherwise: handle
8783            them as ((w)char *) expressions */
8784         if ((tok == TOK_LSTR &&
8785 #ifdef TCC_TARGET_PE
8786              (t1->t & VT_BTYPE) == VT_SHORT && (t1->t & VT_UNSIGNED)
8787 #else
8788              (t1->t & VT_BTYPE) == VT_INT
8789 #endif
8790             ) || (tok == TOK_STR && (t1->t & VT_BTYPE) == VT_BYTE)) {
8791             while (tok == TOK_STR || tok == TOK_LSTR) {
8792                 int cstr_len, ch;
8793                 CString *cstr;
8794
8795                 cstr = tokc.cstr;
8796                 /* compute maximum number of chars wanted */
8797                 if (tok == TOK_STR)
8798                     cstr_len = cstr->size;
8799                 else
8800                     cstr_len = cstr->size / sizeof(nwchar_t);
8801                 cstr_len--;
8802                 nb = cstr_len;
8803                 if (n >= 0 && nb > (n - array_length))
8804                     nb = n - array_length;
8805                 if (!size_only) {
8806                     if (cstr_len > nb)
8807                         warning("initializer-string for array is too long");
8808                     /* in order to go faster for common case (char
8809                        string in global variable, we handle it
8810                        specifically */
8811                     if (sec && tok == TOK_STR && size1 == 1) {
8812                         memcpy(sec->data + c + array_length, cstr->data, nb);
8813                     } else {
8814                         for(i=0;i<nb;i++) {
8815                             if (tok == TOK_STR)
8816                                 ch = ((unsigned char *)cstr->data)[i];
8817                             else
8818                                 ch = ((nwchar_t *)cstr->data)[i];
8819                             init_putv(t1, sec, c + (array_length + i) * size1,
8820                                       ch, EXPR_VAL);
8821                         }
8822                     }
8823                 }
8824                 array_length += nb;
8825                 next();
8826             }
8827             /* only add trailing zero if enough storage (no
8828                warning in this case since it is standard) */
8829             if (n < 0 || array_length < n) {
8830                 if (!size_only) {
8831                     init_putv(t1, sec, c + (array_length * size1), 0, EXPR_VAL);
8832                 }
8833                 array_length++;
8834             }
8835         } else {
8836             index = 0;
8837             while (tok != '}') {
8838                 decl_designator(type, sec, c, &index, NULL, size_only);
8839                 if (n >= 0 && index >= n)
8840                     error("index too large");
8841                 /* must put zero in holes (note that doing it that way
8842                    ensures that it even works with designators) */
8843                 if (!size_only && array_length < index) {
8844                     init_putz(t1, sec, c + array_length * size1,
8845                               (index - array_length) * size1);
8846                 }
8847                 index++;
8848                 if (index > array_length)
8849                     array_length = index;
8850                 /* special test for multi dimensional arrays (may not
8851                    be strictly correct if designators are used at the
8852                    same time) */
8853                 if (index >= n && no_oblock)
8854                     break;
8855                 if (tok == '}')
8856                     break;
8857                 skip(',');
8858             }
8859         }
8860         if (!no_oblock)
8861             skip('}');
8862         /* put zeros at the end */
8863         if (!size_only && n >= 0 && array_length < n) {
8864             init_putz(t1, sec, c + array_length * size1,
8865                       (n - array_length) * size1);
8866         }
8867         /* patch type size if needed */
8868         if (n < 0)
8869             s->c = array_length;
8870     } else if ((type->t & VT_BTYPE) == VT_STRUCT &&
8871                (sec || !first || tok == '{')) {
8872         int par_count;
8873
8874         /* NOTE: the previous test is a specific case for automatic
8875            struct/union init */
8876         /* XXX: union needs only one init */
8877
8878         /* XXX: this test is incorrect for local initializers
8879            beginning with ( without {. It would be much more difficult
8880            to do it correctly (ideally, the expression parser should
8881            be used in all cases) */
8882         par_count = 0;
8883         if (tok == '(') {
8884             AttributeDef ad1;
8885             CType type1;
8886             next();
8887             while (tok == '(') {
8888                 par_count++;
8889                 next();
8890             }
8891             if (!parse_btype(&type1, &ad1))
8892                 expect("cast");
8893             type_decl(&type1, &ad1, &n, TYPE_ABSTRACT);
8894 #if 0
8895             if (!is_assignable_types(type, &type1))
8896                 error("invalid type for cast");
8897 #endif
8898             skip(')');
8899         }
8900         no_oblock = 1;
8901         if (first || tok == '{') {
8902             skip('{');
8903             no_oblock = 0;
8904         }
8905         s = type->ref;
8906         f = s->next;
8907         array_length = 0;
8908         index = 0;
8909         n = s->c;
8910         while (tok != '}') {
8911             decl_designator(type, sec, c, NULL, &f, size_only);
8912             index = f->c;
8913             if (!size_only && array_length < index) {
8914                 init_putz(type, sec, c + array_length,
8915                           index - array_length);
8916             }
8917             index = index + type_size(&f->type, &align1);
8918             if (index > array_length)
8919                 array_length = index;
8920             f = f->next;
8921             if (no_oblock && f == NULL)
8922                 break;
8923             if (tok == '}')
8924                 break;
8925             skip(',');
8926         }
8927         /* put zeros at the end */
8928         if (!size_only && array_length < n) {
8929             init_putz(type, sec, c + array_length,
8930                       n - array_length);
8931         }
8932         if (!no_oblock)
8933             skip('}');
8934         while (par_count) {
8935             skip(')');
8936             par_count--;
8937         }
8938     } else if (tok == '{') {
8939         next();
8940         decl_initializer(type, sec, c, first, size_only);
8941         skip('}');
8942     } else if (size_only) {
8943         /* just skip expression */
8944         parlevel = 0;
8945         while ((parlevel > 0 || (tok != '}' && tok != ',')) &&
8946                tok != -1) {
8947             if (tok == '(')
8948                 parlevel++;
8949             else if (tok == ')')
8950                 parlevel--;
8951             next();
8952         }
8953     } else {
8954         /* currently, we always use constant expression for globals
8955            (may change for scripting case) */
8956         expr_type = EXPR_CONST;
8957         if (!sec)
8958             expr_type = EXPR_ANY;
8959         init_putv(type, sec, c, 0, expr_type);
8960     }
8961 }
8962
8963 /* parse an initializer for type 't' if 'has_init' is non zero, and
8964    allocate space in local or global data space ('r' is either
8965    VT_LOCAL or VT_CONST). If 'v' is non zero, then an associated
8966    variable 'v' of scope 'scope' is declared before initializers are
8967    parsed. If 'v' is zero, then a reference to the new object is put
8968    in the value stack. If 'has_init' is 2, a special parsing is done
8969    to handle string constants. */
8970 static void decl_initializer_alloc(CType *type, AttributeDef *ad, int r,
8971                                    int has_init, int v, int scope)
8972 {
8973     int size, align, addr, data_offset;
8974     int level;
8975     ParseState saved_parse_state;
8976     TokenString init_str;
8977     Section *sec;
8978
8979     size = type_size(type, &align);
8980     /* If unknown size, we must evaluate it before
8981        evaluating initializers because
8982        initializers can generate global data too
8983        (e.g. string pointers or ISOC99 compound
8984        literals). It also simplifies local
8985        initializers handling */
8986     tok_str_new(&init_str);
8987     if (size < 0) {
8988         if (!has_init)
8989             error("unknown type size");
8990         /* get all init string */
8991         if (has_init == 2) {
8992             /* only get strings */
8993             while (tok == TOK_STR || tok == TOK_LSTR) {
8994                 tok_str_add_tok(&init_str);
8995                 next();
8996             }
8997         } else {
8998             level = 0;
8999             while (level > 0 || (tok != ',' && tok != ';')) {
9000                 if (tok < 0)
9001                     error("unexpected end of file in initializer");
9002                 tok_str_add_tok(&init_str);
9003                 if (tok == '{')
9004                     level++;
9005                 else if (tok == '}') {
9006                     if (level == 0)
9007                         break;
9008                     level--;
9009                 }
9010                 next();
9011             }
9012         }
9013         tok_str_add(&init_str, -1);
9014         tok_str_add(&init_str, 0);
9015
9016         /* compute size */
9017         save_parse_state(&saved_parse_state);
9018
9019         macro_ptr = init_str.str;
9020         next();
9021         decl_initializer(type, NULL, 0, 1, 1);
9022         /* prepare second initializer parsing */
9023         macro_ptr = init_str.str;
9024         next();
9025
9026         /* if still unknown size, error */
9027         size = type_size(type, &align);
9028         if (size < 0)
9029             error("unknown type size");
9030     }
9031     /* take into account specified alignment if bigger */
9032     if (ad->aligned) {
9033         if (ad->aligned > align)
9034             align = ad->aligned;
9035     } else if (ad->packed) {
9036         align = 1;
9037     }
9038     if ((r & VT_VALMASK) == VT_LOCAL) {
9039         sec = NULL;
9040         if (do_bounds_check && (type->t & VT_ARRAY))
9041             loc--;
9042         loc = (loc - size) & -align;
9043         addr = loc;
9044         /* handles bounds */
9045         /* XXX: currently, since we do only one pass, we cannot track
9046            '&' operators, so we add only arrays */
9047         if (do_bounds_check && (type->t & VT_ARRAY)) {
9048             unsigned long *bounds_ptr;
9049             /* add padding between regions */
9050             loc--;
9051             /* then add local bound info */
9052             bounds_ptr = section_ptr_add(lbounds_section, 2 * sizeof(unsigned long));
9053             bounds_ptr[0] = addr;
9054             bounds_ptr[1] = size;
9055         }
9056         if (v) {
9057             /* local variable */
9058             sym_push(v, type, r, addr);
9059         } else {
9060             /* push local reference */
9061             vset(type, r, addr);
9062         }
9063     } else {
9064         Sym *sym;
9065
9066         sym = NULL;
9067         if (v && scope == VT_CONST) {
9068             /* see if the symbol was already defined */
9069             sym = sym_find(v);
9070             if (sym) {
9071                 if (!is_compatible_types(&sym->type, type))
9072                     error("incompatible types for redefinition of '%s'",
9073                           get_tok_str(v, NULL));
9074                 if (sym->type.t & VT_EXTERN) {
9075                     /* if the variable is extern, it was not allocated */
9076                     sym->type.t &= ~VT_EXTERN;
9077                     /* set array size if it was ommited in extern
9078                        declaration */
9079                     if ((sym->type.t & VT_ARRAY) &&
9080                         sym->type.ref->c < 0 &&
9081                         type->ref->c >= 0)
9082                         sym->type.ref->c = type->ref->c;
9083                 } else {
9084                     /* we accept several definitions of the same
9085                        global variable. this is tricky, because we
9086                        must play with the SHN_COMMON type of the symbol */
9087                     /* XXX: should check if the variable was already
9088                        initialized. It is incorrect to initialized it
9089                        twice */
9090                     /* no init data, we won't add more to the symbol */
9091                     if (!has_init)
9092                         goto no_alloc;
9093                 }
9094             }
9095         }
9096
9097         /* allocate symbol in corresponding section */
9098         sec = ad->section;
9099         if (!sec) {
9100             if (has_init)
9101                 sec = data_section;
9102             else if (tcc_state->nocommon)
9103                 sec = bss_section;
9104         }
9105         if (sec) {
9106             data_offset = sec->data_offset;
9107             data_offset = (data_offset + align - 1) & -align;
9108             addr = data_offset;
9109             /* very important to increment global pointer at this time
9110                because initializers themselves can create new initializers */
9111             data_offset += size;
9112             /* add padding if bound check */
9113             if (do_bounds_check)
9114                 data_offset++;
9115             sec->data_offset = data_offset;
9116             /* allocate section space to put the data */
9117             if (sec->sh_type != SHT_NOBITS &&
9118                 data_offset > sec->data_allocated)
9119                 section_realloc(sec, data_offset);
9120             /* align section if needed */
9121             if (align > sec->sh_addralign)
9122                 sec->sh_addralign = align;
9123         } else {
9124             addr = 0; /* avoid warning */
9125         }
9126
9127         if (v) {
9128             if (scope != VT_CONST || !sym) {
9129                 sym = sym_push(v, type, r | VT_SYM, 0);
9130             }
9131             /* update symbol definition */
9132             if (sec) {
9133                 put_extern_sym(sym, sec, addr, size);
9134             } else {
9135                 Elf32_Sym *esym;
9136                 /* put a common area */
9137                 put_extern_sym(sym, NULL, align, size);
9138                 /* XXX: find a nicer way */
9139                 esym = &((Elf32_Sym *)symtab_section->data)[sym->c];
9140                 esym->st_shndx = SHN_COMMON;
9141             }
9142         } else {
9143             CValue cval;
9144
9145             /* push global reference */
9146             sym = get_sym_ref(type, sec, addr, size);
9147             cval.ul = 0;
9148             vsetc(type, VT_CONST | VT_SYM, &cval);
9149             vtop->sym = sym;
9150         }
9151
9152         /* handles bounds now because the symbol must be defined
9153            before for the relocation */
9154         if (do_bounds_check) {
9155             unsigned long *bounds_ptr;
9156
9157             greloc(bounds_section, sym, bounds_section->data_offset, R_DATA_32);
9158             /* then add global bound info */
9159             bounds_ptr = section_ptr_add(bounds_section, 2 * sizeof(long));
9160             bounds_ptr[0] = 0; /* relocated */
9161             bounds_ptr[1] = size;
9162         }
9163     }
9164     if (has_init) {
9165         decl_initializer(type, sec, addr, 1, 0);
9166         /* restore parse state if needed */
9167         if (init_str.str) {
9168             tok_str_free(init_str.str);
9169             restore_parse_state(&saved_parse_state);
9170         }
9171     }
9172  no_alloc: ;
9173 }
9174
9175 void put_func_debug(Sym *sym)
9176 {
9177     char buf[512];
9178
9179     /* stabs info */
9180     /* XXX: we put here a dummy type */
9181     snprintf(buf, sizeof(buf), "%s:%c1",
9182              funcname, sym->type.t & VT_STATIC ? 'f' : 'F');
9183     put_stabs_r(buf, N_FUN, 0, file->line_num, 0,
9184                 cur_text_section, sym->c);
9185     /* //gr gdb wants a line at the function */
9186     put_stabn(N_SLINE, 0, file->line_num, 0);
9187     last_ind = 0;
9188     last_line_num = 0;
9189 }
9190
9191 /* parse an old style function declaration list */
9192 /* XXX: check multiple parameter */
9193 static void func_decl_list(Sym *func_sym)
9194 {
9195     AttributeDef ad;
9196     int v;
9197     Sym *s;
9198     CType btype, type;
9199
9200     /* parse each declaration */
9201     while (tok != '{' && tok != ';' && tok != ',' && tok != TOK_EOF) {
9202         if (!parse_btype(&btype, &ad))
9203             expect("declaration list");
9204         if (((btype.t & VT_BTYPE) == VT_ENUM ||
9205              (btype.t & VT_BTYPE) == VT_STRUCT) &&
9206             tok == ';') {
9207             /* we accept no variable after */
9208         } else {
9209             for(;;) {
9210                 type = btype;
9211                 type_decl(&type, &ad, &v, TYPE_DIRECT);
9212                 /* find parameter in function parameter list */
9213                 s = func_sym->next;
9214                 while (s != NULL) {
9215                     if ((s->v & ~SYM_FIELD) == v)
9216                         goto found;
9217                     s = s->next;
9218                 }
9219                 error("declaration for parameter '%s' but no such parameter",
9220                       get_tok_str(v, NULL));
9221             found:
9222                 /* check that no storage specifier except 'register' was given */
9223                 if (type.t & VT_STORAGE)
9224                     error("storage class specified for '%s'", get_tok_str(v, NULL));
9225                 convert_parameter_type(&type);
9226                 /* we can add the type (NOTE: it could be local to the function) */
9227                 s->type = type;
9228                 /* accept other parameters */
9229                 if (tok == ',')
9230                     next();
9231                 else
9232                     break;
9233             }
9234         }
9235         skip(';');
9236     }
9237 }
9238
9239 /* parse a function defined by symbol 'sym' and generate its code in
9240    'cur_text_section' */
9241 static void gen_function(Sym *sym)
9242 {
9243     int saved_nocode_wanted = nocode_wanted;
9244     nocode_wanted = 0;
9245     ind = cur_text_section->data_offset;
9246     /* NOTE: we patch the symbol size later */
9247     put_extern_sym(sym, cur_text_section, ind, 0);
9248     funcname = get_tok_str(sym->v, NULL);
9249     func_ind = ind;
9250     /* put debug symbol */
9251     if (do_debug)
9252         put_func_debug(sym);
9253     /* push a dummy symbol to enable local sym storage */
9254     sym_push2(&local_stack, SYM_FIELD, 0, 0);
9255     gfunc_prolog(&sym->type);
9256     rsym = 0;
9257     block(NULL, NULL, NULL, NULL, 0, 0);
9258     gsym(rsym);
9259     gfunc_epilog();
9260     cur_text_section->data_offset = ind;
9261     label_pop(&global_label_stack, NULL);
9262     sym_pop(&local_stack, NULL); /* reset local stack */
9263     /* end of function */
9264     /* patch symbol size */
9265     ((Elf32_Sym *)symtab_section->data)[sym->c].st_size =
9266         ind - func_ind;
9267     if (do_debug) {
9268         put_stabn(N_FUN, 0, 0, ind - func_ind);
9269     }
9270     /* It's better to crash than to generate wrong code */
9271     cur_text_section = NULL;
9272     funcname = ""; /* for safety */
9273     func_vt.t = VT_VOID; /* for safety */
9274     ind = 0; /* for safety */
9275     nocode_wanted = saved_nocode_wanted;
9276 }
9277
9278 static void gen_inline_functions(void)
9279 {
9280     Sym *sym;
9281     CType *type;
9282     int *str, inline_generated;
9283
9284     /* iterate while inline function are referenced */
9285     for(;;) {
9286         inline_generated = 0;
9287         for(sym = global_stack; sym != NULL; sym = sym->prev) {
9288             type = &sym->type;
9289             if (((type->t & VT_BTYPE) == VT_FUNC) &&
9290                 (type->t & (VT_STATIC | VT_INLINE)) ==
9291                 (VT_STATIC | VT_INLINE) &&
9292                 sym->c != 0) {
9293                 /* the function was used: generate its code and
9294                    convert it to a normal function */
9295                 str = INLINE_DEF(sym->r);
9296                 sym->r = VT_SYM | VT_CONST;
9297                 sym->type.t &= ~VT_INLINE;
9298
9299                 macro_ptr = str;
9300                 next();
9301                 cur_text_section = text_section;
9302                 gen_function(sym);
9303                 macro_ptr = NULL; /* fail safe */
9304
9305                 tok_str_free(str);
9306                 inline_generated = 1;
9307             }
9308         }
9309         if (!inline_generated)
9310             break;
9311     }
9312
9313     /* free all remaining inline function tokens */
9314     for(sym = global_stack; sym != NULL; sym = sym->prev) {
9315         type = &sym->type;
9316         if (((type->t & VT_BTYPE) == VT_FUNC) &&
9317             (type->t & (VT_STATIC | VT_INLINE)) ==
9318             (VT_STATIC | VT_INLINE)) {
9319             //gr printf("sym %d %s\n", sym->r, get_tok_str(sym->v, NULL));
9320             if (sym->r == (VT_SYM | VT_CONST)) //gr beware!
9321                 continue;
9322             str = INLINE_DEF(sym->r);
9323             tok_str_free(str);
9324             sym->r = 0; /* fail safe */
9325         }
9326     }
9327 }
9328
9329 /* 'l' is VT_LOCAL or VT_CONST to define default storage type */
9330 static void decl(int l)
9331 {
9332     int v, has_init, r;
9333     CType type, btype;
9334     Sym *sym;
9335     AttributeDef ad;
9336
9337     while (1) {
9338         if (!parse_btype(&btype, &ad)) {
9339             /* skip redundant ';' */
9340             /* XXX: find more elegant solution */
9341             if (tok == ';') {
9342                 next();
9343                 continue;
9344             }
9345             if (l == VT_CONST &&
9346                 (tok == TOK_ASM1 || tok == TOK_ASM2 || tok == TOK_ASM3)) {
9347                 /* global asm block */
9348                 asm_global_instr();
9349                 continue;
9350             }
9351             /* special test for old K&R protos without explicit int
9352                type. Only accepted when defining global data */
9353             if (l == VT_LOCAL || tok < TOK_DEFINE)
9354                 break;
9355             btype.t = VT_INT;
9356         }
9357         if (((btype.t & VT_BTYPE) == VT_ENUM ||
9358              (btype.t & VT_BTYPE) == VT_STRUCT) &&
9359             tok == ';') {
9360             /* we accept no variable after */
9361             next();
9362             continue;
9363         }
9364         while (1) { /* iterate thru each declaration */
9365             type = btype;
9366             type_decl(&type, &ad, &v, TYPE_DIRECT);
9367 #if 0
9368             {
9369                 char buf[500];
9370                 type_to_str(buf, sizeof(buf), t, get_tok_str(v, NULL));
9371                 printf("type = '%s'\n", buf);
9372             }
9373 #endif
9374             if ((type.t & VT_BTYPE) == VT_FUNC) {
9375                 /* if old style function prototype, we accept a
9376                    declaration list */
9377                 sym = type.ref;
9378                 if (sym->c == FUNC_OLD)
9379                     func_decl_list(sym);
9380             }
9381
9382             if (tok == '{') {
9383                 if (l == VT_LOCAL)
9384                     error("cannot use local functions");
9385                 if ((type.t & VT_BTYPE) != VT_FUNC)
9386                     expect("function definition");
9387
9388                 /* reject abstract declarators in function definition */
9389                 sym = type.ref;
9390                 while ((sym = sym->next) != NULL)
9391                     if (!(sym->v & ~SYM_FIELD))
9392                        expect("identifier");
9393
9394                 /* XXX: cannot do better now: convert extern line to static inline */
9395                 if ((type.t & (VT_EXTERN | VT_INLINE)) == (VT_EXTERN | VT_INLINE))
9396                     type.t = (type.t & ~VT_EXTERN) | VT_STATIC;
9397
9398                 sym = sym_find(v);
9399                 if (sym) {
9400                     if ((sym->type.t & VT_BTYPE) != VT_FUNC)
9401                         goto func_error1;
9402                     /* specific case: if not func_call defined, we put
9403                        the one of the prototype */
9404                     /* XXX: should have default value */
9405                     r = sym->type.ref->r;
9406                     if (FUNC_CALL(r) != FUNC_CDECL
9407                      && FUNC_CALL(type.ref->r) == FUNC_CDECL)
9408                         FUNC_CALL(type.ref->r) = FUNC_CALL(r);
9409                     if (FUNC_EXPORT(r))
9410                         FUNC_EXPORT(type.ref->r) = 1;
9411
9412                     if (!is_compatible_types(&sym->type, &type)) {
9413                     func_error1:
9414                         error("incompatible types for redefinition of '%s'",
9415                               get_tok_str(v, NULL));
9416                     }
9417                     /* if symbol is already defined, then put complete type */
9418                     sym->type = type;
9419                 } else {
9420                     /* put function symbol */
9421                     sym = global_identifier_push(v, type.t, 0);
9422                     sym->type.ref = type.ref;
9423                 }
9424
9425                 /* static inline functions are just recorded as a kind
9426                    of macro. Their code will be emitted at the end of
9427                    the compilation unit only if they are used */
9428                 if ((type.t & (VT_INLINE | VT_STATIC)) ==
9429                     (VT_INLINE | VT_STATIC)) {
9430                     TokenString func_str;
9431                     int block_level;
9432
9433                     tok_str_new(&func_str);
9434
9435                     block_level = 0;
9436                     for(;;) {
9437                         int t;
9438                         if (tok == TOK_EOF)
9439                             error("unexpected end of file");
9440                         tok_str_add_tok(&func_str);
9441                         t = tok;
9442                         next();
9443                         if (t == '{') {
9444                             block_level++;
9445                         } else if (t == '}') {
9446                             block_level--;
9447                             if (block_level == 0)
9448                                 break;
9449                         }
9450                     }
9451                     tok_str_add(&func_str, -1);
9452                     tok_str_add(&func_str, 0);
9453                     INLINE_DEF(sym->r) = func_str.str;
9454                 } else {
9455                     /* compute text section */
9456                     cur_text_section = ad.section;
9457                     if (!cur_text_section)
9458                         cur_text_section = text_section;
9459                     sym->r = VT_SYM | VT_CONST;
9460                     gen_function(sym);
9461                 }
9462                 break;
9463             } else {
9464                 if (btype.t & VT_TYPEDEF) {
9465                     /* save typedefed type  */
9466                     /* XXX: test storage specifiers ? */
9467                     sym = sym_push(v, &type, 0, 0);
9468                     sym->type.t |= VT_TYPEDEF;
9469                 } else if ((type.t & VT_BTYPE) == VT_FUNC) {
9470                     /* external function definition */
9471                     /* specific case for func_call attribute */
9472                     if (ad.func_attr)
9473                         type.ref->r = ad.func_attr;
9474                     external_sym(v, &type, 0);
9475                 } else {
9476                     /* not lvalue if array */
9477                     r = 0;
9478                     if (!(type.t & VT_ARRAY))
9479                         r |= lvalue_type(type.t);
9480                     has_init = (tok == '=');
9481                     if ((btype.t & VT_EXTERN) ||
9482                         ((type.t & VT_ARRAY) && (type.t & VT_STATIC) &&
9483                          !has_init && l == VT_CONST && type.ref->c < 0)) {
9484                         /* external variable */
9485                         /* NOTE: as GCC, uninitialized global static
9486                            arrays of null size are considered as
9487                            extern */
9488                         external_sym(v, &type, r);
9489                     } else {
9490                         type.t |= (btype.t & VT_STATIC); /* Retain "static". */
9491                         if (type.t & VT_STATIC)
9492                             r |= VT_CONST;
9493                         else
9494                             r |= l;
9495                         if (has_init)
9496                             next();
9497                         decl_initializer_alloc(&type, &ad, r,
9498                                                has_init, v, l);
9499                     }
9500                 }
9501                 if (tok != ',') {
9502                     skip(';');
9503                     break;
9504                 }
9505                 next();
9506             }
9507         }
9508     }
9509 }
9510
9511 /* better than nothing, but needs extension to handle '-E' option
9512    correctly too */
9513 static void preprocess_init(TCCState *s1)
9514 {
9515     s1->include_stack_ptr = s1->include_stack;
9516     /* XXX: move that before to avoid having to initialize
9517        file->ifdef_stack_ptr ? */
9518     s1->ifdef_stack_ptr = s1->ifdef_stack;
9519     file->ifdef_stack_ptr = s1->ifdef_stack_ptr;
9520
9521     /* XXX: not ANSI compliant: bound checking says error */
9522     vtop = vstack - 1;
9523     s1->pack_stack[0] = 0;
9524     s1->pack_stack_ptr = s1->pack_stack;
9525 }
9526
9527 /* compile the C file opened in 'file'. Return non zero if errors. */
9528 static int tcc_compile(TCCState *s1)
9529 {
9530     Sym *define_start;
9531     char buf[512];
9532     volatile int section_sym;
9533
9534 #ifdef INC_DEBUG
9535     printf("%s: **** new file\n", file->filename);
9536 #endif
9537     preprocess_init(s1);
9538
9539     cur_text_section = NULL;
9540     funcname = "";
9541     anon_sym = SYM_FIRST_ANOM;
9542
9543     /* file info: full path + filename */
9544     section_sym = 0; /* avoid warning */
9545     if (do_debug) {
9546         section_sym = put_elf_sym(symtab_section, 0, 0,
9547                                   ELF32_ST_INFO(STB_LOCAL, STT_SECTION), 0,
9548                                   text_section->sh_num, NULL);
9549         getcwd(buf, sizeof(buf));
9550 #ifdef _WIN32
9551         normalize_slashes(buf);
9552 #endif
9553         pstrcat(buf, sizeof(buf), "/");
9554         put_stabs_r(buf, N_SO, 0, 0,
9555                     text_section->data_offset, text_section, section_sym);
9556         put_stabs_r(file->filename, N_SO, 0, 0,
9557                     text_section->data_offset, text_section, section_sym);
9558     }
9559     /* an elf symbol of type STT_FILE must be put so that STB_LOCAL
9560        symbols can be safely used */
9561     put_elf_sym(symtab_section, 0, 0,
9562                 ELF32_ST_INFO(STB_LOCAL, STT_FILE), 0,
9563                 SHN_ABS, file->filename);
9564
9565     /* define some often used types */
9566     int_type.t = VT_INT;
9567
9568     char_pointer_type.t = VT_BYTE;
9569     mk_pointer(&char_pointer_type);
9570
9571     func_old_type.t = VT_FUNC;
9572     func_old_type.ref = sym_push(SYM_FIELD, &int_type, FUNC_CDECL, FUNC_OLD);
9573
9574 #if defined(TCC_ARM_EABI) && defined(TCC_ARM_VFP)
9575     float_type.t = VT_FLOAT;
9576     double_type.t = VT_DOUBLE;
9577
9578     func_float_type.t = VT_FUNC;
9579     func_float_type.ref = sym_push(SYM_FIELD, &float_type, FUNC_CDECL, FUNC_OLD);
9580     func_double_type.t = VT_FUNC;
9581     func_double_type.ref = sym_push(SYM_FIELD, &double_type, FUNC_CDECL, FUNC_OLD);
9582 #endif
9583
9584 #if 0
9585     /* define 'void *alloca(unsigned int)' builtin function */
9586     {
9587         Sym *s1;
9588
9589         p = anon_sym++;
9590         sym = sym_push(p, mk_pointer(VT_VOID), FUNC_CDECL, FUNC_NEW);
9591         s1 = sym_push(SYM_FIELD, VT_UNSIGNED | VT_INT, 0, 0);
9592         s1->next = NULL;
9593         sym->next = s1;
9594         sym_push(TOK_alloca, VT_FUNC | (p << VT_STRUCT_SHIFT), VT_CONST, 0);
9595     }
9596 #endif
9597
9598     define_start = define_stack;
9599     nocode_wanted = 1;
9600
9601     if (setjmp(s1->error_jmp_buf) == 0) {
9602         s1->nb_errors = 0;
9603         s1->error_set_jmp_enabled = 1;
9604
9605         ch = file->buf_ptr[0];
9606         tok_flags = TOK_FLAG_BOL | TOK_FLAG_BOF;
9607         parse_flags = PARSE_FLAG_PREPROCESS | PARSE_FLAG_TOK_NUM;
9608         next();
9609         decl(VT_CONST);
9610         if (tok != TOK_EOF)
9611             expect("declaration");
9612
9613         /* end of translation unit info */
9614         if (do_debug) {
9615             put_stabs_r(NULL, N_SO, 0, 0,
9616                         text_section->data_offset, text_section, section_sym);
9617         }
9618     }
9619     s1->error_set_jmp_enabled = 0;
9620
9621     /* reset define stack, but leave -Dsymbols (may be incorrect if
9622        they are undefined) */
9623     free_defines(define_start);
9624
9625     gen_inline_functions();
9626
9627     sym_pop(&global_stack, NULL);
9628     sym_pop(&local_stack, NULL);
9629
9630     return s1->nb_errors != 0 ? -1 : 0;
9631 }
9632
9633 /* Preprocess the current file */
9634 /* XXX: add line and file infos, add options to preserve spaces */
9635 static int tcc_preprocess(TCCState *s1)
9636 {
9637     Sym *define_start;
9638     int last_is_space;
9639
9640     preprocess_init(s1);
9641
9642     define_start = define_stack;
9643
9644     ch = file->buf_ptr[0];
9645     tok_flags = TOK_FLAG_BOL | TOK_FLAG_BOF;
9646     parse_flags = PARSE_FLAG_ASM_COMMENTS | PARSE_FLAG_PREPROCESS |
9647         PARSE_FLAG_LINEFEED;
9648     last_is_space = 1;
9649     next();
9650     for(;;) {
9651         if (tok == TOK_EOF) {
9652             break;
9653         } else if (tok == TOK_LINEFEED) {
9654             last_is_space = 1;
9655         } else {
9656             if (!last_is_space)
9657                 fputc(' ', s1->outfile);
9658             last_is_space = 0;
9659         }
9660         fputs(get_tok_str(tok, &tokc), s1->outfile);
9661         next();
9662     }
9663     free_defines(define_start);
9664     return 0;
9665 }
9666
9667 #ifdef LIBTCC
9668 int tcc_compile_string(TCCState *s, const char *str)
9669 {
9670     BufferedFile bf1, *bf = &bf1;
9671     int ret, len;
9672     char *buf;
9673
9674     /* init file structure */
9675     bf->fd = -1;
9676     /* XXX: avoid copying */
9677     len = strlen(str);
9678     buf = tcc_malloc(len + 1);
9679     if (!buf)
9680         return -1;
9681     memcpy(buf, str, len);
9682     buf[len] = CH_EOB;
9683     bf->buf_ptr = buf;
9684     bf->buf_end = buf + len;
9685     pstrcpy(bf->filename, sizeof(bf->filename), "<string>");
9686     bf->line_num = 1;
9687     file = bf;
9688     ret = tcc_compile(s);
9689     file = NULL;
9690     tcc_free(buf);
9691
9692     /* currently, no need to close */
9693     return ret;
9694 }
9695 #endif
9696
9697 /* define a preprocessor symbol. A value can also be provided with the '=' operator */
9698 void tcc_define_symbol(TCCState *s1, const char *sym, const char *value)
9699 {
9700     BufferedFile bf1, *bf = &bf1;
9701
9702     pstrcpy(bf->buffer, IO_BUF_SIZE, sym);
9703     pstrcat(bf->buffer, IO_BUF_SIZE, " ");
9704     /* default value */
9705     if (!value)
9706         value = "1";
9707     pstrcat(bf->buffer, IO_BUF_SIZE, value);
9708
9709     /* init file structure */
9710     bf->fd = -1;
9711     bf->buf_ptr = bf->buffer;
9712     bf->buf_end = bf->buffer + strlen(bf->buffer);
9713     *bf->buf_end = CH_EOB;
9714     bf->filename[0] = '\0';
9715     bf->line_num = 1;
9716     file = bf;
9717
9718     s1->include_stack_ptr = s1->include_stack;
9719
9720     /* parse with define parser */
9721     ch = file->buf_ptr[0];
9722     next_nomacro();
9723     parse_define();
9724     file = NULL;
9725 }
9726
9727 /* undefine a preprocessor symbol */
9728 void tcc_undefine_symbol(TCCState *s1, const char *sym)
9729 {
9730     TokenSym *ts;
9731     Sym *s;
9732     ts = tok_alloc(sym, strlen(sym));
9733     s = define_find(ts->tok);
9734     /* undefine symbol by putting an invalid name */
9735     if (s)
9736         define_undef(s);
9737 }
9738
9739 #ifdef CONFIG_TCC_ASM
9740
9741 #ifdef TCC_TARGET_I386
9742 #include "i386-asm.c"
9743 #endif
9744 #include "tccasm.c"
9745
9746 #else
9747 static void asm_instr(void)
9748 {
9749     error("inline asm() not supported");
9750 }
9751 static void asm_global_instr(void)
9752 {
9753     error("inline asm() not supported");
9754 }
9755 #endif
9756
9757 #include "tccelf.c"
9758
9759 #ifdef TCC_TARGET_COFF
9760 #include "tcccoff.c"
9761 #endif
9762
9763 #ifdef TCC_TARGET_PE
9764 #include "tccpe.c"
9765 #endif
9766
9767 /* print the position in the source file of PC value 'pc' by reading
9768    the stabs debug information */
9769 static void rt_printline(unsigned long wanted_pc)
9770 {
9771     Stab_Sym *sym, *sym_end;
9772     char func_name[128], last_func_name[128];
9773     unsigned long func_addr, last_pc, pc;
9774     const char *incl_files[INCLUDE_STACK_SIZE];
9775     int incl_index, len, last_line_num, i;
9776     const char *str, *p;
9777
9778     fprintf(stderr, "0x%08lx:", wanted_pc);
9779
9780     func_name[0] = '\0';
9781     func_addr = 0;
9782     incl_index = 0;
9783     last_func_name[0] = '\0';
9784     last_pc = 0xffffffff;
9785     last_line_num = 1;
9786     sym = (Stab_Sym *)stab_section->data + 1;
9787     sym_end = (Stab_Sym *)(stab_section->data + stab_section->data_offset);
9788     while (sym < sym_end) {
9789         switch(sym->n_type) {
9790             /* function start or end */
9791         case N_FUN:
9792             if (sym->n_strx == 0) {
9793                 /* we test if between last line and end of function */
9794                 pc = sym->n_value + func_addr;
9795                 if (wanted_pc >= last_pc && wanted_pc < pc)
9796                     goto found;
9797                 func_name[0] = '\0';
9798                 func_addr = 0;
9799             } else {
9800                 str = stabstr_section->data + sym->n_strx;
9801                 p = strchr(str, ':');
9802                 if (!p) {
9803                     pstrcpy(func_name, sizeof(func_name), str);
9804                 } else {
9805                     len = p - str;
9806                     if (len > sizeof(func_name) - 1)
9807                         len = sizeof(func_name) - 1;
9808                     memcpy(func_name, str, len);
9809                     func_name[len] = '\0';
9810                 }
9811                 func_addr = sym->n_value;
9812             }
9813             break;
9814             /* line number info */
9815         case N_SLINE:
9816             pc = sym->n_value + func_addr;
9817             if (wanted_pc >= last_pc && wanted_pc < pc)
9818                 goto found;
9819             last_pc = pc;
9820             last_line_num = sym->n_desc;
9821             /* XXX: slow! */
9822             strcpy(last_func_name, func_name);
9823             break;
9824             /* include files */
9825         case N_BINCL:
9826             str = stabstr_section->data + sym->n_strx;
9827         add_incl:
9828             if (incl_index < INCLUDE_STACK_SIZE) {
9829                 incl_files[incl_index++] = str;
9830             }
9831             break;
9832         case N_EINCL:
9833             if (incl_index > 1)
9834                 incl_index--;
9835             break;
9836         case N_SO:
9837             if (sym->n_strx == 0) {
9838                 incl_index = 0; /* end of translation unit */
9839             } else {
9840                 str = stabstr_section->data + sym->n_strx;
9841                 /* do not add path */
9842                 len = strlen(str);
9843                 if (len > 0 && str[len - 1] != '/')
9844                     goto add_incl;
9845             }
9846             break;
9847         }
9848         sym++;
9849     }
9850
9851     /* second pass: we try symtab symbols (no line number info) */
9852     incl_index = 0;
9853     {
9854         Elf32_Sym *sym, *sym_end;
9855         int type;
9856
9857         sym_end = (Elf32_Sym *)(symtab_section->data + symtab_section->data_offset);
9858         for(sym = (Elf32_Sym *)symtab_section->data + 1;
9859             sym < sym_end;
9860             sym++) {
9861             type = ELF32_ST_TYPE(sym->st_info);
9862             if (type == STT_FUNC) {
9863                 if (wanted_pc >= sym->st_value &&
9864                     wanted_pc < sym->st_value + sym->st_size) {
9865                     pstrcpy(last_func_name, sizeof(last_func_name),
9866                             strtab_section->data + sym->st_name);
9867                     goto found;
9868                 }
9869             }
9870         }
9871     }
9872     /* did not find any info: */
9873     fprintf(stderr, " ???\n");
9874     return;
9875  found:
9876     if (last_func_name[0] != '\0') {
9877         fprintf(stderr, " %s()", last_func_name);
9878     }
9879     if (incl_index > 0) {
9880         fprintf(stderr, " (%s:%d",
9881                 incl_files[incl_index - 1], last_line_num);
9882         for(i = incl_index - 2; i >= 0; i--)
9883             fprintf(stderr, ", included from %s", incl_files[i]);
9884         fprintf(stderr, ")");
9885     }
9886     fprintf(stderr, "\n");
9887 }
9888
9889 #if !defined(_WIN32) && !defined(CONFIG_TCCBOOT)
9890
9891 #ifdef __i386__
9892
9893 /* fix for glibc 2.1 */
9894 #ifndef REG_EIP
9895 #define REG_EIP EIP
9896 #define REG_EBP EBP
9897 #endif
9898
9899 /* return the PC at frame level 'level'. Return non zero if not found */
9900 static int rt_get_caller_pc(unsigned long *paddr,
9901                             ucontext_t *uc, int level)
9902 {
9903     unsigned long fp;
9904     int i;
9905
9906     if (level == 0) {
9907 #if defined(__FreeBSD__)
9908         *paddr = uc->uc_mcontext.mc_eip;
9909 #elif defined(__dietlibc__)
9910         *paddr = uc->uc_mcontext.eip;
9911 #else
9912         *paddr = uc->uc_mcontext.gregs[REG_EIP];
9913 #endif
9914         return 0;
9915     } else {
9916 #if defined(__FreeBSD__)
9917         fp = uc->uc_mcontext.mc_ebp;
9918 #elif defined(__dietlibc__)
9919         fp = uc->uc_mcontext.ebp;
9920 #else
9921         fp = uc->uc_mcontext.gregs[REG_EBP];
9922 #endif
9923         for(i=1;i<level;i++) {
9924             /* XXX: check address validity with program info */
9925             if (fp <= 0x1000 || fp >= 0xc0000000)
9926                 return -1;
9927             fp = ((unsigned long *)fp)[0];
9928         }
9929         *paddr = ((unsigned long *)fp)[1];
9930         return 0;
9931     }
9932 }
9933 #else
9934
9935 #warning add arch specific rt_get_caller_pc()
9936
9937 static int rt_get_caller_pc(unsigned long *paddr,
9938                             ucontext_t *uc, int level)
9939 {
9940     return -1;
9941 }
9942 #endif
9943
9944 /* emit a run time error at position 'pc' */
9945 void rt_error(ucontext_t *uc, const char *fmt, ...)
9946 {
9947     va_list ap;
9948     unsigned long pc;
9949     int i;
9950
9951     va_start(ap, fmt);
9952     fprintf(stderr, "Runtime error: ");
9953     vfprintf(stderr, fmt, ap);
9954     fprintf(stderr, "\n");
9955     for(i=0;i<num_callers;i++) {
9956         if (rt_get_caller_pc(&pc, uc, i) < 0)
9957             break;
9958         if (i == 0)
9959             fprintf(stderr, "at ");
9960         else
9961             fprintf(stderr, "by ");
9962         rt_printline(pc);
9963     }
9964     exit(255);
9965     va_end(ap);
9966 }
9967
9968 /* signal handler for fatal errors */
9969 static void sig_error(int signum, siginfo_t *siginf, void *puc)
9970 {
9971     ucontext_t *uc = puc;
9972
9973     switch(signum) {
9974     case SIGFPE:
9975         switch(siginf->si_code) {
9976         case FPE_INTDIV:
9977         case FPE_FLTDIV:
9978             rt_error(uc, "division by zero");
9979             break;
9980         default:
9981             rt_error(uc, "floating point exception");
9982             break;
9983         }
9984         break;
9985     case SIGBUS:
9986     case SIGSEGV:
9987         if (rt_bound_error_msg && *rt_bound_error_msg)
9988             rt_error(uc, *rt_bound_error_msg);
9989         else
9990             rt_error(uc, "dereferencing invalid pointer");
9991         break;
9992     case SIGILL:
9993         rt_error(uc, "illegal instruction");
9994         break;
9995     case SIGABRT:
9996         rt_error(uc, "abort() called");
9997         break;
9998     default:
9999         rt_error(uc, "caught signal %d", signum);
10000         break;
10001     }
10002     exit(255);
10003 }
10004 #endif
10005
10006 /* do all relocations (needed before using tcc_get_symbol()) */
10007 int tcc_relocate(TCCState *s1)
10008 {
10009     Section *s;
10010     int i;
10011
10012     s1->nb_errors = 0;
10013
10014 #ifdef TCC_TARGET_PE
10015     pe_add_runtime(s1);
10016 #else
10017     tcc_add_runtime(s1);
10018 #endif
10019
10020     relocate_common_syms();
10021
10022     tcc_add_linker_symbols(s1);
10023 #ifndef TCC_TARGET_PE
10024     build_got_entries(s1);
10025 #endif
10026     /* compute relocation address : section are relocated in place. We
10027        also alloc the bss space */
10028     for(i = 1; i < s1->nb_sections; i++) {
10029         s = s1->sections[i];
10030         if (s->sh_flags & SHF_ALLOC) {
10031             if (s->sh_type == SHT_NOBITS)
10032                 s->data = tcc_mallocz(s->data_offset);
10033             s->sh_addr = (unsigned long)s->data;
10034         }
10035     }
10036
10037     relocate_syms(s1, 1);
10038
10039     if (s1->nb_errors != 0)
10040         return -1;
10041
10042     /* relocate each section */
10043     for(i = 1; i < s1->nb_sections; i++) {
10044         s = s1->sections[i];
10045         if (s->reloc)
10046             relocate_section(s1, s);
10047     }
10048
10049     /* mark executable sections as executable in memory */
10050     for(i = 1; i < s1->nb_sections; i++) {
10051         s = s1->sections[i];
10052         if ((s->sh_flags & (SHF_ALLOC | SHF_EXECINSTR)) ==
10053             (SHF_ALLOC | SHF_EXECINSTR))
10054             set_pages_executable(s->data, s->data_offset);
10055     }
10056     return 0;
10057 }
10058
10059 /* launch the compiled program with the given arguments */
10060 int tcc_run(TCCState *s1, int argc, char **argv)
10061 {
10062     int (*prog_main)(int, char **);
10063
10064     if (tcc_relocate(s1) < 0)
10065         return -1;
10066
10067     prog_main = tcc_get_symbol_err(s1, "main");
10068
10069     if (do_debug) {
10070 #if defined(_WIN32) || defined(CONFIG_TCCBOOT)
10071         error("debug mode currently not available for Windows");
10072 #else
10073         struct sigaction sigact;
10074         /* install TCC signal handlers to print debug info on fatal
10075            runtime errors */
10076         sigact.sa_flags = SA_SIGINFO | SA_RESETHAND;
10077         sigact.sa_sigaction = sig_error;
10078         sigemptyset(&sigact.sa_mask);
10079         sigaction(SIGFPE, &sigact, NULL);
10080         sigaction(SIGILL, &sigact, NULL);
10081         sigaction(SIGSEGV, &sigact, NULL);
10082         sigaction(SIGBUS, &sigact, NULL);
10083         sigaction(SIGABRT, &sigact, NULL);
10084 #endif
10085     }
10086
10087 #ifdef CONFIG_TCC_BCHECK
10088     if (do_bounds_check) {
10089         void (*bound_init)(void);
10090
10091         /* set error function */
10092         rt_bound_error_msg = (void *)tcc_get_symbol_err(s1,
10093                                                         "__bound_error_msg");
10094
10095         /* XXX: use .init section so that it also work in binary ? */
10096         bound_init = (void *)tcc_get_symbol_err(s1, "__bound_init");
10097         bound_init();
10098     }
10099 #endif
10100     return (*prog_main)(argc, argv);
10101 }
10102
10103 void tcc_memstats(void)
10104 {
10105 #ifdef MEM_DEBUG
10106     printf("memory in use: %d\n", mem_cur_size);
10107 #endif
10108 }
10109
10110 static void tcc_cleanup(void)
10111 {
10112     int i, n;
10113
10114     if (NULL == tcc_state)
10115         return;
10116     tcc_state = NULL;
10117
10118     /* free -D defines */
10119     free_defines(NULL);
10120
10121     /* free tokens */
10122     n = tok_ident - TOK_IDENT;
10123     for(i = 0; i < n; i++)
10124         tcc_free(table_ident[i]);
10125     tcc_free(table_ident);
10126
10127     /* free sym_pools */
10128     dynarray_reset(&sym_pools, &nb_sym_pools);
10129     /* string buffer */
10130     cstr_free(&tokcstr);
10131     /* reset symbol stack */
10132     sym_free_first = NULL;
10133     /* cleanup from error/setjmp */
10134     macro_ptr = NULL;
10135 }
10136
10137 TCCState *tcc_new(void)
10138 {
10139     const char *p, *r;
10140     TCCState *s;
10141     TokenSym *ts;
10142     int i, c;
10143
10144     tcc_cleanup();
10145
10146     s = tcc_mallocz(sizeof(TCCState));
10147     if (!s)
10148         return NULL;
10149     tcc_state = s;
10150     s->output_type = TCC_OUTPUT_MEMORY;
10151
10152     /* init isid table */
10153     for(i=CH_EOF;i<256;i++)
10154         isidnum_table[i-CH_EOF] = isid(i) || isnum(i);
10155
10156     /* add all tokens */
10157     table_ident = NULL;
10158     memset(hash_ident, 0, TOK_HASH_SIZE * sizeof(TokenSym *));
10159
10160     tok_ident = TOK_IDENT;
10161     p = tcc_keywords;
10162     while (*p) {
10163         r = p;
10164         for(;;) {
10165             c = *r++;
10166             if (c == '\0')
10167                 break;
10168         }
10169         ts = tok_alloc(p, r - p - 1);
10170         p = r;
10171     }
10172
10173     /* we add dummy defines for some special macros to speed up tests
10174        and to have working defined() */
10175     define_push(TOK___LINE__, MACRO_OBJ, NULL, NULL);
10176     define_push(TOK___FILE__, MACRO_OBJ, NULL, NULL);
10177     define_push(TOK___DATE__, MACRO_OBJ, NULL, NULL);
10178     define_push(TOK___TIME__, MACRO_OBJ, NULL, NULL);
10179
10180     /* standard defines */
10181     tcc_define_symbol(s, "__STDC__", NULL);
10182     tcc_define_symbol(s, "__STDC_VERSION__", "199901L");
10183 #if defined(TCC_TARGET_I386)
10184     tcc_define_symbol(s, "__i386__", NULL);
10185 #endif
10186 #if defined(TCC_TARGET_ARM)
10187     tcc_define_symbol(s, "__ARM_ARCH_4__", NULL);
10188     tcc_define_symbol(s, "__arm_elf__", NULL);
10189     tcc_define_symbol(s, "__arm_elf", NULL);
10190     tcc_define_symbol(s, "arm_elf", NULL);
10191     tcc_define_symbol(s, "__arm__", NULL);
10192     tcc_define_symbol(s, "__arm", NULL);
10193     tcc_define_symbol(s, "arm", NULL);
10194     tcc_define_symbol(s, "__APCS_32__", NULL);
10195 #endif
10196 #ifdef TCC_TARGET_PE
10197     tcc_define_symbol(s, "_WIN32", NULL);
10198 #else
10199     tcc_define_symbol(s, "__unix__", NULL);
10200     tcc_define_symbol(s, "__unix", NULL);
10201 #if defined(__linux)
10202     tcc_define_symbol(s, "__linux__", NULL);
10203     tcc_define_symbol(s, "__linux", NULL);
10204 #endif
10205 #endif
10206     /* tiny C specific defines */
10207     tcc_define_symbol(s, "__TINYC__", NULL);
10208
10209     /* tiny C & gcc defines */
10210     tcc_define_symbol(s, "__SIZE_TYPE__", "unsigned int");
10211     tcc_define_symbol(s, "__PTRDIFF_TYPE__", "int");
10212 #ifdef TCC_TARGET_PE
10213     tcc_define_symbol(s, "__WCHAR_TYPE__", "unsigned short");
10214 #else
10215     tcc_define_symbol(s, "__WCHAR_TYPE__", "int");
10216 #endif
10217
10218 #ifndef TCC_TARGET_PE
10219     /* default library paths */
10220     tcc_add_library_path(s, CONFIG_SYSROOT "/usr/local/lib");
10221     tcc_add_library_path(s, CONFIG_SYSROOT "/usr/lib");
10222     tcc_add_library_path(s, CONFIG_SYSROOT "/lib");
10223 #endif
10224
10225     /* no section zero */
10226     dynarray_add((void ***)&s->sections, &s->nb_sections, NULL);
10227
10228     /* create standard sections */
10229     text_section = new_section(s, ".text", SHT_PROGBITS, SHF_ALLOC | SHF_EXECINSTR);
10230     data_section = new_section(s, ".data", SHT_PROGBITS, SHF_ALLOC | SHF_WRITE);
10231     bss_section = new_section(s, ".bss", SHT_NOBITS, SHF_ALLOC | SHF_WRITE);
10232
10233     /* symbols are always generated for linking stage */
10234     symtab_section = new_symtab(s, ".symtab", SHT_SYMTAB, 0,
10235                                 ".strtab",
10236                                 ".hashtab", SHF_PRIVATE);
10237     strtab_section = symtab_section->link;
10238
10239     /* private symbol table for dynamic symbols */
10240     s->dynsymtab_section = new_symtab(s, ".dynsymtab", SHT_SYMTAB, SHF_PRIVATE,
10241                                       ".dynstrtab",
10242                                       ".dynhashtab", SHF_PRIVATE);
10243     s->alacarte_link = 1;
10244
10245 #ifdef CHAR_IS_UNSIGNED
10246     s->char_is_unsigned = 1;
10247 #endif
10248 #if defined(TCC_TARGET_PE) && 0
10249     /* XXX: currently the PE linker is not ready to support that */
10250     s->leading_underscore = 1;
10251 #endif
10252     return s;
10253 }
10254
10255 void tcc_delete(TCCState *s1)
10256 {
10257     int i;
10258
10259     tcc_cleanup();
10260
10261     /* free all sections */
10262     free_section(s1->dynsymtab_section);
10263
10264     for(i = 1; i < s1->nb_sections; i++)
10265         free_section(s1->sections[i]);
10266     tcc_free(s1->sections);
10267
10268     /* free any loaded DLLs */
10269     for ( i = 0; i < s1->nb_loaded_dlls; i++)
10270     {
10271         DLLReference *ref = s1->loaded_dlls[i];
10272         if ( ref->handle )
10273             dlclose(ref->handle);
10274     }
10275
10276     /* free loaded dlls array */
10277     dynarray_reset(&s1->loaded_dlls, &s1->nb_loaded_dlls);
10278
10279     /* free library paths */
10280     dynarray_reset(&s1->library_paths, &s1->nb_library_paths);
10281
10282     /* free include paths */
10283     dynarray_reset(&s1->cached_includes, &s1->nb_cached_includes);
10284     dynarray_reset(&s1->include_paths, &s1->nb_include_paths);
10285     dynarray_reset(&s1->sysinclude_paths, &s1->nb_sysinclude_paths);
10286
10287     tcc_free(s1);
10288 }
10289
10290 int tcc_add_include_path(TCCState *s1, const char *pathname)
10291 {
10292     char *pathname1;
10293
10294     pathname1 = tcc_strdup(pathname);
10295     dynarray_add((void ***)&s1->include_paths, &s1->nb_include_paths, pathname1);
10296     return 0;
10297 }
10298
10299 int tcc_add_sysinclude_path(TCCState *s1, const char *pathname)
10300 {
10301     char *pathname1;
10302
10303     pathname1 = tcc_strdup(pathname);
10304     dynarray_add((void ***)&s1->sysinclude_paths, &s1->nb_sysinclude_paths, pathname1);
10305     return 0;
10306 }
10307
10308 static int tcc_add_file_internal(TCCState *s1, const char *filename, int flags)
10309 {
10310     const char *ext;
10311     Elf32_Ehdr ehdr;
10312     int fd, ret;
10313     BufferedFile *saved_file;
10314
10315     /* find source file type with extension */
10316     ext = tcc_fileextension(filename);
10317     if (ext[0])
10318         ext++;
10319
10320     /* open the file */
10321     saved_file = file;
10322     file = tcc_open(s1, filename);
10323     if (!file) {
10324         if (flags & AFF_PRINT_ERROR) {
10325             error_noabort("file '%s' not found", filename);
10326         }
10327         ret = -1;
10328         goto fail1;
10329     }
10330
10331     if (flags & AFF_PREPROCESS) {
10332         ret = tcc_preprocess(s1);
10333     } else if (!ext[0] || !strcmp(ext, "c")) {
10334         /* C file assumed */
10335         ret = tcc_compile(s1);
10336     } else
10337 #ifdef CONFIG_TCC_ASM
10338     if (!strcmp(ext, "S")) {
10339         /* preprocessed assembler */
10340         ret = tcc_assemble(s1, 1);
10341     } else if (!strcmp(ext, "s")) {
10342         /* non preprocessed assembler */
10343         ret = tcc_assemble(s1, 0);
10344     } else
10345 #endif
10346 #ifdef TCC_TARGET_PE
10347     if (!strcmp(ext, "def")) {
10348         ret = pe_load_def_file(s1, file->fd);
10349     } else
10350 #endif
10351     {
10352         fd = file->fd;
10353         /* assume executable format: auto guess file type */
10354         ret = read(fd, &ehdr, sizeof(ehdr));
10355         lseek(fd, 0, SEEK_SET);
10356         if (ret <= 0) {
10357             error_noabort("could not read header");
10358             goto fail;
10359         } else if (ret != sizeof(ehdr)) {
10360             goto try_load_script;
10361         }
10362
10363         if (ehdr.e_ident[0] == ELFMAG0 &&
10364             ehdr.e_ident[1] == ELFMAG1 &&
10365             ehdr.e_ident[2] == ELFMAG2 &&
10366             ehdr.e_ident[3] == ELFMAG3) {
10367             file->line_num = 0; /* do not display line number if error */
10368             if (ehdr.e_type == ET_REL) {
10369                 ret = tcc_load_object_file(s1, fd, 0);
10370             } else if (ehdr.e_type == ET_DYN) {
10371                 if (s1->output_type == TCC_OUTPUT_MEMORY) {
10372 #ifdef TCC_TARGET_PE
10373                     ret = -1;
10374 #else
10375                     void *h;
10376                     h = dlopen(filename, RTLD_GLOBAL | RTLD_LAZY);
10377                     if (h)
10378                         ret = 0;
10379                     else
10380                         ret = -1;
10381 #endif
10382                 } else {
10383                     ret = tcc_load_dll(s1, fd, filename,
10384                                        (flags & AFF_REFERENCED_DLL) != 0);
10385                 }
10386             } else {
10387                 error_noabort("unrecognized ELF file");
10388                 goto fail;
10389             }
10390         } else if (memcmp((char *)&ehdr, ARMAG, 8) == 0) {
10391             file->line_num = 0; /* do not display line number if error */
10392             ret = tcc_load_archive(s1, fd);
10393         } else
10394 #ifdef TCC_TARGET_COFF
10395         if (*(uint16_t *)(&ehdr) == COFF_C67_MAGIC) {
10396             ret = tcc_load_coff(s1, fd);
10397         } else
10398 #endif
10399 #ifdef TCC_TARGET_PE
10400         if (pe_test_res_file(&ehdr, ret)) {
10401             ret = pe_load_res_file(s1, fd);
10402         } else
10403 #endif
10404         {
10405             /* as GNU ld, consider it is an ld script if not recognized */
10406         try_load_script:
10407             ret = tcc_load_ldscript(s1);
10408             if (ret < 0) {
10409                 error_noabort("unrecognized file type");
10410                 goto fail;
10411             }
10412         }
10413     }
10414  the_end:
10415     tcc_close(file);
10416  fail1:
10417     file = saved_file;
10418     return ret;
10419  fail:
10420     ret = -1;
10421     goto the_end;
10422 }
10423
10424 int tcc_add_file(TCCState *s, const char *filename)
10425 {
10426     return tcc_add_file_internal(s, filename, AFF_PRINT_ERROR);
10427 }
10428
10429 int tcc_add_library_path(TCCState *s, const char *pathname)
10430 {
10431     char *pathname1;
10432
10433     pathname1 = tcc_strdup(pathname);
10434     dynarray_add((void ***)&s->library_paths, &s->nb_library_paths, pathname1);
10435     return 0;
10436 }
10437
10438 /* find and load a dll. Return non zero if not found */
10439 /* XXX: add '-rpath' option support ? */
10440 static int tcc_add_dll(TCCState *s, const char *filename, int flags)
10441 {
10442     char buf[1024];
10443     int i;
10444
10445     for(i = 0; i < s->nb_library_paths; i++) {
10446         snprintf(buf, sizeof(buf), "%s/%s",
10447                  s->library_paths[i], filename);
10448         if (tcc_add_file_internal(s, buf, flags) == 0)
10449             return 0;
10450     }
10451     return -1;
10452 }
10453
10454 /* the library name is the same as the argument of the '-l' option */
10455 int tcc_add_library(TCCState *s, const char *libraryname)
10456 {
10457     char buf[1024];
10458     int i;
10459
10460     /* first we look for the dynamic library if not static linking */
10461     if (!s->static_link) {
10462 #ifdef TCC_TARGET_PE
10463         snprintf(buf, sizeof(buf), "%s.def", libraryname);
10464 #else
10465         snprintf(buf, sizeof(buf), "lib%s.so", libraryname);
10466 #endif
10467         if (tcc_add_dll(s, buf, 0) == 0)
10468             return 0;
10469     }
10470
10471     /* then we look for the static library */
10472     for(i = 0; i < s->nb_library_paths; i++) {
10473         snprintf(buf, sizeof(buf), "%s/lib%s.a",
10474                  s->library_paths[i], libraryname);
10475         if (tcc_add_file_internal(s, buf, 0) == 0)
10476             return 0;
10477     }
10478     return -1;
10479 }
10480
10481 int tcc_add_symbol(TCCState *s, const char *name, unsigned long val)
10482 {
10483     add_elf_sym(symtab_section, val, 0,
10484                 ELF32_ST_INFO(STB_GLOBAL, STT_NOTYPE), 0,
10485                 SHN_ABS, name);
10486     return 0;
10487 }
10488
10489 int tcc_set_output_type(TCCState *s, int output_type)
10490 {
10491     char buf[1024];
10492
10493     s->output_type = output_type;
10494
10495     if (!s->nostdinc) {
10496         /* default include paths */
10497         /* XXX: reverse order needed if -isystem support */
10498 #ifndef TCC_TARGET_PE
10499         tcc_add_sysinclude_path(s, CONFIG_SYSROOT "/usr/local/include");
10500         tcc_add_sysinclude_path(s, CONFIG_SYSROOT "/usr/include");
10501 #endif
10502         snprintf(buf, sizeof(buf), "%s/include", tcc_lib_path);
10503         tcc_add_sysinclude_path(s, buf);
10504 #ifdef TCC_TARGET_PE
10505         snprintf(buf, sizeof(buf), "%s/include/winapi", tcc_lib_path);
10506         tcc_add_sysinclude_path(s, buf);
10507 #endif
10508     }
10509
10510     /* if bound checking, then add corresponding sections */
10511 #ifdef CONFIG_TCC_BCHECK
10512     if (do_bounds_check) {
10513         /* define symbol */
10514         tcc_define_symbol(s, "__BOUNDS_CHECKING_ON", NULL);
10515         /* create bounds sections */
10516         bounds_section = new_section(s, ".bounds",
10517                                      SHT_PROGBITS, SHF_ALLOC);
10518         lbounds_section = new_section(s, ".lbounds",
10519                                       SHT_PROGBITS, SHF_ALLOC);
10520     }
10521 #endif
10522
10523     if (s->char_is_unsigned) {
10524         tcc_define_symbol(s, "__CHAR_UNSIGNED__", NULL);
10525     }
10526
10527     /* add debug sections */
10528     if (do_debug) {
10529         /* stab symbols */
10530         stab_section = new_section(s, ".stab", SHT_PROGBITS, 0);
10531         stab_section->sh_entsize = sizeof(Stab_Sym);
10532         stabstr_section = new_section(s, ".stabstr", SHT_STRTAB, 0);
10533         put_elf_str(stabstr_section, "");
10534         stab_section->link = stabstr_section;
10535         /* put first entry */
10536         put_stabs("", 0, 0, 0, 0);
10537     }
10538
10539     /* add libc crt1/crti objects */
10540 #ifndef TCC_TARGET_PE
10541     if ((output_type == TCC_OUTPUT_EXE || output_type == TCC_OUTPUT_DLL) &&
10542         !s->nostdlib) {
10543         if (output_type != TCC_OUTPUT_DLL)
10544             tcc_add_file(s, CONFIG_TCC_CRT_PREFIX "/crt1.o");
10545         tcc_add_file(s, CONFIG_TCC_CRT_PREFIX "/crti.o");
10546     }
10547 #endif
10548
10549 #ifdef TCC_TARGET_PE
10550     snprintf(buf, sizeof(buf), "%s/lib", tcc_lib_path);
10551     tcc_add_library_path(s, buf);
10552 #endif
10553
10554     return 0;
10555 }
10556
10557 #define WD_ALL    0x0001 /* warning is activated when using -Wall */
10558 #define FD_INVERT 0x0002 /* invert value before storing */
10559
10560 typedef struct FlagDef {
10561     uint16_t offset;
10562     uint16_t flags;
10563     const char *name;
10564 } FlagDef;
10565
10566 static const FlagDef warning_defs[] = {
10567     { offsetof(TCCState, warn_unsupported), 0, "unsupported" },
10568     { offsetof(TCCState, warn_write_strings), 0, "write-strings" },
10569     { offsetof(TCCState, warn_error), 0, "error" },
10570     { offsetof(TCCState, warn_implicit_function_declaration), WD_ALL,
10571       "implicit-function-declaration" },
10572 };
10573
10574 static int set_flag(TCCState *s, const FlagDef *flags, int nb_flags,
10575                     const char *name, int value)
10576 {
10577     int i;
10578     const FlagDef *p;
10579     const char *r;
10580
10581     r = name;
10582     if (r[0] == 'n' && r[1] == 'o' && r[2] == '-') {
10583         r += 3;
10584         value = !value;
10585     }
10586     for(i = 0, p = flags; i < nb_flags; i++, p++) {
10587         if (!strcmp(r, p->name))
10588             goto found;
10589     }
10590     return -1;
10591  found:
10592     if (p->flags & FD_INVERT)
10593         value = !value;
10594     *(int *)((uint8_t *)s + p->offset) = value;
10595     return 0;
10596 }
10597
10598
10599 /* set/reset a warning */
10600 int tcc_set_warning(TCCState *s, const char *warning_name, int value)
10601 {
10602     int i;
10603     const FlagDef *p;
10604
10605     if (!strcmp(warning_name, "all")) {
10606         for(i = 0, p = warning_defs; i < countof(warning_defs); i++, p++) {
10607             if (p->flags & WD_ALL)
10608                 *(int *)((uint8_t *)s + p->offset) = 1;
10609         }
10610         return 0;
10611     } else {
10612         return set_flag(s, warning_defs, countof(warning_defs),
10613                         warning_name, value);
10614     }
10615 }
10616
10617 static const FlagDef flag_defs[] = {
10618     { offsetof(TCCState, char_is_unsigned), 0, "unsigned-char" },
10619     { offsetof(TCCState, char_is_unsigned), FD_INVERT, "signed-char" },
10620     { offsetof(TCCState, nocommon), FD_INVERT, "common" },
10621     { offsetof(TCCState, leading_underscore), 0, "leading-underscore" },
10622 };
10623
10624 /* set/reset a flag */
10625 int tcc_set_flag(TCCState *s, const char *flag_name, int value)
10626 {
10627     return set_flag(s, flag_defs, countof(flag_defs),
10628                     flag_name, value);
10629 }
10630
10631 #if !defined(LIBTCC)
10632
10633 static int64_t getclock_us(void)
10634 {
10635 #ifdef _WIN32
10636     struct _timeb tb;
10637     _ftime(&tb);
10638     return (tb.time * 1000LL + tb.millitm) * 1000LL;
10639 #else
10640     struct timeval tv;
10641     gettimeofday(&tv, NULL);
10642     return tv.tv_sec * 1000000LL + tv.tv_usec;
10643 #endif
10644 }
10645
10646 void help(void)
10647 {
10648     printf("tcc version " TCC_VERSION " - Tiny C Compiler - Copyright (C) 2001-2006 Fabrice Bellard\n"
10649            "usage: tcc [-v] [-c] [-o outfile] [-Bdir] [-bench] [-Idir] [-Dsym[=val]] [-Usym]\n"
10650            "           [-Wwarn] [-g] [-b] [-bt N] [-Ldir] [-llib] [-shared] [-soname name]\n"
10651            "           [-static] [infile1 infile2...] [-run infile args...]\n"
10652            "\n"
10653            "General options:\n"
10654            "  -v          display current version, increase verbosity\n"
10655            "  -c          compile only - generate an object file\n"
10656            "  -o outfile  set output filename\n"
10657            "  -Bdir       set tcc internal library path\n"
10658            "  -bench      output compilation statistics\n"
10659            "  -run        run compiled source\n"
10660            "  -fflag      set or reset (with 'no-' prefix) 'flag' (see man page)\n"
10661            "  -Wwarning   set or reset (with 'no-' prefix) 'warning' (see man page)\n"
10662            "  -w          disable all warnings\n"
10663            "Preprocessor options:\n"
10664            "  -E          preprocess only\n"
10665            "  -Idir       add include path 'dir'\n"
10666            "  -Dsym[=val] define 'sym' with value 'val'\n"
10667            "  -Usym       undefine 'sym'\n"
10668            "Linker options:\n"
10669            "  -Ldir       add library path 'dir'\n"
10670            "  -llib       link with dynamic or static library 'lib'\n"
10671            "  -shared     generate a shared library\n"
10672            "  -soname     set name for shared library to be used at runtime\n"
10673            "  -static     static linking\n"
10674            "  -rdynamic   export all global symbols to dynamic linker\n"
10675            "  -r          generate (relocatable) object file\n"
10676            "Debugger options:\n"
10677            "  -g          generate runtime debug info\n"
10678 #ifdef CONFIG_TCC_BCHECK
10679            "  -b          compile with built-in memory and bounds checker (implies -g)\n"
10680 #endif
10681            "  -bt N       show N callers in stack traces\n"
10682            );
10683 }
10684
10685 #define TCC_OPTION_HAS_ARG 0x0001
10686 #define TCC_OPTION_NOSEP   0x0002 /* cannot have space before option and arg */
10687
10688 typedef struct TCCOption {
10689     const char *name;
10690     uint16_t index;
10691     uint16_t flags;
10692 } TCCOption;
10693
10694 enum {
10695     TCC_OPTION_HELP,
10696     TCC_OPTION_I,
10697     TCC_OPTION_D,
10698     TCC_OPTION_U,
10699     TCC_OPTION_L,
10700     TCC_OPTION_B,
10701     TCC_OPTION_l,
10702     TCC_OPTION_bench,
10703     TCC_OPTION_bt,
10704     TCC_OPTION_b,
10705     TCC_OPTION_g,
10706     TCC_OPTION_c,
10707     TCC_OPTION_static,
10708     TCC_OPTION_shared,
10709     TCC_OPTION_soname,
10710     TCC_OPTION_o,
10711     TCC_OPTION_r,
10712     TCC_OPTION_Wl,
10713     TCC_OPTION_W,
10714     TCC_OPTION_O,
10715     TCC_OPTION_m,
10716     TCC_OPTION_f,
10717     TCC_OPTION_nostdinc,
10718     TCC_OPTION_nostdlib,
10719     TCC_OPTION_print_search_dirs,
10720     TCC_OPTION_rdynamic,
10721     TCC_OPTION_run,
10722     TCC_OPTION_v,
10723     TCC_OPTION_w,
10724     TCC_OPTION_pipe,
10725     TCC_OPTION_E,
10726 };
10727
10728 static const TCCOption tcc_options[] = {
10729     { "h", TCC_OPTION_HELP, 0 },
10730     { "?", TCC_OPTION_HELP, 0 },
10731     { "I", TCC_OPTION_I, TCC_OPTION_HAS_ARG },
10732     { "D", TCC_OPTION_D, TCC_OPTION_HAS_ARG },
10733     { "U", TCC_OPTION_U, TCC_OPTION_HAS_ARG },
10734     { "L", TCC_OPTION_L, TCC_OPTION_HAS_ARG },
10735     { "B", TCC_OPTION_B, TCC_OPTION_HAS_ARG },
10736     { "l", TCC_OPTION_l, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10737     { "bench", TCC_OPTION_bench, 0 },
10738     { "bt", TCC_OPTION_bt, TCC_OPTION_HAS_ARG },
10739 #ifdef CONFIG_TCC_BCHECK
10740     { "b", TCC_OPTION_b, 0 },
10741 #endif
10742     { "g", TCC_OPTION_g, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10743     { "c", TCC_OPTION_c, 0 },
10744     { "static", TCC_OPTION_static, 0 },
10745     { "shared", TCC_OPTION_shared, 0 },
10746     { "soname", TCC_OPTION_soname, TCC_OPTION_HAS_ARG },
10747     { "o", TCC_OPTION_o, TCC_OPTION_HAS_ARG },
10748     { "run", TCC_OPTION_run, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10749     { "rdynamic", TCC_OPTION_rdynamic, 0 },
10750     { "r", TCC_OPTION_r, 0 },
10751     { "Wl,", TCC_OPTION_Wl, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10752     { "W", TCC_OPTION_W, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10753     { "O", TCC_OPTION_O, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10754     { "m", TCC_OPTION_m, TCC_OPTION_HAS_ARG },
10755     { "f", TCC_OPTION_f, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10756     { "nostdinc", TCC_OPTION_nostdinc, 0 },
10757     { "nostdlib", TCC_OPTION_nostdlib, 0 },
10758     { "print-search-dirs", TCC_OPTION_print_search_dirs, 0 },
10759     { "v", TCC_OPTION_v, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
10760     { "w", TCC_OPTION_w, 0 },
10761     { "pipe", TCC_OPTION_pipe, 0},
10762     { "E", TCC_OPTION_E, 0},
10763     { NULL },
10764 };
10765
10766 /* convert 'str' into an array of space separated strings */
10767 static int expand_args(char ***pargv, const char *str)
10768 {
10769     const char *s1;
10770     char **argv, *arg;
10771     int argc, len;
10772
10773     argc = 0;
10774     argv = NULL;
10775     for(;;) {
10776         while (is_space(*str))
10777             str++;
10778         if (*str == '\0')
10779             break;
10780         s1 = str;
10781         while (*str != '\0' && !is_space(*str))
10782             str++;
10783         len = str - s1;
10784         arg = tcc_malloc(len + 1);
10785         memcpy(arg, s1, len);
10786         arg[len] = '\0';
10787         dynarray_add((void ***)&argv, &argc, arg);
10788     }
10789     *pargv = argv;
10790     return argc;
10791 }
10792
10793 static char **files;
10794 static int nb_files, nb_libraries;
10795 static int multiple_files;
10796 static int print_search_dirs;
10797 static int output_type;
10798 static int reloc_output;
10799 static const char *outfile;
10800
10801 int parse_args(TCCState *s, int argc, char **argv)
10802 {
10803     int optind;
10804     const TCCOption *popt;
10805     const char *optarg, *p1, *r1;
10806     char *r;
10807
10808     optind = 0;
10809     while (optind < argc) {
10810
10811         r = argv[optind++];
10812         if (r[0] != '-' || r[1] == '\0') {
10813             /* add a new file */
10814             dynarray_add((void ***)&files, &nb_files, r);
10815             if (!multiple_files) {
10816                 optind--;
10817                 /* argv[0] will be this file */
10818                 break;
10819             }
10820         } else {
10821             /* find option in table (match only the first chars */
10822             popt = tcc_options;
10823             for(;;) {
10824                 p1 = popt->name;
10825                 if (p1 == NULL)
10826                     error("invalid option -- '%s'", r);
10827                 r1 = r + 1;
10828                 for(;;) {
10829                     if (*p1 == '\0')
10830                         goto option_found;
10831                     if (*r1 != *p1)
10832                         break;
10833                     p1++;
10834                     r1++;
10835                 }
10836                 popt++;
10837             }
10838         option_found:
10839             if (popt->flags & TCC_OPTION_HAS_ARG) {
10840                 if (*r1 != '\0' || (popt->flags & TCC_OPTION_NOSEP)) {
10841                     optarg = r1;
10842                 } else {
10843                     if (optind >= argc)
10844                         error("argument to '%s' is missing", r);
10845                     optarg = argv[optind++];
10846                 }
10847             } else {
10848                 if (*r1 != '\0')
10849                     return 0;
10850                 optarg = NULL;
10851             }
10852
10853             switch(popt->index) {
10854             case TCC_OPTION_HELP:
10855                 return 0;
10856
10857             case TCC_OPTION_I:
10858                 if (tcc_add_include_path(s, optarg) < 0)
10859                     error("too many include paths");
10860                 break;
10861             case TCC_OPTION_D:
10862                 {
10863                     char *sym, *value;
10864                     sym = (char *)optarg;
10865                     value = strchr(sym, '=');
10866                     if (value) {
10867                         *value = '\0';
10868                         value++;
10869                     }
10870                     tcc_define_symbol(s, sym, value);
10871                 }
10872                 break;
10873             case TCC_OPTION_U:
10874                 tcc_undefine_symbol(s, optarg);
10875                 break;
10876             case TCC_OPTION_L:
10877                 tcc_add_library_path(s, optarg);
10878                 break;
10879             case TCC_OPTION_B:
10880                 /* set tcc utilities path (mainly for tcc development) */
10881                 tcc_lib_path = optarg;
10882                 break;
10883             case TCC_OPTION_l:
10884                 dynarray_add((void ***)&files, &nb_files, r);
10885                 nb_libraries++;
10886                 break;
10887             case TCC_OPTION_bench:
10888                 do_bench = 1;
10889                 break;
10890             case TCC_OPTION_bt:
10891                 num_callers = atoi(optarg);
10892                 break;
10893 #ifdef CONFIG_TCC_BCHECK
10894             case TCC_OPTION_b:
10895                 do_bounds_check = 1;
10896                 do_debug = 1;
10897                 break;
10898 #endif
10899             case TCC_OPTION_g:
10900                 do_debug = 1;
10901                 break;
10902             case TCC_OPTION_c:
10903                 multiple_files = 1;
10904                 output_type = TCC_OUTPUT_OBJ;
10905                 break;
10906             case TCC_OPTION_static:
10907                 s->static_link = 1;
10908                 break;
10909             case TCC_OPTION_shared:
10910                 output_type = TCC_OUTPUT_DLL;
10911                 break;
10912             case TCC_OPTION_soname:
10913                 s->soname = optarg;
10914                 break;
10915             case TCC_OPTION_o:
10916                 multiple_files = 1;
10917                 outfile = optarg;
10918                 break;
10919             case TCC_OPTION_r:
10920                 /* generate a .o merging several output files */
10921                 reloc_output = 1;
10922                 output_type = TCC_OUTPUT_OBJ;
10923                 break;
10924             case TCC_OPTION_nostdinc:
10925                 s->nostdinc = 1;
10926                 break;
10927             case TCC_OPTION_nostdlib:
10928                 s->nostdlib = 1;
10929                 break;
10930             case TCC_OPTION_print_search_dirs:
10931                 print_search_dirs = 1;
10932                 break;
10933             case TCC_OPTION_run:
10934                 {
10935                     int argc1;
10936                     char **argv1;
10937                     argc1 = expand_args(&argv1, optarg);
10938                     if (argc1 > 0) {
10939                         parse_args(s, argc1, argv1);
10940                     }
10941                     multiple_files = 0;
10942                     output_type = TCC_OUTPUT_MEMORY;
10943                 }
10944                 break;
10945             case TCC_OPTION_v:
10946                 do {
10947                     if (0 == verbose++)
10948                         printf("tcc version %s\n", TCC_VERSION);
10949                 } while (*optarg++ == 'v');
10950                 break;
10951             case TCC_OPTION_f:
10952                 if (tcc_set_flag(s, optarg, 1) < 0 && s->warn_unsupported)
10953                     goto unsupported_option;
10954                 break;
10955             case TCC_OPTION_W:
10956                 if (tcc_set_warning(s, optarg, 1) < 0 &&
10957                     s->warn_unsupported)
10958                     goto unsupported_option;
10959                 break;
10960             case TCC_OPTION_w:
10961                 s->warn_none = 1;
10962                 break;
10963             case TCC_OPTION_rdynamic:
10964                 s->rdynamic = 1;
10965                 break;
10966             case TCC_OPTION_Wl:
10967                 {
10968                     const char *p;
10969                     if (strstart(optarg, "-Ttext,", &p)) {
10970                         s->text_addr = strtoul(p, NULL, 16);
10971                         s->has_text_addr = 1;
10972                     } else if (strstart(optarg, "--oformat,", &p)) {
10973                         if (strstart(p, "elf32-", NULL)) {
10974                             s->output_format = TCC_OUTPUT_FORMAT_ELF;
10975                         } else if (!strcmp(p, "binary")) {
10976                             s->output_format = TCC_OUTPUT_FORMAT_BINARY;
10977                         } else
10978 #ifdef TCC_TARGET_COFF
10979                         if (!strcmp(p, "coff")) {
10980                             s->output_format = TCC_OUTPUT_FORMAT_COFF;
10981                         } else
10982 #endif
10983                         {
10984                             error("target %s not found", p);
10985                         }
10986                     } else {
10987                         error("unsupported linker option '%s'", optarg);
10988                     }
10989                 }
10990                 break;
10991             case TCC_OPTION_E:
10992                 output_type = TCC_OUTPUT_PREPROCESS;
10993                 break;
10994             default:
10995                 if (s->warn_unsupported) {
10996                 unsupported_option:
10997                     warning("unsupported option '%s'", r);
10998                 }
10999                 break;
11000             }
11001         }
11002     }
11003     return optind + 1;
11004 }
11005
11006 int main(int argc, char **argv)
11007 {
11008     int i;
11009     TCCState *s;
11010     int nb_objfiles, ret, optind;
11011     char objfilename[1024];
11012     int64_t start_time = 0;
11013
11014 #ifdef _WIN32
11015     tcc_lib_path = w32_tcc_lib_path();
11016 #endif
11017
11018     s = tcc_new();
11019     output_type = TCC_OUTPUT_EXE;
11020     outfile = NULL;
11021     multiple_files = 1;
11022     files = NULL;
11023     nb_files = 0;
11024     nb_libraries = 0;
11025     reloc_output = 0;
11026     print_search_dirs = 0;
11027     ret = 0;
11028
11029     optind = parse_args(s, argc - 1, argv + 1);
11030     if (print_search_dirs) {
11031         /* enough for Linux kernel */
11032         printf("install: %s/\n", tcc_lib_path);
11033         return 0;
11034     }
11035     if (optind == 0 || nb_files == 0) {
11036         if (optind && verbose)
11037             return 0;
11038         help();
11039         return 1;
11040     }
11041
11042     nb_objfiles = nb_files - nb_libraries;
11043
11044     /* if outfile provided without other options, we output an
11045        executable */
11046     if (outfile && output_type == TCC_OUTPUT_MEMORY)
11047         output_type = TCC_OUTPUT_EXE;
11048
11049     /* check -c consistency : only single file handled. XXX: checks file type */
11050     if (output_type == TCC_OUTPUT_OBJ && !reloc_output) {
11051         /* accepts only a single input file */
11052         if (nb_objfiles != 1)
11053             error("cannot specify multiple files with -c");
11054         if (nb_libraries != 0)
11055             error("cannot specify libraries with -c");
11056     }
11057
11058
11059     if (output_type == TCC_OUTPUT_PREPROCESS) {
11060         if (!outfile) {
11061             s->outfile = stdout;
11062         } else {
11063             s->outfile = fopen(outfile, "w");
11064             if (!s->outfile)
11065                 error("could not open '%s", outfile);
11066         }
11067     } else if (output_type != TCC_OUTPUT_MEMORY) {
11068         if (!outfile) {
11069             /* compute default outfile name */
11070             char *ext;
11071             const char *name =
11072                 strcmp(files[0], "-") == 0 ? "a" : tcc_basename(files[0]);
11073             pstrcpy(objfilename, sizeof(objfilename), name);
11074             ext = tcc_fileextension(objfilename);
11075 #ifdef TCC_TARGET_PE
11076             if (output_type == TCC_OUTPUT_DLL)
11077                 strcpy(ext, ".dll");
11078             else
11079             if (output_type == TCC_OUTPUT_EXE)
11080                 strcpy(ext, ".exe");
11081             else
11082 #endif
11083             if (output_type == TCC_OUTPUT_OBJ && !reloc_output && *ext)
11084                 strcpy(ext, ".o");
11085             else
11086                 pstrcpy(objfilename, sizeof(objfilename), "a.out");
11087             outfile = objfilename;
11088         }
11089     }
11090
11091     if (do_bench) {
11092         start_time = getclock_us();
11093     }
11094
11095     tcc_set_output_type(s, output_type);
11096
11097     /* compile or add each files or library */
11098     for(i = 0; i < nb_files && ret == 0; i++) {
11099         const char *filename;
11100
11101         filename = files[i];
11102         if (output_type == TCC_OUTPUT_PREPROCESS) {
11103             if (tcc_add_file_internal(s, filename,
11104                     AFF_PRINT_ERROR | AFF_PREPROCESS) < 0)
11105                 ret = 1;
11106         } else if (filename[0] == '-' && filename[1]) {
11107             if (tcc_add_library(s, filename + 2) < 0)
11108                 error("cannot find %s", filename);
11109         } else {
11110             if (1 == verbose)
11111                 printf("-> %s\n", filename);
11112             if (tcc_add_file(s, filename) < 0)
11113                 ret = 1;
11114         }
11115     }
11116
11117     /* free all files */
11118     tcc_free(files);
11119
11120     if (ret)
11121         goto the_end;
11122
11123     if (do_bench) {
11124         double total_time;
11125         total_time = (double)(getclock_us() - start_time) / 1000000.0;
11126         if (total_time < 0.001)
11127             total_time = 0.001;
11128         if (total_bytes < 1)
11129             total_bytes = 1;
11130         printf("%d idents, %d lines, %d bytes, %0.3f s, %d lines/s, %0.1f MB/s\n",
11131                tok_ident - TOK_IDENT, total_lines, total_bytes,
11132                total_time, (int)(total_lines / total_time),
11133                total_bytes / total_time / 1000000.0);
11134     }
11135
11136     if (s->output_type == TCC_OUTPUT_PREPROCESS) {
11137         if (outfile)
11138             fclose(s->outfile);
11139     } else if (s->output_type == TCC_OUTPUT_MEMORY) {
11140         ret = tcc_run(s, argc - optind, argv + optind);
11141     } else
11142         ret = tcc_output_file(s, outfile) ? 1 : 0;
11143  the_end:
11144     /* XXX: cannot do it with bound checking because of the malloc hooks */
11145     if (!do_bounds_check)
11146         tcc_delete(s);
11147
11148 #ifdef MEM_DEBUG
11149     if (do_bench) {
11150         printf("memory: %d bytes, max = %d bytes\n", mem_cur_size, mem_max_size);
11151     }
11152 #endif
11153     return ret;
11154 }
11155
11156 #endif