524 lines
12 KiB
Go
524 lines
12 KiB
Go
package tcp
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/traefik/traefik/v2/pkg/config/dynamic"
|
|
"github.com/traefik/traefik/v2/pkg/config/runtime"
|
|
"github.com/traefik/traefik/v2/pkg/server/service/tcp"
|
|
traefiktls "github.com/traefik/traefik/v2/pkg/tls"
|
|
)
|
|
|
|
func TestRuntimeConfiguration(t *testing.T) {
|
|
testCases := []struct {
|
|
desc string
|
|
httpServiceConfig map[string]*runtime.ServiceInfo
|
|
httpRouterConfig map[string]*runtime.RouterInfo
|
|
tcpServiceConfig map[string]*runtime.TCPServiceInfo
|
|
tcpRouterConfig map[string]*runtime.TCPRouterInfo
|
|
expectedError int
|
|
}{
|
|
{
|
|
desc: "No error",
|
|
tcpServiceConfig: map[string]*runtime.TCPServiceInfo{
|
|
"foo-service": {
|
|
TCPService: &dynamic.TCPService{
|
|
LoadBalancer: &dynamic.TCPServersLoadBalancer{
|
|
Servers: []dynamic.TCPServer{
|
|
{
|
|
Port: "8085",
|
|
Address: "127.0.0.1:8085",
|
|
},
|
|
{
|
|
Address: "127.0.0.1:8086",
|
|
Port: "8086",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
tcpRouterConfig: map[string]*runtime.TCPRouterInfo{
|
|
"foo": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "HostSNI(`bar.foo`)",
|
|
TLS: &dynamic.RouterTCPTLSConfig{
|
|
Passthrough: false,
|
|
Options: "foo",
|
|
},
|
|
},
|
|
},
|
|
"bar": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "HostSNI(`foo.bar`)",
|
|
TLS: &dynamic.RouterTCPTLSConfig{
|
|
Passthrough: false,
|
|
Options: "bar",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedError: 0,
|
|
},
|
|
{
|
|
desc: "HTTP routers with same domain but different TLS options",
|
|
httpServiceConfig: map[string]*runtime.ServiceInfo{
|
|
"foo-service": {
|
|
Service: &dynamic.Service{
|
|
LoadBalancer: &dynamic.ServersLoadBalancer{
|
|
Servers: []dynamic.Server{
|
|
{
|
|
Port: "8085",
|
|
URL: "127.0.0.1:8085",
|
|
},
|
|
{
|
|
URL: "127.0.0.1:8086",
|
|
Port: "8086",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
httpRouterConfig: map[string]*runtime.RouterInfo{
|
|
"foo": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "Host(`bar.foo`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "foo",
|
|
},
|
|
},
|
|
},
|
|
"bar": {
|
|
Router: &dynamic.Router{
|
|
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "Host(`bar.foo`) && PathPrefix(`/path`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "bar",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedError: 2,
|
|
},
|
|
{
|
|
desc: "One router with wrong rule",
|
|
tcpServiceConfig: map[string]*runtime.TCPServiceInfo{
|
|
"foo-service": {
|
|
TCPService: &dynamic.TCPService{
|
|
LoadBalancer: &dynamic.TCPServersLoadBalancer{
|
|
Servers: []dynamic.TCPServer{
|
|
{
|
|
Address: "127.0.0.1:80",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
tcpRouterConfig: map[string]*runtime.TCPRouterInfo{
|
|
"foo": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "WrongRule(`bar.foo`)",
|
|
},
|
|
},
|
|
|
|
"bar": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "HostSNI(`foo.bar`)",
|
|
},
|
|
},
|
|
},
|
|
expectedError: 1,
|
|
},
|
|
{
|
|
desc: "All router with wrong rule",
|
|
tcpServiceConfig: map[string]*runtime.TCPServiceInfo{
|
|
"foo-service": {
|
|
TCPService: &dynamic.TCPService{
|
|
LoadBalancer: &dynamic.TCPServersLoadBalancer{
|
|
Servers: []dynamic.TCPServer{
|
|
{
|
|
Address: "127.0.0.1:80",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
tcpRouterConfig: map[string]*runtime.TCPRouterInfo{
|
|
"foo": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "WrongRule(`bar.foo`)",
|
|
},
|
|
},
|
|
"bar": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "WrongRule(`foo.bar`)",
|
|
},
|
|
},
|
|
},
|
|
expectedError: 2,
|
|
},
|
|
{
|
|
desc: "Router with unknown service",
|
|
tcpServiceConfig: map[string]*runtime.TCPServiceInfo{
|
|
"foo-service": {
|
|
TCPService: &dynamic.TCPService{
|
|
LoadBalancer: &dynamic.TCPServersLoadBalancer{
|
|
Servers: []dynamic.TCPServer{
|
|
{
|
|
Address: "127.0.0.1:80",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
tcpRouterConfig: map[string]*runtime.TCPRouterInfo{
|
|
"foo": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "wrong-service",
|
|
Rule: "HostSNI(`bar.foo`)",
|
|
},
|
|
},
|
|
"bar": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "HostSNI(`foo.bar`)",
|
|
},
|
|
},
|
|
},
|
|
expectedError: 1,
|
|
},
|
|
{
|
|
desc: "Router with broken service",
|
|
tcpServiceConfig: map[string]*runtime.TCPServiceInfo{
|
|
"foo-service": {
|
|
TCPService: &dynamic.TCPService{
|
|
LoadBalancer: nil,
|
|
},
|
|
},
|
|
},
|
|
tcpRouterConfig: map[string]*runtime.TCPRouterInfo{
|
|
"bar": {
|
|
TCPRouter: &dynamic.TCPRouter{
|
|
EntryPoints: []string{"web"},
|
|
Service: "foo-service",
|
|
Rule: "HostSNI(`foo.bar`)",
|
|
},
|
|
},
|
|
},
|
|
expectedError: 2,
|
|
},
|
|
}
|
|
|
|
for _, test := range testCases {
|
|
test := test
|
|
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
entryPoints := []string{"web"}
|
|
|
|
conf := &runtime.Configuration{
|
|
Services: test.httpServiceConfig,
|
|
Routers: test.httpRouterConfig,
|
|
TCPServices: test.tcpServiceConfig,
|
|
TCPRouters: test.tcpRouterConfig,
|
|
}
|
|
serviceManager := tcp.NewManager(conf)
|
|
tlsManager := traefiktls.NewManager()
|
|
tlsManager.UpdateConfigs(
|
|
context.Background(),
|
|
map[string]traefiktls.Store{},
|
|
map[string]traefiktls.Options{
|
|
"default": {
|
|
MinVersion: "VersionTLS10",
|
|
},
|
|
"foo": {
|
|
MinVersion: "VersionTLS12",
|
|
},
|
|
"bar": {
|
|
MinVersion: "VersionTLS11",
|
|
},
|
|
},
|
|
[]*traefiktls.CertAndStores{})
|
|
|
|
routerManager := NewManager(conf, serviceManager,
|
|
nil, nil, tlsManager)
|
|
|
|
_ = routerManager.BuildHandlers(context.Background(), entryPoints)
|
|
|
|
// even though conf was passed by argument to the manager builders above,
|
|
// it's ok to use it as the result we check, because everything worth checking
|
|
// can be accessed by pointers in it.
|
|
var allErrors int
|
|
for _, v := range conf.TCPServices {
|
|
if v.Err != nil {
|
|
allErrors++
|
|
}
|
|
}
|
|
for _, v := range conf.TCPRouters {
|
|
if len(v.Err) > 0 {
|
|
allErrors++
|
|
}
|
|
}
|
|
for _, v := range conf.Services {
|
|
if v.Err != nil {
|
|
allErrors++
|
|
}
|
|
}
|
|
for _, v := range conf.Routers {
|
|
if len(v.Err) > 0 {
|
|
allErrors++
|
|
}
|
|
}
|
|
assert.Equal(t, test.expectedError, allErrors)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestDomainFronting(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
routers map[string]*runtime.RouterInfo
|
|
expectedStatus int
|
|
}{
|
|
{
|
|
desc: "Request is misdirected when TLS options are different",
|
|
routers: map[string]*runtime.RouterInfo{
|
|
"router-1@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
"router-2@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host2.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{},
|
|
},
|
|
},
|
|
},
|
|
expectedStatus: http.StatusMisdirectedRequest,
|
|
},
|
|
{
|
|
desc: "Request is OK when TLS options are the same",
|
|
routers: map[string]*runtime.RouterInfo{
|
|
"router-1@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
"router-2@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host2.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedStatus: http.StatusOK,
|
|
},
|
|
{
|
|
desc: "Default TLS options is used when options are ambiguous for the same host",
|
|
routers: map[string]*runtime.RouterInfo{
|
|
"router-1@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
"router-2@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`) && PathPrefix(`/foo`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "default",
|
|
},
|
|
},
|
|
},
|
|
"router-3@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host2.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedStatus: http.StatusMisdirectedRequest,
|
|
},
|
|
{
|
|
desc: "Default TLS options should not be used when options are the same for the same host",
|
|
routers: map[string]*runtime.RouterInfo{
|
|
"router-1@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
"router-2@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`) && PathPrefix(`/bar`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
"router-3@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host2.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedStatus: http.StatusOK,
|
|
},
|
|
{
|
|
desc: "Request is misdirected when TLS options have the same name but from different providers",
|
|
routers: map[string]*runtime.RouterInfo{
|
|
"router-1@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
"router-2@crd": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host2.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedStatus: http.StatusMisdirectedRequest,
|
|
},
|
|
{
|
|
desc: "Request is OK when TLS options reference from a different provider is the same",
|
|
routers: map[string]*runtime.RouterInfo{
|
|
"router-1@file": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host1.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1@crd",
|
|
},
|
|
},
|
|
},
|
|
"router-2@crd": {
|
|
Router: &dynamic.Router{
|
|
EntryPoints: []string{"web"},
|
|
Rule: "Host(`host2.local`)",
|
|
TLS: &dynamic.RouterTLSConfig{
|
|
Options: "host1@crd",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
expectedStatus: http.StatusOK,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
entryPoints := []string{"web"}
|
|
tlsOptions := map[string]traefiktls.Options{
|
|
"default": {
|
|
MinVersion: "VersionTLS10",
|
|
},
|
|
"host1@file": {
|
|
MinVersion: "VersionTLS12",
|
|
},
|
|
"host1@crd": {
|
|
MinVersion: "VersionTLS12",
|
|
},
|
|
}
|
|
|
|
conf := &runtime.Configuration{
|
|
Routers: test.routers,
|
|
}
|
|
|
|
serviceManager := tcp.NewManager(conf)
|
|
|
|
tlsManager := traefiktls.NewManager()
|
|
tlsManager.UpdateConfigs(context.Background(), map[string]traefiktls.Store{}, tlsOptions, []*traefiktls.CertAndStores{})
|
|
|
|
httpsHandler := map[string]http.Handler{
|
|
"web": http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {}),
|
|
}
|
|
|
|
routerManager := NewManager(conf, serviceManager, nil, httpsHandler, tlsManager)
|
|
|
|
routers := routerManager.BuildHandlers(context.Background(), entryPoints)
|
|
|
|
router, ok := routers["web"]
|
|
require.True(t, ok)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Host = "host1.local"
|
|
req.TLS = &tls.ConnectionState{
|
|
ServerName: "host2.local",
|
|
}
|
|
|
|
rw := httptest.NewRecorder()
|
|
|
|
router.GetHTTPSHandler().ServeHTTP(rw, req)
|
|
|
|
assert.Equal(t, test.expectedStatus, rw.Code)
|
|
})
|
|
}
|
|
}
|