|
1866
|
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 }
|
