diff --git a/libnethogs.cpp b/libnethogs.cpp index 3f3ae6c..a44d941 100644 --- a/libnethogs.cpp +++ b/libnethogs.cpp @@ -160,7 +160,7 @@ void NethogsMonitor::handleUpdate() NethogsAppUpdate &data = it->second; data.action = NethogsAppUpdate::Remove; monitor_udpate_callback(data); - monitor_update_data.erase(it); + monitor_update_data.erase(curproc->getVal()->pid); } } diff --git a/libnethogs.h b/libnethogs.h index 1fe03b1..fbf1f4f 100644 --- a/libnethogs.h +++ b/libnethogs.h @@ -2,6 +2,7 @@ #define LIBNETHOGS_H_ #include +#include #include #define NETHOGS_DSO_VISIBLE __attribute__ ((visibility ("default")))