2018-07-21 15:40:16 -07:00
# include "mainwindow.h"
# include "ui_mainwindow.h"
2018-07-26 22:42:21 -07:00
# include "extensions.h"
2018-07-25 21:48:18 -07:00
# include "misc.h"
2018-08-22 15:11:40 -04:00
# include "../vnrhook/include/const.h"
# include <QCoreApplication>
# include <QInputDialog>
# include <QFileDialog>
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 ) ,
2018-08-21 22:43:30 -04:00
ui ( new Ui : : MainWindow )
2018-07-21 15:40:16 -07:00
{
2018-07-25 10:46:59 -07:00
ui - > setupUi ( this ) ;
2018-08-21 22:43:30 -04:00
2018-08-20 11:00:34 -04:00
QFile settings ( " NHWindow " ) ;
settings . open ( QIODevice : : ReadOnly ) ;
QDataStream reader ( & settings ) ;
QRect rect = QRect ( ) ;
reader > > rect ;
if ( rect . bottom ( ) ) this - > setGeometry ( rect ) ;
2018-07-23 12:25:02 -07:00
2018-08-21 22:43:30 -04:00
processCombo = findChild < QComboBox * > ( " processCombo " ) ;
ttCombo = findChild < QComboBox * > ( " ttCombo " ) ;
extenCombo = findChild < QComboBox * > ( " extenCombo " ) ;
textOutput = findChild < QPlainTextEdit * > ( " textOutput " ) ;
connect ( this , & MainWindow : : SigAddProcess , this , & MainWindow : : AddProcess ) ;
connect ( this , & MainWindow : : SigRemoveProcess , this , & MainWindow : : RemoveProcess ) ;
connect ( this , & MainWindow : : SigAddThread , this , & MainWindow : : AddThread ) ;
connect ( this , & MainWindow : : SigRemoveThread , this , & MainWindow : : RemoveThread ) ;
connect ( this , & MainWindow : : SigThreadOutput , this , & MainWindow : : ThreadOutput ) ;
Host : : Start (
[ & ] ( DWORD processId ) { emit SigAddProcess ( processId ) ; } ,
[ & ] ( DWORD processId ) { emit SigRemoveProcess ( processId ) ; } ,
[ & ] ( TextThread * thread ) { emit SigAddThread ( thread ) ; } ,
[ & ] ( TextThread * thread ) { emit SigRemoveThread ( thread ) ; }
) ;
2018-08-17 14:37:37 -04:00
ReloadExtensions ( ) ;
2018-08-18 15:06:14 -04:00
Host : : AddConsoleOutput ( L " NextHooker beta v2.1.3 by Artikash \r \n Source code and more information available under GPLv3 at https://github.com/Artikash/NextHooker " ) ;
2018-07-21 15:40:16 -07:00
}
MainWindow : : ~ MainWindow ( )
{
2018-08-20 11:00:34 -04:00
QFile settings ( " NHWindow " ) ;
settings . open ( QIODevice : : ReadWrite | QIODevice : : Truncate ) ;
QDataStream writer ( & settings ) ;
writer < < this - > geometry ( ) ;
2018-08-22 01:22:34 -04:00
Host : : Close ( ) ;
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
2018-07-24 10:39:02 -07:00
void MainWindow : : AddProcess ( unsigned int processId )
2018-07-23 22:57:54 -07:00
{
2018-08-22 15:11:40 -04:00
processCombo - > addItem ( QString : : number ( processId ) + " : " + GetModuleName ( 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 ) )
{
QStringList hooks = allProcesses . at ( i ) . split ( " , " ) ;
2018-07-30 20:25:08 -07:00
for ( int j = 1 ; j < hooks . length ( ) ; + + j )
2018-08-04 18:01:59 -04:00
Host : : InsertHook ( processId , ParseCode ( hooks . at ( j ) ) ) ;
2018-07-26 22:42:21 -07:00
return ;
}
2018-07-23 22:57:54 -07:00
}
2018-07-24 10:39:02 -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
}
2018-07-24 10:39:02 -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-08-19 17:55:50 -04:00
QString : : fromStdWString ( Host : : GetHookName ( thread - > GetThreadParameter ( ) . pid , thread - > GetThreadParameter ( ) . hook ) ) +
2018-07-25 21:48:18 -07:00
" ( " +
2018-08-21 22:43:30 -04:00
GenerateCode ( Host : : GetHookParam ( thread - > GetThreadParameter ( ) ) , thread - > GetThreadParameter ( ) . pid ) +
2018-07-25 21:48:18 -07:00
" ) "
2018-07-25 10:46:59 -07:00
) ;
2018-08-03 00:07:25 -04:00
thread - > RegisterOutputCallBack ( [ & ] ( TextThread * thread , std : : wstring output )
{
2018-08-04 03:16:14 -04:00
output = DispatchSentenceToExtensions ( output , GetInfoForExtensions ( thread ) ) ;
2018-08-07 12:50:31 -04:00
output + = L " \r \n " ;
2018-08-21 22:43:30 -04:00
emit SigThreadOutput ( thread , QString : : fromStdWString ( output ) ) ;
2018-08-03 00:07:25 -04:00
return output ;
} ) ;
2018-07-23 22:57:54 -07:00
}
2018-07-24 10:39:02 -07:00
void MainWindow : : RemoveThread ( TextThread * thread )
2018-07-23 22:57:54 -07:00
{
2018-08-21 22:43:30 -04:00
int threadIndex = ttCombo - > findText ( TextThreadString ( thread ) , Qt : : MatchStartsWith ) ;
2018-07-25 10:46:59 -07:00
if ( threadIndex = = ttCombo - > currentIndex ( ) )
{
ttCombo - > setCurrentIndex ( 0 ) ;
on_ttCombo_activated ( 0 ) ;
}
2018-08-02 02:17:20 -04:00
ttCombo - > removeItem ( threadIndex ) ;
2018-08-21 22:43:30 -04:00
delete thread ;
2018-07-24 10:39:02 -07:00
}
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-08-22 15:11:40 -04:00
QString MainWindow : : TextThreadString ( TextThread * thread )
{
ThreadParameter tp = thread - > GetThreadParameter ( ) ;
return QString ( " %1:%2:%3:%4: " ) . arg (
QString : : number ( tp . pid ) ,
QString : : number ( tp . hook , 16 ) ,
QString : : number ( tp . retn , 16 ) ,
QString : : number ( tp . spl , 16 )
) . toUpper ( ) ;
}
ThreadParameter MainWindow : : ParseTextThreadString ( QString textThreadString )
{
QStringList threadParam = textThreadString . split ( " : " ) ;
return { threadParam [ 0 ] . toUInt ( ) , threadParam [ 1 ] . toULongLong ( nullptr , 16 ) , threadParam [ 2 ] . toULongLong ( nullptr , 16 ) , threadParam [ 3 ] . toULongLong ( nullptr , 16 ) } ;
}
DWORD MainWindow : : GetSelectedProcessId ( )
{
return processCombo - > currentText ( ) . split ( " : " ) [ 0 ] . toULong ( ) ;
}
2018-08-17 14:37:37 -04:00
void MainWindow : : ReloadExtensions ( )
{
extenCombo - > clear ( ) ;
std : : map < int , QString > extensions = LoadExtensions ( ) ;
for ( auto i : extensions ) extenCombo - > addItem ( QString : : number ( i . first ) + " : " + i . second ) ;
}
2018-08-04 03:16:14 -04:00
std : : unordered_map < std : : string , int > MainWindow : : GetInfoForExtensions ( TextThread * thread )
{
return
{
2018-08-21 22:43:30 -04:00
{ " current select " , ( int ) ttCombo - > currentText ( ) . startsWith ( TextThreadString ( thread ) ) } ,
{ " text number " , 0 } ,
2018-08-04 03:16:14 -04:00
{ " process id " , thread - > GetThreadParameter ( ) . pid } ,
{ " hook address " , ( int ) thread - > GetThreadParameter ( ) . hook } ,
{ " hook address (upper 32 bits) " , ( int ) ( thread - > GetThreadParameter ( ) . hook > > 32 ) }
} ;
}
2018-07-25 21:48:18 -07:00
QVector < HookParam > MainWindow : : GetAllHooks ( DWORD processId )
{
2018-08-22 15:11:40 -04:00
QSet < DWORD > addresses ;
2018-07-25 21:48:18 -07:00
QVector < HookParam > hooks ;
for ( int i = 0 ; i < ttCombo - > count ( ) ; + + i )
2018-08-21 22:43:30 -04:00
{
ThreadParameter tp = ParseTextThreadString ( ttCombo - > itemText ( i ) ) ;
2018-08-22 15:11:40 -04:00
if ( tp . pid = = processId & & ! addresses . contains ( tp . hook ) )
2018-07-25 21:48:18 -07:00
{
2018-08-21 22:43:30 -04:00
addresses . insert ( tp . hook ) ;
hooks . push_back ( Host : : GetHookParam ( tp ) ) ;
2018-07-25 21:48:18 -07:00
}
2018-08-21 22:43:30 -04:00
}
2018-07-25 21:48:18 -07:00
return hooks ;
}
2018-07-23 12:25:02 -07:00
void MainWindow : : on_attachButton_clicked ( )
2018-07-22 16:53:51 -07:00
{
2018-08-20 14:30:50 -04:00
std : : unordered_map < std : : wstring , DWORD > allProcesses = GetAllProcesses ( ) ;
QStringList processList ;
for ( auto i : allProcesses )
processList . push_back ( QString : : fromStdWString ( i . first ) ) ;
processList . sort ( Qt : : CaseInsensitive ) ;
2018-07-25 21:48:18 -07:00
bool ok ;
2018-07-30 20:25:08 -07:00
QString process = QInputDialog : : getItem ( this , " Select Process " ,
2018-08-18 15:06:14 -04:00
" If you don't see the process you want to inject, try running with admin rights \r \n You can just type in the process id if you know it " ,
2018-08-20 14:30:50 -04:00
processList , 0 , true , & ok ) ;
2018-07-30 20:25:08 -07:00
if ( ! ok ) return ;
2018-08-20 14:30:50 -04:00
if ( process . toInt ( ) )
{
if ( Host : : InjectProcess ( process . toInt ( ) ) ) return ;
}
else if ( Host : : InjectProcess ( allProcesses [ process . toStdWString ( ) ] ) ) return ;
2018-08-22 15:11:40 -04:00
Host : : AddConsoleOutput ( L " failed to attach " ) ;
2018-07-23 22:57:54 -07:00
}
void MainWindow : : on_detachButton_clicked ( )
{
2018-08-21 22:43:30 -04:00
Host : : DetachProcess ( GetSelectedProcessId ( ) ) ;
2018-07-23 22:57:54 -07:00
}
2018-07-25 21:48:18 -07:00
void MainWindow : : on_hookButton_clicked ( )
{
bool ok ;
2018-08-04 18:01:59 -04:00
QString hookCode = QInputDialog : : getText ( this , " Add Hook " , CodeInfoDump , QLineEdit : : Normal , " " , & ok ) ;
2018-07-30 20:25:08 -07:00
if ( ! ok ) return ;
2018-08-04 18:01:59 -04:00
HookParam toInsert = ParseCode ( hookCode ) ;
2018-07-30 20:25:08 -07:00
if ( toInsert . type = = 0 & & toInsert . length_offset = = 0 )
{
2018-08-04 18:01:59 -04:00
Host : : AddConsoleOutput ( L " invalid code " ) ;
2018-07-30 20:25:08 -07:00
return ;
}
2018-08-21 22:43:30 -04:00
Host : : InsertHook ( GetSelectedProcessId ( ) , ParseCode ( hookCode ) ) ;
2018-07-25 21:48:18 -07:00
}
void MainWindow : : on_unhookButton_clicked ( )
{
2018-08-21 22:43:30 -04:00
QVector < HookParam > hooks = GetAllHooks ( GetSelectedProcessId ( ) ) ;
2018-07-25 21:48:18 -07:00
QStringList hookList ;
for ( auto i : hooks ) hookList . push_back (
2018-08-21 22:43:30 -04:00
QString : : fromStdWString ( Host : : GetHookName ( GetSelectedProcessId ( ) , i . address ) ) +
2018-07-25 21:48:18 -07:00
" : " +
2018-08-21 22:43:30 -04:00
GenerateCode ( i , GetSelectedProcessId ( ) )
2018-07-25 21:48:18 -07:00
) ;
bool ok ;
QString hook = QInputDialog : : getItem ( this , " Unhook " , " Which hook to remove? " , hookList , 0 , false , & ok ) ;
2018-08-21 22:43:30 -04:00
if ( ok ) Host : : RemoveHook ( GetSelectedProcessId ( ) , hooks . at ( hookList . indexOf ( hook ) ) . address ) ;
2018-07-25 21:48:18 -07:00
}
2018-07-26 22:42:21 -07:00
void MainWindow : : on_saveButton_clicked ( )
{
2018-08-21 22:43:30 -04:00
QVector < HookParam > hooks = GetAllHooks ( GetSelectedProcessId ( ) ) ;
QString hookList = GetFullModuleName ( GetSelectedProcessId ( ) ) ;
2018-07-26 22:42:21 -07:00
for ( auto i : hooks )
if ( ! ( i . type & HOOK_ENGINE ) )
2018-08-21 22:43:30 -04:00
hookList + = " , " + GenerateCode ( i , GetSelectedProcessId ( ) ) ;
2018-07-26 22:42:21 -07:00
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-08-21 22:43:30 -04:00
textOutput - > setPlainText ( QString : : fromStdWString ( Host : : GetThread ( ParseTextThreadString ( ttCombo - > itemText ( index ) ) ) - > GetStore ( ) ) ) ;
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 ( )
{
2018-08-03 00:07:25 -04:00
QString extenFileName = QFileDialog : : getOpenFileName ( this , " Select Extension dll " , " C: \\ " , " Extensions (*.dll) " ) ;
2018-07-28 12:41:21 -07:00
if ( ! extenFileName . length ( ) ) return ;
QString extenName = extenFileName . split ( " / " ) [ extenFileName . split ( " / " ) . count ( ) - 1 ] ;
extenName . chop ( 4 ) ;
QString copyTo = QString : : number ( extenCombo - > itemText ( extenCombo - > count ( ) - 1 ) . split ( " : " ) [ 0 ] . toInt ( ) + 1 ) + " _ " +
extenName +
" _nexthooker_extension.dll " ;
QFile : : copy ( extenFileName , copyTo ) ;
2018-08-17 14:37:37 -04:00
ReloadExtensions ( ) ;
2018-07-28 12:41:21 -07:00
}
void MainWindow : : on_rmvExtenButton_clicked ( )
{
2018-08-03 00:07:25 -04:00
if ( extenCombo - > currentText ( ) . size ( ) = = 0 ) return ;
2018-08-07 02:19:20 -04:00
QString extenFileName = extenCombo - > currentText ( ) . split ( " : " ) [ 0 ] + " _ " + extenCombo - > currentText ( ) . split ( " : " ) [ 1 ] + " _nexthooker_extension.dll " ;
2018-07-28 12:41:21 -07:00
FreeLibrary ( GetModuleHandleW ( extenFileName . toStdWString ( ) . c_str ( ) ) ) ;
2018-08-19 17:55:50 -04:00
QFile : : remove ( extenFileName ) ;
2018-08-17 14:37:37 -04:00
ReloadExtensions ( ) ;
2018-07-26 22:42:21 -07:00
}