mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-04 04:37:36 +03:00
2dbc29a5bd
There are two ways that an error can occur during the handshake: 1. as a return value from qtls.Handshake() 2. when new data is passed to the crypto setup via HandleData() We need to make sure that the RunHandshake() as well as HandleData() both return if an error occurs at any step during the handshake.
53 lines
1.2 KiB
Go
53 lines
1.2 KiB
Go
package quic
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
"github.com/lucas-clemente/quic-go/internal/wire"
|
|
)
|
|
|
|
type cryptoDataHandler interface {
|
|
HandleData([]byte, protocol.EncryptionLevel)
|
|
}
|
|
|
|
type cryptoStreamManager struct {
|
|
cryptoHandler cryptoDataHandler
|
|
|
|
initialStream cryptoStream
|
|
handshakeStream cryptoStream
|
|
}
|
|
|
|
func newCryptoStreamManager(
|
|
cryptoHandler cryptoDataHandler,
|
|
initialStream cryptoStream,
|
|
handshakeStream cryptoStream,
|
|
) *cryptoStreamManager {
|
|
return &cryptoStreamManager{
|
|
cryptoHandler: cryptoHandler,
|
|
initialStream: initialStream,
|
|
handshakeStream: handshakeStream,
|
|
}
|
|
}
|
|
|
|
func (m *cryptoStreamManager) HandleCryptoFrame(frame *wire.CryptoFrame, encLevel protocol.EncryptionLevel) error {
|
|
var str cryptoStream
|
|
switch encLevel {
|
|
case protocol.EncryptionInitial:
|
|
str = m.initialStream
|
|
case protocol.EncryptionHandshake:
|
|
str = m.handshakeStream
|
|
default:
|
|
return fmt.Errorf("received CRYPTO frame with unexpected encryption level: %s", encLevel)
|
|
}
|
|
if err := str.HandleCryptoFrame(frame); err != nil {
|
|
return err
|
|
}
|
|
for {
|
|
data := str.GetCryptoData()
|
|
if data == nil {
|
|
return nil
|
|
}
|
|
m.cryptoHandler.HandleData(data, encLevel)
|
|
}
|
|
}
|