From f14dcb8344f1cbb1197cdc4f341a6fe4fdcc0cf6 Mon Sep 17 00:00:00 2001
From: Toni <matzeton@googlemail.com>
Date: Fri, 25 Feb 2022 14:26:04 +0100
Subject: Provide some API functions for convenience. (#1456)
* Extended JSON serializsation: risk, risk score, confidence
Signed-off-by: Toni Uhlig <matzeton@googlemail.com>
---
src/lib/ndpi_main.c | 21 +++++++++++++++++++++
1 file changed, 21 insertions(+)
(limited to 'src/lib/ndpi_main.c')
diff --git a/src/lib/ndpi_main.c b/src/lib/ndpi_main.c
index 37d73172c..c0a871cc5 100644
--- a/src/lib/ndpi_main.c
+++ b/src/lib/ndpi_main.c
@@ -6490,6 +6490,27 @@ u_int16_t ndpi_get_flow_masterprotocol(struct ndpi_detection_module_struct *ndpi
/* ********************************************************************************* */
+u_int16_t ndpi_get_flow_appprotocol(struct ndpi_detection_module_struct *ndpi_str, struct ndpi_flow_struct *flow) {
+ return(flow->detected_protocol_stack[0]);
+}
+
+/* ********************************************************************************* */
+
+ndpi_protocol_category_t ndpi_get_flow_category(struct ndpi_detection_module_struct *ndpi_str, struct ndpi_flow_struct *flow)
+{
+ return(flow->category);
+}
+
+void ndpi_get_flow_ndpi_proto(struct ndpi_detection_module_struct *ndpi_str, struct ndpi_flow_struct *flow,
+ struct ndpi_proto * ndpi_proto)
+{
+ ndpi_proto->master_protocol = ndpi_get_flow_masterprotocol(ndpi_str, flow);
+ ndpi_proto->app_protocol = ndpi_get_flow_appprotocol(ndpi_str, flow);
+ ndpi_proto->category = ndpi_get_flow_category(ndpi_str, flow);
+}
+
+/* ********************************************************************************* */
+
static void ndpi_int_change_flow_protocol(struct ndpi_detection_module_struct *ndpi_str, struct ndpi_flow_struct *flow,
u_int16_t upper_detected_protocol, u_int16_t lower_detected_protocol,
ndpi_confidence_t confidence) {
--
cgit v1.2.3