reorganize project

This commit is contained in:
fancy
2020-05-17 21:08:44 +08:00
committed by fancy
parent c5ec1027ad
commit 4525d83a53
19 changed files with 312 additions and 105 deletions

30
tools/cgattach.cpp Normal file
View File

@@ -0,0 +1,30 @@
#include "cgroup_attach.hpp"
#include "common.hpp"
#include <cstdlib>
using namespace std;
void print_usage() { fprintf(stdout, "usage: cgattach <pid> <cgroup>\n"); }
int main(int argc, char *argv[]) {
int flag = setuid(0);
if (flag != 0) {
perror("cgattach need root");
exit(EXIT_FAILURE);
}
if (argc != 3) {
error("need exact 2 paramaters");
print_usage();
exit(EXIT_FAILURE);
}
string pid = string(argv[1]);
string cgroup_target = string(argv[2]);
if (validPid(pid)&&validCgroup(cgroup_target)){
CGPROXY::CGROUP::attach(pid, cgroup_target);
}else{
error("param not valid");
exit(EXIT_FAILURE);
}
}