bugfix:TSG-14802:修复没有将categoryid number传输给tfe的bugv5.1.2

1 files changed, 5 insertions, 1 deletions

diff --git a/entry/src/kni_entry.cpp b/entry/src/kni_entry.cpp
index aab72c2..5b7c2b2 100644
--- a/entry/src/kni_entry.cpp
+++ b/entry/src/kni_entry.cpp
@@ -513,6 +513,7 @@ static int session_attr_fqdn_category_id_cmsg_set(const struct session_runtime_a
 
 	n_category_ids = srt_attribute_get_category_ids((const struct session_runtime_attribute *)pmeinfo->session_attribute, category_ids, sizeof(category_ids)/sizeof(category_ids[0]));
 
+	ret = wrapped_kni_cmsg_set(cmsg, TFE_CMSG_FQDN_CAT_ID_NUM, (const unsigned char*)&n_category_ids, sizeof(unsigned int), pmeinfo);
 	if(n_category_ids <= 0 || n_category_ids > 8)
 	{
 		ret = wrapped_kni_cmsg_set(cmsg, TFE_CMSG_FQDN_CAT_ID_VAL, (const unsigned char*)empty_arr, strlen(empty_arr), pmeinfo);
@@ -1265,7 +1266,10 @@ static struct session_runtime_attribute * kni_pull_session_attribute_results(str
 		KNI_LOG_DEBUG(logger, "share-session-attribute: ja3_fingerprint is %s,stream traceid = %s", ja3_fingerprint, pmeinfo->stream_traceid);
 
 	unsigned int category_ids[8] = {0};
-	int n_category_ids = srt_attribute_set_category_ids(a_stream, category_ids, sizeof(category_ids)/sizeof(category_ids[0]));
+
+	int n_category_ids = srt_attribute_get_category_ids(session_attr, category_ids, sizeof(category_ids)/sizeof(category_ids[0]));
+
+	KNI_LOG_DEBUG(logger, "share-session-attribute: fqdn_category_id number is %d,stream traceid = %s", n_category_ids, pmeinfo->stream_traceid);
 	if(n_category_ids < 0 || n_category_ids > 8)
 	{
 		KNI_LOG_DEBUG(logger, "share-session-attribute: fqdn_category_id_num out of range( 0 <= value <= 8), value = %d,stream traceid = %s", n_category_ids, pmeinfo->stream_traceid);