2 * TCC - Tiny C Compiler - Support for -run switch
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
24 #define ucontext_t CONTEXT
27 static void set_pages_executable(void *ptr
, unsigned long length
);
28 static void set_exception_handler(void);
29 static int rt_get_caller_pc(uplong
*paddr
, ucontext_t
*uc
, int level
);
30 static void rt_error(ucontext_t
*uc
, const char *fmt
, ...);
31 static int tcc_relocate_ex(TCCState
*s1
, void *ptr
);
34 static void win64_add_function_table(TCCState
*s1
);
37 /* ------------------------------------------------------------- */
38 /* Do all relocations (needed before using tcc_get_symbol())
39 Returns -1 on error. */
41 int tcc_relocate(TCCState
*s1
)
45 /* Use mmap instead of malloc for Selinux
46 Ref http://www.gnu.org/s/libc/manual/html_node/File-Size.html */
47 char tmpfname
[] = "/tmp/.tccrunXXXXXX";
48 int fd
= mkstemp (tmpfname
);
49 if ((ret
= tcc_relocate_ex(s1
,NULL
)) < 0)return -1;
51 unlink (tmpfname
); ftruncate (fd
, s1
->mem_size
);
52 s1
->write_mem
= mmap (NULL
, ret
, PROT_READ
|PROT_WRITE
,
54 if(s1
->write_mem
== MAP_FAILED
){
55 error("/tmp not writeable");
58 s1
->runtime_mem
= mmap (NULL
, ret
, PROT_READ
|PROT_EXEC
,
60 if(s1
->runtime_mem
== MAP_FAILED
){
61 error("/tmp not executable");
64 ret
= tcc_relocate_ex(s1
, s1
->write_mem
);
66 ret
= tcc_relocate_ex(s1
, NULL
);
68 s1
->runtime_mem
= tcc_malloc(ret
);
69 ret
= tcc_relocate_ex(s1
, s1
->runtime_mem
);
75 /* launch the compiled program with the given arguments */
76 int tcc_run(TCCState
*s1
, int argc
, char **argv
)
78 int (*prog_main
)(int, char **);
80 if (tcc_relocate(s1
) < 0)
83 prog_main
= tcc_get_symbol_err(s1
, "main");
85 #ifdef CONFIG_TCC_BACKTRACE
87 set_exception_handler();
90 #ifdef CONFIG_TCC_BCHECK
91 if (s1
->do_bounds_check
) {
92 void (*bound_init
)(void);
93 void (*bound_exit
)(void);
95 /* set error function */
96 rt_bound_error_msg
= tcc_get_symbol_err(s1
, "__bound_error_msg");
97 rt_prog_main
= prog_main
;
98 /* XXX: use .init section so that it also work in binary ? */
99 bound_init
= tcc_get_symbol_err(s1
, "__bound_init");
100 bound_exit
= tcc_get_symbol_err(s1
, "__bound_exit");
102 ret
= (*prog_main
)(argc
, argv
);
110 unsigned char *p
= tcc_get_symbol(s1
, "tinyc_no_getbp");
114 return (*prog_main
)(argc
, argv
);
118 /* relocate code. Return -1 on error, required size if ptr is NULL,
119 otherwise copy code into buffer passed by the caller */
120 static int tcc_relocate_ex(TCCState
*s1
, void *ptr
)
123 unsigned long offset
, length
;
127 if (0 == s1
->runtime_added
) {
128 s1
->runtime_added
= 1;
131 pe_output_file(s1
, NULL
);
134 relocate_common_syms();
135 tcc_add_linker_symbols(s1
);
136 build_got_entries(s1
);
142 offset
= 0, mem
= (uplong
)ptr
;
143 for(i
= 1; i
< s1
->nb_sections
; i
++) {
145 if (0 == (s
->sh_flags
& SHF_ALLOC
))
147 length
= s
->data_offset
;
148 s
->sh_addr
= mem
? (mem
+ offset
+ 15) & ~15 : 0;
149 offset
= (offset
+ length
+ 15) & ~15;
153 /* relocate symbols */
154 relocate_syms(s1
, 1);
158 #if (defined TCC_TARGET_X86_64 || defined TCC_TARGET_ARM) && !defined TCC_TARGET_PE
159 s1
->runtime_plt_and_got_offset
= 0;
160 s1
->runtime_plt_and_got
= (char *)(mem
+ offset
);
161 /* double the size of the buffer for got and plt entries
162 XXX: calculate exact size for them? */
169 /* relocate each section */
170 for(i
= 1; i
< s1
->nb_sections
; i
++) {
173 relocate_section(s1
, s
);
176 for(i
= 1; i
< s1
->nb_sections
; i
++) {
178 if (0 == (s
->sh_flags
& SHF_ALLOC
))
180 length
= s
->data_offset
;
181 // printf("%-12s %08x %04x\n", s->name, s->sh_addr, length);
182 ptr
= (void*)(uplong
)s
->sh_addr
;
183 if (NULL
== s
->data
|| s
->sh_type
== SHT_NOBITS
)
184 memset(ptr
, 0, length
);
186 memcpy(ptr
, s
->data
, length
);
187 /* mark executable sections as executable in memory */
188 if (s
->sh_flags
& SHF_EXECINSTR
)
189 set_pages_executable(ptr
, length
);
192 #if (defined TCC_TARGET_X86_64 || defined TCC_TARGET_ARM) && !defined TCC_TARGET_PE
193 set_pages_executable(s1
->runtime_plt_and_got
,
194 s1
->runtime_plt_and_got_offset
);
198 win64_add_function_table(s1
);
203 /* ------------------------------------------------------------- */
204 /* allow to run code in memory */
206 static void set_pages_executable(void *ptr
, unsigned long length
)
209 unsigned long old_protect
;
210 VirtualProtect(ptr
, length
, PAGE_EXECUTE_READWRITE
, &old_protect
);
212 unsigned long start
, end
;
213 start
= (uplong
)ptr
& ~(PAGESIZE
- 1);
214 end
= (uplong
)ptr
+ length
;
215 end
= (end
+ PAGESIZE
- 1) & ~(PAGESIZE
- 1);
216 mprotect((void *)start
, end
- start
, PROT_READ
| PROT_WRITE
| PROT_EXEC
);
220 /* ------------------------------------------------------------- */
221 #ifdef CONFIG_TCC_BACKTRACE
223 /* print the position in the source file of PC value 'pc' by reading
224 the stabs debug information */
225 static uplong
rt_printline(uplong wanted_pc
, const char *msg
)
227 char func_name
[128], last_func_name
[128];
228 uplong func_addr
, last_pc
, pc
;
229 const char *incl_files
[INCLUDE_STACK_SIZE
];
230 int incl_index
, len
, last_line_num
, i
;
233 Stab_Sym
*stab_sym
= NULL
, *stab_sym_end
, *sym
;
235 char *stab_str
= NULL
;
238 stab_len
= stab_section
->data_offset
;
239 stab_sym
= (Stab_Sym
*)stab_section
->data
;
240 stab_str
= stabstr_section
->data
;
246 last_func_name
[0] = '\0';
247 last_pc
= (uplong
)-1;
253 stab_sym_end
= (Stab_Sym
*)((char*)stab_sym
+ stab_len
);
254 for (sym
= stab_sym
+ 1; sym
< stab_sym_end
; ++sym
) {
255 switch(sym
->n_type
) {
256 /* function start or end */
258 if (sym
->n_strx
== 0) {
259 /* we test if between last line and end of function */
260 pc
= sym
->n_value
+ func_addr
;
261 if (wanted_pc
>= last_pc
&& wanted_pc
< pc
)
266 str
= stab_str
+ sym
->n_strx
;
267 p
= strchr(str
, ':');
269 pstrcpy(func_name
, sizeof(func_name
), str
);
272 if (len
> sizeof(func_name
) - 1)
273 len
= sizeof(func_name
) - 1;
274 memcpy(func_name
, str
, len
);
275 func_name
[len
] = '\0';
277 func_addr
= sym
->n_value
;
280 /* line number info */
282 pc
= sym
->n_value
+ func_addr
;
283 if (wanted_pc
>= last_pc
&& wanted_pc
< pc
)
286 last_line_num
= sym
->n_desc
;
288 strcpy(last_func_name
, func_name
);
292 str
= stab_str
+ sym
->n_strx
;
294 if (incl_index
< INCLUDE_STACK_SIZE
) {
295 incl_files
[incl_index
++] = str
;
303 if (sym
->n_strx
== 0) {
304 incl_index
= 0; /* end of translation unit */
306 str
= stab_str
+ sym
->n_strx
;
307 /* do not add path */
309 if (len
> 0 && str
[len
- 1] != '/')
317 /* second pass: we try symtab symbols (no line number info) */
321 ElfW(Sym
) *sym
, *sym_end
;
324 sym_end
= (ElfW(Sym
) *)(symtab_section
->data
+ symtab_section
->data_offset
);
325 for(sym
= (ElfW(Sym
) *)symtab_section
->data
+ 1;
328 type
= ELFW(ST_TYPE
)(sym
->st_info
);
329 if (type
== STT_FUNC
|| type
== STT_GNU_IFUNC
) {
330 if (wanted_pc
>= sym
->st_value
&&
331 wanted_pc
< sym
->st_value
+ sym
->st_size
) {
332 pstrcpy(last_func_name
, sizeof(last_func_name
),
333 strtab_section
->data
+ sym
->st_name
);
334 func_addr
= sym
->st_value
;
340 /* did not find any info: */
341 fprintf(stderr
, "%s %p ???\n", msg
, (void*)wanted_pc
);
347 fprintf(stderr
, "%s:%d: ", incl_files
[--i
], last_line_num
);
348 fprintf(stderr
, "%s %p", msg
, (void*)wanted_pc
);
349 if (last_func_name
[0] != '\0')
350 fprintf(stderr
, " %s()", last_func_name
);
352 fprintf(stderr
, " (included from ");
354 fprintf(stderr
, "%s", incl_files
[i
]);
357 fprintf(stderr
, ", ");
359 fprintf(stderr
, ")");
361 fprintf(stderr
, "\n");
366 /* emit a run time error at position 'pc' */
367 static void rt_error(ucontext_t
*uc
, const char *fmt
, ...)
374 fprintf(stderr
, "Runtime error: ");
375 vfprintf(stderr
, fmt
, ap
);
376 fprintf(stderr
, "\n");
378 for(i
=0;i
<num_callers
;i
++) {
379 if (rt_get_caller_pc(&pc
, uc
, i
) < 0)
381 pc
= rt_printline(pc
, i
? "by" : "at");
382 if (pc
== (uplong
)rt_prog_main
&& pc
)
389 /* ------------------------------------------------------------- */
392 /* signal handler for fatal errors */
393 static void sig_error(int signum
, siginfo_t
*siginf
, void *puc
)
395 ucontext_t
*uc
= puc
;
399 switch(siginf
->si_code
) {
402 rt_error(uc
, "division by zero");
405 rt_error(uc
, "floating point exception");
411 if (rt_bound_error_msg
&& *rt_bound_error_msg
)
412 rt_error(uc
, *rt_bound_error_msg
);
414 rt_error(uc
, "dereferencing invalid pointer");
417 rt_error(uc
, "illegal instruction");
420 rt_error(uc
, "abort() called");
423 rt_error(uc
, "caught signal %d", signum
);
429 /* Generate a stack backtrace when a CPU exception occurs. */
430 static void set_exception_handler(void)
432 struct sigaction sigact
;
433 /* install TCC signal handlers to print debug info on fatal
435 sigact
.sa_flags
= SA_SIGINFO
| SA_RESETHAND
;
436 sigact
.sa_sigaction
= sig_error
;
437 sigemptyset(&sigact
.sa_mask
);
438 sigaction(SIGFPE
, &sigact
, NULL
);
439 sigaction(SIGILL
, &sigact
, NULL
);
440 sigaction(SIGSEGV
, &sigact
, NULL
);
441 sigaction(SIGBUS
, &sigact
, NULL
);
442 sigaction(SIGABRT
, &sigact
, NULL
);
445 /* ------------------------------------------------------------- */
448 /* fix for glibc 2.1 */
454 /* return the PC at frame level 'level'. Return non zero if not found */
455 static int rt_get_caller_pc(unsigned long *paddr
, ucontext_t
*uc
, int level
)
461 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
462 *paddr
= uc
->uc_mcontext
.mc_eip
;
463 #elif defined(__dietlibc__)
464 *paddr
= uc
->uc_mcontext
.eip
;
466 *paddr
= uc
->uc_mcontext
.gregs
[REG_EIP
];
470 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
471 fp
= uc
->uc_mcontext
.mc_ebp
;
472 #elif defined(__dietlibc__)
473 fp
= uc
->uc_mcontext
.ebp
;
475 fp
= uc
->uc_mcontext
.gregs
[REG_EBP
];
477 for(i
=1;i
<level
;i
++) {
478 /* XXX: check address validity with program info */
479 if (fp
<= 0x1000 || fp
>= 0xc0000000)
481 fp
= ((unsigned long *)fp
)[0];
483 *paddr
= ((unsigned long *)fp
)[1];
488 /* ------------------------------------------------------------- */
489 #elif defined(__x86_64__)
491 /* return the PC at frame level 'level'. Return non zero if not found */
492 static int rt_get_caller_pc(unsigned long *paddr
,
493 ucontext_t
*uc
, int level
)
499 /* XXX: only support linux */
500 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
501 *paddr
= uc
->uc_mcontext
.mc_rip
;
503 *paddr
= uc
->uc_mcontext
.gregs
[REG_RIP
];
507 #if defined(__FreeBSD__) || defined(__FreeBSD_kernel__)
508 fp
= uc
->uc_mcontext
.mc_rbp
;
510 fp
= uc
->uc_mcontext
.gregs
[REG_RBP
];
512 for(i
=1;i
<level
;i
++) {
513 /* XXX: check address validity with program info */
516 fp
= ((unsigned long *)fp
)[0];
518 *paddr
= ((unsigned long *)fp
)[1];
523 /* ------------------------------------------------------------- */
524 #elif defined(__arm__)
526 /* return the PC at frame level 'level'. Return negative if not found */
527 static int rt_get_caller_pc(unsigned long *paddr
,
528 ucontext_t
*uc
, int level
)
534 /* XXX: only supports linux */
535 #if defined(__linux__)
536 *paddr
= uc
->uc_mcontext
.arm_pc
;
542 #if defined(__linux__)
543 fp
= uc
->uc_mcontext
.arm_fp
;
544 sp
= uc
->uc_mcontext
.arm_sp
;
550 /* XXX: specific to tinycc stack frames */
551 if (fp
< sp
+ 12 || fp
& 3)
553 for(i
= 1; i
< level
; i
++) {
554 sp
= ((uint32_t *)fp
)[-2];
555 if (sp
< fp
|| sp
- fp
> 16 || sp
& 3)
557 fp
= ((uint32_t *)fp
)[-3];
558 if (fp
<= sp
|| fp
- sp
< 12 || fp
& 3)
561 /* XXX: check address validity with program info */
562 *paddr
= ((uint32_t *)fp
)[-1];
567 /* ------------------------------------------------------------- */
570 #warning add arch specific rt_get_caller_pc()
571 static int rt_get_caller_pc(unsigned long *paddr
,
572 ucontext_t
*uc
, int level
)
577 #endif /* !__i386__ */
579 /* ------------------------------------------------------------- */
582 static long __stdcall
cpu_exception_handler(EXCEPTION_POINTERS
*ex_info
)
584 CONTEXT
*uc
= ex_info
->ContextRecord
;
586 EXCEPTION_RECORD *er = ex_info->ExceptionRecord;
587 printf("CPU exception: code=%08lx addr=%p\n",
588 er->ExceptionCode, er->ExceptionAddress);
590 if (rt_bound_error_msg
&& *rt_bound_error_msg
)
591 rt_error(uc
, *rt_bound_error_msg
);
593 rt_error(uc
, "dereferencing invalid pointer");
595 //return EXCEPTION_CONTINUE_SEARCH;
598 /* Generate a stack backtrace when a CPU exception occurs. */
599 static void set_exception_handler(void)
601 SetUnhandledExceptionFilter(cpu_exception_handler
);
605 static void win64_add_function_table(TCCState
*s1
)
608 (RUNTIME_FUNCTION
*)(uplong
)s1
->uw_pdata
->sh_addr
,
609 s1
->uw_pdata
->data_offset
/ sizeof (RUNTIME_FUNCTION
),
610 (uplong
)text_section
->sh_addr
620 /* return the PC at frame level 'level'. Return non zero if not found */
621 static int rt_get_caller_pc(uplong
*paddr
, CONTEXT
*uc
, int level
)
631 for(i
=1;i
<level
;i
++) {
632 /* XXX: check address validity with program info */
633 if (fp
<= 0x1000 || fp
>= 0xc0000000)
635 fp
= ((uplong
*)fp
)[0];
637 *paddr
= ((uplong
*)fp
)[1];
646 #endif /* CONFIG_TCC_BACKTRACE */
647 /* ------------------------------------------------------------- */
649 #ifdef CONFIG_TCC_STATIC
651 #define RTLD_LAZY 0x001
652 #define RTLD_NOW 0x002
653 #define RTLD_GLOBAL 0x100
654 #define RTLD_DEFAULT NULL
656 /* dummy function for profiling */
657 void *dlopen(const char *filename
, int flag
)
662 void dlclose(void *p
)
666 const char *dlerror(void)
671 typedef struct TCCSyms
{
676 #define TCCSYM(a) { #a, &a, },
678 /* add the symbol you want here if no dynamic linking is done */
679 static TCCSyms tcc_syms
[] = {
680 #if !defined(CONFIG_TCCBOOT)
689 void *resolve_sym(TCCState
*s1
, const char *symbol
)
693 while (p
->str
!= NULL
) {
694 if (!strcmp(p
->str
, symbol
))
701 #elif !defined(_WIN32)
703 void *resolve_sym(TCCState
*s1
, const char *sym
)
705 return dlsym(RTLD_DEFAULT
, sym
);
708 #endif /* CONFIG_TCC_STATIC */
710 /* ------------------------------------------------------------- */