717 lines
24 KiB
C++
717 lines
24 KiB
C++
// DataObj.cpp : Implementation of data object classes
|
|
|
|
#include "stdafx.h"
|
|
#include "compdata.h"
|
|
#include "safetemp.h"
|
|
|
|
#include "macros.h"
|
|
USE_HANDLE_MACROS("FILEMGMT(dataobj.cpp)")
|
|
|
|
#include "FileSvc.h" // FileServiceProvider
|
|
#include "dataobj.h"
|
|
|
|
#include "smb.h"
|
|
#include "sfm.h"
|
|
#include "cmponent.h" // for COLNUM_SESSIONS_COMPUTERNAME
|
|
|
|
#define DONT_WANT_SHELLDEBUG
|
|
#include "shlobjp.h" // ILFree, ILGetSize, ILClone, etc.
|
|
|
|
#include <comstrm.h>
|
|
|
|
#ifdef _DEBUG
|
|
#define new DEBUG_NEW
|
|
#undef THIS_FILE
|
|
static char THIS_FILE[] = __FILE__;
|
|
#endif
|
|
|
|
|
|
#include "stddtobj.cpp"
|
|
|
|
|
|
// IDataObject interface implementation
|
|
HRESULT CFileMgmtDataObject::GetDataHere(
|
|
FORMATETC __RPC_FAR *pFormatEtcIn,
|
|
STGMEDIUM __RPC_FAR *pMedium)
|
|
{
|
|
MFC_TRY;
|
|
|
|
const CLIPFORMAT cf=pFormatEtcIn->cfFormat;
|
|
|
|
if (cf == m_CFInternal)
|
|
{
|
|
CFileMgmtDataObject *pThis = this;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(&pThis, sizeof(CFileMgmtDataObject*));
|
|
}
|
|
else if (cf == m_CFSnapInCLSID)
|
|
{
|
|
stream_ptr s(pMedium);
|
|
return s.Write(&m_SnapInCLSID, sizeof(GUID));
|
|
}
|
|
|
|
if (!m_MultiSelectObjList.empty())
|
|
{
|
|
//
|
|
// this is the multiselect data object, we don't support other clipformats in GetDataHere().
|
|
//
|
|
return DV_E_FORMATETC;
|
|
}
|
|
|
|
if (cf == m_CFNodeType)
|
|
{
|
|
const GUID* pguid = GetObjectTypeGUID( m_objecttype );
|
|
stream_ptr s(pMedium);
|
|
return s.Write(pguid, sizeof(GUID));
|
|
}
|
|
else if (cf == m_CFNodeTypeString)
|
|
{
|
|
const BSTR strGUID = GetObjectTypeString( m_objecttype );
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strGUID);
|
|
}
|
|
else if (cf == m_CFDisplayName)
|
|
{
|
|
return PutDisplayName(pMedium);
|
|
}
|
|
else if (cf == m_CFDataObjectType)
|
|
{
|
|
stream_ptr s(pMedium);
|
|
return s.Write(&m_dataobjecttype, sizeof(m_dataobjecttype));
|
|
}
|
|
else if (cf == m_CFMachineName)
|
|
{
|
|
stream_ptr s(pMedium);
|
|
return s.Write(m_strMachineName);
|
|
}
|
|
else if (cf == m_CFTransport)
|
|
{
|
|
FILEMGMT_TRANSPORT transport = FILEMGMT_OTHER;
|
|
HRESULT hr = m_pcookie->GetTransport( &transport );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(&transport, sizeof(DWORD));
|
|
}
|
|
else if (cf == m_CFShareName)
|
|
{
|
|
ASSERT( NULL != m_pcookie );
|
|
CString strShareName;
|
|
HRESULT hr = m_pcookie->GetShareName( strShareName );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strShareName);
|
|
}
|
|
else if (cf == m_CFSessionClientName)
|
|
{
|
|
ASSERT( NULL != m_pcookie );
|
|
CString strSessionClientName;
|
|
HRESULT hr = m_pcookie->GetSessionClientName( strSessionClientName );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strSessionClientName);
|
|
}
|
|
else if (cf == m_CFSessionUserName)
|
|
{
|
|
ASSERT( NULL != m_pcookie );
|
|
CString strSessionUserName;
|
|
HRESULT hr = m_pcookie->GetSessionUserName( strSessionUserName );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strSessionUserName);
|
|
}
|
|
else if (cf == m_CFSessionID)
|
|
{
|
|
DWORD dwSessionID = 0;
|
|
HRESULT hr = m_pcookie->GetSessionID( &dwSessionID );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(&dwSessionID, sizeof(DWORD));
|
|
}
|
|
else if (cf == m_CFFileID)
|
|
{
|
|
DWORD dwFileID = 0;
|
|
HRESULT hr = m_pcookie->GetFileID( &dwFileID );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(&dwFileID, sizeof(DWORD));
|
|
}
|
|
else if (cf == m_CFServiceName)
|
|
{
|
|
ASSERT( NULL != m_pcookie );
|
|
CString strServiceName;
|
|
HRESULT hr = m_pcookie->GetServiceName( strServiceName );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strServiceName);
|
|
}
|
|
else if (cf == m_CFServiceDisplayName)
|
|
{
|
|
ASSERT( NULL != m_pcookie );
|
|
CString strServiceDisplayName;
|
|
HRESULT hr = m_pcookie->GetServiceDisplayName( strServiceDisplayName );
|
|
if ( FAILED(hr) )
|
|
return hr;
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strServiceDisplayName);
|
|
}
|
|
else if (cf == m_CFRawCookie)
|
|
{
|
|
stream_ptr s(pMedium);
|
|
return s.Write((PBYTE)&m_pcookie, sizeof(m_pcookie));
|
|
}
|
|
else if (cf == m_CFSnapinPreloads) // added JonN 01/19/00
|
|
{
|
|
stream_ptr s(pMedium);
|
|
BOOL bPreload = TRUE;
|
|
return s.Write((PBYTE)&bPreload, sizeof(BOOL));
|
|
}
|
|
|
|
return DV_E_FORMATETC;
|
|
|
|
MFC_CATCH;
|
|
}
|
|
|
|
HRESULT CFileMgmtDataObject::Initialize( CFileMgmtCookie* pcookie,
|
|
CFileMgmtComponentData& refComponentData,
|
|
DATA_OBJECT_TYPES type )
|
|
{
|
|
if (NULL == pcookie)
|
|
{
|
|
ASSERT(FALSE);
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
m_pComponentData = &refComponentData;
|
|
((IComponentData*) m_pComponentData)->AddRef ();
|
|
|
|
m_objecttype = pcookie->QueryObjectType();
|
|
m_fAllowOverrideMachineName = refComponentData.m_fAllowOverrideMachineName;
|
|
#ifdef SNAPIN_PROTOTYPER
|
|
if (m_objecttype != FILEMGMT_PROTOTYPER_LEAF)
|
|
m_strMachineName = pcookie->QueryTargetServer();
|
|
#else
|
|
m_strMachineName = pcookie->QueryTargetServer();
|
|
#endif
|
|
m_dataobjecttype = type;
|
|
m_pcookie = pcookie;
|
|
m_pcookie->AddRefCookie();
|
|
VERIFY( SUCCEEDED(refComponentData.GetClassID(&m_SnapInCLSID)) );
|
|
return S_OK;
|
|
}
|
|
|
|
|
|
CFileMgmtDataObject::~CFileMgmtDataObject()
|
|
{
|
|
FreeMultiSelectObjList();
|
|
|
|
if (NULL != m_pcookie)
|
|
{
|
|
m_pcookie->ReleaseCookie();
|
|
}
|
|
if ( m_pComponentData )
|
|
((IComponentData*) m_pComponentData)->Release ();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
// CFileMgmtDataObject::IDataObject::GetData()
|
|
//
|
|
// Write data into the storage medium.
|
|
// The data will be retrieved by the Send Console Message snapin.
|
|
//
|
|
//
|
|
HRESULT CFileMgmtDataObject::GetData(
|
|
FORMATETC __RPC_FAR * pFormatEtcIn,
|
|
STGMEDIUM __RPC_FAR * pMedium)
|
|
{
|
|
ASSERT(pFormatEtcIn != NULL);
|
|
ASSERT(pMedium != NULL);
|
|
|
|
HRESULT hResult = S_OK;
|
|
const CLIPFORMAT cf = pFormatEtcIn->cfFormat;
|
|
|
|
if (!m_MultiSelectObjList.empty())
|
|
{
|
|
//
|
|
// This is the multiselect dataobject.
|
|
//
|
|
if (cf == m_CFObjectTypesInMultiSelect)
|
|
{
|
|
//
|
|
// We will provide the list of object types of all the currently selected items here.
|
|
// MMC will use this format to determine extensions snapins
|
|
//
|
|
UINT nMultiSelectedObjects = m_MultiSelectObjList.size();
|
|
|
|
// Calculate the size of SMMCObjectTypes.
|
|
int cb = sizeof(DWORD) + sizeof(SMMCObjectTypes) * nMultiSelectedObjects;
|
|
|
|
//Fill out parameters
|
|
pMedium->tymed = TYMED_HGLOBAL;
|
|
pMedium->hGlobal = ::GlobalAlloc(GMEM_SHARE|GMEM_MOVEABLE, cb);
|
|
if (pMedium->hGlobal == NULL)
|
|
return STG_E_MEDIUMFULL;
|
|
|
|
SMMCObjectTypes* pMMCDO = reinterpret_cast<SMMCObjectTypes*>(::GlobalLock(pMedium->hGlobal));
|
|
pMMCDO->count = 0;
|
|
GUID *pMMCDOGuid = pMMCDO->guid;
|
|
|
|
for (CDataObjectList::iterator i = m_MultiSelectObjList.begin(); i != m_MultiSelectObjList.end(); i++)
|
|
{
|
|
CCookie* pbasecookie = NULL;
|
|
hResult = ExtractData(*i, CFileMgmtDataObject::m_CFRawCookie, &pbasecookie, sizeof(pbasecookie));
|
|
if (FAILED(hResult))
|
|
break;
|
|
|
|
pbasecookie = m_pComponentData->ActiveBaseCookie( pbasecookie );
|
|
CFileMgmtCookie* pUseThisCookie = dynamic_cast<CFileMgmtCookie*>(pbasecookie);
|
|
FileMgmtObjectType objecttype = pUseThisCookie->QueryObjectType();
|
|
const GUID* pguid = GetObjectTypeGUID(objecttype);
|
|
memcpy(pMMCDOGuid++, pguid, sizeof(GUID));
|
|
pMMCDO->count++;
|
|
}
|
|
::GlobalUnlock(pMedium->hGlobal);
|
|
|
|
if (FAILED(hResult))
|
|
{
|
|
::GlobalFree(pMedium->hGlobal);
|
|
pMedium->hGlobal = NULL;
|
|
}
|
|
} else
|
|
hResult = DV_E_FORMATETC;
|
|
|
|
return hResult;
|
|
}
|
|
|
|
if (cf == m_cfSendConsoleMessageRecipients)
|
|
{
|
|
ASSERT (m_pComponentData);
|
|
if ( m_pComponentData )
|
|
{
|
|
|
|
// I suppose it doesn't matter what type this cookie is because we
|
|
// are just using to hang enumerated session cookies off of, as long
|
|
// as it's not an abstract class.
|
|
BOOL fContinue = TRUE;
|
|
HRESULT hr = S_OK;
|
|
CSmbSessionCookie* pCookie = new CSmbSessionCookie[1];
|
|
if ( pCookie )
|
|
{
|
|
CSmbCookieBlock cookieBlock (pCookie, 1, (LPCTSTR) m_strMachineName, 0);
|
|
|
|
CBaseCookieBlock* pCookieBlock = 0;
|
|
|
|
//
|
|
// JonN 6/28/01 426224
|
|
// Shared Folder: Send Message access denied error title needs to be changed for localization
|
|
//
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState()); // required for CWaitCursor
|
|
|
|
// Enumerate all the session cookies
|
|
for (INT iTransport = FILEMGMT_FIRST_TRANSPORT;
|
|
fContinue && iTransport < FILEMGMT_NUM_TRANSPORTS;
|
|
iTransport++ )
|
|
{
|
|
hr = m_pComponentData->GetFileServiceProvider(iTransport)->EnumerateSessions(
|
|
NULL, &pCookie[0], false);
|
|
fContinue = SUCCEEDED(hr);
|
|
}
|
|
|
|
// Enumerate all the computer names from the session cookies and store them in
|
|
// the computerList
|
|
CStringList computerList;
|
|
size_t len = 0; // number of WCHARS
|
|
while ( !pCookie[0].m_listResultCookieBlocks.IsEmpty () )
|
|
{
|
|
pCookieBlock = pCookie[0].m_listResultCookieBlocks.RemoveHead ();
|
|
ASSERT (pCookieBlock);
|
|
if ( !pCookieBlock )
|
|
break;
|
|
int nCookies = pCookieBlock->QueryNumCookies ();
|
|
while (nCookies--)
|
|
{
|
|
CCookie* pBaseCookie = pCookieBlock->QueryBaseCookie (nCookies);
|
|
ASSERT (pBaseCookie);
|
|
if ( pBaseCookie )
|
|
{
|
|
CFileMgmtCookie* pFMCookie = dynamic_cast <CFileMgmtCookie*> (pBaseCookie);
|
|
ASSERT (pFMCookie);
|
|
if ( pFMCookie )
|
|
{
|
|
computerList.AddHead (pFMCookie->QueryResultColumnText (
|
|
COLNUM_SESSIONS_COMPUTERNAME, *m_pComponentData));
|
|
len += computerList.GetHead ().GetLength () + 1; // to account for NULL
|
|
}
|
|
}
|
|
}
|
|
pCookieBlock->Release ();
|
|
}
|
|
|
|
|
|
if ( !m_strMachineName.IsEmpty () )
|
|
{
|
|
computerList.AddHead (m_strMachineName);
|
|
len += computerList.GetHead ().GetLength () + 1; // to account for NULL
|
|
}
|
|
|
|
// Run through all the computer names in computerList and add them to the output buffer
|
|
//
|
|
// Write the list of recipients to the storage medium.
|
|
// - The list of recipients is a group of UNICODE strings
|
|
// terminated by TWO null characters.c
|
|
// - Allocated memory must include BOTH null characters.
|
|
//
|
|
len += 1; // to account for extra NULL at end.
|
|
WCHAR* pgrszRecipients = new WCHAR[len];
|
|
WCHAR* ptr = pgrszRecipients;
|
|
CString computerName;
|
|
|
|
if ( pgrszRecipients )
|
|
{
|
|
::ZeroMemory (pgrszRecipients, len * sizeof (WCHAR));
|
|
while ( !computerList.IsEmpty () )
|
|
{
|
|
computerName = computerList.RemoveHead ();
|
|
|
|
// append computer name
|
|
wcscpy (ptr, (LPCTSTR) computerName);
|
|
|
|
// skip past computer name and terminating NULL
|
|
ptr += computerName.GetLength () + 1;
|
|
}
|
|
|
|
// Add the name of this computer
|
|
|
|
HGLOBAL hGlobal = ::GlobalAlloc (GMEM_FIXED, len * sizeof (WCHAR));
|
|
if ( hGlobal )
|
|
{
|
|
memcpy (OUT hGlobal, pgrszRecipients, len * sizeof (WCHAR));
|
|
pMedium->hGlobal = hGlobal;
|
|
}
|
|
else
|
|
hResult = E_OUTOFMEMORY;
|
|
|
|
delete [] pgrszRecipients;
|
|
}
|
|
else
|
|
hResult = E_OUTOFMEMORY;
|
|
|
|
// pCookie deleted in destructor of cookieBlock
|
|
}
|
|
else
|
|
hResult = E_OUTOFMEMORY;
|
|
}
|
|
else
|
|
hResult = E_UNEXPECTED;
|
|
}
|
|
else if (cf == m_CFIDList)
|
|
{
|
|
LPITEMIDLIST pidl = NULL, pidlR = NULL;
|
|
|
|
hResult = m_pcookie->GetSharePIDList( &pidl );
|
|
|
|
if (SUCCEEDED(hResult))
|
|
{
|
|
pidlR = ILClone(ILFindLastID(pidl)); // relative IDList
|
|
ILRemoveLastID(pidl); // folder IDList
|
|
|
|
int cidl = 1;
|
|
UINT offset = sizeof(CIDA) + sizeof(UINT)*cidl;
|
|
UINT cbFolder = ILGetSize(pidl);
|
|
UINT cbRelative = ILGetSize(pidlR);
|
|
UINT cbTotal = offset + cbFolder + cbRelative;
|
|
|
|
HGLOBAL hGlobal = ::GlobalAlloc (GPTR, cbTotal);
|
|
if ( hGlobal )
|
|
{
|
|
LPIDA pida = (LPIDA)hGlobal;
|
|
|
|
pida->cidl = cidl;
|
|
pida->aoffset[0] = offset;
|
|
MoveMemory(((LPBYTE)hGlobal+offset), pidl, cbFolder);
|
|
|
|
offset += cbFolder;
|
|
pida->aoffset[1] = offset;
|
|
MoveMemory(((LPBYTE)hGlobal+offset), pidlR, cbRelative);
|
|
|
|
pMedium->hGlobal = hGlobal;
|
|
}
|
|
else
|
|
hResult = E_OUTOFMEMORY;
|
|
|
|
// free pidl & pidlR
|
|
if (pidl)
|
|
ILFree(pidl);
|
|
if (pidlR)
|
|
ILFree(pidlR);
|
|
} else
|
|
hResult = DV_E_FORMATETC;
|
|
} else if (cf == m_CFNodeID2)
|
|
{
|
|
if (m_objecttype != FILEMGMT_ROOT &&
|
|
m_objecttype != FILEMGMT_SHARES &&
|
|
m_objecttype != FILEMGMT_SESSIONS &&
|
|
m_objecttype != FILEMGMT_RESOURCES &&
|
|
m_objecttype != FILEMGMT_SERVICES)
|
|
{
|
|
hResult = DV_E_FORMATETC;
|
|
} else
|
|
{
|
|
const BSTR strGUID = GetObjectTypeString( m_objecttype );
|
|
|
|
// JonN 12/11/01 502856
|
|
int cbString = (lstrlen(strGUID) + 1) * sizeof(TCHAR);
|
|
|
|
pMedium->tymed = TYMED_HGLOBAL;
|
|
pMedium->hGlobal = ::GlobalAlloc(GMEM_SHARE|GMEM_MOVEABLE,
|
|
sizeof(SNodeID2) + cbString);
|
|
if (!(pMedium->hGlobal))
|
|
{
|
|
hResult = E_OUTOFMEMORY;
|
|
} else
|
|
{
|
|
SNodeID2 *pNodeID = (SNodeID2 *)GlobalLock(pMedium->hGlobal);
|
|
if (!pNodeID)
|
|
{
|
|
hResult = HRESULT_FROM_WIN32(GetLastError());
|
|
GlobalFree(pMedium->hGlobal);
|
|
pMedium->hGlobal = NULL;
|
|
} else
|
|
{
|
|
pNodeID->dwFlags = 0;
|
|
pNodeID->cBytes = cbString;
|
|
CopyMemory(pNodeID->id, strGUID, cbString);
|
|
GlobalUnlock(pMedium->hGlobal);
|
|
}
|
|
}
|
|
}
|
|
} else
|
|
hResult = DV_E_FORMATETC; // Invalid/unknown clipboard format
|
|
|
|
return hResult;
|
|
} // CMyComputerDataObject::GetData()
|
|
|
|
|
|
HRESULT CFileMgmtDataObject::PutDisplayName(STGMEDIUM* pMedium)
|
|
// Writes the "friendly name" to the provided storage medium
|
|
// Returns the result of the write operation
|
|
{
|
|
if ( !IsAutonomousObjectType(m_objecttype) )
|
|
{
|
|
ASSERT(FALSE);
|
|
return DV_E_FORMATETC;
|
|
}
|
|
|
|
CString strDisplayName;
|
|
BOOL fStaticNode = ( NULL != m_pComponentData
|
|
&& !m_pComponentData->IsExtensionSnapin()
|
|
&& m_pComponentData->QueryRootCookie().QueryObjectType()
|
|
== m_objecttype);
|
|
// will only succeed for scope cookies
|
|
m_pcookie->GetDisplayName( strDisplayName, fStaticNode );
|
|
// LoadStringPrintf(nStringId, OUT &strDisplayName, (LPCTSTR)m_strMachineName);
|
|
stream_ptr s(pMedium);
|
|
return s.Write(strDisplayName);
|
|
}
|
|
|
|
void CFileMgmtDataObject::FreeMultiSelectObjList()
|
|
{
|
|
if (!m_MultiSelectObjList.empty())
|
|
{
|
|
for (CDataObjectList::iterator i = m_MultiSelectObjList.begin(); i != m_MultiSelectObjList.end(); i++)
|
|
{
|
|
CComObject<CFileMgmtDataObject>* pDataObject = reinterpret_cast<CComObject<CFileMgmtDataObject>*>(*i);
|
|
delete pDataObject;
|
|
}
|
|
|
|
m_MultiSelectObjList.clear();
|
|
}
|
|
}
|
|
|
|
HRESULT CFileMgmtDataObject::InitMultiSelectDataObjects(CFileMgmtComponentData& refComponentData)
|
|
{
|
|
FreeMultiSelectObjList();
|
|
|
|
ASSERT(!m_pComponentData);
|
|
m_pComponentData = &refComponentData;
|
|
((IComponentData*) m_pComponentData)->AddRef ();
|
|
|
|
VERIFY( SUCCEEDED(refComponentData.GetClassID(&m_SnapInCLSID)) );
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CFileMgmtDataObject::AddMultiSelectDataObjects(CFileMgmtCookie* pCookie, DATA_OBJECT_TYPES type)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
CComObject<CFileMgmtDataObject>* pDataObject = NULL;
|
|
hr = CComObject<CFileMgmtDataObject>::CreateInstance(&pDataObject);
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = pDataObject->Initialize(pCookie, *m_pComponentData, type );
|
|
|
|
IDataObject *piDataObject = NULL;
|
|
if (SUCCEEDED(hr))
|
|
hr = pDataObject->QueryInterface(IID_IDataObject, reinterpret_cast<void**>(&piDataObject));
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
m_MultiSelectObjList.push_back(piDataObject);
|
|
} else
|
|
{
|
|
delete pDataObject;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
// Register the clipboard formats
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFSnapinPreloads =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CCF_SNAPIN_PRELOADS);
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFDisplayName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CCF_DISPLAY_NAME);
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFNodeID2 =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CCF_NODEID2);
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFTransport =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_TRANSPORT");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFMachineName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"MMC_SNAPIN_MACHINE_NAME");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFShareName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_SHARE_NAME");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFSessionClientName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_SESSION_CLIENT_NAME");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFSessionUserName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_SESSION_USER_NAME");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFSessionID =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_SESSION_ID");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFFileID =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_FILE_ID");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFServiceName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_SERVICE_NAME");
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFServiceDisplayName =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_SERVICE_DISPLAYNAME");
|
|
CLIPFORMAT CDataObject::m_CFRawCookie =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_RAW_COOKIE");
|
|
|
|
// Additional clipboard formats for the Send Console Message snapin
|
|
CLIPFORMAT CFileMgmtDataObject::m_cfSendConsoleMessageRecipients =
|
|
(CLIPFORMAT)RegisterClipboardFormat(_T("mmc.sendcmsg.MessageRecipients"));
|
|
|
|
// Additional clipboard formats for the Security Page
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFIDList =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CFSTR_SHELLIDLIST);
|
|
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFObjectTypesInMultiSelect =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CCF_OBJECT_TYPES_IN_MULTI_SELECT);
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFMultiSelectDataObject =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CCF_MMC_MULTISELECT_DATAOBJECT);
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFMultiSelectSnapins =
|
|
(CLIPFORMAT)RegisterClipboardFormat(CCF_MULTI_SELECT_SNAPINS);
|
|
|
|
CLIPFORMAT CFileMgmtDataObject::m_CFInternal =
|
|
(CLIPFORMAT)RegisterClipboardFormat(L"FILEMGMT_SNAPIN_INTERNAL");
|
|
|
|
STDMETHODIMP CFileMgmtComponentData::QueryDataObject(MMC_COOKIE cookie, DATA_OBJECT_TYPES type, LPDATAOBJECT* ppDataObject)
|
|
{
|
|
MFC_TRY;
|
|
|
|
// WARNING cookie cast
|
|
CCookie* pbasecookie = reinterpret_cast<CCookie*>(cookie);
|
|
CFileMgmtCookie* pUseThisCookie = ActiveCookie((CFileMgmtCookie*)pbasecookie);
|
|
ASSERT( IsValidObjectType(pUseThisCookie->QueryObjectType()) );
|
|
|
|
CComObject<CFileMgmtDataObject>* pDataObject = NULL;
|
|
HRESULT hRes = CComObject<CFileMgmtDataObject>::CreateInstance(&pDataObject);
|
|
if ( FAILED(hRes) )
|
|
return hRes;
|
|
|
|
HRESULT hr = pDataObject->Initialize( pUseThisCookie, *this, type );
|
|
if ( SUCCEEDED(hr) )
|
|
{
|
|
hr = pDataObject->QueryInterface(IID_IDataObject,
|
|
reinterpret_cast<void**>(ppDataObject));
|
|
}
|
|
if ( FAILED(hr) )
|
|
{
|
|
delete pDataObject;
|
|
return hr;
|
|
}
|
|
|
|
return hr;
|
|
|
|
MFC_CATCH;
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////
|
|
// FileMgmtObjectTypeFromIDataObject()
|
|
//
|
|
// Find the objecttype of a IDataObject pointer. The purpose
|
|
// of this routine is to combine ExtractObjectTypeGUID() and
|
|
// CheckObjectTypeGUID() into a single function.
|
|
//
|
|
// Return the FILEMGMT_* node type (enum FileMgmtObjectType).
|
|
//
|
|
// HISTORY
|
|
// 30-Jul-97 t-danm Creation.
|
|
//
|
|
FileMgmtObjectType
|
|
FileMgmtObjectTypeFromIDataObject(IN LPDATAOBJECT lpDataObject)
|
|
{
|
|
ASSERT(lpDataObject != NULL);
|
|
GUID guidObjectType = GUID_NULL; // JonN 11/21/00 PREFIX 226042
|
|
HRESULT hr = ExtractObjectTypeGUID( IN lpDataObject, OUT &guidObjectType );
|
|
ASSERT( SUCCEEDED(hr) );
|
|
return (FileMgmtObjectType)CheckObjectTypeGUID(IN &guidObjectType );
|
|
} // FileMgmtObjectTypeFromIDataObject()
|
|
|
|
|
|
HRESULT ExtractBaseCookie(
|
|
IN LPDATAOBJECT piDataObject,
|
|
OUT CCookie** ppcookie,
|
|
OUT FileMgmtObjectType* pobjecttype )
|
|
{
|
|
HRESULT hr = ExtractData( piDataObject,
|
|
CFileMgmtDataObject::m_CFRawCookie,
|
|
(PBYTE)ppcookie,
|
|
sizeof(CCookie*) );
|
|
if ( SUCCEEDED(hr) && NULL != pobjecttype )
|
|
{
|
|
*pobjecttype = FileMgmtObjectTypeFromIDataObject(piDataObject);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
BOOL IsMultiSelectObject(LPDATAOBJECT piDataObject)
|
|
{
|
|
BOOL bMultiSelectObject = FALSE;
|
|
|
|
if (piDataObject)
|
|
{
|
|
//
|
|
// return TRUE if piDataObject is the composite data object (MMC_MS_DO) created by MMC.
|
|
//
|
|
STGMEDIUM stgmedium = {TYMED_HGLOBAL, NULL, NULL};
|
|
FORMATETC formatetc = {CFileMgmtDataObject::m_CFMultiSelectDataObject,
|
|
NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL};
|
|
HRESULT hr = piDataObject->GetData(&formatetc, &stgmedium);
|
|
if (S_OK == hr && stgmedium.hGlobal)
|
|
{
|
|
DWORD* pdwData = reinterpret_cast<DWORD*>(::GlobalLock(stgmedium.hGlobal));
|
|
bMultiSelectObject = (1 == *pdwData);
|
|
::GlobalUnlock(stgmedium.hGlobal);
|
|
::GlobalFree(stgmedium.hGlobal);
|
|
}
|
|
}
|
|
|
|
return bMultiSelectObject;
|
|
}
|
|
|