added nethogsmonitor_loop_devices and moved code to python dir

2022-03-16 18:58:28 +00:00
parent 257b576221
commit fcacd7efdb
4 changed files with 34 additions and 4 deletions
--- a/python/bindings.cpp
+++ b/python/bindings.cpp
@@ -0,0 +1,98 @@
+#include <pybind11/functional.h>
+#include <pybind11/pybind11.h>
+#include <pybind11/stl.h>
+#include <set>
+#include <iostream>
+
+#include "libnethogs.h"
+
+namespace py = pybind11;
+
+//--- for some reason this is a global defined in main.cpp
+std::set<pid_t> pidsToWatch;
+
+//--- hacky way to get callbacks working and handle signals
+std::function<void(int, NethogsMonitorRecord const *)> empty_callback;
+std::function<void(int, NethogsMonitorRecord const *)> loop_callback;
+void loop_callback_wrapper(
+  int action,
+  NethogsMonitorRecord const *record)
+{
+  py::gil_scoped_acquire acquire;
+  if (PyErr_CheckSignals() != 0) {
+    nethogsmonitor_breakloop();
+    PyErr_Clear();
+  }
+  else if (loop_callback) {
+    loop_callback(action, record);
+  }
+}
+
+int nethogsmonitor_loop_py(
+  std::function<void(int, NethogsMonitorRecord const *)> &cb,
+  char *filter,
+  int to_ms)
+{
+    int retval;
+    loop_callback = cb;
+    {
+      py::gil_scoped_release release;
+      retval = nethogsmonitor_loop(loop_callback_wrapper, filter, to_ms);
+    }
+    loop_callback = empty_callback;
+    return retval;
+}
+
+int nethogsmonitor_loop_devices_py(
+  std::function<void(int, NethogsMonitorRecord const *)> &cb,
+  char *filter,
+  std::vector<std::string> __devicenames,
+  bool all,
+  int to_ms)
+{
+    // this is ok because we only use the vector here
+    std::vector<char*> _devicenames;
+    for (auto _dn : __devicenames) _devicenames.push_back(const_cast<char*>(_dn.c_str()));
+    int devc = _devicenames.size();
+    char **devicenames = (_devicenames.empty()) ? NULL : _devicenames.data();
+
+    int retval;
+    loop_callback = cb;
+    {
+      py::gil_scoped_release release;
+      retval = nethogsmonitor_loop_devices(loop_callback_wrapper, filter, devc, devicenames, all, to_ms);
+    }
+    loop_callback = empty_callback;
+    return retval;
+}
+
+//--- python module binding
+PYBIND11_MODULE(nethogs, m) {
+    py::class_<NethogsMonitorRecord>(m, "NethogsMonitorRecord")
+        .def_readwrite("record_id", &NethogsMonitorRecord::record_id)
+        .def_readwrite("name", &NethogsMonitorRecord::name)
+        .def_readwrite("pid", &NethogsMonitorRecord::pid)
+        .def_readwrite("uid", &NethogsMonitorRecord::uid)
+        .def_readwrite("device_name", &NethogsMonitorRecord::device_name)
+        .def_readwrite("sent_bytes", &NethogsMonitorRecord::sent_bytes)
+        .def_readwrite("recv_bytes", &NethogsMonitorRecord::recv_bytes)
+        .def_readwrite("sent_kbs", &NethogsMonitorRecord::sent_kbs)
+        .def_readwrite("recv_kbs", &NethogsMonitorRecord::recv_kbs);
+
+    m.def("nethogsmonitor_loop", &nethogsmonitor_loop_py, R"pbdoc(
+        Nethogs monitor loop
+    )pbdoc");
+    m.def("nethogsmonitor_loop_devices", &nethogsmonitor_loop_devices_py, R"pbdoc(
+        Nethogs monitor loop
+    )pbdoc");
+    m.def("nethogsmonitor_breakloop", &nethogsmonitor_breakloop, R"pbdoc(
+        Nethogs monitor loop break
+    )pbdoc");
+
+#ifdef VERSION
+    m.attr("__version__") = VERSION;
+#else
+    m.attr("__version__") = "unknown";
+#endif
+
+}