mirror of https://github.com/lua/lua
149 lines
2.6 KiB
C
149 lines
2.6 KiB
C
/*
|
|
** undump.c
|
|
** load bytecodes from files
|
|
*/
|
|
|
|
char *rcs_undump="$Id: undump.c,v 1.3 1996/02/24 03:46:57 lhf Exp lhf $";
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include "luac.h"
|
|
|
|
static void warn(char *s) /* TODO: remove */
|
|
{
|
|
fprintf(stderr,"luac: %s\n",s);
|
|
}
|
|
|
|
static void panic(char *s) /* TODO: remove */
|
|
{
|
|
warn(s);
|
|
exit(1);
|
|
}
|
|
|
|
static void Unthread(Byte *p, int i, int v)
|
|
{
|
|
while (i!=0)
|
|
{
|
|
CodeWord c;
|
|
Byte *q=p+i;
|
|
get_word(c,q);
|
|
q=p+i;
|
|
i=c.w;
|
|
c.w=v;
|
|
q[0]=c.m.c1;
|
|
q[1]=c.m.c2;
|
|
}
|
|
}
|
|
|
|
static int LoadWord(FILE *D)
|
|
{
|
|
Word w;
|
|
fread(&w,sizeof(w),1,D);
|
|
return w;
|
|
}
|
|
|
|
static char* LoadBlock(int size, FILE *D)
|
|
{
|
|
char *b=luaI_malloc(size);
|
|
fread(b,size,1,D);
|
|
return b;
|
|
}
|
|
|
|
static char* LoadString(FILE *D)
|
|
{
|
|
return LoadBlock(LoadWord(D),D);
|
|
}
|
|
|
|
static TFunc *Main=NULL;
|
|
static TFunc *lastF=NULL;
|
|
|
|
static void LoadFunction(FILE *D)
|
|
{
|
|
TFunc *tf=new(TFunc);
|
|
tf->size=LoadWord(D);
|
|
tf->marked=LoadWord(D);
|
|
tf->lineDefined=LoadWord(D);
|
|
tf->fileName=LoadString(D);
|
|
tf->code=LoadBlock(tf->size,D);
|
|
tf->next=NULL;
|
|
if (tf->lineDefined==0) /* new main */
|
|
Main=lastF=tf;
|
|
else /* fix PUSHFUNCTION */
|
|
{
|
|
CodeCode c;
|
|
Byte *p=Main->code+tf->marked; /* TODO: tf->marked=? */
|
|
c.tf=tf;
|
|
*p++=c.m.c1; *p++=c.m.c2; *p++=c.m.c3; *p++=c.m.c4;
|
|
lastF->next=tf; lastF=tf;
|
|
}
|
|
while (1) /* unthread */
|
|
{
|
|
int c=getc(D);
|
|
if (c=='V')
|
|
{
|
|
int i=LoadWord(D);
|
|
char *s=LoadString(D);
|
|
int v=luaI_findsymbolbyname(s); /* TODO: free s? */
|
|
Unthread(tf->code,i,v);
|
|
}
|
|
else if (c=='S')
|
|
{
|
|
int i=LoadWord(D);
|
|
char *s=LoadString(D);
|
|
int v=luaI_findconstantbyname(s); /* TODO: free s? */
|
|
Unthread(tf->code,i,v);
|
|
}
|
|
else
|
|
{
|
|
ungetc(c,D);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
static void LoadHeader(FILE *D) /* TODO: error handling */
|
|
{
|
|
char *s=LoadString(D);
|
|
Word w,tw=TEST_WORD;
|
|
float f,tf=TEST_FLOAT;
|
|
if (strcmp(s,SIGNATURE)!=0) panic("bad signature");
|
|
luaI_free(s);
|
|
getc(D); /* skip version */
|
|
fread(&w,sizeof(w),1,D); /* a word for testing byte ordering */
|
|
if (w!=tw) warn("different byte order");
|
|
fread(&f,sizeof(f),1,D); /* a float for testing byte ordering */
|
|
if (f!=tf) warn("different float representation");
|
|
}
|
|
|
|
static void LoadChunk(FILE *D)
|
|
{
|
|
LoadHeader(D);
|
|
while (1)
|
|
{
|
|
int c=getc(D);
|
|
if (c=='F') LoadFunction(D); else { ungetc(c,D); break; }
|
|
}
|
|
{ /* TODO: run Main? */
|
|
TFunc *tf;
|
|
for (tf=Main; tf!=NULL; tf=tf->next)
|
|
PrintFunction(tf);
|
|
}
|
|
}
|
|
|
|
void luaI_undump(FILE *D)
|
|
{
|
|
while (1)
|
|
{
|
|
int c=getc(D);
|
|
if (c==ESC) LoadChunk(D); else
|
|
if (c==EOF) return; else
|
|
panic("not a lua binary file");
|
|
}
|
|
}
|
|
|
|
int main(int argc, char* argv[])
|
|
{
|
|
luaI_undump(stdin);
|
|
return 0;
|
|
}
|