diff --git a/lib/Fuzzer/FuzzerShmemPosix.cpp b/lib/Fuzzer/FuzzerShmemPosix.cpp index 820d64d07c1b..d5f4aabc4269 100644 --- a/lib/Fuzzer/FuzzerShmemPosix.cpp +++ b/lib/Fuzzer/FuzzerShmemPosix.cpp @@ -86,9 +86,13 @@ void SharedMemoryRegion::Post(int Idx) { void SharedMemoryRegion::Wait(int Idx) { assert(Idx == 0 || Idx == 1); - if (sem_wait((sem_t*)Semaphore[Idx])) { - Printf("ERROR: sem_wait failed\n"); - exit(1); + for (int i = 0; i < 10 && sem_wait((sem_t*)Semaphore[Idx]); i++) { + // sem_wait may fail if interrupted by a signal. + sleep(i); + if (i) + Printf("%s: sem_wait[%d] failed %s\n", i < 9 ? "WARNING" : "ERROR", i, + strerror(errno)); + if (i == 9) abort(); } } diff --git a/lib/Fuzzer/test/equivalence.test b/lib/Fuzzer/test/equivalence.test index 8c8b9ba15541..6c9d87888e07 100644 --- a/lib/Fuzzer/test/equivalence.test +++ b/lib/Fuzzer/test/equivalence.test @@ -4,3 +4,13 @@ RUN: not LLVMFuzzer-EquivalenceBTest -use_equivalence_server=EQUIV_TEST 2>&1 | F CHECK: ERROR: libFuzzer: equivalence-mismatch. Sizes: {{.*}}; offset 2 CHECK: SUMMARY: libFuzzer: equivalence-mismatch RUN: kill -9 $APID + + +# Run EquivalenceATest against itself with a small timeout +# to stress the signal handling and ensure that shmem doesn't mind +# the signals. + +RUN: LLVMFuzzer-EquivalenceATest -timeout=1 -run_equivalence_server=EQUIV_TEST & export APID=$! +RUN: sleep 3 +RUN: LLVMFuzzer-EquivalenceATest -timeout=1 -use_equivalence_server=EQUIV_TEST -runs=500000 2>&1 +RUN: kill -9 $APID