LLDB通过Python添加断点回调时触发KeyError问题求助

我完全理解你现在碰到的麻烦：明明已经在write系统调用的回调里成功给上一帧地址添加了断点，却触发了KeyError: 'lldb_autogen_python_bp_callback_func__1'的错误，断点明明在正确地址，但就是跑不起来。我来帮你拆解下问题原因和解决办法。

先理清楚你的需求：每次触发write系统调用断点时，自动给调用栈的上一帧设置新断点，并绑定bp_callback函数打印提示，但LLDB内部生成的回调包装函数找不到对应条目，导致了这个KeyError。

问题根源 #

LLDB处理Python脚本回调时，会自动生成类似lldb_autogen_python_bp_callback_func__X的包装函数，这些函数依赖当前上下文的internal_dict来查找你的自定义回调。但在断点回调内部动态添加新断点并设置回调时，当前的internal_dict是syscall_callback的局部上下文，而非模块的全局上下文，导致LLDB找不到自动生成的包装函数对应的条目，最终抛出KeyError。

解决方案 #

我们可以绕开LLDB自动生成回调包装函数的逻辑，直接给新断点添加命令行脚本调用，确保直接访问到模块里的bp_callback函数。修改你的trace_mvp.py代码如下：

import lldb

def syscall_callback(frame: lldb.SBFrame, bp_loc, internal_dict):
    print("[+] Entering syscall_callback")
    thread = frame.GetThread()

    # 打印调用栈用于调试
    frames = thread.GetThreadFrames()
    for f in frames:
        print(f)

    # 做调用栈边界检查，避免索引越界
    if len(frames) < 2:
        print("[-] 调用栈帧数不足，无法获取上一帧")
        return
    
    # 获取上一帧的地址
    next_frame_pc = frames[1].GetPC()
   
    print("[+] 准备在地址添加断点: ", hex(next_frame_pc))
    target = thread.GetProcess().GetTarget()
    bp = target.BreakpointCreateByAddress(next_frame_pc)
    
    # 替换SetScriptCallbackFunction，改用命令行脚本直接调用回调函数
    bp_cmd = """
script import trace_mvp
trace_mvp.bp_callback(frame, bp_loc, {})
"""
    # 清理命令行脚本的空行，避免LLDB报错
    cleaned_cmds = [line.strip() for line in bp_cmd.splitlines() if line.strip()]
    bp.SetCommandLineCommands(cleaned_cmds)
    
    print(bp)


def bp_callback(frame: lldb.SBFrame, bp_loc, internal_dict):
    print("[+] Hello from bp_callback")

def __lldb_init_module(debugger, internal_dict):
    debugger.HandleCommand('breakpoint set -n write')
    debugger.HandleCommand('breakpoint command add -s python -F trace_mvp.syscall_callback')
    # 建议移除这里的自动run，手动在LLDB里输入run启动程序，避免模块加载时机冲突
    # debugger.HandleCommand('run')  

额外注意事项 #

1. 调用栈边界检查：添加了len(frames) < 2的判断，避免当调用栈只有一帧时出现索引越界错误。
2. 移除自动run：建议注释掉__lldb_init_module里的run命令，手动在LLDB中输入run启动程序，减少模块加载和程序启动的时机冲突问题。
3. 显式导入模块：在命令行脚本里显式import trace_mvp，确保回调触发时模块能被正确访问。

你可以试试修改后的代码，应该能解决KeyError的问题，同时实现你想要的功能。

备注：内容来源于stack exchange，提问作者Billbucket