说到Lua代码调试，最常用的方法应该就是加一堆print进行打印。print大法虽好，但其缺点也是显而易见的。比如效率低下，需要修改原有函数内部代码，在每个需要的地方添加print语句，运行一次只能获取一次信息，下次换个地方又得重新添加print语句。而且有时候，事先并不知道该去哪打印、或者打印什么内容，需要通过运行中获取的信息才能确定。
当print大法无法满足我们的需求时，就需要类似断点调试这样更高级的调试功能。本文将从零开始编写一个Lua调试器，实现类似gdb的断点调试功能。
本文代码已开源至Github，欢迎watch/star😘。
本博客已迁移至CatBro's Blog，那里是我自己搭建的个人博客，欢迎关注。
定义模块及接口
首先，我们来定义模块及接口，创建一个名为luadebug.lua的模块，该模块是基于标准库中的debug库。为了实现最基本的断点调试功能，我们的模块提供了两个接口setbreakpoint和removebreakpoint，分别用于设置断点和删除断点。断点信息通过一个函数和一个行号指定，返回断点的id。后续可以通过这个id来删除相应断点。
#!/usr/bin/env lua
local debug = require "debug"
-- 省略...
local function setbreakpoint(func, line)
-- 省略...
end
local function removebreakpoint(id)
-- 省略...
end
return {
setbreakpoint = setbreakpoint,
removebreakpoint = removebreakpoint,
}
维护状态的数据结构
接着，来定义维护状态的数据结构，表status维护了所有断点相关信息，其中的bpnum元素表示当前总共有多少断点，bpid表示当前的断点id，这个值是不断递增的，bptable则是保存所有断点信息的表。表bptable的键是断点的id，值也是一个表，保存了断点的所在的函数和行号。
-- 省略...
-- 记录断点状态
local status = {}
status.bpnum = 0 -- 当前总断点数
status.bpid = 0 -- 当前断点id
status.bptable = {} -- 保存断点信息的表
-- 省略...
设置断点接口
接下来来定义我们的setbreakpoint接口。设置断点时，首先检查参数有效性，再更新断点id和断点数，然后将参数中传入的函数func和行号line保存到表bptable中下一个断点id的位置。如果只有一个断点（从无到有），那么还需要调用debug.sethook设置钩子。这是实现断点调试的核心函数之一，它使得我们有机会停在断点处。因为是最小实现，简单起见这里只设置了line事件。
-- 设置断点
local function setbreakpoint(func, line)
if type(func) ~= "function" or type(line) ~= "number" then
return nil --> nil表示无效断点
end
status.bpid = status.bpid + 1
status.bpnum = status.bpnum + 1
status.bptable[status.bpid] = {func = func, line = line}
if status.bpnum == 1 then -- 第一个断点
debug.sethook(linehook, "l") -- 设置钩子
end
return status.bpid --> 返回断点id
end
钩子函数
在钩子函数中，通过debug.getinfo获取到闭包信息，注意这里的层次为2，因为debug.getinfo()函数本身的层次是0，钩子函数层次是1，断点所在的函数层次即为2。然后遍历断点表，与获取的闭包信息进行比较，如果函数和行号都匹配，说明命中断点。我们打印一行提示信息，然后调用debug.debug()进入交互调试模式，debug.debug是实现断点调试的另一个核心函数，它使得我们可以在断点处输入任意代码执行。交互调试模式一直持续，直到用户输入cont为止。
-- 钩子函数
local function linehook (event, line)
local info = debug.getinfo(2, "nfS")
for _, v in pairs(status.bptable) do
if v.func == info.func and v.line == line then
local prompt = string.format("(%s)%s %s:%d\n",
info.namewhat, info.name, info.short_src, line)
io.write(prompt)
debug.debug()
end
end
end
删除断点接口
删除断点比较简单，首先检查id参数是否有效，如果无效直接返回，如果有效则将断点表中相应id位置的值置为nil即可，然后更新断点数，如果已经没有断点了，则清除钩子。
-- 删除断点
local function removebreakpoint(id)
if status.bptable[id] == nil then
return
end
status.bptable[id] = nil
status.bpnum = status.bpnum - 1
if status.bpnum == 0 then
debug.sethook() -- 清除钩子
end
end
至此我们的模块就编写好了，下面对这个模块进行测试。
测试脚本
我们编写一个如下的测试脚本test.lua，定义了两个函数foo和bar，然后分别在两个函数中设置了一个断点（注意：注释和空行不是有效的断点行），然后多次调用函数并先后删除断点：
local ldb = require "luadebug"
local setbp = ldb.setbreakpoint
local rmbp = ldb.removebreakpoint
g = 1
local u = 2
local function foo (n)
local a = 3
a = a + 1
u = u + 1
g = g + 1
end
local function bar (n)
n = n + 1
end
local id1 = setbp(foo, 11) -- 设置断点1
local id2 = setbp(bar, 16) -- 设置断点2
foo(10)
bar(10)
rmbp(id1) -- 删除断点1
foo(20)
bar(20)
rmbp(id2) -- 删除断点2
foo(30)
bar(30)
测试验证
然后我们运行测试脚本，可以看到程序停在了foo函数的断点1处。
$ lua test.lua
(local)foo test.lua:11
lua_debug>
我们可以在这里打印调用栈信息
$ lua test.lua
(local)foo test.lua:11
lua_debug> print(debug.traceback())
stack traceback:
(debug command):1: in main chunk
[C]: in function 'debug.debug'
./luadebug.lua:20: in hook '?'
test.lua:11: in local 'foo'
test.lua:22: in main chunk
[C]: in ?
lua_debug>
可以看到foo函数在第4层（第1层是执行我们调试命令的main chunk，第2层是debug.debug函数，第3层是hook函数）。我们打印foo函数中第一个局部变量（即固定参数n）的值
lua_debug> print(debug.getlocal(4, 1))
n 10
lua_debug>
然后打印第二个局部变量（即a）的值
lua_debug> print(debug.getlocal(4, 2))
a 4
lua_debug>
然后我们输入cont继续代码的执行，碰到了bar函数的断点2
lua_debug> cont
(local)bar test.lua:16
lua_debug>
我们打印bar函数的参数n的值，可以看到也是10
lua_debug> print(debug.getlocal(4, 1))
n 10
lua_debug>
然后我们输入cont继续执行代码，因为断点1已经被移除，所以再次停在了bar函数的断点2处
lua_debug> cont
(local)bar test.lua:16
lua_debug>
我们再来打印下参数n的值，此时参数n的值是20
lua_debug> print(debug.getlocal(4, 1))
n 20
lua_debug>
我们再次输入cont，因为断点2也被移除了，所以第三次调用foo函数和bar函数就没有再碰到断点，程序运行结束
lua_debug> cont
$
这样一个最简单的Lua断点调试器就完成了。虽然还比较简陋，但是已经能够应付一些简单的调试了。🎉
