libtcc.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
  21 #if !defined ONE_SOURCE || ONE_SOURCE
  22 #include "tccpp.c"
  23 #include "tccgen.c"
  24 #include "tccelf.c"
  25 #include "tccrun.c"
  26 #ifdef TCC_TARGET_I386
  27 #include "i386-gen.c"
  28 #include "i386-link.c"
  29 #include "i386-asm.c"
  30 #elif defined(TCC_TARGET_ARM)
  31 #include "arm-gen.c"
  32 #include "arm-link.c"
  33 #include "arm-asm.c"
  34 #elif defined(TCC_TARGET_ARM64)
  35 #include "arm64-gen.c"
  36 #include "arm64-link.c"
  37 #include "arm-asm.c"
  38 #elif defined(TCC_TARGET_C67)
  39 #include "c67-gen.c"
  40 #include "c67-link.c"
  41 #include "tcccoff.c"
  42 #elif defined(TCC_TARGET_X86_64)
  43 #include "x86_64-gen.c"
  44 #include "x86_64-link.c"
  45 #include "i386-asm.c"
  46 #elif defined(TCC_TARGET_RISCV64)
  47 #include "riscv64-gen.c"
  48 #include "riscv64-link.c"
  49 #include "riscv64-asm.c"
  50 #else
  51 #error unknown target
  52 #endif
  53 #ifdef CONFIG_TCC_ASM
  54 #include "tccasm.c"
  55 #endif
  56 #ifdef TCC_TARGET_PE
  57 #include "tccpe.c"
  58 #endif
  59 #endif /* ONE_SOURCE */
  60
  61 #include "tcc.h"
  62
  63 /********************************************************/
  64 /* global variables */
  65
  66 /* XXX: get rid of this ASAP (or maybe not) */
  67 ST_DATA struct TCCState *tcc_state;
  68
  69 #ifdef MEM_DEBUG
  70 static int nb_states;
  71 #endif
  72
  73 /********************************************************/
  74 #ifdef _WIN32
  75 ST_FUNC char *normalize_slashes(char *path)
  76 {
  77     char *p;
  78     for (p = path; *p; ++p)
  79         if (*p == '\\')
  80             *p = '/';
  81     return path;
  82 }
  83
  84 static HMODULE tcc_module;
  85
  86 /* on win32, we suppose the lib and includes are at the location of 'tcc.exe' */
  87 static void tcc_set_lib_path_w32(TCCState *s)
  88 {
  89     char path[1024], *p;
  90     GetModuleFileNameA(tcc_module, path, sizeof path);
  91     p = tcc_basename(normalize_slashes(strlwr(path)));
  92     if (p > path)
  93         --p;
  94     *p = 0;
  95     tcc_set_lib_path(s, path);
  96 }
  97
  98 #ifdef TCC_TARGET_PE
  99 static void tcc_add_systemdir(TCCState *s)
 100 {
 101     char buf[1000];
 102     GetSystemDirectory(buf, sizeof buf);
 103     tcc_add_library_path(s, normalize_slashes(buf));
 104 }
 105 #endif
 106
 107 #ifdef LIBTCC_AS_DLL
 108 BOOL WINAPI DllMain (HINSTANCE hDll, DWORD dwReason, LPVOID lpReserved)
 109 {
 110     if (DLL_PROCESS_ATTACH == dwReason)
 111         tcc_module = hDll;
 112     return TRUE;
 113 }
 114 #endif
 115 #endif
 116
 117 /********************************************************/
 118 #ifndef CONFIG_TCC_SEMLOCK
 119 #define WAIT_SEM()
 120 #define POST_SEM()
 121 #elif defined _WIN32
 122 static int tcc_sem_init;
 123 static CRITICAL_SECTION tcc_cr;
 124 static void wait_sem(void)
 125 {
 126     if (!tcc_sem_init)
 127         InitializeCriticalSection(&tcc_cr), tcc_sem_init = 1;
 128     EnterCriticalSection(&tcc_cr);
 129 }
 130 #define WAIT_SEM() wait_sem()
 131 #define POST_SEM() LeaveCriticalSection(&tcc_cr);
 132 #else
 133 #include <semaphore.h>
 134 static int tcc_sem_init;
 135 static sem_t tcc_sem;
 136 static void wait_sem(void)
 137 {
 138     if (!tcc_sem_init)
 139         sem_init(&tcc_sem, 0, 1), tcc_sem_init = 1;
 140     while (sem_wait (&tcc_sem) < 0 && errno == EINTR);
 141 }
 142 #define WAIT_SEM() wait_sem()
 143 #define POST_SEM() sem_post(&tcc_sem)
 144 #endif
 145
 146 /********************************************************/
 147 /* copy a string and truncate it. */
 148 ST_FUNC char *pstrcpy(char *buf, size_t buf_size, const char *s)
 149 {
 150     char *q, *q_end;
 151     int c;
 152
 153     if (buf_size > 0) {
 154         q = buf;
 155         q_end = buf + buf_size - 1;
 156         while (q < q_end) {
 157             c = *s++;
 158             if (c == '\0')
 159                 break;
 160             *q++ = c;
 161         }
 162         *q = '\0';
 163     }
 164     return buf;
 165 }
 166
 167 /* strcat and truncate. */
 168 ST_FUNC char *pstrcat(char *buf, size_t buf_size, const char *s)
 169 {
 170     size_t len;
 171     len = strlen(buf);
 172     if (len < buf_size)
 173         pstrcpy(buf + len, buf_size - len, s);
 174     return buf;
 175 }
 176
 177 ST_FUNC char *pstrncpy(char *out, const char *in, size_t num)
 178 {
 179     memcpy(out, in, num);
 180     out[num] = '\0';
 181     return out;
 182 }
 183
 184 /* extract the basename of a file */
 185 PUB_FUNC char *tcc_basename(const char *name)
 186 {
 187     char *p = strchr(name, 0);
 188     while (p > name && !IS_DIRSEP(p[-1]))
 189         --p;
 190     return p;
 191 }
 192
 193 /* extract extension part of a file
 194  *
 195  * (if no extension, return pointer to end-of-string)
 196  */
 197 PUB_FUNC char *tcc_fileextension (const char *name)
 198 {
 199     char *b = tcc_basename(name);
 200     char *e = strrchr(b, '.');
 201     return e ? e : strchr(b, 0);
 202 }
 203
 204 /********************************************************/
 205 /* memory management */
 206
 207 #undef free
 208 #undef malloc
 209 #undef realloc
 210
 211 #ifndef MEM_DEBUG
 212
 213 PUB_FUNC void tcc_free(void *ptr)
 214 {
 215     free(ptr);
 216 }
 217
 218 PUB_FUNC void *tcc_malloc(unsigned long size)
 219 {
 220     void *ptr;
 221     ptr = malloc(size);
 222     if (!ptr && size)
 223         _tcc_error("memory full (malloc)");
 224     return ptr;
 225 }
 226
 227 PUB_FUNC void *tcc_mallocz(unsigned long size)
 228 {
 229     void *ptr;
 230     ptr = tcc_malloc(size);
 231     memset(ptr, 0, size);
 232     return ptr;
 233 }
 234
 235 PUB_FUNC void *tcc_realloc(void *ptr, unsigned long size)
 236 {
 237     void *ptr1;
 238     ptr1 = realloc(ptr, size);
 239     if (!ptr1 && size)
 240         _tcc_error("memory full (realloc)");
 241     return ptr1;
 242 }
 243
 244 PUB_FUNC char *tcc_strdup(const char *str)
 245 {
 246     char *ptr;
 247     ptr = tcc_malloc(strlen(str) + 1);
 248     strcpy(ptr, str);
 249     return ptr;
 250 }
 251
 252 #else
 253
 254 #define MEM_DEBUG_MAGIC1 0xFEEDDEB1
 255 #define MEM_DEBUG_MAGIC2 0xFEEDDEB2
 256 #define MEM_DEBUG_MAGIC3 0xFEEDDEB3
 257 #define MEM_DEBUG_FILE_LEN 40
 258 #define MEM_DEBUG_CHECK3(header) \
 259     ((mem_debug_header_t*)((char*)header + header->size))->magic3
 260 #define MEM_USER_PTR(header) \
 261     ((char *)header + offsetof(mem_debug_header_t, magic3))
 262 #define MEM_HEADER_PTR(ptr) \
 263     (mem_debug_header_t *)((char*)ptr - offsetof(mem_debug_header_t, magic3))
 264
 265 struct mem_debug_header {
 266     unsigned magic1;
 267     unsigned size;
 268     struct mem_debug_header *prev;
 269     struct mem_debug_header *next;
 270     int line_num;
 271     char file_name[MEM_DEBUG_FILE_LEN + 1];
 272     unsigned magic2;
 273     ALIGNED(16) unsigned magic3;
 274 };
 275
 276 typedef struct mem_debug_header mem_debug_header_t;
 277
 278 static mem_debug_header_t *mem_debug_chain;
 279 static unsigned mem_cur_size;
 280 static unsigned mem_max_size;
 281
 282 static mem_debug_header_t *malloc_check(void *ptr, const char *msg)
 283 {
 284     mem_debug_header_t * header = MEM_HEADER_PTR(ptr);
 285     if (header->magic1 != MEM_DEBUG_MAGIC1 ||
 286         header->magic2 != MEM_DEBUG_MAGIC2 ||
 287         MEM_DEBUG_CHECK3(header) != MEM_DEBUG_MAGIC3 ||
 288         header->size == (unsigned)-1) {
 289         fprintf(stderr, "%s check failed\n", msg);
 290         if (header->magic1 == MEM_DEBUG_MAGIC1)
 291             fprintf(stderr, "%s:%u: block allocated here.\n",
 292                 header->file_name, header->line_num);
 293         exit(1);
 294     }
 295     return header;
 296 }
 297
 298 PUB_FUNC void *tcc_malloc_debug(unsigned long size, const char *file, int line)
 299 {
 300     int ofs;
 301     mem_debug_header_t *header;
 302
 303     header = malloc(sizeof(mem_debug_header_t) + size);
 304     if (!header)
 305         _tcc_error("memory full (malloc)");
 306
 307     header->magic1 = MEM_DEBUG_MAGIC1;
 308     header->magic2 = MEM_DEBUG_MAGIC2;
 309     header->size = size;
 310     MEM_DEBUG_CHECK3(header) = MEM_DEBUG_MAGIC3;
 311     header->line_num = line;
 312     ofs = strlen(file) - MEM_DEBUG_FILE_LEN;
 313     strncpy(header->file_name, file + (ofs > 0 ? ofs : 0), MEM_DEBUG_FILE_LEN);
 314     header->file_name[MEM_DEBUG_FILE_LEN] = 0;
 315
 316     header->next = mem_debug_chain;
 317     header->prev = NULL;
 318     if (header->next)
 319         header->next->prev = header;
 320     mem_debug_chain = header;
 321
 322     mem_cur_size += size;
 323     if (mem_cur_size > mem_max_size)
 324         mem_max_size = mem_cur_size;
 325
 326     return MEM_USER_PTR(header);
 327 }
 328
 329 PUB_FUNC void tcc_free_debug(void *ptr)
 330 {
 331     mem_debug_header_t *header;
 332     if (!ptr)
 333         return;
 334     header = malloc_check(ptr, "tcc_free");
 335     mem_cur_size -= header->size;
 336     header->size = (unsigned)-1;
 337     if (header->next)
 338         header->next->prev = header->prev;
 339     if (header->prev)
 340         header->prev->next = header->next;
 341     if (header == mem_debug_chain)
 342         mem_debug_chain = header->next;
 343     free(header);
 344 }
 345
 346 PUB_FUNC void *tcc_mallocz_debug(unsigned long size, const char *file, int line)
 347 {
 348     void *ptr;
 349     ptr = tcc_malloc_debug(size,file,line);
 350     memset(ptr, 0, size);
 351     return ptr;
 352 }
 353
 354 PUB_FUNC void *tcc_realloc_debug(void *ptr, unsigned long size, const char *file, int line)
 355 {
 356     mem_debug_header_t *header;
 357     int mem_debug_chain_update = 0;
 358     if (!ptr)
 359         return tcc_malloc_debug(size, file, line);
 360     header = malloc_check(ptr, "tcc_realloc");
 361     mem_cur_size -= header->size;
 362     mem_debug_chain_update = (header == mem_debug_chain);
 363     header = realloc(header, sizeof(mem_debug_header_t) + size);
 364     if (!header)
 365         _tcc_error("memory full (realloc)");
 366     header->size = size;
 367     MEM_DEBUG_CHECK3(header) = MEM_DEBUG_MAGIC3;
 368     if (header->next)
 369         header->next->prev = header;
 370     if (header->prev)
 371         header->prev->next = header;
 372     if (mem_debug_chain_update)
 373         mem_debug_chain = header;
 374     mem_cur_size += size;
 375     if (mem_cur_size > mem_max_size)
 376         mem_max_size = mem_cur_size;
 377     return MEM_USER_PTR(header);
 378 }
 379
 380 PUB_FUNC char *tcc_strdup_debug(const char *str, const char *file, int line)
 381 {
 382     char *ptr;
 383     ptr = tcc_malloc_debug(strlen(str) + 1, file, line);
 384     strcpy(ptr, str);
 385     return ptr;
 386 }
 387
 388 PUB_FUNC void tcc_memcheck(void)
 389 {
 390     if (mem_cur_size) {
 391         mem_debug_header_t *header = mem_debug_chain;
 392         fprintf(stderr, "MEM_DEBUG: mem_leak= %d bytes, mem_max_size= %d bytes\n",
 393             mem_cur_size, mem_max_size);
 394         while (header) {
 395             fprintf(stderr, "%s:%u: error: %u bytes leaked\n",
 396                 header->file_name, header->line_num, header->size);
 397             header = header->next;
 398         }
 399 #if MEM_DEBUG-0 == 2
 400         exit(2);
 401 #endif
 402     }
 403 }
 404 #endif /* MEM_DEBUG */
 405
 406 #define free(p) use_tcc_free(p)
 407 #define malloc(s) use_tcc_malloc(s)
 408 #define realloc(p, s) use_tcc_realloc(p, s)
 409
 410 /********************************************************/
 411 /* dynarrays */
 412
 413 ST_FUNC void dynarray_add(void *ptab, int *nb_ptr, void *data)
 414 {
 415     int nb, nb_alloc;
 416     void **pp;
 417
 418     nb = *nb_ptr;
 419     pp = *(void ***)ptab;
 420     /* every power of two we double array size */
 421     if ((nb & (nb - 1)) == 0) {
 422         if (!nb)
 423             nb_alloc = 1;
 424         else
 425             nb_alloc = nb * 2;
 426         pp = tcc_realloc(pp, nb_alloc * sizeof(void *));
 427         *(void***)ptab = pp;
 428     }
 429     pp[nb++] = data;
 430     *nb_ptr = nb;
 431 }
 432
 433 ST_FUNC void dynarray_reset(void *pp, int *n)
 434 {
 435     void **p;
 436     for (p = *(void***)pp; *n; ++p, --*n)
 437         if (*p)
 438             tcc_free(*p);
 439     tcc_free(*(void**)pp);
 440     *(void**)pp = NULL;
 441 }
 442
 443 static void tcc_split_path(TCCState *s, void *p_ary, int *p_nb_ary, const char *in)
 444 {
 445     const char *p;
 446     do {
 447         int c;
 448         CString str;
 449
 450         cstr_new(&str);
 451         for (p = in; c = *p, c != '\0' && c != PATHSEP[0]; ++p) {
 452             if (c == '{' && p[1] && p[2] == '}') {
 453                 c = p[1], p += 2;
 454                 if (c == 'B')
 455                     cstr_cat(&str, s->tcc_lib_path, -1);
 456                 if (c == 'f' && file) {
 457                     /* substitute current file's dir */
 458                     const char *f = file->true_filename;
 459                     const char *b = tcc_basename(f);
 460                     if (b > f)
 461                         cstr_cat(&str, f, b - f - 1);
 462                     else
 463                         cstr_cat(&str, ".", 1);
 464                 }
 465             } else {
 466                 cstr_ccat(&str, c);
 467             }
 468         }
 469         if (str.size) {
 470             cstr_ccat(&str, '\0');
 471             dynarray_add(p_ary, p_nb_ary, tcc_strdup(str.data));
 472         }
 473         cstr_free(&str);
 474         in = p+1;
 475     } while (*p);
 476 }
 477
 478 /********************************************************/
 479
 480 static void strcat_vprintf(char *buf, int buf_size, const char *fmt, va_list ap)
 481 {
 482     int len;
 483     len = strlen(buf);
 484     vsnprintf(buf + len, buf_size - len, fmt, ap);
 485 }
 486
 487 static void strcat_printf(char *buf, int buf_size, const char *fmt, ...)
 488 {
 489     va_list ap;
 490     va_start(ap, fmt);
 491     strcat_vprintf(buf, buf_size, fmt, ap);
 492     va_end(ap);
 493 }
 494
 495 #define ERROR_WARN 0
 496 #define ERROR_NOABORT 1
 497 #define ERROR_ERROR 2
 498
 499 PUB_FUNC void tcc_enter_state(TCCState *s1)
 500 {
 501     WAIT_SEM();
 502     tcc_state = s1;
 503 }
 504
 505 PUB_FUNC void tcc_exit_state(void)
 506 {
 507     tcc_state = NULL;
 508     POST_SEM();
 509 }
 510
 511 static void error1(int mode, const char *fmt, va_list ap)
 512 {
 513     char buf[2048];
 514     BufferedFile **pf, *f;
 515     TCCState *s1 = tcc_state;
 516
 517     buf[0] = '\0';
 518     if (s1 == NULL)
 519         /* can happen only if called from tcc_malloc(): 'out of memory' */
 520         goto no_file;
 521
 522     if (s1 && !s1->error_set_jmp_enabled)
 523         /* tcc_state just was set by tcc_enter_state() */
 524         tcc_exit_state();
 525
 526     if (mode == ERROR_WARN) {
 527         if (s1->warn_none)
 528             return;
 529         if (s1->warn_error)
 530             mode = ERROR_ERROR;
 531     }
 532
 533     f = NULL;
 534     if (s1->error_set_jmp_enabled) { /* we're called while parsing a file */
 535         /* use upper file if inline ":asm:" or token ":paste:" */
 536         for (f = file; f && f->filename[0] == ':'; f = f->prev)
 537             ;
 538     }
 539     if (f) {
 540         for(pf = s1->include_stack; pf < s1->include_stack_ptr; pf++)
 541             strcat_printf(buf, sizeof(buf), "In file included from %s:%d:\n",
 542                 (*pf)->filename, (*pf)->line_num);
 543         strcat_printf(buf, sizeof(buf), "%s:%d: ",
 544             f->filename, f->line_num - !!(tok_flags & TOK_FLAG_BOL));
 545     } else if (s1->current_filename) {
 546         strcat_printf(buf, sizeof(buf), "%s: ", s1->current_filename);
 547     }
 548
 549 no_file:
 550     if (0 == buf[0])
 551         strcat_printf(buf, sizeof(buf), "tcc: ");
 552     if (mode == ERROR_WARN)
 553         strcat_printf(buf, sizeof(buf), "warning: ");
 554     else
 555         strcat_printf(buf, sizeof(buf), "error: ");
 556     strcat_vprintf(buf, sizeof(buf), fmt, ap);
 557     if (!s1 || !s1->error_func) {
 558         /* default case: stderr */
 559         if (s1 && s1->output_type == TCC_OUTPUT_PREPROCESS && s1->ppfp == stdout)
 560             /* print a newline during tcc -E */
 561             printf("\n"), fflush(stdout);
 562         fflush(stdout); /* flush -v output */
 563         fprintf(stderr, "%s\n", buf);
 564         fflush(stderr); /* print error/warning now (win32) */
 565     } else {
 566         s1->error_func(s1->error_opaque, buf);
 567     }
 568     if (s1) {
 569         if (mode != ERROR_WARN)
 570             s1->nb_errors++;
 571         if (mode != ERROR_ERROR)
 572             return;
 573         if (s1->error_set_jmp_enabled)
 574             longjmp(s1->error_jmp_buf, 1);
 575     }
 576     exit(1);
 577 }
 578
 579 LIBTCCAPI void tcc_set_error_func(TCCState *s, void *error_opaque, TCCErrorFunc error_func)
 580 {
 581     s->error_opaque = error_opaque;
 582     s->error_func = error_func;
 583 }
 584
 585 LIBTCCAPI TCCErrorFunc tcc_get_error_func(TCCState *s)
 586 {
 587     return s->error_func;
 588 }
 589
 590 LIBTCCAPI void *tcc_get_error_opaque(TCCState *s)
 591 {
 592     return s->error_opaque;
 593 }
 594
 595 /* error without aborting current compilation */
 596 PUB_FUNC void _tcc_error_noabort(const char *fmt, ...)
 597 {
 598     va_list ap;
 599     va_start(ap, fmt);
 600     error1(ERROR_NOABORT, fmt, ap);
 601     va_end(ap);
 602 }
 603
 604 PUB_FUNC void _tcc_error(const char *fmt, ...)
 605 {
 606     va_list ap;
 607     va_start(ap, fmt);
 608     for (;;) error1(ERROR_ERROR, fmt, ap);
 609 }
 610
 611 PUB_FUNC void _tcc_warning(const char *fmt, ...)
 612 {
 613     va_list ap;
 614     va_start(ap, fmt);
 615     error1(ERROR_WARN, fmt, ap);
 616     va_end(ap);
 617 }
 618
 619 /********************************************************/
 620 /* I/O layer */
 621
 622 ST_FUNC void tcc_open_bf(TCCState *s1, const char *filename, int initlen)
 623 {
 624     BufferedFile *bf;
 625     int buflen = initlen ? initlen : IO_BUF_SIZE;
 626
 627     bf = tcc_mallocz(sizeof(BufferedFile) + buflen);
 628     bf->buf_ptr = bf->buffer;
 629     bf->buf_end = bf->buffer + initlen;
 630     bf->buf_end[0] = CH_EOB; /* put eob symbol */
 631     pstrcpy(bf->filename, sizeof(bf->filename), filename);
 632 #ifdef _WIN32
 633     normalize_slashes(bf->filename);
 634 #endif
 635     bf->true_filename = bf->filename;
 636     bf->line_num = 1;
 637     bf->ifdef_stack_ptr = s1->ifdef_stack_ptr;
 638     bf->fd = -1;
 639     bf->prev = file;
 640     file = bf;
 641     tok_flags = TOK_FLAG_BOL | TOK_FLAG_BOF;
 642 }
 643
 644 ST_FUNC void tcc_close(void)
 645 {
 646     TCCState *s1 = tcc_state;
 647     BufferedFile *bf = file;
 648     if (bf->fd > 0) {
 649         close(bf->fd);
 650         total_lines += bf->line_num;
 651     }
 652     if (bf->true_filename != bf->filename)
 653         tcc_free(bf->true_filename);
 654     file = bf->prev;
 655     tcc_free(bf);
 656 }
 657
 658 static int _tcc_open(TCCState *s1, const char *filename)
 659 {
 660     int fd;
 661     if (strcmp(filename, "-") == 0)
 662         fd = 0, filename = "<stdin>";
 663     else
 664         fd = open(filename, O_RDONLY | O_BINARY);
 665     if ((s1->verbose == 2 && fd >= 0) || s1->verbose == 3)
 666         printf("%s %*s%s\n", fd < 0 ? "nf":"->",
 667                (int)(s1->include_stack_ptr - s1->include_stack), "", filename);
 668     return fd;
 669 }
 670
 671 ST_FUNC int tcc_open(TCCState *s1, const char *filename)
 672 {
 673     int fd = _tcc_open(s1, filename);
 674     if (fd < 0)
 675         return -1;
 676     tcc_open_bf(s1, filename, 0);
 677     file->fd = fd;
 678     return 0;
 679 }
 680
 681 /* compile the file opened in 'file'. Return non zero if errors. */
 682 static int tcc_compile(TCCState *s1, int filetype, const char *str, int fd)
 683 {
 684     /* Here we enter the code section where we use the global variables for
 685        parsing and code generation (tccpp.c, tccgen.c, <target>-gen.c).
 686        Other threads need to wait until we're done.
 687
 688        Alternatively we could use thread local storage for those global
 689        variables, which may or may not have advantages */
 690
 691     tcc_enter_state(s1);
 692
 693     if (setjmp(s1->error_jmp_buf) == 0) {
 694         int is_asm;
 695         s1->error_set_jmp_enabled = 1;
 696         s1->nb_errors = 0;
 697
 698         if (fd == -1) {
 699             int len = strlen(str);
 700             tcc_open_bf(s1, "<string>", len);
 701             memcpy(file->buffer, str, len);
 702         } else {
 703             tcc_open_bf(s1, str, 0);
 704             file->fd = fd;
 705         }
 706
 707         is_asm = !!(filetype & (AFF_TYPE_ASM|AFF_TYPE_ASMPP));
 708         tccelf_begin_file(s1);
 709         preprocess_start(s1, is_asm);
 710         tccgen_init(s1);
 711         if (s1->output_type == TCC_OUTPUT_PREPROCESS) {
 712             tcc_preprocess(s1);
 713         } else if (is_asm) {
 714 #ifdef CONFIG_TCC_ASM
 715             tcc_assemble(s1, !!(filetype & AFF_TYPE_ASMPP));
 716 #else
 717             tcc_error_noabort("asm not supported");
 718 #endif
 719         } else {
 720             tccgen_compile(s1);
 721         }
 722     }
 723     s1->error_set_jmp_enabled = 0;
 724     tccgen_finish(s1);
 725     preprocess_end(s1);
 726     tccelf_end_file(s1);
 727
 728     tcc_exit_state();
 729     return s1->nb_errors != 0 ? -1 : 0;
 730 }
 731
 732 LIBTCCAPI int tcc_compile_string(TCCState *s, const char *str)
 733 {
 734     return tcc_compile(s, s->filetype, str, -1);
 735 }
 736
 737 /* define a preprocessor symbol. A value can also be provided with the '=' operator */
 738 LIBTCCAPI void tcc_define_symbol(TCCState *s1, const char *sym, const char *value)
 739 {
 740     if (!value)
 741         value = "1";
 742     cstr_printf(&s1->cmdline_defs, "#define %s %s\n", sym, value);
 743 }
 744
 745 /* undefine a preprocessor symbol */
 746 LIBTCCAPI void tcc_undefine_symbol(TCCState *s1, const char *sym)
 747 {
 748     cstr_printf(&s1->cmdline_defs, "#undef %s\n", sym);
 749 }
 750
 751
 752 LIBTCCAPI TCCState *tcc_new(void)
 753 {
 754     TCCState *s;
 755
 756     s = tcc_mallocz(sizeof(TCCState));
 757     if (!s)
 758         return NULL;
 759 #ifdef MEM_DEBUG
 760     ++nb_states;
 761 #endif
 762
 763 #undef gnu_ext
 764
 765     s->gnu_ext = 1;
 766     s->tcc_ext = 1;
 767     s->nocommon = 1;
 768     s->dollars_in_identifiers = 1; /*on by default like in gcc/clang*/
 769     s->cversion = 199901; /* default unless -std=c11 is supplied */
 770     s->warn_implicit_function_declaration = 1;
 771     s->ms_extensions = 1;
 772
 773 #ifdef CHAR_IS_UNSIGNED
 774     s->char_is_unsigned = 1;
 775 #endif
 776 #ifdef TCC_TARGET_I386
 777     s->seg_size = 32;
 778 #endif
 779     /* enable this if you want symbols with leading underscore on windows: */
 780 #if 0 /* def TCC_TARGET_PE */
 781     s->leading_underscore = 1;
 782 #endif
 783     s->ppfp = stdout;
 784     /* might be used in error() before preprocess_start() */
 785     s->include_stack_ptr = s->include_stack;
 786
 787     tccelf_new(s);
 788
 789 #ifdef _WIN32
 790     tcc_set_lib_path_w32(s);
 791 #else
 792     tcc_set_lib_path(s, CONFIG_TCCDIR);
 793 #endif
 794
 795     {
 796         /* define __TINYC__ 92X  */
 797         char buffer[32]; int a,b,c;
 798         sscanf(TCC_VERSION, "%d.%d.%d", &a, &b, &c);
 799         sprintf(buffer, "%d", a*10000 + b*100 + c);
 800         tcc_define_symbol(s, "__TINYC__", buffer);
 801     }
 802
 803     /* standard defines */
 804     tcc_define_symbol(s, "__STDC__", NULL);
 805     tcc_define_symbol(s, "__STDC_VERSION__", "199901L");
 806     tcc_define_symbol(s, "__STDC_HOSTED__", NULL);
 807
 808     /* target defines */
 809 #if defined(TCC_TARGET_I386)
 810     tcc_define_symbol(s, "__i386__", NULL);
 811     tcc_define_symbol(s, "__i386", NULL);
 812     tcc_define_symbol(s, "i386", NULL);
 813 #elif defined(TCC_TARGET_X86_64)
 814     tcc_define_symbol(s, "__x86_64__", NULL);
 815 #elif defined(TCC_TARGET_ARM)
 816     tcc_define_symbol(s, "__ARM_ARCH_4__", NULL);
 817     tcc_define_symbol(s, "__arm_elf__", NULL);
 818     tcc_define_symbol(s, "__arm_elf", NULL);
 819     tcc_define_symbol(s, "arm_elf", NULL);
 820     tcc_define_symbol(s, "__arm__", NULL);
 821     tcc_define_symbol(s, "__arm", NULL);
 822     tcc_define_symbol(s, "arm", NULL);
 823     tcc_define_symbol(s, "__APCS_32__", NULL);
 824     tcc_define_symbol(s, "__ARMEL__", NULL);
 825 #if defined(TCC_ARM_EABI)
 826     tcc_define_symbol(s, "__ARM_EABI__", NULL);
 827 #endif
 828 #if defined(TCC_ARM_HARDFLOAT)
 829     s->float_abi = ARM_HARD_FLOAT;
 830     tcc_define_symbol(s, "__ARM_PCS_VFP", NULL);
 831 #else
 832     s->float_abi = ARM_SOFTFP_FLOAT;
 833 #endif
 834 #elif defined(TCC_TARGET_ARM64)
 835     tcc_define_symbol(s, "__aarch64__", NULL);
 836 #elif defined TCC_TARGET_C67
 837     tcc_define_symbol(s, "__C67__", NULL);
 838 #elif defined TCC_TARGET_RISCV64
 839     tcc_define_symbol(s, "__riscv", NULL);
 840     tcc_define_symbol(s, "__riscv_xlen", "64");
 841     tcc_define_symbol(s, "__riscv_flen", "64");
 842     tcc_define_symbol(s, "__riscv_div", NULL);
 843     tcc_define_symbol(s, "__riscv_mul", NULL);
 844     tcc_define_symbol(s, "__riscv_fdiv", NULL);
 845     tcc_define_symbol(s, "__riscv_fsqrt", NULL);
 846     tcc_define_symbol(s, "__riscv_float_abi_double", NULL);
 847 #endif
 848
 849 #ifdef TCC_TARGET_PE
 850     tcc_define_symbol(s, "_WIN32", NULL);
 851     tcc_define_symbol(s, "__declspec(x)", "__attribute__((x))");
 852     tcc_define_symbol(s, "__cdecl", "");
 853 # ifdef TCC_TARGET_X86_64
 854     tcc_define_symbol(s, "_WIN64", NULL);
 855 # endif
 856 #else
 857     tcc_define_symbol(s, "__unix__", NULL);
 858     tcc_define_symbol(s, "__unix", NULL);
 859     tcc_define_symbol(s, "unix", NULL);
 860 # if defined(__linux__)
 861     tcc_define_symbol(s, "__linux__", NULL);
 862     tcc_define_symbol(s, "__linux", NULL);
 863 # endif
 864 # if defined(__FreeBSD__)
 865     tcc_define_symbol(s, "__FreeBSD__", "__FreeBSD__");
 866     /* No 'Thread Storage Local' on FreeBSD with tcc */
 867     tcc_define_symbol(s, "__NO_TLS", NULL);
 868 # endif
 869 # if defined(__FreeBSD_kernel__)
 870     tcc_define_symbol(s, "__FreeBSD_kernel__", NULL);
 871 # endif
 872 # if defined(__NetBSD__)
 873     tcc_define_symbol(s, "__NetBSD__", "__NetBSD__");
 874 # endif
 875 # if defined(__OpenBSD__)
 876     tcc_define_symbol(s, "__OpenBSD__", "__OpenBSD__");
 877 # endif
 878 #endif
 879
 880     /* TinyCC & gcc defines */
 881 #if PTR_SIZE == 4
 882     /* 32bit systems. */
 883     tcc_define_symbol(s, "__SIZE_TYPE__", "unsigned int");
 884     tcc_define_symbol(s, "__PTRDIFF_TYPE__", "int");
 885     tcc_define_symbol(s, "__ILP32__", NULL);
 886 #elif LONG_SIZE == 4
 887     /* 64bit Windows. */
 888     tcc_define_symbol(s, "__SIZE_TYPE__", "unsigned long long");
 889     tcc_define_symbol(s, "__PTRDIFF_TYPE__", "long long");
 890     tcc_define_symbol(s, "__LLP64__", NULL);
 891 #else
 892     /* Other 64bit systems. */
 893     tcc_define_symbol(s, "__SIZE_TYPE__", "unsigned long");
 894     tcc_define_symbol(s, "__PTRDIFF_TYPE__", "long");
 895     tcc_define_symbol(s, "__LP64__", NULL);
 896 #endif
 897     tcc_define_symbol(s, "__SIZEOF_POINTER__", PTR_SIZE == 4 ? "4" : "8");
 898
 899 #ifdef TCC_TARGET_PE
 900     tcc_define_symbol(s, "__WCHAR_TYPE__", "unsigned short");
 901     tcc_define_symbol(s, "__WINT_TYPE__", "unsigned short");
 902 #else
 903     tcc_define_symbol(s, "__WCHAR_TYPE__", "int");
 904     /* wint_t is unsigned int by default, but (signed) int on BSDs
 905        and unsigned short on windows.  Other OSes might have still
 906        other conventions, sigh.  */
 907 # if defined(__FreeBSD__) || defined (__FreeBSD_kernel__) \
 908   || defined(__NetBSD__) || defined(__OpenBSD__)
 909     tcc_define_symbol(s, "__WINT_TYPE__", "int");
 910 #  ifdef __FreeBSD__
 911     /* define __GNUC__ to have some useful stuff from sys/cdefs.h
 912        that are unconditionally used in FreeBSDs other system headers :/ */
 913     tcc_define_symbol(s, "__GNUC__", "2");
 914     tcc_define_symbol(s, "__GNUC_MINOR__", "7");
 915     tcc_define_symbol(s, "__builtin_alloca", "alloca");
 916 #  endif
 917 # else
 918     tcc_define_symbol(s, "__WINT_TYPE__", "unsigned int");
 919     /* glibc defines */
 920     tcc_define_symbol(s, "__REDIRECT(name, proto, alias)",
 921         "name proto __asm__ (#alias)");
 922     tcc_define_symbol(s, "__REDIRECT_NTH(name, proto, alias)",
 923         "name proto __asm__ (#alias) __THROW");
 924 # endif
 925     /* Some GCC builtins that are simple to express as macros.  */
 926     tcc_define_symbol(s, "__builtin_extract_return_addr(x)", "x");
 927 #endif /* ndef TCC_TARGET_PE */
 928 #ifdef TCC_TARGET_MACHO
 929     /* emulate APPLE-GCC to make libc's headerfiles compile: */
 930     tcc_define_symbol(s, "__APPLE__", "1");
 931     tcc_define_symbol(s, "__GNUC__", "4");   /* darwin emits warning on GCC<4 */
 932
 933     /* avoids usage of GCC/clang specific builtins in libc-headerfiles: */
 934     tcc_define_symbol(s, "__FINITE_MATH_ONLY__", "1");
 935     tcc_define_symbol(s, "_FORTIFY_SOURCE", "0");
 936 #endif /* ndef TCC_TARGET_MACHO */
 937     return s;
 938 }
 939
 940 LIBTCCAPI void tcc_delete(TCCState *s1)
 941 {
 942     /* free sections */
 943     tccelf_delete(s1);
 944
 945     /* free library paths */
 946     dynarray_reset(&s1->library_paths, &s1->nb_library_paths);
 947     dynarray_reset(&s1->crt_paths, &s1->nb_crt_paths);
 948
 949     /* free include paths */
 950     dynarray_reset(&s1->include_paths, &s1->nb_include_paths);
 951     dynarray_reset(&s1->sysinclude_paths, &s1->nb_sysinclude_paths);
 952
 953     tcc_free(s1->tcc_lib_path);
 954     tcc_free(s1->soname);
 955     tcc_free(s1->rpath);
 956     tcc_free(s1->init_symbol);
 957     tcc_free(s1->fini_symbol);
 958     tcc_free(s1->outfile);
 959     tcc_free(s1->deps_outfile);
 960     dynarray_reset(&s1->files, &s1->nb_files);
 961     dynarray_reset(&s1->target_deps, &s1->nb_target_deps);
 962     dynarray_reset(&s1->pragma_libs, &s1->nb_pragma_libs);
 963     dynarray_reset(&s1->argv, &s1->argc);
 964
 965     cstr_free(&s1->cmdline_defs);
 966     cstr_free(&s1->cmdline_incl);
 967 #ifdef TCC_IS_NATIVE
 968     /* free runtime memory */
 969     tcc_run_free(s1);
 970 #endif
 971
 972     tcc_free(s1);
 973 #ifdef MEM_DEBUG
 974     if (0 == --nb_states)
 975         tcc_memcheck();
 976 #endif
 977 }
 978
 979 LIBTCCAPI int tcc_set_output_type(TCCState *s, int output_type)
 980 {
 981     s->output_type = output_type;
 982
 983     /* always elf for objects */
 984     if (output_type == TCC_OUTPUT_OBJ)
 985         s->output_format = TCC_OUTPUT_FORMAT_ELF;
 986
 987     if (s->char_is_unsigned)
 988         tcc_define_symbol(s, "__CHAR_UNSIGNED__", NULL);
 989
 990     if (s->cversion == 201112) {
 991         tcc_undefine_symbol(s, "__STDC_VERSION__");
 992         tcc_define_symbol(s, "__STDC_VERSION__", "201112L");
 993         tcc_define_symbol(s, "__STDC_NO_ATOMICS__", NULL);
 994         tcc_define_symbol(s, "__STDC_NO_COMPLEX__", NULL);
 995         tcc_define_symbol(s, "__STDC_NO_THREADS__", NULL);
 996 #ifndef TCC_TARGET_PE
 997         /* on Linux, this conflicts with a define introduced by
 998            /usr/include/stdc-predef.h included by glibc libs
 999         tcc_define_symbol(s, "__STDC_ISO_10646__", "201605L"); */
1000         tcc_define_symbol(s, "__STDC_UTF_16__", NULL);
1001         tcc_define_symbol(s, "__STDC_UTF_32__", NULL);
1002 #endif
1003     }
1004
1005     if (s->optimize > 0)
1006         tcc_define_symbol(s, "__OPTIMIZE__", NULL);
1007
1008     if (s->option_pthread)
1009         tcc_define_symbol(s, "_REENTRANT", NULL);
1010
1011     if (!s->nostdinc) {
1012         /* default include paths */
1013         /* -isystem paths have already been handled */
1014         tcc_add_sysinclude_path(s, CONFIG_TCC_SYSINCLUDEPATHS);
1015     }
1016
1017 #ifdef CONFIG_TCC_BCHECK
1018     if (s->do_bounds_check) {
1019         /* if bound checking, then add corresponding sections */
1020         tccelf_bounds_new(s);
1021         /* define symbol */
1022         tcc_define_symbol(s, "__BOUNDS_CHECKING_ON", NULL);
1023     }
1024 #endif
1025     if (s->do_debug) {
1026         /* add debug sections */
1027         tccelf_stab_new(s);
1028     }
1029
1030     tcc_add_library_path(s, CONFIG_TCC_LIBPATHS);
1031
1032 #ifdef TCC_TARGET_PE
1033 # ifdef _WIN32
1034     if (!s->nostdlib && output_type != TCC_OUTPUT_OBJ)
1035         tcc_add_systemdir(s);
1036 # endif
1037 #else
1038     /* paths for crt objects */
1039     tcc_split_path(s, &s->crt_paths, &s->nb_crt_paths, CONFIG_TCC_CRTPREFIX);
1040     /* add libc crt1/crti objects */
1041     if ((output_type == TCC_OUTPUT_EXE || output_type == TCC_OUTPUT_DLL) &&
1042         !s->nostdlib) {
1043         if (output_type != TCC_OUTPUT_DLL)
1044             tcc_add_crt(s, "crt1.o");
1045         tcc_add_crt(s, "crti.o");
1046     }
1047 #endif
1048     return 0;
1049 }
1050
1051 LIBTCCAPI int tcc_add_include_path(TCCState *s, const char *pathname)
1052 {
1053     tcc_split_path(s, &s->include_paths, &s->nb_include_paths, pathname);
1054     return 0;
1055 }
1056
1057 LIBTCCAPI int tcc_add_sysinclude_path(TCCState *s, const char *pathname)
1058 {
1059     tcc_split_path(s, &s->sysinclude_paths, &s->nb_sysinclude_paths, pathname);
1060     return 0;
1061 }
1062
1063 ST_FUNC int tcc_add_file_internal(TCCState *s1, const char *filename, int flags)
1064 {
1065     int fd, ret;
1066
1067     /* open the file */
1068     fd = _tcc_open(s1, filename);
1069     if (fd < 0) {
1070         if (flags & AFF_PRINT_ERROR)
1071             tcc_error_noabort("file '%s' not found", filename);
1072         return -1;
1073     }
1074
1075     s1->current_filename = filename;
1076     if (flags & AFF_TYPE_BIN) {
1077         ElfW(Ehdr) ehdr;
1078         int obj_type;
1079
1080         obj_type = tcc_object_type(fd, &ehdr);
1081         lseek(fd, 0, SEEK_SET);
1082
1083 #ifdef TCC_TARGET_MACHO
1084         if (0 == obj_type && 0 == strcmp(tcc_fileextension(filename), ".dylib"))
1085             obj_type = AFF_BINTYPE_DYN;
1086 #endif
1087
1088         switch (obj_type) {
1089         case AFF_BINTYPE_REL:
1090             ret = tcc_load_object_file(s1, fd, 0);
1091             break;
1092 #ifndef TCC_TARGET_PE
1093         case AFF_BINTYPE_DYN:
1094             if (s1->output_type == TCC_OUTPUT_MEMORY) {
1095                 ret = 0;
1096 #ifdef TCC_IS_NATIVE
1097                 if (NULL == dlopen(filename, RTLD_GLOBAL | RTLD_LAZY))
1098                     ret = -1;
1099 #endif
1100             } else {
1101 #ifndef TCC_TARGET_MACHO
1102                 ret = tcc_load_dll(s1, fd, filename,
1103                                    (flags & AFF_REFERENCED_DLL) != 0);
1104 #else
1105                 ret = 0;
1106 #endif
1107             }
1108             break;
1109 #endif
1110         case AFF_BINTYPE_AR:
1111             ret = tcc_load_archive(s1, fd, !(flags & AFF_WHOLE_ARCHIVE));
1112             break;
1113 #ifdef TCC_TARGET_COFF
1114         case AFF_BINTYPE_C67:
1115             ret = tcc_load_coff(s1, fd);
1116             break;
1117 #endif
1118         default:
1119 #ifdef TCC_TARGET_PE
1120             ret = pe_load_file(s1, filename, fd);
1121 #elif defined(TCC_TARGET_MACHO)
1122             ret = -1;
1123 #else
1124             /* as GNU ld, consider it is an ld script if not recognized */
1125             ret = tcc_load_ldscript(s1, fd);
1126 #endif
1127             if (ret < 0)
1128                 tcc_error_noabort("%s: unrecognized file type %d", filename,
1129                                   obj_type);
1130             break;
1131         }
1132         close(fd);
1133     } else {
1134         /* update target deps */
1135         dynarray_add(&s1->target_deps, &s1->nb_target_deps, tcc_strdup(filename));
1136         ret = tcc_compile(s1, flags, filename, fd);
1137     }
1138     s1->current_filename = NULL;
1139     return ret;
1140 }
1141
1142 LIBTCCAPI int tcc_add_file(TCCState *s, const char *filename)
1143 {
1144     int filetype = s->filetype;
1145     if (0 == (filetype & AFF_TYPE_MASK)) {
1146         /* use a file extension to detect a filetype */
1147         const char *ext = tcc_fileextension(filename);
1148         if (ext[0]) {
1149             ext++;
1150             if (!strcmp(ext, "S"))
1151                 filetype = AFF_TYPE_ASMPP;
1152             else if (!strcmp(ext, "s"))
1153                 filetype = AFF_TYPE_ASM;
1154             else if (!PATHCMP(ext, "c") || !PATHCMP(ext, "i"))
1155                 filetype = AFF_TYPE_C;
1156             else
1157                 filetype |= AFF_TYPE_BIN;
1158         } else {
1159             filetype = AFF_TYPE_C;
1160         }
1161     }
1162     return tcc_add_file_internal(s, filename, filetype | AFF_PRINT_ERROR);
1163 }
1164
1165 LIBTCCAPI int tcc_add_library_path(TCCState *s, const char *pathname)
1166 {
1167     tcc_split_path(s, &s->library_paths, &s->nb_library_paths, pathname);
1168     return 0;
1169 }
1170
1171 static int tcc_add_library_internal(TCCState *s, const char *fmt,
1172     const char *filename, int flags, char **paths, int nb_paths)
1173 {
1174     char buf[1024];
1175     int i;
1176
1177     for(i = 0; i < nb_paths; i++) {
1178         snprintf(buf, sizeof(buf), fmt, paths[i], filename);
1179         if (tcc_add_file_internal(s, buf, flags | AFF_TYPE_BIN) == 0)
1180             return 0;
1181     }
1182     return -1;
1183 }
1184
1185 /* find and load a dll. Return non zero if not found */
1186 /* XXX: add '-rpath' option support ? */
1187 ST_FUNC int tcc_add_dll(TCCState *s, const char *filename, int flags)
1188 {
1189     return tcc_add_library_internal(s, "%s/%s", filename, flags,
1190         s->library_paths, s->nb_library_paths);
1191 }
1192
1193 #ifndef TCC_TARGET_PE
1194 ST_FUNC int tcc_add_crt(TCCState *s1, const char *filename)
1195 {
1196     if (-1 == tcc_add_library_internal(s1, "%s/%s",
1197         filename, 0, s1->crt_paths, s1->nb_crt_paths))
1198         tcc_error_noabort("file '%s' not found", filename);
1199     return 0;
1200 }
1201 #endif
1202
1203 /* the library name is the same as the argument of the '-l' option */
1204 LIBTCCAPI int tcc_add_library(TCCState *s, const char *libraryname)
1205 {
1206 #if defined TCC_TARGET_PE
1207     const char *libs[] = { "%s/%s.def", "%s/lib%s.def", "%s/%s.dll", "%s/lib%s.dll", "%s/lib%s.a", NULL };
1208     const char **pp = s->static_link ? libs + 4 : libs;
1209 #elif defined TCC_TARGET_MACHO
1210     const char *libs[] = { "%s/lib%s.dylib", "%s/lib%s.a", NULL };
1211     const char **pp = s->static_link ? libs + 1 : libs;
1212 #else
1213     const char *libs[] = { "%s/lib%s.so", "%s/lib%s.a", NULL };
1214     const char **pp = s->static_link ? libs + 1 : libs;
1215 #endif
1216     int flags = s->filetype & AFF_WHOLE_ARCHIVE;
1217     while (*pp) {
1218         if (0 == tcc_add_library_internal(s, *pp,
1219             libraryname, flags, s->library_paths, s->nb_library_paths))
1220             return 0;
1221         ++pp;
1222     }
1223     return -1;
1224 }
1225
1226 PUB_FUNC int tcc_add_library_err(TCCState *s1, const char *libname)
1227 {
1228     int ret = tcc_add_library(s1, libname);
1229     if (ret < 0)
1230         tcc_error_noabort("library '%s' not found", libname);
1231     return ret;
1232 }
1233
1234 /* handle #pragma comment(lib,) */
1235 ST_FUNC void tcc_add_pragma_libs(TCCState *s1)
1236 {
1237     int i;
1238     for (i = 0; i < s1->nb_pragma_libs; i++)
1239         tcc_add_library_err(s1, s1->pragma_libs[i]);
1240 }
1241
1242 LIBTCCAPI int tcc_add_symbol(TCCState *s1, const char *name, const void *val)
1243 {
1244 #ifdef TCC_TARGET_PE
1245     /* On x86_64 'val' might not be reachable with a 32bit offset.
1246        So it is handled here as if it were in a DLL. */
1247     pe_putimport(s1, 0, name, (uintptr_t)val);
1248 #else
1249     set_elf_sym(symtab_section, (uintptr_t)val, 0,
1250         ELFW(ST_INFO)(STB_GLOBAL, STT_NOTYPE), 0,
1251         SHN_ABS, name);
1252 #endif
1253     return 0;
1254 }
1255
1256 LIBTCCAPI void tcc_set_lib_path(TCCState *s, const char *path)
1257 {
1258     tcc_free(s->tcc_lib_path);
1259     s->tcc_lib_path = tcc_strdup(path);
1260 }
1261
1262 #define WD_ALL    0x0001 /* warning is activated when using -Wall */
1263 #define FD_INVERT 0x0002 /* invert value before storing */
1264
1265 typedef struct FlagDef {
1266     uint16_t offset;
1267     uint16_t flags;
1268     const char *name;
1269 } FlagDef;
1270
1271 static int no_flag(const char **pp)
1272 {
1273     const char *p = *pp;
1274     if (*p != 'n' || *++p != 'o' || *++p != '-')
1275         return 0;
1276     *pp = p + 1;
1277     return 1;
1278 }
1279
1280 ST_FUNC int set_flag(TCCState *s, const FlagDef *flags, const char *name)
1281 {
1282     int value, ret;
1283     const FlagDef *p;
1284     const char *r;
1285
1286     value = 1;
1287     r = name;
1288     if (no_flag(&r))
1289         value = 0;
1290
1291     for (ret = -1, p = flags; p->name; ++p) {
1292         if (ret) {
1293             if (strcmp(r, p->name))
1294                 continue;
1295         } else {
1296             if (0 == (p->flags & WD_ALL))
1297                 continue;
1298         }
1299         if (p->offset) {
1300             *((unsigned char *)s + p->offset) =
1301                 p->flags & FD_INVERT ? !value : value;
1302             if (ret)
1303                 return 0;
1304         } else {
1305             ret = 0;
1306         }
1307     }
1308     return ret;
1309 }
1310
1311 static int strstart(const char *val, const char **str)
1312 {
1313     const char *p, *q;
1314     p = *str;
1315     q = val;
1316     while (*q) {
1317         if (*p != *q)
1318             return 0;
1319         p++;
1320         q++;
1321     }
1322     *str = p;
1323     return 1;
1324 }
1325
1326 /* Like strstart, but automatically takes into account that ld options can
1327  *
1328  * - start with double or single dash (e.g. '--soname' or '-soname')
1329  * - arguments can be given as separate or after '=' (e.g. '-Wl,-soname,x.so'
1330  *   or '-Wl,-soname=x.so')
1331  *
1332  * you provide `val` always in 'option[=]' form (no leading -)
1333  */
1334 static int link_option(const char *str, const char *val, const char **ptr)
1335 {
1336     const char *p, *q;
1337     int ret;
1338
1339     /* there should be 1 or 2 dashes */
1340     if (*str++ != '-')
1341         return 0;
1342     if (*str == '-')
1343         str++;
1344
1345     /* then str & val should match (potentially up to '=') */
1346     p = str;
1347     q = val;
1348
1349     ret = 1;
1350     if (q[0] == '?') {
1351         ++q;
1352         if (no_flag(&p))
1353             ret = -1;
1354     }
1355
1356     while (*q != '\0' && *q != '=') {
1357         if (*p != *q)
1358             return 0;
1359         p++;
1360         q++;
1361     }
1362
1363     /* '=' near eos means ',' or '=' is ok */
1364     if (*q == '=') {
1365         if (*p == 0)
1366             *ptr = p;
1367         if (*p != ',' && *p != '=')
1368             return 0;
1369         p++;
1370     } else if (*p) {
1371         return 0;
1372     }
1373     *ptr = p;
1374     return ret;
1375 }
1376
1377 static const char *skip_linker_arg(const char **str)
1378 {
1379     const char *s1 = *str;
1380     const char *s2 = strchr(s1, ',');
1381     *str = s2 ? s2++ : (s2 = s1 + strlen(s1));
1382     return s2;
1383 }
1384
1385 static void copy_linker_arg(char **pp, const char *s, int sep)
1386 {
1387     const char *q = s;
1388     char *p = *pp;
1389     int l = 0;
1390     if (p && sep)
1391         p[l = strlen(p)] = sep, ++l;
1392     skip_linker_arg(&q);
1393     pstrncpy(l + (*pp = tcc_realloc(p, q - s + l + 1)), s, q - s);
1394 }
1395
1396 /* set linker options */
1397 static int tcc_set_linker(TCCState *s, const char *option)
1398 {
1399     TCCState *s1 = s;
1400     while (*option) {
1401
1402         const char *p = NULL;
1403         char *end = NULL;
1404         int ignoring = 0;
1405         int ret;
1406
1407         if (link_option(option, "Bsymbolic", &p)) {
1408             s->symbolic = 1;
1409         } else if (link_option(option, "nostdlib", &p)) {
1410             s->nostdlib = 1;
1411         } else if (link_option(option, "fini=", &p)) {
1412             copy_linker_arg(&s->fini_symbol, p, 0);
1413             ignoring = 1;
1414         } else if (link_option(option, "image-base=", &p)
1415                 || link_option(option, "Ttext=", &p)) {
1416             s->text_addr = strtoull(p, &end, 16);
1417             s->has_text_addr = 1;
1418         } else if (link_option(option, "init=", &p)) {
1419             copy_linker_arg(&s->init_symbol, p, 0);
1420             ignoring = 1;
1421         } else if (link_option(option, "oformat=", &p)) {
1422 #if defined(TCC_TARGET_PE)
1423             if (strstart("pe-", &p)) {
1424 #elif PTR_SIZE == 8
1425             if (strstart("elf64-", &p)) {
1426 #else
1427             if (strstart("elf32-", &p)) {
1428 #endif
1429                 s->output_format = TCC_OUTPUT_FORMAT_ELF;
1430             } else if (!strcmp(p, "binary")) {
1431                 s->output_format = TCC_OUTPUT_FORMAT_BINARY;
1432 #ifdef TCC_TARGET_COFF
1433             } else if (!strcmp(p, "coff")) {
1434                 s->output_format = TCC_OUTPUT_FORMAT_COFF;
1435 #endif
1436             } else
1437                 goto err;
1438
1439         } else if (link_option(option, "as-needed", &p)) {
1440             ignoring = 1;
1441         } else if (link_option(option, "O", &p)) {
1442             ignoring = 1;
1443         } else if (link_option(option, "export-all-symbols", &p)) {
1444             s->rdynamic = 1;
1445         } else if (link_option(option, "export-dynamic", &p)) {
1446             s->rdynamic = 1;
1447         } else if (link_option(option, "rpath=", &p)) {
1448             copy_linker_arg(&s->rpath, p, ':');
1449         } else if (link_option(option, "enable-new-dtags", &p)) {
1450             s->enable_new_dtags = 1;
1451         } else if (link_option(option, "section-alignment=", &p)) {
1452             s->section_align = strtoul(p, &end, 16);
1453         } else if (link_option(option, "soname=", &p)) {
1454             copy_linker_arg(&s->soname, p, 0);
1455 #ifdef TCC_TARGET_PE
1456         } else if (link_option(option, "large-address-aware", &p)) {
1457             s->pe_characteristics |= 0x20;
1458         } else if (link_option(option, "file-alignment=", &p)) {
1459             s->pe_file_align = strtoul(p, &end, 16);
1460         } else if (link_option(option, "stack=", &p)) {
1461             s->pe_stack_size = strtoul(p, &end, 10);
1462         } else if (link_option(option, "subsystem=", &p)) {
1463 #if defined(TCC_TARGET_I386) || defined(TCC_TARGET_X86_64)
1464             if (!strcmp(p, "native")) {
1465                 s->pe_subsystem = 1;
1466             } else if (!strcmp(p, "console")) {
1467                 s->pe_subsystem = 3;
1468             } else if (!strcmp(p, "gui") || !strcmp(p, "windows")) {
1469                 s->pe_subsystem = 2;
1470             } else if (!strcmp(p, "posix")) {
1471                 s->pe_subsystem = 7;
1472             } else if (!strcmp(p, "efiapp")) {
1473                 s->pe_subsystem = 10;
1474             } else if (!strcmp(p, "efiboot")) {
1475                 s->pe_subsystem = 11;
1476             } else if (!strcmp(p, "efiruntime")) {
1477                 s->pe_subsystem = 12;
1478             } else if (!strcmp(p, "efirom")) {
1479                 s->pe_subsystem = 13;
1480 #elif defined(TCC_TARGET_ARM)
1481             if (!strcmp(p, "wince")) {
1482                 s->pe_subsystem = 9;
1483 #endif
1484             } else
1485                 goto err;
1486 #endif
1487         } else if (ret = link_option(option, "?whole-archive", &p), ret) {
1488             if (ret > 0)
1489                 s->filetype |= AFF_WHOLE_ARCHIVE;
1490             else
1491                 s->filetype &= ~AFF_WHOLE_ARCHIVE;
1492         } else if (p) {
1493             return 0;
1494         } else {
1495     err:
1496             tcc_error("unsupported linker option '%s'", option);
1497         }
1498
1499         if (ignoring && s->warn_unsupported)
1500             tcc_warning("unsupported linker option '%s'", option);
1501
1502         option = skip_linker_arg(&p);
1503     }
1504     return 1;
1505 }
1506
1507 typedef struct TCCOption {
1508     const char *name;
1509     uint16_t index;
1510     uint16_t flags;
1511 } TCCOption;
1512
1513 enum {
1514     TCC_OPTION_HELP,
1515     TCC_OPTION_HELP2,
1516     TCC_OPTION_v,
1517     TCC_OPTION_I,
1518     TCC_OPTION_D,
1519     TCC_OPTION_U,
1520     TCC_OPTION_P,
1521     TCC_OPTION_L,
1522     TCC_OPTION_B,
1523     TCC_OPTION_l,
1524     TCC_OPTION_bench,
1525     TCC_OPTION_bt,
1526     TCC_OPTION_b,
1527     TCC_OPTION_ba,
1528     TCC_OPTION_g,
1529     TCC_OPTION_c,
1530     TCC_OPTION_dumpversion,
1531     TCC_OPTION_d,
1532     TCC_OPTION_static,
1533     TCC_OPTION_std,
1534     TCC_OPTION_shared,
1535     TCC_OPTION_soname,
1536     TCC_OPTION_o,
1537     TCC_OPTION_r,
1538     TCC_OPTION_s,
1539     TCC_OPTION_traditional,
1540     TCC_OPTION_Wl,
1541     TCC_OPTION_Wp,
1542     TCC_OPTION_W,
1543     TCC_OPTION_O,
1544     TCC_OPTION_mfloat_abi,
1545     TCC_OPTION_m,
1546     TCC_OPTION_f,
1547     TCC_OPTION_isystem,
1548     TCC_OPTION_iwithprefix,
1549     TCC_OPTION_include,
1550     TCC_OPTION_nostdinc,
1551     TCC_OPTION_nostdlib,
1552     TCC_OPTION_print_search_dirs,
1553     TCC_OPTION_rdynamic,
1554     TCC_OPTION_param,
1555     TCC_OPTION_pedantic,
1556     TCC_OPTION_pthread,
1557     TCC_OPTION_run,
1558     TCC_OPTION_w,
1559     TCC_OPTION_pipe,
1560     TCC_OPTION_E,
1561     TCC_OPTION_MD,
1562     TCC_OPTION_MF,
1563     TCC_OPTION_x,
1564     TCC_OPTION_ar,
1565     TCC_OPTION_impdef
1566 };
1567
1568 #define TCC_OPTION_HAS_ARG 0x0001
1569 #define TCC_OPTION_NOSEP   0x0002 /* cannot have space before option and arg */
1570
1571 static const TCCOption tcc_options[] = {
1572     { "h", TCC_OPTION_HELP, 0 },
1573     { "-help", TCC_OPTION_HELP, 0 },
1574     { "?", TCC_OPTION_HELP, 0 },
1575     { "hh", TCC_OPTION_HELP2, 0 },
1576     { "v", TCC_OPTION_v, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1577     { "-version", TCC_OPTION_v, 0 }, /* handle as verbose, also prints version*/
1578     { "I", TCC_OPTION_I, TCC_OPTION_HAS_ARG },
1579     { "D", TCC_OPTION_D, TCC_OPTION_HAS_ARG },
1580     { "U", TCC_OPTION_U, TCC_OPTION_HAS_ARG },
1581     { "P", TCC_OPTION_P, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1582     { "L", TCC_OPTION_L, TCC_OPTION_HAS_ARG },
1583     { "B", TCC_OPTION_B, TCC_OPTION_HAS_ARG },
1584     { "l", TCC_OPTION_l, TCC_OPTION_HAS_ARG },
1585     { "bench", TCC_OPTION_bench, 0 },
1586 #ifdef CONFIG_TCC_BACKTRACE
1587     { "bt", TCC_OPTION_bt, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1588 #endif
1589 #ifdef CONFIG_TCC_BCHECK
1590     { "b", TCC_OPTION_b, 0 },
1591 #endif
1592     { "g", TCC_OPTION_g, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1593     { "c", TCC_OPTION_c, 0 },
1594     { "dumpversion", TCC_OPTION_dumpversion, 0},
1595     { "d", TCC_OPTION_d, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1596     { "static", TCC_OPTION_static, 0 },
1597     { "std", TCC_OPTION_std, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1598     { "shared", TCC_OPTION_shared, 0 },
1599     { "soname", TCC_OPTION_soname, TCC_OPTION_HAS_ARG },
1600     { "o", TCC_OPTION_o, TCC_OPTION_HAS_ARG },
1601     { "-param", TCC_OPTION_param, TCC_OPTION_HAS_ARG },
1602     { "pedantic", TCC_OPTION_pedantic, 0},
1603     { "pthread", TCC_OPTION_pthread, 0},
1604     { "run", TCC_OPTION_run, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1605     { "rdynamic", TCC_OPTION_rdynamic, 0 },
1606     { "r", TCC_OPTION_r, 0 },
1607     { "s", TCC_OPTION_s, 0 },
1608     { "traditional", TCC_OPTION_traditional, 0 },
1609     { "Wl,", TCC_OPTION_Wl, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1610     { "Wp,", TCC_OPTION_Wp, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1611     { "W", TCC_OPTION_W, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1612     { "O", TCC_OPTION_O, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1613 #ifdef TCC_TARGET_ARM
1614     { "mfloat-abi", TCC_OPTION_mfloat_abi, TCC_OPTION_HAS_ARG },
1615 #endif
1616     { "m", TCC_OPTION_m, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1617     { "f", TCC_OPTION_f, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1618     { "isystem", TCC_OPTION_isystem, TCC_OPTION_HAS_ARG },
1619     { "include", TCC_OPTION_include, TCC_OPTION_HAS_ARG },
1620     { "nostdinc", TCC_OPTION_nostdinc, 0 },
1621     { "nostdlib", TCC_OPTION_nostdlib, 0 },
1622     { "print-search-dirs", TCC_OPTION_print_search_dirs, 0 },
1623     { "w", TCC_OPTION_w, 0 },
1624     { "pipe", TCC_OPTION_pipe, 0},
1625     { "E", TCC_OPTION_E, 0},
1626     { "MD", TCC_OPTION_MD, 0},
1627     { "MF", TCC_OPTION_MF, TCC_OPTION_HAS_ARG },
1628     { "x", TCC_OPTION_x, TCC_OPTION_HAS_ARG },
1629     { "ar", TCC_OPTION_ar, 0},
1630 #ifdef TCC_TARGET_PE
1631     { "impdef", TCC_OPTION_impdef, 0},
1632 #endif
1633     { NULL, 0, 0 },
1634 };
1635
1636 static const FlagDef options_W[] = {
1637     { 0, 0, "all" },
1638     { offsetof(TCCState, warn_unsupported), 0, "unsupported" },
1639     { offsetof(TCCState, warn_write_strings), 0, "write-strings" },
1640     { offsetof(TCCState, warn_error), 0, "error" },
1641     { offsetof(TCCState, warn_gcc_compat), 0, "gcc-compat" },
1642     { offsetof(TCCState, warn_implicit_function_declaration), WD_ALL,
1643       "implicit-function-declaration" },
1644     { 0, 0, NULL }
1645 };
1646
1647 static const FlagDef options_f[] = {
1648     { offsetof(TCCState, char_is_unsigned), 0, "unsigned-char" },
1649     { offsetof(TCCState, char_is_unsigned), FD_INVERT, "signed-char" },
1650     { offsetof(TCCState, nocommon), FD_INVERT, "common" },
1651     { offsetof(TCCState, leading_underscore), 0, "leading-underscore" },
1652     { offsetof(TCCState, ms_extensions), 0, "ms-extensions" },
1653     { offsetof(TCCState, dollars_in_identifiers), 0, "dollars-in-identifiers" },
1654     { 0, 0, NULL }
1655 };
1656
1657 static const FlagDef options_m[] = {
1658     { offsetof(TCCState, ms_bitfields), 0, "ms-bitfields" },
1659 #ifdef TCC_TARGET_X86_64
1660     { offsetof(TCCState, nosse), FD_INVERT, "sse" },
1661 #endif
1662     { 0, 0, NULL }
1663 };
1664
1665 static void parse_option_D(TCCState *s1, const char *optarg)
1666 {
1667     char *sym = tcc_strdup(optarg);
1668     char *value = strchr(sym, '=');
1669     if (value)
1670         *value++ = '\0';
1671     tcc_define_symbol(s1, sym, value);
1672     tcc_free(sym);
1673 }
1674
1675 static void args_parser_add_file(TCCState *s, const char* filename, int filetype)
1676 {
1677     struct filespec *f = tcc_malloc(sizeof *f + strlen(filename));
1678     f->type = filetype;
1679     strcpy(f->name, filename);
1680     dynarray_add(&s->files, &s->nb_files, f);
1681 }
1682
1683 static int args_parser_make_argv(const char *r, int *argc, char ***argv)
1684 {
1685     int ret = 0, q, c;
1686     CString str;
1687     for(;;) {
1688         while (c = (unsigned char)*r, c && c <= ' ')
1689           ++r;
1690         if (c == 0)
1691             break;
1692         q = 0;
1693         cstr_new(&str);
1694         while (c = (unsigned char)*r, c) {
1695             ++r;
1696             if (c == '\\' && (*r == '"' || *r == '\\')) {
1697                 c = *r++;
1698             } else if (c == '"') {
1699                 q = !q;
1700                 continue;
1701             } else if (q == 0 && c <= ' ') {
1702                 break;
1703             }
1704             cstr_ccat(&str, c);
1705         }
1706         cstr_ccat(&str, 0);
1707         //printf("<%s>\n", str.data), fflush(stdout);
1708         dynarray_add(argv, argc, tcc_strdup(str.data));
1709         cstr_free(&str);
1710         ++ret;
1711     }
1712     return ret;
1713 }
1714
1715 /* read list file */
1716 static void args_parser_listfile(TCCState *s,
1717     const char *filename, int optind, int *pargc, char ***pargv)
1718 {
1719     TCCState *s1 = s;
1720     int fd, i;
1721     size_t len;
1722     char *p;
1723     int argc = 0;
1724     char **argv = NULL;
1725
1726     fd = open(filename, O_RDONLY | O_BINARY);
1727     if (fd < 0)
1728         tcc_error("listfile '%s' not found", filename);
1729
1730     len = lseek(fd, 0, SEEK_END);
1731     p = tcc_malloc(len + 1), p[len] = 0;
1732     lseek(fd, 0, SEEK_SET), read(fd, p, len), close(fd);
1733
1734     for (i = 0; i < *pargc; ++i)
1735         if (i == optind)
1736             args_parser_make_argv(p, &argc, &argv);
1737         else
1738             dynarray_add(&argv, &argc, tcc_strdup((*pargv)[i]));
1739
1740     tcc_free(p);
1741     dynarray_reset(&s->argv, &s->argc);
1742     *pargc = s->argc = argc, *pargv = s->argv = argv;
1743 }
1744
1745 PUB_FUNC int tcc_parse_args(TCCState *s, int *pargc, char ***pargv, int optind)
1746 {
1747     TCCState *s1 = s;
1748     const TCCOption *popt;
1749     const char *optarg, *r;
1750     const char *run = NULL;
1751     int x;
1752     CString linker_arg; /* collect -Wl options */
1753     int tool = 0, arg_start = 0, noaction = optind;
1754     char **argv = *pargv;
1755     int argc = *pargc;
1756
1757     cstr_new(&linker_arg);
1758
1759     while (optind < argc) {
1760         r = argv[optind];
1761         if (r[0] == '@' && r[1] != '\0') {
1762             args_parser_listfile(s, r + 1, optind, &argc, &argv);
1763             continue;
1764         }
1765         optind++;
1766         if (tool) {
1767             if (r[0] == '-' && r[1] == 'v' && r[2] == 0)
1768                 ++s->verbose;
1769             continue;
1770         }
1771 reparse:
1772         if (r[0] != '-' || r[1] == '\0') {
1773             if (r[0] != '@') /* allow "tcc file(s) -run @ args ..." */
1774                 args_parser_add_file(s, r, s->filetype);
1775             if (run) {
1776                 tcc_set_options(s, run);
1777                 arg_start = optind - 1;
1778                 break;
1779             }
1780             continue;
1781         }
1782
1783         /* find option in table */
1784         for(popt = tcc_options; ; ++popt) {
1785             const char *p1 = popt->name;
1786             const char *r1 = r + 1;
1787             if (p1 == NULL)
1788                 tcc_error("invalid option -- '%s'", r);
1789             if (!strstart(p1, &r1))
1790                 continue;
1791             optarg = r1;
1792             if (popt->flags & TCC_OPTION_HAS_ARG) {
1793                 if (*r1 == '\0' && !(popt->flags & TCC_OPTION_NOSEP)) {
1794                     if (optind >= argc)
1795                 arg_err:
1796                         tcc_error("argument to '%s' is missing", r);
1797                     optarg = argv[optind++];
1798                 }
1799             } else if (*r1 != '\0')
1800                 continue;
1801             break;
1802         }
1803
1804         switch(popt->index) {
1805         case TCC_OPTION_HELP:
1806             x = OPT_HELP;
1807             goto extra_action;
1808         case TCC_OPTION_HELP2:
1809             x = OPT_HELP2;
1810             goto extra_action;
1811         case TCC_OPTION_I:
1812             tcc_add_include_path(s, optarg);
1813             break;
1814         case TCC_OPTION_D:
1815             parse_option_D(s, optarg);
1816             break;
1817         case TCC_OPTION_U:
1818             tcc_undefine_symbol(s, optarg);
1819             break;
1820         case TCC_OPTION_L:
1821             tcc_add_library_path(s, optarg);
1822             break;
1823         case TCC_OPTION_B:
1824             /* set tcc utilities path (mainly for tcc development) */
1825             tcc_set_lib_path(s, optarg);
1826             break;
1827         case TCC_OPTION_l:
1828             args_parser_add_file(s, optarg, AFF_TYPE_LIB | (s->filetype & ~AFF_TYPE_MASK));
1829             s->nb_libraries++;
1830             break;
1831         case TCC_OPTION_pthread:
1832             s->option_pthread = 1;
1833             break;
1834         case TCC_OPTION_bench:
1835             s->do_bench = 1;
1836             break;
1837 #ifdef CONFIG_TCC_BACKTRACE
1838         case TCC_OPTION_bt:
1839             s->rt_num_callers = atoi(optarg);
1840             s->do_backtrace = 1;
1841             s->do_debug = 1;
1842             break;
1843 #endif
1844 #ifdef CONFIG_TCC_BCHECK
1845         case TCC_OPTION_b:
1846             s->do_bounds_check = 1;
1847             s->do_backtrace = 1;
1848             s->do_debug = 1;
1849             break;
1850 #endif
1851         case TCC_OPTION_g:
1852             s->do_debug = 1;
1853             break;
1854         case TCC_OPTION_c:
1855             x = TCC_OUTPUT_OBJ;
1856         set_output_type:
1857             if (s->output_type)
1858                 tcc_warning("-%s: overriding compiler action already specified", popt->name);
1859             s->output_type = x;
1860             break;
1861         case TCC_OPTION_d:
1862             if (*optarg == 'D')
1863                 s->dflag = 3;
1864             else if (*optarg == 'M')
1865                 s->dflag = 7;
1866             else if (*optarg == 't')
1867                 s->dflag = 16;
1868             else if (isnum(*optarg))
1869                 s->g_debug |= atoi(optarg);
1870             else
1871                 goto unsupported_option;
1872             break;
1873         case TCC_OPTION_static:
1874             s->static_link = 1;
1875             break;
1876         case TCC_OPTION_std:
1877             if (strcmp(optarg, "=c11") == 0)
1878                 s->cversion = 201112;
1879             break;
1880         case TCC_OPTION_shared:
1881             x = TCC_OUTPUT_DLL;
1882             goto set_output_type;
1883         case TCC_OPTION_soname:
1884             s->soname = tcc_strdup(optarg);
1885             break;
1886         case TCC_OPTION_o:
1887             if (s->outfile) {
1888                 tcc_warning("multiple -o option");
1889                 tcc_free(s->outfile);
1890             }
1891             s->outfile = tcc_strdup(optarg);
1892             break;
1893         case TCC_OPTION_r:
1894             /* generate a .o merging several output files */
1895             s->option_r = 1;
1896             x = TCC_OUTPUT_OBJ;
1897             goto set_output_type;
1898         case TCC_OPTION_isystem:
1899             tcc_add_sysinclude_path(s, optarg);
1900             break;
1901         case TCC_OPTION_include:
1902             cstr_printf(&s->cmdline_incl, "#include \"%s\"\n", optarg);
1903             break;
1904         case TCC_OPTION_nostdinc:
1905             s->nostdinc = 1;
1906             break;
1907         case TCC_OPTION_nostdlib:
1908             s->nostdlib = 1;
1909             break;
1910         case TCC_OPTION_run:
1911 #ifndef TCC_IS_NATIVE
1912             tcc_error("-run is not available in a cross compiler");
1913 #endif
1914             run = optarg;
1915             x = TCC_OUTPUT_MEMORY;
1916             goto set_output_type;
1917         case TCC_OPTION_v:
1918             do ++s->verbose; while (*optarg++ == 'v');
1919             ++noaction;
1920             break;
1921         case TCC_OPTION_f:
1922             if (set_flag(s, options_f, optarg) < 0)
1923                 goto unsupported_option;
1924             break;
1925 #ifdef TCC_TARGET_ARM
1926         case TCC_OPTION_mfloat_abi:
1927             /* tcc doesn't support soft float yet */
1928             if (!strcmp(optarg, "softfp")) {
1929                 s->float_abi = ARM_SOFTFP_FLOAT;
1930                 tcc_undefine_symbol(s, "__ARM_PCS_VFP");
1931             } else if (!strcmp(optarg, "hard"))
1932                 s->float_abi = ARM_HARD_FLOAT;
1933             else
1934                 tcc_error("unsupported float abi '%s'", optarg);
1935             break;
1936 #endif
1937         case TCC_OPTION_m:
1938             if (set_flag(s, options_m, optarg) < 0) {
1939                 if (x = atoi(optarg), x != 32 && x != 64)
1940                     goto unsupported_option;
1941                 if (PTR_SIZE != x/8)
1942                     return x;
1943                 ++noaction;
1944             }
1945             break;
1946         case TCC_OPTION_W:
1947             s->warn_none = 0;
1948             if (optarg[0] && set_flag(s, options_W, optarg) < 0)
1949                 goto unsupported_option;
1950             break;
1951         case TCC_OPTION_w:
1952             s->warn_none = 1;
1953             break;
1954         case TCC_OPTION_rdynamic:
1955             s->rdynamic = 1;
1956             break;
1957         case TCC_OPTION_Wl:
1958             if (linker_arg.size)
1959                 --linker_arg.size, cstr_ccat(&linker_arg, ',');
1960             cstr_cat(&linker_arg, optarg, 0);
1961             if (tcc_set_linker(s, linker_arg.data))
1962                 cstr_free(&linker_arg);
1963             break;
1964         case TCC_OPTION_Wp:
1965             r = optarg;
1966             goto reparse;
1967         case TCC_OPTION_E:
1968             x = TCC_OUTPUT_PREPROCESS;
1969             goto set_output_type;
1970         case TCC_OPTION_P:
1971             s->Pflag = atoi(optarg) + 1;
1972             break;
1973         case TCC_OPTION_MD:
1974             s->gen_deps = 1;
1975             break;
1976         case TCC_OPTION_MF:
1977             s->deps_outfile = tcc_strdup(optarg);
1978             break;
1979         case TCC_OPTION_dumpversion:
1980             printf ("%s\n", TCC_VERSION);
1981             exit(0);
1982             break;
1983         case TCC_OPTION_x:
1984             x = 0;
1985             if (*optarg == 'c')
1986                 x = AFF_TYPE_C;
1987             else if (*optarg == 'a')
1988                 x = AFF_TYPE_ASMPP;
1989             else if (*optarg == 'b')
1990                 x = AFF_TYPE_BIN;
1991             else if (*optarg == 'n')
1992                 x = AFF_TYPE_NONE;
1993             else
1994                 tcc_warning("unsupported language '%s'", optarg);
1995             s->filetype = x | (s->filetype & ~AFF_TYPE_MASK);
1996             break;
1997         case TCC_OPTION_O:
1998             s->optimize = atoi(optarg);
1999             break;
2000         case TCC_OPTION_print_search_dirs:
2001             x = OPT_PRINT_DIRS;
2002             goto extra_action;
2003         case TCC_OPTION_impdef:
2004             x = OPT_IMPDEF;
2005             goto extra_action;
2006         case TCC_OPTION_ar:
2007             x = OPT_AR;
2008         extra_action:
2009             arg_start = optind - 1;
2010             if (arg_start != noaction)
2011                 tcc_error("cannot parse %s here", r);
2012             tool = x;
2013             break;
2014         case TCC_OPTION_traditional:
2015         case TCC_OPTION_pedantic:
2016         case TCC_OPTION_pipe:
2017         case TCC_OPTION_s:
2018             /* ignored */
2019             break;
2020         default:
2021 unsupported_option:
2022             if (s->warn_unsupported)
2023                 tcc_warning("unsupported option '%s'", r);
2024             break;
2025         }
2026     }
2027     if (linker_arg.size) {
2028         r = linker_arg.data;
2029         goto arg_err;
2030     }
2031     *pargc = argc - arg_start;
2032     *pargv = argv + arg_start;
2033     if (tool)
2034         return tool;
2035     if (optind != noaction)
2036         return 0;
2037     if (s->verbose == 2)
2038         return OPT_PRINT_DIRS;
2039     if (s->verbose)
2040         return OPT_V;
2041     return OPT_HELP;
2042 }
2043
2044 LIBTCCAPI void tcc_set_options(TCCState *s, const char *r)
2045 {
2046     char **argv = NULL;
2047     int argc = 0;
2048     args_parser_make_argv(r, &argc, &argv);
2049     tcc_parse_args(s, &argc, &argv, 0);
2050     dynarray_reset(&argv, &argc);
2051 }
2052
2053 PUB_FUNC void tcc_print_stats(TCCState *s1, unsigned total_time)
2054 {
2055     if (total_time < 1)
2056         total_time = 1;
2057     if (total_bytes < 1)
2058         total_bytes = 1;
2059     fprintf(stderr, "* %d idents, %d lines, %d bytes\n"
2060                     "* %0.3f s, %u lines/s, %0.1f MB/s\n",
2061            total_idents, total_lines, total_bytes,
2062            (double)total_time/1000,
2063            (unsigned)total_lines*1000/total_time,
2064            (double)total_bytes/1000/total_time);
2065 #ifdef MEM_DEBUG
2066     fprintf(stderr, "* %d bytes memory used\n", mem_max_size);
2067 #endif
2068 }