c5c0da8f41
This reverts commit e10db1594feefbeb0a9642d83b00f4a164f0104c. This adds the MCDisassembler which was missing, breaking shared library builds.
104 lines
3.5 KiB
C++
104 lines
3.5 KiB
C++
//===- SystemZMCDisassemblerTest.cpp - Tests for SystemZ MCDisassembler ---===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/MC/MCAsmInfo.h"
|
|
#include "llvm/MC/MCContext.h"
|
|
#include "llvm/MC/MCDisassembler/MCDisassembler.h"
|
|
#include "llvm/MC/MCDisassembler/MCSymbolizer.h"
|
|
#include "llvm/MC/MCInst.h"
|
|
#include "llvm/MC/MCRegisterInfo.h"
|
|
#include "llvm/MC/MCSubtargetInfo.h"
|
|
#include "llvm/MC/MCTargetOptions.h"
|
|
#include "llvm/MC/TargetRegistry.h"
|
|
#include "llvm/Support/TargetSelect.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
struct Context {
|
|
const char *TripleName = "systemz-unknown";
|
|
std::unique_ptr<MCRegisterInfo> MRI;
|
|
std::unique_ptr<MCAsmInfo> MAI;
|
|
std::unique_ptr<MCContext> Ctx;
|
|
std::unique_ptr<MCSubtargetInfo> STI;
|
|
std::unique_ptr<MCDisassembler> DisAsm;
|
|
|
|
Context() {
|
|
LLVMInitializeSystemZTargetInfo();
|
|
LLVMInitializeSystemZTargetMC();
|
|
LLVMInitializeSystemZDisassembler();
|
|
|
|
// If we didn't build SystemZ, do not run the test.
|
|
std::string Error;
|
|
const Target *TheTarget = TargetRegistry::lookupTarget(TripleName, Error);
|
|
if (!TheTarget)
|
|
return;
|
|
|
|
MRI.reset(TheTarget->createMCRegInfo(TripleName));
|
|
MAI.reset(TheTarget->createMCAsmInfo(*MRI, TripleName, MCTargetOptions()));
|
|
STI.reset(TheTarget->createMCSubtargetInfo(TripleName, "", ""));
|
|
Ctx = std::make_unique<MCContext>(Triple(TripleName), MAI.get(), MRI.get(),
|
|
STI.get());
|
|
|
|
DisAsm.reset(TheTarget->createMCDisassembler(*STI, *Ctx));
|
|
}
|
|
|
|
operator MCContext &() { return *Ctx; };
|
|
};
|
|
|
|
Context &getContext() {
|
|
static Context Ctxt;
|
|
return Ctxt;
|
|
}
|
|
|
|
class SystemZMCSymbolizerTest : public MCSymbolizer {
|
|
public:
|
|
SystemZMCSymbolizerTest(MCContext &MC) : MCSymbolizer(MC, nullptr) {}
|
|
~SystemZMCSymbolizerTest() {}
|
|
|
|
bool tryAddingSymbolicOperand([[maybe_unused]] MCInst &Inst,
|
|
[[maybe_unused]] raw_ostream &CStream,
|
|
[[maybe_unused]] int64_t Value,
|
|
[[maybe_unused]] uint64_t Address,
|
|
[[maybe_unused]] bool IsBranch,
|
|
[[maybe_unused]] uint64_t Offset,
|
|
[[maybe_unused]] uint64_t OpSize,
|
|
[[maybe_unused]] uint64_t InstSize) override {
|
|
return true;
|
|
}
|
|
|
|
void
|
|
tryAddingPcLoadReferenceComment([[maybe_unused]] raw_ostream &cStream,
|
|
[[maybe_unused]] int64_t Value,
|
|
[[maybe_unused]] uint64_t Address) override {}
|
|
};
|
|
|
|
} // namespace
|
|
|
|
TEST(SystemZDisassembler, SystemZMCSymbolizerTest) {
|
|
SystemZMCSymbolizerTest *TestSymbolizer =
|
|
new SystemZMCSymbolizerTest(getContext());
|
|
getContext().DisAsm->setSymbolizer(
|
|
std::unique_ptr<MCSymbolizer>(TestSymbolizer));
|
|
|
|
MCInst Inst;
|
|
uint64_t InstSize;
|
|
|
|
// Check that the SystemZ disassembler sets the comment stream before calling
|
|
// MCDisassembler::tryAddingSymbolicOperand. This will fail an assert if it
|
|
// does not do that.
|
|
MCDisassembler::DecodeStatus Status = getContext().DisAsm->getInstruction(
|
|
Inst, InstSize,
|
|
// lgrl %r1, 0x1234
|
|
{0xc4, 0x18, 0x00, 0x00, 0x9a, 0x1a}, 0, nulls());
|
|
ASSERT_TRUE(Status == MCDisassembler::Success);
|
|
EXPECT_EQ(InstSize, uint64_t{6});
|
|
}
|