llvm/unittests/MC/Disassembler.cpp

   1 //===- llvm/unittest/Object/Disassembler.cpp ------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #include "llvm-c/Disassembler.h"
  10 #include "llvm/Support/TargetSelect.h"
  11 #include "gtest/gtest.h"
  12
  13 using namespace llvm;
  14
  15 static const char *symbolLookupCallback(void *DisInfo, uint64_t ReferenceValue,
  16                                         uint64_t *ReferenceType,
  17                                         uint64_t ReferencePC,
  18                                         const char **ReferenceName) {
  19   *ReferenceType = LLVMDisassembler_ReferenceType_InOut_None;
  20   return nullptr;
  21 }
  22
  23 TEST(Disassembler, X86Test) {
  24   llvm::InitializeAllTargetInfos();
  25   llvm::InitializeAllTargetMCs();
  26   llvm::InitializeAllDisassemblers();
  27
  28   uint8_t Bytes[] = {0x90, 0x90, 0xeb, 0xfd};
  29   uint8_t *BytesP = Bytes;
  30   const char OutStringSize = 100;
  31   char OutString[OutStringSize];
  32   LLVMDisasmContextRef DCR = LLVMCreateDisasm("x86_64-pc-linux", nullptr, 0,
  33                                               nullptr, symbolLookupCallback);
  34   if (!DCR)
  35     GTEST_SKIP();
  36
  37   size_t InstSize;
  38   unsigned NumBytes = sizeof(Bytes);
  39   unsigned PC = 0;
  40
  41   InstSize =
  42       LLVMDisasmInstruction(DCR, BytesP, 0, PC, OutString, OutStringSize);
  43   EXPECT_EQ(InstSize, 0U);
  44
  45   InstSize = LLVMDisasmInstruction(DCR, BytesP, NumBytes, PC, OutString,
  46                                    OutStringSize);
  47   EXPECT_EQ(InstSize, 1U);
  48   EXPECT_EQ(StringRef(OutString), "\tnop");
  49   PC += InstSize;
  50   BytesP += InstSize;
  51   NumBytes -= InstSize;
  52
  53   InstSize = LLVMDisasmInstruction(DCR, BytesP, NumBytes, PC, OutString,
  54                                    OutStringSize);
  55   EXPECT_EQ(InstSize, 1U);
  56   EXPECT_EQ(StringRef(OutString), "\tnop");
  57   PC += InstSize;
  58   BytesP += InstSize;
  59   NumBytes -= InstSize;
  60
  61   InstSize = LLVMDisasmInstruction(DCR, BytesP, NumBytes, PC, OutString,
  62                                    OutStringSize);
  63   EXPECT_EQ(InstSize, 2U);
  64   EXPECT_EQ(StringRef(OutString), "\tjmp\t0x1");
  65
  66   LLVMDisasmDispose(DCR);
  67 }
  68
  69 TEST(Disassembler, WebAssemblyTest) {
  70   llvm::InitializeAllTargetInfos();
  71   llvm::InitializeAllTargetMCs();
  72   llvm::InitializeAllDisassemblers();
  73
  74   uint8_t Bytes[] = {0x6a, 0x42, 0x7F, 0x35, 0x01, 0x10};
  75   uint8_t *BytesP = Bytes;
  76   const char OutStringSize = 100;
  77   char OutString[OutStringSize];
  78   LLVMDisasmContextRef DCR = LLVMCreateDisasm("wasm32-unknown-unknown", nullptr,
  79                                               0, nullptr, symbolLookupCallback);
  80   if (!DCR)
  81     GTEST_SKIP();
  82
  83   size_t InstSize;
  84   unsigned NumBytes = sizeof(Bytes);
  85   unsigned PC = 0;
  86
  87   InstSize = LLVMDisasmInstruction(DCR, BytesP, NumBytes, PC, OutString,
  88                                    OutStringSize);
  89   EXPECT_EQ(InstSize, 1U);
  90   EXPECT_EQ(StringRef(OutString), "\ti32.add ");
  91   PC += InstSize;
  92   BytesP += InstSize;
  93   NumBytes -= InstSize;
  94
  95   InstSize = LLVMDisasmInstruction(DCR, BytesP, NumBytes, PC, OutString,
  96                                    OutStringSize);
  97   EXPECT_EQ(InstSize, 2U);
  98   EXPECT_EQ(StringRef(OutString), "\ti64.const\t-1");
  99
 100   PC += InstSize;
 101   BytesP += InstSize;
 102   NumBytes -= InstSize;
 103
 104   InstSize = LLVMDisasmInstruction(DCR, BytesP, NumBytes, PC, OutString,
 105                                    OutStringSize);
 106   EXPECT_EQ(InstSize, 3U);
 107   EXPECT_EQ(StringRef(OutString), "\ti64.load32_u\t16:p2align=1");
 108
 109   LLVMDisasmDispose(DCR);
 110 }