/*
** $Id: print.c,v 1.55a 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);
size_t i,n=ts->tsv.len;
for (i=0; i<n; i++)
{
int c=s[i];
switch (c)
{
case '"': printf("\\\""); break;
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
))
else
printf("\\%03u",(unsigned char)c
);
}
}
}
static void PrintConstant(const Proto* f, int i)
{
const TValue* o=&f->k[i];
switch (ttype(o))
{
case LUA_TNIL:
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 */
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("%-9s\t",luaP_opnames
[o
]);
switch (getOpMode(o))
{
case iABC:
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))
{
if (ISK
(b
)) PrintConstant
(f
,INDEXK
(b
)); else printf("-");
if (ISK
(c
)) PrintConstant
(f
,INDEXK
(c
)); else printf("-");
}
break;
case OP_JMP:
case OP_FORLOOP:
case OP_FORPREP:
break;
case OP_CLOSURE:
printf("\t; %p",VOID
(f
->p
[bx
]));
break;
case OP_SETLIST:
if (c
==0) printf("\t; %d",(int)code
[++pc
]);
break;
default:
break;
}
}
}
#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++)
{
PrintConstant(f,i);
}
}
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++)
{
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);
}