|
|
/////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 1996-2001 Microsoft Corporation
//
// Module Name:
// BasePage.cpp
//
// Abstract:
// Implementation of the CBasePropertyPage class.
//
// Author:
// David Potter (davidp) June 28, 1996
//
// Revision History:
// 1. Removed the calls to UpdateData from OnWizardNext and OnApply
// since OnKillActive, called before both these functions does a
// data update anyway.
//
// Notes:
//
/////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "CluAdmX.h"
#include "ExtObj.h"
#include "BasePage.h"
#include "BasePage.inl"
#include "PropList.h"
#include "ExcOper.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__; #endif
/////////////////////////////////////////////////////////////////////////////
// CBasePropertyPage property page
/////////////////////////////////////////////////////////////////////////////
IMPLEMENT_DYNCREATE( CBasePropertyPage, CPropertyPage )
/////////////////////////////////////////////////////////////////////////////
// Message Maps
BEGIN_MESSAGE_MAP( CBasePropertyPage, CPropertyPage ) //{{AFX_MSG_MAP(CBasePropertyPage)
ON_WM_CREATE() ON_WM_DESTROY() ON_WM_HELPINFO() ON_WM_CONTEXTMENU() ON_MESSAGE(WM_COMMANDHELP, OnCommandHelp) //}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::CBasePropertyPage
//
// Routine Description:
// Default constructor.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CBasePropertyPage::CBasePropertyPage( void ) { CommonConstruct();
} //*** CBasePropertyPage::CBasePropertyPage()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::CBasePropertyPage
//
// Routine Description:
// Default constructor.
//
// Arguments:
// pdwHelpMap [IN] Control-to-help ID map.
// pdwWizardHelpMap [IN] Control-to-help ID map if this is a wizard page.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CBasePropertyPage::CBasePropertyPage( IN const DWORD * pdwHelpMap, IN const DWORD * pdwWizardHelpMap ) : m_dlghelp( pdwHelpMap, 0 ) { CommonConstruct(); m_pdwWizardHelpMap = pdwWizardHelpMap;
} //*** CBasePropertyPage::CBasePropertyPage()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::CBasePropertyPage
//
// Routine Description:
// Default constructor.
//
// Arguments:
// idd [IN] Dialog template resource ID.
// pdwHelpMap [IN] Control-to-help ID map.
// pdwWizardHelpMap [IN] Control-to-help ID map if this is a wizard page.
// nIDCaption [IN] Caption string resource ID.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CBasePropertyPage::CBasePropertyPage( IN UINT idd, IN const DWORD * pdwHelpMap, IN const DWORD * pdwWizardHelpMap, IN UINT nIDCaption ) : CPropertyPage( idd, nIDCaption ) , m_dlghelp( pdwHelpMap, idd ) { CommonConstruct(); m_pdwWizardHelpMap = pdwWizardHelpMap;
} //*** CBasePropertyPage::CBasePropertyPage(UINT, UINT)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::CommonConstruct
//
// Routine Description:
// Common construction.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::CommonConstruct( void ) { //{{AFX_DATA_INIT(CBasePropertyPage)
//}}AFX_DATA_INIT
m_peo = NULL; m_hpage = NULL; m_bBackPressed = FALSE; m_bSaved = FALSE;
m_iddPropertyPage = NULL; m_iddWizardPage = NULL; m_idsCaption = NULL;
m_pdwWizardHelpMap = NULL;
m_bDoDetach = FALSE;
} //*** CBasePropertyPage::CommonConstruct()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::HrInit
//
// Routine Description:
// Initialize the page.
//
// Arguments:
// peo [IN OUT] Pointer to the extension object.
//
// Return Value:
// S_OK Page initialized successfully.
// hr Error initializing the page.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CBasePropertyPage::HrInit( IN OUT CExtObject * peo ) { ASSERT( peo != NULL );
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
HRESULT hr = S_OK; CWaitCursor wc;
m_peo = peo;
// Change the help map if this is a wizard page.
if ( Peo()->BWizard() ) { m_dlghelp.SetMap( m_pdwWizardHelpMap ); } // if: on wizard page
// Don't display a help button.
m_psp.dwFlags &= ~PSP_HASHELP;
// Construct the property page.
if ( Peo()->BWizard() ) { ASSERT( IddWizardPage() != NULL ); Construct( IddWizardPage(), IdsCaption() ); m_dlghelp.SetHelpMask( IddWizardPage() ); } // if: adding page to wizard
else { ASSERT( IddPropertyPage() != NULL ); Construct( IddPropertyPage(), IdsCaption() ); m_dlghelp.SetHelpMask( IddPropertyPage() ); } // else: adding page to property sheet
// Read the properties private to this resource and parse them.
{ DWORD sc; CClusPropList cpl;
ASSERT( Peo() != NULL ); ASSERT( Peo()->PrdResData() != NULL ); ASSERT( Peo()->PrdResData()->m_hresource != NULL );
// Read the properties.
sc = cpl.ScGetResourceProperties( Peo()->PrdResData()->m_hresource, CLUSCTL_RESOURCE_GET_PRIVATE_PROPERTIES );
// Parse the properties.
if ( sc == ERROR_SUCCESS ) { // Parse the properties.
try { sc = ScParseProperties( cpl ); } // try
catch ( CMemoryException * pme ) { sc = ERROR_NOT_ENOUGH_MEMORY; pme->Delete(); } // catch: CMemoryException
} // if: properties read successfully
if ( sc != ERROR_SUCCESS ) { CNTException nte( sc, IDS_ERROR_GETTING_PROPERTIES, NULL, NULL, FALSE ); nte.ReportError(); hr = HRESULT_FROM_WIN32( sc ); } // if: error parsing getting or parsing properties
} // Read the properties private to this resource and parse them
return hr;
} //*** CBasePropertyPage::HrInit()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScParseProperties
//
// Routine Description:
// Parse the properties of the resource. This is in a separate function
// from HrInit so that the optimizer can do a better job.
//
// Arguments:
// rcpl [IN] Cluster property list to parse.
//
// Return Value:
// ERROR_SUCCESS Properties were parsed successfully.
// Any error returns from ScParseUnknownProperty().
//
// Exceptions Thrown:
// Any exceptions from CString::operator=().
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScParseProperties( IN CClusPropList & rcpl ) { DWORD sc; DWORD cprop; const CObjectProperty * pprop;
ASSERT( rcpl.PbPropList() != NULL );
sc = rcpl.ScMoveToFirstProperty(); while ( sc == ERROR_SUCCESS ) { //
// Parse known properties.
//
for ( pprop = Pprops(), cprop = Cprops() ; cprop > 0 ; pprop++, cprop-- ) { if ( lstrcmpiW( rcpl.PszCurrentPropertyName(), pprop->m_pwszName ) == 0 ) { ASSERT( rcpl.CpfCurrentValueFormat() == pprop->m_propFormat ); switch ( pprop->m_propFormat ) { case CLUSPROP_FORMAT_SZ: case CLUSPROP_FORMAT_EXPAND_SZ: ASSERT( ( rcpl.CbCurrentValueLength() == (lstrlenW( rcpl.CbhCurrentValue().pStringValue->sz ) + 1) * sizeof( WCHAR ) ) || ( ( rcpl.CbCurrentValueLength() == 0 ) && ( rcpl.CbhCurrentValue().pStringValue->sz[ 0 ] == L'\0' ) ) ); *pprop->m_value.pstr = rcpl.CbhCurrentValue().pStringValue->sz; *pprop->m_valuePrev.pstr = rcpl.CbhCurrentValue().pStringValue->sz; break; case CLUSPROP_FORMAT_DWORD: case CLUSPROP_FORMAT_LONG: ASSERT( rcpl.CbCurrentValueLength() == sizeof( DWORD ) ); *pprop->m_value.pdw = rcpl.CbhCurrentValue().pDwordValue->dw; *pprop->m_valuePrev.pdw = rcpl.CbhCurrentValue().pDwordValue->dw; break; case CLUSPROP_FORMAT_BINARY: case CLUSPROP_FORMAT_MULTI_SZ: *pprop->m_value.ppb = rcpl.CbhCurrentValue().pBinaryValue->rgb; *pprop->m_value.pcb = rcpl.CbhCurrentValue().pBinaryValue->cbLength; *pprop->m_valuePrev.ppb = rcpl.CbhCurrentValue().pBinaryValue->rgb; *pprop->m_valuePrev.pcb = rcpl.CbhCurrentValue().pBinaryValue->cbLength; break; default: ASSERT( 0 ); // don't know how to deal with this type
} // switch: property format
// Exit the loop since we found the parameter.
break; } // if: found a match
} // for: each property that we know about
//
// If the property wasn't known, ask the derived class to parse it.
//
if ( cprop == 0 ) { sc = ScParseUnknownProperty( rcpl.CbhCurrentPropertyName().pName->sz, rcpl.CbhCurrentValue(), rcpl.RPvlPropertyValue().CbDataLeft() ); if ( sc != ERROR_SUCCESS ) { return sc; } // if: error parsing the unknown property
} // if: property not parsed
//
// Advance the buffer pointer past the value in the value list.
//
sc = rcpl.ScMoveToNextProperty(); } // while: more properties to parse
//
// If we reached the end of the properties, fix the return code.
//
if ( sc == ERROR_NO_MORE_ITEMS ) { sc = ERROR_SUCCESS; } // if: ended loop after parsing all properties
return sc;
} //*** CBasePropertyPage::ScParseProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnCreate
//
// Routine Description:
// Handler for the WM_CREATE message.
//
// Arguments:
// lpCreateStruct [IN OUT] Window create structure.
//
// Return Value:
// -1 Error.
// 0 Success.
//
//--
/////////////////////////////////////////////////////////////////////////////
int CBasePropertyPage::OnCreate( LPCREATESTRUCT lpCreateStruct ) { AFX_MANAGE_STATE( AfxGetStaticModuleState() );
// Attach the window to the property page structure.
// This has been done once already in the main application, since the
// main application owns the property sheet. It needs to be done here
// so that the window handle can be found in the DLL's handle map.
if ( FromHandlePermanent( m_hWnd ) == NULL ) // is the window handle already in the handle map
{ HWND hWnd = m_hWnd; m_hWnd = NULL; Attach( hWnd ); m_bDoDetach = TRUE; } // if: is the window handle in the handle map
return CPropertyPage::OnCreate( lpCreateStruct );
} //*** CBasePropertyPage::OnCreate()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnDestroy
//
// Routine Description:
// Handler for the WM_DESTROY message.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::OnDestroy( void ) { AFX_MANAGE_STATE( AfxGetStaticModuleState() );
// Detach the window from the property page structure.
// This will be done again by the main application, since it owns the
// property sheet. It needs to be done here so that the window handle
// can be removed from the DLL's handle map.
if ( m_bDoDetach ) { if ( m_hWnd != NULL ) { HWND hWnd = m_hWnd;
Detach(); m_hWnd = hWnd; } // if: do we have a window handle?
} // if: do we need to balance the attach we did with a detach?
CPropertyPage::OnDestroy();
} //*** CBasePropertyPage::OnDestroy()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::DoDataExchange
//
// Routine Description:
// Do data exchange between the dialog and the class.
//
// Arguments:
// pDX [IN OUT] Data exchange object
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::DoDataExchange( CDataExchange * pDX ) { if ( ! pDX->m_bSaveAndValidate || !BSaved() ) { AFX_MANAGE_STATE( AfxGetStaticModuleState() );
//{{AFX_DATA_MAP(CBasePropertyPage)
// NOTE: the ClassWizard will add DDX and DDV calls here
//}}AFX_DATA_MAP
DDX_Control( pDX, IDC_PP_ICON, m_staticIcon ); DDX_Control( pDX, IDC_PP_TITLE, m_staticTitle );
if ( pDX->m_bSaveAndValidate ) { if ( ! BBackPressed() ) { CWaitCursor wc;
// Validate the data.
if ( ! BSetPrivateProps( TRUE /*bValidateOnly*/ ) ) { pDX->Fail(); } // if: error setting private properties
} // if: Back button not pressed
} // if: saving data from dialog
else { // Set the title.
DDX_Text( pDX, IDC_PP_TITLE, m_strTitle ); } // if: not saving data
} // if: not saving or haven't saved yet
CPropertyPage::DoDataExchange( pDX );
} //*** CBasePropertyPage::DoDataExchange()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnInitDialog
//
// Routine Description:
// Handler for the WM_INITDIALOG message.
//
// Arguments:
// None.
//
// Return Value:
// TRUE We need the focus to be set for us.
// FALSE We already set the focus to the proper control.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::OnInitDialog( void ) { ASSERT( Peo() != NULL );
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
// Set the title string.
m_strTitle = Peo()->RrdResData().m_strName;
// Call the base class method.
CPropertyPage::OnInitDialog();
// Display an icon for the object.
if ( Peo()->Hicon() != NULL ) { m_staticIcon.SetIcon( Peo()->Hicon() ); } // if: icon was specified
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
} //*** CBasePropertyPage::OnInitDialog()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnSetActive
//
// Routine Description:
// Handler for the PSN_SETACTIVE message.
//
// Arguments:
// None.
//
// Return Value:
// TRUE Page successfully initialized.
// FALSE Page not initialized.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::OnSetActive( void ) { HRESULT hr;
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
// Reread the data.
hr = Peo()->HrGetObjectInfo(); if ( hr != NOERROR ) { return FALSE; } // if: error getting object info
// Set the title string.
m_strTitle = Peo()->RrdResData().m_strName;
m_bBackPressed = FALSE; m_bSaved = FALSE; return CPropertyPage::OnSetActive();
} //*** CBasePropertyPage::OnSetActive()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnApply
//
// Routine Description:
// Handler for the PSM_APPLY message.
//
// Arguments:
// None.
//
// Return Value:
// TRUE Page successfully applied.
// FALSE Error applying page.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::OnApply( void ) { ASSERT( ! BWizard() );
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
CWaitCursor wc;
if ( ! BApplyChanges() ) { return FALSE; } // if: error applying changes
return CPropertyPage::OnApply();
} //*** CBasePropertyPage::OnApply()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnWizardBack
//
// Routine Description:
// Handler for the PSN_WIZBACK message.
//
// Arguments:
// None.
//
// Return Value:
// -1 Don't change the page.
// 0 Change the page.
//
//--
/////////////////////////////////////////////////////////////////////////////
LRESULT CBasePropertyPage::OnWizardBack( void ) { LRESULT lResult;
ASSERT( BWizard() );
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
lResult = CPropertyPage::OnWizardBack(); if ( lResult != -1 ) { m_bBackPressed = TRUE; } // if: back processing performed successfully
return lResult;
} //*** CBasePropertyPage::OnWizardBack()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnWizardNext
//
// Routine Description:
// Handler for the PSN_WIZNEXT message.
//
// Arguments:
// None.
//
// Return Value:
// -1 Don't change the page.
// 0 Change the page.
//
//--
/////////////////////////////////////////////////////////////////////////////
LRESULT CBasePropertyPage::OnWizardNext( void ) { ASSERT( BWizard() );
AFX_MANAGE_STATE(AfxGetStaticModuleState());
CWaitCursor _wc;
// Update the data in the class from the page.
// This necessary because, while OnKillActive() will call UpdateData(),
// it is called after this method is called, and we need to be sure that
// data has been saved before we apply them.
if ( ! UpdateData( TRUE /*bSaveAndValidate*/ ) ) { return -1; } // if: error updating data
// Save the data in the sheet.
if ( ! BApplyChanges() ) { return -1; } // if: error applying changes
// Create the object.
return CPropertyPage::OnWizardNext();
} //*** CBasePropertyPage::OnWizardNext()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnWizardFinish
//
// Routine Description:
// Handler for the PSN_WIZFINISH message.
//
// Arguments:
// None.
//
// Return Value:
// FALSE Don't change the page.
// TRUE Change the page.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::OnWizardFinish( void ) { ASSERT( BWizard() );
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
CWaitCursor wc;
// BUG! There should be no need to call UpdateData in this function.
// See BUG: Finish Button Fails Data Transfer from Page to Variables
// MSDN Article ID: Q150349
// Update the data in the class from the page.
if ( ! UpdateData( TRUE /*bSaveAndValidate*/ ) ) { return FALSE; } // if: error updating data
// Save the data in the sheet.
if ( ! BApplyChanges() ) { return FALSE; } // if: error applying changes
return CPropertyPage::OnWizardFinish();
} //*** CBasePropertyPage::OnWizardFinish()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnChangeCtrl
//
// Routine Description:
// Handler for the messages sent when a control is changed. This
// method can be specified in a message map if all that needs to be
// done is enable the Apply button.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::OnChangeCtrl( void ) { AFX_MANAGE_STATE( AfxGetStaticModuleState() );
SetModified( TRUE );
} //*** CBasePropertyPage::OnChangeCtrl()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::EnableNext
//
// Routine Description:
// Enables or disables the NEXT or FINISH button.
//
// Arguments:
// bEnable [IN] TRUE = enable the button, FALSE = disable the button.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::EnableNext( IN BOOL bEnable /*TRUE*/ ) { ASSERT( BWizard() ); ASSERT( PiWizardCallback() );
AFX_MANAGE_STATE( AfxGetStaticModuleState() );
PiWizardCallback()->EnableNext( (LONG *) Hpage(), bEnable );
} //*** CBasePropertyPage::EnableNext()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::BApplyChanges
//
// Routine Description:
// Apply changes made on the page.
//
// Arguments:
// None.
//
// Return Value:
// TRUE Page successfully applied.
// FALSE Error applying page.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::BApplyChanges( void ) { AFX_MANAGE_STATE( AfxGetStaticModuleState() );
BOOL bSuccess; CWaitCursor wc;
// Make sure required dependencies have been set.
if ( ! BSetPrivateProps() ) { bSuccess = FALSE; } // if: all required dependencies are not present
else { // Save data.
bSuccess = BRequiredDependenciesPresent(); } // else: all required dependencies are present
return bSuccess;
} //*** CBasePropertyPage::BApplyChanges()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::BBuildPropList
//
// Routine Description:
// Build the property list.
//
// Arguments:
// rcpl [IN OUT] Cluster property list.
// bNoNewProps [IN] TRUE = exclude properties marked with opfNew.
//
// Return Value:
// TRUE Property list built successfully.
// FALSE Error building property list.
//
// Exceptions Thrown:
// Any exceptions thrown by CClusPropList::ScAddProp().
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::BBuildPropList( IN OUT CClusPropList & rcpl, IN BOOL bNoNewProps // = FALSE
) { BOOL bNewPropsFound = FALSE; DWORD cprop; const CObjectProperty * pprop;
for ( pprop = Pprops(), cprop = Cprops() ; cprop > 0 ; pprop++, cprop-- ) { if ( bNoNewProps && ( pprop->m_fFlags & CObjectProperty::opfNew ) ) { bNewPropsFound = TRUE; continue; } // if: no new props allowed and this is a new property
switch ( pprop->m_propFormat ) { case CLUSPROP_FORMAT_SZ: rcpl.ScAddProp( pprop->m_pwszName, *pprop->m_value.pstr, *pprop->m_valuePrev.pstr ); break; case CLUSPROP_FORMAT_EXPAND_SZ: rcpl.ScAddExpandSzProp( pprop->m_pwszName, *pprop->m_value.pstr, *pprop->m_valuePrev.pstr ); break; case CLUSPROP_FORMAT_DWORD: rcpl.ScAddProp( pprop->m_pwszName, *pprop->m_value.pdw, *pprop->m_valuePrev.pdw ); break; case CLUSPROP_FORMAT_LONG: rcpl.ScAddProp( pprop->m_pwszName, *pprop->m_value.pl, *pprop->m_valuePrev.pl ); break; case CLUSPROP_FORMAT_BINARY: case CLUSPROP_FORMAT_MULTI_SZ: rcpl.ScAddProp( pprop->m_pwszName, *pprop->m_value.ppb, *pprop->m_value.pcb, *pprop->m_valuePrev.ppb, *pprop->m_valuePrev.pcb ); break; default: ASSERT( 0 ); // don't know how to deal with this type
return FALSE; } // switch: property format
} // for: each property
return ( ! bNoNewProps || bNewPropsFound );
} //*** CBasePropertyPage::BBuildPropList()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::BSetPrivateProps
//
// Routine Description:
// Set the private properties for this object.
//
// Arguments:
// bValidateOnly [IN] TRUE = only validate the data.
// bNoNewProps [IN] TRUE = exclude properties marked with opfNew.
//
// Return Value:
// ERROR_SUCCESS The operation was completed successfully.
// !0 Failure.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::BSetPrivateProps( IN BOOL bValidateOnly, // = FALSE
IN BOOL bNoNewProps // = FALSE
) { BOOL bSuccess = TRUE; CClusPropList cpl(BWizard() /*bAlwaysAddProp*/);
ASSERT( Peo() != NULL ); ASSERT( Peo()->PrdResData() ); ASSERT( Peo()->PrdResData()->m_hresource );
// Build the property list.
try { bSuccess = BBuildPropList( cpl, bNoNewProps ); } // try
catch ( CException * pe ) { pe->ReportError(); pe->Delete(); bSuccess = FALSE; } // catch: CException
// Set the data.
if ( bSuccess ) { if ( ( cpl.PbPropList() != NULL ) && ( cpl.CbPropList() > 0 ) ) { DWORD sc; DWORD dwControlCode; DWORD cbProps;
// Determine which control code to use.
if ( bValidateOnly ) { dwControlCode = CLUSCTL_RESOURCE_VALIDATE_PRIVATE_PROPERTIES; } // if: only validating
else { dwControlCode = CLUSCTL_RESOURCE_SET_PRIVATE_PROPERTIES; } // else: not just validating
// Set private properties.
sc = ClusterResourceControl( Peo()->PrdResData()->m_hresource, NULL, // hNode
dwControlCode, cpl.PbPropList(), cpl.CbPropList(), NULL, // lpOutBuffer
0, // nOutBufferSize
&cbProps ); if ( sc != ERROR_SUCCESS ) { if ( sc == ERROR_INVALID_PARAMETER ) { if ( ! bNoNewProps ) { bSuccess = BSetPrivateProps( bValidateOnly, TRUE /*bNoNewProps*/ ); } // if: new props are allowed
else { bSuccess = FALSE; } // else: new props are not allowed
} // if: invalid parameter error occurred
else { bSuccess = FALSE; } // else: some other error occurred
} // if: error setting/validating data
//
// If an error occurred, display an error message.
//
if ( ! bSuccess ) { DisplaySetPropsError( sc, bValidateOnly ? IDS_ERROR_VALIDATING_PROPERTIES : IDS_ERROR_SETTING_PROPERTIES ); if ( sc == ERROR_RESOURCE_PROPERTIES_STORED ) { bSuccess = TRUE; } // if: properties only stored
} // if: error occurred
} // if: there is data to set
} // if: no errors building the property list
// Save data locally.
if ( ! bValidateOnly && bSuccess ) { // Save new values as previous values.
try { DWORD cprop; const CObjectProperty * pprop;
for ( pprop = Pprops(), cprop = Cprops() ; cprop > 0 ; pprop++, cprop-- ) { switch ( pprop->m_propFormat ) { case CLUSPROP_FORMAT_SZ: case CLUSPROP_FORMAT_EXPAND_SZ: ASSERT( pprop->m_value.pstr != NULL ); ASSERT( pprop->m_valuePrev.pstr != NULL ); *pprop->m_valuePrev.pstr = *pprop->m_value.pstr; break; case CLUSPROP_FORMAT_DWORD: case CLUSPROP_FORMAT_LONG: ASSERT( pprop->m_value.pdw != NULL ); ASSERT( pprop->m_valuePrev.pdw != NULL ); *pprop->m_valuePrev.pdw = *pprop->m_value.pdw; break; case CLUSPROP_FORMAT_BINARY: case CLUSPROP_FORMAT_MULTI_SZ: ASSERT( pprop->m_value.ppb != NULL ); ASSERT( *pprop->m_value.ppb != NULL ); ASSERT( pprop->m_value.pcb != NULL ); ASSERT( pprop->m_valuePrev.ppb != NULL ); ASSERT( *pprop->m_valuePrev.ppb != NULL ); ASSERT( pprop->m_valuePrev.pcb != NULL ); delete [] *pprop->m_valuePrev.ppb; *pprop->m_valuePrev.ppb = new BYTE[ *pprop->m_value.pcb ]; if ( *pprop->m_valuePrev.ppb == NULL ) { AfxThrowMemoryException(); } // if: error allocating memory
CopyMemory( *pprop->m_valuePrev.ppb, *pprop->m_value.ppb, *pprop->m_value.pcb ); *pprop->m_valuePrev.pcb = *pprop->m_value.pcb; break; default: ASSERT( 0 ); // don't know how to deal with this type
} // switch: property format
} // for: each property
} // try
catch ( CException * pe ) { pe->ReportError(); pe->Delete(); bSuccess = FALSE; } // catch: CException
} // if: not just validating and successful so far
//
// Indicate we successfully saved the properties.
//
if ( ! bValidateOnly && bSuccess ) { m_bSaved = TRUE; } // if: successfully saved data
return bSuccess;
} //*** CBasePropertyPage::BSetPrivateProps()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::DisplaySetPropsError
//
// Routine Description:
// Display an error caused by setting or validating properties.
//
// Arguments:
// sc [IN] Status to display error on.
// idsOper [IN] Operation message.
//
// Return Value:
// nStatus ERROR_SUCCESS = success, !0 = failure
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::DisplaySetPropsError( IN DWORD sc, IN UINT idsOper ) const { CString strErrorMsg; CString strOperMsg; CString strMsgIdFmt; CString strMsgId; CString strMsg;
strOperMsg.LoadString( IDS_ERROR_SETTING_PROPERTIES ); FormatError( strErrorMsg, sc ); strMsgIdFmt.LoadString( IDS_ERROR_MSG_ID ); strMsgId.Format( strMsgIdFmt, sc, sc ); strMsg.Format( _T("%s\n\n%s%s"), strOperMsg, strErrorMsg, strMsgId ); AfxMessageBox( strMsg );
} //*** CBasePropertyPage::DisplaySetPropsError()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::BRequiredDependenciesPresent
//
// Routine Description:
// Determine if the specified list contains each required resource
// for this type of resource.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
// Exceptions Thrown:
// Any exceptions thrown by CString::LoadString() or CString::operator=().
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::BRequiredDependenciesPresent( void ) { BOOL bFound = TRUE; DWORD sc; CClusPropValueList pvl; HRESOURCE hres; PCLUS_RESOURCE_CLASS_INFO prci = NULL; CString strMissing;
do { // Collect the list of required dependencies.
sc = pvl.ScGetResourceValueList( Peo()->PrdResData()->m_hresource, CLUSCTL_RESOURCE_GET_REQUIRED_DEPENDENCIES ); if ( sc != ERROR_SUCCESS ) { CNTException nte( sc, 0, NULL, NULL, FALSE ); nte.ReportError(); break; } // if: error collecting required dependencies
// Move to the first value.
sc = pvl.ScMoveToFirstValue();
while ( sc == ERROR_SUCCESS ) { switch ( pvl.CptCurrentValueType() ) { case CLUSPROP_TYPE_RESCLASS: prci = reinterpret_cast< PCLUS_RESOURCE_CLASS_INFO >( &pvl.CbhCurrentValue().pResourceClassInfoValue->li ); hres = ResUtilGetResourceDependencyByClass( Hcluster(), Peo()->PrdResData()->m_hresource, prci, FALSE // bRecurse
); if ( hres != NULL ) { CloseClusterResource( hres ); } // if: found the resource
else { if ( ! strMissing.LoadString( IDS_RESCLASS_UNKNOWN + prci->rc ) ) { strMissing.LoadString( IDS_RESCLASS_UNKNOWN ); } // if: unknown resource class
bFound = FALSE; } // else: resource not found
break;
case CLUSPROP_TYPE_NAME: hres = ResUtilGetResourceDependencyByName( Hcluster(), Peo()->PrdResData()->m_hresource, pvl.CbhCurrentValue().pName->sz, FALSE // bRecurse
); if ( hres != NULL ) { CloseClusterResource( hres ); } // if: found the resource
else { GetResTypeDisplayOrTypeName( pvl.CbhCurrentValue().pName->sz, &strMissing ); bFound = FALSE; } // else: resource not found
break;
} // switch: value type
// If a match was not found, changes cannot be applied.
if ( ! bFound ) { CExceptionWithOper ewo( IDS_REQUIRED_DEPENDENCY_NOT_FOUND, NULL, NULL, FALSE );
ewo.SetOperation( IDS_REQUIRED_DEPENDENCY_NOT_FOUND, static_cast< LPCWSTR >( strMissing ) ); ewo.ReportError();
break; } // if: not found
sc = pvl.ScMoveToNextValue(); } // while: more values in the value list
} while( 0 );
return bFound;
} //*** CBasePropertyPage::BRequiredDependenciesPresent()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::GetResTypeDisplayOrTypeName
//
// Routine Description:
// Get the display name for a resource type if possible. If any errors
// occur, just return the type name.
//
// Arguments:
// pszResTypeNameIn
// [IN] Name of resource type.
//
// pstrResTypeDisplayNameInOut
// [IN OUT] CString in which to return the display name.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::GetResTypeDisplayOrTypeName( IN LPCWSTR pszResTypeNameIn, IN OUT CString * pstrResTypeDisplayNameInOut ) { DWORD sc; CClusPropList cpl;
// Get resource type properties.
sc = cpl.ScGetResourceTypeProperties( Hcluster(), pszResTypeNameIn, CLUSCTL_RESOURCE_TYPE_GET_COMMON_PROPERTIES ); if ( sc != ERROR_SUCCESS ) goto Error;
// Find the Name property.
sc = cpl.ScMoveToPropertyByName( CLUSREG_NAME_RESTYPE_NAME ); if ( sc != ERROR_SUCCESS ) goto Error;
// Move to the first value for the property.
sc = cpl.ScMoveToFirstPropertyValue(); if ( sc != ERROR_SUCCESS ) goto Error;
// Make sure the name is a string.
if ( ( cpl.CpfCurrentValueFormat() != CLUSPROP_FORMAT_SZ ) && ( cpl.CpfCurrentValueFormat() != CLUSPROP_FORMAT_EXPAND_SZ ) && ( cpl.CpfCurrentValueFormat() != CLUSPROP_FORMAT_EXPANDED_SZ ) ) goto Error;
// Copy the string into the output CString.
*pstrResTypeDisplayNameInOut = cpl.CbhCurrentValue().pStringValue->sz;
Cleanup: return;
Error: *pstrResTypeDisplayNameInOut = pszResTypeNameIn; goto Cleanup;
} //*** CBasePropertyPage::GetResTypeDisplayOrTypeName()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScReadValue
//
// Routine Description:
// Read a REG_SZ value for this item.
//
// Arguments:
// pszValueName [IN] Name of the value to read.
// rstrValue [OUT] String in which to return the value.
// hkey [IN] Handle to the registry key to read from.
//
// Return Value:
// sc ERROR_SUCCESS = success, !0 = failure
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScReadValue( IN LPCTSTR pszValueName, OUT CString & rstrValue, IN HKEY hkey ) { DWORD sc; LPWSTR pwszValue = NULL; DWORD dwValueLen; DWORD dwValueType;
ASSERT( pszValueName != NULL ); ASSERT( hkey != NULL );
rstrValue.Empty();
try { // Get the size of the value.
dwValueLen = 0; sc = ::ClusterRegQueryValue( hkey, pszValueName, &dwValueType, NULL, &dwValueLen ); if ( ( sc == ERROR_SUCCESS ) || ( sc == ERROR_MORE_DATA ) ) { ASSERT( dwValueType == REG_SZ );
// Allocate enough space for the data.
pwszValue = rstrValue.GetBuffer( dwValueLen / sizeof( WCHAR ) ); if ( pwszValue == NULL ) { AfxThrowMemoryException(); } // if: error getting the buffer
ASSERT( pwszValue != NULL ); dwValueLen += 1 * sizeof( WCHAR ); // Don't forget the final null-terminator.
// Read the value.
sc = ::ClusterRegQueryValue( hkey, pszValueName, &dwValueType, (LPBYTE) pwszValue, &dwValueLen ); if ( sc == ERROR_SUCCESS ) { ASSERT( dwValueType == REG_SZ ); } // if: value read successfully
rstrValue.ReleaseBuffer(); } // if: got the size successfully
} // try
catch ( CMemoryException * pme ) { pme->Delete(); sc = ERROR_NOT_ENOUGH_MEMORY; } // catch: CMemoryException
return sc;
} //*** CBasePropertyPage::ScReadValue(LPCTSTR, CString&)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScReadValue
//
// Routine Description:
// Read a REG_DWORD value for this item.
//
// Arguments:
// pszValueName [IN] Name of the value to read.
// pdwValue [OUT] DWORD in which to return the value.
// hkey [IN] Handle to the registry key to read from.
//
// Return Value:
// _sc ERROR_SUCCESS = success, !0 = failure
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScReadValue( IN LPCTSTR pszValueName, OUT DWORD * pdwValue, IN HKEY hkey ) { DWORD _sc; DWORD _dwValue; DWORD _dwValueLen; DWORD _dwValueType;
ASSERT(pszValueName != NULL); ASSERT(pdwValue != NULL); ASSERT(hkey != NULL);
*pdwValue = 0;
// Read the value.
_dwValueLen = sizeof(_dwValue); _sc = ::ClusterRegQueryValue( hkey, pszValueName, &_dwValueType, (LPBYTE) &_dwValue, &_dwValueLen ); if (_sc == ERROR_SUCCESS) { ASSERT(_dwValueType == REG_DWORD); ASSERT(_dwValueLen == sizeof(_dwValue)); *pdwValue = _dwValue; } // if: value read successfully
return _sc;
} //*** CBasePropertyPage::ScReadValue(LPCTSTR, DWORD*)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScReadValue
//
// Routine Description:
// Read a REG_BINARY value for this item.
//
// Arguments:
// pszValueName [IN] Name of the value to read.
// ppbValue [OUT] Pointer in which to return the data. Caller
// is responsible for deallocating the data.
// hkey [IN] Handle to the registry key to read from.
//
// Return Value:
// _sc ERROR_SUCCESS = success, !0 = failure
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScReadValue( IN LPCTSTR pszValueName, OUT LPBYTE * ppbValue, IN HKEY hkey ) { DWORD _sc; DWORD _dwValueLen; DWORD _dwValueType;
ASSERT(pszValueName != NULL); ASSERT(ppbValue != NULL); ASSERT(hkey != NULL);
*ppbValue = NULL;
// Get the length of the value.
_dwValueLen = 0; _sc = ::ClusterRegQueryValue( hkey, pszValueName, &_dwValueType, NULL, &_dwValueLen ); if (_sc != ERROR_SUCCESS) return _sc;
ASSERT(_dwValueType == REG_BINARY);
// Allocate a buffer,
try { *ppbValue = new BYTE[_dwValueLen]; } // try
catch (CMemoryException *) { _sc = ERROR_NOT_ENOUGH_MEMORY; return _sc; } // catch: CMemoryException
// Read the value.
_sc = ::ClusterRegQueryValue( hkey, pszValueName, &_dwValueType, *ppbValue, &_dwValueLen ); if (_sc != ERROR_SUCCESS) { delete [] *ppbValue; *ppbValue = NULL; } // if: value read successfully
return _sc;
} //*** CBasePropertyPage::ScReadValue(LPCTSTR, LPBYTE)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScWriteValue
//
// Routine Description:
// Write a REG_SZ value for this item if it hasn't changed.
//
// Arguments:
// pszValueName [IN] Name of the value to write.
// rstrValue [IN] Value data.
// rstrPrevValue [IN OUT] Previous value.
// hkey [IN] Handle to the registry key to write to.
//
// Return Value:
// _sc
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScWriteValue( IN LPCTSTR pszValueName, IN const CString & rstrValue, IN OUT CString & rstrPrevValue, IN HKEY hkey ) { DWORD _sc;
ASSERT(pszValueName != NULL); ASSERT(hkey != NULL);
// Write the value if it hasn't changed.
if (rstrValue != rstrPrevValue) { _sc = ::ClusterRegSetValue( hkey, pszValueName, REG_SZ, (CONST BYTE *) (LPCTSTR) rstrValue, (rstrValue.GetLength() + 1) * sizeof(TCHAR) ); if (_sc == ERROR_SUCCESS) rstrPrevValue = rstrValue; } // if: value changed
else _sc = ERROR_SUCCESS; return _sc;
} //*** CBasePropertyPage::ScWriteValue(LPCTSTR, CString&)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScWriteValue
//
// Routine Description:
// Write a REG_DWORD value for this item if it hasn't changed.
//
// Arguments:
// pszValueName [IN] Name of the value to write.
// dwValue [IN] Value data.
// pdwPrevValue [IN OUT] Previous value.
// hkey [IN] Handle to the registry key to write to.
//
// Return Value:
// _sc
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScWriteValue( IN LPCTSTR pszValueName, IN DWORD dwValue, IN OUT DWORD * pdwPrevValue, IN HKEY hkey ) { DWORD _sc;
ASSERT(pszValueName != NULL); ASSERT(pdwPrevValue != NULL); ASSERT(hkey != NULL);
// Write the value if it hasn't changed.
if (dwValue != *pdwPrevValue) { _sc = ::ClusterRegSetValue( hkey, pszValueName, REG_DWORD, (CONST BYTE *) &dwValue, sizeof(dwValue) ); if (_sc == ERROR_SUCCESS) *pdwPrevValue = dwValue; } // if: value changed
else _sc = ERROR_SUCCESS; return _sc;
} //*** CBasePropertyPage::ScWriteValue(LPCTSTR, DWORD)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ScWriteValue
//
// Routine Description:
// Write a REG_BINARY value for this item if it hasn't changed.
//
// Arguments:
// pszValueName [IN] Name of the value to write.
// pbValue [IN] Value data.
// cbValue [IN] Size of value data.
// ppbPrevValue [IN OUT] Previous value.
// cbPrevValue [IN] Size of the previous data.
// hkey [IN] Handle to the registry key to write to.
//
// Return Value:
// _sc
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CBasePropertyPage::ScWriteValue( IN LPCTSTR pszValueName, IN const LPBYTE pbValue, IN DWORD cbValue, IN OUT LPBYTE * ppbPrevValue, IN DWORD cbPrevValue, IN HKEY hkey ) { DWORD _sc; LPBYTE _pbPrevValue = NULL;
ASSERT(pszValueName != NULL); ASSERT(pbValue != NULL); ASSERT(ppbPrevValue != NULL); ASSERT(cbValue > 0); ASSERT(hkey != NULL);
// See if the data has changed.
if (cbValue == cbPrevValue) { if (memcmp(pbValue, *ppbPrevValue, cbValue) == 0) return ERROR_SUCCESS; } // if: lengths are the same
// Allocate a new buffer for the previous data pointer.
try { _pbPrevValue = new BYTE[cbValue]; } catch (CMemoryException *) { return ERROR_NOT_ENOUGH_MEMORY; } // catch: CMemoryException
::CopyMemory(_pbPrevValue, pbValue, cbValue);
// Write the value if it hasn't changed.
_sc = ::ClusterRegSetValue( hkey, pszValueName, REG_BINARY, pbValue, cbValue ); if (_sc == ERROR_SUCCESS) { delete [] *ppbPrevValue; *ppbPrevValue = _pbPrevValue; } // if: set was successful
else delete [] _pbPrevValue;
return _sc;
} //*** CBasePropertyPage::ScWriteValue(LPCTSTR, const LPBYTE)
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnContextMenu
//
// Routine Description:
// Handler for the WM_CONTEXTMENU message.
//
// Arguments:
// pWnd Window in which user clicked the right mouse button.
// point Position of the cursor, in screen coordinates.
//
// Return Value:
// TRUE Help processed.
// FALSE Help not processed.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::OnContextMenu(CWnd * pWnd, CPoint point) { AFX_MANAGE_STATE(AfxGetStaticModuleState());
m_dlghelp.OnContextMenu(pWnd, point);
} //*** CBasePropertyPage::OnContextMenu()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnHelpInfo
//
// Routine Description:
// Handler for the WM_HELPINFO message.
//
// Arguments:
// pHelpInfo Structure containing info about displaying help.
//
// Return Value:
// TRUE Help processed.
// FALSE Help not processed.
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::OnHelpInfo(HELPINFO * pHelpInfo) { BOOL _bProcessed;
AFX_MANAGE_STATE(AfxGetStaticModuleState());
_bProcessed = m_dlghelp.OnHelpInfo(pHelpInfo); if (!_bProcessed) _bProcessed = CPropertyPage::OnHelpInfo(pHelpInfo); return _bProcessed;
} //*** CBasePropertyPage::OnHelpInfo()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::OnCommandHelp
//
// Routine Description:
// Handler for the WM_COMMANDHELP message.
//
// Arguments:
// wParam [IN] WPARAM.
// lParam [IN] LPARAM.
//
// Return Value:
// TRUE Help processed.
// FALSE Help not processed.
//
//--
/////////////////////////////////////////////////////////////////////////////
LRESULT CBasePropertyPage::OnCommandHelp(WPARAM wParam, LPARAM lParam) { LRESULT _bProcessed;
AFX_MANAGE_STATE(AfxGetStaticModuleState());
_bProcessed = m_dlghelp.OnCommandHelp(wParam, lParam); if (!_bProcessed) _bProcessed = CPropertyPage::OnCommandHelp(wParam, lParam);
return _bProcessed;
} //*** CBasePropertyPage::OnCommandHelp()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::ConstructDefaultDirectory
//
// Routine Description:
// Get the name of the first partition from the first storage-class
// resource on which this resource is dependent.
//
// Arguments:
// rstrDir [OUT] Directory string.
// idsFormat [IN] Resource ID for the format string.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CBasePropertyPage::ConstructDefaultDirectory( OUT CString & rstrDir, IN IDS idsFormat ) { AFX_MANAGE_STATE(AfxGetStaticModuleState());
HRESOURCE _hres = NULL; DWORD _sc = ERROR_SUCCESS; DWORD _cbDiskInfo = sizeof(CLUSPROP_DWORD) + sizeof(CLUSPROP_SCSI_ADDRESS) + sizeof(CLUSPROP_DISK_NUMBER) + sizeof(CLUSPROP_PARTITION_INFO) + sizeof(CLUSPROP_SYNTAX); PBYTE _pbDiskInfo = NULL; CLUSPROP_BUFFER_HELPER _cbh;
// Get the first partition for the resource..
try { // Get the storage-class resource on which we are dependent.
_hres = GetDependentStorageResource(); if (_hres == NULL) return;
// Get disk info.
_pbDiskInfo = new BYTE[_cbDiskInfo]; _sc = ClusterResourceControl( _hres, NULL, CLUSCTL_RESOURCE_STORAGE_GET_DISK_INFO, NULL, 0, _pbDiskInfo, _cbDiskInfo, &_cbDiskInfo ); if (_sc == ERROR_MORE_DATA) { delete [] _pbDiskInfo; _pbDiskInfo = new BYTE[_cbDiskInfo]; _sc = ClusterResourceControl( _hres, NULL, CLUSCTL_RESOURCE_STORAGE_GET_DISK_INFO, NULL, 0, _pbDiskInfo, _cbDiskInfo, &_cbDiskInfo ); } // if: buffer too small
if (_sc == ERROR_SUCCESS) { // Find the first partition.
_cbh.pb = _pbDiskInfo; while (_cbh.pSyntax->dw != CLUSPROP_SYNTAX_ENDMARK) { if (_cbh.pSyntax->dw == CLUSPROP_SYNTAX_PARTITION_INFO) { rstrDir.FormatMessage( idsFormat, _cbh.pPartitionInfoValue->szDeviceName ); break; } // if: found a partition
_cbh.pb += sizeof(*_cbh.pValue) + ALIGN_CLUSPROP(_cbh.pValue->cbLength); } // while: not at end of list
} // if: no error getting disk info
else { CNTException nte( _sc, IDS_ERROR_CONSTRUCTING_DEF_DIR ); nte.ReportError(); } // else: error getting disk info
} // try
catch (CMemoryException * _pme) { _pme->Delete(); } // catch: CMemoryException
CloseClusterResource(_hres); delete [] _pbDiskInfo;
} //*** CBasePropertyPage::ConstructDefaultDirectory()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::GetDependentStorageResource
//
// Routine Description:
// Construct a default spool directory based on the drive on which
// this resource is dependent and a default value for the directory.
//
// Arguments:
// phres [OUT] Handle to dependent resource.
//
// Return Value:
// HRESOURCE for the open dependent resource, or NULL if error.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESOURCE CBasePropertyPage::GetDependentStorageResource(void) { DWORD _sc = ERROR_SUCCESS; HRESENUM _hresenum; HRESOURCE _hres = NULL; DWORD _ires; DWORD _dwType; DWORD _cchName; DWORD _cchNameSize; LPWSTR _pszName = NULL; CLUS_RESOURCE_CLASS_INFO _classinfo; DWORD _cbClassInfo;
// Open the dependency enumerator.
_hresenum = ClusterResourceOpenEnum( Peo()->PrdResData()->m_hresource, CLUSTER_RESOURCE_ENUM_DEPENDS ); if (_hresenum == NULL) return NULL;
// Allocate a default size name buffer.
_cchNameSize = 512; _pszName = new WCHAR[_cchNameSize];
for (_ires = 0 ; ; _ires++) { // Get the name of the next resource.
_cchName = _cchNameSize; _sc = ClusterResourceEnum( _hresenum, _ires, &_dwType, _pszName, &_cchName ); if (_sc == ERROR_MORE_DATA) { delete [] _pszName; _cchNameSize = _cchName; _pszName = new WCHAR[_cchNameSize]; _sc = ClusterResourceEnum( _hresenum, _ires, &_dwType, _pszName, &_cchName ); } // if: name buffer too small
if (_sc != ERROR_SUCCESS) break;
// Open the resource.
_hres = OpenClusterResource(Hcluster(), _pszName); if (_hres == NULL) { _sc = GetLastError(); break; } // if: error opening the resource
// Get the class of the resource.
_sc = ClusterResourceControl( _hres, NULL, CLUSCTL_RESOURCE_GET_CLASS_INFO, NULL, 0, &_classinfo, sizeof(_classinfo), &_cbClassInfo ); if (_sc != ERROR_SUCCESS) { CNTException nte( _sc, IDS_ERROR_GET_CLASS_INFO, _pszName, NULL, FALSE /*bAutoDelete*/ ); nte.ReportError(); continue; }
// If this is a storage-class resource, we're done.
if (_classinfo.rc == CLUS_RESCLASS_STORAGE) break;
// Not storage-class resource.
CloseClusterResource(_hres); _hres = NULL; } // for each resource on which we are dependent
// Handle errors.
if ((_sc != ERROR_SUCCESS) && (_hres != NULL)) { CloseClusterResource(_hres); _hres = NULL; } // if: error getting resource
// Cleanup.
ClusterResourceCloseEnum(_hresenum); delete [] _pszName;
return _hres;
} //*** CBasePropertyPage::GetDependentStorageResource()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CBasePropertyPage::BGetClusterNetworkNameNode
//
// Routine Description:
// Get the node hosting the Network Name resource.
//
// Arguments:
// rstrNode [OUT] - receives the node name
//
// Return Value:
// BOOL -- TRUE for success, FALSE for error
//
//--
/////////////////////////////////////////////////////////////////////////////
BOOL CBasePropertyPage::BGetClusterNetworkNameNode( OUT CString & rstrNode ) { BOOL _bSuccess = TRUE; DWORD _sc; DWORD _dwFlag; DWORD _cbData; DWORD _ires; DWORD _dwType; DWORD _cchName = 0; DWORD _cchNameCurrent; LPWSTR _pszName = NULL; WCHAR _szResType[sizeof(CLUS_RESTYPE_NAME_NETNAME) / sizeof(WCHAR)]; WCHAR _szNode[MAX_COMPUTERNAME_LENGTH+1]; HCLUSENUM _hclusenum = NULL; HRESOURCE _hresource = NULL; CLUSTER_RESOURCE_STATE _crs; CWaitCursor _wc;
try { // Open a cluster enumerator.
_hclusenum = ClusterOpenEnum( Hcluster(), CLUSTER_ENUM_RESOURCE ); if (_hclusenum == NULL) { ThrowStaticException( GetLastError() ); }
// Allocate an initial buffer.
_cchName = 256; _pszName = new WCHAR[_cchName];
// Loop through each resource.
for ( _ires = 0 ; ; _ires++ ) { // Get the next resource.
_cchNameCurrent = _cchName; _sc = ClusterEnum( _hclusenum, _ires, &_dwType, _pszName, &_cchNameCurrent ); if ( _sc == ERROR_MORE_DATA ) { delete [] _pszName; _cchName = ++_cchNameCurrent; _pszName = new WCHAR[_cchName]; _sc = ClusterEnum(_hclusenum, _ires, &_dwType, _pszName, &_cchNameCurrent); } // if: buffer too small
if (_sc == ERROR_NO_MORE_ITEMS) break; if (_sc != ERROR_SUCCESS) ThrowStaticException(_sc);
// Open the resource.
_hresource = OpenClusterResource(Hcluster(), _pszName); if (_hresource == NULL) ThrowStaticException(GetLastError());
// Get its flags.
_sc = ClusterResourceControl( _hresource, NULL, CLUSCTL_RESOURCE_GET_FLAGS, NULL, 0, &_dwFlag, sizeof(DWORD), &_cbData ); if (_sc != ERROR_SUCCESS) { CNTException nte( _sc, IDS_ERROR_GET_RESOURCE_FLAGS, _pszName, NULL, FALSE /*bAutoDelete*/ ); nte.ReportError(); continue; }
// If this isn't a core resource, skip it.
if ((_dwFlag & CLUS_FLAG_CORE) == 0) continue;
// Get its resource type name. If the buffer is too small,
// it isn't a Network Name resource so skip it.
_sc = ClusterResourceControl( _hresource, NULL, CLUSCTL_RESOURCE_GET_RESOURCE_TYPE, NULL, 0, _szResType, sizeof(_szResType), &_cbData ); if (_sc == ERROR_MORE_DATA) continue; if (_sc != ERROR_SUCCESS) ThrowStaticException(_sc);
// If this is a Network Name resource, get which node it is online on.
if (lstrcmpiW(_szResType, CLUS_RESTYPE_NAME_NETNAME) == 0) { // Get the state of the resource.
_crs = GetClusterResourceState( _hresource, _szNode, &_cchName, NULL, NULL ); if (_crs == ClusterResourceStateUnknown) ThrowStaticException(GetLastError());
// Save the node name in the return argument.
rstrNode = _szNode;
break; } // if: Network Name resource
CloseClusterResource( _hresource ); _hresource = NULL; } // for: each resource
if (rstrNode[0] == _T('\0')) ThrowStaticException(ERROR_FILE_NOT_FOUND, (IDS) 0); } // try
catch (CException * _pe) { _pe->ReportError(); _pe->Delete(); _bSuccess = FALSE; } // catch: CException
delete [] _pszName;
if ( _hresource != NULL ) { CloseClusterResource( _hresource ); } // if: resource is open
if ( _hclusenum != NULL ) { ClusterCloseEnum( _hclusenum ); } // if: enumerator is open
return _bSuccess;
} //*** CBasePropertyPage::BGetClusterNetworkNameNode()
|