maxDcb-C2Core/modules/Cat/tests/testsCat.cpp

#include "../Cat.hpp"

#ifdef __linux__
#elif _WIN32
#include <windows.h>
#endif

bool testCat();

int main()
{
    bool res;

    std::cout << "[+] testCat" << std::endl;
    res = testCat();
    if (res)
       std::cout << "[+] Sucess" << std::endl;
    else
       std::cout << "[-] Failed" << std::endl;

    return !res;
}

bool testCat()
{
    std::string testFile = "test1.txt";
    std::string fileContent = "testCat";

    std::ofstream outfile("test1.txt");
    outfile << "testCat" << std::endl;
    outfile.close();

    std::unique_ptr<Cat> cat = std::make_unique<Cat>();
    {
        std::vector<std::string> splitedCmd;
        splitedCmd.push_back("cat");
        splitedCmd.push_back(testFile);

        C2Message c2Message;
        C2Message c2RetMessage;
        cat->init(splitedCmd, c2Message);
        cat->process(c2Message, c2RetMessage);

        std::string output = "\n\noutput:\n";
        output += c2RetMessage.returnvalue();
        output += "\n";
        std::cout << output << std::endl;

        if (c2RetMessage.returnvalue().compare(0, fileContent.length(), fileContent) == 0) 
        {
        } 
        else 
        {
            std::cout << "[-] fileContent " << fileContent << std::endl;
            std::cout << "[-] c2RetMessage.returnvalue() " << c2RetMessage.returnvalue() << std::endl;
            return false;
        }
    }
    
    {
        std::vector<std::string> splitedCmd;
        splitedCmd.push_back("cat");
        splitedCmd.push_back(".\\test space folder\\test space.txt");

        C2Message c2Message;
        C2Message c2RetMessage;
        cat->init(splitedCmd, c2Message);
        cat->process(c2Message, c2RetMessage);

        std::string output = "\n\noutput:\n";
        output += c2RetMessage.returnvalue();
        output += "\n";
        std::cout << output << std::endl;

        if (c2RetMessage.errorCode()) 
        {
            std::cout << "[+] c2RetMessage.errorCode() " << c2RetMessage.errorCode() << std::endl;
        } 
        else 
        {
            return false;
        }
    }

    return true;
}