46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
package wire
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"io"
|
|
|
|
"github.com/quic-go/quic-go/internal/protocol"
|
|
"github.com/quic-go/quic-go/quicvarint"
|
|
)
|
|
|
|
// A NewTokenFrame is a NEW_TOKEN frame
|
|
type NewTokenFrame struct {
|
|
Token []byte
|
|
}
|
|
|
|
func parseNewTokenFrame(r *bytes.Reader, _ protocol.Version) (*NewTokenFrame, error) {
|
|
tokenLen, err := quicvarint.Read(r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if uint64(r.Len()) < tokenLen {
|
|
return nil, io.EOF
|
|
}
|
|
if tokenLen == 0 {
|
|
return nil, errors.New("token must not be empty")
|
|
}
|
|
token := make([]byte, int(tokenLen))
|
|
if _, err := io.ReadFull(r, token); err != nil {
|
|
return nil, err
|
|
}
|
|
return &NewTokenFrame{Token: token}, nil
|
|
}
|
|
|
|
func (f *NewTokenFrame) Append(b []byte, _ protocol.Version) ([]byte, error) {
|
|
b = append(b, newTokenFrameType)
|
|
b = quicvarint.Append(b, uint64(len(f.Token)))
|
|
b = append(b, f.Token...)
|
|
return b, nil
|
|
}
|
|
|
|
// Length of a written frame
|
|
func (f *NewTokenFrame) Length(protocol.Version) protocol.ByteCount {
|
|
return 1 + quicvarint.Len(uint64(len(f.Token))) + protocol.ByteCount(len(f.Token))
|
|
}
|