aboutsummaryrefslogtreecommitdiff
path: root/src/server/reader.odin
blob: 1a32b6f47c12a250cb13e7246df015f4af49aac6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package server

import "core:os"
import "core:mem"
import "core:strings"

ReaderFn :: proc(_: rawptr, _: []byte) -> (int, int)

Reader :: struct {
	reader_fn:      ReaderFn,
	reader_context: rawptr,
}

make_reader :: proc(reader_fn: ReaderFn, reader_context: rawptr) -> Reader {
	return Reader{reader_context = reader_context, reader_fn = reader_fn}
}

read_u8 :: proc(reader: ^Reader) -> (u8, bool) {
	value: [1]byte

	read, err := reader.reader_fn(reader.reader_context, value[:])

	if (err != 0 || read != 1) {
		return 0, false
	}

	return value[0], true
}

read_until_delimiter :: proc(
	reader: ^Reader,
	delimiter: u8,
	builder: ^strings.Builder,
) -> bool {
	for true {

		value, success := read_u8(reader)

		if (!success) {
			return false
		}

		strings.write_byte(builder, value)

		if (value == delimiter) {
			break
		}
	}

	return true
}

read_sized :: proc(reader: ^Reader, data: []u8) -> (ok: bool) {
	ok = true
	size := len(data)
	n := 0

	for n < size && ok {
		read: int
		err_code: int

		read, err_code = reader.reader_fn(reader.reader_context, data[n:])

		ok = err_code == 0

		n += read
	}

	if n >= size {
		ok = true
	}

	return
}