From ecd1177511eb2a30382f72310c521832cd76e275 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=A8=E7=8E=89=E6=B3=A2?= <yangyubo@geedgenetworks.com>
Date: Mon, 24 Apr 2023 01:42:18 +0000
Subject: [PATCH] =?UTF-8?q?fix=20TSG-14832:Monitor=20SSH=E7=9A=84=E7=AD=96?=
 =?UTF-8?q?=E7=95=A5=EF=BC=8C=E5=AE=89=E5=85=A8=E6=97=A5=E5=BF=97=E4=B8=AD?=
 =?UTF-8?q?=E6=9C=89=E5=AF=B9=E5=BA=94=E7=AD=96=E7=95=A5=E7=9A=84=E6=97=A5?=
 =?UTF-8?q?=E5=BF=97=EF=BC=8C=E4=BD=86=E7=AD=96=E7=95=A5=E8=AF=A6=E6=83=85?=
 =?UTF-8?q?=E9=A1=B5log=20count=E4=B8=BA0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/tsg_entry.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/tsg_entry.cpp b/src/tsg_entry.cpp
index e4363af..3258299 100644
--- a/src/tsg_entry.cpp
+++ b/src/tsg_entry.cpp
@@ -1441,6 +1441,12 @@ static unsigned char matched_security_rules_deal(const struct streaminfo *a_stre
 			{
 				break;
 			}
+
+			if (srt_process_context->proto == PROTO_SSH)
+			{
+				tsg_set_policy_flow(a_stream, p_rule, a_stream->threadnum);
+			}
+
 			session_packet_capture_notify(a_stream, security_rules, n_security_rules, a_stream->threadnum);
 			session_matched_rules_notify(a_stream, TSG_SERVICE_SECURITY, security_rules, n_security_rules, a_stream->threadnum);
 			break;