Nixyri/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2024-09-20 09:31:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/GP-4863_ghidra007_PR-6834_epinter_RTTI_Script_cache_vfunctions'

Browse Source
This commit is contained in:
Ryan Kurtz 2024-08-23 11:50:04 -04:00
commit d87add933e
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
Ghidra/Features/Decompiler/ghidra_scripts/classrecovery/RecoveredClassHelper.java
View File

@ -137,6 +137,7 @@ public class RecoveredClassHelper {
protected final boolean createBookmarks;
protected final boolean useShortTemplates;
protected final boolean nameVfunctions;
public HashMap<Address, Set<Function>> allVfunctions = new HashMap<>();
public RecoveredClassHelper(Program program, ServiceProvider serviceProvider,
FlatProgramAPI api, boolean createBookmarks, boolean useShortTemplates,
@ -478,20 +479,25 @@ public class RecoveredClassHelper {
return functionToLoadPcodeOps.get(function);
}
public Set<Function> getAllVfunctions(List<Address> vftableAddresses)
throws CancelledException {
Set<Function> allVfunctionsSet = new HashSet<Function>();
public Set<Function> getAllVfunctions(List<Address> vftableAddresses) throws CancelledException {
if (vftableAddresses.isEmpty()) {
return allVfunctionsSet;
return Collections.emptySet();
}
Set<Function> vfunctionSet = new HashSet<>();
for (Address vftableAddress : vftableAddresses) {
monitor.checkCancelled();
allVfunctionsSet.addAll(getVfunctions(vftableAddress));
if (!allVfunctions.containsKey(vftableAddress)) {
List<Function> funcList = getVfunctions(vftableAddress);
if (funcList == null) {
funcList = new ArrayList<>();
}
allVfunctions.put(vftableAddress, new HashSet<>(funcList));
}
vfunctionSet.addAll(allVfunctions.get(vftableAddress));
}
return allVfunctionsSet;
return vfunctionSet;
}
public Set<Function> getAllClassFunctionsWithVtableRef(List<Address> vftables)