1 /*
2 * Copyright 2011, The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "Disassembler.h"
18
19 #include "Config.h"
20
21 #include "DebugHelper.h"
22 #include "ExecutionEngine/Compiler.h"
23
24 #include "llvm/MC/MCAsmInfo.h"
25 #include "llvm/MC/MCDisassembler.h"
26 #include "llvm/MC/MCInst.h"
27 #include "llvm/MC/MCInstPrinter.h"
28
29 #include "llvm/Support/MemoryObject.h"
30 #include "llvm/Support/raw_ostream.h"
31
32 #include "llvm/Target/TargetData.h"
33 #include "llvm/Target/TargetMachine.h"
34 #include "llvm/Target/TargetOptions.h"
35 #include "llvm/Support/TargetRegistry.h"
36 #include "llvm/Support/TargetSelect.h"
37
38 #include "llvm/LLVMContext.h"
39
40 #if USE_DISASSEMBLER
41
42 namespace {
43
44 class BufferMemoryObject : public llvm::MemoryObject {
45 private:
46 const uint8_t *mBytes;
47 uint64_t mLength;
48
49 public:
BufferMemoryObject(const uint8_t * Bytes,uint64_t Length)50 BufferMemoryObject(const uint8_t *Bytes, uint64_t Length)
51 : mBytes(Bytes), mLength(Length) {
52 }
53
getBase() const54 virtual uint64_t getBase() const { return 0; }
getExtent() const55 virtual uint64_t getExtent() const { return mLength; }
56
readByte(uint64_t Addr,uint8_t * Byte) const57 virtual int readByte(uint64_t Addr, uint8_t *Byte) const {
58 if (Addr > getExtent())
59 return -1;
60 *Byte = mBytes[Addr];
61 return 0;
62 }
63 };
64
65 } // namespace anonymous
66
67 namespace bcc {
68
InitializeDisassembler()69 void InitializeDisassembler() {
70 #if defined(PROVIDE_ARM_CODEGEN)
71 LLVMInitializeARMDisassembler();
72 #endif
73
74 #if defined(PROVIDE_X86_CODEGEN)
75 LLVMInitializeX86Disassembler();
76 #endif
77 }
78
Disassemble(char const * OutputFileName,llvm::Target const * Target,llvm::TargetMachine * TM,std::string const & Name,unsigned char const * Func,size_t FuncSize)79 void Disassemble(char const *OutputFileName,
80 llvm::Target const *Target,
81 llvm::TargetMachine *TM,
82 std::string const &Name,
83 unsigned char const *Func,
84 size_t FuncSize) {
85
86 std::string ErrorInfo;
87
88 // Open the disassembler output file
89 llvm::raw_fd_ostream OS(OutputFileName, ErrorInfo,
90 llvm::raw_fd_ostream::F_Append);
91
92 if (!ErrorInfo.empty()) {
93 ALOGE("Unable to open disassembler output file: %s\n", OutputFileName);
94 return;
95 }
96
97 // Disassemble the given function
98 OS << "Disassembled code: " << Name << "\n";
99
100 const llvm::MCAsmInfo *AsmInfo;
101 const llvm::MCSubtargetInfo *SubtargetInfo;
102 const llvm::MCDisassembler *Disassmbler;
103 llvm::MCInstPrinter *IP;
104
105 AsmInfo = Target->createMCAsmInfo(Compiler::getTargetTriple());
106 SubtargetInfo = Target->createMCSubtargetInfo(Compiler::getTargetTriple(), "", "");
107 Disassmbler = Target->createMCDisassembler(*SubtargetInfo);
108 IP = Target->createMCInstPrinter(AsmInfo->getAssemblerDialect(),
109 *AsmInfo, *SubtargetInfo);
110
111 const BufferMemoryObject *BufferMObj = new BufferMemoryObject(Func, FuncSize);
112
113 uint64_t Size;
114 uint64_t Index;
115
116 for (Index = 0; Index < FuncSize; Index += Size) {
117 llvm::MCInst Inst;
118
119 if (Disassmbler->getInstruction(Inst, Size, *BufferMObj, Index,
120 /* REMOVED */ llvm::nulls(), llvm::nulls())) {
121 OS.indent(4);
122 OS.write("0x", 2);
123 OS.write_hex((uint32_t)Func + Index);
124 OS.write(": 0x", 4);
125 OS.write_hex(*(uint32_t *)(Func + Index));
126 IP->printInst(&Inst, OS, "");
127 OS << "\n";
128 } else {
129 if (Size == 0)
130 Size = 1; // skip illegible bytes
131 }
132 }
133
134 OS << "\n";
135
136 delete BufferMObj;
137
138 delete AsmInfo;
139 delete Disassmbler;
140 delete IP;
141
142 OS.close();
143 }
144
145 } // namespace bcc
146
147 #endif // USE_DISASSEMBLER
148