79 lines
2.5 KiB
Java
79 lines
2.5 KiB
Java
/* ###
|
|
* 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.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
import ghidra.app.script.GhidraScript;
|
|
import ghidra.program.database.ProgramDB;
|
|
import ghidra.program.model.lang.*;
|
|
import ghidra.program.model.listing.IncompatibleLanguageException;
|
|
import ghidra.program.util.LanguageTranslator;
|
|
import ghidra.program.util.LanguageTranslatorAdapter;
|
|
import ghidra.util.Msg;
|
|
import ghidra.util.exception.AssertException;
|
|
|
|
|
|
public class ForceRedisassembly extends GhidraScript {
|
|
|
|
@Override
|
|
public void run() throws Exception {
|
|
|
|
if (currentProgram == null) {
|
|
Msg.showError(this, null, "No Program Error", "No active program found");
|
|
return;
|
|
}
|
|
ProgramDB program = (ProgramDB)currentProgram;
|
|
|
|
Language lang = program.getLanguage();
|
|
|
|
LanguageTranslator translator = new MyLanguageTranslator(lang.getLanguageID(), lang.getVersion());
|
|
if (!translator.isValid()) {
|
|
return;
|
|
}
|
|
|
|
program.setLanguage(translator, program.getCompilerSpec().getCompilerSpecID(), true, monitor);
|
|
}
|
|
|
|
private static class MyLanguageTranslator extends LanguageTranslatorAdapter {
|
|
protected MyLanguageTranslator(LanguageID languageId, int version) {
|
|
super(languageId, version, languageId, version);
|
|
}
|
|
@Override
|
|
public boolean isValid() {
|
|
if (super.isValid()) {
|
|
try {
|
|
validateDefaultSpaceMap();
|
|
} catch (IncompatibleLanguageException e) {
|
|
throw new AssertException();
|
|
}
|
|
Register newContextReg = getNewLanguage().getContextBaseRegister();
|
|
if (newContextReg != null) {
|
|
Register oldContextReg = getOldLanguage().getContextBaseRegister();
|
|
if (oldContextReg == null || !isSameRegisterConstruction(oldContextReg, newContextReg)) {
|
|
throw new AssertException();
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
return "[" + getOldLanguageID() + " (Version " + getOldVersion() + ")] -> [" +
|
|
getNewLanguageID() + " (Version " + getNewVersion() + ")] {Forced Re-Disassembly Translator}";
|
|
}
|
|
}
|
|
}
|