mirrors/uquic
1
0
Fork 0
mirror of https://github.com/refraction-networking/uquic.git synced 2025-04-06 21:57:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

use the connection ID manager to save the destination connection ID

Browse source
This commit is contained in:
Marten Seemann 2019-10-25 17:58:53 +07:00
parent a321f9faa6
commit 772ffd3d20
7 changed files with 125 additions and 121 deletions
Download patch file Download diff file Expand all files Collapse all files

17
packet_packer.go
View file

@ -21,7 +21,6 @@ type packer interface {
HandleTransportParameters(*handshake.TransportParameters)
SetToken([]byte)
ChangeDestConnectionID(protocol.ConnectionID)
}
type sealer interface {
@ -128,8 +127,8 @@ type ackFrameSource interface {
}
type packetPacker struct {
destConnID protocol.ConnectionID
srcConnID protocol.ConnectionID
srcConnID protocol.ConnectionID
getDestConnID func() protocol.ConnectionID
perspective protocol.Perspective
version protocol.VersionNumber
@ -155,8 +154,8 @@ type packetPacker struct {
var _ packer = &packetPacker{}
func newPacketPacker(
destConnID protocol.ConnectionID,
srcConnID protocol.ConnectionID,
getDestConnID func() protocol.ConnectionID,
initialStream cryptoStream,
handshakeStream cryptoStream,
packetNumberManager packetNumberManager,
@ -170,7 +169,7 @@ func newPacketPacker(
) *packetPacker {
return &packetPacker{
cryptoSetup: cryptoSetup,
destConnID: destConnID,
getDestConnID: getDestConnID,
srcConnID: srcConnID,
initialStream: initialStream,
handshakeStream: handshakeStream,
@ -432,7 +431,7 @@ func (p *packetPacker) getShortHeader(kp protocol.KeyPhaseBit) *wire.ExtendedHea
hdr := &wire.ExtendedHeader{}
hdr.PacketNumber = pn
hdr.PacketNumberLen = pnLen
hdr.DestConnectionID = p.destConnID
hdr.DestConnectionID = p.getDestConnID()
hdr.KeyPhase = kp
return hdr
}
@ -442,7 +441,7 @@ func (p *packetPacker) getLongHeader(encLevel protocol.EncryptionLevel) *wire.Ex
hdr := &wire.ExtendedHeader{}
hdr.PacketNumber = pn
hdr.PacketNumberLen = pnLen
hdr.DestConnectionID = p.destConnID
hdr.DestConnectionID = p.getDestConnID()
switch encLevel {
case protocol.EncryptionInitial:
@ -550,10 +549,6 @@ func (p *packetPacker) writeAndSealPacketWithPadding(
}, nil
}
func (p *packetPacker) ChangeDestConnectionID(connID protocol.ConnectionID) {
p.destConnID = connID
}
func (p *packetPacker) SetToken(token []byte) {
p.token = token
}