From eada82cc75dd0a2ad40ad25552f8e8e2c93891f8 Mon Sep 17 00:00:00 2001 From: Benoit Giannangeli Date: Thu, 2 Mar 2017 21:15:42 +0100 Subject: luaB_load --- src/lapi.js | 29 +++++++++++++++++++++++++++++ src/lbaselib.js | 58 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 87 insertions(+) (limited to 'src') diff --git a/src/lapi.js b/src/lapi.js index 72ff554..32a9873 100644 --- a/src/lapi.js +++ b/src/lapi.js @@ -182,6 +182,11 @@ const lua_insert = function(L, idx) { lua_rotate(L, idx, 1); }; +const lua_replace = function(L, idx) { + lua_copy(L, -1, idx); + lua_pop(L, 1); +}; + /* ** push functions (JS -> stack) */ @@ -424,6 +429,19 @@ const lua_createtable = function(L, narray, nrec) { assert(L.top <= L.ci.top, "stack overflow"); }; +const lua_setupvalue = function(L, funcindex, n) { + let fi = index2addr(L, funcindex); + assert(1 < L.top - L.ci.funcOff, "not enough elements in the stack"); + let aux = aux_upvalue(fi, n); + let name = aux.name; + let val = aux.val; + if (name) { + L.top--; + setobj(L, val, L.top); + } + return name; +}; + const lua_newtable = function(L) { lua_createtable(L, 0, 0); }; @@ -595,6 +613,14 @@ const lua_typename = function(L, t) { return ltm.ttypename(t); }; +const lua_isnil = function(L, n) { + return lua_type(L, n) === CT.LUA_TNIL; +}; + +const lua_isnone = function(L, n) { + return lua_type(L, n) === CT.LUA_TNONE; +}; + const lua_isnoneornil = function(L, n) { return lua_type(L, n) <= 0; }; @@ -798,6 +824,8 @@ module.exports.lua_gettable = lua_gettable; module.exports.lua_gettop = lua_gettop; module.exports.lua_insert = lua_insert; module.exports.lua_isinteger = lua_isinteger; +module.exports.lua_isnil = lua_isnil; +module.exports.lua_isnone = lua_isnone; module.exports.lua_isnoneornil = lua_isnoneornil; module.exports.lua_isnumber = lua_isnumber; module.exports.lua_isstring = lua_isstring; @@ -830,6 +858,7 @@ module.exports.lua_rawgeti = lua_rawgeti; module.exports.lua_rawlen = lua_rawlen; module.exports.lua_rawset = lua_rawset; module.exports.lua_remove = lua_remove; +module.exports.lua_replace = lua_replace; module.exports.lua_rotate = lua_rotate; module.exports.lua_setfield = lua_setfield; module.exports.lua_setglobal = lua_setglobal; diff --git a/src/lbaselib.js b/src/lbaselib.js index 77638d0..a7666b5 100644 --- a/src/lbaselib.js +++ b/src/lbaselib.js @@ -251,6 +251,64 @@ const luaB_xpcall = function(L) { return finishpcall(L, status, 2); }; +const load_aux = function(L, status, envidx) { + if (status === TS.LUA_OK) { + if (envidx !== 0) { /* 'env' parameter? */ + lapi.lua_pushvalue(L, envidx); /* environment for loaded function */ + if (!lapi.lua_setupvalue(L, -2, 1)) /* set it as 1st upvalue */ + lapi.lua_pop(L, 1); /* remove 'env' if not used by previous call */ + } + return 1; + } else { /* error (message is on top of the stack) */ + lapi.lua_pushnil(L); + lapi.lua_insert(L, -2); /* put before error message */ + return 2; /* return nil plus error message */ + } +}; + +/* +** reserved slot, above all arguments, to hold a copy of the returned +** string to avoid it being collected while parsed. 'load' has four +** optional arguments (chunk, source name, mode, and environment). +*/ +const RESERVEDSLOT = 5; + +/* +** Reader for generic 'load' function: 'lua_load' uses the +** stack for internal stuff, so the reader cannot change the +** stack top. Instead, it keeps its resulting string in a +** reserved slot inside the stack. +*/ +const generic_reader = function(L, ud) { + lauxlib.luaL_checkstack(L, 2, "too many nested functions"); + lapi.lua_pushvalue(L, 1); /* get function */ + lapi.lua_call(L, 0, 1); /* call it */ + if (lapi.lua_isnil(L, -1)) { + lapi.lua_pop(L, 1); /* pop result */ + return null; + } else if (!lapi.lua_isstring(L, -1)) + lauxlib.luaL_error(L, "reader function must return a string"); + lapi.lua_replace(L, RESERVEDSLOT); /* save string in reserved slot */ + return lapi.lua_tostring(L, RESERVEDSLOT); +}; + +const luaB_load = function(L) { + let s = lapi.lua_tostring(L, 1); + let mode = lauxlib.luaL_optstring(L, 3, "bt"); + let env = !lapi.lua_isnone(L, 4) ? 4 : 0; /* 'env' index or 0 if no 'env' */ + let status; + if (s !== null) { /* loading a string? */ + let chunkname = lauxlib.luaL_optstring(L, 2, s); + status = lauxlib.luaL_loadbufferx(L, s, chunkname, mode); + } else { /* loading from a reader function */ + let chunkname = lauxlib.luaL_optstring(L, 2, "=(load)"); + lauxlib.luaL_checktype(L, 1, CT.LUA_TFUNCTION); + lapi.lua_settop(L, RESERVEDSLOT); /* create reserved slot */ + status = lapi.lua_load(L, generic_reader, null, chunkname, mode); + } + return load_aux(L, status, env); +}; + const base_funcs = { "collectgarbage": function () {}, "assert": luaB_assert, -- cgit v1.2.3-70-g09d2 From dba35fd2956f9f6b14ea8947baa11c5e483bc407 Mon Sep 17 00:00:00 2001 From: daurnimator Date: Fri, 3 Mar 2017 22:46:39 +1100 Subject: Remove use of 'buffer-dataview' library --- package.json | 1 - src/llex.js | 1 - src/lundump.js | 3 +-- tests/tests.js | 6 +++--- 4 files changed, 4 insertions(+), 7 deletions(-) (limited to 'src') diff --git a/package.json b/package.json index 7496921..d567a62 100644 --- a/package.json +++ b/package.json @@ -33,7 +33,6 @@ "tmp": "0.0.31" }, "dependencies": { - "buffer-dataview": "0.0.2", "seedrandom": "^2.4.2" } } diff --git a/src/llex.js b/src/llex.js index 711f44b..2e79066 100644 --- a/src/llex.js +++ b/src/llex.js @@ -1,6 +1,5 @@ "use strict"; -const DataView = require('buffer-dataview'); const assert = require('assert'); const lapi = require('./lapi.js'); diff --git a/src/lundump.js b/src/lundump.js index 5a6d8c4..92063a2 100644 --- a/src/lundump.js +++ b/src/lundump.js @@ -1,7 +1,6 @@ /*jshint esversion: 6 */ "use strict"; -const DataView = require('buffer-dataview'); const fs = require('fs'); const assert = require('assert'); @@ -110,7 +109,7 @@ class BytecodeParser { } readInstruction() { - let ins = new DataView(new Buffer(this.instructionSize)); + let ins = new DataView(new ArrayBuffer(this.instructionSize)); for (let i = 0; i < this.instructionSize; i++) ins.setUint8(i, this.readByte()); diff --git a/tests/tests.js b/tests/tests.js index a47aee5..d4173c3 100644 --- a/tests/tests.js +++ b/tests/tests.js @@ -3,7 +3,6 @@ const fs = require('fs'); const child_process = require('child_process'); const tmp = require('tmp'); -const DataView = require('buffer-dataview'); const BytecodeParser = require("../src/lundump.js"); const lauxlib = require("../src/lauxlib.js"); @@ -20,10 +19,11 @@ const toByteCode = function (luaCode) { bclist = fs.readFileSync(`${luaFile.name}.bc.txt`, 'utf8'); - // console.log(bclist); + let b = fs.readFileSync(`${luaFile.name}.bc`); + let dv = new DataView(b.buffer.slice(b.byteOffset, b.byteOffset + b.byteLength)); return { - dataView: new DataView(fs.readFileSync(`${luaFile.name}.bc`)), + dataView: dv, bclist: bclist }; }; -- cgit v1.2.3-70-g09d2