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.
 
 
 
 
 
 

1901 lines
57 KiB

/************************************************************************
Copyright (c) Microsoft Corporation 1997-1998
All rights reserved
***************************************************************************/
#include "pch.h"
#include "setup.h"
#include "check.h"
DEFINE_MODULE("Check");
#define SMALL_BUFFER_SIZE 256
#define FILTER_SIZE 128
#define BIG_BUFFER 4096
//
// Ldap_InitializeConnection( )
//
DWORD
Ldap_InitializeConnection(
PLDAP * LdapHandle )
{
TraceFunc( "Ldap_InitializeConnection( ... )\n" );
DWORD LdapError = LDAP_SUCCESS;
if ( !( *LdapHandle ) ) {
ULONG temp = DS_DIRECTORY_SERVICE_REQUIRED |
DS_IP_REQUIRED;
*LdapHandle = ldap_init( NULL, LDAP_PORT);
if ( !*LdapHandle )
{
LdapError = GetLastError( );
goto e0;
}
ldap_set_option( *LdapHandle, LDAP_OPT_GETDSNAME_FLAGS, &temp );
temp = LDAP_VERSION3;
ldap_set_option( *LdapHandle, LDAP_OPT_VERSION, &temp );
//
// our searches should be contained to a single naming context, though
// we should be able to go outside the root of the tree to find the
// naming context we're interested in.
//
temp = LDAP_CHASE_EXTERNAL_REFERRALS;
ldap_set_option( *LdapHandle, LDAP_OPT_REFERRALS, &temp );
LdapError = ldap_connect( *LdapHandle, 0 );
if ( LdapError != LDAP_SUCCESS )
goto e1;
LdapError = ldap_bind_s( *LdapHandle, NULL, NULL, LDAP_AUTH_SSPI );
if ( LdapError != LDAP_SUCCESS )
goto e1;
}
e0:
if ( LdapError != LDAP_SUCCESS ) {
DebugMsg( "!!Error initializing LDAP connection.\n" );
}
RETURN( LdapError );
e1:
ldap_unbind( *LdapHandle );
*LdapHandle = NULL;
goto e0;
}
//
// CheckDSForSCP( )
//
HRESULT
CheckDSForSCP( )
{
TraceFunc( "CheckDSForSCP( )\n" );
HRESULT hr = S_FALSE;
PLDAP LdapHandle = NULL;
DWORD LdapError = LDAP_SUCCESS;
DWORD count;
LPWSTR * ppszPath;
PLDAPMessage CurrentEntry;
PLDAPMessage LdapMessage;
ULONG ulSize = 0;
LPWSTR pMachineDN = NULL;
// Paramters we want from the Computer Object
LPWSTR ComputerAttrs[2];
ComputerAttrs[0] = TEXT("netbootSCPBL");
ComputerAttrs[1] = NULL;
//
// We already detected it
//
if ( g_Options.fBINLSCPFound ) {
Assert( LdapError == LDAP_SUCCESS );
hr = S_OK;
goto e0;
}
if ( !GetComputerObjectName( NameFullyQualifiedDN, NULL, &ulSize ) ) {
DWORD Error = GetLastError( );
DebugMsg( "0 GetComputerObjectName failed (%x)\n", Error);
MessageBoxFromStrings( NULL, IDS_DOMAINMEMBERSHIP_TITLE, IDS_DOMAINMEMBERSHIP_TEXT, MB_OK );
hr = THR(E_FAIL);
goto e0;
}
pMachineDN = (LPWSTR) TraceAlloc( LPTR, ulSize * sizeof(WCHAR) );
if ( !pMachineDN ) {
hr = THR(E_OUTOFMEMORY);
goto e0;
}
if ( !GetComputerObjectName( NameFullyQualifiedDN, pMachineDN, &ulSize ) ) {
DWORD Error = GetLastError( );
DebugMsg( "1 GetComputerObjectName failed (%x)\n", Error);
MessageBoxFromStrings( NULL, IDS_DOMAINMEMBERSHIP_TITLE, IDS_DOMAINMEMBERSHIP_TEXT, MB_OK );
hr = THR(E_FAIL);
goto e0;
}
DebugMsg( "Our MAO's DN is %s.\n", pMachineDN );
LdapError = Ldap_InitializeConnection( &LdapHandle );
if ( LdapError != ERROR_SUCCESS ) {
hr = THR( HRESULT_FROM_WIN32( LdapMapErrorToWin32( LdapError ) ) );
MessageBoxFromError( NULL, NULL, LdapMapErrorToWin32(LdapError) );
LdapError = LDAP_SUCCESS;
goto e0;
}
LdapError = ldap_search_ext_s( LdapHandle,
pMachineDN,
LDAP_SCOPE_BASE,
L"(objectClass=computer)",
ComputerAttrs,
FALSE,
NULL,
NULL,
NULL,
0,
&LdapMessage);
if (LdapError == LDAP_NO_SUCH_ATTRIBUTE ) {
DebugMsg( "SCP not found.\n" );
LdapError = ERROR_SUCCESS;
hr = S_FALSE;
goto e1;
}
count = ldap_count_entries( LdapHandle, LdapMessage );
if (count!= 1) {
//
// What should we do with two entries for the same server?
//
if ( count ) {
Assert(FALSE);
}
DebugMsg( "SCP not found.\n" );
goto e2;
}
CurrentEntry = ldap_first_entry( LdapHandle, LdapMessage );
ppszPath = ldap_get_values( LdapHandle, CurrentEntry, TEXT("netbootSCPBL") );
if ( !ppszPath ) {
DebugMsg( "SCP not found.\n" );
LdapError = LDAP_OTHER;
goto e2;
}
DebugMsg( "SCP found.\n" );
g_Options.fBINLSCPFound = TRUE;
hr = S_OK;
ldap_value_free(ppszPath);
e2:
ldap_msgfree( LdapMessage );
e1:
ldap_unbind( LdapHandle );
e0:
if ( LdapError != LDAP_SUCCESS ) {
hr = S_FALSE;
}
if ( pMachineDN ) {
TraceFree( pMachineDN );
}
HRETURN(hr);
}
//
// This will retrieve information about the remote install share path
//
HRESULT
GetRemoteInstallShareInfo()
{
WCHAR remInstPath[MAX_PATH];
LPSHARE_INFO_502 psi;
HRESULT hr = S_OK;
NET_API_STATUS netStat;
//
// See if we found the share or have setup the directory yet
//
if ( !g_Options.fIMirrorShareFound ||
!g_Options.fIMirrorDirectory) {
//
// Load the share name
//
if (LoadString( g_hinstance,
IDS_REMOTEBOOTSHARENAME,
remInstPath,
ARRAYSIZE(remInstPath )) == 0) {
HRETURN(HRESULT_FROM_WIN32(GetLastError()));
}
//
// See if the share exists, get path name
//
netStat = NetShareGetInfo( NULL, remInstPath, 502, (LPBYTE *)&psi );
if (netStat != NERR_Success) {
if (netStat == ERROR_MORE_DATA) {
NetApiBufferFree( psi );
}
HRETURN(HRESULT_FROM_WIN32(netStat));
}
//
// The share exists, flag it
//
g_Options.fIMirrorShareFound = TRUE;
DebugMsg( "Found the IMIRROR share, using it for the IntelliMirror Directory: %s\n", psi->shi502_path );
//
// Save the share path name
//
hr = StringCbCopy(g_Options.szIntelliMirrorPath,
sizeof(g_Options.szIntelliMirrorPath),
psi->shi502_path);
if (SUCCEEDED(hr)) {
//
// If it fit, flag we got it
//
g_Options.fIMirrorDirectory = TRUE;
}
NetApiBufferFree( psi );
}
HRETURN(hr);
}
//
// CheckDirectoryTree( )
//
HRESULT
CheckDirectoryTree( )
{
HRESULT hr = S_FALSE;
PWSTR szRemInstPath = NULL;
TraceFunc( "CheckDirectoryTree()\n" );
Assert( g_Options.hinf != INVALID_HANDLE_VALUE );
//
// Try to find the IMIRROR share
//
(void)GetRemoteInstallShareInfo();
//
// Try to use the TFTPD's reg key to find the IntelliMirror tree.
//
if ( !g_Options.fIMirrorDirectory
&& !g_Options.fTFTPDDirectoryFound ) {
//
// Try finding TFTPD's regkey to find the IntelliMirror Directory
//
HKEY hkey;
PWSTR szTftpPath = (PWSTR)TraceAlloc( LPTR, MAX_PATH*sizeof(WCHAR));
if (szTftpPath) {
if (LoadString(
g_hinstance,
IDS_TFTPD_SERVICE_PARAMETERS,
szTftpPath,
MAX_PATH) &&
( ERROR_SUCCESS == RegOpenKeyEx(
HKEY_LOCAL_MACHINE,
szTftpPath,
0, // options
KEY_QUERY_VALUE,
&hkey ))) {
ULONG l;
DWORD dwType;
LONG lErr;
l = sizeof(g_Options.szTFTPDDirectory);
lErr = RegQueryValueEx( hkey,
L"Directory",
0, // reserved
&dwType,
(LPBYTE) g_Options.szTFTPDDirectory,
&l );
Assert( wcslen(g_Options.szTFTPDDirectory) < ARRAYSIZE(g_Options.szTFTPDDirectory) );
if ( lErr == ERROR_SUCCESS ) {
DebugMsg( "Found TFTPD's Directory regkey: %s\n", g_Options.szTFTPDDirectory );
g_Options.fTFTPDDirectoryFound = TRUE;
}
RegCloseKey( hkey );
}
TraceFree(szTftpPath);
szTftpPath = NULL;
}
}
if ( !g_Options.fIMirrorDirectory
&& g_Options.fTFTPDDirectoryFound ) {
lstrcpyn( g_Options.szIntelliMirrorPath, g_Options.szTFTPDDirectory, ARRAYSIZE(g_Options.szIntelliMirrorPath) );
g_Options.fIMirrorDirectory = TRUE;
DebugMsg( "Used the TFTPD RegKey to find the IntelliMirror Directory.\n" );
}
//
// Did we find the IntelliMirror directory?
//
if ( !g_Options.fIMirrorDirectory ) {
//
// Nope... so figure out which drive we can put it on.
// Find a non-system, fixed disk, drive to place the
// IntelliMirror directory tree
//
WCHAR chSystemDriveLetter;
WCHAR chLargestDriveLetter = 0;
ULARGE_INTEGER uliBiggestFree = { 0 };
WCHAR szRootDrive[ 4 ] = TEXT("C:\\");
PWSTR DefaultPath;
DebugMsg( "Could not find the IntelliMirror directory.\n" );
DefaultPath = (PWSTR)TraceAlloc( LPTR, MAX_PATH*sizeof(WCHAR));
if (DefaultPath) {
if (LoadString( g_hinstance, IDS_DEFAULTPATH, DefaultPath, MAX_PATH ) &&
ExpandEnvironmentStrings( DefaultPath, g_Options.szIntelliMirrorPath, ARRAYSIZE(g_Options.szIntelliMirrorPath))) {
// The default string actually contains the system drive letter
chSystemDriveLetter = g_Options.szIntelliMirrorPath[0];
DebugMsg( "Searching for suitable drive:" );
for( ; szRootDrive[0] <= TEXT('Z'); szRootDrive[0]++ )
{
ULARGE_INTEGER uliFree;
ULARGE_INTEGER uliTotal;
ULARGE_INTEGER uliBytesFree;
UINT uDriveType;
DebugMsg( "%s ", szRootDrive );
uDriveType = GetDriveType( szRootDrive );
if ( DRIVE_FIXED != uDriveType
|| szRootDrive[0] == chSystemDriveLetter )
continue; // skip non-fixed and non-system drives
if ( !GetDiskFreeSpaceEx( szRootDrive, &uliFree, &uliTotal, &uliBytesFree ) )
continue; // error - skip it.
if ( uliBytesFree.QuadPart > uliBiggestFree.QuadPart )
{
chLargestDriveLetter = szRootDrive[0];
uliBiggestFree = uliBytesFree;
}
}
DebugMsg( "\n" );
if ( !chLargestDriveLetter )
{
g_Options.szIntelliMirrorPath[0] = chSystemDriveLetter;
DebugMsg( "BAD! Using system drive as default.\n" );
}
else
{
g_Options.szIntelliMirrorPath[0] = chLargestDriveLetter;
DebugMsg( "Suggesting %s for the IntelliMirror Drive/Directory.\n", g_Options.szIntelliMirrorPath );
}
} else {
TraceFree(DefaultPath);
DefaultPath = NULL;
goto e0; // skip the tree
}
TraceFree(DefaultPath);
DefaultPath = NULL;
goto e0; // skip the tree
} else {
goto e0; // skip the tree
}
}
//
// Check the directory tree.
// If any of these fail, just recreate the whole tree.
//
DebugMsg( "Checking Directory Tree:\n" );
//
// Create
// "D:\IntelliMirror"
//
DebugMsg( "%s\n", g_Options.szIntelliMirrorPath );
if ( 0xFFFFffff == GetFileAttributes( g_Options.szIntelliMirrorPath ) )
goto e0;
//
// Create
// "D:\IntelliMirror\Setup"
//
szRemInstPath = (PWSTR)TraceAlloc( LPTR, MAX_PATH*sizeof(WCHAR));
if (!szRemInstPath) {
goto e0;
}
if (_snwprintf(
szRemInstPath,
MAX_PATH,
L"%s\\Setup",
g_Options.szIntelliMirrorPath) <0) {
TraceFree(szRemInstPath);
szRemInstPath = NULL;
goto e0;
}
szRemInstPath[MAX_PATH-1] = L'\0';
DebugMsg( "%s\n", szRemInstPath );
if ( 0xFFFFffff == GetFileAttributes( szRemInstPath ) ) {
TraceFree(szRemInstPath);
szRemInstPath = NULL;
goto e0;
}
//
// Create the OS Chooser tree
// "D:\IntelliMirror\OSChooser"
//
lstrcpyn( g_Options.szOSChooserPath, g_Options.szIntelliMirrorPath, ARRAYSIZE(g_Options.szOSChooserPath) );
ConcatenatePaths( g_Options.szOSChooserPath, L"\\OSChooser", ARRAYSIZE(g_Options.szOSChooserPath) );
DebugMsg( "%s\n", g_Options.szOSChooserPath );
if ( 0xFFFFffff == GetFileAttributes( g_Options.szOSChooserPath ) ) {
TraceFree(szRemInstPath);
szRemInstPath = NULL;
goto e0;
}
g_Options.fOSChooserDirectory = TRUE;
if ( !g_Options.fOSChooserDirectoryTreeExists ) {
WCHAR szFile[ MAX_PATH ];
BOOL fFound;
INFCONTEXT context;
fFound = SetupFindFirstLine( g_Options.hinf, L"OSChooser", NULL, &context );
AssertMsg( fFound, "Could not find 'OSChooser' section in REMINST.INF.\n" );
while ( fFound
&& SetupGetStringField( &context, 1, szFile, ARRAYSIZE(szFile), NULL ) )
{
LPWSTR psz = StrChr( szFile, L'\\' );
if ( psz ) {
*psz = L'\0'; // terminate
if (_snwprintf(szRemInstPath,
MAX_PATH,
L"%s\\%s",
g_Options.szOSChooserPath,
szFile ) <0) {
TraceFree(szRemInstPath);
szRemInstPath = NULL;
goto e0;
}
szRemInstPath[MAX_PATH-1] = L'\0';
DebugMsg( "%s\n", szRemInstPath );
if ( 0xFFFFffff == GetFileAttributes( szRemInstPath ) ) {
TraceFree(szRemInstPath);
szRemInstPath = NULL;
goto e0;
}
}
fFound = SetupFindNextLine( &context, &context );
}
g_Options.fOSChooserDirectoryTreeExists = TRUE;
}
g_Options.fDirectoryTreeExists = TRUE;
TraceFree(szRemInstPath);
szRemInstPath = NULL;
e0:
Assert( szRemInstPath == NULL );
if (szRemInstPath) {
TraceFree(szRemInstPath);
szRemInstPath = NULL;
}
if ( g_Options.fIMirrorDirectory
&& g_Options.fIMirrorShareFound
&& g_Options.fDirectoryTreeExists
&& g_Options.fOSChooserDirectory
&& g_Options.fOSChooserDirectoryTreeExists ) {
DebugMsg( "All directories found.\n" );
hr = S_OK;
} else {
DebugMsg( "Directory tree check failed.\n" );
}
HRETURN(hr);
}
BOOL
CheckService(
SC_HANDLE schSystem,
LPWSTR ServiceName,
LPBOOL Started
)
{
SC_HANDLE sch;
SERVICE_STATUS status;
BOOL b;
sch = OpenService( schSystem,
ServiceName,
SERVICE_ALL_ACCESS );
if ( sch == NULL ) {
DebugMsg( "%ws is NOT installed.\n", ServiceName );
return FALSE;
}
DebugMsg( "%ws is installed.\n", ServiceName );
b = QueryServiceStatus( sch, &status);
*Started = (BOOL)(b && (status.dwCurrentState == SERVICE_RUNNING));
DebugMsg( "%ws is %wsrunning.\n", ServiceName, *Started ? L"" : L"not " );
CloseServiceHandle( sch );
return TRUE;
} // CheckService
//
// CheckBINL( )
//
HRESULT
CheckBINL(
SC_HANDLE schSystem )
{
HRESULT hr = S_FALSE;
BOOL started = TRUE;
PWSTR szSystem32Path = NULL;
PWSTR szFile = NULL;
PWSTR szFullPath = NULL;
TraceFunc( "CheckBINL( )\n" );
//
// Check to see if the service manager can find the service
//
if ( !g_Options.fBINLServiceInstalled ) {
if ( CheckService( schSystem, L"BINLSVC", &started ) ) {
g_Options.fBINLServiceInstalled = TRUE;
}
}
//
// Read the REMINST.INF for the files need for the service
//
if ( !g_Options.fBINLFilesFound ) {
UINT index;
INFCONTEXT context;
szSystem32Path = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFile = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFullPath = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!(szSystem32Path) ||
!(szFile) ||
!(szFullPath) ||
!GetSystemDirectory(szSystem32Path,MAX_PATH) ||
!SetupFindFirstLine( g_Options.hinf, L"Service.BINLSVC", L"FilesRequired", &context )) {
Assert(FALSE);
goto BINLCheckSCP;
}
index = 1;
while ( SetupGetStringField( &context, index, szFile, MAX_PATH+1, NULL ) )
{
if (_snwprintf(
szFullPath,
MAX_PATH+1,
L"%s\\%s",
szSystem32Path,
szFile )<0) {
DebugMsg( "%s is missing for Service BINLSVC.(_snwprintf failed)\n", szFile );
goto BINLCheckSCP;
}
szFullPath[MAX_PATH] = L'\0';
if ( 0xFFFFffff == GetFileAttributes( szFullPath ) ) {
DebugMsg( "%s is missing for Service BINLSVC.\n", szFullPath );
goto BINLCheckSCP;
}
DebugMsg( "%s found.\n", szFullPath );
index++;
}
g_Options.fBINLFilesFound = TRUE;
}
BINLCheckSCP:
if (szSystem32Path) {
TraceFree(szSystem32Path);
}
if (szFile) {
TraceFree(szFile);
}
if (szFullPath) {
TraceFree(szFullPath);
}
//
// Check to see if the SCP exists
//
hr = CheckDSForSCP( );
if ( hr == S_OK
&& g_Options.fBINLFilesFound
&& g_Options.fBINLServiceInstalled
&& started ) {
DebugMsg( "All BINL services checked out OK.\n");
hr = S_OK;
} else if ( hr == S_OK ) {
DebugMsg( "BINL check found something strange with the SCP. Ignoring.\n" );
hr = S_FALSE;
} else {
DebugMsg( "BINL check failed.\n" );
}
HRETURN(hr);
}
//
// CheckTFTPD( )
//
HRESULT
CheckTFTPD(
SC_HANDLE schSystem )
{
HRESULT hr = S_FALSE;
HKEY hkey;
BOOL started = TRUE;
PWSTR szSystem32Path = NULL;
PWSTR szFile = NULL;
PWSTR szFullPath = NULL;
TraceFunc( "CheckTFTPD( )\n" );
//
// Check to see if the service manager can find the service
//
if ( !g_Options.fTFTPDServiceInstalled ) {
if ( CheckService( schSystem, L"TFTPD", &started ) ) {
g_Options.fTFTPDServiceInstalled = TRUE;
}
}
//
// Read the REMINST.INF for the files need for the service
//
if ( !g_Options.fTFTPDFilesFound ) {
UINT index;
INFCONTEXT context;
szSystem32Path = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFile = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFullPath = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!(szSystem32Path) ||
!(szFile) ||
!(szFullPath) ||
!GetSystemDirectory(szSystem32Path,MAX_PATH) ||
!SetupFindFirstLine( g_Options.hinf, L"Service.TFTPD", L"FilesRequired", &context )) {
Assert(FALSE);
goto TFTPDCheckReg;
}
index = 1;
while ( SetupGetStringField( &context, index, szFile, ARRAYSIZE( szFile ), NULL ) )
{
if (_snwprintf(
szFullPath,
MAX_PATH+1,
L"%s\\%s",
szSystem32Path,
szFile )<0) {
DebugMsg( "%s is missing for Service TFTPD (_snwprintf failed).\n", szFile );
goto TFTPDCheckReg;
}
szFullPath[MAX_PATH] = L'\0';
if ( 0xFFFFffff == GetFileAttributes( szFullPath ) ) {
DebugMsg( "%s is missing for Service TFTPD.\n", szFullPath );
goto TFTPDCheckReg;
}
DebugMsg( "%s found.\n", szFullPath );
index++;
}
g_Options.fTFTPDFilesFound = TRUE;
}
TFTPDCheckReg:
if (szSystem32Path) {
TraceFree(szSystem32Path);
}
if (szFile) {
TraceFree(szFile);
}
if (szFullPath) {
TraceFree(szFullPath);
szFullPath = NULL;
}
//
// Check to see if the Directory reg key exists
//
if ( !g_Options.fTFTPDDirectoryFound ) {
szFullPath = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!(szFullPath) ||
!LoadString( g_hinstance, IDS_TFTPD_SERVICE_PARAMETERS, szFullPath, MAX_PATH+1)) {
Assert( FALSE );
goto e0;
}
if ( ERROR_SUCCESS == RegOpenKeyEx( HKEY_LOCAL_MACHINE,
szFullPath,
0, // options
KEY_QUERY_VALUE,
&hkey ) ) {
ULONG l;
DWORD dwType;
LONG lErr;
l = sizeof(g_Options.szTFTPDDirectory);
lErr = RegQueryValueEx( hkey,
L"Directory",
0, // reserved
&dwType,
(LPBYTE) g_Options.szTFTPDDirectory,
&l );
if ( lErr == ERROR_SUCCESS ) {
DebugMsg( "TFTPD's Directory RegKey found: %s\n", g_Options.szTFTPDDirectory );
Assert( wcslen(g_Options.szTFTPDDirectory) < ARRAYSIZE(g_Options.szTFTPDDirectory) );
if ( 0xFFFFffff == GetFileAttributes( g_Options.szTFTPDDirectory ) ) {
DebugMsg( "BUT, %s was not found.\n", g_Options.szTFTPDDirectory );
}
g_Options.fTFTPDDirectoryFound = TRUE;
}
RegCloseKey( hkey );
} else {
DebugMsg( "HKLM\\%s not found.\n", szFullPath );
}
}
e0:
if (szFullPath) {
TraceFree(szFullPath);
szFullPath = NULL;
}
if ( g_Options.fTFTPDDirectoryFound
&& g_Options.fTFTPDFilesFound
&& g_Options.fTFTPDServiceInstalled
&& started ) {
DebugMsg( "All TFTPD services checked out OK.\n" );
hr = S_OK;
} else {
DebugMsg( "TFTPD check failed.\n" );
}
HRETURN(hr);
}
//
// CheckSIS( )
//
HRESULT
CheckSIS(
SC_HANDLE schSystem )
{
HRESULT hr = S_FALSE;
BOOL started = TRUE;
PWSTR szSystem32Path = NULL;
PWSTR szFile = NULL;
PWSTR szFullPath = NULL;
TraceFunc( "CheckSIS( )\n" );
//
// Check to see if the service manager can find the service
//
if ( !g_Options.fSISServiceInstalled ) {
if ( CheckService( schSystem, L"SIS", &started ) ) {
g_Options.fSISServiceInstalled = TRUE;
}
}
//
// Read the REMINST.INF for the files need for the service
//
if ( !g_Options.fSISFilesFound ) {
UINT index;
INFCONTEXT context;
szSystem32Path = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFile = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFullPath = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!(szSystem32Path) ||
!(szFile) ||
!(szFullPath) ||
!GetSystemDirectory(szSystem32Path,MAX_PATH) ||
!SetupFindFirstLine( g_Options.hinf, L"Service.SIS", L"FilesRequired", &context )) {
Assert(FALSE);
goto SISCheckVolume;
}
index = 1;
while ( SetupGetStringField( &context, index, szFile, ARRAYSIZE( szFile ), NULL ) )
{
if (_snwprintf(
szFullPath,
MAX_PATH+1,
L"%s\\%s",
szSystem32Path,
szFile )<0)
{
DebugMsg( "%s is missing for Service SIS (_snwprintf failed).\n", szFile );
goto SISCheckVolume;
}
szFullPath[MAX_PATH] = L'\0';
if ( 0xFFFFffff == GetFileAttributes( szFullPath ) ) {
DebugMsg( "%s is missing for Service SIS.\n", szFullPath );
goto SISCheckVolume;
}
DebugMsg( "%s found.\n", szFullPath );
index++;
}
g_Options.fSISFilesFound = TRUE;
}
SISCheckVolume:
if (szSystem32Path) {
TraceFree(szSystem32Path);
szSystem32Path = NULL;
}
if (szFile) {
TraceFree(szFile);
szFile = NULL;
}
if (szFullPath) {
TraceFree(szFullPath);
szFullPath = NULL;
}
//
// If we know the IMirrorDirectory (and hence the volume), check
// to see if the Common Store Directory has been created.
//
if ( g_Options.fIMirrorDirectory ) {
szFullPath = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!(szFullPath) ||
FAILED(GetSisVolumePath(szFullPath,MAX_PATH+1)))
{
DebugMsg( "error building path to sis common store.\n" );
goto e0;
}
if ( !g_Options.fSISVolumeCreated ) {
//
// See if the directory exists
//
if ( 0xFFFFffff == GetFileAttributes( szFullPath ) ) {
DebugMsg( "%s is missing.\n", szFullPath );
goto e0;
}
DebugMsg( "%s found.\n", szFullPath );
g_Options.fSISVolumeCreated = TRUE;
}
//
// The directory exists, check its security
//
if ( !g_Options.fSISSecurityCorrect ) {
if (CheckSISCommonStoreSecurity( szFullPath )) {
g_Options.fSISSecurityCorrect = TRUE;
}
}
}
e0:
if (szFullPath) {
TraceFree(szFullPath);
szFullPath = NULL;
}
if ( g_Options.fSISVolumeCreated
&& g_Options.fSISFilesFound
&& g_Options.fSISServiceInstalled
&& g_Options.fSISSecurityCorrect
&& started ) {
DebugMsg( "All SIS services checked out OK.\n" );
hr = S_OK;
} else {
DebugMsg( "SIS check failed.\n" );
}
HRETURN(hr);
}
//
// CheckSISGroveler( )
//
HRESULT
CheckSISGroveler(
SC_HANDLE schSystem )
{
HRESULT hr = S_FALSE;
BOOL started = TRUE;
PWSTR szSystem32Path = NULL;
PWSTR szFile = NULL;
PWSTR szFullPath = NULL;
TraceFunc( "CheckSISGroveler( )\n" );
//
// Check to see if the service manager can find the service
//
if ( !g_Options.fSISGrovelerServiceInstalled ) {
if ( CheckService( schSystem, L"Groveler", &started ) ) {
g_Options.fSISGrovelerServiceInstalled = TRUE;
}
}
//
// Check to see if the service manager can find the service
//
if ( !g_Options.fSISServiceInstalled ) {
if ( CheckService( schSystem, L"SIS", &started ) ) {
g_Options.fSISServiceInstalled = TRUE;
}
}
//
// Read the REMINST.INF for the files need for the service
//
if ( !g_Options.fSISGrovelerFilesFound ) {
UINT index;
INFCONTEXT context;
szSystem32Path = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFile = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szFullPath = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!(szSystem32Path) ||
!(szFile) ||
!(szFullPath) ||
!GetSystemDirectory(szSystem32Path,MAX_PATH) ||
!SetupFindFirstLine( g_Options.hinf, L"Service.SISGroveler", L"FilesRequired", &context )) {
Assert(FALSE);
goto e0;
}
index = 1;
while ( SetupGetStringField( &context, index, szFile, ARRAYSIZE( szFile ), NULL ) )
{
if (_snwprintf(
szFullPath,
MAX_PATH+1,
L"%s\\%s",
szSystem32Path,
szFile )<0) {
DebugMsg( "%s is missing for Service SIS Groveler (_snwprintf failed).\n", szFile );
goto e0;
}
szFullPath[MAX_PATH] = L'\0';
if ( 0xFFFFffff == GetFileAttributes( szFullPath ) ) {
DebugMsg( "%s is missing for Service SIS Groveler.\n", szFullPath );
goto e0;
}
DebugMsg( "%s found.\n", szFullPath );
index++;
}
g_Options.fSISGrovelerFilesFound = TRUE;
}
if ( g_Options.fSISGrovelerFilesFound
&& g_Options.fSISGrovelerServiceInstalled
&& started ) {
DebugMsg( "All SIS Groveler services checked out OK.\n" );
hr = S_OK;
} else {
DebugMsg( "SIS Groveler check failed.\n" );
}
e0:
if (szSystem32Path) {
TraceFree(szSystem32Path);
}
if (szFile) {
TraceFree(szFile);
}
if (szFullPath) {
TraceFree(szFullPath);
szFullPath = NULL;
}
HRETURN(hr);
}
//
// CheckRegSrvDlls( )
//
HRESULT
CheckRegSrvDlls( )
{
HRESULT hr = S_FALSE;
WCHAR szSection[ SMALL_BUFFER_SIZE ];
DWORD dw;
BOOL fMissingDll = FALSE;
BOOL fRegistrationFailed = FALSE;
BOOL b;
TraceFunc( "CheckRegSrvDlls( )\n" );
if ( !g_Options.fRegSrvDllsFilesFound ||
! g_Options.fRegSrvDllsRegistered ) {
dw = LoadString( g_hinstance, IDS_INF_SECTION, szSection, ARRAYSIZE( szSection ));
Assert( dw );
b = SetupInstallFromInfSection(NULL, // hwndOwner
g_Options.hinf, // inf handle
szSection, // name of component
SPINST_REGSVR,
NULL, // relative key root
NULL, // source root path
0, // copy flags
NULL, // callback routine
NULL, // callback routine context
NULL, // device info set
NULL); // device info struct
if ( !b )
{
DWORD dwErr = GetLastError( );
DebugMsg( "SetupInstallFromInfSection failed - error 0x%08x\n", dwErr );
//
// For now do extra work and try both copying and re-registering the DLL,
// otw if we want to optimize we might be able to use the error code to
// determine which to do. However, this case is extremely rare, so the
// simple extra work is not a burden.
//
fMissingDll = TRUE;
fRegistrationFailed = TRUE;
}
g_Options.fRegSrvDllsFilesFound = !fMissingDll;
g_Options.fRegSrvDllsRegistered = !(fMissingDll || fRegistrationFailed);
}
if ( g_Options.fRegSrvDllsFilesFound
&& g_Options.fRegSrvDllsRegistered ) {
DebugMsg( "All DLLs found and registered.\n" );
hr = S_OK;
}
HRETURN(hr);
}
//
// CheckOSChooser( )
//
HRESULT
CheckOSChooser( )
{
HRESULT hr = S_FALSE;
TraceFunc( "CheckOSChooser( )\n" );
//
// Make sure the OS Chooser files for all platforms
// are installed
//
if ( !g_Options.fOSChooserInstalled
&& g_Options.fOSChooserDirectory ) {
WCHAR szFile[ MAX_PATH ];
BOOL fFound;
INFCONTEXT context;
fFound = SetupFindFirstLine( g_Options.hinf, L"OSChooser", NULL, &context );
Assert( fFound );
if ( !fFound ) {
DebugMsg( "Could not find 'OSChooser' section in REMINST.INF.\n" );
goto OSChooserCheckScreens;
}
while ( fFound
&& SetupGetStringField( &context, 1, szFile, ARRAYSIZE(szFile), NULL ) )
{
WCHAR szPath[ MAX_PATH ];
_snwprintf( szPath,
ARRAYSIZE(szPath),
L"%s\\%s",
g_Options.szOSChooserPath,
szFile );
TERMINATE_BUFFER(szPath);
if ( 0xFFFFffff == GetFileAttributes( szPath ) ) {
DebugMsg( "%s is missing for OS Chooser.\n", szPath );
goto OSChooserCheckScreens;
}
fFound = SetupFindNextLine( &context, &context );
}
DebugMsg( "All OS Chooser files found.\n" );
g_Options.fOSChooserInstalled = TRUE;
}
OSChooserCheckScreens:
//
// Check to see if all the screens are present
//
// Make the check only if we have a language set.
if ( g_Options.fLanguageSet ) {
// First check for the directory
if ( !g_Options.fOSChooserScreensDirectory ) {
WCHAR szOSChooserScreensPath[ MAX_PATH ];
_snwprintf( szOSChooserScreensPath,
ARRAYSIZE(szOSChooserScreensPath),
L"%s\\%s",
g_Options.szOSChooserPath,
g_Options.szLanguage );
TERMINATE_BUFFER(szOSChooserScreensPath);
if ( 0xFFFFffff == GetFileAttributes( szOSChooserScreensPath ) ) {
DebugMsg( "%s directory not found.\n", szOSChooserScreensPath );
goto OSChooserExitCheckScreens;
}
DebugMsg( "%s directory found.\n", szOSChooserScreensPath );
g_Options.fOSChooserScreensDirectory = TRUE;
}
// now check for files
if ( !g_Options.fOSChooserScreens
&& g_Options.fOSChooserScreensDirectory ) {
WCHAR szFile[ MAX_PATH ];
BOOL fFound;
INFCONTEXT context;
fFound = SetupFindFirstLine( g_Options.hinf, L"OSChooser Screens", NULL, &context );
Assert( fFound );
if ( !fFound )
goto OSChooserExitCheckScreens;
while ( fFound
&& SetupGetStringField( &context, 1, szFile, ARRAYSIZE(szFile), NULL ) )
{
WCHAR szPath[ MAX_PATH ];
_snwprintf( szPath,
ARRAYSIZE(szPath),
L"%s\\%s\\%s",
g_Options.szOSChooserPath,
g_Options.szLanguage,
szFile );
TERMINATE_BUFFER(szPath);
if ( 0xFFFFffff == GetFileAttributes( szPath ) ) {
DebugMsg( "%s is missing for OS Chooser.\n", szPath );
goto OSChooserExitCheckScreens;
}
fFound = SetupFindNextLine( &context, &context );
}
g_Options.fOSChooserScreens = TRUE;
DebugMsg( "All OS Chooser screens found for language %s.\n", g_Options.szLanguage );
}
}
OSChooserExitCheckScreens:
if ( g_Options.fOSChooserDirectory
&& g_Options.fOSChooserInstalled
&& ( !g_Options.fLanguageSet
|| ( g_Options.fOSChooserScreensDirectory
&& g_Options.fOSChooserScreens ) ) ) {
DebugMsg( "All OS Chooser services checked out OK.\n" );
hr = S_OK;
} else {
DebugMsg( "OS Chooser check failed.\n" );
}
HRETURN(hr);
}
//
// CheckServerVersion( )
//
HRESULT
CheckServerVersion( )
{
HRESULT hr = S_OK;
WCHAR szMajor[ SMALL_BUFFER_SIZE ];
WCHAR szMinor[ SMALL_BUFFER_SIZE ];
WCHAR szBuild[ SMALL_BUFFER_SIZE ];
WCHAR szServicePack[ SMALL_BUFFER_SIZE ];
PWSTR szSection = NULL;
TraceFunc( "CheckServerVersion( )\n" );
// DebugMsg( "Fudging Server Version check...\n" );
// g_Options.fServerCompatible = TRUE;
if ( !g_Options.fServerCompatible
&& g_Options.szWorkstationRemBootInfPath[0] ) {
OSVERSIONINFO osver;
hr = E_FAIL;
osver.dwOSVersionInfoSize = sizeof( OSVERSIONINFO );
if ( GetVersionEx( &osver ) ) {
WCHAR szServerMajor[ 10 ];
WCHAR szServerMinor[ 10 ];
WCHAR szServerBuild[ 10 ];
BOOL fFound;
HINF hinf = INVALID_HANDLE_VALUE;
UINT uLineNum;
INFCONTEXT context;
szSection = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
if (!szSection ||
!LoadString( g_hinstance, IDS_COMPATIBILITY, szSection, MAX_PATH+1)) {
Assert(FALSE);
goto Error;
}
_snwprintf( szServerMajor, ARRAYSIZE(szServerMajor), L"%u", osver.dwMajorVersion );
TERMINATE_BUFFER(szServerMajor);
_snwprintf( szServerMinor, ARRAYSIZE(szServerMinor), L"%u", osver.dwMinorVersion );
TERMINATE_BUFFER(szServerMinor);
_snwprintf( szServerBuild, ARRAYSIZE(szServerBuild), L"%u", osver.dwBuildNumber );
TERMINATE_BUFFER(szServerBuild);
DebugMsg( "Server Version: %u.%u Build: %u Service Pack: %s\n",
osver.dwMajorVersion,
osver.dwMinorVersion,
osver.dwMinorVersion,
osver.szCSDVersion );
hinf = SetupOpenInfFile( g_Options.szWorkstationRemBootInfPath, NULL, INF_STYLE_WIN4, &uLineNum);
if ( hinf == INVALID_HANDLE_VALUE ) {
ErrorBox( NULL, g_Options.szWorkstationRemBootInfPath );
goto Error;
}
fFound = SetupFindFirstLine( hinf, szSection, NULL, &context );
if ( !fFound ) {
ErrorBox( NULL, g_Options.szWorkstationRemBootInfPath );
goto Error;
}
while ( fFound )
{
fFound = SetupGetStringField( &context, 1, szMajor, ARRAYSIZE(szMajor), NULL );
Assert( fFound );
if (!fFound)
break;
fFound = SetupGetStringField( &context, 2, szMinor, ARRAYSIZE(szMinor), NULL );
Assert( fFound );
if (!fFound)
break;
fFound = SetupGetStringField( &context, 3, szBuild, ARRAYSIZE(szBuild), NULL );
Assert( fFound );
if (!fFound)
break;
fFound = SetupGetStringField( &context, 4, szServicePack, ARRAYSIZE(szServicePack), NULL );
Assert( fFound );
if (!fFound)
break;
// Does it correspond to the server version?
if ( StrCmpI( szServerMajor, szMajor ) >= 0 // must match or greater
&& StrCmpI( szServerMinor, szMinor ) >= 0 // must match or greater
&& StrCmpI( szServerBuild, szBuild ) >= 0 // must match or server build is greater
&& ( szServicePack[0] == L'\0' // ignore service pack if not specified
|| StrCmpI( osver.szCSDVersion, szServicePack ) == 0 ) ) {
// found match
DebugMsg( "Server is compatiable with Workstation.\n");
hr = S_OK;
break;
}
fFound = SetupFindNextLine( &context, &context );
}
if ( FAILED(hr) ) {
PWSTR szTitle = NULL;
PWSTR szCaption = NULL;
PWSTR szBuf = NULL;
INFCONTEXT infcontext;
DebugMsg( "Server doesn't look like it is compatiable with the Workstation Build.\n" );
fFound = SetupFindFirstLine( hinf, szSection, NULL, &infcontext );
Assert( fFound );
fFound = SetupGetStringField( &infcontext, 1, szMajor, ARRAYSIZE(szMajor), NULL );
Assert( fFound );
fFound = SetupGetStringField( &infcontext, 2, szMinor, ARRAYSIZE(szMinor), NULL );
Assert( fFound );
fFound = SetupGetStringField( &infcontext, 3, szBuild, ARRAYSIZE(szBuild), NULL );
Assert( fFound );
fFound = SetupGetStringField( &infcontext, 4, szServicePack, ARRAYSIZE(szServicePack), NULL );
Assert( fFound );
szTitle = (PWSTR)TraceAlloc( LPTR, (MAX_PATH+1)*sizeof(WCHAR));
szCaption = (PWSTR)TraceAlloc( LPTR, (4096)*sizeof(WCHAR));
szBuf = (PWSTR)TraceAlloc( LPTR, (4096)*sizeof(WCHAR));
if ((szTitle) &&
(szCaption) &&
(szBuf) &&
LoadString( g_hinstance, IDS_INCOMPATIBLE_SERVER_VERSION_TITLE, szTitle, MAX_PATH+1) &&
LoadString( g_hinstance, IDS_INCOMPATIBLE_SERVER_VERSION_CAPTION, szBuf, 4096) &&
_snwprintf(
szCaption,
4096,
szBuf,
szMajor,
szMinor,
szBuild,
szServicePack) >=0) {
szCaption[4096-1] = L'\0';
MessageBox( NULL, szCaption, szTitle, MB_OK );
} else {
Assert(FALSE);
}
if (szTitle) {
TraceFree(szTitle);
}
if (szBuf) {
TraceFree(szBuf);
}
if (szCaption) {
TraceFree(szCaption);
}
Assert( hr == E_FAIL );
}
Error:
if ( hinf != INVALID_HANDLE_VALUE ) {
SetupCloseInfFile( hinf );
}
}
g_Options.fServerCompatible = ( hr == S_OK );
}
if (szSection) {
TraceFree(szSection);
}
if (g_Options.fServerCompatible ) {
DebugMsg( "Server compatibility checked out OK.\n" );
hr = S_OK;
} else {
DebugMsg( "Server compatibility failed.\n" );
}
HRETURN(hr);
}
//
// CheckFirstInstall( )
//
// This really checks for the existence of the REMINST directory
// under the system32 directory which was created during OCM setup.
//
HRESULT
CheckFirstInstall( )
{
HRESULT hr = S_OK;
TraceFunc( "CheckFirstInstall( )\n" );
if ( !g_Options.fRemBootDirectory ) {
DWORD dw;
WCHAR szFilePath[ MAX_PATH ];
dw = ExpandEnvironmentStrings( TEXT("%windir%"), g_Options.szRemBootDirectory, ARRAYSIZE(g_Options.szRemBootDirectory) );
Assert( dw );
lstrcpyn( szFilePath, g_Options.szRemBootDirectory, ARRAYSIZE(szFilePath) );
wcsncat( szFilePath, L"\\System32\\reminst.inf", ARRAYSIZE(szFilePath) - lstrlen(szFilePath) );
TERMINATE_BUFFER(szFilePath);
wcsncat( g_Options.szRemBootDirectory, L"\\System32\\REMINST", ARRAYSIZE(g_Options.szRemBootDirectory) - lstrlen(g_Options.szRemBootDirectory) );
TERMINATE_BUFFER(g_Options.szRemBootDirectory);
if ( 0xFFFFffff != GetFileAttributes( g_Options.szRemBootDirectory ) )
{
DebugMsg( "Found the %s directory.\n", g_Options.szRemBootDirectory );
HINF hinf = INVALID_HANDLE_VALUE;
INFCONTEXT context;
// now check to see if the files are still in there.
hinf = SetupOpenInfFile( szFilePath, NULL, INF_STYLE_WIN4, NULL );
if ( hinf != INVALID_HANDLE_VALUE )
{
WCHAR szFile[ MAX_PATH ];
BOOL fFound;
fFound = SetupFindFirstLine( hinf, L"OSChooser", NULL, &context );
Assert( fFound );
if ( !fFound ) {
DebugMsg( "Could not find 'OSChooser' section in REMINST.INF.\n" );
hr = S_FALSE;
goto bail;
}
lstrcpyn( szFilePath, g_Options.szRemBootDirectory, ARRAYSIZE(szFilePath) );
wcsncat( szFilePath, L"\\", ARRAYSIZE(szFilePath) - lstrlen(szFilePath) );
TERMINATE_BUFFER(szFilePath);
dw = (DWORD)wcslen( szFilePath );
while ( fFound )
{
fFound = SetupGetStringField( &context, 2, szFile, ARRAYSIZE(szFile), NULL );
if ( !fFound )
{
fFound = SetupGetStringField( &context, 2, szFile, ARRAYSIZE(szFile), NULL );
if ( !fFound )
goto skipit;
}
lstrcpyn( &szFilePath[ dw ], szFile, ARRAYSIZE(szFilePath) - dw );
if ( 0xFFFFffff == GetFileAttributes( szFilePath ) ) {
DebugMsg( "%s is missing.\n", szFilePath );
goto bail;
}
skipit:
fFound = SetupFindNextLine( &context, &context );
}
DebugMsg( "All REMINST files found.\n" );
g_Options.fRemBootDirectory = TRUE;
}
else
{
DebugMsg( "File not found. Error 0x%08x\n", GetLastError( ) );
}
bail:
if ( hinf != INVALID_HANDLE_VALUE )
{
SetupCloseInfFile( hinf );
}
}
}
if ( g_Options.fRemBootDirectory )
{
DebugMsg( "REMINST directory checked out OK.\n" );
}
else
{
DebugMsg( "REMINST directory is missing or missing files. This may require the server CD.\n ");
}
HRETURN(hr);
}
//
// CheckRegistry( )
//
HRESULT
CheckRegistry( )
{
HRESULT hr = S_FALSE;
WCHAR szSection[ SMALL_BUFFER_SIZE ];
INFCONTEXT context;
BOOL fFound;
DWORD dw;
INT index;
TraceFunc( "CheckRegistry( )\n" );
if ( !g_Options.fRegistryIntact ) {
dw = LoadString( g_hinstance, IDS_INF_SECTION, szSection, ARRAYSIZE( szSection ) );
Assert( dw );
fFound = SetupFindFirstLine( g_Options.hinf, szSection, L"AddReg", &context );
Assert( fFound );
if ( !fFound )
goto Error;
// Assume SUCCESS and use it to exit the loop early
g_Options.fRegistryIntact = TRUE;
DebugMsg( "Checking Registry:\n" );
index = 1;
while ( g_Options.fRegistryIntact
&& SetupGetStringField( &context, index, szSection, ARRAYSIZE(szSection), NULL ) )
{
INFCONTEXT subcontext;
fFound = SetupFindFirstLine( g_Options.hinf, szSection, NULL, &subcontext );
Assert( fFound );
while ( fFound && g_Options.fRegistryIntact )
{
WCHAR szRootKey[ 10 ];
WCHAR szBaseKey[ MAX_PATH ];
WCHAR szValueName[ MAX_PATH ];
HKEY hkeyRoot = NULL;
HKEY hkeyBase;
LONG lResult;
SetupGetStringField( &subcontext, 1, szRootKey, ARRAYSIZE(szRootKey), NULL );
SetupGetStringField( &subcontext, 2, szBaseKey, ARRAYSIZE(szBaseKey), NULL );
SetupGetStringField( &subcontext, 3, szValueName, ARRAYSIZE(szValueName), NULL );
DebugMsg( "%s, %s, %s - ", szRootKey, szBaseKey, szValueName );
if ( _wcsicmp( szRootKey, L"HKLM" ) == 0 ) {
hkeyRoot = HKEY_LOCAL_MACHINE;
} else if ( _wcsicmp( szRootKey, L"HKCU" ) == 0 ) {
hkeyRoot = HKEY_CURRENT_USER;
} else if ( _wcsicmp( szRootKey, L"HKCC" ) == 0 ) {
hkeyRoot = HKEY_CURRENT_CONFIG;
} else if ( _wcsicmp( szRootKey, L"HKCR" ) == 0 ) {
hkeyRoot = HKEY_CLASSES_ROOT;
}
//
// If this Assert() fires, it is because the INF is malformed
//
Assert( hkeyRoot );
if ( hkeyRoot == NULL )
continue;
lResult = RegOpenKeyEx( hkeyRoot, szBaseKey, 0, KEY_QUERY_VALUE, &hkeyBase );
if ( lResult == ERROR_SUCCESS )
{
lResult = RegQueryValueEx( hkeyBase, szValueName, 0, NULL, NULL, NULL );
if ( lResult != ERROR_SUCCESS ) {
DebugMsg("NOT " );
g_Options.fRegistryIntact = FALSE;
hr = S_FALSE;
}
RegCloseKey( hkeyBase );
} else {
DebugMsg("NOT " );
g_Options.fRegistryIntact = FALSE;
hr = S_FALSE;
}
DebugMsg("found.\n" );
fFound = SetupFindNextLine( &subcontext, &subcontext );
}
index++;
}
}
if ( g_Options.fRegistryIntact ) {
DebugMsg( "Registry checked out OK.\n" );
hr = S_OK;
} else {
DebugMsg( "Registry check failed.\n" );
}
Error:
HRETURN(hr);
}
//
// Paranoid( )
//
// Catch all for double checking that common things across services
// match up.
//
HRESULT
Paranoid( )
{
HRESULT hr = S_OK;
TraceFunc( "Paranoid( )\n" );
// Make sure that the IntelliMirror directory are the same
// for TFTPD and for the IMIRROR share (binlsvc uses this).
if ( g_Options.fIMirrorDirectory
&& g_Options.fTFTPDDirectoryFound
&& StrCmpI( g_Options.szIntelliMirrorPath, g_Options.szTFTPDDirectory ) != 0 ) {
//
// If they are not the same, force the registry to modify the TFTPD key.
//
DebugMsg( "TFTPD doesn't agree with IMIRROR share. Forcing TFTPD registry update.\n" );
g_Options.fTFTPDDirectoryFound = FALSE;
hr = S_FALSE;
}
if ( hr != S_OK ) {
DebugMsg( "Paranoid found a problem.\n" );
} else {
DebugMsg( "Paranoid is happy. 8~)\n" );
}
HRETURN( hr );
}
//
// CheckInstallation( )
//
HRESULT
CheckInstallation( )
{
HRESULT hr = S_OK;
BOOL fSomethingbroke = FALSE;
BOOL fSomethingFailed = FALSE;
DWORD dw;
DWORD dwLen;
UINT uLineNum;
SC_HANDLE schSystem;
WCHAR szServerRemBootInfPath[ MAX_PATH ];
CWaitCursor Wait;
TraceFunc( "CheckInstallation( )\n");
szServerRemBootInfPath[0] = L'\0';
dw = ExpandEnvironmentStrings( TEXT("%SystemRoot%"), szServerRemBootInfPath, ARRAYSIZE( szServerRemBootInfPath ));
Assert( dw );
ConcatenatePaths( szServerRemBootInfPath, L"\\System32\\", ARRAYSIZE(szServerRemBootInfPath) );
dwLen = lstrlen( szServerRemBootInfPath );
dw = LoadString( g_hinstance, IDS_REMBOOTINF, &szServerRemBootInfPath[dwLen], ARRAYSIZE( szServerRemBootInfPath ) - dwLen );
Assert( dw );
if ( 0xFFFFffff == GetFileAttributes( szServerRemBootInfPath ) ) {
MessageBoxFromStrings( NULL,
IDS_MISSING_INF_TITLE,
IDS_MISSING_INF_MESSAGE,
MB_OK );
fSomethingbroke = TRUE;
goto e0;
}
if ( g_Options.hinf == INVALID_HANDLE_VALUE ) {
g_Options.hinf = SetupOpenInfFile( szServerRemBootInfPath, NULL, INF_STYLE_WIN4, &uLineNum);
Assert( g_Options.hinf != INVALID_HANDLE_VALUE );
if ( g_Options.hinf == INVALID_HANDLE_VALUE ) {
fSomethingbroke = TRUE;
goto e0;
}
}
hr = CheckServerVersion( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
goto e0;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckFirstInstall( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckDirectoryTree( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
schSystem = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS );
if ( !schSystem ) {
hr = THR(E_FAIL);
fSomethingbroke = TRUE;
WCHAR szCaption[SMALL_BUFFER_SIZE];
DWORD LoadStringResult;
szCaption[0] = L'\0';
LoadStringResult = LoadString( g_hinstance, IDS_OPENING_SERVICE_MANAGER_TITLE, szCaption, ARRAYSIZE( szCaption ));
Assert( LoadStringResult );
ErrorBox( NULL, szCaption );
goto e0;
}
Assert( schSystem );
if ( schSystem ) {
hr = CheckBINL( schSystem );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckTFTPD( schSystem );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckSIS( schSystem );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckSISGroveler( schSystem );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
CloseServiceHandle( schSystem );
}
hr = CheckRegSrvDlls( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckRegistry( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = CheckOSChooser( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
hr = Paranoid( );
if ( FAILED(hr) ) {
fSomethingbroke = TRUE;
}
if ( hr == S_FALSE ) {
fSomethingFailed = TRUE;
}
e0:
if ( fSomethingbroke ) {
DebugMsg( "Something is broken. Installation check failed to complete.\n" );
hr = E_FAIL;
} else if ( fSomethingFailed ) {
DebugMsg( "Installation check found something wrong.\n" );
hr = S_FALSE;
} else {
DebugMsg( "Installation check succeeded.\n" );
hr = S_OK;
}
HRETURN(hr);
}