From 16ca1bc87c5a3d832f0412ae106dcc3e44b06f3a Mon Sep 17 00:00:00 2001
From: Konstantin Tsoy <ktsoy@meta.com>
Date: Tue, 8 Oct 2024 15:50:47 -0700
Subject: [PATCH] Expose PTO count and overall packets sent count for the
 client

Summary: Expose PTO count and overall packets sent count for the client

Reviewed By: sharmafb

Differential Revision: D64006928

fbshipit-source-id: f28b29c175271683391375fb97bb43fef8598034
---
 quic/client/QuicClientTransport.cpp | 8 ++++++++
 quic/client/QuicClientTransport.h   | 3 +++
 2 files changed, 11 insertions(+)

diff --git a/quic/client/QuicClientTransport.cpp b/quic/client/QuicClientTransport.cpp
index 4ba71a6f6..2403e5ce6 100644
--- a/quic/client/QuicClientTransport.cpp
+++ b/quic/client/QuicClientTransport.cpp
@@ -1891,4 +1891,12 @@ uint64_t QuicClientTransport::getEnobufsCount() const {
   return conn_->enobufsCount;
 }
 
+uint64_t QuicClientTransport::getPtoCount() const {
+  return conn_->lossState.ptoCount;
+}
+
+uint64_t QuicClientTransport::getPacketsSentCount() const {
+  return conn_->lossState.totalPacketsSent;
+}
+
 } // namespace quic
diff --git a/quic/client/QuicClientTransport.h b/quic/client/QuicClientTransport.h
index ae5f1500d..71165412b 100644
--- a/quic/client/QuicClientTransport.h
+++ b/quic/client/QuicClientTransport.h
@@ -226,6 +226,9 @@ class QuicClientTransport
 
   uint64_t getEnobufsCount() const;
 
+  uint64_t getPtoCount() const;
+  uint64_t getPacketsSentCount() const;
+
   class HappyEyeballsConnAttemptDelayTimeout : public QuicTimerCallback {
    public:
     explicit HappyEyeballsConnAttemptDelayTimeout(