From f71422e1fb6ea975edfdc25bf65aabe9cc0abed5 Mon Sep 17 00:00:00 2001
From: Sijie Yang <iyangsj@gmail.com>
Date: Fri, 12 Jan 2024 19:41:40 +0800
Subject: [PATCH] Add `cid_len` option to tquic tools (#139)

---
 tools/src/bin/tquic_client.rs | 5 +++++
 tools/src/bin/tquic_server.rs | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/tools/src/bin/tquic_client.rs b/tools/src/bin/tquic_client.rs
index 35795985..9fed94a4 100644
--- a/tools/src/bin/tquic_client.rs
+++ b/tools/src/bin/tquic_client.rs
@@ -212,6 +212,10 @@ pub struct ClientOpt {
     #[clap(long, value_name = "FILE")]
     pub qlog_file: Option<String>,
 
+    /// Length of connection id in bytes.
+    #[clap(long, default_value = "8", value_name = "NUM")]
+    pub cid_len: usize,
+
     /// Batch size for sending packets.
     #[clap(long, default_value = "1", value_name = "NUM")]
     pub send_batch_size: usize,
@@ -422,6 +426,7 @@ impl Worker {
         config.set_max_pto(option.max_pto);
         config.set_max_concurrent_conns(option.max_concurrent_conns);
         config.set_initial_max_streams_bidi(option.max_concurrent_requests);
+        config.set_cid_len(option.cid_len);
         config.set_send_batch_size(option.send_batch_size);
         config.set_recv_udp_payload_size(option.recv_udp_payload_size);
         config.set_send_udp_payload_size(option.send_udp_payload_size);
diff --git a/tools/src/bin/tquic_server.rs b/tools/src/bin/tquic_server.rs
index b2ddec8f..1fff8c80 100644
--- a/tools/src/bin/tquic_server.rs
+++ b/tools/src/bin/tquic_server.rs
@@ -150,6 +150,10 @@ pub struct ServerOpt {
     #[clap(long, value_name = "FILE")]
     pub qlog_file: Option<String>,
 
+    /// Length of connection id in bytes.
+    #[clap(long, default_value = "8", value_name = "NUM")]
+    pub cid_len: usize,
+
     /// Batch size for sending packets.
     #[clap(long, default_value = "16", value_name = "NUM")]
     pub send_batch_size: usize,
@@ -185,6 +189,7 @@ impl Server {
         config.set_initial_rtt(option.initial_rtt);
         config.set_pto_linear_factor(option.pto_linear_factor);
         config.set_max_pto(option.max_pto);
+        config.set_cid_len(option.cid_len);
         config.set_send_batch_size(option.send_batch_size);
         config.set_congestion_control_algorithm(option.congestion_control_algor);
         config.set_initial_congestion_window(option.initial_congestion_window);