Merge pull request #269 from AndreasGocht/add_udp

allow the library to enable UDP recording
raboof · Jan 8, 2024 · e6afe4e · e6afe4e
2 parents 89c1951 + 52dbb17
commit e6afe4e
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 0 deletions.
diff --git a/python/bindings.cpp b/python/bindings.cpp
@@ -84,6 +84,12 @@ std::vector<NethogsPackageStats> nethogs_packet_stats_py()
   return stats_vector;
 }
 
+void nethogs_enable_udp_py(bool state)
+{
+  nethogs_enable_udp(state);
+}
+
+
 //--- python module binding
 PYBIND11_MODULE(nethogs, m) {
     py::class_<NethogsMonitorRecord>(m, "NethogsMonitorRecord")
@@ -117,6 +123,9 @@ PYBIND11_MODULE(nethogs, m) {
     m.def("nethogs_packet_stats", &nethogs_packet_stats_py, R"pbdoc(
         Nethogs pcap packet stats
     )pbdoc");
+    m.def("nethogs_enable_udp", &nethogs_enable_udp_py, R"pbdoc(
+        Enables or disables the recording of UDP, default is False.
+    )pbdoc");
 
 
 #ifdef VERSION

diff --git a/src/libnethogs.cpp b/src/libnethogs.cpp
@@ -355,4 +355,9 @@ void nethogs_packet_stats(NethogsPackageStats **stats, int *stats_size)
     i++;
   }
   *stats_size = handles.size();
+}
+
+void nethogs_enable_udp(bool state)
+{
+  catchall = state;
 }
diff --git a/src/libnethogs.h b/src/libnethogs.h
@@ -112,6 +112,13 @@ NETHOGS_DSO_VISIBLE void nethogsmonitor_breakloop();
  */
 NETHOGS_DSO_VISIBLE void nethogs_packet_stats(NethogsPackageStats **stats, int *stats_size);
 
+/**
+ * @brief Enables or disables the UDP recording. Default is False.
+ * 
+ * @param state state to set
+*/
+NETHOGS_DSO_VISIBLE void nethogs_enable_udp(bool state);
+
 #undef NETHOGS_DSO_VISIBLE
 #undef NETHOGS_DSO_HIDDEN