73 lines
2.4 KiB
Java
73 lines
2.4 KiB
Java
package re3lib;
|
|
|
|
import java.io.File;
|
|
import java.io.PrintWriter;
|
|
import java.util.ArrayList;
|
|
import java.util.HashSet;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
|
|
import ghidra.app.script.GhidraScript;
|
|
import ghidra.program.model.data.DataType;
|
|
import ghidra.program.model.data.EnumDataType;
|
|
import ghidra.program.model.data.ProgramBasedDataTypeManager;
|
|
import ghidra.program.model.data.Structure;
|
|
import ghidra.program.model.data.TypedefDataType;
|
|
import ghidra.program.model.listing.Program;
|
|
|
|
public class TypeDumper {
|
|
Program currentProgram;
|
|
GhidraScript script;
|
|
|
|
public TypeDumper(GhidraScript script) {
|
|
this.script = script;
|
|
currentProgram = script.getCurrentProgram();
|
|
RecompileConfig.INSTANCE = new RecompileConfig(script);
|
|
}
|
|
|
|
public void run() throws Exception {
|
|
ProgramBasedDataTypeManager dtm = currentProgram.getDataTypeManager();
|
|
|
|
HashSet<String> typeBlacklist = Utils.loadStructBlacklist(RecompileConfig.INSTANCE.typeBlacklistPath);
|
|
|
|
if (typeBlacklist == null) {
|
|
script.println("Building struct blacklist from existing data types");
|
|
typeBlacklist = new HashSet<>();
|
|
Iterator<DataType> it = dtm.getAllDataTypes();
|
|
while (it.hasNext()) {
|
|
DataType dt = it.next();
|
|
if (dt instanceof Structure || dt instanceof TypedefDataType) {
|
|
typeBlacklist.add(dt.getDisplayName());
|
|
}
|
|
}
|
|
Utils.saveStructBlacklist(typeBlacklist, RecompileConfig.INSTANCE.typeBlacklistPath);
|
|
}
|
|
|
|
List<DataType> filteredTypes = new ArrayList<>();
|
|
Iterator<DataType> it = dtm.getAllDataTypes();
|
|
while (it.hasNext()) {
|
|
DataType dt = it.next();
|
|
if (dt instanceof Structure || dt instanceof TypedefDataType || dt instanceof EnumDataType) {
|
|
if (typeBlacklist.contains(dt.getDisplayName()))
|
|
continue;
|
|
// println("Adding: " + dt.getDisplayName() + " - " +
|
|
// dt.getClass().getSimpleName());
|
|
filteredTypes.add(dt);
|
|
}
|
|
}
|
|
|
|
try (PrintWriter writer = new PrintWriter(new File(RecompileConfig.INSTANCE.outputDir, "gh_types.h"),
|
|
"UTF-8")) {
|
|
Utils.headerGuardPre(writer, "STRUCTS");
|
|
writer.println("// AUTO-GENERATED FILE ");
|
|
writer.println("#include <r3/binders/type.h>");
|
|
|
|
DataTypeWriter dtw = new DataTypeWriter(dtm, writer);
|
|
dtw.blacklistedTypes = typeBlacklist;
|
|
dtw.write(filteredTypes, script.getMonitor());
|
|
|
|
Utils.headerGuardPost(writer, "STRUCTS");
|
|
}
|
|
}
|
|
}
|