From 3e0e901a18d8170d0df91b0d067e965b7dc7a77c Mon Sep 17 00:00:00 2001 From: Kostya Serebryany Date: Tue, 11 Oct 2016 01:13:32 +0000 Subject: [PATCH] [libFuzzer] add switch tests llvm-svn: 283840 --- llvm/lib/Fuzzer/test/Switch2Test.cpp | 35 ++++++++++++++++++++++++++ llvm/lib/Fuzzer/test/value-profile-switch.test | 3 +++ 2 files changed, 38 insertions(+) create mode 100644 llvm/lib/Fuzzer/test/Switch2Test.cpp create mode 100644 llvm/lib/Fuzzer/test/value-profile-switch.test diff --git a/llvm/lib/Fuzzer/test/Switch2Test.cpp b/llvm/lib/Fuzzer/test/Switch2Test.cpp new file mode 100644 index 0000000..3c6a300 --- /dev/null +++ b/llvm/lib/Fuzzer/test/Switch2Test.cpp @@ -0,0 +1,35 @@ +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. + +// Simple test for a fuzzer. The fuzzer must find the interesting switch value. +#include +#include +#include +#include +#include + +int Switch(int a) { + switch(a) { + case 100001: return 1; + case 100002: return 2; + case 100003: return 4; + } + return 0; +} + +extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) { + const int N = 3; + if (Size < N * sizeof(int)) return 0; + int Res = 0; + for (int i = 0; i < N; i++) { + int X; + memcpy(&X, Data + i * sizeof(int), sizeof(int)); + Res += Switch(X); + } + if (Res == 5 || Res == 3 || Res == 6 || Res == 7) { + fprintf(stderr, "BINGO; Found the target, exiting; Res=%d\n", Res); + exit(1); + } + return 0; +} + diff --git a/llvm/lib/Fuzzer/test/value-profile-switch.test b/llvm/lib/Fuzzer/test/value-profile-switch.test new file mode 100644 index 0000000..7e66e51 --- /dev/null +++ b/llvm/lib/Fuzzer/test/value-profile-switch.test @@ -0,0 +1,3 @@ +CHECK: BINGO +RUN: not LLVMFuzzer-SwitchTest-TracePC -use_value_profile=1 -runs=100000000 -seed=1 2>&1 | FileCheck %s +RUN: not LLVMFuzzer-Switch2Test-TracePC -use_value_profile=1 -runs=100000000 -seed=1 2>&1 | FileCheck %s -- 2.7.4