about summary refs log tree commit diff
path: root/instrumentation/afl-compiler-rt.o.c
diff options
context:
space:
mode:
authorvan Hauser <vh@thc.org>2023-01-27 13:40:48 +0100
committerGitHub <noreply@github.com>2023-01-27 13:40:48 +0100
commit4a54555a1a73dd6be4f494ef67155ed41a81f0f4 (patch)
treedc29dd420c9d1d0f071447c91c0a0ab85780ad05 /instrumentation/afl-compiler-rt.o.c
parent1b4e1d75b32c6024765ab27b36591ae97cb33f6b (diff)
parent33eba1fc5652060e8d877b02135fce2325813d0c (diff)
downloadafl++-4a54555a1a73dd6be4f494ef67155ed41a81f0f4.tar.gz
Merge pull request #1628 from AFLplusplus/dev
Dev
Diffstat (limited to 'instrumentation/afl-compiler-rt.o.c')
-rw-r--r--instrumentation/afl-compiler-rt.o.c8
1 files changed, 8 insertions, 0 deletions
diff --git a/instrumentation/afl-compiler-rt.o.c b/instrumentation/afl-compiler-rt.o.c
index 9c6345b6..d6d6c38c 100644
--- a/instrumentation/afl-compiler-rt.o.c
+++ b/instrumentation/afl-compiler-rt.o.c
@@ -149,6 +149,7 @@ u32 __afl_already_initialized_shm;
 u32 __afl_already_initialized_forkserver;
 u32 __afl_already_initialized_first;
 u32 __afl_already_initialized_second;
+u32 __afl_already_initialized_early;
 u32 __afl_already_initialized_init;
 
 /* Dummy pipe for area_is_valid() */
@@ -1373,6 +1374,9 @@ __attribute__((constructor(EARLY_FS_PRIO))) void __early_forkserver(void) {
 
 __attribute__((constructor(CTOR_PRIO))) void __afl_auto_early(void) {
 
+  if (__afl_already_initialized_early) return;
+  __afl_already_initialized_early = 1;
+
   is_persistent = !!getenv(PERSIST_ENV_VAR);
 
   if (getenv("AFL_DISABLE_LLVM_INSTRUMENTATION")) return;
@@ -1514,6 +1518,10 @@ void __sanitizer_cov_trace_pc_guard_init(uint32_t *start, uint32_t *stop) {
 
   _is_sancov = 1;
 
+  __afl_auto_first();
+  __afl_auto_second();
+  __afl_auto_early();
+
   if (__afl_debug) {
 
     fprintf(stderr,