func startTestServer() (string, error) {
l, err := net.Listen("tcp", "localhost:0")
if err != nil {
return "", err
}
err = waitforserver.WaitForServer("tcp", l.Addr().String(), 1*time.Second)
if err != nil {
return "", err
}
go func() {
for {
c, err := l.Accept()
if err != nil {
log.Fatalf("Error listening: %s", err)
}
if _, err = c.Write(msg); err != nil {
log.Fatalf("Unable to write message: %s", err)
}
if err := c.Close(); err != nil {
log.Fatalf("Unable to close connection: %v", err)
}
}
}()
return l.Addr().String(), nil
}
// httpClient creates an http.Client. If rootCA is specified, the client will
// validate the server's certificate on TLS connections against that RootCA. If
// proxyAddr is specified, the client will proxy through the given http proxy.
func httpClient(rootCA string, proxyAddr string, persistent bool) (*http.Client, error) {
log.Debugf("Creating new HTTPClient with proxy: %v", proxyAddr)
tr := &http.Transport{
Dial: (&net.Dialer{
Timeout: 60 * time.Second,
KeepAlive: 30 * time.Second,
}).Dial,
TLSHandshakeTimeout: 10 * time.Second,
// This method is typically used for creating a one-off HTTP client
// that we don't want to keep around for future calls, making
// persistent connections a potential source of file descriptor
// leaks. Note the name of this variable is misleading -- it would
// be clearer to call it DisablePersistentConnections -- i.e. it has
// nothing to do with TCP keep alives along the lines of the KeepAlive
// variable in net.Dialer.
DisableKeepAlives: !persistent,
}
if rootCA != "" {
caCert, err := keyman.LoadCertificateFromPEMBytes([]byte(rootCA))
if err != nil {
return nil, fmt.Errorf("Unable to decode rootCA: %s", err)
}
tr.TLSClientConfig = &tls.Config{
RootCAs: caCert.PoolContainingCert(),
}
}
if proxyAddr != "" {
host, _, err := net.SplitHostPort(proxyAddr)
if err != nil {
return nil, fmt.Errorf("Unable to split host and port for %v: %v", proxyAddr, err)
}
noHostSpecified := host == ""
if noHostSpecified {
// For addresses of the form ":8080", prepend the loopback IP
host = "127.0.0.1"
proxyAddr = host + proxyAddr
}
if isLoopback(host) {
log.Debugf("Waiting for loopback proxy server to came online...")
// Waiting for proxy server to came online.
err := waitforserver.WaitForServer("tcp", proxyAddr, 60*time.Second)
if err != nil {
// Instead of finishing here we just log the error and continue, the client
// we are going to create will surely fail when used and return errors,
// those errors should be handled by the code that depends on such client.
log.Errorf("Proxy never came online at %v: %q", proxyAddr, err)
}
log.Debugf("Connected to proxy on localhost")
} else {
log.Errorf("Attempting to proxy through server other than loopback %v", host)
}
tr.Proxy = func(req *http.Request) (*url.URL, error) {
return url.Parse("http://" + proxyAddr)
}
} else {
log.Errorf("Using direct http client with no proxyAddr")
}
return &http.Client{Transport: tr}, nil
}
