130 lines
3.8 KiB
C++
130 lines
3.8 KiB
C++
#include "mainwindow.h"
|
|
#include "ui_mainwindow.h"
|
|
#include "QCoreApplication"
|
|
#include "QTextBrowser"
|
|
#include "QMessageBox"
|
|
#include "QComboBox"
|
|
#include "QLineEdit"
|
|
#include "QInputDialog"
|
|
#include <QCursor>
|
|
#include <Qt>
|
|
#include <Windows.h>
|
|
#include <qdebug.h>
|
|
#include <Psapi.h>
|
|
#include "../texthook/host.h"
|
|
|
|
QMainWindow* mainWindow;
|
|
QComboBox* processCombo;
|
|
QComboBox* ttCombo;
|
|
QTextBrowser* textOutput;
|
|
|
|
QString GetModuleName(DWORD processId, HMODULE module = NULL)
|
|
{
|
|
HANDLE handle = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, processId);
|
|
wchar_t buffer[MAX_PATH];
|
|
GetModuleFileNameExW(handle, module, buffer, MAX_PATH);
|
|
CloseHandle(handle);
|
|
return QString::fromWCharArray(wcsrchr(buffer, L'\\') + 1);
|
|
}
|
|
|
|
QString ProcessString(DWORD processId)
|
|
{
|
|
return QString("%1: %2").arg(QString::number(processId), GetModuleName(processId));
|
|
}
|
|
|
|
QString TextThreadString(TextThread* thread)
|
|
{
|
|
ThreadParameter tp = thread->GetThreadParameter();
|
|
return QString("%1:%2:%3:%4:%5: ").arg(
|
|
QString::number(thread->Number()),
|
|
QString::number(tp.pid),
|
|
QString::number(tp.hook, 16),
|
|
QString::number(tp.retn, 16),
|
|
QString::number(tp.spl, 16)
|
|
).toUpper() + QString::fromWCharArray(Host::GetHookName(tp.pid, tp.hook).c_str());
|
|
}
|
|
|
|
MainWindow::MainWindow(QWidget *parent) :
|
|
QMainWindow(parent),
|
|
ui(new Ui::MainWindow),
|
|
hostSignaller(new HostSignaller)
|
|
{
|
|
ui->setupUi(this);
|
|
mainWindow = this;
|
|
processCombo = mainWindow->findChild<QComboBox*>("processCombo");
|
|
ttCombo = mainWindow->findChild<QComboBox*>("ttCombo");
|
|
textOutput = mainWindow->findChild<QTextBrowser*>("textOutput");
|
|
|
|
Host::Start();
|
|
hostSignaller->Initialize();
|
|
connect(hostSignaller, &HostSignaller::AddProcess, this, &MainWindow::AddProcess);
|
|
connect(hostSignaller, &HostSignaller::RemoveProcess, this, &MainWindow::RemoveProcess);
|
|
connect(hostSignaller, &HostSignaller::AddThread, this, &MainWindow::AddThread);
|
|
connect(hostSignaller, &HostSignaller::RemoveThread, this, &MainWindow::RemoveThread);
|
|
connect(hostSignaller, &HostSignaller::ThreadOutput, this, &MainWindow::ThreadOutput);
|
|
Host::Open();
|
|
}
|
|
|
|
MainWindow::~MainWindow()
|
|
{
|
|
Host::Close();
|
|
delete hostSignaller;
|
|
delete ui;
|
|
}
|
|
|
|
void MainWindow::AddProcess(unsigned int processId)
|
|
{
|
|
processCombo->addItem(ProcessString(processId), Qt::AlignHCenter);
|
|
}
|
|
|
|
void MainWindow::RemoveProcess(unsigned int processId)
|
|
{
|
|
for (int i = 0; i < processCombo->count(); ++i)
|
|
if (processCombo->itemText(i).split(":")[0] == QString::number(processId))
|
|
processCombo->removeItem(i);
|
|
}
|
|
|
|
void MainWindow::AddThread(TextThread* thread)
|
|
{
|
|
ttCombo->addItem(TextThreadString(thread));
|
|
}
|
|
|
|
void MainWindow::RemoveThread(TextThread* thread)
|
|
{
|
|
for (int i = 0; i < ttCombo->count(); ++i)
|
|
if (ttCombo->itemText(i).split(":")[0] == QString::number(thread->Number()))
|
|
{
|
|
ttCombo->removeItem(i);
|
|
if (i == ttCombo->currentIndex())
|
|
{
|
|
ttCombo->setCurrentIndex(0);
|
|
on_ttCombo_activated(0);
|
|
}
|
|
}
|
|
}
|
|
|
|
void MainWindow::ThreadOutput(TextThread* thread, QString output)
|
|
{
|
|
if (TextThreadString(thread) == ttCombo->currentText())
|
|
{
|
|
textOutput->moveCursor(QTextCursor::End);
|
|
textOutput->insertPlainText(output);
|
|
textOutput->moveCursor(QTextCursor::End);
|
|
}
|
|
}
|
|
|
|
void MainWindow::on_attachButton_clicked()
|
|
{
|
|
Host::InjectProcess(QInputDialog::getInt(this, "Process ID?", "You can find this under Task Manager -> Details"));
|
|
}
|
|
|
|
void MainWindow::on_detachButton_clicked()
|
|
{
|
|
Host::DetachProcess(processCombo->currentText().split(":")[0].toInt());
|
|
}
|
|
|
|
void MainWindow::on_ttCombo_activated(int index)
|
|
{
|
|
textOutput->setText(QString::fromWCharArray(Host::GetThread(ttCombo->itemText(index).split(":")[0].toInt())->GetStore().c_str()));
|
|
}
|