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