/*** $Id: ldump.c,v 2.17 2012/01/23 23:02:10 roberto Exp $** save precompiled Lua chunks** See Copyright Notice in lua.h*/ #include <stddef.h> #define ldump_c#define LUA_CORE #include "lua.h" #include "lobject.h"#include "lstate.h"#include "lundump.h" typedef struct { lua_State* L; lua_Writer writer; void* data; int strip; int status;} DumpState; #define DumpMem(b,n,size,D)	DumpBlock(b,(n)*(size),D)#define DumpVar(x,D)		DumpMem(&x,1,sizeof(x),D) 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 DumpChar(int y, DumpState* D){ char x=(char)y; DumpVar(x,D);} static void DumpInt(int x, DumpState* D){ DumpVar(x,D);} static void DumpNumber(lua_Number x, DumpState* D){ DumpVar(x,D);} static void DumpVector(const void* b, int n, size_t size, DumpState* D){ DumpInt(n,D); DumpMem(b,n,size,D);} static void DumpString(const TString* s, DumpState* D){ if (s==NULL) {  size_t size=0;  DumpVar(size,D); } else {  size_t size=s->tsv.len+1;		/* include trailing '\0' */  DumpVar(size,D);  DumpBlock(getstr(s),size*sizeof(char),D); }} #define DumpCode(f,D)	 DumpVector(f->code,f->sizecode,sizeof(Instruction),D) static void DumpFunction(const Proto* f, DumpState* D); static void DumpConstants(const Proto* f, DumpState* D){ int i,n=f->sizek; DumpInt(n,D); for (i=0; i<n; i++) {  const TValue* o=&f->k[i];  DumpChar(ttypenv(o),D);  switch (ttypenv(o))  {   case LUA_TNIL:	break;   case LUA_TBOOLEAN:	DumpChar(bvalue(o),D);	break;   case LUA_TNUMBER:	DumpNumber(nvalue(o),D);	break;   case LUA_TSTRING:	DumpString(rawtsvalue(o),D);	break;    default: lua_assert(0);  } } n=f->sizep; DumpInt(n,D); for (i=0; i<n; i++) DumpFunction(f->p[i],D);} static void DumpUpvalues(const Proto* f, DumpState* D){ int i,n=f->sizeupvalues; DumpInt(n,D); for (i=0; i<n; i++) {  DumpChar(f->upvalues[i].instack,D);  DumpChar(f->upvalues[i].idx,D); }} static void DumpDebug(const Proto* f, DumpState* D){ int i,n; DumpString((D->strip) ? NULL : f->source,D); n= (D->strip) ? 0 : f->sizelineinfo; DumpVector(f->lineinfo,n,sizeof(int),D); n= (D->strip) ? 0 : 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); } n= (D->strip) ? 0 : f->sizeupvalues; DumpInt(n,D); for (i=0; i<n; i++) DumpString(f->upvalues[i].name,D);} static void DumpFunction(const Proto* f, DumpState* D){ DumpInt(f->linedefined,D); DumpInt(f->lastlinedefined,D); DumpChar(f->numparams,D); DumpChar(f->is_vararg,D); DumpChar(f->maxstacksize,D); DumpCode(f,D); DumpConstants(f,D); DumpUpvalues(f,D); DumpDebug(f,D);} static void DumpHeader(DumpState* D){ lu_byte h[LUAC_HEADERSIZE]; luaU_header(h); DumpBlock(h,LUAC_HEADERSIZE,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,&D); return D.status;}