324 lines
10 KiB
C++
324 lines
10 KiB
C++
|
#include <string.h>
|
||
|
|
#include <ntverp.h>
|
||
|
|
#include "priv.h"
|
||
|
|
#include "ids.h"
|
||
|
|
|
||
|
|
#define SECURITY_WIN32
|
||
|
|
#include <schnlsp.h> //for UNISP_NAME_A
|
||
|
|
#include <sspi.h> //for SCHANNEL.dll api -- to obtain encryption key size
|
||
|
|
|
||
|
|
#include <mluisupp.h>
|
||
|
|
#include <wininet.h> // INTERNET_MAX_URL_LENGTH
|
||
|
|
|
||
|
|
#define MAX_REG_VALUE 256
|
||
|
|
|
||
|
|
// Some Static strings that we use to read from the registry
|
||
|
|
|
||
|
|
#ifdef UNIX
|
||
|
|
#define VERSION "IEUNIX Version"
|
||
|
|
#define UNIX_IE_PRODUCT_ID TEXT("86139-999-2001594-12504")
|
||
|
|
#ifndef ux10
|
||
|
|
#define UNIX_IE_PRODUCT_FILE TEXT("%MWDEV%/ie/setup/sunos5/.iepid")
|
||
|
|
#else
|
||
|
|
#define UNIX_IE_PRODUCT_FILE TEXT("%MWDEV%/ie/setup/ux10/.iepid")
|
||
|
|
#endif
|
||
|
|
#endif
|
||
|
|
|
||
|
|
typedef PSecurityFunctionTable (APIENTRY *INITSECURITYINTERFACE_FN_A) (VOID);
|
||
|
|
|
||
|
|
// Returns the maximum cipher strength
|
||
|
|
DWORD GetCipherStrength()
|
||
|
|
{
|
||
|
|
DWORD dwKeySize = 0;
|
||
|
|
HINSTANCE hSecurity;
|
||
|
|
INITSECURITYINTERFACE_FN_A pfnInitSecurityInterfaceA;
|
||
|
|
PSecurityFunctionTable pSecFuncTable;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Can't go directly to schannel on NT5. (Note that g_bRunningOnNT5OrHigher
|
||
|
|
// may not be initialized when fUseSChannel is initialized!)
|
||
|
|
//
|
||
|
|
static BOOL fUseSChannel = TRUE;
|
||
|
|
if (fUseSChannel && !g_bRunningOnNT5OrHigher)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// This is better for performance. Rather than call through
|
||
|
|
// SSPI, we go right to the DLL doing the work.
|
||
|
|
//
|
||
|
|
hSecurity = LoadLibrary("schannel");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// Use SSPI
|
||
|
|
//
|
||
|
|
if (g_bRunningOnNT)
|
||
|
|
{
|
||
|
|
hSecurity = LoadLibrary("security");
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hSecurity = LoadLibrary("secur32");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (hSecurity == NULL)
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Get the SSPI dispatch table
|
||
|
|
//
|
||
|
|
pfnInitSecurityInterfaceA =
|
||
|
|
(INITSECURITYINTERFACE_FN_A)GetProcAddress(hSecurity, "InitSecurityInterfaceA");
|
||
|
|
|
||
|
|
if (pfnInitSecurityInterfaceA == NULL)
|
||
|
|
{
|
||
|
|
goto exit;
|
||
|
|
}
|
||
|
|
|
||
|
|
pSecFuncTable = (PSecurityFunctionTable)((*pfnInitSecurityInterfaceA)());
|
||
|
|
if (pSecFuncTable == NULL)
|
||
|
|
{
|
||
|
|
goto exit;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pSecFuncTable->AcquireCredentialsHandleA && pSecFuncTable->QueryCredentialsAttributesA)
|
||
|
|
{
|
||
|
|
TimeStamp tsExpiry;
|
||
|
|
CredHandle chCred;
|
||
|
|
SecPkgCred_CipherStrengths cs;
|
||
|
|
|
||
|
|
if (SEC_E_OK == (*pSecFuncTable->AcquireCredentialsHandleA)(NULL,
|
||
|
|
UNISP_NAME_A, // Package
|
||
|
|
SECPKG_CRED_OUTBOUND,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
&chCred, // Handle
|
||
|
|
&tsExpiry ))
|
||
|
|
{
|
||
|
|
if (SEC_E_OK == (*pSecFuncTable->QueryCredentialsAttributesA)(&chCred, SECPKG_ATTR_CIPHER_STRENGTHS, &cs))
|
||
|
|
{
|
||
|
|
dwKeySize = cs.dwMaximumCipherStrength;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Free the handle if we can
|
||
|
|
if (pSecFuncTable->FreeCredentialsHandle)
|
||
|
|
{
|
||
|
|
(*pSecFuncTable->FreeCredentialsHandle)(&chCred);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
exit:
|
||
|
|
FreeLibrary(hSecurity);
|
||
|
|
|
||
|
|
if (dwKeySize == 0 && fUseSChannel)
|
||
|
|
{
|
||
|
|
// Failed, so retry using SSPI
|
||
|
|
fUseSChannel = FALSE;
|
||
|
|
dwKeySize = GetCipherStrength();
|
||
|
|
}
|
||
|
|
return dwKeySize;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
BOOL SHAboutInfoA(LPSTR lpszInfo, DWORD cchSize)
|
||
|
|
{
|
||
|
|
HKEY hkey;
|
||
|
|
char szVersion[64];
|
||
|
|
char szUserName[MAX_REG_VALUE];
|
||
|
|
char szCompanyName[MAX_REG_VALUE];
|
||
|
|
char szKeySize[11];
|
||
|
|
char szProductId[MAX_REG_VALUE];
|
||
|
|
char szUpdateUrl[INTERNET_MAX_URL_LENGTH];
|
||
|
|
char szIEAKStr[MAX_REG_VALUE];
|
||
|
|
LPSTR lpszAboutKey;
|
||
|
|
DWORD dwKeySize = 0;
|
||
|
|
DWORD cb;
|
||
|
|
DWORD dwType;
|
||
|
|
BOOL fIEOrShell = TRUE;
|
||
|
|
|
||
|
|
lpszInfo[0] = '\0';
|
||
|
|
szKeySize[0] = '\0';
|
||
|
|
|
||
|
|
#ifndef UNIX
|
||
|
|
// Are we in the explorer or IE process?
|
||
|
|
fIEOrShell = GetModuleHandle("EXPLORER.EXE") || GetModuleHandle("IEXPLORE.EXE");
|
||
|
|
|
||
|
|
if (g_bRunningOnNT)
|
||
|
|
lpszAboutKey = "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion";
|
||
|
|
else
|
||
|
|
#endif
|
||
|
|
lpszAboutKey = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion";
|
||
|
|
|
||
|
|
if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, lpszAboutKey, 0, KEY_READ, &hkey) == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
// get the encription key size
|
||
|
|
dwKeySize = GetCipherStrength();
|
||
|
|
wsprintf(szKeySize, "~%d", dwKeySize);
|
||
|
|
|
||
|
|
// get the custom IEAK update url
|
||
|
|
// (always get from Windows\CurrentVersion because IEAK policy file must be platform
|
||
|
|
// independent
|
||
|
|
|
||
|
|
cb = sizeof(szUpdateUrl);
|
||
|
|
if(SHGetValueA(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Windows\\CurrentVersion",
|
||
|
|
"IEAKUpdateUrl", &dwType, (LPBYTE)szUpdateUrl, &cb) != ERROR_SUCCESS)
|
||
|
|
szUpdateUrl[0] = '\0';
|
||
|
|
|
||
|
|
#ifndef UNIX
|
||
|
|
// Extra Whistler/IE6 code to get the VBL version info
|
||
|
|
{
|
||
|
|
char szVBLVersion[64];
|
||
|
|
char* pszBuildLabKey;
|
||
|
|
DWORD cbSize;
|
||
|
|
DWORD dwType;
|
||
|
|
|
||
|
|
if (g_bRunningOnNT)
|
||
|
|
{
|
||
|
|
pszBuildLabKey = "SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion";
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
pszBuildLabKey = "SOFTWARE\\Microsoft\\Internet Explorer";
|
||
|
|
}
|
||
|
|
|
||
|
|
cbSize = sizeof(szVBLVersion);
|
||
|
|
if ((SHGetValueA(HKEY_LOCAL_MACHINE,
|
||
|
|
pszBuildLabKey,
|
||
|
|
"BuildLab",
|
||
|
|
&dwType,
|
||
|
|
szVBLVersion,
|
||
|
|
&cbSize) == ERROR_SUCCESS) && (dwType == REG_SZ))
|
||
|
|
{
|
||
|
|
// Now szVBLVersion contains the VBL buildnumber in the format: "2204.reinerf.010700"
|
||
|
|
// Since we are only interested in the latter part, we remove the build # (first 4 digits)
|
||
|
|
memmove((void*)szVBLVersion, (void*)&szVBLVersion[4], (lstrlenA(szVBLVersion) + 1) * sizeof(char));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
szVBLVersion[0] = '\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
// get the Version number (version string is in the following format 5.00.xxxx.x)
|
||
|
|
szVersion[0] = '\0';
|
||
|
|
cb = sizeof(szVersion);
|
||
|
|
if (ERROR_SUCCESS == SHGetValueA(HKEY_LOCAL_MACHINE,
|
||
|
|
"SOFTWARE\\Microsoft\\Internet Explorer",
|
||
|
|
"Version", &dwType, (LPVOID)szVersion, &cb))
|
||
|
|
{
|
||
|
|
DWORD dwLen;
|
||
|
|
|
||
|
|
// if we have a szVBLVersion, tack it onto the end of the IE major/minor version string
|
||
|
|
if (*szVBLVersion)
|
||
|
|
{
|
||
|
|
lstrcatnA(szVersion, szVBLVersion, ARRAYSIZE(szVersion));
|
||
|
|
}
|
||
|
|
|
||
|
|
// added by pritobla on 9/1/98
|
||
|
|
// CustomizedVersion contains a 2-letter code that identifies what mode was used
|
||
|
|
// (CORP, ICP, ISP, etc.) in building this version IE using the IEAK.
|
||
|
|
dwLen = lstrlenA(szVersion);
|
||
|
|
cb = sizeof(szVersion) - (dwLen * sizeof(char));
|
||
|
|
SHGetValueA(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Internet Explorer",
|
||
|
|
"CustomizedVersion", &dwType, (LPVOID)&szVersion[dwLen], &cb);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
// Get the version details from ntverp.h / ieverp.h
|
||
|
|
sprintf(szVersion, "%s.%04d.%04d",VER_PRODUCTVERSION_STRING,
|
||
|
|
VER_PRODUCTBUILD,
|
||
|
|
VER_PRODUCTBUILD_QFE);
|
||
|
|
#endif // UNIX
|
||
|
|
|
||
|
|
if (!fIEOrShell)
|
||
|
|
{
|
||
|
|
// Not in the explorer or iexplore process so we are doing some side by side stuff so
|
||
|
|
// reflect this in the version string. Maybe we should get the version out of MSHTML
|
||
|
|
// but not sure since this still doesn't reflect IE4 or IE5 properly anyway.
|
||
|
|
MLLoadString(IDS_SIDEBYSIDE, szVersion, ARRAYSIZE(szVersion));
|
||
|
|
}
|
||
|
|
|
||
|
|
// get the custom IEAK branded help string
|
||
|
|
cb = sizeof(szIEAKStr);
|
||
|
|
if(RegQueryValueExA(hkey, "IEAKHelpString", 0, &dwType, (LPBYTE)szIEAKStr, &cb) != ERROR_SUCCESS)
|
||
|
|
szIEAKStr[0] = '\0';
|
||
|
|
|
||
|
|
// get the User name.
|
||
|
|
cb = sizeof(szUserName);
|
||
|
|
if(RegQueryValueExA(hkey, "RegisteredOwner", 0, &dwType, (LPBYTE)szUserName, &cb) != ERROR_SUCCESS)
|
||
|
|
szUserName[0] = '\0';
|
||
|
|
|
||
|
|
// get the Organization name.
|
||
|
|
cb = sizeof(szCompanyName);
|
||
|
|
if(RegQueryValueExA(hkey, "RegisteredOrganization", 0, &dwType, (LPBYTE)szCompanyName, &cb) != ERROR_SUCCESS)
|
||
|
|
szCompanyName[0] = '\0';
|
||
|
|
|
||
|
|
#ifndef UNIX
|
||
|
|
cb = sizeof(szProductId);
|
||
|
|
if (SHGetValueA(HKEY_LOCAL_MACHINE, "SOFTWARE\\Microsoft\\Internet Explorer\\Registration", "ProductId", &dwType, (LPVOID)szProductId, &cb) != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
szProductId[0] = '\0';
|
||
|
|
}
|
||
|
|
#else
|
||
|
|
HANDLE hPidFile;
|
||
|
|
char szPidFileName[MAX_PATH];
|
||
|
|
DWORD dwRead;
|
||
|
|
SHExpandEnvironmentStrings(UNIX_IE_PRODUCT_FILE, szPidFileName, MAX_PATH);
|
||
|
|
if ((hPidFile = CreateFileA(szPidFileName, GENERIC_READ, FILE_SHARE_READ, NULL,
|
||
|
|
OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL))
|
||
|
|
== INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
sprintf(szProductId, "%s", UNIX_IE_PRODUCT_ID);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
ReadFile(hPidFile, (LPVOID)szProductId, 23, &dwRead, NULL);
|
||
|
|
szProductId[dwRead] = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseHandle(hPidFile);
|
||
|
|
#endif
|
||
|
|
lstrcatn(lpszInfo, szVersion, cchSize);
|
||
|
|
lstrcatn(lpszInfo, "~", cchSize);
|
||
|
|
lstrcatn(lpszInfo, szUserName, cchSize);
|
||
|
|
lstrcatn(lpszInfo, "~", cchSize);
|
||
|
|
lstrcatn(lpszInfo, szCompanyName, cchSize);
|
||
|
|
lstrcatn(lpszInfo, szKeySize, cchSize);
|
||
|
|
lstrcatn(lpszInfo, "~", cchSize);
|
||
|
|
lstrcatn(lpszInfo, szProductId, cchSize);
|
||
|
|
lstrcatn(lpszInfo, "~", cchSize);
|
||
|
|
lstrcatn(lpszInfo, szUpdateUrl, cchSize);
|
||
|
|
lstrcatn(lpszInfo, "~", cchSize);
|
||
|
|
lstrcatn(lpszInfo, szIEAKStr, cchSize);
|
||
|
|
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL SHAboutInfoW(LPWSTR lpszInfo, DWORD cchSize)
|
||
|
|
{
|
||
|
|
LPSTR lpszTmp;
|
||
|
|
BOOL bRet = FALSE;
|
||
|
|
|
||
|
|
lpszInfo[0] = L'\0';
|
||
|
|
|
||
|
|
if(NULL != (lpszTmp = (LPSTR)LocalAlloc(LPTR, cchSize)))
|
||
|
|
if(SHAboutInfoA(lpszTmp, cchSize))
|
||
|
|
if(MultiByteToWideChar(CP_ACP, 0, lpszTmp, -1, lpszInfo, cchSize) != 0)
|
||
|
|
bRet = TRUE;
|
||
|
|
|
||
|
|
if(lpszTmp)
|
||
|
|
LocalFree(lpszTmp);
|
||
|
|
|
||
|
|
return bRet;
|
||
|
|
}
|