mirror of
https://github.com/HIllya51/LunaHook.git
synced 2024-10-23 07:28:16 +08:00
234 lines
8.1 KiB
C++
234 lines
8.1 KiB
C++
|
|
#include <CommCtrl.h>
|
|
#include <TlHelp32.h>
|
|
#include <commdlg.h>
|
|
#include<stdio.h>
|
|
#include"host.h"
|
|
#include"hookcode.h"
|
|
#include"textthread.h"
|
|
#include"LunaHost.h"
|
|
#include"processlistwindow.h"
|
|
#include"Lang/Lang.h"
|
|
void LunaHost::toclipboard(std::wstring& sentence){
|
|
|
|
for (int loop = 0; loop < 10; loop++) {
|
|
if (OpenClipboard(winId)) {
|
|
HGLOBAL hMem = GlobalAlloc(GMEM_MOVEABLE, (sentence.size() + 2) * sizeof(wchar_t));
|
|
memcpy(GlobalLock(hMem), sentence.c_str(), (sentence.size() + 2) * sizeof(wchar_t));
|
|
EmptyClipboard();
|
|
SetClipboardData(CF_UNICODETEXT, hMem);
|
|
GlobalUnlock(hMem);
|
|
CloseClipboard();
|
|
break;
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
|
}
|
|
}
|
|
void LunaHost::on_close(){
|
|
for(auto pid:attachedprocess){
|
|
Host::DetachProcess(pid);
|
|
}
|
|
}
|
|
void LunaHost::on_size(int w,int h){
|
|
int height = h-140;
|
|
g_hListBox_listtext->setgeo(10, 110, w - 20, height/2);
|
|
g_showtexts->setgeo(10, 120+height/2, w - 20, height/2);
|
|
}
|
|
std::optional<std::wstring>SelectFile(HWND hwnd){
|
|
OPENFILENAME ofn;
|
|
wchar_t szFileName[MAX_PATH] = { 0 };
|
|
|
|
ZeroMemory(&ofn, sizeof(ofn));
|
|
ofn.lStructSize = sizeof(ofn);
|
|
ofn.hwndOwner = hwnd;
|
|
ofn.lpstrFilter = L"Plugin Files\0*.dll;*.xdll\0";
|
|
ofn.lpstrFile = szFileName;
|
|
ofn.nMaxFile = sizeof(szFileName);
|
|
ofn.Flags = OFN_EXPLORER | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY;
|
|
|
|
if (GetOpenFileName(&ofn))
|
|
{
|
|
return szFileName;
|
|
}
|
|
else return {};
|
|
}
|
|
LunaHost::LunaHost(){
|
|
settext(WndLunaHostGui);
|
|
g_selectprocessbutton =new button(this,BtnSelectProcess,830, 10, 200, 40) ;
|
|
|
|
g_hEdit_userhook = new textedit(this,L"",10, 60, 600, 40,ES_AUTOHSCROLL);
|
|
btnaddplugin=new button(this,BtnAddPlugin,830,60,200,40);
|
|
btnaddplugin->onclick=[&](){
|
|
auto f=SelectFile(winId);
|
|
if(f.has_value()){
|
|
plugins->addplugin(f.value());
|
|
}
|
|
};
|
|
g_hButton_insert = new button(this,BtnInsertUserHook,610, 60, 200, 40) ;
|
|
|
|
g_selectprocessbutton->onclick=[&](){
|
|
if(_processlistwindow==0) _processlistwindow=new processlistwindow(this);
|
|
_processlistwindow->show();
|
|
};
|
|
g_hButton_insert->onclick=[&](){
|
|
auto hp = HookCode::Parse(std::move(g_hEdit_userhook->text()));
|
|
if(hp){
|
|
for(auto _:attachedprocess){
|
|
Host::InsertHook(_,hp.value());
|
|
}
|
|
}
|
|
else{
|
|
g_showtexts->appendtext(NotifyInvalidHookCode);
|
|
}
|
|
};
|
|
g_check_clipboard =new checkbox(this,BtnToClipboard,550, 10, 200, 40) ;
|
|
g_check_clipboard->onclick=[&](){
|
|
check_toclipboard=g_check_clipboard->ischecked();
|
|
};
|
|
new label(this,LblFlushDelay,10, 10, 150, 40);
|
|
new label(this,LblCodePage,270, 10, 150, 40);
|
|
|
|
g_timeout = new textedit(this,std::to_wstring(TextThread::flushDelay).c_str(),160, 10, 100, 40) ;
|
|
g_codepage = new textedit(this,L"932",420, 10, 100, 40) ;
|
|
|
|
g_timeout->ontextchange=[&](const std::wstring &text){
|
|
TextThread::flushDelay=std::stoi(text);
|
|
};
|
|
g_codepage->ontextchange=[&](const std::wstring &text){
|
|
try {
|
|
auto cp=std::stoi(text);
|
|
if(IsValidCodePage(cp))
|
|
Host::defaultCodepage= cp;
|
|
}
|
|
catch (const std::invalid_argument& e) {
|
|
}
|
|
};
|
|
|
|
g_hListBox_listtext = new listbox(this,10, 120, 200, 200);
|
|
g_hListBox_listtext->oncurrentchange=[&](int idx){
|
|
uint64_t handle = g_hListBox_listtext->getdata(idx);
|
|
std::wstring get;
|
|
for(auto& _:savetext.at(handle)){
|
|
get+=_;
|
|
get+=L"\r\n";
|
|
}
|
|
currentselect=handle;
|
|
g_showtexts->settext(get);
|
|
g_showtexts->scrolltoend();
|
|
};
|
|
|
|
#define IDM_REMOVE_HOOK 1001
|
|
#define IDM_DETACH_PROCESS 1002
|
|
#define IDM_COPY_HOOKCODE 1003
|
|
g_hListBox_listtext->oncontextmenu=[](){
|
|
HMENU hMenu = CreatePopupMenu();
|
|
AppendMenu(hMenu, MF_STRING, IDM_COPY_HOOKCODE, MenuCopyHookCode);
|
|
AppendMenu(hMenu, MF_STRING, IDM_REMOVE_HOOK, MenuRemoveHook);
|
|
AppendMenu(hMenu, MF_STRING, IDM_DETACH_PROCESS, MenuDetachProcess);
|
|
return hMenu;
|
|
};
|
|
g_hListBox_listtext->oncontextmenucallback=[&](WPARAM wparam){
|
|
|
|
auto handle = g_hListBox_listtext->getdata(g_hListBox_listtext->currentidx());
|
|
auto tt=Host::GetThread(handle);
|
|
if(tt==0)return;
|
|
switch (LOWORD(wparam)) {
|
|
|
|
case IDM_COPY_HOOKCODE:
|
|
toclipboard(std::wstring(tt->hp.hookcode));
|
|
break;
|
|
case IDM_DETACH_PROCESS:
|
|
Host::DetachProcess(tt->tp.processId);
|
|
break;
|
|
case IDM_REMOVE_HOOK:
|
|
Host::RemoveHook(tt->tp.processId,tt->tp.addr);
|
|
break;
|
|
}
|
|
};
|
|
g_showtexts = new textedit(this,L"",10, 330, 200, 200,ES_READONLY|ES_MULTILINE |ES_AUTOVSCROLL| WS_VSCROLL);
|
|
|
|
plugins=new pluginmanager;
|
|
|
|
Host::Start(
|
|
[&](DWORD pid) {attachedprocess.push_back(pid);},
|
|
[&](DWORD pid) {
|
|
attachedprocess.erase(std::remove(attachedprocess.begin(), attachedprocess.end(), pid), attachedprocess.end());
|
|
},
|
|
std::bind(&LunaHost::on_thread_create,this,std::placeholders::_1),
|
|
std::bind(&LunaHost::on_thread_delete,this,std::placeholders::_1),
|
|
std::bind(&LunaHost::on_text_recv,this,std::placeholders::_1,std::placeholders::_2)
|
|
);
|
|
}
|
|
|
|
std::array<InfoForExtension, 20> LunaHost::GetSentenceInfo(TextThread& thread)
|
|
{
|
|
void (*AddText)(int64_t, const wchar_t*) = [](int64_t number, const wchar_t* text)
|
|
{
|
|
if (TextThread* thread = Host::GetThread(number)) thread->Push(text);
|
|
};
|
|
void (*AddSentence)(int64_t, const wchar_t*) = [](int64_t number, const wchar_t* sentence)
|
|
{
|
|
if (TextThread* thread = Host::GetThread(number)) thread->AddSentence(sentence);;
|
|
};
|
|
static DWORD SelectedProcessId;
|
|
auto currthread=Host::GetThread(currentselect);
|
|
SelectedProcessId=(currthread!=0)?currthread->tp.processId:0;
|
|
DWORD (*GetSelectedProcessId)() = [] { return SelectedProcessId; };
|
|
|
|
return
|
|
{ {
|
|
{ "HostHWND",(int64_t)winId },
|
|
{ "toclipboard", check_toclipboard },
|
|
{ "current select", &thread == currthread },
|
|
{ "text number", thread.handle },
|
|
{ "process id", thread.tp.processId },
|
|
{ "hook address", (int64_t)thread.tp.addr },
|
|
{ "text handle", thread.handle },
|
|
{ "text name", (int64_t)thread.name.c_str() },
|
|
{ "add sentence", (int64_t)AddSentence },
|
|
{ "add text", (int64_t)AddText },
|
|
{ "get selected process id", (int64_t)GetSelectedProcessId },
|
|
{ "void (*AddSentence)(int64_t number, const wchar_t* sentence)", (int64_t)AddSentence },
|
|
{ "void (*AddText)(int64_t number, const wchar_t* text)", (int64_t)AddText },
|
|
{ "DWORD (*GetSelectedProcessId)()", (int64_t)GetSelectedProcessId },
|
|
{ nullptr, 0 } // nullptr marks end of info array
|
|
} };
|
|
}
|
|
bool LunaHost::on_text_recv(TextThread& thread, std::wstring& output){
|
|
std::lock_guard _(settextmutex);
|
|
std::wstring lfoutput=output;
|
|
if(!plugins->dispatch(GetSentenceInfo(thread).data(),output))return false;
|
|
strReplace(lfoutput,L"\n",L"\r\n");
|
|
savetext.at(thread.handle).push_back(lfoutput);
|
|
if(currentselect==thread.handle){
|
|
g_showtexts->scrolltoend();
|
|
g_showtexts->appendtext(lfoutput);
|
|
}
|
|
return true;
|
|
}
|
|
void LunaHost::on_thread_create(TextThread& thread){
|
|
wchar_t buff[65535];
|
|
swprintf_s(buff,L"[%I64X:%I32X:%I64X:%I64X:%I64X:%s:%s]",
|
|
thread.handle,
|
|
thread.tp.processId,
|
|
thread.tp.addr,
|
|
thread.tp.ctx,
|
|
thread.tp.ctx2,
|
|
thread.name.c_str(),
|
|
thread.hp.hookcode
|
|
);
|
|
savetext.insert({thread.handle,{}});
|
|
int index=g_hListBox_listtext->additem(buff);
|
|
g_hListBox_listtext->setdata(index,thread.handle);
|
|
}
|
|
void LunaHost::on_thread_delete(TextThread& thread){
|
|
int count = g_hListBox_listtext->count();
|
|
for (int i = 0; i < count; i++) {
|
|
uint64_t handle = g_hListBox_listtext->getdata(i);
|
|
|
|
if (handle== thread.handle) {
|
|
g_hListBox_listtext->deleteitem(i);
|
|
break;
|
|
}
|
|
}
|
|
} |