yetangitu/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2025-10-03 09:49:23 +02:00
Code Issues Releases Wiki Activity

GP-4865/PR-6833 added missing javadoc, minor format improvement

Browse source
This commit is contained in:
ghidra007 2024-09-11 19:31:12 +00:00
parent 527cee1b76
commit 150fe43efd
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
Ghidra/Features/Decompiler/ghidra_scripts/classrecovery/RecoveredClassHelper.java
View file

@ -404,11 +404,14 @@ public class RecoveredClassHelper {
* @param function the given function * @param function the given function
* @param getThunkedFunction if true, use the thunked function in the map, if false use the * @param getThunkedFunction if true, use the thunked function in the map, if false use the
* directly called function from the calling function even if it is a thunk * directly called function from the calling function even if it is a thunk
* @param visited the set of function entry point addresses already processed
* @return a map of the given functions calling addresses to the called functions * @return a map of the given functions calling addresses to the called functions
* @throws CancelledException if cancelled * @throws CancelledException if cancelled
*/ */
public Map<Address, Function> getFunctionCallMap(Function function, boolean getThunkedFunction, Set<Address> visited) public Map<Address, Function> getFunctionCallMap(Function function, boolean getThunkedFunction,
Set<Address> visited)
throws CancelledException { throws CancelledException {
visited.add(function.getEntryPoint()); visited.add(function.getEntryPoint());
Map<Address, Function> functionCallMap = new HashMap<Address, Function>(); Map<Address, Function> functionCallMap = new HashMap<Address, Function>();
@ -435,7 +438,8 @@ public class RecoveredClassHelper {
Address functionAddress = reference.getFromAddress(); Address functionAddress = reference.getFromAddress();
Function secondHalfOfFunction = Function secondHalfOfFunction =
extendedFlatAPI.getReferencedFunction(functionAddress); extendedFlatAPI.getReferencedFunction(functionAddress);
if (secondHalfOfFunction != null && !visited.contains(secondHalfOfFunction.getEntryPoint())) { if (secondHalfOfFunction != null &&
!visited.contains(secondHalfOfFunction.getEntryPoint())) {
Map<Address, Function> functionCallMap2 = Map<Address, Function> functionCallMap2 =
getFunctionCallMap(secondHalfOfFunction, false, visited); getFunctionCallMap(secondHalfOfFunction, false, visited);
for (Address addr : functionCallMap2.keySet()) { for (Address addr : functionCallMap2.keySet()) {
@ -449,7 +453,8 @@ public class RecoveredClassHelper {
return functionCallMap; return functionCallMap;
} }
public Map<Address, Function> getFunctionCallMap(Function function, boolean getThunkedFunction) throws CancelledException { public Map<Address, Function> getFunctionCallMap(Function function, boolean getThunkedFunction)
throws CancelledException {
return getFunctionCallMap(function, getThunkedFunction, new HashSet<>()); return getFunctionCallMap(function, getThunkedFunction, new HashSet<>());
} }