cvs.zerfleddert.de Git - proxmark3-svn/blame_incremental

... / ...

Commit	Line	Data
	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
	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)
	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
	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
	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
	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
	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
	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	*/
	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	}