module ssl

import io
import net
import net.mbedtls

const issue_16557_server_cert = @VMODROOT + '/examples/ssl_server/cert/server.crt'
const issue_16557_server_key = @VMODROOT + '/examples/ssl_server/cert/server.key'
const issue_16557_request = 'ping\r\n'
const issue_16557_response_chunk = 'hello over tls'
const issue_16557_response_repeat = 2_000

fn issue_16557_serve_once(mut listener mbedtls.SSLListener) {
	defer {
		listener.shutdown() or {}
	}
	mut conn := listener.accept() or { panic(err) }
	defer {
		conn.shutdown() or {}
	}
	mut request_buf := []u8{len: issue_16557_request.len}
	_ = conn.read(mut request_buf) or { panic(err) }
	response := issue_16557_expected_response()
	mut start := 0
	for start < response.len {
		end := if start + 257 > response.len { response.len } else { start + 257 }
		conn.write_string(response[start..end]) or { panic(err) }
		start = end
	}
}

fn issue_16557_expected_response() string {
	return issue_16557_response_chunk.repeat(issue_16557_response_repeat)
}

fn test_io_read_all_reads_ssl_conn_until_eof() {
	mut port_listener := net.listen_tcp(.ip, '127.0.0.1:0') or { panic(err) }
	port := port_listener.addr() or { panic(err) }.port() or { panic(err) }
	port_listener.close() or { panic(err) }

	mut listener := mbedtls.new_ssl_listener('127.0.0.1:${port}', mbedtls.SSLConnectConfig{
		cert:     issue_16557_server_cert
		cert_key: issue_16557_server_key
		validate: false
	}) or { panic(err) }
	server := spawn issue_16557_serve_once(mut listener)

	mut client := new_ssl_conn(validate: false) or { panic(err) }
	defer {
		client.shutdown() or {}
	}
	client.dial('127.0.0.1', port) or { panic(err) }
	client.write_string(issue_16557_request) or { panic(err) }

	bytes := io.read_all(reader: client) or { panic(err) }
	server.wait()

	assert bytes == issue_16557_expected_response().bytes()
}