1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
/*
** $Id: luacomp.c,v 1.5 2004-11-27 21:35:20 pixel Exp $
** Lua compiler (saves bytecodes to files; also list bytecodes)
** Highly hacked by Nicolas "Pixel" Noble to be transformed into a
** small form-factor LUA compiler.
** See Copyright Notice in lua.h
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "lua.h"
#include "lauxlib.h"
#include "lfunc.h"
#include "lmem.h"
#include "lobject.h"
#include "lopcodes.h"
#include "lstring.h"
#include "lundump.h"
#ifndef PROGNAME
#define PROGNAME "luacomp" /* program name */
#endif
const char * progname = PROGNAME; /* actual program name */
static Proto * toproto(lua_State * L, int i) {
const Closure * c = (const Closure *) lua_topointer(L, i);
return c->l.p;
}
static Proto * combine(lua_State * L, int n) {
if (n == 1)
return toproto(L, -1);
else {
int i, pc = 0;
Proto * f = luaF_newproto(L);
f->source = luaS_newliteral(L, "=(" PROGNAME ")");
f->maxstacksize = 1;
f->p = luaM_newvector(L, n, Proto *);
f->sizep = n;
f->sizecode = 2 * n + 1;
f->code = luaM_newvector(L, f->sizecode, Instruction);
for (i = 0; i < n; i++) {
f->p[i] = toproto(L, i - n);
f->code[pc++] = CREATE_ABx(OP_CLOSURE, 0, i);
f->code[pc++] = CREATE_ABC(OP_CALL, 0, 1, 1);
}
f->code[pc++] = CREATE_ABC(OP_RETURN, 0, 1, 0);
return f;
}
}
static void strip(lua_State * L, Proto * f) {
int i, n = f->sizep;
luaM_freearray(L, f->lineinfo, f->sizelineinfo, int);
luaM_freearray(L, f->locvars, f->sizelocvars, struct LocVar);
luaM_freearray(L, f->upvalues, f->sizeupvalues, TString *);
f->lineinfo = NULL;
f->sizelineinfo = 0;
f->locvars = NULL;
f->sizelocvars = 0;
f->upvalues = NULL;
f->sizeupvalues = 0;
f->source = luaS_newliteral(L, "=(none)");
for (i = 0; i < n; i++)
strip(L, f->p[i]);
}
void luacmain(lua_State * L, int stripping, lua_Chunkwriter w, void *uD) {
Proto * f;
f = combine(L, lua_gettop(L));
if (stripping)
strip(L, f);
luaU_dump(L, f, w, uD);
}
|
