commit bb93c604544dd63f12c0be4c805363d22fdf0b0b
author fayang <fayang@google.com> Mon Feb 24 08:31:42 2020 -0800
committer Copybara-Service <copybara-worker@google.com> Mon Feb 24 08:32:22 2020 -0800
tree 513ec4e664829f34ca62bdb4ffcb36343bd515ad
parent 6265bac9c2471df7f749c767f1933b2906d1fc1f

Add DCHECKs to make sure data does not get sent after connection closed.

gfe-relnote: n/a (DCHECK only)
PiperOrigin-RevId: 296897669
Change-Id: Ib333d6c3df430719bc92700e0eba68bd51c81367

quic/core/quic_session.cc

diff --git a/quic/core/quic_session.cc b/quic/core/quic_session.cc
index a47c265..073758e 100644
--- a/quic/core/quic_session.cc
+++ b/quic/core/quic_session.cc
@@ -739,6 +739,8 @@
                                          QuicStreamOffset offset,
                                          StreamSendingState state,
                                          bool is_retransmission) {
+  DCHECK(connection_->connected())
+      << ENDPOINT << "Try to write stream data when connection is closed.";
   if (!IsEncryptionEstablished() &&
       !QuicUtils::IsCryptoStreamId(transport_version(), id)) {
     // Do not let streams write without encryption. The calling stream will end
@@ -2094,6 +2096,8 @@
 }
 
 MessageResult QuicSession::SendMessage(QuicMemSliceSpan message, bool flush) {
+  DCHECK(connection_->connected())
+      << ENDPOINT << "Try to write messages when connection is closed.";
   if (!IsEncryptionEstablished()) {
     return {MESSAGE_STATUS_ENCRYPTION_NOT_ESTABLISHED, 0};
   }