gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

drop initial packets when the handshake is confirmed

Browse Source
This commit is contained in:
Marten Seemann
2025-09-23 00:26:32 +08:00
parent c385cd10f1
commit 275c172fec
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
connection.go
View File

@@ -789,6 +789,9 @@ func (s *connection) handleHandshakeComplete(now time.Time) error {
}
func (s *connection) handleHandshakeConfirmed(now time.Time) error {
if err := s.dropEncryptionLevel(protocol.EncryptionInitial, now); err != nil {
return err
}
if err := s.dropEncryptionLevel(protocol.EncryptionHandshake, now); err != nil {
return err
}

3
connection_test.go
View File

@@ -1102,7 +1102,7 @@ func TestConnectionHandshakeServer(t *testing.T) {
data, err := (&wire.CryptoFrame{Data: []byte("foobar")}).Append(nil, protocol.Version1)
require.NoError(t, err)
cs.EXPECT().DiscardInitialKeys()
cs.EXPECT().DiscardInitialKeys().Times(2)
tc.connRunner.EXPECT().Retire(gomock.Any())
gomock.InOrder(
cs.EXPECT().StartHandshake(gomock.Any()),
@@ -1254,6 +1254,7 @@ func testConnectionHandshakeClient(t *testing.T, usePreferredAddress bool) {
unpacker.EXPECT().UnpackLongHeader(gomock.Any(), gomock.Any()).Return(
&unpackedPacket{hdr: hdr, encryptionLevel: protocol.Encryption1RTT, data: data}, nil,
),
cs.EXPECT().DiscardInitialKeys(),
cs.EXPECT().SetHandshakeConfirmed(),
tc.packer.EXPECT().AppendPacket(gomock.Any(), gomock.Any(), gomock.Any(), gomock.Any()).DoAndReturn(
func(buf *packetBuffer, _ protocol.ByteCount, _ time.Time, _ protocol.Version) (shortHeaderPacket, error) {