Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add ssl configuration to fleet server http. #98

Merged
merged 4 commits into from
Feb 22, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
40 changes: 20 additions & 20 deletions cmd/fleet/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,15 @@ package fleet

import (
"context"
"github.com/elastic/fleet-server/v7/internal/pkg/config"
"crypto/tls"
slog "log"
"net"
"net/http"

"github.com/elastic/fleet-server/v7/internal/pkg/config"
"github.com/elastic/fleet-server/v7/internal/pkg/rate"

"github.com/elastic/beats/v7/libbeat/common/transport/tlscommon"
"github.com/julienschmidt/httprouter"
"github.com/rs/zerolog/log"
)
Expand All @@ -37,7 +39,7 @@ func runServer(ctx context.Context, router *httprouter.Router, cfg *config.Serve
Str("bind", addr).
Dur("rdTimeout", rdto).
Dur("wrTimeout", wrto).
Msg("Server listening")
Msg("server listening")

server := http.Server{
Addr: addr,
Expand All @@ -57,53 +59,51 @@ func runServer(ctx context.Context, router *httprouter.Router, cfg *config.Serve
go func() {
select {
case <-ctx.Done():
log.Debug().Msg("Force server close on ctx.Done()")
log.Debug().Msg("force server close on ctx.Done()")
server.Close()
case <-forceCh:
log.Debug().Msg("Go routine forced closed on exit")
log.Debug().Msg("go routine forced closed on exit")
}
}()

ln, err := makeListener(ctx, addr, cfg)
ln, err := net.Listen("tcp", addr)
if err != nil {
return err
}

defer ln.Close()

// TODO: Use tls.Config to properly mux down tls connection
keyFile := cfg.TLS.Key
certFile := cfg.TLS.Cert

if keyFile != "" || certFile != "" {
return server.ServeTLS(ln, certFile, keyFile)
if cfg.TLS != nil && cfg.TLS.IsEnabled() {
tlsCfg, err := tlscommon.LoadTLSConfig(cfg.TLS)
if err != nil {
return err
}
server.TLSConfig = tlsCfg.ToConfig()
ln = tls.NewListener(ln, server.TLSConfig)
} else {
log.Warn().Msg("exposed over insecure HTTP; enablement of TLS is strongly recommended")
}

ln = wrapRateLimitter(ctx, ln, cfg)
if err := server.Serve(ln); err != nil && err != context.Canceled {
return err
}

return nil
}

func makeListener(ctx context.Context, addr string, cfg *config.Server) (net.Listener, error) {
// Create listener
ln, err := net.Listen("tcp", addr)
if err != nil {
return nil, err
}

func wrapRateLimitter(ctx context.Context, ln net.Listener, cfg *config.Server) net.Listener {
rateLimitBurst := cfg.RateLimitBurst
rateLimitInterval := cfg.RateLimitInterval

if rateLimitInterval != 0 {
log.Info().Dur("interval", rateLimitInterval).Int("burst", rateLimitBurst).Msg("Server rate limiter installed")
ln = rate.NewRateListener(ctx, ln, rateLimitBurst, rateLimitInterval)
} else {
log.Info().Msg("Server connection rate limiter disabled")
log.Info().Msg("server connection rate limiter disabled")
}

return ln, err
return ln
}

type stubLogger struct {
Expand Down
2 changes: 1 addition & 1 deletion cmd/fleet/server_integration_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ func (s *tserver) baseUrl() string {
input := s.cfg.Inputs[0]
tls := input.Server.TLS
schema := "http"
if tls.Key != "" || tls.Cert != "" {
if tls != nil && tls.IsEnabled() {
schema = "https"
}
return fmt.Sprintf("%s://%s:%d", schema, input.Server.Host, input.Server.Port)
Expand Down
20 changes: 11 additions & 9 deletions internal/pkg/config/input.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@ import (
"fmt"
"strings"
"time"

"github.com/elastic/beats/v7/libbeat/common/transport/tlscommon"
)

const kDefaultHost = "localhost"
Expand Down Expand Up @@ -48,15 +50,15 @@ type ServerTLS struct {

// Server is the configuration for the server
type Server struct {
Host string `config:"host"`
Port uint16 `config:"port"`
TLS ServerTLS `config:"tls"`
Timeouts ServerTimeouts `config:"timeouts"`
MaxHeaderByteSize int `config:"max_header_byte_size"`
RateLimitBurst int `config:"rate_limit_burst"`
RateLimitInterval time.Duration `config:"rate_limit_interval"`
MaxEnrollPending int64 `config:"max_enroll_pending"`
Profile ServerProfile `config:"profile"`
Host string `config:"host"`
Port uint16 `config:"port"`
TLS *tlscommon.Config `config:"ssl"`
Timeouts ServerTimeouts `config:"timeouts"`
MaxHeaderByteSize int `config:"max_header_byte_size"`
RateLimitBurst int `config:"rate_limit_burst"`
RateLimitInterval time.Duration `config:"rate_limit_interval"`
MaxEnrollPending int64 `config:"max_enroll_pending"`
Profile ServerProfile `config:"profile"`
}

// InitDefaults initializes the defaults for the configuration.
Expand Down