Subversion Repositories NedoOS

Rev

Blame | Last modification | View Log | Download

  1. /*
  2. ** $Id: ldump.c,v 2.17.1.1 2013/04/12 18:48:47 roberto Exp $
  3. ** save precompiled Lua chunks
  4. ** See Copyright Notice in lua.h
  5. */
  6.  
  7. #include <stddef.h>
  8.  
  9. #define ldump_c
  10. #define LUA_CORE
  11.  
  12. #include "lua.h"
  13.  
  14. #include "lobject.h"
  15. #include "lstate.h"
  16. #include "lundump.h"
  17.  
  18. typedef struct {
  19.  lua_State* L;
  20.  lua_Writer writer;
  21.  void* data;
  22.  int strip;
  23.  int status;
  24. } DumpState;
  25.  
  26. #define DumpMem(b,n,size,D)     DumpBlock(b,(n)*(size),D)
  27. #define DumpVar(x,D)            DumpMem(&x,1,sizeof(x),D)
  28.  
  29. static void DumpBlock(const void* b, size_t size, DumpState* D)
  30. {
  31.  if (D->status==0)
  32.  {
  33.   lua_unlock(D->L);
  34.   D->status=(*D->writer)(D->L,b,size,D->data);
  35.   lua_lock(D->L);
  36.  }
  37. }
  38.  
  39. static void DumpChar(int y, DumpState* D)
  40. {
  41.  char x=(char)y;
  42.  DumpVar(x,D);
  43. }
  44.  
  45. static void DumpInt(int x, DumpState* D)
  46. {
  47.  DumpVar(x,D);
  48. }
  49.  
  50. static void DumpNumber(lua_Number x, DumpState* D)
  51. {
  52.  DumpVar(x,D);
  53. }
  54.  
  55. static void DumpVector(const void* b, int n, size_t size, DumpState* D)
  56. {
  57.  DumpInt(n,D);
  58.  DumpMem(b,n,size,D);
  59. }
  60.  
  61. static void DumpString(const TString* s, DumpState* D)
  62. {
  63.  if (s==NULL)
  64.  {
  65.   size_t size=0;
  66.   DumpVar(size,D);
  67.  }
  68.  else
  69.  {
  70.   size_t size=s->tsv.len+1;             /* include trailing '\0' */
  71.   DumpVar(size,D);
  72.   DumpBlock(getstr(s),size*sizeof(char),D);
  73.  }
  74. }
  75.  
  76. #define DumpCode(f,D)    DumpVector(f->code,f->sizecode,sizeof(Instruction),D)
  77.  
  78. static void DumpFunction(const Proto* f, DumpState* D);
  79.  
  80. static void DumpConstants(const Proto* f, DumpState* D)
  81. {
  82.  int i,n=f->sizek;
  83.  DumpInt(n,D);
  84.  for (i=0; i<n; i++)
  85.  {
  86.   const TValue* o=&f->k[i];
  87.   DumpChar(ttypenv(o),D);
  88.   switch (ttypenv(o))
  89.   {
  90.    case LUA_TNIL:
  91.         break;
  92.    case LUA_TBOOLEAN:
  93.         DumpChar(bvalue(o),D);
  94.         break;
  95.    case LUA_TNUMBER:
  96.         DumpNumber(nvalue(o),D);
  97.         break;
  98.    case LUA_TSTRING:
  99.         DumpString(rawtsvalue(o),D);
  100.         break;
  101.     default: lua_assert(0);
  102.   }
  103.  }
  104.  n=f->sizep;
  105.  DumpInt(n,D);
  106.  for (i=0; i<n; i++) DumpFunction(f->p[i],D);
  107. }
  108.  
  109. static void DumpUpvalues(const Proto* f, DumpState* D)
  110. {
  111.  int i,n=f->sizeupvalues;
  112.  DumpInt(n,D);
  113.  for (i=0; i<n; i++)
  114.  {
  115.   DumpChar(f->upvalues[i].instack,D);
  116.   DumpChar(f->upvalues[i].idx,D);
  117.  }
  118. }
  119.  
  120. static void DumpDebug(const Proto* f, DumpState* D)
  121. {
  122.  int i,n;
  123.  DumpString((D->strip) ? NULL : f->source,D);
  124.  n= (D->strip) ? 0 : f->sizelineinfo;
  125.  DumpVector(f->lineinfo,n,sizeof(int),D);
  126.  n= (D->strip) ? 0 : f->sizelocvars;
  127.  DumpInt(n,D);
  128.  for (i=0; i<n; i++)
  129.  {
  130.   DumpString(f->locvars[i].varname,D);
  131.   DumpInt(f->locvars[i].startpc,D);
  132.   DumpInt(f->locvars[i].endpc,D);
  133.  }
  134.  n= (D->strip) ? 0 : f->sizeupvalues;
  135.  DumpInt(n,D);
  136.  for (i=0; i<n; i++) DumpString(f->upvalues[i].name,D);
  137. }
  138.  
  139. static void DumpFunction(const Proto* f, DumpState* D)
  140. {
  141.  DumpInt(f->linedefined,D);
  142.  DumpInt(f->lastlinedefined,D);
  143.  DumpChar(f->numparams,D);
  144.  DumpChar(f->is_vararg,D);
  145.  DumpChar(f->maxstacksize,D);
  146.  DumpCode(f,D);
  147.  DumpConstants(f,D);
  148.  DumpUpvalues(f,D);
  149.  DumpDebug(f,D);
  150. }
  151.  
  152. static void DumpHeader(DumpState* D)
  153. {
  154.  lu_byte h[LUAC_HEADERSIZE];
  155.  luaU_header(h);
  156.  DumpBlock(h,LUAC_HEADERSIZE,D);
  157. }
  158.  
  159. /*
  160. ** dump Lua function as precompiled chunk
  161. */
  162. int luaU_dump (lua_State* L, const Proto* f, lua_Writer w, void* data, int strip)
  163. {
  164.  DumpState D;
  165.  D.L=L;
  166.  D.writer=w;
  167.  D.data=data;
  168.  D.strip=strip;
  169.  D.status=0;
  170.  DumpHeader(&D);
  171.  DumpFunction(f,&D);
  172.  return D.status;
  173. }
  174.