ba8b5c17 |
1 | /* |
2 | ** $Id: ldblib.c,v 1.132 2012/01/19 20:14:44 roberto Exp $ |
3 | ** Interface from Lua to its debug API |
4 | ** See Copyright Notice in lua.h |
5 | */ |
6 | |
7 | |
8 | #include <stdio.h> |
9 | #include <stdlib.h> |
10 | #include <string.h> |
11 | |
12 | #define ldblib_c |
13 | #define LUA_LIB |
14 | |
15 | #include "lua.h" |
16 | |
17 | #include "lauxlib.h" |
18 | #include "lualib.h" |
19 | |
20 | |
21 | #define HOOKKEY "_HKEY" |
22 | |
23 | |
24 | |
25 | static int db_getregistry (lua_State *L) { |
26 | lua_pushvalue(L, LUA_REGISTRYINDEX); |
27 | return 1; |
28 | } |
29 | |
30 | |
31 | static int db_getmetatable (lua_State *L) { |
32 | luaL_checkany(L, 1); |
33 | if (!lua_getmetatable(L, 1)) { |
34 | lua_pushnil(L); /* no metatable */ |
35 | } |
36 | return 1; |
37 | } |
38 | |
39 | |
40 | static int db_setmetatable (lua_State *L) { |
41 | int t = lua_type(L, 2); |
42 | luaL_argcheck(L, t == LUA_TNIL || t == LUA_TTABLE, 2, |
43 | "nil or table expected"); |
44 | lua_settop(L, 2); |
45 | lua_setmetatable(L, 1); |
46 | return 1; /* return 1st argument */ |
47 | } |
48 | |
49 | |
50 | static int db_getuservalue (lua_State *L) { |
51 | if (lua_type(L, 1) != LUA_TUSERDATA) |
52 | lua_pushnil(L); |
53 | else |
54 | lua_getuservalue(L, 1); |
55 | return 1; |
56 | } |
57 | |
58 | |
59 | static int db_setuservalue (lua_State *L) { |
60 | if (lua_type(L, 1) == LUA_TLIGHTUSERDATA) |
61 | luaL_argerror(L, 1, "full userdata expected, got light userdata"); |
62 | luaL_checktype(L, 1, LUA_TUSERDATA); |
63 | if (!lua_isnoneornil(L, 2)) |
64 | luaL_checktype(L, 2, LUA_TTABLE); |
65 | lua_settop(L, 2); |
66 | lua_setuservalue(L, 1); |
67 | return 1; |
68 | } |
69 | |
70 | |
71 | static void settabss (lua_State *L, const char *i, const char *v) { |
72 | lua_pushstring(L, v); |
73 | lua_setfield(L, -2, i); |
74 | } |
75 | |
76 | |
77 | static void settabsi (lua_State *L, const char *i, int v) { |
78 | lua_pushinteger(L, v); |
79 | lua_setfield(L, -2, i); |
80 | } |
81 | |
82 | |
83 | static void settabsb (lua_State *L, const char *i, int v) { |
84 | lua_pushboolean(L, v); |
85 | lua_setfield(L, -2, i); |
86 | } |
87 | |
88 | |
89 | static lua_State *getthread (lua_State *L, int *arg) { |
90 | if (lua_isthread(L, 1)) { |
91 | *arg = 1; |
92 | return lua_tothread(L, 1); |
93 | } |
94 | else { |
95 | *arg = 0; |
96 | return L; |
97 | } |
98 | } |
99 | |
100 | |
101 | static void treatstackoption (lua_State *L, lua_State *L1, const char *fname) { |
102 | if (L == L1) { |
103 | lua_pushvalue(L, -2); |
104 | lua_remove(L, -3); |
105 | } |
106 | else |
107 | lua_xmove(L1, L, 1); |
108 | lua_setfield(L, -2, fname); |
109 | } |
110 | |
111 | |
112 | static int db_getinfo (lua_State *L) { |
113 | lua_Debug ar; |
114 | int arg; |
115 | lua_State *L1 = getthread(L, &arg); |
116 | const char *options = luaL_optstring(L, arg+2, "flnStu"); |
117 | if (lua_isnumber(L, arg+1)) { |
118 | if (!lua_getstack(L1, (int)lua_tointeger(L, arg+1), &ar)) { |
119 | lua_pushnil(L); /* level out of range */ |
120 | return 1; |
121 | } |
122 | } |
123 | else if (lua_isfunction(L, arg+1)) { |
124 | lua_pushfstring(L, ">%s", options); |
125 | options = lua_tostring(L, -1); |
126 | lua_pushvalue(L, arg+1); |
127 | lua_xmove(L, L1, 1); |
128 | } |
129 | else |
130 | return luaL_argerror(L, arg+1, "function or level expected"); |
131 | if (!lua_getinfo(L1, options, &ar)) |
132 | return luaL_argerror(L, arg+2, "invalid option"); |
133 | lua_createtable(L, 0, 2); |
134 | if (strchr(options, 'S')) { |
135 | settabss(L, "source", ar.source); |
136 | settabss(L, "short_src", ar.short_src); |
137 | settabsi(L, "linedefined", ar.linedefined); |
138 | settabsi(L, "lastlinedefined", ar.lastlinedefined); |
139 | settabss(L, "what", ar.what); |
140 | } |
141 | if (strchr(options, 'l')) |
142 | settabsi(L, "currentline", ar.currentline); |
143 | if (strchr(options, 'u')) { |
144 | settabsi(L, "nups", ar.nups); |
145 | settabsi(L, "nparams", ar.nparams); |
146 | settabsb(L, "isvararg", ar.isvararg); |
147 | } |
148 | if (strchr(options, 'n')) { |
149 | settabss(L, "name", ar.name); |
150 | settabss(L, "namewhat", ar.namewhat); |
151 | } |
152 | if (strchr(options, 't')) |
153 | settabsb(L, "istailcall", ar.istailcall); |
154 | if (strchr(options, 'L')) |
155 | treatstackoption(L, L1, "activelines"); |
156 | if (strchr(options, 'f')) |
157 | treatstackoption(L, L1, "func"); |
158 | return 1; /* return table */ |
159 | } |
160 | |
161 | |
162 | static int db_getlocal (lua_State *L) { |
163 | int arg; |
164 | lua_State *L1 = getthread(L, &arg); |
165 | lua_Debug ar; |
166 | const char *name; |
167 | int nvar = luaL_checkint(L, arg+2); /* local-variable index */ |
168 | if (lua_isfunction(L, arg + 1)) { /* function argument? */ |
169 | lua_pushvalue(L, arg + 1); /* push function */ |
170 | lua_pushstring(L, lua_getlocal(L, NULL, nvar)); /* push local name */ |
171 | return 1; |
172 | } |
173 | else { /* stack-level argument */ |
174 | if (!lua_getstack(L1, luaL_checkint(L, arg+1), &ar)) /* out of range? */ |
175 | return luaL_argerror(L, arg+1, "level out of range"); |
176 | name = lua_getlocal(L1, &ar, nvar); |
177 | if (name) { |
178 | lua_xmove(L1, L, 1); /* push local value */ |
179 | lua_pushstring(L, name); /* push name */ |
180 | lua_pushvalue(L, -2); /* re-order */ |
181 | return 2; |
182 | } |
183 | else { |
184 | lua_pushnil(L); /* no name (nor value) */ |
185 | return 1; |
186 | } |
187 | } |
188 | } |
189 | |
190 | |
191 | static int db_setlocal (lua_State *L) { |
192 | int arg; |
193 | lua_State *L1 = getthread(L, &arg); |
194 | lua_Debug ar; |
195 | if (!lua_getstack(L1, luaL_checkint(L, arg+1), &ar)) /* out of range? */ |
196 | return luaL_argerror(L, arg+1, "level out of range"); |
197 | luaL_checkany(L, arg+3); |
198 | lua_settop(L, arg+3); |
199 | lua_xmove(L, L1, 1); |
200 | lua_pushstring(L, lua_setlocal(L1, &ar, luaL_checkint(L, arg+2))); |
201 | return 1; |
202 | } |
203 | |
204 | |
205 | static int auxupvalue (lua_State *L, int get) { |
206 | const char *name; |
207 | int n = luaL_checkint(L, 2); |
208 | luaL_checktype(L, 1, LUA_TFUNCTION); |
209 | name = get ? lua_getupvalue(L, 1, n) : lua_setupvalue(L, 1, n); |
210 | if (name == NULL) return 0; |
211 | lua_pushstring(L, name); |
212 | lua_insert(L, -(get+1)); |
213 | return get + 1; |
214 | } |
215 | |
216 | |
217 | static int db_getupvalue (lua_State *L) { |
218 | return auxupvalue(L, 1); |
219 | } |
220 | |
221 | |
222 | static int db_setupvalue (lua_State *L) { |
223 | luaL_checkany(L, 3); |
224 | return auxupvalue(L, 0); |
225 | } |
226 | |
227 | |
228 | static int checkupval (lua_State *L, int argf, int argnup) { |
229 | lua_Debug ar; |
230 | int nup = luaL_checkint(L, argnup); |
231 | luaL_checktype(L, argf, LUA_TFUNCTION); |
232 | lua_pushvalue(L, argf); |
233 | lua_getinfo(L, ">u", &ar); |
234 | luaL_argcheck(L, 1 <= nup && nup <= ar.nups, argnup, "invalid upvalue index"); |
235 | return nup; |
236 | } |
237 | |
238 | |
239 | static int db_upvalueid (lua_State *L) { |
240 | int n = checkupval(L, 1, 2); |
241 | lua_pushlightuserdata(L, lua_upvalueid(L, 1, n)); |
242 | return 1; |
243 | } |
244 | |
245 | |
246 | static int db_upvaluejoin (lua_State *L) { |
247 | int n1 = checkupval(L, 1, 2); |
248 | int n2 = checkupval(L, 3, 4); |
249 | luaL_argcheck(L, !lua_iscfunction(L, 1), 1, "Lua function expected"); |
250 | luaL_argcheck(L, !lua_iscfunction(L, 3), 3, "Lua function expected"); |
251 | lua_upvaluejoin(L, 1, n1, 3, n2); |
252 | return 0; |
253 | } |
254 | |
255 | |
256 | #define gethooktable(L) luaL_getsubtable(L, LUA_REGISTRYINDEX, HOOKKEY) |
257 | |
258 | |
259 | static void hookf (lua_State *L, lua_Debug *ar) { |
260 | static const char *const hooknames[] = |
261 | {"call", "return", "line", "count", "tail call"}; |
262 | gethooktable(L); |
263 | lua_pushthread(L); |
264 | lua_rawget(L, -2); |
265 | if (lua_isfunction(L, -1)) { |
266 | lua_pushstring(L, hooknames[(int)ar->event]); |
267 | if (ar->currentline >= 0) |
268 | lua_pushinteger(L, ar->currentline); |
269 | else lua_pushnil(L); |
270 | lua_assert(lua_getinfo(L, "lS", ar)); |
271 | lua_call(L, 2, 0); |
272 | } |
273 | } |
274 | |
275 | |
276 | static int makemask (const char *smask, int count) { |
277 | int mask = 0; |
278 | if (strchr(smask, 'c')) mask |= LUA_MASKCALL; |
279 | if (strchr(smask, 'r')) mask |= LUA_MASKRET; |
280 | if (strchr(smask, 'l')) mask |= LUA_MASKLINE; |
281 | if (count > 0) mask |= LUA_MASKCOUNT; |
282 | return mask; |
283 | } |
284 | |
285 | |
286 | static char *unmakemask (int mask, char *smask) { |
287 | int i = 0; |
288 | if (mask & LUA_MASKCALL) smask[i++] = 'c'; |
289 | if (mask & LUA_MASKRET) smask[i++] = 'r'; |
290 | if (mask & LUA_MASKLINE) smask[i++] = 'l'; |
291 | smask[i] = '\0'; |
292 | return smask; |
293 | } |
294 | |
295 | |
296 | static int db_sethook (lua_State *L) { |
297 | int arg, mask, count; |
298 | lua_Hook func; |
299 | lua_State *L1 = getthread(L, &arg); |
300 | if (lua_isnoneornil(L, arg+1)) { |
301 | lua_settop(L, arg+1); |
302 | func = NULL; mask = 0; count = 0; /* turn off hooks */ |
303 | } |
304 | else { |
305 | const char *smask = luaL_checkstring(L, arg+2); |
306 | luaL_checktype(L, arg+1, LUA_TFUNCTION); |
307 | count = luaL_optint(L, arg+3, 0); |
308 | func = hookf; mask = makemask(smask, count); |
309 | } |
310 | if (gethooktable(L) == 0) { /* creating hook table? */ |
311 | lua_pushstring(L, "k"); |
312 | lua_setfield(L, -2, "__mode"); /** hooktable.__mode = "k" */ |
313 | lua_pushvalue(L, -1); |
314 | lua_setmetatable(L, -2); /* setmetatable(hooktable) = hooktable */ |
315 | } |
316 | lua_pushthread(L1); lua_xmove(L1, L, 1); |
317 | lua_pushvalue(L, arg+1); |
318 | lua_rawset(L, -3); /* set new hook */ |
319 | lua_sethook(L1, func, mask, count); /* set hooks */ |
320 | return 0; |
321 | } |
322 | |
323 | |
324 | static int db_gethook (lua_State *L) { |
325 | int arg; |
326 | lua_State *L1 = getthread(L, &arg); |
327 | char buff[5]; |
328 | int mask = lua_gethookmask(L1); |
329 | lua_Hook hook = lua_gethook(L1); |
330 | if (hook != NULL && hook != hookf) /* external hook? */ |
331 | lua_pushliteral(L, "external hook"); |
332 | else { |
333 | gethooktable(L); |
334 | lua_pushthread(L1); lua_xmove(L1, L, 1); |
335 | lua_rawget(L, -2); /* get hook */ |
336 | lua_remove(L, -2); /* remove hook table */ |
337 | } |
338 | lua_pushstring(L, unmakemask(mask, buff)); |
339 | lua_pushinteger(L, lua_gethookcount(L1)); |
340 | return 3; |
341 | } |
342 | |
343 | |
344 | static int db_debug (lua_State *L) { |
345 | for (;;) { |
346 | char buffer[250]; |
347 | luai_writestringerror("%s", "lua_debug> "); |
348 | if (fgets(buffer, sizeof(buffer), stdin) == 0 || |
349 | strcmp(buffer, "cont\n") == 0) |
350 | return 0; |
351 | if (luaL_loadbuffer(L, buffer, strlen(buffer), "=(debug command)") || |
352 | lua_pcall(L, 0, 0, 0)) |
353 | luai_writestringerror("%s\n", lua_tostring(L, -1)); |
354 | lua_settop(L, 0); /* remove eventual returns */ |
355 | } |
356 | } |
357 | |
358 | |
359 | static int db_traceback (lua_State *L) { |
360 | int arg; |
361 | lua_State *L1 = getthread(L, &arg); |
362 | const char *msg = lua_tostring(L, arg + 1); |
363 | if (msg == NULL && !lua_isnoneornil(L, arg + 1)) /* non-string 'msg'? */ |
364 | lua_pushvalue(L, arg + 1); /* return it untouched */ |
365 | else { |
366 | int level = luaL_optint(L, arg + 2, (L == L1) ? 1 : 0); |
367 | luaL_traceback(L, L1, msg, level); |
368 | } |
369 | return 1; |
370 | } |
371 | |
372 | |
373 | static const luaL_Reg dblib[] = { |
374 | {"debug", db_debug}, |
375 | {"getuservalue", db_getuservalue}, |
376 | {"gethook", db_gethook}, |
377 | {"getinfo", db_getinfo}, |
378 | {"getlocal", db_getlocal}, |
379 | {"getregistry", db_getregistry}, |
380 | {"getmetatable", db_getmetatable}, |
381 | {"getupvalue", db_getupvalue}, |
382 | {"upvaluejoin", db_upvaluejoin}, |
383 | {"upvalueid", db_upvalueid}, |
384 | {"setuservalue", db_setuservalue}, |
385 | {"sethook", db_sethook}, |
386 | {"setlocal", db_setlocal}, |
387 | {"setmetatable", db_setmetatable}, |
388 | {"setupvalue", db_setupvalue}, |
389 | {"traceback", db_traceback}, |
390 | {NULL, NULL} |
391 | }; |
392 | |
393 | |
394 | LUAMOD_API int luaopen_debug (lua_State *L) { |
395 | luaL_newlib(L, dblib); |
396 | return 1; |
397 | } |
398 | |