Refactor as Rust workspace

This commit is contained in:
Rerumu
2021-11-28 03:33:00 -05:00
parent 185799eecc
commit a1894e492c
22 changed files with 11 additions and 9 deletions
+393
View File
@@ -0,0 +1,393 @@
local module = {}
local bit = require('bit')
local ffi = require('ffi')
local jit = require('jit')
local u32 = ffi.typeof('uint32_t')
local u64 = ffi.typeof('uint64_t')
local i64 = ffi.typeof('int64_t')
ffi.cdef [[
typedef union {
int32_t i32;
int64_t i64;
float f32;
double f64;
} Reinterpret;
]]
if jit and jit.opt then jit.opt.start("maxsnap=1000", "loopunroll=500", "maxmcode=2048") end
local function truncate(num)
if num >= 0 then
return math.floor(num)
else
return math.ceil(num)
end
end
do
local add = {}
local sub = {}
local mul = {}
local div = {}
local to_signed = bit.tobit
module.add = add
module.sub = sub
module.mul = mul
module.div = div
function add.i32(a, b) return to_signed(a + b) end
function add.i64(a, b) return a + b end
function sub.i32(a, b) return to_signed(a - b) end
function sub.i64(a, b) return a - b end
function mul.i32(a, b) return to_signed(a * b) end
function mul.i64(a, b) return a * b end
function div.i32(lhs, rhs)
if rhs == 0 then error('division by zero') end
return truncate(lhs / rhs)
end
function div.u32(lhs, rhs)
if rhs == 0 then error('division by zero') end
lhs = tonumber(u32(lhs))
rhs = tonumber(u32(rhs))
return to_signed(math.floor(lhs / rhs))
end
function div.u64(lhs, rhs)
if rhs == 0 then error('division by zero') end
return i64(u64(lhs) / u64(rhs))
end
end
do
local clz = {}
local ctz = {}
local popcnt = {}
local lj_band = bit.band
local lj_lshift = bit.lshift
module.clz = clz
module.ctz = ctz
module.popcnt = popcnt
function clz.i32(num)
for i = 0, 31 do
local mask = lj_lshift(1, 31 - i)
if lj_band(num, mask) ~= 0 then return i end
end
return 32
end
function ctz.i32(num)
for i = 0, 31 do
local mask = lj_lshift(1, i)
if lj_band(num, mask) ~= 0 then return i end
end
return 32
end
function popcnt.i32(num)
local count = 0
while num ~= 0 do
num = lj_band(num, num - 1)
count = count + 1
end
return count
end
end
do
local eqz = {}
local eq = {}
local ne = {}
local le = {}
local lt = {}
local ge = {}
local gt = {}
module.eqz = eqz
module.eq = eq
module.ne = ne
module.le = le
module.lt = lt
module.ge = ge
module.gt = gt
local function to_boolean(cond)
if cond then
return 1
else
return 0
end
end
function eq.i32(lhs, rhs) return to_boolean(lhs == rhs) end
function eq.i64(lhs, rhs) return to_boolean(lhs == rhs) end
function eqz.i32(lhs) return to_boolean(lhs == 0) end
function eqz.i64(lhs) return to_boolean(lhs == 0) end
function ne.i32(lhs, rhs) return to_boolean(lhs ~= rhs) end
function ne.i64(lhs, rhs) return to_boolean(lhs ~= rhs) end
function ge.i32(lhs, rhs) return to_boolean(lhs >= rhs) end
function ge.i64(lhs, rhs) return to_boolean(lhs >= rhs) end
function ge.u32(lhs, rhs) return to_boolean(u32(lhs) >= u32(rhs)) end
function ge.u64(lhs, rhs) return to_boolean(u64(lhs) >= u64(rhs)) end
function gt.i32(lhs, rhs) return to_boolean(lhs > rhs) end
function gt.i64(lhs, rhs) return to_boolean(lhs > rhs) end
function gt.u32(lhs, rhs) return to_boolean(u32(lhs) > u32(rhs)) end
function gt.u64(lhs, rhs) return to_boolean(u64(lhs) > u64(rhs)) end
function le.i32(lhs, rhs) return to_boolean(lhs <= rhs) end
function le.i64(lhs, rhs) return to_boolean(lhs <= rhs) end
function le.u32(lhs, rhs) return to_boolean(u32(lhs) <= u32(rhs)) end
function le.u64(lhs, rhs) return to_boolean(u64(lhs) <= u64(rhs)) end
function lt.i32(lhs, rhs) return to_boolean(lhs < rhs) end
function lt.i64(lhs, rhs) return to_boolean(lhs < rhs) end
function lt.u32(lhs, rhs) return to_boolean(u32(lhs) < u32(rhs)) end
function lt.u64(lhs, rhs) return to_boolean(u64(lhs) < u64(rhs)) end
end
do
local band = {}
local bor = {}
local bxor = {}
local bnot = {}
module.band = band
module.bor = bor
module.bxor = bxor
module.bnot = bnot
band.i32 = bit.band
band.i64 = bit.band
bnot.i32 = bit.bnot
bnot.i64 = bit.bnot
bor.i32 = bit.bor
bor.i64 = bit.bor
bxor.i32 = bit.bxor
bxor.i64 = bit.bxor
end
do
local shl = {}
local shr = {}
local rotl = {}
local rotr = {}
local lj_lshift = bit.lshift
local lj_rshift = bit.rshift
local lj_arshift = bit.arshift
module.shl = shl
module.shr = shr
module.rotl = rotl
module.rotr = rotr
rotl.i32 = bit.rol
rotl.i64 = bit.rol
rotr.i32 = bit.ror
rotr.i64 = bit.ror
shl.i32 = lj_lshift
shl.i64 = lj_lshift
shl.u32 = lj_lshift
shl.u64 = lj_lshift
shr.i32 = lj_arshift
shr.i64 = lj_arshift
shr.u32 = lj_rshift
shr.u64 = lj_rshift
end
do
local wrap = {}
local trunc = {}
local extend = {}
local convert = {}
local reinterpret = {}
-- This would surely be an issue in a multi-thread environment...
-- ... thankfully this isn't one.
local RE_INSTANCE = ffi.new('Reinterpret')
local function truncate_i64(num) return i64(truncate(num)) end
module.wrap = wrap
module.trunc = trunc
module.extend = extend
module.convert = convert
module.reinterpret = reinterpret
function wrap.i32_i64(num)
RE_INSTANCE.i64 = num
return RE_INSTANCE.i32
end
trunc.i32_f32 = truncate
trunc.i32_f64 = truncate
trunc.u32_f32 = truncate
trunc.u32_f64 = truncate
trunc.i64_f32 = truncate_i64
trunc.i64_f64 = truncate_i64
trunc.u64_f32 = truncate_i64
trunc.u64_f64 = truncate_i64
function extend.u64_i32(num)
RE_INSTANCE.i64 = 0
RE_INSTANCE.i32 = num
return RE_INSTANCE.i64
end
function convert.f32_i32(num) return num end
function convert.f32_u32(num) return tonumber(u32(num)) end
function convert.f32_i64(num) return tonumber(num) end
function convert.f32_u64(num) return tonumber(u64(num)) end
function convert.f64_i32(num) return num end
function convert.f64_u32(num) return tonumber(u32(num)) end
function convert.f64_i64(num) return tonumber(num) end
function convert.f64_u64(num) return tonumber(u64(num)) end
function reinterpret.i32_f32(num)
RE_INSTANCE.f32 = num
return RE_INSTANCE.i32
end
function reinterpret.i64_f64(num)
RE_INSTANCE.f64 = num
return RE_INSTANCE.i64
end
function reinterpret.f32_i32(num)
RE_INSTANCE.i32 = num
return RE_INSTANCE.f32
end
function reinterpret.f64_i64(num)
RE_INSTANCE.i64 = num
return RE_INSTANCE.f64
end
end
do
local load = {}
local store = {}
local cast = ffi.cast
local ptr_i8 = ffi.typeof('int8_t *')
local ptr_i16 = ffi.typeof('int16_t *')
local ptr_i32 = ffi.typeof('int32_t *')
local ptr_i64 = ffi.typeof('int64_t *')
local ptr_u16 = ffi.typeof('uint16_t *')
local ptr_u32 = ffi.typeof('uint32_t *')
local ptr_f32 = ffi.typeof('float *')
local ptr_f64 = ffi.typeof('double *')
module.load = load
module.store = store
function load.i32_i8(memory, addr) return cast(ptr_i8, memory.data)[addr] end
function load.i32_u8(memory, addr) return memory.data[addr] end
function load.i32_i16(memory, addr) return cast(ptr_i16, memory.data + addr)[0] end
function load.i32_u16(memory, addr) return cast(ptr_u16, memory.data + addr)[0] end
function load.i32(memory, addr) return cast(ptr_i32, memory.data + addr)[0] end
function load.i64_i8(memory, addr) return i64(cast(ptr_i8, memory.data)[addr]) end
function load.i64_u8(memory, addr) return i64(memory.data[addr]) end
function load.i64_i16(memory, addr) return i64(cast(ptr_i16, memory.data + addr)[0]) end
function load.i64_u16(memory, addr) return i64(cast(ptr_u16, memory.data + addr)[0]) end
function load.i64_i32(memory, addr) return i64(cast(ptr_i32, memory.data + addr)[0]) end
function load.i64_u32(memory, addr) return i64(cast(ptr_u32, memory.data + addr)[0]) end
function load.i64(memory, addr) return cast(ptr_i64, memory.data + addr)[0] end
function load.f32(memory, addr) return cast(ptr_f32, memory.data + addr)[0] end
function load.f64(memory, addr) return cast(ptr_f64, memory.data + addr)[0] end
function store.i32_n8(memory, addr, value) memory.data[addr] = value end
function store.i32_n16(memory, addr, value) cast(ptr_i16, memory.data + addr)[0] = value end
function store.i32(memory, addr, value) cast(ptr_i32, memory.data + addr)[0] = value end
function store.i64_n8(memory, addr, value) memory.data[addr] = value end
function store.i64_n16(memory, addr, value) cast(ptr_i16, memory.data + addr)[0] = value end
function store.i64_n32(memory, addr, value) cast(ptr_i32, memory.data + addr)[0] = value end
function store.i64(memory, addr, value) cast(ptr_i64, memory.data + addr)[0] = value end
function store.f32(memory, addr, value) cast(ptr_f32, memory.data + addr)[0] = value end
function store.f64(memory, addr, value) cast(ptr_f64, memory.data + addr)[0] = value end
end
do
local memory = {}
local vla_u8 = ffi.typeof('uint8_t[?]')
local WASM_PAGE_SIZE = 65536
module.memory = memory
local function grow_unchecked(memory, old, new)
local data = vla_u8(new * WASM_PAGE_SIZE, 0)
ffi.copy(data, memory.data, old * WASM_PAGE_SIZE)
memory.min = new
memory.data = data
end
function memory.new(min, max)
local memory = {}
memory.min = min
memory.max = max
memory.data = vla_u8(min * WASM_PAGE_SIZE, 0)
return memory
end
function memory.init(memory, offset, data) ffi.copy(memory.data + offset, data) end
function memory.size(memory) return memory.min end
function memory.grow(memory, num)
local old = memory.min
local new = old + num
if new > memory.max then
return -1
else
grow_unchecked(memory, old, new)
return old
end
end
end
return module
+286
View File
@@ -0,0 +1,286 @@
local module = {}
local function no_op(x) return x end
do
local div = {}
module.div = div
function div.i32(lhs, rhs)
if rhs == 0 then error('division by zero') end
return math.floor(lhs / rhs)
end
end
do
local clz = {}
local ctz = {}
local popcnt = {}
module.clz = clz
module.ctz = ctz
module.popcnt = popcnt
clz.i32 = bit32.countlz
ctz.i32 = bit32.countrz
function popcnt.i32(num)
local count = 0
while num ~= 0 do
num = bit.band(num, num - 1)
count = count + 1
end
return count
end
end
do
local eqz = {}
local eq = {}
local ne = {}
local le = {}
local lt = {}
local ge = {}
local gt = {}
module.eqz = eqz
module.eq = eq
module.ne = ne
module.le = le
module.lt = lt
module.ge = ge
module.gt = gt
local function unsign_i32(x)
if x < 0 then x = x + 0x100000000 end
return x
end
local function unsign_i64(x)
if x < 0 then x = x + 0x10000000000000000 end
return x
end
function eq.i32(lhs, rhs) return lhs == rhs and 1 or 0 end
function eq.i64(lhs, rhs) return lhs == rhs and 1 or 0 end
function eqz.i32(lhs) return lhs == 0 and 1 or 0 end
function eqz.i64(lhs) return lhs == 0 and 1 or 0 end
function ne.i32(lhs, rhs) return lhs ~= rhs and 1 or 0 end
function ne.i64(lhs, rhs) return lhs ~= rhs and 1 or 0 end
function ge.i32(lhs, rhs) return lhs >= rhs and 1 or 0 end
function ge.i64(lhs, rhs) return lhs >= rhs and 1 or 0 end
function ge.u32(lhs, rhs) return unsign_i32(lhs) >= unsign_i32(rhs) and 1 or 0 end
function ge.u64(lhs, rhs) return unsign_i64(lhs) >= unsign_i64(rhs) and 1 or 0 end
function gt.i32(lhs, rhs) return lhs > rhs and 1 or 0 end
function gt.i64(lhs, rhs) return lhs > rhs and 1 or 0 end
function gt.u32(lhs, rhs) return unsign_i32(lhs) > unsign_i32(rhs) and 1 or 0 end
function gt.u64(lhs, rhs) return unsign_i64(lhs) > unsign_i64(rhs) and 1 or 0 end
function le.i32(lhs, rhs) return lhs <= rhs and 1 or 0 end
function le.i64(lhs, rhs) return lhs <= rhs and 1 or 0 end
function le.u32(lhs, rhs) return unsign_i32(lhs) <= unsign_i32(rhs) and 1 or 0 end
function le.u64(lhs, rhs) return unsign_i64(lhs) <= unsign_i64(rhs) and 1 or 0 end
function lt.i32(lhs, rhs) return lhs < rhs and 1 or 0 end
function lt.i64(lhs, rhs) return lhs < rhs and 1 or 0 end
function lt.u32(lhs, rhs) return unsign_i32(lhs) < unsign_i32(rhs) and 1 or 0 end
function lt.u64(lhs, rhs) return unsign_i64(lhs) < unsign_i64(rhs) and 1 or 0 end
end
do
local band = {}
local bor = {}
local bxor = {}
local bnot = {}
module.band = band
module.bor = bor
module.bxor = bxor
module.bnot = bnot
band.i32 = bit32.band
band.i64 = bit32.band
bnot.i32 = bit32.bnot
bnot.i64 = bit32.bnot
bor.i32 = bit32.bor
bor.i64 = bit32.bor
bxor.i32 = bit32.bxor
bxor.i64 = bit32.bxor
end
do
local shl = {}
local shr = {}
local rotl = {}
local rotr = {}
module.shl = shl
module.shr = shr
module.rotl = rotl
module.rotr = rotr
rotl.i32 = bit32.lrotate
rotr.i32 = bit32.rrotate
shl.i32 = bit32.lshift
shl.i64 = bit32.lshift
shl.u32 = bit32.lshift
shl.u64 = bit32.lshift
shr.i32 = bit32.rshift
shr.i64 = bit32.rshift
shr.u32 = bit32.rshift
shr.u64 = bit32.rshift
end
do
local extend = {}
local wrap = {}
module.extend = extend
module.wrap = wrap
extend.i32_u64 = no_op
function wrap.i64_i32(i) return i % 2 ^ 32 end
end
do
local load = {}
local store = {}
module.load = load
module.store = store
local function rip_u64(x) return math.floor(x / 0x100000000), x % 0x100000000 end
local function merge_u64(hi, lo) return hi * 0x100000000 + lo end
local function black_mask_byte(value, offset)
local mask = bit32.lshift(0xFF, offset * 8)
return bit32.band(value, bit32.bnot(mask))
end
local function load_byte(memory, addr)
local offset = addr % 4
local value = memory.data[(addr - offset) / 4] or 0
return bit32.band(bit32.rshift(value, offset * 8), 0xFF)
end
local function store_byte(memory, addr, value)
local offset = addr % 4
local adjust = (addr - offset) / 4
local lhs = bit32.lshift(bit32.band(value, 0xFF), offset * 8)
local rhs = black_mask_byte(memory.data[adjust] or 0, offset)
memory.data[adjust] = bit32.bor(lhs, rhs)
end
function load.i32_i8(memory, addr)
local b = load_byte(memory, addr)
if b > 0x7F then b = b - 0x100 end
return b
end
load.i32_u8 = load_byte
function load.i32(memory, addr)
if addr % 4 == 0 then
-- aligned read
return memory.data[addr / 4] or 0
else
-- unaligned read
local b1 = load_byte(memory, addr)
local b2 = bit32.lshift(load_byte(memory, addr + 1), 8)
local b3 = bit32.lshift(load_byte(memory, addr + 2), 16)
local b4 = bit32.lshift(load_byte(memory, addr + 3), 24)
return bit32.bor(b1, b2, b3, b4)
end
end
function load.i64(memory, addr)
local hi = load.i32(memory, addr + 4)
local lo = load.i32(memory, addr)
return merge_u64(hi, lo)
end
store.i32_n8 = store_byte
function store.i32(memory, addr, value)
if addr % 4 == 0 then
-- aligned write
memory.data[addr / 4] = value
else
-- unaligned write
store_byte(memory, addr, value)
store_byte(memory, addr + 1, bit32.rshift(value, 8))
store_byte(memory, addr + 2, bit32.rshift(value, 16))
store_byte(memory, addr + 3, bit32.rshift(value, 24))
end
end
function store.i64(memory, addr, value)
local hi, lo = rip_u64(value)
store.i32(memory, addr, lo)
store.i32(memory, addr + 4, hi)
end
end
do
local memory = {}
module.memory = memory
function memory.new(min, max) return {min = min, max = max, data = {}} end
function memory.init(memory, offset, data)
local store_i8 = module.store.i32_n8
local store_i32 = module.store.i32
local len = #data
local rem = len % 4
for i = 1, len - rem, 4 do
local v = string.unpack('<I4', data, i)
store_i32(memory, offset + i - 1, v)
end
for i = len - rem + 1, len do
local v = string.byte(data, i)
store_i8(memory, offset + i - 1, v)
end
end
function memory.size(memory) return memory.min end
function memory.grow(memory, num)
local old = memory.min
local new = old + num
if new > memory.max then
return -1
else
memory.min = new
return old
end
end
end
return module