Mirrors_Framework/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2026-05-10 08:19:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

GP-0 Updated FixupCompositeDataTypesScript to support archives

Browse Source
This commit is contained in:
ghidra1
2021-10-05 13:08:07 -04:00
parent 32ebac6048
commit 086714df5f
1 changed files with 64 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Ghidra/Features/Base/ghidra_scripts/FixupCompositeDataTypesScript.java
+64 -5
View File
@@ -23,25 +23,84 @@
//
// This script can be run multiple times without harm
//@category Data Types
import java.util.ArrayList;
import ghidra.app.script.GhidraScript;
import ghidra.program.database.data.DataTypeManagerDB;
import ghidra.app.services.DataTypeManagerService;
import ghidra.framework.model.DomainFile;
import ghidra.framework.plugintool.PluginTool;
import ghidra.program.database.data.*;
import ghidra.program.model.data.BuiltInDataTypeManager;
import ghidra.program.model.data.DataTypeManager;
import ghidra.program.model.listing.Program;
public class FixupCompositeDataTypesScript extends GhidraScript {
@Override
protected void run() throws Exception {
if (currentProgram == null) {
PluginTool tool = state.getTool();
if (tool == null) {
print("This script does not support headless use");
}
DataTypeManagerService service = tool.getService(DataTypeManagerService.class);
if (service == null) {
popup("This script requires the DataTypeManagerService");
return;
}
if (!currentProgram.hasExclusiveAccess()) {
popup("This script requires an exclusive checkout of the program");
ArrayList<DTMWrapper> dtms = new ArrayList<>();
for (DataTypeManager dtm : service.getDataTypeManagers()) {
if (dtm instanceof BuiltInDataTypeManager) {
continue;
}
if (dtm instanceof ProgramDataTypeManager) {
dtms.add(0, new DTMWrapper((ProgramDataTypeManager) dtm));
}
else if (dtm instanceof DataTypeManagerDB) {
dtms.add(new DTMWrapper((DataTypeManagerDB) dtm));
}
}
DataTypeManagerDB dtm =
askChoice("Fixup All Composites", "Select Data Type Manager: ", dtms, dtms.get(0)).dtm;
if (dtm instanceof ProgramDataTypeManager) {
Program program = ((ProgramDataTypeManager) dtm).getProgram();
if (!program.hasExclusiveAccess()) {
popup("Shared program must have an exclusive checkout.");
return;
}
}
else if (dtm instanceof ProjectDataTypeManager) {
DomainFile df = ((ProjectDataTypeManager) dtm).getDomainFile();
if (df.isVersioned() && !df.isCheckedOutExclusive()) {
popup("Shared project archive must have an exclusive checkout.");
return;
}
}
if (!dtm.isUpdatable()) {
popup("Selected archive must be open for update.");
return;
}
DataTypeManagerDB dtm = (DataTypeManagerDB) currentProgram.getDataTypeManager();
dtm.fixupComposites(monitor);
}
private class DTMWrapper {
DataTypeManagerDB dtm;
DTMWrapper(DataTypeManagerDB dtm) {
this.dtm = dtm;
}
@Override
public String toString() {
return dtm.getName();
}
}
}