Textractor/GUI/mainwindow.cpp

203 lines
6.5 KiB
C++
Raw Normal View History

2018-07-21 15:40:16 -07:00
#include "mainwindow.h"
#include "ui_mainwindow.h"
2018-07-25 21:48:18 -07:00
#include <QCoreApplication>
2018-07-23 22:57:54 -07:00
#include "QTextBrowser"
2018-07-22 16:53:51 -07:00
#include "QMessageBox"
2018-07-23 22:57:54 -07:00
#include "QComboBox"
2018-07-23 12:25:02 -07:00
#include "QLineEdit"
#include "QInputDialog"
#include <QCursor>
#include <Qt>
2018-07-26 22:42:21 -07:00
#include <QPlainTextEdit>
#include <QDateTime>
#include <QFileDialog>
2018-07-25 21:48:18 -07:00
#include <unordered_set>
2018-07-26 22:42:21 -07:00
#include <map>
#include <unordered_map>
2018-07-22 16:53:51 -07:00
#include <Windows.h>
2018-07-23 12:25:02 -07:00
#include <qdebug.h>
#include <Psapi.h>
2018-07-26 22:42:21 -07:00
#include "extensions.h"
#include "../vnrhook/include/const.h"
2018-07-25 21:48:18 -07:00
#include "misc.h"
2018-07-21 15:40:16 -07:00
2018-07-23 22:57:54 -07:00
QMainWindow* mainWindow;
QComboBox* processCombo;
QComboBox* ttCombo;
2018-07-26 22:42:21 -07:00
QComboBox* extenCombo;
QPlainTextEdit* textOutput;
2018-07-23 12:25:02 -07:00
2018-07-23 22:57:54 -07:00
QString ProcessString(DWORD processId)
2018-07-23 12:25:02 -07:00
{
2018-07-25 10:46:59 -07:00
return QString("%1: %2").arg(QString::number(processId), GetModuleName(processId));
2018-07-23 22:57:54 -07:00
}
QString TextThreadString(TextThread* thread)
{
2018-07-25 10:46:59 -07:00
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();
2018-07-23 12:25:02 -07:00
}
2018-07-21 15:40:16 -07:00
MainWindow::MainWindow(QWidget *parent) :
2018-07-25 10:46:59 -07:00
QMainWindow(parent),
ui(new Ui::MainWindow),
hostSignaller(new HostSignaller)
2018-07-21 15:40:16 -07:00
{
2018-07-25 10:46:59 -07:00
ui->setupUi(this);
mainWindow = this;
processCombo = mainWindow->findChild<QComboBox*>("processCombo");
ttCombo = mainWindow->findChild<QComboBox*>("ttCombo");
2018-07-26 22:42:21 -07:00
extenCombo = mainWindow->findChild<QComboBox*>("extenCombo");
textOutput = mainWindow->findChild<QPlainTextEdit*>("textOutput");
2018-07-23 12:25:02 -07:00
2018-07-25 10:46:59 -07:00
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);
2018-07-26 22:42:21 -07:00
std::map<int, std::wstring> extensions = LoadExtensions();
for (auto i : extensions) extenCombo->addItem(QString::number(i.first) + ":" + QString::fromWCharArray(i.second.c_str()));
2018-07-25 10:46:59 -07:00
Host::Open();
2018-07-21 15:40:16 -07:00
}
MainWindow::~MainWindow()
{
2018-07-25 10:46:59 -07:00
delete ui;
2018-07-21 15:40:16 -07:00
}
2018-07-22 16:53:51 -07:00
void MainWindow::AddProcess(unsigned int processId)
2018-07-23 22:57:54 -07:00
{
2018-07-25 10:46:59 -07:00
processCombo->addItem(ProcessString(processId));
2018-07-26 22:42:21 -07:00
QFile file("SavedHooks.txt");
if (!file.open(QIODevice::ReadOnly)) return;
QString processName = GetFullModuleName(processId);
QString allData = file.readAll();
QStringList allProcesses = allData.split("\r", QString::SkipEmptyParts);
for (int i = allProcesses.length() - 1; i >= 0; --i)
if (allProcesses.at(i).contains(processName))
{
Sleep(50);
QStringList hooks = allProcesses.at(i).split(" , ");
for (int j = 1; j < hooks.length(); ++j) Host::InsertHook(processId, ParseHCode(hooks.at(j)));
return;
}
2018-07-23 22:57:54 -07:00
}
void MainWindow::RemoveProcess(unsigned int processId)
2018-07-23 22:57:54 -07:00
{
2018-07-25 10:46:59 -07:00
processCombo->removeItem(processCombo->findText(QString::number(processId) + ":", Qt::MatchStartsWith));
2018-07-23 22:57:54 -07:00
}
void MainWindow::AddThread(TextThread* thread)
2018-07-23 22:57:54 -07:00
{
2018-07-25 10:46:59 -07:00
ttCombo->addItem(
TextThreadString(thread) +
2018-07-25 21:48:18 -07:00
QString::fromWCharArray(Host::GetHookName(thread->GetThreadParameter().pid, thread->GetThreadParameter().hook).c_str()) +
" (" +
GenerateHCode(Host::GetHookParam(thread->GetThreadParameter().pid, thread->GetThreadParameter().hook), thread->GetThreadParameter().pid) +
")"
2018-07-25 10:46:59 -07:00
);
2018-07-23 22:57:54 -07:00
}
void MainWindow::RemoveThread(TextThread* thread)
2018-07-23 22:57:54 -07:00
{
2018-07-25 10:46:59 -07:00
int threadIndex = ttCombo->findText(QString::number(thread->Number()) + ":", Qt::MatchStartsWith);
ttCombo->removeItem(threadIndex);
if (threadIndex == ttCombo->currentIndex())
{
ttCombo->setCurrentIndex(0);
on_ttCombo_activated(0);
}
}
void MainWindow::ThreadOutput(TextThread* thread, QString output)
{
2018-07-25 10:46:59 -07:00
if (ttCombo->currentText().startsWith(TextThreadString(thread)))
{
textOutput->moveCursor(QTextCursor::End);
textOutput->insertPlainText(output);
textOutput->moveCursor(QTextCursor::End);
}
2018-07-23 22:57:54 -07:00
}
2018-07-25 21:48:18 -07:00
QVector<HookParam> MainWindow::GetAllHooks(DWORD processId)
{
std::unordered_set<DWORD> addresses;
QVector<HookParam> hooks;
for (int i = 0; i < ttCombo->count(); ++i)
if (ttCombo->itemText(i).split(":")[1].toInt() == processId &&
!addresses.count(ttCombo->itemText(i).split(":")[2].toInt(nullptr, 16)))
{
addresses.insert(ttCombo->itemText(i).split(":")[2].toInt(nullptr, 16));
hooks.push_back(Host::GetHookParam(ttCombo->itemText(i).split(":")[1].toInt(), ttCombo->itemText(i).split(":")[2].toInt(nullptr, 16)));
}
return hooks;
}
2018-07-23 12:25:02 -07:00
void MainWindow::on_attachButton_clicked()
2018-07-22 16:53:51 -07:00
{
2018-07-25 21:48:18 -07:00
bool ok;
int processId = QInputDialog::getInt(this, "Attach Process", "Process ID?\r\nYou can find this under Task Manager -> Details", 0, 0, 100000, 1, &ok);
if (ok) Host::InjectProcess(processId);
2018-07-23 22:57:54 -07:00
}
void MainWindow::on_detachButton_clicked()
{
2018-07-25 10:46:59 -07:00
Host::DetachProcess(processCombo->currentText().split(":")[0].toInt());
2018-07-23 22:57:54 -07:00
}
2018-07-25 21:48:18 -07:00
void MainWindow::on_hookButton_clicked()
{
bool ok;
QString hookCode = QInputDialog::getText(this, "Add Hook",
"Enter hook code\r\n/H{A|B|W|S|Q}[N][data_offset[*drdo]][:sub_offset[*drso]]@addr[:module]",
QLineEdit::Normal, "/H", &ok
);
2018-07-26 22:42:21 -07:00
if (ok) Host::InsertHook(processCombo->currentText().split(":")[0].toInt(), ParseHCode(hookCode));
2018-07-25 21:48:18 -07:00
}
void MainWindow::on_unhookButton_clicked()
{
QVector<HookParam> hooks = GetAllHooks(processCombo->currentText().split(":")[0].toInt());
QStringList hookList;
for (auto i : hooks) hookList.push_back(
QString::fromWCharArray(Host::GetHookName(processCombo->currentText().split(":")[0].toInt(), i.address).c_str()) +
": " +
GenerateHCode(i, processCombo->currentText().split(":")[0].toInt())
);
bool ok;
QString hook = QInputDialog::getItem(this, "Unhook", "Which hook to remove?", hookList, 0, false, &ok);
if (ok) Host::RemoveHook(processCombo->currentText().split(":")[0].toInt(), hooks.at(hookList.indexOf(hook)).address);
}
2018-07-26 22:42:21 -07:00
void MainWindow::on_saveButton_clicked()
{
QVector<HookParam> hooks = GetAllHooks(processCombo->currentText().split(":")[0].toInt());
QString hookList = GetFullModuleName(processCombo->currentText().split(":")[0].toInt());;
for (auto i : hooks)
if (!(i.type & HOOK_ENGINE))
hookList += " , " + GenerateHCode(i, processCombo->currentText().split(":")[0].toInt());
QFile file("SavedHooks.txt");
if (!file.open(QIODevice::Append | QIODevice::Text)) return;
file.write((hookList + "\r\n").toUtf8());
}
2018-07-23 22:57:54 -07:00
void MainWindow::on_ttCombo_activated(int index)
{
2018-07-26 22:42:21 -07:00
textOutput->setPlainText(QString::fromWCharArray(Host::GetThread(ttCombo->itemText(index).split(":")[0].toInt())->GetStore().c_str()));
2018-07-25 10:46:59 -07:00
textOutput->moveCursor(QTextCursor::End);
2018-07-22 16:53:51 -07:00
}
2018-07-26 22:42:21 -07:00
void MainWindow::on_addExtenButton_clicked()
{
QFileDialog extenSelector;
}