108 lines
2.9 KiB
Java
108 lines
2.9 KiB
Java
/* ###
|
|
* IP: GHIDRA
|
|
*
|
|
* 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.
|
|
*/
|
|
//Rid us of those pesky FF's that become bad instructions
|
|
//@category Cleanup
|
|
|
|
import ghidra.app.script.GhidraScript;
|
|
import ghidra.program.model.address.Address;
|
|
import ghidra.program.model.data.CategoryPath;
|
|
import ghidra.program.model.data.DataType;
|
|
|
|
import java.util.Arrays;
|
|
|
|
public class FFsBeGoneScript extends GhidraScript {
|
|
private Address addr;
|
|
byte[] bytes = new byte[10];
|
|
byte[] masks = new byte[10];
|
|
|
|
@Override
|
|
public void run() throws Exception {
|
|
DataType wordDataType =
|
|
currentProgram.getDataTypeManager().getDataType(new CategoryPath("/"), "word");
|
|
if (wordDataType == null) {
|
|
throw new Exception("Data type 'word' could not be found in the program.");
|
|
}
|
|
Arrays.fill(bytes, (byte) 0xff);
|
|
Arrays.fill(masks, (byte) 0xff);
|
|
addr = currentProgram.getMemory().getMinAddress();
|
|
advance();
|
|
while (addr != null && !monitor.isCancelled()) {
|
|
byte byt = 0;
|
|
try {
|
|
byt = currentProgram.getMemory().getByte(addr);
|
|
}
|
|
catch (Exception e) {
|
|
advance();
|
|
continue;
|
|
}
|
|
while ((byt & 0xff) == 0xff) {
|
|
if (isUndefinedData(addr)) {
|
|
try {
|
|
createData(addr, wordDataType);
|
|
}
|
|
catch (Exception e) {
|
|
println("Could not create data at " + addr.toString());
|
|
addr = addr.next();
|
|
break;
|
|
}
|
|
}
|
|
else {
|
|
break;
|
|
}
|
|
addr = currentProgram.getListing().getDefinedDataAt(addr).getMaxAddress().next();
|
|
try {
|
|
byt = currentProgram.getMemory().getByte(addr);
|
|
}
|
|
catch (Exception e) {
|
|
break;
|
|
}
|
|
}
|
|
advance();
|
|
}
|
|
}
|
|
|
|
private void advance() {
|
|
if (addr == null) {
|
|
return;
|
|
}
|
|
addr = currentProgram.getMemory().findBytes(addr, bytes, masks, true, monitor);
|
|
if (addr != null) {
|
|
if (!isUndefinedData(addr)) {
|
|
if (currentProgram.getListing().getInstructionContaining(addr) != null) {
|
|
addr =
|
|
currentProgram.getListing().getInstructionContaining(addr).getMaxAddress().next();
|
|
}
|
|
else if (currentProgram.getListing().getDefinedDataContaining(addr) != null) {
|
|
addr =
|
|
currentProgram.getListing().getDefinedDataContaining(addr).getMaxAddress().next();
|
|
}
|
|
advance();
|
|
}
|
|
}
|
|
}
|
|
|
|
private boolean isUndefinedData(Address address) {
|
|
if (currentProgram.getListing().getInstructionContaining(address) != null) {
|
|
return false;
|
|
}
|
|
if (currentProgram.getListing().getDefinedDataContaining(address) != null) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
}
|