656 lines
17 KiB
Plaintext
656 lines
17 KiB
Plaintext
extern malloc
|
|
extern realloc
|
|
extern free
|
|
extern gethostbyname
|
|
|
|
func dbg.panic[msg: str] : void
|
|
io.print("PANIC: ")
|
|
io.println(msg)
|
|
0/0 // crashes program which is kinda better since you get a gdb backtrace
|
|
os.exit(1)
|
|
|
|
func mem.alloc[x: i64] : ptr
|
|
return malloc(x)
|
|
|
|
func mem.free[x: ptr] : void
|
|
free(x)
|
|
|
|
func mem.zero[x: i64, size: i64] : void
|
|
for i in 0..size
|
|
mem.write8(x + i, 0)
|
|
|
|
func mem.read8[x: ptr] : u8
|
|
return _builtin_read8(x)
|
|
|
|
func mem.read16[x: ptr] : i64
|
|
let low: i64 = mem.read8(x)
|
|
let high: i64 = mem.read8(x + 1)
|
|
return low | (high << 8)
|
|
|
|
func mem.read64[x: ptr] : i64
|
|
return _builtin_read64(x)
|
|
|
|
func mem.write8[x: ptr, d: u8] : void
|
|
_builtin_set8(x, d)
|
|
|
|
func mem.write64[x: ptr, d: i64] : void
|
|
_builtin_set64(x, d)
|
|
|
|
func io.print_sized[x: str, size: i64] : void
|
|
_builtin_syscall(1, 1, x, size) // write
|
|
|
|
func io.print[x: str] : void
|
|
io.print_sized(x, str.len(x))
|
|
|
|
func io.println[x: str] : void
|
|
io.print(x)
|
|
io.print("\n")
|
|
|
|
func io.print_char[x: u8] : void
|
|
io.print_sized(^x, 1)
|
|
|
|
func io.print_i64[x: i64] : void
|
|
let s: str = str.from_i64(x)
|
|
io.print(s)
|
|
mem.free(s)
|
|
|
|
func io.println_i64[x: i64] : void
|
|
let s: str = str.from_i64(x)
|
|
io.println(s)
|
|
mem.free(s)
|
|
|
|
func io.read_char[] : u8
|
|
let c: u8 = 0
|
|
_builtin_syscall(0, 0, ^c, 1) // read
|
|
return c
|
|
|
|
func io.read_line[]: str
|
|
let MAX_SIZE = 60000
|
|
let buffer: str = mem.alloc(MAX_SIZE + 1)
|
|
let n: i64 = _builtin_syscall(0, 0, buffer, MAX_SIZE) // read
|
|
if n < 0
|
|
return ""
|
|
str.set(buffer, n, 0)
|
|
return buffer
|
|
|
|
func io.read_file[path: str]: str
|
|
let fd: i64 = _builtin_syscall(257, -100, path, 0, 0) // openat
|
|
if fd <= 0
|
|
dbg.panic("failed to open file")
|
|
|
|
let size: i64 = _builtin_syscall(8, fd, 0, 2) // lseek to the end
|
|
_builtin_syscall(8, fd, 0, 0) // lseek back to start
|
|
|
|
let buffer: str = mem.alloc(size + 1)
|
|
let n: i64 = _builtin_syscall(0, fd, buffer, size) // read
|
|
str.set(buffer, n, 0)
|
|
_builtin_syscall(3, fd) // close
|
|
return buffer
|
|
|
|
func io.write_file[path: str, content: str] : void
|
|
let fd: ptr = _builtin_syscall(257, -100, path, 0x241, 0o644) // openat
|
|
if fd < 0
|
|
dbg.panic("failed to open file")
|
|
|
|
_builtin_syscall(1, fd, content, str.len(content)) // write
|
|
_builtin_syscall(3, fd) // close
|
|
|
|
func str.len[s: str] : i64
|
|
let i = 0
|
|
while mem.read8(s + i)
|
|
i = i + 1
|
|
return i
|
|
|
|
func str.copy[s: str] : str
|
|
let size: i64 = str.len(s) + 1
|
|
let dup: str = mem.alloc(size)
|
|
for i in 0..size
|
|
str.set(dup, i, s[i])
|
|
return dup
|
|
|
|
func str.set[s: str, n: i64, c: u8] : void
|
|
mem.write8(s + n, c)
|
|
|
|
func str.equal[a: str, b: str] : bool
|
|
let i = 0
|
|
while a[i] != 0 && b[i] != 0
|
|
if a[i] != b[i]
|
|
return false
|
|
i = i + 1
|
|
return a[i] == b[i]
|
|
|
|
func str.is_whitespace[x: u8] : bool
|
|
return x == ' ' || x == 10 || x == 13 || x == 9
|
|
|
|
func str.is_digit[x: u8] : bool
|
|
return x >= '0' && x <= '9'
|
|
|
|
func str.is_hex_digit[x: u8] : bool
|
|
return (x >= '0' && x <= '9') || (x >= 'a' && x <= 'f') || (x >= 'A' && x <= 'F')
|
|
|
|
func str.is_lowercase[x: u8] : bool
|
|
return x >= 'a' && x <= 'z'
|
|
|
|
func str.is_uppercase[x: u8] : bool
|
|
return x >= 'A' && x <= 'Z'
|
|
|
|
func str.is_letter[x: u8] : bool
|
|
return str.is_uppercase(x) || str.is_lowercase(x)
|
|
|
|
func str.is_alphanumeric[x: u8] : bool
|
|
return str.is_letter(x) || str.is_digit(x)
|
|
|
|
func str.concat[a: str, b: str] : str
|
|
let a_len: i64 = str.len(a)
|
|
let b_len: i64 = str.len(b)
|
|
let out: str = mem.alloc(a_len + b_len + 1)
|
|
for i in 0..a_len
|
|
str.set(out, i, a[i])
|
|
for i in 0..b_len
|
|
str.set(out, a_len + i, b[i])
|
|
str.set(out, a_len + b_len, 0)
|
|
return out
|
|
|
|
func str.find[haystack: str, needle: str] : i64
|
|
let haystack_len: i64 = str.len(haystack)
|
|
let needle_len: i64 = str.len(needle)
|
|
|
|
if needle_len == 0
|
|
return 0
|
|
|
|
for i in 0..(haystack_len - needle_len + 1)
|
|
let match: bool = true
|
|
for j in 0..needle_len
|
|
if haystack[i + j] != needle[j]
|
|
match = false
|
|
break
|
|
if match
|
|
return i
|
|
return -1
|
|
|
|
func str.substr[s: str, start: i64, length: i64] : str
|
|
if start < 0 || length < 0 || start + length > str.len(s)
|
|
dbg.panic("str.substr out of bounds")
|
|
|
|
let out: str = mem.alloc(length + 1)
|
|
for i in 0..length
|
|
str.set(out, i, s[start + i])
|
|
str.set(out, length, 0)
|
|
return out
|
|
|
|
func str.trim[s: str] : str
|
|
let len: i64 = str.len(s)
|
|
if len == 0
|
|
return ""
|
|
|
|
let start = 0
|
|
let end: i64 = len - 1
|
|
|
|
while start <= end && str.is_whitespace(s[start])
|
|
start = start + 1
|
|
|
|
while end >= start && str.is_whitespace(s[end])
|
|
end = end - 1
|
|
|
|
return str.substr(s, start, end - start + 1)
|
|
|
|
func str.split[haystack: str, needle: str]: array
|
|
let haystack_len: i64 = str.len(haystack)
|
|
let needle_len: i64 = str.len(needle)
|
|
let result: array = []
|
|
|
|
if !needle_len
|
|
if !haystack_len
|
|
return result
|
|
else
|
|
for i in 0..haystack_len
|
|
array.push(result, str.substr(haystack, i, 1))
|
|
return result
|
|
|
|
let start = 0
|
|
let i = 0
|
|
while i < haystack_len
|
|
if i <= haystack_len - needle_len
|
|
let match: bool = true
|
|
for j in 0..needle_len
|
|
if haystack[i + j] != needle[j]
|
|
match = false
|
|
break
|
|
if match
|
|
array.push(result, str.substr(haystack, start, i - start))
|
|
start = i + needle_len
|
|
i = i + needle_len
|
|
continue
|
|
i = i + 1
|
|
|
|
array.push(result, str.substr(haystack, start, haystack_len - start))
|
|
return result
|
|
|
|
func str.reverse[s: str] : str
|
|
let len: i64 = str.len(s)
|
|
let out: str = mem.alloc(len + 1)
|
|
|
|
for i in 0..len
|
|
str.set(out, i, s[len - i - 1])
|
|
str.set(out, len, 0)
|
|
return out
|
|
|
|
// not sure this covers all wacky edge cases
|
|
func str.from_i64[n: i64] : str
|
|
if n == 0
|
|
return str.copy("0")
|
|
|
|
let neg: bool = n < 0
|
|
if neg
|
|
n = -n
|
|
let buf: str = mem.alloc(21) // enough to fit -MAX_I64
|
|
let i = 0
|
|
while n > 0
|
|
let d: u8 = n % 10
|
|
str.set(buf, i, '0' + d)
|
|
n = n / 10
|
|
i = i + 1
|
|
if neg
|
|
str.set(buf, i, '-')
|
|
i = i + 1
|
|
str.set(buf, i, 0)
|
|
let s: str = str.reverse(buf)
|
|
mem.free(buf)
|
|
return s
|
|
|
|
func str.from_char[c: u8] : str
|
|
let s: str = mem.alloc(2)
|
|
str.set(s, 0, c)
|
|
str.set(s, 1, 0)
|
|
return s
|
|
|
|
func str.parse_i64[s: str] : i64
|
|
let len: i64 = str.len(s)
|
|
let i = 0
|
|
|
|
let sign = 1
|
|
if i < len && s[i] == '-'
|
|
sign = -1
|
|
i = i + 1
|
|
|
|
let num = 0
|
|
while i < len
|
|
let d: u8 = s[i]
|
|
if d < '0' || d > '9'
|
|
break
|
|
num = num * 10 + (d - '0')
|
|
i = i + 1
|
|
return num * sign
|
|
|
|
func str.hex_encode[s: str] : str
|
|
let hex_chars: str = "0123456789abcdef"
|
|
let s_len: i64 = str.len(s)
|
|
let j = 0
|
|
let out: str = mem.alloc(s_len * 2 + 1)
|
|
|
|
for i in 0..s_len
|
|
let high: u8 = (s[i] >> 4) & 15
|
|
let low: u8 = s[i] & 15
|
|
str.set(out, j, hex_chars[high])
|
|
str.set(out, j + 1, hex_chars[low])
|
|
j = j + 2
|
|
|
|
str.set(out, j, 0)
|
|
return out
|
|
|
|
func str._hex_digit_to_int[d: u8] : i64
|
|
if d >= 'a' && d <= 'f'
|
|
return d - 'a' + 10
|
|
if d >= 'A' && d <= 'F'
|
|
return d - 'A' + 10
|
|
return d - '0'
|
|
|
|
func str.hex_decode[s: str] : str
|
|
let s_len: i64 = str.len(s)
|
|
let i = 0
|
|
let j = 0
|
|
let out: str = mem.alloc(s_len / 2 + 1)
|
|
|
|
while i < s_len
|
|
str.set(out, j, str._hex_digit_to_int(s[i]) * 16 + str._hex_digit_to_int(s[i + 1]))
|
|
i = i + 2
|
|
j = j + 1
|
|
|
|
str.set(out, j, 0)
|
|
return out
|
|
|
|
func math.gcd[a: i64, b: i64] : i64
|
|
while b != 0
|
|
let tmp: i64 = b
|
|
b = a % b
|
|
a = tmp
|
|
return a
|
|
|
|
func math.min[a: i64, b: i64] : i64
|
|
if a < b
|
|
return a
|
|
return b
|
|
|
|
func math.max[a: i64, b: i64] : i64
|
|
if a > b
|
|
return a
|
|
return b
|
|
|
|
func math.abs[n: i64] : i64
|
|
if n < 0
|
|
return -n
|
|
return n
|
|
|
|
func math.sign[n: i64] : i64
|
|
if n < 0
|
|
return -1
|
|
else if n > 0
|
|
return 1
|
|
else
|
|
return 0
|
|
|
|
func math.pow[b: i64, e: i64] : i64
|
|
let out = 1
|
|
for i in 0..e
|
|
out = out * b
|
|
return out
|
|
|
|
func math.lcm[a: i64, b: i64] : i64
|
|
return (a * b) / math.gcd(a, b)
|
|
|
|
func math.isqrt[n: i64] : i64
|
|
if n < 0
|
|
dbg.panic("negative number passed to math.isqrt")
|
|
if n == 0 || n == 1
|
|
return n
|
|
|
|
let guess: i64 = n
|
|
let next_guess: i64 = (guess + n / guess) / 2
|
|
|
|
while next_guess < guess
|
|
guess = next_guess
|
|
next_guess = (guess + n / guess) / 2
|
|
|
|
return guess
|
|
|
|
func math.is_prime[n: i64]: bool
|
|
if n <= 1
|
|
return false
|
|
if n == 2 || n == 3
|
|
return true
|
|
if n % 2 == 0 || n % 3 == 0
|
|
return false
|
|
|
|
let i = 5
|
|
while i * i <= n
|
|
if n % i == 0 || n % (i + 2) == 0
|
|
return false
|
|
i = i + 6
|
|
return true
|
|
|
|
func array.new[] : array
|
|
// [ 8 bytes - ptr to data ] [ 8 bytes - size ] [ 8 bytes - capacity ]
|
|
let arr: ptr = mem.alloc(24)
|
|
mem.zero(arr, 24)
|
|
return arr
|
|
|
|
func array.nth[xs: array, n: i64] : i64
|
|
if n < 0 || n >= array.size(xs)
|
|
dbg.panic("array.nth out of bounds")
|
|
let data: ptr = mem.read64(xs)
|
|
return mem.read64(data + n * 8)
|
|
|
|
func array.set[xs: array, n: i64, x: i64] : void
|
|
if n < 0 || n >= array.size(xs)
|
|
dbg.panic("array.set out of bounds")
|
|
let data: ptr = mem.read64(xs)
|
|
mem.write64(data + n * 8, x)
|
|
|
|
func array.push[xs: array, x: i64] : void
|
|
let data: ptr = mem.read64(xs)
|
|
let capacity: i64 = mem.read64(xs + 8)
|
|
let size: i64 = mem.read64(xs + 16)
|
|
|
|
if size == capacity
|
|
let new_capacity = 4
|
|
if capacity != 0
|
|
new_capacity = capacity * 2
|
|
let new_data: ptr = realloc(data, new_capacity * 8)
|
|
mem.write64(xs, new_data)
|
|
mem.write64(xs + 8, new_capacity)
|
|
data = new_data
|
|
|
|
mem.write64(data + size * 8, x)
|
|
mem.write64(xs + 16, size + 1)
|
|
|
|
func array.size[xs: array] : i64
|
|
return mem.read64(xs + 16)
|
|
|
|
func array.free[xs: array] : void
|
|
let data: ptr = mem.read64(xs)
|
|
if data != 0
|
|
mem.free(data)
|
|
mem.free(xs)
|
|
|
|
func array.pop[xs: array] : i64
|
|
let size: i64 = array.size(xs)
|
|
if size == 0
|
|
dbg.panic("array.pop on empty array")
|
|
let x: i64 = array.nth(xs, size - 1)
|
|
mem.write64(xs + 16, size - 1)
|
|
return x
|
|
|
|
func array.slice[xs: array, start: i64, length: i64] : array
|
|
if start < 0 || length < 0 || start + length > array.size(xs)
|
|
dbg.panic("array.slice out of bounds")
|
|
|
|
let new_array: array = []
|
|
for i in 0..length
|
|
array.push(new_array, array.nth(xs, start + i))
|
|
return new_array
|
|
|
|
func array.concat[a: array, b: array] : array
|
|
let new_array: array = []
|
|
for i in 0..array.size(a)
|
|
array.push(new_array, array.nth(a, i))
|
|
for i in 0..array.size(b)
|
|
array.push(new_array, array.nth(b, i))
|
|
return new_array
|
|
|
|
func alg.quicksort[arr: array] : void
|
|
alg._do_quicksort(arr, 0, array.size(arr) - 1)
|
|
|
|
func alg._do_quicksort[arr: array, low: i64, high: i64] : void
|
|
if low < high
|
|
let i: i64 = alg._partition(arr, low, high)
|
|
alg._do_quicksort(arr, low, i - 1)
|
|
alg._do_quicksort(arr, i + 1, high)
|
|
|
|
func alg._partition[arr: array, low: i64, high: i64] : i64
|
|
let pivot: i64 = array.nth(arr, high)
|
|
let i: i64 = low - 1
|
|
for j in (low)..high
|
|
if array.nth(arr, j) <= pivot
|
|
i = i + 1
|
|
let temp: i64 = array.nth(arr ,i)
|
|
array.set(arr, i, array.nth(arr, j))
|
|
array.set(arr, j, temp)
|
|
let temp: i64 = array.nth(arr, i + 1)
|
|
array.set(arr, i + 1, array.nth(arr, high))
|
|
array.set(arr, high, temp)
|
|
return i + 1
|
|
|
|
func alg.count[arr: array, item: i64] : i64
|
|
let count = 0
|
|
let size: i64 = array.size(arr)
|
|
for i in 0..size
|
|
if array.nth(arr, i) == item
|
|
count = count + 1
|
|
return count
|
|
|
|
func alg.map[arr: array, fn: ptr] : array
|
|
let out: array = []
|
|
for i in 0..array.size(arr)
|
|
array.push(out, fn(array.nth(arr, i)))
|
|
return out
|
|
|
|
func alg.filter[arr: array, fn: ptr] : array
|
|
let out: array = []
|
|
for i in 0..array.size(arr)
|
|
if fn(array.nth(arr, i))
|
|
array.push(out, array.nth(arr, i))
|
|
return out
|
|
|
|
func alg.reduce[arr: array, fn: ptr, acc: i64] : i64
|
|
for i in 0..array.size(arr)
|
|
acc = fn(acc, array.nth(arr, i))
|
|
return acc
|
|
|
|
func os.exit[code: i64] : void
|
|
_builtin_syscall(60, code)
|
|
|
|
func os.urandom[]: i64
|
|
let n = 0
|
|
let fd: i64 = _builtin_syscall(257, -100, "/dev/urandom", 0, 0) // openat
|
|
_builtin_syscall(0, fd, ^n, 8) // read
|
|
_builtin_syscall(3, fd) // close
|
|
return n
|
|
|
|
func os.time[] : i64
|
|
let tv: ptr = mem.alloc(16)
|
|
_builtin_syscall(96, tv, 0) // gettimeofday
|
|
let seconds: i64 = mem.read64(tv)
|
|
let microseconds: i64 = mem.read64(tv + 8)
|
|
mem.free(tv)
|
|
return seconds * 1000 + microseconds / 1000
|
|
|
|
// voodoo magic
|
|
func os.shell[command: str] : i64
|
|
let pid: i64 = _builtin_syscall(57) // fork
|
|
if pid == 0
|
|
// leaky but not sure where can i free it
|
|
let argv: array = ["sh", "-c", command, 0]
|
|
_builtin_syscall(59, "/bin/sh", mem.read64(argv), _builtin_environ()) // execve
|
|
_builtin_syscall(60, 1) // exit
|
|
else
|
|
let status = 0
|
|
let wp: i64 = _builtin_syscall(61, pid, ^status, 0, 0) // waitpid
|
|
if wp == -1
|
|
return -1
|
|
let st: i64 = status & 0xffffffff
|
|
|
|
if (st & 0x7f) == 0
|
|
return (st >> 8) & 0xff
|
|
else
|
|
return -(st & 0x7f)
|
|
|
|
func os.listdir[path: str] : array
|
|
let fd: i64 = _builtin_syscall(257, -100, path, 0, 0) // openat
|
|
if fd < 0
|
|
return []
|
|
|
|
let files: array = []
|
|
let buf: ptr = mem.alloc(1024)
|
|
while true
|
|
let n: i64 = _builtin_syscall(217, fd, buf, 1024) // getdents64
|
|
if n <= 0
|
|
break
|
|
|
|
let pos = 0
|
|
while pos < n
|
|
let len: i64 = mem.read16(buf + pos + 16)
|
|
let name: str = buf + pos + 19
|
|
if name[0]
|
|
let skip: bool = false
|
|
// skip if name is exactly '.' or '..'
|
|
if name[0] == '.'
|
|
if name[1] == 0
|
|
skip = true
|
|
else if name[1] == '.'
|
|
if name[2] == 0
|
|
skip = true
|
|
if !skip
|
|
array.push(files, str.copy(name))
|
|
pos = pos + len
|
|
|
|
mem.free(buf)
|
|
_builtin_syscall(3, fd) // close
|
|
return files
|
|
|
|
func net.listen[packed_host: i64, port: i64] : i64
|
|
let s: i64 = _builtin_syscall(41, 2, 1, 0) // socket
|
|
if s < 0
|
|
return -1
|
|
|
|
let optval = 1
|
|
if _builtin_syscall(54, s, 1, 2, ^optval, 8) < 0 // setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
|
|
_builtin_syscall(3, s) // close
|
|
return -1
|
|
|
|
let sa: ptr = mem.alloc(16)
|
|
mem.zero(sa, 16)
|
|
mem.write8(sa + 0, 2)
|
|
mem.write8(sa + 1, 0)
|
|
mem.write8(sa + 2, (port >> 8) & 255)
|
|
mem.write8(sa + 3, port & 255)
|
|
mem.write8(sa + 4, (packed_host >> 24) & 255)
|
|
mem.write8(sa + 5, (packed_host >> 16) & 255)
|
|
mem.write8(sa + 6, (packed_host >> 8) & 255)
|
|
mem.write8(sa + 7, packed_host & 255)
|
|
|
|
if _builtin_syscall(49, s, sa, 16) < 0 // bind
|
|
_builtin_syscall(3, s) // close
|
|
mem.free(sa)
|
|
return -1
|
|
mem.free(sa)
|
|
|
|
if _builtin_syscall(50, s, 1) < 0 // listen
|
|
_builtin_syscall(3, s) // close
|
|
return -1
|
|
|
|
return s
|
|
|
|
func net.connect[host: str, port: i64] : i64
|
|
let he: ptr = gethostbyname(host)
|
|
if he == 0
|
|
return -1
|
|
|
|
let ip_ptr: ptr = mem.read64(mem.read64(he + 24))
|
|
|
|
let s: i64 = _builtin_syscall(41, 2, 1, 0) // socket
|
|
if s < 0
|
|
return -1
|
|
|
|
let sa: ptr = mem.alloc(16)
|
|
mem.zero(sa, 16)
|
|
mem.write8(sa + 0, 2)
|
|
mem.write8(sa + 2, (port >> 8) & 255)
|
|
mem.write8(sa + 3, port & 255)
|
|
mem.write8(sa + 4, ip_ptr[0])
|
|
mem.write8(sa + 5, ip_ptr[1])
|
|
mem.write8(sa + 6, ip_ptr[2])
|
|
mem.write8(sa + 7, ip_ptr[3])
|
|
|
|
if _builtin_syscall(42, s, sa, 16) < 0 // connect
|
|
mem.free(sa)
|
|
_builtin_syscall(3, s) // close
|
|
return -1
|
|
|
|
mem.free(sa)
|
|
return s
|
|
|
|
func net.accept[s: i64] : i64
|
|
return _builtin_syscall(43, s, 0, 0)
|
|
|
|
func net.send[s: i64, data: str, size: i64] : void
|
|
_builtin_syscall(44, s, data, size, 0, 0, 0)
|
|
|
|
func net.read[s: i64, buffer: ptr, size: i64] : i64
|
|
return _builtin_syscall(0, s, buffer, size)
|
|
|
|
func net.close[s: i64] : void
|
|
_builtin_syscall(3, s)
|
|
|
|
func net.pack_addr[a: i64, b: i64, c: i64, d: i64] : i64
|
|
return (a << 24) | (b << 16) | (c << 8) | d
|