mirror of
https://github.com/lua/lua
synced 2024-11-22 21:01:26 +03:00
280 lines
6.5 KiB
C
280 lines
6.5 KiB
C
/*
|
|
** $Id: lundump.c,v 1.22 2000/06/12 13:52:05 roberto Exp roberto $
|
|
** load bytecodes from files
|
|
** See Copyright Notice in lua.h
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#define LUA_REENTRANT
|
|
|
|
#include "lua.h"
|
|
|
|
#include "lauxlib.h"
|
|
#include "lfunc.h"
|
|
#include "lmem.h"
|
|
#include "lopcodes.h"
|
|
#include "lstring.h"
|
|
#include "lundump.h"
|
|
|
|
#define LoadBlock(L,b,size,Z) ezread(L,Z,b,size)
|
|
#define LoadByte ezgetc
|
|
|
|
static const char* ZNAME(ZIO* Z)
|
|
{
|
|
const char* s=zname(Z);
|
|
return (*s=='@') ? s+1 : s;
|
|
}
|
|
|
|
static void unexpectedEOZ (lua_State* L, ZIO* Z)
|
|
{
|
|
luaL_verror(L,"unexpected end of file in `%.255s'",ZNAME(Z));
|
|
}
|
|
|
|
static int ezgetc (lua_State* L, ZIO* Z)
|
|
{
|
|
int c=zgetc(Z);
|
|
if (c==EOZ) unexpectedEOZ(L,Z);
|
|
return c;
|
|
}
|
|
|
|
static void ezread (lua_State* L, ZIO* Z, void* b, int n)
|
|
{
|
|
int r=zread(Z,b,n);
|
|
if (r!=0) unexpectedEOZ(L,Z);
|
|
}
|
|
|
|
static unsigned int LoadWord (lua_State* L, ZIO* Z)
|
|
{
|
|
unsigned int hi=ezgetc(L,Z);
|
|
unsigned int lo=ezgetc(L,Z);
|
|
return (hi<<8)|lo;
|
|
}
|
|
|
|
static unsigned long LoadLong (lua_State* L, ZIO* Z)
|
|
{
|
|
unsigned long hi=LoadWord(L,Z);
|
|
unsigned long lo=LoadWord(L,Z);
|
|
return (hi<<16)|lo;
|
|
}
|
|
|
|
static Number LoadNumber (lua_State* L, ZIO* Z)
|
|
{
|
|
char b[256];
|
|
int size=ezgetc(L,Z);
|
|
LoadBlock(L,b,size,Z);
|
|
b[size]=0;
|
|
return luaU_str2d(L,b,ZNAME(Z));
|
|
}
|
|
|
|
static int LoadInt (lua_State* L, ZIO* Z, const char* message)
|
|
{
|
|
unsigned long l=LoadLong(L,Z);
|
|
unsigned int i=l;
|
|
if (i!=l) luaL_verror(L,"%s in `%.255s';\n"
|
|
" read %lu; expected %u",message,ZNAME(Z),l,-1);
|
|
return i;
|
|
}
|
|
|
|
static TString* LoadString (lua_State* L, ZIO* Z)
|
|
{
|
|
long size=LoadLong(L,Z);
|
|
if (size==0)
|
|
return NULL;
|
|
else
|
|
{
|
|
char* s=luaL_openspace(L,size);
|
|
LoadBlock(L,s,size,Z);
|
|
return luaS_newlstr(L,s,size-1); /* remove trailing '\0' */
|
|
}
|
|
}
|
|
|
|
static void SwapCode (lua_State* L, Instruction* code, int size, ZIO* Z)
|
|
{
|
|
unsigned char* p;
|
|
unsigned char c;
|
|
if (sizeof(Instruction)==4)
|
|
while (size--)
|
|
{
|
|
p=(unsigned char *) code++;
|
|
c=p[0]; p[0]=p[3]; p[3]=c;
|
|
c=p[1]; p[1]=p[2]; p[2]=c;
|
|
}
|
|
else if (sizeof(Instruction)==2)
|
|
while (size--)
|
|
{
|
|
p=(unsigned char *) code++;
|
|
c=p[0]; p[0]=p[1]; p[1]=c;
|
|
}
|
|
else
|
|
luaL_verror(L,"cannot swap code with %d-byte instructions in `%.255s'",
|
|
(int)sizeof(Instruction),ZNAME(Z));
|
|
}
|
|
|
|
static void LoadCode (lua_State* L, Proto* tf, ZIO* Z)
|
|
{
|
|
int size=LoadInt(L,Z,"code too long");
|
|
Instruction t=0,tt=TEST_CODE;
|
|
tf->code=luaM_newvector(L,size,Instruction);
|
|
LoadBlock(L,tf->code,size*sizeof(*tf->code),Z);
|
|
if (tf->code[size-1]!=OP_END) luaL_verror(L,"bad code in `%.255s'",ZNAME(Z));
|
|
LoadBlock(L,&t,sizeof(t),Z);
|
|
if (t!=tt)
|
|
{
|
|
Instruction ot=t;
|
|
SwapCode(L,&t,1,Z);
|
|
if (t!=tt) luaL_verror(L,"cannot swap code in `%.255s';\n"
|
|
" read 0x%08X; swapped to 0x%08X; expected 0x%08X",
|
|
ZNAME(Z),(unsigned long)ot,(unsigned long)t,(unsigned long)tt);
|
|
SwapCode(L,tf->code,size,Z);
|
|
}
|
|
}
|
|
|
|
static void LoadLocals (lua_State* L, Proto* tf, ZIO* Z)
|
|
{
|
|
int i,n=LoadInt(L,Z,"too many locals");
|
|
if (n==0) return;
|
|
tf->locvars=luaM_newvector(L,n+1,LocVar);
|
|
for (i=0; i<n; i++)
|
|
{
|
|
tf->locvars[i].pc=LoadInt(L,Z,"too many lines");
|
|
tf->locvars[i].varname=LoadString(L,Z);
|
|
}
|
|
tf->locvars[i].pc=-1; /* flag end of vector */
|
|
tf->locvars[i].varname=NULL;
|
|
}
|
|
|
|
static Proto* LoadFunction (lua_State* L, ZIO* Z, int native);
|
|
|
|
static void LoadConstants (lua_State* L, Proto* tf, ZIO* Z, int native)
|
|
{
|
|
int i,n;
|
|
tf->nkstr=n=LoadInt(L,Z,"too many strings");
|
|
if (n>0)
|
|
{
|
|
tf->kstr=luaM_newvector(L,n,TString*);
|
|
for (i=0; i<n; i++)
|
|
{
|
|
TString* s=LoadString(L,Z);
|
|
LoadByte(L,Z);
|
|
tf->kstr[i]=s;
|
|
}
|
|
}
|
|
tf->nknum=n=LoadInt(L,Z,"too many numbers");
|
|
if (n>0)
|
|
{
|
|
tf->knum=luaM_newvector(L,n,Number);
|
|
if (native)
|
|
LoadBlock(L,tf->knum,n*sizeof(*tf->knum),Z);
|
|
else
|
|
for (i=0; i<n; i++) tf->knum[i]=LoadNumber(L,Z);
|
|
}
|
|
tf->nkproto=n=LoadInt(L,Z,"too many functions");
|
|
if (n>0)
|
|
{
|
|
tf->kproto=luaM_newvector(L,n,Proto*);
|
|
for (i=0; i<n; i++) tf->kproto[i]=LoadFunction(L,Z,native);
|
|
}
|
|
}
|
|
|
|
static Proto* LoadFunction (lua_State* L, ZIO* Z, int native)
|
|
{
|
|
Proto* tf=luaF_newproto(L);
|
|
tf->source=LoadString(L,Z);
|
|
if (tf->source==NULL) tf->source=luaS_new(L,zname(Z));
|
|
tf->lineDefined=LoadInt(L,Z,"lineDefined too large");
|
|
tf->numparams=LoadInt(L,Z,"too many parameters");
|
|
tf->is_vararg=LoadByte(L,Z);
|
|
tf->maxstacksize=LoadInt(L,Z,"too much stack needed");
|
|
LoadCode(L,tf,Z);
|
|
LoadLocals(L,tf,Z);
|
|
LoadConstants(L,tf,Z,native);
|
|
return tf;
|
|
}
|
|
|
|
static void LoadSignature (lua_State* L, ZIO* Z)
|
|
{
|
|
const char* s=SIGNATURE;
|
|
while (*s!=0 && ezgetc(L,Z)==*s)
|
|
++s;
|
|
if (*s!=0) luaL_verror(L,"bad signature in `%.255s'",ZNAME(Z));
|
|
}
|
|
|
|
#define V(v) v/16,v%16
|
|
|
|
static int LoadHeader (lua_State* L, ZIO* Z)
|
|
{
|
|
int version,sizeofR,native;
|
|
LoadSignature(L,Z);
|
|
version=ezgetc(L,Z);
|
|
if (version>VERSION)
|
|
luaL_verror(L,"`%.255s' too new:\n"
|
|
" read version %d.%d; expected at most %d.%d",
|
|
ZNAME(Z),V(version),V(VERSION));
|
|
if (version<VERSION0) /* check last major change */
|
|
luaL_verror(L,"`%.255s' too old:\n"
|
|
" read version %d.%d; expected at least %d.%d",
|
|
ZNAME(Z),V(version),V(VERSION));
|
|
{
|
|
int I=ezgetc(L,Z);
|
|
int i=ezgetc(L,Z);
|
|
int op=ezgetc(L,Z);
|
|
int b=ezgetc(L,Z);
|
|
if (I!=sizeof(Instruction) || i!=SIZE_INSTRUCTION || op!=SIZE_OP || b!=SIZE_B)
|
|
luaL_verror(L,"virtual machine mismatch in `%.255s':\n"
|
|
" read %d-bit,%d-byte instructions, %d-bit opcodes, %d-bit b-arguments;\n"
|
|
" expected %d-bit,%d-byte instructions, %d-bit opcodes, %d-bit b-arguments",
|
|
ZNAME(Z),i,I,op,b,SIZE_INSTRUCTION,(int)sizeof(Instruction),SIZE_OP,SIZE_B);
|
|
}
|
|
sizeofR=ezgetc(L,Z);
|
|
native=(sizeofR!=0);
|
|
if (native) /* test number representation */
|
|
{
|
|
if (sizeofR!=sizeof(Number))
|
|
luaL_verror(L,"native number size mismatch in `%.255s':\n"
|
|
" read %d; expected %d",
|
|
ZNAME(Z),sizeofR,(int)sizeof(Number));
|
|
else
|
|
{
|
|
Number f=0,tf=TEST_NUMBER;
|
|
LoadBlock(L,&f,sizeof(f),Z);
|
|
if ((long)f!=(long)tf) /* disregard errors in last bit of fraction */
|
|
luaL_verror(L,"unknown number format in `%.255s':\n"
|
|
" read " NUMBER_FMT "; expected " NUMBER_FMT,
|
|
ZNAME(Z),f,tf);
|
|
}
|
|
}
|
|
return native;
|
|
}
|
|
|
|
static Proto* LoadChunk (lua_State* L, ZIO* Z)
|
|
{
|
|
return LoadFunction(L,Z,LoadHeader(L,Z));
|
|
}
|
|
|
|
/*
|
|
** load one chunk from a file or buffer
|
|
** return main if ok and NULL at EOF
|
|
*/
|
|
Proto* luaU_undump1 (lua_State* L, ZIO* Z)
|
|
{
|
|
int c=zgetc(Z);
|
|
if (c==ID_CHUNK)
|
|
return LoadChunk(L,Z);
|
|
else if (c!=EOZ)
|
|
luaL_verror(L,"`%.255s' is not a Lua binary file",ZNAME(Z));
|
|
return NULL;
|
|
}
|
|
|
|
/*
|
|
* convert number from text
|
|
*/
|
|
double luaU_str2d (lua_State* L, const char* b, const char* where)
|
|
{
|
|
double x;
|
|
if (!luaO_str2d(b,&x))
|
|
luaL_verror(L,"cannot convert number `%.255s' in `%.255s'",b,where);
|
|
return x;
|
|
}
|