From 34517c340d0bb25717b2fe5545373d01f6853edc Mon Sep 17 00:00:00 2001
From: WoodyAtHome <mzmail@gmx.net>
Date: Mon, 27 Jun 2022 10:01:55 +0200
Subject: [PATCH] net: support blocking sockets in TcpSocket.connect (#14849)

---
 vlib/net/tcp.v | 62 +++++++++++++++++++++++++++-----------------------
 1 file changed, 33 insertions(+), 29 deletions(-)

diff --git a/vlib/net/tcp.v b/vlib/net/tcp.v
index 8cdf6c5a4d..aa2e46ab07 100644
--- a/vlib/net/tcp.v
+++ b/vlib/net/tcp.v
@@ -379,35 +379,39 @@ const (
 )
 
 fn (mut s TcpSocket) connect(a Addr) ? {
-	res := C.connect(s.handle, voidptr(&a), a.len())
-	if res == 0 {
-		return
-	}
-
-	// The  socket  is  nonblocking and the connection cannot be completed
-	// immediately.  (UNIX domain sockets failed with EAGAIN instead.)
-	// It is possible to select(2) or poll(2) for completion by selecting
-	// the socket for  writing.   After  select(2) indicates  writability,
-	// use getsockopt(2) to read the SO_ERROR option at level SOL_SOCKET to
-	// determine whether connect() completed successfully (SO_ERROR is zero) or
-	// unsuccessfully (SO_ERROR is one of the usual error codes  listed  here,
-	// ex‐ plaining the reason for the failure).
-	write_result := s.@select(.write, net.connect_timeout)?
-	if write_result {
-		err := 0
-		len := sizeof(err)
-		socket_error(C.getsockopt(s.handle, C.SOL_SOCKET, C.SO_ERROR, &err, &len))?
-
-		if err != 0 {
-			return wrap_error(err)
+	$if !net_blocking_sockets ? {
+		res := C.connect(s.handle, voidptr(&a), a.len())
+		if res == 0 {
+			return
 		}
-		// Succeeded
-		return
+
+		// The  socket  is  nonblocking and the connection cannot be completed
+		// immediately.  (UNIX domain sockets failed with EAGAIN instead.)
+		// It is possible to select(2) or poll(2) for completion by selecting
+		// the socket for  writing.   After  select(2) indicates  writability,
+		// use getsockopt(2) to read the SO_ERROR option at level SOL_SOCKET to
+		// determine whether connect() completed successfully (SO_ERROR is zero) or
+		// unsuccessfully (SO_ERROR is one of the usual error codes  listed  here,
+		// ex‐ plaining the reason for the failure).
+		write_result := s.@select(.write, net.connect_timeout)?
+		if write_result {
+			err := 0
+			len := sizeof(err)
+			socket_error(C.getsockopt(s.handle, C.SOL_SOCKET, C.SO_ERROR, &err, &len))?
+
+			if err != 0 {
+				return wrap_error(err)
+			}
+			// Succeeded
+			return
+		}
+
+		// Get the error
+		socket_error(C.connect(s.handle, voidptr(&a), a.len()))?
+
+		// otherwise we timed out
+		return err_connect_timed_out
+	} $else {
+		socket_error(C.connect(s.handle, voidptr(&a), a.len()))?
 	}
-
-	// Get the error
-	socket_error(C.connect(s.handle, voidptr(&a), a.len()))?
-
-	// otherwise we timed out
-	return err_connect_timed_out
 }