6 * Copyright (C) 1999 - 2001 Brian Palmer <brianp@reactos.org>
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 #define WIN32_LEAN_AND_MEAN /* Exclude rarely-used stuff from Windows headers */
38 HANDLE hProcessAffinityHandle
;
40 LRESULT CALLBACK
AffinityDialogWndProc(HWND hDlg
, UINT message
, WPARAM wParam
, LPARAM lParam
);
42 void ProcessPage_OnSetAffinity(void)
47 TCHAR strErrorText
[260];
49 for (Index
=0; Index
<(ULONG
)ListView_GetItemCount(hProcessPageListCtrl
); Index
++) {
50 memset(&lvitem
, 0, sizeof(LV_ITEM
));
51 lvitem
.mask
= LVIF_STATE
;
52 lvitem
.stateMask
= LVIS_SELECTED
;
54 ListView_GetItem(hProcessPageListCtrl
, &lvitem
);
55 if (lvitem
.state
& LVIS_SELECTED
)
58 dwProcessId
= PerfDataGetProcessId(Index
);
59 if ((ListView_GetSelectedCount(hProcessPageListCtrl
) != 1) || (dwProcessId
== 0))
61 hProcessAffinityHandle
= OpenProcess(PROCESS_QUERY_INFORMATION
|PROCESS_SET_INFORMATION
, FALSE
, dwProcessId
);
62 if (!hProcessAffinityHandle
) {
63 GetLastErrorText(strErrorText
, 260);
64 MessageBox(hMainWnd
, strErrorText
, _T("Unable to Access or Set Process Affinity"), MB_OK
|MB_ICONSTOP
);
67 DialogBox(hInst
, MAKEINTRESOURCE(IDD_AFFINITY_DIALOG
), hMainWnd
, (DLGPROC
)AffinityDialogWndProc
);
68 if (hProcessAffinityHandle
) {
69 CloseHandle(hProcessAffinityHandle
);
70 hProcessAffinityHandle
= NULL
;
74 LRESULT CALLBACK
AffinityDialogWndProc(HWND hDlg
, UINT message
, WPARAM wParam
, LPARAM lParam
)
76 DWORD dwProcessAffinityMask
= 0;
77 DWORD dwSystemAffinityMask
= 0;
78 TCHAR strErrorText
[260];
84 * Get the current affinity mask for the process and
85 * the number of CPUs present in the system
87 if (!GetProcessAffinityMask(hProcessAffinityHandle
, &dwProcessAffinityMask
, &dwSystemAffinityMask
)) {
88 GetLastErrorText(strErrorText
, 260);
90 MessageBox(hMainWnd
, strErrorText
, _T("Unable to Access or Set Process Affinity"), MB_OK
|MB_ICONSTOP
);
94 * Enable a checkbox for each processor present in the system
96 if (dwSystemAffinityMask
& 0x00000001)
97 EnableWindow(GetDlgItem(hDlg
, IDC_CPU0
), TRUE
);
98 if (dwSystemAffinityMask
& 0x00000002)
99 EnableWindow(GetDlgItem(hDlg
, IDC_CPU1
), TRUE
);
100 if (dwSystemAffinityMask
& 0x00000004)
101 EnableWindow(GetDlgItem(hDlg
, IDC_CPU2
), TRUE
);
102 if (dwSystemAffinityMask
& 0x00000008)
103 EnableWindow(GetDlgItem(hDlg
, IDC_CPU3
), TRUE
);
104 if (dwSystemAffinityMask
& 0x00000010)
105 EnableWindow(GetDlgItem(hDlg
, IDC_CPU4
), TRUE
);
106 if (dwSystemAffinityMask
& 0x00000020)
107 EnableWindow(GetDlgItem(hDlg
, IDC_CPU5
), TRUE
);
108 if (dwSystemAffinityMask
& 0x00000040)
109 EnableWindow(GetDlgItem(hDlg
, IDC_CPU6
), TRUE
);
110 if (dwSystemAffinityMask
& 0x00000080)
111 EnableWindow(GetDlgItem(hDlg
, IDC_CPU7
), TRUE
);
112 if (dwSystemAffinityMask
& 0x00000100)
113 EnableWindow(GetDlgItem(hDlg
, IDC_CPU8
), TRUE
);
114 if (dwSystemAffinityMask
& 0x00000200)
115 EnableWindow(GetDlgItem(hDlg
, IDC_CPU9
), TRUE
);
116 if (dwSystemAffinityMask
& 0x00000400)
117 EnableWindow(GetDlgItem(hDlg
, IDC_CPU10
), TRUE
);
118 if (dwSystemAffinityMask
& 0x00000800)
119 EnableWindow(GetDlgItem(hDlg
, IDC_CPU11
), TRUE
);
120 if (dwSystemAffinityMask
& 0x00001000)
121 EnableWindow(GetDlgItem(hDlg
, IDC_CPU12
), TRUE
);
122 if (dwSystemAffinityMask
& 0x00002000)
123 EnableWindow(GetDlgItem(hDlg
, IDC_CPU13
), TRUE
);
124 if (dwSystemAffinityMask
& 0x00004000)
125 EnableWindow(GetDlgItem(hDlg
, IDC_CPU14
), TRUE
);
126 if (dwSystemAffinityMask
& 0x00008000)
127 EnableWindow(GetDlgItem(hDlg
, IDC_CPU15
), TRUE
);
128 if (dwSystemAffinityMask
& 0x00010000)
129 EnableWindow(GetDlgItem(hDlg
, IDC_CPU16
), TRUE
);
130 if (dwSystemAffinityMask
& 0x00020000)
131 EnableWindow(GetDlgItem(hDlg
, IDC_CPU17
), TRUE
);
132 if (dwSystemAffinityMask
& 0x00040000)
133 EnableWindow(GetDlgItem(hDlg
, IDC_CPU18
), TRUE
);
134 if (dwSystemAffinityMask
& 0x00080000)
135 EnableWindow(GetDlgItem(hDlg
, IDC_CPU19
), TRUE
);
136 if (dwSystemAffinityMask
& 0x00100000)
137 EnableWindow(GetDlgItem(hDlg
, IDC_CPU20
), TRUE
);
138 if (dwSystemAffinityMask
& 0x00200000)
139 EnableWindow(GetDlgItem(hDlg
, IDC_CPU21
), TRUE
);
140 if (dwSystemAffinityMask
& 0x00400000)
141 EnableWindow(GetDlgItem(hDlg
, IDC_CPU22
), TRUE
);
142 if (dwSystemAffinityMask
& 0x00800000)
143 EnableWindow(GetDlgItem(hDlg
, IDC_CPU23
), TRUE
);
144 if (dwSystemAffinityMask
& 0x01000000)
145 EnableWindow(GetDlgItem(hDlg
, IDC_CPU24
), TRUE
);
146 if (dwSystemAffinityMask
& 0x02000000)
147 EnableWindow(GetDlgItem(hDlg
, IDC_CPU25
), TRUE
);
148 if (dwSystemAffinityMask
& 0x04000000)
149 EnableWindow(GetDlgItem(hDlg
, IDC_CPU26
), TRUE
);
150 if (dwSystemAffinityMask
& 0x08000000)
151 EnableWindow(GetDlgItem(hDlg
, IDC_CPU27
), TRUE
);
152 if (dwSystemAffinityMask
& 0x10000000)
153 EnableWindow(GetDlgItem(hDlg
, IDC_CPU28
), TRUE
);
154 if (dwSystemAffinityMask
& 0x20000000)
155 EnableWindow(GetDlgItem(hDlg
, IDC_CPU29
), TRUE
);
156 if (dwSystemAffinityMask
& 0x40000000)
157 EnableWindow(GetDlgItem(hDlg
, IDC_CPU30
), TRUE
);
158 if (dwSystemAffinityMask
& 0x80000000)
159 EnableWindow(GetDlgItem(hDlg
, IDC_CPU31
), TRUE
);
163 * Check each checkbox that the current process
166 if (dwProcessAffinityMask
& 0x00000001)
167 SendMessage(GetDlgItem(hDlg
, IDC_CPU0
), BM_SETCHECK
, BST_CHECKED
, 0);
168 if (dwProcessAffinityMask
& 0x00000002)
169 SendMessage(GetDlgItem(hDlg
, IDC_CPU1
), BM_SETCHECK
, BST_CHECKED
, 0);
170 if (dwProcessAffinityMask
& 0x00000004)
171 SendMessage(GetDlgItem(hDlg
, IDC_CPU2
), BM_SETCHECK
, BST_CHECKED
, 0);
172 if (dwProcessAffinityMask
& 0x00000008)
173 SendMessage(GetDlgItem(hDlg
, IDC_CPU3
), BM_SETCHECK
, BST_CHECKED
, 0);
174 if (dwProcessAffinityMask
& 0x00000010)
175 SendMessage(GetDlgItem(hDlg
, IDC_CPU4
), BM_SETCHECK
, BST_CHECKED
, 0);
176 if (dwProcessAffinityMask
& 0x00000020)
177 SendMessage(GetDlgItem(hDlg
, IDC_CPU5
), BM_SETCHECK
, BST_CHECKED
, 0);
178 if (dwProcessAffinityMask
& 0x00000040)
179 SendMessage(GetDlgItem(hDlg
, IDC_CPU6
), BM_SETCHECK
, BST_CHECKED
, 0);
180 if (dwProcessAffinityMask
& 0x00000080)
181 SendMessage(GetDlgItem(hDlg
, IDC_CPU7
), BM_SETCHECK
, BST_CHECKED
, 0);
182 if (dwProcessAffinityMask
& 0x00000100)
183 SendMessage(GetDlgItem(hDlg
, IDC_CPU8
), BM_SETCHECK
, BST_CHECKED
, 0);
184 if (dwProcessAffinityMask
& 0x00000200)
185 SendMessage(GetDlgItem(hDlg
, IDC_CPU9
), BM_SETCHECK
, BST_CHECKED
, 0);
186 if (dwProcessAffinityMask
& 0x00000400)
187 SendMessage(GetDlgItem(hDlg
, IDC_CPU10
), BM_SETCHECK
, BST_CHECKED
, 0);
188 if (dwProcessAffinityMask
& 0x00000800)
189 SendMessage(GetDlgItem(hDlg
, IDC_CPU11
), BM_SETCHECK
, BST_CHECKED
, 0);
190 if (dwProcessAffinityMask
& 0x00001000)
191 SendMessage(GetDlgItem(hDlg
, IDC_CPU12
), BM_SETCHECK
, BST_CHECKED
, 0);
192 if (dwProcessAffinityMask
& 0x00002000)
193 SendMessage(GetDlgItem(hDlg
, IDC_CPU13
), BM_SETCHECK
, BST_CHECKED
, 0);
194 if (dwProcessAffinityMask
& 0x00004000)
195 SendMessage(GetDlgItem(hDlg
, IDC_CPU14
), BM_SETCHECK
, BST_CHECKED
, 0);
196 if (dwProcessAffinityMask
& 0x00008000)
197 SendMessage(GetDlgItem(hDlg
, IDC_CPU15
), BM_SETCHECK
, BST_CHECKED
, 0);
198 if (dwProcessAffinityMask
& 0x00010000)
199 SendMessage(GetDlgItem(hDlg
, IDC_CPU16
), BM_SETCHECK
, BST_CHECKED
, 0);
200 if (dwProcessAffinityMask
& 0x00020000)
201 SendMessage(GetDlgItem(hDlg
, IDC_CPU17
), BM_SETCHECK
, BST_CHECKED
, 0);
202 if (dwProcessAffinityMask
& 0x00040000)
203 SendMessage(GetDlgItem(hDlg
, IDC_CPU18
), BM_SETCHECK
, BST_CHECKED
, 0);
204 if (dwProcessAffinityMask
& 0x00080000)
205 SendMessage(GetDlgItem(hDlg
, IDC_CPU19
), BM_SETCHECK
, BST_CHECKED
, 0);
206 if (dwProcessAffinityMask
& 0x00100000)
207 SendMessage(GetDlgItem(hDlg
, IDC_CPU20
), BM_SETCHECK
, BST_CHECKED
, 0);
208 if (dwProcessAffinityMask
& 0x00200000)
209 SendMessage(GetDlgItem(hDlg
, IDC_CPU21
), BM_SETCHECK
, BST_CHECKED
, 0);
210 if (dwProcessAffinityMask
& 0x00400000)
211 SendMessage(GetDlgItem(hDlg
, IDC_CPU22
), BM_SETCHECK
, BST_CHECKED
, 0);
212 if (dwProcessAffinityMask
& 0x00800000)
213 SendMessage(GetDlgItem(hDlg
, IDC_CPU23
), BM_SETCHECK
, BST_CHECKED
, 0);
214 if (dwProcessAffinityMask
& 0x01000000)
215 SendMessage(GetDlgItem(hDlg
, IDC_CPU24
), BM_SETCHECK
, BST_CHECKED
, 0);
216 if (dwProcessAffinityMask
& 0x02000000)
217 SendMessage(GetDlgItem(hDlg
, IDC_CPU25
), BM_SETCHECK
, BST_CHECKED
, 0);
218 if (dwProcessAffinityMask
& 0x04000000)
219 SendMessage(GetDlgItem(hDlg
, IDC_CPU26
), BM_SETCHECK
, BST_CHECKED
, 0);
220 if (dwProcessAffinityMask
& 0x08000000)
221 SendMessage(GetDlgItem(hDlg
, IDC_CPU27
), BM_SETCHECK
, BST_CHECKED
, 0);
222 if (dwProcessAffinityMask
& 0x10000000)
223 SendMessage(GetDlgItem(hDlg
, IDC_CPU28
), BM_SETCHECK
, BST_CHECKED
, 0);
224 if (dwProcessAffinityMask
& 0x20000000)
225 SendMessage(GetDlgItem(hDlg
, IDC_CPU29
), BM_SETCHECK
, BST_CHECKED
, 0);
226 if (dwProcessAffinityMask
& 0x40000000)
227 SendMessage(GetDlgItem(hDlg
, IDC_CPU30
), BM_SETCHECK
, BST_CHECKED
, 0);
228 if (dwProcessAffinityMask
& 0x80000000)
229 SendMessage(GetDlgItem(hDlg
, IDC_CPU31
), BM_SETCHECK
, BST_CHECKED
, 0);
236 * If the user has cancelled the dialog box
239 if (LOWORD(wParam
) == IDCANCEL
) {
240 EndDialog(hDlg
, LOWORD(wParam
));
245 * The user has clicked OK -- so now we have
246 * to adjust the process affinity mask
248 if (LOWORD(wParam
) == IDOK
) {
250 * First we have to create a mask out of each
251 * checkbox that the user checked.
253 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU0
), BM_GETCHECK
, 0, 0))
254 dwProcessAffinityMask
|= 0x00000001;
255 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU1
), BM_GETCHECK
, 0, 0))
256 dwProcessAffinityMask
|= 0x00000002;
257 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU2
), BM_GETCHECK
, 0, 0))
258 dwProcessAffinityMask
|= 0x00000004;
259 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU3
), BM_GETCHECK
, 0, 0))
260 dwProcessAffinityMask
|= 0x00000008;
261 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU4
), BM_GETCHECK
, 0, 0))
262 dwProcessAffinityMask
|= 0x00000010;
263 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU5
), BM_GETCHECK
, 0, 0))
264 dwProcessAffinityMask
|= 0x00000020;
265 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU6
), BM_GETCHECK
, 0, 0))
266 dwProcessAffinityMask
|= 0x00000040;
267 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU7
), BM_GETCHECK
, 0, 0))
268 dwProcessAffinityMask
|= 0x00000080;
269 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU8
), BM_GETCHECK
, 0, 0))
270 dwProcessAffinityMask
|= 0x00000100;
271 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU9
), BM_GETCHECK
, 0, 0))
272 dwProcessAffinityMask
|= 0x00000200;
273 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU10
), BM_GETCHECK
, 0, 0))
274 dwProcessAffinityMask
|= 0x00000400;
275 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU11
), BM_GETCHECK
, 0, 0))
276 dwProcessAffinityMask
|= 0x00000800;
277 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU12
), BM_GETCHECK
, 0, 0))
278 dwProcessAffinityMask
|= 0x00001000;
279 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU13
), BM_GETCHECK
, 0, 0))
280 dwProcessAffinityMask
|= 0x00002000;
281 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU14
), BM_GETCHECK
, 0, 0))
282 dwProcessAffinityMask
|= 0x00004000;
283 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU15
), BM_GETCHECK
, 0, 0))
284 dwProcessAffinityMask
|= 0x00008000;
285 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU16
), BM_GETCHECK
, 0, 0))
286 dwProcessAffinityMask
|= 0x00010000;
287 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU17
), BM_GETCHECK
, 0, 0))
288 dwProcessAffinityMask
|= 0x00020000;
289 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU18
), BM_GETCHECK
, 0, 0))
290 dwProcessAffinityMask
|= 0x00040000;
291 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU19
), BM_GETCHECK
, 0, 0))
292 dwProcessAffinityMask
|= 0x00080000;
293 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU20
), BM_GETCHECK
, 0, 0))
294 dwProcessAffinityMask
|= 0x00100000;
295 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU21
), BM_GETCHECK
, 0, 0))
296 dwProcessAffinityMask
|= 0x00200000;
297 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU22
), BM_GETCHECK
, 0, 0))
298 dwProcessAffinityMask
|= 0x00400000;
299 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU23
), BM_GETCHECK
, 0, 0))
300 dwProcessAffinityMask
|= 0x00800000;
301 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU24
), BM_GETCHECK
, 0, 0))
302 dwProcessAffinityMask
|= 0x01000000;
303 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU25
), BM_GETCHECK
, 0, 0))
304 dwProcessAffinityMask
|= 0x02000000;
305 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU26
), BM_GETCHECK
, 0, 0))
306 dwProcessAffinityMask
|= 0x04000000;
307 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU27
), BM_GETCHECK
, 0, 0))
308 dwProcessAffinityMask
|= 0x08000000;
309 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU28
), BM_GETCHECK
, 0, 0))
310 dwProcessAffinityMask
|= 0x10000000;
311 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU29
), BM_GETCHECK
, 0, 0))
312 dwProcessAffinityMask
|= 0x20000000;
313 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU30
), BM_GETCHECK
, 0, 0))
314 dwProcessAffinityMask
|= 0x40000000;
315 if (SendMessage(GetDlgItem(hDlg
, IDC_CPU31
), BM_GETCHECK
, 0, 0))
316 dwProcessAffinityMask
|= 0x80000000;
319 * Make sure they are giving the process affinity
320 * with at least one processor. I'd hate to see a
321 * process that is not in a wait state get deprived
324 if (!dwProcessAffinityMask
) {
325 MessageBox(hDlg
, _T("The process must have affinity with at least one processor."), _T("Invalid Option"), MB_OK
|MB_ICONSTOP
);
330 * Try to set the process affinity
332 if (!SetProcessAffinityMask(hProcessAffinityHandle
, dwProcessAffinityMask
)) {
333 GetLastErrorText(strErrorText
, 260);
334 EndDialog(hDlg
, LOWORD(wParam
));
335 MessageBox(hMainWnd
, strErrorText
, _T("Unable to Access or Set Process Affinity"), MB_OK
|MB_ICONSTOP
);
338 EndDialog(hDlg
, LOWORD(wParam
));