mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Giacomo Vercesi
87543f835a
Remove leftover shebangs from python files in `revng.cli._commands` as those files are no longer executable on their own
114 lines
3.2 KiB
Python
114 lines
3.2 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import logging
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
import idb
|
|
import yaml
|
|
|
|
from revng.cli.commands_registry import Command, Options, commands_registry
|
|
from revng.cli.revng import run_revng_command
|
|
from revng.model import YamlDumper # type: ignore
|
|
|
|
from .idb_converter import IDBConverter
|
|
|
|
|
|
class ImportIDBCommand(Command):
|
|
def __init__(self):
|
|
super().__init__(
|
|
("model", "import", "idb"), "Extract a rev.ng model from an IDB/I64 database"
|
|
)
|
|
|
|
def register_arguments(self, parser):
|
|
parser.add_argument("idb", help="Path to the IDB/I64 file to import")
|
|
parser.add_argument(
|
|
"--output", "-o", default="/dev/stdout", help="Output filepath (default stdout)"
|
|
)
|
|
parser.add_argument(
|
|
"--base", default=0x0, help="base address where dynamic objects should be loaded"
|
|
)
|
|
|
|
def run(self, options: Options):
|
|
# Suppress warnings from python-idb
|
|
logging.basicConfig(level=logging.ERROR)
|
|
|
|
idb_path = options.parsed_args.idb
|
|
output_path = options.parsed_args.output
|
|
|
|
# NOTE: This should be used for PIC only.
|
|
base_address = options.parsed_args.base
|
|
|
|
with idb.from_file(idb_path) as db:
|
|
idb_converter = IDBConverter(db, base_address, options.parsed_args.verbose)
|
|
revng_model = idb_converter.get_model()
|
|
|
|
yaml_model = yaml.dump(revng_model, Dumper=YamlDumper)
|
|
|
|
def temporary_file(suffix="", mode="w+"):
|
|
return NamedTemporaryFile(
|
|
prefix="revng-import-idb-",
|
|
suffix=suffix,
|
|
mode=mode,
|
|
delete=not options.keep_temporaries,
|
|
)
|
|
|
|
with temporary_file(suffix=".yml") as model_file:
|
|
model_file.write(yaml_model)
|
|
model_file.flush()
|
|
|
|
# Fix the model and do the clean-up.
|
|
run_revng_command(
|
|
[
|
|
"model",
|
|
"opt",
|
|
"-fix",
|
|
model_file.name,
|
|
"-o",
|
|
model_file.name,
|
|
],
|
|
options,
|
|
)
|
|
|
|
run_revng_command(
|
|
[
|
|
"model",
|
|
"opt",
|
|
"-deduplicate-equivalent-types",
|
|
model_file.name,
|
|
"-o",
|
|
model_file.name,
|
|
],
|
|
options,
|
|
)
|
|
|
|
run_revng_command(
|
|
[
|
|
"model",
|
|
"opt",
|
|
"-promote-original-name",
|
|
model_file.name,
|
|
"-o",
|
|
model_file.name,
|
|
],
|
|
options,
|
|
)
|
|
|
|
run_revng_command(
|
|
[
|
|
"model",
|
|
"opt",
|
|
"-purge-unnamed-and-unreachable-types",
|
|
model_file.name,
|
|
"-o",
|
|
output_path,
|
|
],
|
|
options,
|
|
)
|
|
|
|
return 0
|
|
|
|
|
|
commands_registry.register_command(ImportIDBCommand())
|