/*
** $Id: ldump.c,v 2.17.1.1 2013/04/12 18:48:47 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;
}