toni/zern
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

net.pack_addr, SO_REUSEADDR

Browse Source
This commit is contained in:
Antoni Piasecki
2025-12-21 17:28:19 +01:00
parent d55ca11658
commit 7c23e57ec0
3 changed files with 45 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
examples/sqlite_todo.zr
View File

@@ -12,16 +12,16 @@ extern sqlite3_finalize
func main[] : I64
let rc: I64 = 0
let db: Ptr = mem.alloc(8)
let stmt: Ptr = mem.alloc(8)
let db: I64 = 0
let stmt: I64 = 0
rc = sqlite3_open("todo.db", db)
rc = sqlite3_open("todo.db", @db)
if rc
dbg.panic("failed to open db")
rc = sqlite3_exec(mem.read64(db), "CREATE TABLE IF NOT EXISTS todo(id INTEGER PRIMARY KEY AUTOINCREMENT, task TEXT NOT NULL);", 0, 0, 0)
rc = sqlite3_exec(db, "CREATE TABLE IF NOT EXISTS todo(id INTEGER PRIMARY KEY AUTOINCREMENT, task TEXT NOT NULL);", 0, 0, 0)
if rc
dbg.panic(sqlite3_errmsg(mem.read64(db)))
dbg.panic(sqlite3_errmsg(db))
while true
io.println("1. List tasks")
@@ -36,11 +36,11 @@ func main[] : I64
break
else if choice == 1
io.println("============")
sqlite3_prepare_v2(mem.read64(db), "SELECT * FROM todo", -1, stmt, 0)
sqlite3_prepare_v2(db, "SELECT * FROM todo", -1, @stmt, 0)
while sqlite3_step(mem.read64(stmt)) == 100
let id: I64 = sqlite3_column_int(mem.read64(stmt), 0)
let task: String = sqlite3_column_text(mem.read64(stmt), 1)
while sqlite3_step(stmt) == 100
let id: I64 = sqlite3_column_int(stmt, 0)
let task: String = sqlite3_column_text(stmt, 1)
io.print_i64(id)
io.print(" - ")
@@ -51,21 +51,21 @@ func main[] : I64
io.print("\nEnter new task: ")
let task: String = io.read_line() |> str.trim()
sqlite3_prepare_v2(mem.read64(db), "INSERT INTO todo(task) VALUES (?);", -1, stmt, 0)
sqlite3_bind_text(mem.read64(stmt), 1, task, -1, 0)
if sqlite3_step(mem.read64(stmt)) != 101
dbg.panic(sqlite3_errmsg(mem.read64(db)))
sqlite3_prepare_v2(db, "INSERT INTO todo(task) VALUES (?);", -1, @stmt, 0)
sqlite3_bind_text(stmt, 1, task, -1, 0)
if sqlite3_step(stmt) != 101
dbg.panic(sqlite3_errmsg(db))
io.println("\nTask added\n")
else if choice == 3
io.print("\nEnter task id: ")
let id: I64 = io.read_line() |> str.parse_i64()
sqlite3_prepare_v2(mem.read64(db), "DELETE FROM todo WHERE id = ?;", -1, stmt, 0)
sqlite3_bind_int(mem.read64(stmt), 1, id, -1, 0)
if sqlite3_step(mem.read64(stmt)) != 101
dbg.panic(sqlite3_errmsg(mem.read64(db)))
sqlite3_prepare_v2(db, "DELETE FROM todo WHERE id = ?;", -1, @stmt, 0)
sqlite3_bind_int(stmt, 1, id, -1, 0)
if sqlite3_step(stmt) != 101
dbg.panic(sqlite3_errmsg(db))
io.println("\nTask deleted\n")
sqlite3_finalize(mem.read64(stmt))
sqlite3_finalize(stmt)

3
examples/tcp_server.zr
View File

@@ -1,5 +1,6 @@
func main[] : I64
let s: I64 = net.listen(8000)
let host: I64 = net.pack_addr(127, 0, 0, 1)
let s: I64 = net.listen(host, 8000)
let resp: String = mem.alloc(60000)
while true

37
src/std.zr
View File

@@ -72,8 +72,9 @@ func io.read_char[] : U8
return c
func io.read_line[]: String
let buffer: String = mem.alloc(1025)
let n: I64 = _builtin_syscall(0, 0, buffer, 1024)
let MAX_SIZE: I64 = 60000
let buffer: String = mem.alloc(MAX_SIZE + 1)
let n: I64 = _builtin_syscall(0, 0, buffer, MAX_SIZE)
if n < 0
return ""
str.set(buffer, n, 0)
@@ -220,7 +221,7 @@ func str.from_i64[n: I64] : String
let neg: Bool = n < 0
if neg
n = -n
let buf: String = mem.alloc(21)
let buf: String = mem.alloc(21) // enough to fit -MAX_I64
let i: I64 = 0
while n > 0
let d: U8 = n % 10
@@ -275,7 +276,7 @@ func str.hex_encode[s: String] : String
str.set(out, j, 0)
return out
func str._from_hex_digit[d: U8] : I64
func str._hex_digit_to_int[d: U8] : I64
if d >= 'a' & d <= 'f'
return d - 'a' + 10
if d >= 'A' & d <= 'F'
@@ -289,7 +290,7 @@ func str.hex_decode[s: String] : String
let out: String = mem.alloc(s_len / 2 + 1)
while i < s_len
str.set(out, j, str._from_hex_digit(s[i]) * 16 + str._from_hex_digit(s[i + 1]))
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
@@ -329,7 +330,7 @@ func math.lcm[a: I64, b: I64] : I64
func math.isqrt[n: I64] : I64
if n < 0
return -1
dbg.panic("negative number passed to math.isqrt")
if n == 0 | n == 1
return n
@@ -358,6 +359,7 @@ func math.is_prime[n: I64]: Bool
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
@@ -453,12 +455,10 @@ func os.exit[code: I64] : Void
_builtin_syscall(60, code)
func os.urandom[]: I64
let buffer: Ptr = mem.alloc(8)
let n: I64 = 0
let fd: I64 = _builtin_syscall(257, -100, "/dev/urandom", 0, 0) // openat
_builtin_syscall(0, fd, buffer, 8) // read
_builtin_syscall(0, fd, @n, 8) // read
_builtin_syscall(3, fd) // close
let n: I64 = mem.read64(buffer)
mem.free(buffer)
return n
func os.time[] : I64
@@ -523,10 +523,15 @@ func os.listdir[path: String] : Array
_builtin_syscall(3, fd) // close
return files
func net.listen[port: I64] : I64
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: I64 = 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)
@@ -534,9 +539,14 @@ func net.listen[port: I64] : I64
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)
@@ -585,4 +595,7 @@ 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)
_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