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.
316 lines
9.5 KiB
316 lines
9.5 KiB
/////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright(C) Microsoft Corporation all rights reserved.
|
|
//
|
|
// Module: DoUpgrade.cpp
|
|
//
|
|
// Description: Implementation of CDoNT4OrCleanUpgrade, CMigrateOrUpgradeWindowsDB
|
|
// and CUpgradeNT4
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
#include "stdafx.h"
|
|
#include "DoUpgrade.h"
|
|
#include "migratemdb.h"
|
|
#include "migratecontent.h"
|
|
#include "Version.h"
|
|
#include "ias.h"
|
|
#include "iasdb.h"
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
// Execute
|
|
// Does either a clean NT4 upgrade or nothing (clean install)
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
void CDoNT4OrCleanUpgrade::Execute()
|
|
{
|
|
if ( m_Utils.IsNT4Corp() || m_Utils.IsNT4Isp() )
|
|
{
|
|
CUpgradeNT4 UpgradeNT4;
|
|
UpgradeNT4.Execute();
|
|
}
|
|
// else this is a clean install: nothing to do
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
// GetVersionNumber
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
LONG CMigrateOrUpgradeWindowsDB::GetVersionNumber()
|
|
{
|
|
return m_GlobalData.m_pRefVersion->GetVersion();
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
// CLASS CMigrateOrUpgradeWindowsDB
|
|
// expects the files:
|
|
// iasold.mdb as a Whistler before proxy or a Win2k mdb file
|
|
// ias.mdb to be a Whistler file (already good)
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////
|
|
// Constructor
|
|
///////////////
|
|
CMigrateOrUpgradeWindowsDB::CMigrateOrUpgradeWindowsDB(IAS_SHOW_TOKEN_LIST configType)
|
|
:m_Outcome(E_FAIL),
|
|
m_GlobalTransaction(CGlobalTransaction::Instance()),
|
|
m_Utils(CUtils::Instance()),
|
|
m_ConfigType(configType)
|
|
{
|
|
/////////////////////////////////////////////////
|
|
// Check %TMP% and create a directory if needed
|
|
// That's to fix a bug with JET
|
|
/////////////////////////////////////////////////
|
|
IASCreateTmpDirectory();
|
|
|
|
// Create the paths to the mdb files
|
|
LONG Result = m_Utils.GetIAS2Directory(m_IASWhistlerPath);
|
|
if ( Result != ERROR_SUCCESS )
|
|
{
|
|
_com_util::CheckError(HRESULT_FROM_WIN32(Result));
|
|
}
|
|
m_IASOldPath = m_IASWhistlerPath;
|
|
|
|
m_IASOldPath += L"\\iasold.mdb";
|
|
m_IASWhistlerPath += L"\\ias.mdb";
|
|
|
|
//////////////////
|
|
// Database inits
|
|
//////////////////
|
|
|
|
// Open the DataSource and session for ias.mdb
|
|
// and Initialize the GlobalTransaction
|
|
m_GlobalTransaction.OpenStdDataSource(m_IASWhistlerPath);
|
|
|
|
// Open the DataSource and session for iasnew.mdb
|
|
m_GlobalTransaction.OpenRefDataSource(m_IASOldPath);
|
|
|
|
// create instances of CObjects and CProperties
|
|
m_GlobalData.InitStandard(m_GlobalTransaction.GetStdSession());
|
|
|
|
// create instances of CObjects and CProperties for the Ref database
|
|
m_GlobalData.InitRef(m_GlobalTransaction.GetRefSession());
|
|
};
|
|
|
|
|
|
//////////////
|
|
// Destructor
|
|
//////////////
|
|
CMigrateOrUpgradeWindowsDB::~CMigrateOrUpgradeWindowsDB()
|
|
{
|
|
m_GlobalData.Clean();
|
|
|
|
if ( FAILED(m_Outcome) )
|
|
{
|
|
/////////////////////////
|
|
// Abort the transaction
|
|
/////////////////////////
|
|
m_GlobalTransaction.Abort();
|
|
|
|
SetLastError(E_FAIL);
|
|
}
|
|
else
|
|
{
|
|
///////////
|
|
// Success
|
|
///////////
|
|
m_GlobalTransaction.Commit();
|
|
|
|
}
|
|
/////////////////////////////////////////////
|
|
// close the session and then the datasource
|
|
/////////////////////////////////////////////
|
|
m_GlobalTransaction.MyCloseDataSources();
|
|
};
|
|
|
|
|
|
/////////////////////////////
|
|
// Execute
|
|
// IAS_WIN2K_VERSION = 0;
|
|
// IAS_WHISTLER1_VERSION = 1;
|
|
// IAS_WHISTLER_BETA1_VERSION = 2;
|
|
// IAS_WHISTLER_BETA2_VERSION = 3;
|
|
// IAS_WHISTLER_RC1_VERSION = 4;
|
|
// IAS_WHISTLER_RC1A_VERSION = 5;
|
|
// IAS_WHISTLER_RC1B_VERSION = 6;
|
|
// IAS_WHISTLER_RC2_VERSION = 7;
|
|
// IAS_CURRENT_VERSION = IAS_WHISTLER_RC2_VERSION;
|
|
//
|
|
/////////////////////////////
|
|
void CMigrateOrUpgradeWindowsDB::Execute()
|
|
{
|
|
CMigrateContent MigrateContent(m_Utils, m_GlobalData, m_ConfigType);
|
|
|
|
////////////////////////////////////////////////////
|
|
// Check the version number (of iasold.mdb)
|
|
////////////////////////////////////////////////////
|
|
LONG CurrentVersion = GetVersionNumber();
|
|
////////////////////////////////////////////////////
|
|
// Migrate the content from iasold.mdb into ias.mdb
|
|
////////////////////////////////////////////////////
|
|
switch (CurrentVersion)
|
|
{
|
|
case IAS_WIN2K_VERSION:
|
|
case IAS_WHISTLER1_VERSION:
|
|
{
|
|
MigrateContent.Migrate();
|
|
// Everything is ok. Set m_Outcome = S_OK to Commit the IAS.mdb
|
|
m_Outcome = S_OK;
|
|
break;
|
|
}
|
|
case IAS_WHISTLER_BETA1_VERSION:
|
|
{
|
|
MigrateContent.UpdateWhistler(
|
|
(CMigrateContent::updateChangePassword |
|
|
CMigrateContent::migrateEapConfig)
|
|
);
|
|
// Everything is ok. Set m_Outcome = S_OK to Commit the IAS.mdb
|
|
m_Outcome = S_OK;
|
|
break;
|
|
}
|
|
case IAS_WHISTLER_BETA2_VERSION:
|
|
case IAS_WHISTLER_RC1_VERSION:
|
|
case IAS_WHISTLER_RC1A_VERSION:
|
|
{
|
|
MigrateContent.UpdateWhistler(
|
|
CMigrateContent::migrateEapConfig
|
|
);
|
|
// Everything is ok. Set m_Outcome = S_OK to Commit the IAS.mdb
|
|
m_Outcome = S_OK;
|
|
break;
|
|
}
|
|
case IAS_WHISTLER_RC1B_VERSION:
|
|
case IAS_WHISTLER_RC2_VERSION:
|
|
{
|
|
MigrateContent.UpdateWhistler(0);
|
|
// Everything is ok. Set m_Outcome = S_OK to Commit the IAS.mdb
|
|
m_Outcome = S_OK;
|
|
break;
|
|
}
|
|
|
|
default:
|
|
{
|
|
// should never be there unless the version was increased and
|
|
// the code in the switch statement above not upgraded
|
|
_com_issue_error(E_FAIL);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
// CLASS CUpgradeNT4
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
///////////////
|
|
// Constructor
|
|
///////////////
|
|
CUpgradeNT4::CUpgradeNT4()
|
|
:m_Outcome(E_FAIL),
|
|
m_GlobalTransaction(CGlobalTransaction::Instance()),
|
|
m_Utils(CUtils::Instance())
|
|
{
|
|
const WCHAR IAS_MDB_NAME[] = L"\\ias.mdb";
|
|
const WCHAR DNARY_MDB_NAME[] = L"\\dnary.mdb";
|
|
|
|
/////////////////////////////////////////////////
|
|
// Check %TMP% and create a directory if needed
|
|
// That's to fix a bug with JET
|
|
/////////////////////////////////////////////////
|
|
IASCreateTmpDirectory();
|
|
|
|
///////////////////////////////
|
|
// Backup the pristine ias.mdb
|
|
///////////////////////////////
|
|
LONG Result = m_Utils.GetIAS2Directory(m_Ias2MdbString);
|
|
if ( Result != ERROR_SUCCESS )
|
|
{
|
|
_com_issue_error(HRESULT_FROM_WIN32(Result));
|
|
}
|
|
|
|
m_DnaryMdbString = m_Ias2MdbString;
|
|
m_DnaryMdbString += DNARY_MDB_NAME;
|
|
|
|
m_Ias2MdbString += IAS_MDB_NAME;
|
|
|
|
Result = m_Utils.GetAuthSrvDirectory(m_AuthSrvMdbString);
|
|
if ( Result != ERROR_SUCCESS )
|
|
{
|
|
_com_issue_error(HRESULT_FROM_WIN32(Result));
|
|
}
|
|
m_AuthSrvMdbString += L"\\_adminui.mdb";
|
|
|
|
m_IASNT4Path = m_AuthSrvMdbString;
|
|
m_IASWhistlerPath = m_Ias2MdbString;
|
|
|
|
// Open the DataSource and session for _adminui.mdb
|
|
m_GlobalTransaction.OpenNT4DataSource(m_IASNT4Path);
|
|
|
|
// create instances of CRemoteRadiusServers and CRealms
|
|
m_GlobalData.InitNT4(m_GlobalTransaction.GetNT4Session());
|
|
|
|
// Open the DataSource and session for ias.mdb
|
|
// and Initialize the GlobalTransaction
|
|
m_GlobalTransaction.OpenStdDataSource(m_IASWhistlerPath);
|
|
|
|
// create instances of CObjects and CProperties
|
|
m_GlobalData.InitStandard(m_GlobalTransaction.GetStdSession());
|
|
|
|
m_GlobalTransaction.OpenDnaryDataSource(m_DnaryMdbString);
|
|
|
|
m_GlobalData.InitDnary(m_GlobalTransaction.GetDnarySession());
|
|
};
|
|
|
|
|
|
//////////////
|
|
// Destructor
|
|
//////////////
|
|
CUpgradeNT4::~CUpgradeNT4()
|
|
{
|
|
//////////
|
|
// Clean
|
|
//////////
|
|
|
|
// Abort or Commit depending on Result
|
|
|
|
m_GlobalData.Clean();
|
|
|
|
if ( SUCCEEDED(m_Outcome) )
|
|
{
|
|
m_GlobalTransaction.Commit(); // ignore return value
|
|
}
|
|
else
|
|
{
|
|
m_GlobalTransaction.Abort(); // ignore return value
|
|
SetLastError(E_FAIL);
|
|
}
|
|
|
|
// close the sessions and then the datasources
|
|
// for ias.mdb iasnew.mdb and _adminui.mdb
|
|
m_GlobalTransaction.MyCloseDataSources();
|
|
|
|
// Restore the pristine ias.mdb if the migration failed
|
|
if ( SUCCEEDED(m_Outcome) )
|
|
{
|
|
// Success, the old files are deleted
|
|
m_Utils.DeleteOldIASFiles();
|
|
}
|
|
};
|
|
|
|
|
|
///////////
|
|
// Execute
|
|
///////////
|
|
void CUpgradeNT4::Execute()
|
|
{
|
|
CMigrateMdb MigrateMdb(m_Utils, m_GlobalData);
|
|
////////////////////////////////////////
|
|
// Migrate the MDB file
|
|
// including the proxy servers
|
|
////////////////////////////////////////
|
|
// Perform the migration into ias.mdb
|
|
MigrateMdb.NewMigrate();
|
|
|
|
// Set m_Outcome = S_OK to allow the Commit on the DB... in the
|
|
// destructor
|
|
m_Outcome = S_OK;
|
|
}
|