Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
5205 | clevermous | 1 | /* |
2 | ** $Id: ldump.c,v 1.19 2011/11/23 17:48:18 lhf Exp $ |
||
3 | ** save precompiled Lua chunks |
||
4 | ** See Copyright Notice in lua.h |
||
5 | */ |
||
6 | |||
7 | #include |
||
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 |
||
85 | { |
||
86 | const TValue* o=&f->k[i]; |
||
87 | DumpChar(ttype(o),D); |
||
88 | switch (ttype(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 | } |
||
102 | } |
||
103 | n=f->sizep; |
||
104 | DumpInt(n,D); |
||
105 | for (i=0; i |
||
106 | } |
||
107 | |||
108 | static void DumpUpvalues(const Proto* f, DumpState* D) |
||
109 | { |
||
110 | int i,n=f->sizeupvalues; |
||
111 | DumpInt(n,D); |
||
112 | for (i=0; i |
||
113 | { |
||
114 | DumpChar(f->upvalues[i].instack,D); |
||
115 | DumpChar(f->upvalues[i].idx,D); |
||
116 | } |
||
117 | } |
||
118 | |||
119 | static void DumpDebug(const Proto* f, DumpState* D) |
||
120 | { |
||
121 | int i,n; |
||
122 | DumpString((D->strip) ? NULL : f->source,D); |
||
123 | n= (D->strip) ? 0 : f->sizelineinfo; |
||
124 | DumpVector(f->lineinfo,n,sizeof(int),D); |
||
125 | n= (D->strip) ? 0 : f->sizelocvars; |
||
126 | DumpInt(n,D); |
||
127 | for (i=0; i |
||
128 | { |
||
129 | DumpString(f->locvars[i].varname,D); |
||
130 | DumpInt(f->locvars[i].startpc,D); |
||
131 | DumpInt(f->locvars[i].endpc,D); |
||
132 | } |
||
133 | n= (D->strip) ? 0 : f->sizeupvalues; |
||
134 | DumpInt(n,D); |
||
135 | for (i=0; i |
||
136 | } |
||
137 | |||
138 | static void DumpFunction(const Proto* f, DumpState* D) |
||
139 | { |
||
140 | DumpInt(f->linedefined,D); |
||
141 | DumpInt(f->lastlinedefined,D); |
||
142 | DumpChar(f->numparams,D); |
||
143 | DumpChar(f->is_vararg,D); |
||
144 | DumpChar(f->maxstacksize,D); |
||
145 | DumpCode(f,D); |
||
146 | DumpConstants(f,D); |
||
147 | DumpUpvalues(f,D); |
||
148 | DumpDebug(f,D); |
||
149 | } |
||
150 | |||
151 | static void DumpHeader(DumpState* D) |
||
152 | { |
||
153 | lu_byte h[LUAC_HEADERSIZE]; |
||
154 | luaU_header(h); |
||
155 | DumpBlock(h,LUAC_HEADERSIZE,D); |
||
156 | } |
||
157 | |||
158 | /* |
||
159 | ** dump Lua function as precompiled chunk |
||
160 | */ |
||
161 | int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip) |
||
162 | { |
||
163 | DumpState D; |
||
164 | D.L=L; |
||
165 | D.writer=w; |
||
166 | D.data=data; |
||
167 | D.strip=strip; |
||
168 | D.status=0; |
||
169 | DumpHeader(&D); |
||
170 | DumpFunction(f,&D); |
||
171 | return D.status; |
||
172 | } |