commit d6db04af4b7e1c5d0de30dabd3baa0d883a7f2b5
author fayang <fayang@google.com> Wed Oct 02 14:21:42 2019 -0700
committer Copybara-Service <copybara-worker@google.com> Thu Oct 03 12:36:46 2019 -0700
tree 047f90f9ebb6bb736d44c3506d45bcb14cc98d29
parent ffbc29361a62c2681e5e57af1357cfc0868b4059

gfe-relnote: Check for partial write when crypto frame gets retransmitted. This only affects T099 which supports amplification factor.

PiperOrigin-RevId: 272518334
Change-Id: Ia79b1ae533824e0c4a06de53a24451ea5f697e2b

quic/core/quic_crypto_stream.cc

diff --git a/quic/core/quic_crypto_stream.cc b/quic/core/quic_crypto_stream.cc
index 3081516..d62417f 100644
--- a/quic/core/quic_crypto_stream.cc
+++ b/quic/core/quic_crypto_stream.cc
@@ -256,6 +256,9 @@
       size_t bytes_consumed = session()->connection()->SendCryptoData(
           level, pending.length, pending.offset);
       send_buffer->OnStreamDataRetransmitted(pending.offset, bytes_consumed);
+      if (bytes_consumed < pending.length) {
+        break;
+      }
     }
   }
   session()->connection()->SetDefaultEncryptionLevel(current_encryption_level);
@@ -401,6 +404,9 @@
         crypto_frame->level, retransmission_length, retransmission_offset);
     send_buffer->OnStreamDataRetransmitted(retransmission_offset,
                                            bytes_consumed);
+    if (bytes_consumed < retransmission_length) {
+      break;
+    }
   }
   session()->connection()->SetDefaultEncryptionLevel(current_encryption_level);
 }