mark xref to specified address with tag feature in binaryninja

marker.py

from binaryninja import *

class ReferenceMarker(object):

    def __init__(self, bv):
        self.bv = bv

    def xref_to(self, addr):
        return self.bv.get_code_refs(addr)

    def mark_all_references_to(self, addr):
    	tt = self.bv.create_tag_type("link to %x" % (addr), "x")
        for r in self.xref_to(addr):
            xref_func = r.function
            xref_addr = r.address
            xref_func.set_auto_instr_highlight(xref_addr, HighlightStandardColor.YellowHighlightColor)
            self.bv.create_user_data_tag(xref_addr, tt, xref_func.name)

def mark(bv, addr):
    ReferenceMarker(bv).mark_all_references_to(addr)

PluginCommand.register_for_address("marker", "mark and tag all reference to specified addr", mark)