487 lines
16 KiB
C++
487 lines
16 KiB
C++
//**********************************************************************
|
|
// File name: desktop.cpp
|
|
//
|
|
// Desktop manipulation functions
|
|
//
|
|
// Functions:
|
|
//
|
|
// Copyright (c) 1992 - 1998 Microsoft Corporation. All rights reserved.
|
|
//**********************************************************************
|
|
|
|
#include "pre.h"
|
|
#include "regstr.h"
|
|
#include "inetreg.h"
|
|
#include <shlobj.h>
|
|
#include <shfolder.h> // latest platform SDK has this
|
|
|
|
#define MAX_USER_NAME 255
|
|
#define REGSTR_PATH_SETUPKEY REGSTR_PATH_SETUP REGSTR_KEY_SETUP
|
|
#define REGSTR_PATH_IEONDESKTOP REGSTR_PATH_IEXPLORER TEXT("\\AdvancedOptions\\BROWSE\\IEONDESKTOP")
|
|
#define InternetConnectionWiz "Internet Connection Wizard"
|
|
#define NOICWICON "NoICWIcon"
|
|
|
|
static const TCHAR g_szRegPathWelcomeICW[] = TEXT("Welcome\\ICW");
|
|
static const TCHAR g_szAllUsers[] = TEXT("All Users");
|
|
static const TCHAR g_szConnectApp[] = TEXT("ICWCONN1.EXE");
|
|
static const TCHAR g_szConnectLink[] = TEXT("Connect to the Internet");
|
|
static const TCHAR g_szOEApp[] = TEXT("MSINM.EXE");
|
|
static const TCHAR g_szOELink[] = TEXT("Outlook Express");
|
|
static const TCHAR g_szRegPathICWSettings[] = TEXT("Software\\Microsoft\\Internet Connection Wizard");
|
|
static const TCHAR g_szRegValICWCompleted[] = TEXT("Completed");
|
|
static const TCHAR g_szRegValNoIcon[] = TEXT("NoIcon");
|
|
|
|
extern BOOL MyIsSmartStartEx(LPTSTR lpszConnectionName, DWORD dwBufLen);
|
|
extern BOOL IsNT();
|
|
extern BOOL IsNT5();
|
|
|
|
void QuickCompleteSignup()
|
|
{
|
|
// Set the welcome state
|
|
UpdateWelcomeRegSetting(TRUE);
|
|
|
|
// Restore the desktop
|
|
UndoDesktopChanges(g_hInstance);
|
|
|
|
// Mark the ICW as being complete
|
|
SetICWComplete();
|
|
}
|
|
|
|
|
|
void UpdateWelcomeRegSetting
|
|
(
|
|
BOOL bSetBit
|
|
)
|
|
{
|
|
HKEY hkey;
|
|
HKEY hkeyCurVer;
|
|
DWORD dwValue = bSetBit;
|
|
|
|
|
|
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
|
REGSTR_PATH_SETUP, // ...\Windows\CurrentVersion
|
|
0,
|
|
KEY_ALL_ACCESS,
|
|
&hkeyCurVer) == ERROR_SUCCESS)
|
|
{
|
|
|
|
DWORD dwDisposition;
|
|
if (ERROR_SUCCESS == RegCreateKeyEx(hkeyCurVer,
|
|
g_szRegPathWelcomeICW,
|
|
0,
|
|
NULL,
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS,
|
|
NULL,
|
|
&hkey,
|
|
&dwDisposition))
|
|
{
|
|
RegSetValueEx(hkey,
|
|
TEXT("@"),
|
|
0,
|
|
REG_DWORD,
|
|
(LPBYTE) &dwValue,
|
|
sizeof(DWORD));
|
|
|
|
RegCloseKey(hkey);
|
|
}
|
|
RegCloseKey(hkeyCurVer);
|
|
}
|
|
}
|
|
|
|
BOOL GetCompletedBit( )
|
|
{
|
|
HKEY hkey;
|
|
DWORD dwValue;
|
|
DWORD dwRet = ERROR_GEN_FAILURE;
|
|
DWORD dwType = REG_DWORD;
|
|
BOOL bBit = FALSE;
|
|
|
|
if (RegOpenKeyEx(HKEY_CURRENT_USER,
|
|
g_szRegPathICWSettings, // ...Software\\Microsoft\\Internet Connection Wizard
|
|
0,
|
|
KEY_ALL_ACCESS,
|
|
&hkey) == ERROR_SUCCESS)
|
|
{
|
|
DWORD dwDataSize = sizeof (dwValue);
|
|
if (ERROR_SUCCESS == RegQueryValueEx(hkey, g_szRegValICWCompleted, NULL, &dwType, (LPBYTE) &dwValue, &dwDataSize))
|
|
{
|
|
bBit = (1 == dwValue);
|
|
}
|
|
RegCloseKey(hkey);
|
|
}
|
|
return bBit;
|
|
}
|
|
|
|
// shlwapi!StrCatBuff is not be available in Win95/NT without IE5. Not sure
|
|
// if ICW has to run in these environment or not.
|
|
//
|
|
// return pszDestination. always NULL terminated.
|
|
void MyStrCatBuff(
|
|
LPTSTR pszDestination, // in/out. null terminated string
|
|
LPCTSTR pszSource, // in. null terminated string
|
|
int cchDestBuffSize // in. size of pzDestination in TCHAR
|
|
)
|
|
{
|
|
int nDestLen = lstrlen(pszDestination);
|
|
int nRemainLen = cchDestBuffSize - nDestLen;
|
|
|
|
if (nRemainLen > 1)
|
|
{
|
|
// has room in additional to '\0'
|
|
lstrcpyn(&(pszDestination[nDestLen]), pszSource, nRemainLen);
|
|
}
|
|
}
|
|
|
|
|
|
void GetDesktopDirectory(TCHAR* pszPath)
|
|
{
|
|
LPITEMIDLIST lpItemDList = NULL;
|
|
IMalloc* pMalloc = NULL;
|
|
HRESULT hr = E_FAIL;
|
|
|
|
if(IsNT5()) // Bug 81444 in IE DB
|
|
hr = SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOPDIRECTORY, &lpItemDList);
|
|
else if (IsNT())
|
|
hr = SHGetSpecialFolderLocation(NULL, CSIDL_COMMON_DESKTOPDIRECTORY, &lpItemDList);
|
|
else
|
|
{
|
|
|
|
|
|
TCHAR pszFolder[MAX_PATH];
|
|
*pszFolder = 0;
|
|
HRESULT hRet = S_FALSE;
|
|
|
|
HMODULE hmod = LoadLibrary(TEXT("shfolder.dll"));
|
|
PFNSHGETFOLDERPATH pfn = NULL;
|
|
if (hmod)
|
|
{
|
|
pfn = (PFNSHGETFOLDERPATH)GetProcAddress(hmod, "SHGetFolderPathA"); // or W if you are unicode
|
|
if (pfn)
|
|
{
|
|
hRet = pfn(NULL, CSIDL_COMMON_DESKTOPDIRECTORY, NULL, 0, pszFolder);
|
|
if (S_OK != hRet)
|
|
hRet = pfn(NULL, CSIDL_DESKTOPDIRECTORY, NULL, 0, pszFolder);
|
|
if (S_OK == hRet)
|
|
lstrcpy(pszPath ,pszFolder);
|
|
}
|
|
FreeLibrary(hmod);
|
|
}
|
|
|
|
if (S_OK != hRet)
|
|
{
|
|
|
|
FARPROC hShell32VersionProc = NULL;
|
|
HMODULE hShell32Mod = (HMODULE)LoadLibrary(TEXT("shell32.dll"));
|
|
|
|
if (hShell32Mod)
|
|
hShell32VersionProc = GetProcAddress(hShell32Mod, "DllGetVersion");
|
|
|
|
if(hShell32VersionProc)
|
|
{
|
|
TCHAR szDir [MAX_PATH] = TEXT("\0");
|
|
|
|
//ok, we're not NT, but we may be multiuser windows.
|
|
GetWindowsDirectory(szDir, MAX_PATH);
|
|
if (szDir)
|
|
{
|
|
MyStrCatBuff(szDir, TEXT("\\"), MAX_PATH);
|
|
MyStrCatBuff(szDir, g_szAllUsers, MAX_PATH);
|
|
|
|
TCHAR szTemp [MAX_MESSAGE_LEN] = TEXT("\0");
|
|
|
|
LoadString(g_hInstance, IDS_DESKTOP, szTemp, MAX_MESSAGE_LEN);
|
|
if (szTemp)
|
|
{
|
|
MyStrCatBuff(szDir, TEXT("\\"), MAX_PATH);
|
|
MyStrCatBuff(szDir, szTemp, MAX_PATH);
|
|
lstrcpy(pszPath ,szDir);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
hr = SHGetSpecialFolderLocation(NULL, CSIDL_DESKTOPDIRECTORY, &lpItemDList);
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
SHGetPathFromIDList(lpItemDList, pszPath);
|
|
|
|
if (SUCCEEDED(SHGetMalloc (&pMalloc)))
|
|
{
|
|
pMalloc->Free (lpItemDList);
|
|
pMalloc->Release ();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void RemoveDesktopShortCut
|
|
(
|
|
LPTSTR lpszShortcutName
|
|
)
|
|
{
|
|
TCHAR szShortcutPath[MAX_PATH] = TEXT("\0");
|
|
|
|
GetDesktopDirectory(szShortcutPath);
|
|
|
|
if(szShortcutPath[0] != TEXT('\0'))
|
|
{
|
|
lstrcat(szShortcutPath, TEXT("\\"));
|
|
lstrcat(szShortcutPath, lpszShortcutName);
|
|
lstrcat(szShortcutPath, TEXT(".LNK"));
|
|
DeleteFile(szShortcutPath);
|
|
}
|
|
}
|
|
|
|
// This function will add a desktop shortcut
|
|
void AddDesktopShortCut
|
|
(
|
|
LPTSTR lpszAppName,
|
|
LPTSTR lpszLinkName
|
|
)
|
|
{
|
|
TCHAR szConnectPath [MAX_PATH] = TEXT("\0");
|
|
TCHAR szAppPath [MAX_PATH] = TEXT("\0");
|
|
TCHAR szConnectLinkPath [MAX_PATH] = TEXT("\0"); // Path the where the Shortcut file will livE
|
|
TCHAR szdrive [_MAX_DRIVE] = TEXT("\0");
|
|
TCHAR szdir [_MAX_DIR] = TEXT("\0");
|
|
TCHAR szfname [_MAX_FNAME] = TEXT("\0");
|
|
TCHAR szext [_MAX_EXT] = TEXT("\0");
|
|
TCHAR szRegPath [MAX_PATH] = TEXT("\0");
|
|
HRESULT hres = E_FAIL;
|
|
IShellLink* psl = NULL;
|
|
HKEY hkey = NULL;
|
|
|
|
// first get the app path
|
|
lstrcpy(szRegPath, REGSTR_PATH_APPPATHS);
|
|
lstrcat(szRegPath, TEXT("\\"));
|
|
lstrcat(szRegPath, lpszAppName);
|
|
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
|
szRegPath,
|
|
0,
|
|
IsNT5()? KEY_QUERY_VALUE : KEY_ALL_ACCESS,
|
|
&hkey) == ERROR_SUCCESS)
|
|
{
|
|
DWORD dwTmp = sizeof(szConnectPath);
|
|
DWORD dwType = 0;
|
|
if(RegQueryValueEx(hkey,
|
|
NULL,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE) szConnectPath,
|
|
&dwTmp) != ERROR_SUCCESS)
|
|
{
|
|
RegCloseKey(hkey);
|
|
return;
|
|
}
|
|
RegQueryValueEx(hkey,
|
|
TEXT("Path"),
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE) szAppPath,
|
|
&dwTmp);
|
|
|
|
RegCloseKey(hkey);
|
|
}
|
|
else
|
|
{
|
|
return;
|
|
}
|
|
|
|
GetDesktopDirectory(szConnectLinkPath);
|
|
|
|
if(szConnectLinkPath[0] != TEXT('\0'))
|
|
{
|
|
// Append on the connect EXE name
|
|
lstrcat(szConnectLinkPath, TEXT("\\"));
|
|
lstrcat(szConnectLinkPath, lpszAppName);
|
|
|
|
//
|
|
int nLastChar = lstrlen(szAppPath)-1;
|
|
if ((nLastChar > 0) && (';' == szAppPath[nLastChar]))
|
|
szAppPath[nLastChar] = 0;
|
|
|
|
// Split the path, and the reassemble with the .LNK extension
|
|
_tsplitpath( szConnectLinkPath, szdrive, szdir, szfname, szext );
|
|
_tmakepath(szConnectLinkPath, szdrive, szdir, lpszLinkName, TEXT(".LNK"));
|
|
|
|
// Create an IShellLink object and get a pointer to the IShellLink
|
|
// interface (returned from CoCreateInstance).
|
|
hres = CoCreateInstance (CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER,
|
|
IID_IShellLink, (void **)&psl);
|
|
if (SUCCEEDED (hres))
|
|
{
|
|
IPersistFile *ppf;
|
|
|
|
// Query IShellLink for the IPersistFile interface for
|
|
// saving the shortcut in persistent storage.
|
|
hres = psl->QueryInterface (IID_IPersistFile, (void **)&ppf);
|
|
if (SUCCEEDED (hres))
|
|
{
|
|
WORD wsz [MAX_PATH]; // buffer for Unicode string
|
|
|
|
do
|
|
{
|
|
|
|
// Set the path to the shortcut target.
|
|
if (!SUCCEEDED(psl->SetPath (szConnectPath)))
|
|
break;
|
|
|
|
// Set the working dir to the shortcut target.
|
|
if (!SUCCEEDED(psl->SetWorkingDirectory (szAppPath)))
|
|
break;
|
|
|
|
// Set the args.
|
|
if (!SUCCEEDED(psl->SetArguments (SHORTCUTENTRY_CMD)))
|
|
break;
|
|
|
|
// Set the description of the shortcut.
|
|
TCHAR szDescription[MAX_MESSAGE_LEN];
|
|
if (!LoadString(g_hInstance, IDS_SHORTCUT_DESC, szDescription, MAX_MESSAGE_LEN))
|
|
lstrcpy(szDescription, lpszLinkName);
|
|
|
|
if (!SUCCEEDED(psl->SetDescription (szDescription)))
|
|
break;
|
|
|
|
// Ensure that the string consists of ANSI TCHARacters.
|
|
#ifdef UNICODE
|
|
lstrcpy(wsz, szConnectLinkPath);
|
|
#else
|
|
MultiByteToWideChar (CP_ACP, 0, szConnectLinkPath, -1, wsz, MAX_PATH);
|
|
#endif
|
|
|
|
// Save the shortcut via the IPersistFile::Save member function.
|
|
if (!SUCCEEDED(ppf->Save (wsz, TRUE)))
|
|
break;
|
|
|
|
// Release the pointer to IPersistFile.
|
|
ppf->Release ();
|
|
break;
|
|
|
|
} while (1);
|
|
}
|
|
// Release the pointer to IShellLink.
|
|
psl->Release ();
|
|
}
|
|
}
|
|
}
|
|
|
|
// This function will apply the appropriate desktop changes based on the following
|
|
// algorithm. This code below assumes that the machine is NOT internet capable.
|
|
// If the machine was upgraded from a previous OS, then
|
|
// Add the connect to the internet ICON
|
|
// ELSE (clean install or OEM pre install)
|
|
// Add the connect to the internet ICON
|
|
//
|
|
void DoDesktopChanges
|
|
(
|
|
HINSTANCE hAppInst
|
|
)
|
|
{
|
|
TCHAR szAppName[MAX_PATH];
|
|
TCHAR szLinkName[MAX_PATH];
|
|
HKEY hkey;
|
|
|
|
if (!LoadString(hAppInst, IDS_CONNECT_FNAME, szAppName, ARRAYSIZE(szAppName)))
|
|
lstrcpy(szAppName, g_szConnectApp);
|
|
|
|
if (!LoadString(hAppInst, IDS_CONNECT_DESKTOP_TITLE, szLinkName, ARRAYSIZE(szLinkName)))
|
|
lstrcpy(szLinkName, g_szConnectLink);
|
|
|
|
// We always add the connect shortcut
|
|
AddDesktopShortCut(szAppName, szLinkName);
|
|
|
|
// Set a registry value indicating that we messed with the desktop
|
|
DWORD dwDisposition;
|
|
if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER,
|
|
ICWSETTINGSPATH,
|
|
0,
|
|
NULL,
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_ALL_ACCESS,
|
|
NULL,
|
|
&hkey,
|
|
&dwDisposition))
|
|
{
|
|
DWORD dwDesktopChanged = 1;
|
|
RegSetValueEx(hkey,
|
|
ICWDESKTOPCHANGED,
|
|
0,
|
|
REG_DWORD,
|
|
(LPBYTE)&dwDesktopChanged,
|
|
sizeof(DWORD));
|
|
RegCloseKey(hkey);
|
|
}
|
|
}
|
|
|
|
// This undoes what DoDesktopChanges did
|
|
void UndoDesktopChanges
|
|
(
|
|
HINSTANCE hAppInst
|
|
)
|
|
{
|
|
|
|
TCHAR szConnectTotheInternetTitle[MAX_PATH];
|
|
HKEY hkey;
|
|
|
|
// Verify that we really changed the desktop
|
|
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_CURRENT_USER,
|
|
ICWSETTINGSPATH,
|
|
0,
|
|
KEY_ALL_ACCESS,
|
|
&hkey))
|
|
{
|
|
DWORD dwDesktopChanged = 0;
|
|
DWORD dwTmp = sizeof(DWORD);
|
|
DWORD dwType = 0;
|
|
|
|
RegQueryValueEx(hkey,
|
|
ICWDESKTOPCHANGED,
|
|
NULL,
|
|
&dwType,
|
|
(LPBYTE)&dwDesktopChanged,
|
|
&dwTmp);
|
|
RegCloseKey(hkey);
|
|
|
|
// Bail if the desktop was not changed by us
|
|
if(!dwDesktopChanged)
|
|
return;
|
|
}
|
|
|
|
// Always nuke the Connect to the internet icon
|
|
if (!LoadString(hAppInst,
|
|
IDS_CONNECT_DESKTOP_TITLE,
|
|
szConnectTotheInternetTitle,
|
|
ARRAYSIZE(szConnectTotheInternetTitle)))
|
|
{
|
|
lstrcpy(szConnectTotheInternetTitle, g_szConnectLink);
|
|
}
|
|
|
|
RemoveDesktopShortCut(szConnectTotheInternetTitle);
|
|
}
|
|
|
|
void UpdateDesktop
|
|
(
|
|
HINSTANCE hAppInst
|
|
)
|
|
{
|
|
if(MyIsSmartStartEx(NULL, 0))
|
|
{
|
|
// VYUNG NT5 bug See if IEAK wants to stop GETCONN icon creation
|
|
//if (!SHGetRestriction(NULL, TEXT("Internet Connection Wizard"), TEXT("NoICWIcon")))
|
|
// CHUNHOC NT5.1 bug Don't create icon at desktop in any case.
|
|
/*
|
|
if (!SHGetRestriction(NULL, L"Internet Connection Wizard", L"NoICWIcon"))
|
|
DoDesktopChanges(hAppInst);
|
|
*/
|
|
}
|
|
else
|
|
{
|
|
// We are internet ready, so set the appropriate Welcome show bit
|
|
// and replace the IE and OE links
|
|
UpdateWelcomeRegSetting(TRUE);
|
|
}
|
|
}
|
|
|