src/luac/print.c

   1 /*
   2 ** $Id: print.c,v 1.13 1998/07/12 00:17:37 lhf Exp $
   3 ** print bytecodes
   4 ** See Copyright Notice in lua.h
   5 */
   6
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include "luac.h"
  10
  11 #ifdef DEBUG
  12 void PrintConstant1(TProtoFunc* tf, int i)
  13 {
  14  TObject* o=tf->consts+i;
  15  printf("%6d ",i);
  16  if (i<0 || i>=tf->nconsts)
  17   printf("(bad constant #%d: max=%d)",i,tf->nconsts);
  18  else
  19   switch (ttype(o))
  20   {
  21    case LUA_T_NUMBER:
  22          printf("N " NUMBER_FMT "\n",nvalue(o));        /* LUA_NUMBER */
  23          break;
  24    case LUA_T_STRING:
  25          printf("S %p\t\"%s\"\n",(void*)tsvalue(o),svalue(o));
  26          break;
  27    case LUA_T_PROTO:
  28          printf("F %p\n",(void*)tfvalue(o));
  29          break;
  30    default:                             /* cannot happen */
  31          printf("? %d\n",ttype(o));
  32    break;
  33   }
  34 }
  35
  36 static void PrintConstants(TProtoFunc* tf)
  37 {
  38  int i,n=tf->nconsts;
  39  printf("constants (%d):\n",n);
  40  for (i=0; i<n; i++) PrintConstant1(tf,i);
  41 }
  42 #endif
  43
  44 static void PrintConstant(TProtoFunc* tf, int i)
  45 {
  46  if (i<0 || i>=tf->nconsts)
  47   printf("(bad constant #%d: max=%d)",i,tf->nconsts);
  48  else
  49  {
  50   TObject* o=tf->consts+i;
  51   switch (ttype(o))
  52   {
  53    case LUA_T_NUMBER:
  54         printf(NUMBER_FMT,nvalue(o));           /* LUA_NUMBER */
  55         break;
  56    case LUA_T_STRING:
  57         printf("\"%s\"",svalue(o));
  58         break;
  59    case LUA_T_PROTO:
  60         printf("function at %p",(void*)tfvalue(o));
  61         break;
  62    case LUA_T_NIL:
  63         printf("(nil)");
  64         break;
  65    default:                             /* cannot happen */
  66         printf("(bad constant #%d: type=%d [%s])\n",i,ttype(o),luaO_typename(o));
  67         break;
  68   }
  69  }
  70 }
  71
  72 #define VarStr(i)       svalue(tf->consts+i)
  73
  74 static void PrintCode(TProtoFunc* tf)
  75 {
  76  Byte* code=tf->code;
  77  Byte* p=code;
  78  int line=0;
  79  while (1)
  80  {
  81         Opcode OP;
  82         int n=INFO(tf,p,&OP);
  83         int op=OP.op;
  84         int i=OP.arg;
  85         printf("%6d  ",(int)(p-code));
  86         {
  87          Byte* q=p;
  88          int j=n;
  89          while (j--) printf("%02X",*q++);
  90         }
  91         printf("%*s%-13s",2*(5-n),"",OP.name);
  92
  93         if (n!=1 || op<0) printf("\t%d",i); else if (i>=0) printf("\t");
  94
  95         switch (OP.class)
  96         {
  97
  98         case ENDCODE:
  99                 printf("\n");
 100                 return;
 101
 102         case CLOSURE:
 103                 printf(" %d",OP.arg2);
 104         case PUSHCONSTANT:
 105         case GETDOTTED:
 106         case PUSHSELF:
 107                 printf("\t; ");
 108                 PrintConstant(tf,i);
 109                 break;
 110
 111         case PUSHLOCAL:
 112         case SETLOCAL:
 113         {
 114                 char* s=luaF_getlocalname(tf,i+1,line);
 115                 if (s) printf("\t; %s",s);
 116                 break;
 117         }
 118
 119         case GETGLOBAL:
 120         case SETGLOBAL:
 121                 printf("\t; %s",VarStr(i));
 122                 break;
 123
 124         case SETLIST:
 125         case CALLFUNC:
 126                 if (n>=3) printf(" %d",OP.arg2);
 127                 break;
 128
 129         case SETLINE:
 130                 printf("\t; \"%s\":%d",fileName(tf),line=i);
 131                 break;
 132
 133 /* suggested by Norman Ramsey <nr@cs.virginia.edu> */
 134         case IFTUPJMP:
 135         case IFFUPJMP:
 136                 i=-i;
 137         case ONTJMP:
 138         case ONFJMP:
 139         case JMP:
 140         case IFFJMP:
 141                 printf("\t; to %d",(int)(p-code)+i+n);
 142                 break;
 143
 144         }
 145         printf("\n");
 146         p+=n;
 147  }
 148 }
 149
 150 static void PrintLocals(TProtoFunc* tf)
 151 {
 152  LocVar* v=tf->locvars;
 153  int n,i=0;
 154  if (v==NULL || v->varname==NULL) return;
 155  n=tf->code[1]; if (n>=ZEROVARARG) n-=ZEROVARARG;
 156
 157  printf("locals:");
 158  if (n>0)
 159  {
 160   for (i=0; i<n; v++,i++) printf(" %s",v->varname->str);
 161  }
 162  if (v->varname!=NULL)
 163  {
 164   for (; v->line>=0; v++)
 165   {
 166    if (v->varname==NULL)
 167    {
 168     printf(")"); --i;
 169    }
 170    else
 171    {
 172     printf(" (%s",v->varname->str); i++;
 173    }
 174   }
 175   i-=n;
 176   while (i--) printf(")");
 177  }
 178  printf("\n");
 179 }
 180
 181 static void PrintHeader(TProtoFunc* tf, TProtoFunc* Main, int at)
 182 {
 183  int size=CodeSize(tf);
 184  if (IsMain(tf))
 185   printf("\nmain of \"%s\" (%d bytes at %p)\n",fileName(tf),size,(void*)tf);
 186  else if (Main)
 187  {
 188   printf("\nfunction defined at \"%s\":%d (%d bytes at %p); used at ",
 189         fileName(tf),tf->lineDefined,size,(void*)tf);
 190   if (IsMain(Main))
 191    printf("main");
 192   else
 193    printf("%p",(void*)Main);
 194   printf("+%d\n",at);
 195  }
 196 }
 197
 198 static void PrintFunction(TProtoFunc* tf, TProtoFunc* Main, int at);
 199
 200 static void PrintFunctions(TProtoFunc* Main)
 201 {
 202  Byte* code=Main->code;
 203  Byte* p=code;
 204  while (1)
 205  {
 206   Opcode OP;
 207   int n=INFO(Main,p,&OP);
 208   if (OP.class==ENDCODE) break;
 209   if (OP.class==PUSHCONSTANT || OP.class==CLOSURE)
 210   {
 211    int i=OP.arg;
 212    TObject* o=Main->consts+i;
 213    if (ttype(o)==LUA_T_PROTO) PrintFunction(tfvalue(o),Main,(int)(p-code));
 214   }
 215   p+=n;
 216  }
 217 }
 218
 219 static void PrintFunction(TProtoFunc* tf, TProtoFunc* Main, int at)
 220 {
 221  PrintHeader(tf,Main,at);
 222  PrintLocals(tf);
 223  PrintCode(tf);
 224 #ifdef DEBUG
 225  PrintConstants(tf);
 226 #endif
 227  PrintFunctions(tf);
 228 }
 229
 230 void PrintChunk(TProtoFunc* Main)
 231 {
 232  PrintFunction(Main,0,0);
 233 }