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
  21 #ifdef ONE_SOURCE
  22 #include "libtcc.c"
  23 #else
  24 #include "tcc.h"
  25 #endif
  26
  27 static char **files;
  28 static int nb_files, nb_libraries;
  29 static int multiple_files;
  30 static int print_search_dirs;
  31 static int output_type;
  32 static int reloc_output;
  33 static const char *outfile;
  34 static int do_bench = 0;
  35 static int gen_deps;
  36 static const char *deps_outfile;
  37
  38 #define TCC_OPTION_HAS_ARG 0x0001
  39 #define TCC_OPTION_NOSEP   0x0002 /* cannot have space before option and arg */
  40
  41 static void help(void)
  42 {
  43     printf("tcc version " TCC_VERSION " - Tiny C Compiler - Copyright (C) 2001-2006 Fabrice Bellard\n"
  44            "Usage: tcc [options...] [-o outfile] [-c] infile(s)...\n"
  45            "       tcc [options...] -run infile [arguments...]\n"
  46            "General options:\n"
  47            "  -v          display current version, increase verbosity\n"
  48            "  -c          compile only - generate an object file\n"
  49            "  -o outfile  set output filename\n"
  50            "  -Bdir       set tcc internal library and include path\n"
  51            "  -bench      output compilation statistics\n"
  52            "  -run        run compiled source\n"
  53            "  -fflag      set or reset (with 'no-' prefix) 'flag' (see man page)\n"
  54            "  -Wwarning   set or reset (with 'no-' prefix) 'warning' (see man page)\n"
  55            "  -w          disable all warnings\n"
  56            "Preprocessor options:\n"
  57            "  -E          preprocess only\n"
  58            "  -Idir       add include path 'dir'\n"
  59            "  -Dsym[=val] define 'sym' with value 'val'\n"
  60            "  -Usym       undefine 'sym'\n"
  61            "Linker options:\n"
  62            "  -Ldir       add library path 'dir'\n"
  63            "  -llib       link with dynamic or static library 'lib'\n"
  64            "  -pthread    link with -lpthread and -D_REENTRANT (POSIX Linux)\n"
  65            "  -shared     generate a shared library\n"
  66            "  -soname     set name for shared library to be used at runtime\n"
  67            "  -static     static linking\n"
  68            "  -rdynamic   export all global symbols to dynamic linker\n"
  69            "  -r          generate (relocatable) object file\n"
  70            "Debugger options:\n"
  71            "  -g          generate runtime debug info\n"
  72 #ifdef CONFIG_TCC_BCHECK
  73            "  -b          compile with built-in memory and bounds checker (implies -g)\n"
  74 #endif
  75 #ifdef CONFIG_TCC_BACKTRACE
  76            "  -bt N       show N callers in stack traces\n"
  77 #endif
  78            "Misc options:\n"
  79            "  -MD         generate target dependencies for make\n"
  80            "  -MF depfile put generated dependencies here\n"
  81            );
  82 }
  83
  84 typedef struct TCCOption {
  85     const char *name;
  86     uint16_t index;
  87     uint16_t flags;
  88 } TCCOption;
  89
  90 enum {
  91     TCC_OPTION_HELP,
  92     TCC_OPTION_I,
  93     TCC_OPTION_D,
  94     TCC_OPTION_U,
  95     TCC_OPTION_L,
  96     TCC_OPTION_B,
  97     TCC_OPTION_l,
  98     TCC_OPTION_bench,
  99     TCC_OPTION_bt,
 100     TCC_OPTION_b,
 101     TCC_OPTION_g,
 102     TCC_OPTION_c,
 103     TCC_OPTION_static,
 104     TCC_OPTION_shared,
 105     TCC_OPTION_soname,
 106     TCC_OPTION_o,
 107     TCC_OPTION_r,
 108     TCC_OPTION_s,
 109     TCC_OPTION_Wl,
 110     TCC_OPTION_W,
 111     TCC_OPTION_O,
 112     TCC_OPTION_m,
 113     TCC_OPTION_f,
 114     TCC_OPTION_isystem,
 115     TCC_OPTION_nostdinc,
 116     TCC_OPTION_nostdlib,
 117     TCC_OPTION_print_search_dirs,
 118     TCC_OPTION_rdynamic,
 119     TCC_OPTION_pedantic,
 120     TCC_OPTION_pthread,
 121     TCC_OPTION_run,
 122     TCC_OPTION_v,
 123     TCC_OPTION_w,
 124     TCC_OPTION_pipe,
 125     TCC_OPTION_E,
 126     TCC_OPTION_MD,
 127     TCC_OPTION_MF,
 128     TCC_OPTION_x,
 129 };
 130
 131 static const TCCOption tcc_options[] = {
 132     { "h", TCC_OPTION_HELP, 0 },
 133     { "-help", TCC_OPTION_HELP, 0 },
 134     { "?", TCC_OPTION_HELP, 0 },
 135     { "I", TCC_OPTION_I, TCC_OPTION_HAS_ARG },
 136     { "D", TCC_OPTION_D, TCC_OPTION_HAS_ARG },
 137     { "U", TCC_OPTION_U, TCC_OPTION_HAS_ARG },
 138     { "L", TCC_OPTION_L, TCC_OPTION_HAS_ARG },
 139     { "B", TCC_OPTION_B, TCC_OPTION_HAS_ARG },
 140     { "l", TCC_OPTION_l, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 141     { "bench", TCC_OPTION_bench, 0 },
 142 #ifdef CONFIG_TCC_BACKTRACE
 143     { "bt", TCC_OPTION_bt, TCC_OPTION_HAS_ARG },
 144 #endif
 145 #ifdef CONFIG_TCC_BCHECK
 146     { "b", TCC_OPTION_b, 0 },
 147 #endif
 148     { "g", TCC_OPTION_g, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 149     { "c", TCC_OPTION_c, 0 },
 150     { "static", TCC_OPTION_static, 0 },
 151     { "shared", TCC_OPTION_shared, 0 },
 152     { "soname", TCC_OPTION_soname, TCC_OPTION_HAS_ARG },
 153     { "o", TCC_OPTION_o, TCC_OPTION_HAS_ARG },
 154     { "pedantic", TCC_OPTION_pedantic, 0},
 155     { "pthread", TCC_OPTION_pthread, 0},
 156     { "run", TCC_OPTION_run, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 157     { "rdynamic", TCC_OPTION_rdynamic, 0 },
 158     { "r", TCC_OPTION_r, 0 },
 159     { "s", TCC_OPTION_s, 0 },
 160     { "Wl,", TCC_OPTION_Wl, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 161     { "W", TCC_OPTION_W, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 162     { "O", TCC_OPTION_O, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 163     { "m", TCC_OPTION_m, TCC_OPTION_HAS_ARG },
 164     { "f", TCC_OPTION_f, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 165     { "isystem", TCC_OPTION_isystem, TCC_OPTION_HAS_ARG },
 166     { "nostdinc", TCC_OPTION_nostdinc, 0 },
 167     { "nostdlib", TCC_OPTION_nostdlib, 0 },
 168     { "print-search-dirs", TCC_OPTION_print_search_dirs, 0 },
 169     { "v", TCC_OPTION_v, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
 170     { "w", TCC_OPTION_w, 0 },
 171     { "pipe", TCC_OPTION_pipe, 0},
 172     { "E", TCC_OPTION_E, 0},
 173     { "MD", TCC_OPTION_MD, 0},
 174     { "MF", TCC_OPTION_MF, TCC_OPTION_HAS_ARG },
 175     { "x", TCC_OPTION_x, TCC_OPTION_HAS_ARG },
 176     { NULL },
 177 };
 178
 179 static int64_t getclock_us(void)
 180 {
 181 #ifdef _WIN32
 182     struct _timeb tb;
 183     _ftime(&tb);
 184     return (tb.time * 1000LL + tb.millitm) * 1000LL;
 185 #else
 186     struct timeval tv;
 187     gettimeofday(&tv, NULL);
 188     return tv.tv_sec * 1000000LL + tv.tv_usec;
 189 #endif
 190 }
 191
 192 /* convert 'str' into an array of space separated strings */
 193 static int expand_args(char ***pargv, const char *str)
 194 {
 195     const char *s1;
 196     char **argv, *arg;
 197     int argc, len;
 198
 199     argc = 0;
 200     argv = NULL;
 201     for(;;) {
 202         while (is_space(*str))
 203             str++;
 204         if (*str == '\0')
 205             break;
 206         s1 = str;
 207         while (*str != '\0' && !is_space(*str))
 208             str++;
 209         len = str - s1;
 210         arg = tcc_malloc(len + 1);
 211         memcpy(arg, s1, len);
 212         arg[len] = '\0';
 213         dynarray_add((void ***)&argv, &argc, arg);
 214     }
 215     *pargv = argv;
 216     return argc;
 217 }
 218
 219 #if defined(TCC_TARGET_X86_64) || defined (TCC_TARGET_I386)
 220
 221 #ifdef _WIN32
 222 #define CAST (const char * const *)
 223 #else
 224 #define CAST (char * const *)
 225 #endif
 226
 227 #if defined(TCC_TARGET_X86_64)
 228 #define ARG 32
 229 #define CHILD "i386"
 230 #elif defined (TCC_TARGET_I386)
 231 #define ARG 64
 232 #define CHILD "x86_64"
 233 #endif
 234
 235 static char *ssuffix(const char *oldname, const char sep)
 236 {
 237     char *p, *name = tcc_strdup(oldname);
 238     p = strchr(name, sep);
 239     if (p)
 240         return p + 1;
 241     /* prefix "win32-" when file extension is present */
 242     if (*tcc_fileextension(name)){
 243         name = tcc_realloc(name, strlen(oldname + 7));
 244         sprintf(name, "win32-%s", oldname);
 245     }
 246     return name;
 247 }
 248
 249 static void exec_other_tcc(TCCState *s, int argc,
 250     char **argv,const char *optarg, int optind)
 251 {
 252     char child_path[4096],child_name[4096];
 253     char *parent,*child_tcc;
 254     int opt = atoi(optarg);
 255     if (strlen(argv[0]) > 4000)
 256         error("-m%i unsafe path length", ARG);
 257     switch (opt) {
 258         case ARG + 1: /* oops we called ourselves */
 259             error("-m%i cross compiler not installed", ARG);
 260             break;
 261         case ARG:
 262         {
 263             parent = tcc_basename(argv[0]);
 264             child_tcc = ssuffix(parent,'-');
 265             sprintf(child_name, CHILD "-%s", child_tcc);
 266             tcc_free(child_tcc);
 267             if (strcmp(parent, child_name)) {
 268                 /* child_path = dirname */
 269                 pstrcpy(child_path, parent - argv[0] + 1, argv[0]);
 270                 child_tcc = strchr(child_path, 0);
 271                 strcpy(child_tcc, child_name);
 272                 if (0 < s->verbose)
 273                     printf("%s -m%i -> %s\n",
 274                          parent, ARG, child_path);
 275                 sprintf(argv[optind],"-m%i", ARG + 1); /* no loop */
 276                 execvp(child_path, CAST argv);
 277                 sprintf(child_tcc,"tcc%s",tcc_fileextension(parent));
 278                 execvp(child_path, CAST argv);
 279                 error("-m%i cross compiler not found", ARG);
 280             } else error("-m%i unsupported configuration", ARG);
 281         }
 282         case 96 ^  ARG     : break;
 283         case 96 ^ (ARG + 1): break;
 284         default:
 285             warning("usupported option \"-m%s\"",optarg);
 286     }
 287 }
 288 #undef CAST
 289 #undef ARG
 290 #undef CHILD
 291 #endif
 292
 293 static int parse_args(TCCState *s, int argc, char **argv)
 294 {
 295     int optind;
 296     const TCCOption *popt;
 297     const char *optarg, *p1, *r1;
 298     char *r;
 299     int was_pthread;
 300
 301     was_pthread = 0; /* is set if commandline contains -pthread key */
 302
 303     optind = 0;
 304     while (optind < argc) {
 305
 306         r = argv[optind++];
 307         if (r[0] != '-' || r[1] == '\0') {
 308             /* add a new file */
 309             dynarray_add((void ***)&files, &nb_files, r);
 310             if (!multiple_files) {
 311                 optind--;
 312                 /* argv[0] will be this file */
 313                 break;
 314             }
 315         } else {
 316             /* find option in table (match only the first chars */
 317             popt = tcc_options;
 318             for(;;) {
 319                 p1 = popt->name;
 320                 if (p1 == NULL)
 321                     error("invalid option -- '%s'", r);
 322                 r1 = r + 1;
 323                 for(;;) {
 324                     if (*p1 == '\0')
 325                         goto option_found;
 326                     if (*r1 != *p1)
 327                         break;
 328                     p1++;
 329                     r1++;
 330                 }
 331                 popt++;
 332             }
 333         option_found:
 334             if (popt->flags & TCC_OPTION_HAS_ARG) {
 335                 if (*r1 != '\0' || (popt->flags & TCC_OPTION_NOSEP)) {
 336                     optarg = r1;
 337                 } else {
 338                     if (optind >= argc)
 339                         error("argument to '%s' is missing", r);
 340                     optarg = argv[optind++];
 341                 }
 342             } else {
 343                 if (*r1 != '\0')
 344                     return 0;
 345                 optarg = NULL;
 346             }
 347
 348             switch(popt->index) {
 349             case TCC_OPTION_HELP:
 350                 return 0;
 351
 352             case TCC_OPTION_I:
 353                 if (tcc_add_include_path(s, optarg) < 0)
 354                     error("too many include paths");
 355                 break;
 356             case TCC_OPTION_D:
 357                 {
 358                     char *sym, *value;
 359                     sym = (char *)optarg;
 360                     value = strchr(sym, '=');
 361                     if (value) {
 362                         *value = '\0';
 363                         value++;
 364                     }
 365                     tcc_define_symbol(s, sym, value);
 366                 }
 367                 break;
 368             case TCC_OPTION_U:
 369                 tcc_undefine_symbol(s, optarg);
 370                 break;
 371             case TCC_OPTION_L:
 372                 tcc_add_library_path(s, optarg);
 373                 break;
 374             case TCC_OPTION_B:
 375                 /* set tcc utilities path (mainly for tcc development) */
 376                 tcc_set_lib_path(s, optarg);
 377                 /* append /include and add it as -isystem, as gcc does */
 378                 r = tcc_strdup(optarg);
 379                 r = tcc_realloc(r, strlen(r) + 8 + 1);
 380                 strcat(r, "/include");
 381                 tcc_add_sysinclude_path(s, r);
 382                 tcc_free(r);
 383                 break;
 384             case TCC_OPTION_l:
 385                 dynarray_add((void ***)&files, &nb_files, r);
 386                 nb_libraries++;
 387                 break;
 388             case TCC_OPTION_pthread:
 389                 was_pthread = 1;
 390                 tcc_define_symbol(s, "_REENTRANT", "1");
 391                 break;
 392             case TCC_OPTION_bench:
 393                 do_bench = 1;
 394                 break;
 395 #ifdef CONFIG_TCC_BACKTRACE
 396             case TCC_OPTION_bt:
 397                 set_num_callers(atoi(optarg));
 398                 break;
 399 #endif
 400 #ifdef CONFIG_TCC_BCHECK
 401             case TCC_OPTION_b:
 402                 s->do_bounds_check = 1;
 403                 s->do_debug = 1;
 404                 break;
 405 #endif
 406             case TCC_OPTION_g:
 407                 s->do_debug = 1;
 408                 break;
 409             case TCC_OPTION_c:
 410                 multiple_files = 1;
 411                 output_type = TCC_OUTPUT_OBJ;
 412                 break;
 413             case TCC_OPTION_static:
 414                 s->static_link = 1;
 415                 break;
 416             case TCC_OPTION_shared:
 417                 output_type = TCC_OUTPUT_DLL;
 418                 break;
 419             case TCC_OPTION_soname:
 420                 s->soname = optarg;
 421                 break;
 422 #if defined(TCC_TARGET_X86_64) || defined (TCC_TARGET_I386)
 423             case TCC_OPTION_m:
 424                     exec_other_tcc(s, argc+1, argv-1, optarg, optind);
 425                 break;
 426 #endif
 427             case TCC_OPTION_o:
 428                 multiple_files = 1;
 429                 outfile = optarg;
 430                 break;
 431             case TCC_OPTION_r:
 432                 /* generate a .o merging several output files */
 433                 reloc_output = 1;
 434                 output_type = TCC_OUTPUT_OBJ;
 435                 break;
 436             case TCC_OPTION_isystem:
 437                 tcc_add_sysinclude_path(s, optarg);
 438                 break;
 439             case TCC_OPTION_nostdinc:
 440                 s->nostdinc = 1;
 441                 break;
 442             case TCC_OPTION_nostdlib:
 443                 s->nostdlib = 1;
 444                 break;
 445             case TCC_OPTION_print_search_dirs:
 446                 print_search_dirs = 1;
 447                 break;
 448             case TCC_OPTION_run:
 449                 {
 450                     int argc1;
 451                     char **argv1;
 452                     argc1 = expand_args(&argv1, optarg);
 453                     if (argc1 > 0) {
 454                         parse_args(s, argc1, argv1);
 455                     }
 456                     multiple_files = 0;
 457                     output_type = TCC_OUTPUT_MEMORY;
 458                 }
 459                 break;
 460             case TCC_OPTION_v:
 461                 do {
 462                     if (0 == s->verbose++)
 463                         printf("tcc version %s\n", TCC_VERSION);
 464                 } while (*optarg++ == 'v');
 465                 break;
 466             case TCC_OPTION_f:
 467                 if (tcc_set_flag(s, optarg, 1) < 0 && s->warn_unsupported)
 468                     goto unsupported_option;
 469                 break;
 470             case TCC_OPTION_W:
 471                 if (tcc_set_warning(s, optarg, 1) < 0 &&
 472                     s->warn_unsupported)
 473                     goto unsupported_option;
 474                 break;
 475             case TCC_OPTION_w:
 476                 s->warn_none = 1;
 477                 break;
 478             case TCC_OPTION_rdynamic:
 479                 s->rdynamic = 1;
 480                 break;
 481             case TCC_OPTION_Wl:
 482                 {
 483                     if ((r = (char *) tcc_set_linker(s, (char *)optarg, TRUE)))
 484                         error("unsupported linker option '%s'", r);
 485                 }
 486                 break;
 487             case TCC_OPTION_E:
 488                 output_type = TCC_OUTPUT_PREPROCESS;
 489                 break;
 490             case TCC_OPTION_MD:
 491                 gen_deps = 1;
 492                 break;
 493             case TCC_OPTION_MF:
 494                 deps_outfile = optarg;
 495                 break;
 496             case TCC_OPTION_x:
 497                 break;
 498             default:
 499                 if (s->warn_unsupported) {
 500                 unsupported_option:
 501                     warning("unsupported option '%s'", r);
 502                 }
 503                 break;
 504             }
 505         }
 506     }
 507     /* fixme: these options could be different on your platform */
 508     if (was_pthread
 509         && output_type != TCC_OUTPUT_OBJ)
 510     {
 511         dynarray_add((void ***)&files, &nb_files, "-lpthread");
 512         nb_libraries++;
 513     }
 514     return optind + 1;
 515 }
 516
 517 int main(int argc, char **argv)
 518 {
 519     int i;
 520     TCCState *s;
 521     int nb_objfiles, ret, optind;
 522     int64_t start_time = 0;
 523
 524     s = tcc_new();
 525
 526     output_type = TCC_OUTPUT_EXE;
 527     outfile = NULL;
 528     multiple_files = 1;
 529     files = NULL;
 530     nb_files = 0;
 531     nb_libraries = 0;
 532     reloc_output = 0;
 533     print_search_dirs = 0;
 534     ret = 0;
 535
 536     optind = parse_args(s, argc - 1, argv + 1);
 537     if (print_search_dirs) {
 538         /* enough for Linux kernel */
 539         printf("install: %s/\n", s->tcc_lib_path);
 540         return 0;
 541     }
 542     if (optind == 0 || nb_files == 0) {
 543         if (optind && s->verbose)
 544             return 0;
 545         help();
 546         return 1;
 547     }
 548
 549     nb_objfiles = nb_files - nb_libraries;
 550
 551     /* if outfile provided without other options, we output an
 552        executable */
 553     if (outfile && output_type == TCC_OUTPUT_MEMORY)
 554         output_type = TCC_OUTPUT_EXE;
 555
 556     /* check -c consistency : only single file handled. XXX: checks file type */
 557     if (output_type == TCC_OUTPUT_OBJ && !reloc_output) {
 558         /* accepts only a single input file */
 559         if (nb_objfiles != 1)
 560             error("cannot specify multiple files with -c");
 561         if (nb_libraries != 0)
 562             error("cannot specify libraries with -c");
 563     }
 564
 565
 566     if (output_type == TCC_OUTPUT_PREPROCESS) {
 567         if (!outfile) {
 568             s->outfile = stdout;
 569         } else {
 570             s->outfile = fopen(outfile, "w");
 571             if (!s->outfile)
 572                 error("could not open '%s'", outfile);
 573         }
 574     }
 575
 576     if (do_bench) {
 577         start_time = getclock_us();
 578     }
 579
 580     tcc_set_output_type(s, output_type);
 581     s->reloc_output = reloc_output;
 582
 583     /* compile or add each files or library */
 584     for(i = 0; i < nb_files && ret == 0; i++) {
 585         const char *filename;
 586
 587         filename = files[i];
 588         if (filename[0] == '-' && filename[1] == 'l') {
 589             if (tcc_add_library(s, filename + 2) < 0) {
 590                 error_noabort("cannot find %s", filename);
 591                 ret = 1;
 592             }
 593         } else {
 594             if (1 == s->verbose)
 595                 printf("-> %s\n", filename);
 596             if (tcc_add_file(s, filename) < 0)
 597                 ret = 1;
 598         }
 599     }
 600
 601     /* free all files */
 602     tcc_free(files);
 603
 604     if (0 == ret) {
 605         if (do_bench)
 606             tcc_print_stats(s, getclock_us() - start_time);
 607
 608         if (s->output_type == TCC_OUTPUT_MEMORY)
 609             ret = tcc_run(s, argc - optind, argv + optind);
 610         else {
 611             if (s->output_type == TCC_OUTPUT_PREPROCESS) {
 612                 if (outfile)
 613                     fclose(s->outfile);
 614             } else {
 615                 ret = tcc_output_file(s, outfile ? outfile : tcc_default_target(s));
 616                 ret = ret ? 1 : 0;
 617             }
 618
 619             /* dump collected dependencies */
 620             if (gen_deps && !ret)
 621                 tcc_gen_makedeps(s, outfile, deps_outfile);
 622         }
 623     }
 624
 625     tcc_delete(s);
 626
 627 #ifdef MEM_DEBUG
 628     if (do_bench) {
 629         printf("memory: %d bytes, max = %d bytes\n", mem_cur_size, mem_max_size);
 630     }
 631 #endif
 632     return ret;
 633 }