Lua 5.1.4: print.c
L0001
L0006
L0007 #include <ctype.h>
L0008 #include <stdio.h>
L0009
L0010 #define luac_c
L0011 #define LUA_CORE
L0012
L0013 #include "ldebug.h"
L0014 #include "lobject.h"
L0015 #include "lopcodes.h"
L0016 #include "lundump.h"
L0017
L0018 #define PrintFunction luaU_print
L0019
L0020 #define Sizeof(x) ((int)sizeof(x))
L0021 #define VOID(p) ((const void*)(p))
L0022
L0023 static void PrintString(const TString* ts)
L0024 {
L0025 const char* s=getstr(ts);
L0026 size_t i,n=ts->tsv.len;
L0027 putchar('"');
L0028 for (i=0; i<n; i++)
L0029 {
L0030 int c=s[i];
L0031 switch (c)
L0032 {
L0033 case '"': printf("\\\""); break;
L0034 case '\\': printf("\\\\"); break;
L0035 case '\a': printf("\\a"); break;
L0036 case '\b': printf("\\b"); break;
L0037 case '\f': printf("\\f"); break;
L0038 case '\n': printf("\\n"); break;
L0039 case '\r': printf("\\r"); break;
L0040 case '\t': printf("\\t"); break;
L0041 case '\v': printf("\\v"); break;
L0042 default: if (isprint((unsigned char)c))
L0043 putchar(c);
L0044 else
L0045 printf("\\%03u",(unsigned char)c);
L0046 }
L0047 }
L0048 putchar('"');
L0049 }
L0050
L0051 static void PrintConstant(const Proto* f, int i)
L0052 {
L0053 const TValue* o=&f->k[i];
L0054 switch (ttype(o))
L0055 {
L0056 case LUA_TNIL:
L0057 printf("nil");
L0058 break;
L0059 case LUA_TBOOLEAN:
L0060 printf(bvalue(o) ? "true" : "false");
L0061 break;
L0062 case LUA_TNUMBER:
L0063 printf(LUA_NUMBER_FMT,nvalue(o));
L0064 break;
L0065 case LUA_TSTRING:
L0066 PrintString(rawtsvalue(o));
L0067 break;
L0068 default:
L0069 printf("? type=%d",ttype(o));
L0070 break;
L0071 }
L0072 }
L0073
L0074 static void PrintCode(const Proto* f)
L0075 {
L0076 const Instruction* code=f->code;
L0077 int pc,n=f->sizecode;
L0078 for (pc=0; pc<n; pc++)
L0079 {
L0080 Instruction i=code[pc];
L0081 OpCode o=GET_OPCODE(i);
L0082 int a=GETARG_A(i);
L0083 int b=GETARG_B(i);
L0084 int c=GETARG_C(i);
L0085 int bx=GETARG_Bx(i);
L0086 int sbx=GETARG_sBx(i);
L0087 int line=getline(f,pc);
L0088 printf("\t%d\t",pc+1);
L0089 if (line>0) printf("[%d]\t",line); else printf("[-]\t");
L0090 printf("%-9s\t",luaP_opnames[o]);
L0091 switch (getOpMode(o))
L0092 {
L0093 case iABC:
L0094 printf("%d",a);
L0095 if (getBMode(o)!=OpArgN) printf(" %d",ISK(b) ? (-1-INDEXK(b)) : b);
L0096 if (getCMode(o)!=OpArgN) printf(" %d",ISK(c) ? (-1-INDEXK(c)) : c);
L0097 break;
L0098 case iABx:
L0099 if (getBMode(o)==OpArgK) printf("%d %d",a,-1-bx); else printf("%d %d",a,bx);
L0100 break;
L0101 case iAsBx:
L0102 if (o==OP_JMP) printf("%d",sbx); else printf("%d %d",a,sbx);
L0103 break;
L0104 }
L0105 switch (o)
L0106 {
L0107 case OP_LOADK:
L0108 printf("\t; "); PrintConstant(f,bx);
L0109 break;
L0110 case OP_GETUPVAL:
L0111 case OP_SETUPVAL:
L0112 printf("\t; %s", (f->sizeupvalues>0) ? getstr(f->upvalues[b]) : "-");
L0113 break;
L0114 case OP_GETGLOBAL:
L0115 case OP_SETGLOBAL:
L0116 printf("\t; %s",svalue(&f->k[bx]));
L0117 break;
L0118 case OP_GETTABLE:
L0119 case OP_SELF:
L0120 if (ISK(c)) { printf("\t; "); PrintConstant(f,INDEXK(c)); }
L0121 break;
L0122 case OP_SETTABLE:
L0123 case OP_ADD:
L0124 case OP_SUB:
L0125 case OP_MUL:
L0126 case OP_DIV:
L0127 case OP_POW:
L0128 case OP_EQ:
L0129 case OP_LT:
L0130 case OP_LE:
L0131 if (ISK(b) || ISK(c))
L0132 {
L0133 printf("\t; ");
L0134 if (ISK(b)) PrintConstant(f,INDEXK(b)); else printf("-");
L0135 printf(" ");
L0136 if (ISK(c)) PrintConstant(f,INDEXK(c)); else printf("-");
L0137 }
L0138 break;
L0139 case OP_JMP:
L0140 case OP_FORLOOP:
L0141 case OP_FORPREP:
L0142 printf("\t; to %d",sbx+pc+2);
L0143 break;
L0144 case OP_CLOSURE:
L0145 printf("\t; %p",VOID(f->p[bx]));
L0146 break;
L0147 case OP_SETLIST:
L0148 if (c==0) printf("\t; %d",(int)code[++pc]);
L0149 else printf("\t; %d",c);
L0150 break;
L0151 default:
L0152 break;
L0153 }
L0154 printf("\n");
L0155 }
L0156 }
L0157
L0158 #define SS(x) (x==1)?"":"s"
L0159 #define S(x) x,SS(x)
L0160
L0161 static void PrintHeader(const Proto* f)
L0162 {
L0163 const char* s=getstr(f->source);
L0164 if (*s=='@' || *s=='=')
L0165 s++;
L0166 else if (*s==LUA_SIGNATURE[0])
L0167 s="(bstring)";
L0168 else
L0169 s="(string)";
L0170 printf("\n%s <%s:%d,%d> (%d instruction%s, %d bytes at %p)\n",
L0171 (f->linedefined==0)?"main":"function",s,
L0172 f->linedefined,f->lastlinedefined,
L0173 S(f->sizecode),f->sizecode*Sizeof(Instruction),VOID(f));
L0174 printf("%d%s param%s, %d slot%s, %d upvalue%s, ",
L0175 f->numparams,f->is_vararg?"+":"",SS(f->numparams),
L0176 S(f->maxstacksize),S(f->nups));
L0177 printf("%d local%s, %d constant%s, %d function%s\n",
L0178 S(f->sizelocvars),S(f->sizek),S(f->sizep));
L0179 }
L0180
L0181 static void PrintConstants(const Proto* f)
L0182 {
L0183 int i,n=f->sizek;
L0184 printf("constants (%d) for %p:\n",n,VOID(f));
L0185 for (i=0; i<n; i++)
L0186 {
L0187 printf("\t%d\t",i+1);
L0188 PrintConstant(f,i);
L0189 printf("\n");
L0190 }
L0191 }
L0192
L0193 static void PrintLocals(const Proto* f)
L0194 {
L0195 int i,n=f->sizelocvars;
L0196 printf("locals (%d) for %p:\n",n,VOID(f));
L0197 for (i=0; i<n; i++)
L0198 {
L0199 printf("\t%d\t%s\t%d\t%d\n",
L0200 i,getstr(f->locvars[i].varname),f->locvars[i].startpc+1,f->locvars[i].endpc+1);
L0201 }
L0202 }
L0203
L0204 static void PrintUpvalues(const Proto* f)
L0205 {
L0206 int i,n=f->sizeupvalues;
L0207 printf("upvalues (%d) for %p:\n",n,VOID(f));
L0208 if (f->upvalues==NULL) return;
L0209 for (i=0; i<n; i++)
L0210 {
L0211 printf("\t%d\t%s\n",i,getstr(f->upvalues[i]));
L0212 }
L0213 }
L0214
L0215 void PrintFunction(const Proto* f, int full)
L0216 {
L0217 int i,n=f->sizep;
L0218 PrintHeader(f);
L0219 PrintCode(f);
L0220 if (full)
L0221 {
L0222 PrintConstants(f);
L0223 PrintLocals(f);
L0224 PrintUpvalues(f);
L0225 }
L0226 for (i=0; i<n; i++) PrintFunction(f->p[i],full);
L0227 }
Generated by pretty.lua