Merge pull request #4 from maxDcb/codex/add-mkdir-module-with-tests

Add MkDir module

Author: maxDcb

modules/CMakeLists.txt

@@ -22,6 +22,7 @@ add_subdirectory(SpawnAs)
 add_subdirectory(StealToken)
 add_subdirectory(Upload)
 add_subdirectory(Cat)
+add_subdirectory(MkDir)
 add_subdirectory(Tree)
 add_subdirectory(WmiExec)
 add_subdirectory(KeyLogger)

modules/MkDir/CMakeLists.txt

@@ -0,0 +1,14 @@
+include_directories(../)
+add_library(MkDir SHARED MkDir.cpp)
+set_property(TARGET MkDir PROPERTY MSVC_RUNTIME_LIBRARY "MultiThreaded")
+target_link_libraries(MkDir )
+add_custom_command(TARGET MkDir POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy
+    $<TARGET_FILE:MkDir> "${CMAKE_SOURCE_DIR}/Release/Modules/$<TARGET_FILE_NAME:MkDir>")
+
+if(WITH_TESTS)
+    add_executable(testsMkDir tests/testsMkDir.cpp MkDir.cpp)
+    target_link_libraries(testsMkDir )
+    add_custom_command(TARGET testsMkDir POST_BUILD COMMAND ${CMAKE_COMMAND} -E copy
+        $<TARGET_FILE:testsMkDir> "${CMAKE_SOURCE_DIR}/Tests/$<TARGET_FILE_NAME:testsMkDir>")
+    add_test(NAME testsMkDir COMMAND "${CMAKE_SOURCE_DIR}/Tests/$<TARGET_FILE_NAME:testsMkDir>")
+endif()

modules/MkDir/MkDir.cpp

@@ -0,0 +1,107 @@
+#include "MkDir.hpp"
+
+#include <filesystem>
+#include <cstring>
+
+#include "Common.hpp"
+
+using namespace std;
+namespace fs = std::filesystem;
+
+constexpr std::string_view moduleName = "mkDir";
+constexpr unsigned long long moduleHash = djb2(moduleName);
+
+#ifdef _WIN32
+__declspec(dllexport) MkDir* MkDirConstructor()
+{
+    return new MkDir();
+}
+#else
+__attribute__((visibility("default"))) MkDir* MkDirConstructor()
+{
+    return new MkDir();
+}
+#endif
+
+MkDir::MkDir()
+#ifdef BUILD_TEAMSERVER
+    : ModuleCmd(std::string(moduleName), moduleHash)
+#else
+    : ModuleCmd("", moduleHash)
+#endif
+{
+}
+
+MkDir::~MkDir()
+{
+}
+
+std::string MkDir::getInfo()
+{
+    std::string info;
+#ifdef BUILD_TEAMSERVER
+    info += "mkDir Module:\n";
+    info += "Create a new directory on the target system.\n";
+    info += "\nUsage:\n";
+    info += " - mkDir <path>\n";
+#endif
+    return info;
+}
+
+int MkDir::init(std::vector<std::string> &splitedCmd, C2Message &c2Message)
+{
+#if defined(BUILD_TEAMSERVER) || defined(BUILD_TESTS)
+    if (splitedCmd.size() >= 2)
+    {
+        std::string path;
+        for (size_t i = 1; i < splitedCmd.size(); ++i)
+        {
+            if (!path.empty())
+                path += " ";
+            path += splitedCmd[i];
+        }
+        c2Message.set_instruction(splitedCmd[0]);
+        c2Message.set_cmd(path);
+    }
+    else
+    {
+        c2Message.set_returnvalue(getInfo());
+        return -1;
+    }
+#endif
+    return 0;
+}
+
+#define ERROR_CREATE_DIRECTORY 1
+
+int MkDir::process(C2Message &c2Message, C2Message &c2RetMessage)
+{
+    std::string path = c2Message.cmd();
+    c2RetMessage.set_instruction(c2RetMessage.instruction());
+    c2RetMessage.set_cmd(path);
+    std::error_code ec;
+    bool created = fs::create_directories(path, ec);
+    if (!ec)
+    {
+        c2RetMessage.set_returnvalue(created ? "Directory created." : "Already exists.");
+    }
+    else
+    {
+        c2RetMessage.set_errorCode(ERROR_CREATE_DIRECTORY);
+    }
+    return 0;
+}
+
+int MkDir::errorCodeToMsg(const C2Message &c2RetMessage, std::string& errorMsg)
+{
+#ifdef BUILD_TEAMSERVER
+    int errorCode = c2RetMessage.errorCode();
+    if (errorCode > 0)
+    {
+        if (errorCode == ERROR_CREATE_DIRECTORY)
+            errorMsg = "Failed: Could not create directory";
+    }
+#endif
+    return 0;
+}
+

modules/MkDir/MkDir.hpp

@@ -0,0 +1,29 @@
+#pragma once
+
+#include "ModuleCmd.hpp"
+
+class MkDir : public ModuleCmd
+{
+public:
+    MkDir();
+    ~MkDir();
+
+    std::string getInfo();
+
+    int init(std::vector<std::string>& splitedCmd, C2Message& c2Message);
+    int process(C2Message& c2Message, C2Message& c2RetMessage);
+    int errorCodeToMsg(const C2Message &c2RetMessage, std::string& errorMsg);
+    int osCompatibility()
+    {
+        return OS_LINUX | OS_WINDOWS;
+    }
+
+private:
+};
+
+#ifdef _WIN32
+extern "C" __declspec(dllexport) MkDir * MkDirConstructor();
+#else
+extern "C"  __attribute__((visibility("default"))) MkDir * MkDirConstructor();
+#endif
+

modules/MkDir/tests/testsMkDir.cpp

@@ -0,0 +1,49 @@
+#include "../MkDir.hpp"
+#include <filesystem>
+
+bool testMkDir();
+
+int main()
+{
+    bool res;
+    std::cout << "[+] testMkDir" << std::endl;
+    res = testMkDir();
+    if (res)
+        std::cout << "[+] Sucess" << std::endl;
+    else
+        std::cout << "[-] Failed" << std::endl;
+
+    return !res;
+}
+
+bool testMkDir()
+{
+    namespace fs = std::filesystem;
+    fs::path temp = fs::temp_directory_path() / "c2core_mkdir_test";
+    fs::remove_all(temp);
+    bool ok = true;
+
+    {
+        MkDir mk;
+        std::vector<std::string> cmd = {"mkDir", temp.string()};
+        C2Message msg, ret;
+        mk.init(cmd, msg);
+        msg.set_cmd(temp.string());
+        mk.process(msg, ret);
+        ok &= ret.returnvalue().find("Directory created") != std::string::npos;
+        ok &= fs::exists(temp);
+    }
+
+    {
+        MkDir mk;
+        std::vector<std::string> cmd = {"mkDir", temp.string()};
+        C2Message msg, ret;
+        mk.init(cmd, msg);
+        msg.set_cmd(temp.string());
+        mk.process(msg, ret);
+        ok &= ret.returnvalue().find("Already exists") != std::string::npos;
+    }
+
+    fs::remove_all(temp);
+    return ok;
+}