Rework API for getting at instructions

[frida/frida.git] / src / disassembler / llvm / LLVMDisassembler.hxx

diff --git a/src/disassembler/llvm/LLVMDisassembler.hxx b/src/disassembler/llvm/LLVMDisassembler.hxx
index d00381b7ed962862c8be6e26008cdf94d06f9da8..a64dc2c66e451131f9187ab82ca5a70342a9e1ab 100644 (file)
--- a/src/disassembler/llvm/LLVMDisassembler.hxx
+++ b/src/disassembler/llvm/LLVMDisassembler.hxx
@@ -14,35 +14,31 @@ class BasicBlock;
 
 Disassembler * createLLVMDisassembler(const std::string& filename, InformationManager* manager);
 
-template <typename ELFT>
+template <typename ObjectType>
 class LLVMDisassembler
        : public Disassembler {
 public:
        LLVMDisassembler(const std::string& filename, InformationManager* manager,
-                        llvm::object::ELFObjectFile<ELFT>* file = NULL);
+                        llvm::object::ObjectFile* file = NULL);
        virtual ~LLVMDisassembler();
 
        void start();
        void getSymbols() {}
-       uint64_t entryAddress() {return _entryAddress;}
+       uint64_t entryAddress();
 
        void printEachInstruction(uint64_t start, uint64_t end,
                                  std::function<void (uint8_t*, size_t, const std::string&,
                                                      const std::string&)> fun);
 
        Function * disassembleFunctionAt(uint64_t address, const std::string& name = "");
-
-protected:
-       bool isFunctionCall(uint64_t address) {return false;}
-       bool isJump(uint64_t address) {return false;}
+       std::list<Instruction> getInstructions(const BasicBlock* block);
 
 private:
-       typedef llvm::object::ELFFile<ELFT> ELFO;
-
        // http://llvm.org/docs/doxygen/html/MCObjectDisassembler_8cpp_source.html +197
        void disassembleFunction(Function* function);
        void splitBlocks(Function* fun);
        void disassemble();
+       llvm::object::SectionRef getTextSection();
 
        void readSymbols();
        void readSections();
@@ -54,7 +50,7 @@ private:
        std::shared_ptr<llvm::object::Binary> binary;
 
        const llvm::Target * target;
-       llvm::object::ELFObjectFile<ELFT> * o;
+       llvm::object::ObjectFile * o;
 
        std::unique_ptr<const llvm::MCRegisterInfo> MRI;
        std::unique_ptr<const llvm::MCAsmInfo> AsmInfo;