83 lines
2.1 KiB
Go
83 lines
2.1 KiB
Go
package quic
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/quic-go/quic-go/internal/handshake"
|
|
"github.com/quic-go/quic-go/internal/protocol"
|
|
"github.com/quic-go/quic-go/internal/wire"
|
|
)
|
|
|
|
type cryptoDataHandler interface {
|
|
HandleMessage([]byte, protocol.EncryptionLevel) error
|
|
NextEvent() handshake.Event
|
|
}
|
|
|
|
type cryptoStreamManager struct {
|
|
cryptoHandler cryptoDataHandler
|
|
|
|
initialStream cryptoStream
|
|
handshakeStream cryptoStream
|
|
oneRTTStream cryptoStream
|
|
}
|
|
|
|
func newCryptoStreamManager(
|
|
cryptoHandler cryptoDataHandler,
|
|
initialStream cryptoStream,
|
|
handshakeStream cryptoStream,
|
|
oneRTTStream cryptoStream,
|
|
) *cryptoStreamManager {
|
|
return &cryptoStreamManager{
|
|
cryptoHandler: cryptoHandler,
|
|
initialStream: initialStream,
|
|
handshakeStream: handshakeStream,
|
|
oneRTTStream: oneRTTStream,
|
|
}
|
|
}
|
|
|
|
func (m *cryptoStreamManager) HandleCryptoFrame(frame *wire.CryptoFrame, encLevel protocol.EncryptionLevel) error {
|
|
var str cryptoStream
|
|
//nolint:exhaustive // CRYPTO frames cannot be sent in 0-RTT packets.
|
|
switch encLevel {
|
|
case protocol.EncryptionInitial:
|
|
str = m.initialStream
|
|
case protocol.EncryptionHandshake:
|
|
str = m.handshakeStream
|
|
case protocol.Encryption1RTT:
|
|
str = m.oneRTTStream
|
|
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
|
|
}
|
|
if err := m.cryptoHandler.HandleMessage(data, encLevel); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
|
|
func (m *cryptoStreamManager) GetPostHandshakeData(maxSize protocol.ByteCount) *wire.CryptoFrame {
|
|
if !m.oneRTTStream.HasData() {
|
|
return nil
|
|
}
|
|
return m.oneRTTStream.PopCryptoFrame(maxSize)
|
|
}
|
|
|
|
func (m *cryptoStreamManager) Drop(encLevel protocol.EncryptionLevel) error {
|
|
//nolint:exhaustive // 1-RTT keys should never get dropped.
|
|
switch encLevel {
|
|
case protocol.EncryptionInitial:
|
|
return m.initialStream.Finish()
|
|
case protocol.EncryptionHandshake:
|
|
return m.handshakeStream.Finish()
|
|
default:
|
|
panic(fmt.Sprintf("dropped unexpected encryption level: %s", encLevel))
|
|
}
|
|
}
|