/*** $Id: print.c,v 1.55 2006/05/31 13:30:05 lhf Exp $** print bytecodes** See Copyright Notice in lua.h*/#include <ctype.h>#include <stdio.h>#define luac_c#define LUA_CORE#include "ldebug.h"#include "lobject.h"#include "lopcodes.h"#include "lundump.h"#define PrintFunction	luaU_print#define Sizeof(x)	((int)sizeof(x))#define VOID(p)		((const void*)(p))static void PrintString(const TString* ts){ const char* s=getstr(ts); int n=ts->tsv.len; int i; putchar('"'); for (i=0; i<n; i++) {  int c=s[i];  switch (c)  {   case '"': printf("\\\""); break;   case '\a': printf("\\a"); break;   case '\b': printf("\\b"); break;   case '\f': printf("\\f"); break;   case '\n': printf("\\n"); break;   case '\r': printf("\\r"); break;   case '\t': printf("\\t"); break;   case '\v': printf("\\v"); break;   default:	if (isprint((unsigned char)c))   			putchar(c);		else			printf("\\%03u",(unsigned char)c);  } } putchar('"');}static void PrintConstant(const Proto* f, int i){ const TValue* o=&f->k[i]; switch (ttype(o)) {  case LUA_TNIL:	printf("nil");	break;  case LUA_TBOOLEAN:	printf(bvalue(o) ? "true" : "false");	break;  case LUA_TNUMBER:	printf(LUA_NUMBER_FMT,nvalue(o));	break;  case LUA_TSTRING:	PrintString(rawtsvalue(o));	break;  default:				/* cannot happen */	printf("? type=%d",ttype(o));	break; }}static void PrintCode(const Proto* f){ const Instruction* code=f->code; int pc,n=f->sizecode; for (pc=0; pc<n; pc++) {  Instruction i=code[pc];  OpCode o=GET_OPCODE(i);  int a=GETARG_A(i);  int b=GETARG_B(i);  int c=GETARG_C(i);  int bx=GETARG_Bx(i);  int sbx=GETARG_sBx(i);  int line=getline(f,pc);  printf("\t%d\t",pc+1);  if (line>0) printf("[%d]\t",line); else printf("[-]\t");  printf("%-9s\t",luaP_opnames[o]);  switch (getOpMode(o))  {   case iABC:    printf("%d",a);    if (getBMode(o)!=OpArgN) printf(" %d",ISK(b) ? (-1-INDEXK(b)) : b);    if (getCMode(o)!=OpArgN) printf(" %d",ISK(c) ? (-1-INDEXK(c)) : c);    break;   case iABx:    if (getBMode(o)==OpArgK) printf("%d %d",a,-1-bx); else printf("%d %d",a,bx);    break;   case iAsBx:    if (o==OP_JMP) printf("%d",sbx); else printf("%d %d",a,sbx);    break;  }  switch (o)  {   case OP_LOADK:    printf("\t; "); PrintConstant(f,bx);    break;   case OP_GETUPVAL:   case OP_SETUPVAL:    printf("\t; %s", (f->sizeupvalues>0) ? getstr(f->upvalues[b]) : "-");    break;   case OP_GETGLOBAL:   case OP_SETGLOBAL:    printf("\t; %s",svalue(&f->k[bx]));    break;   case OP_GETTABLE:   case OP_SELF:    if (ISK(c)) { printf("\t; "); PrintConstant(f,INDEXK(c)); }    break;   case OP_SETTABLE:   case OP_ADD:   case OP_SUB:   case OP_MUL:   case OP_DIV:   case OP_POW:   case OP_EQ:   case OP_LT:   case OP_LE:    if (ISK(b) || ISK(c))    {     printf("\t; ");     if (ISK(b)) PrintConstant(f,INDEXK(b)); else printf("-");     printf(" ");     if (ISK(c)) PrintConstant(f,INDEXK(c)); else printf("-");    }    break;   case OP_JMP:   case OP_FORLOOP:   case OP_FORPREP:    printf("\t; to %d",sbx+pc+2);    break;   case OP_CLOSURE:    printf("\t; %p",VOID(f->p[bx]));    break;   case OP_SETLIST:    if (c==0) printf("\t; %d",(int)code[++pc]);    else printf("\t; %d",c);    break;   default:    break;  }  printf("\n"); }}#define SS(x)	(x==1)?"":"s"#define S(x)	x,SS(x)static void PrintHeader(const Proto* f){ const char* s=getstr(f->source); if (*s=='@' || *s=='=')  s++; else if (*s==LUA_SIGNATURE[0])  s="(bstring)"; else  s="(string)"; printf("\n%s <%s:%d,%d> (%d instruction%s, %d bytes at %p)\n", 	(f->linedefined==0)?"main":"function",s,	f->linedefined,f->lastlinedefined,	S(f->sizecode),f->sizecode*Sizeof(Instruction),VOID(f)); printf("%d%s param%s, %d slot%s, %d upvalue%s, ",	f->numparams,f->is_vararg?"+":"",SS(f->numparams),	S(f->maxstacksize),S(f->nups)); printf("%d local%s, %d constant%s, %d function%s\n",	S(f->sizelocvars),S(f->sizek),S(f->sizep));}static void PrintConstants(const Proto* f){ int i,n=f->sizek; printf("constants (%d) for %p:\n",n,VOID(f)); for (i=0; i<n; i++) {  printf("\t%d\t",i+1);  PrintConstant(f,i);  printf("\n"); }}static void PrintLocals(const Proto* f){ int i,n=f->sizelocvars; printf("locals (%d) for %p:\n",n,VOID(f)); for (i=0; i<n; i++) {  printf("\t%d\t%s\t%d\t%d\n",  i,getstr(f->locvars[i].varname),f->locvars[i].startpc+1,f->locvars[i].endpc+1); }}static void PrintUpvalues(const Proto* f){ int i,n=f->sizeupvalues; printf("upvalues (%d) for %p:\n",n,VOID(f)); if (f->upvalues==NULL) return; for (i=0; i<n; i++) {  printf("\t%d\t%s\n",i,getstr(f->upvalues[i])); }}void PrintFunction(const Proto* f, int full){ int i,n=f->sizep; PrintHeader(f); PrintCode(f); if (full) {  PrintConstants(f);  PrintLocals(f);  PrintUpvalues(f); } for (i=0; i<n; i++) PrintFunction(f->p[i],full);}