mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-04 12:47:36 +03:00
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package quic
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
|
|
"github.com/lucas-clemente/quic-go/crypto"
|
|
"github.com/lucas-clemente/quic-go/errorcodes"
|
|
"github.com/lucas-clemente/quic-go/frames"
|
|
"github.com/lucas-clemente/quic-go/protocol"
|
|
)
|
|
|
|
type unpackedPacket struct {
|
|
entropyBit bool
|
|
frames []frames.Frame
|
|
}
|
|
|
|
type packetUnpacker struct {
|
|
version protocol.VersionNumber
|
|
aead crypto.AEAD
|
|
}
|
|
|
|
func (u *packetUnpacker) Unpack(publicHeaderBinary []byte, hdr *publicHeader, r *bytes.Reader) (*unpackedPacket, error) {
|
|
ciphertext, _ := ioutil.ReadAll(r)
|
|
plaintext, err := u.aead.Open(hdr.PacketNumber, publicHeaderBinary, ciphertext)
|
|
if err != nil {
|
|
// Wrap err in quicError so that public reset is sent by session
|
|
return nil, protocol.NewQuicError(errorcodes.QUIC_DECRYPTION_FAILURE, err.Error())
|
|
}
|
|
r = bytes.NewReader(plaintext)
|
|
|
|
privateFlag, err := r.ReadByte()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
entropyBit := privateFlag&0x01 > 0
|
|
|
|
fs := []frames.Frame{}
|
|
|
|
// Read all frames in the packet
|
|
ReadLoop:
|
|
for r.Len() > 0 {
|
|
typeByte, _ := r.ReadByte()
|
|
r.UnreadByte()
|
|
|
|
var frame frames.Frame
|
|
if typeByte&0x80 == 0x80 {
|
|
frame, err = frames.ParseStreamFrame(r)
|
|
} else if typeByte&0xc0 == 0x40 {
|
|
frame, err = frames.ParseAckFrame(r, u.version)
|
|
} else if typeByte&0xe0 == 0x20 {
|
|
err = errors.New("unimplemented: CONGESTION_FEEDBACK")
|
|
} else {
|
|
switch typeByte {
|
|
case 0x0: // PAD, end of frames
|
|
break ReadLoop
|
|
case 0x01:
|
|
frame, err = frames.ParseRstStreamFrame(r)
|
|
case 0x02:
|
|
frame, err = frames.ParseConnectionCloseFrame(r)
|
|
case 0x03:
|
|
err = errors.New("unimplemented: GOAWAY")
|
|
case 0x04:
|
|
frame, err = frames.ParseWindowUpdateFrame(r)
|
|
case 0x05:
|
|
frame, err = frames.ParseBlockedFrame(r)
|
|
case 0x06:
|
|
frame, err = frames.ParseStopWaitingFrame(r, hdr.PacketNumber, hdr.PacketNumberLen)
|
|
case 0x07:
|
|
frame, err = frames.ParsePingFrame(r)
|
|
default:
|
|
err = protocol.NewQuicError(errorcodes.QUIC_INVALID_FRAME_DATA, fmt.Sprintf("unknown type byte 0x%x", typeByte))
|
|
}
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// TODO: Remove once all frames are implemented
|
|
if frame != nil {
|
|
fs = append(fs, frame)
|
|
}
|
|
}
|
|
|
|
return &unpackedPacket{
|
|
entropyBit: entropyBit,
|
|
frames: fs,
|
|
}, nil
|
|
}
|