ntdll: Use PAGE_EXECUTE_READWRITE protection when allocating stubs.
[wine.git] / dlls / shell32 / dialogs.c
blob84d87991a24821dfe0c7654fcedb0ce05d7d8a89
1 /*
2 * common shell dialogs
4 * Copyright 2000 Juergen Schmied
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.
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.
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
21 #include "config.h"
22 #include "wine/port.h"
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"
36 #include "shellapi.h"
37 #include "shlobj.h"
38 #include "shell32_main.h"
39 #include "shresdef.h"
40 #include "undocshell.h"
42 typedef struct
44 HWND hwndOwner ;
45 HICON hIcon ;
46 LPCWSTR lpstrDirectory ;
47 LPCWSTR lpstrTitle ;
48 LPCWSTR lpstrDescription ;
49 UINT uFlags ;
50 } RUNFILEDLGPARAMS ;
52 typedef BOOL (WINAPI * LPFNOFN) (OPENFILENAMEW *) ;
54 WINE_DEFAULT_DEBUG_CHANNEL(shell);
55 static INT_PTR CALLBACK RunDlgProc (HWND, UINT, WPARAM, LPARAM) ;
56 static void FillList (HWND, char *, BOOL) ;
59 /*************************************************************************
60 * PickIconDlg [SHELL32.62]
63 INT WINAPI PickIconDlg(
64 HWND hwndOwner,
65 LPSTR lpstrFile,
66 DWORD nMaxFile,
67 LPDWORD lpdwIconIndex)
69 FIXME("(%p,%s,%08x,%p):stub.\n",
70 hwndOwner, lpstrFile, nMaxFile,lpdwIconIndex);
71 return 0xffffffff;
74 /*************************************************************************
75 * RunFileDlgW [internal]
77 * The Unicode function that is available as ordinal 61 on Windows NT/2000/XP/...
79 * SEE ALSO
80 * RunFileDlgAW
82 static void RunFileDlgW(
83 HWND hwndOwner,
84 HICON hIcon,
85 LPCWSTR lpstrDirectory,
86 LPCWSTR lpstrTitle,
87 LPCWSTR lpstrDescription,
88 UINT uFlags)
90 static const WCHAR resnameW[] = {'S','H','E','L','L','_','R','U','N','_','D','L','G',0};
91 RUNFILEDLGPARAMS rfdp;
92 HRSRC hRes;
93 LPVOID template;
94 TRACE("\n");
96 rfdp.hwndOwner = hwndOwner;
97 rfdp.hIcon = hIcon;
98 rfdp.lpstrDirectory = lpstrDirectory;
99 rfdp.lpstrTitle = lpstrTitle;
100 rfdp.lpstrDescription = lpstrDescription;
101 rfdp.uFlags = uFlags;
103 if (!(hRes = FindResourceW(shell32_hInstance, resnameW, (LPWSTR)RT_DIALOG)) ||
104 !(template = LoadResource(shell32_hInstance, hRes)))
106 ERR("Couldn't load SHELL_RUN_DLG resource\n");
107 ShellMessageBoxW(shell32_hInstance, hwndOwner, MAKEINTRESOURCEW(IDS_RUNDLG_ERROR), NULL, MB_OK | MB_ICONERROR);
108 return;
111 DialogBoxIndirectParamW(shell32_hInstance,
112 template, hwndOwner, RunDlgProc, (LPARAM)&rfdp);
116 /* find the directory that contains the file being run */
117 static LPWSTR RunDlg_GetParentDir(LPCWSTR cmdline)
119 const WCHAR *src;
120 WCHAR *dest, *result, *result_end=NULL;
121 static const WCHAR dotexeW[] = {'.','e','x','e',0};
123 result = HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR)*(strlenW(cmdline)+5));
125 src = cmdline;
126 dest = result;
128 if (*src == '"')
130 src++;
131 while (*src && *src != '"')
133 if (*src == '\\')
134 result_end = dest;
135 *dest++ = *src++;
138 else {
139 while (*src)
141 if (isspaceW(*src))
143 *dest = 0;
144 if (INVALID_FILE_ATTRIBUTES != GetFileAttributesW(result))
145 break;
146 strcatW(dest, dotexeW);
147 if (INVALID_FILE_ATTRIBUTES != GetFileAttributesW(result))
148 break;
150 else if (*src == '\\')
151 result_end = dest;
152 *dest++ = *src++;
156 if (result_end)
158 *result_end = 0;
159 return result;
161 else
163 HeapFree(GetProcessHeap(), 0, result);
164 return NULL;
168 /* Dialog procedure for RunFileDlg */
169 static INT_PTR CALLBACK RunDlgProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
171 RUNFILEDLGPARAMS *prfdp = (RUNFILEDLGPARAMS *)GetWindowLongPtrW(hwnd, DWLP_USER);
173 switch (message)
175 case WM_INITDIALOG :
176 prfdp = (RUNFILEDLGPARAMS *)lParam ;
177 SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)prfdp);
179 if (prfdp->lpstrTitle)
180 SetWindowTextW(hwnd, prfdp->lpstrTitle);
181 if (prfdp->lpstrDescription)
182 SetWindowTextW(GetDlgItem(hwnd, IDC_RUNDLG_DESCRIPTION), prfdp->lpstrDescription);
183 if (prfdp->uFlags & RFF_NOBROWSE)
185 HWND browse = GetDlgItem(hwnd, IDC_RUNDLG_BROWSE);
186 ShowWindow(browse, SW_HIDE);
187 EnableWindow(browse, FALSE);
189 if (prfdp->uFlags & RFF_NOLABEL)
190 ShowWindow(GetDlgItem(hwnd, IDC_RUNDLG_LABEL), SW_HIDE);
191 if (prfdp->uFlags & RFF_CALCDIRECTORY)
192 FIXME("RFF_CALCDIRECTORY not supported\n");
194 if (prfdp->hIcon == NULL)
195 prfdp->hIcon = LoadIconW(NULL, (LPCWSTR)IDI_WINLOGO);
196 SendMessageW(hwnd, WM_SETICON, ICON_BIG, (LPARAM)prfdp->hIcon);
197 SendMessageW(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)prfdp->hIcon);
198 SendMessageW(GetDlgItem(hwnd, IDC_RUNDLG_ICON), STM_SETICON, (WPARAM)prfdp->hIcon, 0);
200 FillList (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), NULL, (prfdp->uFlags & RFF_NODEFAULT) == 0) ;
201 SetFocus (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH)) ;
202 return TRUE ;
204 case WM_COMMAND :
205 switch (LOWORD (wParam))
207 case IDOK :
209 int ic ;
210 HWND htxt = GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH);
211 if ((ic = GetWindowTextLengthW (htxt)))
213 WCHAR *psz, *parent=NULL ;
214 SHELLEXECUTEINFOW sei ;
216 ZeroMemory (&sei, sizeof(sei)) ;
217 sei.cbSize = sizeof(sei) ;
218 psz = HeapAlloc( GetProcessHeap(), 0, (ic + 1)*sizeof(WCHAR) );
219 GetWindowTextW (htxt, psz, ic + 1) ;
221 /* according to http://www.codeproject.com/KB/shell/runfiledlg.aspx we should send a
222 * WM_NOTIFY before execution */
224 sei.hwnd = hwnd;
225 sei.nShow = SW_SHOWNORMAL;
226 sei.lpFile = psz;
228 if (prfdp->lpstrDirectory)
229 sei.lpDirectory = prfdp->lpstrDirectory;
230 else
231 sei.lpDirectory = parent = RunDlg_GetParentDir(sei.lpFile);
233 if (!ShellExecuteExW( &sei ))
235 HeapFree(GetProcessHeap(), 0, psz);
236 HeapFree(GetProcessHeap(), 0, parent);
237 SendMessageA (htxt, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
238 return TRUE ;
241 /* FillList is still ANSI */
242 GetWindowTextA (htxt, (LPSTR)psz, ic + 1) ;
243 FillList (htxt, (LPSTR)psz, FALSE) ;
245 HeapFree(GetProcessHeap(), 0, psz);
246 HeapFree(GetProcessHeap(), 0, parent);
247 EndDialog (hwnd, 0);
250 return TRUE;
252 case IDCANCEL :
253 EndDialog (hwnd, 0) ;
254 return TRUE ;
256 case IDC_RUNDLG_BROWSE :
258 static const WCHAR filterW[] = {'%','s','%','c','*','.','e','x','e','%','c','%','s','%','c','*','.','*','%','c',0};
259 HMODULE hComdlg = NULL ;
260 LPFNOFN ofnProc = NULL ;
261 static const WCHAR comdlg32W[] = {'c','o','m','d','l','g','3','2',0};
262 WCHAR szFName[1024] = {0};
263 WCHAR filter_exe[256], filter_all[256], filter[MAX_PATH], szCaption[MAX_PATH];
264 OPENFILENAMEW ofn;
266 LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_FILTER_EXE, filter_exe, 256);
267 LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_FILTER_ALL, filter_all, 256);
268 LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_CAPTION, szCaption, MAX_PATH);
269 snprintfW( filter, MAX_PATH, filterW, filter_exe, 0, 0, filter_all, 0 );
271 ZeroMemory(&ofn, sizeof(ofn));
272 ofn.lStructSize = sizeof(OPENFILENAMEW);
273 ofn.hwndOwner = hwnd;
274 ofn.lpstrFilter = filter;
275 ofn.lpstrFile = szFName;
276 ofn.nMaxFile = 1023;
277 ofn.lpstrTitle = szCaption;
278 ofn.Flags = OFN_ENABLESIZING | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST;
279 ofn.lpstrInitialDir = prfdp->lpstrDirectory;
281 if (NULL == (hComdlg = LoadLibraryExW (comdlg32W, NULL, 0)) ||
282 NULL == (ofnProc = (LPFNOFN)GetProcAddress (hComdlg, "GetOpenFileNameW")))
284 ERR("Couldn't get GetOpenFileName function entry (lib=%p, proc=%p)\n", hComdlg, ofnProc);
285 ShellMessageBoxW(shell32_hInstance, hwnd, MAKEINTRESOURCEW(IDS_RUNDLG_BROWSE_ERROR), NULL, MB_OK | MB_ICONERROR);
286 return TRUE ;
289 if (ofnProc(&ofn))
291 SetFocus (GetDlgItem (hwnd, IDOK)) ;
292 SetWindowTextW (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), szFName) ;
293 SendMessageW (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
294 SetFocus (GetDlgItem (hwnd, IDOK)) ;
297 FreeLibrary (hComdlg) ;
299 return TRUE ;
302 return TRUE ;
304 return FALSE ;
307 /* This grabs the MRU list from the registry and fills the combo for the "Run" dialog above */
308 /* fShowDefault ignored if pszLatest != NULL */
309 static void FillList (HWND hCb, char *pszLatest, BOOL fShowDefault)
311 HKEY hkey ;
312 /* char szDbgMsg[256] = "" ; */
313 char *pszList = NULL, *pszCmd = NULL, cMatch = 0, cMax = 0x60, szIndex[2] = "-" ;
314 DWORD icList = 0, icCmd = 0 ;
315 UINT Nix ;
317 SendMessageA (hCb, CB_RESETCONTENT, 0, 0) ;
319 if (ERROR_SUCCESS != RegCreateKeyExA (
320 HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\RunMRU",
321 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, NULL))
322 MessageBoxA (hCb, "Unable to open registry key !", "Nix", MB_OK) ;
324 RegQueryValueExA (hkey, "MRUList", NULL, NULL, NULL, &icList) ;
326 if (icList > 0)
328 pszList = HeapAlloc( GetProcessHeap(), 0, icList) ;
329 if (ERROR_SUCCESS != RegQueryValueExA (hkey, "MRUList", NULL, NULL, (LPBYTE)pszList, &icList))
330 MessageBoxA (hCb, "Unable to grab MRUList !", "Nix", MB_OK) ;
332 else
334 icList = 1 ;
335 pszList = HeapAlloc( GetProcessHeap(), 0, icList) ;
336 pszList[0] = 0 ;
339 for (Nix = 0 ; Nix < icList - 1 ; Nix++)
341 if (pszList[Nix] > cMax)
342 cMax = pszList[Nix] ;
344 szIndex[0] = pszList[Nix] ;
346 if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, NULL, &icCmd))
347 MessageBoxA (hCb, "Unable to grab size of index", "Nix", MB_OK) ;
348 if( pszCmd )
349 pszCmd = HeapReAlloc(GetProcessHeap(), 0, pszCmd, icCmd) ;
350 else
351 pszCmd = HeapAlloc(GetProcessHeap(), 0, icCmd) ;
352 if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, (LPBYTE)pszCmd, &icCmd))
353 MessageBoxA (hCb, "Unable to grab index", "Nix", MB_OK) ;
355 if (NULL != pszLatest)
357 if (!lstrcmpiA(pszCmd, pszLatest))
360 sprintf (szDbgMsg, "Found existing (%d).\n", Nix) ;
361 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
363 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszCmd) ;
364 SetWindowTextA (hCb, pszCmd) ;
365 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
367 cMatch = pszList[Nix] ;
368 memmove (&pszList[1], pszList, Nix) ;
369 pszList[0] = cMatch ;
370 continue ;
374 if (26 != icList - 1 || icList - 2 != Nix || cMatch || NULL == pszLatest)
377 sprintf (szDbgMsg, "Happily appending (%d).\n", Nix) ;
378 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
380 SendMessageA (hCb, CB_ADDSTRING, 0, (LPARAM)pszCmd) ;
381 if (!Nix && fShowDefault)
383 SetWindowTextA (hCb, pszCmd) ;
384 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
388 else
391 sprintf (szDbgMsg, "Doing loop thing.\n") ;
392 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
394 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
395 SetWindowTextA (hCb, pszLatest) ;
396 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
398 cMatch = pszList[Nix] ;
399 memmove (&pszList[1], pszList, Nix) ;
400 pszList[0] = cMatch ;
401 szIndex[0] = cMatch ;
402 RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
406 if (!cMatch && NULL != pszLatest)
409 sprintf (szDbgMsg, "Simply inserting (increasing list).\n") ;
410 MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
412 SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
413 SetWindowTextA (hCb, pszLatest) ;
414 SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
416 cMatch = ++cMax ;
417 if( pszList )
418 pszList = HeapReAlloc(GetProcessHeap(), 0, pszList, ++icList) ;
419 else
420 pszList = HeapAlloc(GetProcessHeap(), 0, ++icList) ;
421 memmove (&pszList[1], pszList, icList - 1) ;
422 pszList[0] = cMatch ;
423 szIndex[0] = cMatch ;
424 RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
427 RegSetValueExA (hkey, "MRUList", 0, REG_SZ, (LPBYTE)pszList, strlen (pszList) + 1) ;
429 HeapFree( GetProcessHeap(), 0, pszCmd) ;
430 HeapFree( GetProcessHeap(), 0, pszList) ;
433 /*************************************************************************
434 * RunFileDlgA [internal]
436 * The ANSI function that is available as ordinal 61 on Windows 9x/Me
438 * SEE ALSO
439 * RunFileDlgAW
441 static void RunFileDlgA(
442 HWND hwndOwner,
443 HICON hIcon,
444 LPCSTR lpstrDirectory,
445 LPCSTR lpstrTitle,
446 LPCSTR lpstrDescription,
447 UINT uFlags)
449 WCHAR title[MAX_PATH]; /* longer string wouldn't be visible in the dialog anyway */
450 WCHAR description[MAX_PATH];
451 WCHAR directory[MAX_PATH];
453 MultiByteToWideChar(CP_ACP, 0, lpstrTitle, -1, title, MAX_PATH);
454 title[MAX_PATH - 1] = 0;
455 MultiByteToWideChar(CP_ACP, 0, lpstrDescription, -1, description, MAX_PATH);
456 description[MAX_PATH - 1] = 0;
457 if (!MultiByteToWideChar(CP_ACP, 0, lpstrDirectory, -1, directory, MAX_PATH))
458 directory[0] = 0;
460 RunFileDlgW(hwndOwner, hIcon,
461 lpstrDirectory ? directory : NULL,
462 lpstrTitle ? title : NULL,
463 lpstrDescription ? description : NULL,
464 uFlags);
467 /*************************************************************************
468 * RunFileDlgAW [SHELL32.61]
470 * An undocumented way to open the Run File dialog. A documented way is to use
471 * CLSID_Shell, IID_IShellDispatch (as of Wine 1.0, not implemented under Wine)
473 * Exported by ordinal. ANSI on Windows 9x and Unicode on Windows NT/2000/XP/etc
476 void WINAPI RunFileDlgAW(
477 HWND hwndOwner,
478 HICON hIcon,
479 LPCVOID lpstrDirectory,
480 LPCVOID lpstrTitle,
481 LPCVOID lpstrDescription,
482 UINT uFlags)
484 if (SHELL_OsIsUnicode())
485 RunFileDlgW(hwndOwner, hIcon, lpstrDirectory, lpstrTitle, lpstrDescription, uFlags);
486 else
487 RunFileDlgA(hwndOwner, hIcon, lpstrDirectory, lpstrTitle, lpstrDescription, uFlags);
491 /*************************************************************************
492 * ConfirmDialog [internal]
494 * Put up a confirm box, return TRUE if the user confirmed
496 static BOOL ConfirmDialog(HWND hWndOwner, UINT PromptId, UINT TitleId)
498 WCHAR Prompt[256];
499 WCHAR Title[256];
501 LoadStringW(shell32_hInstance, PromptId, Prompt, sizeof(Prompt) / sizeof(WCHAR));
502 LoadStringW(shell32_hInstance, TitleId, Title, sizeof(Title) / sizeof(WCHAR));
503 return MessageBoxW(hWndOwner, Prompt, Title, MB_YESNO|MB_ICONQUESTION) == IDYES;
507 /*************************************************************************
508 * RestartDialogEx [SHELL32.730]
511 int WINAPI RestartDialogEx(HWND hWndOwner, LPCWSTR lpwstrReason, DWORD uFlags, DWORD uReason)
513 TRACE("(%p)\n", hWndOwner);
515 /* FIXME: use lpwstrReason */
516 if (ConfirmDialog(hWndOwner, IDS_RESTART_PROMPT, IDS_RESTART_TITLE))
518 HANDLE hToken;
519 TOKEN_PRIVILEGES npr;
521 /* enable the shutdown privilege for the current process */
522 if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
524 LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
525 npr.PrivilegeCount = 1;
526 npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
527 AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
528 CloseHandle(hToken);
530 ExitWindowsEx(EWX_REBOOT, uReason);
533 return 0;
537 /*************************************************************************
538 * RestartDialog [SHELL32.59]
541 int WINAPI RestartDialog(HWND hWndOwner, LPCWSTR lpstrReason, DWORD uFlags)
543 return RestartDialogEx(hWndOwner, lpstrReason, uFlags, 0);
547 /*************************************************************************
548 * ExitWindowsDialog [SHELL32.60]
550 * NOTES
551 * exported by ordinal
553 void WINAPI ExitWindowsDialog (HWND hWndOwner)
555 TRACE("(%p)\n", hWndOwner);
557 if (ConfirmDialog(hWndOwner, IDS_SHUTDOWN_PROMPT, IDS_SHUTDOWN_TITLE))
559 HANDLE hToken;
560 TOKEN_PRIVILEGES npr;
562 /* enable shutdown privilege for current process */
563 if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
565 LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
566 npr.PrivilegeCount = 1;
567 npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
568 AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
569 CloseHandle(hToken);
571 ExitWindowsEx(EWX_SHUTDOWN, 0);