Simplify the QuicDispatcher::OnCanWrite logic

net/quic/quic_dispatcher.cc

Index: net/quic/quic_dispatcher.cc
diff --git a/net/quic/quic_dispatcher.cc b/net/quic/quic_dispatcher.cc
index ffdce7c9286214c57e85fe6d942b300d573d3a0e..1f843eead244d0c91902b67366cc9d6ab4e3ca9f 100644
--- a/net/quic/quic_dispatcher.cc
+++ b/net/quic/quic_dispatcher.cc
@@ -292,8 +292,8 @@ void QuicDispatcher::OnCanWrite() {
   // We finished a write: the socket should not be blocked.
   writer_->SetWritable();
 
-  // Let all the blocked writers try to write, until we're blocked again or
-  // there's no work left.
+  // Give all the blocked writers one chance to write, until we're blocked again
+  // or there's no work left.
   while (!write_blocked_list_.empty() && !writer_->IsWriteBlocked()) {
     QuicBlockedWriterInterface* blocked_writer =
         write_blocked_list_.begin()->first;
@@ -337,9 +337,15 @@ void QuicDispatcher::OnConnectionClosed(QuicConnectionId connection_id,
   CleanUpSession(it);
 }
 
-void QuicDispatcher::OnWriteBlocked(QuicBlockedWriterInterface* writer) {
-  DCHECK(writer_->IsWriteBlocked());
-  write_blocked_list_.insert(make_pair(writer, true));
+void QuicDispatcher::OnWriteBlocked(QuicBlockedWriterInterface* blocked) {
+  if (!writer_->IsWriteBlocked()) {
+    LOG(DFATAL) <<
+        "QuicDispatcher::OnWriteBlocked called when the writer is not blocked.";
+    // Return without adding the session to the blocked list, to avoid infinite
+    // loops in OnCanWrite
+    return;
+  }
+  write_blocked_list_.insert(make_pair(blocked, true));
 }
 
 QuicSession* QuicDispatcher::CreateQuicSession(