able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/shell/osshell/migrate/ocinst.cpp

1531 lines
49 KiB
C++
Raw Permalink Normal View History

Add source files
2020-09-26 03:20:57 -05:00
#include <windows.h>
#include <shlobj.h>
#include <shlwapi.h>
#include <shlwapip.h>
#include <inetreg.h>
#include <advpub.h>
#include <debug.h>
#include <inetreg.h>
#include <shlobj.h>
#include <advpub.h>
#include <xpsp1res.h>
#include <regstr.h>
#include "shmgdefs.h"
HINSTANCE g_hInstResLib = NULL;
const WCHAR c_szIEApp[] = L"iexplore.exe";
const WCHAR c_szOEApp[] = L"msimn.exe";
const WCHAR c_szWMP[] = L"wmplayer.exe";
const WCHAR c_szMessenger[] = L"msmsgs.exe";
const WCHAR c_szIECanonicalName[] = L"IEXPLORE.EXE";
const WCHAR c_szOECanonicalName[] = L"Outlook Express";
#ifndef INTERNET_MAX_PATH_LENGTH
#define INTERNET_MAX_PATH_LENGTH 2048
#endif
#ifdef _TNOONAN_TEST_WIN64
#define _WIN64
#undef CSIDL_PROGRAM_FILESX86
#define CSIDL_PROGRAM_FILESX86 CSIDL_PROGRAM_FILES
#endif
//
// We must create our shortcuts in the same language as the install
// language. If the install language is USEnglish, we cannot use
// LoadString because MUI might redirect us to the localized string,
// so we have a hard-coded list of USEnglish strings. We need that
// anyway in the Hide case, where we may need to delete old USEnglish
// versions of the shortcut left over from previous versions of IE.
//
enum { // Localized Shortcut Name
LSN_SM_IE,
LSN_QL_IE,
LSN_SM_OE,
LSN_QL_OE,
LSN_NOSHOW_OE,
LSN_NOSHOW_IE,
#ifdef _WIN64
LSN_SM_IE32,
#endif
// special sentinel value
LSN_NONE = -1,
};
struct LOCALIZEDSHORTCUTINFO
{
LPCWSTR pszExe; // The application to run
LPCWSTR pszUSEnglish; // Use if native OS is USEnglish or MUI-localized
UINT idsLocalized; // Use if native OS is fully-localized (relative to xpsp1res)
UINT idsDescription; // Shortcut description (relative to xpsp1res)
BOOL fNeverShowShortcut; // Use if we should never show (only hide)
};
const LOCALIZEDSHORTCUTINFO c_rglsi[] = {
{ // LSN_SM_IE
c_szIEApp,
L"Internet Explorer",
IDS_OC_IESHORTCUTNAME_SM,
IDS_OC_IEDESCRIPTION,
FALSE,
},
{ // LSN_QL_IE
c_szIEApp,
L"Launch Internet Explorer Browser",
IDS_OC_IESHORTCUTNAME_QL,
IDS_OC_IEDESCRIPTION,
FALSE,
},
{ // LSN_SM_OE
c_szOEApp,
L"Outlook Express",
IDS_OC_OESHORTCUTNAME_SM,
IDS_OC_OEDESCRIPTION,
FALSE,
},
{ // LSN_QL_OE
c_szOEApp,
L"Launch Outlook Express",
IDS_OC_OESHORTCUTNAME_QL,
IDS_OC_OEDESCRIPTION,
TRUE,
},
{ // LSN_NOSHOW_OE
c_szOEApp,
L"Outlook Express",
IDS_OC_OESHORTCUTNAME_SM,
IDS_OC_OEDESCRIPTION,
TRUE,
},
{ // LSN_NOSHOW_IE
c_szIEApp,
L"Internet Explorer",
IDS_OC_OESHORTCUTNAME_SM,
IDS_OC_OEDESCRIPTION,
TRUE,
},
#ifdef _WIN64
{ // LSN_SM_IE32
NULL, // Special-cased in CreateShortcut since it's not in App Paths
L"Internet Explorer (32-bit)",
IDS_OC_IESHORTCUTNAME_SM64,
IDS_OC_IEDESCRIPTION,
FALSE,
},
#endif
};
#ifdef _WIN64
WCHAR g_szIE32Path[MAX_PATH];
#endif
const WCHAR c_szIMN[] = L"Internet Mail and News";
const WCHAR c_szNTOS[] = L"Microsoft(R) Windows NT(TM) Operating System";
const WCHAR c_szHotmail[] = L"Hotmail";
const WCHAR c_szUrlDll[] = L"url.dll";
const WCHAR c_szMailNewsDll[] = L"mailnews.dll";
const WCHAR c_szIEInstallInfoKey[] = L"Software\\Clients\\StartMenuInternet\\IEXPLORE.EXE\\InstallInfo";
const WCHAR c_szOEInstallInfoKey[] = L"Software\\Clients\\Mail\\Outlook Express\\InstallInfo";
const WCHAR c_szIconsVisible[] = L"IconsVisible";
const WCHAR c_szReinstallCommand[] = L"ReinstallCommand";
const WCHAR c_szHideIconsCommand[] = L"HideIconsCommand";
const WCHAR c_szShowIconsCommand[] = L"ShowIconsCommand";
const WCHAR c_szReinstallCommandOE[] = L"%SystemRoot%\\system32\\shmgrate.exe OCInstallReinstallOE";
const WCHAR c_szHideIconsCommandOE[] = L"%SystemRoot%\\system32\\shmgrate.exe OCInstallHideOE";
const WCHAR c_szShowIconsCommandOE[] = L"%SystemRoot%\\system32\\shmgrate.exe OCInstallShowOE";
const WCHAR c_szStartMenuInternetClientKey[] = L"Software\\Clients\\StartMenuInternet";
const WCHAR c_szMailClientKey[] = L"Software\\Clients\\Mail";
#ifdef _WIN64
const WCHAR c_szMailClientKeyWOW32[] = L"Software\\Wow6432Node\\Clients\\Mail";
#endif
const WCHAR c_szJavaVMKey[] = L"CLSID\\{08B0E5C0-4FCB-11CF-AAA5-00401C608501}";
const WCHAR c_szInitiallyClearKey[] = L"Software\\Microsoft\\Active Setup\\Installed Components\\InitiallyClear";
const WCHAR c_szIEAccessKey[] = L"Software\\Microsoft\\Active Setup\\Installed Components\\>{26923b43-4d38-484f-9b9e-de460746276c}";
const WCHAR c_szOEAccessKey[] = L"Software\\Microsoft\\Active Setup\\Installed Components\\>{881dd1c5-3dcf-431b-b061-f3f88e8be88a}";
const WCHAR c_szIsInstalled[] = L"IsInstalled";
const WCHAR c_szLocale[] = L"Locale";
const WCHAR c_szVersion[] = L"Version";
const WCHAR c_szOCManagerSubComponents[] = L"Software\\Microsoft\\Windows\\CurrentVersion\\Setup\\OC Manager\\Subcomponents";
const WCHAR c_szIEAccess[] = L"IEAccess";
const WCHAR c_szOEAccess[] = L"OEAccess";
const WCHAR c_szKeyComponentFormat[] = L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\CLSID\\%s\\ShellFolder";
const WCHAR c_szIEDesktopCLSID[] = L"{871C5380-42A0-1069-A2EA-08002B30309D}";
const WCHAR c_szShellFolder[] = L"ShellFolder";
const WCHAR c_szAttribute[] = L"Attributes";
void ShellFree(LPITEMIDLIST pidl)
{
if (pidl)
{
IMalloc *pMalloc;
if (SUCCEEDED(SHGetMalloc(&pMalloc)))
{
pMalloc->Free(pidl);
pMalloc->Release();
}
}
}
HINSTANCE GetResLibInstance()
{
if (NULL == g_hInstResLib)
{
g_hInstResLib = LoadLibraryEx(L"xpsp1res.dll", NULL, LOAD_LIBRARY_AS_DATAFILE);
}
return g_hInstResLib;
}
BOOL IsServer()
{
OSVERSIONINFOEX osvi;
osvi.dwOSVersionInfoSize = sizeof(osvi);
return (GetVersionEx((OSVERSIONINFO *)&osvi) &&
((VER_NT_SERVER == osvi.wProductType) || (VER_NT_DOMAIN_CONTROLLER == osvi.wProductType)));
}
LONG GetStringValue(
IN HKEY hkey,
IN LPCWSTR pwszSubKey, OPTIONAL
IN LPCWSTR pwszValue, OPTIONAL
OUT LPVOID pvData, OPTIONAL
OUT LPDWORD pcbData) OPTIONAL
{
DWORD dwType;
DWORD cbData = pcbData ? *pcbData : 0;
LONG lResult = SHGetValueW(hkey, pwszSubKey, pwszValue, &dwType, pvData, pcbData);
if ((ERROR_SUCCESS == lResult) && (REG_SZ == dwType))
{
// NULL terminate this puppy...
if (pvData && cbData)
{
WCHAR *psz = (WCHAR *)pvData;
psz[(cbData / sizeof(WCHAR)) - 1] = 0;
}
}
else
{
lResult = ERROR_BADKEY;
}
return lResult;
}
inline HRESULT FailedHresultFromWin32()
{
DWORD dwGLE = GetLastError();
return (dwGLE != NOERROR) ? HRESULT_FROM_WIN32(dwGLE) : E_UNEXPECTED;
}
void SendChangeNotification(int csidl)
{
LPITEMIDLIST pidl;
if (SUCCEEDED(SHGetSpecialFolderLocation(NULL, csidl, &pidl)))
{
SHChangeNotify(SHCNE_UPDATEDIR, SHCNF_IDLIST, pidl, NULL);
ShellFree(pidl);
}
}
BOOL IsNtSetupRunning()
{
BOOL fSetupRunning = FALSE;
DWORD dwSetupRunning;
DWORD cbValue = sizeof(dwSetupRunning);
long lResult = SHGetValue(HKEY_LOCAL_MACHINE, L"system\\Setup", L"SystemSetupInProgress", NULL, &dwSetupRunning, &cbValue);
if ((ERROR_SUCCESS == lResult) && (dwSetupRunning))
{
fSetupRunning = TRUE;
}
else
{
cbValue = sizeof(dwSetupRunning);
lResult = SHGetValue(HKEY_LOCAL_MACHINE, L"system\\Setup", L"UpgradeInProgress", NULL, &dwSetupRunning, &cbValue);
if ((ERROR_SUCCESS == lResult) && (dwSetupRunning))
{
fSetupRunning = TRUE;
}
}
return fSetupRunning;
}
BOOL IsIgnorableClientProgram(LPCWSTR pszCurrentClient, LPCWSTR *ppszIgnoreList)
{
BOOL fResult = (lstrlen(pszCurrentClient) == 0);
if (!fResult && (NULL != ppszIgnoreList))
{
while (NULL != *ppszIgnoreList)
{
if (0 == StrCmpI(pszCurrentClient, *ppszIgnoreList))
{
fResult = TRUE;
break;
}
ppszIgnoreList++;
}
}
return fResult;
}
void SetDefaultClientProgram(HKEY hkeyRoot, LPCWSTR pszClientKey, LPCWSTR pszCanonicalName, LPCWSTR *ppszIgnoreList, BOOL fShow, BOOL fForce)
{
HKEY hKey;
LONG lResult = RegOpenKeyEx(hkeyRoot, pszClientKey, 0, KEY_QUERY_VALUE | KEY_SET_VALUE, &hKey);
if (ERROR_SUCCESS == lResult)
{
WCHAR szCurrentClient[MAX_PATH];
DWORD cbCurrentClient = sizeof(szCurrentClient);
lResult = GetStringValue(hKey, NULL, NULL, szCurrentClient, &cbCurrentClient);
// If we're meant to show ourselves then just show set the key
if (fShow)
{
if (fForce ||
((ERROR_SUCCESS != lResult) || IsIgnorableClientProgram(szCurrentClient, ppszIgnoreList)))
{
RegSetValue(hKey, NULL, REG_SZ, pszCanonicalName, lstrlen(pszCanonicalName) * sizeof(WCHAR));
}
}
else
{
// If we're meant to disappear then clear the key only if we're the one that is currently set.
if (fForce || ((ERROR_SUCCESS == lResult) && (0 == StrCmpI(pszCanonicalName, szCurrentClient))))
{
RegSetValue(hKey, NULL, REG_SZ, L"", 0);
}
}
RegCloseKey(hKey);
}
else
{
TraceMsg(TF_ERROR, "Error opening client key %s - 0x%08x", pszClientKey, lResult);
}
}
BOOL IsInstalled(LPCWSTR pszComponent)
{
DWORD dwType;
DWORD dwValue;
DWORD cbValue = sizeof(dwValue);
DWORD dwResult = SHGetValueW(HKEY_LOCAL_MACHINE, pszComponent, c_szIsInstalled, &dwType, &dwValue, &cbValue);
BOOL fIsInstalled = ((ERROR_SUCCESS == dwResult) && (REG_DWORD == dwType) && (dwValue));
return fIsInstalled;
}
void SetBool(LPCWSTR pszKey, LPCWSTR pszValue, BOOL fValue)
{
HKEY hKey;
DWORD dwDisposition;
LONG lResult = RegCreateKeyEx(HKEY_LOCAL_MACHINE, pszKey, 0, NULL, REG_OPTION_NON_VOLATILE,
KEY_SET_VALUE, NULL, &hKey, &dwDisposition);
if (ERROR_SUCCESS == lResult)
{
DWORD dwValue = fValue ? TRUE : FALSE;
lResult = RegSetValueEx(hKey, pszValue, 0, REG_DWORD, (LPBYTE)&dwValue, sizeof(dwValue));
if (ERROR_SUCCESS != lResult)
{
TraceMsg(TF_ERROR, "Error setting value %s - 0x%08x", pszValue, lResult);
}
RegCloseKey(hKey);
}
else
{
TraceMsg(TF_ERROR, "Error creating %s key - 0x%08x", pszKey, lResult);
}
}
//Copy data from HKLM to HKCU
LONG CopyRegValue(LPCWSTR pszSubKey, LPCWSTR pszValue)
{
HKEY hKeySrc;
LONG lResult;
lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, pszSubKey, 0, KEY_QUERY_VALUE, &hKeySrc);
if (ERROR_SUCCESS == lResult)
{
BYTE buffer[128];
DWORD dwSize = sizeof(buffer);
lResult = RegQueryValueEx(hKeySrc, pszValue, NULL, NULL, (LPBYTE)buffer, &dwSize);
RegCloseKey(hKeySrc);
if (ERROR_SUCCESS == lResult)
{
HKEY hKeyDst;
lResult = RegCreateKeyEx(HKEY_CURRENT_USER, pszSubKey, 0, NULL,
REG_OPTION_NON_VOLATILE, KEY_SET_VALUE, NULL, &hKeyDst, NULL);
if (ERROR_SUCCESS == lResult)
{
lResult = RegSetValueEx(hKeyDst, pszValue, NULL, REG_SZ, (CONST BYTE *)buffer, dwSize);
}
RegCloseKey(hKeyDst);
}
}
return lResult;
}
void UpdateActiveSetupValues(LPCWSTR pszKeyName, BOOL fInstall)
{
if (fInstall)
{
CopyRegValue(pszKeyName, c_szLocale);
CopyRegValue(pszKeyName, c_szVersion);
}
else
{
RegDeleteKey(HKEY_CURRENT_USER, pszKeyName);
}
}
LONG GetExeAppPathWorker(LPCWSTR pszAppName, LPWSTR pszExePath, UINT cchExePath, LPCWSTR pszFmt)
{
WCHAR szExeKey[MAX_PATH];
HKEY hKey;
wnsprintf(szExeKey, ARRAYSIZE(szExeKey), pszFmt, pszAppName);
LONG lResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, szExeKey, 0, KEY_QUERY_VALUE, &hKey);
if (ERROR_SUCCESS == lResult)
{
LONG cbExePath = cchExePath * sizeof(WCHAR);
lResult = RegQueryValue(hKey, NULL, pszExePath, &cbExePath);
if (ERROR_SUCCESS != lResult)
{
TraceMsg(TF_ERROR, "Error querying for app path value - 0x%08x", lResult);
}
RegCloseKey(hKey);
}
return lResult;
}
LONG GetExeAppPath(LPCWSTR pszAppName, LPWSTR pszExePath, UINT cchExePath)
{
return GetExeAppPathWorker(pszAppName, pszExePath, cchExePath, L"Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\%s");
}
#ifdef _WIN64
LONG GetWow32ExeAppPath(LPCWSTR pszAppName, LPWSTR pszExePath, UINT cchExePath)
{
return GetExeAppPathWorker(pszAppName, pszExePath, cchExePath, L"Software\\Wow6432Node\\Microsoft\\Windows\\CurrentVersion\\App Paths\\%s");
}
#endif
void ShowHideFile(LPCWSTR pszFileName, BOOL fShow)
{
DWORD dwAttrs = GetFileAttributes(pszFileName);
if (fShow)
{
dwAttrs &= ~(FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM);
}
else
{
dwAttrs |= FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM;
}
SetFileAttributes(pszFileName, dwAttrs);
}
void ShowHideExe(LPCWSTR pszAppName, BOOL fShow)
{
WCHAR szExePath[MAX_PATH];
if (ERROR_SUCCESS == GetExeAppPath(pszAppName, szExePath, ARRAYSIZE(szExePath)))
{
ShowHideFile(szExePath, fShow);
}
}
#ifdef _WIN64
void ShowHideWow32Exe(LPCWSTR pszAppName, BOOL fShow)
{
WCHAR szExePath[MAX_PATH];
if (ERROR_SUCCESS == GetWow32ExeAppPath(pszAppName, szExePath, ARRAYSIZE(szExePath)))
{
ShowHideFile(szExePath, fShow);
}
}
#endif
void CreateShortcut(LPCWSTR pszExePath, LPWSTR pszLinkFullFilePath, int idName, int idDescription)
{
IShellLink *pShellLink;
HRESULT hr = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (void **)&pShellLink);
if (SUCCEEDED(hr))
{
if (idDescription)
{
WCHAR szDescription[MAX_PATH];
wnsprintf(szDescription, ARRAYSIZE(szDescription), L"@xpsp1res.dll,-%d", idDescription);
pShellLink->SetDescription(szDescription);
}
pShellLink->SetWorkingDirectory(L"%HOMEDRIVE%%HOMEPATH%");
if (SUCCEEDED(pShellLink->SetPath(pszExePath)))
{
IPersistFile *pPersistFile;
if (SUCCEEDED(pShellLink->QueryInterface(IID_IPersistFile, (void **)&pPersistFile)))
{
if (SUCCEEDED(pPersistFile->Save(pszLinkFullFilePath, TRUE)))
{
SHSetLocalizedName(pszLinkFullFilePath, L"xpsp1res.dll", idName);
}
pPersistFile->Release();
}
}
pShellLink->Release();
}
else
{
TraceMsg(TF_ERROR, "Couldn't create shell link object - hr = 0x%08x", hr);
}
}
void FindAndNukeIcons(LPCWSTR pszLinkPath, LPCWSTR pszLongExePath)
{
WCHAR szStartDir[MAX_PATH];
WCHAR szShortExePath[MAX_PATH];
DWORD cch = GetShortPathName(pszLongExePath, szShortExePath, ARRAYSIZE(szShortExePath));
if (!cch || (cch >= ARRAYSIZE(szShortExePath)))
{
szShortExePath[0] = 0;
}
if (GetCurrentDirectory(ARRAYSIZE(szStartDir), szStartDir))
{
if (SetCurrentDirectory(pszLinkPath))
{
WIN32_FIND_DATA fd;
HANDLE hFind = FindFirstFile(L"*.lnk", &fd);
if (INVALID_HANDLE_VALUE != hFind)
{
IShellLink *pShellLink;
HRESULT hr = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (void **)&pShellLink);
if (SUCCEEDED(hr))
{
IPersistFile *pPersistFile;
hr = pShellLink->QueryInterface(IID_IPersistFile, (void **)&pPersistFile);
if (SUCCEEDED(hr))
{
do
{
WCHAR szLinkPath[MAX_PATH];
WCHAR szArgs[INTERNET_MAX_PATH_LENGTH];
if (SUCCEEDED(pPersistFile->Load(fd.cFileName, STGM_READ)) &&
SUCCEEDED(pShellLink->GetPath(szLinkPath, ARRAYSIZE(szLinkPath), NULL, 0)) &&
((0 == StrCmpI(pszLongExePath, szLinkPath)) || (0 == StrCmpI(szShortExePath, szLinkPath))) &&
SUCCEEDED(pShellLink->GetArguments(szArgs, ARRAYSIZE(szArgs))))
{
PathRemoveBlanks(szArgs);
if (!szArgs[0])
{
SetFileAttributes(fd.cFileName, FILE_ATTRIBUTE_NORMAL);
DeleteFile(fd.cFileName);
}
}
}
while (FindNextFile(hFind, &fd));
pPersistFile->Release();
}
pShellLink->Release();
}
FindClose(hFind);
}
}
SetCurrentDirectory(szStartDir);
}
}
void ShowShortcut(LPCWSTR pszLinkPath, int lsn, BOOL fShow)
{
WCHAR szLinkFullFilePath[MAX_PATH];
WCHAR szLinkFileName[MAX_PATH];
szLinkFileName[0] = 0;
// If we should use the localized name as the filename, then try to get it.
// We should use the localized name if the install language is not USEnglish.
if (LANGIDFROMLCID(GetSystemDefaultUILanguage()) != MAKELANGID(LANG_ENGLISH, SUBLANG_ENGLISH_US))
{
LoadString(GetResLibInstance(), c_rglsi[lsn].idsLocalized,
szLinkFileName, ARRAYSIZE(szLinkFileName));
}
// If we couldn't (or shouldn't) get the localized name, then
// use the USEnglish name instead.
if (!szLinkFileName[0])
{
lstrcpyn(szLinkFileName, c_rglsi[lsn].pszUSEnglish, ARRAYSIZE(szLinkFileName));
}
PathAddExtension(szLinkFileName, L".lnk");
PathCombine(szLinkFullFilePath, pszLinkPath, szLinkFileName);
WCHAR szExePath[MAX_PATH];
LONG lResult;
#ifdef _WIN64
if (lsn == LSN_SM_IE32)
{
StrCpyN(szExePath, g_szIE32Path, ARRAYSIZE(szExePath));
lResult = ERROR_SUCCESS;
}
else
#endif
{
lResult = GetExeAppPath(c_rglsi[lsn].pszExe, szExePath, ARRAYSIZE(szExePath));
}
if (ERROR_SUCCESS == lResult)
{
if (fShow)
{
if (!c_rglsi[lsn].fNeverShowShortcut)
{
CreateShortcut(szExePath, szLinkFullFilePath, c_rglsi[lsn].idsLocalized, c_rglsi[lsn].idsDescription);
}
}
else
{
FindAndNukeIcons(pszLinkPath, szExePath);
}
}
}
LONG QueryExplorerCLSIDFlags(LPCWSTR pszCLSID, DWORD *pdwFlags)
{
WCHAR szKeyComponent[MAX_PATH];
wnsprintf(szKeyComponent, ARRAYSIZE(szKeyComponent), c_szKeyComponentFormat, pszCLSID);
DWORD dwSize = sizeof(*pdwFlags);
return SHGetValueW(HKEY_CURRENT_USER, szKeyComponent, c_szAttribute, NULL, pdwFlags, &dwSize);
}
void SetExplorerCLSIDFlags(LPCWSTR pszCLSID, DWORD dwAndFlags, DWORD dwOrFlags)
{
DWORD dwFlags = 0;
WCHAR szKeyComponent[MAX_PATH];
wnsprintf(szKeyComponent, ARRAYSIZE(szKeyComponent), c_szKeyComponentFormat, pszCLSID);
DWORD dwSize = sizeof(dwFlags);
SHGetValueW(HKEY_CURRENT_USER, szKeyComponent, c_szAttribute, NULL, &dwFlags, &dwSize);
dwFlags &= dwAndFlags;
dwFlags |= dwOrFlags;
SHSetValue(HKEY_CURRENT_USER, szKeyComponent, c_szAttribute, REG_DWORD, &dwFlags, sizeof(dwFlags));
}
void NukeFiles(LPCWSTR pszPath, LPCWSTR pszFileSpec)
{
WCHAR szStartDir[MAX_PATH];
if (GetCurrentDirectory(ARRAYSIZE(szStartDir), szStartDir))
{
if (SetCurrentDirectory(pszPath))
{
WIN32_FIND_DATA fd;
HANDLE hFind = FindFirstFile(pszFileSpec, &fd);
if (INVALID_HANDLE_VALUE != hFind)
{
do
{
SetFileAttributes(fd.cFileName, FILE_ATTRIBUTE_NORMAL);
DeleteFile(fd.cFileName);
}
while (FindNextFile(hFind, &fd));
FindClose(hFind);
}
}
SetCurrentDirectory(szStartDir);
}
}
void NukeDesktopCleanupIcons()
{
WCHAR szPath[MAX_PATH];
HINSTANCE hInstCleaner = LoadLibraryEx(L"fldrclnr.dll", NULL, LOAD_LIBRARY_AS_DATAFILE);
WCHAR szUnusedShortcutFolder[MAX_PATH];
// From nt\shell\applets\cleanup\fldrclnr\resource.h:
#define IDS_ARCHIVEFOLDER 8
#define IDS_ARCHIVEFOLDER_FIRSTBOOT 15
// From nt\shell\applets\cleanup\fldrclnr\cleanupwiz.h:
#define REGSTR_OEM_PATH REGSTR_PATH_SETUP TEXT("\\OemStartMenuData")
#define REGSTR_OEM_TITLEVAL TEXT("DesktopShortcutsFolderName")
if (SHGetSpecialFolderPath(NULL, szPath, CSIDL_DESKTOPDIRECTORY, FALSE))
{
NukeFiles(szPath, L"*.{871C5380-42A0-1069-A2EA-08002B30309D}");
if (hInstCleaner &&
LoadString(hInstCleaner, IDS_ARCHIVEFOLDER, szUnusedShortcutFolder, ARRAYSIZE(szUnusedShortcutFolder)))
{
PathAppend(szPath, szUnusedShortcutFolder);
NukeFiles(szPath, L"*.{871C5380-42A0-1069-A2EA-08002B30309D}");
}
}
DWORD cb = sizeof(szUnusedShortcutFolder);
DWORD dwType;
// Get the folder name from either the registry or fldrclnr.dll and get the startmenu\programs folder
if (
(
(ERROR_SUCCESS == SHGetValue(HKEY_LOCAL_MACHINE, REGSTR_OEM_PATH, REGSTR_OEM_TITLEVAL, &dwType, szUnusedShortcutFolder, &cb))
||
(hInstCleaner && LoadString(hInstCleaner, IDS_ARCHIVEFOLDER_FIRSTBOOT, szUnusedShortcutFolder, ARRAYSIZE(szUnusedShortcutFolder)))
)
&&
SHGetSpecialFolderPath(NULL, szPath, CSIDL_PROGRAMS, FALSE)
)
{
PathAppend(szPath, szUnusedShortcutFolder);
NukeFiles(szPath, L"*.{871C5380-42A0-1069-A2EA-08002B30309D}");
}
if (hInstCleaner)
{
FreeLibrary(hInstCleaner);
}
}
void RemoveIcons(int lsn, int *pcsidlList, DWORD nEntries)
{
for (DWORD i = 0; i < nEntries; i++)
{
WCHAR szPath[MAX_PATH];
if (SHGetSpecialFolderPath(NULL, szPath, pcsidlList[i], FALSE))
{
ShowShortcut(szPath, lsn, FALSE);
}
}
}
void RemoveAnAnnoyingIESubFolderIcon(int csidl)
{
WCHAR szPath[MAX_PATH];
if (SHGetSpecialFolderPath(NULL, szPath, csidl, FALSE))
{
WCHAR szIEFolder[MAX_PATH];
PathCombine(szIEFolder, szPath, L"Internet Explorer");
ShowShortcut(szIEFolder, LSN_NOSHOW_IE, FALSE);
WCHAR szLocalizedName[MAX_PATH];
LoadString(GetResLibInstance(), IDS_OC_IESHORTCUTNAME_SM, szLocalizedName, ARRAYSIZE(szLocalizedName));
PathRemoveExtension(szLocalizedName);
PathCombine(szIEFolder, szPath, szLocalizedName);
ShowShortcut(szIEFolder, LSN_NOSHOW_IE, FALSE);
LoadString(GetResLibInstance(), IDS_OC_IESHORTCUTNAME_QL, szLocalizedName, ARRAYSIZE(szLocalizedName));
PathRemoveExtension(szLocalizedName);
PathCombine(szIEFolder, szPath, szLocalizedName);
ShowShortcut(szIEFolder, LSN_NOSHOW_IE, FALSE);
}
}
void RemoveAnnoyingIESubFolderIcons()
{
RemoveAnAnnoyingIESubFolderIcon(CSIDL_PROGRAMS);
RemoveAnAnnoyingIESubFolderIcon(CSIDL_COMMON_PROGRAMS);
}
void ShowIEDesktopIcon(BOOL fShow)
{
if (fShow)
{
SetExplorerCLSIDFlags(c_szIEDesktopCLSID, ~SFGAO_NONENUMERATED, 0);
}
else
{
NukeDesktopCleanupIcons();
SetExplorerCLSIDFlags(c_szIEDesktopCLSID, 0xffffffff, SFGAO_NONENUMERATED);
}
SendChangeNotification(CSIDL_DESKTOP);
SendChangeNotification(CSIDL_DESKTOPDIRECTORY);
}
// Ensure a handler is in place for all the important things
void FixupIEAssociations(BOOL fForceAssociations)
{
// In order for shdocvw to do its magic, IEXPLORE.EXE must be properly
// registered. Setup runs OC Manager before it runs IE.INF, so don't
// call shdocvw to try to do something he can't do. (He'll assert if
// you try.)
HKEY hk;
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("Software\\Microsoft\\Windows\\CurrentVersion\\App Paths\\IEXPLORE.EXE"),
0, KEY_READ, &hk) != ERROR_SUCCESS)
{
// ie.inf hasn't run yet - shdocvw can't do its thing
return;
}
RegCloseKey(hk);
// Shdocvw does a pretty good job of this, let's give him a crack at it
HINSTANCE hinst = LoadLibrary(L"shdocvw.dll");
if (NULL != hinst)
{
typedef HRESULT (WINAPI *DLLINSTALL)(BOOL bInstall, LPCWSTR pszCmdLine);
DLLINSTALL pfnDllInstall = (DLLINSTALL)GetProcAddress(hinst, "DllInstall");
if (pfnDllInstall)
{
pfnDllInstall(TRUE, (fForceAssociations ? L"ForceAssoc" : L""));
}
else
{
TraceMsg(TF_ERROR, "Error getting DllInstall entry point - GLE = 0x%08x", GetLastError());
}
FreeLibrary(hinst);
}
else
{
TraceMsg(TF_ERROR, "Error loading shdocvw - GLE = 0x%08x", GetLastError());
}
}
// This will create or delete shortcuts in Start Menu\Programs and the Quick Launch bar
void ShowUserShortcuts(int lsnSM, int lsnQL, BOOL fShow)
{
WCHAR szPath[MAX_PATH];
if (lsnSM >= 0 && SHGetSpecialFolderPath(NULL, szPath, CSIDL_PROGRAMS, fShow))
{
ShowShortcut(szPath, lsnSM, fShow);
}
if (lsnQL >= 0 && SHGetSpecialFolderPath(NULL, szPath, CSIDL_APPDATA, fShow))
{
WCHAR szQuickLaunchPath[MAX_PATH];
LoadString(GetResLibInstance(), IDS_OC_QLAUNCHAPPDATAPATH, szQuickLaunchPath, ARRAYSIZE(szQuickLaunchPath));
PathAppend(szPath, szQuickLaunchPath);
// In case we're the first ones through, create the Quick Launch dir
CreateDirectory(szPath, NULL);
ShowShortcut(szPath, lsnQL, fShow);
}
}
// Internet Explorer optional component goo
void UserConfigIE()
{
BOOL fIsInstalled = IsInstalled(c_szIEAccessKey);
ShowIEDesktopIcon(fIsInstalled);
ShowUserShortcuts(LSN_SM_IE, LSN_QL_IE, fIsInstalled);
#ifdef _WIN64
WCHAR szLinkDir[MAX_PATH];
if (SHGetSpecialFolderPath(NULL, szLinkDir, CSIDL_PROGRAMS, fIsInstalled))
{
WCHAR szExePath[MAX_PATH];
if (SHGetSpecialFolderPath(NULL, szExePath, CSIDL_PROGRAM_FILESX86, FALSE))
{
PathCombine(g_szIE32Path, szExePath, L"Internet Explorer\\iexplore.exe");
ShowShortcut(szLinkDir, LSN_SM_IE32, fIsInstalled);
}
}
#endif
if (!fIsInstalled)
{
int csidlList[] = { CSIDL_DESKTOPDIRECTORY, CSIDL_COMMON_DESKTOPDIRECTORY, CSIDL_COMMON_PROGRAMS };
RemoveIcons(LSN_NOSHOW_IE, csidlList, ARRAYSIZE(csidlList));
RemoveAnnoyingIESubFolderIcons();
}
SHSendMessageBroadcast(WM_SETTINGCHANGE, 0, (LPARAM)L"Software\\Clients\\StartMenuInternet");
}
void SetIEShowHideFlags(BOOL fShow)
{
SetBool(c_szIEAccessKey, c_szIsInstalled, fShow);
SetBool(c_szIEInstallInfoKey, c_szIconsVisible, fShow);
SetBool(c_szOCManagerSubComponents, c_szIEAccess, fShow);
}
void ShowHideIE(BOOL fShow, BOOL fForceAssociations, BOOL fNonUpdateInstall)
{
SetIEShowHideFlags(fShow);
if (fShow)
{
if (fNonUpdateInstall)
{
FixupIEAssociations(fForceAssociations);
}
SetDefaultClientProgram(HKEY_LOCAL_MACHINE, c_szStartMenuInternetClientKey, c_szIECanonicalName, NULL, TRUE, fForceAssociations);
}
else
{
SetDefaultClientProgram(HKEY_LOCAL_MACHINE, c_szStartMenuInternetClientKey, c_szIECanonicalName, NULL, FALSE, FALSE);
int csidlList[] = { CSIDL_COMMON_DESKTOPDIRECTORY, CSIDL_COMMON_PROGRAMS };
RemoveIcons(LSN_NOSHOW_IE, csidlList, ARRAYSIZE(csidlList));
RemoveAnAnnoyingIESubFolderIcon(CSIDL_COMMON_PROGRAMS);
}
if (fNonUpdateInstall && !IsNtSetupRunning())
{
SetDefaultClientProgram(HKEY_CURRENT_USER, c_szStartMenuInternetClientKey, c_szIECanonicalName, NULL, FALSE, fShow);
UserConfigIE();
UpdateActiveSetupValues(c_szIEAccessKey, fShow);
SendChangeNotification(CSIDL_DESKTOP);
}
else
{
SHSendMessageBroadcast(WM_SETTINGCHANGE, 0, (LPARAM)L"Software\\Clients\\StartMenuInternet");
}
ShowHideExe(c_szIEApp, fShow);
#ifdef _WIN64
ShowHideWow32Exe(c_szIEApp, fShow);
#endif
}
// Outlook Express optional component goo
void UserConfigOE()
{
BOOL fIsInstalled = IsInstalled(c_szOEAccessKey);
ShowUserShortcuts(LSN_SM_OE, LSN_QL_OE, fIsInstalled);
SHSendMessageBroadcastW(WM_WININICHANGE, 0, (LPARAM)REGSTR_PATH_MAILCLIENTS);
if (!fIsInstalled)
{
// ShowUserShortcuts(fInstalled=FALSE) already cleaned up any
// old OE Quick Launch shortcuts, so all that's left to remove
// is the desktop icon
int csidlList[] = { CSIDL_DESKTOPDIRECTORY, CSIDL_COMMON_DESKTOPDIRECTORY, CSIDL_COMMON_PROGRAMS };
RemoveIcons(LSN_NOSHOW_OE, csidlList, ARRAYSIZE(csidlList));
}
}
void SetOEShowHideFlags(BOOL fShow)
{
SetBool(c_szOEAccessKey, c_szIsInstalled, fShow);
SetBool(c_szOEInstallInfoKey, c_szIconsVisible, fShow);
SetBool(c_szOCManagerSubComponents, c_szOEAccess, fShow);
}
void FixupOEAssociations(BOOL fForceAssociations, LPCWSTR *ppszIgnoreList)
{
if (!IsNtSetupRunning())
{
long lResult;
if (!fForceAssociations)
{
WCHAR szValue[MAX_PATH];
DWORD cbValue = sizeof(szValue);
DWORD dwType;
lResult = SHGetValueW(HKEY_CLASSES_ROOT, L"mailto\\shell\\open\\command", NULL, &dwType, szValue, &cbValue);
if ((REG_SZ == dwType) &&
((ERROR_MORE_DATA == lResult) || ((ERROR_SUCCESS == lResult) && (cbValue >= (2 * sizeof(WCHAR))))))
{
// Some sort of valid string data at least character in length -- see if it's one we should stomp,
// otherwise leave it alone
if (NULL != ppszIgnoreList)
{
while (NULL != *ppszIgnoreList)
{
if (NULL != StrStrIW(szValue, *ppszIgnoreList))
{
fForceAssociations = TRUE;
break;
}
ppszIgnoreList++;
}
}
}
else
{
// Either it's not a string or it's zero length
fForceAssociations = TRUE;
}
}
if (fForceAssociations)
{
HKEY hkey;
DWORD dwDisposition;
SHDeleteKey(HKEY_CLASSES_ROOT, L"mailto");
lResult = RegCreateKeyEx(HKEY_CLASSES_ROOT, L"mailto", 0, NULL, REG_OPTION_NON_VOLATILE, KEY_WRITE,
NULL, &hkey, &dwDisposition);
if (ERROR_SUCCESS == lResult)
{
SHCopyKey(HKEY_LOCAL_MACHINE, L"Software\\Clients\\Mail\\Outlook Express\\Protocols\\mailto", hkey, 0);
RegCloseKey(hkey);
}
}
}
}
void ShowHideOE(BOOL fShow, BOOL fForceAssociations, BOOL fNonUpdateInstall)
{
if (!IsServer())
{
static LPCWSTR ppszIgnoreClientsNormal[] = { c_szIMN, c_szNTOS, NULL };
static LPCWSTR ppszIgnoreClientsSetup[] = { c_szIMN, c_szNTOS, c_szHotmail, NULL };
LPCWSTR *ppszIgnoreClients = !IsNtSetupRunning() ? ppszIgnoreClientsNormal : ppszIgnoreClientsSetup;
static LPCWSTR ppszIgnoreMailto[] = { c_szUrlDll, c_szMailNewsDll, NULL };
SetOEShowHideFlags(fShow);
if (fShow)
{
FixupOEAssociations(fForceAssociations, ppszIgnoreMailto);
SetDefaultClientProgram(HKEY_LOCAL_MACHINE, c_szMailClientKey, c_szOECanonicalName, ppszIgnoreClients, TRUE, fForceAssociations);
#ifdef _WIN64
SetDefaultClientProgram(HKEY_LOCAL_MACHINE, c_szMailClientKeyWOW32, c_szOECanonicalName, ppszIgnoreClients, TRUE, fForceAssociations);
#endif
}
else
{
FixupOEAssociations(FALSE, ppszIgnoreMailto);
int csidlList[] = { CSIDL_COMMON_DESKTOPDIRECTORY, CSIDL_COMMON_PROGRAMS };
RemoveIcons(LSN_NOSHOW_OE, csidlList, ARRAYSIZE(csidlList));
SetDefaultClientProgram(HKEY_LOCAL_MACHINE, c_szMailClientKey, c_szOECanonicalName, ppszIgnoreClients, FALSE, FALSE);
#ifdef _WIN64
SetDefaultClientProgram(HKEY_LOCAL_MACHINE, c_szMailClientKeyWOW32, c_szOECanonicalName, ppszIgnoreClients, FALSE, FALSE);
#endif
}
if (fNonUpdateInstall && !IsNtSetupRunning())
{
SetDefaultClientProgram(HKEY_CURRENT_USER, c_szMailClientKey, c_szOECanonicalName, ppszIgnoreClients, FALSE, fShow);
UserConfigOE();
UpdateActiveSetupValues(c_szOEAccessKey, fShow);
SendChangeNotification(CSIDL_DESKTOP);
}
else
{
SHSendMessageBroadcastW(WM_WININICHANGE, 0, (LPARAM)REGSTR_PATH_MAILCLIENTS);
}
}
ShowHideExe(c_szOEApp, fShow);
#ifdef _WIN64
ShowHideWow32Exe(c_szOEApp, fShow);
#endif
}
HRESULT CallRegisterServer(LPCWSTR pszModule, BOOL fRegister)
{
HRESULT hr;
HINSTANCE hinst = LoadLibrary(pszModule);
if (NULL != hinst)
{
typedef HRESULT (WINAPI *DLLREGISTERSERVER)();
LPCSTR pszRegFuncName = fRegister ? "DllRegisterServer" : "DllUnregisterServer";
DLLREGISTERSERVER pfnDllRegisterServer = (DLLREGISTERSERVER)GetProcAddress(hinst, pszRegFuncName);
if (pfnDllRegisterServer)
{
hr = pfnDllRegisterServer();
}
else
{
TraceMsg(TF_ERROR, "Error getting DllRegisterServer entry point - GLE = 0x%08x", GetLastError());
hr = FailedHresultFromWin32();
}
FreeLibrary(hinst);
}
else
{
TraceMsg(TF_ERROR, "Error loading %s - GLE = 0x%08x", pszModule, GetLastError());
hr = FailedHresultFromWin32();
}
return hr;
}
void ReinstallVM()
{
#if 0
SHDeleteKey(HKEY_CLASSES_ROOT, c_szJavaVMKey);
CallRegisterServer(L"msjava.dll", TRUE);
#endif
}
void FixupMailClientKey()
{
SHSetValueW(HKEY_LOCAL_MACHINE, c_szOEInstallInfoKey, c_szReinstallCommand, REG_EXPAND_SZ, c_szReinstallCommandOE, sizeof(c_szReinstallCommandOE));
SHSetValueW(HKEY_LOCAL_MACHINE, c_szOEInstallInfoKey, c_szHideIconsCommand, REG_EXPAND_SZ, c_szHideIconsCommandOE, sizeof(c_szHideIconsCommandOE));
SHSetValueW(HKEY_LOCAL_MACHINE, c_szOEInstallInfoKey, c_szShowIconsCommand, REG_EXPAND_SZ, c_szShowIconsCommandOE, sizeof(c_szShowIconsCommandOE));
SetBool(c_szOEInstallInfoKey, c_szIconsVisible, IsInstalled(c_szOEAccessKey));
}
extern "C" void FixupOptionalComponents()
{
// Need to do this here since the app paths weren't set during setup.
ShowHideExe(c_szIEApp, IsInstalled(c_szIEAccessKey));
ShowHideExe(c_szOEApp, IsInstalled(c_szOEAccessKey));
#ifdef _WIN64
ShowHideWow32Exe(c_szIEApp, IsInstalled(c_szIEAccessKey));
ShowHideWow32Exe(c_szOEApp, IsInstalled(c_szOEAccessKey));
#endif
// OE likes to nuke the whole branch and start from scratch.
FixupMailClientKey();
}
#define FC_QUICKLAUNCH 0x00000001
#define FC_ALLPROGRAMS 0x00000002
#define FC_DESKTOP 0x00000004
#define FC_EXPLORERCLSID 0x00000008
#define FC_MAX 0x00000008
struct SHORTCUTFINDENTRY
{
LPCTSTR pszExe;
LPCTSTR pszExplorerCLSID;
DWORD dwFoldersToCheck;
DWORD dwFoldersToRecurseCheck;
};
SHORTCUTFINDENTRY scfEntries[] =
{
{ c_szIEApp, c_szIEDesktopCLSID, FC_QUICKLAUNCH | FC_ALLPROGRAMS | FC_EXPLORERCLSID, 0 },
{ c_szOEApp, NULL, FC_QUICKLAUNCH | FC_ALLPROGRAMS, 0 },
{ c_szWMP, NULL, FC_QUICKLAUNCH | FC_ALLPROGRAMS, FC_ALLPROGRAMS },
{ c_szMessenger, NULL, FC_QUICKLAUNCH | FC_ALLPROGRAMS, FC_ALLPROGRAMS },
};
HRESULT FindAnyIcons(LPCWSTR pszLinkPath, LPCWSTR pszLongExePath, BOOL fRecurse, BOOL *pfFound)
{
// Pretty straightforward here. Loop through all the .lnk files in a directory and see if we
// find a shortcut pointing to the EXE in question. This function will also recurse if asked.
WCHAR szStartDir[MAX_PATH];
HRESULT hr = E_FAIL;
*pfFound = FALSE;
WCHAR szShortExePath[MAX_PATH];
DWORD cch = GetShortPathName(pszLongExePath, szShortExePath, ARRAYSIZE(szShortExePath));
if (!cch || (cch >= ARRAYSIZE(szShortExePath)))
{
szShortExePath[0] = 0;
}
if (GetCurrentDirectory(ARRAYSIZE(szStartDir), szStartDir))
{
if (SetCurrentDirectory(pszLinkPath))
{
WIN32_FIND_DATA fd;
HANDLE hFind = FindFirstFile(fRecurse ? L"*" : L"*.lnk", &fd);
if (INVALID_HANDLE_VALUE != hFind)
{
IShellLink *pShellLink;
hr = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (void **)&pShellLink);
if (SUCCEEDED(hr))
{
IPersistFile *pPersistFile;
hr = pShellLink->QueryInterface(IID_IPersistFile, (void **)&pPersistFile);
if (SUCCEEDED(hr))
{
do
{
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
if ((0 != StrCmpI(fd.cFileName, L".")) && (0 != StrCmpI(fd.cFileName, L"..")) && fRecurse)
{
hr = FindAnyIcons(fd.cFileName, pszLongExePath, fRecurse, pfFound);
}
}
else if (0 == StrCmpIC(PathFindExtension(fd.cFileName), L".lnk"))
{
WCHAR szLinkPath[MAX_PATH];
if (SUCCEEDED(pPersistFile->Load(fd.cFileName, STGM_READ)) &&
SUCCEEDED(pShellLink->GetPath(szLinkPath, ARRAYSIZE(szLinkPath), NULL, 0)) &&
((0 == StrCmpI(pszLongExePath, szLinkPath)) || (0 == StrCmpI(szShortExePath, szLinkPath))))
{
*pfFound = TRUE;
}
}
}
while (SUCCEEDED(hr) && (FALSE == *pfFound) && FindNextFile(hFind, &fd));
pPersistFile->Release();
}
pShellLink->Release();
}
FindClose(hFind);
}
}
SetCurrentDirectory(szStartDir);
}
return hr;
}
void GetCurrentIconInvisibility()
{
// Loop through our list of "interesting" .EXEs. For each one, look in "interesting" places to see if
// the shortcut is NOT present. If the shortcut is missing, then presumably the user did this on purpose.
// The problem is that a bunch of per-user install stubs will get rerun after an SP and recreate the
// shortcuts. This is a no-no.
for (int i = 0; i < ARRAYSIZE(scfEntries); i++)
{
WCHAR szExePath[MAX_PATH];
if (ERROR_SUCCESS == GetExeAppPath(scfEntries[i].pszExe, szExePath, ARRAYSIZE(szExePath)))
{
DWORD dwInitiallyClear = 0;
// Loop through each of the bits, look to see if the icon is NOT there. If it's not there, then
// we'll set that bit in dwInitiallyClear.
//
// Note that we err on the side of leaving the bit unset so in failure cases we may allow an icon
// to be recreated -- this is generally a better approach since most of the time the icons will
// not be deleted.
for (DWORD dwFolderClass = FC_QUICKLAUNCH; dwFolderClass <= FC_MAX; dwFolderClass <<= 1)
{
if (dwFolderClass & scfEntries[i].dwFoldersToCheck)
{
if (FC_EXPLORERCLSID != dwFolderClass)
{
HRESULT hr = E_FAIL;
WCHAR szPath[MAX_PATH];
switch (dwFolderClass)
{
case FC_QUICKLAUNCH:
if (SHGetSpecialFolderPath(NULL, szPath, CSIDL_APPDATA, FALSE))
{
WCHAR szQuickLaunchPath[MAX_PATH];
LoadString(GetResLibInstance(), IDS_OC_QLAUNCHAPPDATAPATH, szQuickLaunchPath, ARRAYSIZE(szQuickLaunchPath));
PathAppend(szPath, szQuickLaunchPath);
hr = S_OK;
}
break;
case FC_ALLPROGRAMS:
hr = SHGetSpecialFolderPath(NULL, szPath, CSIDL_PROGRAMS, FALSE) ? S_OK : E_FAIL;
break;
case FC_DESKTOP:
hr = SHGetSpecialFolderPath(NULL, szPath, CSIDL_DESKTOPDIRECTORY, FALSE) ? S_OK : E_FAIL;
break;
default:
hr = E_UNEXPECTED;
break;
}
if (SUCCEEDED(hr))
{
BOOL fFound;
hr = FindAnyIcons(szPath, szExePath,
dwFolderClass & scfEntries[i].dwFoldersToRecurseCheck, &fFound);
if (SUCCEEDED(hr) && !fFound)
{
dwInitiallyClear |= dwFolderClass;
}
}
}
else
{
// This will only work for the currently logged in user...
DWORD dwFlags;
ASSERT(NULL != scfEntries[i].pszExplorerCLSID);
if (ERROR_SUCCESS == QueryExplorerCLSIDFlags(scfEntries[i].pszExplorerCLSID, &dwFlags))
{
if (dwFlags & SFGAO_NONENUMERATED)
{
dwInitiallyClear |= FC_EXPLORERCLSID;
}
}
}
}
}
// Only record a value for this app if there is a shortcut missing...
if (dwInitiallyClear)
{
SHSetValue(HKEY_CURRENT_USER, c_szInitiallyClearKey, scfEntries[i].pszExe, REG_DWORD, &dwInitiallyClear, sizeof(dwInitiallyClear));
}
}
}
}
extern "C" void OCInstallUpdate()
{
DWORD dwType;
DWORD dwValue;
DWORD cbValue = sizeof(dwValue);
DWORD dwResult = SHGetValueW(HKEY_LOCAL_MACHINE, c_szOCManagerSubComponents, c_szIEAccess, &dwType, &dwValue, &cbValue);
BOOL fIEAccessIsOn = ((ERROR_SUCCESS == dwResult) && (REG_DWORD == dwType) && (dwValue));
ShowHideIE(fIEAccessIsOn, FALSE, FALSE);
// If IE is on, then we go ahead and write the active setup values in case the IE version is not updated as well.
// This way we don't recreate icons for this user when there is no change to IE.
if (fIEAccessIsOn)
{
UpdateActiveSetupValues(c_szIEAccessKey, TRUE);
}
cbValue = sizeof(dwValue);
dwResult = SHGetValueW(HKEY_LOCAL_MACHINE, c_szOCManagerSubComponents, c_szOEAccess, &dwType, &dwValue, &cbValue);
BOOL fOEAccessIsOn = TRUE;
if ((ERROR_SUCCESS == dwResult) && (REG_DWORD == dwType) && !dwValue)
{
fOEAccessIsOn = FALSE;
}
ShowHideOE(fOEAccessIsOn, FALSE, FALSE);
// If OE is on, then we go ahead and write the active setup values in case the OE version is not updated as well.
// This way we don't recreate icons for this user when there is no change to OE.
if (fOEAccessIsOn)
{
UpdateActiveSetupValues(c_szOEAccessKey, TRUE);
}
GetCurrentIconInvisibility();
}
extern "C" void OCInstallCleanupInitiallyClear()
{
HKEY hKey;
// This code runs through and sees if we need to delete any shortcuts that were created by an SP update install
// that the user had previously deleted. The idea is that we shouldn't be recreating them.
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_CURRENT_USER, c_szInitiallyClearKey, 0, KEY_QUERY_VALUE, &hKey))
{
int nRegEnum = 0;
while (1)
{
WCHAR szExe[MAX_PATH];
DWORD cchExe = ARRAYSIZE(szExe);
DWORD dwType;
DWORD dwFlags;
DWORD cbFlags = sizeof(dwFlags);
LONG lResult = RegEnumValue(hKey, nRegEnum, szExe, &cchExe, NULL, &dwType, (LPBYTE)&dwFlags, &cbFlags);
if (ERROR_NO_MORE_ITEMS == lResult)
{
break;
}
else if ((ERROR_SUCCESS == lResult) && (REG_DWORD == dwType))
{
WCHAR szExePath[MAX_PATH];
if (ERROR_SUCCESS == GetExeAppPath(szExe, szExePath, ARRAYSIZE(szExePath)))
{
// Once we have the EXE, loop through each bit and see if it was initially clear (meaning
// the shorcut was not present before update ran).
for (DWORD dwFolderClass = FC_QUICKLAUNCH; dwFolderClass <= FC_MAX; dwFolderClass <<= 1)
{
if (dwFlags & dwFolderClass)
{
if (FC_EXPLORERCLSID != dwFolderClass)
{
HRESULT hr = E_FAIL;
WCHAR szPath[MAX_PATH];
switch (dwFolderClass)
{
case FC_QUICKLAUNCH:
if (SHGetSpecialFolderPath(NULL, szPath, CSIDL_APPDATA, FALSE))
{
WCHAR szQuickLaunchPath[MAX_PATH];
LoadString(GetResLibInstance(), IDS_OC_QLAUNCHAPPDATAPATH, szQuickLaunchPath, ARRAYSIZE(szQuickLaunchPath));
PathAppend(szPath, szQuickLaunchPath);
hr = S_OK;
}
break;
case FC_ALLPROGRAMS:
hr = SHGetSpecialFolderPath(NULL, szPath, CSIDL_PROGRAMS, FALSE) ? S_OK : E_FAIL;
break;
case FC_DESKTOP:
hr = SHGetSpecialFolderPath(NULL, szPath, CSIDL_DESKTOPDIRECTORY, FALSE) ? S_OK : E_FAIL;
break;
default:
hr = E_UNEXPECTED;
break;
}
if (SUCCEEDED(hr))
{
FindAndNukeIcons(szPath, szExePath);
}
}
else
{
// need to grovel for the CLSID in this case.
for (int i = 0; i < ARRAYSIZE(scfEntries); i++)
{
if (0 == StrCmpI(szExe, scfEntries[i].pszExe))
{
SetExplorerCLSIDFlags(scfEntries[i].pszExplorerCLSID, 0xffffffff, SFGAO_NONENUMERATED);
break;
}
}
}
}
}
}
}
nRegEnum++;
}
RegCloseKey(hKey);
}
// This makes sure that we only get called once.
SHDeleteKey(HKEY_CURRENT_USER, c_szInitiallyClearKey);
}
Reference in a new issue Copy permalink