net/openssl/websocket: implement io.Writer (#8980)

vlib/net/ftp/ftp.v

@@ -74,11 +74,11 @@ pub fn new() FTP {
 	return f
 }
 
-fn (mut zftp FTP) write(data string) ? {
+fn (mut zftp FTP) write(data string) ?int {
 	$if debug {
 		println('FTP.v >>> $data')
 	}
-	zftp.conn.write('$data\r\n'.bytes()) ?
+	return zftp.conn.write('$data\r\n'.bytes())
 }
 
 fn (mut zftp FTP) read() ?(int, string) {

vlib/net/tcp.v

@@ -33,7 +33,7 @@ pub fn (mut c TcpConn) close() ? {
 }
 
 // write_ptr blocks and attempts to write all data
-pub fn (mut c TcpConn) write_ptr(b byteptr, len int) ? {
+pub fn (mut c TcpConn) write_ptr(b byteptr, len int) ?int {
 	$if trace_tcp ? {
 		eprintln(
 			'>>> TcpConn.write_ptr | c.sock.handle: $c.sock.handle | b: ${ptr_str(b)} len: $len |\n' +
@@ -57,17 +57,17 @@ pub fn (mut c TcpConn) write_ptr(b byteptr, len int) ? {
 			}
 			total_sent += sent
 		}
+		return total_sent
 	}
-	return none
 }
 
 // write blocks and attempts to write all data
-pub fn (mut c TcpConn) write(bytes []byte) ? {
+pub fn (mut c TcpConn) write(bytes []byte) ?int {
 	return c.write_ptr(bytes.data, bytes.len)
 }
 
 // write_str blocks and attempts to write all data
-pub fn (mut c TcpConn) write_str(s string) ? {
+pub fn (mut c TcpConn) write_str(s string) ?int {
 	return c.write_ptr(s.str, s.len)
 }
 

vlib/net/udp.v

@@ -45,23 +45,23 @@ fn resolve_wrapper(raddr string) ?Addr {
 	return x
 }
 
-pub fn (mut c UdpConn) write_ptr(b byteptr, len int) ? {
+pub fn (mut c UdpConn) write_ptr(b byteptr, len int) ?int {
 	remote := c.sock.remote() or { return err_no_udp_remote }
 	return c.write_to_ptr(remote, b, len)
 }
 
-pub fn (mut c UdpConn) write(buf []byte) ? {
+pub fn (mut c UdpConn) write(buf []byte) ?int {
 	return c.write_ptr(buf.data, buf.len)
 }
 
-pub fn (mut c UdpConn) write_str(s string) ? {
+pub fn (mut c UdpConn) write_str(s string) ?int {
 	return c.write_ptr(s.str, s.len)
 }
 
-pub fn (mut c UdpConn) write_to_ptr(addr Addr, b byteptr, len int) ? {
+pub fn (mut c UdpConn) write_to_ptr(addr Addr, b byteptr, len int) ?int {
 	res := C.sendto(c.sock.handle, b, len, 0, &addr.addr, addr.len)
 	if res >= 0 {
-		return none
+		return res
 	}
 	code := error_code()
 	if code == int(error_ewouldblock) {
@@ -74,12 +74,12 @@ pub fn (mut c UdpConn) write_to_ptr(addr Addr, b byteptr, len int) ? {
 }
 
 // write_to blocks and writes the buf to the remote addr specified
-pub fn (mut c UdpConn) write_to(addr Addr, buf []byte) ? {
+pub fn (mut c UdpConn) write_to(addr Addr, buf []byte) ?int {
 	return c.write_to_ptr(addr, buf.data, buf.len)
 }
 
 // write_to_string blocks and writes the buf to the remote addr specified
-pub fn (mut c UdpConn) write_to_string(addr Addr, s string) ? {
+pub fn (mut c UdpConn) write_to_string(addr Addr, s string) ?int {
 	return c.write_to_ptr(addr, s.str, s.len)
 }
 

vlib/net/unix/stream_nix.v

@@ -170,7 +170,7 @@ pub fn (mut c StreamConn) close() ? {
 }
 
 // write_ptr blocks and attempts to write all data
-pub fn (mut c StreamConn) write_ptr(b byteptr, len int) ? {
+pub fn (mut c StreamConn) write_ptr(b byteptr, len int) ?int {
 	$if trace_unix ? {
 		eprintln(
 			'>>> StreamConn.write_ptr | c.sock.handle: $c.sock.handle | b: ${ptr_str(b)} len: $len |\n' +
@@ -194,17 +194,17 @@ pub fn (mut c StreamConn) write_ptr(b byteptr, len int) ? {
 			}
 			total_sent += sent
 		}
+		return total_sent
 	}
-	return none
 }
 
 // write blocks and attempts to write all data
-pub fn (mut c StreamConn) write(bytes []byte) ? {
+pub fn (mut c StreamConn) write(bytes []byte) ?int {
 	return c.write_ptr(bytes.data, bytes.len)
 }
 
 // write_str blocks and attempts to write all data
-pub fn (mut c StreamConn) write_str(s string) ? {
+pub fn (mut c StreamConn) write_str(s string) ?int {
 	return c.write_ptr(s.str, s.len)
 }
 

vlib/x/openssl/openssl.v

@@ -186,7 +186,7 @@ pub fn (mut s SSLConn) read_into(mut buffer []byte) ?int {
 }
 
 // write number of bytes to SSL connection
-pub fn (mut s SSLConn) write(bytes []byte) ? {
+pub fn (mut s SSLConn) write(bytes []byte) ?int {
 	unsafe {
 		mut ptr_base := byteptr(bytes.data)
 		mut total_sent := 0
@@ -218,6 +218,7 @@ pub fn (mut s SSLConn) write(bytes []byte) ? {
 			}
 			total_sent += sent
 		}
+		return total_sent
 	}
 }
 

vlib/x/websocket/io.v

@@ -52,24 +52,25 @@ fn (mut ws Client) socket_read_ptr(buf_ptr byteptr, len int) ?int {
 }
 
 // socket_write writes the provided byte array to the socket
-fn (mut ws Client) socket_write(bytes []byte) ? {
+fn (mut ws Client) socket_write(bytes []byte) ?int {
 	lock  {
 		if ws.state == .closed || ws.conn.sock.handle <= 1 {
 			ws.debug_log('socket_write: Socket allready closed')
 			return error('socket_write: trying to write on a closed socket')
 		}
 		if ws.is_ssl {
-			ws.ssl_conn.write(bytes) ?
+			return ws.ssl_conn.write(bytes)
 		} else {
 			for {
-				ws.conn.write(bytes) or {
+				n := ws.conn.write(bytes) or {
 					if errcode == net.err_timed_out_code {
 						continue
 					}
 					return error(err)
 				}
-				return
+				return n
 			}
+			panic('reached unreachable code')
 		}
 	}
 }

vlib/x/websocket/websocket_client.v

@@ -226,7 +226,7 @@ pub fn (mut ws Client) pong() ? {
 }
 
 // write_ptr writes len bytes provided a byteptr with a websocket messagetype
-pub fn (mut ws Client) write_ptr(bytes byteptr, payload_len int, code OPCode) ? {
+pub fn (mut ws Client) write_ptr(bytes byteptr, payload_len int, code OPCode) ?int {
 	// ws.debug_log('write_ptr code: $code')
 	if ws.state != .open || ws.conn.sock.handle < 1 {
 		// todo: send error here later
@@ -293,22 +293,23 @@ pub fn (mut ws Client) write_ptr(bytes byteptr, payload_len int, code OPCode) ?
 			frame_buf[header_len + i] ^= masking_key[i % 4] & 0xff
 		}
 	}
-	ws.socket_write(frame_buf) ?
+	written_len := ws.socket_write(frame_buf) ?
 	unsafe {
 		frame_buf.free()
 		masking_key.free()
 		header.free()
 	}
+	return written_len
 }
 
 // write writes a byte array with a websocket messagetype to socket
-pub fn (mut ws Client) write(bytes []byte, code OPCode) ? {
+pub fn (mut ws Client) write(bytes []byte, code OPCode) ?int {
-	ws.write_ptr(byteptr(bytes.data), bytes.len, code) ?
+	return ws.write_ptr(byteptr(bytes.data), bytes.len, code)
 }
 
 // write_str, writes a string with a websocket texttype to socket
-pub fn (mut ws Client) write_str(str string) ? {
+pub fn (mut ws Client) write_str(str string) ?int {
-	ws.write_ptr(str.str, str.len, .text_frame) ?
+	return ws.write_ptr(str.str, str.len, .text_frame)
 }
 
 // close closes the websocket connection