commit 81f5d44aafb9a5ef3ed5897ff21c174cceb036b4
author danzh <danzh@google.com> Fri Jul 02 14:24:37 2021 -0700
committer Copybara-Service <copybara-worker@google.com> Fri Jul 02 14:25:05 2021 -0700
tree d7d01c3ef895168f6a04b4293980a9115774bb9f
parent a1363dcb9c2b25780c7e4a527b83fc8392a681a5

Internal change

PiperOrigin-RevId: 382812505

quic/tools/quic_toy_client.cc

diff --git a/quic/tools/quic_toy_client.cc b/quic/tools/quic_toy_client.cc
index 8d22791..81a0202 100644
--- a/quic/tools/quic_toy_client.cc
+++ b/quic/tools/quic_toy_client.cc
@@ -211,6 +211,10 @@
                               -1,
                               "Length of the client connection ID used.");
 
+DEFINE_QUIC_COMMAND_LINE_FLAG(int32_t, max_time_before_crypto_handshake_ms,
+                              10000,
+                              "Max time to wait before handshake completes.");
+
 namespace quic {
 
 QuicToyClient::QuicToyClient(ClientFactory* client_factory)
@@ -293,6 +297,8 @@
     config.custom_transport_parameters_to_send()[kCustomParameter] =
         custom_value;
   }
+  config.set_max_time_before_crypto_handshake(QuicTime::Delta::FromMilliseconds(
+      GetQuicFlag(FLAGS_max_time_before_crypto_handshake_ms)));
 
   int address_family_for_lookup = AF_UNSPEC;
   if (GetQuicFlag(FLAGS_ip_version_for_host_lookup) == "4") {