add pcap packet stats

python/bindings.cpp

@@ -66,6 +66,21 @@ int nethogsmonitor_loop_devices_py(
     return retval;
 }
 
+std::vector<NethogsPackageStats> nethogs_packet_stats_py()
+{
+  NethogsPackageStats* stats; 
+  int stat_count;
+
+  nethogs_packet_stats(&stats, &stat_count);
+
+  std::vector<NethogsPackageStats> stats_vector(stat_count);
+  std::copy_n(stats,stat_count, stats_vector.begin());
+
+  free(stats);
+
+  return stats_vector;
+}
+
 //--- python module binding
 PYBIND11_MODULE(nethogs, m) {
     py::class_<NethogsMonitorRecord>(m, "NethogsMonitorRecord")
@@ -81,6 +96,12 @@ PYBIND11_MODULE(nethogs, m) {
         .def_readwrite("sent_kbs", &NethogsMonitorRecord::sent_kbs)
         .def_readwrite("recv_kbs", &NethogsMonitorRecord::recv_kbs);
 
+    py::class_<NethogsPackageStats>(m, "NethogsPackageStats")
+        .def_readonly("ps_recv", &NethogsPackageStats::ps_recv)
+        .def_readonly("ps_drop", &NethogsPackageStats::ps_drop)
+        .def_readonly("ps_ifdrop", &NethogsPackageStats::ps_ifdrop)
+        .def_readonly("devicename", &NethogsPackageStats::devicename);
+
     m.def("nethogsmonitor_loop", &nethogsmonitor_loop_py, R"pbdoc(
         Nethogs monitor loop
     )pbdoc");
@@ -90,6 +111,10 @@ PYBIND11_MODULE(nethogs, m) {
     m.def("nethogsmonitor_breakloop", &nethogsmonitor_breakloop, R"pbdoc(
         Nethogs monitor loop break
     )pbdoc");
+    m.def("nethogs_packet_stats", &nethogs_packet_stats_py, R"pbdoc(
+        Nethogs pcap packet stats
+    )pbdoc");
+
 
 #ifdef VERSION
     m.attr("__version__") = VERSION;

src/decpcap.c

@@ -110,6 +110,23 @@ struct dp_handle *dp_open_live(const char *device, int snaplen, int promisc,
   return dp_fillhandle(temp);
 }
 
+/* function to get packet statistics, e.g. dropped packets */
+
+dp_stat dp_stats(struct dp_handle* handle)
+{
+  struct pcap_stat ps;
+  if(pcap_stats(handle->pcap_handle, &ps) == PCAP_ERROR)
+  {
+      fprintf(stderr, "Error getting pcap_stats: %s\n", 
+              pcap_geterr(handle->pcap_handle));
+      ps.ps_recv   = 0;
+      ps.ps_drop   = 0;
+      ps.ps_ifdrop = 0;
+      return ps;
+  }
+  return ps;
+}
+
 /* functions to add callbacks */
 
 void dp_addcb(struct dp_handle *handle, enum dp_packet_type type,

src/decpcap.h

@@ -52,6 +52,7 @@ enum dp_packet_type {
  * pcap
 };*/
 typedef struct pcap_pkthdr dp_header;
+typedef struct pcap_stat dp_stat;
 
 typedef int (*dp_callback)(u_char *, const dp_header *, const u_char *);
 
@@ -69,6 +70,10 @@ struct dp_handle *dp_open_live(const char *device, int snaplen, int promisc,
                                int to_ms, char *filter, char *errbuf);
 struct dp_handle *dp_open_offline(char *fname, char *ebuf);
 
+/* function to get packet statistics, e.g. dropped packets */
+
+dp_stat dp_stats(struct dp_handle* handle);
+
 /* functions to add callbacks */
 
 void dp_addcb(struct dp_handle *handle, enum dp_packet_type type,

src/libnethogs.cpp

@@ -339,3 +339,20 @@ void nethogsmonitor_breakloop() {
   monitor_run_flag = false;
   write(self_pipe.second, "x", 1);
 }
+
+void nethogs_packet_stats(NethogsPackageStats **stats, int *stats_size)
+{
+  
+  *stats = static_cast<NethogsPackageStats *>(malloc(handles.size() * sizeof(NethogsPackageStats)));
+  int i = 0;
+
+  for(auto current_handle = handles.begin(); current_handle != handles.end(); current_handle ++){
+    dp_stat stat = dp_stats(current_handle->content);
+    stats[i]->ps_recv = stat.ps_recv;
+    stats[i]->ps_drop = stat.ps_drop;
+    stats[i]->ps_ifdrop = stat.ps_ifdrop;
+    stats[i]->devicename = current_handle->devicename;
+    i++;
+  }
+  *stats_size = handles.size();
+}

src/libnethogs.h

@@ -7,6 +7,7 @@ extern "C" {
 
 #include <stdbool.h>
 #include <stdint.h>
+#include <sys/types.h>
 
 #define NETHOGS_DSO_VISIBLE __attribute__((visibility("default")))
 #define NETHOGS_DSO_HIDDEN __attribute__((visibility("hidden")))
@@ -32,6 +33,14 @@ typedef struct NethogsMonitorRecord {
   float recv_kbs;
 } NethogsMonitorRecord;
 
+typedef struct NethogsPackageStats
+{
+  u_int ps_recv;           /** number of packets received */
+  u_int ps_drop;           /** number of packets dropped because there was no room in the operating system's buffer when they arrived, because packets weren't being read fast enough */
+  u_int ps_ifdrop;         /** number of packets dropped by the network interface or its driver.  */
+  const char *devicename; /** name of the network interface */
+} NethogsPackageStats;
+
 /**
  * @brief Defines a callback to handle updates about applications
  * @param action NETHOGS_APP_ACTION_SET if data is being added or updated,
@@ -95,6 +104,14 @@ NETHOGS_DSO_VISIBLE int nethogsmonitor_loop_devices(NethogsMonitorCallback cb,
  */
 NETHOGS_DSO_VISIBLE void nethogsmonitor_breakloop();
 
+/**
+ * @brief returns the pcap packet stats per device
+ * 
+ * @param stats C-Style array the will hold the stats
+ * @param stats_size elements and therefore devices in stats
+ */
+NETHOGS_DSO_VISIBLE void nethogs_packet_stats(NethogsPackageStats **stats, int *stats_size);
+
 #undef NETHOGS_DSO_VISIBLE
 #undef NETHOGS_DSO_HIDDEN