summaryrefslogtreecommitdiff
path: root/src/luac/print.c
diff options
context:
space:
mode:
Diffstat (limited to 'src/luac/print.c')
-rw-r--r--src/luac/print.c233
1 files changed, 213 insertions, 20 deletions
diff --git a/src/luac/print.c b/src/luac/print.c
index 0e1d2569..fb7bf8a7 100644
--- a/src/luac/print.c
+++ b/src/luac/print.c
@@ -3,7 +3,7 @@
** print bytecodes
*/
-char* rcs_print="$Id: print.c,v 1.11 1996/11/18 11:24:16 lhf Exp $";
+char* rcs_print="$Id: print.c,v 1.17 1997/06/25 17:07:28 lhf Exp $";
#include <stdio.h>
#include <stdlib.h>
@@ -11,20 +11,134 @@ char* rcs_print="$Id: print.c,v 1.11 1996/11/18 11:24:16 lhf Exp $";
#include "luac.h"
#include "print.h"
-static LocVar* V=NULL;
-
-static char* LocStr(int i)
+void LinkFunctions(TFunc* m)
{
- if (V==NULL) return ""; else return V[i].varname->str;
+ static TFunc* lastF; /* list of functions seen in code */
+ Byte* code=m->code;
+ Byte* end=code+m->size;
+ Byte* p;
+ if (IsMain(m)) lastF=m;
+ for (p=code; p!=end;)
+ {
+ int op=*p;
+ int at=p-code+1;
+ switch (op)
+ {
+ case PUSHNIL:
+ case PUSH0:
+ case PUSH1:
+ case PUSH2:
+ case PUSHLOCAL0:
+ case PUSHLOCAL1:
+ case PUSHLOCAL2:
+ case PUSHLOCAL3:
+ case PUSHLOCAL4:
+ case PUSHLOCAL5:
+ case PUSHLOCAL6:
+ case PUSHLOCAL7:
+ case PUSHLOCAL8:
+ case PUSHLOCAL9:
+ case PUSHINDEXED:
+ case STORELOCAL0:
+ case STORELOCAL1:
+ case STORELOCAL2:
+ case STORELOCAL3:
+ case STORELOCAL4:
+ case STORELOCAL5:
+ case STORELOCAL6:
+ case STORELOCAL7:
+ case STORELOCAL8:
+ case STORELOCAL9:
+ case STOREINDEXED0:
+ case ADJUST0:
+ case EQOP:
+ case LTOP:
+ case LEOP:
+ case GTOP:
+ case GEOP:
+ case ADDOP:
+ case SUBOP:
+ case MULTOP:
+ case DIVOP:
+ case POWOP:
+ case CONCOP:
+ case MINUSOP:
+ case NOTOP:
+ case POP:
+ case RETCODE0:
+ p++;
+ break;
+ case PUSHBYTE:
+ case PUSHLOCAL:
+ case STORELOCAL:
+ case STOREINDEXED:
+ case STORELIST0:
+ case ADJUST:
+ case RETCODE:
+ case VARARGS:
+ case STOREMAP:
+ p+=2;
+ break;
+ case PUSHWORD:
+ case PUSHSTRING:
+ case PUSHGLOBAL:
+ case PUSHSELF:
+ case STOREGLOBAL:
+ case CREATEARRAY:
+ case ONTJMP:
+ case ONFJMP:
+ case JMP:
+ case UPJMP:
+ case IFFJMP:
+ case IFFUPJMP:
+ case CALLFUNC:
+ case SETLINE:
+ case STORELIST:
+ p+=3;
+ break;
+ case PUSHFLOAT:
+ p+=5; /* assumes sizeof(float)==4 */
+ break;
+ case PUSHFUNCTION:
+ {
+ TFunc* tf;
+ p++;
+ get_code(tf,p);
+ tf->marked=at;
+ tf->next=NULL; /* TODO: remove? */
+ lastF=lastF->next=tf;
+ break;
+ }
+ case STORERECORD:
+ {
+ int n=*++p;
+ p+=2*n+1;
+ break;
+ }
+ default: /* cannot happen */
+ fprintf(stderr,"luac: bad opcode %d at %d\n",*p,(int)(p-code));
+ exit(1);
+ break;
+ }
+ }
}
-static void PrintCode(Byte* code, Byte* end)
+#define LocStr(i) luaI_getlocalname(tf,i+1,line)
+
+static void PrintCode(TFunc* tf)
{
+ Byte* code=tf->code;
+ Byte* end=code+tf->size;
Byte* p;
+ int line=0;
for (p=code; p!=end;)
{
- OpCode op=(OpCode)*p;
- if (op>SETLINE) op=SETLINE+1;
+ int op=*p;
+ if (op>=NOPCODES)
+ {
+ fprintf(stderr,"luac: bad opcode %d at %d\n",op,(int)(p-code));
+ exit(1);
+ }
printf("%6d\t%s",(int)(p-code),OpCodeName[op]);
switch (op)
{
@@ -64,7 +178,7 @@ static void PrintCode(Byte* code, Byte* end)
case PUSHLOCAL9:
{
int i=op-PUSHLOCAL0;
- printf("\t%d\t; %s",i,LocStr(i));
+ if (tf->locvars) printf("\t\t; %s",LocStr(i));
p++;
break;
}
@@ -80,7 +194,7 @@ static void PrintCode(Byte* code, Byte* end)
case STORELOCAL9:
{
int i=op-STORELOCAL0;
- printf("\t%d\t; %s",i,LocStr(i));
+ if (tf->locvars) printf("\t\t; %s",LocStr(i));
p++;
break;
}
@@ -88,7 +202,7 @@ static void PrintCode(Byte* code, Byte* end)
case STORELOCAL:
{
int i=*(p+1);
- printf("\t%d\t; %s",i,LocStr(i));
+ if (tf->locvars) printf("\t%d\t; %s",i,LocStr(i));
p+=2;
break;
}
@@ -97,23 +211,40 @@ static void PrintCode(Byte* code, Byte* end)
case STORELIST0:
case ADJUST:
case RETCODE:
+ case VARARGS:
+ case STOREMAP:
printf("\t%d",*(p+1));
p+=2;
break;
case PUSHWORD:
case CREATEARRAY:
+ case SETLINE:
+ {
+ Word w;
+ p++;
+ get_word(w,p);
+ printf("\t%d",w);
+ if (op==SETLINE) line=w;
+ break;
+ }
case ONTJMP:
case ONFJMP:
case JMP:
- case UPJMP:
case IFFJMP:
+ { /* suggested by Norman Ramsey <nr@cs.virginia.edu> */
+ Word w;
+ p++;
+ get_word(w,p);
+ printf("\t%d\t\t; to %d",w,(int)(p-code)+w);
+ break;
+ }
+ case UPJMP:
case IFFUPJMP:
- case SETLINE:
- {
+ { /* suggested by Norman Ramsey <nr@cs.virginia.edu> */
Word w;
p++;
get_word(w,p);
- printf("\t%d",w);
+ printf("\t%d\t\t; to %d",w,(int)(p-code)-w);
break;
}
case PUSHFLOAT:
@@ -170,7 +301,8 @@ static void PrintCode(Byte* code, Byte* end)
break;
}
default:
- printf("\tcannot happen: opcode=%d",*p);
+ printf("\tcannot happen: opcode=%d\n",*p);
+ fprintf(stderr,"luac: bad opcode %d at %d\n",op,(int)(p-code));
exit(1);
break;
}
@@ -178,13 +310,74 @@ static void PrintCode(Byte* code, Byte* end)
}
}
-void PrintFunction(TFunc* tf)
+#undef LocStr
+
+static void PrintLocals(LocVar* v, int n)
{
+ int i=0;
+ if (v==NULL || v->varname==NULL) return;
+ if (n>0)
+ {
+ printf("parameters:");
+ for (i=0; i<n; v++,i++) printf(" %s[%d@%d]",v->varname->str,i,v->line);
+ printf("\n");
+ }
+ if (v->varname!=NULL)
+ {
+ printf("locals:");
+ for (; v->line>=0; v++)
+ {
+ if (v->varname==NULL)
+#if 0
+ printf(" %s[%d@%d]","*",--i,v->line);
+#else
+ --i;
+#endif
+ else
+ printf(" %s[%d@%d]",v->varname->str,i++,v->line);
+ }
+ printf("\n");
+ }
+}
+
+void PrintFunction(TFunc* tf, TFunc* Main)
+{
+ int n=0;
if (IsMain(tf))
printf("\nmain of \"%s\" (%d bytes at %p)\n",tf->fileName,tf->size,tf);
else
- printf("\nfunction \"%s\":%d (%d bytes at %p); used at main+%d\n",
+ {
+ Byte* p;
+ p=tf->code; /* get number of parameters */
+ while (*p==SETLINE) p+=3;
+ if (*p==ADJUST) n=p[1];
+ p=Main->code+tf->marked+sizeof(TFunc*);
+ printf("\nfunction ");
+ switch (*p) /* try to get name */
+ {
+ case STOREGLOBAL:
+ {
+ Word w;
+ p++; get_word(w,p); printf("%s defined at ",VarStr(w));
+ break;
+ }
+ case STOREINDEXED0: /* try method definition */
+ {
+ if (p[-11]==PUSHGLOBAL && p[-8]==PUSHSTRING)
+ {
+ Word w;
+ Byte* op=p;
+ int c=(tf->locvars && n>0 && strcmp(tf->locvars->varname->str,"self")==0)
+ ? ':' : '.';
+ p=op-11; p++; get_word(w,p); printf("%s%c",VarStr(w),c);
+ p=op-8; p++; get_word(w,p); printf("%s defined at ",StrStr(w));
+ }
+ break;
+ }
+ }
+ printf("\"%s\":%d (%d bytes at %p); used at main+%d\n",
tf->fileName,tf->lineDefined,tf->size,tf,tf->marked);
- V=tf->locvars;
- PrintCode(tf->code,tf->code+tf->size);
+ }
+ PrintLocals(tf->locvars,n);
+ PrintCode(tf);
}