Retry wsarecv: An existing connection was forcibly closed by the remote host.

ste/xferRetryHelper.go

@@ -30,6 +30,8 @@ import (
 
 var RetryStatusCodes RetryCodes
 
+var platformRetryPolicy func(response *http.Response, err error) bool
+
 func getShouldRetry() func(*http.Response, error) bool {
 	if len(RetryStatusCodes) == 0 {
 		return nil
@@ -48,7 +50,13 @@ func getShouldRetry() func(*http.Response, error) bool {
 				}
 			}
 		}
-		return false
+
+		// fall back to our platform retry policy
+		if platformRetryPolicy != nil {
+			return platformRetryPolicy(resp, err)
+		} else {
+			return false // If we have none, don't retry.
+		}
 	}
 }
 

ste/xferRetryHelper_test.go

@@ -1,8 +1,11 @@
 package ste
 
 import (
+	"errors"
 	"github.com/stretchr/testify/assert"
 	"net/http"
+	"runtime"
+	"syscall"
 	"testing"
 )
 
@@ -38,6 +41,14 @@ func TestGetShouldRetry(t *testing.T) {
 	header["x-ms-error-code"] = []string{"ContainerBeingDeleted"}
 	response = &http.Response{Header: header, StatusCode: 409}
 	a.False(shouldRetry(response, nil))
+
+	if runtime.GOOS == "windows" {
+		rawErr := syscall.Errno(10054) // magic number, in reference to windows.WSAECONNRESET, preventing OS specific shenanigans
+		strErr := errors.New("wsarecv: An existing connection was forcibly closed by the remote host.")
+
+		a.True(shouldRetry(nil, rawErr))
+		a.True(shouldRetry(nil, strErr))
+	}
 }
 
 func TestGetErrorCode(t *testing.T) {

ste/xferRetryHelper_windows.go

@@ -0,0 +1,22 @@
+//go:build windows
+
+package ste
+
+import (
+	"errors"
+	"golang.org/x/sys/windows"
+	"net/http"
+	"strings"
+)
+
+func init() {
+	platformRetryPolicy = func(response *http.Response, err error) bool {
+		// It's entirely possible something in between closed our connection.
+		if errors.Is(err, windows.WSAECONNRESET) || // Catch it in the idiomatic way
+			strings.Contains(strings.ToLower(err.Error()), "an existing connection was forcibly closed by the remote host.") { // But just in case something funny happened along the line, let's listen for the string we expect.
+			return true
+		}
+
+		return false
+	}
+}