From cc39ca2398b9f0b8b0109ab40c32b6a47b820a5b Mon Sep 17 00:00:00 2001
From: Kevin Krakauer <krakauer@google.com>
Date: Thu, 8 Jun 2023 10:37:40 -0700
Subject: [PATCH] tcp_benchmark: only output the relevant CPU usage

This just makes it easier to script and automate benchmarking.

PiperOrigin-RevId: 538827909
---
 test/benchmarks/tcp/tcp_benchmark.sh | 17 ++++++++++++++++-
 1 file changed, 16 insertions(+), 1 deletion(-)

diff --git a/test/benchmarks/tcp/tcp_benchmark.sh b/test/benchmarks/tcp/tcp_benchmark.sh
index ee03f4e59..933f8ff1f 100755
--- a/test/benchmarks/tcp/tcp_benchmark.sh
+++ b/test/benchmarks/tcp/tcp_benchmark.sh
@@ -461,5 +461,20 @@ server_cpu_ticks=\$(cat /proc/\$server_pid/stat \\
 ticks_per_sec=\$(getconf CLK_TCK)
 client_cpu_load=\$(bc -l <<< \$client_cpu_ticks/\$ticks_per_sec/${duration})
 server_cpu_load=\$(bc -l <<< \$server_cpu_ticks/\$ticks_per_sec/${duration})
-echo \$mbits \$client_cpu_load \$server_cpu_load
+
+hostgso=true
+if [[ "${disable_linux_gso}" ]]; then
+  hostgso=false
+fi
+hostgro=true
+if [[ "${disable_linux_gro}" ]]; then
+  hostgro=false
+fi
+
+if ${client}; then
+  echo "BenchmarkTCPClient/role=client/host-gso=\$hostgso/host-gro=\$hostgro 1 \$mbits Mb/s \$client_cpu_load cpu-time"
+elif ${server}; then
+  echo "BenchmarkTCPClient/role=server/host-gso=\$hostgso/host-gro=\$hostgro 1 \$mbits Mb/s \$server_cpu_load cpu-time"
+fi
+
 EOF