mirror of
https://github.com/RPCSX/llvm.git
synced 2024-12-27 14:45:24 +00:00
8be7707c14
At the same time, fixes InstructionsTest::CastInst unittest: yes you can leave the IR in an invalid state and exit when you don't destroy the context (like the global one), no longer now. This is the first part of http://reviews.llvm.org/D19094 From: Mehdi Amini <mehdi.amini@apple.com> git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@266379 91177308-0d34-0410-b5e6-96231b3b80d8
77 lines
2.4 KiB
C++
77 lines
2.4 KiB
C++
//===--- fuzz-llvm-as.cpp - Fuzzer for llvm-as using lib/Fuzzer -----------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Build tool to fuzz the LLVM assembler (llvm-as) using
|
|
// lib/Fuzzer. The main reason for using this tool is that it is much
|
|
// faster than using afl-fuzz, since it is run in-process.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/AsmParser/Parser.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Module.h"
|
|
#include "llvm/IR/Verifier.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "llvm/Support/SourceMgr.h"
|
|
|
|
#include <csetjmp>
|
|
|
|
using namespace llvm;
|
|
|
|
static jmp_buf JmpBuf;
|
|
|
|
namespace {
|
|
|
|
void MyFatalErrorHandler(void *user_data, const std::string& reason,
|
|
bool gen_crash_diag) {
|
|
// Don't bother printing reason, just return to the test function,
|
|
// since a fatal error represents a successful parse (i.e. it correctly
|
|
// terminated with an error message to the user).
|
|
longjmp(JmpBuf, 1);
|
|
}
|
|
|
|
static bool InstalledHandler = false;
|
|
|
|
} // end of anonymous namespace
|
|
|
|
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
|
|
|
|
// Allocate space for locals before setjmp so that memory can be collected
|
|
// if parse exits prematurely (via longjmp).
|
|
StringRef Input((const char *)Data, Size);
|
|
// Note: We need to create a buffer to add a null terminator to the
|
|
// end of the input string. The parser assumes that the string
|
|
// parsed is always null terminated.
|
|
std::unique_ptr<MemoryBuffer> MemBuf = MemoryBuffer::getMemBufferCopy(Input);
|
|
SMDiagnostic Err;
|
|
LLVMContext Context;
|
|
std::unique_ptr<Module> M;
|
|
|
|
if (setjmp(JmpBuf))
|
|
// If reached, we have returned with non-zero status, so exit.
|
|
return 0;
|
|
|
|
// TODO(kschimpf) Write a main to do this initialization.
|
|
if (!InstalledHandler) {
|
|
llvm::install_fatal_error_handler(::MyFatalErrorHandler, nullptr);
|
|
InstalledHandler = true;
|
|
}
|
|
|
|
M = parseAssembly(MemBuf->getMemBufferRef(), Err, Context);
|
|
|
|
if (!M.get())
|
|
return 0;
|
|
|
|
verifyModule(*M.get());
|
|
return 0;
|
|
}
|