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
DumpBlock(const void * b,size_t size,DumpState * D)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
DumpChar(int y,DumpState * D)39 static void DumpChar(int y, DumpState* D)
40 {
41 char x=(char)y;
42 DumpVar(x,D);
43 }
44
DumpInt(int x,DumpState * D)45 static void DumpInt(int x, DumpState* D)
46 {
47 DumpVar(x,D);
48 }
49
DumpNumber(lua_Number x,DumpState * D)50 static void DumpNumber(lua_Number x, DumpState* D)
51 {
52 DumpVar(x,D);
53 }
54
DumpVector(const void * b,int n,size_t size,DumpState * D)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
DumpString(const TString * s,DumpState * D)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
DumpConstants(const Proto * f,DumpState * D)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
DumpUpvalues(const Proto * f,DumpState * D)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
DumpDebug(const Proto * f,DumpState * D)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
DumpFunction(const Proto * f,DumpState * D)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
DumpHeader(DumpState * D)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 */
luaU_dump(lua_State * L,const Proto * f,lua_Writer w,void * data,int strip)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 }
174