mirror of
https://github.com/maxDcb/C2Core
synced 2026-06-08 15:48:01 +00:00
maxdcb
65 lines
1.3 KiB
C++
65 lines
1.3 KiB
C++
#include "../KillProcess.hpp"
|
|
#include "../../ModuleCmd/Tools.hpp"
|
|
#include <chrono>
|
|
#include <thread>
|
|
#ifdef __linux__
|
|
#include <signal.h>
|
|
#include <sys/wait.h>
|
|
#elif _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
bool testKillProcess();
|
|
|
|
int main()
|
|
{
|
|
bool res;
|
|
std::cout << "[+] testKillProcess" << std::endl;
|
|
res = testKillProcess();
|
|
if (res)
|
|
std::cout << "[+] Sucess" << std::endl;
|
|
else
|
|
std::cout << "[-] Failed" << std::endl;
|
|
|
|
return !res;
|
|
}
|
|
|
|
bool testKillProcess()
|
|
{
|
|
bool ok = true;
|
|
|
|
int pid;
|
|
#ifdef __linux__
|
|
pid = launchProcess("sleep 1");
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(100));
|
|
#elif _WIN32
|
|
pid = launchProcess("C:\\Windows\\System32\\notepad.exe");
|
|
Sleep(1000);
|
|
#endif
|
|
|
|
KillProcess kp;
|
|
std::vector<std::string> cmd = {"killProcess", std::to_string(pid)};
|
|
C2Message msg, ret;
|
|
kp.init(cmd, msg);
|
|
msg.set_pid(pid);
|
|
kp.process(msg, ret);
|
|
|
|
#ifdef __linux__
|
|
ok &= ret.errorCode() == -1;
|
|
#elif _WIN32
|
|
HANDLE h = OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pid);
|
|
if (h)
|
|
{
|
|
DWORD code = 0;
|
|
bool alive = GetExitCodeProcess(h, &code) && code == STILL_ACTIVE;
|
|
CloseHandle(h);
|
|
ok &= !alive;
|
|
}
|
|
else
|
|
{
|
|
ok &= true;
|
|
}
|
|
#endif
|
|
return ok;
|
|
}
|