mirror of
https://github.com/frida/tinycc
synced 2024-12-24 05:56:49 +03:00
make --with-selinux work with libtcc, too
This commit is contained in:
parent
a4ed587f61
commit
2ab42855cb
5
libtcc.c
5
libtcc.c
@ -1038,7 +1038,12 @@ LIBTCCAPI void tcc_delete(TCCState *s1)
|
||||
dynarray_reset(&s1->sysinclude_paths, &s1->nb_sysinclude_paths);
|
||||
|
||||
tcc_free(s1->tcc_lib_path);
|
||||
#ifdef HAVE_SELINUX
|
||||
munmap (s1->write_mem, s1->mem_size);
|
||||
munmap (s1->runtime_mem, s1->mem_size);
|
||||
#else
|
||||
tcc_free(s1->runtime_mem);
|
||||
#endif
|
||||
tcc_free(s1);
|
||||
}
|
||||
|
||||
|
4
tcc.h
4
tcc.h
@ -521,6 +521,10 @@ struct TCCState {
|
||||
/* for tcc_relocate */
|
||||
int runtime_added;
|
||||
void *runtime_mem;
|
||||
#ifdef HAVE_SELINUX
|
||||
void *write_mem;
|
||||
unsigned long mem_size;
|
||||
#endif
|
||||
|
||||
struct InlineFunc **inline_fns;
|
||||
int nb_inline_fns;
|
||||
|
60
tccrun.c
60
tccrun.c
@ -37,12 +37,33 @@ static int tcc_relocate_ex(TCCState *s1, void *ptr);
|
||||
int tcc_relocate(TCCState *s1)
|
||||
{
|
||||
int ret;
|
||||
|
||||
#ifdef HAVE_SELINUX
|
||||
char tmpfname[] = "/tmp/.tccrunXXXXXX";
|
||||
int fd = mkstemp (tmpfname);
|
||||
unlink (tmpfname); ftruncate (fd, 1000);
|
||||
if ((ret= tcc_relocate_ex(s1,NULL)) < 0)return -1;
|
||||
s1->mem_size=ret;
|
||||
/* Use mmap instead of malloc for Selinux */
|
||||
s1->write_mem = mmap (NULL, ret, PROT_READ|PROT_WRITE,
|
||||
MAP_SHARED, fd, 0);
|
||||
if(s1->write_mem == MAP_FAILED){
|
||||
error("/tmp not writeable");
|
||||
return -1;
|
||||
}
|
||||
s1->runtime_mem = mmap (NULL, ret, PROT_READ|PROT_EXEC,
|
||||
MAP_SHARED, fd, 0);
|
||||
if(s1->runtime_mem == MAP_FAILED){
|
||||
error("/tmp not executable");
|
||||
return -1;
|
||||
}
|
||||
ret = tcc_relocate_ex(s1, s1->write_mem);
|
||||
#else
|
||||
ret = tcc_relocate_ex(s1, NULL);
|
||||
if (-1 != ret) {
|
||||
s1->runtime_mem = tcc_malloc(ret);
|
||||
ret = tcc_relocate_ex(s1, s1->runtime_mem);
|
||||
}
|
||||
#endif
|
||||
return ret;
|
||||
}
|
||||
|
||||
@ -50,34 +71,10 @@ int tcc_relocate(TCCState *s1)
|
||||
int tcc_run(TCCState *s1, int argc, char **argv)
|
||||
{
|
||||
int (*prog_main)(int, char **);
|
||||
int ret;
|
||||
#ifdef HAVE_SELINUX
|
||||
int rret;
|
||||
void *ptr,*writep;
|
||||
char tmpfname[] = "/tmp/.tccrunXXXXXX";
|
||||
int fd = mkstemp (tmpfname);
|
||||
unlink (tmpfname);
|
||||
ftruncate (fd, 1000);
|
||||
if ((rret= tcc_relocate_ex(s1,NULL)) < 0)
|
||||
return -1;
|
||||
/* Use mmap instead of malloc for Selinux */
|
||||
writep = mmap (NULL, rret, PROT_READ|PROT_WRITE,
|
||||
MAP_SHARED, fd, 0);
|
||||
if(writep == MAP_FAILED){
|
||||
error("/tmp not writeable");
|
||||
return -1;
|
||||
}
|
||||
ptr = mmap (NULL, rret, PROT_READ|PROT_EXEC,
|
||||
MAP_SHARED, fd, 0);
|
||||
if(ptr == MAP_FAILED){
|
||||
error("/tmp not executable");
|
||||
return -1;
|
||||
}
|
||||
tcc_relocate_ex(s1, writep);
|
||||
#else
|
||||
|
||||
if (tcc_relocate(s1) < 0)
|
||||
return -1;
|
||||
#endif
|
||||
|
||||
prog_main = tcc_get_symbol_err(s1, "main");
|
||||
|
||||
#ifdef CONFIG_TCC_BACKTRACE
|
||||
@ -89,6 +86,7 @@ int tcc_run(TCCState *s1, int argc, char **argv)
|
||||
if (s1->do_bounds_check) {
|
||||
void (*bound_init)(void);
|
||||
void (*bound_exit)(void);
|
||||
int ret;
|
||||
/* set error function */
|
||||
rt_bound_error_msg = tcc_get_symbol_err(s1, "__bound_error_msg");
|
||||
rt_prog_main = prog_main;
|
||||
@ -108,13 +106,7 @@ int tcc_run(TCCState *s1, int argc, char **argv)
|
||||
if (p) *p = 0;
|
||||
}
|
||||
#endif
|
||||
ret=(*prog_main)(argc, argv);
|
||||
#ifdef HAVE_SELINUX
|
||||
munmap (writep, rret);
|
||||
munmap (ptr, rret);
|
||||
|
||||
#endif
|
||||
return ret;
|
||||
return (*prog_main)(argc, argv);
|
||||
}
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user