From c034dc68ed650b782a668dda35bfd74918c04bc1 Mon Sep 17 00:00:00 2001
From: Vern Paxson <vern@corelight.com>
Date: Mon, 1 Aug 2022 13:36:24 -0700
Subject: [PATCH] fix for profiling "when" statements

---
 src/Stmt.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/Stmt.cc b/src/Stmt.cc
index 6b3dd99542..9dfd7ea1f4 100644
--- a/src/Stmt.cc
+++ b/src/Stmt.cc
@@ -2185,6 +2185,12 @@ TraversalCode WhenStmt::Traverse(TraversalCallback* cb) const
 			}
 		}
 
+	if ( wi->TimeoutExpr() )
+		{
+		tc = wi->TimeoutExpr()->Traverse(cb);
+		HANDLE_TC_STMT_PRE(tc);
+		}
+
 	tc = cb->PostStmt(this);
 	HANDLE_TC_STMT_POST(tc);
 	}