| /* |
| ** $Id: ldump.c,v 2.5 2005/05/05 20:47:02 roberto Exp roberto $ |
| ** save pre-compiled Lua chunks |
| ** See Copyright Notice in lua.h |
| */ |
| |
| #include <stddef.h> |
| |
| #define ldump_c |
| #define LUA_CORE |
| |
| #include "lua.h" |
| |
| #include "lobject.h" |
| #include "lopcodes.h" |
| #include "lstate.h" |
| #include "lundump.h" |
| |
| #define DumpVector(b,n,size,D) DumpBlock(b,(n)*(size),D) |
| #define DumpLiteral(s,D) DumpBlock("" s,(sizeof(s))-1,D) |
| |
| typedef struct { |
| lua_State* L; |
| lua_Writer writer; |
| void* data; |
| int strip; |
| int status; |
| } DumpState; |
| |
| static void DumpBlock(const void* b, size_t size, DumpState* D) |
| { |
| if (D->status==0) |
| { |
| lua_unlock(D->L); |
| D->status=(*D->writer)(D->L,b,size,D->data); |
| lua_lock(D->L); |
| } |
| } |
| |
| static void DumpByte(int y, DumpState* D) |
| { |
| char x=(char)y; |
| DumpBlock(&x,sizeof(x),D); |
| } |
| |
| static void DumpInt(int x, DumpState* D) |
| { |
| DumpBlock(&x,sizeof(x),D); |
| } |
| |
| static void DumpSize(size_t x, DumpState* D) |
| { |
| DumpBlock(&x,sizeof(x),D); |
| } |
| |
| static void DumpNumber(lua_Number x, DumpState* D) |
| { |
| DumpBlock(&x,sizeof(x),D); |
| } |
| |
| static void DumpString(const TString* s, DumpState* D) |
| { |
| if (s==NULL || getstr(s)==NULL) |
| DumpSize(0,D); |
| else |
| { |
| size_t size=s->tsv.len+1; /* include trailing '\0' */ |
| DumpSize(size,D); |
| DumpBlock(getstr(s),size,D); |
| } |
| } |
| |
| static void DumpCode(const Proto* f, DumpState* D) |
| { |
| DumpInt(f->sizecode,D); |
| DumpVector(f->code,f->sizecode,sizeof(*f->code),D); |
| } |
| |
| static void DumpLocals(const Proto* f, DumpState* D) |
| { |
| int i,n=f->sizelocvars; |
| DumpInt(n,D); |
| for (i=0; i<n; i++) |
| { |
| DumpString(f->locvars[i].varname,D); |
| DumpInt(f->locvars[i].startpc,D); |
| DumpInt(f->locvars[i].endpc,D); |
| } |
| } |
| |
| static void DumpLines(const Proto* f, DumpState* D) |
| { |
| DumpInt(f->sizelineinfo,D); |
| DumpVector(f->lineinfo,f->sizelineinfo,sizeof(*f->lineinfo),D); |
| } |
| |
| static void DumpUpvalues(const Proto* f, DumpState* D) |
| { |
| int i,n=f->sizeupvalues; |
| DumpInt(n,D); |
| for (i=0; i<n; i++) DumpString(f->upvalues[i],D); |
| } |
| |
| static void DumpFunction(const Proto* f, const TString* p, DumpState* D); |
| |
| static void DumpConstants(const Proto* f, DumpState* D) |
| { |
| int i,n; |
| DumpInt(n=f->sizek,D); |
| for (i=0; i<n; i++) |
| { |
| const TValue* o=&f->k[i]; |
| DumpByte(ttype(o),D); |
| switch (ttype(o)) |
| { |
| case LUA_TNUMBER: |
| DumpNumber(nvalue(o),D); |
| break; |
| case LUA_TSTRING: |
| DumpString(rawtsvalue(o),D); |
| break; |
| case LUA_TNIL: |
| break; |
| case LUA_TBOOLEAN: |
| DumpByte(bvalue(o),D); |
| break; |
| default: |
| lua_assert(0); /* cannot happen */ |
| break; |
| } |
| } |
| DumpInt(n=f->sizep,D); |
| for (i=0; i<n; i++) DumpFunction(f->p[i],f->source,D); |
| } |
| |
| static void DumpFunction(const Proto* f, const TString* p, DumpState* D) |
| { |
| DumpString((f->source==p) ? NULL : f->source,D); |
| DumpInt(f->linedefined,D); |
| DumpInt(f->lastlinedefined,D); |
| DumpByte(f->nups,D); |
| DumpByte(f->numparams,D); |
| DumpByte(f->is_vararg,D); |
| DumpByte(f->maxstacksize,D); |
| if (D->strip) DumpInt(0,D); else DumpLines(f,D); |
| if (D->strip) DumpInt(0,D); else DumpLocals(f,D); |
| if (D->strip) DumpInt(0,D); else DumpUpvalues(f,D); |
| DumpConstants(f,D); |
| DumpCode(f,D); |
| } |
| |
| static void DumpHeader(DumpState* D) |
| { |
| DumpLiteral(LUA_SIGNATURE,D); |
| DumpByte(VERSION,D); |
| DumpByte(luaU_endianness(),D); |
| DumpByte(sizeof(int),D); |
| DumpByte(sizeof(size_t),D); |
| DumpByte(sizeof(Instruction),D); |
| DumpByte(sizeof(lua_Number),D); |
| DumpNumber(TEST_NUMBER,D); |
| } |
| |
| /* |
| ** dump Lua function as precompiled chunk |
| */ |
| int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, |
| int strip) |
| { |
| DumpState D; |
| D.L=L; |
| D.writer=w; |
| D.data=data; |
| D.strip=strip; |
| D.status=0; |
| DumpHeader(&D); |
| DumpFunction(f,NULL,&D); |
| return D.status; |
| } |