diff --git a/lundump.c b/lundump.c index 3dd01ba4..f9d90168 100644 --- a/lundump.c +++ b/lundump.c @@ -1,5 +1,5 @@ /* -** $Id: lundump.c,v 1.18 2000/03/03 14:58:26 roberto Exp roberto $ +** $Id: lundump.c,v 1.28 2000/04/24 19:32:58 lhf Exp $ ** load bytecodes from files ** See Copyright Notice in lua.h */ @@ -17,10 +17,17 @@ #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)); + luaL_verror(L,"unexpected end of file in `%.255s'",ZNAME(Z)); } static int ezgetc (lua_State* L, ZIO* Z) @@ -56,25 +63,18 @@ static Number LoadNumber (lua_State* L, ZIO* Z) int size=ezgetc(L,Z); LoadBlock(L,b,size,Z); 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) { unsigned long l=LoadLong(L,Z); unsigned int i=l; - if (i!=l) luaL_verror(L,message,l,zname(Z)); + if (i!=l) luaL_verror(L,"%s in `%.255s';\n" + " read %lu; expected %u",message,ZNAME(Z),l,-1); return i; } -static void LoadCode (lua_State* L, Proto* tf, ZIO* Z) -{ - int size=LoadInt(L,Z,"code too long (%lu bytes) in %.255s"); - tf->code=luaM_newvector(L,size,Instruction); - LoadBlock(L,tf->code,size*sizeof(tf->code[0]),Z); - if (tf->code[size-1]!=OP_END) luaL_verror(L,"bad code in %.255s",zname(Z)); -} - static TString* LoadString (lua_State* L, ZIO* Z) { long size=LoadLong(L,Z); @@ -88,14 +88,55 @@ static TString* LoadString (lua_State* L, ZIO* Z) } } +static void SwapCode (lua_State* L, Instruction* code, int size, ZIO* Z) +{ + unsigned char* p; + int 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 (%lu) in %.255s"); + int i,n=LoadInt(L,Z,"too many locals"); if (n==0) return; tf->locvars=luaM_newvector(L,n+1,LocVar); for (i=0; ilocvars[i].line=LoadInt(L,Z,"too many lines (%lu) in %.255s"); + tf->locvars[i].line=LoadInt(L,Z,"too many lines"); tf->locvars[i].varname=LoadString(L,Z); } tf->locvars[i].line=-1; /* flag end of vector */ @@ -107,22 +148,28 @@ 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 (%lu) in %.255s"); + tf->nkstr=n=LoadInt(L,Z,"too many strings"); if (n>0) { tf->kstr=luaM_newvector(L,n,TString*); - for (i=0; ikstr[i]=LoadString(L,Z); + for (i=0; ikstr[i]=s; + } } - tf->nknum=n=LoadInt(L,Z,"too many numbers (%lu) in %.255s"); + 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[0]),Z); + LoadBlock(L,tf->knum,n*sizeof(*tf->knum),Z); else for (i=0; iknum[i]=LoadNumber(L,Z); } - tf->nkproto=n=LoadInt(L,Z,"too many functions (%lu) in %.255s"); + tf->nkproto=n=LoadInt(L,Z,"too many functions"); if (n>0) { tf->kproto=luaM_newvector(L,n,Proto*); @@ -133,12 +180,12 @@ static void LoadConstants (lua_State* L, Proto* tf, ZIO* Z, int native) static Proto* LoadFunction (lua_State* L, ZIO* Z, int native) { Proto* tf=luaF_newproto(L); - tf->lineDefined=LoadInt(L,Z,"lineDefined too large (%lu) in %.255s"); tf->source=LoadString(L,Z); if (tf->source==NULL) tf->source=luaS_new(L,zname(Z)); - tf->numparams=LoadInt(L,Z,"numparams too large (%lu) in %.255s"); - tf->is_vararg=LoadInt(L,Z,"is_vararg too large (%lu) in %.255s"); - tf->maxstacksize=LoadInt(L,Z,"maxstacksize too large (%lu) in %.255s"); + 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); @@ -150,7 +197,7 @@ 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)); + if (*s!=0) luaL_verror(L,"bad signature in `%.255s'",ZNAME(Z)); } #define V(v) v/16,v%16 @@ -161,28 +208,40 @@ static int LoadHeader (lua_State* L, ZIO* Z) LoadSignature(L,Z); version=ezgetc(L,Z); if (version>VERSION) - luaL_verror(L, - "%.255s too new: its version is %d.%d; expected at most %d.%d", - zname(Z),V(version),V(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