如何为IDA Pro伪代码变量设置自定义颜色？

要在IDA伪代码窗口中使用自定义RGB颜色（比如#00FF00），不能直接依赖ida_lines.COLSTR（它仅支持预定义的SCOLOR_*常量），需要直接嵌入IDA支持的ANSI颜色控制序列来实现。

核心原理 #

IDA的伪代码行支持嵌入ANSI转义码指定自定义颜色，格式规则：

• 开启自定义前景色：{COLOR_ON}\x1B[38;2;R;G;Bm（R/G/B为0-255的RGB分量值）
• 关闭颜色：{COLOR_OFF}
  其中COLOR_ON对应chr(1)或ida_lines.COLOR_ON，COLOR_OFF对应chr(2)或ida_lines.COLOR_OFF。

修改后的插件代码 #

以下是调整后的完整代码，替换了原有的颜色设置逻辑，实现自定义RGB颜色：

import idaapi, ida_kernwin, ida_lines

class ColorizeVariable(ida_kernwin.action_handler_t):

    def __init__(self):
        ida_kernwin.action_handler_t.__init__(self)

    def activate(self, ctx):
        if ida_kernwin.get_widget_type(ctx.widget) == ida_kernwin.BWN_PSEUDOCODE:
            vu = idaapi.get_widget_vdui(ctx.widget)
            pc = vu.cfunc.get_pseudocode()
            find = "MyVar"

            # 定义#00FF00（纯绿）的自定义颜色控制序列
            custom_color_on = f"{ida_lines.COLOR_ON}\x1B[38;2;0;255;0m"
            custom_color_off = ida_lines.COLOR_OFF

            for sl in pc:
                # 替换目标文本并应用自定义颜色
                sl.line = sl.line.replace(find, f"{custom_color_on}{find}{custom_color_off}")
            
            # 强制刷新视图确保颜色生效
            vu.refresh_view(True)
        return 0

    def update(self, ctx):
        return ida_kernwin.AST_ENABLE_ALWAYS

class ida_plugin_container(idaapi.plugin_t):
    flags         = idaapi.PLUGIN_UNL
    comment       = '自定义伪代码变量颜色插件'
    help          = '为伪代码中的指定变量设置自定义RGB颜色'
    wanted_name   = "自定义变量着色"
    wanted_hotkey = 'Shift-Q'

    def init(self):
        action_desc = idaapi.action_desc_t(
            'myAction',
            '着色指定变量',
            ColorizeVariable(),
            'Ctrl+R',
            '为伪代码中的MyVar设置自定义颜色',
            10)
        ida_kernwin.unregister_action('myAction')
        ida_kernwin.register_action(action_desc)
        ida_kernwin.attach_action_to_toolbar("SearchToolBar", 'myAction')
        print('自定义变量着色插件已加载')
        return idaapi.PLUGIN_OK

    def run(self, arg):
        pass

    def term(self):
        pass

def PLUGIN_ENTRY():
    return ida_plugin_container()

注意事项 #

• 确保使用IDA 7.0及以上版本，更早版本可能不支持ANSI颜色转义码
• 若要修改背景色，可将转义码改为\x1B[48;2;R;G;Bm
• 调整RGB分量值即可切换为其他自定义颜色

内容的提问来源于stack exchange，提问作者Roman Toasov