mirror of
https://github.com/HIllya51/LunaHook.git
synced 2024-12-27 13:44:11 +08:00
109 lines
3.4 KiB
C++
109 lines
3.4 KiB
C++
|
|
#include <CommCtrl.h>
|
|
#include <TlHelp32.h>
|
|
#include"host.h"
|
|
#include"processlistwindow.h"
|
|
#include"Lang/Lang.h"
|
|
#include<shellapi.h>
|
|
std::unordered_map<std::wstring,std::vector<int>> getprocesslist()
|
|
{
|
|
std::unordered_map<std::wstring,std::vector<int>>exe_pid;
|
|
AutoHandle<> hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
|
|
if (hSnapshot == INVALID_HANDLE_VALUE)
|
|
return {};
|
|
|
|
PROCESSENTRY32 pe32;
|
|
pe32.dwSize = sizeof(PROCESSENTRY32);
|
|
wchar_t buff[65535];
|
|
if (Process32First(hSnapshot, &pe32))
|
|
{
|
|
do
|
|
{
|
|
auto PROCESS_INJECT_ACCESS = (
|
|
PROCESS_CREATE_THREAD |
|
|
PROCESS_QUERY_INFORMATION |
|
|
PROCESS_VM_OPERATION |
|
|
PROCESS_VM_WRITE |
|
|
PROCESS_VM_READ);
|
|
AutoHandle<> handle = OpenProcess(PROCESS_INJECT_ACCESS, 0, pe32.th32ProcessID);
|
|
if (handle == 0)continue;
|
|
DWORD sz = 65535;
|
|
QueryFullProcessImageNameW(handle, 0, buff, &sz);
|
|
|
|
auto buffs=std::wstring(buff);
|
|
auto str=stolower(buffs);
|
|
if(str.find(L":\\windows\\")!=str.npos || str.find(L"\\microsoft")!=str.npos|| str.find(L"\\windowsapps")!=str.npos)continue;
|
|
|
|
|
|
if(exe_pid.find(buffs)==exe_pid.end()){
|
|
exe_pid.insert({buffs,{}});
|
|
}
|
|
exe_pid[buffs].push_back(pe32.th32ProcessID);
|
|
} while (Process32Next(hSnapshot, &pe32));
|
|
}
|
|
return exe_pid;
|
|
}
|
|
|
|
void processlistwindow::PopulateProcessList(listview* _listbox,std::unordered_map<std::wstring,std::vector<int>>&exe_pid){
|
|
_listbox->clear();
|
|
for(auto& exe:exe_pid){
|
|
auto hicon=GetExeIcon(exe.first);
|
|
_listbox->additem(exe.first,hicon);
|
|
DestroyIcon(hicon);
|
|
}
|
|
}
|
|
|
|
processlistwindow::processlistwindow(mainwindow* p):mainwindow(p){
|
|
g_hEdit = new textedit(this,L"",10, 10, 400, 30,ES_AUTOHSCROLL);
|
|
g_hButton=new button(this,BtnAttach,420, 10, 100, 30);
|
|
g_refreshbutton =new button(this,BtnRefresh,530, 10, 100, 30);
|
|
g_hButton->onclick=[&](){
|
|
auto str=g_hEdit->text();
|
|
if(str.size()){
|
|
close();
|
|
|
|
for(auto _s:strSplit(str,L",")){
|
|
DWORD pid=0;
|
|
try{
|
|
pid=std::stoi(_s);
|
|
}catch(std::exception&){}
|
|
if(pid)
|
|
Host::InjectProcess(pid);
|
|
}
|
|
}
|
|
};
|
|
g_refreshbutton->onclick=[&](){
|
|
g_exe_pid=getprocesslist();
|
|
PopulateProcessList(g_hListBox,g_exe_pid);
|
|
};
|
|
g_hListBox = new listview(this,10, 60, 310, 200);
|
|
g_hListBox->setheader({L""});
|
|
g_hListBox->oncurrentchange=[&](int idx){
|
|
auto pids=g_exe_pid[g_hListBox->text(idx)];
|
|
|
|
std::wstring _;
|
|
bool _1=false;
|
|
for(auto &p:pids){
|
|
if(_1)_+=L",";
|
|
_+=std::to_wstring(p);
|
|
_1=true;
|
|
}
|
|
g_hEdit->settext(_);
|
|
};
|
|
settext(WndSelectProcess);
|
|
setcentral(800,400);
|
|
}
|
|
void processlistwindow::on_show(){
|
|
g_hEdit->settext(L"");
|
|
g_exe_pid=getprocesslist();
|
|
PopulateProcessList(g_hListBox,g_exe_pid);
|
|
}
|
|
void processlistwindow::on_size(int w,int h){
|
|
w=w-20;
|
|
auto _w=w-20;
|
|
g_hEdit->setgeo(10,10,_w/2,30);
|
|
g_hButton->setgeo(10+_w/2+10,10,_w/4,30);
|
|
g_refreshbutton->setgeo(10+_w/2+_w/4+20,10,_w/4,30);
|
|
g_hListBox->setgeo(10,50,w,h-60);
|
|
g_hListBox->autosize();
|
|
} |