Subversion Repositories WoWGM

Rev

Details | Last modification | View Log | RSS feed

Rev Author Line No. Line
3 tristanc 1 
/*
2 
** $Id: ldump.c,v 1.15 2006/02/16 15:53:49 lhf 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 || getstr(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,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, const TString* p, 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(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 
   default:
102 
	lua_assert(0);			/* cannot happen */
103 
	break;
104 
  }
105 
 }
106 
 n=f->sizep;
107 
 DumpInt(n,D);
108 
 for (i=0; i<n; i++) DumpFunction(f->p[i],f->source,D);
109 
}
110 
 
111 
static void DumpDebug(const Proto* f, DumpState* D)
112 
{
113 
 int i,n;
114 
 n= (D->strip) ? 0 : f->sizelineinfo;
115 
 DumpVector(f->lineinfo,n,sizeof(int),D);
116 
 n= (D->strip) ? 0 : f->sizelocvars;
117 
 DumpInt(n,D);
118 
 for (i=0; i<n; i++)
119 
 {
120 
  DumpString(f->locvars[i].varname,D);
121 
  DumpInt(f->locvars[i].startpc,D);
122 
  DumpInt(f->locvars[i].endpc,D);
123 
 }
124 
 n= (D->strip) ? 0 : f->sizeupvalues;
125 
 DumpInt(n,D);
126 
 for (i=0; i<n; i++) DumpString(f->upvalues[i],D);
127 
}
128 
 
129 
static void DumpFunction(const Proto* f, const TString* p, DumpState* D)
130 
{
131 
 DumpString((f->source==p || D->strip) ? NULL : f->source,D);
132 
 DumpInt(f->linedefined,D);
133 
 DumpInt(f->lastlinedefined,D);
134 
 DumpChar(f->nups,D);
135 
 DumpChar(f->numparams,D);
136 
 DumpChar(f->is_vararg,D);
137 
 DumpChar(f->maxstacksize,D);
138 
 DumpCode(f,D);
139 
 DumpConstants(f,D);
140 
 DumpDebug(f,D);
141 
}
142 
 
143 
static void DumpHeader(DumpState* D)
144 
{
145 
 char h[LUAC_HEADERSIZE];
146 
 luaU_header(h);
147 
 DumpBlock(h,LUAC_HEADERSIZE,D);
148 
}
149 
 
150 
/*
151 
** dump Lua function as precompiled chunk
152 
*/
153 
int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip)
154 
{
155 
 DumpState D;
156 
 D.L=L;
157 
 D.writer=w;
158 
 D.data=data;
159 
 D.strip=strip;
160 
 D.status=0;
161 
 DumpHeader(&D);
162 
 DumpFunction(f,NULL,&D);
163 
 return D.status;
164 
}