116a5fed65808adf648004b34f98718301d718fa2darylm/* 216a5fed65808adf648004b34f98718301d718fa2darylm** $Id: luac.c,v 1.69 2011/11/29 17:46:33 lhf Exp $ 316a5fed65808adf648004b34f98718301d718fa2darylm** Lua compiler (saves bytecodes to files; also list bytecodes) 416a5fed65808adf648004b34f98718301d718fa2darylm** See Copyright Notice in lua.h 516a5fed65808adf648004b34f98718301d718fa2darylm*/ 616a5fed65808adf648004b34f98718301d718fa2darylm 716a5fed65808adf648004b34f98718301d718fa2darylm#include <errno.h> 816a5fed65808adf648004b34f98718301d718fa2darylm#include <stdio.h> 916a5fed65808adf648004b34f98718301d718fa2darylm#include <stdlib.h> 1016a5fed65808adf648004b34f98718301d718fa2darylm#include <string.h> 1116a5fed65808adf648004b34f98718301d718fa2darylm 1216a5fed65808adf648004b34f98718301d718fa2darylm#define luac_c 1316a5fed65808adf648004b34f98718301d718fa2darylm#define LUA_CORE 1416a5fed65808adf648004b34f98718301d718fa2darylm 1516a5fed65808adf648004b34f98718301d718fa2darylm#include "lua.h" 1616a5fed65808adf648004b34f98718301d718fa2darylm#include "lauxlib.h" 1716a5fed65808adf648004b34f98718301d718fa2darylm 1816a5fed65808adf648004b34f98718301d718fa2darylm#include "lobject.h" 1916a5fed65808adf648004b34f98718301d718fa2darylm#include "lstate.h" 2016a5fed65808adf648004b34f98718301d718fa2darylm#include "lundump.h" 2116a5fed65808adf648004b34f98718301d718fa2darylm 2216a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintFunction(const Proto* f, int full); 2316a5fed65808adf648004b34f98718301d718fa2darylm#define luaU_print PrintFunction 2416a5fed65808adf648004b34f98718301d718fa2darylm 2516a5fed65808adf648004b34f98718301d718fa2darylm#define PROGNAME "luac" /* default program name */ 2616a5fed65808adf648004b34f98718301d718fa2darylm#define OUTPUT PROGNAME ".out" /* default output file */ 2716a5fed65808adf648004b34f98718301d718fa2darylm 2816a5fed65808adf648004b34f98718301d718fa2darylmstatic int listing=0; /* list bytecodes? */ 2916a5fed65808adf648004b34f98718301d718fa2darylmstatic int dumping=1; /* dump bytecodes? */ 3016a5fed65808adf648004b34f98718301d718fa2darylmstatic int stripping=0; /* strip debug information? */ 3116a5fed65808adf648004b34f98718301d718fa2darylmstatic char Output[]={ OUTPUT }; /* default output file name */ 3216a5fed65808adf648004b34f98718301d718fa2darylmstatic const char* output=Output; /* actual output file name */ 3316a5fed65808adf648004b34f98718301d718fa2darylmstatic const char* progname=PROGNAME; /* actual program name */ 3416a5fed65808adf648004b34f98718301d718fa2darylm 3516a5fed65808adf648004b34f98718301d718fa2darylmstatic void fatal(const char* message) 3616a5fed65808adf648004b34f98718301d718fa2darylm{ 3716a5fed65808adf648004b34f98718301d718fa2darylm fprintf(stderr,"%s: %s\n",progname,message); 3816a5fed65808adf648004b34f98718301d718fa2darylm exit(EXIT_FAILURE); 3916a5fed65808adf648004b34f98718301d718fa2darylm} 4016a5fed65808adf648004b34f98718301d718fa2darylm 4116a5fed65808adf648004b34f98718301d718fa2darylmstatic void cannot(const char* what) 4216a5fed65808adf648004b34f98718301d718fa2darylm{ 4316a5fed65808adf648004b34f98718301d718fa2darylm fprintf(stderr,"%s: cannot %s %s: %s\n",progname,what,output,strerror(errno)); 4416a5fed65808adf648004b34f98718301d718fa2darylm exit(EXIT_FAILURE); 4516a5fed65808adf648004b34f98718301d718fa2darylm} 4616a5fed65808adf648004b34f98718301d718fa2darylm 4716a5fed65808adf648004b34f98718301d718fa2darylmstatic void usage(const char* message) 4816a5fed65808adf648004b34f98718301d718fa2darylm{ 4916a5fed65808adf648004b34f98718301d718fa2darylm if (*message=='-') 5016a5fed65808adf648004b34f98718301d718fa2darylm fprintf(stderr,"%s: unrecognized option " LUA_QS "\n",progname,message); 5116a5fed65808adf648004b34f98718301d718fa2darylm else 5216a5fed65808adf648004b34f98718301d718fa2darylm fprintf(stderr,"%s: %s\n",progname,message); 5316a5fed65808adf648004b34f98718301d718fa2darylm fprintf(stderr, 5416a5fed65808adf648004b34f98718301d718fa2darylm "usage: %s [options] [filenames]\n" 5516a5fed65808adf648004b34f98718301d718fa2darylm "Available options are:\n" 5616a5fed65808adf648004b34f98718301d718fa2darylm " -l list (use -l -l for full listing)\n" 5716a5fed65808adf648004b34f98718301d718fa2darylm " -o name output to file " LUA_QL("name") " (default is \"%s\")\n" 5816a5fed65808adf648004b34f98718301d718fa2darylm " -p parse only\n" 5916a5fed65808adf648004b34f98718301d718fa2darylm " -s strip debug information\n" 6016a5fed65808adf648004b34f98718301d718fa2darylm " -v show version information\n" 6116a5fed65808adf648004b34f98718301d718fa2darylm " -- stop handling options\n" 6216a5fed65808adf648004b34f98718301d718fa2darylm " - stop handling options and process stdin\n" 6316a5fed65808adf648004b34f98718301d718fa2darylm ,progname,Output); 6416a5fed65808adf648004b34f98718301d718fa2darylm exit(EXIT_FAILURE); 6516a5fed65808adf648004b34f98718301d718fa2darylm} 6616a5fed65808adf648004b34f98718301d718fa2darylm 6716a5fed65808adf648004b34f98718301d718fa2darylm#define IS(s) (strcmp(argv[i],s)==0) 6816a5fed65808adf648004b34f98718301d718fa2darylm 6916a5fed65808adf648004b34f98718301d718fa2darylmstatic int doargs(int argc, char* argv[]) 7016a5fed65808adf648004b34f98718301d718fa2darylm{ 7116a5fed65808adf648004b34f98718301d718fa2darylm int i; 7216a5fed65808adf648004b34f98718301d718fa2darylm int version=0; 7316a5fed65808adf648004b34f98718301d718fa2darylm if (argv[0]!=NULL && *argv[0]!=0) progname=argv[0]; 7416a5fed65808adf648004b34f98718301d718fa2darylm for (i=1; i<argc; i++) 7516a5fed65808adf648004b34f98718301d718fa2darylm { 7616a5fed65808adf648004b34f98718301d718fa2darylm if (*argv[i]!='-') /* end of options; keep it */ 7716a5fed65808adf648004b34f98718301d718fa2darylm break; 7816a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("--")) /* end of options; skip it */ 7916a5fed65808adf648004b34f98718301d718fa2darylm { 8016a5fed65808adf648004b34f98718301d718fa2darylm ++i; 8116a5fed65808adf648004b34f98718301d718fa2darylm if (version) ++version; 8216a5fed65808adf648004b34f98718301d718fa2darylm break; 8316a5fed65808adf648004b34f98718301d718fa2darylm } 8416a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("-")) /* end of options; use stdin */ 8516a5fed65808adf648004b34f98718301d718fa2darylm break; 8616a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("-l")) /* list */ 8716a5fed65808adf648004b34f98718301d718fa2darylm ++listing; 8816a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("-o")) /* output file */ 8916a5fed65808adf648004b34f98718301d718fa2darylm { 9016a5fed65808adf648004b34f98718301d718fa2darylm output=argv[++i]; 9116a5fed65808adf648004b34f98718301d718fa2darylm if (output==NULL || *output==0 || (*output=='-' && output[1]!=0)) 9216a5fed65808adf648004b34f98718301d718fa2darylm usage(LUA_QL("-o") " needs argument"); 9316a5fed65808adf648004b34f98718301d718fa2darylm if (IS("-")) output=NULL; 9416a5fed65808adf648004b34f98718301d718fa2darylm } 9516a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("-p")) /* parse only */ 9616a5fed65808adf648004b34f98718301d718fa2darylm dumping=0; 9716a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("-s")) /* strip debug information */ 9816a5fed65808adf648004b34f98718301d718fa2darylm stripping=1; 9916a5fed65808adf648004b34f98718301d718fa2darylm else if (IS("-v")) /* show version */ 10016a5fed65808adf648004b34f98718301d718fa2darylm ++version; 10116a5fed65808adf648004b34f98718301d718fa2darylm else /* unknown option */ 10216a5fed65808adf648004b34f98718301d718fa2darylm usage(argv[i]); 10316a5fed65808adf648004b34f98718301d718fa2darylm } 10416a5fed65808adf648004b34f98718301d718fa2darylm if (i==argc && (listing || !dumping)) 10516a5fed65808adf648004b34f98718301d718fa2darylm { 10616a5fed65808adf648004b34f98718301d718fa2darylm dumping=0; 10716a5fed65808adf648004b34f98718301d718fa2darylm argv[--i]=Output; 10816a5fed65808adf648004b34f98718301d718fa2darylm } 10916a5fed65808adf648004b34f98718301d718fa2darylm if (version) 11016a5fed65808adf648004b34f98718301d718fa2darylm { 11116a5fed65808adf648004b34f98718301d718fa2darylm printf("%s\n",LUA_COPYRIGHT); 11216a5fed65808adf648004b34f98718301d718fa2darylm if (version==argc-1) exit(EXIT_SUCCESS); 11316a5fed65808adf648004b34f98718301d718fa2darylm } 11416a5fed65808adf648004b34f98718301d718fa2darylm return i; 11516a5fed65808adf648004b34f98718301d718fa2darylm} 11616a5fed65808adf648004b34f98718301d718fa2darylm 11716a5fed65808adf648004b34f98718301d718fa2darylm#define FUNCTION "(function()end)();" 11816a5fed65808adf648004b34f98718301d718fa2darylm 11916a5fed65808adf648004b34f98718301d718fa2darylmstatic const char* reader(lua_State *L, void *ud, size_t *size) 12016a5fed65808adf648004b34f98718301d718fa2darylm{ 12116a5fed65808adf648004b34f98718301d718fa2darylm UNUSED(L); 12216a5fed65808adf648004b34f98718301d718fa2darylm if ((*(int*)ud)--) 12316a5fed65808adf648004b34f98718301d718fa2darylm { 12416a5fed65808adf648004b34f98718301d718fa2darylm *size=sizeof(FUNCTION)-1; 12516a5fed65808adf648004b34f98718301d718fa2darylm return FUNCTION; 12616a5fed65808adf648004b34f98718301d718fa2darylm } 12716a5fed65808adf648004b34f98718301d718fa2darylm else 12816a5fed65808adf648004b34f98718301d718fa2darylm { 12916a5fed65808adf648004b34f98718301d718fa2darylm *size=0; 13016a5fed65808adf648004b34f98718301d718fa2darylm return NULL; 13116a5fed65808adf648004b34f98718301d718fa2darylm } 13216a5fed65808adf648004b34f98718301d718fa2darylm} 13316a5fed65808adf648004b34f98718301d718fa2darylm 13416a5fed65808adf648004b34f98718301d718fa2darylm#define toproto(L,i) getproto(L->top+(i)) 13516a5fed65808adf648004b34f98718301d718fa2darylm 13616a5fed65808adf648004b34f98718301d718fa2darylmstatic const Proto* combine(lua_State* L, int n) 13716a5fed65808adf648004b34f98718301d718fa2darylm{ 13816a5fed65808adf648004b34f98718301d718fa2darylm if (n==1) 13916a5fed65808adf648004b34f98718301d718fa2darylm return toproto(L,-1); 14016a5fed65808adf648004b34f98718301d718fa2darylm else 14116a5fed65808adf648004b34f98718301d718fa2darylm { 14216a5fed65808adf648004b34f98718301d718fa2darylm Proto* f; 14316a5fed65808adf648004b34f98718301d718fa2darylm int i=n; 14416a5fed65808adf648004b34f98718301d718fa2darylm if (lua_load(L,reader,&i,"=(" PROGNAME ")",NULL)!=LUA_OK) fatal(lua_tostring(L,-1)); 14516a5fed65808adf648004b34f98718301d718fa2darylm f=toproto(L,-1); 14616a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<n; i++) 14716a5fed65808adf648004b34f98718301d718fa2darylm { 14816a5fed65808adf648004b34f98718301d718fa2darylm f->p[i]=toproto(L,i-n-1); 14916a5fed65808adf648004b34f98718301d718fa2darylm if (f->p[i]->sizeupvalues>0) f->p[i]->upvalues[0].instack=0; 15016a5fed65808adf648004b34f98718301d718fa2darylm } 15116a5fed65808adf648004b34f98718301d718fa2darylm f->sizelineinfo=0; 15216a5fed65808adf648004b34f98718301d718fa2darylm return f; 15316a5fed65808adf648004b34f98718301d718fa2darylm } 15416a5fed65808adf648004b34f98718301d718fa2darylm} 15516a5fed65808adf648004b34f98718301d718fa2darylm 15616a5fed65808adf648004b34f98718301d718fa2darylmstatic int writer(lua_State* L, const void* p, size_t size, void* u) 15716a5fed65808adf648004b34f98718301d718fa2darylm{ 15816a5fed65808adf648004b34f98718301d718fa2darylm UNUSED(L); 15916a5fed65808adf648004b34f98718301d718fa2darylm return (fwrite(p,size,1,(FILE*)u)!=1) && (size!=0); 16016a5fed65808adf648004b34f98718301d718fa2darylm} 16116a5fed65808adf648004b34f98718301d718fa2darylm 16216a5fed65808adf648004b34f98718301d718fa2darylmstatic int pmain(lua_State* L) 16316a5fed65808adf648004b34f98718301d718fa2darylm{ 16416a5fed65808adf648004b34f98718301d718fa2darylm int argc=(int)lua_tointeger(L,1); 16516a5fed65808adf648004b34f98718301d718fa2darylm char** argv=(char**)lua_touserdata(L,2); 16616a5fed65808adf648004b34f98718301d718fa2darylm const Proto* f; 16716a5fed65808adf648004b34f98718301d718fa2darylm int i; 16816a5fed65808adf648004b34f98718301d718fa2darylm if (!lua_checkstack(L,argc)) fatal("too many input files"); 16916a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<argc; i++) 17016a5fed65808adf648004b34f98718301d718fa2darylm { 17116a5fed65808adf648004b34f98718301d718fa2darylm const char* filename=IS("-") ? NULL : argv[i]; 17216a5fed65808adf648004b34f98718301d718fa2darylm if (luaL_loadfile(L,filename)!=LUA_OK) fatal(lua_tostring(L,-1)); 17316a5fed65808adf648004b34f98718301d718fa2darylm } 17416a5fed65808adf648004b34f98718301d718fa2darylm f=combine(L,argc); 17516a5fed65808adf648004b34f98718301d718fa2darylm if (listing) luaU_print(f,listing>1); 17616a5fed65808adf648004b34f98718301d718fa2darylm if (dumping) 17716a5fed65808adf648004b34f98718301d718fa2darylm { 17816a5fed65808adf648004b34f98718301d718fa2darylm FILE* D= (output==NULL) ? stdout : fopen(output,"wb"); 17916a5fed65808adf648004b34f98718301d718fa2darylm if (D==NULL) cannot("open"); 18016a5fed65808adf648004b34f98718301d718fa2darylm lua_lock(L); 18116a5fed65808adf648004b34f98718301d718fa2darylm luaU_dump(L,f,writer,D,stripping); 18216a5fed65808adf648004b34f98718301d718fa2darylm lua_unlock(L); 18316a5fed65808adf648004b34f98718301d718fa2darylm if (ferror(D)) cannot("write"); 18416a5fed65808adf648004b34f98718301d718fa2darylm if (fclose(D)) cannot("close"); 18516a5fed65808adf648004b34f98718301d718fa2darylm } 18616a5fed65808adf648004b34f98718301d718fa2darylm return 0; 18716a5fed65808adf648004b34f98718301d718fa2darylm} 18816a5fed65808adf648004b34f98718301d718fa2darylm 18916a5fed65808adf648004b34f98718301d718fa2darylmint main(int argc, char* argv[]) 19016a5fed65808adf648004b34f98718301d718fa2darylm{ 19116a5fed65808adf648004b34f98718301d718fa2darylm lua_State* L; 19216a5fed65808adf648004b34f98718301d718fa2darylm int i=doargs(argc,argv); 19316a5fed65808adf648004b34f98718301d718fa2darylm argc-=i; argv+=i; 19416a5fed65808adf648004b34f98718301d718fa2darylm if (argc<=0) usage("no input files given"); 19516a5fed65808adf648004b34f98718301d718fa2darylm L=luaL_newstate(); 19616a5fed65808adf648004b34f98718301d718fa2darylm if (L==NULL) fatal("cannot create state: not enough memory"); 19716a5fed65808adf648004b34f98718301d718fa2darylm lua_pushcfunction(L,&pmain); 19816a5fed65808adf648004b34f98718301d718fa2darylm lua_pushinteger(L,argc); 19916a5fed65808adf648004b34f98718301d718fa2darylm lua_pushlightuserdata(L,argv); 20016a5fed65808adf648004b34f98718301d718fa2darylm if (lua_pcall(L,2,0,0)!=LUA_OK) fatal(lua_tostring(L,-1)); 20116a5fed65808adf648004b34f98718301d718fa2darylm lua_close(L); 20216a5fed65808adf648004b34f98718301d718fa2darylm return EXIT_SUCCESS; 20316a5fed65808adf648004b34f98718301d718fa2darylm} 20416a5fed65808adf648004b34f98718301d718fa2darylm 20516a5fed65808adf648004b34f98718301d718fa2darylm/* 20616a5fed65808adf648004b34f98718301d718fa2darylm** $Id: print.c,v 1.69 2013/07/04 01:03:46 lhf Exp $ 20716a5fed65808adf648004b34f98718301d718fa2darylm** print bytecodes 20816a5fed65808adf648004b34f98718301d718fa2darylm** See Copyright Notice in lua.h 20916a5fed65808adf648004b34f98718301d718fa2darylm*/ 21016a5fed65808adf648004b34f98718301d718fa2darylm 21116a5fed65808adf648004b34f98718301d718fa2darylm#include <ctype.h> 21216a5fed65808adf648004b34f98718301d718fa2darylm#include <stdio.h> 21316a5fed65808adf648004b34f98718301d718fa2darylm 21416a5fed65808adf648004b34f98718301d718fa2darylm#define luac_c 21516a5fed65808adf648004b34f98718301d718fa2darylm#define LUA_CORE 21616a5fed65808adf648004b34f98718301d718fa2darylm 21716a5fed65808adf648004b34f98718301d718fa2darylm#include "ldebug.h" 21816a5fed65808adf648004b34f98718301d718fa2darylm#include "lobject.h" 21916a5fed65808adf648004b34f98718301d718fa2darylm#include "lopcodes.h" 22016a5fed65808adf648004b34f98718301d718fa2darylm 22116a5fed65808adf648004b34f98718301d718fa2darylm#define VOID(p) ((const void*)(p)) 22216a5fed65808adf648004b34f98718301d718fa2darylm 22316a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintString(const TString* ts) 22416a5fed65808adf648004b34f98718301d718fa2darylm{ 22516a5fed65808adf648004b34f98718301d718fa2darylm const char* s=getstr(ts); 22616a5fed65808adf648004b34f98718301d718fa2darylm size_t i,n=ts->tsv.len; 22716a5fed65808adf648004b34f98718301d718fa2darylm printf("%c",'"'); 22816a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<n; i++) 22916a5fed65808adf648004b34f98718301d718fa2darylm { 23016a5fed65808adf648004b34f98718301d718fa2darylm int c=(int)(unsigned char)s[i]; 23116a5fed65808adf648004b34f98718301d718fa2darylm switch (c) 23216a5fed65808adf648004b34f98718301d718fa2darylm { 23316a5fed65808adf648004b34f98718301d718fa2darylm case '"': printf("\\\""); break; 23416a5fed65808adf648004b34f98718301d718fa2darylm case '\\': printf("\\\\"); break; 23516a5fed65808adf648004b34f98718301d718fa2darylm case '\a': printf("\\a"); break; 23616a5fed65808adf648004b34f98718301d718fa2darylm case '\b': printf("\\b"); break; 23716a5fed65808adf648004b34f98718301d718fa2darylm case '\f': printf("\\f"); break; 23816a5fed65808adf648004b34f98718301d718fa2darylm case '\n': printf("\\n"); break; 23916a5fed65808adf648004b34f98718301d718fa2darylm case '\r': printf("\\r"); break; 24016a5fed65808adf648004b34f98718301d718fa2darylm case '\t': printf("\\t"); break; 24116a5fed65808adf648004b34f98718301d718fa2darylm case '\v': printf("\\v"); break; 24216a5fed65808adf648004b34f98718301d718fa2darylm default: if (isprint(c)) 24316a5fed65808adf648004b34f98718301d718fa2darylm printf("%c",c); 24416a5fed65808adf648004b34f98718301d718fa2darylm else 24516a5fed65808adf648004b34f98718301d718fa2darylm printf("\\%03d",c); 24616a5fed65808adf648004b34f98718301d718fa2darylm } 24716a5fed65808adf648004b34f98718301d718fa2darylm } 24816a5fed65808adf648004b34f98718301d718fa2darylm printf("%c",'"'); 24916a5fed65808adf648004b34f98718301d718fa2darylm} 25016a5fed65808adf648004b34f98718301d718fa2darylm 25116a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintConstant(const Proto* f, int i) 25216a5fed65808adf648004b34f98718301d718fa2darylm{ 25316a5fed65808adf648004b34f98718301d718fa2darylm const TValue* o=&f->k[i]; 25416a5fed65808adf648004b34f98718301d718fa2darylm switch (ttypenv(o)) 25516a5fed65808adf648004b34f98718301d718fa2darylm { 25616a5fed65808adf648004b34f98718301d718fa2darylm case LUA_TNIL: 25716a5fed65808adf648004b34f98718301d718fa2darylm printf("nil"); 25816a5fed65808adf648004b34f98718301d718fa2darylm break; 25916a5fed65808adf648004b34f98718301d718fa2darylm case LUA_TBOOLEAN: 26016a5fed65808adf648004b34f98718301d718fa2darylm printf(bvalue(o) ? "true" : "false"); 26116a5fed65808adf648004b34f98718301d718fa2darylm break; 26216a5fed65808adf648004b34f98718301d718fa2darylm case LUA_TNUMBER: 26316a5fed65808adf648004b34f98718301d718fa2darylm printf(LUA_NUMBER_FMT,nvalue(o)); 26416a5fed65808adf648004b34f98718301d718fa2darylm break; 26516a5fed65808adf648004b34f98718301d718fa2darylm case LUA_TSTRING: 26616a5fed65808adf648004b34f98718301d718fa2darylm PrintString(rawtsvalue(o)); 26716a5fed65808adf648004b34f98718301d718fa2darylm break; 26816a5fed65808adf648004b34f98718301d718fa2darylm default: /* cannot happen */ 26916a5fed65808adf648004b34f98718301d718fa2darylm printf("? type=%d",ttype(o)); 27016a5fed65808adf648004b34f98718301d718fa2darylm break; 27116a5fed65808adf648004b34f98718301d718fa2darylm } 27216a5fed65808adf648004b34f98718301d718fa2darylm} 27316a5fed65808adf648004b34f98718301d718fa2darylm 27416a5fed65808adf648004b34f98718301d718fa2darylm#define UPVALNAME(x) ((f->upvalues[x].name) ? getstr(f->upvalues[x].name) : "-") 27516a5fed65808adf648004b34f98718301d718fa2darylm#define MYK(x) (-1-(x)) 27616a5fed65808adf648004b34f98718301d718fa2darylm 27716a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintCode(const Proto* f) 27816a5fed65808adf648004b34f98718301d718fa2darylm{ 27916a5fed65808adf648004b34f98718301d718fa2darylm const Instruction* code=f->code; 28016a5fed65808adf648004b34f98718301d718fa2darylm int pc,n=f->sizecode; 28116a5fed65808adf648004b34f98718301d718fa2darylm for (pc=0; pc<n; pc++) 28216a5fed65808adf648004b34f98718301d718fa2darylm { 28316a5fed65808adf648004b34f98718301d718fa2darylm Instruction i=code[pc]; 28416a5fed65808adf648004b34f98718301d718fa2darylm OpCode o=GET_OPCODE(i); 28516a5fed65808adf648004b34f98718301d718fa2darylm int a=GETARG_A(i); 28616a5fed65808adf648004b34f98718301d718fa2darylm int b=GETARG_B(i); 28716a5fed65808adf648004b34f98718301d718fa2darylm int c=GETARG_C(i); 28816a5fed65808adf648004b34f98718301d718fa2darylm int ax=GETARG_Ax(i); 28916a5fed65808adf648004b34f98718301d718fa2darylm int bx=GETARG_Bx(i); 29016a5fed65808adf648004b34f98718301d718fa2darylm int sbx=GETARG_sBx(i); 29116a5fed65808adf648004b34f98718301d718fa2darylm int line=getfuncline(f,pc); 29216a5fed65808adf648004b34f98718301d718fa2darylm printf("\t%d\t",pc+1); 29316a5fed65808adf648004b34f98718301d718fa2darylm if (line>0) printf("[%d]\t",line); else printf("[-]\t"); 29416a5fed65808adf648004b34f98718301d718fa2darylm printf("%-9s\t",luaP_opnames[o]); 29516a5fed65808adf648004b34f98718301d718fa2darylm switch (getOpMode(o)) 29616a5fed65808adf648004b34f98718301d718fa2darylm { 29716a5fed65808adf648004b34f98718301d718fa2darylm case iABC: 29816a5fed65808adf648004b34f98718301d718fa2darylm printf("%d",a); 29916a5fed65808adf648004b34f98718301d718fa2darylm if (getBMode(o)!=OpArgN) printf(" %d",ISK(b) ? (MYK(INDEXK(b))) : b); 30016a5fed65808adf648004b34f98718301d718fa2darylm if (getCMode(o)!=OpArgN) printf(" %d",ISK(c) ? (MYK(INDEXK(c))) : c); 30116a5fed65808adf648004b34f98718301d718fa2darylm break; 30216a5fed65808adf648004b34f98718301d718fa2darylm case iABx: 30316a5fed65808adf648004b34f98718301d718fa2darylm printf("%d",a); 30416a5fed65808adf648004b34f98718301d718fa2darylm if (getBMode(o)==OpArgK) printf(" %d",MYK(bx)); 30516a5fed65808adf648004b34f98718301d718fa2darylm if (getBMode(o)==OpArgU) printf(" %d",bx); 30616a5fed65808adf648004b34f98718301d718fa2darylm break; 30716a5fed65808adf648004b34f98718301d718fa2darylm case iAsBx: 30816a5fed65808adf648004b34f98718301d718fa2darylm printf("%d %d",a,sbx); 30916a5fed65808adf648004b34f98718301d718fa2darylm break; 31016a5fed65808adf648004b34f98718301d718fa2darylm case iAx: 31116a5fed65808adf648004b34f98718301d718fa2darylm printf("%d",MYK(ax)); 31216a5fed65808adf648004b34f98718301d718fa2darylm break; 31316a5fed65808adf648004b34f98718301d718fa2darylm } 31416a5fed65808adf648004b34f98718301d718fa2darylm switch (o) 31516a5fed65808adf648004b34f98718301d718fa2darylm { 31616a5fed65808adf648004b34f98718301d718fa2darylm case OP_LOADK: 31716a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; "); PrintConstant(f,bx); 31816a5fed65808adf648004b34f98718301d718fa2darylm break; 31916a5fed65808adf648004b34f98718301d718fa2darylm case OP_GETUPVAL: 32016a5fed65808adf648004b34f98718301d718fa2darylm case OP_SETUPVAL: 32116a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; %s",UPVALNAME(b)); 32216a5fed65808adf648004b34f98718301d718fa2darylm break; 32316a5fed65808adf648004b34f98718301d718fa2darylm case OP_GETTABUP: 32416a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; %s",UPVALNAME(b)); 32516a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(c)) { printf(" "); PrintConstant(f,INDEXK(c)); } 32616a5fed65808adf648004b34f98718301d718fa2darylm break; 32716a5fed65808adf648004b34f98718301d718fa2darylm case OP_SETTABUP: 32816a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; %s",UPVALNAME(a)); 32916a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(b)) { printf(" "); PrintConstant(f,INDEXK(b)); } 33016a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(c)) { printf(" "); PrintConstant(f,INDEXK(c)); } 33116a5fed65808adf648004b34f98718301d718fa2darylm break; 33216a5fed65808adf648004b34f98718301d718fa2darylm case OP_GETTABLE: 33316a5fed65808adf648004b34f98718301d718fa2darylm case OP_SELF: 33416a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(c)) { printf("\t; "); PrintConstant(f,INDEXK(c)); } 33516a5fed65808adf648004b34f98718301d718fa2darylm break; 33616a5fed65808adf648004b34f98718301d718fa2darylm case OP_SETTABLE: 33716a5fed65808adf648004b34f98718301d718fa2darylm case OP_ADD: 33816a5fed65808adf648004b34f98718301d718fa2darylm case OP_SUB: 33916a5fed65808adf648004b34f98718301d718fa2darylm case OP_MUL: 34016a5fed65808adf648004b34f98718301d718fa2darylm case OP_DIV: 34116a5fed65808adf648004b34f98718301d718fa2darylm case OP_POW: 34216a5fed65808adf648004b34f98718301d718fa2darylm case OP_EQ: 34316a5fed65808adf648004b34f98718301d718fa2darylm case OP_LT: 34416a5fed65808adf648004b34f98718301d718fa2darylm case OP_LE: 34516a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(b) || ISK(c)) 34616a5fed65808adf648004b34f98718301d718fa2darylm { 34716a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; "); 34816a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(b)) PrintConstant(f,INDEXK(b)); else printf("-"); 34916a5fed65808adf648004b34f98718301d718fa2darylm printf(" "); 35016a5fed65808adf648004b34f98718301d718fa2darylm if (ISK(c)) PrintConstant(f,INDEXK(c)); else printf("-"); 35116a5fed65808adf648004b34f98718301d718fa2darylm } 35216a5fed65808adf648004b34f98718301d718fa2darylm break; 35316a5fed65808adf648004b34f98718301d718fa2darylm case OP_JMP: 35416a5fed65808adf648004b34f98718301d718fa2darylm case OP_FORLOOP: 35516a5fed65808adf648004b34f98718301d718fa2darylm case OP_FORPREP: 35616a5fed65808adf648004b34f98718301d718fa2darylm case OP_TFORLOOP: 35716a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; to %d",sbx+pc+2); 35816a5fed65808adf648004b34f98718301d718fa2darylm break; 35916a5fed65808adf648004b34f98718301d718fa2darylm case OP_CLOSURE: 36016a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; %p",VOID(f->p[bx])); 36116a5fed65808adf648004b34f98718301d718fa2darylm break; 36216a5fed65808adf648004b34f98718301d718fa2darylm case OP_SETLIST: 36316a5fed65808adf648004b34f98718301d718fa2darylm if (c==0) printf("\t; %d",(int)code[++pc]); else printf("\t; %d",c); 36416a5fed65808adf648004b34f98718301d718fa2darylm break; 36516a5fed65808adf648004b34f98718301d718fa2darylm case OP_EXTRAARG: 36616a5fed65808adf648004b34f98718301d718fa2darylm printf("\t; "); PrintConstant(f,ax); 36716a5fed65808adf648004b34f98718301d718fa2darylm break; 36816a5fed65808adf648004b34f98718301d718fa2darylm default: 36916a5fed65808adf648004b34f98718301d718fa2darylm break; 37016a5fed65808adf648004b34f98718301d718fa2darylm } 37116a5fed65808adf648004b34f98718301d718fa2darylm printf("\n"); 37216a5fed65808adf648004b34f98718301d718fa2darylm } 37316a5fed65808adf648004b34f98718301d718fa2darylm} 37416a5fed65808adf648004b34f98718301d718fa2darylm 37516a5fed65808adf648004b34f98718301d718fa2darylm#define SS(x) ((x==1)?"":"s") 37616a5fed65808adf648004b34f98718301d718fa2darylm#define S(x) (int)(x),SS(x) 37716a5fed65808adf648004b34f98718301d718fa2darylm 37816a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintHeader(const Proto* f) 37916a5fed65808adf648004b34f98718301d718fa2darylm{ 38016a5fed65808adf648004b34f98718301d718fa2darylm const char* s=f->source ? getstr(f->source) : "=?"; 38116a5fed65808adf648004b34f98718301d718fa2darylm if (*s=='@' || *s=='=') 38216a5fed65808adf648004b34f98718301d718fa2darylm s++; 38316a5fed65808adf648004b34f98718301d718fa2darylm else if (*s==LUA_SIGNATURE[0]) 38416a5fed65808adf648004b34f98718301d718fa2darylm s="(bstring)"; 38516a5fed65808adf648004b34f98718301d718fa2darylm else 38616a5fed65808adf648004b34f98718301d718fa2darylm s="(string)"; 38716a5fed65808adf648004b34f98718301d718fa2darylm printf("\n%s <%s:%d,%d> (%d instruction%s at %p)\n", 38816a5fed65808adf648004b34f98718301d718fa2darylm (f->linedefined==0)?"main":"function",s, 38916a5fed65808adf648004b34f98718301d718fa2darylm f->linedefined,f->lastlinedefined, 39016a5fed65808adf648004b34f98718301d718fa2darylm S(f->sizecode),VOID(f)); 39116a5fed65808adf648004b34f98718301d718fa2darylm printf("%d%s param%s, %d slot%s, %d upvalue%s, ", 39216a5fed65808adf648004b34f98718301d718fa2darylm (int)(f->numparams),f->is_vararg?"+":"",SS(f->numparams), 39316a5fed65808adf648004b34f98718301d718fa2darylm S(f->maxstacksize),S(f->sizeupvalues)); 39416a5fed65808adf648004b34f98718301d718fa2darylm printf("%d local%s, %d constant%s, %d function%s\n", 39516a5fed65808adf648004b34f98718301d718fa2darylm S(f->sizelocvars),S(f->sizek),S(f->sizep)); 39616a5fed65808adf648004b34f98718301d718fa2darylm} 39716a5fed65808adf648004b34f98718301d718fa2darylm 39816a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintDebug(const Proto* f) 39916a5fed65808adf648004b34f98718301d718fa2darylm{ 40016a5fed65808adf648004b34f98718301d718fa2darylm int i,n; 40116a5fed65808adf648004b34f98718301d718fa2darylm n=f->sizek; 40216a5fed65808adf648004b34f98718301d718fa2darylm printf("constants (%d) for %p:\n",n,VOID(f)); 40316a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<n; i++) 40416a5fed65808adf648004b34f98718301d718fa2darylm { 40516a5fed65808adf648004b34f98718301d718fa2darylm printf("\t%d\t",i+1); 40616a5fed65808adf648004b34f98718301d718fa2darylm PrintConstant(f,i); 40716a5fed65808adf648004b34f98718301d718fa2darylm printf("\n"); 40816a5fed65808adf648004b34f98718301d718fa2darylm } 40916a5fed65808adf648004b34f98718301d718fa2darylm n=f->sizelocvars; 41016a5fed65808adf648004b34f98718301d718fa2darylm printf("locals (%d) for %p:\n",n,VOID(f)); 41116a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<n; i++) 41216a5fed65808adf648004b34f98718301d718fa2darylm { 41316a5fed65808adf648004b34f98718301d718fa2darylm printf("\t%d\t%s\t%d\t%d\n", 41416a5fed65808adf648004b34f98718301d718fa2darylm i,getstr(f->locvars[i].varname),f->locvars[i].startpc+1,f->locvars[i].endpc+1); 41516a5fed65808adf648004b34f98718301d718fa2darylm } 41616a5fed65808adf648004b34f98718301d718fa2darylm n=f->sizeupvalues; 41716a5fed65808adf648004b34f98718301d718fa2darylm printf("upvalues (%d) for %p:\n",n,VOID(f)); 41816a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<n; i++) 41916a5fed65808adf648004b34f98718301d718fa2darylm { 42016a5fed65808adf648004b34f98718301d718fa2darylm printf("\t%d\t%s\t%d\t%d\n", 42116a5fed65808adf648004b34f98718301d718fa2darylm i,UPVALNAME(i),f->upvalues[i].instack,f->upvalues[i].idx); 42216a5fed65808adf648004b34f98718301d718fa2darylm } 42316a5fed65808adf648004b34f98718301d718fa2darylm} 42416a5fed65808adf648004b34f98718301d718fa2darylm 42516a5fed65808adf648004b34f98718301d718fa2darylmstatic void PrintFunction(const Proto* f, int full) 42616a5fed65808adf648004b34f98718301d718fa2darylm{ 42716a5fed65808adf648004b34f98718301d718fa2darylm int i,n=f->sizep; 42816a5fed65808adf648004b34f98718301d718fa2darylm PrintHeader(f); 42916a5fed65808adf648004b34f98718301d718fa2darylm PrintCode(f); 43016a5fed65808adf648004b34f98718301d718fa2darylm if (full) PrintDebug(f); 43116a5fed65808adf648004b34f98718301d718fa2darylm for (i=0; i<n; i++) PrintFunction(f->p[i],full); 43216a5fed65808adf648004b34f98718301d718fa2darylm} 433