commit a6a85a8059f31263fdae813e5d47e304cf293cad
author fayang <fayang@google.com> Mon May 04 08:58:53 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Mon May 04 08:59:37 2020 -0700
tree 91c71df34d34d989ce637080d6b786d6fded0fed
parent 9a31bd55ff106cbe33c9d93861aa80855a04aefa

gfe-relnote: In QUIC with TLS, do not proceed in SetWriteSecret if connection has been closed. Protected by gfe2_reloadable_flag_quic_notify_handshaker_on_connection_close.

PiperOrigin-RevId: 309753183
Change-Id: I829b92d82ca84f85ab60aa09940a3b205641a34b

quic/core/quic_crypto_client_stream.h

diff --git a/quic/core/quic_crypto_client_stream.h b/quic/core/quic_crypto_client_stream.h
index 82b0ef9..23f83c7 100644
--- a/quic/core/quic_crypto_client_stream.h
+++ b/quic/core/quic_crypto_client_stream.h
@@ -159,6 +159,10 @@
     // Called when a packet of ENCRYPTION_HANDSHAKE gets sent.
     virtual void OnHandshakePacketSent() = 0;
 
+    // Called when connection gets closed.
+    virtual void OnConnectionClosed(QuicErrorCode error,
+                                    ConnectionCloseSource source) = 0;
+
     // Called when handshake done has been received.
     virtual void OnHandshakeDoneReceived() = 0;
 
@@ -215,6 +219,8 @@
   void OnPacketDecrypted(EncryptionLevel /*level*/) override {}
   void OnOneRttPacketAcknowledged() override;
   void OnHandshakePacketSent() override;
+  void OnConnectionClosed(QuicErrorCode error,
+                          ConnectionCloseSource source) override;
   void OnHandshakeDoneReceived() override;
   HandshakeState GetHandshakeState() const override;
   size_t BufferSizeLimitForLevel(EncryptionLevel level) const override;