annotate src/lua/ldump.c @ 512:7ba07a6adb0c

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