1168 lines
29 KiB
C++
1168 lines
29 KiB
C++
|
// VirtualRoot.cpp : Implementation of CNntpVirtualRoot & CNntpVirtualRoots.
|
||
|
|
||
|
#include "stdafx.h"
|
||
|
#include "nntpcmn.h"
|
||
|
#include "cmultisz.h"
|
||
|
#include "vroots.h"
|
||
|
#include "oleutil.h"
|
||
|
#include "metautil.h"
|
||
|
#include "metakey.h"
|
||
|
|
||
|
#define THIS_FILE_HELP_CONTEXT 0
|
||
|
#define THIS_FILE_PROG_ID _T("Nntpadm.VirtualServer.1")
|
||
|
#define THIS_FILE_IID IID_INntpVirtualServer
|
||
|
|
||
|
#define DEFAULT_ACCESS_PERM ( 0 )
|
||
|
|
||
|
CVRoot::CVRoot () :
|
||
|
m_dwWin32Error ( 0 ),
|
||
|
m_bvAccess ( 0 ),
|
||
|
m_bvSslAccess ( 0 ),
|
||
|
m_fLogAccess ( TRUE ),
|
||
|
m_fIndexContent ( TRUE ),
|
||
|
m_dwUseAccount ( 0 ),
|
||
|
m_fDoExpire ( FALSE ),
|
||
|
m_fOwnModerator ( FALSE )
|
||
|
{
|
||
|
}
|
||
|
|
||
|
HRESULT CVRoot::SetProperties (
|
||
|
BSTR strNewsgroupSubtree,
|
||
|
BSTR strDirectory,
|
||
|
DWORD dwWin32Error,
|
||
|
DWORD bvAccess,
|
||
|
DWORD bvSslAccess,
|
||
|
BOOL fLogAccess,
|
||
|
BOOL fIndexContent,
|
||
|
BSTR strUNCUsername,
|
||
|
BSTR strUNCPassword,
|
||
|
BSTR strDriverProgId,
|
||
|
BSTR strGroupPropFile,
|
||
|
DWORD dwUseAccount,
|
||
|
BOOL fDoExpire,
|
||
|
BOOL fOwnModerator,
|
||
|
BSTR strMdbGuid
|
||
|
)
|
||
|
{
|
||
|
_ASSERT ( IS_VALID_STRING ( strNewsgroupSubtree ) );
|
||
|
_ASSERT ( IS_VALID_STRING ( strDirectory ) );
|
||
|
_ASSERT ( IS_VALID_STRING ( strUNCUsername ) );
|
||
|
_ASSERT ( IS_VALID_STRING ( strUNCPassword ) );
|
||
|
_ASSERT ( IS_VALID_STRING ( strDriverProgId ) );
|
||
|
_ASSERT ( IS_VALID_STRING ( strGroupPropFile ) );
|
||
|
_ASSERT ( IS_VALID_STRING ( strMdbGuid ) );
|
||
|
|
||
|
m_strNewsgroupSubtree = strNewsgroupSubtree;
|
||
|
m_strDirectory = strDirectory;
|
||
|
m_dwWin32Error = dwWin32Error;
|
||
|
m_bvAccess = bvAccess;
|
||
|
m_bvSslAccess = bvSslAccess;
|
||
|
m_fLogAccess = fLogAccess;
|
||
|
m_fIndexContent = fIndexContent;
|
||
|
m_strUNCUsername = strUNCUsername;
|
||
|
m_strUNCPassword = strUNCPassword;
|
||
|
m_strDriverProgId = strDriverProgId;
|
||
|
m_strGroupPropFile = strGroupPropFile;
|
||
|
m_dwUseAccount = dwUseAccount;
|
||
|
m_fDoExpire = fDoExpire;
|
||
|
m_fOwnModerator = fOwnModerator,
|
||
|
m_strMdbGuid = strMdbGuid;
|
||
|
|
||
|
if ( !m_strNewsgroupSubtree || !m_strDirectory || !m_strUNCUsername || !m_strUNCPassword ) {
|
||
|
return E_OUTOFMEMORY;
|
||
|
}
|
||
|
|
||
|
return NOERROR;
|
||
|
}
|
||
|
|
||
|
HRESULT CVRoot::SetProperties ( INntpVirtualRoot * pVirtualRoot )
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
|
||
|
BOOL fAllowPosting;
|
||
|
BOOL fRestrictVisibility;
|
||
|
BOOL fRequireSsl;
|
||
|
BOOL fRequire128BitSsl;
|
||
|
|
||
|
m_strDirectory.Empty ();
|
||
|
m_strUNCUsername.Empty ();
|
||
|
m_strUNCPassword.Empty ();
|
||
|
m_strDriverProgId.Empty();
|
||
|
m_strGroupPropFile.Empty();
|
||
|
m_strMdbGuid.Empty();
|
||
|
|
||
|
hr = pVirtualRoot->get_NewsgroupSubtree ( &m_strNewsgroupSubtree );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = pVirtualRoot->get_Directory ( &m_strDirectory );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = pVirtualRoot->get_Win32Error ( (long *) &m_dwWin32Error );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = pVirtualRoot->get_UNCUsername ( &m_strUNCUsername );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = pVirtualRoot->get_UNCPassword ( &m_strUNCPassword );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = pVirtualRoot->get_DriverProgId ( &m_strDriverProgId );
|
||
|
BAIL_ON_FAILURE( hr );
|
||
|
|
||
|
hr = pVirtualRoot->get_GroupPropFile ( &m_strGroupPropFile );
|
||
|
BAIL_ON_FAILURE( hr );
|
||
|
|
||
|
hr = pVirtualRoot->get_MdbGuid ( &m_strMdbGuid );
|
||
|
BAIL_ON_FAILURE( hr );
|
||
|
|
||
|
hr = pVirtualRoot->get_UseAccount( &m_dwUseAccount );
|
||
|
BAIL_ON_FAILURE( hr );
|
||
|
|
||
|
hr = pVirtualRoot->get_OwnExpire( &m_fDoExpire );
|
||
|
BAIL_ON_FAILURE( hr );
|
||
|
|
||
|
hr = pVirtualRoot->get_OwnModerator( &m_fOwnModerator );
|
||
|
BAIL_ON_FAILURE( hr );
|
||
|
|
||
|
hr = pVirtualRoot->get_LogAccess ( &m_fLogAccess );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
hr = pVirtualRoot->get_IndexContent ( &m_fIndexContent );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
|
||
|
hr = pVirtualRoot->get_AllowPosting ( &fAllowPosting );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
hr = pVirtualRoot->get_RestrictGroupVisibility ( &fRestrictVisibility );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
hr = pVirtualRoot->get_RequireSsl ( &fRequireSsl );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
hr = pVirtualRoot->get_Require128BitSsl ( &fRequire128BitSsl );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
SetBitFlag ( &m_bvAccess, MD_ACCESS_ALLOW_POSTING, fAllowPosting );
|
||
|
SetBitFlag ( &m_bvAccess, MD_ACCESS_RESTRICT_VISIBILITY, fRestrictVisibility );
|
||
|
|
||
|
SetBitFlag ( &m_bvSslAccess, MD_ACCESS_SSL, fRequireSsl );
|
||
|
SetBitFlag ( &m_bvSslAccess, MD_ACCESS_SSL128, fRequire128BitSsl );
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT CVRoot::GetFromMetabase ( CMetabaseKey * pMB,
|
||
|
LPCWSTR wszName,
|
||
|
DWORD dwInstanceId,
|
||
|
LPWSTR wszServerName )
|
||
|
{
|
||
|
HRESULT hr = NOERROR;
|
||
|
DWORD dwDontLog = TRUE;
|
||
|
|
||
|
m_strDirectory.Empty();
|
||
|
m_strUNCUsername.Empty();
|
||
|
m_strUNCPassword.Empty();
|
||
|
|
||
|
StdGetMetabaseProp ( pMB, MD_ACCESS_PERM, 0, &m_bvAccess, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_SSL_ACCESS_PERM, 0, &m_bvSslAccess, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_DONT_LOG, FALSE, &dwDontLog, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_IS_CONTENT_INDEXED, FALSE, &m_fIndexContent, wszName, IIS_MD_UT_FILE);
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_PATH, _T(""), &m_strDirectory, wszName );
|
||
|
//StdGetMetabaseProp ( pMB, MD_WIN32_ERROR, 0, &m_dwWin32Error, wszName, IIS_MD_UT_FILE, METADATA_NO_ATTRIBUTES );
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_USERNAME, _T(""), &m_strUNCUsername, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_PASSWORD, _T(""), &m_strUNCPassword, wszName, IIS_MD_UT_SERVER, METADATA_SECURE );
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_USE_ACCOUNT, 0, &m_dwUseAccount, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_DO_EXPIRE, FALSE, &m_fDoExpire, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_OWN_MODERATOR, FALSE, &m_fOwnModerator, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_VR_DRIVER_PROGID, _T("NNTP.FSPrepare"), &m_strDriverProgId, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_FS_PROPERTY_PATH, m_strDirectory, &m_strGroupPropFile, wszName );
|
||
|
StdGetMetabaseProp ( pMB, MD_EX_MDB_GUID, _T(""), &m_strMdbGuid, wszName );
|
||
|
|
||
|
//
|
||
|
// Get the win32 error code from RPC
|
||
|
//
|
||
|
DWORD dw = NntpGetVRootWin32Error( (LPWSTR)wszServerName, dwInstanceId, (LPWSTR)wszName, (LPDWORD)&m_dwWin32Error );
|
||
|
switch (dw) {
|
||
|
case ERROR_SERVICE_NOT_ACTIVE:
|
||
|
m_dwWin32Error = dw;
|
||
|
break;
|
||
|
|
||
|
case NOERROR:
|
||
|
break;
|
||
|
|
||
|
default:
|
||
|
hr = HRESULT_FROM_WIN32( dw );
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
m_strNewsgroupSubtree = wszName;
|
||
|
m_fLogAccess = !dwDontLog;
|
||
|
|
||
|
if ( !m_strNewsgroupSubtree || !m_strDirectory || !m_strUNCUsername || !m_strUNCPassword ) {
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT CVRoot::SendToMetabase ( CMetabaseKey * pMB, LPCWSTR wszName )
|
||
|
{
|
||
|
HRESULT hr = NOERROR;
|
||
|
|
||
|
hr = pMB->SetString ( wszName, MD_KEY_TYPE, _T("IIsNntpVirtualDir"), METADATA_NO_ATTRIBUTES, IIS_MD_UT_SERVER );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = pMB->SetString ( wszName, MD_VR_PATH, m_strDirectory, METADATA_INHERIT, IIS_MD_UT_FILE );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
StdPutMetabaseProp ( pMB, MD_ACCESS_PERM, m_bvAccess, wszName, IIS_MD_UT_FILE );
|
||
|
StdPutMetabaseProp ( pMB, MD_SSL_ACCESS_PERM, m_bvSslAccess, wszName, IIS_MD_UT_FILE );
|
||
|
StdPutMetabaseProp ( pMB, MD_DONT_LOG, !m_fLogAccess, wszName, IIS_MD_UT_FILE );
|
||
|
StdPutMetabaseProp ( pMB, MD_IS_CONTENT_INDEXED, m_fIndexContent, wszName, IIS_MD_UT_FILE );
|
||
|
StdPutMetabaseProp ( pMB, MD_WIN32_ERROR, m_dwWin32Error, wszName, IIS_MD_UT_SERVER, METADATA_VOLATILE );
|
||
|
StdPutMetabaseProp ( pMB, MD_VR_USERNAME, m_strUNCUsername, wszName, IIS_MD_UT_FILE );
|
||
|
StdPutMetabaseProp ( pMB, MD_VR_PASSWORD, m_strUNCPassword, wszName, IIS_MD_UT_FILE, METADATA_SECURE | METADATA_INHERIT );
|
||
|
StdPutMetabaseProp ( pMB, MD_VR_USE_ACCOUNT, m_dwUseAccount, wszName, IIS_MD_UT_SERVER );
|
||
|
StdPutMetabaseProp ( pMB, MD_VR_DO_EXPIRE, m_fDoExpire, wszName, IIS_MD_UT_SERVER );
|
||
|
StdPutMetabaseProp ( pMB, MD_VR_OWN_MODERATOR, m_fOwnModerator, wszName, IIS_MD_UT_SERVER );
|
||
|
StdPutMetabaseProp ( pMB, MD_VR_DRIVER_PROGID, m_strDriverProgId, wszName, IIS_MD_UT_SERVER );
|
||
|
StdPutMetabaseProp ( pMB, MD_FS_PROPERTY_PATH, m_strGroupPropFile, wszName, IIS_MD_UT_SERVER );
|
||
|
StdPutMetabaseProp ( pMB, MD_EX_MDB_GUID, m_strMdbGuid, wszName, IIS_MD_UT_SERVER );
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::InterfaceSupportsErrorInfo(REFIID riid)
|
||
|
{
|
||
|
static const IID* arr[] =
|
||
|
{
|
||
|
&IID_INntpVirtualRoot,
|
||
|
};
|
||
|
|
||
|
for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++)
|
||
|
{
|
||
|
if (InlineIsEqualGUID(*arr[i],riid))
|
||
|
return S_OK;
|
||
|
}
|
||
|
return S_FALSE;
|
||
|
}
|
||
|
|
||
|
CNntpVirtualRoot::CNntpVirtualRoot ()
|
||
|
// CComBSTR's are initialized to NULL by default.
|
||
|
{
|
||
|
}
|
||
|
|
||
|
CNntpVirtualRoot::~CNntpVirtualRoot ()
|
||
|
{
|
||
|
// All CComBSTR's are freed automatically.
|
||
|
}
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////
|
||
|
// Properties:
|
||
|
//////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_NewsgroupSubtree ( BSTR * pstrNewsgroupSubtree )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_strNewsgroupSubtree, pstrNewsgroupSubtree );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_NewsgroupSubtree ( BSTR strNewsgroupSubtree )
|
||
|
{
|
||
|
if ( wcslen( strNewsgroupSubtree ) > METADATA_MAX_NAME_LEN - 20 ) return CO_E_PATHTOOLONG;
|
||
|
return StdPropertyPut ( &m_vroot.m_strNewsgroupSubtree, strNewsgroupSubtree );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_Directory ( BSTR * pstrDirectory )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_strDirectory, pstrDirectory );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_Directory ( BSTR strDirectory )
|
||
|
{
|
||
|
return StdPropertyPut ( &m_vroot.m_strDirectory, strDirectory );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_Win32Error ( long * plWin32Error )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_dwWin32Error, plWin32Error );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_LogAccess ( BOOL * pfLogAccess )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_fLogAccess, pfLogAccess );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_LogAccess ( BOOL fLogAccess )
|
||
|
{
|
||
|
return StdPropertyPut ( &m_vroot.m_fLogAccess, fLogAccess );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_IndexContent ( BOOL * pfIndexContent )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_fIndexContent, pfIndexContent );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_IndexContent ( BOOL fIndexContent )
|
||
|
{
|
||
|
return StdPropertyPut ( &m_vroot.m_fIndexContent, fIndexContent );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_AllowPosting ( BOOL * pfAllowPosting )
|
||
|
{
|
||
|
return StdPropertyGetBit ( m_vroot.m_bvAccess, MD_ACCESS_ALLOW_POSTING, pfAllowPosting );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_AllowPosting ( BOOL fAllowPosting )
|
||
|
{
|
||
|
return StdPropertyPutBit ( &m_vroot.m_bvAccess, MD_ACCESS_ALLOW_POSTING, fAllowPosting );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_RestrictGroupVisibility ( BOOL * pfRestrictGroupVisibility )
|
||
|
{
|
||
|
return StdPropertyGetBit ( m_vroot.m_bvAccess, MD_ACCESS_RESTRICT_VISIBILITY, pfRestrictGroupVisibility );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_RestrictGroupVisibility ( BOOL fRestrictGroupVisibility )
|
||
|
{
|
||
|
return StdPropertyPutBit ( &m_vroot.m_bvAccess, MD_ACCESS_RESTRICT_VISIBILITY, fRestrictGroupVisibility );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP SSLNegotiateCert ( BOOL * pfSSLNegotiateCert );
|
||
|
STDMETHODIMP SSLNegotiateCert ( BOOL fSSLNegotiateCert );
|
||
|
|
||
|
STDMETHODIMP SSLRequireCert ( BOOL * pfSSLRequireCert );
|
||
|
STDMETHODIMP SSLRequireCert ( BOOL fSSLRequireCert );
|
||
|
|
||
|
STDMETHODIMP SSLMapCert ( BOOL * pfSSLMapCert );
|
||
|
STDMETHODIMP SSLMapCert ( BOOL fSSLMapCert );
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_RequireSsl ( BOOL * pfRequireSsl )
|
||
|
{
|
||
|
return StdPropertyGetBit ( m_vroot.m_bvSslAccess, MD_ACCESS_SSL, pfRequireSsl );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_RequireSsl ( BOOL fRequireSsl )
|
||
|
{
|
||
|
return StdPropertyPutBit ( &m_vroot.m_bvSslAccess, MD_ACCESS_SSL, fRequireSsl );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_Require128BitSsl ( BOOL * pfRequire128BitSsl )
|
||
|
{
|
||
|
return StdPropertyGetBit ( m_vroot.m_bvSslAccess, MD_ACCESS_SSL128, pfRequire128BitSsl );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_Require128BitSsl ( BOOL fRequire128BitSsl )
|
||
|
{
|
||
|
return StdPropertyPutBit ( &m_vroot.m_bvSslAccess, MD_ACCESS_SSL128, fRequire128BitSsl );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_UNCUsername ( BSTR * pstrUNCUsername )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_strUNCUsername, pstrUNCUsername );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_UNCUsername ( BSTR strUNCUsername )
|
||
|
{
|
||
|
return StdPropertyPut ( &m_vroot.m_strUNCUsername, strUNCUsername );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_UNCPassword ( BSTR * pstrUNCPassword )
|
||
|
{
|
||
|
return StdPropertyGet ( m_vroot.m_strUNCPassword, pstrUNCPassword );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_UNCPassword ( BSTR strUNCPassword )
|
||
|
{
|
||
|
return StdPropertyPut ( &m_vroot.m_strUNCPassword, strUNCPassword );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_DriverProgId( BSTR *pstrDriverProgId )
|
||
|
{
|
||
|
return StdPropertyGet( m_vroot.m_strDriverProgId, pstrDriverProgId );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_DriverProgId( BSTR strDriverProgId )
|
||
|
{
|
||
|
return StdPropertyPut( &m_vroot.m_strDriverProgId, strDriverProgId );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_GroupPropFile( BSTR *pstrGroupPropFile )
|
||
|
{
|
||
|
return StdPropertyGet( m_vroot.m_strGroupPropFile, pstrGroupPropFile );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_GroupPropFile( BSTR strGroupPropFile )
|
||
|
{
|
||
|
return StdPropertyPut( &m_vroot.m_strGroupPropFile, strGroupPropFile );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_MdbGuid( BSTR *pstrMdbGuid )
|
||
|
{
|
||
|
return StdPropertyGet( m_vroot.m_strMdbGuid, pstrMdbGuid );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_MdbGuid( BSTR strMdbGuid )
|
||
|
{
|
||
|
return StdPropertyPut( &m_vroot.m_strMdbGuid, strMdbGuid );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_UseAccount( DWORD *pdwUseAccount )
|
||
|
{
|
||
|
return StdPropertyGet( m_vroot.m_dwUseAccount, pdwUseAccount );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_UseAccount( DWORD dwUseAccount )
|
||
|
{
|
||
|
return StdPropertyPut( &m_vroot.m_dwUseAccount, dwUseAccount );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_OwnExpire( BOOL *pfOwnExpire )
|
||
|
{
|
||
|
return StdPropertyGet( m_vroot.m_fDoExpire, pfOwnExpire );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_OwnExpire( BOOL fOwnExpire )
|
||
|
{
|
||
|
return StdPropertyPut( &m_vroot.m_fDoExpire, fOwnExpire );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::get_OwnModerator( BOOL *pfOwnModerator )
|
||
|
{
|
||
|
return StdPropertyGet( m_vroot.m_fOwnModerator, pfOwnModerator );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoot::put_OwnModerator( BOOL fOwnModerator )
|
||
|
{
|
||
|
return StdPropertyPut( &m_vroot.m_fOwnModerator, fOwnModerator );
|
||
|
}
|
||
|
|
||
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::InterfaceSupportsErrorInfo(REFIID riid)
|
||
|
{
|
||
|
static const IID* arr[] =
|
||
|
{
|
||
|
&IID_INntpVirtualRoots,
|
||
|
};
|
||
|
|
||
|
for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++)
|
||
|
{
|
||
|
if (InlineIsEqualGUID(*arr[i],riid))
|
||
|
return S_OK;
|
||
|
}
|
||
|
return S_FALSE;
|
||
|
}
|
||
|
|
||
|
CNntpVirtualRoots::CNntpVirtualRoots () :
|
||
|
m_dwCount ( 0 ),
|
||
|
m_rgVRoots ( NULL ),
|
||
|
m_dwServiceInstance ( 0 )
|
||
|
// CComBSTR's are initialized to NULL by default.
|
||
|
{
|
||
|
}
|
||
|
|
||
|
CNntpVirtualRoots::~CNntpVirtualRoots ()
|
||
|
{
|
||
|
// All CComBSTR's are freed automatically.
|
||
|
|
||
|
delete [] m_rgVRoots;
|
||
|
}
|
||
|
|
||
|
HRESULT CNntpVirtualRoots::Init ( BSTR strServer, DWORD dwServiceInstance )
|
||
|
{
|
||
|
m_strServer = strServer;
|
||
|
m_dwServiceInstance = dwServiceInstance;
|
||
|
|
||
|
if ( !m_strServer ) {
|
||
|
return E_OUTOFMEMORY;
|
||
|
}
|
||
|
|
||
|
return NOERROR;
|
||
|
}
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////
|
||
|
// Properties:
|
||
|
//////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
// Which service to configure:
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::get_Server ( BSTR * pstrServer )
|
||
|
{
|
||
|
return StdPropertyGet ( m_strServer, pstrServer );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::put_Server ( BSTR strServer )
|
||
|
{
|
||
|
_ASSERT ( strServer );
|
||
|
_ASSERT ( IS_VALID_STRING ( strServer ) );
|
||
|
|
||
|
if ( strServer == NULL ) {
|
||
|
return E_POINTER;
|
||
|
}
|
||
|
|
||
|
if ( lstrcmp ( strServer, _T("") ) == 0 ) {
|
||
|
m_strServer.Empty();
|
||
|
|
||
|
return NOERROR;
|
||
|
}
|
||
|
else {
|
||
|
return StdPropertyPutServerName ( &m_strServer, strServer );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::get_ServiceInstance ( long * plServiceInstance )
|
||
|
{
|
||
|
return StdPropertyGet ( m_dwServiceInstance, plServiceInstance );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::put_ServiceInstance ( long lServiceInstance )
|
||
|
{
|
||
|
return StdPropertyPut ( &m_dwServiceInstance, lServiceInstance );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::get_Count ( long * pdwCount )
|
||
|
{
|
||
|
return StdPropertyGet ( m_dwCount, pdwCount );
|
||
|
}
|
||
|
|
||
|
//////////////////////////////////////////////////////////////////////
|
||
|
// Methods:
|
||
|
//////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::Enumerate ( )
|
||
|
{
|
||
|
TraceFunctEnter ( "CNntpVirtualRoots::Enumerate" );
|
||
|
|
||
|
HRESULT hr = NOERROR;
|
||
|
CComPtr<IMSAdminBase> pMetabase;
|
||
|
|
||
|
// Clear the old enumeration:
|
||
|
delete [] m_rgVRoots;
|
||
|
m_rgVRoots = NULL;
|
||
|
m_dwCount = 0;
|
||
|
|
||
|
hr = m_mbFactory.GetMetabaseObject ( m_strServer, &pMetabase );
|
||
|
if ( FAILED(hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
hr = GetVRootsFromMetabase ( pMetabase );
|
||
|
if ( FAILED(hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
Exit:
|
||
|
TRACE_HRESULT(hr);
|
||
|
TraceFunctLeave ();
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::Item (
|
||
|
long index,
|
||
|
INntpVirtualRoot ** ppVirtualRoot
|
||
|
)
|
||
|
{
|
||
|
TraceFunctEnter ( "CNntpVirtualRoots::Item" );
|
||
|
|
||
|
_ASSERT ( IS_VALID_OUT_PARAM ( ppVirtualRoot ) );
|
||
|
|
||
|
*ppVirtualRoot = NULL;
|
||
|
|
||
|
HRESULT hr = NOERROR;
|
||
|
CComObject<CNntpVirtualRoot> * pVirtualRoot = NULL;
|
||
|
|
||
|
if ( index < 0 || index >= m_dwCount ) {
|
||
|
hr = NntpCreateException ( IDS_NNTPEXCEPTION_INVALID_INDEX );
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
hr = CComObject<CNntpVirtualRoot>::CreateInstance ( &pVirtualRoot );
|
||
|
if ( FAILED(hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
_ASSERT ( pVirtualRoot );
|
||
|
hr = pVirtualRoot->SetProperties ( m_rgVRoots[index] );
|
||
|
if ( FAILED(hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
hr = pVirtualRoot->QueryInterface ( IID_INntpVirtualRoot, (void **) ppVirtualRoot );
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
|
||
|
Exit:
|
||
|
if ( FAILED(hr) ) {
|
||
|
delete pVirtualRoot;
|
||
|
}
|
||
|
|
||
|
TRACE_HRESULT(hr);
|
||
|
TraceFunctLeave ();
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::Add (
|
||
|
INntpVirtualRoot * pVirtualRoot
|
||
|
)
|
||
|
{
|
||
|
TraceFunctEnter ( "CNntpVirtualRoots::Add" );
|
||
|
|
||
|
_ASSERT ( IS_VALID_IN_PARAM ( pVirtualRoot ) );
|
||
|
|
||
|
HRESULT hr = NOERROR;
|
||
|
CVRoot * rgNewVirtualRoots = NULL;
|
||
|
long i;
|
||
|
CComPtr<IMSAdminBase> pMetabase;
|
||
|
CVRoot newVroot;
|
||
|
WCHAR wszName [ 2*METADATA_MAX_NAME_LEN ];
|
||
|
WCHAR wszVRootPath [ 2*METADATA_MAX_NAME_LEN ];
|
||
|
|
||
|
// Validate the new Virtual Root:
|
||
|
hr = newVroot.SetProperties ( pVirtualRoot );
|
||
|
if ( FAILED(hr) ) {
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
if ( !newVroot.m_strNewsgroupSubtree ||
|
||
|
!newVroot.m_strNewsgroupSubtree[0] ) {
|
||
|
|
||
|
return RETURNCODETOHRESULT ( ERROR_INVALID_PARAMETER );
|
||
|
}
|
||
|
|
||
|
// Add the new virtual root to the metabase:
|
||
|
GetVRootName ( newVroot.m_strNewsgroupSubtree, wszName );
|
||
|
|
||
|
//
|
||
|
// The sub tree itself should not exceed length METADATA_MAX_NAME_LEN
|
||
|
//
|
||
|
_ASSERT( wcslen( wszName ) <= METADATA_MAX_NAME_LEN );
|
||
|
if ( wcslen( wszName ) > METADATA_MAX_NAME_LEN ) {
|
||
|
return CO_E_PATHTOOLONG;
|
||
|
}
|
||
|
|
||
|
hr = m_mbFactory.GetMetabaseObject ( m_strServer, &pMetabase );
|
||
|
if ( FAILED(hr) ) {
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
CMetabaseKey mb ( pMetabase );
|
||
|
|
||
|
GetMDVRootPath ( wszVRootPath );
|
||
|
|
||
|
//
|
||
|
// I trust wszVRootPath, it's shorter than METADATA_MAX_NAME_LEN
|
||
|
//
|
||
|
|
||
|
//
|
||
|
// Test to see if this item exists already:
|
||
|
//
|
||
|
|
||
|
WCHAR wszTempPath [ 2 * METADATA_MAX_NAME_LEN + 1];
|
||
|
|
||
|
wsprintf ( wszTempPath, _T("%s%s"), wszVRootPath, wszName );
|
||
|
if ( wcslen( wszTempPath ) > METADATA_MAX_NAME_LEN ) {
|
||
|
|
||
|
//
|
||
|
// I can't handle it either
|
||
|
//
|
||
|
return CO_E_PATHTOOLONG;
|
||
|
}
|
||
|
|
||
|
hr = mb.Open ( wszTempPath );
|
||
|
if ( SUCCEEDED(hr) ) {
|
||
|
DWORD cbVrootDir;
|
||
|
|
||
|
if ( SUCCEEDED ( mb.GetDataSize (
|
||
|
_T(""),
|
||
|
MD_VR_PATH,
|
||
|
STRING_METADATA,
|
||
|
&cbVrootDir,
|
||
|
METADATA_NO_ATTRIBUTES )
|
||
|
) ) {
|
||
|
|
||
|
mb.Close();
|
||
|
|
||
|
hr = NntpCreateException ( IDS_NNTPEXCEPTION_VROOT_ALREADY_EXISTS );
|
||
|
goto Exit;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
hr = mb.Open ( wszVRootPath, METADATA_PERMISSION_WRITE );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
/*
|
||
|
hr = mb.CreateChild ( wszName );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
*/
|
||
|
|
||
|
hr = newVroot.SendToMetabase ( &mb, wszName );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = mb.Save ();
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
// Allocate the new VirtualRoot array:
|
||
|
rgNewVirtualRoots = new CVRoot [ m_dwCount + 1 ];
|
||
|
if ( !rgNewVirtualRoots ) {
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
// Copy the old VirtualRoots to the new array:
|
||
|
for ( i = 0; i < m_dwCount; i++ ) {
|
||
|
hr = rgNewVirtualRoots[i].SetProperties ( m_rgVRoots[i] );
|
||
|
if ( FAILED (hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Add the new VirtualRoot to the end of the array:
|
||
|
hr = rgNewVirtualRoots[m_dwCount].SetProperties ( pVirtualRoot );
|
||
|
if ( FAILED(hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
delete [] m_rgVRoots;
|
||
|
m_rgVRoots = rgNewVirtualRoots;
|
||
|
m_dwCount++;
|
||
|
|
||
|
Exit:
|
||
|
TraceFunctLeave ();
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::Set (
|
||
|
long index,
|
||
|
INntpVirtualRoot * pVirtualRoot
|
||
|
)
|
||
|
{
|
||
|
TraceFunctEnter ( "CNntpVirtualRoots::ChangeVirtualRoot" );
|
||
|
|
||
|
HRESULT hr = NOERROR;
|
||
|
|
||
|
CComBSTR strNewsgroupSubtree;
|
||
|
CComBSTR strDirectory;
|
||
|
CVRoot temp;
|
||
|
CComPtr<IMSAdminBase> pMetabase;
|
||
|
WCHAR wszVRootPath [ METADATA_MAX_NAME_LEN ];
|
||
|
WCHAR wszName [ METADATA_MAX_NAME_LEN ];
|
||
|
|
||
|
// Send the new virtual root to the metabase:
|
||
|
hr = m_mbFactory.GetMetabaseObject ( m_strServer, &pMetabase );
|
||
|
if ( FAILED(hr) ) {
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
CMetabaseKey mb ( pMetabase );
|
||
|
|
||
|
if ( index < 0 || index >= m_dwCount ) {
|
||
|
hr = NntpCreateException ( IDS_NNTPEXCEPTION_INVALID_INDEX );
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
hr = temp.SetProperties ( pVirtualRoot );
|
||
|
if ( FAILED(hr) ) {
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
GetMDVRootPath ( wszVRootPath );
|
||
|
|
||
|
hr = mb.Open ( wszVRootPath, METADATA_PERMISSION_WRITE );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
GetVRootName ( temp.m_strNewsgroupSubtree, wszName );
|
||
|
|
||
|
hr = temp.SendToMetabase ( &mb, wszName );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = m_rgVRoots[index].SetProperties ( pVirtualRoot );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
Exit:
|
||
|
TraceFunctLeave ();
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::Remove ( long index )
|
||
|
{
|
||
|
TraceFunctEnter ( "CNntpVirtualRoots::Remove" );
|
||
|
|
||
|
HRESULT hr = NOERROR;
|
||
|
CVRoot temp;
|
||
|
long cPositionsToSlide;
|
||
|
CComPtr<IMSAdminBase> pMetabase;
|
||
|
WCHAR wszName [ METADATA_MAX_NAME_LEN ];
|
||
|
WCHAR wszVRootPath [ METADATA_MAX_NAME_LEN ];
|
||
|
|
||
|
if ( index < 0 || index >= m_dwCount ) {
|
||
|
return NntpCreateException ( IDS_NNTPEXCEPTION_INVALID_INDEX );
|
||
|
}
|
||
|
|
||
|
// Delete the virtual root from the metabase:
|
||
|
|
||
|
hr = m_mbFactory.GetMetabaseObject ( m_strServer, &pMetabase );
|
||
|
if ( FAILED(hr) ) {
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
CMetabaseKey mb ( pMetabase );
|
||
|
|
||
|
GetMDVRootPath ( wszVRootPath );
|
||
|
|
||
|
hr = mb.Open ( wszVRootPath, METADATA_PERMISSION_WRITE );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
GetVRootName ( m_rgVRoots[index].m_strNewsgroupSubtree, wszName );
|
||
|
|
||
|
hr = mb.DeleteAllData ( wszName );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
/*
|
||
|
hr = mb.DestroyChild ( wszName );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
*/
|
||
|
|
||
|
hr = mb.Save ();
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
mb.Close ();
|
||
|
|
||
|
// Slide the array down by one position:
|
||
|
|
||
|
_ASSERT ( m_rgVRoots );
|
||
|
|
||
|
cPositionsToSlide = (m_dwCount - 1) - index;
|
||
|
|
||
|
_ASSERT ( cPositionsToSlide < m_dwCount );
|
||
|
|
||
|
if ( cPositionsToSlide > 0 ) {
|
||
|
// Save the deleted VirtualRoot in temp:
|
||
|
CopyMemory ( &temp, &m_rgVRoots[index], sizeof ( CVRoot ) );
|
||
|
|
||
|
// Move the array down one:
|
||
|
MoveMemory ( &m_rgVRoots[index], &m_rgVRoots[index + 1], sizeof ( CVRoot ) * cPositionsToSlide );
|
||
|
|
||
|
// Put the deleted VirtualRoot on the end (so it gets destructed):
|
||
|
CopyMemory ( &m_rgVRoots[m_dwCount - 1], &temp, sizeof ( CVRoot ) );
|
||
|
|
||
|
// Zero out the temp VirtualRoot:
|
||
|
ZeroMemory ( &temp, sizeof ( CVRoot ) );
|
||
|
}
|
||
|
|
||
|
m_dwCount--;
|
||
|
|
||
|
Exit:
|
||
|
TraceFunctLeave ();
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::Find ( BSTR strNewsgroupSubtree, long * pIndex )
|
||
|
{
|
||
|
HRESULT hr = NOERROR;
|
||
|
long i;
|
||
|
|
||
|
_ASSERT ( pIndex );
|
||
|
*pIndex = -1;
|
||
|
|
||
|
for ( i = 0; i < m_dwCount; i++ ) {
|
||
|
|
||
|
if ( lstrcmp ( m_rgVRoots[i].m_strNewsgroupSubtree, strNewsgroupSubtree ) == 0 ) {
|
||
|
*pIndex = i;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return NOERROR;
|
||
|
}
|
||
|
|
||
|
static HRESULT CountVrootsIterator (
|
||
|
CMetabaseKey * pMB,
|
||
|
LPCWSTR wszVrootPath,
|
||
|
LPARAM lParam
|
||
|
)
|
||
|
{
|
||
|
DWORD * pcVroots = (DWORD *) lParam;
|
||
|
|
||
|
(*pcVroots)++;
|
||
|
|
||
|
return NOERROR;
|
||
|
}
|
||
|
|
||
|
typedef struct tagAddVrootsParms
|
||
|
{
|
||
|
DWORD cCount;
|
||
|
CVRoot * rgVroots;
|
||
|
DWORD dwCurrentIndex;
|
||
|
LPWSTR wszServerName;
|
||
|
DWORD dwInstanceId;
|
||
|
} ADD_VROOTS_PARMS;
|
||
|
|
||
|
static HRESULT AddVrootsIterator (
|
||
|
CMetabaseKey * pMB,
|
||
|
LPCWSTR wszVrootPath,
|
||
|
LPARAM lParam
|
||
|
)
|
||
|
{
|
||
|
_ASSERT ( pMB );
|
||
|
_ASSERT ( wszVrootPath );
|
||
|
_ASSERT ( lParam );
|
||
|
|
||
|
HRESULT hr;
|
||
|
ADD_VROOTS_PARMS * pParms = (ADD_VROOTS_PARMS *) lParam;
|
||
|
LPWSTR wszServerName = pParms->wszServerName;
|
||
|
DWORD dwInstanceId = pParms->dwInstanceId;
|
||
|
|
||
|
_ASSERT ( pParms->dwCurrentIndex < pParms->cCount );
|
||
|
|
||
|
hr = pParms->rgVroots[pParms->dwCurrentIndex].GetFromMetabase (
|
||
|
pMB,
|
||
|
wszVrootPath,
|
||
|
dwInstanceId,
|
||
|
wszServerName
|
||
|
);
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
pParms->dwCurrentIndex++;
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT CNntpVirtualRoots::GetVRootsFromMetabase ( IMSAdminBase * pMetabase )
|
||
|
{
|
||
|
HRESULT hr = NOERROR;
|
||
|
CMetabaseKey mb ( pMetabase );
|
||
|
WCHAR wszVRootPath [ METADATA_MAX_NAME_LEN ];
|
||
|
DWORD cCount = 0;
|
||
|
CVRoot * rgVroots = NULL;
|
||
|
ADD_VROOTS_PARMS parms;
|
||
|
|
||
|
//
|
||
|
// Initialize the metabase:
|
||
|
//
|
||
|
|
||
|
GetMDVRootPath ( wszVRootPath );
|
||
|
|
||
|
hr = mb.Open ( wszVRootPath );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
//
|
||
|
// Count the virtual roots:
|
||
|
//
|
||
|
|
||
|
hr = IterateOverVroots ( &mb, CountVrootsIterator, (LPARAM) &cCount );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
//
|
||
|
// Create the virtual roots array:
|
||
|
//
|
||
|
|
||
|
rgVroots = new CVRoot [ cCount ];
|
||
|
if ( !rgVroots ) {
|
||
|
hr = E_OUTOFMEMORY;
|
||
|
goto Exit;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Add the virtual roots to the array:
|
||
|
//
|
||
|
|
||
|
parms.cCount = cCount;
|
||
|
parms.rgVroots = rgVroots;
|
||
|
parms.dwCurrentIndex = 0;
|
||
|
parms.wszServerName = m_strServer;
|
||
|
parms.dwInstanceId = m_dwServiceInstance;
|
||
|
|
||
|
hr = IterateOverVroots ( &mb, AddVrootsIterator, (LPARAM) &parms );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
_ASSERT ( SUCCEEDED(hr) );
|
||
|
_ASSERT ( m_rgVRoots == NULL );
|
||
|
m_rgVRoots = rgVroots;
|
||
|
m_dwCount = cCount;
|
||
|
|
||
|
Exit:
|
||
|
if ( FAILED(hr) ) {
|
||
|
delete [] rgVroots;
|
||
|
}
|
||
|
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
HRESULT CNntpVirtualRoots::IterateOverVroots (
|
||
|
CMetabaseKey * pMB,
|
||
|
VROOT_ITERATOR fpIterator,
|
||
|
LPARAM lParam,
|
||
|
LPCWSTR wszPath // = _T("")
|
||
|
)
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
WCHAR wszSubKey[ METADATA_MAX_NAME_LEN ];
|
||
|
WCHAR wszSubPath[ METADATA_MAX_NAME_LEN ];
|
||
|
BOOL fRealVroot;
|
||
|
DWORD cbVrootDir;
|
||
|
DWORD i;
|
||
|
|
||
|
//
|
||
|
// Is this a real vroot?
|
||
|
//
|
||
|
|
||
|
fRealVroot =
|
||
|
!*wszPath || // Always count the home directory
|
||
|
SUCCEEDED ( pMB->GetDataSize (
|
||
|
wszPath,
|
||
|
MD_VR_PATH,
|
||
|
STRING_METADATA,
|
||
|
&cbVrootDir,
|
||
|
METADATA_NO_ATTRIBUTES )
|
||
|
);
|
||
|
|
||
|
if ( fRealVroot ) {
|
||
|
// Call the iterator on this key:
|
||
|
|
||
|
hr = (*fpIterator) ( pMB, wszPath, lParam );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Recurse down the tree:
|
||
|
//
|
||
|
|
||
|
for ( i = 0; ; ) {
|
||
|
hr = pMB->EnumObjects ( wszPath, wszSubKey, i );
|
||
|
|
||
|
if ( HRESULTTOWIN32(hr) == ERROR_NO_MORE_ITEMS ) {
|
||
|
// This is expected, end the loop:
|
||
|
|
||
|
if ( !fRealVroot && i == 0 ) {
|
||
|
//
|
||
|
// This key isn't a vroot and has no children, so delete it.
|
||
|
//
|
||
|
|
||
|
hr = pMB->ChangePermissions ( METADATA_PERMISSION_WRITE );
|
||
|
if ( SUCCEEDED(hr) ) {
|
||
|
hr = pMB->DeleteKey ( wszPath );
|
||
|
}
|
||
|
|
||
|
pMB->ChangePermissions ( METADATA_PERMISSION_READ );
|
||
|
if ( SUCCEEDED(hr) ) {
|
||
|
hr = pMB->Save ();
|
||
|
}
|
||
|
|
||
|
if ( SUCCEEDED(hr) ) {
|
||
|
//
|
||
|
// Tell our parent that this key was deleted.
|
||
|
//
|
||
|
|
||
|
hr = S_FALSE;
|
||
|
}
|
||
|
else {
|
||
|
//
|
||
|
// Ignore any deleting problems:
|
||
|
//
|
||
|
|
||
|
hr = NOERROR;
|
||
|
}
|
||
|
}
|
||
|
else {
|
||
|
hr = NOERROR;
|
||
|
}
|
||
|
break;
|
||
|
}
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
if ( *wszPath ) {
|
||
|
wsprintf ( wszSubPath, _T("%s/%s"), wszPath, wszSubKey );
|
||
|
}
|
||
|
else {
|
||
|
wcscpy ( wszSubPath, wszSubKey );
|
||
|
}
|
||
|
|
||
|
hr = IterateOverVroots ( pMB, fpIterator, lParam, wszSubPath );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
if ( hr != S_FALSE ) {
|
||
|
//
|
||
|
// This means the child key still exists, so increment
|
||
|
// the counter and go on to the next one.
|
||
|
//
|
||
|
|
||
|
i++;
|
||
|
}
|
||
|
//
|
||
|
// Else the return code is S_FALSE, which means the current key
|
||
|
// has been deleted, shifting all indicies down by one. Therefore,
|
||
|
// there is no need to increment i.
|
||
|
//
|
||
|
}
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
void CNntpVirtualRoots::GetVRootName ( LPWSTR wszDisplayName, LPWSTR wszPathName )
|
||
|
{
|
||
|
wcscpy ( wszPathName, wszDisplayName );
|
||
|
}
|
||
|
|
||
|
void CNntpVirtualRoots::GetMDVRootPath ( LPWSTR wszPath )
|
||
|
{
|
||
|
GetMDInstancePath ( wszPath, m_dwServiceInstance );
|
||
|
|
||
|
wcscat ( wszPath, _T("Root/") );
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::ItemDispatch ( long index, IDispatch ** ppDispatch )
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
CComPtr<INntpVirtualRoot> pVirtualRoot;
|
||
|
|
||
|
hr = Item ( index, &pVirtualRoot );
|
||
|
BAIL_ON_FAILURE ( hr );
|
||
|
|
||
|
hr = pVirtualRoot->QueryInterface ( IID_IDispatch, (void **) ppDispatch );
|
||
|
BAIL_ON_FAILURE ( hr );
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::AddDispatch ( IDispatch * pVirtualRoot )
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
CComPtr<INntpVirtualRoot> pINntpVirtualRoot;
|
||
|
|
||
|
hr = pVirtualRoot->QueryInterface ( IID_INntpVirtualRoot, (void **) &pINntpVirtualRoot );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = Add ( pINntpVirtualRoot );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|
||
|
STDMETHODIMP CNntpVirtualRoots::SetDispatch ( long lIndex, IDispatch * pVirtualRoot )
|
||
|
{
|
||
|
HRESULT hr;
|
||
|
CComPtr<INntpVirtualRoot> pINntpVirtualRoot;
|
||
|
|
||
|
hr = pVirtualRoot->QueryInterface ( IID_INntpVirtualRoot, (void **) &pINntpVirtualRoot );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
hr = Set ( lIndex, pINntpVirtualRoot );
|
||
|
BAIL_ON_FAILURE(hr);
|
||
|
|
||
|
Exit:
|
||
|
return hr;
|
||
|
}
|
||
|
|