/src/middleware/lua/print.c

  1/*
  2** $Id: print.c,v 1.55a 2006/05/31 13:30:05 lhf Exp $
  3** print bytecodes
  4** See Copyright Notice in lua.h
  5*/
  6
  7#include <ctype.h>
  8#include <stdio.h>
  9
 10#define luac_c
 11#define LUA_CORE
 12
 13#include "ldebug.h"
 14#include "lobject.h"
 15#include "lopcodes.h"
 16#include "lundump.h"
 17
 18#define PrintFunction	luaU_print
 19
 20#define Sizeof(x)	((int)sizeof(x))
 21#define VOID(p)		((const void*)(p))
 22
 23static void PrintString(const TString* ts)
 24{
 25 const char* s=getstr(ts);
 26 size_t i,n=ts->tsv.len;
 27 putchar('"');
 28 for (i=0; i<n; i++)
 29 {
 30  int c=s[i];
 31  switch (c)
 32  {
 33   case '"': printf("\\\""); break;
 34   case '\\': printf("\\\\"); break;
 35   case '\a': printf("\\a"); break;
 36   case '\b': printf("\\b"); break;
 37   case '\f': printf("\\f"); break;
 38   case '\n': printf("\\n"); break;
 39   case '\r': printf("\\r"); break;
 40   case '\t': printf("\\t"); break;
 41   case '\v': printf("\\v"); break;
 42   default:	if (isprint((unsigned char)c))
 43   			putchar(c);
 44		else
 45			printf("\\%03u",(unsigned char)c);
 46  }
 47 }
 48 putchar('"');
 49}
 50
 51static void PrintConstant(const Proto* f, int i)
 52{
 53 const TValue* o=&f->k[i];
 54 switch (ttype(o))
 55 {
 56  case LUA_TNIL:
 57	printf("nil");
 58	break;
 59  case LUA_TBOOLEAN:
 60	printf(bvalue(o) ? "true" : "false");
 61	break;
 62  case LUA_TNUMBER:
 63	printf(LUA_NUMBER_FMT,nvalue(o));
 64	break;
 65  case LUA_TSTRING:
 66	PrintString(rawtsvalue(o));
 67	break;
 68  default:				/* cannot happen */
 69	printf("? type=%d",ttype(o));
 70	break;
 71 }
 72}
 73
 74static void PrintCode(const Proto* f)
 75{
 76 const Instruction* code=f->code;
 77 int pc,n=f->sizecode;
 78 for (pc=0; pc<n; pc++)
 79 {
 80  Instruction i=code[pc];
 81  OpCode o=GET_OPCODE(i);
 82  int a=GETARG_A(i);
 83  int b=GETARG_B(i);
 84  int c=GETARG_C(i);
 85  int bx=GETARG_Bx(i);
 86  int sbx=GETARG_sBx(i);
 87  int line=getline(f,pc);
 88  printf("\t%d\t",pc+1);
 89  if (line>0) printf("[%d]\t",line); else printf("[-]\t");
 90  printf("%-9s\t",luaP_opnames[o]);
 91  switch (getOpMode(o))
 92  {
 93   case iABC:
 94    printf("%d",a);
 95    if (getBMode(o)!=OpArgN) printf(" %d",ISK(b) ? (-1-INDEXK(b)) : b);
 96    if (getCMode(o)!=OpArgN) printf(" %d",ISK(c) ? (-1-INDEXK(c)) : c);
 97    break;
 98   case iABx:
 99    if (getBMode(o)==OpArgK) printf("%d %d",a,-1-bx); else printf("%d %d",a,bx);
100    break;
101   case iAsBx:
102    if (o==OP_JMP) printf("%d",sbx); else printf("%d %d",a,sbx);
103    break;
104  }
105  switch (o)
106  {
107   case OP_LOADK:
108    printf("\t; "); PrintConstant(f,bx);
109    break;
110   case OP_GETUPVAL:
111   case OP_SETUPVAL:
112    printf("\t; %s", (f->sizeupvalues>0) ? getstr(f->upvalues[b]) : "-");
113    break;
114   case OP_GETGLOBAL:
115   case OP_SETGLOBAL:
116    printf("\t; %s",svalue(&f->k[bx]));
117    break;
118   case OP_GETTABLE:
119   case OP_SELF:
120    if (ISK(c)) { printf("\t; "); PrintConstant(f,INDEXK(c)); }
121    break;
122   case OP_SETTABLE:
123   case OP_ADD:
124   case OP_SUB:
125   case OP_MUL:
126   case OP_DIV:
127   case OP_POW:
128   case OP_EQ:
129   case OP_LT:
130   case OP_LE:
131    if (ISK(b) || ISK(c))
132    {
133     printf("\t; ");
134     if (ISK(b)) PrintConstant(f,INDEXK(b)); else printf("-");
135     printf(" ");
136     if (ISK(c)) PrintConstant(f,INDEXK(c)); else printf("-");
137    }
138    break;
139   case OP_JMP:
140   case OP_FORLOOP:
141   case OP_FORPREP:
142    printf("\t; to %d",sbx+pc+2);
143    break;
144   case OP_CLOSURE:
145    printf("\t; %p",VOID(f->p[bx]));
146    break;
147   case OP_SETLIST:
148    if (c==0) printf("\t; %d",(int)code[++pc]);
149    else printf("\t; %d",c);
150    break;
151   default:
152    break;
153  }
154  printf("\n");
155 }
156}
157
158#define SS(x)	(x==1)?"":"s"
159#define S(x)	x,SS(x)
160
161static void PrintHeader(const Proto* f)
162{
163 const char* s=getstr(f->source);
164 if (*s=='@' || *s=='=')
165  s++;
166 else if (*s==LUA_SIGNATURE[0])
167  s="(bstring)";
168 else
169  s="(string)";
170 printf("\n%s <%s:%d,%d> (%d instruction%s, %d bytes at %p)\n",
171 	(f->linedefined==0)?"main":"function",s,
172	f->linedefined,f->lastlinedefined,
173	S(f->sizecode),f->sizecode*Sizeof(Instruction),VOID(f));
174 printf("%d%s param%s, %d slot%s, %d upvalue%s, ",
175	f->numparams,f->is_vararg?"+":"",SS(f->numparams),
176	S(f->maxstacksize),S(f->nups));
177 printf("%d local%s, %d constant%s, %d function%s\n",
178	S(f->sizelocvars),S(f->sizek),S(f->sizep));
179}
180
181static void PrintConstants(const Proto* f)
182{
183 int i,n=f->sizek;
184 printf("constants (%d) for %p:\n",n,VOID(f));
185 for (i=0; i<n; i++)
186 {
187  printf("\t%d\t",i+1);
188  PrintConstant(f,i);
189  printf("\n");
190 }
191}
192
193static void PrintLocals(const Proto* f)
194{
195 int i,n=f->sizelocvars;
196 printf("locals (%d) for %p:\n",n,VOID(f));
197 for (i=0; i<n; i++)
198 {
199  printf("\t%d\t%s\t%d\t%d\n",
200  i,getstr(f->locvars[i].varname),f->locvars[i].startpc+1,f->locvars[i].endpc+1);
201 }
202}
203
204static void PrintUpvalues(const Proto* f)
205{
206 int i,n=f->sizeupvalues;
207 printf("upvalues (%d) for %p:\n",n,VOID(f));
208 if (f->upvalues==NULL) return;
209 for (i=0; i<n; i++)
210 {
211  printf("\t%d\t%s\n",i,getstr(f->upvalues[i]));
212 }
213}
214
215void PrintFunction(const Proto* f, int full)
216{
217 int i,n=f->sizep;
218 PrintHeader(f);
219 PrintCode(f);
220 if (full)
221 {
222  PrintConstants(f);
223  PrintLocals(f);
224  PrintUpvalues(f);
225 }
226 for (i=0; i<n; i++) PrintFunction(f->p[i],full);
227}