mirror of
https://github.com/revng/revng
synced 2026-06-21 14:07:57 +00:00
Alessandro Di Federico
103 lines
3.5 KiB
Python
103 lines
3.5 KiB
Python
#
|
|
# This file is distributed under the MIT License. See LICENSE.md for details.
|
|
#
|
|
|
|
import sys
|
|
from shutil import copyfileobj
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
from revng.cli.commands_registry import Command, CommandsRegistry, Options
|
|
from revng.cli.revng import run_revng_command
|
|
from revng.cli.support import log_error
|
|
|
|
|
|
class IRPipelineCommand(Command):
|
|
def __init__(self):
|
|
super().__init__(("llvm", "pipeline"), "revng-pipeline working on LLVM IR")
|
|
|
|
def register_arguments(self, parser):
|
|
parser.add_argument("input", type=str, default="/dev/stdin")
|
|
parser.add_argument("output", type=str, default="/dev/stdout")
|
|
parser.add_argument("from", type=str)
|
|
parser.add_argument("to", type=str)
|
|
parser.add_argument("target", type=str)
|
|
parser.add_argument("--analysis", type=str, default="")
|
|
parser.add_argument("--binary", type=str, default="")
|
|
|
|
def run(self, options: Options):
|
|
if options.remaining_args:
|
|
log_error("Unknown arguments passed in")
|
|
return 1
|
|
|
|
args = options.parsed_args
|
|
|
|
with NamedTemporaryFile(
|
|
prefix="revng-llvm-pipeline-", suffix=".yml", delete=not options.keep_temporaries
|
|
) as model, NamedTemporaryFile(
|
|
prefix="revng-llvm-pipeline-", suffix=".bc", delete=not options.keep_temporaries
|
|
) as module, open( # noqa: SIM115
|
|
args.input, "rb"
|
|
) if args.input != "-" else sys.stdin.buffer as input_file, NamedTemporaryFile(
|
|
prefix="revng-llvm-pipeline-",
|
|
suffix=".bc",
|
|
delete=not options.keep_temporaries,
|
|
mode="wb",
|
|
) as saved_input_file:
|
|
# Copy so we can work with stdin too
|
|
copyfileobj(input_file, saved_input_file) # type: ignore
|
|
saved_input_file.flush()
|
|
|
|
# Extract model
|
|
result = run_revng_command(
|
|
["model", "dump", saved_input_file.name, "-o", model.name], options
|
|
)
|
|
if result != 0:
|
|
return result
|
|
|
|
# Run revng-pipeline
|
|
target = f"--produce={args.to}/module.ll/{args.target}"
|
|
if args.analysis:
|
|
target = f"--analyze={args.to}/{args.analysis}/module.ll/{args.target}"
|
|
|
|
load_binary = []
|
|
if args.binary != "":
|
|
from_step = args.__dict__["from"]
|
|
load_binary = [
|
|
"-i",
|
|
f"{args.binary}:{from_step}/input",
|
|
"-i",
|
|
f"{args.binary}:begin/input",
|
|
]
|
|
|
|
result = run_revng_command(
|
|
[
|
|
"pipeline",
|
|
"-m",
|
|
model.name,
|
|
"-i",
|
|
f"""{saved_input_file.name}:{args.__dict__["from"]}/module.ll""",
|
|
"-o",
|
|
f"{module.name}:{args.to}/module.ll",
|
|
"--save-model",
|
|
model.name,
|
|
target,
|
|
*load_binary,
|
|
],
|
|
options,
|
|
)
|
|
if result != 0:
|
|
return result
|
|
|
|
# Re-inject the model
|
|
result = run_revng_command(
|
|
["model", "inject", model.name, module.name, "-o", args.output], options
|
|
)
|
|
if result != 0:
|
|
return result
|
|
|
|
return 0
|
|
|
|
|
|
def setup(commands_registry: CommandsRegistry):
|
|
commands_registry.register_command(IRPipelineCommand())
|