FOSS/nethogs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

* src/main.cpp: Include header files for capabilities and attrs.

Browse Source 
* src/nethogs.cpp: Add code to read and parse file attrs, and check for the required capabilities in the non-root case.
This commit is contained in:
rain1
2016-04-14 11:49:55 +00:00
parent ef74014094
commit 43c98c68be
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.cpp
View File

@@ -173,8 +173,20 @@ int main(int argc, char **argv) {
init_ui();
}
if (NEEDROOT && (geteuid() != 0))
forceExit(false, "You need to be root to run NetHogs!");
if (NEEDROOT) {
if (geteuid() != 0)
forceExit(false, "You need to be root to run NetHogs!");
}
else {
unsigned int caps[5] = {0};
getxattr(argv[0], "security.capability", (char *)caps, sizeof(caps));
if ((val[1] >> CAP_NET_ADMIN) & 1 != 1)
forceExit(false, "You need to enable cap_net_admin (and cap_net_raw) to run NetHogs!");
if ((val[1] >> CAP_NET_RAW) & 1 != 1)
forceExit(false, "You need to enable cap_net_raw to run NetHogs!");
}
// use the Self-Pipe trick to interrupt the select() in the main loop
self_pipe = create_self_pipe();