From c49d126d3642d5b1f5168d049e3ebf0ee3451edc Mon Sep 17 00:00:00 2001
From: Leonardo Teixeira Alves <49043232+LTxAlves@users.noreply.github.com>
Date: Thu, 6 Mar 2025 10:47:17 -0300
Subject: Add Autonomous System Organization to geoip (#2763)

Co-authored-by: Leonardo Teixeira Alves <leonardo.alves@zerum.com>
---
 src/include/ndpi_api.h |  2 ++
 src/lib/ndpi_geoip.c   | 43 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 45 insertions(+)

(limited to 'src')

diff --git a/src/include/ndpi_api.h b/src/include/ndpi_api.h
index 3772be89e..0a85e812a 100644
--- a/src/include/ndpi_api.h
+++ b/src/include/ndpi_api.h
@@ -2087,6 +2087,8 @@ extern "C" {
   void ndpi_free_geoip(struct ndpi_detection_module_struct *ndpi_str);
   int ndpi_get_geoip_asn(struct ndpi_detection_module_struct *ndpi_str,
 			 char *ip, u_int32_t *asn);
+  int ndpi_get_geoip_aso(struct ndpi_detection_module_struct *ndpi_str,
+        char *ip, char *aso, u_int8_t aso_len);
   int ndpi_get_geoip_country_continent(struct ndpi_detection_module_struct *ndpi_str, char *ip,
 				       char *country_code, u_int8_t country_code_len,
 				       char *continent, u_int8_t continent_len);
diff --git a/src/lib/ndpi_geoip.c b/src/lib/ndpi_geoip.c
index 0bdd67f78..dadcf7eb3 100644
--- a/src/lib/ndpi_geoip.c
+++ b/src/lib/ndpi_geoip.c
@@ -123,6 +123,49 @@ int ndpi_get_geoip_asn(struct ndpi_detection_module_struct *ndpi_str, char *ip,
   return (-2);
 }
 
+int ndpi_get_geoip_aso(struct ndpi_detection_module_struct *ndpi_str, char *ip, char *aso, u_int8_t aso_len)
+{
+#ifdef HAVE_MAXMINDDB
+  int gai_error, mmdb_error, status;
+  MMDB_lookup_result_s result;
+  MMDB_entry_data_s entry_data;
+
+  if (ndpi_str->mmdb_as_loaded && aso_len > 0)
+  {
+    result = MMDB_lookup_string((MMDB_s *)ndpi_str->mmdb_as, ip, &gai_error, &mmdb_error);
+
+    if ((gai_error != 0) || (mmdb_error != MMDB_SUCCESS) || (!result.found_entry))
+      aso[0] = '\0';
+    else
+    {
+      /* Get the ASO */
+      if (aso_len > 0)
+      {
+        status = MMDB_get_value(&result.entry, &entry_data, "autonomous_system_organization", NULL);
+        if (status != MMDB_SUCCESS || !entry_data.has_data)
+          aso[0] = '\0';
+        else
+        {
+          int str_len = ndpi_min(entry_data.data_size, aso_len);
+
+          memcpy(aso, entry_data.utf8_string, str_len);
+          aso[str_len] = '\0';
+        }
+      }
+    }
+
+    return (0);
+  }
+#else
+  (void)ndpi_str;
+  (void)ip;
+  (void)aso;
+  (void)aso_len;
+#endif
+
+  return (-2);
+}
+
 /* ********************************************************************************* */
 
 int ndpi_get_geoip_country_continent(struct ndpi_detection_module_struct *ndpi_str, char *ip,
-- 
cgit v1.2.3