src/luac/dump.c

   1 /*
   2 ** $Id: dump.c,v 1.30 2000/10/31 16:57:23 lhf Exp $
   3 ** save bytecodes to file
   4 ** See Copyright Notice in lua.h
   5 */
   6
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <string.h>
  10
  11 #include "luac.h"
  12
  13 #define DumpVector(b,n,size,D)  fwrite(b,size,n,D)
  14 #define DumpBlock(b,size,D)     fwrite(b,size,1,D)
  15 #define DumpByte                fputc
  16
  17 static void DumpInt(int x, FILE* D)
  18 {
  19  DumpBlock(&x,sizeof(x),D);
  20 }
  21
  22 static void DumpSize(size_t x, FILE* D)
  23 {
  24  DumpBlock(&x,sizeof(x),D);
  25 }
  26
  27 static void DumpNumber(Number x, FILE* D)
  28 {
  29  DumpBlock(&x,sizeof(x),D);
  30 }
  31
  32 static void DumpString(const TString* s, FILE* D)
  33 {
  34  if (s==NULL || s->str==NULL)
  35   DumpSize(0,D);
  36  else
  37  {
  38   size_t size=s->len+1;                 /* include trailing '\0' */
  39   DumpSize(size,D);
  40   DumpBlock(s->str,size,D);
  41  }
  42 }
  43
  44 static void DumpCode(const Proto* tf, FILE* D)
  45 {
  46  DumpInt(tf->ncode,D);
  47  DumpVector(tf->code,tf->ncode,sizeof(*tf->code),D);
  48 }
  49
  50 static void DumpLocals(const Proto* tf, FILE* D)
  51 {
  52  int i,n=tf->nlocvars;
  53  DumpInt(n,D);
  54  for (i=0; i<n; i++)
  55  {
  56   DumpString(tf->locvars[i].varname,D);
  57   DumpInt(tf->locvars[i].startpc,D);
  58   DumpInt(tf->locvars[i].endpc,D);
  59  }
  60 }
  61
  62 static void DumpLines(const Proto* tf, FILE* D)
  63 {
  64  DumpInt(tf->nlineinfo,D);
  65  DumpVector(tf->lineinfo,tf->nlineinfo,sizeof(*tf->lineinfo),D);
  66 }
  67
  68 static void DumpFunction(const Proto* tf, FILE* D);
  69
  70 static void DumpConstants(const Proto* tf, FILE* D)
  71 {
  72  int i,n;
  73  DumpInt(n=tf->nkstr,D);
  74  for (i=0; i<n; i++)
  75   DumpString(tf->kstr[i],D);
  76  DumpInt(tf->nknum,D);
  77  DumpVector(tf->knum,tf->nknum,sizeof(*tf->knum),D);
  78  DumpInt(n=tf->nkproto,D);
  79  for (i=0; i<n; i++)
  80   DumpFunction(tf->kproto[i],D);
  81 }
  82
  83 static void DumpFunction(const Proto* tf, FILE* D)
  84 {
  85  DumpString(tf->source,D);
  86  DumpInt(tf->lineDefined,D);
  87  DumpInt(tf->numparams,D);
  88  DumpByte(tf->is_vararg,D);
  89  DumpInt(tf->maxstacksize,D);
  90  DumpLocals(tf,D);
  91  DumpLines(tf,D);
  92  DumpConstants(tf,D);
  93  DumpCode(tf,D);
  94  if (ferror(D))
  95  {
  96   perror("luac: write error");
  97   exit(1);
  98  }
  99 }
 100
 101 static void DumpHeader(FILE* D)
 102 {
 103  DumpByte(ID_CHUNK,D);
 104  fputs(SIGNATURE,D);
 105  DumpByte(VERSION,D);
 106  DumpByte(luaU_endianess(),D);
 107  DumpByte(sizeof(int),D);
 108  DumpByte(sizeof(size_t),D);
 109  DumpByte(sizeof(Instruction),D);
 110  DumpByte(SIZE_INSTRUCTION,D);
 111  DumpByte(SIZE_OP,D);
 112  DumpByte(SIZE_B,D);
 113  DumpByte(sizeof(Number),D);
 114  DumpNumber(TEST_NUMBER,D);
 115 }
 116
 117 void luaU_dumpchunk(const Proto* Main, FILE* D)
 118 {
 119  DumpHeader(D);
 120  DumpFunction(Main,D);
 121 }