mirror of
https://github.com/maxDcb/C2Core
synced 2026-06-08 15:48:01 +00:00
maxdcb
47 lines
944 B
C++
47 lines
944 B
C++
#pragma once
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
#include "ModuleCmd.hpp"
|
|
|
|
class Shell : public ModuleCmd
|
|
{
|
|
public:
|
|
Shell();
|
|
~Shell();
|
|
|
|
std::string getInfo();
|
|
|
|
int init(std::vector<std::string>& splitedCmd, C2Message& c2Message);
|
|
int process(C2Message& c2Message, C2Message& c2RetMessage);
|
|
int followUp(const C2Message &c2RetMessage);
|
|
int errorCodeToMsg(const C2Message &c2RetMessage, std::string& errorMsg);
|
|
int osCompatibility()
|
|
{
|
|
return OS_LINUX | OS_WINDOWS;
|
|
}
|
|
|
|
private:
|
|
int startShell();
|
|
void stopShell();
|
|
|
|
#ifdef _WIN32
|
|
HANDLE m_hChildStdoutRd;
|
|
HANDLE m_hChildStdinWr;
|
|
PROCESS_INFORMATION m_pi;
|
|
#else
|
|
int m_masterFd;
|
|
pid_t m_pid;
|
|
#endif
|
|
std::string m_program;
|
|
bool m_started;
|
|
};
|
|
|
|
#ifdef _WIN32
|
|
extern "C" __declspec(dllexport) Shell * ShellConstructor();
|
|
#else
|
|
extern "C" __attribute__((visibility("default"))) Shell * ShellConstructor();
|
|
#endif
|