9e689792b2
Summary: Implements https://github.com/google/sanitizers/issues/835. Flush stdout before exiting in test cases. Since the atexit hook is used for exit reports, pending prints to stdout can be lost if they aren't flushed before calling exit(). Expect tests to have non-zero exit code if exit() is called. Reviewers: vitalybuka, kcc Reviewed By: kcc Subscribers: eraman, llvm-commits, hiraditya Differential Revision: https://reviews.llvm.org/D35602 llvm-svn: 308669
40 lines
995 B
C++
40 lines
995 B
C++
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
|
|
// Simple test for a cutom mutator.
|
|
#include <assert.h>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <cstdlib>
|
|
#include <iostream>
|
|
#include <ostream>
|
|
|
|
#include "FuzzerInterface.h"
|
|
|
|
static volatile int Sink;
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
|
|
assert(Data);
|
|
if (Size > 0 && Data[0] == 'H') {
|
|
Sink = 1;
|
|
if (Size > 1 && Data[1] == 'i') {
|
|
Sink = 2;
|
|
if (Size > 2 && Data[2] == '!') {
|
|
std::cout << "BINGO; Found the target, exiting\n" << std::flush;
|
|
exit(1);
|
|
}
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
extern "C" size_t LLVMFuzzerCustomMutator(uint8_t *Data, size_t Size,
|
|
size_t MaxSize, unsigned int Seed) {
|
|
static bool Printed;
|
|
if (!Printed) {
|
|
std::cerr << "In LLVMFuzzerCustomMutator\n";
|
|
Printed = true;
|
|
}
|
|
return LLVMFuzzerMutate(Data, Size, MaxSize);
|
|
}
|