fix "tcc test.c -Wl,,--oformat,binary"
[tinycc.git] / libtcc.c
Commit [+]AuthorDateLineData
a93bcdff grischka2009-05-05 20:17:11 +02001/*
2 * TCC - Tiny C Compiler
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06003 *
a93bcdff grischka2009-05-05 20:17:11 +02004 * 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
9dc9cbf3 grischka2009-05-05 20:18:53 +020021#include "tcc.h"
22
0a35f9d6 grischka2009-05-11 18:45:56 +020023/********************************************************/
24/* global variables */
25
92204e88 grischka2009-05-05 20:30:13 +020026/* use GNU C extensions */
88a3ccab grischka2009-12-20 01:53:49 +010027ST_DATA int gnu_ext = 1;
92204e88 grischka2009-05-05 20:30:13 +020028
88a3ccab grischka2009-12-20 01:53:49 +010029/* use TinyCC extensions */
30ST_DATA int tcc_ext = 1;
92204e88 grischka2009-05-05 20:30:13 +020031
32/* XXX: get rid of this ASAP */
88a3ccab grischka2009-12-20 01:53:49 +010033ST_DATA struct TCCState *tcc_state;
0a35f9d6 grischka2009-05-11 18:45:56 +020034
5e83b64d grischka2009-07-06 21:13:20 +020035/********************************************************/
0a35f9d6 grischka2009-05-11 18:45:56 +020036
5280293d grischka2011-07-14 18:45:37 +020037#ifdef ONE_SOURCE
88a3ccab grischka2009-12-20 01:53:49 +010038#include "tccpp.c"
39#include "tccgen.c"
40#include "tccelf.c"
41#include "tccrun.c"
9dc9cbf3 grischka2009-05-05 20:18:53 +020042#ifdef TCC_TARGET_I386
43#include "i386-gen.c"
44#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +020045#ifdef TCC_TARGET_ARM
46#include "arm-gen.c"
47#endif
b14ef0e2
EGE
Edmund Grimley Evans2015-02-13 18:58:31 +000048#ifdef TCC_TARGET_ARM64
49#include "arm64-gen.c"
50#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +020051#ifdef TCC_TARGET_C67
52#include "c67-gen.c"
53#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +020054#ifdef TCC_TARGET_X86_64
55#include "x86_64-gen.c"
56#endif
5e83b64d grischka2009-07-06 21:13:20 +020057#ifdef CONFIG_TCC_ASM
88a3ccab grischka2009-12-20 01:53:49 +010058#include "tccasm.c"
1308e8eb grischka2009-12-19 22:08:37 +010059#if defined TCC_TARGET_I386 || defined TCC_TARGET_X86_64
5e83b64d grischka2009-07-06 21:13:20 +020060#include "i386-asm.c"
61#endif
5e83b64d grischka2009-07-06 21:13:20 +020062#endif
5e83b64d grischka2009-07-06 21:13:20 +020063#ifdef TCC_TARGET_COFF
64#include "tcccoff.c"
65#endif
5e83b64d grischka2009-07-06 21:13:20 +020066#ifdef TCC_TARGET_PE
67#include "tccpe.c"
68#endif
339f961c Joe Soroka2011-07-22 02:12:49 -070069#endif /* ONE_SOURCE */
5e83b64d grischka2009-07-06 21:13:20 +020070
88a3ccab grischka2009-12-20 01:53:49 +010071/********************************************************/
72#ifndef CONFIG_TCC_ASM
73ST_FUNC void asm_instr(void)
74{
bf374a5f grischka2011-08-11 17:07:56 +020075 tcc_error("inline asm() not supported");
88a3ccab grischka2009-12-20 01:53:49 +010076}
77ST_FUNC void asm_global_instr(void)
78{
bf374a5f grischka2011-08-11 17:07:56 +020079 tcc_error("inline asm() not supported");
88a3ccab grischka2009-12-20 01:53:49 +010080}
81#endif
3db21947 grischka2009-12-19 22:10:26 +010082
5e83b64d grischka2009-07-06 21:13:20 +020083/********************************************************/
1df662c1 grischka2009-07-18 22:07:25 +020084#ifdef _WIN32
88a3ccab grischka2009-12-20 01:53:49 +010085static char *normalize_slashes(char *path)
1df662c1 grischka2009-07-18 22:07:25 +020086{
87 char *p;
73faaea2 grischka2013-08-28 22:55:05 +020088 for (p = path; *p; ++p)
89 if (*p == '\\')
90 *p = '/';
1df662c1 grischka2009-07-18 22:07:25 +020091 return path;
92}
93
88a3ccab grischka2009-12-20 01:53:49 +010094static HMODULE tcc_module;
1df662c1 grischka2009-07-18 22:07:25 +020095
96/* on win32, we suppose the lib and includes are at the location of 'tcc.exe' */
88a3ccab grischka2009-12-20 01:53:49 +010097static void tcc_set_lib_path_w32(TCCState *s)
1df662c1 grischka2009-07-18 22:07:25 +020098{
99 char path[1024], *p;
100 GetModuleFileNameA(tcc_module, path, sizeof path);
101 p = tcc_basename(normalize_slashes(strlwr(path)));
102 if (p - 5 > path && 0 == strncmp(p - 5, "/bin/", 5))
103 p -= 5;
104 else if (p > path)
105 p--;
106 *p = 0;
107 tcc_set_lib_path(s, path);
108}
109
7a477d70 grischka2013-02-06 19:01:07 +0100110#ifdef TCC_TARGET_PE
df9cce24 grischka2011-08-01 01:10:36 +0200111static void tcc_add_systemdir(TCCState *s)
112{
113 char buf[1000];
114 GetSystemDirectory(buf, sizeof buf);
74a24d77 grischka2011-08-11 16:55:30 +0200115 tcc_add_library_path(s, normalize_slashes(buf));
df9cce24 grischka2011-08-01 01:10:36 +0200116}
7a477d70 grischka2013-02-06 19:01:07 +0100117#endif
df9cce24 grischka2011-08-01 01:10:36 +0200118
0de95730 grischka2009-12-20 20:33:41 +0100119#ifndef CONFIG_TCC_STATIC
120void dlclose(void *p)
121{
122 FreeLibrary((HMODULE)p);
123}
124#endif
125
1df662c1 grischka2009-07-18 22:07:25 +0200126#ifdef LIBTCC_AS_DLL
df0267b2 minux2014-04-11 23:49:53 -0400127BOOL WINAPI DllMain (HINSTANCE hDll, DWORD dwReason, LPVOID lpReserved)
1df662c1 grischka2009-07-18 22:07:25 +0200128{
129 if (DLL_PROCESS_ATTACH == dwReason)
130 tcc_module = hDll;
131 return TRUE;
132}
133#endif
134#endif
135
136/********************************************************/
9dc9cbf3 grischka2009-05-05 20:18:53 +0200137/* copy a string and truncate it. */
88a3ccab grischka2009-12-20 01:53:49 +0100138PUB_FUNC char *pstrcpy(char *buf, int buf_size, const char *s)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200139{
140 char *q, *q_end;
141 int c;
142
143 if (buf_size > 0) {
144 q = buf;
145 q_end = buf + buf_size - 1;
146 while (q < q_end) {
147 c = *s++;
148 if (c == '\0')
149 break;
150 *q++ = c;
151 }
152 *q = '\0';
153 }
154 return buf;
155}
156
157/* strcat and truncate. */
88a3ccab grischka2009-12-20 01:53:49 +0100158PUB_FUNC char *pstrcat(char *buf, int buf_size, const char *s)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200159{
160 int len;
161 len = strlen(buf);
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600162 if (len < buf_size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200163 pstrcpy(buf + len, buf_size - len, s);
164 return buf;
165}
166
df9cce24 grischka2011-08-01 01:10:36 +0200167PUB_FUNC char *pstrncpy(char *out, const char *in, size_t num)
168{
169 memcpy(out, in, num);
170 out[num] = '\0';
171 return out;
172}
173
9dc9cbf3 grischka2009-05-05 20:18:53 +0200174/* extract the basename of a file */
88a3ccab grischka2009-12-20 01:53:49 +0100175PUB_FUNC char *tcc_basename(const char *name)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200176{
177 char *p = strchr(name, 0);
df9cce24 grischka2011-08-01 01:10:36 +0200178 while (p > name && !IS_DIRSEP(p[-1]))
9dc9cbf3 grischka2009-05-05 20:18:53 +0200179 --p;
180 return p;
181}
182
441a089a
KS
Kirill Smelkov2010-06-20 19:17:42 +0400183/* extract extension part of a file
184 *
185 * (if no extension, return pointer to end-of-string)
186 */
88a3ccab grischka2009-12-20 01:53:49 +0100187PUB_FUNC char *tcc_fileextension (const char *name)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200188{
189 char *b = tcc_basename(name);
190 char *e = strrchr(b, '.');
191 return e ? e : strchr(b, 0);
192}
193
88a3ccab grischka2009-12-20 01:53:49 +0100194/********************************************************/
9dc9cbf3 grischka2009-05-05 20:18:53 +0200195/* memory management */
88a3ccab grischka2009-12-20 01:53:49 +0100196
197#undef free
198#undef malloc
199#undef realloc
200
9dc9cbf3 grischka2009-05-05 20:18:53 +0200201#ifdef MEM_DEBUG
05108a3b grischka2013-02-12 19:13:28 +0100202ST_DATA int mem_cur_size;
203ST_DATA int mem_max_size;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200204unsigned malloc_usable_size(void*);
205#endif
206
88a3ccab grischka2009-12-20 01:53:49 +0100207PUB_FUNC void tcc_free(void *ptr)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200208{
209#ifdef MEM_DEBUG
210 mem_cur_size -= malloc_usable_size(ptr);
211#endif
212 free(ptr);
213}
214
88a3ccab grischka2009-12-20 01:53:49 +0100215PUB_FUNC void *tcc_malloc(unsigned long size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200216{
217 void *ptr;
218 ptr = malloc(size);
219 if (!ptr && size)
2bd0daab grischka2014-01-06 19:56:26 +0100220 tcc_error("memory full (malloc)");
9dc9cbf3 grischka2009-05-05 20:18:53 +0200221#ifdef MEM_DEBUG
222 mem_cur_size += malloc_usable_size(ptr);
223 if (mem_cur_size > mem_max_size)
224 mem_max_size = mem_cur_size;
225#endif
226 return ptr;
227}
228
88a3ccab grischka2009-12-20 01:53:49 +0100229PUB_FUNC void *tcc_mallocz(unsigned long size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200230{
231 void *ptr;
232 ptr = tcc_malloc(size);
233 memset(ptr, 0, size);
234 return ptr;
235}
236
88a3ccab grischka2009-12-20 01:53:49 +0100237PUB_FUNC void *tcc_realloc(void *ptr, unsigned long size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200238{
239 void *ptr1;
240#ifdef MEM_DEBUG
241 mem_cur_size -= malloc_usable_size(ptr);
242#endif
243 ptr1 = realloc(ptr, size);
f1b5c2ef grischka2012-04-18 18:43:55 +0200244 if (!ptr1 && size)
2bd0daab grischka2014-01-06 19:56:26 +0100245 tcc_error("memory full (realloc)");
9dc9cbf3 grischka2009-05-05 20:18:53 +0200246#ifdef MEM_DEBUG
247 /* NOTE: count not correct if alloc error, but not critical */
248 mem_cur_size += malloc_usable_size(ptr1);
249 if (mem_cur_size > mem_max_size)
250 mem_max_size = mem_cur_size;
251#endif
252 return ptr1;
253}
254
88a3ccab grischka2009-12-20 01:53:49 +0100255PUB_FUNC char *tcc_strdup(const char *str)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200256{
257 char *ptr;
258 ptr = tcc_malloc(strlen(str) + 1);
259 strcpy(ptr, str);
260 return ptr;
261}
262
88a3ccab grischka2009-12-20 01:53:49 +0100263PUB_FUNC void tcc_memstats(void)
3db21947 grischka2009-12-19 22:10:26 +0100264{
265#ifdef MEM_DEBUG
05108a3b grischka2013-02-12 19:13:28 +0100266 printf("memory: %d bytes, max = %d bytes\n", mem_cur_size, mem_max_size);
3db21947 grischka2009-12-19 22:10:26 +0100267#endif
268}
269
9dc9cbf3 grischka2009-05-05 20:18:53 +0200270#define free(p) use_tcc_free(p)
271#define malloc(s) use_tcc_malloc(s)
272#define realloc(p, s) use_tcc_realloc(p, s)
273
88a3ccab grischka2009-12-20 01:53:49 +0100274/********************************************************/
275/* dynarrays */
276
05108a3b grischka2013-02-12 19:13:28 +0100277ST_FUNC void dynarray_add(void ***ptab, int *nb_ptr, void *data)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200278{
279 int nb, nb_alloc;
280 void **pp;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600281
9dc9cbf3 grischka2009-05-05 20:18:53 +0200282 nb = *nb_ptr;
283 pp = *ptab;
284 /* every power of two we double array size */
285 if ((nb & (nb - 1)) == 0) {
286 if (!nb)
287 nb_alloc = 1;
288 else
289 nb_alloc = nb * 2;
290 pp = tcc_realloc(pp, nb_alloc * sizeof(void *));
9dc9cbf3 grischka2009-05-05 20:18:53 +0200291 *ptab = pp;
292 }
293 pp[nb++] = data;
294 *nb_ptr = nb;
295}
296
05108a3b grischka2013-02-12 19:13:28 +0100297ST_FUNC void dynarray_reset(void *pp, int *n)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200298{
299 void **p;
300 for (p = *(void***)pp; *n; ++p, --*n)
301 if (*p)
302 tcc_free(*p);
303 tcc_free(*(void**)pp);
304 *(void**)pp = NULL;
305}
306
df9cce24 grischka2011-08-01 01:10:36 +0200307static void tcc_split_path(TCCState *s, void ***p_ary, int *p_nb_ary, const char *in)
13121e22 Thomas Preud'homme2011-07-08 10:51:26 +0200308{
df9cce24 grischka2011-08-01 01:10:36 +0200309 const char *p;
13121e22 Thomas Preud'homme2011-07-08 10:51:26 +0200310 do {
df9cce24 grischka2011-08-01 01:10:36 +0200311 int c;
312 CString str;
313
314 cstr_new(&str);
81cd0cf6 grischka2011-08-06 16:11:12 +0200315 for (p = in; c = *p, c != '\0' && c != PATHSEP; ++p) {
74a24d77 grischka2011-08-11 16:55:30 +0200316 if (c == '{' && p[1] && p[2] == '}') {
317 c = p[1], p += 2;
318 if (c == 'B')
319 cstr_cat(&str, s->tcc_lib_path);
81cd0cf6 grischka2011-08-06 16:11:12 +0200320 } else {
321 cstr_ccat(&str, c);
df9cce24 grischka2011-08-01 01:10:36 +0200322 }
13121e22 Thomas Preud'homme2011-07-08 10:51:26 +0200323 }
81cd0cf6 grischka2011-08-06 16:11:12 +0200324 cstr_ccat(&str, '\0');
df9cce24 grischka2011-08-01 01:10:36 +0200325 dynarray_add(p_ary, p_nb_ary, str.data);
81cd0cf6 grischka2011-08-06 16:11:12 +0200326 in = p+1;
327 } while (*p);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200328}
329
88a3ccab grischka2009-12-20 01:53:49 +0100330/********************************************************/
9dc9cbf3 grischka2009-05-05 20:18:53 +0200331
88a3ccab grischka2009-12-20 01:53:49 +0100332ST_FUNC Section *new_section(TCCState *s1, const char *name, int sh_type, int sh_flags)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200333{
334 Section *sec;
335
700c2f76 mingodad2014-03-28 20:25:39 +0000336 sec = tcc_mallocz(sizeof(Section) + strlen(name));
9dc9cbf3 grischka2009-05-05 20:18:53 +0200337 strcpy(sec->name, name);
338 sec->sh_type = sh_type;
339 sec->sh_flags = sh_flags;
340 switch(sh_type) {
341 case SHT_HASH:
342 case SHT_REL:
343 case SHT_RELA:
344 case SHT_DYNSYM:
345 case SHT_SYMTAB:
346 case SHT_DYNAMIC:
347 sec->sh_addralign = 4;
348 break;
349 case SHT_STRTAB:
350 sec->sh_addralign = 1;
351 break;
352 default:
353 sec->sh_addralign = 32; /* default conservative alignment */
354 break;
355 }
356
357 if (sh_flags & SHF_PRIVATE) {
358 dynarray_add((void ***)&s1->priv_sections, &s1->nb_priv_sections, sec);
359 } else {
360 sec->sh_num = s1->nb_sections;
361 dynarray_add((void ***)&s1->sections, &s1->nb_sections, sec);
362 }
363
364 return sec;
365}
366
367static void free_section(Section *s)
368{
369 tcc_free(s->data);
370}
371
372/* realloc section and set its content to zero */
88a3ccab grischka2009-12-20 01:53:49 +0100373ST_FUNC void section_realloc(Section *sec, unsigned long new_size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200374{
375 unsigned long size;
376 unsigned char *data;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600377
9dc9cbf3 grischka2009-05-05 20:18:53 +0200378 size = sec->data_allocated;
379 if (size == 0)
380 size = 1;
381 while (size < new_size)
382 size = size * 2;
383 data = tcc_realloc(sec->data, size);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200384 memset(data + sec->data_allocated, 0, size - sec->data_allocated);
385 sec->data = data;
386 sec->data_allocated = size;
387}
388
389/* reserve at least 'size' bytes in section 'sec' from
390 sec->data_offset. */
3b7f5008 seyko2015-03-26 11:28:11 +0300391ST_FUNC void *section_ptr_add(Section *sec, addr_t size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200392{
acef4ff2 seyko2015-03-26 07:47:45 +0300393 size_t offset, offset1;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200394
395 offset = sec->data_offset;
396 offset1 = offset + size;
397 if (offset1 > sec->data_allocated)
398 section_realloc(sec, offset1);
399 sec->data_offset = offset1;
400 return sec->data + offset;
401}
402
d63ec6f2
AGR
Ali Gholami Rudi2010-02-04 23:08:01 +0330403/* reserve at least 'size' bytes from section start */
404ST_FUNC void section_reserve(Section *sec, unsigned long size)
405{
406 if (size > sec->data_allocated)
407 section_realloc(sec, size);
408 if (size > sec->data_offset)
409 sec->data_offset = size;
410}
411
9dc9cbf3 grischka2009-05-05 20:18:53 +0200412/* return a reference to a section, and create it if it does not
413 exists */
88a3ccab grischka2009-12-20 01:53:49 +0100414ST_FUNC Section *find_section(TCCState *s1, const char *name)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200415{
416 Section *sec;
417 int i;
418 for(i = 1; i < s1->nb_sections; i++) {
419 sec = s1->sections[i];
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600420 if (!strcmp(name, sec->name))
9dc9cbf3 grischka2009-05-05 20:18:53 +0200421 return sec;
422 }
423 /* sections are created as PROGBITS */
424 return new_section(s1, name, SHT_PROGBITS, SHF_ALLOC);
425}
426
9dc9cbf3 grischka2009-05-05 20:18:53 +0200427/* update sym->c so that it points to an external symbol in section
428 'section' with value 'value' */
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600429ST_FUNC void put_extern_sym2(Sym *sym, Section *section,
82bcbd02 grischka2013-02-04 16:08:06 +0100430 addr_t value, unsigned long size,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200431 int can_add_underscore)
432{
2341ee51 grischka2010-01-14 20:55:51 +0100433 int sym_type, sym_bind, sh_num, info, other;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200434 ElfW(Sym) *esym;
435 const char *name;
436 char buf1[256];
e7a60e4d seyko2015-04-12 04:47:15 +0300437
438#ifdef CONFIG_TCC_BCHECK
4bb9dd44 seyko2015-04-11 16:22:34 +0300439 char buf[32];
e7a60e4d seyko2015-04-12 04:47:15 +0300440#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +0200441
442 if (section == NULL)
443 sh_num = SHN_UNDEF;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600444 else if (section == SECTION_ABS)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200445 sh_num = SHN_ABS;
446 else
447 sh_num = section->sh_num;
448
9dc9cbf3 grischka2009-05-05 20:18:53 +0200449 if ((sym->type.t & VT_BTYPE) == VT_FUNC) {
450 sym_type = STT_FUNC;
b5e22384 grischka2009-12-19 19:12:33 +0100451 } else if ((sym->type.t & VT_BTYPE) == VT_VOID) {
452 sym_type = STT_NOTYPE;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200453 } else {
cf02f920 Thomas Preud'homme2013-11-03 18:55:54 +0800454 sym_type = STT_OBJECT;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200455 }
456
457 if (sym->type.t & VT_STATIC)
458 sym_bind = STB_LOCAL;
95b9a477 Manuel Simoni2010-02-27 17:37:59 +0100459 else {
4d5105c8 Joe Soroka2011-02-01 00:37:53 -0800460 if (sym->type.t & VT_WEAK)
95b9a477
MS
Manuel Simoni2010-02-27 17:37:59 +0100461 sym_bind = STB_WEAK;
462 else
463 sym_bind = STB_GLOBAL;
464 }
9dc9cbf3 grischka2009-05-05 20:18:53 +0200465
466 if (!sym->c) {
32a682b8 Thomas Preud'homme2011-02-07 22:42:38 +0100467 name = get_tok_str(sym->v, NULL);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200468#ifdef CONFIG_TCC_BCHECK
f9181416 grischka2009-05-11 18:45:44 +0200469 if (tcc_state->do_bounds_check) {
9dc9cbf3 grischka2009-05-05 20:18:53 +0200470 /* XXX: avoid doing that for statics ? */
471 /* if bound checking is activated, we change some function
472 names by adding the "__bound" prefix */
473 switch(sym->v) {
7fa712e0 grischka2009-12-19 22:22:43 +0100474#ifdef TCC_TARGET_PE
9dc9cbf3 grischka2009-05-05 20:18:53 +0200475 /* XXX: we rely only on malloc hooks */
0650ab01
JP
Joseph Poirier2013-11-08 13:24:15 -0600476 case TOK_malloc:
477 case TOK_free:
478 case TOK_realloc:
479 case TOK_memalign:
480 case TOK_calloc:
9dc9cbf3 grischka2009-05-05 20:18:53 +0200481#endif
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600482 case TOK_memcpy:
9dc9cbf3 grischka2009-05-05 20:18:53 +0200483 case TOK_memmove:
484 case TOK_memset:
485 case TOK_strlen:
486 case TOK_strcpy:
110a4edc grischka2009-05-16 22:30:13 +0200487 case TOK_alloca:
9dc9cbf3 grischka2009-05-05 20:18:53 +0200488 strcpy(buf, "__bound_");
489 strcat(buf, name);
490 name = buf;
491 break;
492 }
493 }
494#endif
2341ee51 grischka2010-01-14 20:55:51 +0100495 other = 0;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200496
497#ifdef TCC_TARGET_PE
2341ee51 grischka2010-01-14 20:55:51 +0100498 if (sym->type.t & VT_EXPORT)
2ac238fc grischka2014-04-17 17:01:28 +0200499 other |= ST_PE_EXPORT;
2341ee51 grischka2010-01-14 20:55:51 +0100500 if (sym_type == STT_FUNC && sym->type.ref) {
3fe2a95d grischka2014-01-07 14:57:07 +0100501 Sym *ref = sym->type.ref;
502 if (ref->a.func_export)
2ac238fc grischka2014-04-17 17:01:28 +0200503 other |= ST_PE_EXPORT;
3fe2a95d grischka2014-01-07 14:57:07 +0100504 if (ref->a.func_call == FUNC_STDCALL && can_add_underscore) {
505 sprintf(buf1, "_%s@%d", name, ref->a.func_args * PTR_SIZE);
2341ee51 grischka2010-01-14 20:55:51 +0100506 name = buf1;
2ac238fc grischka2014-04-17 17:01:28 +0200507 other |= ST_PE_STDCALL;
2341ee51 grischka2010-01-14 20:55:51 +0100508 can_add_underscore = 0;
509 }
510 } else {
511 if (find_elf_sym(tcc_state->dynsymtab_section, name))
2ac238fc grischka2014-04-17 17:01:28 +0200512 other |= ST_PE_IMPORT;
2341ee51 grischka2010-01-14 20:55:51 +0100513 if (sym->type.t & VT_IMPORT)
2ac238fc grischka2014-04-17 17:01:28 +0200514 other |= ST_PE_IMPORT;
2341ee51 grischka2010-01-14 20:55:51 +0100515 }
a9fda392
MM
Michael Matz2014-04-14 03:33:50 +0200516#else
517 if (! (sym->type.t & VT_STATIC))
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -0400518 other = (sym->type.t & VT_VIS_MASK) >> VT_VIS_SHIFT;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200519#endif
520 if (tcc_state->leading_underscore && can_add_underscore) {
521 buf1[0] = '_';
522 pstrcpy(buf1 + 1, sizeof(buf1) - 1, name);
523 name = buf1;
524 }
32a682b8
TP
Thomas Preud'homme2011-02-07 22:42:38 +0100525 if (sym->asm_label) {
526 name = sym->asm_label;
527 }
9dc9cbf3 grischka2009-05-05 20:18:53 +0200528 info = ELFW(ST_INFO)(sym_bind, sym_type);
529 sym->c = add_elf_sym(symtab_section, value, size, info, other, sh_num, name);
530 } else {
531 esym = &((ElfW(Sym) *)symtab_section->data)[sym->c];
532 esym->st_value = value;
533 esym->st_size = size;
534 esym->st_shndx = sh_num;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200535 }
536}
537
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600538ST_FUNC void put_extern_sym(Sym *sym, Section *section,
82bcbd02 grischka2013-02-04 16:08:06 +0100539 addr_t value, unsigned long size)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200540{
541 put_extern_sym2(sym, section, value, size, 1);
542}
543
544/* add a new relocation entry to symbol 'sym' in section 's' */
738606db Edmund Grimley Evans2015-02-21 21:29:03 +0000545ST_FUNC void greloca(Section *s, Sym *sym, unsigned long offset, int type,
8f639006 seyko2015-03-23 19:24:55 +0300546 addr_t addend)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200547{
1308e8eb grischka2009-12-19 22:08:37 +0100548 int c = 0;
549 if (sym) {
550 if (0 == sym->c)
551 put_extern_sym(sym, NULL, 0, 0);
552 c = sym->c;
553 }
9dc9cbf3 grischka2009-05-05 20:18:53 +0200554 /* now we can add ELF relocation info */
738606db
EGE
Edmund Grimley Evans2015-02-21 21:29:03 +0000555 put_elf_reloca(symtab_section, s, offset, type, c, addend);
556}
557
558ST_FUNC void greloc(Section *s, Sym *sym, unsigned long offset, int type)
559{
560 greloca(s, sym, offset, type, 0);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200561}
562
88a3ccab grischka2009-12-20 01:53:49 +0100563/********************************************************/
564
9dc9cbf3 grischka2009-05-05 20:18:53 +0200565static void strcat_vprintf(char *buf, int buf_size, const char *fmt, va_list ap)
566{
567 int len;
568 len = strlen(buf);
569 vsnprintf(buf + len, buf_size - len, fmt, ap);
570}
571
572static void strcat_printf(char *buf, int buf_size, const char *fmt, ...)
573{
574 va_list ap;
575 va_start(ap, fmt);
576 strcat_vprintf(buf, buf_size, fmt, ap);
577 va_end(ap);
578}
579
88a3ccab grischka2009-12-20 01:53:49 +0100580static void error1(TCCState *s1, int is_warning, const char *fmt, va_list ap)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200581{
582 char buf[2048];
944627c4 grischka2013-02-14 06:53:07 +0100583 BufferedFile **pf, *f;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600584
9dc9cbf3 grischka2009-05-05 20:18:53 +0200585 buf[0] = '\0';
944627c4 grischka2013-02-14 06:53:07 +0100586 /* use upper file if inline ":asm:" or token ":paste:" */
0db7f616
UJ
Urs Janssen2013-02-18 15:44:18 +0100587 for (f = file; f && f->filename[0] == ':'; f = f->prev)
588 ;
944627c4 grischka2013-02-14 06:53:07 +0100589 if (f) {
590 for(pf = s1->include_stack; pf < s1->include_stack_ptr; pf++)
591 strcat_printf(buf, sizeof(buf), "In file included from %s:%d:\n",
592 (*pf)->filename, (*pf)->line_num);
593 if (f->line_num > 0) {
594 strcat_printf(buf, sizeof(buf), "%s:%d: ",
595 f->filename, f->line_num);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200596 } else {
944627c4 grischka2013-02-14 06:53:07 +0100597 strcat_printf(buf, sizeof(buf), "%s: ",
598 f->filename);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200599 }
600 } else {
944627c4 grischka2013-02-14 06:53:07 +0100601 strcat_printf(buf, sizeof(buf), "tcc: ");
9dc9cbf3 grischka2009-05-05 20:18:53 +0200602 }
603 if (is_warning)
604 strcat_printf(buf, sizeof(buf), "warning: ");
ec54c34b grischka2009-06-17 02:10:56 +0200605 else
606 strcat_printf(buf, sizeof(buf), "error: ");
9dc9cbf3 grischka2009-05-05 20:18:53 +0200607 strcat_vprintf(buf, sizeof(buf), fmt, ap);
608
609 if (!s1->error_func) {
610 /* default case: stderr */
611 fprintf(stderr, "%s\n", buf);
612 } else {
613 s1->error_func(s1->error_opaque, buf);
614 }
615 if (!is_warning || s1->warn_error)
616 s1->nb_errors++;
617}
618
88a3ccab grischka2009-12-20 01:53:49 +0100619LIBTCCAPI void tcc_set_error_func(TCCState *s, void *error_opaque,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200620 void (*error_func)(void *opaque, const char *msg))
621{
622 s->error_opaque = error_opaque;
623 s->error_func = error_func;
624}
9dc9cbf3 grischka2009-05-05 20:18:53 +0200625
626/* error without aborting current compilation */
bf374a5f grischka2011-08-11 17:07:56 +0200627PUB_FUNC void tcc_error_noabort(const char *fmt, ...)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200628{
629 TCCState *s1 = tcc_state;
630 va_list ap;
631
632 va_start(ap, fmt);
633 error1(s1, 0, fmt, ap);
634 va_end(ap);
635}
636
bf374a5f grischka2011-08-11 17:07:56 +0200637PUB_FUNC void tcc_error(const char *fmt, ...)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200638{
639 TCCState *s1 = tcc_state;
640 va_list ap;
641
642 va_start(ap, fmt);
643 error1(s1, 0, fmt, ap);
644 va_end(ap);
645 /* better than nothing: in some cases, we accept to handle errors */
646 if (s1->error_set_jmp_enabled) {
647 longjmp(s1->error_jmp_buf, 1);
648 } else {
649 /* XXX: eliminate this someday */
650 exit(1);
651 }
652}
653
bf374a5f grischka2011-08-11 17:07:56 +0200654PUB_FUNC void tcc_warning(const char *fmt, ...)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200655{
656 TCCState *s1 = tcc_state;
657 va_list ap;
658
659 if (s1->warn_none)
660 return;
661
662 va_start(ap, fmt);
663 error1(s1, 1, fmt, ap);
664 va_end(ap);
665}
666
88a3ccab grischka2009-12-20 01:53:49 +0100667/********************************************************/
9dc9cbf3 grischka2009-05-05 20:18:53 +0200668/* I/O layer */
669
e97bf88b grischka2010-11-25 13:29:15 +0100670ST_FUNC void tcc_open_bf(TCCState *s1, const char *filename, int initlen)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200671{
9dc9cbf3 grischka2009-05-05 20:18:53 +0200672 BufferedFile *bf;
e97bf88b grischka2010-11-25 13:29:15 +0100673 int buflen = initlen ? initlen : IO_BUF_SIZE;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200674
e97bf88b grischka2010-11-25 13:29:15 +0100675 bf = tcc_malloc(sizeof(BufferedFile) + buflen);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200676 bf->buf_ptr = bf->buffer;
e97bf88b grischka2010-11-25 13:29:15 +0100677 bf->buf_end = bf->buffer + initlen;
678 bf->buf_end[0] = CH_EOB; /* put eob symbol */
9dc9cbf3 grischka2009-05-05 20:18:53 +0200679 pstrcpy(bf->filename, sizeof(bf->filename), filename);
680#ifdef _WIN32
681 normalize_slashes(bf->filename);
682#endif
683 bf->line_num = 1;
684 bf->ifndef_macro = 0;
685 bf->ifdef_stack_ptr = s1->ifdef_stack_ptr;
8c56b0cf Thomas Preud'homme2013-01-14 17:34:07 +0100686 bf->fd = -1;
e97bf88b grischka2010-11-25 13:29:15 +0100687 bf->prev = file;
688 file = bf;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200689}
690
e97bf88b grischka2010-11-25 13:29:15 +0100691ST_FUNC void tcc_close(void)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200692{
e97bf88b grischka2010-11-25 13:29:15 +0100693 BufferedFile *bf = file;
8c56b0cf
TP
Thomas Preud'homme2013-01-14 17:34:07 +0100694 if (bf->fd > 0) {
695 close(bf->fd);
e97bf88b grischka2010-11-25 13:29:15 +0100696 total_lines += bf->line_num;
697 }
698 file = bf->prev;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200699 tcc_free(bf);
700}
701
8c56b0cf Thomas Preud'homme2013-01-14 17:34:07 +0100702ST_FUNC int tcc_open(TCCState *s1, const char *filename)
e97bf88b grischka2010-11-25 13:29:15 +0100703{
8c56b0cf
TP
Thomas Preud'homme2013-01-14 17:34:07 +0100704 int fd;
705 if (strcmp(filename, "-") == 0)
706 fd = 0, filename = "stdin";
e97bf88b grischka2010-11-25 13:29:15 +0100707 else
8c56b0cf
TP
Thomas Preud'homme2013-01-14 17:34:07 +0100708 fd = open(filename, O_RDONLY | O_BINARY);
709 if ((s1->verbose == 2 && fd >= 0) || s1->verbose == 3)
710 printf("%s %*s%s\n", fd < 0 ? "nf":"->",
e97bf88b grischka2010-11-25 13:29:15 +0100711 (int)(s1->include_stack_ptr - s1->include_stack), "", filename);
8c56b0cf
TP
Thomas Preud'homme2013-01-14 17:34:07 +0100712 if (fd < 0)
713 return -1;
e97bf88b grischka2010-11-25 13:29:15 +0100714
715 tcc_open_bf(s1, filename, 0);
716 file->fd = fd;
717 return fd;
718}
719
9dc9cbf3 grischka2009-05-05 20:18:53 +0200720/* compile the C file opened in 'file'. Return non zero if errors. */
721static int tcc_compile(TCCState *s1)
722{
723 Sym *define_start;
6eb97c70 Joe Soroka2011-04-08 00:41:55 -0700724 SValue *pvtop;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200725 char buf[512];
726 volatile int section_sym;
727
728#ifdef INC_DEBUG
729 printf("%s: **** new file\n", file->filename);
730#endif
731 preprocess_init(s1);
732
733 cur_text_section = NULL;
734 funcname = "";
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600735 anon_sym = SYM_FIRST_ANOM;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200736
737 /* file info: full path + filename */
738 section_sym = 0; /* avoid warning */
f9181416 grischka2009-05-11 18:45:44 +0200739 if (s1->do_debug) {
0650ab01
JP
Joseph Poirier2013-11-08 13:24:15 -0600740 section_sym = put_elf_sym(symtab_section, 0, 0,
741 ELFW(ST_INFO)(STB_LOCAL, STT_SECTION), 0,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200742 text_section->sh_num, NULL);
743 getcwd(buf, sizeof(buf));
744#ifdef _WIN32
745 normalize_slashes(buf);
746#endif
747 pstrcat(buf, sizeof(buf), "/");
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600748 put_stabs_r(buf, N_SO, 0, 0,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200749 text_section->data_offset, text_section, section_sym);
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600750 put_stabs_r(file->filename, N_SO, 0, 0,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200751 text_section->data_offset, text_section, section_sym);
752 }
753 /* an elf symbol of type STT_FILE must be put so that STB_LOCAL
754 symbols can be safely used */
0650ab01
JP
Joseph Poirier2013-11-08 13:24:15 -0600755 put_elf_sym(symtab_section, 0, 0,
756 ELFW(ST_INFO)(STB_LOCAL, STT_FILE), 0,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200757 SHN_ABS, file->filename);
758
759 /* define some often used types */
760 int_type.t = VT_INT;
761
762 char_pointer_type.t = VT_BYTE;
763 mk_pointer(&char_pointer_type);
764
718fd591
MM
Michael Matz2012-04-16 01:13:25 +0200765#if PTR_SIZE == 4
766 size_type.t = VT_INT;
767#else
768 size_type.t = VT_LLONG;
769#endif
770
9dc9cbf3 grischka2009-05-05 20:18:53 +0200771 func_old_type.t = VT_FUNC;
772 func_old_type.ref = sym_push(SYM_FIELD, &int_type, FUNC_CDECL, FUNC_OLD);
d6d7686b grischka2013-02-08 19:07:11 +0100773#ifdef TCC_TARGET_ARM
b6247d1f Thomas Preud'homme2014-01-07 15:23:54 +0800774 arm_init(s1);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200775#endif
776
777#if 0
778 /* define 'void *alloca(unsigned int)' builtin function */
779 {
780 Sym *s1;
781
782 p = anon_sym++;
783 sym = sym_push(p, mk_pointer(VT_VOID), FUNC_CDECL, FUNC_NEW);
784 s1 = sym_push(SYM_FIELD, VT_UNSIGNED | VT_INT, 0, 0);
785 s1->next = NULL;
786 sym->next = s1;
787 sym_push(TOK_alloca, VT_FUNC | (p << VT_STRUCT_SHIFT), VT_CONST, 0);
788 }
789#endif
790
791 define_start = define_stack;
c45a8695 seyko2015-03-03 15:44:29 +0300792 nocode_wanted = 1;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200793
794 if (setjmp(s1->error_jmp_buf) == 0) {
795 s1->nb_errors = 0;
796 s1->error_set_jmp_enabled = 1;
797
798 ch = file->buf_ptr[0];
799 tok_flags = TOK_FLAG_BOL | TOK_FLAG_BOF;
800 parse_flags = PARSE_FLAG_PREPROCESS | PARSE_FLAG_TOK_NUM;
7391cf01 Joe Soroka2011-05-02 00:05:36 -0700801 pvtop = vtop;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200802 next();
803 decl(VT_CONST);
804 if (tok != TOK_EOF)
805 expect("declaration");
7391cf01 Joe Soroka2011-05-02 00:05:36 -0700806 if (pvtop != vtop)
bf374a5f grischka2011-08-11 17:07:56 +0200807 tcc_warning("internal compiler error: vstack leak? (%d)", vtop - pvtop);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200808
809 /* end of translation unit info */
f9181416 grischka2009-05-11 18:45:44 +0200810 if (s1->do_debug) {
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600811 put_stabs_r(NULL, N_SO, 0, 0,
9dc9cbf3 grischka2009-05-05 20:18:53 +0200812 text_section->data_offset, text_section, section_sym);
813 }
814 }
48d81a79 Joe Soroka2011-04-14 01:02:42 -0700815
9dc9cbf3 grischka2009-05-05 20:18:53 +0200816 s1->error_set_jmp_enabled = 0;
817
818 /* reset define stack, but leave -Dsymbols (may be incorrect if
819 they are undefined) */
0650ab01 Joseph Poirier2013-11-08 13:24:15 -0600820 free_defines(define_start);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200821
822 gen_inline_functions();
823
824 sym_pop(&global_stack, NULL);
825 sym_pop(&local_stack, NULL);
826
827 return s1->nb_errors != 0 ? -1 : 0;
828}
829
8c56b0cf Thomas Preud'homme2013-01-14 17:34:07 +0100830LIBTCCAPI int tcc_compile_string(TCCState *s, const char *str)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200831{
aeaff94e Steven G. Messervey2015-04-15 22:56:21 -0400832 int i;
e97bf88b grischka2010-11-25 13:29:15 +0100833 int len, ret;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200834 len = strlen(str);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200835
8c56b0cf Thomas Preud'homme2013-01-14 17:34:07 +0100836 tcc_open_bf(s, "<string>", len);
e97bf88b grischka2010-11-25 13:29:15 +0100837 memcpy(file->buffer, str, len);
aeaff94e
SM
Steven G. Messervey2015-04-15 22:56:21 -0400838
839 len = s->nb_files;
e97bf88b grischka2010-11-25 13:29:15 +0100840 ret = tcc_compile(s);
841 tcc_close();
aeaff94e
SM
Steven G. Messervey2015-04-15 22:56:21 -0400842
843 /* habdle #pragma comment(lib,) */
844 for(i = len; i < s->nb_files; i++) {
845 /* int filetype = *(unsigned char *)s->files[i]; */
846 const char *filename = s->files[i] + 1;
847 if (filename[0] == '-' && filename[1] == 'l') {
848 if (tcc_add_library(s, filename + 2) < 0) {
b472d536 seyko2015-04-16 07:30:24 +0300849 tcc_warning("cannot find library 'lib%s'", filename+2);
aeaff94e
SM
Steven G. Messervey2015-04-15 22:56:21 -0400850 ret++;
851 }
852 }
853 tcc_free(s->files[i]);
854 }
855 s->nb_files = len;
856
9dc9cbf3 grischka2009-05-05 20:18:53 +0200857 return ret;
858}
9dc9cbf3 grischka2009-05-05 20:18:53 +0200859
860/* define a preprocessor symbol. A value can also be provided with the '=' operator */
88a3ccab grischka2009-12-20 01:53:49 +0100861LIBTCCAPI void tcc_define_symbol(TCCState *s1, const char *sym, const char *value)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200862{
e97bf88b grischka2010-11-25 13:29:15 +0100863 int len1, len2;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200864 /* default value */
e97bf88b grischka2010-11-25 13:29:15 +0100865 if (!value)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200866 value = "1";
e97bf88b grischka2010-11-25 13:29:15 +0100867 len1 = strlen(sym);
868 len2 = strlen(value);
869
9dc9cbf3 grischka2009-05-05 20:18:53 +0200870 /* init file structure */
e97bf88b grischka2010-11-25 13:29:15 +0100871 tcc_open_bf(s1, "<define>", len1 + len2 + 1);
872 memcpy(file->buffer, sym, len1);
873 file->buffer[len1] = ' ';
874 memcpy(file->buffer + len1 + 1, value, len2);
9dc9cbf3 grischka2009-05-05 20:18:53 +0200875
876 /* parse with define parser */
877 ch = file->buf_ptr[0];
878 next_nomacro();
879 parse_define();
e97bf88b grischka2010-11-25 13:29:15 +0100880
881 tcc_close();
9dc9cbf3 grischka2009-05-05 20:18:53 +0200882}
883
884/* undefine a preprocessor symbol */
88a3ccab grischka2009-12-20 01:53:49 +0100885LIBTCCAPI void tcc_undefine_symbol(TCCState *s1, const char *sym)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200886{
887 TokenSym *ts;
888 Sym *s;
889 ts = tok_alloc(sym, strlen(sym));
890 s = define_find(ts->tok);
891 /* undefine symbol by putting an invalid name */
892 if (s)
893 define_undef(s);
894}
895
05108a3b grischka2013-02-12 19:13:28 +0100896/* cleanup all static data used during compilation */
9dc9cbf3 grischka2009-05-05 20:18:53 +0200897static void tcc_cleanup(void)
898{
899 int i, n;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200900 if (NULL == tcc_state)
901 return;
902 tcc_state = NULL;
903
904 /* free -D defines */
905 free_defines(NULL);
906
907 /* free tokens */
908 n = tok_ident - TOK_IDENT;
909 for(i = 0; i < n; i++)
910 tcc_free(table_ident[i]);
911 tcc_free(table_ident);
a429d40f seyko2015-03-03 14:54:46 +0300912 table_ident = NULL;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200913
914 /* free sym_pools */
915 dynarray_reset(&sym_pools, &nb_sym_pools);
916 /* string buffer */
917 cstr_free(&tokcstr);
918 /* reset symbol stack */
919 sym_free_first = NULL;
920 /* cleanup from error/setjmp */
921 macro_ptr = NULL;
922}
923
88a3ccab grischka2009-12-20 01:53:49 +0100924LIBTCCAPI TCCState *tcc_new(void)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200925{
9dc9cbf3 grischka2009-05-05 20:18:53 +0200926 TCCState *s;
dd70d192 grischka2009-05-25 02:07:30 +0200927 char buffer[100];
928 int a,b,c;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200929
930 tcc_cleanup();
931
932 s = tcc_mallocz(sizeof(TCCState));
933 if (!s)
934 return NULL;
935 tcc_state = s;
94ae3984 grischka2009-07-18 22:07:33 +0200936#ifdef _WIN32
937 tcc_set_lib_path_w32(s);
938#else
939 tcc_set_lib_path(s, CONFIG_TCCDIR);
940#endif
c334b591 seyko2015-01-06 22:19:45 +0300941 s->output_type = 0;
b8f6e1ae grischka2009-05-05 20:30:39 +0200942 preprocess_new();
c4eef480 grischka2010-11-26 20:46:25 +0100943 s->include_stack_ptr = s->include_stack;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200944
945 /* we add dummy defines for some special macros to speed up tests
946 and to have working defined() */
947 define_push(TOK___LINE__, MACRO_OBJ, NULL, NULL);
948 define_push(TOK___FILE__, MACRO_OBJ, NULL, NULL);
949 define_push(TOK___DATE__, MACRO_OBJ, NULL, NULL);
950 define_push(TOK___TIME__, MACRO_OBJ, NULL, NULL);
951
05108a3b grischka2013-02-12 19:13:28 +0100952 /* define __TINYC__ 92X */
953 sscanf(TCC_VERSION, "%d.%d.%d", &a, &b, &c);
954 sprintf(buffer, "%d", a*10000 + b*100 + c);
955 tcc_define_symbol(s, "__TINYC__", buffer);
956
9dc9cbf3 grischka2009-05-05 20:18:53 +0200957 /* standard defines */
958 tcc_define_symbol(s, "__STDC__", NULL);
959 tcc_define_symbol(s, "__STDC_VERSION__", "199901L");
0db7f616 Urs Janssen2013-02-18 15:44:18 +0100960 tcc_define_symbol(s, "__STDC_HOSTED__", NULL);
05108a3b grischka2013-02-12 19:13:28 +0100961
962 /* target defines */
9dc9cbf3 grischka2009-05-05 20:18:53 +0200963#if defined(TCC_TARGET_I386)
e20bf69a grischka2010-01-14 20:56:35 +0100964 tcc_define_symbol(s, "__i386__", NULL);
965 tcc_define_symbol(s, "__i386", NULL);
966 tcc_define_symbol(s, "i386", NULL);
05108a3b grischka2013-02-12 19:13:28 +0100967#elif defined(TCC_TARGET_X86_64)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200968 tcc_define_symbol(s, "__x86_64__", NULL);
05108a3b grischka2013-02-12 19:13:28 +0100969#elif defined(TCC_TARGET_ARM)
9dc9cbf3 grischka2009-05-05 20:18:53 +0200970 tcc_define_symbol(s, "__ARM_ARCH_4__", NULL);
971 tcc_define_symbol(s, "__arm_elf__", NULL);
972 tcc_define_symbol(s, "__arm_elf", NULL);
973 tcc_define_symbol(s, "arm_elf", NULL);
974 tcc_define_symbol(s, "__arm__", NULL);
975 tcc_define_symbol(s, "__arm", NULL);
976 tcc_define_symbol(s, "arm", NULL);
977 tcc_define_symbol(s, "__APCS_32__", NULL);
3676f619
TP
Thomas Preud'homme2013-11-26 12:06:21 +0800978 tcc_define_symbol(s, "__ARMEL__", NULL);
979#if defined(TCC_ARM_EABI)
980 tcc_define_symbol(s, "__ARM_EABI__", NULL);
b6247d1f Thomas Preud'homme2014-01-07 15:23:54 +0800981#endif
37b03489 Thomas Preud'homme2013-06-11 18:13:04 +0200982#if defined(TCC_ARM_HARDFLOAT)
b6247d1f Thomas Preud'homme2014-01-07 15:23:54 +0800983 s->float_abi = ARM_HARD_FLOAT;
37b03489 Thomas Preud'homme2013-06-11 18:13:04 +0200984 tcc_define_symbol(s, "__ARM_PCS_VFP", NULL);
b6247d1f
TP
Thomas Preud'homme2014-01-07 15:23:54 +0800985#else
986 s->float_abi = ARM_SOFTFP_FLOAT;
9dc9cbf3 grischka2009-05-05 20:18:53 +0200987#endif
b14ef0e2
EGE
Edmund Grimley Evans2015-02-13 18:58:31 +0000988#elif defined(TCC_TARGET_ARM64)
989 tcc_define_symbol(s, "__aarch64__", NULL);
3676f619 Thomas Preud'homme2013-11-26 12:06:21 +0800990#endif
05108a3b grischka2013-02-12 19:13:28 +0100991
9dc9cbf3 grischka2009-05-05 20:18:53 +0200992#ifdef TCC_TARGET_PE
993 tcc_define_symbol(s, "_WIN32", NULL);
05108a3b grischka2013-02-12 19:13:28 +0100994# ifdef TCC_TARGET_X86_64
3ea4acb9 grischka2009-07-18 22:05:27 +0200995 tcc_define_symbol(s, "_WIN64", NULL);
05108a3b grischka2013-02-12 19:13:28 +0100996# endif
9dc9cbf3 grischka2009-05-05 20:18:53 +0200997#else
e20bf69a grischka2010-01-14 20:56:35 +0100998 tcc_define_symbol(s, "__unix__", NULL);
999 tcc_define_symbol(s, "__unix", NULL);
1000 tcc_define_symbol(s, "unix", NULL);
05108a3b grischka2013-02-12 19:13:28 +01001001# if defined(__linux)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001002 tcc_define_symbol(s, "__linux__", NULL);
1003 tcc_define_symbol(s, "__linux", NULL);
05108a3b grischka2013-02-12 19:13:28 +01001004# endif
1005# if defined(__FreeBSD__)
1006# define str(s) #s
1007 tcc_define_symbol(s, "__FreeBSD__", str( __FreeBSD__));
1008# undef str
1009# endif
1010# if defined(__FreeBSD_kernel__)
1011 tcc_define_symbol(s, "__FreeBSD_kernel__", NULL);
1012# endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001013#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001014
05108a3b grischka2013-02-12 19:13:28 +01001015 /* TinyCC & gcc defines */
e20bf69a grischka2010-01-14 20:56:35 +01001016#if defined TCC_TARGET_PE && defined TCC_TARGET_X86_64
1017 tcc_define_symbol(s, "__SIZE_TYPE__", "unsigned long long");
1018 tcc_define_symbol(s, "__PTRDIFF_TYPE__", "long long");
1019#else
3a138012
SH
Shinichiro Hamaji2009-07-19 04:26:19 +09001020 tcc_define_symbol(s, "__SIZE_TYPE__", "unsigned long");
1021 tcc_define_symbol(s, "__PTRDIFF_TYPE__", "long");
e20bf69a grischka2010-01-14 20:56:35 +01001022#endif
1023
9dc9cbf3 grischka2009-05-05 20:18:53 +02001024#ifdef TCC_TARGET_PE
1025 tcc_define_symbol(s, "__WCHAR_TYPE__", "unsigned short");
0961a384 Michael Matz2014-04-07 00:26:36 +02001026 tcc_define_symbol(s, "__WINT_TYPE__", "unsigned short");
9dc9cbf3 grischka2009-05-05 20:18:53 +02001027#else
1028 tcc_define_symbol(s, "__WCHAR_TYPE__", "int");
0961a384
MM
Michael Matz2014-04-07 00:26:36 +02001029 /* wint_t is unsigned int by default, but (signed) int on BSDs
1030 and unsigned short on windows. Other OSes might have still
1031 other conventions, sigh. */
1032#if defined(__FreeBSD__) || defined (__FreeBSD_kernel__)
1033 tcc_define_symbol(s, "__WINT_TYPE__", "int");
1034#else
1035 tcc_define_symbol(s, "__WINT_TYPE__", "unsigned int");
1036#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001037#endif
2596273f Thomas Preud'homme2010-09-11 17:57:15 +02001038
05108a3b grischka2013-02-12 19:13:28 +01001039#ifndef TCC_TARGET_PE
2596273f Thomas Preud'homme2010-09-11 17:57:15 +02001040 /* glibc defines */
e343b1df Ramsay Jones2012-02-12 22:27:26 +00001041 tcc_define_symbol(s, "__REDIRECT(name, proto, alias)", "name proto __asm__ (#alias)");
2596273f Thomas Preud'homme2010-09-11 17:57:15 +02001042 tcc_define_symbol(s, "__REDIRECT_NTH(name, proto, alias)", "name proto __asm__ (#alias) __THROW");
e844fb11 grischka2011-08-06 16:49:30 +02001043 /* paths for crt objects */
1044 tcc_split_path(s, (void ***)&s->crt_paths, &s->nb_crt_paths, CONFIG_TCC_CRTPREFIX);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001045#endif
1046
1047 /* no section zero */
1048 dynarray_add((void ***)&s->sections, &s->nb_sections, NULL);
1049
1050 /* create standard sections */
1051 text_section = new_section(s, ".text", SHT_PROGBITS, SHF_ALLOC | SHF_EXECINSTR);
1052 data_section = new_section(s, ".data", SHT_PROGBITS, SHF_ALLOC | SHF_WRITE);
1053 bss_section = new_section(s, ".bss", SHT_NOBITS, SHF_ALLOC | SHF_WRITE);
1054
1055 /* symbols are always generated for linking stage */
1056 symtab_section = new_symtab(s, ".symtab", SHT_SYMTAB, 0,
1057 ".strtab",
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001058 ".hashtab", SHF_PRIVATE);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001059 strtab_section = symtab_section->link;
f98c2306 grischka2012-04-18 18:48:26 +02001060 s->symtab = symtab_section;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001061
9dc9cbf3 grischka2009-05-05 20:18:53 +02001062 /* private symbol table for dynamic symbols */
1063 s->dynsymtab_section = new_symtab(s, ".dynsymtab", SHT_SYMTAB, SHF_PRIVATE,
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001064 ".dynstrtab",
9dc9cbf3 grischka2009-05-05 20:18:53 +02001065 ".dynhashtab", SHF_PRIVATE);
1066 s->alacarte_link = 1;
88a3ccab grischka2009-12-20 01:53:49 +01001067 s->nocommon = 1;
b5f88b59 seyko2015-03-03 16:32:25 +03001068 s->warn_implicit_function_declaration = 1;
9dc9cbf3 grischka2009-05-05 20:18:53 +02001069
1070#ifdef CHAR_IS_UNSIGNED
1071 s->char_is_unsigned = 1;
1072#endif
a3ebdd0a grischka2011-02-13 17:41:05 +01001073 /* enable this if you want symbols with leading underscore on windows: */
0bdbd49e Urs Janssen2013-02-17 00:48:51 +01001074#if 0 /* def TCC_TARGET_PE */
9dc9cbf3 grischka2009-05-05 20:18:53 +02001075 s->leading_underscore = 1;
1076#endif
0d768b97
FF
Frederic Feret2009-08-27 09:34:35 +02001077#ifdef TCC_TARGET_I386
1078 s->seg_size = 32;
1079#endif
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001080#ifdef TCC_IS_NATIVE
73faaea2 grischka2013-08-28 22:55:05 +02001081 s->runtime_main = "main";
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001082#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001083 return s;
1084}
1085
88a3ccab grischka2009-12-20 01:53:49 +01001086LIBTCCAPI void tcc_delete(TCCState *s1)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001087{
1088 int i;
1089
1090 tcc_cleanup();
1091
1092 /* free all sections */
1093 for(i = 1; i < s1->nb_sections; i++)
1094 free_section(s1->sections[i]);
1095 dynarray_reset(&s1->sections, &s1->nb_sections);
1096
1097 for(i = 0; i < s1->nb_priv_sections; i++)
1098 free_section(s1->priv_sections[i]);
1099 dynarray_reset(&s1->priv_sections, &s1->nb_priv_sections);
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001100
9dc9cbf3 grischka2009-05-05 20:18:53 +02001101 /* free any loaded DLLs */
1caee8ab James Lyon2013-04-25 22:30:53 +01001102#ifdef TCC_IS_NATIVE
9dc9cbf3 grischka2009-05-05 20:18:53 +02001103 for ( i = 0; i < s1->nb_loaded_dlls; i++) {
1104 DLLReference *ref = s1->loaded_dlls[i];
1105 if ( ref->handle )
1106 dlclose(ref->handle);
1107 }
1caee8ab James Lyon2013-04-25 22:30:53 +01001108#endif
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001109
9dc9cbf3 grischka2009-05-05 20:18:53 +02001110 /* free loaded dlls array */
1111 dynarray_reset(&s1->loaded_dlls, &s1->nb_loaded_dlls);
1112
1113 /* free library paths */
1114 dynarray_reset(&s1->library_paths, &s1->nb_library_paths);
e844fb11 grischka2011-08-06 16:49:30 +02001115 dynarray_reset(&s1->crt_paths, &s1->nb_crt_paths);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001116
1117 /* free include paths */
1118 dynarray_reset(&s1->cached_includes, &s1->nb_cached_includes);
1119 dynarray_reset(&s1->include_paths, &s1->nb_include_paths);
1120 dynarray_reset(&s1->sysinclude_paths, &s1->nb_sysinclude_paths);
1121
e6f3bf7f grischka2011-08-06 16:08:46 +02001122 tcc_free(s1->tcc_lib_path);
05108a3b grischka2013-02-12 19:13:28 +01001123 tcc_free(s1->soname);
1124 tcc_free(s1->rpath);
1125 tcc_free(s1->init_symbol);
1126 tcc_free(s1->fini_symbol);
1127 tcc_free(s1->outfile);
1128 tcc_free(s1->deps_outfile);
1129 dynarray_reset(&s1->files, &s1->nb_files);
1130 dynarray_reset(&s1->target_deps, &s1->nb_target_deps);
e6f3bf7f grischka2011-08-06 16:08:46 +02001131
05108a3b grischka2013-02-12 19:13:28 +01001132#ifdef TCC_IS_NATIVE
1133# ifdef HAVE_SELINUX
2ab42855 Henry Kroll III2010-04-24 03:28:54 -07001134 munmap (s1->write_mem, s1->mem_size);
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001135 munmap (s1->runtime_mem, s1->mem_size);
05108a3b grischka2013-02-12 19:13:28 +01001136# else
8bbde91f grischka2009-12-19 22:11:12 +01001137 tcc_free(s1->runtime_mem);
05108a3b grischka2013-02-12 19:13:28 +01001138# endif
2ab42855 Henry Kroll III2010-04-24 03:28:54 -07001139#endif
05108a3b grischka2013-02-12 19:13:28 +01001140
ad956806 mingodad2014-03-25 18:06:14 +00001141 if(s1->sym_attrs) tcc_free(s1->sym_attrs);
1142
9dc9cbf3 grischka2009-05-05 20:18:53 +02001143 tcc_free(s1);
1144}
1145
df9cce24 grischka2011-08-01 01:10:36 +02001146LIBTCCAPI int tcc_add_include_path(TCCState *s, const char *pathname)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001147{
df9cce24 grischka2011-08-01 01:10:36 +02001148 tcc_split_path(s, (void ***)&s->include_paths, &s->nb_include_paths, pathname);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001149 return 0;
1150}
1151
df9cce24 grischka2011-08-01 01:10:36 +02001152LIBTCCAPI int tcc_add_sysinclude_path(TCCState *s, const char *pathname)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001153{
df9cce24 grischka2011-08-01 01:10:36 +02001154 tcc_split_path(s, (void ***)&s->sysinclude_paths, &s->nb_sysinclude_paths, pathname);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001155 return 0;
1156}
1157
05364072 seyko2015-04-12 15:35:37 +03001158ST_FUNC int tcc_add_file_internal(TCCState *s1, const char *filename, int flags, int filetype)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001159{
9dc9cbf3 grischka2009-05-05 20:18:53 +02001160 ElfW(Ehdr) ehdr;
8c56b0cf Thomas Preud'homme2013-01-14 17:34:07 +01001161 int fd, ret, size;
1df662c1 grischka2009-07-18 22:07:25 +02001162
d81611b6 seyko2015-04-10 16:53:29 +03001163 parse_flags = 0;
a25325e9
JS
Joe Soroka2011-02-01 15:26:21 -08001164#ifdef CONFIG_TCC_ASM
1165 /* if .S file, define __ASSEMBLER__ like gcc does */
05364072 seyko2015-04-12 15:35:37 +03001166 if ((filetype == TCC_FILETYPE_ASM) || (filetype == TCC_FILETYPE_ASM_PP)) {
a25325e9 Joe Soroka2011-02-01 15:26:21 -08001167 tcc_define_symbol(s1, "__ASSEMBLER__", NULL);
bbcb54a1 seyko2015-04-27 16:36:58 +03001168 parse_flags = PARSE_FLAG_ASM_FILE;
d81611b6 seyko2015-04-10 16:53:29 +03001169 }
a25325e9
JS
Joe Soroka2011-02-01 15:26:21 -08001170#endif
1171
9dc9cbf3 grischka2009-05-05 20:18:53 +02001172 /* open the file */
8c56b0cf
TP
Thomas Preud'homme2013-01-14 17:34:07 +01001173 ret = tcc_open(s1, filename);
1174 if (ret < 0) {
1df662c1 grischka2009-07-18 22:07:25 +02001175 if (flags & AFF_PRINT_ERROR)
bf374a5f grischka2011-08-11 17:07:56 +02001176 tcc_error_noabort("file '%s' not found", filename);
8c56b0cf Thomas Preud'homme2013-01-14 17:34:07 +01001177 return ret;
9dc9cbf3 grischka2009-05-05 20:18:53 +02001178 }
1179
0c928da9
KS
Kirill Smelkov2010-06-20 20:08:12 +04001180 /* update target deps */
1181 dynarray_add((void ***)&s1->target_deps, &s1->nb_target_deps,
1182 tcc_strdup(filename));
1183
9dc9cbf3 grischka2009-05-05 20:18:53 +02001184 if (flags & AFF_PREPROCESS) {
1185 ret = tcc_preprocess(s1);
1df662c1 grischka2009-07-18 22:07:25 +02001186 goto the_end;
1187 }
1188
05364072 seyko2015-04-12 15:35:37 +03001189 if (filetype == TCC_FILETYPE_C) {
9dc9cbf3 grischka2009-05-05 20:18:53 +02001190 /* C file assumed */
1191 ret = tcc_compile(s1);
1df662c1 grischka2009-07-18 22:07:25 +02001192 goto the_end;
1193 }
1194
9dc9cbf3 grischka2009-05-05 20:18:53 +02001195#ifdef CONFIG_TCC_ASM
05364072 seyko2015-04-12 15:35:37 +03001196 if (filetype == TCC_FILETYPE_ASM_PP) {
70dbe169 seyko2015-04-10 16:31:12 +03001197 /* non preprocessed assembler */
9dc9cbf3 grischka2009-05-05 20:18:53 +02001198 ret = tcc_assemble(s1, 1);
1df662c1 grischka2009-07-18 22:07:25 +02001199 goto the_end;
1200 }
1201
05364072 seyko2015-04-12 15:35:37 +03001202 if (filetype == TCC_FILETYPE_ASM) {
70dbe169 seyko2015-04-10 16:31:12 +03001203 /* preprocessed assembler */
9dc9cbf3 grischka2009-05-05 20:18:53 +02001204 ret = tcc_assemble(s1, 0);
1df662c1 grischka2009-07-18 22:07:25 +02001205 goto the_end;
1206 }
9dc9cbf3 grischka2009-05-05 20:18:53 +02001207#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001208
1df662c1 grischka2009-07-18 22:07:25 +02001209 fd = file->fd;
1210 /* assume executable format: auto guess file type */
8c56b0cf
TP
Thomas Preud'homme2013-01-14 17:34:07 +01001211 size = read(fd, &ehdr, sizeof(ehdr));
1212 lseek(fd, 0, SEEK_SET);
1df662c1 grischka2009-07-18 22:07:25 +02001213 if (size <= 0) {
bf374a5f grischka2011-08-11 17:07:56 +02001214 tcc_error_noabort("could not read header");
1df662c1 grischka2009-07-18 22:07:25 +02001215 goto the_end;
1216 }
1217
1218 if (size == sizeof(ehdr) &&
1219 ehdr.e_ident[0] == ELFMAG0 &&
1220 ehdr.e_ident[1] == ELFMAG1 &&
1221 ehdr.e_ident[2] == ELFMAG2 &&
1222 ehdr.e_ident[3] == ELFMAG3) {
1223
1224 /* do not display line number if error */
1225 file->line_num = 0;
1226 if (ehdr.e_type == ET_REL) {
1227 ret = tcc_load_object_file(s1, fd, 0);
1228 goto the_end;
1229
1230 }
1231#ifndef TCC_TARGET_PE
1232 if (ehdr.e_type == ET_DYN) {
1233 if (s1->output_type == TCC_OUTPUT_MEMORY) {
7a477d70 grischka2013-02-06 19:01:07 +01001234#ifdef TCC_IS_NATIVE
1df662c1 grischka2009-07-18 22:07:25 +02001235 void *h;
1236 h = dlopen(filename, RTLD_GLOBAL | RTLD_LAZY);
1237 if (h)
7a477d70 grischka2013-02-06 19:01:07 +01001238#endif
1df662c1 grischka2009-07-18 22:07:25 +02001239 ret = 0;
9dc9cbf3 grischka2009-05-05 20:18:53 +02001240 } else {
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001241 ret = tcc_load_dll(s1, fd, filename,
1df662c1 grischka2009-07-18 22:07:25 +02001242 (flags & AFF_REFERENCED_DLL) != 0);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001243 }
1df662c1 grischka2009-07-18 22:07:25 +02001244 goto the_end;
1245 }
1246#endif
bf374a5f grischka2011-08-11 17:07:56 +02001247 tcc_error_noabort("unrecognized ELF file");
1df662c1 grischka2009-07-18 22:07:25 +02001248 goto the_end;
1249 }
1250
1251 if (memcmp((char *)&ehdr, ARMAG, 8) == 0) {
1252 file->line_num = 0; /* do not display line number if error */
1253 ret = tcc_load_archive(s1, fd);
1254 goto the_end;
1255 }
1256
9dc9cbf3 grischka2009-05-05 20:18:53 +02001257#ifdef TCC_TARGET_COFF
1df662c1 grischka2009-07-18 22:07:25 +02001258 if (*(uint16_t *)(&ehdr) == COFF_C67_MAGIC) {
1259 ret = tcc_load_coff(s1, fd);
1260 goto the_end;
1261 }
9dc9cbf3 grischka2009-05-05 20:18:53 +02001262#endif
1df662c1 grischka2009-07-18 22:07:25 +02001263
9dc9cbf3 grischka2009-05-05 20:18:53 +02001264#ifdef TCC_TARGET_PE
1df662c1 grischka2009-07-18 22:07:25 +02001265 ret = pe_load_file(s1, filename, fd);
1266#else
1267 /* as GNU ld, consider it is an ld script if not recognized */
1268 ret = tcc_load_ldscript(s1);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001269#endif
1df662c1 grischka2009-07-18 22:07:25 +02001270 if (ret < 0)
bf374a5f grischka2011-08-11 17:07:56 +02001271 tcc_error_noabort("unrecognized file type");
1df662c1 grischka2009-07-18 22:07:25 +02001272
1273the_end:
e97bf88b grischka2010-11-25 13:29:15 +01001274 tcc_close();
9dc9cbf3 grischka2009-05-05 20:18:53 +02001275 return ret;
9dc9cbf3 grischka2009-05-05 20:18:53 +02001276}
1277
05364072 seyko2015-04-12 15:35:37 +03001278LIBTCCAPI int tcc_add_file(TCCState *s, const char *filename, int filetype)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001279{
67aebdd5 grischka2009-05-11 18:46:02 +02001280 if (s->output_type == TCC_OUTPUT_PREPROCESS)
05364072 seyko2015-04-12 15:35:37 +03001281 return tcc_add_file_internal(s, filename, AFF_PRINT_ERROR | AFF_PREPROCESS, filetype);
67aebdd5 grischka2009-05-11 18:46:02 +02001282 else
05364072 seyko2015-04-12 15:35:37 +03001283 return tcc_add_file_internal(s, filename, AFF_PRINT_ERROR, filetype);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001284}
1285
88a3ccab grischka2009-12-20 01:53:49 +01001286LIBTCCAPI int tcc_add_library_path(TCCState *s, const char *pathname)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001287{
df9cce24 grischka2011-08-01 01:10:36 +02001288 tcc_split_path(s, (void ***)&s->library_paths, &s->nb_library_paths, pathname);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001289 return 0;
1290}
1291
e844fb11 grischka2011-08-06 16:49:30 +02001292static int tcc_add_library_internal(TCCState *s, const char *fmt,
1293 const char *filename, int flags, char **paths, int nb_paths)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001294{
1295 char buf[1024];
1296 int i;
1297
e844fb11 grischka2011-08-06 16:49:30 +02001298 for(i = 0; i < nb_paths; i++) {
1299 snprintf(buf, sizeof(buf), fmt, paths[i], filename);
05364072 seyko2015-04-12 15:35:37 +03001300 if (tcc_add_file_internal(s, buf, flags, TCC_FILETYPE_BINARY) == 0)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001301 return 0;
1302 }
1303 return -1;
1304}
1305
e844fb11 grischka2011-08-06 16:49:30 +02001306/* find and load a dll. Return non zero if not found */
1307/* XXX: add '-rpath' option support ? */
1308ST_FUNC int tcc_add_dll(TCCState *s, const char *filename, int flags)
1309{
1310 return tcc_add_library_internal(s, "%s/%s", filename, flags,
1311 s->library_paths, s->nb_library_paths);
1312}
1313
1314ST_FUNC int tcc_add_crt(TCCState *s, const char *filename)
1315{
1316 if (-1 == tcc_add_library_internal(s, "%s/%s",
1317 filename, 0, s->crt_paths, s->nb_crt_paths))
bf374a5f grischka2011-08-11 17:07:56 +02001318 tcc_error_noabort("file '%s' not found", filename);
e844fb11 grischka2011-08-06 16:49:30 +02001319 return 0;
1320}
1321
9dc9cbf3 grischka2009-05-05 20:18:53 +02001322/* the library name is the same as the argument of the '-l' option */
88a3ccab grischka2009-12-20 01:53:49 +01001323LIBTCCAPI int tcc_add_library(TCCState *s, const char *libraryname)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001324{
9dc9cbf3 grischka2009-05-05 20:18:53 +02001325#ifdef TCC_TARGET_PE
e844fb11 grischka2011-08-06 16:49:30 +02001326 const char *libs[] = { "%s/%s.def", "%s/lib%s.def", "%s/%s.dll", "%s/lib%s.dll", "%s/lib%s.a", NULL };
1327 const char **pp = s->static_link ? libs + 4 : libs;
9dc9cbf3 grischka2009-05-05 20:18:53 +02001328#else
e844fb11 grischka2011-08-06 16:49:30 +02001329 const char *libs[] = { "%s/lib%s.so", "%s/lib%s.a", NULL };
1330 const char **pp = s->static_link ? libs + 1 : libs;
1df662c1 grischka2009-07-18 22:07:25 +02001331#endif
e844fb11 grischka2011-08-06 16:49:30 +02001332 while (*pp) {
1333 if (0 == tcc_add_library_internal(s, *pp,
1334 libraryname, 0, s->library_paths, s->nb_library_paths))
9dc9cbf3 grischka2009-05-05 20:18:53 +02001335 return 0;
e844fb11 grischka2011-08-06 16:49:30 +02001336 ++pp;
9dc9cbf3 grischka2009-05-05 20:18:53 +02001337 }
1338 return -1;
1339}
1340
52991422 grischka2010-01-14 20:56:04 +01001341LIBTCCAPI int tcc_add_symbol(TCCState *s, const char *name, const void *val)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001342{
2341ee51 grischka2010-01-14 20:55:51 +01001343#ifdef TCC_TARGET_PE
05108a3b grischka2013-02-12 19:13:28 +01001344 /* On x86_64 'val' might not be reachable with a 32bit offset.
1345 So it is handled here as if it were in a DLL. */
1346 pe_putimport(s, 0, name, (uintptr_t)val);
2341ee51 grischka2010-01-14 20:55:51 +01001347#else
05108a3b grischka2013-02-12 19:13:28 +01001348 add_elf_sym(symtab_section, (uintptr_t)val, 0,
1349 ELFW(ST_INFO)(STB_GLOBAL, STT_NOTYPE), 0,
1350 SHN_ABS, name);
2341ee51 grischka2010-01-14 20:55:51 +01001351#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001352 return 0;
1353}
1354
88a3ccab grischka2009-12-20 01:53:49 +01001355LIBTCCAPI int tcc_set_output_type(TCCState *s, int output_type)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001356{
9dc9cbf3 grischka2009-05-05 20:18:53 +02001357 s->output_type = output_type;
e2650608 seyko2015-03-19 08:07:35 +03001358 if (output_type == TCC_OUTPUT_PREPROCESS)
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04001359 print_defines();
9dc9cbf3 grischka2009-05-05 20:18:53 +02001360
1361 if (!s->nostdinc) {
1362 /* default include paths */
e939c407 Joe Soroka2011-02-01 23:32:53 -08001363 /* -isystem paths have already been handled */
81cd0cf6 grischka2011-08-06 16:11:12 +02001364 tcc_add_sysinclude_path(s, CONFIG_TCC_SYSINCLUDEPATHS);
9dc9cbf3 grischka2009-05-05 20:18:53 +02001365 }
1366
1367 /* if bound checking, then add corresponding sections */
1368#ifdef CONFIG_TCC_BCHECK
f9181416 grischka2009-05-11 18:45:44 +02001369 if (s->do_bounds_check) {
9dc9cbf3 grischka2009-05-05 20:18:53 +02001370 /* define symbol */
1371 tcc_define_symbol(s, "__BOUNDS_CHECKING_ON", NULL);
1372 /* create bounds sections */
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001373 bounds_section = new_section(s, ".bounds",
9dc9cbf3 grischka2009-05-05 20:18:53 +02001374 SHT_PROGBITS, SHF_ALLOC);
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001375 lbounds_section = new_section(s, ".lbounds",
9dc9cbf3 grischka2009-05-05 20:18:53 +02001376 SHT_PROGBITS, SHF_ALLOC);
1377 }
1378#endif
1379
1380 if (s->char_is_unsigned) {
1381 tcc_define_symbol(s, "__CHAR_UNSIGNED__", NULL);
1382 }
1383
1384 /* add debug sections */
f9181416 grischka2009-05-11 18:45:44 +02001385 if (s->do_debug) {
9dc9cbf3 grischka2009-05-05 20:18:53 +02001386 /* stab symbols */
1387 stab_section = new_section(s, ".stab", SHT_PROGBITS, 0);
1388 stab_section->sh_entsize = sizeof(Stab_Sym);
1389 stabstr_section = new_section(s, ".stabstr", SHT_STRTAB, 0);
1390 put_elf_str(stabstr_section, "");
1391 stab_section->link = stabstr_section;
1392 /* put first entry */
1393 put_stabs("", 0, 0, 0, 0);
1394 }
1395
81cd0cf6 grischka2011-08-06 16:11:12 +02001396 tcc_add_library_path(s, CONFIG_TCC_LIBPATHS);
45b35a3d Vittorio Giovara2013-09-23 09:40:06 +02001397#ifdef TCC_TARGET_PE
df9cce24 grischka2011-08-01 01:10:36 +02001398# ifdef _WIN32
1399 tcc_add_systemdir(s);
1400# endif
1401#else
81cd0cf6 grischka2011-08-06 16:11:12 +02001402 /* add libc crt1/crti objects */
9dc9cbf3 grischka2009-05-05 20:18:53 +02001403 if ((output_type == TCC_OUTPUT_EXE || output_type == TCC_OUTPUT_DLL) &&
1404 !s->nostdlib) {
0b665200 Thomas Preud'homme2011-08-03 22:26:39 +02001405 if (output_type != TCC_OUTPUT_DLL)
e844fb11 grischka2011-08-06 16:49:30 +02001406 tcc_add_crt(s, "crt1.o");
1407 tcc_add_crt(s, "crti.o");
9dc9cbf3 grischka2009-05-05 20:18:53 +02001408 }
1409#endif
7e7e6148 seyko2015-04-10 07:37:31 +03001410
1411#ifdef CONFIG_TCC_BCHECK
1412 if (s->do_bounds_check && (output_type == TCC_OUTPUT_EXE))
1413 {
e7a60e4d seyko2015-04-12 04:47:15 +03001414 /* force a bcheck.o linking */
1415 addr_t func = TOK___bound_init;
1416 Sym *sym = external_global_sym(func, &func_old_type, 0);
1417 if (!sym->c)
1418 put_extern_sym(sym, NULL, 0, 0);
7e7e6148 seyko2015-04-10 07:37:31 +03001419 }
1420#endif
9dc9cbf3 grischka2009-05-05 20:18:53 +02001421 return 0;
1422}
1423
05108a3b grischka2013-02-12 19:13:28 +01001424LIBTCCAPI void tcc_set_lib_path(TCCState *s, const char *path)
1425{
1426 tcc_free(s->tcc_lib_path);
1427 s->tcc_lib_path = tcc_strdup(path);
1428}
1429
9dc9cbf3 grischka2009-05-05 20:18:53 +02001430#define WD_ALL 0x0001 /* warning is activated when using -Wall */
1431#define FD_INVERT 0x0002 /* invert value before storing */
1432
1433typedef struct FlagDef {
1434 uint16_t offset;
1435 uint16_t flags;
1436 const char *name;
1437} FlagDef;
1438
1439static const FlagDef warning_defs[] = {
1440 { offsetof(TCCState, warn_unsupported), 0, "unsupported" },
1441 { offsetof(TCCState, warn_write_strings), 0, "write-strings" },
1442 { offsetof(TCCState, warn_error), 0, "error" },
1443 { offsetof(TCCState, warn_implicit_function_declaration), WD_ALL,
1444 "implicit-function-declaration" },
1445};
1446
88a3ccab grischka2009-12-20 01:53:49 +01001447ST_FUNC int set_flag(TCCState *s, const FlagDef *flags, int nb_flags,
9dc9cbf3 grischka2009-05-05 20:18:53 +02001448 const char *name, int value)
1449{
1450 int i;
1451 const FlagDef *p;
1452 const char *r;
1453
1454 r = name;
1455 if (r[0] == 'n' && r[1] == 'o' && r[2] == '-') {
1456 r += 3;
1457 value = !value;
1458 }
1459 for(i = 0, p = flags; i < nb_flags; i++, p++) {
1460 if (!strcmp(r, p->name))
1461 goto found;
1462 }
1463 return -1;
1464 found:
1465 if (p->flags & FD_INVERT)
1466 value = !value;
1467 *(int *)((uint8_t *)s + p->offset) = value;
1468 return 0;
1469}
1470
9dc9cbf3 grischka2009-05-05 20:18:53 +02001471/* set/reset a warning */
05108a3b grischka2013-02-12 19:13:28 +01001472static int tcc_set_warning(TCCState *s, const char *warning_name, int value)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001473{
1474 int i;
1475 const FlagDef *p;
1476
1477 if (!strcmp(warning_name, "all")) {
1478 for(i = 0, p = warning_defs; i < countof(warning_defs); i++, p++) {
1479 if (p->flags & WD_ALL)
1480 *(int *)((uint8_t *)s + p->offset) = 1;
1481 }
1482 return 0;
1483 } else {
1484 return set_flag(s, warning_defs, countof(warning_defs),
1485 warning_name, value);
1486 }
1487}
1488
1489static const FlagDef flag_defs[] = {
1490 { offsetof(TCCState, char_is_unsigned), 0, "unsigned-char" },
1491 { offsetof(TCCState, char_is_unsigned), FD_INVERT, "signed-char" },
1492 { offsetof(TCCState, nocommon), FD_INVERT, "common" },
1493 { offsetof(TCCState, leading_underscore), 0, "leading-underscore" },
dec95935 seyko2015-04-10 06:31:58 +03001494 { offsetof(TCCState, ms_extensions), 0, "ms-extensions" },
5c9dde72 seyko2015-04-10 23:44:10 +03001495 { offsetof(TCCState, old_struct_init_code), 0, "old-struct-init-code" },
dcb36587 seyko2015-04-12 15:32:03 +03001496 { offsetof(TCCState, dollars_in_identifiers), 0, "dollars-in-identifiers" },
9dc9cbf3 grischka2009-05-05 20:18:53 +02001497};
1498
1499/* set/reset a flag */
05108a3b grischka2013-02-12 19:13:28 +01001500static int tcc_set_flag(TCCState *s, const char *flag_name, int value)
9dc9cbf3 grischka2009-05-05 20:18:53 +02001501{
1502 return set_flag(s, flag_defs, countof(flag_defs),
1503 flag_name, value);
1504}
1505
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001506
05108a3b grischka2013-02-12 19:13:28 +01001507static int strstart(const char *val, const char **str)
78e83d87
DR
Detlef Riekenberg2010-04-05 21:21:58 +02001508{
1509 const char *p, *q;
05108a3b grischka2013-02-12 19:13:28 +01001510 p = *str;
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001511 q = val;
05108a3b grischka2013-02-12 19:13:28 +01001512 while (*q) {
78e83d87
DR
Detlef Riekenberg2010-04-05 21:21:58 +02001513 if (*p != *q)
1514 return 0;
1515 p++;
1516 q++;
1517 }
05108a3b grischka2013-02-12 19:13:28 +01001518 *str = p;
78e83d87
DR
Detlef Riekenberg2010-04-05 21:21:58 +02001519 return 1;
1520}
1521
7901d1e3
KS
Kirill Smelkov2010-08-12 23:30:21 +04001522/* Like strstart, but automatically takes into account that ld options can
1523 *
1524 * - start with double or single dash (e.g. '--soname' or '-soname')
1525 * - arguments can be given as separate or after '=' (e.g. '-Wl,-soname,x.so'
1526 * or '-Wl,-soname=x.so')
1527 *
1528 * you provide `val` always in 'option[=]' form (no leading -)
1529 */
05108a3b grischka2013-02-12 19:13:28 +01001530static int link_option(const char *str, const char *val, const char **ptr)
7901d1e3
KS
Kirill Smelkov2010-08-12 23:30:21 +04001531{
1532 const char *p, *q;
1533
1534 /* there should be 1 or 2 dashes */
1535 if (*str++ != '-')
1536 return 0;
1537 if (*str == '-')
1538 str++;
1539
1540 /* then str & val should match (potentialy up to '=') */
1541 p = str;
1542 q = val;
1543
1544 while (*q != '\0' && *q != '=') {
1545 if (*p != *q)
1546 return 0;
1547 p++;
1548 q++;
1549 }
1550
1551 /* '=' near eos means ',' or '=' is ok */
1552 if (*q == '=') {
1553 if (*p != ',' && *p != '=')
1554 return 0;
1555 p++;
1556 q++;
1557 }
1558
1559 if (ptr)
05108a3b grischka2013-02-12 19:13:28 +01001560 *ptr = p;
7901d1e3
KS
Kirill Smelkov2010-08-12 23:30:21 +04001561 return 1;
1562}
1563
05108a3b grischka2013-02-12 19:13:28 +01001564static const char *skip_linker_arg(const char **str)
1565{
1566 const char *s1 = *str;
1567 const char *s2 = strchr(s1, ',');
1568 *str = s2 ? s2++ : (s2 = s1 + strlen(s1));
1569 return s2;
1570}
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001571
05108a3b grischka2013-02-12 19:13:28 +01001572static char *copy_linker_arg(const char *p)
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001573{
05108a3b grischka2013-02-12 19:13:28 +01001574 const char *q = p;
1575 skip_linker_arg(&q);
1576 return pstrncpy(tcc_malloc(q - p + 1), p, q - p);
1577}
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001578
05108a3b grischka2013-02-12 19:13:28 +01001579/* set linker options */
1580static int tcc_set_linker(TCCState *s, const char *option)
1581{
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001582 while (option && *option) {
05108a3b grischka2013-02-12 19:13:28 +01001583
1584 const char *p = option;
1585 char *end = NULL;
1586 int ignoring = 0;
1587
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001588 if (link_option(option, "Bsymbolic", &p)) {
263dc93c grischka2013-02-04 15:10:08 +01001589 s->symbolic = 1;
176876e5 Gabriel Corneanu2012-04-18 09:48:57 +02001590 } else if (link_option(option, "nostdlib", &p)) {
263dc93c grischka2013-02-04 15:10:08 +01001591 s->nostdlib = 1;
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001592 } else if (link_option(option, "fini=", &p)) {
05108a3b grischka2013-02-12 19:13:28 +01001593 s->fini_symbol = copy_linker_arg(p);
1594 ignoring = 1;
1595 } else if (link_option(option, "image-base=", &p)
1596 || link_option(option, "Ttext=", &p)) {
a0db7162 Andrew Mulbrook2012-03-03 11:10:15 -06001597 s->text_addr = strtoull(p, &end, 16);
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001598 s->has_text_addr = 1;
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001599 } else if (link_option(option, "init=", &p)) {
05108a3b grischka2013-02-12 19:13:28 +01001600 s->init_symbol = copy_linker_arg(p);
1601 ignoring = 1;
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001602 } else if (link_option(option, "oformat=", &p)) {
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001603#if defined(TCC_TARGET_PE)
05108a3b grischka2013-02-12 19:13:28 +01001604 if (strstart("pe-", &p)) {
b14ef0e2 Edmund Grimley Evans2015-02-13 18:58:31 +00001605#elif defined(TCC_TARGET_ARM64) || defined(TCC_TARGET_X86_64)
05108a3b grischka2013-02-12 19:13:28 +01001606 if (strstart("elf64-", &p)) {
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001607#else
05108a3b grischka2013-02-12 19:13:28 +01001608 if (strstart("elf32-", &p)) {
78e83d87
DR
Detlef Riekenberg2010-04-05 21:21:58 +02001609#endif
1610 s->output_format = TCC_OUTPUT_FORMAT_ELF;
1611 } else if (!strcmp(p, "binary")) {
1612 s->output_format = TCC_OUTPUT_FORMAT_BINARY;
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001613#ifdef TCC_TARGET_COFF
05108a3b grischka2013-02-12 19:13:28 +01001614 } else if (!strcmp(p, "coff")) {
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001615 s->output_format = TCC_OUTPUT_FORMAT_COFF;
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001616#endif
05108a3b grischka2013-02-12 19:13:28 +01001617 } else
1618 goto err;
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001619
ba286136
AE
Austin English2014-03-06 12:29:19 -08001620 } else if (link_option(option, "as-needed", &p)) {
1621 ignoring = 1;
1622 } else if (link_option(option, "O", &p)) {
1623 ignoring = 1;
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001624 } else if (link_option(option, "rpath=", &p)) {
05108a3b grischka2013-02-12 19:13:28 +01001625 s->rpath = copy_linker_arg(p);
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001626 } else if (link_option(option, "section-alignment=", &p)) {
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001627 s->section_align = strtoul(p, &end, 16);
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001628 } else if (link_option(option, "soname=", &p)) {
05108a3b grischka2013-02-12 19:13:28 +01001629 s->soname = copy_linker_arg(p);
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001630#ifdef TCC_TARGET_PE
45184e01 grischka2011-07-11 18:47:16 +02001631 } else if (link_option(option, "file-alignment=", &p)) {
1632 s->pe_file_align = strtoul(p, &end, 16);
1633 } else if (link_option(option, "stack=", &p)) {
1634 s->pe_stack_size = strtoul(p, &end, 10);
7901d1e3 Kirill Smelkov2010-08-12 23:30:21 +04001635 } else if (link_option(option, "subsystem=", &p)) {
78e83d87
DR
Detlef Riekenberg2010-04-05 21:21:58 +02001636#if defined(TCC_TARGET_I386) || defined(TCC_TARGET_X86_64)
1637 if (!strcmp(p, "native")) {
1638 s->pe_subsystem = 1;
1639 } else if (!strcmp(p, "console")) {
1640 s->pe_subsystem = 3;
1641 } else if (!strcmp(p, "gui")) {
1642 s->pe_subsystem = 2;
1643 } else if (!strcmp(p, "posix")) {
1644 s->pe_subsystem = 7;
1645 } else if (!strcmp(p, "efiapp")) {
1646 s->pe_subsystem = 10;
1647 } else if (!strcmp(p, "efiboot")) {
1648 s->pe_subsystem = 11;
1649 } else if (!strcmp(p, "efiruntime")) {
1650 s->pe_subsystem = 12;
1651 } else if (!strcmp(p, "efirom")) {
1652 s->pe_subsystem = 13;
1653#elif defined(TCC_TARGET_ARM)
1654 if (!strcmp(p, "wince")) {
1655 s->pe_subsystem = 9;
1656#endif
05108a3b grischka2013-02-12 19:13:28 +01001657 } else
1658 goto err;
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001659#endif
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001660 } else
05108a3b grischka2013-02-12 19:13:28 +01001661 goto err;
1662
1663 if (ignoring && s->warn_unsupported) err: {
1664 char buf[100], *e;
1665 pstrcpy(buf, sizeof buf, e = copy_linker_arg(option)), tcc_free(e);
1666 if (ignoring)
1667 tcc_warning("unsupported linker option '%s'", buf);
1668 else
1669 tcc_error("unsupported linker option '%s'", buf);
1670 }
1671 option = skip_linker_arg(&p);
78e83d87 Detlef Riekenberg2010-04-05 21:21:58 +02001672 }
05108a3b grischka2013-02-12 19:13:28 +01001673 return 0;
78e83d87
DR
Detlef Riekenberg2010-04-05 21:21:58 +02001674}
1675
05108a3b grischka2013-02-12 19:13:28 +01001676typedef struct TCCOption {
1677 const char *name;
1678 uint16_t index;
1679 uint16_t flags;
1680} TCCOption;
1681
1682enum {
1683 TCC_OPTION_HELP,
1684 TCC_OPTION_I,
1685 TCC_OPTION_D,
1686 TCC_OPTION_U,
50cdccf3 seyko2015-03-03 14:19:14 +03001687 TCC_OPTION_P,
05108a3b grischka2013-02-12 19:13:28 +01001688 TCC_OPTION_L,
1689 TCC_OPTION_B,
1690 TCC_OPTION_l,
1691 TCC_OPTION_bench,
1692 TCC_OPTION_bt,
1693 TCC_OPTION_b,
1694 TCC_OPTION_g,
1695 TCC_OPTION_c,
724425ad seyko2015-03-23 20:58:27 +03001696 TCC_OPTION_dumpversion,
b7b9f9f5 seyko2015-03-03 14:25:57 +03001697 TCC_OPTION_d,
b6247d1f Thomas Preud'homme2014-01-07 15:23:54 +08001698 TCC_OPTION_float_abi,
05108a3b grischka2013-02-12 19:13:28 +01001699 TCC_OPTION_static,
e374a733 seyko2015-03-03 16:25:02 +03001700 TCC_OPTION_std,
05108a3b grischka2013-02-12 19:13:28 +01001701 TCC_OPTION_shared,
1702 TCC_OPTION_soname,
1703 TCC_OPTION_o,
1704 TCC_OPTION_r,
1705 TCC_OPTION_s,
1706 TCC_OPTION_Wl,
1707 TCC_OPTION_W,
1708 TCC_OPTION_O,
1709 TCC_OPTION_m,
1710 TCC_OPTION_f,
1711 TCC_OPTION_isystem,
1712 TCC_OPTION_nostdinc,
1713 TCC_OPTION_nostdlib,
1714 TCC_OPTION_print_search_dirs,
1715 TCC_OPTION_rdynamic,
1716 TCC_OPTION_pedantic,
1717 TCC_OPTION_pthread,
1718 TCC_OPTION_run,
1719 TCC_OPTION_v,
1720 TCC_OPTION_w,
1721 TCC_OPTION_pipe,
1722 TCC_OPTION_E,
1723 TCC_OPTION_MD,
1724 TCC_OPTION_MF,
1725 TCC_OPTION_x,
1726};
1727
1728#define TCC_OPTION_HAS_ARG 0x0001
1729#define TCC_OPTION_NOSEP 0x0002 /* cannot have space before option and arg */
1730
1731static const TCCOption tcc_options[] = {
1732 { "h", TCC_OPTION_HELP, 0 },
1733 { "-help", TCC_OPTION_HELP, 0 },
1734 { "?", TCC_OPTION_HELP, 0 },
1735 { "I", TCC_OPTION_I, TCC_OPTION_HAS_ARG },
1736 { "D", TCC_OPTION_D, TCC_OPTION_HAS_ARG },
1737 { "U", TCC_OPTION_U, TCC_OPTION_HAS_ARG },
50cdccf3 seyko2015-03-03 14:19:14 +03001738 { "P", TCC_OPTION_P, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
05108a3b grischka2013-02-12 19:13:28 +01001739 { "L", TCC_OPTION_L, TCC_OPTION_HAS_ARG },
1740 { "B", TCC_OPTION_B, TCC_OPTION_HAS_ARG },
1741 { "l", TCC_OPTION_l, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1742 { "bench", TCC_OPTION_bench, 0 },
1743#ifdef CONFIG_TCC_BACKTRACE
1744 { "bt", TCC_OPTION_bt, TCC_OPTION_HAS_ARG },
1745#endif
1746#ifdef CONFIG_TCC_BCHECK
1747 { "b", TCC_OPTION_b, 0 },
1748#endif
1749 { "g", TCC_OPTION_g, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1750 { "c", TCC_OPTION_c, 0 },
724425ad seyko2015-03-23 20:58:27 +03001751 { "dumpversion", TCC_OPTION_dumpversion, 0},
b7b9f9f5 seyko2015-03-03 14:25:57 +03001752 { "d", TCC_OPTION_d, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
b6247d1f
TP
Thomas Preud'homme2014-01-07 15:23:54 +08001753#ifdef TCC_TARGET_ARM
1754 { "mfloat-abi", TCC_OPTION_float_abi, TCC_OPTION_HAS_ARG },
1755#endif
05108a3b grischka2013-02-12 19:13:28 +01001756 { "static", TCC_OPTION_static, 0 },
e374a733 seyko2015-03-03 16:25:02 +03001757 { "std", TCC_OPTION_std, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
05108a3b grischka2013-02-12 19:13:28 +01001758 { "shared", TCC_OPTION_shared, 0 },
1759 { "soname", TCC_OPTION_soname, TCC_OPTION_HAS_ARG },
1760 { "o", TCC_OPTION_o, TCC_OPTION_HAS_ARG },
1761 { "pedantic", TCC_OPTION_pedantic, 0},
1762 { "pthread", TCC_OPTION_pthread, 0},
1763 { "run", TCC_OPTION_run, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1764 { "rdynamic", TCC_OPTION_rdynamic, 0 },
1765 { "r", TCC_OPTION_r, 0 },
1766 { "s", TCC_OPTION_s, 0 },
1767 { "Wl,", TCC_OPTION_Wl, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1768 { "W", TCC_OPTION_W, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1769 { "O", TCC_OPTION_O, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1770 { "m", TCC_OPTION_m, TCC_OPTION_HAS_ARG },
1771 { "f", TCC_OPTION_f, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1772 { "isystem", TCC_OPTION_isystem, TCC_OPTION_HAS_ARG },
1773 { "nostdinc", TCC_OPTION_nostdinc, 0 },
1774 { "nostdlib", TCC_OPTION_nostdlib, 0 },
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001775 { "print-search-dirs", TCC_OPTION_print_search_dirs, 0 },
05108a3b grischka2013-02-12 19:13:28 +01001776 { "v", TCC_OPTION_v, TCC_OPTION_HAS_ARG | TCC_OPTION_NOSEP },
1777 { "w", TCC_OPTION_w, 0 },
1778 { "pipe", TCC_OPTION_pipe, 0},
1779 { "E", TCC_OPTION_E, 0},
1780 { "MD", TCC_OPTION_MD, 0},
1781 { "MF", TCC_OPTION_MF, TCC_OPTION_HAS_ARG },
1782 { "x", TCC_OPTION_x, TCC_OPTION_HAS_ARG },
1783 { NULL, 0, 0 },
1784};
1785
1786static void parse_option_D(TCCState *s1, const char *optarg)
88a3ccab grischka2009-12-20 01:53:49 +01001787{
05108a3b grischka2013-02-12 19:13:28 +01001788 char *sym = tcc_strdup(optarg);
1789 char *value = strchr(sym, '=');
1790 if (value)
1791 *value++ = '\0';
1792 tcc_define_symbol(s1, sym, value);
1793 tcc_free(sym);
88a3ccab grischka2009-12-20 01:53:49 +01001794}
1795
05364072 seyko2015-04-12 15:35:37 +03001796static void args_parser_add_file(TCCState *s, const char* filename, int filetype)
1797{
1798 int len = strlen(filename);
1799 char *p = tcc_malloc(len + 2);
1800 if (filetype) {
1801 *p = filetype;
1802 }
1803 else {
1804 /* use a file extension to detect a filetype */
1805 const char *ext = tcc_fileextension(filename);
1806 if (ext[0]) {
1807 ext++;
1808 if (!strcmp(ext, "S"))
1809 *p = TCC_FILETYPE_ASM_PP;
1810 else
1811 if (!strcmp(ext, "s"))
1812 *p = TCC_FILETYPE_ASM;
1813 else
1814 if (!PATHCMP(ext, "c") || !PATHCMP(ext, "i"))
1815 *p = TCC_FILETYPE_C;
1816 else
1817 *p = TCC_FILETYPE_BINARY;
1818 }
1819 else {
1820 *p = TCC_FILETYPE_C;
1821 }
1822 }
1823 strcpy(p+1, filename);
1824 dynarray_add((void ***)&s->files, &s->nb_files, p);
1825}
1826
05108a3b grischka2013-02-12 19:13:28 +01001827PUB_FUNC int tcc_parse_args(TCCState *s, int argc, char **argv)
bdae4a59 Kirill Smelkov2010-06-16 16:54:24 +04001828{
05108a3b grischka2013-02-12 19:13:28 +01001829 const TCCOption *popt;
1830 const char *optarg, *r;
1831 int run = 0;
1832 int pthread = 0;
1833 int optind = 0;
05364072 seyko2015-04-12 15:35:37 +03001834 int filetype = 0;
e6f3bf7f grischka2011-08-06 16:08:46 +02001835
05108a3b grischka2013-02-12 19:13:28 +01001836 /* collect -Wl options for input such as "-Wl,-rpath -Wl,<path>" */
1837 CString linker_arg;
1838 cstr_new(&linker_arg);
bdae4a59 Kirill Smelkov2010-06-16 16:54:24 +04001839
05108a3b grischka2013-02-12 19:13:28 +01001840 while (optind < argc) {
0c928da9 Kirill Smelkov2010-06-20 20:08:12 +04001841
05108a3b grischka2013-02-12 19:13:28 +01001842 r = argv[optind++];
1843 if (r[0] != '-' || r[1] == '\0') {
05364072 seyko2015-04-12 15:35:37 +03001844 args_parser_add_file(s, r, filetype);
05108a3b grischka2013-02-12 19:13:28 +01001845 if (run) {
1846 optind--;
1847 /* argv[0] will be this file */
1848 break;
1849 }
1850 continue;
1851 }
0c928da9 Kirill Smelkov2010-06-20 20:08:12 +04001852
05108a3b grischka2013-02-12 19:13:28 +01001853 /* find option in table */
1854 for(popt = tcc_options; ; ++popt) {
1855 const char *p1 = popt->name;
1856 const char *r1 = r + 1;
1857 if (p1 == NULL)
1858 tcc_error("invalid option -- '%s'", r);
1859 if (!strstart(p1, &r1))
1860 continue;
1861 optarg = r1;
1862 if (popt->flags & TCC_OPTION_HAS_ARG) {
1863 if (*r1 == '\0' && !(popt->flags & TCC_OPTION_NOSEP)) {
1864 if (optind >= argc)
1865 tcc_error("argument to '%s' is missing", r);
1866 optarg = argv[optind++];
1867 }
1868 } else if (*r1 != '\0')
1869 continue;
1870 break;
1871 }
0c928da9 Kirill Smelkov2010-06-20 20:08:12 +04001872
05108a3b grischka2013-02-12 19:13:28 +01001873 switch(popt->index) {
1874 case TCC_OPTION_HELP:
1875 return 0;
1876 case TCC_OPTION_I:
1877 if (tcc_add_include_path(s, optarg) < 0)
1878 tcc_error("too many include paths");
1879 break;
1880 case TCC_OPTION_D:
1881 parse_option_D(s, optarg);
1882 break;
1883 case TCC_OPTION_U:
1884 tcc_undefine_symbol(s, optarg);
1885 break;
1886 case TCC_OPTION_L:
1887 tcc_add_library_path(s, optarg);
1888 break;
1889 case TCC_OPTION_B:
1890 /* set tcc utilities path (mainly for tcc development) */
1891 tcc_set_lib_path(s, optarg);
1892 break;
1893 case TCC_OPTION_l:
05364072 seyko2015-04-12 15:35:37 +03001894 args_parser_add_file(s, r, TCC_FILETYPE_BINARY);
05108a3b grischka2013-02-12 19:13:28 +01001895 s->nb_libraries++;
1896 break;
1897 case TCC_OPTION_pthread:
1898 parse_option_D(s, "_REENTRANT");
1899 pthread = 1;
1900 break;
1901 case TCC_OPTION_bench:
1902 s->do_bench = 1;
1903 break;
1904#ifdef CONFIG_TCC_BACKTRACE
1905 case TCC_OPTION_bt:
1906 tcc_set_num_callers(atoi(optarg));
1907 break;
1908#endif
1909#ifdef CONFIG_TCC_BCHECK
1910 case TCC_OPTION_b:
1911 s->do_bounds_check = 1;
1912 s->do_debug = 1;
1913 break;
1914#endif
1915 case TCC_OPTION_g:
1916 s->do_debug = 1;
1917 break;
1918 case TCC_OPTION_c:
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04001919 if (s->output_type)
c334b591 seyko2015-01-06 22:19:45 +03001920 tcc_warning("-c: some compiler action already specified (%d)", s->output_type);
05108a3b grischka2013-02-12 19:13:28 +01001921 s->output_type = TCC_OUTPUT_OBJ;
1922 break;
b7b9f9f5 seyko2015-03-03 14:25:57 +03001923 case TCC_OPTION_d:
e50d68e4
SM
Steven G. Messervey2015-04-15 21:24:15 -04001924 if (*optarg != 'D') {
1925 if (s->warn_unsupported)
1926 goto unsupported_option;
b7b9f9f5 seyko2015-03-03 14:25:57 +03001927 tcc_error("invalid option -- '%s'", r);
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04001928 }
b7b9f9f5 seyko2015-03-03 14:25:57 +03001929 s->dflag = 1;
1930 break;
b6247d1f
TP
Thomas Preud'homme2014-01-07 15:23:54 +08001931#ifdef TCC_TARGET_ARM
1932 case TCC_OPTION_float_abi:
1933 /* tcc doesn't support soft float yet */
1934 if (!strcmp(optarg, "softfp")) {
1935 s->float_abi = ARM_SOFTFP_FLOAT;
1936 tcc_undefine_symbol(s, "__ARM_PCS_VFP");
1937 } else if (!strcmp(optarg, "hard"))
1938 s->float_abi = ARM_HARD_FLOAT;
1939 else
1940 tcc_error("unsupported float abi '%s'", optarg);
1941 break;
1942#endif
05108a3b grischka2013-02-12 19:13:28 +01001943 case TCC_OPTION_static:
1944 s->static_link = 1;
1945 break;
e374a733 seyko2015-03-03 16:25:02 +03001946 case TCC_OPTION_std:
e50d68e4
SM
Steven G. Messervey2015-04-15 21:24:15 -04001947 /* silently ignore, a current purpose:
1948 allow to use a tcc as a reference compiler for "make test" */
e374a733 seyko2015-03-03 16:25:02 +03001949 break;
05108a3b grischka2013-02-12 19:13:28 +01001950 case TCC_OPTION_shared:
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04001951 if (s->output_type)
c334b591 seyko2015-01-06 22:19:45 +03001952 tcc_warning("-shared: some compiler action already specified (%d)", s->output_type);
05108a3b grischka2013-02-12 19:13:28 +01001953 s->output_type = TCC_OUTPUT_DLL;
1954 break;
1955 case TCC_OPTION_soname:
1956 s->soname = tcc_strdup(optarg);
1957 break;
1958 case TCC_OPTION_m:
1959 s->option_m = tcc_strdup(optarg);
1960 break;
1961 case TCC_OPTION_o:
1962 s->outfile = tcc_strdup(optarg);
1963 break;
1964 case TCC_OPTION_r:
1965 /* generate a .o merging several output files */
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04001966 if (s->output_type)
c334b591 seyko2015-01-06 22:19:45 +03001967 tcc_warning("-r: some compiler action already specified (%d)", s->output_type);
05108a3b grischka2013-02-12 19:13:28 +01001968 s->option_r = 1;
1969 s->output_type = TCC_OUTPUT_OBJ;
1970 break;
1971 case TCC_OPTION_isystem:
1972 tcc_add_sysinclude_path(s, optarg);
1973 break;
1974 case TCC_OPTION_nostdinc:
1975 s->nostdinc = 1;
1976 break;
1977 case TCC_OPTION_nostdlib:
1978 s->nostdlib = 1;
1979 break;
1980 case TCC_OPTION_print_search_dirs:
1981 s->print_search_dirs = 1;
1982 break;
1983 case TCC_OPTION_run:
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04001984 if (s->output_type)
c334b591 seyko2015-01-06 22:19:45 +03001985 tcc_warning("-run: some compiler action already specified (%d)", s->output_type);
05108a3b grischka2013-02-12 19:13:28 +01001986 s->output_type = TCC_OUTPUT_MEMORY;
1987 tcc_set_options(s, optarg);
1988 run = 1;
1989 break;
1990 case TCC_OPTION_v:
1991 do ++s->verbose; while (*optarg++ == 'v');
1992 break;
1993 case TCC_OPTION_f:
1994 if (tcc_set_flag(s, optarg, 1) < 0 && s->warn_unsupported)
1995 goto unsupported_option;
1996 break;
1997 case TCC_OPTION_W:
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06001998 if (tcc_set_warning(s, optarg, 1) < 0 &&
05108a3b grischka2013-02-12 19:13:28 +01001999 s->warn_unsupported)
2000 goto unsupported_option;
2001 break;
2002 case TCC_OPTION_w:
2003 s->warn_none = 1;
2004 break;
2005 case TCC_OPTION_rdynamic:
2006 s->rdynamic = 1;
2007 break;
2008 case TCC_OPTION_Wl:
2009 if (linker_arg.size)
2010 --linker_arg.size, cstr_ccat(&linker_arg, ',');
2011 cstr_cat(&linker_arg, optarg);
2012 cstr_ccat(&linker_arg, '\0');
2013 break;
2014 case TCC_OPTION_E:
e50d68e4 Steven G. Messervey2015-04-15 21:24:15 -04002015 if (s->output_type)
c334b591 seyko2015-01-06 22:19:45 +03002016 tcc_warning("-E: some compiler action already specified (%d)", s->output_type);
05108a3b grischka2013-02-12 19:13:28 +01002017 s->output_type = TCC_OUTPUT_PREPROCESS;
2018 break;
50cdccf3 seyko2015-03-03 14:19:14 +03002019 case TCC_OPTION_P:
2020 s->Pflag = atoi(optarg) + 1;
2021 break;
05108a3b grischka2013-02-12 19:13:28 +01002022 case TCC_OPTION_MD:
2023 s->gen_deps = 1;
2024 break;
2025 case TCC_OPTION_MF:
2026 s->deps_outfile = tcc_strdup(optarg);
2027 break;
062efe6a
UJ
Urs Janssen2013-02-14 08:24:51 +01002028 case TCC_OPTION_dumpversion:
2029 printf ("%s\n", TCC_VERSION);
2030 exit(0);
5cd4393a seyko2015-04-10 06:53:48 +03002031 case TCC_OPTION_s:
2032 s->do_strip = 1;
2033 break;
05364072 seyko2015-04-12 15:35:37 +03002034 case TCC_OPTION_x:
2035 if (*optarg == 'c')
2036 filetype = TCC_FILETYPE_C;
2037 else
2038 if (*optarg == 'a')
2039 filetype = TCC_FILETYPE_ASM_PP;
2040 else
2041 if (*optarg == 'n')
2042 filetype = 0;
2043 else
2044 tcc_warning("unsupported language '%s'", optarg);
2045 break;
05108a3b grischka2013-02-12 19:13:28 +01002046 case TCC_OPTION_O:
2047 case TCC_OPTION_pedantic:
2048 case TCC_OPTION_pipe:
05108a3b grischka2013-02-12 19:13:28 +01002049 /* ignored */
2050 break;
2051 default:
2052 if (s->warn_unsupported) {
2053 unsupported_option:
2054 tcc_warning("unsupported option '%s'", r);
2055 }
2056 break;
2057 }
0c928da9
KS
Kirill Smelkov2010-06-20 20:08:12 +04002058 }
2059
576aeb3a seyko2015-05-03 14:17:18 +03002060 if (s->output_type == 0)
2061 s->output_type = TCC_OUTPUT_EXE;
2062
05108a3b grischka2013-02-12 19:13:28 +01002063 if (pthread && s->output_type != TCC_OUTPUT_OBJ)
2064 tcc_set_options(s, "-lpthread");
0c928da9 Kirill Smelkov2010-06-20 20:08:12 +04002065
576aeb3a seyko2015-05-03 14:17:18 +03002066 if (s->output_type == TCC_OUTPUT_EXE)
2067 tcc_set_linker(s, (const char *)linker_arg.data);
05108a3b grischka2013-02-12 19:13:28 +01002068 cstr_free(&linker_arg);
0c928da9 Kirill Smelkov2010-06-20 20:08:12 +04002069
05108a3b grischka2013-02-12 19:13:28 +01002070 return optind;
2071}
2072
2073LIBTCCAPI int tcc_set_options(TCCState *s, const char *str)
2074{
2075 const char *s1;
2076 char **argv, *arg;
2077 int argc, len;
2078 int ret;
2079
2080 argc = 0, argv = NULL;
2081 for(;;) {
2082 while (is_space(*str))
2083 str++;
2084 if (*str == '\0')
2085 break;
2086 s1 = str;
2087 while (*str != '\0' && !is_space(*str))
2088 str++;
2089 len = str - s1;
2090 arg = tcc_malloc(len + 1);
2091 pstrncpy(arg, s1, len);
2092 dynarray_add((void ***)&argv, &argc, arg);
2093 }
2094 ret = tcc_parse_args(s, argc, argv);
2095 dynarray_reset(&argv, &argc);
2096 return ret;
0c928da9 Kirill Smelkov2010-06-20 20:08:12 +04002097}
8042121d grischka2013-02-10 00:38:40 +01002098
2099PUB_FUNC void tcc_print_stats(TCCState *s, int64_t total_time)
2100{
2101 double tt;
2102 tt = (double)total_time / 1000000.0;
2103 if (tt < 0.001)
2104 tt = 0.001;
2105 if (total_bytes < 1)
2106 total_bytes = 1;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06002107 printf("%d idents, %d lines, %d bytes, %0.3f s, %d lines/s, %0.1f MB/s\n",
8042121d grischka2013-02-10 00:38:40 +01002108 tok_ident - TOK_IDENT, total_lines, total_bytes,
2109 tt, (int)(total_lines / tt),
2110 total_bytes / tt / 1000000.0);
2111}
0ad857c8
AA
Andrew Aladjev2013-02-19 14:47:36 +03002112
2113PUB_FUNC void tcc_set_environment(TCCState *s)
2114{
2115 char * path;
0650ab01 Joseph Poirier2013-11-08 13:24:15 -06002116
0ad857c8
AA
Andrew Aladjev2013-02-19 14:47:36 +03002117 path = getenv("C_INCLUDE_PATH");
2118 if(path != NULL) {
2119 tcc_add_include_path(s, path);
2120 }
2121 path = getenv("CPATH");
2122 if(path != NULL) {
2123 tcc_add_include_path(s, path);
2124 }
183b2ab1 Urs Janssen2013-02-19 14:41:58 +01002125 path = getenv("LIBRARY_PATH");
0ad857c8
AA
Andrew Aladjev2013-02-19 14:47:36 +03002126 if(path != NULL) {
2127 tcc_add_library_path(s, path);
2128 }
183b2ab1 Urs Janssen2013-02-19 14:41:58 +01002129}