diff options
Diffstat (limited to 'arch/arm64/arch_arm64.cpp')
| -rw-r--r-- | arch/arm64/arch_arm64.cpp | 19 |
1 files changed, 8 insertions, 11 deletions
diff --git a/arch/arm64/arch_arm64.cpp b/arch/arm64/arch_arm64.cpp index 68cfae5a..be856e79 100644 --- a/arch/arm64/arch_arm64.cpp +++ b/arch/arm64/arch_arm64.cpp @@ -2448,20 +2448,17 @@ class Arm64ImportedFunctionRecognizer : public FunctionRecognizer if (jumpOperand.GetSourceRegister<LLIL_REG>() != targetReg) return false; - Ref<Symbol> funcSym = Symbol::ImportedFunctionFromImportAddressSymbol(sym, func->GetStart()); // func is plt function + Ref<Symbol> funcSym = Symbol::ImportedFunctionFromImportAddressSymbol(sym, func->GetStart()); data->DefineAutoSymbol(funcSym); - for (auto& extSym : data->GetSymbolsByName(funcSym->GetRawName())) - { - if (extSym->GetType() == ExternalSymbol) - { - DataVariable var; - if (data->GetDataVariableAtAddress(extSym->GetAddress(), var)) - { - func->ApplyImportedTypes(funcSym, var.type); - } - return true; + auto extSym = data->GetSymbolsByName(funcSym->GetRawName(), data->GetExternalNameSpace()); + if (!extSym.empty()) { + DataVariable var; + if (data->GetDataVariableAtAddress(extSym.front()->GetAddress(), var)) + { + func->ApplyImportedTypes(funcSym, var.type); } + return true; } return false; } |