|
|
/*++
� 1998 Seagate Software, Inc. All rights reserved.
Module Name:
DataObj.cpp
Abstract:
Implementation of IDataObject interface, which is supported by the CBaseHsm COM object.
Author:
Rohde Wakefield [rohde] 19-Aug-1997
Revision History:
--*/
#include "stdafx.h"
// Declare Snap-in NodeType formats:
// - GUID format
// - string GUID format
// - display name format.
// - internal format.
UINT CSakNode::m_cfNodeType = RegisterClipboardFormat(CCF_NODETYPE);UINT CSakNode::m_cfNodeTypeString = RegisterClipboardFormat(CCF_SZNODETYPE); UINT CSakNode::m_cfDisplayName = RegisterClipboardFormat(CCF_DISPLAY_NAME); UINT CSakNode::m_cfInternal = RegisterClipboardFormat(SAKSNAP_INTERNAL); UINT CSakNode::m_cfClassId = RegisterClipboardFormat(CCF_SNAPIN_CLASSID); UINT CSakNode::m_cfComputerName = RegisterClipboardFormat(MMC_SNAPIN_MACHINE_NAME); UINT CSakNode::m_cfEventLogViews = RegisterClipboardFormat(CF_EV_VIEWS);
HRESULTCSakNode::GetDataGeneric( IN LPFORMATETC lpFormatetcIn, IN OUT LPSTGMEDIUM lpMedium, IN BOOL DoAlloc )/*++
Routine Description:
Retrieve information FROM the dataobject and put INTO lpMedium.
Arguments:
lpFormatetc - Format to retreive.
lpMedium - Storage to put information into.
Return Value:
S_OK - Storage filled in.
E_xxxxxxxxxxx - Failure occurred.
--*/{ WsbTraceIn( L"CSakNode::GetDataGeneric", L"lpFormatetc->cfFormat = <%ls>", RsClipFormatAsString( lpFormatetcIn->cfFormat ), WsbBoolAsString( DoAlloc ) ); HRESULT hr = DV_E_CLIPFORMAT;
try {
WsbAffirmPointer( lpMedium );
if( DoAlloc ) {
lpMedium->hGlobal = 0; lpMedium->tymed = TYMED_HGLOBAL;
} else {
WsbAffirm( TYMED_HGLOBAL == lpMedium->tymed, DV_E_TYMED ); WsbAffirmPointer( lpMedium->hGlobal );
}
// Based on the CLIPFORMAT write data to "lpMediam" in the correct format.
const CLIPFORMAT cf = lpFormatetcIn->cfFormat;
// clip format is the GUID node type
if(cf == m_cfNodeType) { hr = RetrieveNodeTypeData(lpMedium);
// clip format is the string "spelling" of the GUID node type
} else if(cf == m_cfNodeTypeString) { hr = RetrieveNodeTypeStringData(lpMedium);
// clip format is the computer represented
} else if (cf == m_cfComputerName) { hr = RetrieveComputerName(lpMedium);
// clip format is the event viewer setup
} else if (cf == m_cfEventLogViews) { hr = RetrieveEventLogViews(lpMedium);
// clip format is the display name of the node
} else if (cf == m_cfDisplayName) { hr = RetrieveDisplayName(lpMedium);
// clip format is the ClassId
} else if( cf == m_cfClassId ) { hr = RetrieveClsid( lpMedium );
// clip format is an INTERNAL format
} else if (cf == m_cfInternal) { hr = RetrieveInternal(lpMedium);
} else { hr = DV_E_CLIPFORMAT; }
} WsbCatch( hr );
WsbTraceOut( L"CSakNode::GetDataGeneric", L"hr = <%ls>", WsbHrAsString( hr ) ); return( hr );} STDMETHODIMPCSakNode::GetData( IN LPFORMATETC lpFormatetcIn, OUT LPSTGMEDIUM lpMedium )/*++
Routine Description:
Retrieve information FROM the dataobject and put INTO lpMedium. Storage allocated and returned.
Arguments:
lpFormatetc - Format to retreive.
lpMedium - Storage to put information into.
Return Value:
S_OK - Storage filled in.
E_xxxxxxxxxxx - Failure occurred.
--*/{ WsbTraceIn( L"CSakNode::GetData", L"lpFormatetc->cfFormat = <%ls>", RsClipFormatAsString( lpFormatetcIn->cfFormat ) );
HRESULT hr = S_OK;
hr = GetDataGeneric( lpFormatetcIn, lpMedium, TRUE );
WsbTraceOut( L"CSakNode::GetData", L"hr = <%ls>", WsbHrAsString( hr ) ); return( hr );} STDMETHODIMPCSakNode::GetDataHere( IN LPFORMATETC lpFormatetc, IN OUT LPSTGMEDIUM lpMedium )/*++
Routine Description:
Retrieve information FROM the dataobject and put INTO lpMedium.
Arguments:
lpFormatetc - Format to retreive.
lpMedium - Storage to put information into.
Return Value:
S_OK - Storage filled in.
E_xxxxxxxxxxx - Failure occurred.
--*/{ WsbTraceIn( L"CSakNode::GetDataHere", L"lpFormatetc->cfFormat = <%ls>", RsClipFormatAsString( lpFormatetc->cfFormat ) ); HRESULT hr = S_OK;
hr = GetDataGeneric( lpFormatetc, lpMedium, FALSE );
WsbTraceOut( L"CSakNode::GetDataHere", L"hr = <%ls>", WsbHrAsString( hr ) ); return( hr );} STDMETHODIMPCSakNode::SetData( LPFORMATETC lpFormatetc, LPSTGMEDIUM lpMedium, BOOL /*fRelease*/ )/*++
Routine Description:
Put data INTO a dataobject FROM the information in the lpMedium. We do not allow any data to be set.
Arguments:
lpFormatetc - Format to set.
lpMedium - Storage to get information from.
fRelease - Indicates who owns storage after call.
Return Value:
S_OK - Storage retreived.
E_xxxxxxxxxxx - Failure occurred.
--*/{ WsbTraceIn( L"CSakNode::SetData", L"lpFormatetc->cfFormat = <%ls>", RsClipFormatAsString( lpFormatetc->cfFormat ) ); HRESULT hr = DV_E_CLIPFORMAT;
// Based on the CLIPFORMAT write data to "lpMediam" in the correct format.
const CLIPFORMAT cf = lpFormatetc->cfFormat;
//clip format is an INTERNAL format
if( cf == m_cfInternal ) {
hr = StoreInternal( lpMedium );
}
WsbTraceOut( L"CSakNode::SetData", L"hr = <%ls>", WsbHrAsString( hr ) ); return( hr );} ///////////////////////////////////////////////////////////////////////
// Note - CSakNode does not implement these
///////////////////////////////////////////////////////////////////////
STDMETHODIMP CSakNode::EnumFormatEtc(DWORD /*dwDirection*/, LPENUMFORMATETC* /*ppEnumFormatEtc*/){ WsbTraceIn( L"CSakNode::EnumFormatEtc", L"" );
HRESULT hr = E_NOTIMPL;
WsbTraceOut( L"CSakNode::EnumFormatEtc", L"hr = <%ls>", WsbHrAsString( hr ) ); return( hr );}
// Retrieve from a dataobject with the NodeType (GUID) data in it.
HRESULT CSakNode::RetrieveNodeTypeData(LPSTGMEDIUM lpMedium){ return Retrieve((const void*)(m_rTypeGuid), sizeof(GUID), lpMedium);}
// Retrieve from a dataobject with the node type object in GUID string format
HRESULT CSakNode::RetrieveNodeTypeStringData(LPSTGMEDIUM lpMedium){ CWsbStringPtr guidString = *m_rTypeGuid; return Retrieve(guidString, ((wcslen(guidString)+1) * sizeof(wchar_t)), lpMedium);}
// Retrieve from a dataobject with the display named used in the scope pane
HRESULT CSakNode::RetrieveDisplayName(LPSTGMEDIUM lpMedium){ // Load the name the data object
return Retrieve(m_szName, ((wcslen(m_szName)+1) * sizeof(wchar_t)), lpMedium);}
// Retrieve from a dataobject with the CLSID data in it.
HRESULT CSakNode::RetrieveClsid(LPSTGMEDIUM lpMedium){ // zzzz
return Retrieve( (const void*)(&CLSID_HsmAdminDataSnapin), sizeof(CLSID), lpMedium );}
// Retrieve INTERNAL data from the dataobject's m_internal member INTO the lpMedium
HRESULT CSakNode::RetrieveInternal(LPSTGMEDIUM lpMedium){ return Retrieve(&m_internal, sizeof(INTERNAL), lpMedium);}
// Retrieve data from the dataobject's hsm name
HRESULT CSakNode::RetrieveComputerName(LPSTGMEDIUM lpMedium){ HRESULT hr = S_OK;
try {
CWsbStringPtr computerName; HRESULT hrTemp = m_pSakSnapAsk->GetHsmName( &computerName ); WsbAffirmHr( hrTemp );
if( S_FALSE == hrTemp ) {
computerName = L"";
}
WsbAffirmHr( Retrieve( (WCHAR*)computerName, ( wcslen( computerName ) + 1 ) * sizeof(WCHAR), lpMedium ) );
} WsbCatch( hr );
return( hr );}
// Retrieve event setup info
HRESULT CSakNode::RetrieveEventLogViews(LPSTGMEDIUM lpMedium){ HRESULT hr = S_OK;
AFX_MANAGE_STATE(AfxGetStaticModuleState());
try {
WsbAffirmPointer( lpMedium ); ULONG strLength = 0;
CWsbStringPtr hsmName; CString appName, sysName; CString nullString; CString appPath, sysPath;
appName.LoadString( IDS_EVENT_LOG_APP_TITLE ); sysName.LoadString( IDS_EVENT_LOG_SYS_TITLE ); nullString = L"";
HRESULT hrTemp = m_pSakSnapAsk->GetHsmName( &hsmName ); WsbAffirmHr( hrTemp ); if( S_FALSE == hrTemp ) {
hsmName = L""; appPath = L""; sysPath = L"";
} else {
CString configPath = L"\\\\"; configPath += hsmName; configPath += L"\\Admin$\\System32\\config\\"; appPath = configPath; sysPath = configPath; appPath += L"AppEvent.Evt"; sysPath += L"SysEvent.Evt";
} #define ADD_TYPE(data, type) \
{type x = data; \ WsbAffirmHr( spStm->Write(&x, sizeof(type), NULL) ) };
#define ADD_USHORT(us) ADD_TYPE(us, USHORT)
#define ADD_BOOL(b) ADD_TYPE(b, BOOL)
#define ADD_ULONG(ul) ADD_TYPE(ul, ULONG)
#define ADD_STRING(str) \
{strLength = (ULONG)wcslen((LPCWSTR)(str)) + 1; \ ADD_USHORT((USHORT)strLength); \ WsbAffirmHr( spStm->Write(str, strLength * sizeof(WCHAR), NULL) ) };
CComPtr<IStream> spStm;
WsbAffirmHr( CreateStreamOnHGlobal(NULL, FALSE, &spStm) );
//
// Add header info
//
ADD_BOOL( TRUE ); // fOnlyTheseViews
ADD_USHORT( 2 ); // cViews
//
// Add application log filtered for our services
//
ADD_ULONG( ELT_APPLICATION ); // Type;
ADD_USHORT( VIEWINFO_FILTERED | LOGINFO_DONT_PERSIST); // flViewFlags
ADD_STRING( hsmName ); // ServerName
PCWSTR pwz = L"Application"; ADD_STRING( pwz ); // SourceName
ADD_STRING( appPath ); // FileName
ADD_STRING( appName ); // DisplayName
ADD_ULONG( EVENTLOG_ALL_EVENTS ); // flRecType (could filter warning, error, etc.)
ADD_USHORT( 0 ); // usCategory
ADD_BOOL( FALSE ); // fEventID
ADD_ULONG( 0 ); // ulEventID
ADD_STRING( WSB_LOG_SOURCE_NAME ); // szSourceName
ADD_STRING( nullString ); // szUser
ADD_STRING( hsmName ); // szComputer
ADD_ULONG( 0 ); // ulFrom
ADD_ULONG( 0 ); // ulTo
//
// Add system log filtered for our device
//
ADD_ULONG( ELT_SYSTEM ); // Type;
ADD_USHORT( VIEWINFO_FILTERED | LOGINFO_DONT_PERSIST); // flViewFlags
ADD_STRING( hsmName ); // ServerName
pwz = L"System"; ADD_STRING( pwz ); // SourceName
ADD_STRING( sysPath ); // FileName
ADD_STRING( sysName ); // DisplayName
ADD_ULONG( EVENTLOG_ALL_EVENTS ); // flRecType (could filter warning, error, etc.)
ADD_USHORT( 0 ); // usCategory
ADD_BOOL( FALSE ); // fEventID
ADD_ULONG( 0 ); // ulEventID
ADD_STRING( WSB_LOG_FILTER_NAME ); // szSourceName
ADD_STRING( nullString ); // szUser
ADD_STRING( hsmName ); // szComputer
ADD_ULONG( 0 ); // ulFrom
ADD_ULONG( 0 ); // ulTo
HGLOBAL hMem = NULL;
WsbAffirmHr( GetHGlobalFromStream(spStm, &hMem) );
lpMedium->hGlobal = hMem; // StgMedium variables
lpMedium->tymed = TYMED_HGLOBAL; lpMedium->pUnkForRelease = NULL;
} WsbCatch( hr );
return( hr );}
// Store the INTERNAL data FROM the lpMedium->hGlobal INTO the dataobject's m_internal member
HRESULT CSakNode::StoreInternal(LPSTGMEDIUM lpMedium){ return Store(&m_internal, sizeof(INTERNAL), lpMedium);}
// Retrieve FROM a dataobject INTO a lpMedium. The data object can be one of
// several types of data in it (nodetype, nodetype string, display name, or
// INTERNAL data).
// This function moves data from pBuffer to the lpMedium->hGlobal
//
HRESULT CSakNode::Retrieve(const void* pBuffer, DWORD len, LPSTGMEDIUM lpMedium){ HRESULT hr = S_OK;
try {
WsbAffirmPointer( pBuffer ); WsbAffirmPointer( lpMedium ); WsbAffirm( TYMED_HGLOBAL == lpMedium->tymed, DV_E_TYMED );
//
// Check to see if we need to allocate the global memory here
//
if( 0 == lpMedium->hGlobal ) {
lpMedium->hGlobal = ::GlobalAlloc( GPTR, len );
} else {
WsbAffirm( GlobalSize( lpMedium->hGlobal ) >= (DWORD)len, STG_E_MEDIUMFULL );
}
WsbAffirmPointer( lpMedium->hGlobal );
// Create the stream on the hGlobal passed in. When we write to the stream,
// it simultaneously writes to the hGlobal the same information.
LPSTREAM lpStream; WsbAffirmHr( CreateStreamOnHGlobal(lpMedium->hGlobal, FALSE, &lpStream ));
// Write 'len' number of bytes from pBuffer into the stream. When we write
// to the stream, it simultaneously writes to the global memory we
// associated it with above.
ULONG numBytesWritten; WsbAffirmHr( lpStream->Write(pBuffer, len, &numBytesWritten ));
// Because we told CreateStreamOnHGlobal with 'FALSE', only the stream is released here.
// Note - the caller (i.e. snap-in, object) will free the HGLOBAL
// at the correct time. This is according to the IDataObject specification.
lpStream->Release();
} WsbCatch( hr );
return hr;}
// Store INTO a dataobject FROM an lpMedium. The data object can be one of
// several types of data in it (nodetype, nodetype string, display name, or
// INTERNAL data).
// This function moves data INTO pBuffer FROM the lpMedium->hGlobal
//
HRESULT CSakNode::Store( void* pBuffer, DWORD len, LPSTGMEDIUM lpMedium ){ HRESULT hr = S_OK;
try { WsbAffirmPointer( pBuffer ); WsbAffirmPointer( lpMedium ); WsbAffirm( lpMedium->tymed == TYMED_HGLOBAL, E_INVALIDARG );
// Use memcpy, because the lpStream->Read is failing to read any bytes.
memcpy(pBuffer, &(lpMedium->hGlobal), len);
} WsbCatch( hr );
return hr;}
|
