Binary Ninja: Highlight ARM32 bl, blx, b calls - an example script

highlight_calls_example.py

"""
binary ninja script to highlight all "bl" instructions in 
an ARM32 binary. makes assembly easier to read.
"""

from binaryninja import *

TARGET_INSTRS = ['bl', 'blx', 'b']

# RGB color for bright red
RED_RGB = (255, 69, 29)

# build a highlight color from that
red = binaryninja.highlight.HighlightColor(
    red=RED_RGB[0],
    green=RED_RGB[1],
    blue=RED_RGB[2]
)

# get all instructions
instrs = bv.instructions
# go through them, check if an instruction is in the 
# target instructions
for instr in instrs:
    (mnemonic, addr) = (instr[0][0].text, instr[1])
    # if we find a target instruction, highlight it
    if mnemonic in TARGET_INSTRS:
        # optional, remove if necessary
        print("highlighting {} @ {}".format(mnemonic, hex(addr)))
        # get the basic block at address x
        block = bv.get_basic_blocks_at(addr)
        # get the function at the basic block to get access to
        # the set_auto_instr_highlight function
        f = block[0].function
        f.set_auto_instr_highlight(addr, red)