]> git.zerfleddert.de Git - proxmark3-svn/blob - liblua/ldblib.c
fix length test
[proxmark3-svn] / 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
Impressum, Datenschutz