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

GP-5545_5600: ProgramLoader and more flexible loader args

Browse source
This commit is contained in:
Ryan Kurtz 2025-07-01 08:18:36 -04:00
parent 4aa78ae6d0
commit d3aed2c4b3
47 changed files with 2026 additions and 976 deletions
Download patch file Download diff file Expand all files Collapse all files

60
Ghidra/Features/FunctionID/ghidra_scripts/MSLibBatchImportWorker.java
View file

@ -4,9 +4,9 @@
* 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.
@ -53,7 +53,6 @@ import java.io.File;
import java.io.IOException;
import java.lang.management.ManagementFactory;
import java.util.*;
import java.util.function.Predicate;
import org.apache.commons.io.FileUtils;
@ -63,19 +62,19 @@ import ghidra.app.util.bin.*;
import ghidra.app.util.bin.format.coff.*;
import ghidra.app.util.bin.format.coff.archive.CoffArchiveHeader;
import ghidra.app.util.bin.format.coff.archive.CoffArchiveMemberHeader;
import ghidra.app.util.importer.*;
import ghidra.app.util.importer.MessageLog;
import ghidra.app.util.importer.ProgramLoader;
import ghidra.app.util.opinion.*;
import ghidra.framework.model.*;
import ghidra.framework.store.local.LocalFileSystem;
import ghidra.program.model.listing.Program;
import ghidra.util.InvalidNameException;
import ghidra.util.exception.*;
import ghidra.util.exception.CancelledException;
import ghidra.util.exception.VersionException;
import ghidra.util.task.TaskMonitor;
import utilities.util.FileUtilities;
public class MSLibBatchImportWorker extends GhidraScript {
final static Predicate<Loader> LOADER_FILTER = new SingleLoaderFilter(MSCoffLoader.class);
final static LoadSpecChooser LOADSPEC_CHOOSER = new CsHintLoadSpecChooser("windows");
private static String getProcessId(String fallback) {
// something like '<pid>@<hostname>', at least in SUN / Oracle JVMs
@ -173,8 +172,7 @@ public class MSLibBatchImportWorker extends GhidraScript {
}
private void importLibrary(DomainFolder currentLibraryFolder, File file, MessageLog log)
throws CancelledException, DuplicateNameException, InvalidNameException,
VersionException, IOException {
throws CancelledException, InvalidNameException, VersionException, IOException {
try (RandomAccessByteProvider provider = new RandomAccessByteProvider(file)) {
if (!CoffArchiveHeader.isMatch(provider)) {
return;
@ -196,30 +194,34 @@ public class MSLibBatchImportWorker extends GhidraScript {
Pair<DomainFolder, String> pair =
getFolderAndUniqueFile(currentLibraryFolder, preferredName);
LoadResults<? extends DomainObject> loadResults = null;
try {
loadResults = AutoImporter.importFresh(coffProvider,
state.getProject(), pair.first.getPathname(), this, log,
monitor, LOADER_FILTER, LOADSPEC_CHOOSER, pair.second,
OptionChooser.DEFAULT_OPTIONS);
try (LoadResults<? extends DomainObject> loadResults =
ProgramLoader.builder()
.source(coffProvider)
.project(state.getProject())
.projectFolderPath(pair.first.getPathname())
.loaders(MSCoffLoader.class)
.compiler("windows")
.name(pair.second)
.log(log)
.monitor(monitor)
.load()) {
for (Loaded<? extends DomainObject> loaded : loadResults) {
if (loaded.getDomainObject() instanceof Program program) {
loaded.save(state.getProject(), log, monitor);
println(
"Imported " + program.getDomainFile().getPathname());
DomainFile progFile = program.getDomainFile();
if (!progFile.isVersioned()) {
progFile.addToVersionControl(initalCheckInComment,
false, monitor);
DomainObject obj = loaded.getDomainObject(this);
try {
if (obj instanceof Program) {
loaded.save(monitor);
DomainFile progFile = obj.getDomainFile();
println("Imported " + progFile.getPathname());
if (!progFile.isVersioned()) {
progFile.addToVersionControl(initalCheckInComment,
false, monitor);
}
}
}
}
}
finally {
if (loadResults != null) {
loadResults.release(this);
finally {
obj.release(this);
}
}
}
}