[ACPPAGE] Make use of the PCH.
[reactos.git] / dll / cpl / inetcpl / security.c
1 /*
2 * Internet control panel applet: security propsheet
3 *
4 * Copyright 2011 Detlef Riekenberg
5 *
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19 *
20 */
21
22 #include "inetcpl.h"
23
24 typedef struct secdlg_data_s {
25 HWND hsec; /* security propsheet */
26 HWND hlv; /* listview */
27 HWND htb; /* trackbar */
28 IInternetSecurityManager *sec_mgr;
29 IInternetZoneManager *zone_mgr;
30 DWORD zone_enumerator;
31 DWORD num_zones;
32 ZONEATTRIBUTES *zone_attr;
33 DWORD *zones;
34 DWORD *levels;
35 HIMAGELIST himages;
36 DWORD last_lv_index;
37 DWORD last_level;
38 } secdlg_data;
39
40 #define NUM_TRACKBAR_POS 5
41
42 static DWORD url_templates[] = {URLTEMPLATE_CUSTOM,
43 URLTEMPLATE_LOW,
44 URLTEMPLATE_MEDLOW,
45 URLTEMPLATE_MEDIUM,
46 URLTEMPLATE_MEDHIGH,
47 URLTEMPLATE_HIGH};
48
49 /*********************************************************************
50 * index_from_urltemplate [internal]
51 *
52 */
53 static DWORD index_from_urltemplate(URLTEMPLATE value)
54 {
55
56 DWORD index = sizeof(url_templates) / sizeof(url_templates[0]);
57
58 while((index > 0) && (url_templates[index-1] != value))
59 index--;
60
61 index--; /* table entries are 0 based */
62 if (!index && value)
63 FIXME("URLTEMPLATE 0x%x not supported\n", value);
64
65 TRACE("URLTEMPLATE 0x%08x=> Level %d\n", value, index);
66 return index;
67 }
68
69 /*********************************************************************
70 * update_security_level [internal]
71 *
72 */
73 static void update_security_level(secdlg_data *sd, DWORD lv_index, DWORD tb_index)
74 {
75 WCHAR name[512];
76 DWORD current_index;
77
78 TRACE("(%p, lv_index: %u, tb_index: %u)\n", sd, lv_index, tb_index);
79
80 if ((sd->levels[lv_index] != sd->last_level) || (tb_index > 0)) {
81 /* show or hide the trackbar */
82 if (!sd->levels[lv_index] || !sd->last_level)
83 ShowWindow(sd->htb, sd->levels[lv_index] ? SW_NORMAL : SW_HIDE);
84
85 current_index = (tb_index > 0) ? tb_index : index_from_urltemplate(sd->levels[lv_index]);
86
87 name[0] = 0;
88 LoadStringW(hcpl, IDS_SEC_LEVEL0 + current_index, name, sizeof(name)/sizeof(name[0]));
89 TRACE("new level #%d: %s\n", current_index, debugstr_w(name));
90 SetWindowTextW(GetDlgItem(sd->hsec, IDC_SEC_LEVEL), name);
91
92 name[0] = 0;
93 LoadStringW(hcpl, IDS_SEC_LEVEL0_INFO + (current_index * 0x10), name, sizeof(name)/sizeof(name[0]));
94 TRACE("new level info: %s\n", debugstr_w(name));
95 SetWindowTextW(GetDlgItem(sd->hsec, IDC_SEC_LEVEL_INFO), name);
96
97 if (current_index)
98 SendMessageW(sd->htb, TBM_SETPOS, TRUE, NUM_TRACKBAR_POS - current_index);
99
100 sd->last_level = sd->levels[lv_index];
101
102 }
103 }
104
105 /*********************************************************************
106 * update_zone_info [internal]
107 *
108 */
109 static void update_zone_info(secdlg_data *sd, DWORD lv_index)
110 {
111 ZONEATTRIBUTES *za = &sd->zone_attr[lv_index];
112 WCHAR name[MAX_PATH];
113 DWORD len;
114
115 SetWindowTextW(GetDlgItem(sd->hsec, IDC_SEC_ZONE_INFO), za->szDescription);
116
117 len = LoadStringW(hcpl, IDS_SEC_SETTINGS, name, sizeof(name)/sizeof(*name));
118 lstrcpynW(&name[len], za->szDisplayName, sizeof(name)/sizeof(*name) - len - 1);
119
120 TRACE("new title: %s\n", debugstr_w(name));
121 SetWindowTextW(GetDlgItem(sd->hsec, IDC_SEC_GROUP), name);
122
123 update_security_level(sd, lv_index, 0);
124 sd->last_lv_index = lv_index;
125 }
126
127 /*********************************************************************
128 * add_zone_to_listview [internal]
129 *
130 */
131 static void add_zone_to_listview(secdlg_data *sd, DWORD *pindex, DWORD zone)
132 {
133 DWORD lv_index = *pindex;
134 ZONEATTRIBUTES *za = &sd->zone_attr[lv_index];
135 LVITEMW lvitem;
136 HRESULT hr;
137 INT iconid = 0;
138 HMODULE hdll = NULL;
139 WCHAR * ptr;
140 HICON icon;
141
142 TRACE("item %d (zone %d)\n", lv_index, zone);
143
144 sd->zones[lv_index] = zone;
145
146 memset(&lvitem, 0, sizeof(LVITEMW));
147 memset(za, 0, sizeof(ZONEATTRIBUTES));
148 za->cbSize = sizeof(ZONEATTRIBUTES);
149 hr = IInternetZoneManager_GetZoneAttributes(sd->zone_mgr, zone, za);
150 if (SUCCEEDED(hr)) {
151 TRACE("displayname: %s\n", debugstr_w(za->szDisplayName));
152 TRACE("description: %s\n", debugstr_w(za->szDescription));
153 TRACE("minlevel: 0x%x, recommended: 0x%x, current: 0x%x (flags: 0x%x)\n", za->dwTemplateMinLevel,
154 za->dwTemplateRecommended, za->dwTemplateCurrentLevel, za->dwFlags);
155
156 if (za->dwFlags & ZAFLAGS_NO_UI ) {
157 TRACE("item %d (zone %d): UI disabled for %s\n", lv_index, zone, debugstr_w(za->szDisplayName));
158 return;
159 }
160
161 sd->levels[lv_index] = za->dwTemplateCurrentLevel;
162
163 lvitem.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;
164 lvitem.iItem = lv_index;
165 lvitem.iSubItem = 0;
166 lvitem.pszText = za->szDisplayName;
167 lvitem.lParam = (LPARAM) zone;
168
169 /* format is "filename.ext#iconid" */
170 ptr = StrChrW(za->szIconPath, '#');
171 if (ptr) {
172 *ptr = 0;
173 ptr++;
174 iconid = StrToIntW(ptr);
175 hdll = LoadLibraryExW(za->szIconPath, NULL, LOAD_LIBRARY_AS_DATAFILE);
176 TRACE("%p: icon #%d from %s\n", hdll, iconid, debugstr_w(za->szIconPath));
177
178 icon = LoadImageW(hdll, MAKEINTRESOURCEW(iconid), IMAGE_ICON, GetSystemMetrics(SM_CXICON),
179 GetSystemMetrics(SM_CYICON), LR_SHARED);
180
181 if (!icon) {
182 FIXME("item %d (zone %d): missing icon #%d in %s\n", lv_index, zone, iconid, debugstr_w(za->szIconPath));
183 }
184
185 /* the failure result (NULL) from LoadImageW let ImageList_AddIcon fail
186 with -1, which is reused in ListView_InsertItemW to disable the image */
187 lvitem.iImage = ImageList_AddIcon(sd->himages, icon);
188 }
189 else
190 FIXME("item %d (zone %d): malformed szIconPath %s\n", lv_index, zone, debugstr_w(za->szIconPath));
191
192 if (ListView_InsertItemW(sd->hlv, &lvitem) >= 0) {
193 /* activate first item in the listview */
194 if (! lv_index) {
195 lvitem.state = LVIS_FOCUSED | LVIS_SELECTED;
196 lvitem.stateMask = LVIS_FOCUSED | LVIS_SELECTED;
197 SendMessageW(sd->hlv, LVM_SETITEMSTATE, 0, (LPARAM) &lvitem);
198 sd->last_level = ~0;
199 update_zone_info(sd, lv_index);
200 }
201 (*pindex)++;
202 }
203 FreeLibrary(hdll);
204 }
205 else
206 FIXME("item %d (zone %d): GetZoneAttributes failed with 0x%x\n", lv_index, zone, hr);
207 }
208
209 /*********************************************************************
210 * security_cleanup_zones [internal]
211 *
212 */
213 static void security_cleanup_zones(secdlg_data *sd)
214 {
215 if (sd->zone_enumerator) {
216 IInternetZoneManager_DestroyZoneEnumerator(sd->zone_mgr, sd->zone_enumerator);
217 }
218
219 if (sd->zone_mgr) {
220 IInternetZoneManager_Release(sd->zone_mgr);
221 }
222
223 if (sd->sec_mgr) {
224 IInternetSecurityManager_Release(sd->sec_mgr);
225 }
226 }
227
228 /*********************************************************************
229 * security_enum_zones [internal]
230 *
231 */
232 static HRESULT security_enum_zones(secdlg_data * sd)
233 {
234 HRESULT hr;
235
236 hr = CoInternetCreateSecurityManager(NULL, &sd->sec_mgr, 0);
237 if (SUCCEEDED(hr)) {
238 hr = CoInternetCreateZoneManager(NULL, &sd->zone_mgr, 0);
239 if (SUCCEEDED(hr)) {
240 hr = IInternetZoneManager_CreateZoneEnumerator(sd->zone_mgr, &sd->zone_enumerator, &sd->num_zones, 0);
241 }
242 }
243 return hr;
244 }
245
246 /*********************************************************************
247 * security_on_destroy [internal]
248 *
249 * handle WM_NCDESTROY
250 *
251 */
252 static INT_PTR security_on_destroy(secdlg_data * sd)
253 {
254 TRACE("(%p)\n", sd);
255
256 heap_free(sd->zone_attr);
257 heap_free(sd->zones);
258 if (sd->himages) {
259 SendMessageW(sd->hlv, LVM_SETIMAGELIST, LVSIL_NORMAL, 0);
260 ImageList_Destroy(sd->himages);
261 }
262
263 security_cleanup_zones(sd);
264 SetWindowLongPtrW(sd->hsec, DWLP_USER, 0);
265 heap_free(sd);
266 return TRUE;
267 }
268
269 /*********************************************************************
270 * security_on_initdialog [internal]
271 *
272 * handle WM_INITDIALOG
273 *
274 */
275 static INT_PTR security_on_initdialog(HWND hsec)
276 {
277 secdlg_data *sd;
278 HRESULT hr;
279 DWORD current_zone;
280 DWORD lv_index = 0;
281 DWORD i;
282
283 sd = heap_alloc_zero(sizeof(secdlg_data));
284 SetWindowLongPtrW(hsec, DWLP_USER, (LONG_PTR) sd);
285 if (!sd) {
286 return FALSE;
287 }
288
289 sd->hsec = hsec;
290 sd->hlv = GetDlgItem(hsec, IDC_SEC_LISTVIEW);
291 sd->htb = GetDlgItem(hsec, IDC_SEC_TRACKBAR);
292
293 EnableWindow(sd->htb, FALSE); /* not changeable yet */
294
295 TRACE("(%p) (data: %p, listview: %p, trackbar: %p)\n", hsec, sd, sd->hlv, sd->htb);
296
297 SendMessageW(sd->htb, TBM_SETRANGE, FALSE, MAKELONG(0, NUM_TRACKBAR_POS - 1));
298 SendMessageW(sd->htb, TBM_SETTICFREQ, 1, 0 );
299
300 /* Create the image lists for the listview */
301 sd->himages = ImageList_Create(GetSystemMetrics(SM_CXICON), GetSystemMetrics(SM_CYICON), ILC_COLOR32 | ILC_MASK, 1, 1);
302
303 TRACE("using imagelist: %p\n", sd->himages);
304 if (!sd->himages) {
305 ERR("ImageList_Create failed!\n");
306 return FALSE;
307 }
308 SendMessageW(sd->hlv, LVM_SETIMAGELIST, LVSIL_NORMAL, (LPARAM)sd->himages);
309
310 hr = security_enum_zones(sd);
311 if (FAILED(hr)) {
312 ERR("got 0x%x\n", hr);
313 security_on_destroy(sd);
314 return FALSE;
315 }
316
317 TRACE("found %d zones\n", sd->num_zones);
318
319 /* remember ZONEATTRIBUTES for a listview entry */
320 sd->zone_attr = heap_alloc(sizeof(ZONEATTRIBUTES) * sd->num_zones);
321 if (!sd->zone_attr) {
322 security_on_destroy(sd);
323 return FALSE;
324 }
325
326 /* remember zone number and current security level for a listview entry */
327 sd->zones = heap_alloc((sizeof(DWORD) + sizeof(DWORD)) * sd->num_zones);
328 if (!sd->zones) {
329 security_on_destroy(sd);
330 return FALSE;
331 }
332 sd->levels = &sd->zones[sd->num_zones];
333
334 /* use the same order as visible with native inetcpl.cpl */
335 add_zone_to_listview(sd, &lv_index, URLZONE_INTERNET);
336 add_zone_to_listview(sd, &lv_index, URLZONE_INTRANET);
337 add_zone_to_listview(sd, &lv_index, URLZONE_TRUSTED);
338 add_zone_to_listview(sd, &lv_index, URLZONE_UNTRUSTED);
339
340 for (i = 0; i < sd->num_zones; i++)
341 {
342 hr = IInternetZoneManager_GetZoneAt(sd->zone_mgr, sd->zone_enumerator, i, &current_zone);
343 if (SUCCEEDED(hr) && (current_zone != (DWORD)URLZONE_INVALID)) {
344 if (!current_zone || (current_zone > URLZONE_UNTRUSTED)) {
345 add_zone_to_listview(sd, &lv_index, current_zone);
346 }
347 }
348 }
349 return TRUE;
350 }
351
352 /*********************************************************************
353 * security_on_notify [internal]
354 *
355 * handle WM_NOTIFY
356 *
357 */
358 static INT_PTR security_on_notify(secdlg_data *sd, WPARAM wparam, LPARAM lparam)
359 {
360 NMLISTVIEW *nm;
361
362 nm = (NMLISTVIEW *) lparam;
363 switch (nm->hdr.code)
364 {
365 case LVN_ITEMCHANGED:
366 TRACE("LVN_ITEMCHANGED (0x%lx, 0x%lx) from %p with code: %d (item: %d, uNewState: %u)\n",
367 wparam, lparam, nm->hdr.hwndFrom, nm->hdr.code, nm->iItem, nm->uNewState);
368 if ((nm->uNewState & LVIS_SELECTED) == LVIS_SELECTED) {
369 update_zone_info(sd, nm->iItem);
370 }
371 break;
372
373 case PSN_APPLY:
374 TRACE("PSN_APPLY (0x%lx, 0x%lx) from %p with code: %d\n", wparam, lparam,
375 nm->hdr.hwndFrom, nm->hdr.code);
376 break;
377
378 default:
379 TRACE("WM_NOTIFY (0x%lx, 0x%lx) from %p with code: %d\n", wparam, lparam,
380 nm->hdr.hwndFrom, nm->hdr.code);
381
382 }
383 return FALSE;
384 }
385
386 /*********************************************************************
387 * security_dlgproc [internal]
388 *
389 */
390 INT_PTR CALLBACK security_dlgproc(HWND hwnd, UINT msg, WPARAM wparam, LPARAM lparam)
391 {
392 secdlg_data *sd;
393
394 if (msg == WM_INITDIALOG) {
395 return security_on_initdialog(hwnd);
396 }
397
398 sd = (secdlg_data *)GetWindowLongPtrW(hwnd, DWLP_USER);
399 if (sd) {
400 switch (msg)
401 {
402 case WM_NOTIFY:
403 return security_on_notify(sd, wparam, lparam);
404
405 case WM_NCDESTROY:
406 return security_on_destroy(sd);
407
408 default:
409 /* do not flood the log */
410 if ((msg == WM_SETCURSOR) || (msg == WM_NCHITTEST) ||
411 (msg == WM_MOUSEMOVE) || (msg == WM_MOUSEACTIVATE) || (msg == WM_PARENTNOTIFY))
412 return FALSE;
413
414 TRACE("(%p, 0x%08x/%03d, 0x%08lx, 0x%08lx)\n", hwnd, msg, msg, wparam, lparam);
415 }
416 }
417 return FALSE;
418 }