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.
 
 
 
 
 
 

378 lines
11 KiB

//+---------------------------------------------------------------------------
//
// Microsoft Windows
// Copyright (C) Microsoft Corporation, 1997.
//
// File: M A I N . C P P
//
// Contents: Code to provide a simple cmdline interface to
// the sample code functions
//
// Notes: The code in this file is not required to access any
// netcfg functionality. It merely provides a simple cmdline
// interface to the sample code functions provided in
// file snetcfg.cpp.
//
// Author: kumarp 28-September-98
//
//----------------------------------------------------------------------------
#include "pch.h"
#pragma hdrstop
#include "snetcfg.h"
#include "error.h"
#include <wbemcli.h>
#include <winnls.h>
#include "tracelog.h"
BOOL g_fVerbose = FALSE;
BOOL WlbsCheckSystemVersion ();
BOOL WlbsCheckFiles ();
HRESULT WlbsRegisterDlls ();
HRESULT WlbsCompileMof ();
// ----------------------------------------------------------------------
//
// Function: wmain
//
// Purpose: This is the main function for NLBWizard, which is for W2K
// installations of NLB. Making this application compatible
// with post-W2K releases will require changes, e.g., in
// functions WlbsCheckFiles and WlbsCheckSystemVersion.
//
// Arguments: standard main args
//
// Returns: 0 on success, non-zero otherwise
//
// Author: kumarp 25-December-97
//
// Notes:
//
EXTERN_C int __cdecl wmain (int argc, WCHAR * argv[]) {
HRESULT hr = S_OK;
WCHAR ch;
enum NetClass nc = NC_Unknown;
WCHAR szFileFullPath[MAX_PATH+1];
WCHAR szFileFullPathDest[MAX_PATH+1];
PWCHAR pwc;
PWSTR szFileComponent;
TRACELogRegister(L"wlbs");
LOG_INFO("Checking Windows version information.");
if (!WlbsCheckSystemVersion()) {
hr = MAKE_HRESULT(SEVERITY_ERROR, FACILITY_NLB, NLB_E_INVALID_OS);
LOG_ERROR("The NLB install pack can only be used on Windows 2000 Server Service Pack 1 or higher.");
goto error;
}
LOG_INFO("Checking for previous NLB installations.");
hr = FindIfComponentInstalled(_TEXT("ms_wlbs"));
if (hr == S_OK) {
/* AppCenter request on 1.9.01 to revert to S_FALSE in this case due to RTM proximity. */
/* hr = MAKE_HRESULT(SEVERITY_ERROR, FACILITY_NLB, NLB_E_ALREADY_INSTALLED); */
hr = S_FALSE;
LOG_ERROR("Network Load Balancing Service is already installed.");
goto error;
}
if (FAILED(hr)) {
LOG_ERROR("Warning: Error querying for Network Load Balancing Service. There may be errors in this installtion.");
}
LOG_INFO("Checking for necessary files.");
if (!WlbsCheckFiles()) {
hr = MAKE_HRESULT(SEVERITY_ERROR, FACILITY_NLB, NLB_E_FILES_MISSING);
LOG_ERROR1("Please install the NLB hotfix before running %ls", argv[0]);
goto error;
}
GetModuleFileName(NULL, szFileFullPath, MAX_PATH + 1);
pwc = wcsrchr(szFileFullPath, L'\\');
* (pwc + 1) = L'\0';
wcscat(szFileFullPath, L"netwlbs.inf");
LOG_INFO("Checking language version.");
switch (GetSystemDefaultLangID()) {
case 0x0409: case 0x0809: case 0x0c09: case 0x1009: case 0x1409: case 0x1809: case 0x1c09:
case 0x2009: case 0x2409: case 0x2809: case 0x2c09:
LOG_INFO1("English version detected 0x%x.", GetSystemDefaultLangID());
wcscat (szFileFullPath, L".eng");
break;
case 0x0411:
LOG_INFO1("Japanese version detected 0x%x.", GetSystemDefaultLangID());
wcscat (szFileFullPath, L".jpn");
break;
case 0x0407: case 0x0807: case 0x0c07: case 0x1007: case 0x1407:
LOG_INFO1("German version detected 0x%x.", GetSystemDefaultLangID());
wcscat (szFileFullPath, L".ger");
break;
case 0x040c: case 0x080c: case 0x0c0c: case 0x100c: case 0x140c:
LOG_INFO1("French version detected 0x%x.", GetSystemDefaultLangID());
wcscat (szFileFullPath, L".fr");
break;
default:
LOG_INFO1("Unsupported Language.Please contact PSS for a new %ls.", argv[0]);
wcscat (szFileFullPath, L".eng");
break;
}
/* First copy the .inf file. */
if (GetWindowsDirectory(szFileFullPathDest, MAX_PATH + 1) == 0)
{
hr = HRESULT_FROM_WIN32(GetLastError());
goto error;
}
wcscat(szFileFullPathDest, L"\\INF\\netwlbs.inf");
LOG_INFO("Copying the NLB .inf file.");
if (!CopyFile(szFileFullPath, szFileFullPathDest, FALSE)) {
hr = MAKE_HRESULT(SEVERITY_ERROR, FACILITY_NLB, NLB_E_INF_FAILURE);
LOG_ERROR2("Warning: Unable to copy the inf file %ls %ls.", szFileFullPath, szFileFullPathDest);
goto error;
}
/* Now install the service. */
hr = HrInstallNetComponent(L"ms_wlbs", NC_NetService, szFileFullPathDest);
if (!SUCCEEDED(hr)) {
LOG_ERROR("Error installing Network Load Balancing.");
goto error;
} else {
LOG_INFO("Installation of Network Load Balancing done.");
}
/* Change working directory to %TEMP%. Needed because of a IMofCopiler::CompileFile error on Win2K. */
WCHAR * szTempDir = _wgetenv(L"TEMP");
_wchdir(szTempDir);
LOG_INFO("Registering NLB Dlls.");
/* Register the provider .dll here. */
hr = WlbsRegisterDlls();
if (!SUCCEEDED(hr)) {
hr = MAKE_HRESULT(SEVERITY_ERROR, FACILITY_NLB, NLB_E_REGISTER_DLL);
LOG_ERROR("Error registering NLB Dlls.");
goto error;
}
LOG_INFO("Compiling the NLB MOF.");
/* Compile wlbsprov.mof here */
hr = WlbsCompileMof();
if (!SUCCEEDED(hr)) {
hr = MAKE_HRESULT(SEVERITY_ERROR, FACILITY_NLB, NLB_E_COMPILE_MOF);
LOG_ERROR("Error compiling the NLB MOF.");
goto error;
}
LOG_INFO("WLBS Setup successful.");
return hr;
error:
LOG_ERROR1("WLBS Setup failed 0x%x", hr);
return hr;
}
/* This checks whether the system on which NLB is being installed is a W2K Server or not. */
BOOL WlbsCheckSystemVersion () {
OSVERSIONINFOEX osinfo;
osinfo.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);
if (!GetVersionEx((LPOSVERSIONINFO)&osinfo)) return FALSE;
/* For install, we return TRUE only if its Windows 2000 Server. */
if ((osinfo.dwMajorVersion == 5) &&
(osinfo.dwMinorVersion == 0) &&
(osinfo.wProductType == VER_NT_SERVER) &&
!(osinfo.wSuiteMask & VER_SUITE_ENTERPRISE) &&
!(osinfo.wSuiteMask & VER_SUITE_DATACENTER))
return TRUE;
return FALSE;
}
BOOL WlbsCheckFiles () {
WCHAR * FileList [] = {
L"\\system32\\drivers\\wlbs.sys",
L"\\system32\\wlbs.exe",
L"\\help\\wlbs.chm",
L"\\help\\wlbs.hlp",
L"\\system32\\wlbsctrl.dll",
L"\\system32\\wbem\\wlbsprov.dll",
L"\\system32\\wbem\\wlbsprov.mof",
L"\\system32\\wbem\\wlbsprov.mfl",
L"\\inf\\netwlbsm.inf",
NULL
};
WCHAR wszPath [MAX_PATH + 1];
PWCHAR pwc;
INT i = 0;
BOOL first = FALSE;
WIN32_FIND_DATA FileFind;
HANDLE hdl;
while (FileList [i] != NULL) {
if (GetWindowsDirectory(wszPath, MAX_PATH + 1) == 0)
{
//
// This function returns the inverse of "first" as the status. Since we have a failure here, we set first = TRUE.
//
first = TRUE;
break;
}
wcscat(wszPath, FileList [i]);
hdl = FindFirstFile(wszPath, & FileFind);
if (hdl == INVALID_HANDLE_VALUE) {
if (!first) {
first = TRUE;
LOG_ERROR("Error: The following files were not found:");
}
LOG_ERROR1("%\tls",wszPath);
}
if (hdl != INVALID_HANDLE_VALUE)
FindClose(hdl);
i++;
}
return !first;
}
HRESULT WlbsRegisterDlls () {
WCHAR * DllList [] = {
L"\\wbem\\wlbsprov.dll",
NULL
};
INT i = 0;
WCHAR pszDllPath [MAX_PATH + 1];
HINSTANCE hLib;
CHAR * pszDllEntryPoint = "DllRegisterServer";
HRESULT (STDAPICALLTYPE * lpDllEntryPoint)(void);
HRESULT hr = S_OK;
if (!GetSystemDirectory(pszDllPath, MAX_PATH + 1)) {
hr = E_UNEXPECTED;
LOG_ERROR("GetSystemDirectoryFailed.");
return hr;
}
wcscat(pszDllPath, DllList [0]);
if (FAILED(hr = OleInitialize(NULL))) {
LOG_ERROR("OleInitialize Failed.");
return hr;
}
hLib = LoadLibrary(pszDllPath);
if (hLib == NULL) {
hr = E_UNEXPECTED;
LOG_ERROR("LoadLibrary for wlbsprov.dll Failed.");
goto CleanOle;
}
(FARPROC &)lpDllEntryPoint = GetProcAddress(hLib, pszDllEntryPoint);
if (lpDllEntryPoint == NULL) {
hr = E_UNEXPECTED;
LOG_ERROR("DllRegisterServer was not found.");
goto CleanLib;
}
if (FAILED(hr = (*lpDllEntryPoint)())) {
LOG_ERROR("DllRegisterServer failed.");
goto CleanLib;
}
LOG_INFO("Dll Registration Succeeded.");
CleanLib:
FreeLibrary(hLib);
CleanOle:
OleUninitialize();
return hr;
}
HRESULT WlbsCompileMof () {
WCHAR * MofList [] = {
L"\\wbem\\wlbsprov.mof",
NULL
};
IMofCompiler * pMofComp = NULL;
WBEM_COMPILE_STATUS_INFO Info;
HRESULT hr = S_OK;
WCHAR pszMofPath [MAX_PATH + 1];
if (!GetSystemDirectory(pszMofPath, MAX_PATH + 1)) {
hr = E_UNEXPECTED;
LOG_ERROR("GetSystemDirectoryFailed.");
return hr;
}
wcscat(pszMofPath, MofList [0]);
hr = CoInitialize(NULL);
if (FAILED(hr)) {
LOG_ERROR("CoInitialize failed.");
return hr;
}
hr = CoCreateInstance(CLSID_MofCompiler, NULL, CLSCTX_INPROC_SERVER, IID_IMofCompiler, (LPVOID *)&pMofComp);
if (FAILED(hr)) {
LOG_ERROR("CoCreateInstance Failed.");
switch (hr) {
case REGDB_E_CLASSNOTREG:
LOG_ERROR("Not registered.");
break;
case CLASS_E_NOAGGREGATION:
LOG_ERROR("No aggregration.");
break;
default:
LOG_ERROR1("Error ox%x.", hr);
break;
}
CoUninitialize();
return hr;
}
hr = pMofComp->CompileFile(pszMofPath, NULL, NULL, NULL, NULL, 0, 0, 0, &Info);
if (hr != WBEM_S_NO_ERROR)
LOG_ERROR("Compile Failed.");
pMofComp->Release();
CoUninitialize();
return Info.hRes;
}