100 lines
2.3 KiB
Python
100 lines
2.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import binaryninja
|
||
|
|
import argparse
|
||
|
|
import angr
|
||
|
|
import os
|
||
|
|
import json
|
||
|
|
|
||
|
|
from cle.backends import Region, Regions
|
||
|
|
from IPython import embed
|
||
|
|
from typing import List, Dict
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("file", help="File to analyze")
|
||
|
|
parser.add_argument("-d", "--debug", help="Enable debug mode", action = "store_true", default = False)
|
||
|
|
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
class WrappedBin:
|
||
|
|
def __init__(self, filename: str):
|
||
|
|
self.angr: angr.Project = angr.Project(filename)
|
||
|
|
|
||
|
|
"""
|
||
|
|
Interface
|
||
|
|
"""
|
||
|
|
class Analysis:
|
||
|
|
def __init__(self, target: WrappedBin):
|
||
|
|
self.target = target
|
||
|
|
|
||
|
|
def name(self) -> str:
|
||
|
|
pass
|
||
|
|
def execute(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def __get_relevant_vars__(self) -> Dict:
|
||
|
|
# get all attributes of the instance
|
||
|
|
instance_vars = vars(self)
|
||
|
|
# filter out attributes that belong to the parent class
|
||
|
|
return {k: str(v) for k, v in instance_vars.items() if not k.startswith("__") and k != "target"}
|
||
|
|
|
||
|
|
def __json_dumps__(self) -> str:
|
||
|
|
return json.dumps(self.__get_relevant_vars__())
|
||
|
|
|
||
|
|
class AnalysisManager:
|
||
|
|
def __init__(self, target: WrappedBin, tasks: List[Analysis] = []):
|
||
|
|
self.target: WrappedBin = target
|
||
|
|
self.tasks: List[Analysis] = []
|
||
|
|
|
||
|
|
for t in tasks:
|
||
|
|
self.add_task(t)
|
||
|
|
|
||
|
|
def add_task(self, task: Analysis):
|
||
|
|
self.tasks.append(task(self.target))
|
||
|
|
|
||
|
|
def run_tasks(self):
|
||
|
|
n_tasks = len(self.tasks)
|
||
|
|
|
||
|
|
print("Starting analyses...")
|
||
|
|
|
||
|
|
for n, t in enumerate(self.tasks):
|
||
|
|
print(f"\t[{n + 1}/{n_tasks}] Running {t.name()}")
|
||
|
|
|
||
|
|
ret = t.execute()
|
||
|
|
|
||
|
|
if ret:
|
||
|
|
print(ret)
|
||
|
|
|
||
|
|
print("Done")
|
||
|
|
|
||
|
|
|
||
|
|
class LoaderFacts(Analysis):
|
||
|
|
def name(self) -> str:
|
||
|
|
return "Loader Facts"
|
||
|
|
|
||
|
|
def execute(self):
|
||
|
|
self.sections: Regions = self.target.angr.loader.all_objects[0].sections
|
||
|
|
self.aslr: bool = self.target.angr.loader.aslr
|
||
|
|
|
||
|
|
return self.__json_dumps__()
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
if not os.path.exists(args.file):
|
||
|
|
print("File does not exist")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
target = WrappedBin(args.file)
|
||
|
|
tasks = [ LoaderFacts ]
|
||
|
|
am = AnalysisManager(target, tasks)
|
||
|
|
|
||
|
|
am.run_tasks()
|
||
|
|
|
||
|
|
if args.debug:
|
||
|
|
embed()
|
||
|
|
|
||
|
|
return
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|