windows-nt/Source/XPSP1/NT/shell/comdlg32/util.cpp

525 lines
17 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
/*++
Copyright (c) 1990-1998, Microsoft Corporation All rights reserved.
Module Name:
util.cpp
Abstract:
This module implements utility functions for the common dialog.
Author:
Arul Kumaravel (arulk@microsoft.com)
History:
Mar-07-2001 - Lazar Ivanov (LazarI)
reimplemented ThunkDevNamesW2A & ThunkDevNamesA2W
--*/
// precompiled headers
#include "precomp.h"
#pragma hdrstop
#include "cdids.h"
#include "fileopen.h"
#include "filenew.h"
#include "util.h"
// crtfree.h is located in shell\inc and it defines new and delete
// operators to do LocalAlloc and LocalFree, so you don't have to
// link to MSVCRT in order to get those. i tried to remove this code
// and link to MSVCRT, but there are some ugly written code here
// which relies on the new operator to zero initialize the returned
// memory block so the class don't bother to initialize its members
// in the constructor. as i said this is quite ugly, but nothing i can
// do about this at the moment.
//
// LazarI - 2/21/2001
//
#define DECL_CRTFREE
#include <crtfree.h>
#ifndef ASSERT
#define ASSERT Assert
#endif
#define EVAL(x) x
#define USE_AUTOCOMPETE_DEFAULT TRUE
#define SZ_REGKEY_USEAUTOCOMPLETE TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\AutoComplete")
#define SZ_REGVALUE_FILEDLGAUTOCOMPLETE TEXT("AutoComplete In File Dialog")
#define BOOL_NOT_SET 0x00000005
#define SZ_REGVALUE_AUTOCOMPLETE_TAB TEXT("Always Use Tab")
/****************************************************\
FUNCTION: AutoComplete
DESCRIPTION:
This function will have AutoComplete take over
an editbox to help autocomplete DOS paths.
\****************************************************/
HRESULT AutoComplete(HWND hwndEdit, ICurrentWorkingDirectory ** ppcwd, DWORD dwFlags)
{
HRESULT hr = S_OK;
IUnknown * punkACLISF;
static BOOL fUseAutoComplete = -10; // Not inited.
if (-10 == fUseAutoComplete)
fUseAutoComplete = (SHRegGetBoolUSValue(SZ_REGKEY_USEAUTOCOMPLETE, SZ_REGVALUE_FILEDLGAUTOCOMPLETE, FALSE, USE_AUTOCOMPETE_DEFAULT));
// WARNING: If you want to disable AutoComplete by default,
// turn USE_AUTOCOMPETE_DEFAULT to FALSE
if (fUseAutoComplete)
{
Assert(!dwFlags); // Not yet used.
hr = SHCoCreateInstance(NULL, &CLSID_ACListISF, NULL, IID_PPV_ARG(IUnknown, &punkACLISF));
if (SUCCEEDED(hr))
{
IAutoComplete2 * pac;
// Create the AutoComplete Object
hr = SHCoCreateInstance(NULL, &CLSID_AutoComplete, NULL, IID_PPV_ARG(IAutoComplete2, &pac));
if (SUCCEEDED(hr))
{
DWORD dwOptions = 0;
hr = pac->Init(hwndEdit, punkACLISF, NULL, NULL);
// Set the autocomplete options
if (SHRegGetBoolUSValue(REGSTR_PATH_AUTOCOMPLETE, REGSTR_VAL_USEAUTOAPPEND, FALSE, /*default:*/FALSE))
{
dwOptions |= ACO_AUTOAPPEND;
}
if (SHRegGetBoolUSValue(REGSTR_PATH_AUTOCOMPLETE, REGSTR_VAL_USEAUTOSUGGEST, FALSE, /*default:*/TRUE))
{
dwOptions |= ACO_AUTOSUGGEST;
}
// Windows uses the TAB key to move between controls in a dialog. UNIX and other
// operating systems that use AutoComplete have traditionally used the TAB key to
// iterate thru the AutoComplete possibilities. We need to default to disable the
// TAB key (ACO_USETAB) unless the caller specifically wants it. We will also
// turn it on
static BOOL s_fAlwaysUseTab = BOOL_NOT_SET;
if (BOOL_NOT_SET == s_fAlwaysUseTab)
s_fAlwaysUseTab = SHRegGetBoolUSValue(SZ_REGKEY_USEAUTOCOMPLETE, SZ_REGVALUE_AUTOCOMPLETE_TAB, FALSE, FALSE);
if (s_fAlwaysUseTab)
dwOptions |= ACO_USETAB;
EVAL(SUCCEEDED(pac->SetOptions(dwOptions)));
pac->Release();
}
if (ppcwd)
{
punkACLISF->QueryInterface(IID_PPV_ARG(ICurrentWorkingDirectory, ppcwd));
}
punkACLISF->Release();
}
}
return hr;
}
////////////////////////////////////////////////////////////////////////////
//
// Common Dialog Administrator Restrictions
//
////////////////////////////////////////////////////////////////////////////
const SHRESTRICTIONITEMS c_rgRestItems[] =
{
{REST_NOBACKBUTTON, L"Comdlg32", L"NoBackButton"},
{REST_NOFILEMRU , L"Comdlg32", L"NoFileMru"},
{REST_NOPLACESBAR, L"Comdlg32", L"NoPlacesBar"},
{0, NULL, NULL},
};
#define NUMRESTRICTIONS ARRAYSIZE(c_rgRestItems)
DWORD g_rgRestItemValues[NUMRESTRICTIONS - 1 ] = { -1 };
DWORD IsRestricted(COMMDLG_RESTRICTIONS rest)
{
static BOOL bFirstTime = TRUE;
if (bFirstTime)
{
memset((LPBYTE)g_rgRestItemValues, (BYTE)-1, sizeof(g_rgRestItemValues));
bFirstTime = FALSE;
}
return SHRestrictionLookup(rest, NULL, c_rgRestItems, g_rgRestItemValues);
}
#define MODULE_NAME_SIZE 128
#define MODULE_VERSION_SIZE 15
typedef struct tagAPPCOMPAT
{
LPCTSTR pszModule;
LPCTSTR pszVersion;
DWORD dwFlags;
} APPCOMPAT, FAR* LPAPPCOMPAT;
DWORD CDGetAppCompatFlags()
{
static BOOL bInitialized = FALSE;
static DWORD dwCachedFlags = 0;
static const APPCOMPAT aAppCompat[] =
{ //Mathcad
{TEXT("MCAD.EXE"), TEXT("6.00b"), CDACF_MATHCAD},
//Picture Publisher
{TEXT("PP70.EXE"),NULL, CDACF_NT40TOOLBAR},
{TEXT("PP80.EXE"),NULL, CDACF_NT40TOOLBAR},
//Code Wright
{TEXT("CW32.exe"),TEXT("5.1"), CDACF_NT40TOOLBAR},
//Designer.exe
{TEXT("ds70.exe"),NULL, CDACF_FILETITLE}
};
if (!bInitialized)
{
TCHAR szModulePath[MODULE_NAME_SIZE];
TCHAR* pszModuleName;
DWORD dwHandle;
int i;
GetModuleFileName(GetModuleHandle(NULL), szModulePath, ARRAYSIZE(szModulePath));
pszModuleName = PathFindFileName(szModulePath);
if (pszModuleName)
{
for (i=0; i < ARRAYSIZE(aAppCompat); i++)
{
if (lstrcmpi(aAppCompat[i].pszModule, pszModuleName) == 0)
{
if (aAppCompat[i].pszVersion == NULL)
{
dwCachedFlags = aAppCompat[i].dwFlags;
}
else
{
CHAR chBuffer[3072]; // hopefully this is enough... lotus smart center needs 3000
TCHAR* pszVersion = NULL;
UINT cb;
// get module version here!
cb = GetFileVersionInfoSize(szModulePath, &dwHandle);
if (cb <= ARRAYSIZE(chBuffer) &&
GetFileVersionInfo(szModulePath, dwHandle, ARRAYSIZE(chBuffer), (LPVOID)chBuffer) &&
VerQueryValue((LPVOID)chBuffer, TEXT("\\StringFileInfo\\040904E4\\ProductVersion"), (void **) &pszVersion, &cb))
{
DebugMsg(0x0004, TEXT("product: %s\n version: %s"), pszModuleName, pszVersion);
if (lstrcmpi(pszVersion, aAppCompat[i].pszVersion) == 0)
{
dwCachedFlags = aAppCompat[i].dwFlags;
break;
}
}
}
}
}
}
bInitialized = TRUE;
}
return dwCachedFlags;
}
BOOL ILIsFTP(LPCITEMIDLIST pidl)
{
IShellFolder * psf;
BOOL fIsFTPFolder = FALSE;
if (SUCCEEDED(CDBindToObject(NULL, IID_X_PPV_ARG(IShellFolder, pidl, &psf))))
{
CLSID clsid;
if (SUCCEEDED(IUnknown_GetClassID(psf, &clsid)) &&
(IsEqualIID(clsid, CLSID_FtpFolder)))
{
fIsFTPFolder = TRUE;
}
psf->Release();
}
return fIsFTPFolder;
}
#ifdef __cplusplus
extern "C" {
#endif
// this is weak.
// a long time ago somebody changed all the FindResources to call FindResourceEx, specifying
// a language. thatd be cool except FindResource already has logic to get the right language.
// whatever was busted should have probably been fixed some other way.
// not only that but it's broken because MUI needs to fall back to US if it can't get the resource
// from the MUI language-specific files.
// thus force a fallback to US. really everything should be rewritten to be normal like every other
// DLL but there's a lot of weird TLS stuff that would break and its risky for this late in XP.
HRSRC FindResourceExFallback(HMODULE hModule, LPCTSTR lpType, LPCTSTR lpName, WORD wLanguage)
{
HRSRC hrsrc = FindResourceEx(hModule, lpType, lpName, wLanguage);
if (!hrsrc)
{
LANGID langid = GetSystemDefaultUILanguage();
if (langid && (langid != wLanguage))
{
hrsrc = FindResourceEx(hModule, lpType, lpName, langid);
}
}
return hrsrc;
}
// Win32Error2HRESULT: converts Win32 error to HRESULT
inline HRESULT Win32Error2HRESULT(DWORD dwError = GetLastError())
{
return (ERROR_SUCCESS == dwError) ? E_FAIL : HRESULT_FROM_WIN32(dwError);
}
/*++
Routine Name:
ThunkDevNamesA2W
Routine Description:
Converts ANSI DEVNAMES structure to UNICODE
on failure we don't release *phDevNamesW
Arguments:
hDevNamesA - [in] handle to ANSI DEVNAMES
phDevNamesW - [in, out] handle to UNICODE DEVNAMES
Return Value:
S_OK if succeded and OLE error otherwise
History:
Lazar Ivanov (LazarI), Mar-07-2001 - created.
--*/
HRESULT
ThunkDevNamesA2W(
IN HGLOBAL hDevNamesA,
IN OUT HGLOBAL *phDevNamesW
)
{
HRESULT hr = E_FAIL;
if (hDevNamesA && phDevNamesW)
{
LPDEVNAMES pDNA = (LPDEVNAMES )GlobalLock(hDevNamesA);
if (pDNA)
{
// calculate the input string pointers
LPSTR pszDriver = reinterpret_cast<LPSTR>(pDNA) + pDNA->wDriverOffset;
LPSTR pszDevice = reinterpret_cast<LPSTR>(pDNA) + pDNA->wDeviceOffset;
LPSTR pszOutput = reinterpret_cast<LPSTR>(pDNA) + pDNA->wOutputOffset;
// calculate the lengths of the ANSI strings
SIZE_T iDriverLenW = MultiByteToWideChar(CP_ACP, 0, pszDriver, -1, NULL, 0);
SIZE_T iDeviceLenW = MultiByteToWideChar(CP_ACP, 0, pszDevice, -1, NULL, 0);
SIZE_T iOutputLenW = MultiByteToWideChar(CP_ACP, 0, pszOutput, -1, NULL, 0);
// calculate the output buffer length
SIZE_T iBytesTotal = sizeof(DEVNAMES) + sizeof(WCHAR) *
((iDriverLenW + 1) + (iDeviceLenW + 1) + (iOutputLenW + 1) + DN_PADDINGCHARS);
HGLOBAL hDevNamesW = (*phDevNamesW) ?
GlobalReAlloc(*phDevNamesW, iBytesTotal, GHND) :
GlobalAlloc(GHND, iBytesTotal);
if (hDevNamesW)
{
// thunk DEVNAMES...
LPDEVNAMES pDNW = (LPDEVNAMES )GlobalLock(hDevNamesW);
if (pDNW)
{
// calculate the offsets
// note: the offsets are in chars not bytes!!
pDNW->wDriverOffset = sizeof(DEVNAMES) / sizeof(WCHAR);
pDNW->wDeviceOffset = pDNW->wDriverOffset + iDriverLenW + 1;
pDNW->wOutputOffset = pDNW->wDeviceOffset + iDeviceLenW + 1;
pDNW->wDefault = pDNA->wDefault;
// calculate the output string pointers
LPWSTR pwszDriver = reinterpret_cast<LPWSTR>(pDNW) + pDNW->wDriverOffset;
LPWSTR pwszDevice = reinterpret_cast<LPWSTR>(pDNW) + pDNW->wDeviceOffset;
LPWSTR pwszOutput = reinterpret_cast<LPWSTR>(pDNW) + pDNW->wOutputOffset;
// convert from ansi to uniciode
MultiByteToWideChar(CP_ACP, 0, pszDriver, -1, pwszDriver, iDriverLenW + 1);
MultiByteToWideChar(CP_ACP, 0, pszDevice, -1, pwszDevice, iDeviceLenW + 1);
MultiByteToWideChar(CP_ACP, 0, pszOutput, -1, pwszOutput, iOutputLenW + 1);
// unlock hDevNamesW
GlobalUnlock(hDevNamesW);
// declare success
*phDevNamesW = hDevNamesW;
hr = S_OK;
}
else
{
// GlobalLock failed
hr = Win32Error2HRESULT(GetLastError());
GlobalFree(hDevNamesW);
}
}
else
{
// GlobalAlloc failed
hr = E_OUTOFMEMORY;
}
// unlock hDevNamesA
GlobalUnlock(hDevNamesA);
}
else
{
// GlobalLock failed
hr = Win32Error2HRESULT(GetLastError());
}
}
else
{
// some of the arguments are invalid (NULL)
hr = E_INVALIDARG;
}
return hr;
}
/*++
Routine Name:
ThunkDevNamesW2A
Routine Description:
Converts UNICODE DEVNAMES structure to ANSI
on failure we don't release *phDevNamesA
Arguments:
hDevNamesW - [in] handle to UNICODE DEVNAMES
phDevNamesA - [in, out] handle to ANSI DEVNAMES
Return Value:
S_OK if succeded and OLE error otherwise
History:
Lazar Ivanov (LazarI), Mar-07-2001 - created.
--*/
HRESULT
ThunkDevNamesW2A(
IN HGLOBAL hDevNamesW,
IN OUT HGLOBAL *phDevNamesA
)
{
HRESULT hr = E_FAIL;
if (hDevNamesW && phDevNamesA)
{
LPDEVNAMES pDNW = (LPDEVNAMES)GlobalLock(hDevNamesW);
if (pDNW)
{
// calculate the input string pointers
LPWSTR pwszDriver = reinterpret_cast<LPWSTR>(pDNW) + pDNW->wDriverOffset;
LPWSTR pwszDevice = reinterpret_cast<LPWSTR>(pDNW) + pDNW->wDeviceOffset;
LPWSTR pwszOutput = reinterpret_cast<LPWSTR>(pDNW) + pDNW->wOutputOffset;
// calculate the lengths of the ANSI strings
SIZE_T iDriverLenA = WideCharToMultiByte(CP_ACP, 0, pwszDriver, -1, NULL, 0, NULL, NULL);
SIZE_T iDeviceLenA = WideCharToMultiByte(CP_ACP, 0, pwszDevice, -1, NULL, 0, NULL, NULL);
SIZE_T iOutputLenA = WideCharToMultiByte(CP_ACP, 0, pwszOutput, -1, NULL, 0, NULL, NULL);
// calculate the output buffer length
SIZE_T iBytesTotal = sizeof(DEVNAMES) + sizeof(CHAR) *
((iDriverLenA + 1) + (iDeviceLenA + 1) + (iOutputLenA + 1) + DN_PADDINGCHARS);
HGLOBAL hDevNamesA = (*phDevNamesA) ?
GlobalReAlloc(*phDevNamesA, iBytesTotal, GHND) :
GlobalAlloc(GHND, iBytesTotal);
if (hDevNamesA)
{
// thunk DEVNAMES...
LPDEVNAMES pDNA = (LPDEVNAMES )GlobalLock(hDevNamesA);
if (pDNA)
{
// calculate the offsets
// note: the offsets are in chars not bytes!!
pDNA->wDriverOffset = sizeof(DEVNAMES) / sizeof(CHAR);
pDNA->wDeviceOffset = pDNA->wDriverOffset + iDriverLenA + 1;
pDNA->wOutputOffset = pDNA->wDeviceOffset + iDeviceLenA + 1;
pDNA->wDefault = pDNW->wDefault;
// calculate the output string pointers
LPSTR pszDriver = reinterpret_cast<LPSTR>(pDNA) + pDNA->wDriverOffset;
LPSTR pszDevice = reinterpret_cast<LPSTR>(pDNA) + pDNA->wDeviceOffset;
LPSTR pszOutput = reinterpret_cast<LPSTR>(pDNA) + pDNA->wOutputOffset;
// convert from uniciode to ansi
WideCharToMultiByte(CP_ACP, 0, pwszDriver, -1, pszDriver, iDriverLenA + 1, NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pwszDevice, -1, pszDevice, iDeviceLenA + 1, NULL, NULL);
WideCharToMultiByte(CP_ACP, 0, pwszOutput, -1, pszOutput, iOutputLenA + 1, NULL, NULL);
// unlock hDevNamesA
GlobalUnlock(hDevNamesA);
// declare success
*phDevNamesA = hDevNamesA;
hr = S_OK;
}
else
{
// GlobalLock failed
hr = Win32Error2HRESULT(GetLastError());
GlobalFree(hDevNamesW);
}
}
else
{
// GlobalAlloc failed
hr = E_OUTOFMEMORY;
}
// unlock hDevNamesW
GlobalUnlock(hDevNamesW);
}
else
{
// GlobalLock failed
hr = Win32Error2HRESULT(GetLastError());
}
}
else
{
// some of the arguments are invalid (NULL)
hr = E_INVALIDARG;
}
return hr;
}
#ifdef __cplusplus
}; // extern "C"
#endif