OLD | NEW |
(Empty) | |
| 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 |
OLD | NEW |