[475] | 1 | #include "stdafx.h"
|
---|
| 2 |
|
---|
[3] | 3 | #ifndef THETEXT
|
---|
| 4 |
|
---|
| 5 | #include "common.h"
|
---|
| 6 | #include <psapi.h>
|
---|
| 7 | #include <tlhelp32.h>
|
---|
| 8 |
|
---|
| 9 | #pragma comment(lib, "psapi.lib")
|
---|
| 10 |
|
---|
[475] | 11 | using namespace ActiveBasic::IDE;
|
---|
[3] | 12 |
|
---|
[475] | 13 |
|
---|
[3] | 14 | int GetSelectingProcessId(HWND hListView,DWORD *lpdwPlatform){
|
---|
| 15 | int nCount;
|
---|
| 16 | nCount=ListView_GetItemCount(hListView);
|
---|
| 17 |
|
---|
| 18 | char temporary[255];
|
---|
| 19 | DWORD dwProcessId;
|
---|
| 20 | int i;
|
---|
| 21 | for(i=0;i<nCount;i++){
|
---|
| 22 | if(ListView_GetItemState(hListView,i,LVIS_SELECTED)){
|
---|
| 23 | ListView_GetItemText(hListView,i,1,temporary,255);
|
---|
| 24 | dwProcessId=atoi(temporary);
|
---|
| 25 |
|
---|
| 26 | ListView_GetItemText(hListView,i,2,temporary,255);
|
---|
| 27 | if(lstrcmpi(temporary,"Win32")==0) *lpdwPlatform=IMAGE_FILE_MACHINE_I386;
|
---|
| 28 | else if(lstrcmpi(temporary,"Win64")==0) *lpdwPlatform=IMAGE_FILE_MACHINE_AMD64;
|
---|
| 29 | break;
|
---|
| 30 | }
|
---|
| 31 | }
|
---|
| 32 | if(i==nCount) return 0;
|
---|
| 33 |
|
---|
| 34 | return dwProcessId;
|
---|
| 35 | }
|
---|
| 36 |
|
---|
| 37 | void InsertProcess_ToListView(HWND hListView,char *lpszAppPath,int id,char *lpszPlatform,char *lpszTitle){
|
---|
| 38 | char szAppName[MAX_PATH],temp2[MAX_PATH];
|
---|
| 39 | _splitpath(lpszAppPath,NULL,NULL,szAppName,temp2);
|
---|
| 40 | lstrcat(szAppName,temp2);
|
---|
| 41 |
|
---|
| 42 | //イメージリストを取得
|
---|
| 43 | HIMAGELIST hImageList;
|
---|
| 44 | hImageList=ListView_GetImageList(hListView,LVSIL_SMALL );
|
---|
| 45 |
|
---|
| 46 | //アイコンを取得
|
---|
| 47 | int iImage;
|
---|
| 48 | SHFILEINFO shfi;
|
---|
| 49 | SHGetFileInfo(lpszAppPath, FILE_ATTRIBUTE_ARCHIVE, &shfi, sizeof(SHFILEINFO), SHGFI_ICON | SHGFI_SMALLICON );
|
---|
| 50 | if(shfi.hIcon){
|
---|
| 51 | iImage=ImageList_AddIcon(hImageList,shfi.hIcon);
|
---|
| 52 | DestroyIcon(shfi.hIcon);
|
---|
| 53 | }
|
---|
| 54 | else iImage=-1;
|
---|
| 55 |
|
---|
| 56 | char szId[255];
|
---|
| 57 | sprintf(szId,"%d",id);
|
---|
| 58 |
|
---|
| 59 | //リストビューに追加
|
---|
| 60 | LV_ITEM item;
|
---|
| 61 | if(iImage==-1) item.mask=LVIF_TEXT;
|
---|
| 62 | else item.mask=LVIF_TEXT|LVIF_IMAGE;
|
---|
| 63 | item.iItem=0;
|
---|
| 64 |
|
---|
| 65 | item.pszText=szAppName;
|
---|
| 66 | item.iSubItem=0;
|
---|
| 67 | item.iImage=iImage;
|
---|
| 68 | ListView_InsertItem(hListView,&item);
|
---|
| 69 |
|
---|
| 70 | item.mask=LVIF_TEXT ;
|
---|
| 71 | item.pszText=szId;
|
---|
| 72 | item.iSubItem++;
|
---|
| 73 | ListView_SetItem(hListView,&item);
|
---|
| 74 |
|
---|
| 75 | item.pszText=lpszPlatform;
|
---|
| 76 | item.iSubItem++;
|
---|
| 77 | ListView_SetItem(hListView,&item);
|
---|
| 78 |
|
---|
| 79 | item.pszText=lpszTitle;
|
---|
| 80 | item.iSubItem++;
|
---|
| 81 | ListView_SetItem(hListView,&item);
|
---|
| 82 | }
|
---|
| 83 |
|
---|
[475] | 84 | void ListupProcesses64(HWND hListView)
|
---|
| 85 | {
|
---|
[3] | 86 | ListView_DeleteAllItems(hListView);
|
---|
| 87 |
|
---|
| 88 |
|
---|
| 89 | /////////////////////////////////////////////////////
|
---|
| 90 | // 64ビット版プロセス列挙モジュールを呼び出す
|
---|
| 91 | /////////////////////////////////////////////////////
|
---|
| 92 |
|
---|
[477] | 93 | std::string enumProcess64ExePath = ActiveBasic::Common::Environment::GetAbdevSystemDirPath() + "\\enum_process64\\enum_process64.exe";
|
---|
[3] | 94 |
|
---|
| 95 | STARTUPINFO si;
|
---|
| 96 | PROCESS_INFORMATION pi;
|
---|
| 97 | memset(&si,0,sizeof(STARTUPINFO));
|
---|
| 98 | si.cb=sizeof(STARTUPINFO);
|
---|
[475] | 99 | CreateProcess(NULL,const_cast<char *>(enumProcess64ExePath.c_str()),NULL,NULL,0,NORMAL_PRIORITY_CLASS,NULL,NULL,&si,&pi);
|
---|
[3] | 100 |
|
---|
| 101 | WaitForSingleObject(pi.hProcess,INFINITE);
|
---|
| 102 |
|
---|
| 103 |
|
---|
| 104 | //////////////////////////////////
|
---|
| 105 | // 生成されたファイルの読み込み
|
---|
| 106 | //////////////////////////////////
|
---|
| 107 |
|
---|
[477] | 108 | std::string listPath = ActiveBasic::Common::Environment::GetAbdevSystemDirPath() + "\\enum_process64\\list.dat";
|
---|
[3] | 109 |
|
---|
| 110 | char *buffer;
|
---|
[475] | 111 | buffer=ReadBuffer_NonErrMsg( listPath );
|
---|
[3] | 112 | if(!buffer) return;
|
---|
| 113 |
|
---|
| 114 | int i=0;
|
---|
[475] | 115 | char temporary[MAX_PATH];
|
---|
[3] | 116 | while(1){
|
---|
| 117 | //ファイルパス
|
---|
| 118 | char szAppPath[MAX_PATH];
|
---|
| 119 | i=GetOneParameter(buffer,i,szAppPath);
|
---|
[475] | 120 | if(szAppPath[0]=='\0') break;
|
---|
[3] | 121 | RemoveStringQuotes(szAppPath);
|
---|
| 122 |
|
---|
| 123 | //プロセスID
|
---|
| 124 | DWORD dwProcessId;
|
---|
| 125 | i=GetOneParameter(buffer,i,temporary);
|
---|
| 126 | if(temporary[0]=='\0') break;
|
---|
| 127 | dwProcessId=(DWORD)atoi(temporary);
|
---|
| 128 |
|
---|
| 129 | //プラットフォーム
|
---|
| 130 | char szPlatform[255];
|
---|
| 131 | i=GetOneParameter(buffer,i,szPlatform);
|
---|
| 132 | if(temporary[0]=='\0') break;
|
---|
| 133 |
|
---|
| 134 | //タイトル
|
---|
| 135 | char szTitle[255];
|
---|
| 136 | int i2;
|
---|
| 137 | for(i2=0;;i++,i2++){
|
---|
| 138 | if(IS_RETURN(buffer,i)||buffer[i]=='\0'){
|
---|
| 139 | szTitle[i2]=0;
|
---|
| 140 | break;
|
---|
| 141 | }
|
---|
| 142 | szTitle[i2]=buffer[i];
|
---|
| 143 | }
|
---|
| 144 |
|
---|
| 145 |
|
---|
| 146 | //リストビューへ追加
|
---|
| 147 | InsertProcess_ToListView(hListView,szAppPath,dwProcessId,szPlatform,szTitle);
|
---|
| 148 |
|
---|
| 149 | if(buffer[i]=='\r'&&buffer[i+1]=='\n') i+=2;
|
---|
| 150 | if(buffer[i]=='\0') break;
|
---|
| 151 | }
|
---|
| 152 |
|
---|
| 153 | HeapDefaultFree(buffer);
|
---|
| 154 | }
|
---|
| 155 |
|
---|
| 156 | // EnumWindowsProcコールバック関数
|
---|
| 157 | BOOL CALLBACK FindWindowProc(HWND hwnd, LPARAM lParam)
|
---|
| 158 | {
|
---|
| 159 | HWND* pphWnd = (HWND*)lParam;
|
---|
| 160 | *pphWnd = hwnd;
|
---|
| 161 | return FALSE; /* 列挙を中断 */
|
---|
| 162 | }
|
---|
| 163 | void ListupProcesses(HWND hListView){
|
---|
| 164 | char temporary[MAX_PATH];
|
---|
| 165 |
|
---|
| 166 | ListView_DeleteAllItems(hListView);
|
---|
| 167 |
|
---|
| 168 | DWORD *pdwProcessId;
|
---|
| 169 | pdwProcessId=(DWORD *)HeapAlloc(hHeap,0,8192*sizeof(DWORD));
|
---|
| 170 |
|
---|
| 171 | //プロセスを列挙
|
---|
| 172 | DWORD cbNeeded;
|
---|
| 173 | EnumProcesses(pdwProcessId,8192*sizeof(DWORD),&cbNeeded);
|
---|
| 174 |
|
---|
| 175 | char szMyAppPath[MAX_PATH];
|
---|
| 176 | GetModuleFileName(GetModuleHandle(0),szMyAppPath,MAX_PATH);
|
---|
| 177 |
|
---|
| 178 | int i;
|
---|
| 179 | for(i=0;i<(int)(cbNeeded/sizeof(DWORD));i++){
|
---|
| 180 |
|
---|
| 181 | //プロセスIDを元にハンドルを取得
|
---|
| 182 | HANDLE hProcess;
|
---|
| 183 | hProcess=OpenProcess(PROCESS_ALL_ACCESS,0,pdwProcessId[i]);
|
---|
| 184 | if(!hProcess) continue;
|
---|
| 185 |
|
---|
| 186 | //そのプロセスにおける実行モジュールのインスタンスハンドルを取得
|
---|
| 187 | HINSTANCE hModule;
|
---|
| 188 | DWORD cbReturned;
|
---|
| 189 | if(!EnumProcessModules( hProcess, &hModule, sizeof(HINSTANCE), &cbReturned ))
|
---|
| 190 | continue;
|
---|
| 191 |
|
---|
| 192 | //実行ファイル名を取得
|
---|
| 193 | GetModuleFileNameEx(hProcess,hModule,temporary,MAX_PATH);
|
---|
| 194 |
|
---|
| 195 |
|
---|
| 196 | //////////////////////////////
|
---|
| 197 | // ウィンドウタイトルを取得
|
---|
| 198 | //////////////////////////////
|
---|
| 199 |
|
---|
| 200 | //スナップショットを取得
|
---|
| 201 | HANDLE hSnapshot;
|
---|
| 202 | hSnapshot=CreateToolhelp32Snapshot(TH32CS_SNAPALL,0);
|
---|
| 203 |
|
---|
| 204 | THREADENTRY32 te;
|
---|
| 205 | memset(&te,0,sizeof(THREADENTRY32));
|
---|
| 206 | te.dwSize=sizeof(THREADENTRY32);
|
---|
| 207 | if(Thread32First(hSnapshot,&te)){
|
---|
| 208 | do{
|
---|
| 209 | if(te.th32OwnerProcessID==pdwProcessId[i]){
|
---|
| 210 | break;
|
---|
| 211 | }
|
---|
| 212 | }while(Thread32Next(hSnapshot,&te));
|
---|
| 213 | }
|
---|
| 214 |
|
---|
| 215 | HWND hwnd;
|
---|
| 216 | hwnd=0;
|
---|
| 217 | EnumThreadWindows(te.th32ThreadID,FindWindowProc,(LPARAM)&hwnd);
|
---|
| 218 |
|
---|
| 219 | while(GetParent(hwnd)) hwnd=GetParent(hwnd);
|
---|
| 220 |
|
---|
| 221 | char szWndTitle[1024];
|
---|
| 222 | szWndTitle[0]=0;
|
---|
| 223 | if(hwnd) GetWindowText(hwnd,szWndTitle,1024);
|
---|
| 224 |
|
---|
| 225 | CloseHandle(hSnapshot);
|
---|
| 226 |
|
---|
| 227 |
|
---|
| 228 |
|
---|
| 229 | CloseHandle(hProcess);
|
---|
| 230 |
|
---|
| 231 | if(lstrcmpi(szMyAppPath,temporary)==0){
|
---|
| 232 | //自分自身(ProjectEditor.exe)は無視
|
---|
| 233 | continue;
|
---|
| 234 | }
|
---|
| 235 |
|
---|
| 236 | //リストビューへ追加
|
---|
| 237 | InsertProcess_ToListView(hListView,temporary,pdwProcessId[i],"Win32",szWndTitle);
|
---|
| 238 | }
|
---|
| 239 |
|
---|
| 240 | HeapDefaultFree(pdwProcessId);
|
---|
| 241 | }
|
---|
| 242 |
|
---|
| 243 | BOOL CALLBACK DlgAttach(HWND hwnd,UINT message,WPARAM wParam,LPARAM lParam){
|
---|
| 244 | static HWND hListView;
|
---|
| 245 | static HIMAGELIST hImageList;
|
---|
| 246 | static DWORD *lpdwPlatform;
|
---|
| 247 | switch(message){
|
---|
| 248 | case WM_INITDIALOG:
|
---|
| 249 | SetPosCenter(hwnd);
|
---|
| 250 |
|
---|
| 251 | lpdwPlatform=(DWORD *)lParam;
|
---|
| 252 |
|
---|
| 253 | hListView=GetDlgItem(hwnd,IDC_PROCESSLIST);
|
---|
| 254 |
|
---|
| 255 | //拡張スタイル
|
---|
| 256 | DWORD dwStyle;
|
---|
| 257 | dwStyle=ListView_GetExtendedListViewStyle(hListView);
|
---|
| 258 | dwStyle|=LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES;
|
---|
| 259 | ListView_SetExtendedListViewStyle(hListView,dwStyle);
|
---|
| 260 |
|
---|
| 261 |
|
---|
| 262 | //イメージリストの生成と設定
|
---|
| 263 | hImageList=ImageList_Create(16,16,ILC_COLOR24|ILC_MASK,0,0);
|
---|
| 264 | ListView_SetImageList(hListView,hImageList,LVSIL_SMALL );
|
---|
| 265 |
|
---|
| 266 |
|
---|
| 267 | ///////////////////////////
|
---|
| 268 | // カラムをセット
|
---|
| 269 | ///////////////////////////
|
---|
| 270 |
|
---|
| 271 | LV_COLUMN ListView_Column;
|
---|
| 272 | ListView_Column.mask=LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM;
|
---|
| 273 | ListView_Column.fmt=LVCFMT_LEFT;
|
---|
| 274 |
|
---|
| 275 | ListView_Column.cx=150;
|
---|
| 276 | ListView_Column.pszText="プロセス名";
|
---|
| 277 | ListView_Column.iSubItem=0;
|
---|
| 278 | ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);
|
---|
| 279 |
|
---|
| 280 | ListView_Column.cx=70;
|
---|
| 281 | ListView_Column.pszText="ID";
|
---|
| 282 | ListView_Column.iSubItem++;
|
---|
| 283 | ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);
|
---|
| 284 |
|
---|
| 285 | ListView_Column.cx=50;
|
---|
| 286 | ListView_Column.pszText="種別";
|
---|
| 287 | ListView_Column.iSubItem++;
|
---|
| 288 | ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);
|
---|
| 289 |
|
---|
| 290 | ListView_Column.cx=200;
|
---|
| 291 | ListView_Column.pszText="タイトル";
|
---|
| 292 | ListView_Column.iSubItem++;
|
---|
| 293 | ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);
|
---|
| 294 |
|
---|
| 295 |
|
---|
| 296 | //リスビューを更新
|
---|
| 297 | if(IsWow64()) ListupProcesses64(hListView);
|
---|
| 298 | else ListupProcesses(hListView);
|
---|
| 299 |
|
---|
| 300 | break;
|
---|
| 301 | case WM_COMMAND:
|
---|
| 302 | switch(LOWORD(wParam)){
|
---|
| 303 | case IDOK:
|
---|
| 304 | EndDialog(hwnd,GetSelectingProcessId(hListView,lpdwPlatform));
|
---|
| 305 | return 1;
|
---|
| 306 | case IDCANCEL:
|
---|
| 307 | EndDialog(hwnd,0);
|
---|
| 308 | return 1;
|
---|
| 309 | }
|
---|
| 310 | return 0;
|
---|
| 311 |
|
---|
| 312 | case WM_NOTIFY:
|
---|
| 313 | NMLISTVIEW *nmListView;
|
---|
| 314 | nmListView=(NMLISTVIEW *)lParam;
|
---|
| 315 | if(nmListView->hdr.hwndFrom==hListView){
|
---|
| 316 | if(nmListView->hdr.code==NM_DBLCLK){
|
---|
| 317 | if(nmListView->iItem==-1) return 1;
|
---|
| 318 | SendMessage(hwnd,WM_COMMAND,IDOK,0);
|
---|
| 319 | return 1;
|
---|
| 320 | }
|
---|
| 321 | }
|
---|
| 322 | break;
|
---|
| 323 |
|
---|
| 324 | case WM_DESTROY:
|
---|
| 325 | ImageList_Destroy(hImageList);
|
---|
| 326 | return 1;
|
---|
| 327 | }
|
---|
| 328 | return 0;
|
---|
| 329 | }
|
---|
| 330 |
|
---|
| 331 |
|
---|
| 332 | #endif //THETEXT
|
---|