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

GP-3765 Updated AutoVersionTracking script and task to process implied matches when running from the script if user wants them.

Browse source
This commit is contained in:
ghidra007 2023-09-19 16:39:44 +00:00
parent 180b09041c
commit 7f5eb65e55
3 changed files with 206 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

26
Ghidra/Features/VersionTracking/ghidra_scripts/AutoVersionTrackingScript.java
View file

@ -23,6 +23,7 @@ import ghidra.feature.vt.api.main.VTSession;
import ghidra.feature.vt.api.util.VTOptions;
import ghidra.feature.vt.gui.actions.AutoVersionTrackingTask;
import ghidra.feature.vt.gui.plugin.VTController;
import ghidra.feature.vt.gui.util.VTOptionDefines;
import ghidra.framework.model.DomainFolder;
import ghidra.framework.options.ToolOptions;
import ghidra.framework.plugintool.PluginTool;
@ -68,6 +69,9 @@ public class AutoVersionTrackingScript extends GhidraScript {
return;
}
boolean autoCreateImpliedMatches = askYesNo("Implied Matches?",
"Would you like the script to figure out implied matches from any matches it creates?");
// Need to end the script transaction or it interferes with vt things that need locks
end(true);
@ -76,12 +80,32 @@ public class AutoVersionTrackingScript extends GhidraScript {
folder.createFile(name, session, monitor);
ToolOptions options = getOptions();
boolean originalImpliedMatchSetting =
options.getBoolean(VTOptionDefines.AUTO_CREATE_IMPLIED_MATCH, false);
options.setBoolean(VTOptionDefines.AUTO_CREATE_IMPLIED_MATCH, autoCreateImpliedMatches);
AutoVersionTrackingTask autoVtTask =
new AutoVersionTrackingTask(session, getOptions(), 1.0, 10.0);
new AutoVersionTrackingTask(session, options, 0.95, 10.0);
TaskLauncher.launch(autoVtTask);
// if not running headless user can decide whether to save or not
// if running headless - must save here or nothing that was done in this script will be
// accessible later.
if (isRunningHeadless()) {
session.save();
}
options.setBoolean(VTOptionDefines.AUTO_CREATE_IMPLIED_MATCH, originalImpliedMatchSetting);
println(autoVtTask.getStatusMsg());
}
private ToolOptions getOptions() {
ToolOptions vtOptions = new VTOptions("Dummy");
PluginTool tool = state.getTool();