net: bypass concurrency-limited WinAPI TransmitFile function

Author: shibijm

src/internal/syscall/windows/types_windows.go

@@ -256,3 +256,7 @@ type FILE_COMPLETION_INFORMATION struct {
 	Port syscall.Handle
 	Key  uintptr
 }
+
+// https://learn.microsoft.com/en-us/windows/win32/api/winnt/ns-winnt-osversioninfoexa
+// https://learn.microsoft.com/en-us/windows-hardware/drivers/ddi/wdm/ns-wdm-_osversioninfoexw
+const VER_NT_WORKSTATION = 0x0000001

src/internal/syscall/windows/version_windows.go

@@ -11,28 +11,52 @@ import (
 	"unsafe"
 )
 
-// https://learn.microsoft.com/en-us/windows-hardware/drivers/ddi/wdm/ns-wdm-_osversioninfow
-type _OSVERSIONINFOW struct {
+// https://learn.microsoft.com/en-us/windows-hardware/drivers/ddi/wdm/ns-wdm-_osversioninfoexw
+type _OSVERSIONINFOEXW struct {
 	osVersionInfoSize uint32
 	majorVersion      uint32
 	minorVersion      uint32
 	buildNumber       uint32
 	platformId        uint32
 	csdVersion        [128]uint16
+	servicePackMajor  uint16
+	servicePackMinor  uint16
+	suiteMask         uint16
+	productType       byte
+	reserved          byte
 }
 
 // According to documentation, RtlGetVersion function always succeeds.
-//sys	rtlGetVersion(info *_OSVERSIONINFOW) = ntdll.RtlGetVersion
+//sys	rtlGetVersion(info *_OSVERSIONINFOEXW) = ntdll.RtlGetVersion
+
+// Retrieves version information of the current Windows OS
+// from the RtlGetVersion API.
+func getVersionInfo() *_OSVERSIONINFOEXW {
+	info := _OSVERSIONINFOEXW{}
+	info.osVersionInfoSize = uint32(unsafe.Sizeof(info))
+	rtlGetVersion(&info)
+	return &info
+}
 
 // Version retrieves the major, minor, and build version numbers
 // of the current Windows OS from the RtlGetVersion API.
 func Version() (major, minor, build uint32) {
-	info := _OSVERSIONINFOW{}
-	info.osVersionInfoSize = uint32(unsafe.Sizeof(info))
-	rtlGetVersion(&info)
+	info := getVersionInfo()
 	return info.majorVersion, info.minorVersion, info.buildNumber
 }
 
+// SupportUnlimitedTransmitFile indicates whether the current
+// Windows version's TransmitFile function imposes any
+// concurrent operation limits.
+// Workstation and client versions of Windows limit the number
+// of concurrent TransmitFile operations allowed on the system
+// to a maximum of two.
+// See remarks under https://learn.microsoft.com/en-us/windows/win32/api/mswsock/nf-mswsock-transmitfile
+var SupportUnlimitedTransmitFile = sync.OnceValue(func() bool {
+	info := getVersionInfo()
+	return info.productType != VER_NT_WORKSTATION
+})
+
 var (
 	supportTCPKeepAliveIdle     bool
 	supportTCPKeepAliveInterval bool

src/internal/syscall/windows/zsyscall_windows.go

@@ -35,7 +35,7 @@ const (
 // expect a call to SendFile.
 func expectSendfile(t *testing.T, wantConn Conn, f func()) {
 	t.Helper()
-	if !supportsSendfile {
+	if !supportsSendfile || !osSupportsUnlimitedSendFile() {
 		f()
 		return
 	}

src/net/sendfile_test.go

@@ -48,8 +48,10 @@ func (c *TCPConn) readFrom(r io.Reader) (int64, error) {
 	if n, err, handled := spliceFrom(c.fd, r); handled {
 		return n, err
 	}
-	if n, err, handled := sendFile(c.fd, r); handled {
-		return n, err
+	if osSupportsUnlimitedSendFile() {
+		if n, err, handled := sendFile(c.fd, r); handled {
+			return n, err
+		}
 	}
 	return genericReadFrom(c, r)
 }

src/net/tcpsock_posix.go

@@ -36,3 +36,8 @@ func (c *TCPConn) SetKeepAliveConfig(config KeepAliveConfig) error {
 
 	return nil
 }
+
+// Always true except for workstation and client versions of Windows
+func osSupportsUnlimitedSendFile() bool {
+	return true
+}

src/net/tcpsock_solaris.go

@@ -29,3 +29,8 @@ func (c *TCPConn) SetKeepAliveConfig(config KeepAliveConfig) error {
 
 	return nil
 }
+
+// Always true except for workstation and client versions of Windows
+func osSupportsUnlimitedSendFile() bool {
+	return true
+}

src/net/tcpsock_unix.go

@@ -34,3 +34,11 @@ func (c *TCPConn) SetKeepAliveConfig(config KeepAliveConfig) error {
 
 	return nil
 }
+
+// Workstation and client versions of Windows limit the number
+// of concurrent TransmitFile operations allowed on the system
+// to a maximum of two.
+// See remarks under https://learn.microsoft.com/en-us/windows/win32/api/mswsock/nf-mswsock-transmitfile
+func osSupportsUnlimitedSendFile() bool {
+	return windows.SupportUnlimitedTransmitFile()
+}