ldebug.c
上传用户:yisoukefu
上传日期:2020-08-09
资源大小:39506k
文件大小:16k
源码类别:

其他游戏

开发平台:

Visual C++

  1. /*
  2. ** $Id: ldebug.c,v 2.29 2005/12/22 16:19:56 roberto Exp $
  3. ** Debug Interface
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <stdarg.h>
  7. #include <stddef.h>
  8. #include <string.h>
  9. #define ldebug_c
  10. #define LUA_CORE
  11. #include "lua.h"
  12. #include "lapi.h"
  13. #include "lcode.h"
  14. #include "ldebug.h"
  15. #include "ldo.h"
  16. #include "lfunc.h"
  17. #include "lobject.h"
  18. #include "lopcodes.h"
  19. #include "lstate.h"
  20. #include "lstring.h"
  21. #include "ltable.h"
  22. #include "ltm.h"
  23. #include "lvm.h"
  24. static const char *getfuncname (lua_State *L, CallInfo *ci, const char **name);
  25. static int currentpc (lua_State *L, CallInfo *ci) {
  26.   if (!isLua(ci)) return -1;  /* function is not a Lua function? */
  27.   if (ci == L->ci)
  28.     ci->savedpc = L->savedpc;
  29.   return pcRel(ci->savedpc, ci_func(ci)->l.p);
  30. }
  31. static int currentline (lua_State *L, CallInfo *ci) {
  32.   int pc = currentpc(L, ci);
  33.   if (pc < 0)
  34.     return -1;  /* only active lua functions have current-line information */
  35.   else
  36.     return getline(ci_func(ci)->l.p, pc);
  37. }
  38. /*
  39. ** this function can be called asynchronous (e.g. during a signal)
  40. */
  41. LUA_API int lua_sethook (lua_State *L, lua_Hook func, int mask, int count) {
  42.   if (func == NULL || mask == 0) {  /* turn off hooks? */
  43.     mask = 0;
  44.     func = NULL;
  45.   }
  46.   L->hook = func;
  47.   L->basehookcount = count;
  48.   resethookcount(L);
  49.   L->hookmask = cast_byte(mask);
  50.   return 1;
  51. }
  52. LUA_API lua_Hook lua_gethook (lua_State *L) {
  53.   return L->hook;
  54. }
  55. LUA_API int lua_gethookmask (lua_State *L) {
  56.   return L->hookmask;
  57. }
  58. LUA_API int lua_gethookcount (lua_State *L) {
  59.   return L->basehookcount;
  60. }
  61. LUA_API int lua_getstack (lua_State *L, int level, lua_Debug *ar) {
  62.   int status;
  63.   CallInfo *ci;
  64.   lua_lock(L);
  65.   for (ci = L->ci; level > 0 && ci > L->base_ci; ci--) {
  66.     level--;
  67.     if (f_isLua(ci))  /* Lua function? */
  68.       level -= ci->tailcalls;  /* skip lost tail calls */
  69.   }
  70.   if (level == 0 && ci > L->base_ci) {  /* level found? */
  71.     status = 1;
  72.     ar->i_ci = cast_int(ci - L->base_ci);
  73.   }
  74.   else if (level < 0) {  /* level is of a lost tail call? */
  75.     status = 1;
  76.     ar->i_ci = 0;
  77.   }
  78.   else status = 0;  /* no such level */
  79.   lua_unlock(L);
  80.   return status;
  81. }
  82. static Proto *getluaproto (CallInfo *ci) {
  83.   return (isLua(ci) ? ci_func(ci)->l.p : NULL);
  84. }
  85. static const char *findlocal (lua_State *L, CallInfo *ci, int n) {
  86.   const char *name;
  87.   Proto *fp = getluaproto(ci);
  88.   if (fp && (name = luaF_getlocalname(fp, n, currentpc(L, ci))) != NULL)
  89.     return name;  /* is a local variable in a Lua function */
  90.   else {
  91.     StkId limit = (ci == L->ci) ? L->top : (ci+1)->func;
  92.     if (limit - ci->base >= n && n > 0)  /* is 'n' inside 'ci' stack? */
  93.       return "(*temporary)";
  94.     else
  95.       return NULL;
  96.   }
  97. }
  98. LUA_API const char *lua_getlocal (lua_State *L, const lua_Debug *ar, int n) {
  99.   CallInfo *ci = L->base_ci + ar->i_ci;
  100.   const char *name = findlocal(L, ci, n);
  101.   lua_lock(L);
  102.   if (name)
  103.       luaA_pushobject(L, ci->base + (n - 1));
  104.   lua_unlock(L);
  105.   return name;
  106. }
  107. LUA_API const char *lua_setlocal (lua_State *L, const lua_Debug *ar, int n) {
  108.   CallInfo *ci = L->base_ci + ar->i_ci;
  109.   const char *name = findlocal(L, ci, n);
  110.   lua_lock(L);
  111.   if (name)
  112.       setobjs2s(L, ci->base + (n - 1), L->top - 1);
  113.   L->top--;  /* pop value */
  114.   lua_unlock(L);
  115.   return name;
  116. }
  117. static void funcinfo (lua_Debug *ar, Closure *cl) {
  118.   if (cl->c.isC) {
  119.     ar->source = "=[C]";
  120.     ar->linedefined = -1;
  121.     ar->lastlinedefined = -1;
  122.     ar->what = "C";
  123.   }
  124.   else {
  125.     ar->source = getstr(cl->l.p->source);
  126.     ar->linedefined = cl->l.p->linedefined;
  127.     ar->lastlinedefined = cl->l.p->lastlinedefined;
  128.     ar->what = (ar->linedefined == 0) ? "main" : "Lua";
  129.   }
  130.   luaO_chunkid(ar->short_src, ar->source, LUA_IDSIZE);
  131. }
  132. static void info_tailcall (lua_Debug *ar) {
  133.   ar->name = ar->namewhat = "";
  134.   ar->what = "tail";
  135.   ar->lastlinedefined = ar->linedefined = ar->currentline = -1;
  136.   ar->source = "=(tail call)";
  137.   luaO_chunkid(ar->short_src, ar->source, LUA_IDSIZE);
  138.   ar->nups = 0;
  139. }
  140. static void collectvalidlines (lua_State *L, Closure *f) {
  141.   if (f == NULL || f->c.isC) {
  142.     setnilvalue(L->top);
  143.   }
  144.   else {
  145.     Table *t = luaH_new(L, 0, 0);
  146.     int *lineinfo = f->l.p->lineinfo;
  147.     int i;
  148.     for (i=0; i<f->l.p->sizelineinfo; i++)
  149.       setbvalue(luaH_setnum(L, t, lineinfo[i]), 1);
  150.     sethvalue(L, L->top, t); 
  151.   }
  152.   incr_top(L);
  153. }
  154. static int auxgetinfo (lua_State *L, const char *what, lua_Debug *ar,
  155.                     Closure *f, CallInfo *ci) {
  156.   int status = 1;
  157.   if (f == NULL) {
  158.     info_tailcall(ar);
  159.     return status;
  160.   }
  161.   for (; *what; what++) {
  162.     switch (*what) {
  163.       case 'S': {
  164.         funcinfo(ar, f);
  165.         break;
  166.       }
  167.       case 'l': {
  168.         ar->currentline = (ci) ? currentline(L, ci) : -1;
  169.         break;
  170.       }
  171.       case 'u': {
  172.         ar->nups = f->c.nupvalues;
  173.         break;
  174.       }
  175.       case 'n': {
  176.         ar->namewhat = (ci) ? getfuncname(L, ci, &ar->name) : NULL;
  177.         if (ar->namewhat == NULL) {
  178.           ar->namewhat = "";  /* not found */
  179.           ar->name = NULL;
  180.         }
  181.         break;
  182.       }
  183.       case 'L':
  184.       case 'f':  /* handled by lua_getinfo */
  185.         break;
  186.       default: status = 0;  /* invalid option */
  187.     }
  188.   }
  189.   return status;
  190. }
  191. LUA_API int lua_getinfo (lua_State *L, const char *what, lua_Debug *ar) {
  192.   int status;
  193.   Closure *f = NULL;
  194.   CallInfo *ci = NULL;
  195.   lua_lock(L);
  196.   if (*what == '>') {
  197.     StkId func = L->top - 1;
  198.     luai_apicheck(L, ttisfunction(func));
  199.     what++;  /* skip the '>' */
  200.     f = clvalue(func);
  201.     L->top--;  /* pop function */
  202.   }
  203.   else if (ar->i_ci != 0) {  /* no tail call? */
  204.     ci = L->base_ci + ar->i_ci;
  205.     lua_assert(ttisfunction(ci->func));
  206.     f = clvalue(ci->func);
  207.   }
  208.   status = auxgetinfo(L, what, ar, f, ci);
  209.   if (strchr(what, 'f')) {
  210.     if (f == NULL) setnilvalue(L->top);
  211.     else setclvalue(L, L->top, f);
  212.     incr_top(L);
  213.   }
  214.   if (strchr(what, 'L'))
  215.     collectvalidlines(L, f);
  216.   lua_unlock(L);
  217.   return status;
  218. }
  219. /*
  220. ** {======================================================
  221. ** Symbolic Execution and code checker
  222. ** =======================================================
  223. */
  224. #define check(x) if (!(x)) return 0;
  225. #define checkjump(pt,pc) check(0 <= pc && pc < pt->sizecode)
  226. #define checkreg(pt,reg) check((reg) < (pt)->maxstacksize)
  227. static int precheck (const Proto *pt) {
  228.   check(pt->maxstacksize <= MAXSTACK);
  229.   lua_assert(pt->numparams+(pt->is_vararg & VARARG_HASARG) <= pt->maxstacksize);
  230.   lua_assert(!(pt->is_vararg & VARARG_NEEDSARG) ||
  231.               (pt->is_vararg & VARARG_HASARG));
  232.   check(pt->sizeupvalues <= pt->nups);
  233.   check(pt->sizelineinfo == pt->sizecode || pt->sizelineinfo == 0);
  234.   check(GET_OPCODE(pt->code[pt->sizecode-1]) == OP_RETURN);
  235.   return 1;
  236. }
  237. #define checkopenop(pt,pc) luaG_checkopenop((pt)->code[(pc)+1])
  238. int luaG_checkopenop (Instruction i) {
  239.   switch (GET_OPCODE(i)) {
  240.     case OP_CALL:
  241.     case OP_TAILCALL:
  242.     case OP_RETURN:
  243.     case OP_SETLIST: {
  244.       check(GETARG_B(i) == 0);
  245.       return 1;
  246.     }
  247.     default: return 0;  /* invalid instruction after an open call */
  248.   }
  249. }
  250. static int checkArgMode (const Proto *pt, int r, enum OpArgMask mode) {
  251.   switch (mode) {
  252.     case OpArgN: check(r == 0); break;
  253.     case OpArgU: break;
  254.     case OpArgR: checkreg(pt, r); break;
  255.     case OpArgK:
  256.       check(ISK(r) ? INDEXK(r) < pt->sizek : r < pt->maxstacksize);
  257.       break;
  258.   }
  259.   return 1;
  260. }
  261. static Instruction symbexec (const Proto *pt, int lastpc, int reg) {
  262.   int pc;
  263.   int last;  /* stores position of last instruction that changed `reg' */
  264.   last = pt->sizecode-1;  /* points to final return (a `neutral' instruction) */
  265.   check(precheck(pt));
  266.   for (pc = 0; pc < lastpc; pc++) {
  267.     Instruction i = pt->code[pc];
  268.     OpCode op = GET_OPCODE(i);
  269.     int a = GETARG_A(i);
  270.     int b = 0;
  271.     int c = 0;
  272.     check(op < NUM_OPCODES);
  273.     checkreg(pt, a);
  274.     switch (getOpMode(op)) {
  275.       case iABC: {
  276.         b = GETARG_B(i);
  277.         c = GETARG_C(i);
  278.         check(checkArgMode(pt, b, getBMode(op)));
  279.         check(checkArgMode(pt, c, getCMode(op)));
  280.         break;
  281.       }
  282.       case iABx: {
  283.         b = GETARG_Bx(i);
  284.         if (getBMode(op) == OpArgK) check(b < pt->sizek);
  285.         break;
  286.       }
  287.       case iAsBx: {
  288.         b = GETARG_sBx(i);
  289.         if (getBMode(op) == OpArgR) {
  290.           int dest = pc+1+b;
  291.           check(0 <= dest && dest < pt->sizecode);
  292.           if (dest > 0) {
  293.             /* cannot jump to a setlist count */
  294.             Instruction d = pt->code[dest-1];
  295.             check(!(GET_OPCODE(d) == OP_SETLIST && GETARG_C(d) == 0));
  296.           }
  297.         }
  298.         break;
  299.       }
  300.     }
  301.     if (testAMode(op)) {
  302.       if (a == reg) last = pc;  /* change register `a' */
  303.     }
  304.     if (testTMode(op)) {
  305.       check(pc+2 < pt->sizecode);  /* check skip */
  306.       check(GET_OPCODE(pt->code[pc+1]) == OP_JMP);
  307.     }
  308.     switch (op) {
  309.       case OP_LOADBOOL: {
  310.         check(c == 0 || pc+2 < pt->sizecode);  /* check its jump */
  311.         break;
  312.       }
  313.       case OP_LOADNIL: {
  314.         if (a <= reg && reg <= b)
  315.           last = pc;  /* set registers from `a' to `b' */
  316.         break;
  317.       }
  318.       case OP_GETUPVAL:
  319.       case OP_SETUPVAL: {
  320.         check(b < pt->nups);
  321.         break;
  322.       }
  323.       case OP_GETGLOBAL:
  324.       case OP_SETGLOBAL: {
  325.         check(ttisstring(&pt->k[b]));
  326.         break;
  327.       }
  328.       case OP_SELF: {
  329.         checkreg(pt, a+1);
  330.         if (reg == a+1) last = pc;
  331.         break;
  332.       }
  333.       case OP_CONCAT: {
  334.         check(b < c);  /* at least two operands */
  335.         break;
  336.       }
  337.       case OP_TFORLOOP: {
  338.         check(c >= 1);  /* at least one result (control variable) */
  339.         checkreg(pt, a+2+c);  /* space for results */
  340.         if (reg >= a+2) last = pc;  /* affect all regs above its base */
  341.         break;
  342.       }
  343.       case OP_FORLOOP:
  344.       case OP_FORPREP:
  345.         checkreg(pt, a+3);
  346.         /* go through */
  347.       case OP_JMP: {
  348.         int dest = pc+1+b;
  349.         /* not full check and jump is forward and do not skip `lastpc'? */
  350.         if (reg != NO_REG && pc < dest && dest <= lastpc)
  351.           pc += b;  /* do the jump */
  352.         break;
  353.       }
  354.       case OP_CALL:
  355.       case OP_TAILCALL: {
  356.         if (b != 0) {
  357.           checkreg(pt, a+b-1);
  358.         }
  359.         c--;  /* c = num. returns */
  360.         if (c == LUA_MULTRET) {
  361.           check(checkopenop(pt, pc));
  362.         }
  363.         else if (c != 0)
  364.           checkreg(pt, a+c-1);
  365.         if (reg >= a) last = pc;  /* affect all registers above base */
  366.         break;
  367.       }
  368.       case OP_RETURN: {
  369.         b--;  /* b = num. returns */
  370.         if (b > 0) checkreg(pt, a+b-1);
  371.         break;
  372.       }
  373.       case OP_SETLIST: {
  374.         if (b > 0) checkreg(pt, a + b);
  375.         if (c == 0) pc++;
  376.         break;
  377.       }
  378.       case OP_CLOSURE: {
  379.         int nup;
  380.         check(b < pt->sizep);
  381.         nup = pt->p[b]->nups;
  382.         check(pc + nup < pt->sizecode);
  383.         for (; nup>0; nup--) {
  384.           OpCode op1 = GET_OPCODE(pt->code[pc+nup]);
  385.           check(op1 == OP_GETUPVAL || op1 == OP_MOVE);
  386.         }
  387.         break;
  388.       }
  389.       case OP_VARARG: {
  390.         check((pt->is_vararg & VARARG_ISVARARG) &&
  391.              !(pt->is_vararg & VARARG_NEEDSARG));
  392.         b--;
  393.         if (b == LUA_MULTRET) check(checkopenop(pt, pc));
  394.         checkreg(pt, a+b-1);
  395.         break;
  396.       }
  397.       default: break;
  398.     }
  399.   }
  400.   return pt->code[last];
  401. }
  402. #undef check
  403. #undef checkjump
  404. #undef checkreg
  405. /* }====================================================== */
  406. int luaG_checkcode (const Proto *pt) {
  407.   return (symbexec(pt, pt->sizecode, NO_REG) != 0);
  408. }
  409. static const char *kname (Proto *p, int c) {
  410.   if (ISK(c) && ttisstring(&p->k[INDEXK(c)]))
  411.     return svalue(&p->k[INDEXK(c)]);
  412.   else
  413.     return "?";
  414. }
  415. static const char *getobjname (lua_State *L, CallInfo *ci, int stackpos,
  416.                                const char **name) {
  417.   if (isLua(ci)) {  /* a Lua function? */
  418.     Proto *p = ci_func(ci)->l.p;
  419.     int pc = currentpc(L, ci);
  420.     Instruction i;
  421.     *name = luaF_getlocalname(p, stackpos+1, pc);
  422.     if (*name)  /* is a local? */
  423.       return "local";
  424.     i = symbexec(p, pc, stackpos);  /* try symbolic execution */
  425.     lua_assert(pc != -1);
  426.     switch (GET_OPCODE(i)) {
  427.       case OP_GETGLOBAL: {
  428.         int g = GETARG_Bx(i);  /* global index */
  429.         lua_assert(ttisstring(&p->k[g]));
  430.         *name = svalue(&p->k[g]);
  431.         return "global";
  432.       }
  433.       case OP_MOVE: {
  434.         int a = GETARG_A(i);
  435.         int b = GETARG_B(i);  /* move from `b' to `a' */
  436.         if (b < a)
  437.           return getobjname(L, ci, b, name);  /* get name for `b' */
  438.         break;
  439.       }
  440.       case OP_GETTABLE: {
  441.         int k = GETARG_C(i);  /* key index */
  442.         *name = kname(p, k);
  443.         return "field";
  444.       }
  445.       case OP_GETUPVAL: {
  446.         int u = GETARG_B(i);  /* upvalue index */
  447.         *name = p->upvalues ? getstr(p->upvalues[u]) : "?";
  448.         return "upvalue";
  449.       }
  450.       case OP_SELF: {
  451.         int k = GETARG_C(i);  /* key index */
  452.         *name = kname(p, k);
  453.         return "method";
  454.       }
  455.       default: break;
  456.     }
  457.   }
  458.   return NULL;  /* no useful name found */
  459. }
  460. static const char *getfuncname (lua_State *L, CallInfo *ci, const char **name) {
  461.   Instruction i;
  462.   if ((isLua(ci) && ci->tailcalls > 0) || !isLua(ci - 1))
  463.     return NULL;  /* calling function is not Lua (or is unknown) */
  464.   ci--;  /* calling function */
  465.   i = ci_func(ci)->l.p->code[currentpc(L, ci)];
  466.   if (GET_OPCODE(i) == OP_CALL || GET_OPCODE(i) == OP_TAILCALL ||
  467.       GET_OPCODE(i) == OP_TFORLOOP)
  468.     return getobjname(L, ci, GETARG_A(i), name);
  469.   else
  470.     return NULL;  /* no useful name can be found */
  471. }
  472. /* only ANSI way to check whether a pointer points to an array */
  473. static int isinstack (CallInfo *ci, const TValue *o) {
  474.   StkId p;
  475.   for (p = ci->base; p < ci->top; p++)
  476.     if (o == p) return 1;
  477.   return 0;
  478. }
  479. void luaG_typeerror (lua_State *L, const TValue *o, const char *op) {
  480.   const char *name = NULL;
  481.   const char *t = luaT_typenames[ttype(o)];
  482.   const char *kind = (isinstack(L->ci, o)) ?
  483.                          getobjname(L, L->ci, cast_int(o - L->base), &name) :
  484.                          NULL;
  485.   if (kind)
  486.     luaG_runerror(L, "attempt to %s %s " LUA_QS " (a %s value)",
  487.                 op, kind, name, t);
  488.   else
  489.     luaG_runerror(L, "attempt to %s a %s value", op, t);
  490. }
  491. void luaG_concaterror (lua_State *L, StkId p1, StkId p2) {
  492.   if (ttisstring(p1)) p1 = p2;
  493.   lua_assert(!ttisstring(p1));
  494.   luaG_typeerror(L, p1, "concatenate");
  495. }
  496. void luaG_aritherror (lua_State *L, const TValue *p1, const TValue *p2) {
  497.   TValue temp;
  498.   if (luaV_tonumber(p1, &temp) == NULL)
  499.     p2 = p1;  /* first operand is wrong */
  500.   luaG_typeerror(L, p2, "perform arithmetic on");
  501. }
  502. int luaG_ordererror (lua_State *L, const TValue *p1, const TValue *p2) {
  503.   const char *t1 = luaT_typenames[ttype(p1)];
  504.   const char *t2 = luaT_typenames[ttype(p2)];
  505.   if (t1[2] == t2[2])
  506.     luaG_runerror(L, "attempt to compare two %s values", t1);
  507.   else
  508.     luaG_runerror(L, "attempt to compare %s with %s", t1, t2);
  509.   return 0;
  510. }
  511. static void addinfo (lua_State *L, const char *msg) {
  512.   CallInfo *ci = L->ci;
  513.   if (isLua(ci)) {  /* is Lua code? */
  514.     char buff[LUA_IDSIZE];  /* add file:line information */
  515.     int line = currentline(L, ci);
  516.     luaO_chunkid(buff, getstr(getluaproto(ci)->source), LUA_IDSIZE);
  517.     luaO_pushfstring(L, "%s:%d: %s", buff, line, msg);
  518.   }
  519. }
  520. void luaG_errormsg (lua_State *L) {
  521.   if (L->errfunc != 0) {  /* is there an error handling function? */
  522.     StkId errfunc = restorestack(L, L->errfunc);
  523.     if (!ttisfunction(errfunc)) luaD_throw(L, LUA_ERRERR);
  524.     setobjs2s(L, L->top, L->top - 1);  /* move argument */
  525.     setobjs2s(L, L->top - 1, errfunc);  /* push function */
  526.     incr_top(L);
  527.     luaD_call(L, L->top - 2, 1);  /* call it */
  528.   }
  529.   luaD_throw(L, LUA_ERRRUN);
  530. }
  531. void luaG_runerror (lua_State *L, const char *fmt, ...) {
  532.   va_list argp;
  533.   va_start(argp, fmt);
  534.   addinfo(L, luaO_pushvfstring(L, fmt, argp));
  535.   va_end(argp);
  536.   luaG_errormsg(L);
  537. }