51 lines
1.4 KiB
Go
51 lines
1.4 KiB
Go
package handshake
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/lucas-clemente/quic-go/crypto"
|
|
"github.com/lucas-clemente/quic-go/protocol"
|
|
"github.com/lucas-clemente/quic-go/utils"
|
|
)
|
|
|
|
var (
|
|
kexLifetime = protocol.EphermalKeyLifetime
|
|
kexCurrent crypto.KeyExchange
|
|
kexCurrentTime time.Time
|
|
kexMutex sync.RWMutex
|
|
)
|
|
|
|
// getEphermalKEX returns the currently active KEX, which changes every protocol.EphermalKeyLifetime
|
|
// See the explanation from the QUIC crypto doc:
|
|
//
|
|
// A single connection is the usual scope for forward security, but the security
|
|
// difference between an ephemeral key used for a single connection, and one
|
|
// used for all connections for 60 seconds is negligible. Thus we can amortise
|
|
// the Diffie-Hellman key generation at the server over all the connections in a
|
|
// small time span.
|
|
func getEphermalKEX() (res crypto.KeyExchange) {
|
|
kexMutex.RLock()
|
|
res = kexCurrent
|
|
t := kexCurrentTime
|
|
kexMutex.RUnlock()
|
|
if res != nil && time.Now().Sub(t) < kexLifetime {
|
|
return res
|
|
}
|
|
|
|
kexMutex.Lock()
|
|
defer kexMutex.Unlock()
|
|
// Check if still unfulfilled
|
|
if kexCurrent == nil || time.Now().Sub(kexCurrentTime) > kexLifetime {
|
|
kex, err := crypto.NewCurve25519KEX()
|
|
if err != nil {
|
|
utils.Errorf("could not set KEX: %s", err.Error())
|
|
return kexCurrent
|
|
}
|
|
kexCurrent = kex
|
|
kexCurrentTime = time.Now()
|
|
return kexCurrent
|
|
}
|
|
return kexCurrent
|
|
}
|