src/luac/luac.c

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