src/lib/ldblib.c

   1 /*
   2 ** $Id: ldblib.c,v 1.5 1999/03/04 21:17:26 roberto Exp $
   3 ** Interface from Lua to its debug API
   4 ** See Copyright Notice in lua.h
   5 */
   6
   7
   8 #include <stdlib.h>
   9 #include <string.h>
  10
  11 #include "lauxlib.h"
  12 #include "lua.h"
  13 #include "luadebug.h"
  14 #include "lualib.h"
  15
  16
  17
  18 static void settabss (lua_Object t, char *i, char *v) {
  19   lua_pushobject(t);
  20   lua_pushstring(i);
  21   lua_pushstring(v);
  22   lua_settable();
  23 }
  24
  25
  26 static void settabsi (lua_Object t, char *i, int v) {
  27   lua_pushobject(t);
  28   lua_pushstring(i);
  29   lua_pushnumber(v);
  30   lua_settable();
  31 }
  32
  33
  34 static lua_Object getfuncinfo (lua_Object func) {
  35   lua_Object result = lua_createtable();
  36   char *str;
  37   int line;
  38   lua_funcinfo(func, &str, &line);
  39   if (line == -1)  /* C function? */
  40     settabss(result, "kind", "C");
  41   else if (line == 0) {  /* "main"? */
  42       settabss(result, "kind", "chunk");
  43       settabss(result, "source", str);
  44     }
  45   else {  /* Lua function */
  46     settabss(result, "kind", "Lua");
  47     settabsi(result, "def_line", line);
  48     settabss(result, "source", str);
  49   }
  50   if (line != 0) {  /* is it not a "main"? */
  51     char *kind = lua_getobjname(func, &str);
  52     if (*kind) {
  53       settabss(result, "name", str);
  54       settabss(result, "where", kind);
  55     }
  56   }
  57   return result;
  58 }
  59
  60
  61 static void getstack (void) {
  62   lua_Object func = lua_stackedfunction(luaL_check_int(1));
  63   if (func == LUA_NOOBJECT)  /* level out of range? */
  64     return;
  65   else {
  66     lua_Object result = getfuncinfo(func);
  67     int currline = lua_currentline(func);
  68     if (currline > 0)
  69       settabsi(result, "current", currline);
  70     lua_pushobject(result);
  71     lua_pushstring("func");
  72     lua_pushobject(func);
  73     lua_settable();  /* result.func = func */
  74     lua_pushobject(result);
  75   }
  76 }
  77
  78
  79 static void funcinfo (void) {
  80   lua_pushobject(getfuncinfo(luaL_functionarg(1)));
  81 }
  82
  83
  84 static int findlocal (lua_Object func, int arg) {
  85   lua_Object v = lua_getparam(arg);
  86   if (lua_isnumber(v))
  87     return (int)lua_getnumber(v);
  88   else {
  89     char *name = luaL_check_string(arg);
  90     int i = 0;
  91     int result = -1;
  92     char *vname;
  93     while (lua_getlocal(func, ++i, &vname) != LUA_NOOBJECT) {
  94       if (strcmp(name, vname) == 0)
  95         result = i;  /* keep looping to get the last var with this name */
  96     }
  97     if (result == -1)
  98       luaL_verror("no local variable `%.50s' at given level", name);
  99     return result;
 100   }
 101 }
 102
 103
 104 static void getlocal (void) {
 105   lua_Object func = lua_stackedfunction(luaL_check_int(1));
 106   lua_Object val;
 107   char *name;
 108   if (func == LUA_NOOBJECT)  /* level out of range? */
 109     return;  /* return nil */
 110   else if (lua_getparam(2) != LUA_NOOBJECT) {  /* 2nd argument? */
 111     if ((val = lua_getlocal(func, findlocal(func, 2), &name)) != LUA_NOOBJECT) {
 112       lua_pushobject(val);
 113       lua_pushstring(name);
 114     }
 115     /* else return nil */
 116   }
 117   else {  /* collect all locals in a table */
 118     lua_Object result = lua_createtable();
 119     int i;
 120     for (i=1; ;i++) {
 121       if ((val = lua_getlocal(func, i, &name)) == LUA_NOOBJECT)
 122         break;
 123       lua_pushobject(result);
 124       lua_pushstring(name);
 125       lua_pushobject(val);
 126       lua_settable();  /* result[name] = value */
 127     }
 128     lua_pushobject(result);
 129   }
 130 }
 131
 132
 133 static void setlocal (void) {
 134   lua_Object func = lua_stackedfunction(luaL_check_int(1));
 135   int numvar;
 136   luaL_arg_check(func != LUA_NOOBJECT, 1, "level out of range");
 137   numvar = findlocal(func, 2);
 138   lua_pushobject(luaL_nonnullarg(3));
 139   if (!lua_setlocal(func, numvar))
 140     lua_error("no such local variable");
 141 }
 142
 143
 144
 145 static int linehook = -1;  /* Lua reference to line hook function */
 146 static int callhook = -1;  /* Lua reference to call hook function */
 147
 148
 149 static void dohook (int ref) {
 150   lua_LHFunction oldlinehook = lua_setlinehook(NULL);
 151   lua_CHFunction oldcallhook = lua_setcallhook(NULL);
 152   lua_callfunction(lua_getref(ref));
 153   lua_setlinehook(oldlinehook);
 154   lua_setcallhook(oldcallhook);
 155 }
 156
 157
 158 static void linef (int line) {
 159   lua_pushnumber(line);
 160   dohook(linehook);
 161 }
 162
 163
 164 static void callf (lua_Function func, char *file, int line) {
 165   if (func != LUA_NOOBJECT) {
 166     lua_pushobject(func);
 167     lua_pushstring(file);
 168     lua_pushnumber(line);
 169   }
 170   dohook(callhook);
 171 }
 172
 173
 174 static void setcallhook (void) {
 175   lua_Object f = lua_getparam(1);
 176   lua_unref(callhook);
 177   if (f == LUA_NOOBJECT) {
 178     callhook = -1;
 179     lua_setcallhook(NULL);
 180   }
 181   else {
 182     lua_pushobject(f);
 183     callhook = lua_ref(1);
 184     lua_setcallhook(callf);
 185   }
 186 }
 187
 188
 189 static void setlinehook (void) {
 190   lua_Object f = lua_getparam(1);
 191   lua_unref(linehook);
 192   if (f == LUA_NOOBJECT) {
 193     linehook = -1;
 194     lua_setlinehook(NULL);
 195   }
 196   else {
 197     lua_pushobject(f);
 198     linehook = lua_ref(1);
 199     lua_setlinehook(linef);
 200   }
 201 }
 202
 203
 204 static struct luaL_reg dblib[] = {
 205   {"funcinfo", funcinfo},
 206   {"getlocal", getlocal},
 207   {"getstack", getstack},
 208   {"setcallhook", setcallhook},
 209   {"setlinehook", setlinehook},
 210   {"setlocal", setlocal}
 211 };
 212
 213
 214 void lua_dblibopen (void) {
 215   luaL_openlib(dblib, (sizeof(dblib)/sizeof(dblib[0])));
 216 }
 217