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.
 
 
 
 
 
 

499 lines
16 KiB

//* Microsoft Windows **
//* Copyright(c) Microsoft Corp., 1994 **
//*********************************************************************
//
// UTIL.C - common utility functions
//
// HISTORY:
//
// 12/21/94 jeremys Created.
//
#include "inetcplp.h"
#include <advpub.h> // For REGINSTALL
#include <mluisupp.h>
#include "brutil.h"
#include <mlang.h>
#include <inetreg.h>
// function prototypes
VOID _cdecl FormatErrorMessage(TCHAR * pszMsg,DWORD cbMsg,TCHAR * pszFmt,va_list ArgList);
extern VOID GetRNAErrorText(UINT uErr,CHAR * pszErrText,DWORD cbErrText);
extern VOID GetMAPIErrorText(UINT uErr,CHAR * pszErrText,DWORD cbErrText);
/*******************************************************************
NAME: MsgBox
SYNOPSIS: Displays a message box with the specified string ID
********************************************************************/
int MsgBox(HWND hWnd,UINT nMsgID,UINT uIcon,UINT uButtons)
{
TCHAR szMsgBuf[MAX_RES_LEN+1];
TCHAR szSmallBuf[SMALL_BUF_LEN+1];
MLLoadShellLangString(IDS_APPNAME,szSmallBuf,sizeof(szSmallBuf));
MLLoadShellLangString(nMsgID,szMsgBuf,sizeof(szMsgBuf));
MessageBeep(uIcon);
return (MessageBox(hWnd,szMsgBuf,szSmallBuf,uIcon | uButtons));
}
/*******************************************************************
NAME: MsgBoxSz
SYNOPSIS: Displays a message box with the specified text
********************************************************************/
int MsgBoxSz(HWND hWnd,LPTSTR szText,UINT uIcon,UINT uButtons)
{
TCHAR szSmallBuf[SMALL_BUF_LEN+1];
MLLoadShellLangString(IDS_APPNAME,szSmallBuf,sizeof(szSmallBuf));
MessageBeep(uIcon);
return (MessageBox(hWnd,szText,szSmallBuf,uIcon | uButtons));
}
/*******************************************************************
NAME: MsgBoxParam
SYNOPSIS: Displays a message box with the specified string ID
NOTES: extra parameters are string pointers inserted into nMsgID.
********************************************************************/
int _cdecl MsgBoxParam(HWND hWnd,UINT nMsgID,UINT uIcon,UINT uButtons,...)
{
va_list nextArg;
BUFFER Msg(3*MAX_RES_LEN+1); // nice n' big for room for inserts
BUFFER MsgFmt(MAX_RES_LEN+1);
if (!Msg || !MsgFmt) {
return MsgBox(hWnd,IDS_ERROutOfMemory,MB_ICONSTOP,MB_OK);
}
MLLoadShellLangString(nMsgID,MsgFmt.QueryPtr(),MsgFmt.QuerySize());
va_start(nextArg, uButtons);
FormatErrorMessage(Msg.QueryPtr(),Msg.QuerySize(),
MsgFmt.QueryPtr(),nextArg);
va_end(nextArg);
return MsgBoxSz(hWnd,Msg.QueryPtr(),uIcon,uButtons);
}
BOOL EnableDlgItem(HWND hDlg,UINT uID,BOOL fEnable)
{
return EnableWindow(GetDlgItem(hDlg,uID),fEnable);
}
/*******************************************************************
NAME: LoadSz
SYNOPSIS: Loads specified string resource into buffer
EXIT: returns a pointer to the passed-in buffer
NOTES: If this function fails (most likely due to low
memory), the returned buffer will have a leading NULL
so it is generally safe to use this without checking for
failure.
********************************************************************/
LPTSTR LoadSz(UINT idString,LPTSTR lpszBuf,UINT cbBuf)
{
ASSERT(lpszBuf);
// Clear the buffer and load the string
if ( lpszBuf )
{
*lpszBuf = '\0';
MLLoadString( idString, lpszBuf, cbBuf );
}
return lpszBuf;
}
/*******************************************************************
NAME: FormatErrorMessage
SYNOPSIS: Builds an error message by calling FormatMessage
NOTES: Worker function for DisplayErrorMessage
********************************************************************/
VOID _cdecl FormatErrorMessage(TCHAR * pszMsg,DWORD cbMsg,TCHAR * pszFmt,va_list ArgList)
{
ASSERT(pszMsg);
ASSERT(pszFmt);
// build the message into the pszMsg buffer
DWORD dwCount = FormatMessage(FORMAT_MESSAGE_FROM_STRING,
pszFmt,0,0,pszMsg,cbMsg,&ArgList);
ASSERT(dwCount > 0);
}
/*----------------------------------------------------------
Purpose: Calls the ADVPACK entry-point which executes an inf
file section.
*/
HRESULT CallRegInstall(LPSTR szSection)
{
HRESULT hr = E_FAIL;
STRENTRY seReg[] = {
#ifdef WINNT
{ "CHANNELBARINIT", "no" }, // channel bar off by default on NT
#else
{ "CHANNELBARINIT", "yes" } // channel bar on by default on Win95/98
#endif
};
STRTABLE stReg = { ARRAYSIZE(seReg), seReg };
RegInstall(ghInstance, szSection, &stReg);
return hr;
}
//
// Code page to Script mapping table
// Can't load MLang during setup, so, we port this table from MLANG
//
typedef struct tagCPTOSCRIPT{
UINT uiCodePage;
SCRIPT_ID sid;
} CPTOSCRIPT;
const CPTOSCRIPT CpToScript [] =
{
{1252, sidAsciiLatin},
{1250, sidAsciiLatin},
{1254, sidAsciiLatin},
{1257, sidAsciiLatin},
{1258, sidAsciiLatin},
{1251, sidCyrillic },
{1253, sidGreek },
{1255, sidHebrew },
{1256, sidArabic },
{874, sidThai },
{932, sidKana },
{936, sidHan },
{949, sidHangul },
{950, sidBopomofo },
{50000, sidUserDefined},
};
/*******************************************************************
NAME: MigrateIEFontSetting
SYNOPSIS: Port IE4 font setting data to IE5 script settings
NOTES:
********************************************************************/
VOID MigrateIEFontSetting(void)
{
HKEY hKeyInternational;
HKEY hKeyScripts;
// Open IE international setting registry key
if (ERROR_SUCCESS ==
RegOpenKeyEx(HKEY_CURRENT_USER, REGSTR_PATH_INTERNATIONAL, NULL, KEY_READ, &hKeyInternational))
{
DWORD dwIndex = 0;
DWORD dwCreate = 0;
TCHAR szCodePage[1024] = {0};
// Open/Create scripts key
if (ERROR_SUCCESS == RegCreateKeyEx(hKeyInternational, REGSTR_VAL_FONT_SCRIPTS, 0, NULL,
REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS,
NULL, &hKeyScripts, &dwCreate))
{
// If scripts already exists, we're upgrading from IE5, no data porting needed
if (dwCreate == REG_CREATED_NEW_KEY)
{
DWORD dwSize = ARRAYSIZE(szCodePage);
TCHAR szFont[LF_FACESIZE];
while (ERROR_SUCCESS == RegEnumKeyEx(hKeyInternational, dwIndex, szCodePage,
&dwSize, NULL, NULL, NULL, NULL))
{
UINT uiCP = StrToInt(szCodePage);
for (int i=0; i<ARRAYSIZE(CpToScript); i++)
{
if (uiCP == CpToScript[i].uiCodePage)
{
HKEY hKeyCodePage;
if ( ERROR_SUCCESS == RegOpenKeyEx(hKeyInternational, szCodePage,
NULL, KEY_READ, &hKeyCodePage))
{
HKEY hKeyScript;
CHAR szScript[1024];
wsprintfA(szScript, "%d", CpToScript[i].sid);
// Port code page font data to script font data
// If CP == 1252, we always need to use it to update Latin CP font info
if ((ERROR_SUCCESS == RegCreateKeyExA(hKeyScripts, szScript, 0, NULL,
REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS,
NULL, &hKeyScript, &dwCreate)) &&
((dwCreate == REG_CREATED_NEW_KEY) || (uiCP == 1252)))
{
DWORD cb = sizeof(szFont);
if (ERROR_SUCCESS == RegQueryValueEx(hKeyCodePage,
REGSTR_VAL_FIXED_FONT, NULL, NULL,
(LPBYTE)szFont, &cb))
{
RegSetValueEx(hKeyScript, REGSTR_VAL_FIXED_FONT, 0,
REG_SZ, (LPBYTE)szFont, cb);
}
cb = sizeof(szFont);
if (ERROR_SUCCESS == RegQueryValueEx(hKeyCodePage,
REGSTR_VAL_PROP_FONT, NULL, NULL,
(LPBYTE)szFont, &cb))
{
RegSetValueEx(hKeyScript, REGSTR_VAL_PROP_FONT, 0,
REG_SZ, (LPBYTE)szFont, cb);
}
RegCloseKey(hKeyScript);
}
RegCloseKey(hKeyCodePage);
}
}
}
dwIndex++;
dwSize = ARRAYSIZE(szCodePage);
}
}
RegCloseKey(hKeyScripts);
}
RegCloseKey(hKeyInternational);
}
}
#define TSZUNATTENDEDINTRANET TSZWININETPATH TEXT("\\Unattend\\LocalIntranetSites")
#define TSZUNATTENDEDTRUSTED TSZWININETPATH TEXT("\\Unattend\\TrustedSites")
//Copy trusted/Intranet sites list from HKLM into ZoneMap
void CopySitesList(DWORD dwZone, TCHAR *szSubKey)
{
HRESULT hr;
HKEY hKey = NULL;
IInternetSecurityManager* pInternetSecurityManager = NULL;
TCHAR szValueName[32];
TCHAR szUrl[128];
DWORD cValueName, cbUrl;
if (FAILED(CoInternetCreateSecurityManager(NULL, &pInternetSecurityManager,0)))
{
goto Cleanup; // no zone manager?
}
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE, szSubKey, 0, KEY_QUERY_VALUE, &hKey))
{
for (DWORD i=0; ; i++)
{
cValueName = sizeof(szValueName)/sizeof(szValueName[0]);
cbUrl = sizeof(szUrl);
hr = RegEnumValue(hKey, i, szValueName, &cValueName, NULL, NULL, (LPBYTE)szUrl, &cbUrl);
if (hr == ERROR_NO_MORE_ITEMS)
{
break;
}
else if (hr == ERROR_SUCCESS)
{
pInternetSecurityManager->SetZoneMapping(dwZone, szUrl, SZM_CREATE);
}
}
RegCloseKey(hKey);
}
Cleanup:
if (pInternetSecurityManager)
pInternetSecurityManager->Release();
return;
}
#if 0
//Remove sites list from ZoneMap;
void ClearSitesList(DWORD dwZone)
{
IInternetSecurityManager* pInternetSecurityManager = NULL;
IEnumString *pEnumString = NULL;
LPOLESTR pOleStr;
if (FAILED(CoInternetCreateSecurityManager(NULL, &pInternetSecurityManager,0)))
{
goto Cleanup; // no zone manager?
}
if (S_OK != pInternetSecurityManager->GetZoneMappings(dwZone, &pEnumString, 0))
{
goto Cleanup;
}
while (S_OK == pEnumString->Next(1, &pOleStr, NULL))
{
pInternetSecurityManager->SetZoneMapping(dwZone, pOleStr, SZM_DELETE);
}
Cleanup:
if (pInternetSecurityManager)
pInternetSecurityManager->Release();
if (pEnumString)
pEnumString->Release();
return;
}
#endif
/*----------------------------------------------------------
Purpose: Install/uninstall user settings
*/
STDAPI DllInstall(BOOL bInstall, LPCWSTR pszCmdLine)
{
#ifdef DEBUG
if (IsFlagSet(g_dwBreakFlags, BF_ONAPIENTER))
{
TraceMsg(TF_ALWAYS, "Stopping in DllInstall");
DEBUG_BREAK;
}
#endif
if (bInstall)
{
if (pszCmdLine)
{
if(!StrCmpIW(pszCmdLine, L"HKCUHard"))
{
CallRegInstall("RegDll.HKCUHard");
CopySitesList(URLZONE_TRUSTED, TSZUNATTENDEDTRUSTED);
CopySitesList(URLZONE_INTRANET, TSZUNATTENDEDINTRANET);
return S_OK;
}
if(!StrCmpIW(pszCmdLine, L"HKCUSoft"))
{
CallRegInstall("RegDll.HKCUSoft");
return S_OK;
}
if(!StrCmpIW(pszCmdLine, L"HKLMHard"))
{
CallRegInstall("RegDll.HKLMHard");
return S_OK;
}
if(!StrCmpIW(pszCmdLine, L"HKLMSoft"))
{
CallRegInstall("RegDll.HKLMSoft");
return S_OK;
}
}
//
// We use to delete the whole key here - that doesn't work anymore
// because other people write to this key and we don't want
// to crush them. If you need to explicity delete a value
// add it to ao_2 value
// CallRegInstall("UnregDll");
CallRegInstall("RegDll");
// If we also have the integrated shell installed, throw in the options
// related to the Integrated Shell.
if (WhichPlatform() == PLATFORM_INTEGRATED)
CallRegInstall("RegDll.IntegratedShell");
// NT5's new shell has special reg key settings
if (GetUIVersion() >= 5)
CallRegInstall("RegDll.NT5");
// Run Whistler-specific settings.
if (IsOS(OS_WHISTLERORGREATER))
{
CallRegInstall("RegDll.Whistler");
}
// Port IE4 code page font setting
MigrateIEFontSetting();
}
else
{
CallRegInstall("UnregDll");
}
return S_OK;
}
#define REGSTR_CCS_CONTROL_WINDOWS REGSTR_PATH_CURRENT_CONTROL_SET TEXT("\\WINDOWS")
#define CSDVERSION TEXT("CSDVersion")
BOOL IsNTSPx(BOOL fEqualOrGreater, UINT uMajorVer, UINT uSPVer)
{
HKEY hKey;
DWORD dwSPVersion;
DWORD dwSize;
BOOL fResult = FALSE;
OSVERSIONINFO VerInfo;
VerInfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFO);
GetVersionEx(&VerInfo);
// make sure we're on NT4 or greater (or specifically NT4, if required)
if (VER_PLATFORM_WIN32_NT != VerInfo.dwPlatformId ||
(!fEqualOrGreater && VerInfo.dwMajorVersion != uMajorVer) ||
(fEqualOrGreater && VerInfo.dwMajorVersion < uMajorVer))
return FALSE;
if (fEqualOrGreater && VerInfo.dwMajorVersion > uMajorVer)
return TRUE;
// check for installed SP
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, REGSTR_CCS_CONTROL_WINDOWS, 0, KEY_QUERY_VALUE, &hKey) == ERROR_SUCCESS)
{
dwSize = sizeof(dwSPVersion);
if (RegQueryValueEx(hKey, CSDVERSION, NULL, NULL, (unsigned char*)&dwSPVersion, &dwSize) == ERROR_SUCCESS)
{
dwSPVersion = dwSPVersion >> 8;
}
RegCloseKey(hKey);
if (fEqualOrGreater)
fResult = (dwSPVersion >= uSPVer ? TRUE : FALSE);
else
fResult = (dwSPVersion == uSPVer ? TRUE : FALSE);
}
return fResult;
}