add TCP User Timeout param to s3 driver

registry/storage/driver/s3-aws/s3.go

@@ -19,13 +19,15 @@ import (
 	"fmt"
 	"io"
 	"math"
+	"net"
 	"net/http"
 	"path/filepath"
 	"reflect"
 	"sort"
 	"strconv"
 	"strings"
 	"sync"
+	"syscall"
 	"time"
 
 	"github.com/aws/aws-sdk-go/aws"
@@ -120,6 +122,7 @@ type DriverParameters struct {
 	UseDualStack                bool
 	Accelerate                  bool
 	LogLevel                    aws.LogLevelType
+	TCPUserTimeout              time.Duration
 }
 
 func init() {
@@ -440,6 +443,21 @@ func FromParameters(ctx context.Context, parameters map[string]interface{}) (*Dr
 		return nil, fmt.Errorf("the accelerate parameter should be a boolean")
 	}
 
+	tcpUserTimeout := parameters["tcpusertimeout"]
+	tcpUserTimeoutDuration := time.Duration(0)
+	switch tcpUserTimeout := tcpUserTimeout.(type) {
+	case string:
+		d, err := time.ParseDuration(tcpUserTimeout)
+		if err != nil {
+			return nil, fmt.Errorf("the tcpusertimeout parameter should be a duration")
+		}
+		tcpUserTimeoutDuration = d
+	case nil:
+		// do nothing
+	default:
+		return nil, fmt.Errorf("the tcpusertimeout parameter should be a duration")
+	}
+
 	params := DriverParameters{
 		fmt.Sprint(accessKey),
 		fmt.Sprint(secretKey),
@@ -465,6 +483,7 @@ func FromParameters(ctx context.Context, parameters map[string]interface{}) (*Dr
 		useDualStackBool,
 		accelerateBool,
 		getS3LogLevelFromParam(parameters["loglevel"]),
+		tcpUserTimeoutDuration,
 	}
 
 	return New(ctx, params)
@@ -561,6 +580,25 @@ func New(ctx context.Context, params DriverParameters) (*Driver, error) {
 	httpTransportModified := false
 	httpTransport := http.DefaultTransport.(*http.Transport).Clone()
 
+	if params.TCPUserTimeout > 0 {
+		dialer := &net.Dialer{
+			Timeout:   30 * time.Second,
+			KeepAlive: 30 * time.Second,
+		}
+		dialer.Control = func(network, address string, c syscall.RawConn) error {
+			var err error
+			controlErr := c.Control(func(fd uintptr) {
+				err = setTCPUserTimeout(fd, params.TCPUserTimeout)
+			})
+			if controlErr != nil {
+				return controlErr
+			}
+			return err
+		}
+		httpTransport.DialContext = dialer.DialContext
+		httpTransportModified = true
+	}
+
 	if params.SkipVerify {
 		httpTransport.TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
 		httpTransportModified = true

registry/storage/driver/s3-aws/tcp_default.go

@@ -0,0 +1,12 @@
+//go:build !linux
+// +build !linux
+
+package s3
+
+import (
+	"time"
+)
+
+func setTCPUserTimeout(fd uintptr, timeout time.Duration) error {
+	return nil
+}

registry/storage/driver/s3-aws/tcp_linux.go

@@ -0,0 +1,12 @@
+package s3
+
+import (
+	"syscall"
+	"time"
+
+	"golang.org/x/sys/unix"
+)
+
+func setTCPUserTimeout(fd uintptr, timeout time.Duration) error {
+	return syscall.SetsockoptInt(int(fd), syscall.IPPROTO_TCP, unix.TCP_USER_TIMEOUT, int(timeout.Milliseconds()))
+}