break up llvm rt afl init

This commit is contained in:
van Hauser
2020-08-11 03:37:02 +02:00
parent fee58a4d1b
commit 9cf8637fab
2 changed files with 18 additions and 7 deletions

View File

@ -35,6 +35,8 @@
#include <string.h>
#include <assert.h>
#include <stdint.h>
#include <stddef.h>
#include <limits.h>
#include <errno.h>
#include <sys/mman.h>
@ -848,7 +850,6 @@ void __afl_manual_init(void) {
if (!init_done) {
__afl_map_shm();
__afl_start_forkserver();
init_done = 1;
@ -856,17 +857,27 @@ void __afl_manual_init(void) {
}
/* Proper initialization routine. */
/* Initialization of the forkserver - latest possible */
__attribute__((constructor(CONST_PRIO))) void __afl_auto_init(void) {
__attribute__((constructor())) void __afl_auto_init(void) {
if (getenv("AFL_DISABLE_LLVM_INSTRUMENTATION")) return;
if (getenv(DEFER_ENV_VAR)) return;
__afl_manual_init();
}
/* Initialization of the shmem - earliest possible because of LTO fixed mem. */
__attribute__((constructor(0))) void __afl_auto_early(void) {
if (getenv("AFL_DISABLE_LLVM_INSTRUMENTATION")) return;
is_persistent = !!getenv(PERSIST_ENV_VAR);
if (getenv(DEFER_ENV_VAR)) return;
__afl_manual_init();
__afl_map_shm();
}