gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

simplify the crypto setup constructors

Browse Source
This commit is contained in:
Marten Seemann
2018-11-10 16:58:01 +07:00
parent c2aac727b9
commit 8c103be876
1 changed files with 36 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
internal/handshake/crypto_setup.go
View File

@@ -95,12 +95,6 @@ type cryptoSetup struct {
var _ qtls.RecordLayer = &cryptoSetup{}
var _ CryptoSetup = &cryptoSetup{}
type versionInfo struct {
initialVersion protocol.VersionNumber
supportedVersions []protocol.VersionNumber
currentVersion protocol.VersionNumber
}
// NewCryptoSetupClient creates a new crypto setup for the client
func NewCryptoSetupClient(
initialStream io.Writer,
@@ -115,18 +109,21 @@ func NewCryptoSetupClient(
logger utils.Logger,
perspective protocol.Perspective,
) (CryptoSetup, <-chan struct{} /* ClientHello written */, error) {
extHandler, receivedTransportParams := newExtensionHandlerClient(
params,
initialVersion,
supportedVersions,
currentVersion,
logger,
)
return newCryptoSetup(
initialStream,
handshakeStream,
connID,
params,
extHandler,
receivedTransportParams,
handleParams,
tlsConf,
versionInfo{
currentVersion: currentVersion,
initialVersion: initialVersion,
supportedVersions: supportedVersions,
},
logger,
perspective,
)
@@ -145,17 +142,20 @@ func NewCryptoSetupServer(
logger utils.Logger,
perspective protocol.Perspective,
) (CryptoSetup, error) {
extHandler, receivedTransportParams := newExtensionHandlerServer(
params,
supportedVersions,
currentVersion,
logger,
)
cs, _, err := newCryptoSetup(
initialStream,
handshakeStream,
connID,
params,
extHandler,
receivedTransportParams,
handleParams,
tlsConf,
versionInfo{
currentVersion: currentVersion,
supportedVersions: supportedVersions,
},
logger,
perspective,
)
@@ -166,10 +166,10 @@ func newCryptoSetup(
initialStream io.Writer,
handshakeStream io.Writer,
connID protocol.ConnectionID,
params *TransportParameters,
extHandler tlsExtensionHandler,
transportParamChan <-chan TransportParameters,
handleParams func(*TransportParameters),
tlsConf *tls.Config,
versionInfo versionInfo,
logger utils.Logger,
perspective protocol.Perspective,
) (CryptoSetup, <-chan struct{} /* ClientHello written */, error) {
@@ -178,40 +178,23 @@ func newCryptoSetup(
return nil, nil, err
}
cs := &cryptoSetup{
initialStream: initialStream,
initialAEAD: initialAEAD,
handshakeStream: handshakeStream,
readEncLevel: protocol.EncryptionInitial,
writeEncLevel: protocol.EncryptionInitial,
handleParamsCallback: handleParams,
logger: logger,
perspective: perspective,
handshakeDone: make(chan struct{}),
handshakeErrChan: make(chan struct{}),
messageErrChan: make(chan error, 1),
clientHelloWrittenChan: make(chan struct{}),
messageChan: make(chan []byte, 100),
receivedReadKey: make(chan struct{}),
receivedWriteKey: make(chan struct{}),
closeChan: make(chan struct{}),
}
var extHandler tlsExtensionHandler
switch perspective {
case protocol.PerspectiveClient:
extHandler, cs.receivedTransportParams = newExtensionHandlerClient(
params,
versionInfo.initialVersion,
versionInfo.supportedVersions,
versionInfo.currentVersion,
logger,
)
case protocol.PerspectiveServer:
extHandler, cs.receivedTransportParams = newExtensionHandlerServer(
params,
versionInfo.supportedVersions,
versionInfo.currentVersion,
logger,
)
initialStream: initialStream,
initialAEAD: initialAEAD,
handshakeStream: handshakeStream,
readEncLevel: protocol.EncryptionInitial,
writeEncLevel: protocol.EncryptionInitial,
handleParamsCallback: handleParams,
receivedTransportParams: transportParamChan,
logger: logger,
perspective: perspective,
handshakeDone: make(chan struct{}),
handshakeErrChan: make(chan struct{}),
messageErrChan: make(chan error, 1),
clientHelloWrittenChan: make(chan struct{}),
messageChan: make(chan []byte, 100),
receivedReadKey: make(chan struct{}),
receivedWriteKey: make(chan struct{}),
closeChan: make(chan struct{}),
}
qtlsConf := tlsConfigToQtlsConfig(tlsConf)
qtlsConf.AlternativeRecordLayer = cs