46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
|
// Copyright 2016 go-dockerclient authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
// +build windows
|
||
|
|
|
||
|
|
package docker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
"net/http"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/Microsoft/go-winio"
|
||
|
|
)
|
||
|
|
|
||
|
|
const namedPipeConnectTimeout = 2 * time.Second
|
||
|
|
|
||
|
|
type pipeDialer struct {
|
||
|
|
dialFunc func(network, addr string) (net.Conn, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p pipeDialer) Dial(network, address string) (net.Conn, error) {
|
||
|
|
return p.dialFunc(network, address)
|
||
|
|
}
|
||
|
|
|
||
|
|
// initializeNativeClient initializes the native Named Pipe client for Windows
|
||
|
|
func (c *Client) initializeNativeClient(trFunc func() *http.Transport) {
|
||
|
|
if c.endpointURL.Scheme != namedPipeProtocol {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
namedPipePath := c.endpointURL.Path
|
||
|
|
dialFunc := func(network, addr string) (net.Conn, error) {
|
||
|
|
timeout := namedPipeConnectTimeout
|
||
|
|
return winio.DialPipe(namedPipePath, &timeout)
|
||
|
|
}
|
||
|
|
tr := trFunc()
|
||
|
|
tr.Dial = dialFunc
|
||
|
|
tr.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
|
||
|
|
return dialFunc(network, addr)
|
||
|
|
}
|
||
|
|
c.Dialer = &pipeDialer{dialFunc}
|
||
|
|
c.HTTPClient.Transport = tr
|
||
|
|
}
|