mirror/tlstunnel
1
1
Fork 0
mirror of https://git.sr.ht/~emersion/tlstunnel synced 2024-11-19 15:53:50 +01:00
Code

Add downstream TLS handshake timeout

Browse Source
This commit is contained in:
Simon Ser 2021-02-18 17:49:52 +01:00
parent 8ce6fc38f2
commit 14bdfb49f3
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
server.go
View File

@ -9,6 +9,7 @@ import (
"net"
"strings"
"sync/atomic"
"time"
"git.sr.ht/~emersion/go-scfg"
"github.com/caddyserver/certmagic"
@ -16,6 +17,8 @@ import (
"github.com/pires/go-proxyproto/tlvparse"
)
const tlsHandshakeTimeout = 10 * time.Second
type acmeCache struct {
config *certmagic.Config
cache *certmagic.Cache
@ -245,7 +248,6 @@ func (ln *Listener) handle(conn net.Conn) error {
defer conn.Close()
srv := ln.atomic.Load().(*listenerHandles).Server
// TODO: setup timeouts
tlsConfig := srv.ACMEConfig.TLSConfig()
getConfigForClient := tlsConfig.GetConfigForClient
tlsConfig.GetConfigForClient = func(hello *tls.ClientHelloInfo) (*tls.Config, error) {
@ -270,9 +272,17 @@ func (ln *Listener) handle(conn net.Conn) error {
return tlsConfig, nil
}
tlsConn := tls.Server(conn, tlsConfig)
if err := tlsConn.SetDeadline(time.Now().Add(tlsHandshakeTimeout)); err != nil {
return fmt.Errorf("failed to set TLS handshake timeout: %v", err)
}
if err := tlsConn.Handshake(); err != nil {
return fmt.Errorf("TLS handshake failed: %v", err)
}
if err := tlsConn.SetDeadline(time.Time{}); err != nil {
return fmt.Errorf("failed to reset TLS handshake timeout: %v", err)
}
// TODO: allow setting custom downstream timeouts
tlsState := tlsConn.ConnectionState()
fe, err := ln.matchFrontend(tlsState.ServerName)
@ -314,6 +324,8 @@ type Frontend struct {
func (fe *Frontend) handle(downstream net.Conn, tlsState *tls.ConnectionState) error {
defer downstream.Close()
// TODO: setup upstream timeouts
be := &fe.Backend
upstream, err := net.Dial(be.Network, be.Address)
if err != nil {