Fix modules for windows compatibility

Author: maxDcb

modules/EnumerateShares/tests/testsEnumerateShares.cpp

@@ -21,5 +21,8 @@ bool testEnumerateShares()
     C2Message msg, ret;
     mod->init(cmd, msg);
     mod->process(msg, ret);
+
+    std::cout <<  "[+] enumerateShares output: " << ret.returnvalue() << std::endl;
+
     return !ret.returnvalue().empty();
 }

modules/GetEnv/tests/testsGetEnv.cpp

@@ -22,5 +22,8 @@ bool testGetEnv()
     C2Message msg, ret;
     mod->init(cmd, msg);
     mod->process(msg, ret);
+
+    std::cout << "[+] getEnv output: " << ret.returnvalue() << std::endl;
+
     return !ret.returnvalue().empty();
 }

modules/IpConfig/IpConfig.cpp

@@ -69,6 +69,22 @@ int IpConfig::process(C2Message& c2Message, C2Message& c2RetMessage)
     return 0;
 }
 
+
+#ifdef _WIN32
+std::string WideToUTF8(const PWCHAR wideStr) {
+    if (!wideStr) return "";
+
+    int len = WideCharToMultiByte(CP_UTF8, 0, wideStr, -1, nullptr, 0, nullptr, nullptr);
+    if (len == 0) return "";
+
+    std::string result(len - 1, 0); // exclude null terminator
+    WideCharToMultiByte(CP_UTF8, 0, wideStr, -1, &result[0], len, nullptr, nullptr);
+
+    return result;
+}
+#endif
+
+
 std::string IpConfig::runIpconfig()
 {
 #ifdef _WIN32
@@ -81,7 +97,7 @@ std::string IpConfig::runIpconfig()
     {
         for(auto p = addr; p; p = p->Next)
         {
-            result += p->FriendlyName; result += "\n";
+            result += WideToUTF8(p->FriendlyName) + "\n";
             for(PIP_ADAPTER_UNICAST_ADDRESS unicast = p->FirstUnicastAddress; unicast; unicast = unicast->Next)
             {
                 char ip[INET6_ADDRSTRLEN];

modules/IpConfig/tests/testsIpConfig.cpp

@@ -21,5 +21,8 @@ bool testIpConfig()
     C2Message msg, ret;
     mod->init(cmd, msg);
     mod->process(msg, ret);
+
+    std::cout << "[+] ipConfig output: " << ret.returnvalue() << std::endl;
+
     return !ret.returnvalue().empty();
 }

modules/Netstat/tests/testsNetstat.cpp

@@ -21,5 +21,8 @@ bool testNetstat()
     C2Message msg, ret;
     mod->init(cmd, msg);
     mod->process(msg, ret);
+
+    std::cout << "[+] netstat output: " << ret.returnvalue() << std::endl;
+
     return !ret.returnvalue().empty();
 }

modules/Shell/Shell.cpp

@@ -2,10 +2,12 @@
 #include "Common.hpp"
 
 #include <cstring>
+#ifdef __linux__
 #include <pty.h>
 #include <unistd.h>
 #include <sys/select.h>
 #include <sys/wait.h>
+#endif
 #include <chrono>
 
 using namespace std;

modules/Shell/Shell.hpp

@@ -24,7 +24,12 @@ class Shell : public ModuleCmd
     void stopShell();
 
     int m_masterFd;
+
+#ifdef _WIN32
+    int m_pid;
+#else
     pid_t m_pid;
+#endif
     std::string m_program;
     bool m_started;
 };

modules/Whoami/Whoami.cpp

@@ -3,12 +3,11 @@
 
 #include <cstring>
 #include <sstream>
-#ifdef _WIN32
-#include <sddl.h>
-#endif
 
 #ifdef _WIN32
 #include <windows.h>
+#include <sddl.h>
+#pragma comment(lib, "Advapi32.lib")
 #else
 #include <pwd.h>
 #include <unistd.h>

modules/Whoami/tests/testsWhoami.cpp

@@ -21,5 +21,8 @@ bool testWhoami()
     C2Message msg, ret;
     mod->init(cmd, msg);
     mod->process(msg, ret);
+
+    std::cout << "[+] whoami output: " << ret.returnvalue() << std::endl;
+
     return !ret.returnvalue().empty();
 }