forked from OSchip/llvm-project
128 lines
4.3 KiB
C++
128 lines
4.3 KiB
C++
//===- unittests/Interpreter/InterpreterExceptionTest.cpp -----------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Unit tests for Clang's Interpreter library.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/Interpreter/Interpreter.h"
|
|
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/AST/Decl.h"
|
|
#include "clang/AST/DeclGroup.h"
|
|
#include "clang/Basic/TargetInfo.h"
|
|
#include "clang/Basic/Version.h"
|
|
#include "clang/Config/config.h"
|
|
#include "clang/Frontend/CompilerInstance.h"
|
|
#include "clang/Frontend/TextDiagnosticPrinter.h"
|
|
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/ExecutionEngine/Orc/LLJIT.h"
|
|
#include "llvm/Support/ManagedStatic.h"
|
|
#include "llvm/Support/TargetSelect.h"
|
|
#include "llvm-c/Error.h"
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace clang;
|
|
|
|
namespace {
|
|
using Args = std::vector<const char *>;
|
|
static std::unique_ptr<Interpreter>
|
|
createInterpreter(const Args &ExtraArgs = {},
|
|
DiagnosticConsumer *Client = nullptr) {
|
|
Args ClangArgs = {"-Xclang", "-emit-llvm-only"};
|
|
ClangArgs.insert(ClangArgs.end(), ExtraArgs.begin(), ExtraArgs.end());
|
|
auto CI = cantFail(clang::IncrementalCompilerBuilder::create(ClangArgs));
|
|
if (Client)
|
|
CI->getDiagnostics().setClient(Client, /*ShouldOwnClient=*/false);
|
|
return cantFail(clang::Interpreter::create(std::move(CI)));
|
|
}
|
|
|
|
TEST(InterpreterTest, CatchException) {
|
|
llvm::InitializeNativeTarget();
|
|
llvm::InitializeNativeTargetAsmPrinter();
|
|
|
|
{
|
|
auto J = llvm::orc::LLJITBuilder().create();
|
|
if (!J) {
|
|
// The platform does not support JITs.
|
|
// Using llvm::consumeError will require typeinfo for ErrorInfoBase, we
|
|
// can avoid that by going via the C interface.
|
|
LLVMConsumeError(llvm::wrap(J.takeError()));
|
|
return;
|
|
}
|
|
}
|
|
|
|
#define Stringify(s) Stringifyx(s)
|
|
#define Stringifyx(s) #s
|
|
|
|
// We define a custom exception to avoid #include-ing the <exception> header
|
|
// which would require this test to know about the libstdc++ location.
|
|
// its own header file.
|
|
#define CUSTOM_EXCEPTION \
|
|
struct custom_exception { \
|
|
custom_exception(const char *Msg) : Message(Msg) {} \
|
|
const char *Message; \
|
|
};
|
|
|
|
CUSTOM_EXCEPTION;
|
|
|
|
std::string ExceptionCode = Stringify(CUSTOM_EXCEPTION);
|
|
ExceptionCode +=
|
|
R"(
|
|
extern "C" int printf(const char*, ...);
|
|
static void ThrowerAnError(const char* Name) {
|
|
throw custom_exception(Name);
|
|
}
|
|
|
|
extern "C" int throw_exception() {
|
|
try {
|
|
ThrowerAnError("To be caught in JIT");
|
|
} catch (const custom_exception& E) {
|
|
printf("Caught: '%s'\n", E.Message);
|
|
} catch (...) {
|
|
printf("Unknown exception\n");
|
|
}
|
|
ThrowerAnError("To be caught in binary");
|
|
return 0;
|
|
}
|
|
)";
|
|
std::unique_ptr<Interpreter> Interp = createInterpreter();
|
|
// FIXME: Re-enable the excluded target triples.
|
|
const clang::CompilerInstance *CI = Interp->getCompilerInstance();
|
|
const llvm::Triple &Triple = CI->getASTContext().getTargetInfo().getTriple();
|
|
// FIXME: PPC fails due to `Symbols not found: [DW.ref.__gxx_personality_v0]`
|
|
// The current understanding is that the JIT should emit this symbol if it was
|
|
// not (eg. the way passing clang -fPIC does it).
|
|
if (Triple.isPPC())
|
|
return;
|
|
|
|
// FIXME: ARM fails due to `Not implemented relocation type!`
|
|
if (Triple.isARM())
|
|
return;
|
|
|
|
// FIXME: libunwind on darwin is broken, see PR49692.
|
|
if (Triple.isOSDarwin() && (Triple.getArch() == llvm::Triple::aarch64 ||
|
|
Triple.getArch() == llvm::Triple::aarch64_32))
|
|
return;
|
|
|
|
llvm::cantFail(Interp->ParseAndExecute(ExceptionCode));
|
|
testing::internal::CaptureStdout();
|
|
auto ThrowException =
|
|
(int (*)())llvm::cantFail(Interp->getSymbolAddress("throw_exception"));
|
|
EXPECT_ANY_THROW(ThrowException());
|
|
std::string CapturedStdOut = testing::internal::GetCapturedStdout();
|
|
EXPECT_EQ(CapturedStdOut, "Caught: 'To be caught in JIT'\n");
|
|
|
|
llvm::llvm_shutdown();
|
|
}
|
|
|
|
} // end anonymous namespace
|