ida_highlight

ida_highlight.py

from idaapi import *
from idc import *

TRACE_FILE = AskFile(0, "*.txt", "Select trace file")
if not TRACE_FILE:
    Warning("Failed to select trace file")

else:
    HIGHLIGHT = AskLong(0, "Choose action: 0 = clear, 1 = highlight")

    if HIGHLIGHT != 0 and HIGHLIGHT != 1:
        Warning("Invalid action: {}".format(HIGHLIGHT))
    else:
        color = 0xff0000 if HIGHLIGHT else 0xffffff

        with open(TRACE_FILE) as lines:
            for i, line in enumerate(lines):
                if i == 0:  # skip the header
                    continue

                addr = int(line, 16)
                idc.SetColor(addr, idc.CIC_ITEM, color)