diff options
Diffstat (limited to 'llvm/lib/Fuzzer/test/DSOTestMain.cpp')
-rw-r--r-- | llvm/lib/Fuzzer/test/DSOTestMain.cpp | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/llvm/lib/Fuzzer/test/DSOTestMain.cpp b/llvm/lib/Fuzzer/test/DSOTestMain.cpp index 49cd185e97d..3e225d88612 100644 --- a/llvm/lib/Fuzzer/test/DSOTestMain.cpp +++ b/llvm/lib/Fuzzer/test/DSOTestMain.cpp @@ -11,17 +11,21 @@ extern int DSO1(int a); extern int DSO2(int a); extern int DSOTestExtra(int a); +static volatile int *nil = 0; extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) { - if (Size < sizeof(int) * 3) return 0; int x, y, z; - memcpy(&x, Data + 0 * sizeof(int), sizeof(int)); - memcpy(&y, Data + 1 * sizeof(int), sizeof(int)); - memcpy(&z, Data + 2 * sizeof(int), sizeof(int)); - int sum = DSO1(x) + DSO2(y) + DSOTestExtra(z); + if (Size < sizeof(int) * 3) { + x = y = z = 0; + } else { + memcpy(&x, Data + 0 * sizeof(int), sizeof(int)); + memcpy(&y, Data + 1 * sizeof(int), sizeof(int)); + memcpy(&z, Data + 2 * sizeof(int), sizeof(int)); + } + int sum = DSO1(x) + DSO2(y) + (z ? DSOTestExtra(z) : 0); if (sum == 3) { fprintf(stderr, "BINGO %d %d %d\n", x, y, z); - exit(1); + *nil = 0; } return 0; } |