#!/usr/bin/env python

import difflib
import logging
import sys
import os
from sortedcontainers import SortedDict

import angr
from angr.analyses.decompiler.decompilation_options import (
    PARAM_TO_OPTION,
    DecompilationOption,
)

logging.basicConfig(level=logging.CRITICAL, force=True)


def analyze_binary(binary_path):
    decompilation = SortedDict()

    try:
        project = angr.Project(binary_path, auto_load_libs=False)
        cfg = project.analyses.CFGFast(normalize=True)

        for function in cfg.functions.values():
            function.normalize()

            try:
                decomp = project.analyses.Decompiler(
                    func=function, cfg=cfg, options=[(PARAM_TO_OPTION["show_casts"], False)]
                )
                if decomp.codegen:
                    decompilation[function.addr] = decomp.codegen.text
            except Exception as e:
                print(e)

    except Exception as e:
        pass

    return decompilation


def check_file(file):
    result = None
    for i in range(3):
        print(f"{file} attempt {i}")
        latest = analyze_binary(file)
        if result is None:
            result = latest
        elif result != latest:
            for k in result:
                if k not in latest:
                    continue

                if result[k] == latest[k]:
                    continue

                for diff in difflib.unified_diff(
                    result[k].split("\n"),
                    latest[k].split("\n"),
                    fromfile=f"{file} func {k} baseline",
                    tofile=f"{file} func {k} attempt {i}",
                    n=3,
                ):
                    print(diff)
            return


def main():
    for file in sys.argv[1:]:
        if not os.path.isfile(file):
            continue
        check_file(file)


if __name__ == "__main__":
    main()