Initial project import

This commit is contained in:
Agila Forge Importer committed 2026-07-13 15:20:32 +00:00
commit 23f2d29e32
11957 files changed
+1754577

No files matched your search

+197
View File
@@ -0,0 +1,197 @@
module openssl
#define OPENSSL_API_COMPAT 0x30000000L
// On Linux, prefer a locally built openssl, because it is
// much more likely for it to be newer, than the system
// openssl from libssl-dev. If there is no local openssl,
// the next #pkgconfig flag is harmless, since it will still
// use the (older) system openssl.
#flag linux -I/usr/local/include/openssl
#flag linux -L/usr/local/lib64
// On FreeBSD, prefer openssl from the ports collection, because
// it is much more likely for it to be newer, than the system
// openssl.
#flag freebsd -I/usr/local/include
#flag freebsd -L/usr/local/lib
// Installed through choco:
#flag windows -IC:/Program Files/OpenSSL-Win64/include
#flag windows -LC:/Program Files/OpenSSL-Win64/lib/VC/x64/MD
// Installed on the CI:
#flag windows -IC:/Program Files/OpenSSL/include
#flag windows -LC:/Program Files/OpenSSL/lib/VC/x64/MD
$if $pkgconfig('openssl') {
#pkgconfig --cflags --libs openssl
} $else {
#flag windows -l libssl -l libcrypto
$if !windows {
#flag -lssl -lcrypto
}
#flag linux -ldl -lpthread
// Prefer a single matching macOS OpenSSL prefix to avoid mixing Intel and arm64 installs.
$if arm64 {
#flag darwin -I$when_first_existing('/opt/local/include','/opt/homebrew/opt/openssl/include','/opt/homebrew/opt/openssl@3/include','/opt/procursus/include','/usr/local/opt/openssl/include','/usr/local/opt/openssl@3/include')
#flag darwin -L$when_first_existing('/opt/local/lib','/opt/homebrew/opt/openssl/lib','/opt/homebrew/opt/openssl@3/lib','/opt/procursus/lib','/usr/local/opt/openssl/lib','/usr/local/opt/openssl@3/lib')
} $else {
#flag darwin -I$when_first_existing('/opt/local/include','/usr/local/opt/openssl/include','/usr/local/opt/openssl@3/include','/opt/procursus/include','/opt/homebrew/opt/openssl/include','/opt/homebrew/opt/openssl@3/include')
#flag darwin -L$when_first_existing('/opt/local/lib','/usr/local/opt/openssl/lib','/usr/local/opt/openssl@3/lib','/opt/procursus/lib','/opt/homebrew/opt/openssl/lib','/opt/homebrew/opt/openssl@3/lib')
}
}
#include <openssl/rand.h> # Please install OpenSSL development headers
#include <openssl/ssl.h>
#include <openssl/err.h>
#insert "@VEXEROOT/vlib/net/openssl/openssl_compat.h"
@[typedef]
pub struct C.SSL {
}
@[typedef]
pub struct C.BIO {
}
@[typedef]
pub struct C.SSL_METHOD {
}
@[typedef]
pub struct C.X509 {
}
@[typedef]
pub struct C.SSL_CTX {
}
// The above C structs, have incomplete declarations in the OpenSSL headers.
// For this reason, we have to prevent the automatic str() generation for them,
// by adding manual implementations of their .str() methods, that are defined on
// pointers to them:
fn (s &C.SSL) str() string {
return 'C.SSL(0x${voidptr(s)})'
}
fn (c &C.SSL_CTX) str() string {
return 'C.SSL_CTX(0x${voidptr(c)})'
}
fn C.BIO_new_ssl_connect(ctx &C.SSL_CTX) &C.BIO
fn C.BIO_set_conn_hostname(b &C.BIO, name &char) i32
// there are actually 2 macros for BIO_get_ssl
// fn C.BIO_get_ssl(bp &C.BIO, ssl charptr, c int)
// fn C.BIO_get_ssl(bp &C.BIO, sslp charptr)
fn C.BIO_get_ssl(bp &C.BIO, vargs ...voidptr)
fn C.BIO_do_connect(b &C.BIO) i32
fn C.BIO_do_handshake(b &C.BIO) i32
fn C.BIO_puts(b &C.BIO, buf &char)
fn C.BIO_read(b &C.BIO, buf voidptr, len i32) i32
fn C.BIO_free_all(a &C.BIO)
fn C.SSL_CTX_new(method &C.SSL_METHOD) &C.SSL_CTX
fn C.SSL_CTX_set_options(ctx &C.SSL_CTX, options i32)
fn C.SSL_CTX_set_verify_depth(s &C.SSL_CTX, depth i32)
fn C.SSL_CTX_load_verify_locations(ctx &C.SSL_CTX, const_file &char, ca_path &char) i32
fn C.SSL_CTX_free(ctx &C.SSL_CTX)
fn C.SSL_CTX_use_certificate_file(ctx &C.SSL_CTX, const_file &char, file_type i32) i32
fn C.SSL_CTX_use_PrivateKey_file(ctx &C.SSL_CTX, const_file &char, file_type i32) i32
fn C.SSL_new(&C.SSL_CTX) &C.SSL
fn C.SSL_set_fd(ssl &C.SSL, fd i32) i32
fn C.SSL_connect(&C.SSL) i32
fn C.SSL_do_handshake(&C.SSL) i32
fn C.SSL_set_cipher_list(ctx &C.SSL, str &char) i32
fn C.v_net_openssl_get1_peer_certificate(ssl &C.SSL) &C.X509
fn C.X509_free(const_cert &C.X509)
fn C.ERR_clear_error()
fn C.SSL_get_error(ssl &C.SSL, ret i32) i32
fn C.SSL_get_verify_result(ssl &C.SSL) i32
fn C.SSL_set_tlsext_host_name(s &C.SSL, name &char) i32
// The ALPN calls go through small version-guarded shims in openssl_compat.h,
// so the module still links against OpenSSL versions older than 1.0.2 that
// predate ALPN. v_net_openssl_set_alpn_protos returns 0 on success (like the
// underlying SSL_set_alpn_protos) and non-zero when ALPN is unavailable.
// `data` is `const unsigned char **`; declared as voidptr so V emits a clean
// `(void*)` cast and avoids -cstrict nested-pointer const warnings.
fn C.v_net_openssl_set_alpn_protos(ssl &C.SSL, protos &u8, protos_len u32) i32
fn C.v_net_openssl_get0_alpn_selected(ssl &C.SSL, data voidptr, len &u32)
fn C.SSL_shutdown(&C.SSL) i32
fn C.SSL_free(&C.SSL)
fn C.SSL_write(ssl &C.SSL, buf voidptr, buflen i32) i32
fn C.SSL_read(ssl &C.SSL, buf voidptr, buflen i32) i32
fn C.SSLv23_client_method() &C.SSL_METHOD
fn C.TLS_method() voidptr
fn C.TLSv1_2_method() voidptr
fn C.v_net_openssl_init_ssl() i32
fn init() {
C.v_net_openssl_init_ssl()
}
// ssl_error returns non error ssl code or error if unrecoverable and we should panic
fn ssl_error(ret int, ssl voidptr) !SSLError {
res := C.SSL_get_error(ssl, ret)
$if trace_ssl ? {
eprintln('${@METHOD} ret: ${ret} | ssl: ${ssl:x} | res: ${res}')
}
match unsafe { SSLError(res) } {
.ssl_error_syscall {
return error_with_code('net.openssl unrecoverable syscall (${res})', res)
}
.ssl_error_ssl {
return error_with_code('net.openssl unrecoverable ssl protocol error (${res})', res)
}
else {
return unsafe { SSLError(res) }
}
}
}
enum SSLError {
ssl_error_none = C.SSL_ERROR_NONE
ssl_error_ssl = C.SSL_ERROR_SSL
ssl_error_want_read = C.SSL_ERROR_WANT_READ
ssl_error_want_write = C.SSL_ERROR_WANT_WRITE
ssl_error_want_x509_lookup = C.SSL_ERROR_WANT_X509_LOOKUP
ssl_error_syscall = C.SSL_ERROR_SYSCALL
ssl_error_zero_return = C.SSL_ERROR_ZERO_RETURN
ssl_error_want_connect = C.SSL_ERROR_WANT_CONNECT
ssl_error_want_accept = C.SSL_ERROR_WANT_ACCEPT
ssl_error_want_async = C.SSL_ERROR_WANT_ASYNC
ssl_error_want_async_job = C.SSL_ERROR_WANT_ASYNC_JOB
}
+60
View File
@@ -0,0 +1,60 @@
// Match the init API to the OpenSSL headers that are actually available.
#if defined(LIBRESSL_VERSION_NUMBER) || !defined(OPENSSL_VERSION_NUMBER) \
|| OPENSSL_VERSION_NUMBER < 0x10100000L
static int v_net_openssl_init_ssl(void) {
SSL_load_error_strings();
return SSL_library_init();
}
#else
static int v_net_openssl_init_ssl(void) {
return OPENSSL_init_ssl(OPENSSL_INIT_LOAD_SSL_STRINGS, 0);
}
#endif
// SSL_get1_peer_certificate is only available in OpenSSL 3.x.
#if defined(LIBRESSL_VERSION_NUMBER) || !defined(OPENSSL_VERSION_NUMBER) \
|| OPENSSL_VERSION_NUMBER < 0x30000000L
static X509 *v_net_openssl_get1_peer_certificate(SSL *ssl) {
return SSL_get_peer_certificate(ssl);
}
#else
static X509 *v_net_openssl_get1_peer_certificate(SSL *ssl) {
return SSL_get1_peer_certificate(ssl);
}
#endif
// ALPN (SSL_set_alpn_protos / SSL_get0_alpn_selected) is only available in
// OpenSSL 1.0.2 and later. On older OpenSSL-compatible headers, fall back to
// no-op shims so the module still links; ALPN is simply unavailable there.
// LibreSSL reports a high OPENSSL_VERSION_NUMBER and provides ALPN, so it uses
// the native path below.
#if !defined(OPENSSL_VERSION_NUMBER) || OPENSSL_VERSION_NUMBER < 0x10002000L
static int v_net_openssl_set_alpn_protos(SSL *ssl, const unsigned char *protos, unsigned int protos_len) {
(void)ssl;
(void)protos;
(void)protos_len;
return -1; // ALPN unsupported on this OpenSSL version
}
static void v_net_openssl_get0_alpn_selected(SSL *ssl, const unsigned char **data, unsigned int *len) {
(void)ssl;
*data = NULL;
*len = 0;
}
#else
static int v_net_openssl_set_alpn_protos(SSL *ssl, const unsigned char *protos, unsigned int protos_len) {
return SSL_set_alpn_protos(ssl, protos, protos_len);
}
static void v_net_openssl_get0_alpn_selected(SSL *ssl, const unsigned char **data, unsigned int *len) {
SSL_get0_alpn_selected(ssl, data, len);
}
#endif
// LibreSSL and older OpenSSL-compatible headers may not expose the async
// SSL_ERROR constants, but V's SSLError enum needs stable values for them.
#ifndef SSL_ERROR_WANT_ASYNC
#define SSL_ERROR_WANT_ASYNC 9
#endif
#ifndef SSL_ERROR_WANT_ASYNC_JOB
#define SSL_ERROR_WANT_ASYNC_JOB 10
#endif
@@ -0,0 +1,13 @@
// vtest build: present_openssl?
import net.openssl as _
struct Abc {
x &C.SSL_CTX
}
fn test_printing_struct_with_reference_field_of_type_ssl_ctx() {
a := unsafe { Abc{&C.SSL_CTX(123)} }
dump(a)
sa := a.str()
assert sa.contains('&C.SSL_CTX(0x7b)')
}
@@ -0,0 +1,24 @@
// vtest build: present_openssl?
import net
import net.openssl as _
fn test_importing_openssl_does_not_break_tcp_listener_accept() {
mut listener := net.listen_tcp(.ip, '127.0.0.1:0', net.ListenOptions{}) or { panic(err) }
defer {
listener.close() or {}
}
address := listener.addr() or { panic(err) }
mut client := net.dial_tcp(address.str()) or { panic(err) }
defer {
client.close() or {}
}
mut server := listener.accept() or { panic(err) }
defer {
server.close() or {}
}
peer := server.peer_addr() or { panic(err) }
assert peer.str().len > 0
}
@@ -0,0 +1,5 @@
module openssl
// SSL_get_peer1_certificate not defined in LibreSSL (OpenSSL fork) on OpenBSD,
// use SSL_get_peer_certificate instead.
fn C.SSL_get_peer_certificate(ssl &SSL) &C.X509
+557
View File
@@ -0,0 +1,557 @@
module openssl
import io
import net
import time
import os
// SSLConn is the current connection
pub struct SSLConn {
pub:
config SSLConnectConfig
pub mut:
sslctx &C.SSL_CTX = unsafe { nil }
ssl &C.SSL = unsafe { nil }
handle int
duration time.Duration
owns_socket bool
// last_write_sent reports the most recent write_ptr's progress for retry
// decisions: 0 = provably nothing was sent (safe to replay), or -1 = the
// count is indeterminate because a failed/retryable write may have already
// flushed complete records to the peer (TLS cannot prove zero). On full
// success it equals the bytes written.
last_write_sent int
}
@[params]
pub struct SSLConnectConfig {
pub:
verify string // the path to a rootca.pem file, containing trusted CA certificate(s)
cert string // the path to a cert.pem file, containing client certificate(s) for the request
cert_key string // the path to a key.pem file, containing private keys for the client certificate(s)
validate bool // set this to true, if you want to stop requests, when their certificates are found to be invalid
in_memory_verification bool // if true, verify, cert, and cert_key are read from memory, not from a file
alpn_protocols []string // the list of ALPN protocols to advertise, e.g. ['h2', 'http/1.1']; empty means no ALPN extension is sent
}
// new_ssl_conn instance an new SSLCon struct
pub fn new_ssl_conn(config SSLConnectConfig) !&SSLConn {
$if trace_ssl ? {
eprintln(@METHOD)
}
mut conn := &SSLConn{
config: config
sslctx: unsafe { nil }
ssl: unsafe { nil }
handle: 0
}
conn.init() or { return err }
return conn
}
// Select operation
enum Select {
read
write
except
}
fn ssl_timeout_deadline(timeout time.Duration) time.Time {
if timeout <= 0 || timeout == net.infinite_timeout {
return time.unix(0)
}
return time.now().add(timeout)
}
fn ssl_remaining_timeout(deadline time.Time) time.Duration {
if deadline.unix() == 0 {
return net.infinite_timeout
}
return deadline - time.now()
}
// close closes the ssl connection and does cleanup
pub fn (mut s SSLConn) close() ! {
s.shutdown()!
}
// negotiated_alpn returns the ALPN protocol selected during the TLS
// handshake (e.g. 'h2' or 'http/1.1'), or an empty string if no protocol
// was negotiated.
pub fn (s &SSLConn) negotiated_alpn() string {
mut data := &u8(unsafe { nil })
mut length := u32(0)
C.v_net_openssl_get0_alpn_selected(voidptr(s.ssl), voidptr(&data), &length)
if length == 0 || data == unsafe { nil } {
return ''
}
// data points into OpenSSL-owned memory and is not NUL-terminated; copy it.
return unsafe { data.vbytes(int(length)).bytestr() }
}
// shutdown closes the ssl connection and does cleanup
pub fn (mut s SSLConn) shutdown() ! {
$if trace_ssl ? {
eprintln(@METHOD)
}
if s.ssl != 0 {
deadline := ssl_timeout_deadline(s.duration)
for {
mut res := C.SSL_shutdown(voidptr(s.ssl))
if res == 1 {
break
}
err_res := ssl_error(res, s.ssl) or {
break // We break to free rest of resources
}
if err_res == .ssl_error_want_read {
s.wait_for_read(ssl_remaining_timeout(deadline))!
continue
} else if err_res == .ssl_error_want_write {
s.wait_for_write(ssl_remaining_timeout(deadline))!
continue
}
if s.ssl != 0 {
unsafe { C.SSL_free(voidptr(s.ssl)) }
}
if s.sslctx != 0 {
C.SSL_CTX_free(s.sslctx)
}
return error('net.openssl Could not connect using SSL. (${err_res}),err')
}
C.SSL_free(voidptr(s.ssl))
}
if s.sslctx != 0 {
C.SSL_CTX_free(s.sslctx)
}
if s.owns_socket {
net.shutdown(s.handle)
net.close(s.handle)!
}
}
fn (mut s SSLConn) init() ! {
$if trace_ssl ? {
eprintln(@METHOD)
}
s.sslctx = unsafe { C.SSL_CTX_new(C.SSLv23_client_method()) }
if s.sslctx == 0 {
return error('net.openssl Could not get ssl context')
}
if s.config.validate {
C.SSL_CTX_set_verify_depth(s.sslctx, 4)
C.SSL_CTX_set_options(s.sslctx, C.SSL_OP_NO_COMPRESSION)
}
s.ssl = unsafe { &C.SSL(C.SSL_new(s.sslctx)) }
if s.ssl == 0 {
return error('net.openssl Could not create OpenSSL instance')
}
mut res := 0
// Advertise ALPN protocols (e.g. ['h2', 'http/1.1']) when requested.
// OpenSSL expects the length-prefixed wire format: each protocol is a
// single length byte followed by that many bytes of protocol name.
if s.config.alpn_protocols.len > 0 {
mut wire := []u8{cap: 64}
for proto in s.config.alpn_protocols {
if proto.len == 0 || proto.len > 255 {
return error('net.openssl SSLConn.init, invalid ALPN protocol "${proto}"')
}
wire << u8(proto.len)
wire << proto.bytes()
}
// Returns 0 on success (opposite of most OpenSSL calls); non-zero also
// means ALPN is unavailable on OpenSSL versions older than 1.0.2.
if C.v_net_openssl_set_alpn_protos(voidptr(s.ssl), wire.data, u32(wire.len)) != 0 {
return error('net.openssl SSLConn.init, failed to set ALPN protocols (requires OpenSSL >= 1.0.2)')
}
}
if s.config.validate {
mut verify := s.config.verify
mut cert := s.config.cert
mut cert_key := s.config.cert_key
if s.config.in_memory_verification {
now := time.now().unix().str()
verify = os.temp_dir() + '/v_verify' + now
cert = os.temp_dir() + '/v_cert' + now
cert_key = os.temp_dir() + '/v_cert_key' + now
if s.config.verify != '' {
os.write_file(verify, s.config.verify)!
}
if s.config.cert != '' {
os.write_file(cert, s.config.cert)!
}
if s.config.cert_key != '' {
os.write_file(cert_key, s.config.cert_key)!
}
}
if s.config.verify != '' {
res = C.SSL_CTX_load_verify_locations(voidptr(s.sslctx), &char(verify.str), 0)
if s.config.validate && res != 1 {
return error('net.openssl SSLConn.init, SSL_CTX_load_verify_locations failed')
}
}
if s.config.cert != '' {
res = C.SSL_CTX_use_certificate_file(voidptr(s.sslctx), &char(cert.str),
C.SSL_FILETYPE_PEM)
if s.config.validate && res != 1 {
return error('net.openssl SSLConn.init, SSL_CTX_use_certificate_file failed, res: ${res}')
}
}
if s.config.cert_key != '' {
res = C.SSL_CTX_use_PrivateKey_file(voidptr(s.sslctx), &char(cert_key.str),
C.SSL_FILETYPE_PEM)
if s.config.validate && res != 1 {
return error('net.openssl SSLConn.init, SSL_CTX_use_PrivateKey_file failed, res: ${res}')
}
}
preferred_ciphers := 'HIGH:!aNULL:!kRSA:!PSK:!SRP:!MD5:!RC4'
res = C.SSL_set_cipher_list(voidptr(s.ssl), &char(preferred_ciphers.str))
if s.config.validate && res != 1 {
println('net.openssl: set cipher failed')
}
}
}
// connect to server using OpenSSL
pub fn (mut s SSLConn) connect(mut tcp_conn net.TcpConn, hostname string) ! {
$if trace_ssl ? {
eprintln('${@METHOD} hostname: ${hostname}')
}
s.handle = tcp_conn.sock.handle
s.duration = tcp_conn.read_timeout()
mut res := C.SSL_set_tlsext_host_name(voidptr(s.ssl), voidptr(hostname.str))
if res != 1 {
return error('net.openssl SSLConn.connect, could not set host name')
}
if C.SSL_set_fd(voidptr(s.ssl), tcp_conn.sock.handle) != 1 {
return error('net.openssl SSLConn.connect, could not assign ssl to socket.')
}
s.complete_connect()!
}
// dial opens an ssl connection on hostname:port
pub fn (mut s SSLConn) dial(hostname string, port int) ! {
$if trace_ssl ? {
eprintln('${@METHOD} hostname: ${hostname} | port: ${port}')
}
mut tcp_conn := net.dial_tcp('${hostname}:${port}') or { return err }
mut connected := false
defer {
if !connected {
tcp_conn.close() or {}
if s.ssl != 0 {
unsafe { C.SSL_free(voidptr(s.ssl)) }
s.ssl = unsafe { nil }
}
if s.sslctx != 0 {
C.SSL_CTX_free(s.sslctx)
s.sslctx = unsafe { nil }
}
s.handle = 0
s.owns_socket = false
}
}
s.owns_socket = true
s.connect(mut tcp_conn, hostname) or { return err }
connected = true
}
fn (mut s SSLConn) complete_connect() ! {
$if trace_ssl ? {
eprintln(@METHOD)
}
deadline := ssl_timeout_deadline(s.duration)
for {
mut res := C.SSL_connect(voidptr(s.ssl))
if res == 1 {
break
}
err_res := ssl_error(res, s.ssl)!
if err_res == .ssl_error_want_read {
s.wait_for_read(ssl_remaining_timeout(deadline))!
continue
}
if err_res == .ssl_error_want_write {
s.wait_for_write(ssl_remaining_timeout(deadline))!
continue
}
return error('net.openssl SSLConn.complete_connect, could not connect using SSL. (${err_res}),err')
}
if s.config.validate {
mut pcert := &C.X509(unsafe { nil })
for {
mut res := C.SSL_do_handshake(voidptr(s.ssl))
if res == 1 {
break
}
err_res := ssl_error(res, s.ssl)!
if err_res == .ssl_error_want_read {
s.wait_for_read(ssl_remaining_timeout(deadline))!
continue
} else if err_res == .ssl_error_want_write {
s.wait_for_write(ssl_remaining_timeout(deadline))!
continue
}
return error('net.openssl SSLConn.complete_connect, could not validate SSL certificate. (${err_res}),err')
}
pcert = C.v_net_openssl_get1_peer_certificate(s.ssl)
defer {
if pcert != 0 {
C.X509_free(pcert)
}
}
res := C.SSL_get_verify_result(voidptr(s.ssl))
if res != C.X509_V_OK {
return error('net.openssl SSLConn.complete_connect, failed SSL handshake (OpenSSL SSL_get_verify_result = ${res})')
}
}
}
// read_timeout returns the current SSL read timeout.
pub fn (s &SSLConn) read_timeout() time.Duration {
return s.duration
}
// set_read_timeout sets the read timeout used for subsequent SSL reads.
// A value of 0, or net.infinite_timeout means "wait forever".
pub fn (mut s SSLConn) set_read_timeout(timeout time.Duration) {
s.duration = timeout
}
// addr retrieves the local ip address and port number for this connection
pub fn (s &SSLConn) addr() !net.Addr {
return net.addr_from_socket_handle(s.handle)
}
// peer_addr retrieves the ip address and port number used by the peer
pub fn (s &SSLConn) peer_addr() !net.Addr {
return net.peer_addr_from_socket_handle(s.handle)
}
pub fn (mut s SSLConn) socket_read_into_ptr(buf_ptr &u8, len int) !int {
mut res := 0
$if trace_ssl ? {
defer(fn) {
if len > 0 {
eprintln('${@METHOD} res: ${res}: buf_ptr: ${voidptr(buf_ptr):x}, len: ${len}, hex: ${unsafe { buf_ptr.vbytes(len).hex() }} data: `${unsafe { buf_ptr.vstring_with_len(len) }}`')
}
}
}
deadline := ssl_timeout_deadline(s.duration)
// s.wait_for_read(deadline - time.now())!
for {
res = C.SSL_read(voidptr(s.ssl), buf_ptr, len)
if res > 0 {
return res
} else if res == 0 {
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: io.Eof')
}
return io.Eof{}
} else {
err_res := ssl_error(res, s.ssl)!
match err_res {
.ssl_error_want_read {
s.wait_for_read(ssl_remaining_timeout(deadline)) or {
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: ${err} .ssl_error_want_read')
}
return err
}
}
.ssl_error_want_write {
s.wait_for_write(ssl_remaining_timeout(deadline)) or {
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: ${err} .ssl_error_want_write')
}
return err
}
}
.ssl_error_zero_return {
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: 0 .ssl_error_zero_return')
}
return 0
}
else {
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: could not read, err_res: ${err_res}')
}
return error('net.openssl SSLConn.socket_read_into_ptr, could not read. (${err_res})')
}
}
}
}
// Dead code, to satisfy the compiler
return error('net.openssl SSLConn.socket_read_into_ptr, unknown error')
}
pub fn (mut s SSLConn) read(mut buffer []u8) !int {
$if trace_ssl ? {
eprintln('${@METHOD} buffer.len: ${buffer.len}')
}
return s.socket_read_into_ptr(&u8(buffer.data), buffer.len)
}
// write_ptr writes `len` bytes from `bytes` to the ssl connection
pub fn (mut s SSLConn) write_ptr(bytes &u8, len int) !int {
mut total_sent := 0
$if trace_ssl ? {
defer(fn) {
eprintln('${@METHOD} total_sent: ${total_sent}, bytes: ${voidptr(bytes):x}, len: ${len}, hex: ${unsafe { bytes.vbytes(len).hex() }}, data:-=-=-=-\n${unsafe { bytes.vstring_with_len(len) }}\n-=-=-=-')
}
}
s.last_write_sent = 0
deadline := ssl_timeout_deadline(s.duration)
unsafe {
mut ptr_base := bytes
for total_sent < len {
ptr := ptr_base + total_sent
remaining := len - total_sent
mut sent := C.SSL_write(voidptr(s.ssl), ptr, remaining)
if sent <= 0 {
// SSL_write did not fully complete: OpenSSL may already have
// flushed one or more complete records (which the peer can
// decrypt and act on) before returning a retryable error, so the
// sent count is no longer provable. Mark it indeterminate; a
// later full success below resets it to the exact length.
s.last_write_sent = -1
err_res := ssl_error(sent, s.ssl)!
if err_res == .ssl_error_want_read {
s.wait_for_read(ssl_remaining_timeout(deadline))!
continue
} else if err_res == .ssl_error_want_write {
s.wait_for_write(ssl_remaining_timeout(deadline))!
continue
} else if err_res == .ssl_error_zero_return {
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: ssl write on closed connection .ssl_error_zero_return')
}
return error('net.openssl SSLConn.write_ptr, ssl write on closed connection') // TODO: error_with_code close
}
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: could not write SSL, err_res: ${err_res}')
}
return error_with_code('net.openssl SSLConn.write_ptr, could not write. (${err_res}),err',
int(err_res))
}
total_sent += sent
s.last_write_sent = total_sent
}
}
return total_sent
}
// write writes data from `bytes` to the ssl connection
pub fn (mut s SSLConn) write(bytes []u8) !int {
return s.write_ptr(&u8(bytes.data), bytes.len)!
}
// write_string writes a string to the ssl connection
pub fn (mut s SSLConn) write_string(str string) !int {
$if trace_ssl ? {
eprintln('${@METHOD} str: ${str}')
}
return s.write_ptr(str.str, str.len)
}
// Select waits for an io operation (specified by parameter `test`) to be available
fn select(handle int, test Select, timeout time.Duration) !bool {
$if trace_ssl ? {
eprintln('${@METHOD} handle: ${handle}, timeout: ${timeout}')
}
set := C.fd_set{}
C.FD_ZERO(&set)
C.FD_SET(handle, &set)
is_infinite := timeout <= 0 || timeout == net.infinite_timeout
deadline := ssl_timeout_deadline(timeout)
mut remaining_time := if is_infinite { i64(0) } else { timeout.milliseconds() }
for is_infinite || remaining_time > 0 {
seconds := remaining_time / 1000
microseconds := (remaining_time % 1000) * 1000
tt := C.timeval{
tv_sec: u64(seconds)
tv_usec: u64(microseconds)
}
timeval_timeout := if is_infinite {
&C.timeval(unsafe { nil })
} else {
&tt
}
mut res := -1
match test {
.read {
res = net.socket_error(C.select(handle + 1, &set, C.NULL, C.NULL, timeval_timeout))!
}
.write {
res = net.socket_error(C.select(handle + 1, C.NULL, &set, C.NULL, timeval_timeout))!
}
.except {
res = net.socket_error(C.select(handle + 1, C.NULL, C.NULL, &set, timeval_timeout))!
}
}
if res < 0 {
if C.errno == C.EINTR {
// errno is 4, Spurious wakeup from signal, keep waiting
if !is_infinite {
remaining_time = ssl_remaining_timeout(deadline).milliseconds()
}
continue
}
cerr := C.errno
return error_with_code('net.openssl select, failed: ${res}', cerr)
} else if res == 0 {
return net.err_timed_out
}
res = C.FD_ISSET(handle, &set)
$if trace_ssl ? {
eprintln('${@METHOD} ---> res: ${res}')
}
return res != 0
}
return net.err_timed_out
}
// wait_for wraps the common wait code
fn wait_for(handle int, what Select, timeout time.Duration) ! {
ready := select(handle, what, timeout)!
if ready {
return
}
return net.err_timed_out
}
// wait_for_write waits for a write io operation to be available
fn (mut s SSLConn) wait_for_write(timeout time.Duration) ! {
return wait_for(s.handle, .write, timeout)
}
// wait_for_read waits for a read io operation to be available
fn (mut s SSLConn) wait_for_read(timeout time.Duration) ! {
return wait_for(s.handle, .read, timeout)
}