Binary ninja snippet script to add trace based function detection on gba roms (through mgba)

trace_function_detection.py

# Trace function detection
#
# Binary ninja script for trace based function detection on gba using
# the 'tracing' branch of this fork: https://github.com/SiD3W4y/mgba
path = get_open_filename_input("mgba trace file")

if not path:
    show_message_box("Function detection", "Please specify a file")

lines = open(path, "r").readlines()
funcs = []

for line in lines:
    line = line.strip()

    if len(line) > 5:
        chks = line.split(" ")

        if len(chks) == 2:
            funcs.append(int(chks[0], 16))

duplicates = 0

for fnaddr in funcs:
    fns = bv.get_functions_containing(fnaddr)

    # We found a new function
    if not fns:
        bv.create_user_function(fnaddr, plat=Architecture["thumb2"].standalone_platform)
    else:
        duplicates += 1

show_message_box("Function detection", "{} functions found ({} duplicates)".format(len(funcs), duplicates))