mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-04 12:47:36 +03:00
221 lines
8.5 KiB
Go
221 lines
8.5 KiB
Go
package handshake
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/bifurcation/mint"
|
|
"github.com/lucas-clemente/quic-go/internal/crypto"
|
|
"github.com/lucas-clemente/quic-go/internal/mocks/crypto"
|
|
"github.com/lucas-clemente/quic-go/internal/mocks/handshake"
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
func mockKeyDerivation(crypto.TLSExporter, protocol.Perspective) (crypto.AEAD, error) {
|
|
return mockcrypto.NewMockAEAD(mockCtrl), nil
|
|
}
|
|
|
|
var _ = Describe("TLS Crypto Setup", func() {
|
|
var (
|
|
cs *cryptoSetupTLS
|
|
handshakeEvent chan struct{}
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
handshakeEvent = make(chan struct{}, 2)
|
|
cs = NewCryptoSetupTLSServer(
|
|
nil,
|
|
NewCryptoStreamConn(nil),
|
|
nil, // AEAD
|
|
handshakeEvent,
|
|
protocol.VersionTLS,
|
|
).(*cryptoSetupTLS)
|
|
cs.nullAEAD = mockcrypto.NewMockAEAD(mockCtrl)
|
|
})
|
|
|
|
It("errors when the handshake fails", func() {
|
|
alert := mint.AlertBadRecordMAC
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().Handshake().Return(alert)
|
|
err := cs.HandleCryptoStream()
|
|
Expect(err).To(MatchError(fmt.Errorf("TLS handshake error: %s (Alert %d)", alert.String(), alert)))
|
|
})
|
|
|
|
It("derives keys", func() {
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().Handshake().Return(mint.AlertNoAlert)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().State().Return(mint.StateServerConnected)
|
|
cs.keyDerivation = mockKeyDerivation
|
|
err := cs.HandleCryptoStream()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(handshakeEvent).To(Receive())
|
|
Expect(handshakeEvent).To(BeClosed())
|
|
})
|
|
|
|
It("handshakes until it is connected", func() {
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().Handshake().Return(mint.AlertNoAlert).Times(10)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().State().Return(mint.StateServerNegotiated).Times(9)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().State().Return(mint.StateServerConnected)
|
|
cs.keyDerivation = mockKeyDerivation
|
|
err := cs.HandleCryptoStream()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(handshakeEvent).To(Receive())
|
|
})
|
|
|
|
Context("reporting the handshake state", func() {
|
|
It("reports before the handshake compeletes", func() {
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().ConnectionState().Return(mint.ConnectionState{})
|
|
state := cs.ConnectionState()
|
|
Expect(state.HandshakeComplete).To(BeFalse())
|
|
Expect(state.PeerCertificates).To(BeNil())
|
|
})
|
|
|
|
It("reports after the handshake completes", func() {
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().ConnectionState().Return(mint.ConnectionState{})
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().Handshake().Return(mint.AlertNoAlert)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().State().Return(mint.StateServerConnected)
|
|
cs.keyDerivation = mockKeyDerivation
|
|
err := cs.HandleCryptoStream()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
state := cs.ConnectionState()
|
|
Expect(state.HandshakeComplete).To(BeTrue())
|
|
Expect(state.PeerCertificates).To(BeNil())
|
|
})
|
|
})
|
|
|
|
Context("escalating crypto", func() {
|
|
doHandshake := func() {
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().Handshake().Return(mint.AlertNoAlert)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().State().Return(mint.StateServerConnected)
|
|
cs.keyDerivation = mockKeyDerivation
|
|
err := cs.HandleCryptoStream()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
}
|
|
|
|
Context("null encryption", func() {
|
|
It("is used initially", func() {
|
|
cs.nullAEAD.(*mockcrypto.MockAEAD).EXPECT().Seal(nil, []byte("foobar"), protocol.PacketNumber(5), []byte{}).Return([]byte("foobar signed"))
|
|
enc, sealer := cs.GetSealer()
|
|
Expect(enc).To(Equal(protocol.EncryptionUnencrypted))
|
|
d := sealer.Seal(nil, []byte("foobar"), 5, []byte{})
|
|
Expect(d).To(Equal([]byte("foobar signed")))
|
|
})
|
|
|
|
It("is used for opening", func() {
|
|
cs.nullAEAD.(*mockcrypto.MockAEAD).EXPECT().Open(nil, []byte("foobar enc"), protocol.PacketNumber(10), []byte{}).Return([]byte("foobar"), nil)
|
|
d, err := cs.OpenHandshake(nil, []byte("foobar enc"), 10, []byte{})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(d).To(Equal([]byte("foobar")))
|
|
})
|
|
|
|
It("is used for crypto stream", func() {
|
|
cs.nullAEAD.(*mockcrypto.MockAEAD).EXPECT().Seal(nil, []byte("foobar"), protocol.PacketNumber(20), []byte{}).Return([]byte("foobar signed"))
|
|
enc, sealer := cs.GetSealerForCryptoStream()
|
|
Expect(enc).To(Equal(protocol.EncryptionUnencrypted))
|
|
d := sealer.Seal(nil, []byte("foobar"), 20, []byte{})
|
|
Expect(d).To(Equal([]byte("foobar signed")))
|
|
})
|
|
|
|
It("errors if the has the wrong hash", func() {
|
|
cs.nullAEAD.(*mockcrypto.MockAEAD).EXPECT().Open(nil, []byte("foobar enc"), protocol.PacketNumber(10), []byte{}).Return(nil, errors.New("authentication failed"))
|
|
_, err := cs.OpenHandshake(nil, []byte("foobar enc"), 10, []byte{})
|
|
Expect(err).To(MatchError("authentication failed"))
|
|
})
|
|
})
|
|
|
|
Context("forward-secure encryption", func() {
|
|
It("is used for sealing after the handshake completes", func() {
|
|
doHandshake()
|
|
cs.aead.(*mockcrypto.MockAEAD).EXPECT().Seal(nil, []byte("foobar"), protocol.PacketNumber(5), []byte{}).Return([]byte("foobar forward sec"))
|
|
enc, sealer := cs.GetSealer()
|
|
Expect(enc).To(Equal(protocol.EncryptionForwardSecure))
|
|
d := sealer.Seal(nil, []byte("foobar"), 5, []byte{})
|
|
Expect(d).To(Equal([]byte("foobar forward sec")))
|
|
})
|
|
|
|
It("is used for opening", func() {
|
|
doHandshake()
|
|
cs.aead.(*mockcrypto.MockAEAD).EXPECT().Open(nil, []byte("encrypted"), protocol.PacketNumber(6), []byte{}).Return([]byte("decrypted"), nil)
|
|
d, err := cs.Open1RTT(nil, []byte("encrypted"), 6, []byte{})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(d).To(Equal([]byte("decrypted")))
|
|
})
|
|
})
|
|
|
|
Context("forcing encryption levels", func() {
|
|
It("forces null encryption", func() {
|
|
doHandshake()
|
|
cs.nullAEAD.(*mockcrypto.MockAEAD).EXPECT().Seal(nil, []byte("foobar"), protocol.PacketNumber(5), []byte{}).Return([]byte("foobar signed"))
|
|
sealer, err := cs.GetSealerWithEncryptionLevel(protocol.EncryptionUnencrypted)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
d := sealer.Seal(nil, []byte("foobar"), 5, []byte{})
|
|
Expect(d).To(Equal([]byte("foobar signed")))
|
|
})
|
|
|
|
It("forces forward-secure encryption", func() {
|
|
doHandshake()
|
|
cs.aead.(*mockcrypto.MockAEAD).EXPECT().Seal(nil, []byte("foobar"), protocol.PacketNumber(5), []byte{}).Return([]byte("foobar forward sec"))
|
|
sealer, err := cs.GetSealerWithEncryptionLevel(protocol.EncryptionForwardSecure)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
d := sealer.Seal(nil, []byte("foobar"), 5, []byte{})
|
|
Expect(d).To(Equal([]byte("foobar forward sec")))
|
|
})
|
|
|
|
It("errors if the forward-secure AEAD is not available", func() {
|
|
sealer, err := cs.GetSealerWithEncryptionLevel(protocol.EncryptionForwardSecure)
|
|
Expect(err).To(MatchError("CryptoSetup: no sealer with encryption level forward-secure"))
|
|
Expect(sealer).To(BeNil())
|
|
})
|
|
|
|
It("never returns a secure AEAD (they don't exist with TLS)", func() {
|
|
doHandshake()
|
|
sealer, err := cs.GetSealerWithEncryptionLevel(protocol.EncryptionSecure)
|
|
Expect(err).To(MatchError("CryptoSetup: no sealer with encryption level encrypted (not forward-secure)"))
|
|
Expect(sealer).To(BeNil())
|
|
})
|
|
|
|
It("errors if no encryption level is specified", func() {
|
|
seal, err := cs.GetSealerWithEncryptionLevel(protocol.EncryptionUnspecified)
|
|
Expect(err).To(MatchError("CryptoSetup: no sealer with encryption level unknown"))
|
|
Expect(seal).To(BeNil())
|
|
})
|
|
})
|
|
})
|
|
})
|
|
|
|
var _ = Describe("TLS Crypto Setup, for the client", func() {
|
|
var (
|
|
cs *cryptoSetupTLS
|
|
handshakeEvent chan struct{}
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
handshakeEvent = make(chan struct{})
|
|
csInt, err := NewCryptoSetupTLSClient(
|
|
nil,
|
|
protocol.ConnectionID{1, 2, 3, 4, 5, 6, 7, 8},
|
|
"quic.clemente.io",
|
|
handshakeEvent,
|
|
nil, // mintTLS
|
|
protocol.VersionTLS,
|
|
)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
cs = csInt.(*cryptoSetupTLS)
|
|
})
|
|
|
|
It("returns when a retry is performed", func() {
|
|
cs.tls = mockhandshake.NewMockMintTLS(mockCtrl)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().Handshake().Return(mint.AlertNoAlert)
|
|
cs.tls.(*mockhandshake.MockMintTLS).EXPECT().State().Return(mint.StateClientStart)
|
|
err := cs.HandleCryptoStream()
|
|
Expect(err).To(MatchError(ErrCloseSessionForRetry))
|
|
})
|
|
|
|
})
|