mirror of https://github.com/portainer/portainer
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package proxy
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/http/httputil"
|
|
"net/url"
|
|
|
|
"github.com/portainer/portainer"
|
|
"github.com/portainer/portainer/crypto"
|
|
)
|
|
|
|
// AzureAPIBaseURL is the URL where Azure API requests will be proxied.
|
|
const AzureAPIBaseURL = "https://management.azure.com"
|
|
|
|
// proxyFactory is a factory to create reverse proxies to Docker endpoints
|
|
type proxyFactory struct {
|
|
ResourceControlService portainer.ResourceControlService
|
|
TeamMembershipService portainer.TeamMembershipService
|
|
SettingsService portainer.SettingsService
|
|
RegistryService portainer.RegistryService
|
|
DockerHubService portainer.DockerHubService
|
|
SignatureService portainer.DigitalSignatureService
|
|
}
|
|
|
|
func (factory *proxyFactory) newHTTPProxy(u *url.URL) http.Handler {
|
|
u.Scheme = "http"
|
|
return newSingleHostReverseProxyWithHostHeader(u)
|
|
}
|
|
|
|
func newAzureProxy(credentials *portainer.AzureCredentials) (http.Handler, error) {
|
|
url, err := url.Parse(AzureAPIBaseURL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
proxy := newSingleHostReverseProxyWithHostHeader(url)
|
|
proxy.Transport = NewAzureTransport(credentials)
|
|
|
|
return proxy, nil
|
|
}
|
|
|
|
func (factory *proxyFactory) newDockerHTTPSProxy(u *url.URL, tlsConfig *portainer.TLSConfiguration, enableSignature bool) (http.Handler, error) {
|
|
u.Scheme = "https"
|
|
|
|
proxy := factory.createDockerReverseProxy(u, enableSignature)
|
|
config, err := crypto.CreateTLSConfigurationFromDisk(tlsConfig.TLSCACertPath, tlsConfig.TLSCertPath, tlsConfig.TLSKeyPath, tlsConfig.TLSSkipVerify)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
proxy.Transport.(*proxyTransport).dockerTransport.TLSClientConfig = config
|
|
return proxy, nil
|
|
}
|
|
|
|
func (factory *proxyFactory) newDockerHTTPProxy(u *url.URL, enableSignature bool) http.Handler {
|
|
u.Scheme = "http"
|
|
return factory.createDockerReverseProxy(u, enableSignature)
|
|
}
|
|
|
|
func (factory *proxyFactory) newDockerSocketProxy(path string) http.Handler {
|
|
proxy := &socketProxy{}
|
|
transport := &proxyTransport{
|
|
enableSignature: false,
|
|
ResourceControlService: factory.ResourceControlService,
|
|
TeamMembershipService: factory.TeamMembershipService,
|
|
SettingsService: factory.SettingsService,
|
|
RegistryService: factory.RegistryService,
|
|
DockerHubService: factory.DockerHubService,
|
|
dockerTransport: newSocketTransport(path),
|
|
}
|
|
proxy.Transport = transport
|
|
return proxy
|
|
}
|
|
|
|
func (factory *proxyFactory) createDockerReverseProxy(u *url.URL, enableSignature bool) *httputil.ReverseProxy {
|
|
proxy := newSingleHostReverseProxyWithHostHeader(u)
|
|
transport := &proxyTransport{
|
|
enableSignature: enableSignature,
|
|
ResourceControlService: factory.ResourceControlService,
|
|
TeamMembershipService: factory.TeamMembershipService,
|
|
SettingsService: factory.SettingsService,
|
|
RegistryService: factory.RegistryService,
|
|
DockerHubService: factory.DockerHubService,
|
|
dockerTransport: &http.Transport{},
|
|
}
|
|
|
|
if enableSignature {
|
|
transport.SignatureService = factory.SignatureService
|
|
}
|
|
|
|
proxy.Transport = transport
|
|
return proxy
|
|
}
|
|
|
|
func newSocketTransport(socketPath string) *http.Transport {
|
|
return &http.Transport{
|
|
Dial: func(proto, addr string) (conn net.Conn, err error) {
|
|
return net.Dial("unix", socketPath)
|
|
},
|
|
}
|
|
}
|