X-Git-Url: https://git.siccegge.de//index.cgi?p=frida%2Ffrida.git;a=blobdiff_plain;f=src%2Fdisassembler%2Fllvm%2FLLVMDisassembler.cxx;h=1f1cdff1d17670d6b87b900523ef33b7f85bbc6a;hp=cdcd3cb2cd516883400d08a9b47bef5c94a0341d;hb=f9ae4391568cfcbf4e6de0475bddcdd68e9b31d2;hpb=4b2336f2066f76d2be017ad9731865ac6743abcb

diff --git a/src/disassembler/llvm/LLVMDisassembler.cxx b/src/disassembler/llvm/LLVMDisassembler.cxx
index cdcd3cb..1f1cdff 100644
--- a/src/disassembler/llvm/LLVMDisassembler.cxx
+++ b/src/disassembler/llvm/LLVMDisassembler.cxx
@@ -133,10 +133,10 @@ LLVMDisassembler<ELFT>::LLVMDisassembler(const std::string& filename,
 	RelInfo.reset(
 		target->createMCRelocationInfo(tripleName, Ctx));
 	if (RelInfo) {
-		Symzer.reset(
-			MCObjectSymbolizer::createObjectSymbolizer(Ctx, std::move(RelInfo), o));
-		if (Symzer)
-			DisAsm->setSymbolizer(std::move(Symzer));
+		// Symzer.reset(
+		// 	MCObjectSymbolizer::createObjectSymbolizer(Ctx, std::move(RelInfo), o));
+		// if (Symzer)
+		// 	DisAsm->setSymbolizer(std::move(Symzer));
 	}
 	RelInfo.release();
 	Symzer.release();
@@ -199,7 +199,6 @@ Function* LLVMDisassembler<ELFT>::disassembleFunctionAt(uint64_t address, const
 			function->setName(name);
 		}
 		disassembleFunction(function);
-		manager->finishFunction(function);
 	}
 
 	return function;
@@ -207,6 +206,7 @@ Function* LLVMDisassembler<ELFT>::disassembleFunctionAt(uint64_t address, const
 
 template <typename ELFT>
 void LLVMDisassembler<ELFT>::disassembleFunction(Function* function) {
+	std::vector<uint64_t> called_functions;
 	std::stack<BasicBlock*> remaining_blocks;
 	/* TODO:
 	 * Do all blocks get added properly? We should take care to remove
@@ -250,7 +250,7 @@ void LLVMDisassembler<ELFT>::disassembleFunction(Function* function) {
 					if (!MIA->isIndirectBranch(inst)) {
 						if (MIA->isCall(inst)) {
 							if (NULL == manager->getFunction(jmptarget))
-								disassembleFunctionAt(jmptarget);
+								called_functions.push_back(jmptarget);
 						} else {
 							current_block->setNextBlock(0, jmptarget);
 							if (new_blocks.find(jmptarget) == new_blocks.end()) {
@@ -298,7 +298,9 @@ void LLVMDisassembler<ELFT>::disassembleFunction(Function* function) {
 	}
 	splitBlocks(function);
 	LOG4CXX_DEBUG(logger, "Finished function " << function->getName());
-	manager->signal_new_function(function);
+	manager->finishFunction(function);
+	for (uint64_t address : called_functions)
+		disassembleFunctionAt(address);
 }
 
 template <typename ELFT>