Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

245 lines
12 KiB

#include "stdafx.h"
#include <LMCONS.H> // 97/07/22 vtan: for UNLEN
#pragma hdrstop
BOOL OnUpgradeDisableActiveDesktopFeatures();
void CreateMyCurHomeComponent(BOOL fChecked)
{
// If there is a policy to prevent adding desktop components, then we should not attempt to add.
// Otherwise, a new user gets an error message when they login for the first time.
// Bug #21300 -- Fixed on 2/28/2001 -- Sankar
if (SHRestricted(REST_NOADDDESKCOMP))
return;
//Add the base components!
TCHAR szBuf[MAX_PATH];
ISubscriptionMgr * psm;
// Add a component that points to "about:home"
LoadString(HINST_THISDLL, IDS_MY_CURRENT_HOMEPAGE, szBuf, ARRAYSIZE(szBuf));
// 98/07/14 vtan #176721: Changed the following to pass default component
// positions to AddRemoveDesktopComponentNoUI so that the restored position may
// be set to the default component position.
AddRemoveDesktopComponentNoUI(TRUE, AD_APPLY_SAVE, MY_HOMEPAGE_SOURCE, szBuf, COMP_TYPE_WEBSITE, COMPONENT_DEFAULT_LEFT, COMPONENT_DEFAULT_TOP, COMPONENT_DEFAULT_WIDTH, COMPONENT_DEFAULT_HEIGHT, fChecked, IS_SPLIT);
if (SUCCEEDED(CoCreateInstance(CLSID_SubscriptionMgr, NULL,
CLSCTX_INPROC_SERVER, IID_ISubscriptionMgr, (void**)&psm)))
{
WCHAR wszName[MAX_PATH];
//We need to zero init this structure except the cbSize field.
SUBSCRIPTIONINFO siDefault = {sizeof(SUBSCRIPTIONINFO)};
SHTCharToUnicode(szBuf, wszName, ARRAYSIZE(wszName));
//This field is already initialized above.
//siDefault.cbSize = sizeof(siDefault);
psm->CreateSubscription(NULL, MY_HOMEPAGE_SOURCEW, wszName, CREATESUBS_NOUI, SUBSTYPE_DESKTOPURL, &siDefault);
psm->Release();
}
}
STDAPI CDeskHtmlProp_RegUnReg(BOOL bReg)
{
HKEY hKey;
DWORD userNameSize;
TCHAR szDeskcomp[MAX_PATH];
TCHAR userName[UNLEN];
// 98/07/22 vtan #202707: Problem: This code gets called for the first time when NT
// runs. This sets up the default user profile. Anything that is added to this profile
// is propagated to any current user when upgrading from NT 4.0 to NT 5.0. This
// causes the DeskHtmlVersion, DeskHtmlMinorVersion and Component\0 to be replaced
// with the default component. By replacing the version registry entries any old
// components will not get correctly upgraded by the component reading code. It also
// destroys the first component.
// Solution: Prevent a default component being added at setup time by checking the
// logged on user is "SYSTEM". If the user is anybody other than system then perform
// the update or addition of the default component.
userNameSize = ARRAYSIZE(userName);
if ((GetUserName(userName, &userNameSize) != 0) && (lstrcmp(userName, TEXT("SYSTEM")) == 0))
return(S_OK); // an ungracious exit right here and now!
if(bReg)
{
DWORD dwDisposition;
DWORD dwDeskHtmlVersion = 0;
DWORD dwDeskHtmlMinorVersion = 0;
DWORD dwType;
GetRegLocation(szDeskcomp, ARRAYSIZE(szDeskcomp), REG_DESKCOMP_COMPONENTS, NULL);
if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp,
0, NULL, 0, KEY_CREATE_SUB_KEY|KEY_QUERY_VALUE, NULL, &hKey,
&dwDisposition))
{
//Get the version stamp from the registry
if(dwDisposition == REG_OPENED_EXISTING_KEY)
{
DWORD dwDataLength = sizeof(DWORD);
SHQueryValueEx(hKey, REG_VAL_COMP_VERSION, NULL, &dwType, (LPBYTE)(&dwDeskHtmlVersion), &dwDataLength);
SHQueryValueEx(hKey, REG_VAL_COMP_MINOR_VERSION, NULL, &dwType, (LPBYTE)(&dwDeskHtmlMinorVersion), &dwDataLength);
}
//We need to close this key before we delete it
RegCloseKey(hKey);
// If this branch is already there, don't set default comp.
// Note: The differences between IE4_DESKHTML_VERSION and CUR_DESKHTML_VERSION are
// automatically taken care of when we read the components. So, we need to check only
// for very old versions here.
if (dwDeskHtmlVersion < IE4_DESKHTML_VERSION)
{
//Delete the existing components.
SHDeleteKey(HKEY_CURRENT_USER, szDeskcomp);
// Create the default active desktop configuration
if(RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 0, NULL, 0,
(KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
{
//We need an initial state
DWORD dw;
dw = CUR_DESKHTML_VERSION;
RegSetValueEx(hKey, REG_VAL_COMP_VERSION, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));
dw = CUR_DESKHTML_MINOR_VERSION;
RegSetValueEx(hKey, REG_VAL_COMP_MINOR_VERSION, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));
dw = COMPSETTING_ENABLE;
RegSetValueEx(hKey, REG_VAL_COMP_SETTINGS, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));
//Add the home page component
CreateMyCurHomeComponent(FALSE); //For millennium we want to disable the component.
RegCloseKey(hKey);
}
// Create the default active desktop safemode configuration
if(RegCreateKeyEx(HKEY_CURRENT_USER, REG_DESKCOMP_SAFEMODE,
0, NULL, 0, KEY_CREATE_SUB_KEY, NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
{
DWORD dwDisposition;
HKEY hKey2;
if(RegCreateKeyEx(hKey, REG_DESKCOMP_GENERAL_SUFFIX, 0, NULL, 0,
(KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey2, &dwDisposition) == ERROR_SUCCESS)
{
TCHAR szSafeMode[MAX_PATH];
GetWindowsDirectory(szSafeMode, ARRAYSIZE(szSafeMode));
HRESULT hr = StringCchCat(szSafeMode, ARRAYSIZE(szSafeMode), DESKTOPHTML_DEFAULT_SAFEMODE);
if (SUCCEEDED(hr))
{
// Show safemode.htx
SHRegSetPath(hKey2, NULL, REG_VAL_GENERAL_WALLPAPER, szSafeMode, 0);
// Don't bring up the gallery dialog box
dwDisposition = 0;
RegSetValueEx(hKey2, REG_VAL_GENERAL_VISITGALLERY, 0, REG_DWORD, (LPBYTE)&dwDisposition, sizeof(dwDisposition));
}
RegCloseKey(hKey2);
}
RegCloseKey(hKey);
}
// Create the default scheme key
if(RegCreateKeyEx(HKEY_CURRENT_USER, REG_DESKCOMP_SCHEME, 0, NULL, 0,
(KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
{
RegSetValueEx(hKey, REG_VAL_SCHEME_EDIT, 0, REG_SZ, (LPBYTE)TEXT(""), sizeof(TCHAR));
RegSetValueEx(hKey, REG_VAL_SCHEME_DISPLAY, 0, REG_SZ, (LPBYTE)TEXT(""), sizeof(TCHAR));
RegCloseKey(hKey);
}
// Set the components to be dirty sothat we re-generate desktop.htm
// the first boot after installing IE4.0.
SetDesktopFlags(COMPONENTS_DIRTY, COMPONENTS_DIRTY);
GetRegLocation(szDeskcomp, ARRAYSIZE(szDeskcomp), REG_DESKCOMP_GENERAL, NULL);
if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp,
0, NULL, 0, KEY_CREATE_SUB_KEY, NULL, &hKey, &dwDisposition))
{
RegCloseKey(hKey);
}
}
else
{
//See if we are upgrading from an older version like IE4.
if (dwDeskHtmlVersion < CUR_DESKHTML_VERSION)
{
// If so, save the DESKHTML_VERSION we are upgrading from.
// We use this later in SHGetSetSettings to decide if active desktop is ON/OFF.
// NOTE: The "UpgradedFrom" value is at "...\Desktop" and NOT at "..\Desktop\Components"
// This is because the "Components" key gets destroyed very often.
SHSetValue(HKEY_CURRENT_USER, REG_DESKCOMP, REG_VAL_COMP_UPGRADED_FROM,
REG_DWORD, (LPBYTE)&dwDeskHtmlVersion, sizeof(dwDeskHtmlVersion));
}
// The major version numbers match. So check if the minor version numbers
// match too!
if(dwDeskHtmlMinorVersion < CUR_DESKHTML_MINOR_VERSION)
{
//Update the new Minor version number!
if(RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 0, NULL, 0,
(KEY_CREATE_SUB_KEY | KEY_SET_VALUE), NULL, &hKey, &dwDisposition) == ERROR_SUCCESS)
{
DWORD dw;
dw = CUR_DESKHTML_MINOR_VERSION;
RegSetValueEx(hKey, REG_VAL_COMP_MINOR_VERSION, 0, REG_DWORD, (LPBYTE)&dw, sizeof(dw));
RegCloseKey(hKey);
}
// Add the new home page component
if((dwDeskHtmlVersion <= 0x10f) && (dwDeskHtmlMinorVersion <= 0x0001))
CreateMyCurHomeComponent(FALSE);
//
// If this is an upgrade from W2K or earlier, we need to check if the
// active desktop is OFF. If so, we need to turn off all the desktop components
// sothat the active desktop continues to be OFF.
if((dwDeskHtmlVersion <= NT5_DESKHTML_VERSION) &&
(dwDeskHtmlMinorVersion <= NT5_DESKHTML_MINOR_VERSION))
{
OnUpgradeDisableActiveDesktopFeatures();
}
// Minor version numbers do not match. So, set the dirty bit to force
// the regeneration of desktop.htt later when needed.
SetDesktopFlags(COMPONENTS_DIRTY, COMPONENTS_DIRTY);
// 98/07/16 vtan #176721/#202707: Added the following code to delete HKCU\Software\
// Microsoft\Internet Explorer\Desktop\General\ComponentsPositioned because in NT 4.0
// with IE 4.0 SP1 this registry entry is incorrectly incremented when the components
// are iterated rather than when they are positioned. This resets the counter
// in NT 5.0 where the bug has been fixed.
DWORD dw;
GetRegLocation(szDeskcomp, ARRAYSIZE(szDeskcomp), REG_DESKCOMP_GENERAL, NULL);
if (ERROR_SUCCESS == RegCreateKeyEx(HKEY_CURRENT_USER, szDeskcomp, 0,
NULL, 0, KEY_SET_VALUE, NULL, &hKey, &dw))
{
(LONG)RegDeleteValue(hKey, REG_VAL_GENERAL_CCOMPPOS);
(LONG)RegCloseKey(hKey);
}
}
}
}
}
else
{
SHDeleteKey(HKEY_LOCAL_MACHINE, c_szRegDeskHtmlProp);
SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_COMPONENTS_ROOT);
SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_GENERAL_ROOT);
SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_SAFEMODE);
SHDeleteKey(HKEY_CURRENT_USER, REG_DESKCOMP_SCHEME);
}
return S_OK;
}