#include #include #include"host.h" #include"processlistwindow.h" #include"Lang/Lang.h" std::unordered_map> getprocesslist() { std::unordered_map>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(listbox* _listbox,std::unordered_map>&exe_pid){ _listbox->clear(); for(auto& exe:exe_pid){ _listbox->additem(exe.first.c_str()); } } processlistwindow::processlistwindow(mainwindow* p):mainwindow(p){ g_hEdit = new textedit(this,L"",10, 10, 400, 40,ES_AUTOHSCROLL); g_hButton=new button(this,BtnAttach,420, 10, 100, 40); g_refreshbutton =new button(this,BtnRefresh,530, 10, 100, 40); 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 listbox(this,10, 60, 310, 200); 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){ g_hListBox->setgeo(10,60,w-20,h-70); }