diff options
author | Roberto Ierusalimschy <roberto@inf.puc-rio.br> | 2013-04-12 15:48:47 -0300 |
---|---|---|
committer | Roberto Ierusalimschy <roberto@inf.puc-rio.br> | 2013-04-12 15:48:47 -0300 |
commit | 2ca518141350667e38941eac35b76980095d2fdc (patch) | |
tree | 82595f0dfd57df5782905621298c8aae68b46389 /ldump.c | |
parent | f5ae26ec6c2eb69f03f316e59c1e1977c52c7c23 (diff) | |
download | lua-2ca518141350667e38941eac35b76980095d2fdc.tar.gz lua-2ca518141350667e38941eac35b76980095d2fdc.tar.bz2 lua-2ca518141350667e38941eac35b76980095d2fdc.zip |
Branch 5.2 - new releases for 5.2 go from here, main trunk goes
for next version
Diffstat (limited to 'ldump.c')
-rw-r--r-- | ldump.c | 173 |
1 files changed, 173 insertions, 0 deletions
diff --git a/ldump.c b/ldump.c new file mode 100644 index 00000000..d5e6a47c --- /dev/null +++ b/ldump.c | |||
@@ -0,0 +1,173 @@ | |||
1 | /* | ||
2 | ** $Id: ldump.c,v 2.17 2012/01/23 23:02:10 roberto Exp $ | ||
3 | ** save precompiled Lua chunks | ||
4 | ** See Copyright Notice in lua.h | ||
5 | */ | ||
6 | |||
7 | #include <stddef.h> | ||
8 | |||
9 | #define ldump_c | ||
10 | #define LUA_CORE | ||
11 | |||
12 | #include "lua.h" | ||
13 | |||
14 | #include "lobject.h" | ||
15 | #include "lstate.h" | ||
16 | #include "lundump.h" | ||
17 | |||
18 | typedef struct { | ||
19 | lua_State* L; | ||
20 | lua_Writer writer; | ||
21 | void* data; | ||
22 | int strip; | ||
23 | int status; | ||
24 | } DumpState; | ||
25 | |||
26 | #define DumpMem(b,n,size,D) DumpBlock(b,(n)*(size),D) | ||
27 | #define DumpVar(x,D) DumpMem(&x,1,sizeof(x),D) | ||
28 | |||
29 | static void DumpBlock(const void* b, size_t size, DumpState* D) | ||
30 | { | ||
31 | if (D->status==0) | ||
32 | { | ||
33 | lua_unlock(D->L); | ||
34 | D->status=(*D->writer)(D->L,b,size,D->data); | ||
35 | lua_lock(D->L); | ||
36 | } | ||
37 | } | ||
38 | |||
39 | static void DumpChar(int y, DumpState* D) | ||
40 | { | ||
41 | char x=(char)y; | ||
42 | DumpVar(x,D); | ||
43 | } | ||
44 | |||
45 | static void DumpInt(int x, DumpState* D) | ||
46 | { | ||
47 | DumpVar(x,D); | ||
48 | } | ||
49 | |||
50 | static void DumpNumber(lua_Number x, DumpState* D) | ||
51 | { | ||
52 | DumpVar(x,D); | ||
53 | } | ||
54 | |||
55 | static void DumpVector(const void* b, int n, size_t size, DumpState* D) | ||
56 | { | ||
57 | DumpInt(n,D); | ||
58 | DumpMem(b,n,size,D); | ||
59 | } | ||
60 | |||
61 | static void DumpString(const TString* s, DumpState* D) | ||
62 | { | ||
63 | if (s==NULL) | ||
64 | { | ||
65 | size_t size=0; | ||
66 | DumpVar(size,D); | ||
67 | } | ||
68 | else | ||
69 | { | ||
70 | size_t size=s->tsv.len+1; /* include trailing '\0' */ | ||
71 | DumpVar(size,D); | ||
72 | DumpBlock(getstr(s),size*sizeof(char),D); | ||
73 | } | ||
74 | } | ||
75 | |||
76 | #define DumpCode(f,D) DumpVector(f->code,f->sizecode,sizeof(Instruction),D) | ||
77 | |||
78 | static void DumpFunction(const Proto* f, DumpState* D); | ||
79 | |||
80 | static void DumpConstants(const Proto* f, DumpState* D) | ||
81 | { | ||
82 | int i,n=f->sizek; | ||
83 | DumpInt(n,D); | ||
84 | for (i=0; i<n; i++) | ||
85 | { | ||
86 | const TValue* o=&f->k[i]; | ||
87 | DumpChar(ttypenv(o),D); | ||
88 | switch (ttypenv(o)) | ||
89 | { | ||
90 | case LUA_TNIL: | ||
91 | break; | ||
92 | case LUA_TBOOLEAN: | ||
93 | DumpChar(bvalue(o),D); | ||
94 | break; | ||
95 | case LUA_TNUMBER: | ||
96 | DumpNumber(nvalue(o),D); | ||
97 | break; | ||
98 | case LUA_TSTRING: | ||
99 | DumpString(rawtsvalue(o),D); | ||
100 | break; | ||
101 | default: lua_assert(0); | ||
102 | } | ||
103 | } | ||
104 | n=f->sizep; | ||
105 | DumpInt(n,D); | ||
106 | for (i=0; i<n; i++) DumpFunction(f->p[i],D); | ||
107 | } | ||
108 | |||
109 | static void DumpUpvalues(const Proto* f, DumpState* D) | ||
110 | { | ||
111 | int i,n=f->sizeupvalues; | ||
112 | DumpInt(n,D); | ||
113 | for (i=0; i<n; i++) | ||
114 | { | ||
115 | DumpChar(f->upvalues[i].instack,D); | ||
116 | DumpChar(f->upvalues[i].idx,D); | ||
117 | } | ||
118 | } | ||
119 | |||
120 | static void DumpDebug(const Proto* f, DumpState* D) | ||
121 | { | ||
122 | int i,n; | ||
123 | DumpString((D->strip) ? NULL : f->source,D); | ||
124 | n= (D->strip) ? 0 : f->sizelineinfo; | ||
125 | DumpVector(f->lineinfo,n,sizeof(int),D); | ||
126 | n= (D->strip) ? 0 : f->sizelocvars; | ||
127 | DumpInt(n,D); | ||
128 | for (i=0; i<n; i++) | ||
129 | { | ||
130 | DumpString(f->locvars[i].varname,D); | ||
131 | DumpInt(f->locvars[i].startpc,D); | ||
132 | DumpInt(f->locvars[i].endpc,D); | ||
133 | } | ||
134 | n= (D->strip) ? 0 : f->sizeupvalues; | ||
135 | DumpInt(n,D); | ||
136 | for (i=0; i<n; i++) DumpString(f->upvalues[i].name,D); | ||
137 | } | ||
138 | |||
139 | static void DumpFunction(const Proto* f, DumpState* D) | ||
140 | { | ||
141 | DumpInt(f->linedefined,D); | ||
142 | DumpInt(f->lastlinedefined,D); | ||
143 | DumpChar(f->numparams,D); | ||
144 | DumpChar(f->is_vararg,D); | ||
145 | DumpChar(f->maxstacksize,D); | ||
146 | DumpCode(f,D); | ||
147 | DumpConstants(f,D); | ||
148 | DumpUpvalues(f,D); | ||
149 | DumpDebug(f,D); | ||
150 | } | ||
151 | |||
152 | static void DumpHeader(DumpState* D) | ||
153 | { | ||
154 | lu_byte h[LUAC_HEADERSIZE]; | ||
155 | luaU_header(h); | ||
156 | DumpBlock(h,LUAC_HEADERSIZE,D); | ||
157 | } | ||
158 | |||
159 | /* | ||
160 | ** dump Lua function as precompiled chunk | ||
161 | */ | ||
162 | int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip) | ||
163 | { | ||
164 | DumpState D; | ||
165 | D.L=L; | ||
166 | D.writer=w; | ||
167 | D.data=data; | ||
168 | D.strip=strip; | ||
169 | D.status=0; | ||
170 | DumpHeader(&D); | ||
171 | DumpFunction(f,&D); | ||
172 | return D.status; | ||
173 | } | ||