modifications by lhf

This commit is contained in:
Roberto Ierusalimschy
1999-12-02 17:11:51 -02:00
parent b097076678
commit 72afb6debb
2 changed files with 104 additions and 100 deletions

178
lundump.c
View File

@@ -1,13 +1,14 @@
/* /*
** $Id: lundump.c,v 1.14 1999/09/06 13:55:09 roberto Exp roberto $ ** $Id: lundump.c,v 1.24 1999/12/02 18:45:03 lhf Exp $
** load bytecodes from files ** load bytecodes from files
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
#define LUA_REENTRANT
#include <stdio.h> #include <stdio.h>
#include <string.h> #include <string.h>
#define LUA_REENTRANT
#include "lauxlib.h" #include "lauxlib.h"
#include "lfunc.h" #include "lfunc.h"
#include "lmem.h" #include "lmem.h"
@@ -15,195 +16,183 @@
#include "lstring.h" #include "lstring.h"
#include "lundump.h" #include "lundump.h"
#define LoadBlock(L, b,size,Z) ezread(L, Z,b,size) #define LoadBlock(L,b,size,Z) ezread(L,Z,b,size)
static void unexpectedEOZ (lua_State *L, ZIO* Z) static void unexpectedEOZ (lua_State* L, ZIO* Z)
{ {
luaL_verror(L, "unexpected end of file in %s",zname(Z)); luaL_verror(L,"unexpected end of file in %.255s",zname(Z));
} }
static int ezgetc (lua_State *L, ZIO* Z) static int ezgetc (lua_State* L, ZIO* Z)
{ {
int c=zgetc(Z); int c=zgetc(Z);
if (c==EOZ) unexpectedEOZ(L, Z); if (c==EOZ) unexpectedEOZ(L,Z);
return c; return c;
} }
static void ezread (lua_State *L, ZIO* Z, void* b, int n) static void ezread (lua_State* L, ZIO* Z, void* b, int n)
{ {
int r=zread(Z,b,n); int r=zread(Z,b,n);
if (r!=0) unexpectedEOZ(L, Z); if (r!=0) unexpectedEOZ(L,Z);
} }
static unsigned int LoadWord (lua_State *L, ZIO* Z) static unsigned int LoadWord (lua_State* L, ZIO* Z)
{ {
unsigned int hi=ezgetc(L, Z); unsigned int hi=ezgetc(L,Z);
unsigned int lo=ezgetc(L, Z); unsigned int lo=ezgetc(L,Z);
return (hi<<8)|lo; return (hi<<8)|lo;
} }
static unsigned long LoadLong (lua_State *L, ZIO* Z) static unsigned long LoadLong (lua_State* L, ZIO* Z)
{ {
unsigned long hi=LoadWord(L, Z); unsigned long hi=LoadWord(L,Z);
unsigned long lo=LoadWord(L, Z); unsigned long lo=LoadWord(L,Z);
return (hi<<16)|lo; return (hi<<16)|lo;
} }
/* static real LoadNumber (lua_State* L, ZIO* Z, int native)
* convert number from text
*/
real luaU_str2d (lua_State *L, const char* b, const char* where)
{ {
real x;
if (!luaO_str2d(b, &x))
luaL_verror(L, "cannot convert number '%s' in %s",b,where);
return x;
}
static real LoadNumber (lua_State *L, ZIO* Z, int native)
{
real x;
if (native) if (native)
{ {
LoadBlock(L, &x,sizeof(x),Z); real x;
LoadBlock(L,&x,sizeof(x),Z);
return x; return x;
} }
else else
{ {
char b[256]; char b[256];
int size=ezgetc(L, Z); int size=ezgetc(L,Z);
LoadBlock(L, b,size,Z); LoadBlock(L,b,size,Z);
b[size]=0; b[size]=0;
return luaU_str2d(L, b,zname(Z)); return luaU_str2d(L,b,zname(Z));
} }
} }
static int LoadInt (lua_State *L, ZIO* Z, const char* message) static int LoadInt (lua_State* L, ZIO* Z, const char* message)
{ {
unsigned long l=LoadLong(L, Z); unsigned long l=LoadLong(L,Z);
unsigned int i=l; unsigned int i=l;
if (i!=l) luaL_verror(L, message,l,zname(Z)); if (i!=l) luaL_verror(L,message,l,zname(Z));
return i; return i;
} }
#define PAD 5 /* two word operands plus opcode */ #define PAD 5 /* two word operands plus opcode */
static Byte* LoadCode (lua_State *L, ZIO* Z) static Byte* LoadCode (lua_State* L, ZIO* Z)
{ {
int size=LoadInt(L, Z,"code too long (%ld bytes) in %s"); int size=LoadInt(L,Z,"code too long (%lu bytes) in %.255s");
Byte* b=luaM_malloc(L, size+PAD); Byte* b=luaM_malloc(L,size+PAD);
LoadBlock(L, b,size,Z); LoadBlock(L,b,size,Z);
if (b[size-1]!=ENDCODE) luaL_verror(L, "bad code in %s",zname(Z)); if (b[size-1]!=ENDCODE) luaL_verror(L,"bad code in %.255s",zname(Z));
memset(b+size,ENDCODE,PAD); /* pad code for safety */ memset(b+size,ENDCODE,PAD); /* pad code for safety */
return b; return b;
} }
static TaggedString* LoadTString (lua_State *L, ZIO* Z) static TaggedString* LoadTString (lua_State* L, ZIO* Z)
{ {
long size=LoadLong(L, Z); long size=LoadLong(L,Z);
if (size==0) if (size==0)
return NULL; return NULL;
else else
{ {
char* s=luaL_openspace(L, size); char* s=luaL_openspace(L,size);
LoadBlock(L, s,size,Z); LoadBlock(L,s,size,Z);
return luaS_newlstr(L, s,size-1); return luaS_newlstr(L,s,size-1);
} }
} }
static void LoadLocals (lua_State *L, TProtoFunc* tf, ZIO* Z) static void LoadLocals (lua_State* L, TProtoFunc* tf, ZIO* Z)
{ {
int i,n=LoadInt(L, Z,"too many locals (%ld) in %s"); int i,n=LoadInt(L,Z,"too many locals (%lu) in %.255s");
if (n==0) return; if (n==0) return;
tf->locvars=luaM_newvector(L, n+1,LocVar); tf->locvars=luaM_newvector(L,n+1,LocVar);
for (i=0; i<n; i++) for (i=0; i<n; i++)
{ {
tf->locvars[i].line=LoadInt(L, Z,"too many lines (%ld) in %s"); tf->locvars[i].line=LoadInt(L,Z,"too many lines (%lu) in %.255s");
tf->locvars[i].varname=LoadTString(L, Z); tf->locvars[i].varname=LoadTString(L,Z);
} }
tf->locvars[i].line=-1; /* flag end of vector */ tf->locvars[i].line=-1; /* flag end of vector */
tf->locvars[i].varname=NULL; tf->locvars[i].varname=NULL;
} }
static TProtoFunc* LoadFunction (lua_State *L, ZIO* Z, int native); static TProtoFunc* LoadFunction (lua_State* L, ZIO* Z, int native);
static void LoadConstants (lua_State *L, TProtoFunc* tf, ZIO* Z, int native) static void LoadConstants (lua_State* L, TProtoFunc* tf, ZIO* Z, int native)
{ {
int i,n=LoadInt(L, Z,"too many constants (%ld) in %s"); int i,n=LoadInt(L,Z,"too many constants (%lu) in %.255s");
tf->nconsts=n; tf->nconsts=n;
if (n==0) return; if (n==0) return;
tf->consts=luaM_newvector(L, n,TObject); tf->consts=luaM_newvector(L,n,TObject);
for (i=0; i<n; i++) for (i=0; i<n; i++)
{ {
TObject* o=tf->consts+i; TObject* o=tf->consts+i;
ttype(o)=-ezgetc(L, Z); /* ttype(o) is negative - ORDER LUA_T */ ttype(o)=-ezgetc(L,Z); /* ttype(o) is negative - ORDER LUA_T */
switch (ttype(o)) switch (ttype(o))
{ {
case LUA_T_NUMBER: case LUA_T_NUMBER:
nvalue(o)=LoadNumber(L, Z,native); nvalue(o)=LoadNumber(L,Z,native);
break; break;
case LUA_T_STRING: case LUA_T_STRING:
tsvalue(o)=LoadTString(L, Z); tsvalue(o)=LoadTString(L,Z);
break; break;
case LUA_T_PROTO: case LUA_T_PROTO:
tfvalue(o)=LoadFunction(L, Z,native); tfvalue(o)=LoadFunction(L,Z,native);
break; break;
case LUA_T_NIL: case LUA_T_NIL:
break; break;
default: /* cannot happen */ default: /* cannot happen */
luaU_badconstant(L, "load",i,o,tf); luaU_badconstant(L,"load",i,o,tf);
break; break;
} }
} }
} }
static TProtoFunc* LoadFunction (lua_State *L, ZIO* Z, int native) static TProtoFunc* LoadFunction (lua_State* L, ZIO* Z, int native)
{ {
TProtoFunc* tf=luaF_newproto(L); TProtoFunc* tf=luaF_newproto(L);
tf->lineDefined=LoadInt(L, Z,"lineDefined too large (%ld) in %s"); tf->lineDefined=LoadInt(L,Z,"lineDefined too large (%lu) in %.255s");
tf->source=LoadTString(L, Z); tf->source=LoadTString(L,Z);
if (tf->source==NULL) tf->source=luaS_new(L, zname(Z)); if (tf->source==NULL) tf->source=luaS_new(L,zname(Z));
tf->code=LoadCode(L, Z); tf->code=LoadCode(L,Z);
LoadLocals(L, tf,Z); LoadLocals(L,tf,Z);
LoadConstants(L, tf,Z,native); LoadConstants(L,tf,Z,native);
return tf; return tf;
} }
static void LoadSignature (lua_State *L, ZIO* Z) static void LoadSignature (lua_State* L, ZIO* Z)
{ {
const char* s=SIGNATURE; const char* s=SIGNATURE;
while (*s!=0 && ezgetc(L, Z)==*s) while (*s!=0 && ezgetc(L,Z)==*s)
++s; ++s;
if (*s!=0) luaL_verror(L, "bad signature in %s",zname(Z)); if (*s!=0) luaL_verror(L,"bad signature in %.255s",zname(Z));
} }
static int LoadHeader (lua_State *L, ZIO* Z) static int LoadHeader (lua_State* L, ZIO* Z)
{ {
int version,sizeofR; int version,sizeofR,native;
int native; LoadSignature(L,Z);
LoadSignature(L, Z); version=ezgetc(L,Z);
version=ezgetc(L, Z);
if (version>VERSION) if (version>VERSION)
luaL_verror(L, luaL_verror(L,
"%s too new: version=0x%02x; expected at most 0x%02x", "%.255s too new: version=0x%02x; expected at most 0x%02x",
zname(Z),version,VERSION); zname(Z),version,VERSION);
if (version<VERSION0) /* check last major change */ if (version<VERSION0) /* check last major change */
luaL_verror(L, luaL_verror(L,
"%s too old: version=0x%02x; expected at least 0x%02x", "%.255s too old: version=0x%02x; expected at least 0x%02x",
zname(Z),version,VERSION0); zname(Z),version,VERSION0);
sizeofR=ezgetc(L, Z); sizeofR=ezgetc(L,Z);
native=(sizeofR!=0); native=(sizeofR!=0);
if (native) /* test number representation */ if (native) /* test number representation */
{ {
if (sizeofR!=sizeof(real)) if (sizeofR!=sizeof(real))
luaL_verror(L, "unknown number size in %s: read %d; expected %d", luaL_verror(L,"unknown number size in %.255s: read %d; expected %d",
zname(Z),sizeofR,sizeof(real)); zname(Z),sizeofR,(int)sizeof(real));
else else
{ {
real tf=TEST_NUMBER; real tf=TEST_NUMBER;
real f=LoadNumber(L, Z,native); real f=LoadNumber(L,Z,native);
if ((long)f!=(long)tf) if ((long)f!=(long)tf)
luaL_verror(L, "unknown number format in %s: " luaL_verror(L,"unknown number format in %.255s: "
"read " NUMBER_FMT "; expected " NUMBER_FMT, "read " NUMBER_FMT "; expected " NUMBER_FMT,
zname(Z),f,tf); zname(Z),f,tf);
} }
@@ -211,31 +200,42 @@ static int LoadHeader (lua_State *L, ZIO* Z)
return native; return native;
} }
static TProtoFunc* LoadChunk (lua_State *L, ZIO* Z) static TProtoFunc* LoadChunk (lua_State* L, ZIO* Z)
{ {
return LoadFunction(L, Z,LoadHeader(L, Z)); return LoadFunction(L,Z,LoadHeader(L,Z));
} }
/* /*
** load one chunk from a file or buffer ** load one chunk from a file or buffer
** return main if ok and NULL at EOF ** return main if ok and NULL at EOF
*/ */
TProtoFunc* luaU_undump1 (lua_State *L, ZIO* Z) TProtoFunc* luaU_undump1 (lua_State* L, ZIO* Z)
{ {
int c=zgetc(Z); int c=zgetc(Z);
if (c==ID_CHUNK) if (c==ID_CHUNK)
return LoadChunk(L, Z); return LoadChunk(L,Z);
else if (c!=EOZ) else if (c!=EOZ)
luaL_verror(L, "%s is not a Lua binary file",zname(Z)); luaL_verror(L,"%.255s is not a Lua binary file",zname(Z));
return NULL; return NULL;
} }
/* /*
* handle constants that cannot happen * handle constants that cannot happen
*/ */
void luaU_badconstant (lua_State *L, const char* s, int i, const TObject* o, TProtoFunc* tf) void luaU_badconstant (lua_State* L, const char* s, int i, const TObject* o, const TProtoFunc* tf)
{ {
int t=ttype(o); int t=ttype(o);
const char* name= (t>0 || t<LUA_T_LINE) ? "?" : luaO_typenames[-t]; const char* name= (t>0 || t<LUA_T_LINE) ? "?" : luaO_typenames[-t];
luaL_verror(L, "cannot %s constant #%d: type=%d [%s]" IN,s,i,t,name,INLOC); luaL_verror(L,"cannot %.255s constant #%d: type=%d [%s]" IN,s,i,t,name,INLOC);
}
/*
* 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;
} }

View File

@@ -1,5 +1,5 @@
/* /*
** $Id: lundump.h,v 1.10 1999/08/16 20:52:00 roberto Exp roberto $ ** $Id: lundump.h,v 1.17 1999/12/02 18:51:09 lhf Exp $
** load pre-compiled Lua chunks ** load pre-compiled Lua chunks
** See Copyright Notice in lua.h ** See Copyright Notice in lua.h
*/ */
@@ -10,20 +10,24 @@
#include "lobject.h" #include "lobject.h"
#include "lzio.h" #include "lzio.h"
TProtoFunc* luaU_undump1 (lua_State *L, ZIO* Z); /* load one chunk */ /* load one chunk */
void luaU_badconstant (lua_State *L, const char* s, int i, const TObject* o, TProtoFunc* tf); TProtoFunc* luaU_undump1 (lua_State* L, ZIO* Z);
/* handle cases that cannot happen */
double luaU_str2d (lua_State *L, const char* b, const char* where); /* handle cases that cannot happen */
/* convert number from text */ void luaU_badconstant (lua_State* L, const char* s, int i,
const TObject* o, const TProtoFunc* tf);
/* convert number from text */
double luaU_str2d (lua_State* L, const char* b, const char* where);
/* definitions for headers of binary files */ /* definitions for headers of binary files */
#define VERSION 0x32 /* last format change was in 3.2 */ #define VERSION 0x33 /* last format change was in 3.3 */
#define VERSION0 0x32 /* last major change was in 3.2 */ #define VERSION0 0x33 /* last major change was in 3.3 */
#define ID_CHUNK 27 /* binary files start with ESC... */ #define ID_CHUNK 27 /* binary files start with ESC... */
#define SIGNATURE "Lua" /* ...followed by this signature */ #define SIGNATURE "Lua" /* ...followed by this signature */
/* formats for error messages */ /* formats for error messages */
#define SOURCE "<%s:%d>" #define SOURCE "<%.255s:%d>"
#define IN " in %p " SOURCE #define IN " in %p " SOURCE
#define INLOC tf,tf->source->str,tf->lineDefined #define INLOC tf,tf->source->str,tf->lineDefined