Merge pull request #16494 from calavera/fix_plugin_url_scheme

Do not hardcode http as plugin URL scheme for secure connections.
This commit is contained in:
Alexandre Beslic 2015-09-23 16:18:00 -07:00
commit 9e0bd3aca0
2 changed files with 26 additions and 4 deletions

View file

@ -40,12 +40,18 @@ func NewClient(addr string, tlsConfig tlsconfig.Options) (*Client, error) {
protoAndAddr := strings.Split(addr, "://")
sockets.ConfigureTCPTransport(tr, protoAndAddr[0], protoAndAddr[1])
return &Client{&http.Client{Transport: tr}, protoAndAddr[1]}, nil
scheme := protoAndAddr[0]
if scheme != "https" {
scheme = "http"
}
return &Client{&http.Client{Transport: tr}, scheme, protoAndAddr[1]}, nil
}
// Client represents a plugin client.
type Client struct {
http *http.Client // http client to use
scheme string // scheme protocol of the plugin
addr string // http address of the plugin
}
@ -66,7 +72,7 @@ func (c *Client) callWithRetry(serviceMethod string, args interface{}, ret inter
return err
}
req.Header.Add("Accept", versionMimetype)
req.URL.Scheme = "http"
req.URL.Scheme = c.scheme
req.URL.Host = c.addr
var retries int

View file

@ -105,3 +105,19 @@ func TestAbortRetry(t *testing.T) {
}
}
}
func TestClientScheme(t *testing.T) {
cases := map[string]string{
"tcp://127.0.0.1:8080": "http",
"unix:///usr/local/plugins/foo": "http",
"http://127.0.0.1:8080": "http",
"https://127.0.0.1:8080": "https",
}
for addr, scheme := range cases {
c, _ := NewClient(addr, tlsconfig.Options{InsecureSkipVerify: true})
if c.scheme != scheme {
t.Fatalf("URL scheme mismatch, expected %s, got %s", scheme, c.scheme)
}
}
}