git.zerfleddert.de Git - proxmark3-svn/blob - liblua/ldblib.c

   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