cli: Report TCP prober stats with onion stats.

1 files changed, 6 insertions, 1 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index 1223edc8..e4e32f28 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -683,6 +683,9 @@ clientSession s@Session{..} sock cnum h = do
             icnt <- HashMap.size <$> readTVar (trampolineIds onionRouter)
             rs <- getAssocs (pendingRoutes onionRouter)
             pqs <- readTVar (pendingQueries onionRouter)
+            tcp_spill <- readTVar (TCP.probeSpill $ tcpProber onionRouter)
+            tcp_cache <- readTVar (TCP.probeCache $ tcpProber onionRouter)
+            tcp_queue <- readTVar (TCP.probeQueue $ tcpProber onionRouter)
             let showRecord :: Int -> Int -> [String]
                 showRecord n wanted_ver
                     | Just RouteRecord{responseCount,timeoutCount,routeVersion,routeBirthTime} <- IntMap.lookup n rm
@@ -694,7 +697,9 @@ clientSession s@Session{..} sock cnum h = do
                 r = map (uncurry showRecord) rs
             return $ do
                 hPutClientChunk h $ unlines [ "trampolines: " ++ show (IntMap.size ts,tcnt,icnt)
-                                            , "pending: " ++ show (W64.size pqs) ]
+                                            , "pending: " ++ show (W64.size pqs)
+                                            , "TCP spill,cache,queue: "
+                                                ++ show (PSQ.size tcp_spill, PSQ.size tcp_cache, PSQ.size tcp_queue)]
                 hPutClient h $ showColumns $ ["","responses","timeouts", "age", "version"]:r
 
         ("g", s) | Just DHT{..} <- Map.lookup netname dhts