52 lines
		
	
	
		
			919 B
		
	
	
	
		
			V
		
	
	
			
		
		
	
	
			52 lines
		
	
	
		
			919 B
		
	
	
	
		
			V
		
	
	
import net
 | 
						|
 | 
						|
const (
 | 
						|
	test_port = 45123
 | 
						|
)
 | 
						|
 | 
						|
fn handle_conn(mut c net.TcpConn) {
 | 
						|
	for {
 | 
						|
		mut buf := []byte{len: 100, init: 0}
 | 
						|
		read := c.read(mut buf) or {
 | 
						|
			println('Server: connection dropped')
 | 
						|
			return
 | 
						|
		}
 | 
						|
		c.write(buf[..read]) or {
 | 
						|
			println('Server: connection dropped')
 | 
						|
			return
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
fn echo_server(mut l net.TcpListener) ? {
 | 
						|
	for {
 | 
						|
		mut new_conn := l.accept() or { continue }
 | 
						|
		go handle_conn(mut new_conn)
 | 
						|
	}
 | 
						|
	return none
 | 
						|
}
 | 
						|
 | 
						|
fn echo() ? {
 | 
						|
	mut c := net.dial_tcp('127.0.0.1:$test_port') ?
 | 
						|
	defer {
 | 
						|
		c.close() or { }
 | 
						|
	}
 | 
						|
	data := 'Hello from vlib/net!'
 | 
						|
	c.write_str(data) ?
 | 
						|
	mut buf := []byte{len: 4096}
 | 
						|
	read := c.read(mut buf) ?
 | 
						|
	assert read == data.len
 | 
						|
	for i := 0; i < read; i++ {
 | 
						|
		assert buf[i] == data[i]
 | 
						|
	}
 | 
						|
	println('Got "$buf.bytestr()"')
 | 
						|
	return none
 | 
						|
}
 | 
						|
 | 
						|
fn test_tcp() {
 | 
						|
	mut l := net.listen_tcp(test_port) or { panic(err) }
 | 
						|
	go echo_server(mut l)
 | 
						|
	echo() or { panic(err) }
 | 
						|
	l.close() or { }
 | 
						|
}
 |