Stéphane Busso
87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
package factory
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
portainer "github.com/portainer/portainer/api"
|
|
"github.com/portainer/portainer/api/crypto"
|
|
"github.com/portainer/portainer/api/http/proxy/factory/dockercompose"
|
|
)
|
|
|
|
// ProxyServer provide an extedned proxy with a local server to forward requests
|
|
type ProxyServer struct {
|
|
server http.Server
|
|
Port int
|
|
}
|
|
|
|
func (factory *ProxyFactory) NewDockerComposeAgentProxy(endpoint *portainer.Endpoint) (*ProxyServer, error) {
|
|
|
|
endpointURL, err := url.Parse(endpoint.URL)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
endpointURL.Scheme = "http"
|
|
httpTransport := &http.Transport{}
|
|
|
|
if endpoint.TLSConfig.TLS || endpoint.TLSConfig.TLSSkipVerify {
|
|
config, err := crypto.CreateTLSConfigurationFromDisk(endpoint.TLSConfig.TLSCACertPath, endpoint.TLSConfig.TLSCertPath, endpoint.TLSConfig.TLSKeyPath, endpoint.TLSConfig.TLSSkipVerify)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
httpTransport.TLSClientConfig = config
|
|
endpointURL.Scheme = "https"
|
|
}
|
|
|
|
proxy := newSingleHostReverseProxyWithHostHeader(endpointURL)
|
|
|
|
proxy.Transport = dockercompose.NewAgentTransport(factory.signatureService, httpTransport)
|
|
|
|
proxyServer := &ProxyServer{
|
|
http.Server{
|
|
Handler: proxy,
|
|
},
|
|
0,
|
|
}
|
|
|
|
return proxyServer, proxyServer.Start()
|
|
}
|
|
|
|
func (factory *ProxyFactory) GetReverseTunnel(endpoint *portainer.Endpoint) *portainer.TunnelDetails {
|
|
return factory.reverseTunnelService.GetTunnelDetails(endpoint.ID)
|
|
}
|
|
|
|
func (proxy *ProxyServer) Start() error {
|
|
listener, err := net.Listen("tcp", ":0")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
shutdownChan := make(chan error, 1)
|
|
proxy.Port = listener.Addr().(*net.TCPAddr).Port
|
|
go func() {
|
|
|
|
log.Printf("Starting Proxy server on %s...\n", fmt.Sprintf("http://127.0.0.1:%d", proxy.Port))
|
|
|
|
err := proxy.server.Serve(listener)
|
|
log.Printf("Proxy Server exited with '%v' error\n", err)
|
|
|
|
if err != http.ErrServerClosed {
|
|
log.Printf("Put '%v' error returned by Proxy Server to shutdown channel\n", proxy.Port)
|
|
shutdownChan <- err
|
|
}
|
|
}()
|
|
|
|
return nil
|
|
}
|
|
|
|
// Close the server proxy
|
|
func (proxy *ProxyServer) Close() {
|
|
proxy.server.Close()
|
|
}
|