8958ef0b3e
Without the ref only the last element of __devicenames is stored in _devicenames: The temporary is deleted at the end of the for body, therefore the pointer to the c_str is no longer valid and might be reused by the implementation. By using a reference in the for head the c_str does not return a ref to the temporary but to the original array wich's lifetime is till the end of the function.
129 lines
3.9 KiB
C++
129 lines
3.9 KiB
C++
#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
|
|
extern 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;
|
|
}
|
|
|
|
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")
|
|
.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_bytes_last", &NethogsMonitorRecord::sent_bytes_last)
|
|
.def_readwrite("recv_bytes_last", &NethogsMonitorRecord::recv_bytes_last)
|
|
.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");
|
|
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");
|
|
m.def("nethogs_packet_stats", &nethogs_packet_stats_py, R"pbdoc(
|
|
Nethogs pcap packet stats
|
|
)pbdoc");
|
|
|
|
|
|
#ifdef VERSION
|
|
m.attr("__version__") = VERSION;
|
|
#else
|
|
m.attr("__version__") = "unknown";
|
|
#endif
|
|
|
|
}
|