Avoid crash with "Avoid a crash with weak symbols for "make test""
[tinycc.git] / tcc.c
blob136612e6a5b5041703f60780e342c476fe103194
1 /*
2 * TCC - Tiny C Compiler
3 *
4 * Copyright (c) 2001-2004 Fabrice Bellard
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.
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.
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
21 #if defined NOTALLINONE || defined TCC_USE_LIBTCC
22 #include "tcc.h"
23 #else
24 #include "libtcc.c"
25 #endif
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;
38 #define TCC_OPTION_HAS_ARG 0x0001
39 #define TCC_OPTION_NOSEP 0x0002 /* cannot have space before option and arg */
41 static void help(void)
43 printf("tcc version " TCC_VERSION " - Tiny C Compiler - Copyright (C) 2001-2006 Fabrice Bellard\n"
44 "usage: tcc [-v] [-c] [-o outfile] [-Bdir] [-bench] [-Idir] [-Dsym[=val]] [-Usym]\n"
45 " [-Wwarn] [-g] [-b] [-bt N] [-Ldir] [-llib] [-shared] [-soname name]\n"
46 " [-static] [infile1 infile2...] [-run infile args...]\n"
47 "\n"
48 "General options:\n"
49 " -v display current version, increase verbosity\n"
50 " -c compile only - generate an object file\n"
51 " -o outfile set output filename\n"
52 " -Bdir set tcc internal library path\n"
53 " -bench output compilation statistics\n"
54 " -run run compiled source\n"
55 " -fflag set or reset (with 'no-' prefix) 'flag' (see man page)\n"
56 " -Wwarning set or reset (with 'no-' prefix) 'warning' (see man page)\n"
57 " -w disable all warnings\n"
58 "Preprocessor options:\n"
59 " -E preprocess only\n"
60 " -Idir add include path 'dir'\n"
61 " -Dsym[=val] define 'sym' with value 'val'\n"
62 " -Usym undefine 'sym'\n"
63 "Linker options:\n"
64 " -Ldir add library path 'dir'\n"
65 " -llib link with dynamic or static library 'lib'\n"
66 " -shared generate a shared library\n"
67 " -soname set name for shared library to be used at runtime\n"
68 " -static static linking\n"
69 " -rdynamic export all global symbols to dynamic linker\n"
70 " -r generate (relocatable) object file\n"
71 "Debugger options:\n"
72 " -g generate runtime debug info\n"
73 #ifdef CONFIG_TCC_BCHECK
74 " -b compile with built-in memory and bounds checker (implies -g)\n"
75 #endif
76 #ifdef CONFIG_TCC_BACKTRACE
77 " -bt N show N callers in stack traces\n"
78 #endif
79 "Misc options:\n"
80 " -MD generate target dependencies for make\n"
81 " -MF depfile put generated dependencies here\n"
85 typedef struct TCCOption {
86 const char *name;
87 uint16_t index;
88 uint16_t flags;
89 } TCCOption;
91 enum {
92 TCC_OPTION_HELP,
93 TCC_OPTION_I,
94 TCC_OPTION_D,
95 TCC_OPTION_U,
96 TCC_OPTION_L,
97 TCC_OPTION_B,
98 TCC_OPTION_l,
99 TCC_OPTION_bench,
100 TCC_OPTION_bt,
101 TCC_OPTION_b,
102 TCC_OPTION_g,
103 TCC_OPTION_c,
104 TCC_OPTION_static,
105 TCC_OPTION_shared,
106 TCC_OPTION_soname,
107 TCC_OPTION_o,
108 TCC_OPTION_r,
109 TCC_OPTION_Wl,
110 TCC_OPTION_W,
111 TCC_OPTION_O,
112 TCC_OPTION_m,
113 TCC_OPTION_f,
114 TCC_OPTION_nostdinc,
115 TCC_OPTION_nostdlib,
116 TCC_OPTION_print_search_dirs,
117 TCC_OPTION_rdynamic,
118 TCC_OPTION_run,
119 TCC_OPTION_v,
120 TCC_OPTION_w,
121 TCC_OPTION_pipe,
122 TCC_OPTION_E,
123 TCC_OPTION_MD,
124 TCC_OPTION_MF,
125 TCC_OPTION_x,
128 static const TCCOption tcc_options[] = {
129 { "h", TCC_OPTION_HELP, 0 },
130 { "-help", TCC_OPTION_HELP, 0 },
131 { "?", TCC_OPTION_HELP, 0 },
132 { "I", TCC_OPTION_I, TCC_OPTION_HAS_ARG },
133 { "D", TCC_OPTION_D, TCC_OPTION_HAS_ARG },
134 { "U", TCC_OPTION_U, TCC_OPTION_HAS_ARG },
135 { "L", TCC_OPTION_L, TCC_OPTION_HAS_ARG },
136 { "B", TCC_OPTION_B, TCC_OPTION_HAS_ARG },
137 { "l", TCC_OPTION_l, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
138 { "bench", TCC_OPTION_bench, 0 },
139 { "bt", TCC_OPTION_bt, TCC_OPTION_HAS_ARG },
140 #ifdef CONFIG_TCC_BCHECK
141 { "b", TCC_OPTION_b, 0 },
142 #endif
143 { "g", TCC_OPTION_g, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
144 { "c", TCC_OPTION_c, 0 },
145 { "static", TCC_OPTION_static, 0 },
146 { "shared", TCC_OPTION_shared, 0 },
147 { "soname", TCC_OPTION_soname, TCC_OPTION_HAS_ARG },
148 { "o", TCC_OPTION_o, TCC_OPTION_HAS_ARG },
149 { "run", TCC_OPTION_run, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
150 { "rdynamic", TCC_OPTION_rdynamic, 0 },
151 { "r", TCC_OPTION_r, 0 },
152 { "Wl,", TCC_OPTION_Wl, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
153 { "W", TCC_OPTION_W, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
154 { "O", TCC_OPTION_O, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
155 { "m", TCC_OPTION_m, TCC_OPTION_HAS_ARG },
156 { "f", TCC_OPTION_f, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
157 { "nostdinc", TCC_OPTION_nostdinc, 0 },
158 { "nostdlib", TCC_OPTION_nostdlib, 0 },
159 { "print-search-dirs", TCC_OPTION_print_search_dirs, 0 },
160 { "v", TCC_OPTION_v, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
161 { "w", TCC_OPTION_w, 0 },
162 { "pipe", TCC_OPTION_pipe, 0},
163 { "E", TCC_OPTION_E, 0},
164 { "MD", TCC_OPTION_MD, 0},
165 { "MF", TCC_OPTION_MF, TCC_OPTION_HAS_ARG },
166 { "x", TCC_OPTION_x, TCC_OPTION_HAS_ARG },
167 { NULL },
170 static int64_t getclock_us(void)
172 #ifdef _WIN32
173 struct _timeb tb;
174 _ftime(&tb);
175 return (tb.time * 1000LL + tb.millitm) * 1000LL;
176 #else
177 struct timeval tv;
178 gettimeofday(&tv, NULL);
179 return tv.tv_sec * 1000000LL + tv.tv_usec;
180 #endif
183 /* convert 'str' into an array of space separated strings */
184 static int expand_args(char ***pargv, const char *str)
186 const char *s1;
187 char **argv, *arg;
188 int argc, len;
190 argc = 0;
191 argv = NULL;
192 for(;;) {
193 while (is_space(*str))
194 str++;
195 if (*str == '\0')
196 break;
197 s1 = str;
198 while (*str != '\0' && !is_space(*str))
199 str++;
200 len = str - s1;
201 arg = tcc_malloc(len + 1);
202 memcpy(arg, s1, len);
203 arg[len] = '\0';
204 dynarray_add((void ***)&argv, &argc, arg);
206 *pargv = argv;
207 return argc;
210 static int parse_args(TCCState *s, int argc, char **argv)
212 int optind;
213 const TCCOption *popt;
214 const char *optarg, *p1, *r1;
215 char *r;
217 optind = 0;
218 while (optind < argc) {
220 r = argv[optind++];
221 if (r[0] != '-' || r[1] == '\0') {
222 /* add a new file */
223 dynarray_add((void ***)&files, &nb_files, r);
224 if (!multiple_files) {
225 optind--;
226 /* argv[0] will be this file */
227 break;
229 } else {
230 /* find option in table (match only the first chars */
231 popt = tcc_options;
232 for(;;) {
233 p1 = popt->name;
234 if (p1 == NULL)
235 error("invalid option -- '%s'", r);
236 r1 = r + 1;
237 for(;;) {
238 if (*p1 == '\0')
239 goto option_found;
240 if (*r1 != *p1)
241 break;
242 p1++;
243 r1++;
245 popt++;
247 option_found:
248 if (popt->flags & TCC_OPTION_HAS_ARG) {
249 if (*r1 != '\0' || (popt->flags & TCC_OPTION_NOSEP)) {
250 optarg = r1;
251 } else {
252 if (optind >= argc)
253 error("argument to '%s' is missing", r);
254 optarg = argv[optind++];
256 } else {
257 if (*r1 != '\0')
258 return 0;
259 optarg = NULL;
262 switch(popt->index) {
263 case TCC_OPTION_HELP:
264 return 0;
266 case TCC_OPTION_I:
267 if (tcc_add_include_path(s, optarg) < 0)
268 error("too many include paths");
269 break;
270 case TCC_OPTION_D:
272 char *sym, *value;
273 sym = (char *)optarg;
274 value = strchr(sym, '=');
275 if (value) {
276 *value = '\0';
277 value++;
279 tcc_define_symbol(s, sym, value);
281 break;
282 case TCC_OPTION_U:
283 tcc_undefine_symbol(s, optarg);
284 break;
285 case TCC_OPTION_L:
286 tcc_add_library_path(s, optarg);
287 break;
288 case TCC_OPTION_B:
289 /* set tcc utilities path (mainly for tcc development) */
290 tcc_set_lib_path(s, optarg);
291 break;
292 case TCC_OPTION_l:
293 dynarray_add((void ***)&files, &nb_files, r);
294 nb_libraries++;
295 break;
296 case TCC_OPTION_bench:
297 do_bench = 1;
298 break;
299 #ifdef CONFIG_TCC_BACKTRACE
300 case TCC_OPTION_bt:
301 set_num_callers(atoi(optarg));
302 break;
303 #endif
304 #ifdef CONFIG_TCC_BCHECK
305 case TCC_OPTION_b:
306 s->do_bounds_check = 1;
307 s->do_debug = 1;
308 break;
309 #endif
310 case TCC_OPTION_g:
311 s->do_debug = 1;
312 break;
313 case TCC_OPTION_c:
314 multiple_files = 1;
315 output_type = TCC_OUTPUT_OBJ;
316 break;
317 case TCC_OPTION_static:
318 s->static_link = 1;
319 break;
320 case TCC_OPTION_shared:
321 output_type = TCC_OUTPUT_DLL;
322 break;
323 case TCC_OPTION_soname:
324 s->soname = optarg;
325 break;
326 case TCC_OPTION_o:
327 multiple_files = 1;
328 outfile = optarg;
329 break;
330 case TCC_OPTION_r:
331 /* generate a .o merging several output files */
332 reloc_output = 1;
333 output_type = TCC_OUTPUT_OBJ;
334 break;
335 case TCC_OPTION_nostdinc:
336 s->nostdinc = 1;
337 break;
338 case TCC_OPTION_nostdlib:
339 s->nostdlib = 1;
340 break;
341 case TCC_OPTION_print_search_dirs:
342 print_search_dirs = 1;
343 break;
344 case TCC_OPTION_run:
346 int argc1;
347 char **argv1;
348 argc1 = expand_args(&argv1, optarg);
349 if (argc1 > 0) {
350 parse_args(s, argc1, argv1);
352 multiple_files = 0;
353 output_type = TCC_OUTPUT_MEMORY;
355 break;
356 case TCC_OPTION_v:
357 do {
358 if (0 == s->verbose++)
359 printf("tcc version %s\n", TCC_VERSION);
360 } while (*optarg++ == 'v');
361 break;
362 case TCC_OPTION_f:
363 if (tcc_set_flag(s, optarg, 1) < 0 && s->warn_unsupported)
364 goto unsupported_option;
365 break;
366 case TCC_OPTION_W:
367 if (tcc_set_warning(s, optarg, 1) < 0 &&
368 s->warn_unsupported)
369 goto unsupported_option;
370 break;
371 case TCC_OPTION_w:
372 s->warn_none = 1;
373 break;
374 case TCC_OPTION_rdynamic:
375 s->rdynamic = 1;
376 break;
377 case TCC_OPTION_Wl:
379 if ((r = (char *) tcc_set_linker(s, (char *)optarg, TRUE)))
380 error("unsupported linker option '%s'", r);
382 break;
383 case TCC_OPTION_E:
384 output_type = TCC_OUTPUT_PREPROCESS;
385 break;
386 case TCC_OPTION_MD:
387 gen_deps = 1;
388 break;
389 case TCC_OPTION_MF:
390 deps_outfile = optarg;
391 break;
392 case TCC_OPTION_x:
393 break;
394 default:
395 if (s->warn_unsupported) {
396 unsupported_option:
397 warning("unsupported option '%s'", r);
399 break;
403 return optind + 1;
406 int main(int argc, char **argv)
408 int i;
409 TCCState *s;
410 int nb_objfiles, ret, optind;
411 int64_t start_time = 0;
413 s = tcc_new();
415 output_type = TCC_OUTPUT_EXE;
416 outfile = NULL;
417 multiple_files = 1;
418 files = NULL;
419 nb_files = 0;
420 nb_libraries = 0;
421 reloc_output = 0;
422 print_search_dirs = 0;
423 ret = 0;
425 optind = parse_args(s, argc - 1, argv + 1);
426 if (print_search_dirs) {
427 /* enough for Linux kernel */
428 printf("install: %s/\n", s->tcc_lib_path);
429 return 0;
431 if (optind == 0 || nb_files == 0) {
432 if (optind && s->verbose)
433 return 0;
434 help();
435 return 1;
438 nb_objfiles = nb_files - nb_libraries;
440 /* if outfile provided without other options, we output an
441 executable */
442 if (outfile && output_type == TCC_OUTPUT_MEMORY)
443 output_type = TCC_OUTPUT_EXE;
445 /* check -c consistency : only single file handled. XXX: checks file type */
446 if (output_type == TCC_OUTPUT_OBJ && !reloc_output) {
447 /* accepts only a single input file */
448 if (nb_objfiles != 1)
449 error("cannot specify multiple files with -c");
450 if (nb_libraries != 0)
451 error("cannot specify libraries with -c");
455 if (output_type == TCC_OUTPUT_PREPROCESS) {
456 if (!outfile) {
457 s->outfile = stdout;
458 } else {
459 s->outfile = fopen(outfile, "w");
460 if (!s->outfile)
461 error("could not open '%s'", outfile);
465 if (do_bench) {
466 start_time = getclock_us();
469 tcc_set_output_type(s, output_type);
470 s->reloc_output = reloc_output;
472 /* compile or add each files or library */
473 for(i = 0; i < nb_files && ret == 0; i++) {
474 const char *filename;
476 filename = files[i];
477 if (filename[0] == '-' && filename[1] == 'l') {
478 if (tcc_add_library(s, filename + 2) < 0) {
479 error_noabort("cannot find %s", filename);
480 ret = 1;
482 } else {
483 if (1 == s->verbose)
484 printf("-> %s\n", filename);
485 if (tcc_add_file(s, filename) < 0)
486 ret = 1;
490 /* free all files */
491 tcc_free(files);
493 if (0 == ret) {
494 if (do_bench)
495 tcc_print_stats(s, getclock_us() - start_time);
497 if (s->output_type == TCC_OUTPUT_MEMORY)
498 ret = tcc_run(s, argc - optind, argv + optind);
499 else {
500 if (s->output_type == TCC_OUTPUT_PREPROCESS) {
501 if (outfile)
502 fclose(s->outfile);
503 } else {
504 ret = tcc_output_file(s, outfile ? outfile : tcc_default_target(s));
505 ret = ret ? 1 : 0;
508 /* dump collected dependencies */
509 if (gen_deps && !ret)
510 tcc_gen_makedeps(s, outfile, deps_outfile);
514 tcc_delete(s);
516 #ifdef MEM_DEBUG
517 if (do_bench) {
518 printf("memory: %d bytes, max = %d bytes\n", mem_cur_size, mem_max_size);
520 #endif
521 return ret;