From 4079491d5e003c424f1bfa2a3f682185cd0c54a7 Mon Sep 17 00:00:00 2001 From: ptitSeb Date: Mon, 1 Mar 2021 16:05:58 +0100 Subject: Main elf memory allocated --- src/main.c | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'src') diff --git a/src/main.c b/src/main.c index f5bbf9d6..c049a27d 100755 --- a/src/main.c +++ b/src/main.c @@ -732,5 +732,25 @@ int main(int argc, const char **argv, const char **env) { return execvp(argv[1], (char * const*)(argv+1)); } AddElfHeader(my_context, elf_header); + + if(CalcLoadAddr(elf_header)) { + printf_log(LOG_NONE, "Error: reading elf header of %s\n", my_context->argv[0]); + fclose(f); + free_contextargv(); + FreeBox64Context(&my_context); + FreeCollection(&ld_preload); + return -1; + } + // allocate memory + if(AllocElfMemory(my_context, elf_header, 1)) { + printf_log(LOG_NONE, "Error: allocating memory for elf %s\n", my_context->argv[0]); + fclose(f); + free_contextargv(); + FreeBox64Context(&my_context); + FreeCollection(&ld_preload); + return -1; + } + + return 0; } -- cgit 1.4.1