luac.c
上传用户:yisoukefu
上传日期:2020-08-09
资源大小:39506k
文件大小:4k
源码类别:

其他游戏

开发平台:

Visual C++

  1. /*
  2. ** $Id: luac.c,v 1.52 2005/11/11 14:03:13 lhf Exp $
  3. ** Lua compiler (saves bytecodes to files; also list bytecodes)
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <errno.h>
  7. #include <stdio.h>
  8. #include <stdlib.h>
  9. #include <string.h>
  10. #define luac_c
  11. #define LUA_CORE
  12. #include "lua.h"
  13. #include "lauxlib.h"
  14. #include "ldo.h"
  15. #include "lfunc.h"
  16. #include "lmem.h"
  17. #include "lobject.h"
  18. #include "lopcodes.h"
  19. #include "lstring.h"
  20. #include "lundump.h"
  21. #define PROGNAME "luac" /* default program name */
  22. #define OUTPUT PROGNAME ".out" /* default output file */
  23. static int listing=0; /* list bytecodes? */
  24. static int dumping=1; /* dump bytecodes? */
  25. static int stripping=0; /* strip debug information? */
  26. static char Output[]={ OUTPUT }; /* default output file name */
  27. static const char* output=Output; /* actual output file name */
  28. static const char* progname=PROGNAME; /* actual program name */
  29. static void fatal(const char* message)
  30. {
  31.  fprintf(stderr,"%s: %sn",progname,message);
  32.  exit(EXIT_FAILURE);
  33. }
  34. static void cannot(const char* what)
  35. {
  36.  fprintf(stderr,"%s: cannot %s %s: %sn",progname,what,output,strerror(errno));
  37.  exit(EXIT_FAILURE);
  38. }
  39. static void usage(const char* message)
  40. {
  41.  if (*message=='-')
  42.   fprintf(stderr,"%s: unrecognized option " LUA_QS "n",progname,message);
  43.  else
  44.   fprintf(stderr,"%s: %sn",progname,message);
  45.  fprintf(stderr,
  46.  "usage: %s [options] [filenames].n"
  47.  "Available options are:n"
  48.  "  -        process stdinn"
  49.  "  -l       listn"
  50.  "  -o name  output to file " LUA_QL("name") " (default is "%s")n"
  51.  "  -p       parse onlyn"
  52.  "  -s       strip debug informationn"
  53.  "  -v       show version informationn"
  54.  "  --       stop handling optionsn",
  55.  progname,Output);
  56.  exit(EXIT_FAILURE);
  57. }
  58. #define IS(s) (strcmp(argv[i],s)==0)
  59. static int doargs(int argc, char* argv[])
  60. {
  61.  int i;
  62.  if (argv[0]!=NULL && *argv[0]!=0) progname=argv[0];
  63.  for (i=1; i<argc; i++)
  64.  {
  65.   if (*argv[i]!='-') /* end of options; keep it */
  66.    break;
  67.   else if (IS("--")) /* end of options; skip it */
  68.   {
  69.    ++i;
  70.    break;
  71.   }
  72.   else if (IS("-")) /* end of options; use stdin */
  73.    break;
  74.   else if (IS("-l")) /* list */
  75.    ++listing;
  76.   else if (IS("-o")) /* output file */
  77.   {
  78.    output=argv[++i];
  79.    if (output==NULL || *output==0) usage(LUA_QL("-o") " needs argument");
  80.    if (IS("-")) output=NULL;
  81.   }
  82.   else if (IS("-p")) /* parse only */
  83.    dumping=0;
  84.   else if (IS("-s")) /* strip debug information */
  85.    stripping=1;
  86.   else if (IS("-v")) /* show version */
  87.   {
  88.    printf("%s  %sn",LUA_VERSION,LUA_COPYRIGHT);
  89.    if (argc==2) exit(EXIT_SUCCESS);
  90.   }
  91.   else /* unknown option */
  92.    usage(argv[i]);
  93.  }
  94.  if (i==argc && (listing || !dumping))
  95.  {
  96.   dumping=0;
  97.   argv[--i]=Output;
  98.  }
  99.  return i;
  100. }
  101. #define toproto(L,i) (clvalue(L->top+(i))->l.p)
  102. static Proto* combine(lua_State* L, int n)
  103. {
  104.  if (n==1)
  105.   return toproto(L,-1);
  106.  else
  107.  {
  108.   int i,pc;
  109.   Proto* f=luaF_newproto(L);
  110.   setptvalue2s(L,L->top,f); incr_top(L);
  111.   f->source=luaS_newliteral(L,"=(" PROGNAME ")");
  112.   f->maxstacksize=1;
  113.   pc=2*n+1;
  114.   f->code=luaM_newvector(L,pc,Instruction);
  115.   f->sizecode=pc;
  116.   f->p=luaM_newvector(L,n,Proto*);
  117.   f->sizep=n;
  118.   pc=0;
  119.   for (i=0; i<n; i++)
  120.   {
  121.    f->p[i]=toproto(L,i-n-1);
  122.    f->code[pc++]=CREATE_ABx(OP_CLOSURE,0,i);
  123.    f->code[pc++]=CREATE_ABC(OP_CALL,0,1,1);
  124.   }
  125.   f->code[pc++]=CREATE_ABC(OP_RETURN,0,1,0);
  126.   return f;
  127.  }
  128. }
  129. static int writer(lua_State* L, const void* p, size_t size, void* u)
  130. {
  131.  UNUSED(L);
  132.  return (fwrite(p,size,1,(FILE*)u)!=1) && (size!=0);
  133. }
  134. struct Smain {
  135.  int argc;
  136.  char** argv;
  137. };
  138. static int pmain(lua_State* L)
  139. {
  140.  struct Smain* s = (struct Smain*)lua_touserdata(L, 1);
  141.  int argc=s->argc;
  142.  char** argv=s->argv;
  143.  Proto* f;
  144.  int i;
  145.  if (!lua_checkstack(L,argc)) fatal("too many input files");
  146.  for (i=0; i<argc; i++)
  147.  {
  148.   const char* filename=IS("-") ? NULL : argv[i];
  149.   if (luaL_loadfile(L,filename)!=0) fatal(lua_tostring(L,-1));
  150.  }
  151.  f=combine(L,argc);
  152.  if (listing) luaU_print(f,listing>1);
  153.  if (dumping)
  154.  {
  155.   FILE* D= (output==NULL) ? stdout : fopen(output,"wb");
  156.   if (D==NULL) cannot("open");
  157.   lua_lock(L);
  158.   luaU_dump(L,f,writer,D,stripping);
  159.   lua_unlock(L);
  160.   if (ferror(D)) cannot("write");
  161.   if (fclose(D)) cannot("close");
  162.  }
  163.  return 0;
  164. }
  165. int main(int argc, char* argv[])
  166. {
  167.  lua_State* L;
  168.  struct Smain s;
  169.  int i=doargs(argc,argv);
  170.  argc-=i; argv+=i;
  171.  if (argc<=0) usage("no input files given");
  172.  L=lua_open();
  173.  if (L==NULL) fatal("not enough memory for state");
  174.  s.argc=argc;
  175.  s.argv=argv;
  176.  if (lua_cpcall(L,pmain,&s)!=0) fatal(lua_tostring(L,-1));
  177.  lua_close(L);
  178.  return EXIT_SUCCESS;
  179. }