helpers.py

from binaryninja import (
    log_info,
    log_debug,
    log_error,
    BackgroundTaskThread,
)

from .constants import (
    DEBUG,
    HL_BP_COLOR,
    HL_NO_COLOR,
)


def expose(f):
    "Decorator to set exposed flag on a function."
    f.exposed = True
    return f


def is_exposed(f):
    "Test whether another function should be publicly exposed."
    return getattr(f, 'exposed', False)


def ishex(s):
    return s.lower().startswith("0x") and map(lambda c: c in "0123456789abcdef", s[2:].lower())


def info(x):
    log_info("[+] {:s}".format(x))

def err(x):
    log_error("[-] {:s}".format(x))

def dbg(x):
    if DEBUG:
        log_debug("[*] {:s}".format(x))



class RunInBackground(BackgroundTaskThread):
    def __init__(self, target, cancel_cb=None, *args, **kwargs):
            BackgroundTaskThread.__init__(self, '', cancel_cb is not None)
            self.target = target
            self.args = args
            self.kwargs = kwargs
            self.cancel_cb = cancel_cb
            return

    def run(self):
        self.target(self, *self.args, **self.kwargs)
        return

    def cancel(self):
        self.cancel_cb()
        return