mirror of
https://github.com/bitdefender/bddisasm.git
synced 2024-12-04 21:48:09 +00:00
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
|
#
|
||
|
|
# Copyright (c) 2024 Bitdefender
|
||
|
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
#
|
||
|
|
import os
|
||
|
|
|
||
|
|
from core import util
|
||
|
|
|
||
|
|
|
||
|
|
class Generate(object):
|
||
|
|
def __init__(self, f_disasm: str, f_nasm: str, verbose: bool):
|
||
|
|
self._f_disasm = f_disasm
|
||
|
|
self._f_nasm = f_nasm
|
||
|
|
self._verbse = verbose
|
||
|
|
|
||
|
|
def run(self, data: str, d_path: str, overwrite=False) -> type(None):
|
||
|
|
"""
|
||
|
|
Generates the '.result' and '.test' files.
|
||
|
|
|
||
|
|
If the 'overwrite' options is used, the '.result' file is regenerated even if it exists.
|
||
|
|
If the path to nasm is not provided, the '.test' file is not generated.
|
||
|
|
"""
|
||
|
|
if not self._f_disasm or not os.path.exists(self._f_disasm):
|
||
|
|
return
|
||
|
|
|
||
|
|
f_source = data["f_source"]
|
||
|
|
f_result = data["f_result"]
|
||
|
|
f_test = data["f_test"]
|
||
|
|
options = data["options"]
|
||
|
|
command = data["command"]
|
||
|
|
|
||
|
|
f_source = os.path.join(d_path, f_source)
|
||
|
|
f_result = os.path.join(d_path, f_result)
|
||
|
|
f_test = os.path.join(d_path, f_test)
|
||
|
|
|
||
|
|
if self._f_nasm and os.path.exists(self._f_nasm):
|
||
|
|
if not os.path.exists(f_test):
|
||
|
|
if os.path.exists(f_source):
|
||
|
|
util.pr_debug("[+] generate .test file ... (%s)" % (f_test))
|
||
|
|
cmd = [self._f_nasm, f_source, "-o", f_test]
|
||
|
|
if self._verbse:
|
||
|
|
util.pr_debug("[*] run subprocess using cmd : %s" % (cmd))
|
||
|
|
_, _, _ = util.rsubprocess(cmd)
|
||
|
|
else:
|
||
|
|
util.pr_warn(
|
||
|
|
"[-] .asm file is missing (%s) - skip ..." % (f_source)
|
||
|
|
)
|
||
|
|
|
||
|
|
if overwrite or not os.path.exists(f_result):
|
||
|
|
util.pr_debug("[+] generate .result file ... (%s)" % (f_result))
|
||
|
|
cmd = [self._f_disasm, command, "-exi", *options.split(" "), "-f", f_test]
|
||
|
|
if self._verbse:
|
||
|
|
util.pr_debug("[*] run subprocess using cmd : %s" % (cmd))
|
||
|
|
stdout, _, returncode = util.rsubprocess(cmd)
|
||
|
|
if returncode != 0:
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
stdout = stdout.replace("\r", "")
|
||
|
|
with open(f_result, "w") as fd:
|
||
|
|
fd.write(stdout)
|