From 60b4095c75b9ef417e5ad525832f3e37dab9acc0 Mon Sep 17 00:00:00 2001
From: macros <andrianjardan@gmail.com>
Date: Wed, 8 Aug 2018 19:12:03 +0200
Subject: [PATCH] Set keepalive on TCP socket so idleTimeout works

---
 server/server.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/server/server.go b/server/server.go
index adc384d7c..c15fb6927 100644
--- a/server/server.go
+++ b/server/server.go
@@ -90,6 +90,22 @@ type serverEntryPoint struct {
 	onDemandListener func(string) (*tls.Certificate, error)
 }
 
+// tcpKeepAliveListener sets TCP keep-alive timeouts on accepted
+// connections.
+type tcpKeepAliveListener struct {
+	*net.TCPListener
+}
+
+func (ln tcpKeepAliveListener) Accept() (net.Conn, error) {
+	tc, err := ln.AcceptTCP()
+	if err != nil {
+		return nil, err
+	}
+	tc.SetKeepAlive(true)
+	tc.SetKeepAlivePeriod(3 * time.Minute)
+	return tc, nil
+}
+
 // NewServer returns an initialized Server.
 func NewServer(globalConfiguration configuration.GlobalConfiguration, provider provider.Provider) *Server {
 	server := new(Server)
@@ -803,6 +819,8 @@ func (s *Server) prepareServer(entryPointName string, entryPoint *configuration.
 		return nil, nil, err
 	}
 
+	listener = tcpKeepAliveListener{listener.(*net.TCPListener)}
+
 	if entryPoint.ProxyProtocol != nil {
 		IPs, err := whitelist.NewIP(entryPoint.ProxyProtocol.TrustedIPs, entryPoint.ProxyProtocol.Insecure, false)
 		if err != nil {