mirror of
https://github.com/Maschell/GhidraRPXLoader.git
synced 2024-12-01 20:24:15 +01:00
Add references to imports. Currently the imports are handled as data, but can be converted to functions manually.
This commit is contained in:
parent
7066bed843
commit
81a082edb1
@ -14,11 +14,22 @@ import ghidra.app.util.bin.ByteArrayProvider;
|
|||||||
import ghidra.app.util.bin.ByteProvider;
|
import ghidra.app.util.bin.ByteProvider;
|
||||||
import ghidra.app.util.bin.format.elf.ElfException;
|
import ghidra.app.util.bin.format.elf.ElfException;
|
||||||
import ghidra.app.util.bin.format.elf.ElfHeader;
|
import ghidra.app.util.bin.format.elf.ElfHeader;
|
||||||
|
import ghidra.app.util.bin.format.elf.ElfRelocation;
|
||||||
|
import ghidra.app.util.bin.format.elf.ElfRelocationTable;
|
||||||
|
import ghidra.app.util.bin.format.elf.ElfSectionHeader;
|
||||||
|
import ghidra.app.util.bin.format.elf.ElfSymbol;
|
||||||
import ghidra.app.util.importer.MemoryConflictHandler;
|
import ghidra.app.util.importer.MemoryConflictHandler;
|
||||||
import ghidra.app.util.importer.MessageLog;
|
import ghidra.app.util.importer.MessageLog;
|
||||||
import ghidra.app.util.importer.MessageLogContinuesFactory;
|
import ghidra.app.util.importer.MessageLogContinuesFactory;
|
||||||
|
import ghidra.program.model.address.Address;
|
||||||
|
import ghidra.program.model.address.AddressSpace;
|
||||||
import ghidra.program.model.listing.Program;
|
import ghidra.program.model.listing.Program;
|
||||||
|
import ghidra.program.model.symbol.RefType;
|
||||||
|
import ghidra.program.model.symbol.Reference;
|
||||||
|
import ghidra.program.model.symbol.SourceType;
|
||||||
import ghidra.util.exception.CancelledException;
|
import ghidra.util.exception.CancelledException;
|
||||||
|
import ghidra.util.exception.DuplicateNameException;
|
||||||
|
import ghidra.util.exception.InvalidInputException;
|
||||||
import ghidra.util.task.TaskMonitor;
|
import ghidra.util.task.TaskMonitor;
|
||||||
|
|
||||||
public class RPXLoader extends ElfLoader {
|
public class RPXLoader extends ElfLoader {
|
||||||
@ -65,7 +76,40 @@ public class RPXLoader extends ElfLoader {
|
|||||||
GenericFactory factory = MessageLogContinuesFactory.create(log);
|
GenericFactory factory = MessageLogContinuesFactory.create(log);
|
||||||
byte[] data = RPXUtils.convertRPX(provider, monitor);
|
byte[] data = RPXUtils.convertRPX(provider, monitor);
|
||||||
ElfHeader elf = ElfHeader.createElfHeader(factory, new ByteArrayProvider(data));
|
ElfHeader elf = ElfHeader.createElfHeader(factory, new ByteArrayProvider(data));
|
||||||
|
|
||||||
ElfProgramBuilder.loadElf(elf, program, options, log, handler, monitor);
|
ElfProgramBuilder.loadElf(elf, program, options, log, handler, monitor);
|
||||||
|
|
||||||
|
AddressSpace aspace = program.getAddressFactory().getDefaultAddressSpace();
|
||||||
|
|
||||||
|
for (ElfRelocationTable table : elf.getRelocationTables()) {
|
||||||
|
for (ElfRelocation reloc : table.getRelocations()) {
|
||||||
|
int sindex = reloc.getSymbolIndex();
|
||||||
|
|
||||||
|
ElfSymbol symbol = table.getAssociatedSymbolTable().getSymbols()[sindex];
|
||||||
|
ElfSectionHeader section = elf.getSections()[symbol.getSectionHeaderIndex()];
|
||||||
|
if (section.getType() == RPXUtils.SHT_RPL_IMPORTS) {
|
||||||
|
int offset = (int) (reloc.getOffset() & ~3);
|
||||||
|
|
||||||
|
String rplName = section.getNameAsString();
|
||||||
|
if (rplName.contains("import_")) {
|
||||||
|
rplName = rplName.split("import_")[1];
|
||||||
|
if (!rplName.endsWith(".rpl")) {
|
||||||
|
rplName += ".rpl";
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
Address addr = aspace.getAddress(offset);
|
||||||
|
Reference r = program.getReferenceManager().addExternalReference(addr, rplName,
|
||||||
|
symbol.getNameAsString(), aspace.getAddress(0), SourceType.USER_DEFINED, 1,
|
||||||
|
RefType.DATA);
|
||||||
|
program.getReferenceManager().setPrimary(r, true);
|
||||||
|
program.getListing().setComment(addr, 0, rplName + "::" + symbol.getNameAsString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
} catch (ElfException e) {
|
} catch (ElfException e) {
|
||||||
throw new IOException(e.getMessage());
|
throw new IOException(e.getMessage());
|
||||||
} catch (CancelledException e) {
|
} catch (CancelledException e) {
|
||||||
@ -73,6 +117,10 @@ public class RPXLoader extends ElfLoader {
|
|||||||
throw new IOException(e.getMessage());
|
throw new IOException(e.getMessage());
|
||||||
} catch (DataFormatException e) {
|
} catch (DataFormatException e) {
|
||||||
throw new IOException(e.getMessage());
|
throw new IOException(e.getMessage());
|
||||||
|
} catch (DuplicateNameException e) {
|
||||||
|
throw new IOException(e.getMessage());
|
||||||
|
} catch (InvalidInputException e) {
|
||||||
|
throw new IOException(e.getMessage());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user