mirror of
https://github.com/reactos/reactos.git
synced 2024-06-22 22:11:39 +00:00
6f7115b541
- Display file size in bytes in file properties dialog - Don't use ULONGLONG if ULONG is enough - Change LARGE_INTEGER to ULARGE_INTEGER where it makes sense - Fix calculation of file size in CountFolderAndFiles - WM_GETTEXT receives size in bytes (not characters count) - Use Get(Set)DlgItemTextW instead of SendDlgItemMessageW Patch by Carlo Bramix. Improvements by me. See issue #6294 for more details. svn path=/trunk/; revision=54529
739 lines
25 KiB
C++
739 lines
25 KiB
C++
/*
|
|
* common shell dialogs
|
|
*
|
|
* Copyright 2000 Juergen Schmied
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <precomp.h>
|
|
|
|
|
|
typedef struct
|
|
{
|
|
HWND hwndOwner ;
|
|
HICON hIcon ;
|
|
LPCWSTR lpstrDirectory ;
|
|
LPCWSTR lpstrTitle ;
|
|
LPCWSTR lpstrDescription ;
|
|
UINT uFlags ;
|
|
} RUNFILEDLGPARAMS ;
|
|
|
|
typedef BOOL (WINAPI * LPFNOFN) (OPENFILENAMEW *) ;
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(shell);
|
|
static INT_PTR CALLBACK RunDlgProc (HWND, UINT, WPARAM, LPARAM) ;
|
|
static void FillList (HWND, char *, BOOL) ;
|
|
|
|
|
|
/*************************************************************************
|
|
* PickIconDlg [SHELL32.62]
|
|
*
|
|
*/
|
|
|
|
typedef struct
|
|
{
|
|
HMODULE hLibrary;
|
|
HWND hDlgCtrl;
|
|
WCHAR szName[MAX_PATH];
|
|
INT Index;
|
|
}PICK_ICON_CONTEXT, *PPICK_ICON_CONTEXT;
|
|
|
|
BOOL CALLBACK EnumPickIconResourceProc(HMODULE hModule,
|
|
LPCWSTR lpszType,
|
|
LPWSTR lpszName,
|
|
LONG_PTR lParam
|
|
)
|
|
{
|
|
WCHAR szName[100];
|
|
int index;
|
|
HICON hIcon;
|
|
PPICK_ICON_CONTEXT pIconContext = (PPICK_ICON_CONTEXT)lParam;
|
|
|
|
if (IS_INTRESOURCE(lpszName))
|
|
swprintf(szName, L"%u", lpszName);
|
|
else
|
|
wcscpy(szName, (WCHAR*)lpszName);
|
|
|
|
|
|
hIcon = LoadIconW(pIconContext->hLibrary, (LPCWSTR)lpszName);
|
|
if (hIcon == NULL)
|
|
return TRUE;
|
|
|
|
index = SendMessageW(pIconContext->hDlgCtrl, LB_ADDSTRING, 0, (LPARAM)szName);
|
|
if (index != LB_ERR)
|
|
SendMessageW(pIconContext->hDlgCtrl, LB_SETITEMDATA, index, (LPARAM)hIcon);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
void
|
|
DestroyIconList(HWND hDlgCtrl)
|
|
{
|
|
int count;
|
|
int index;
|
|
|
|
count = SendMessage(hDlgCtrl, LB_GETCOUNT, 0, 0);
|
|
if (count == LB_ERR)
|
|
return;
|
|
|
|
for(index = 0; index < count; index++)
|
|
{
|
|
HICON hIcon = (HICON)SendMessageW(hDlgCtrl, LB_GETITEMDATA, index, 0);
|
|
DestroyIcon(hIcon);
|
|
}
|
|
}
|
|
|
|
INT_PTR CALLBACK PickIconProc(HWND hwndDlg,
|
|
UINT uMsg,
|
|
WPARAM wParam,
|
|
LPARAM lParam
|
|
)
|
|
{
|
|
LPMEASUREITEMSTRUCT lpmis;
|
|
LPDRAWITEMSTRUCT lpdis;
|
|
HICON hIcon;
|
|
INT index, count;
|
|
WCHAR szText[MAX_PATH], szTitle[100], szFilter[100];
|
|
OPENFILENAMEW ofn = {0};
|
|
|
|
PPICK_ICON_CONTEXT pIconContext = (PPICK_ICON_CONTEXT)GetWindowLongPtr(hwndDlg, DWLP_USER);
|
|
|
|
switch(uMsg)
|
|
{
|
|
case WM_INITDIALOG:
|
|
pIconContext = (PPICK_ICON_CONTEXT)lParam;
|
|
SetWindowLongPtr(hwndDlg, DWLP_USER, (LONG)pIconContext);
|
|
pIconContext->hDlgCtrl = GetDlgItem(hwndDlg, IDC_PICKICON_LIST);
|
|
EnumResourceNamesW(pIconContext->hLibrary, RT_ICON, EnumPickIconResourceProc, (LPARAM)pIconContext);
|
|
if (PathUnExpandEnvStringsW(pIconContext->szName, szText, MAX_PATH))
|
|
SetDlgItemTextW(hwndDlg, IDC_EDIT_PATH, szText);
|
|
else
|
|
SetDlgItemTextW(hwndDlg, IDC_EDIT_PATH, pIconContext->szName);
|
|
|
|
count = SendMessage(pIconContext->hDlgCtrl, LB_GETCOUNT, 0, 0);
|
|
if (count != LB_ERR)
|
|
{
|
|
if (count > pIconContext->Index)
|
|
SendMessageW(pIconContext->hDlgCtrl, LB_SETCURSEL, pIconContext->Index, 0);
|
|
else
|
|
SendMessageW(pIconContext->hDlgCtrl, LB_SETCURSEL, 0, 0);
|
|
}
|
|
return TRUE;
|
|
case WM_COMMAND:
|
|
switch(LOWORD(wParam))
|
|
{
|
|
case IDOK:
|
|
index = SendMessageW(pIconContext->hDlgCtrl, LB_GETCURSEL, 0, 0);
|
|
pIconContext->Index = index;
|
|
GetDlgItemTextW(hwndDlg, IDC_EDIT_PATH, pIconContext->szName, MAX_PATH);
|
|
DestroyIconList(pIconContext->hDlgCtrl);
|
|
EndDialog(hwndDlg, 1);
|
|
break;
|
|
case IDCANCEL:
|
|
DestroyIconList(pIconContext->hDlgCtrl);
|
|
EndDialog(hwndDlg, 0);
|
|
break;
|
|
case IDC_PICKICON_LIST:
|
|
if (HIWORD(wParam) == LBN_SELCHANGE)
|
|
InvalidateRect((HWND)lParam, NULL, TRUE); // FIXME USE UPDATE RECT
|
|
break;
|
|
case IDC_BUTTON_PATH:
|
|
szText[0] = 0;
|
|
szTitle[0] = 0;
|
|
szFilter[0] = 0;
|
|
ofn.lStructSize = sizeof(ofn);
|
|
ofn.hwndOwner = hwndDlg;
|
|
ofn.lpstrFile = szText;
|
|
ofn.nMaxFile = MAX_PATH;
|
|
LoadStringW(shell32_hInstance, IDS_PICK_ICON_TITLE, szTitle, sizeof(szTitle) / sizeof(WCHAR));
|
|
ofn.lpstrTitle = szTitle;
|
|
LoadStringW(shell32_hInstance, IDS_PICK_ICON_FILTER, szFilter, sizeof(szFilter) / sizeof(WCHAR));
|
|
ofn.lpstrFilter = szFilter;
|
|
if (GetOpenFileNameW(&ofn))
|
|
{
|
|
HMODULE hLibrary;
|
|
|
|
if (!wcsicmp(pIconContext->szName, szText))
|
|
break;
|
|
|
|
DestroyIconList(pIconContext->hDlgCtrl);
|
|
|
|
hLibrary = LoadLibraryExW(szText, NULL, LOAD_LIBRARY_AS_DATAFILE | LOAD_LIBRARY_AS_IMAGE_RESOURCE);
|
|
if (hLibrary == NULL)
|
|
break;
|
|
FreeLibrary(pIconContext->hLibrary);
|
|
pIconContext->hLibrary = hLibrary;
|
|
wcscpy(pIconContext->szName, szText);
|
|
EnumResourceNamesW(pIconContext->hLibrary, RT_ICON, EnumPickIconResourceProc, (LPARAM)pIconContext);
|
|
if (PathUnExpandEnvStringsW(pIconContext->szName, szText, MAX_PATH))
|
|
SetDlgItemTextW(hwndDlg, IDC_EDIT_PATH, szText);
|
|
else
|
|
SetDlgItemTextW(hwndDlg, IDC_EDIT_PATH, pIconContext->szName);
|
|
|
|
SendMessageW(pIconContext->hDlgCtrl, LB_SETCURSEL, 0, 0);
|
|
}
|
|
break;
|
|
}
|
|
break;
|
|
case WM_MEASUREITEM:
|
|
lpmis = (LPMEASUREITEMSTRUCT) lParam;
|
|
lpmis->itemHeight = 32;
|
|
lpmis->itemWidth = 64;
|
|
return TRUE;
|
|
case WM_DRAWITEM:
|
|
lpdis = (LPDRAWITEMSTRUCT) lParam;
|
|
if (lpdis->itemID == (UINT)-1)
|
|
{
|
|
break;
|
|
}
|
|
switch (lpdis->itemAction)
|
|
{
|
|
case ODA_SELECT:
|
|
case ODA_DRAWENTIRE:
|
|
index = SendMessageW(pIconContext->hDlgCtrl, LB_GETCURSEL, 0, 0);
|
|
hIcon =(HICON)SendMessage(lpdis->hwndItem, LB_GETITEMDATA, lpdis->itemID, (LPARAM) 0);
|
|
|
|
if (lpdis->itemID == (UINT)index)
|
|
{
|
|
HBRUSH hBrush;
|
|
hBrush = CreateSolidBrush(RGB(0, 0, 255));
|
|
FillRect(lpdis->hDC, &lpdis->rcItem, hBrush);
|
|
DeleteObject(hBrush);
|
|
}
|
|
else
|
|
{
|
|
HBRUSH hBrush;
|
|
hBrush = CreateSolidBrush(RGB(255, 255, 255));
|
|
FillRect(lpdis->hDC, &lpdis->rcItem, hBrush);
|
|
DeleteObject(hBrush);
|
|
}
|
|
DrawIconEx(lpdis->hDC, lpdis->rcItem.left,lpdis->rcItem.top, hIcon,
|
|
0,
|
|
0,
|
|
0,
|
|
NULL,
|
|
DI_NORMAL);
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL WINAPI PickIconDlg(
|
|
HWND hwndOwner,
|
|
LPWSTR lpstrFile,
|
|
UINT nMaxFile,
|
|
INT* lpdwIconIndex)
|
|
{
|
|
HMODULE hLibrary;
|
|
int res;
|
|
PICK_ICON_CONTEXT IconContext;
|
|
|
|
hLibrary = LoadLibraryExW(lpstrFile, NULL, LOAD_LIBRARY_AS_DATAFILE | LOAD_LIBRARY_AS_IMAGE_RESOURCE);
|
|
IconContext.hLibrary = hLibrary;
|
|
IconContext.Index = *lpdwIconIndex;
|
|
wcscpy(IconContext.szName, lpstrFile);
|
|
|
|
res = DialogBoxParamW(shell32_hInstance, MAKEINTRESOURCEW(IDD_PICK_ICON_DIALOG), hwndOwner, PickIconProc, (LPARAM)&IconContext);
|
|
if (res)
|
|
{
|
|
wcscpy(lpstrFile, IconContext.szName);
|
|
*lpdwIconIndex = IconContext.Index;
|
|
}
|
|
|
|
FreeLibrary(hLibrary);
|
|
return res;
|
|
}
|
|
|
|
/*************************************************************************
|
|
* RunFileDlg [internal]
|
|
*
|
|
* The Unicode function that is available as ordinal 61 on Windows NT/2000/XP/...
|
|
*/
|
|
void WINAPI RunFileDlg(
|
|
HWND hwndOwner,
|
|
HICON hIcon,
|
|
LPCWSTR lpstrDirectory,
|
|
LPCWSTR lpstrTitle,
|
|
LPCWSTR lpstrDescription,
|
|
UINT uFlags)
|
|
{
|
|
static const WCHAR resnameW[] = {'S','H','E','L','L','_','R','U','N','_','D','L','G',0};
|
|
RUNFILEDLGPARAMS rfdp;
|
|
HRSRC hRes;
|
|
LPVOID tmplate;
|
|
TRACE("\n");
|
|
|
|
rfdp.hwndOwner = hwndOwner;
|
|
rfdp.hIcon = hIcon;
|
|
rfdp.lpstrDirectory = lpstrDirectory;
|
|
rfdp.lpstrTitle = lpstrTitle;
|
|
rfdp.lpstrDescription = lpstrDescription;
|
|
rfdp.uFlags = uFlags;
|
|
|
|
if (!(hRes = FindResourceW(shell32_hInstance, resnameW, (LPWSTR)RT_DIALOG)) ||
|
|
!(tmplate = LoadResource(shell32_hInstance, hRes)))
|
|
{
|
|
ERR("Couldn't load SHELL_RUN_DLG resource\n");
|
|
ShellMessageBoxW(shell32_hInstance, hwndOwner, MAKEINTRESOURCEW(IDS_RUNDLG_ERROR), NULL, MB_OK | MB_ICONERROR);
|
|
return;
|
|
}
|
|
|
|
DialogBoxIndirectParamW(shell32_hInstance,
|
|
(LPCDLGTEMPLATEW)tmplate, hwndOwner, RunDlgProc, (LPARAM)&rfdp);
|
|
|
|
}
|
|
|
|
|
|
/* find the directory that contains the file being run */
|
|
static LPWSTR RunDlg_GetParentDir(LPCWSTR cmdline)
|
|
{
|
|
const WCHAR *src;
|
|
WCHAR *dest, *result, *result_end=NULL;
|
|
static const WCHAR dotexeW[] = {'.','e','x','e',0};
|
|
|
|
result = (WCHAR *)HeapAlloc(GetProcessHeap(), 0, sizeof(WCHAR)*(strlenW(cmdline)+5));
|
|
|
|
if (NULL == result)
|
|
{
|
|
TRACE("HeapAlloc couldn't allocate %d bytes\n", sizeof(WCHAR)*(strlenW(cmdline)+5));
|
|
return NULL;
|
|
}
|
|
|
|
src = cmdline;
|
|
dest = result;
|
|
|
|
if (*src == '"')
|
|
{
|
|
src++;
|
|
while (*src && *src != '"')
|
|
{
|
|
if (*src == '\\')
|
|
result_end = dest;
|
|
*dest++ = *src++;
|
|
}
|
|
}
|
|
else {
|
|
while (*src)
|
|
{
|
|
if (isspaceW(*src))
|
|
{
|
|
*dest = 0;
|
|
if (INVALID_FILE_ATTRIBUTES != GetFileAttributesW(result))
|
|
break;
|
|
strcatW(dest, dotexeW);
|
|
if (INVALID_FILE_ATTRIBUTES != GetFileAttributesW(result))
|
|
break;
|
|
}
|
|
else if (*src == '\\')
|
|
result_end = dest;
|
|
*dest++ = *src++;
|
|
}
|
|
}
|
|
|
|
if (result_end)
|
|
{
|
|
*result_end = 0;
|
|
return result;
|
|
}
|
|
else
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, result);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
|
|
/* Dialog procedure for RunFileDlg */
|
|
static INT_PTR CALLBACK RunDlgProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
RUNFILEDLGPARAMS *prfdp = (RUNFILEDLGPARAMS *)GetWindowLongPtrW(hwnd, DWLP_USER);
|
|
|
|
switch (message)
|
|
{
|
|
case WM_INITDIALOG :
|
|
prfdp = (RUNFILEDLGPARAMS *)lParam ;
|
|
SetWindowLongPtrW(hwnd, DWLP_USER, (LONG_PTR)prfdp);
|
|
|
|
if (prfdp->lpstrTitle)
|
|
SetWindowTextW(hwnd, prfdp->lpstrTitle);
|
|
if (prfdp->lpstrDescription)
|
|
SetWindowTextW(GetDlgItem(hwnd, IDC_RUNDLG_DESCRIPTION), prfdp->lpstrDescription);
|
|
if (prfdp->uFlags & RFF_NOBROWSE)
|
|
{
|
|
HWND browse = GetDlgItem(hwnd, IDC_RUNDLG_BROWSE);
|
|
ShowWindow(browse, SW_HIDE);
|
|
EnableWindow(browse, FALSE);
|
|
}
|
|
if (prfdp->uFlags & RFF_NOLABEL)
|
|
ShowWindow(GetDlgItem(hwnd, IDC_RUNDLG_LABEL), SW_HIDE);
|
|
if (prfdp->uFlags & RFF_CALCDIRECTORY)
|
|
FIXME("RFF_CALCDIRECTORY not supported\n");
|
|
|
|
if (prfdp->hIcon == NULL)
|
|
prfdp->hIcon = LoadIconW(NULL, (LPCWSTR)IDI_WINLOGO);
|
|
SendMessageW(hwnd, WM_SETICON, ICON_BIG, (LPARAM)prfdp->hIcon);
|
|
SendMessageW(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)prfdp->hIcon);
|
|
SendMessageW(GetDlgItem(hwnd, IDC_RUNDLG_ICON), STM_SETICON, (WPARAM)prfdp->hIcon, 0);
|
|
|
|
FillList (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), NULL, (prfdp->uFlags & RFF_NODEFAULT) == 0) ;
|
|
SetFocus (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH)) ;
|
|
return TRUE ;
|
|
|
|
case WM_COMMAND :
|
|
switch (LOWORD (wParam))
|
|
{
|
|
case IDOK :
|
|
{
|
|
int ic ;
|
|
HWND htxt = GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH);
|
|
if ((ic = GetWindowTextLengthW (htxt)))
|
|
{
|
|
WCHAR *psz, *parent=NULL ;
|
|
SHELLEXECUTEINFOW sei ;
|
|
|
|
ZeroMemory (&sei, sizeof(sei)) ;
|
|
sei.cbSize = sizeof(sei) ;
|
|
psz = (WCHAR *)HeapAlloc( GetProcessHeap(), 0, (ic + 1)*sizeof(WCHAR) );
|
|
|
|
if (psz)
|
|
{
|
|
GetWindowTextW (htxt, psz, ic + 1) ;
|
|
|
|
/* according to http://www.codeproject.com/KB/shell/runfiledlg.aspx we should send a
|
|
* WM_NOTIFY before execution */
|
|
|
|
sei.hwnd = hwnd;
|
|
sei.nShow = SW_SHOWNORMAL;
|
|
sei.lpFile = psz;
|
|
|
|
if (prfdp->lpstrDirectory)
|
|
sei.lpDirectory = prfdp->lpstrDirectory;
|
|
else
|
|
sei.lpDirectory = parent = RunDlg_GetParentDir(sei.lpFile);
|
|
|
|
if (!ShellExecuteExW( &sei ))
|
|
{
|
|
HeapFree(GetProcessHeap(), 0, psz);
|
|
HeapFree(GetProcessHeap(), 0, parent);
|
|
SendMessageA (htxt, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
|
|
return TRUE ;
|
|
}
|
|
|
|
/* FillList is still ANSI */
|
|
GetWindowTextA (htxt, (LPSTR)psz, ic + 1) ;
|
|
FillList (htxt, (LPSTR)psz, FALSE) ;
|
|
|
|
HeapFree(GetProcessHeap(), 0, psz);
|
|
HeapFree(GetProcessHeap(), 0, parent);
|
|
EndDialog (hwnd, 0);
|
|
}
|
|
}
|
|
}
|
|
|
|
case IDCANCEL :
|
|
EndDialog (hwnd, 0) ;
|
|
return TRUE ;
|
|
|
|
case IDC_RUNDLG_BROWSE :
|
|
{
|
|
HMODULE hComdlg = NULL ;
|
|
LPFNOFN ofnProc = NULL ;
|
|
static const WCHAR comdlg32W[] = {'c','o','m','d','l','g','3','2',0};
|
|
WCHAR szFName[1024] = {0};
|
|
WCHAR filter[MAX_PATH], szCaption[MAX_PATH];
|
|
OPENFILENAMEW ofn;
|
|
|
|
LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_FILTER, filter, MAX_PATH);
|
|
LoadStringW(shell32_hInstance, IDS_RUNDLG_BROWSE_CAPTION, szCaption, MAX_PATH);
|
|
|
|
ZeroMemory(&ofn, sizeof(ofn));
|
|
ofn.lStructSize = sizeof(OPENFILENAMEW);
|
|
ofn.hwndOwner = hwnd;
|
|
ofn.lpstrFilter = filter;
|
|
ofn.lpstrFile = szFName;
|
|
ofn.nMaxFile = 1023;
|
|
ofn.lpstrTitle = szCaption;
|
|
ofn.Flags = OFN_ENABLESIZING | OFN_FILEMUSTEXIST | OFN_HIDEREADONLY | OFN_PATHMUSTEXIST;
|
|
ofn.lpstrInitialDir = prfdp->lpstrDirectory;
|
|
|
|
if (NULL == (hComdlg = LoadLibraryExW (comdlg32W, NULL, 0)) ||
|
|
NULL == (ofnProc = (LPFNOFN)GetProcAddress (hComdlg, "GetOpenFileNameW")))
|
|
{
|
|
ERR("Couldn't get GetOpenFileName function entry (lib=%p, proc=%p)\n", hComdlg, ofnProc);
|
|
ShellMessageBoxW(shell32_hInstance, hwnd, MAKEINTRESOURCEW(IDS_RUNDLG_BROWSE_ERROR), NULL, MB_OK | MB_ICONERROR);
|
|
return TRUE ;
|
|
}
|
|
|
|
if (ofnProc(&ofn))
|
|
{
|
|
SetFocus (GetDlgItem (hwnd, IDOK)) ;
|
|
SetWindowTextW (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), szFName) ;
|
|
SendMessageW (GetDlgItem (hwnd, IDC_RUNDLG_EDITPATH), CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
|
|
SetFocus (GetDlgItem (hwnd, IDOK)) ;
|
|
}
|
|
|
|
FreeLibrary (hComdlg) ;
|
|
|
|
return TRUE ;
|
|
}
|
|
}
|
|
return TRUE ;
|
|
}
|
|
return FALSE ;
|
|
}
|
|
|
|
/* This grabs the MRU list from the registry and fills the combo for the "Run" dialog above */
|
|
/* fShowDefault ignored if pszLatest != NULL */
|
|
static void FillList (HWND hCb, char *pszLatest, BOOL fShowDefault)
|
|
{
|
|
HKEY hkey ;
|
|
/* char szDbgMsg[256] = "" ; */
|
|
char *pszList = NULL, *pszCmd = NULL, cMatch = 0, cMax = 0x60, szIndex[2] = "-" ;
|
|
DWORD icList = 0, icCmd = 0 ;
|
|
UINT Nix ;
|
|
|
|
SendMessageA (hCb, CB_RESETCONTENT, 0, 0) ;
|
|
|
|
if (ERROR_SUCCESS != RegCreateKeyExA (
|
|
HKEY_CURRENT_USER, "Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\RunMRU",
|
|
0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &hkey, NULL))
|
|
MessageBoxA (hCb, "Unable to open registry key !", "Nix", MB_OK) ;
|
|
|
|
RegQueryValueExA (hkey, "MRUList", NULL, NULL, NULL, &icList) ;
|
|
|
|
if (icList > 0)
|
|
{
|
|
pszList = (char *)HeapAlloc( GetProcessHeap(), 0, icList) ;
|
|
|
|
if (pszList)
|
|
{
|
|
if (ERROR_SUCCESS != RegQueryValueExA (hkey, "MRUList", NULL, NULL, (LPBYTE)pszList, &icList))
|
|
MessageBoxA (hCb, "Unable to grab MRUList !", "Nix", MB_OK);
|
|
}
|
|
else
|
|
{
|
|
TRACE("HeapAlloc failed to allocate %d bytes\n", icList);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
icList = 1 ;
|
|
pszList = (char *)HeapAlloc( GetProcessHeap(), 0, icList) ;
|
|
pszList[0] = 0 ;
|
|
}
|
|
|
|
for (Nix = 0 ; Nix < icList - 1 ; Nix++)
|
|
{
|
|
if (pszList[Nix] > cMax)
|
|
cMax = pszList[Nix] ;
|
|
|
|
szIndex[0] = pszList[Nix] ;
|
|
|
|
if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, NULL, &icCmd))
|
|
MessageBoxA (hCb, "Unable to grab size of index", "Nix", MB_OK) ;
|
|
if( pszCmd )
|
|
pszCmd = (char *)HeapReAlloc(GetProcessHeap(), 0, pszCmd, icCmd) ;
|
|
else
|
|
pszCmd = (char *)HeapAlloc(GetProcessHeap(), 0, icCmd) ;
|
|
if (ERROR_SUCCESS != RegQueryValueExA (hkey, szIndex, NULL, NULL, (LPBYTE)pszCmd, &icCmd))
|
|
MessageBoxA (hCb, "Unable to grab index", "Nix", MB_OK) ;
|
|
|
|
if (NULL != pszLatest)
|
|
{
|
|
if (!lstrcmpiA(pszCmd, pszLatest))
|
|
{
|
|
/*
|
|
sprintf (szDbgMsg, "Found existing (%d).\n", Nix) ;
|
|
MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
|
|
*/
|
|
SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszCmd) ;
|
|
SetWindowTextA (hCb, pszCmd) ;
|
|
SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
|
|
|
|
cMatch = pszList[Nix] ;
|
|
memmove (&pszList[1], pszList, Nix) ;
|
|
pszList[0] = cMatch ;
|
|
continue ;
|
|
}
|
|
}
|
|
|
|
if (26 != icList - 1 || icList - 2 != Nix || cMatch || NULL == pszLatest)
|
|
{
|
|
/*
|
|
sprintf (szDbgMsg, "Happily appending (%d).\n", Nix) ;
|
|
MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
|
|
*/
|
|
SendMessageA (hCb, CB_ADDSTRING, 0, (LPARAM)pszCmd) ;
|
|
if (!Nix && fShowDefault)
|
|
{
|
|
SetWindowTextA (hCb, pszCmd) ;
|
|
SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
/*
|
|
sprintf (szDbgMsg, "Doing loop thing.\n") ;
|
|
MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
|
|
*/
|
|
SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
|
|
SetWindowTextA (hCb, pszLatest) ;
|
|
SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
|
|
|
|
cMatch = pszList[Nix] ;
|
|
memmove (&pszList[1], pszList, Nix) ;
|
|
pszList[0] = cMatch ;
|
|
szIndex[0] = cMatch ;
|
|
RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
|
|
}
|
|
}
|
|
|
|
if (!cMatch && NULL != pszLatest)
|
|
{
|
|
/*
|
|
sprintf (szDbgMsg, "Simply inserting (increasing list).\n") ;
|
|
MessageBoxA (hCb, szDbgMsg, "Nix", MB_OK) ;
|
|
*/
|
|
SendMessageA (hCb, CB_INSERTSTRING, 0, (LPARAM)pszLatest) ;
|
|
SetWindowTextA (hCb, pszLatest) ;
|
|
SendMessageA (hCb, CB_SETEDITSEL, 0, MAKELPARAM (0, -1)) ;
|
|
|
|
cMatch = ++cMax ;
|
|
if (pszList)
|
|
pszList = (char *)HeapReAlloc(GetProcessHeap(), 0, pszList, ++icList) ;
|
|
else
|
|
pszList = (char *)HeapAlloc(GetProcessHeap(), 0, ++icList) ;
|
|
|
|
if (pszList)
|
|
{
|
|
memmove (&pszList[1], pszList, icList - 1) ;
|
|
pszList[0] = cMatch ;
|
|
szIndex[0] = cMatch ;
|
|
RegSetValueExA (hkey, szIndex, 0, REG_SZ, (LPBYTE)pszLatest, strlen (pszLatest) + 1) ;
|
|
}
|
|
else
|
|
{
|
|
TRACE("HeapAlloc or HeapReAlloc failed to allocate enough bytes\n");
|
|
}
|
|
}
|
|
|
|
RegSetValueExA (hkey, "MRUList", 0, REG_SZ, (LPBYTE)pszList, strlen (pszList) + 1) ;
|
|
|
|
HeapFree( GetProcessHeap(), 0, pszCmd) ;
|
|
HeapFree( GetProcessHeap(), 0, pszList) ;
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* ConfirmDialog [internal]
|
|
*
|
|
* Put up a confirm box, return TRUE if the user confirmed
|
|
*/
|
|
static BOOL ConfirmDialog(HWND hWndOwner, UINT PromptId, UINT TitleId)
|
|
{
|
|
WCHAR Prompt[256];
|
|
WCHAR Title[256];
|
|
|
|
LoadStringW(shell32_hInstance, PromptId, Prompt, sizeof(Prompt) / sizeof(WCHAR));
|
|
LoadStringW(shell32_hInstance, TitleId, Title, sizeof(Title) / sizeof(WCHAR));
|
|
return MessageBoxW(hWndOwner, Prompt, Title, MB_YESNO|MB_ICONQUESTION) == IDYES;
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* RestartDialogEx [SHELL32.730]
|
|
*/
|
|
|
|
int WINAPI RestartDialogEx(HWND hWndOwner, LPCWSTR lpwstrReason, DWORD uFlags, DWORD uReason)
|
|
{
|
|
TRACE("(%p)\n", hWndOwner);
|
|
|
|
/* FIXME: use lpwstrReason */
|
|
if (ConfirmDialog(hWndOwner, IDS_RESTART_PROMPT, IDS_RESTART_TITLE))
|
|
{
|
|
HANDLE hToken;
|
|
TOKEN_PRIVILEGES npr;
|
|
|
|
/* enable the shutdown privilege for the current process */
|
|
if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
|
|
{
|
|
LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
|
|
npr.PrivilegeCount = 1;
|
|
npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
|
|
AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
|
|
CloseHandle(hToken);
|
|
}
|
|
ExitWindowsEx(EWX_REBOOT, uReason);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* LogoffWindowsDialog [SHELL32.54]
|
|
*/
|
|
|
|
EXTERN_C int WINAPI LogoffWindowsDialog(HWND hWndOwner)
|
|
{
|
|
if (ConfirmDialog(hWndOwner, IDS_LOGOFF_PROMPT, IDS_LOGOFF_TITLE))
|
|
{
|
|
ExitWindowsEx(EWX_LOGOFF, 0);
|
|
}
|
|
return 0;}
|
|
|
|
|
|
/*************************************************************************
|
|
* RestartDialog [SHELL32.59]
|
|
*/
|
|
|
|
int WINAPI RestartDialog(HWND hWndOwner, LPCWSTR lpstrReason, DWORD uFlags)
|
|
{
|
|
return RestartDialogEx(hWndOwner, lpstrReason, uFlags, 0);
|
|
}
|
|
|
|
|
|
/*************************************************************************
|
|
* ExitWindowsDialog [SHELL32.60]
|
|
*
|
|
* NOTES
|
|
* exported by ordinal
|
|
*/
|
|
void WINAPI ExitWindowsDialog (HWND hWndOwner)
|
|
{
|
|
TRACE("(%p)\n", hWndOwner);
|
|
|
|
if (ConfirmDialog(hWndOwner, IDS_SHUTDOWN_PROMPT, IDS_SHUTDOWN_TITLE))
|
|
{
|
|
HANDLE hToken;
|
|
TOKEN_PRIVILEGES npr;
|
|
|
|
/* enable shutdown privilege for current process */
|
|
if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
|
|
{
|
|
LookupPrivilegeValueA(0, "SeShutdownPrivilege", &npr.Privileges[0].Luid);
|
|
npr.PrivilegeCount = 1;
|
|
npr.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
|
|
AdjustTokenPrivileges(hToken, FALSE, &npr, 0, 0, 0);
|
|
CloseHandle(hToken);
|
|
}
|
|
ExitWindowsEx(EWX_SHUTDOWN, 0);
|
|
}
|
|
}
|