dlls/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 "config.h"
  22 #include "wine/port.h"
  23
  24 #include <string.h>
  25 #include <stdarg.h>
  26 #include <stdio.h>
  27 #include "winerror.h"
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "winreg.h"
  31 #include "wingdi.h"
  32 #include "winuser.h"
  33 #include "commdlg.h"
  34 #include "wine/debug.h"
  35
  36 #include "shellapi.h"
  37 #include "shlobj.h"
  38 #include "shell32_main.h"
  39 #include "shresdef.h"
  40 #include "undocshell.h"
  41
  42 typedef struct
  43     {
  44         HWND hwndOwner ;
  45         HICON hIcon ;
  46         LPCWSTR lpstrDirectory ;
  47         LPCWSTR lpstrTitle ;
  48         LPCWSTR lpstrDescription ;
  49         UINT uFlags ;
  50     } RUNFILEDLGPARAMS ;
  51
  52 typedef BOOL (WINAPI * LPFNOFN) (OPENFILENAMEW *) ;
  53
  54 WINE_DEFAULT_DEBUG_CHANNEL(shell);
  55 static INT_PTR CALLBACK RunDlgProc (HWND, UINT, WPARAM, LPARAM) ;
  56 static void FillList (HWND, char *, BOOL) ;
  57
  58
  59 /*************************************************************************
  60  * PickIconDlg                                  [SHELL32.62]
  61  *
  62  */
  63 INT WINAPI PickIconDlg(HWND hwndOwner, WCHAR *path, UINT path_len, INT *index)
  64 {
  65     FIXME("(%p,%s,%u,%p):stub.\n", hwndOwner, debugstr_w(path), path_len, index);
  66     return 0xffffffff;
  67 }
  68
  69 HRESULT WINAPI SHOpenWithDialog(HWND parent, const OPENASINFO *info)
  70 {
  71     FIXME("stub\n");
  72     return E_NOTIMPL;
  73 }
  74
  75 /*************************************************************************
  76  * RunFileDlgW                                  [internal]
  77  *
  78  * The Unicode function that is available as ordinal 61 on Windows NT/2000/XP/...
  79  *
  80  * SEE ALSO
  81  *   RunFileDlgAW
  82  */
  83 static void RunFileDlgW(
  84         HWND hwndOwner,
  85         HICON hIcon,
  86         LPCWSTR lpstrDirectory,
  87         LPCWSTR lpstrTitle,
  88         LPCWSTR lpstrDescription,
  89         UINT uFlags)
  90 {
  91     static const WCHAR resnameW[] = {'S','H','E','L','L','_','R','U','N','_','D','L','G',0};
  92     RUNFILEDLGPARAMS rfdp;
  93     HRSRC hRes;
  94     LPVOID template;
  95     TRACE("\n");
  96
  97     rfdp.hwndOwner        = hwndOwner;
  98     rfdp.hIcon            = hIcon;
  99     rfdp.lpstrDirectory   = lpstrDirectory;
 100     rfdp.lpstrTitle       = lpstrTitle;
 101     rfdp.lpstrDescription = lpstrDescription;
 102     rfdp.uFlags           = uFlags;
 103
 104     if (!(hRes = FindResourceW(shell32_hInstance, resnameW, (LPWSTR)RT_DIALOG)) ||
 105         !(template = LoadResource(shell32_hInstance, hRes)))
 106     {
 107         ERR("Couldn't load SHELL_RUN_DLG resource\n");
 108         ShellMessageBoxW(shell32_hInstance, hwndOwner, MAKEINTRESOURCEW(IDS_RUNDLG_ERROR), NULL, MB_OK | MB_ICONERROR);
 109         return;
 110     }
 111
 112     DialogBoxIndirectParamW(shell32_hInstance,
 113                             template, hwndOwner, RunDlgProc, (LPARAM)&rfdp);
 114
 115 }
 116
 117 /* find the directory that contains the file being run */
 118 static LPWSTR RunDlg_GetParentDir(LPCWSTR cmdline)
 119 {
 120     const WCHAR *src;
 121     WCHAR *dest, *result, *result_end=NULL;
 122     static const WCHAR dotexeW[] = {'.','e','x','e',0};
 123
 124     result = heap_alloc(sizeof(WCHAR)*(strlenW(cmdline)+5));
 125
 126     src = cmdline;
 127     dest = result;
 128
 129     if (*src == '"')
 130     {
 131         src++;
 132         while (*src && *src != '"')
 133         {
 134             if (*src == '\\')
 135                 result_end = dest;
 136             *dest++ = *src++;
 137         }
 138     }
 139     else {
 140         while (*src)
 141         {
 142             if (isspaceW(*src))
 143             {
 144                 *dest = 0;
 145                 if (INVALID_FILE_ATTRIBUTES != GetFileAttributesW(result))
 146                     break;
 147                 strcatW(dest, dotexeW);
 148                 if (INVALID_FILE_ATTRIBUTES != GetFileAttributesW(result))
 149                     break;
 150             }
 151             else if (*src == '\\')
 152                 result_end = dest;
 153             *dest++ = *src++;
 154         }
 155     }
 156
 157     if (result_end)
 158     {
 159         *result_end = 0;
 160         return result;
 161     }
 162     else
 163     {
 164         heap_free(result);
 165         return NULL;
 166     }
 167 }
 168
 169 /* Dialog procedure for RunFileDlg */
 170 static INT_PTR CALLBACK RunDlgProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
 171     {
 172     RUNFILEDLGPARAMS *prfdp = (RUNFILEDLGPARAMS *)GetWindowLongPtrW(hwnd, DWLP_USER);
 173
 174     switch (message)
 175         {
 176         case WM_INITDIALOG :
 177             prfdp = (RUNFILEDLGPARAMS *)lParam ;
 178             SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)prfdp);
 179
 180             if (prfdp->lpstrTitle)
 181                 SetWindowTextW(hwnd, prfdp->lpstrTitle);
 182             if (prfdp->lpstrDescription)
 183                 SetWindowTextW(GetDlgItem(hwnd, IDC_RUNDLG_DESCRIPTION), prfdp->lpstrDescription);
 184             if (prfdp->uFlags & RFF_NOBROWSE)
 185             {
 186                 HWND browse = GetDlgItem(hwnd, IDC_RUNDLG_BROWSE);
 187                 ShowWindow(browse, SW_HIDE);
 188                 EnableWindow(browse, FALSE);
 189             }
 190             if (prfdp->uFlags & RFF_NOLABEL)
 191                 ShowWindow(GetDlgItem(hwnd, IDC_RUNDLG_LABEL), SW_HIDE);
 192             if (prfdp->uFlags & RFF_CALCDIRECTORY)
 193                 FIXME("RFF_CALCDIRECTORY not supported\n");
 194
 195             if (prfdp->hIcon == NULL)
 196                 prfdp->hIcon = LoadIconW(NULL, (LPCWSTR)IDI_WINLOGO);
 197             SendMessageW(hwnd, WM_SETICON, ICON_BIG, (LPARAM)prfdp->hIcon);
 198             SendMessageW(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)prfdp->hIcon);
 199             SendMessageW(GetDlgItem(hwnd, IDC_RUNDLG_ICON), STM_SETICON, (WPARAM)prfdp->hIcon, 0);
 200
 201             FillList (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), NULL, (prfdp->uFlags & RFF_NODEFAULT) == 0) ;
 202             SetFocus (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH)) ;
 203             return TRUE ;
 204
 205         case WM_COMMAND :
 206             switch (LOWORD (wParam))
 207                 {
 208                 case IDOK :
 209                     {
 210                     int ic ;
 211                     HWND htxt = GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH);
 212                     if ((ic = GetWindowTextLengthW (htxt)))
 213                         {
 214                         WCHAR *psz, *parent=NULL ;
 215                         SHELLEXECUTEINFOW sei ;
 216
 217                         ZeroMemory (&sei, sizeof(sei)) ;
 218                         sei.cbSize = sizeof(sei) ;
 219                         psz = heap_alloc( (ic + 1)*sizeof(WCHAR) );
 220                         GetWindowTextW (htxt, psz, ic + 1) ;
 221
 222                         /* according to http://www.codeproject.com/KB/shell/runfiledlg.aspx we should send a
 223                          * WM_NOTIFY before execution */
 224
 225                         sei.hwnd = hwnd;
 226                         sei.nShow = SW_SHOWNORMAL;
 227                         sei.lpFile = psz;
 228
 229                         if (prfdp->lpstrDirectory)
 230                             sei.lpDirectory = prfdp->lpstrDirectory;
 231                         else
 232                             sei.lpDirectory = parent = RunDlg_GetParentDir(sei.lpFile);
 233
 234                         if (!ShellExecuteExW( &sei ))
 235                         {
 236                             heap_free(psz);
 237                             heap_free(parent);
 238                             SendMessageA (htxt, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
 239                             return TRUE ;
 240                         }
 241
 242                         /* FillList is still ANSI */
 243                         GetWindowTextA (htxt, (LPSTR)psz, ic + 1) ;
 244                         FillList (htxt, (LPSTR)psz, FALSE) ;
 245
 246                         heap_free(psz);
 247                         heap_free(parent);
 248                         EndDialog (hwnd, 0);
 249                         }
 250                     }
 251                     return TRUE;
 252
 253                 case IDCANCEL :
 254                     EndDialog (hwnd, 0) ;
 255                     return TRUE ;
 256
 257                 case IDC_RUNDLG_BROWSE :
 258                     {
 259                     static const WCHAR filterW[] = {'%','s','%','c','*','.','e','x','e','%','c','%','s','%','c','*','.','*','%','c',0};
 260                     HMODULE hComdlg = NULL ;
 261                     LPFNOFN ofnProc = NULL ;
 262                     static const WCHAR comdlg32W[] = {'c','o','m','d','l','g','3','2',0};
 263                     WCHAR szFName[1024] = {0};
 264                     WCHAR filter_exe[256], filter_all[256], filter[MAX_PATH], szCaption[MAX_PATH];
 265                     OPENFILENAMEW ofn;
 266
 267                     LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_FILTER_EXE, filter_exe, 256);
 268                     LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_FILTER_ALL, filter_all, 256);
 269                     LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_CAPTION, szCaption, MAX_PATH);
 270                     snprintfW( filter, MAX_PATH, filterW, filter_exe, 0, 0, filter_all, 0, 0 );
 271
 272                     ZeroMemory(&ofn, sizeof(ofn));
 273                     ofn.lStructSize = sizeof(OPENFILENAMEW);
 274                     ofn.hwndOwner = hwnd;
 275                     ofn.lpstrFilter = filter;
 276                     ofn.lpstrFile = szFName;
 277                     ofn.nMaxFile = 1023;
 278                     ofn.lpstrTitle = szCaption;
 279                     ofn.Flags = OFN_ENABLESIZING | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST;
 280                     ofn.lpstrInitialDir = prfdp->lpstrDirectory;
 281
 282                     if (NULL == (hComdlg = LoadLibraryExW (comdlg32W, NULL, 0)) ||
 283                         NULL == (ofnProc = (LPFNOFN)GetProcAddress (hComdlg, "GetOpenFileNameW")))
 284                     {
 285                         ERR("Couldn't get GetOpenFileName function entry (lib=%p, proc=%p)\n", hComdlg, ofnProc);
 286                         ShellMessageBoxW(shell32_hInstance, hwnd, MAKEINTRESOURCEW(IDS_RUNDLG_BROWSE_ERROR), NULL, MB_OK | MB_ICONERROR);
 287                         return TRUE ;
 288                     }
 289
 290                     if (ofnProc(&ofn))
 291                     {
 292                         SetFocus (GetDlgItem (hwnd, IDOK)) ;
 293                         SetWindowTextW (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), szFName) ;
 294                         SendMessageW (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
 295                         SetFocus (GetDlgItem (hwnd, IDOK)) ;
 296                     }
 297
 298                     FreeLibrary (hComdlg) ;
 299
 300                     return TRUE ;
 301                     }
 302                 }
 303             return TRUE ;
 304         }
 305     return FALSE ;
 306     }
 307
 308 /* This grabs the MRU list from the registry and fills the combo for the "Run" dialog above */
 309 /* fShowDefault ignored if pszLatest != NULL */
 310 static void FillList (HWND hCb, char *pszLatest, BOOL fShowDefault)
 311     {
 312     HKEY hkey ;
 313 /*    char szDbgMsg[256] = "" ; */
 314     char *pszList = NULL, *pszCmd = NULL, cMatch = 0, cMax = 0x60, szIndex[2] = "-" ;
 315     DWORD icList = 0, icCmd = 0 ;
 316     UINT Nix ;
 317
 318     SendMessageA (hCb, CB_RESETCONTENT, 0, 0) ;
 319
 320     if (ERROR_SUCCESS != RegCreateKeyExA (
 321         HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\RunMRU",
 322         0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, NULL))
 323         MessageBoxA (hCb, "Unable to open registry key !", "Nix", MB_OK) ;
 324
 325     RegQueryValueExA (hkey, "MRUList", NULL, NULL, NULL, &icList) ;
 326
 327     if (icList > 0)
 328         {
 329         pszList = heap_alloc(icList) ;
 330         if (ERROR_SUCCESS != RegQueryValueExA (hkey, "MRUList", NULL, NULL, (LPBYTE)pszList, &icList))
 331             MessageBoxA (hCb, "Unable to grab MRUList !", "Nix", MB_OK) ;
 332         }
 333     else
 334         {
 335         icList = 1 ;
 336         pszList = heap_alloc(icList) ;
 337         pszList[0] = 0 ;
 338         }
 339
 340     for (Nix = 0 ; Nix < icList - 1 ; Nix++)
 341         {
 342         if (pszList[Nix] > cMax)
 343             cMax = pszList[Nix] ;
 344
 345         szIndex[0] = pszList[Nix] ;
 346
 347         if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, NULL, &icCmd))
 348             MessageBoxA (hCb, "Unable to grab size of index", "Nix", MB_OK) ;
 349         if( pszCmd )
 350             pszCmd = heap_realloc(pszCmd, icCmd) ;
 351         else
 352             pszCmd = heap_alloc(icCmd) ;
 353         if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, (LPBYTE)pszCmd, &icCmd))
 354             MessageBoxA (hCb, "Unable to grab index", "Nix", MB_OK) ;
 355
 356         if (NULL != pszLatest)
 357             {
 358             if (!lstrcmpiA(pszCmd, pszLatest))
 359                 {
 360                 /*
 361                 sprintf (szDbgMsg, "Found existing (%d).\n", Nix) ;
 362                 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
 363                 */
 364                 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszCmd) ;
 365                 SetWindowTextA (hCb, pszCmd) ;
 366                 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
 367
 368                 cMatch = pszList[Nix] ;
 369                 memmove (&pszList[1], pszList, Nix) ;
 370                 pszList[0] = cMatch ;
 371                 continue ;
 372                 }
 373             }
 374
 375         if (26 != icList - 1 || icList - 2 != Nix || cMatch || NULL == pszLatest)
 376             {
 377             /*
 378             sprintf (szDbgMsg, "Happily appending (%d).\n", Nix) ;
 379             MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
 380             */
 381             SendMessageA (hCb, CB_ADDSTRING, 0, (LPARAM)pszCmd) ;
 382             if (!Nix && fShowDefault)
 383                 {
 384                 SetWindowTextA (hCb, pszCmd) ;
 385                 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
 386                 }
 387
 388             }
 389         else
 390             {
 391             /*
 392             sprintf (szDbgMsg, "Doing loop thing.\n") ;
 393             MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
 394             */
 395             SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
 396             SetWindowTextA (hCb, pszLatest) ;
 397             SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
 398
 399             cMatch = pszList[Nix] ;
 400             memmove (&pszList[1], pszList, Nix) ;
 401             pszList[0] = cMatch ;
 402             szIndex[0] = cMatch ;
 403             RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
 404             }
 405         }
 406
 407     if (!cMatch && NULL != pszLatest)
 408         {
 409         /*
 410         sprintf (szDbgMsg, "Simply inserting (increasing list).\n") ;
 411         MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
 412         */
 413         SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
 414         SetWindowTextA (hCb, pszLatest) ;
 415         SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
 416
 417         cMatch = ++cMax ;
 418         if( pszList )
 419             pszList = heap_realloc(pszList, ++icList) ;
 420         else
 421             pszList = heap_alloc(++icList) ;
 422         memmove (&pszList[1], pszList, icList - 1) ;
 423         pszList[0] = cMatch ;
 424         szIndex[0] = cMatch ;
 425         RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
 426         }
 427
 428     RegSetValueExA (hkey, "MRUList", 0, REG_SZ, (LPBYTE)pszList, strlen (pszList) + 1) ;
 429
 430     heap_free(pszCmd) ;
 431     heap_free(pszList) ;
 432 }
 433
 434 /*************************************************************************
 435  * RunFileDlgA                                  [internal]
 436  *
 437  * The ANSI function that is available as ordinal 61 on Windows 9x/Me
 438  *
 439  * SEE ALSO
 440  *   RunFileDlgAW
 441  */
 442 static void RunFileDlgA(
 443         HWND hwndOwner,
 444         HICON hIcon,
 445         LPCSTR lpstrDirectory,
 446         LPCSTR lpstrTitle,
 447         LPCSTR lpstrDescription,
 448         UINT uFlags)
 449 {
 450     WCHAR title[MAX_PATH];       /* longer string wouldn't be visible in the dialog anyway */
 451     WCHAR description[MAX_PATH];
 452     WCHAR directory[MAX_PATH];
 453
 454     MultiByteToWideChar(CP_ACP, 0, lpstrTitle, -1, title, MAX_PATH);
 455     title[MAX_PATH - 1] = 0;
 456     MultiByteToWideChar(CP_ACP, 0, lpstrDescription, -1, description, MAX_PATH);
 457     description[MAX_PATH - 1] = 0;
 458     if (!MultiByteToWideChar(CP_ACP, 0, lpstrDirectory, -1, directory, MAX_PATH))
 459         directory[0] = 0;
 460
 461     RunFileDlgW(hwndOwner, hIcon,
 462         lpstrDirectory ? directory : NULL,
 463         lpstrTitle ? title : NULL,
 464         lpstrDescription ? description : NULL,
 465         uFlags);
 466 }
 467
 468 /*************************************************************************
 469  * RunFileDlgAW                                 [SHELL32.61]
 470  *
 471  * An undocumented way to open the Run File dialog. A documented way is to use
 472  * CLSID_Shell, IID_IShellDispatch (as of Wine 1.0, not implemented under Wine)
 473  *
 474  * Exported by ordinal. ANSI on Windows 9x and Unicode on Windows NT/2000/XP/etc
 475  *
 476  */
 477 void WINAPI RunFileDlgAW(
 478         HWND hwndOwner,
 479         HICON hIcon,
 480         LPCVOID lpstrDirectory,
 481         LPCVOID lpstrTitle,
 482         LPCVOID lpstrDescription,
 483         UINT uFlags)
 484 {
 485     if (SHELL_OsIsUnicode())
 486         RunFileDlgW(hwndOwner, hIcon, lpstrDirectory, lpstrTitle, lpstrDescription, uFlags);
 487     else
 488         RunFileDlgA(hwndOwner, hIcon, lpstrDirectory, lpstrTitle, lpstrDescription, uFlags);
 489 }
 490
 491
 492 /*************************************************************************
 493  * ConfirmDialog                                [internal]
 494  *
 495  * Put up a confirm box, return TRUE if the user confirmed
 496  */
 497 static BOOL ConfirmDialog(HWND hWndOwner, UINT PromptId, UINT TitleId)
 498 {
 499   WCHAR Prompt[256];
 500   WCHAR Title[256];
 501
 502   LoadStringW(shell32_hInstance, PromptId, Prompt, ARRAY_SIZE(Prompt));
 503   LoadStringW(shell32_hInstance, TitleId, Title, ARRAY_SIZE(Title));
 504   return MessageBoxW(hWndOwner, Prompt, Title, MB_YESNO|MB_ICONQUESTION) == IDYES;
 505 }
 506
 507
 508 /*************************************************************************
 509  * RestartDialogEx                              [SHELL32.730]
 510  */
 511
 512 int WINAPI RestartDialogEx(HWND hWndOwner, LPCWSTR lpwstrReason, DWORD uFlags, DWORD uReason)
 513 {
 514     TRACE("(%p)\n", hWndOwner);
 515
 516     /* FIXME: use lpwstrReason */
 517     if (ConfirmDialog(hWndOwner, IDS_RESTART_PROMPT, IDS_RESTART_TITLE))
 518     {
 519         HANDLE hToken;
 520         TOKEN_PRIVILEGES npr;
 521
 522         /* enable the shutdown privilege for the current process */
 523         if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
 524         {
 525             LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
 526             npr.PrivilegeCount = 1;
 527             npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 528             AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
 529             CloseHandle(hToken);
 530         }
 531         ExitWindowsEx(EWX_REBOOT, uReason);
 532     }
 533
 534     return 0;
 535 }
 536
 537
 538 /*************************************************************************
 539  * RestartDialog                                [SHELL32.59]
 540  */
 541
 542 int WINAPI RestartDialog(HWND hWndOwner, LPCWSTR lpstrReason, DWORD uFlags)
 543 {
 544     return RestartDialogEx(hWndOwner, lpstrReason, uFlags, 0);
 545 }
 546
 547
 548 /*************************************************************************
 549  * ExitWindowsDialog                            [SHELL32.60]
 550  *
 551  * NOTES
 552  *     exported by ordinal
 553  */
 554 void WINAPI ExitWindowsDialog (HWND hWndOwner)
 555 {
 556     TRACE("(%p)\n", hWndOwner);
 557
 558     if (ConfirmDialog(hWndOwner, IDS_SHUTDOWN_PROMPT, IDS_SHUTDOWN_TITLE))
 559     {
 560         HANDLE hToken;
 561         TOKEN_PRIVILEGES npr;
 562
 563         /* enable shutdown privilege for current process */
 564         if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
 565         {
 566             LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
 567             npr.PrivilegeCount = 1;
 568             npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 569             AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
 570             CloseHandle(hToken);
 571         }
 572         ExitWindowsEx(EWX_SHUTDOWN, 0);
 573     }
 574 }