blob: 8f2af5a66166b85766ff15aacf1bba2b2484561c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
#!/usr/bin/env python3
import sys
from pathlib import Path
from binaryninja import load
from binaryninja.warp import WarpContainer, WarpFunction, WarpTarget
def process_binary(input_file: str, output_dir: str) -> None:
input_path = Path(input_file)
output_dir = Path(output_dir)
output_dir.mkdir(parents=True, exist_ok=True)
bv = load(input_path)
bv.update_analysis_and_wait()
if not bv:
return
container = WarpContainer.add("Python")
output_file = output_dir / f"{input_path.stem}_analysis.warp"
# Add the source so we can add functions to it and then commit it (write to disk)
source = container.add_source(str(output_file))
if source is None:
print(f"Failed to create source {str(output_file)}")
return
# NOTE: You probably want to pull the platform from the function, but for this example it's fine.
target = WarpTarget(bv.platform)
# NOTE: You probably want to filter for functions with actual annotations, no point to signature a function with no symbol.
functions_to_warp = [WarpFunction(func) for func in bv.functions]
container.add_functions(target, source, functions_to_warp)
# Actually write the warp file to disk.
container.commit_source(source)
bv.file.close()
print(f"committed {len(functions_to_warp)} functions to {output_file}")
if __name__ == "__main__":
if len(sys.argv) != 3:
print(f"Usage: {sys.argv[0]} <input_binary> <output_directory>")
sys.exit(1)
process_binary(sys.argv[1], sys.argv[2])
|
