diff options
| -rw-r--r-- | src/ldump.js | 184 | ||||
| -rw-r--r-- | src/lstrlib.js | 2 |
2 files changed, 185 insertions, 1 deletions
diff --git a/src/ldump.js b/src/ldump.js new file mode 100644 index 0000000..795bd19 --- /dev/null +++ b/src/ldump.js @@ -0,0 +1,184 @@ +"use strict"; + +const assert = require('assert'); + +const lua = require('./lua.js'); +const lstate = require('./lstate.js'); +const lobject = require('./lobject.js'); +const CT = lua.constant_types; + +const LUAC_DATA = "\x19\x93\r\n\x1a\n"; +const LUAC_INT = 0x5678; +const LUAC_NUM = 370.5; +const LUAC_VERSION = lua.LUA_VERSION_MAJOR.charCodeAt(0) * 16 + lua.LUA_VERSION_MINOR.charCodeAt(0); +const LUAC_FORMAT = 0; /* this is the official format */ + +class DumpState { + constructor() { + this.L = null; + this.write = null; + this.data = null; + this.strip = NaN; + this.status = NaN; + } +} + +const DumpBlock = function(b, size, D) { + if (D.status === 0 && size > 0) + D.status = D.writer(D.L, b, size, D.data); +}; + +const DumpVector = function(v, n, D) { + DumpBlock(v, n, D); +}; + +const DumpLiteral = function(s, D) { + DumpBlock(s, s.length, D); +}; + +const DumpVar = function (x,D) { + DumpVector(x, 1, D); +}; + +const DumpByte = function(y, D) { + DumpVar(y, D); +}; + +const DumpInt = function(x, D) { + DumpVar(x, D); +}; + +const DumpInteger = DumpInt; + +const DumpNumber = function(x, D) { + DumpVar(x, D); +}; + +const DumpString = function(s, D) { + if (s === null) + DumpByte(0, D); + else { + let size = s.value.length + 1; + let str = s.value; + if (size < 0xFF) + DumpByte(size, D); + else { + DumpByte(0xFF, D); + DumpVar(size, D); + } + DumpVector(str, size - 1, D); /* no need to save '\0' */ + } +}; + +const DumpCode = function(f, D) { + DumpInt(f.code.length, D); + DumpVector(f.code, f.code.length, D); +}; + +const DumpConstants = function(f, D) { + let n = f.k.length; + DumpInt(n, D); + for (let i = 0; i < n; i++) { + let o = f.k[i]; + DumpByte(o.ttype(), D); + switch (o.ttype()) { + case CT.LUA_TNIL: + break; + case CT.LUA_TBOOLEAN: + DumpByte(o.value ? 1 : 0, D); + break; + case CT.LUA_TNUMFLT: + DumpNumber(o.value, D); + break; + case CT.LUA_TNUMINT: + DumpInteger(o.value, D); + break; + case CT.LUA_TSHRSTR: + case CT.LUA_TLNGSTR: + DumpString(o.value, D); + break; + } + } +}; + +const DumpProtos = function(f, D) { + let n = f.p.length; + DumpInt(n, D); + for (let i = 0; i < n; i++) + DumpFunction(f.p[i], f.source, D); +}; + +const DumpUpvalues = function(f, D) { + let n = f.upvalues.length; + DumpInt(n, D); + for (let i = 0; i < n; i++) { + DumpByte(f.upvalues[i].instack ? 1 : 0, D); + DumpByte(f.upvalues[i].idx, D); + } +}; + +const DumpDebug = function(f, D) { + let n = D.strip ? 0 : f.lineinfo.length; + DumpInt(n, D); + DumpVector(f.lineinfo, n, D); + n = D.strip ? 0 : f.locvars.length; + DumpInt(n, D); + for (let i = 0; i < n; i++) { + DumpString(f.locvars[i].varname, D); + DumpInt(f.locvars[i].startpc, D); + DumpInt(f.locvars[i].endpc, D); + } + n = D.strip ? 0 : f.upvalues.length; + DumpInt(n, D); + for (let i = 0; i < n; i++) + DumpString(f.upvalues[i].name, D); +}; + +const DumpFunction = function(f, psource, D) { + if (D.strip || f.source === psource) + DumpString(null, D); /* no debug info or same source as its parent */ + else + DumpString(f.source, D); + DumpInt(f.linedefined, D); + DumpInt(f.lastlinedefined, D); + DumpByte(f.numparams, D); + DumpByte(f.is_vararg, D); + DumpByte(f.maxstacksize, D); + DumpCode(f, D); + DumpConstants(f, D); + DumpUpvalues(f, D); + DumpProtos(f, D); + DumpDebug(f, D); +}; + +const DumpHeader = function(D) { + DumpLiteral(lua.LUA_SIGNATURE, D); + DumpByte(LUAC_VERSION, D); + DumpByte(LUAC_FORMAT, D); + DumpLiteral(LUAC_DATA, D); + DumpByte(4, D); // intSize + DumpByte(8, D); // size_tSize + DumpByte(4, D); // instructionSize + DumpByte(8, D); // integerSize + DumpByte(8, D); // numberSize + DumpInteger(LUAC_INT, D); + DumpNumber(LUAC_NUM, D); +}; + +/* +** dump Lua function as precompiled chunk +*/ +const luaU_dump = function(L, f, w, data, strip) { + let D = new DumpState(); + D.L = L; + D.writer = w; + D.data = data; + D.strip = strip; + D.status = 0; + DumpHeader(D); + DumpByte(f.upvalues.length, D); + DumpFunction(f, null, D); + return D.status; +}; + +module.exports.luaU_dump = luaU_dump;
\ No newline at end of file diff --git a/src/lstrlib.js b/src/lstrlib.js index 1d3a7cc..b47138f 100644 --- a/src/lstrlib.js +++ b/src/lstrlib.js @@ -405,4 +405,4 @@ const luaopen_string = function(L) { return 1; }; -module.exports.luaopen_string = luaopen_string;
\ No newline at end of file +module.exports.luaopen_string = luaopen_string; |