/*++
© 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);
HRESULT
CSakNode::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 );
}
STDMETHODIMP
CSakNode::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 );
}
STDMETHODIMP
CSakNode::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 );
}
STDMETHODIMP
CSakNode::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;
}