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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
import io
import net
import strings
const (
server_port = ':22443'
)
fn accept(mut server net.TcpListener, c chan &net.TcpConn) {
c <- server.accept() or { panic(err) }
}
fn setup() (&net.TcpListener, &net.TcpConn, &net.TcpConn) {
mut server := net.listen_tcp(.ip6, server_port) or { panic(err) }
c := chan &net.TcpConn{}
go accept(mut server, c)
mut client := net.dial_tcp('localhost$server_port') or { panic(err) }
socket := <-c
$if debug_peer_ip ? {
eprintln('$server.addr()\n$client.peer_addr(), $client.addr()\n$socket.peer_addr(), $socket.addr()')
}
assert true
return server, client, socket
}
fn cleanup(mut server net.TcpListener, mut client net.TcpConn, mut socket net.TcpConn) {
server.close() or {}
client.close() or {}
socket.close() or {}
}
fn test_socket() {
mut server, mut client, mut socket := setup()
defer {
cleanup(mut server, mut client, mut socket)
}
message := 'Hello World'
socket.write_string(message) or {
assert false
return
}
assert true
$if debug {
println('message send: $message')
}
$if debug {
println('send socket: $socket.sock.handle')
}
mut buf := []byte{len: 1024}
nbytes := client.read(mut buf) or {
assert false
return
}
received := buf[0..nbytes].bytestr()
$if debug {
println('message received: $received')
}
$if debug {
println('client: $client.sock.handle')
}
assert message == received
}
fn test_socket_write_and_read() {
mut server, mut client, mut socket := setup()
defer {
cleanup(mut server, mut client, mut socket)
}
message1 := 'a message 1'
socket.write_string(message1) or { assert false }
mut rbuf := []byte{len: message1.len}
client.read(mut rbuf) or {
assert false
return
}
line := rbuf.bytestr()
assert line == message1
}
fn test_socket_read_line() {
mut server, mut client, mut socket := setup()
mut reader := io.new_buffered_reader(reader: client)
defer {
cleanup(mut server, mut client, mut socket)
}
message1, message2 := 'message1', 'message2'
message := '$message1\n$message2\n'
socket.write_string(message) or { assert false }
assert true
//
line1 := reader.read_line() or {
// println(reader.buf)
assert false
return
}
line2 := reader.read_line() or {
// println(reader.buf)
assert false
return
}
assert line1 == message1
assert line2 == message2
}
fn test_socket_write_fail_without_panic() {
mut server, mut client, mut socket := setup()
defer {
cleanup(mut server, mut client, mut socket)
}
message2 := 'a message 2'
// ensure that socket.write (i.e. done on the server side)
// continues to work, even when the client side has been disconnected
// this test is important for a stable long standing server
client.close() or {}
$if solaris {
return
}
// TODO: fix segfaulting on Solaris
for i := 0; i < 3; i++ {
socket.write_string(message2) or {
println('write to a socket without a recipient should produce an option fail: $err | $message2')
assert true
}
}
}
fn test_socket_read_line_long_line_without_eol() {
mut server, mut client, mut socket := setup()
mut reader := io.new_buffered_reader(reader: client)
defer {
cleanup(mut server, mut client, mut socket)
}
message := strings.repeat_string('123', 400)
socket.write_string(message) or {
assert false
return
}
socket.write_string('\n') or {
assert false
return
}
line := reader.read_line() or {
assert false
return
}
assert line == message
}
|
