9955e83c7a7887187d78c79acbbf717d5aaa8654
[reactos.git] / reactos / dll / win32 / shell32 / dialogs.c
1 /*
2 * common shell dialogs
3 *
4 * Copyright 2000 Juergen Schmied
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 #include <precomp.h>
22
23
24 typedef struct
25 {
26 HWND hwndOwner ;
27 HICON hIcon ;
28 LPCSTR lpstrDirectory ;
29 LPCSTR lpstrTitle ;
30 LPCSTR lpstrDescription ;
31 UINT uFlags ;
32 } RUNFILEDLGPARAMS ;
33
34 typedef BOOL (*LPFNOFN) (OPENFILENAMEA *) ;
35
36 WINE_DEFAULT_DEBUG_CHANNEL(shell);
37 static INT_PTR CALLBACK RunDlgProc (HWND, UINT, WPARAM, LPARAM) ;
38 static void FillList (HWND, char *) ;
39
40
41 /*************************************************************************
42 * PickIconDlg [SHELL32.62]
43 *
44 */
45
46 typedef struct
47 {
48 HMODULE hLibrary;
49 HWND hDlgCtrl;
50 WCHAR szName[MAX_PATH];
51 INT Index;
52 }PICK_ICON_CONTEXT, *PPICK_ICON_CONTEXT;
53
54 BOOL CALLBACK EnumPickIconResourceProc(HMODULE hModule,
55 LPCWSTR lpszType,
56 LPWSTR lpszName,
57 LONG_PTR lParam
58 )
59 {
60 WCHAR szName[100];
61 int index;
62 HICON hIcon;
63 PPICK_ICON_CONTEXT pIconContext = (PPICK_ICON_CONTEXT)lParam;
64
65 if (IS_INTRESOURCE(lpszName))
66 swprintf(szName, L"%u", lpszName);
67 else
68 wcscpy(szName, (WCHAR*)lpszName);
69
70
71 hIcon = LoadIconW(pIconContext->hLibrary, (LPCWSTR)lpszName);
72 if (hIcon == NULL)
73 return TRUE;
74
75 index = SendMessageW(pIconContext->hDlgCtrl, LB_ADDSTRING, 0, (LPARAM)szName);
76 if (index != LB_ERR)
77 SendMessageW(pIconContext->hDlgCtrl, LB_SETITEMDATA, index, (LPARAM)hIcon);
78
79 return TRUE;
80 }
81
82 void
83 DestroyIconList(HWND hDlgCtrl)
84 {
85 int count;
86 int index;
87
88 count = SendMessage(hDlgCtrl, LB_GETCOUNT, 0, 0);
89 if (count == LB_ERR)
90 return;
91
92 for(index = 0; index < count; index++)
93 {
94 HICON hIcon = (HICON)SendMessageW(hDlgCtrl, LB_GETITEMDATA, index, 0);
95 DestroyIcon(hIcon);
96 }
97 }
98
99 INT_PTR CALLBACK PickIconProc(HWND hwndDlg,
100 UINT uMsg,
101 WPARAM wParam,
102 LPARAM lParam
103 )
104 {
105 LPMEASUREITEMSTRUCT lpmis;
106 LPDRAWITEMSTRUCT lpdis;
107 HICON hIcon;
108 INT index, count;
109 WCHAR szText[MAX_PATH], szTitle[100], szFilter[100];
110 OPENFILENAMEW ofn = {0};
111
112 PPICK_ICON_CONTEXT pIconContext = (PPICK_ICON_CONTEXT)GetWindowLongPtr(hwndDlg, DWLP_USER);
113
114 switch(uMsg)
115 {
116 case WM_INITDIALOG:
117 pIconContext = (PPICK_ICON_CONTEXT)lParam;
118 SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG)pIconContext);
119 pIconContext->hDlgCtrl = GetDlgItem(hwndDlg, IDC_PICKICON_LIST);
120 EnumResourceNamesW(pIconContext->hLibrary, RT_ICON, EnumPickIconResourceProc, (LPARAM)pIconContext);
121 if (PathUnExpandEnvStringsW(pIconContext->szName, szText, MAX_PATH))
122 SendDlgItemMessageW(hwndDlg, IDC_EDIT_PATH, WM_SETTEXT, 0, (LPARAM)szText);
123 else
124 SendDlgItemMessageW(hwndDlg, IDC_EDIT_PATH, WM_SETTEXT, 0, (LPARAM)pIconContext->szName);
125
126 count = SendMessage(pIconContext->hDlgCtrl, LB_GETCOUNT, 0, 0);
127 if (count != LB_ERR)
128 {
129 if (count > pIconContext->Index)
130 SendMessageW(pIconContext->hDlgCtrl, LB_SETCURSEL, pIconContext->Index, 0);
131 else
132 SendMessageW(pIconContext->hDlgCtrl, LB_SETCURSEL, 0, 0);
133 }
134 return TRUE;
135 case WM_COMMAND:
136 switch(LOWORD(wParam))
137 {
138 case IDOK:
139 index = SendMessageW(pIconContext->hDlgCtrl, LB_GETCURSEL, 0, 0);
140 pIconContext->Index = index;
141 SendDlgItemMessageW(hwndDlg, IDC_EDIT_PATH, WM_GETTEXT, MAX_PATH, (LPARAM)pIconContext->szName);
142 DestroyIconList(pIconContext->hDlgCtrl);
143 EndDialog(hwndDlg, 1);
144 break;
145 case IDCANCEL:
146 DestroyIconList(pIconContext->hDlgCtrl);
147 EndDialog(hwndDlg, 0);
148 break;
149 case IDC_PICKICON_LIST:
150 if (HIWORD(wParam) == LBN_SELCHANGE)
151 InvalidateRect((HWND)lParam, NULL, TRUE); // FIXME USE UPDATE RECT
152 break;
153 case IDC_BUTTON_PATH:
154 szText[0] = 0;
155 szTitle[0] = 0;
156 szFilter[0] = 0;
157 ofn.lStructSize = sizeof(ofn);
158 ofn.hwndOwner = hwndDlg;
159 ofn.lpstrFile = szText;
160 ofn.nMaxFile = MAX_PATH;
161 LoadStringW(shell32_hInstance, IDS_PICK_ICON_TITLE, szTitle, sizeof(szTitle) / sizeof(WCHAR));
162 ofn.lpstrTitle = szTitle;
163 LoadStringW(shell32_hInstance, IDS_PICK_ICON_FILTER, szFilter, sizeof(szFilter) / sizeof(WCHAR));
164 ofn.lpstrFilter = szFilter;
165 if (GetOpenFileNameW(&ofn))
166 {
167 HMODULE hLibrary;
168
169 if (!wcsicmp(pIconContext->szName, szText))
170 break;
171
172 DestroyIconList(pIconContext->hDlgCtrl);
173
174 hLibrary = LoadLibraryExW(szText, NULL, LOAD_LIBRARY_AS_DATAFILE | LOAD_LIBRARY_AS_IMAGE_RESOURCE);
175 if (hLibrary == NULL)
176 break;
177 FreeLibrary(pIconContext->hLibrary);
178 pIconContext->hLibrary = hLibrary;
179 wcscpy(pIconContext->szName, szText);
180 EnumResourceNamesW(pIconContext->hLibrary, RT_ICON, EnumPickIconResourceProc, (LPARAM)pIconContext);
181 if (PathUnExpandEnvStringsW(pIconContext->szName, szText, MAX_PATH))
182 SendDlgItemMessageW(hwndDlg, IDC_EDIT_PATH, WM_SETTEXT, 0, (LPARAM)szText);
183 else
184 SendDlgItemMessageW(hwndDlg, IDC_EDIT_PATH, WM_SETTEXT, 0, (LPARAM)pIconContext->szName);
185
186 SendMessageW(pIconContext->hDlgCtrl, LB_SETCURSEL, 0, 0);
187 }
188 break;
189 }
190 break;
191 case WM_MEASUREITEM:
192 lpmis = (LPMEASUREITEMSTRUCT) lParam;
193 lpmis->itemHeight = 32;
194 lpmis->itemWidth = 64;
195 return TRUE;
196 case WM_DRAWITEM:
197 lpdis = (LPDRAWITEMSTRUCT) lParam;
198 if (lpdis->itemID == -1)
199 {
200 break;
201 }
202 switch (lpdis->itemAction)
203 {
204 case ODA_SELECT:
205 case ODA_DRAWENTIRE:
206 index = SendMessageW(pIconContext->hDlgCtrl, LB_GETCURSEL, 0, 0);
207 hIcon =(HICON)SendMessage(lpdis->hwndItem, LB_GETITEMDATA, lpdis->itemID, (LPARAM) 0);
208
209 if (lpdis->itemID == index)
210 {
211 HBRUSH hBrush;
212 hBrush = CreateSolidBrush(RGB(0, 0, 255));
213 FillRect(lpdis->hDC, &lpdis->rcItem, hBrush);
214 DeleteObject(hBrush);
215 }
216 else
217 {
218 HBRUSH hBrush;
219 hBrush = CreateSolidBrush(RGB(255, 255, 255));
220 FillRect(lpdis->hDC, &lpdis->rcItem, hBrush);
221 DeleteObject(hBrush);
222 }
223 DrawIconEx(lpdis->hDC, lpdis->rcItem.left,lpdis->rcItem.top, hIcon,
224 0,
225 0,
226 0,
227 NULL,
228 DI_NORMAL);
229 break;
230 }
231 break;
232 }
233
234 return FALSE;
235 }
236
237 BOOL WINAPI PickIconDlg(
238 HWND hwndOwner,
239 LPWSTR lpstrFile,
240 UINT nMaxFile,
241 INT* lpdwIconIndex)
242 {
243 HMODULE hLibrary;
244 int res;
245 PICK_ICON_CONTEXT IconContext;
246
247 hLibrary = LoadLibraryExW(lpstrFile, NULL, LOAD_LIBRARY_AS_DATAFILE | LOAD_LIBRARY_AS_IMAGE_RESOURCE);
248 IconContext.hLibrary = hLibrary;
249 IconContext.Index = *lpdwIconIndex;
250 wcscpy(IconContext.szName, lpstrFile);
251
252 res = DialogBoxParamW(shell32_hInstance, MAKEINTRESOURCEW(IDD_PICK_ICON_DIALOG), hwndOwner, PickIconProc, (LPARAM)&IconContext);
253 if (res)
254 {
255 wcscpy(lpstrFile, IconContext.szName);
256 *lpdwIconIndex = IconContext.Index;
257 }
258
259 FreeLibrary(hLibrary);
260 return res;
261 }
262
263 /*************************************************************************
264 * RunFileDlg [SHELL32.61]
265 *
266 * NOTES
267 * Original name: RunFileDlg (exported by ordinal)
268 */
269 void WINAPI RunFileDlg(
270 HWND hwndOwner,
271 HICON hIcon,
272 LPCSTR lpstrDirectory,
273 LPCSTR lpstrTitle,
274 LPCSTR lpstrDescription,
275 UINT uFlags)
276 {
277
278 RUNFILEDLGPARAMS rfdp;
279 HRSRC hRes;
280 LPVOID template;
281 TRACE("\n");
282
283 rfdp.hwndOwner = hwndOwner;
284 rfdp.hIcon = hIcon;
285 rfdp.lpstrDirectory = lpstrDirectory;
286 rfdp.lpstrTitle = lpstrTitle;
287 rfdp.lpstrDescription = lpstrDescription;
288 rfdp.uFlags = uFlags;
289
290 if(!(hRes = FindResourceA(shell32_hInstance, "SHELL_RUN_DLG", (LPSTR)RT_DIALOG)))
291 {
292 MessageBoxA (hwndOwner, "Couldn't find dialog.", "Nix", MB_OK) ;
293 return;
294 }
295 if(!(template = (LPVOID)LoadResource(shell32_hInstance, hRes)))
296 {
297 MessageBoxA (hwndOwner, "Couldn't load dialog.", "Nix", MB_OK) ;
298 return;
299 }
300
301 DialogBoxIndirectParamA((HINSTANCE)GetWindowLongPtrW( hwndOwner,
302 GWLP_HINSTANCE ),
303 template, hwndOwner, RunDlgProc, (LPARAM)&rfdp);
304
305 }
306
307 /* Dialog procedure for RunFileDlg */
308 static INT_PTR CALLBACK RunDlgProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
309 {
310 int ic ;
311 char *psz, *pdir, szMsg[256];
312 static RUNFILEDLGPARAMS *prfdp = NULL ;
313
314 switch (message)
315 {
316 case WM_INITDIALOG :
317 prfdp = (RUNFILEDLGPARAMS *)lParam ;
318
319 if (prfdp->lpstrTitle)
320 SetWindowTextA (hwnd, prfdp->lpstrTitle) ;
321
322 SetClassLongPtrW (hwnd, GCLP_HICON, (LPARAM)prfdp->hIcon) ;
323 SendMessageW (GetDlgItem (hwnd, 12297), STM_SETICON,
324 (WPARAM)LoadIconW (NULL, (LPCWSTR)IDI_WINLOGO), 0);
325 FillList (GetDlgItem (hwnd, 12298), NULL) ;
326 SetFocus (GetDlgItem (hwnd, 12298)) ;
327 return TRUE ;
328
329 case WM_COMMAND :
330 switch (LOWORD (wParam))
331 {
332 case IDOK :
333 {
334 HWND htxt = NULL ;
335 if ((ic = GetWindowTextLengthA (htxt = GetDlgItem (hwnd, 12298))))
336 {
337 psz = HeapAlloc( GetProcessHeap(), 0, (ic + 2) );
338 GetWindowTextA (htxt, psz, ic + 1) ;
339 pdir = HeapAlloc( GetProcessHeap(), 0, (ic + 2) );
340 if (pdir)
341 {
342 char * ptr;
343 strcpy(pdir, psz);
344 ptr = strrchr(pdir + 4, '\\');
345 if(ptr)
346 ptr[0] = '\0';
347 else
348 pdir[3] = '\0';
349 }
350 if (ShellExecuteA(NULL, NULL, psz, NULL, pdir, SW_SHOWNORMAL) < (HINSTANCE)33)
351 {
352 char *pszSysMsg = NULL ;
353 FormatMessageA (
354 FORMAT_MESSAGE_ALLOCATE_BUFFER |
355 FORMAT_MESSAGE_FROM_SYSTEM |
356 FORMAT_MESSAGE_IGNORE_INSERTS,
357 NULL, GetLastError (),
358 MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
359 (LPSTR)&pszSysMsg, 0, NULL
360 ) ;
361 sprintf (szMsg, "Error: %s", pszSysMsg) ;
362 LocalFree ((HLOCAL)pszSysMsg) ;
363 MessageBoxA (hwnd, szMsg, NULL, MB_OK | MB_ICONEXCLAMATION) ;
364
365 HeapFree(GetProcessHeap(), 0, psz);
366 HeapFree(GetProcessHeap(), 0, pdir);
367 SendMessageA (htxt, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
368 return TRUE ;
369 }
370 FillList (htxt, psz) ;
371 HeapFree(GetProcessHeap(), 0, psz);
372 HeapFree(GetProcessHeap(), 0, pdir);
373 EndDialog (hwnd, 0) ;
374 }
375 }
376
377 case IDCANCEL :
378 EndDialog (hwnd, 0) ;
379 return TRUE ;
380
381 case 12288 :
382 {
383 HMODULE hComdlg = NULL ;
384 LPFNOFN ofnProc = NULL ;
385 static char szFName[1024] = "", szFileTitle[256] = "", szInitDir[768] = "" ;
386 static OPENFILENAMEA ofn =
387 {
388 sizeof (OPENFILENAMEA),
389 NULL,
390 NULL,
391 "Executable Files\0*.exe\0All Files\0*.*\0\0\0\0",
392 NULL,
393 0,
394 0,
395 szFName,
396 1023,
397 szFileTitle,
398 255,
399 (LPCSTR)szInitDir,
400 "Browse",
401 OFN_ENABLESIZING | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST,
402 0,
403 0,
404 NULL,
405 0,
406 (LPOFNHOOKPROC)NULL,
407 NULL
408 } ;
409
410 ofn.hwndOwner = hwnd ;
411
412 if (NULL == (hComdlg = LoadLibraryExA ("comdlg32", NULL, 0)))
413 {
414 MessageBoxA (hwnd, "Unable to display dialog box (LoadLibraryEx) !", "Nix", MB_OK | MB_ICONEXCLAMATION) ;
415 return TRUE ;
416 }
417
418 if ((LPFNOFN)NULL == (ofnProc = (LPFNOFN)GetProcAddress (hComdlg, "GetOpenFileNameA")))
419 {
420 MessageBoxA (hwnd, "Unable to display dialog box (GetProcAddress) !", "Nix", MB_OK | MB_ICONEXCLAMATION) ;
421 return TRUE ;
422 }
423
424 ofnProc (&ofn) ;
425
426 SetFocus (GetDlgItem (hwnd, IDOK)) ;
427 SetWindowTextA (GetDlgItem (hwnd, 12298), szFName) ;
428 SendMessageA (GetDlgItem (hwnd, 12298), CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
429 SetFocus (GetDlgItem (hwnd, IDOK)) ;
430
431 FreeLibrary (hComdlg) ;
432
433 return TRUE ;
434 }
435 }
436 return TRUE ;
437 }
438 return FALSE ;
439 }
440
441 /* This grabs the MRU list from the registry and fills the combo for the "Run" dialog above */
442 static void FillList (HWND hCb, char *pszLatest)
443 {
444 HKEY hkey ;
445 /* char szDbgMsg[256] = "" ; */
446 char *pszList = NULL, *pszCmd = NULL, cMatch = 0, cMax = 0x60, szIndex[2] = "-" ;
447 DWORD icList = 0, icCmd = 0 ;
448 UINT Nix ;
449
450 SendMessageA (hCb, CB_RESETCONTENT, 0, 0) ;
451
452 if (ERROR_SUCCESS != RegCreateKeyExA (
453 HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\RunMRU",
454 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, NULL))
455 MessageBoxA (hCb, "Unable to open registry key !", "Nix", MB_OK) ;
456
457 RegQueryValueExA (hkey, "MRUList", NULL, NULL, NULL, &icList) ;
458
459 if (icList > 0)
460 {
461 pszList = HeapAlloc( GetProcessHeap(), 0, icList) ;
462 if (ERROR_SUCCESS != RegQueryValueExA (hkey, "MRUList", NULL, NULL, (LPBYTE)pszList, &icList))
463 MessageBoxA (hCb, "Unable to grab MRUList !", "Nix", MB_OK) ;
464 }
465 else
466 {
467 icList = 1 ;
468 pszList = HeapAlloc( GetProcessHeap(), 0, icList) ;
469 pszList[0] = 0 ;
470 }
471
472 for (Nix = 0 ; Nix < icList - 1 ; Nix++)
473 {
474 if (pszList[Nix] > cMax)
475 cMax = pszList[Nix] ;
476
477 szIndex[0] = pszList[Nix] ;
478
479 if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, NULL, &icCmd))
480 MessageBoxA (hCb, "Unable to grab size of index", "Nix", MB_OK) ;
481 if( pszCmd )
482 pszCmd = HeapReAlloc(GetProcessHeap(), 0, pszCmd, icCmd) ;
483 else
484 pszCmd = HeapAlloc(GetProcessHeap(), 0, icCmd) ;
485 if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, (LPBYTE)pszCmd, &icCmd))
486 MessageBoxA (hCb, "Unable to grab index", "Nix", MB_OK) ;
487
488 if (NULL != pszLatest)
489 {
490 if (!lstrcmpiA(pszCmd, pszLatest))
491 {
492 /*
493 sprintf (szDbgMsg, "Found existing (%d).\n", Nix) ;
494 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
495 */
496 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszCmd) ;
497 SetWindowTextA (hCb, pszCmd) ;
498 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
499
500 cMatch = pszList[Nix] ;
501 memmove (&pszList[1], pszList, Nix) ;
502 pszList[0] = cMatch ;
503 continue ;
504 }
505 }
506
507 if (26 != icList - 1 || icList - 2 != Nix || cMatch || NULL == pszLatest)
508 {
509 /*
510 sprintf (szDbgMsg, "Happily appending (%d).\n", Nix) ;
511 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
512 */
513 SendMessageA (hCb, CB_ADDSTRING, 0, (LPARAM)pszCmd) ;
514 if (!Nix)
515 {
516 SetWindowTextA (hCb, pszCmd) ;
517 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
518 }
519
520 }
521 else
522 {
523 /*
524 sprintf (szDbgMsg, "Doing loop thing.\n") ;
525 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
526 */
527 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
528 SetWindowTextA (hCb, pszLatest) ;
529 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
530
531 cMatch = pszList[Nix] ;
532 memmove (&pszList[1], pszList, Nix) ;
533 pszList[0] = cMatch ;
534 szIndex[0] = cMatch ;
535 RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
536 }
537 }
538
539 if (!cMatch && NULL != pszLatest)
540 {
541 /*
542 sprintf (szDbgMsg, "Simply inserting (increasing list).\n") ;
543 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
544 */
545 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
546 SetWindowTextA (hCb, pszLatest) ;
547 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
548
549 cMatch = ++cMax ;
550 if( pszList )
551 pszList = HeapReAlloc(GetProcessHeap(), 0, pszList, ++icList) ;
552 else
553 pszList = HeapAlloc(GetProcessHeap(), 0, ++icList) ;
554 memmove (&pszList[1], pszList, icList - 1) ;
555 pszList[0] = cMatch ;
556 szIndex[0] = cMatch ;
557 RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
558 }
559
560 RegSetValueExA (hkey, "MRUList", 0, REG_SZ, (LPBYTE)pszList, strlen (pszList) + 1) ;
561
562 HeapFree( GetProcessHeap(), 0, pszCmd) ;
563 HeapFree( GetProcessHeap(), 0, pszList) ;
564 }
565
566
567 /*************************************************************************
568 * ConfirmDialog [internal]
569 *
570 * Put up a confirm box, return TRUE if the user confirmed
571 */
572 static BOOL ConfirmDialog(HWND hWndOwner, UINT PromptId, UINT TitleId)
573 {
574 WCHAR Prompt[256];
575 WCHAR Title[256];
576
577 LoadStringW(shell32_hInstance, PromptId, Prompt, sizeof(Prompt) / sizeof(WCHAR));
578 LoadStringW(shell32_hInstance, TitleId, Title, sizeof(Title) / sizeof(WCHAR));
579 return MessageBoxW(hWndOwner, Prompt, Title, MB_YESNO|MB_ICONQUESTION) == IDYES;
580 }
581
582
583 /*************************************************************************
584 * RestartDialogEx [SHELL32.730]
585 */
586
587 int WINAPI RestartDialogEx(HWND hWndOwner, LPCWSTR lpwstrReason, DWORD uFlags, DWORD uReason)
588 {
589 TRACE("(%p)\n", hWndOwner);
590
591 /* FIXME: use lpwstrReason */
592 if (ConfirmDialog(hWndOwner, IDS_RESTART_PROMPT, IDS_RESTART_TITLE))
593 {
594 HANDLE hToken;
595 TOKEN_PRIVILEGES npr;
596
597 /* enable the shutdown privilege for the current process */
598 if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
599 {
600 LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
601 npr.PrivilegeCount = 1;
602 npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
603 AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
604 CloseHandle(hToken);
605 }
606 ExitWindowsEx(EWX_REBOOT, uReason);
607 }
608
609 return 0;
610 }
611
612 /*************************************************************************
613 * LogoffWindowsDialog [SHELL32.54]
614 */
615
616 int WINAPI LogoffWindowsDialog(HWND hWndOwner)
617 {
618 if (ConfirmDialog(hWndOwner, IDS_LOGOFF_PROMPT, IDS_LOGOFF_TITLE))
619 {
620 ExitWindowsEx(EWX_LOGOFF, 0);
621 }
622 return 0;
623 }
624
625 /*************************************************************************
626 * RestartDialog [SHELL32.59]
627 */
628
629 int WINAPI RestartDialog(HWND hWndOwner, LPCWSTR lpstrReason, DWORD uFlags)
630 {
631 return RestartDialogEx(hWndOwner, lpstrReason, uFlags, 0);
632 }
633
634
635 /*************************************************************************
636 * ExitWindowsDialog [SHELL32.60]
637 *
638 * NOTES
639 * exported by ordinal
640 */
641 void WINAPI ExitWindowsDialog (HWND hWndOwner)
642 {
643 TRACE("(%p)\n", hWndOwner);
644
645 if (ConfirmDialog(hWndOwner, IDS_SHUTDOWN_PROMPT, IDS_SHUTDOWN_TITLE))
646 {
647 HANDLE hToken;
648 TOKEN_PRIVILEGES npr;
649
650 /* enable shutdown privilege for current process */
651 if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
652 {
653 LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
654 npr.PrivilegeCount = 1;
655 npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
656 AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
657 CloseHandle(hToken);
658 }
659 ExitWindowsEx(EWX_SHUTDOWN, 0);
660 }
661 }