backup/cgproxy
1
0
Fork 0
mirror of https://github.com/springzfx/cgproxy.git synced 2026-01-07 13:07:56 +08:00
Code Issues 1 Projects Releases Wiki Activity

check again after small period(100ms) to avoid kde cgroup override

Browse Source
This commit is contained in:
springzfx
2020-06-12 00:20:57 +08:00
parent 840a8338a8
commit bc94e58cb1
4 changed files with 30 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/cgroup_attach.cpp
View File

@@ -9,6 +9,7 @@
#include <string>
#include <sys/stat.h>
#include <sys/types.h>
#include <thread>
#include <unistd.h>
namespace CGPROXY::CGROUP {
@@ -68,9 +69,19 @@ int attach(const string pid, const string cgroup_target) {
}
// put pid to target cgroup
ofstream procs(cgroup_target_procs, ofstream::app);
if (write2procs(pid, cgroup_target_procs) != 0) return_error;
// wait for small period and check again
this_thread::sleep_for(std::chrono::milliseconds(100));
if (getCgroup(pid) != cgroup_target && write2procs(pid, cgroup_target_procs) != 0)
return_error;
return_success;
}
int write2procs(string pid, string procspath) {
ofstream procs(procspath, ofstream::app);
if (!procs.is_open()) {
error("open file %s failed", cgroup_target_procs.c_str());
error("open file %s failed", procspath.c_str());
return_error;
}
procs << pid.c_str() << endl;
@@ -79,7 +90,7 @@ int attach(const string pid, const string cgroup_target) {
// maybe there some write error, for example process pid may not exist
if (!procs) {
error("write %s to %s failed, maybe process %s not exist", pid.c_str(),
cgroup_target_procs.c_str(), pid.c_str());
procspath.c_str(), pid.c_str());
return_error;
}
return_success;