Implement AF_UNIX sockets on Windows

See moby/moby#36442

Signed-off-by: Marat Radchenko <[email protected]>

sockets/sockets.go

@@ -2,8 +2,13 @@
 package sockets
 
 import (
+	"context"
 	"errors"
+	"fmt"
+	"net"
 	"net/http"
+	"syscall"
+	"time"
 )
 
 // ErrProtocolNotAvailable is returned when a given transport protocol is not provided by the operating system.
@@ -24,3 +29,23 @@ func ConfigureTransport(tr *http.Transport, proto, addr string) error {
 	}
 	return nil
 }
+
+const (
+	defaultTimeout        = 10 * time.Second
+	maxUnixSocketPathSize = len(syscall.RawSockaddrUnix{}.Path)
+)
+
+func configureUnixTransport(tr *http.Transport, proto, addr string) error {
+	if len(addr) > maxUnixSocketPathSize {
+		return fmt.Errorf("Unix socket path %q is too long", addr)
+	}
+	// No need for compression in local communications.
+	tr.DisableCompression = true
+	dialer := &net.Dialer{
+		Timeout: defaultTimeout,
+	}
+	tr.DialContext = func(ctx context.Context, _, _ string) (net.Conn, error) {
+		return dialer.DialContext(ctx, proto, addr)
+	}
+	return nil
+}

sockets/sockets_unix.go

@@ -1,36 +1,15 @@
+//go:build !windows
 // +build !windows
 
 package sockets
 
 import (
-	"context"
-	"fmt"
 	"net"
 	"net/http"
 	"syscall"
 	"time"
 )
 
-const (
-	defaultTimeout        = 10 * time.Second
-	maxUnixSocketPathSize = len(syscall.RawSockaddrUnix{}.Path)
-)
-
-func configureUnixTransport(tr *http.Transport, proto, addr string) error {
-	if len(addr) > maxUnixSocketPathSize {
-		return fmt.Errorf("Unix socket path %q is too long", addr)
-	}
-	// No need for compression in local communications.
-	tr.DisableCompression = true
-	dialer := &net.Dialer{
-		Timeout: defaultTimeout,
-	}
-	tr.DialContext = func(ctx context.Context, _, _ string) (net.Conn, error) {
-		return dialer.DialContext(ctx, proto, addr)
-	}
-	return nil
-}
-
 func configureNpipeTransport(tr *http.Transport, proto, addr string) error {
 	return ErrProtocolNotAvailable
 }

sockets/sockets_windows.go

@@ -9,10 +9,6 @@ import (
 	"github.com/Microsoft/go-winio"
 )
 
-func configureUnixTransport(tr *http.Transport, proto, addr string) error {
-	return ErrProtocolNotAvailable
-}
-
 func configureNpipeTransport(tr *http.Transport, proto, addr string) error {
 	// No need for compression in local communications.
 	tr.DisableCompression = true

sockets/unix_socket.go

@@ -1,5 +1,3 @@
-// +build !windows
-
 /*
 Package sockets is a simple unix domain socket wrapper.
 
@@ -103,9 +101,9 @@ func NewUnixSocketWithOpts(path string, opts ...SockOption) (net.Listener, error
 	// We don't use "defer" here, to reset the umask to its original value as soon
 	// as possible. Ideally we'd be able to detect if WithChmod() was passed as
 	// an option, and skip changing umask if default permissions are used.
-	origUmask := syscall.Umask(0777)
+	origUmask := umask(0777)
 	l, err := net.Listen("unix", path)
-	syscall.Umask(origUmask)
+	umask(origUmask)
 	if err != nil {
 		return nil, err
 	}

sockets/unix_socket_test.go

@@ -1,12 +1,9 @@
-// +build !windows
-
 package sockets
 
 import (
 	"fmt"
 	"net"
 	"os"
-	"syscall"
 	"testing"
 )
 
@@ -52,26 +49,8 @@ func TestNewUnixSocket(t *testing.T) {
 }
 
 func TestUnixSocketWithOpts(t *testing.T) {
-	uid, gid := os.Getuid(), os.Getgid()
-	perms := os.FileMode(0660)
-	path := "/tmp/test.sock"
 	echoStr := "hello"
-	l, err := NewUnixSocketWithOpts(path, WithChown(uid, gid), WithChmod(perms))
-	if err != nil {
-		t.Fatal(err)
-	}
+	l, path := createTestUnixSocket(t)
 	defer l.Close()
-	p, err := os.Stat(path)
-	if err != nil {
-		t.Fatal(err)
-	}
-	if p.Mode().Perm() != perms {
-		t.Fatalf("unexpected file permissions: expected: %#o, got: %#o", perms, p.Mode().Perm())
-	}
-	if stat, ok := p.Sys().(*syscall.Stat_t); ok {
-		if stat.Uid != uint32(uid) || stat.Gid != uint32(gid) {
-			t.Fatalf("unexpected file ownership: expected: %d:%d, got: %d:%d", uid, gid, stat.Uid, stat.Gid)
-		}
-	}
 	runTest(t, path, l, echoStr)
 }

sockets/unix_socket_test_unix.go

@@ -0,0 +1,34 @@
+//go:build !windows
+// +build !windows
+
+package sockets
+
+import (
+	"os"
+	"syscall"
+	"testing"
+)
+
+func createTestUnixSocket(t *testing.T) (listener net.Listener, path string) {
+	uid, gid := os.Getuid(), os.Getgid()
+	perms := os.FileMode(0660)
+	path = "/tmp/test.sock"
+	echoStr := "hello"
+	l, err := NewUnixSocketWithOpts(path, WithChown(uid, gid), WithChmod(perms))
+	if err != nil {
+		t.Fatal(err)
+	}
+	p, err := os.Stat(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if p.Mode().Perm() != perms {
+		t.Fatalf("unexpected file permissions: expected: %#o, got: %#o", perms, p.Mode().Perm())
+	}
+	if stat, ok := p.Sys().(*syscall.Stat_t); ok {
+		if stat.Uid != uint32(uid) || stat.Gid != uint32(gid) {
+			t.Fatalf("unexpected file ownership: expected: %d:%d, got: %d:%d", uid, gid, stat.Uid, stat.Gid)
+		}
+	}
+	return l, path
+}

sockets/unix_socket_test_windows.go

@@ -0,0 +1,25 @@
+package sockets
+
+import (
+	"io/ioutil"
+	"net"
+	"testing"
+)
+
+func testSocketPath() string {
+	file, err := ioutil.TempFile("", "test*.sock")
+	if err != nil {
+		panic(err)
+	}
+	defer file.Close()
+	return file.Name()
+}
+
+func createTestUnixSocket(t *testing.T) (listener net.Listener, path string) {
+	path = testSocketPath()
+	l, err := NewUnixSocketWithOpts(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	return l, path
+}

sockets/unix_socket_unix.go

@@ -0,0 +1,10 @@
+//go:build !windows
+// +build !windows
+
+package sockets
+
+import "syscall"
+
+func umask(newmask int) (oldmask int) {
+	return syscall.Umask(0777)
+}

sockets/unix_socket_windows.go

@@ -0,0 +1,5 @@
+package sockets
+
+func umask(newmask int) (oldmask int) {
+	return newmask
+}