From 4297a65ce881c8c5462c520f3b3dc987879bccce Mon Sep 17 00:00:00 2001
From: Luca Deri <deri@ntop.org>
Date: Mon, 10 May 2021 22:43:05 +0200
Subject: Implemented flow score in Wireshark integration

---
 src/lib/ndpi_utils.c | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

(limited to 'src/lib/ndpi_utils.c')

diff --git a/src/lib/ndpi_utils.c b/src/lib/ndpi_utils.c
index 07b1cbb02..91055f694 100644
--- a/src/lib/ndpi_utils.c
+++ b/src/lib/ndpi_utils.c
@@ -1839,6 +1839,39 @@ ndpi_risk_severity ndpi_risk2severity(ndpi_risk_enum risk) {
 
 /* ******************************************************************** */
 
+u_int16_t ndpi_risk2score(ndpi_risk_enum risk) {
+  u_int16_t score = 0;
+  u_int32_t i;
+  
+  for(i = 0; i < NDPI_MAX_RISK; i++) {
+    ndpi_risk_enum r = (ndpi_risk_enum)i;
+    
+    if(NDPI_ISSET_BIT(risk, r)) {
+      switch(ndpi_risk2severity(r)) {
+      case NDPI_RISK_LOW:
+	score += NDPI_SCORE_RISK_LOW;
+	break;
+
+      case NDPI_RISK_MEDIUM:
+	score += NDPI_SCORE_RISK_MEDIUM;
+	break;
+
+      case NDPI_RISK_HIGH:
+	score += NDPI_SCORE_RISK_HIGH;
+	break;
+
+      case NDPI_RISK_SEVERE:
+	score += NDPI_SCORE_RISK_SEVERE;
+	break;
+      }
+    }
+  }
+
+  return(score);
+}
+
+/* ******************************************************************** */
+
 const char* ndpi_http_method2str(ndpi_http_method m) {
   switch(m) {
   case NDPI_HTTP_METHOD_UNKNOWN: break;
-- 
cgit v1.2.3