675 lines
17 KiB
C++
675 lines
17 KiB
C++
|
/****************************************************************************
|
||
|
|
|
||
|
|
Copyright (c) Microsoft Corporation 1998
|
||
|
|
All rights reserved
|
||
|
|
|
||
|
|
File: SERVERDLG.CPP
|
||
|
|
|
||
|
|
|
||
|
|
***************************************************************************/
|
||
|
|
|
||
|
|
#include "pch.h"
|
||
|
|
#include "callback.h"
|
||
|
|
#include "utils.h"
|
||
|
|
#include <winver.h>
|
||
|
|
#include <sputils.h>
|
||
|
|
|
||
|
|
DEFINE_MODULE( "RIPREP" )
|
||
|
|
|
||
|
|
|
||
|
|
BOOLEAN
|
||
|
|
GetInstalledProductType(
|
||
|
|
OUT PDWORD ProductType,
|
||
|
|
OUT PDWORD ProductSuite
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
retrieves the product type and suite from a running system
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
ProductType - receives a VER_NT_* constant.
|
||
|
|
ProductSuite - receives a VER_SUITE_* mask for the system.
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
TRUE indicates success
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
OSVERSIONINFOEX VersionInfo;
|
||
|
|
|
||
|
|
VersionInfo.dwOSVersionInfoSize = sizeof(VersionInfo);
|
||
|
|
|
||
|
|
if (GetVersionEx((OSVERSIONINFO *)&VersionInfo)) {
|
||
|
|
//
|
||
|
|
// make domain controllers and servers look the same
|
||
|
|
//
|
||
|
|
*ProductType = (VersionInfo.wProductType == VER_NT_DOMAIN_CONTROLLER)
|
||
|
|
? VER_NT_SERVER
|
||
|
|
: VersionInfo.wProductType;
|
||
|
|
//
|
||
|
|
// we only care about suites that have a SKU associated with them.
|
||
|
|
//
|
||
|
|
*ProductSuite = (VersionInfo.wSuiteMask
|
||
|
|
& (VER_SUITE_ENTERPRISE | VER_SUITE_DATACENTER | VER_SUITE_PERSONAL)) ;
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
return(TRUE);
|
||
|
|
}
|
||
|
|
|
||
|
|
return(FALSE);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
pSetupEnablePrivilegeW(
|
||
|
|
IN PCWSTR PrivilegeName,
|
||
|
|
IN BOOL Enable
|
||
|
|
)
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Enable or disable a given named privilege.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
PrivilegeName - supplies the name of a system privilege.
|
||
|
|
|
||
|
|
Enable - flag indicating whether to enable or disable the privilege.
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
Boolean value indicating whether the operation was successful.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
{
|
||
|
|
HANDLE Token;
|
||
|
|
BOOL b;
|
||
|
|
TOKEN_PRIVILEGES NewPrivileges;
|
||
|
|
LUID Luid;
|
||
|
|
|
||
|
|
if(!OpenProcessToken(GetCurrentProcess(),TOKEN_ADJUST_PRIVILEGES,&Token)) {
|
||
|
|
return(FALSE);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(!LookupPrivilegeValue(NULL,PrivilegeName,&Luid)) {
|
||
|
|
CloseHandle(Token);
|
||
|
|
return(FALSE);
|
||
|
|
}
|
||
|
|
|
||
|
|
NewPrivileges.PrivilegeCount = 1;
|
||
|
|
NewPrivileges.Privileges[0].Luid = Luid;
|
||
|
|
NewPrivileges.Privileges[0].Attributes = Enable ? SE_PRIVILEGE_ENABLED : 0;
|
||
|
|
|
||
|
|
b = AdjustTokenPrivileges(
|
||
|
|
Token,
|
||
|
|
FALSE,
|
||
|
|
&NewPrivileges,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
NULL
|
||
|
|
);
|
||
|
|
|
||
|
|
CloseHandle(Token);
|
||
|
|
|
||
|
|
return(b);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
GetSuiteMaskFromPath(
|
||
|
|
IN PCWSTR PathToSearch,
|
||
|
|
OUT PDWORD SuiteMask
|
||
|
|
)
|
||
|
|
{
|
||
|
|
#define HIVENAME L"riprepsetupreg"
|
||
|
|
TCHAR lpszSetupReg[MAX_PATH] = HIVENAME L"\\ControlSet001\\Services\\setupdd";
|
||
|
|
|
||
|
|
WCHAR Path[MAX_PATH];
|
||
|
|
WCHAR DestPath[MAX_PATH];
|
||
|
|
LONG rslt;
|
||
|
|
HKEY hKey;
|
||
|
|
DWORD Type;
|
||
|
|
DWORD Buffer[4];
|
||
|
|
DWORD BufferSize = sizeof(Buffer);
|
||
|
|
DWORD i;
|
||
|
|
BOOLEAN RetVal = FALSE;
|
||
|
|
|
||
|
|
GetTempPath(ARRAYSIZE(Path),Path);
|
||
|
|
GetTempFileName( Path, L"ripr", 0, DestPath);
|
||
|
|
|
||
|
|
wcscpy(Path, PathToSearch);
|
||
|
|
wcscat(Path, L"\\setupreg.hiv");
|
||
|
|
|
||
|
|
if (!CopyFile(Path,DestPath,FALSE)) {
|
||
|
|
goto e0;
|
||
|
|
}
|
||
|
|
|
||
|
|
SetFileAttributes(DestPath,FILE_ATTRIBUTE_NORMAL);
|
||
|
|
|
||
|
|
//
|
||
|
|
// need SE_RESTORE_NAME priviledge to call this API!
|
||
|
|
//
|
||
|
|
pSetupEnablePrivilegeW( SE_RESTORE_NAME, TRUE );
|
||
|
|
|
||
|
|
//
|
||
|
|
// try to unload this first in case we faulted or something and the key is still loaded
|
||
|
|
//
|
||
|
|
RegUnLoadKey( HKEY_LOCAL_MACHINE, HIVENAME );
|
||
|
|
|
||
|
|
rslt = RegLoadKey( HKEY_LOCAL_MACHINE, HIVENAME, DestPath );
|
||
|
|
if (rslt != ERROR_SUCCESS) {
|
||
|
|
goto e1;
|
||
|
|
}
|
||
|
|
|
||
|
|
rslt = RegOpenKey(HKEY_LOCAL_MACHINE,lpszSetupReg,&hKey);
|
||
|
|
if (rslt != ERROR_SUCCESS) {
|
||
|
|
goto e2;
|
||
|
|
}
|
||
|
|
|
||
|
|
rslt = RegQueryValueEx(hKey, NULL, NULL, &Type, (LPBYTE) Buffer, &BufferSize);
|
||
|
|
if (rslt != ERROR_SUCCESS || Type != REG_BINARY) {
|
||
|
|
goto e3;
|
||
|
|
}
|
||
|
|
|
||
|
|
*SuiteMask=Buffer[3];
|
||
|
|
|
||
|
|
RetVal = TRUE;
|
||
|
|
|
||
|
|
e3:
|
||
|
|
RegCloseKey( hKey );
|
||
|
|
e2:
|
||
|
|
RegUnLoadKey( HKEY_LOCAL_MACHINE, HIVENAME );
|
||
|
|
|
||
|
|
e1:
|
||
|
|
if (GetFileAttributes(DestPath) != 0xFFFFFFFF) {
|
||
|
|
SetFileAttributes(DestPath,FILE_ATTRIBUTE_NORMAL);
|
||
|
|
DeleteFile(DestPath);
|
||
|
|
wcscat(DestPath, L".LOG");
|
||
|
|
SetFileAttributes(DestPath,FILE_ATTRIBUTE_NORMAL);
|
||
|
|
DeleteFile(DestPath);
|
||
|
|
}
|
||
|
|
|
||
|
|
pSetupEnablePrivilegeW( SE_RESTORE_NAME, FALSE );
|
||
|
|
e0:
|
||
|
|
return(RetVal);
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOLEAN
|
||
|
|
GetProductTypeFromPath(
|
||
|
|
OUT PDWORD ProductType,
|
||
|
|
OUT PDWORD ProductSuite,
|
||
|
|
IN PCWSTR PathToSearch )
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
retrieves the product type and suite by looking in txtsetup.sif
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
ProductType - receives a VER_NT_* constant.
|
||
|
|
ProductSuite - receives a VER_SUITE_* mask for the system.
|
||
|
|
PathToSearch - specifies the path to the txtsetup.sif to be searched
|
||
|
|
|
||
|
|
Return Value:
|
||
|
|
|
||
|
|
TRUE indicates success
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
WCHAR Path[MAX_PATH];
|
||
|
|
UINT DontCare;
|
||
|
|
HINF hInf;
|
||
|
|
WCHAR Type[20];
|
||
|
|
BOOLEAN RetVal = TRUE;
|
||
|
|
INFCONTEXT Context;
|
||
|
|
|
||
|
|
wcscpy(Path, PathToSearch);
|
||
|
|
wcscat(Path, L"\\txtsetup.sif");
|
||
|
|
|
||
|
|
hInf = SetupOpenInfFile( Path, NULL, INF_STYLE_WIN4, &DontCare );
|
||
|
|
if (hInf != INVALID_HANDLE_VALUE) {
|
||
|
|
if (SetupFindFirstLine( hInf, L"SetupData", L"ProductType", &Context) &&
|
||
|
|
SetupGetStringField( &Context, 1, Type, ARRAYSIZE(Type), NULL)) {
|
||
|
|
switch (Type[0]) {
|
||
|
|
case L'0':
|
||
|
|
*ProductType = VER_NT_WORKSTATION;
|
||
|
|
*ProductSuite = 0;
|
||
|
|
break;
|
||
|
|
case L'1':
|
||
|
|
*ProductType = VER_NT_SERVER;
|
||
|
|
//
|
||
|
|
// HACK alert: we have to call this API because txtsetup.sif
|
||
|
|
// didn't have the correct product type in it in win2k.
|
||
|
|
// So we do it the hard way.
|
||
|
|
//
|
||
|
|
if (!GetSuiteMaskFromPath( PathToSearch, ProductSuite)) {
|
||
|
|
*ProductSuite = 0;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case L'2':
|
||
|
|
*ProductType = VER_NT_SERVER;
|
||
|
|
*ProductSuite = VER_SUITE_ENTERPRISE;
|
||
|
|
break;
|
||
|
|
case L'3':
|
||
|
|
*ProductType = VER_NT_SERVER;
|
||
|
|
*ProductSuite = VER_SUITE_ENTERPRISE | VER_SUITE_DATACENTER;
|
||
|
|
break;
|
||
|
|
case L'4':
|
||
|
|
*ProductType = VER_NT_WORKSTATION;
|
||
|
|
*ProductSuite = VER_SUITE_PERSONAL;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
ASSERT( FALSE && L"Unknown type in txtsetup.sif ProductType" );
|
||
|
|
RetVal = FALSE;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
SetupCloseInfFile(hInf);
|
||
|
|
} else {
|
||
|
|
RetVal = FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
return(RetVal);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// GetNtVersionInfo( )
|
||
|
|
//
|
||
|
|
// Retrieves the build version from the kernel
|
||
|
|
//
|
||
|
|
BOOLEAN
|
||
|
|
GetNtVersionInfo(
|
||
|
|
PULONGLONG Version,
|
||
|
|
PWCHAR SearchDir
|
||
|
|
)
|
||
|
|
{
|
||
|
|
DWORD Error = ERROR_SUCCESS;
|
||
|
|
DWORD FileVersionInfoSize;
|
||
|
|
DWORD VersionHandle;
|
||
|
|
ULARGE_INTEGER TmpVersion;
|
||
|
|
PVOID VersionInfo;
|
||
|
|
VS_FIXEDFILEINFO * FixedFileInfo;
|
||
|
|
UINT FixedFileInfoLength;
|
||
|
|
WCHAR Path[MAX_PATH];
|
||
|
|
BOOLEAN fResult = FALSE;
|
||
|
|
|
||
|
|
TraceFunc("GetNtVersionInfo( )\n");
|
||
|
|
|
||
|
|
// Resulting string should be something like:
|
||
|
|
// "\\server\reminst\Setup\English\Images\nt50.wks\i386\ntoskrnl.exe"
|
||
|
|
|
||
|
|
if (!SearchDir) {
|
||
|
|
goto e0;
|
||
|
|
}
|
||
|
|
wcscpy(Path, SearchDir);
|
||
|
|
wcscat(Path, L"\\ntoskrnl.exe");
|
||
|
|
|
||
|
|
FileVersionInfoSize = GetFileVersionInfoSize(Path, &VersionHandle);
|
||
|
|
if (FileVersionInfoSize == 0)
|
||
|
|
goto e0;
|
||
|
|
|
||
|
|
VersionInfo = LocalAlloc( LPTR, FileVersionInfoSize );
|
||
|
|
if (VersionInfo == NULL)
|
||
|
|
goto e0;
|
||
|
|
|
||
|
|
if (!GetFileVersionInfo(
|
||
|
|
Path,
|
||
|
|
VersionHandle,
|
||
|
|
FileVersionInfoSize,
|
||
|
|
VersionInfo))
|
||
|
|
goto e1;
|
||
|
|
|
||
|
|
if (!VerQueryValue(
|
||
|
|
VersionInfo,
|
||
|
|
L"\\",
|
||
|
|
(LPVOID*)&FixedFileInfo,
|
||
|
|
&FixedFileInfoLength))
|
||
|
|
goto e1;
|
||
|
|
|
||
|
|
TmpVersion.HighPart = FixedFileInfo->dwFileVersionMS;
|
||
|
|
TmpVersion.LowPart = FixedFileInfo->dwFileVersionLS;
|
||
|
|
*Version = TmpVersion.QuadPart;
|
||
|
|
|
||
|
|
fResult = TRUE;
|
||
|
|
|
||
|
|
e1:
|
||
|
|
LocalFree( VersionInfo );
|
||
|
|
e0:
|
||
|
|
RETURN(fResult);
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// VerifyMatchingFlatImage( )
|
||
|
|
//
|
||
|
|
BOOLEAN
|
||
|
|
VerifyMatchingFlatImage(
|
||
|
|
PULONGLONG VersionNeeded OPTIONAL
|
||
|
|
)
|
||
|
|
{
|
||
|
|
TraceFunc( "VerifyMatchingFlatImage( )\n" );
|
||
|
|
|
||
|
|
BOOLEAN fResult = FALSE; // assume failure
|
||
|
|
DWORD dwLen;
|
||
|
|
WCHAR szPath[ MAX_PATH ];
|
||
|
|
WIN32_FIND_DATA fd;
|
||
|
|
HANDLE hFind;
|
||
|
|
ULONGLONG OurVersion;
|
||
|
|
DWORD OurProductType = 0, OurProductSuiteMask = 0;
|
||
|
|
|
||
|
|
GetSystemDirectory( szPath, ARRAYSIZE( szPath ));
|
||
|
|
GetNtVersionInfo( &OurVersion, szPath );
|
||
|
|
|
||
|
|
GetInstalledProductType( &OurProductType, &OurProductSuiteMask );
|
||
|
|
|
||
|
|
if (VersionNeeded) {
|
||
|
|
*VersionNeeded = OurVersion;
|
||
|
|
}
|
||
|
|
|
||
|
|
DebugMsg(
|
||
|
|
"Our NTOSKRNL verion: %u.%u:%u.%u Type: %d Suite: %d\n",
|
||
|
|
HIWORD(((PULARGE_INTEGER)&OurVersion)->HighPart),
|
||
|
|
LOWORD(((PULARGE_INTEGER)&OurVersion)->HighPart),
|
||
|
|
HIWORD(((PULARGE_INTEGER)&OurVersion)->LowPart),
|
||
|
|
LOWORD(((PULARGE_INTEGER)&OurVersion)->LowPart),
|
||
|
|
OurProductType,
|
||
|
|
OurProductSuiteMask);
|
||
|
|
|
||
|
|
wsprintf( szPath,
|
||
|
|
L"\\\\%s\\REMINST\\Setup\\%s\\%s\\",
|
||
|
|
g_ServerName,
|
||
|
|
g_Language,
|
||
|
|
REMOTE_INSTALL_IMAGE_DIR_W );
|
||
|
|
|
||
|
|
dwLen = wcslen( szPath );
|
||
|
|
|
||
|
|
wcscat( szPath, L"*" );
|
||
|
|
|
||
|
|
hFind = FindFirstFile( szPath, &fd );
|
||
|
|
if ( hFind != INVALID_HANDLE_VALUE )
|
||
|
|
{
|
||
|
|
do {
|
||
|
|
if ( (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
|
||
|
|
&& StrCmp( fd.cFileName, L"." ) != 0
|
||
|
|
&& StrCmp( fd.cFileName, L".." ) != 0 )
|
||
|
|
{
|
||
|
|
ULONGLONG Version;
|
||
|
|
DWORD ProductType = 0, ProductSuiteMask = 0;
|
||
|
|
wsprintf( &szPath[dwLen],
|
||
|
|
L"%s\\%s",
|
||
|
|
fd.cFileName,
|
||
|
|
g_Architecture );
|
||
|
|
|
||
|
|
if ( GetNtVersionInfo( &Version, szPath ) &&
|
||
|
|
GetProductTypeFromPath(
|
||
|
|
&ProductType,
|
||
|
|
&ProductSuiteMask,
|
||
|
|
szPath ))
|
||
|
|
{
|
||
|
|
DebugMsg(
|
||
|
|
"%s's verion: %u.%u:%u.%u Type: %d Suite: %d\n",
|
||
|
|
fd.cFileName,
|
||
|
|
HIWORD(((PULARGE_INTEGER)&Version)->HighPart),
|
||
|
|
LOWORD(((PULARGE_INTEGER)&Version)->HighPart),
|
||
|
|
HIWORD(((PULARGE_INTEGER)&Version)->LowPart),
|
||
|
|
LOWORD(((PULARGE_INTEGER)&Version)->LowPart),
|
||
|
|
ProductType,
|
||
|
|
ProductSuiteMask);
|
||
|
|
|
||
|
|
if ( OurVersion == Version &&
|
||
|
|
OurProductType == ProductType &&
|
||
|
|
OurProductSuiteMask == ProductSuiteMask )
|
||
|
|
{
|
||
|
|
wcscpy( g_ImageName, szPath );
|
||
|
|
fResult = TRUE;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} while ( FindNextFile( hFind, &fd ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
FindClose( hFind );
|
||
|
|
|
||
|
|
RETURN(fResult);
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// VerifyServerName( )
|
||
|
|
//
|
||
|
|
// Check to see if the server is a Remote Installation Server by
|
||
|
|
// checking for the existance of the "REMINST" share.
|
||
|
|
//
|
||
|
|
DWORD
|
||
|
|
VerifyServerName( )
|
||
|
|
{
|
||
|
|
TraceFunc( "VerifyServerName( )\n" );
|
||
|
|
|
||
|
|
NET_API_STATUS netStatus;
|
||
|
|
SHARE_INFO_0 * psi;
|
||
|
|
|
||
|
|
HCURSOR oldCursor = SetCursor( LoadCursor( NULL, IDC_WAIT ) );
|
||
|
|
|
||
|
|
netStatus = NetShareGetInfo( g_ServerName, L"REMINST", 0, (LPBYTE *) &psi );
|
||
|
|
if ( netStatus == NERR_Success )
|
||
|
|
{
|
||
|
|
NetApiBufferFree( psi );
|
||
|
|
}
|
||
|
|
|
||
|
|
SetCursor( oldCursor );
|
||
|
|
|
||
|
|
RETURN(netStatus);
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
VerifyServerAccess(
|
||
|
|
PCWSTR ServerShareName,
|
||
|
|
PCWSTR ServerLanguage
|
||
|
|
)
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Routine Description:
|
||
|
|
|
||
|
|
Checks permissions on the RIPREP server machine by trying to create a file on the
|
||
|
|
server. The temp file is then deleted.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
|
||
|
|
ServerShareName - path that we want to check permissions on.
|
||
|
|
ServerLanguage - indicates the language subdirectory to check for access in.
|
||
|
|
|
||
|
|
|
||
|
|
Return value:
|
||
|
|
|
||
|
|
TRUE if the user has access to the server, FALSE otherwise.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
{
|
||
|
|
TraceFunc( "VerifyServerAccess( )\n" );
|
||
|
|
|
||
|
|
WCHAR FileName[MAX_PATH];
|
||
|
|
WCHAR FilePath[MAX_PATH];
|
||
|
|
BOOL RetVal;
|
||
|
|
|
||
|
|
HCURSOR oldCursor = SetCursor( LoadCursor( NULL, IDC_WAIT ) );
|
||
|
|
|
||
|
|
wsprintf(FilePath,
|
||
|
|
L"\\\\%s\\reminst\\Setup\\%s\\Images",
|
||
|
|
ServerShareName,
|
||
|
|
ServerLanguage );
|
||
|
|
|
||
|
|
RetVal = GetTempFileName( FilePath , L"ACC", 0, FileName );
|
||
|
|
if (RetVal) {
|
||
|
|
//
|
||
|
|
// delete the file, we don't want to leave turds on the server
|
||
|
|
//
|
||
|
|
DeleteFile(FileName);
|
||
|
|
RetVal = TRUE;
|
||
|
|
} else if (GetLastError() == ERROR_ACCESS_DENIED) {
|
||
|
|
RetVal = FALSE;
|
||
|
|
} else {
|
||
|
|
//
|
||
|
|
// GetTempFileName failed, but not because of an access problem, so
|
||
|
|
// return success
|
||
|
|
//
|
||
|
|
RetVal = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
SetCursor( oldCursor );
|
||
|
|
|
||
|
|
RETURN(RetVal);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// ServerDlgCheckNextButtonActivation( )
|
||
|
|
//
|
||
|
|
VOID
|
||
|
|
ServerDlgCheckNextButtonActivation(
|
||
|
|
HWND hDlg )
|
||
|
|
{
|
||
|
|
TraceFunc( "ServerDlgCheckNextButtonActivation( )\n" );
|
||
|
|
WCHAR szName[ MAX_PATH ];
|
||
|
|
GetDlgItemText( hDlg, IDC_E_SERVER, szName, ARRAYSIZE(szName));
|
||
|
|
PropSheet_SetWizButtons( GetParent( hDlg ), PSWIZB_BACK | (wcslen(szName) ? PSWIZB_NEXT : 0 ) );
|
||
|
|
TraceFuncExit( );
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// ServerDlgProc()
|
||
|
|
//
|
||
|
|
INT_PTR CALLBACK
|
||
|
|
ServerDlgProc(
|
||
|
|
HWND hDlg,
|
||
|
|
UINT uMsg,
|
||
|
|
WPARAM wParam,
|
||
|
|
LPARAM lParam )
|
||
|
|
{
|
||
|
|
ULARGE_INTEGER ImageVersion;
|
||
|
|
WCHAR szTemp[ 1024 ];
|
||
|
|
WCHAR szCaption[ 1024 ];
|
||
|
|
WCHAR ErrorText[ 1024 ];
|
||
|
|
DWORD dw;
|
||
|
|
|
||
|
|
switch (uMsg)
|
||
|
|
{
|
||
|
|
default:
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
case WM_INITDIALOG:
|
||
|
|
CenterDialog( GetParent( hDlg ) );
|
||
|
|
return FALSE;
|
||
|
|
|
||
|
|
case WM_COMMAND:
|
||
|
|
switch ( LOWORD( wParam ) )
|
||
|
|
{
|
||
|
|
case IDC_E_SERVER:
|
||
|
|
if ( HIWORD( wParam ) == EN_CHANGE )
|
||
|
|
{
|
||
|
|
ServerDlgCheckNextButtonActivation( hDlg );
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case WM_NOTIFY:
|
||
|
|
SetWindowLongPtr( hDlg, DWLP_MSGRESULT, FALSE );
|
||
|
|
LPNMHDR lpnmhdr = (LPNMHDR) lParam;
|
||
|
|
switch ( lpnmhdr->code )
|
||
|
|
{
|
||
|
|
case PSN_WIZNEXT:
|
||
|
|
{
|
||
|
|
GetDlgItemText( hDlg, IDC_E_SERVER, g_ServerName, ARRAYSIZE(g_ServerName) );
|
||
|
|
|
||
|
|
//remove the wackwack if found
|
||
|
|
if ( g_ServerName[0] == L'\\' && g_ServerName[1] == L'\\' )
|
||
|
|
{
|
||
|
|
wcscpy( g_ServerName, &g_ServerName[2] );
|
||
|
|
}
|
||
|
|
|
||
|
|
Assert( wcslen( g_ServerName ) );
|
||
|
|
DWORD dwErr = VerifyServerName( );
|
||
|
|
if ( dwErr != ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
switch (dwErr)
|
||
|
|
{
|
||
|
|
case NERR_NetNameNotFound:
|
||
|
|
MessageBoxFromStrings( hDlg, IDS_NOT_A_BINL_SERVER_TITLE, IDS_NOT_A_BINL_SERVER_TEXT, MB_OK );
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
dw = LoadString( g_hinstance, IDS_FAILED_TO_CONTACT_SERVER_TITLE, szTemp, ARRAYSIZE(szTemp) );
|
||
|
|
Assert( dw );
|
||
|
|
MessageBoxFromError( hDlg, szTemp, dwErr, NULL, MB_OK );
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
SetWindowLongPtr( hDlg, DWLP_MSGRESULT, -1 ); // don't go on
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL fFoundMatchingFlatImage = VerifyMatchingFlatImage( (PULONGLONG)&ImageVersion );
|
||
|
|
if ( !fFoundMatchingFlatImage )
|
||
|
|
{
|
||
|
|
dw = LoadString( g_hinstance, IDS_MISSING_BACKING_FLAT_IMAGE_TEXT, szTemp, ARRAYSIZE(szTemp) );
|
||
|
|
ASSERT(dw);
|
||
|
|
|
||
|
|
dw = LoadString( g_hinstance, IDS_MISSING_BACKING_FLAT_IMAGE_TITLE, szCaption, ARRAYSIZE(szCaption) );
|
||
|
|
ASSERT(dw);
|
||
|
|
|
||
|
|
wsprintf(
|
||
|
|
ErrorText,
|
||
|
|
szTemp,
|
||
|
|
HIWORD(ImageVersion.HighPart),
|
||
|
|
LOWORD(ImageVersion.HighPart),
|
||
|
|
HIWORD(ImageVersion.LowPart),
|
||
|
|
g_Language );
|
||
|
|
|
||
|
|
MessageBox( hDlg, ErrorText, szCaption, MB_OK );
|
||
|
|
|
||
|
|
SetWindowLongPtr( hDlg, DWLP_MSGRESULT, -1 ); // don't go on
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!VerifyServerAccess(g_ServerName,g_Language)) {
|
||
|
|
dw = LoadString( g_hinstance, IDS_SERVER_ACCESS_DESC, ErrorText, ARRAYSIZE(ErrorText) );
|
||
|
|
ASSERT(dw);
|
||
|
|
|
||
|
|
dw = LoadString( g_hinstance, IDS_SERVER_ACCESS, szCaption, ARRAYSIZE(szCaption) );
|
||
|
|
ASSERT(dw);
|
||
|
|
|
||
|
|
MessageBox( hDlg, ErrorText, szCaption, MB_OK );
|
||
|
|
|
||
|
|
SetWindowLongPtr( hDlg, DWLP_MSGRESULT, -1 ); // don't go on
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case PSN_QUERYCANCEL:
|
||
|
|
return VerifyCancel( hDlg );
|
||
|
|
|
||
|
|
case PSN_SETACTIVE:
|
||
|
|
SetDlgItemText( hDlg, IDC_E_SERVER, g_ServerName );
|
||
|
|
ServerDlgCheckNextButtonActivation( hDlg );
|
||
|
|
ClearMessageQueue( );
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|