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
26 #define PROGNAME "luac" /* default program name */
27 #define OUTPUT PROGNAME ".out" /* default output file */
29 static int listing
=0; /* list bytecodes? */
30 static int dumping
=1; /* dump bytecodes? */
31 static int stripping
=0; /* strip debug information? */
32 static char Output
[]={ OUTPUT
}; /* default output file name */
33 static const char* output
=Output
; /* actual output file name */
34 static const char* progname
=PROGNAME
; /* actual program name */
36 static void fatal(const char* message
)
38 fprintf(stderr
,"%s: %s\n",progname
,message
);
42 static void cannot(const char* what
)
44 fprintf(stderr
,"%s: cannot %s %s: %s\n",progname
,what
,output
,strerror(errno
));
48 static void usage(const char* message
)
51 fprintf(stderr
,"%s: unrecognized option " LUA_QS
"\n",progname
,message
);
53 fprintf(stderr
,"%s: %s\n",progname
,message
);
55 "usage: %s [options] [filenames].\n"
56 "Available options are:\n"
59 " -o name output to file " LUA_QL("name") " (default is \"%s\")\n"
61 " -s strip debug information\n"
62 " -v show version information\n"
63 " -- stop handling options\n",
68 #define IS(s) (strcmp(argv[i],s)==0)
70 static int doargs(int argc
, char* argv
[])
73 if (argv
[0]!=NULL
&& *argv
[0]!=0) progname
=argv
[0];
74 for (i
=1; i
<argc
; i
++)
76 if (*argv
[i
]!='-') /* end of options; keep it */
78 else if (IS("--")) /* end of options; skip it */
83 else if (IS("-")) /* end of options; use stdin */
85 else if (IS("-l")) /* list */
87 else if (IS("-o")) /* output file */
90 if (output
==NULL
|| *output
==0) usage(LUA_QL("-o") " needs argument");
91 if (IS("-")) output
=NULL
;
93 else if (IS("-p")) /* parse only */
95 else if (IS("-s")) /* strip debug information */
97 else if (IS("-v")) /* show version */
99 printf("%s %s\n",LUA_VERSION
,LUA_COPYRIGHT
);
100 if (argc
==2) exit(EXIT_SUCCESS
);
102 else /* unknown option */
105 if (i
==argc
&& (listing
|| !dumping
))
113 #define toproto(L,i) (clvalue(L->top+(i))->l.p)
115 static Proto
* combine(lua_State
* L
, int n
)
118 return toproto(L
,-1);
122 Proto
* f
=luaF_newproto(L
);
123 setptvalue2s(L
,L
->top
,f
); incr_top(L
);
124 f
->source
=luaS_newliteral(L
,"=(" PROGNAME
")");
127 f
->code
=luaM_newvector(L
,pc
,Instruction
);
129 f
->p
=luaM_newvector(L
,n
,Proto
*);
134 f
->p
[i
]=toproto(L
,i
-n
-1);
135 f
->code
[pc
++]=CREATE_ABx(OP_CLOSURE
,0,i
);
136 f
->code
[pc
++]=CREATE_ABC(OP_CALL
,0,1,1);
138 f
->code
[pc
++]=CREATE_ABC(OP_RETURN
,0,1,0);
143 static int writer(lua_State
* L
, const void* p
, size_t size
, void* u
)
146 return (fwrite(p
,size
,1,(FILE*)u
)!=1) && (size
!=0);
154 static int pmain(lua_State
* L
)
156 struct Smain
* s
= (struct Smain
*)lua_touserdata(L
, 1);
161 if (!lua_checkstack(L
,argc
)) fatal("too many input files");
162 for (i
=0; i
<argc
; i
++)
164 const char* filename
=IS("-") ? NULL
: argv
[i
];
165 if (luaL_loadfile(L
,filename
)!=0) fatal(lua_tostring(L
,-1));
168 if (listing
) luaU_print(f
,listing
>1);
171 FILE* D
= (output
==NULL
) ? stdout
: fopen(output
,"wb");
172 if (D
==NULL
) cannot("open");
174 luaU_dump(L
,f
,writer
,D
,stripping
);
176 if (ferror(D
)) cannot("write");
177 if (fclose(D
)) cannot("close");
182 int main(int argc
, char* argv
[])
186 int i
=doargs(argc
,argv
);
188 if (argc
<=0) usage("no input files given");
190 if (L
==NULL
) fatal("not enough memory for state");
193 if (lua_cpcall(L
,pmain
,&s
)!=0) fatal(lua_tostring(L
,-1));