174 lines
3.1 KiB
C
174 lines
3.1 KiB
C
/*
|
|
** $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;
|
|
}
|