ldump.c
上传用户:jxpjxmjjw
上传日期:2009-12-07
资源大小:5877k
文件大小:3k
源码类别:

模拟服务器

开发平台:

Visual C++

  1. /*
  2. ** $Id: ldump.c,v 1.1 2004/08/20 02:26:56 JH Exp $
  3. ** save bytecodes
  4. ** See Copyright Notice in lua.h
  5. */
  6. #include <stddef.h>
  7. #define ldump_c
  8. #include "lua.h"
  9. #include "lobject.h"
  10. #include "lopcodes.h"
  11. #include "lstate.h"
  12. #include "lundump.h"
  13. #define DumpVector(b,n,size,D) DumpBlock(b,(n)*(size),D)
  14. #define DumpLiteral(s,D) DumpBlock("" s,(sizeof(s))-1,D)
  15. typedef struct {
  16.  lua_State* L;
  17.  lua_Chunkwriter write;
  18.  void* data;
  19. } DumpState;
  20. static void DumpBlock(const void* b, size_t size, DumpState* D)
  21. {
  22.  lua_unlock(D->L);
  23.  (*D->write)(D->L,b,size,D->data);
  24.  lua_lock(D->L);
  25. }
  26. static void DumpByte(int y, DumpState* D)
  27. {
  28.  char x=(char)y;
  29.  DumpBlock(&x,sizeof(x),D);
  30. }
  31. static void DumpInt(int x, DumpState* D)
  32. {
  33.  DumpBlock(&x,sizeof(x),D);
  34. }
  35. static void DumpSize(size_t x, DumpState* D)
  36. {
  37.  DumpBlock(&x,sizeof(x),D);
  38. }
  39. static void DumpNumber(lua_Number x, DumpState* D)
  40. {
  41.  DumpBlock(&x,sizeof(x),D);
  42. }
  43. static void DumpString(TString* s, DumpState* D)
  44. {
  45.  if (s==NULL || getstr(s)==NULL)
  46.   DumpSize(0,D);
  47.  else
  48.  {
  49.   size_t size=s->tsv.len+1; /* include trailing '' */
  50.   DumpSize(size,D);
  51.   DumpBlock(getstr(s),size,D);
  52.  }
  53. }
  54. static void DumpCode(const Proto* f, DumpState* D)
  55. {
  56.  DumpInt(f->sizecode,D);
  57.  DumpVector(f->code,f->sizecode,sizeof(*f->code),D);
  58. }
  59. static void DumpLocals(const Proto* f, DumpState* D)
  60. {
  61.  int i,n=f->sizelocvars;
  62.  DumpInt(n,D);
  63.  for (i=0; i<n; i++)
  64.  {
  65.   DumpString(f->locvars[i].varname,D);
  66.   DumpInt(f->locvars[i].startpc,D);
  67.   DumpInt(f->locvars[i].endpc,D);
  68.  }
  69. }
  70. static void DumpLines(const Proto* f, DumpState* D)
  71. {
  72.  DumpInt(f->sizelineinfo,D);
  73.  DumpVector(f->lineinfo,f->sizelineinfo,sizeof(*f->lineinfo),D);
  74. }
  75. static void DumpUpvalues(const Proto* f, DumpState* D)
  76. {
  77.  int i,n=f->sizeupvalues;
  78.  DumpInt(n,D);
  79.  for (i=0; i<n; i++) DumpString(f->upvalues[i],D);
  80. }
  81. static void DumpFunction(const Proto* f, const TString* p, DumpState* D);
  82. static void DumpConstants(const Proto* f, DumpState* D)
  83. {
  84.  int i,n;
  85.  DumpInt(n=f->sizek,D);
  86.  for (i=0; i<n; i++)
  87.  {
  88.   const TObject* o=&f->k[i];
  89.   DumpByte(ttype(o),D);
  90.   switch (ttype(o))
  91.   {
  92.    case LUA_TNUMBER:
  93. DumpNumber(nvalue(o),D);
  94. break;
  95.    case LUA_TSTRING:
  96. DumpString(tsvalue(o),D);
  97. break;
  98.    case LUA_TNIL:
  99. break;
  100.    default:
  101. lua_assert(0); /* cannot happen */
  102. break;
  103.   }
  104.  }
  105.  DumpInt(n=f->sizep,D);
  106.  for (i=0; i<n; i++) DumpFunction(f->p[i],f->source,D);
  107. }
  108. static void DumpFunction(const Proto* f, const TString* p, DumpState* D)
  109. {
  110.  DumpString((f->source==p) ? NULL : f->source,D);
  111.  DumpInt(f->lineDefined,D);
  112.  DumpByte(f->nups,D);
  113.  DumpByte(f->numparams,D);
  114.  DumpByte(f->is_vararg,D);
  115.  DumpByte(f->maxstacksize,D);
  116.  DumpLines(f,D);
  117.  DumpLocals(f,D);
  118.  DumpUpvalues(f,D);
  119.  DumpConstants(f,D);
  120.  DumpCode(f,D);
  121. }
  122. static void DumpHeader(DumpState* D)
  123. {
  124.  DumpLiteral(LUA_SIGNATURE,D);
  125.  DumpByte(VERSION,D);
  126.  DumpByte(luaU_endianness(),D);
  127.  DumpByte(sizeof(int),D);
  128.  DumpByte(sizeof(size_t),D);
  129.  DumpByte(sizeof(Instruction),D);
  130.  DumpByte(SIZE_OP,D);
  131.  DumpByte(SIZE_A,D);
  132.  DumpByte(SIZE_B,D);
  133.  DumpByte(SIZE_C,D);
  134.  DumpByte(sizeof(lua_Number),D);
  135.  DumpNumber(TEST_NUMBER,D);
  136. }
  137. /*
  138. ** dump function as precompiled chunk
  139. */
  140. void luaU_dump (lua_State* L, const Proto* Main, lua_Chunkwriter w, void* data)
  141. {
  142.  DumpState D;
  143.  D.L=L;
  144.  D.write=w;
  145.  D.data=data;
  146.  DumpHeader(&D);
  147.  DumpFunction(Main,NULL,&D);
  148. }