diff --git a/runtime/luajit.lua b/runtime/luajit.lua index 6929218..986d026 100644 --- a/runtime/luajit.lua +++ b/runtime/luajit.lua @@ -8,8 +8,12 @@ local module = {} local vla_u8 = ffi.typeof('uint8_t[?]') local ptr_i8 = ffi.typeof('int8_t *') -local ptr_i64 = ffi.typeof('int64_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 u32 = ffi.typeof('uint32_t') local u64 = ffi.typeof('uint64_t') @@ -26,6 +30,15 @@ do return math.floor(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 math.floor(lhs / rhs) + end end do @@ -98,24 +111,28 @@ do 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 - function ne.i32(lhs, rhs) return to_boolean(lhs ~= rhs) end - function ne.i64(lhs, rhs) return to_boolean(lhs ~= rhs) end end do @@ -166,6 +183,7 @@ do rotl.i64 = bit.rol rotr.i32 = bit.ror rotr.i64 = bit.ror + shl.i32 = bit.lshift shl.i64 = bit.lshift shl.u32 = bit.lshift @@ -182,7 +200,7 @@ do module.wrap = wrap extend.i32_u64 = i64 - wrap.i64_i32 = i32 + wrap.i64_i32 = function(num) return tonumber(i32(num)) end end do @@ -194,10 +212,20 @@ do function load.i32_i8(memory, addr) return ffi.cast(ptr_i8, memory.data)[addr] end function load.i32_u8(memory, addr) return memory.data[addr] end + function load.i32_u16(memory, addr) return ffi.cast(ptr_u16, memory.data + addr)[0] end function load.i32(memory, addr) return ffi.cast(ptr_i32, memory.data + addr)[0] end + + function load.i64_u8(memory, addr) return memory.data[addr] end + function load.i64_u16(memory, addr) return ffi.cast(ptr_u16, memory.data + addr)[0] end + function load.i64_u32(memory, addr) return ffi.cast(ptr_u32, memory.data + addr)[0] end function load.i64(memory, addr) return ffi.cast(ptr_i64, memory.data + addr)[0] end + function store.i32_n8(memory, addr, value) memory.data[addr] = value end + function store.i32_n16(memory, addr, value) ffi.cast(ptr_i16, memory.data + addr)[0] = value end function store.i32(memory, addr, value) ffi.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) ffi.cast(ptr_i16, memory.data + addr)[0] = value end function store.i64_n32(memory, addr, value) ffi.cast(ptr_i32, memory.data + addr)[0] = value end function store.i64(memory, addr, value) ffi.cast(ptr_i64, memory.data + addr)[0] = value end end