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

GT-3294 Added support for DB FixedField with improved indexing.

Browse source
This commit is contained in:
ghidra1 2020-02-24 18:02:01 -05:00
parent 14d4c87ef4
commit fcb3151f94
224 changed files with 9574 additions and 7913 deletions
Download patch file Download diff file Expand all files Collapse all files

24
Ghidra/Features/Base/developer_scripts/ConsistencyCheck.java
View file

@ -1,6 +1,5 @@
/* ###
* IP: GHIDRA
* REVIEWED: YES
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@ -15,12 +14,11 @@
* limitations under the License.
*/
// Performs database consistency check on the current program
import db.DBHandle;
import ghidra.app.script.GhidraScript;
import ghidra.app.services.ProgramManager;
import ghidra.framework.model.DomainFile;
import ghidra.program.database.ProgramDB;
import ghidra.program.model.listing.Program;
import db.DBHandle;
public class ConsistencyCheck extends GhidraScript {
@ -56,6 +54,8 @@ public class ConsistencyCheck extends GhidraScript {
return;
}
monitor.checkCanceled();
if (!df.canSave() || !currentProgram.hasExclusiveAccess()) {
popup("Program database is NOT consistent!\nRebuild requires exclusive checkout.");
return;
@ -67,19 +67,22 @@ public class ConsistencyCheck extends GhidraScript {
}
end(false);
ProgramDB program = (ProgramDB) df.getDomainObject(this, false, false, monitor);
programMgr.closeProgram(currentProgram, true);
currentProgram = (Program) df.getDomainObject(this, false, false, monitor);
dbh = ((ProgramDB) currentProgram).getDBHandle();
monitor.clearCanceled(); // compensate for Script Manager cancelling task on program close
dbh = program.getDBHandle();
try {
boolean success = false;
long txId = dbh.startTransaction();
int txId = program.startTransaction("Rebuild DB Indexes");
try {
success = dbh.rebuild(monitor);
}
finally {
dbh.endTransaction(txId, success);
program.endTransaction(txId, success);
}
if (!success) {
@ -92,11 +95,12 @@ public class ConsistencyCheck extends GhidraScript {
return;
}
currentProgram.save("DB Rebuild", monitor);
program.save("DB Rebuild", monitor);
}
finally {
currentProgram.release(this);
currentProgram = programMgr.openProgram(df);
programMgr.openProgram(program);
program.release(this);
currentProgram = program;
start();
}
}