mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1200 lines
28 KiB
1200 lines
28 KiB
//*******************************************************************************************
|
|
//
|
|
// Filename : DataObj.cpp
|
|
//
|
|
// Implementation file for CObjFormats and CCabObj
|
|
//
|
|
// Copyright (c) 1994 - 1996 Microsoft Corporation. All rights reserved
|
|
//
|
|
//*******************************************************************************************
|
|
|
|
#include "pch.h"
|
|
#include "ccstock.h"
|
|
#include "thisdll.h"
|
|
#include "folder.h"
|
|
#include "dataobj.h"
|
|
|
|
#include "cabitms.h"
|
|
|
|
UINT CCabObj::s_uFileGroupDesc = 0;
|
|
UINT CCabObj::s_uFileContents = 0;
|
|
UINT CCabObj::s_uPersistedDataObject = 0;
|
|
UINT CCabObj::s_uHIDA = 0;
|
|
|
|
// {dfe49cfe-cd09-11d2-9643-00c04f79adf0}
|
|
const GUID CLSID_CabViewDataObject = {0xdfe49cfe, 0xcd09, 0x11d2, 0x96, 0x43, 0x00, 0xc0, 0x4f, 0x79, 0xad, 0xf0};
|
|
|
|
#define MAX_CHUNK (60*1024) /* max mouthful to CopyTo from CCabStream */
|
|
|
|
class CCabStream : public IStream
|
|
{
|
|
public:
|
|
CCabStream(HGLOBAL hStream,DWORD dwStreamLength);
|
|
~CCabStream(void);
|
|
|
|
// *** IUnknown methods ***
|
|
STDMETHODIMP QueryInterface(REFIID riid, LPVOID * ppvObj);
|
|
STDMETHODIMP_(ULONG) AddRef();
|
|
STDMETHODIMP_(ULONG) Release();
|
|
|
|
// *** IStream methods ***
|
|
STDMETHODIMP Read(void *pv,ULONG cb,ULONG *pcbRead);
|
|
STDMETHODIMP Write(const void *pv,ULONG cb,ULONG *pcbWritten);
|
|
STDMETHODIMP Seek(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER * plibNewPosition);
|
|
STDMETHODIMP SetSize(ULARGE_INTEGER libNewSize);
|
|
STDMETHODIMP CopyTo(IStream * pstm, ULARGE_INTEGER cb, ULARGE_INTEGER * pcbRead, ULARGE_INTEGER * pcbWritten);
|
|
STDMETHODIMP Commit(DWORD grfCommitFlags);
|
|
STDMETHODIMP Revert(void);
|
|
STDMETHODIMP LockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
|
|
STDMETHODIMP UnlockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
|
|
STDMETHODIMP Stat(STATSTG * pstatstg, DWORD grfStatFlag);
|
|
STDMETHODIMP Clone(IStream ** ppstm);
|
|
|
|
private:
|
|
CRefCount m_cRef;
|
|
CRefDll m_cRefDll;
|
|
HGLOBAL m_hStream;
|
|
DWORD m_dwStreamLength;
|
|
DWORD m_dwStreamPosition;
|
|
};
|
|
|
|
CCabStream::CCabStream(HGLOBAL hStream,DWORD dwStreamLength)
|
|
{
|
|
m_hStream = hStream;
|
|
m_dwStreamLength = dwStreamLength;
|
|
m_dwStreamPosition = 0;
|
|
|
|
AddRef();
|
|
}
|
|
|
|
CCabStream::~CCabStream(void)
|
|
{
|
|
GlobalFree(m_hStream);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::QueryInterface(
|
|
REFIID riid,
|
|
LPVOID FAR* ppvObj)
|
|
{
|
|
*ppvObj = NULL;
|
|
|
|
LPUNKNOWN pObj;
|
|
|
|
if (riid == IID_IUnknown)
|
|
{
|
|
pObj = (IUnknown*)((IStream *)this);
|
|
}
|
|
else if (riid == IID_IStream)
|
|
{
|
|
pObj = (IUnknown*)((IStream *)this);
|
|
}
|
|
else
|
|
{
|
|
return(E_NOINTERFACE);
|
|
}
|
|
|
|
pObj->AddRef();
|
|
*ppvObj = pObj;
|
|
|
|
return(NOERROR);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CCabStream::AddRef(void)
|
|
{
|
|
return(m_cRef.AddRef());
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CCabStream::Release(void)
|
|
{
|
|
if (!m_cRef.Release())
|
|
{
|
|
delete this;
|
|
return(0);
|
|
}
|
|
|
|
return(m_cRef.GetRef());
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Read(void * pv, ULONG cb, ULONG * pcbRead)
|
|
{
|
|
*pcbRead = 0;
|
|
|
|
if (m_dwStreamPosition < m_dwStreamLength)
|
|
{
|
|
if (cb > (m_dwStreamLength - m_dwStreamPosition))
|
|
{
|
|
*pcbRead = (m_dwStreamLength - m_dwStreamPosition);
|
|
}
|
|
else
|
|
{
|
|
*pcbRead = cb;
|
|
}
|
|
|
|
CopyMemory(pv,(char *) m_hStream + m_dwStreamPosition,*pcbRead);
|
|
m_dwStreamPosition += *pcbRead;
|
|
}
|
|
|
|
return(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Write(const void * pv, ULONG cb, ULONG * pcbWritten)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Seek(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER * plibNewPosition)
|
|
{
|
|
switch (dwOrigin)
|
|
{
|
|
case STREAM_SEEK_SET:
|
|
m_dwStreamPosition = dlibMove.LowPart;
|
|
break;
|
|
|
|
case STREAM_SEEK_CUR:
|
|
m_dwStreamPosition += dlibMove.LowPart;
|
|
break;
|
|
|
|
case STREAM_SEEK_END:
|
|
m_dwStreamPosition = m_dwStreamLength + dlibMove.LowPart;
|
|
break;
|
|
|
|
default:
|
|
return(STG_E_INVALIDFUNCTION);
|
|
}
|
|
|
|
if (plibNewPosition)
|
|
{
|
|
(*plibNewPosition).LowPart = m_dwStreamPosition;
|
|
(*plibNewPosition).HighPart = 0;
|
|
}
|
|
|
|
return(S_OK);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::SetSize(ULARGE_INTEGER libNewSize)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::CopyTo(IStream * pstm, ULARGE_INTEGER cb, ULARGE_INTEGER * pcbRead, ULARGE_INTEGER * pcbWritten)
|
|
{
|
|
HRESULT hRes;
|
|
unsigned long cbActual = cb.LowPart;
|
|
unsigned long cbWritten;
|
|
unsigned long cbChunk;
|
|
|
|
if (pcbRead)
|
|
{
|
|
(*pcbRead).LowPart = 0;
|
|
(*pcbRead).HighPart = 0;
|
|
}
|
|
|
|
if (pcbWritten)
|
|
{
|
|
(*pcbWritten).LowPart = 0;
|
|
(*pcbWritten).HighPart = 0;
|
|
}
|
|
|
|
hRes = S_OK;
|
|
|
|
if (m_dwStreamPosition < m_dwStreamLength)
|
|
{
|
|
if (cbActual > (m_dwStreamLength - m_dwStreamPosition))
|
|
{
|
|
cbActual = (m_dwStreamLength - m_dwStreamPosition);
|
|
}
|
|
|
|
while (cbActual)
|
|
{
|
|
if (cbActual > MAX_CHUNK)
|
|
{
|
|
cbChunk = MAX_CHUNK;
|
|
}
|
|
else
|
|
{
|
|
cbChunk = cbActual;
|
|
}
|
|
|
|
hRes = pstm->Write((char *) m_hStream + m_dwStreamPosition,cbChunk,&cbWritten);
|
|
if (FAILED(hRes))
|
|
{
|
|
break;
|
|
}
|
|
|
|
m_dwStreamPosition += cbChunk;
|
|
|
|
if (pcbRead)
|
|
{
|
|
(*pcbRead).LowPart += cbChunk;
|
|
(*pcbRead).HighPart = 0;
|
|
}
|
|
|
|
if (pcbWritten)
|
|
{
|
|
(*pcbWritten).LowPart += cbWritten;
|
|
(*pcbWritten).HighPart = 0;
|
|
}
|
|
|
|
cbActual -= cbChunk;
|
|
}
|
|
}
|
|
|
|
return(hRes);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Commit(DWORD grfCommitFlags)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Revert(void)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::LockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::UnlockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Stat(STATSTG * pstatstg, DWORD grfStatFlag)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
STDMETHODIMP CCabStream::Clone(IStream ** ppstm)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
|
|
class CObjFormats : public IEnumFORMATETC
|
|
{
|
|
public:
|
|
CObjFormats(UINT cfmt, const FORMATETC afmt[]);
|
|
~CObjFormats();
|
|
|
|
// *** IUnknown methods ***
|
|
STDMETHODIMP QueryInterface(REFIID riid, LPVOID * ppvObj);
|
|
STDMETHODIMP_(ULONG) AddRef();
|
|
STDMETHODIMP_(ULONG) Release();
|
|
|
|
// *** IEnumFORMATETC methods ***
|
|
STDMETHODIMP Next(ULONG celt, FORMATETC *rgelt, ULONG *pceltFethed);
|
|
STDMETHODIMP Skip(ULONG celt);
|
|
STDMETHODIMP Reset();
|
|
STDMETHODIMP Clone(IEnumFORMATETC ** ppenum);
|
|
|
|
private:
|
|
CRefCount m_cRef;
|
|
|
|
CRefDll m_cRefDll;
|
|
|
|
UINT m_iFmt;
|
|
UINT m_cFmt;
|
|
FORMATETC *m_aFmt;
|
|
} ;
|
|
|
|
|
|
CObjFormats::CObjFormats(UINT cfmt, const FORMATETC afmt[])
|
|
{
|
|
m_iFmt = 0;
|
|
m_cFmt = cfmt;
|
|
m_aFmt = new FORMATETC[cfmt];
|
|
|
|
if (m_aFmt)
|
|
{
|
|
CopyMemory(m_aFmt, afmt, cfmt*sizeof(afmt[0]));
|
|
}
|
|
}
|
|
|
|
|
|
CObjFormats::~CObjFormats()
|
|
{
|
|
if (m_aFmt)
|
|
{
|
|
delete m_aFmt;
|
|
}
|
|
}
|
|
|
|
|
|
// *** IUnknown methods ***
|
|
STDMETHODIMP CObjFormats::QueryInterface(
|
|
REFIID riid,
|
|
LPVOID FAR* ppvObj)
|
|
{
|
|
*ppvObj = NULL;
|
|
|
|
if (!m_aFmt)
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
LPUNKNOWN pObj;
|
|
|
|
if (riid == IID_IUnknown)
|
|
{
|
|
pObj = (IUnknown*)((IEnumFORMATETC*)this);
|
|
}
|
|
else if (riid == IID_IEnumFORMATETC)
|
|
{
|
|
pObj = (IUnknown*)((IEnumFORMATETC*)this);
|
|
}
|
|
else
|
|
{
|
|
return(E_NOINTERFACE);
|
|
}
|
|
|
|
pObj->AddRef();
|
|
*ppvObj = pObj;
|
|
|
|
return(NOERROR);
|
|
}
|
|
|
|
|
|
STDMETHODIMP_(ULONG) CObjFormats::AddRef(void)
|
|
{
|
|
return(m_cRef.AddRef());
|
|
}
|
|
|
|
|
|
STDMETHODIMP_(ULONG) CObjFormats::Release(void)
|
|
{
|
|
if (!m_cRef.Release())
|
|
{
|
|
delete this;
|
|
return(0);
|
|
}
|
|
|
|
return(m_cRef.GetRef());
|
|
}
|
|
|
|
|
|
STDMETHODIMP CObjFormats::Next(ULONG celt, FORMATETC *rgelt, ULONG *pceltFethed)
|
|
{
|
|
UINT cfetch;
|
|
HRESULT hres = S_FALSE; // assume less numbers
|
|
|
|
if (m_iFmt < m_cFmt)
|
|
{
|
|
cfetch = m_cFmt - m_iFmt;
|
|
if (cfetch >= celt)
|
|
{
|
|
cfetch = celt;
|
|
hres = S_OK;
|
|
}
|
|
|
|
CopyMemory(rgelt, &m_aFmt[m_iFmt], cfetch * sizeof(FORMATETC));
|
|
m_iFmt += cfetch;
|
|
}
|
|
else
|
|
{
|
|
cfetch = 0;
|
|
}
|
|
|
|
if (pceltFethed)
|
|
{
|
|
*pceltFethed = cfetch;
|
|
}
|
|
|
|
return hres;
|
|
}
|
|
|
|
STDMETHODIMP CObjFormats::Skip(ULONG celt)
|
|
{
|
|
m_iFmt += celt;
|
|
if (m_iFmt > m_cFmt)
|
|
{
|
|
m_iFmt = m_cFmt;
|
|
return S_FALSE;
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CObjFormats::Reset()
|
|
{
|
|
m_iFmt = 0;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CObjFormats::Clone(IEnumFORMATETC ** ppenum)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
HRESULT CabViewDataObject_CreateInstance(REFIID riid, LPVOID *ppvObj)
|
|
{
|
|
HRESULT hr;
|
|
CCabObj* pco = new CCabObj();
|
|
if (NULL != pco)
|
|
{
|
|
pco->AddRef();
|
|
hr = pco->QueryInterface(riid, ppvObj);
|
|
pco->Release();
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
CCabObj::CCabObj(HWND hwndOwner, CCabFolder *pcf, LPCABITEM *apit, UINT cpit)
|
|
: m_lSel(8), m_lContents(NULL)
|
|
{
|
|
m_pcfHere = pcf;
|
|
pcf->AddRef();
|
|
|
|
m_hwndOwner = hwndOwner;
|
|
m_lSel.AddItems(apit, cpit);
|
|
}
|
|
|
|
// constructor used when we're co-created:
|
|
CCabObj::CCabObj() : m_pcfHere(NULL), m_hwndOwner(NULL), m_lSel(8), m_lContents(NULL)
|
|
{
|
|
};
|
|
|
|
CCabObj::~CCabObj()
|
|
{
|
|
if (m_lContents != NULL)
|
|
{
|
|
int cItems = m_lSel.GetCount();
|
|
|
|
while (cItems--)
|
|
{
|
|
if (m_lContents[cItems] != NULL)
|
|
{
|
|
GlobalFree(m_lContents[cItems]);
|
|
m_lContents[cItems] = NULL;
|
|
}
|
|
}
|
|
|
|
GlobalFree(m_lContents);
|
|
m_lContents = NULL;
|
|
}
|
|
|
|
if (m_pcfHere)
|
|
{
|
|
m_pcfHere->Release();
|
|
}
|
|
}
|
|
|
|
|
|
// *** IUnknown methods ***
|
|
STDMETHODIMP CCabObj::QueryInterface(
|
|
REFIID riid,
|
|
LPVOID FAR* ppvObj)
|
|
{
|
|
*ppvObj = NULL;
|
|
|
|
if (m_lSel.GetState() == CCabItemList::State_OutOfMem)
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
LPUNKNOWN pObj;
|
|
|
|
if (riid == IID_IUnknown)
|
|
{
|
|
pObj = (IUnknown*)((IDataObject*)this);
|
|
}
|
|
else if (riid == IID_IDataObject)
|
|
{
|
|
pObj = (IUnknown*)((IDataObject*)this);
|
|
}
|
|
else if (riid == IID_IPersist)
|
|
{
|
|
pObj = (IUnknown*)((IPersist*)this);
|
|
}
|
|
else if (riid == IID_IPersistStream)
|
|
{
|
|
pObj = (IUnknown*)((IPersistStream*)this);
|
|
}
|
|
else
|
|
{
|
|
return(E_NOINTERFACE);
|
|
}
|
|
|
|
pObj->AddRef();
|
|
*ppvObj = pObj;
|
|
|
|
return(NOERROR);
|
|
}
|
|
|
|
|
|
STDMETHODIMP_(ULONG) CCabObj::AddRef(void)
|
|
{
|
|
return(m_cRef.AddRef());
|
|
}
|
|
|
|
|
|
STDMETHODIMP_(ULONG) CCabObj::Release(void)
|
|
{
|
|
if (!m_cRef.Release())
|
|
{
|
|
delete this;
|
|
return(0);
|
|
}
|
|
|
|
return(m_cRef.GetRef());
|
|
}
|
|
|
|
/////////////////////////////////
|
|
////// IPersistStream Impl
|
|
/////////////////////////////////
|
|
|
|
|
|
typedef struct
|
|
{
|
|
DWORD dwVersion;
|
|
DWORD dwExtraSize; // After pidl list
|
|
DWORD dwReserved1;
|
|
DWORD dwReserved2;
|
|
} CABVIEWDATAOBJ_PERSISTSTRUCT;
|
|
|
|
|
|
/*****************************************************************************\
|
|
FUNCTION: IPersistStream::Load
|
|
|
|
DESCRIPTION:
|
|
See IPersistStream::Save() for the layout of the stream.
|
|
\*****************************************************************************/
|
|
HRESULT CCabObj::Load(IStream *pStm)
|
|
{
|
|
HRESULT hr = E_INVALIDARG;
|
|
|
|
if (pStm)
|
|
{
|
|
CABVIEWDATAOBJ_PERSISTSTRUCT cvdops;
|
|
DWORD dwNumPidls;
|
|
|
|
hr = pStm->Read(&cvdops, SIZEOF(cvdops), NULL); // #1
|
|
// If we rev the version, read it now (cvdops.dwVersion)
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
LPITEMIDLIST pidl = NULL; // ILLoadFromStream frees the param
|
|
|
|
// ASSERT(!m_pff);
|
|
|
|
hr = ILLoadFromStream(pStm, &pidl); // #2
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IShellFolder* psf;
|
|
hr = SHGetDesktopFolder(&psf);
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (m_pcfHere)
|
|
{
|
|
m_pcfHere->Release();
|
|
m_pcfHere = NULL;
|
|
}
|
|
hr = psf->BindToObject(pidl, NULL, CLSID_CabFolder, (void **)&m_pcfHere);
|
|
psf->Release();
|
|
}
|
|
|
|
ILFree(pidl);
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pStm->Read(&dwNumPidls, SIZEOF(dwNumPidls), NULL); // #3
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
for (int nIndex = 0; (nIndex < (int)dwNumPidls) && SUCCEEDED(hr); nIndex++)
|
|
{
|
|
LPITEMIDLIST pidl = NULL; // ILLoadFromStream frees the param
|
|
|
|
hr = ILLoadFromStream(pStm, &pidl); // #4
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = m_lSel.AddItems((LPCABITEM*) &pidl, 1);
|
|
ILFree(pidl);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// We may be reading a version newer than us, so skip their data.
|
|
if (0 != cvdops.dwExtraSize)
|
|
{
|
|
LARGE_INTEGER li = {0};
|
|
|
|
li.LowPart = cvdops.dwExtraSize;
|
|
hr = pStm->Seek(li, STREAM_SEEK_CUR, NULL);
|
|
}
|
|
}
|
|
}
|
|
|
|
// don't return success codes other than S_OK:
|
|
return SUCCEEDED(hr) ? S_OK : hr;
|
|
}
|
|
|
|
|
|
/*****************************************************************************\
|
|
FUNCTION: IPersistStream::Save
|
|
|
|
DESCRIPTION:
|
|
The stream will be layed out in the following way:
|
|
|
|
Version 1:
|
|
1. CABVIEWDATAOBJ_PERSISTSTRUCT - Constant sized data.
|
|
<PidlList BEGIN>
|
|
2. PIDL pidl - Pidl for m_pcfHere. It will be a public pidl (fully qualified
|
|
from the shell root)
|
|
3. DWORD dwNumPidls - Number of pidls coming.
|
|
4. PIDL pidl(n) - Pidl in slot (n) of m_lSel
|
|
<PidlList END>
|
|
\*****************************************************************************/
|
|
HRESULT CCabObj::Save(IStream *pStm, BOOL fClearDirty)
|
|
{
|
|
HRESULT hr = E_INVALIDARG;
|
|
|
|
if (pStm)
|
|
{
|
|
CABVIEWDATAOBJ_PERSISTSTRUCT cvdops = {0};
|
|
DWORD dwNumPidls = m_lSel.GetCount();
|
|
|
|
cvdops.dwVersion = 1;
|
|
hr = pStm->Write(&cvdops, SIZEOF(cvdops), NULL); // #1
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (m_pcfHere)
|
|
{
|
|
LPITEMIDLIST pidl;
|
|
hr = m_pcfHere->GetCurFolder(&pidl);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = ILSaveToStream(pStm, pidl); // #2
|
|
ILFree(pidl);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = E_FAIL;
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
hr = pStm->Write(&dwNumPidls, SIZEOF(dwNumPidls), NULL); // #3
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
for (int nIndex = 0; (nIndex < (int)dwNumPidls) && SUCCEEDED(hr); nIndex++)
|
|
{
|
|
hr = ILSaveToStream(pStm, (LPCITEMIDLIST) m_lSel[nIndex]); // #4
|
|
}
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
#define MAX_STREAM_SIZE (500 * 1024) // 500k
|
|
/*****************************************************************************\
|
|
FUNCTION: IPersistStream::GetSizeMax
|
|
|
|
DESCRIPTION:
|
|
Now this is tough. I can't calculate the real value because I don't know
|
|
how big the hglobals are going to be for the user provided data. I will
|
|
assume everything fits in
|
|
\*****************************************************************************/
|
|
HRESULT CCabObj::GetSizeMax(ULARGE_INTEGER * pcbSize)
|
|
{
|
|
if (pcbSize)
|
|
{
|
|
pcbSize->HighPart = 0;
|
|
pcbSize->LowPart = MAX_STREAM_SIZE;
|
|
}
|
|
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
|
|
/*****************************************************************************\
|
|
DESCRIPTION:
|
|
Allocate an hglobal of the indicated size, initialized from the
|
|
specified buffer.
|
|
\*****************************************************************************/
|
|
HRESULT StgMediumWriteHGlobal(HGLOBAL *phglob, LPVOID pv, SIZE_T cb)
|
|
{
|
|
HRESULT hres = E_OUTOFMEMORY;
|
|
|
|
*phglob = 0; // Rules are rules
|
|
if (cb)
|
|
{
|
|
*phglob = (HGLOBAL) LocalAlloc(LPTR, cb);
|
|
if (*phglob)
|
|
{
|
|
hres = S_OK;
|
|
CopyMemory(*phglob, pv, cb);
|
|
}
|
|
}
|
|
else
|
|
hres = E_INVALIDARG; // Can't clone a discardable block
|
|
|
|
return hres;
|
|
}
|
|
|
|
|
|
/*****************************************************************************\
|
|
DESCRIPTION:
|
|
When the copy source goes away (the process shuts down), it calls
|
|
OleFlushClipboard. OLE will then copy our data, release us, and then
|
|
give out our data later. This works for most things except for:
|
|
1. When lindex needs to very. This doesn't work because ole doesn't know
|
|
how to ask us how may lindexs they need to copy.
|
|
2. If this object has a private interface OLE doesn't know about. For us,
|
|
it's IAsyncOperation.
|
|
|
|
To get around this problem, we want OLE to recreate us when some possible
|
|
paste target calls OleGetClipboard. We want OLE to call OleLoadFromStream()
|
|
to have us CoCreated and reload our persisted data via IPersistStream.
|
|
OLE doesn't want to do this by default or they may have backward compat
|
|
problems so they want a sign from the heavens, or at least from us, that
|
|
we will work. They ping our "OleClipboardPersistOnFlush" clipboard format
|
|
to ask this.
|
|
\*****************************************************************************/
|
|
HRESULT _RenderOlePersist(STGMEDIUM * pStgMedium)
|
|
{
|
|
// The actual cookie value is opaque to the outside world. Since
|
|
// we don't use it either, we just leave it at zero in case we use
|
|
// it in the future. It's mere existence will cause OLE to do the
|
|
// use our IPersistStream, which is what we want.
|
|
DWORD dwCookie = 0;
|
|
return StgMediumWriteHGlobal(&pStgMedium->hGlobal, &dwCookie, sizeof(dwCookie));
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::GetData(FORMATETC *pfmt, STGMEDIUM *pmedium)
|
|
{
|
|
ZeroMemory(pmedium, sizeof(*pmedium));
|
|
|
|
if (!InitFileGroupDesc())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uFileGroupDesc)
|
|
{
|
|
if (pfmt->ptd==NULL && (pfmt->dwAspect&DVASPECT_CONTENT) && pfmt->lindex==-1
|
|
&& (pfmt->tymed&TYMED_HGLOBAL))
|
|
{
|
|
int cItems = m_lSel.GetCount();
|
|
if (cItems < 1)
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
FILEGROUPDESCRIPTOR *pfgd = (FILEGROUPDESCRIPTOR *)GlobalAlloc(GMEM_FIXED,
|
|
sizeof(FILEGROUPDESCRIPTOR) + (cItems-1)*sizeof(FILEDESCRIPTOR));
|
|
if (!pfgd)
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
pfgd->cItems = cItems;
|
|
for (--cItems; cItems>=0; --cItems)
|
|
{
|
|
LPCABITEM pItem = m_lSel[cItems];
|
|
FILETIME ft;
|
|
|
|
pfgd->fgd[cItems].dwFlags = FD_ATTRIBUTES|FD_WRITESTIME|FD_FILESIZE|FD_PROGRESSUI;
|
|
pfgd->fgd[cItems].dwFileAttributes = pItem->uFileAttribs;
|
|
DosDateTimeToFileTime(pItem->uFileDate, pItem->uFileTime,&ft);
|
|
LocalFileTimeToFileTime(&ft, &pfgd->fgd[cItems].ftLastWriteTime);
|
|
pfgd->fgd[cItems].nFileSizeHigh = 0;
|
|
pfgd->fgd[cItems].nFileSizeLow = pItem->dwFileSize;
|
|
|
|
LPCWSTR pszName;
|
|
WSTR_ALIGNED_STACK_COPY(&pszName, pItem->szName);
|
|
|
|
lstrcpyn(pfgd->fgd[cItems].cFileName, PathFindFileName(pszName),
|
|
ARRAYSIZE(pfgd->fgd[cItems].cFileName));
|
|
}
|
|
|
|
pmedium->tymed = TYMED_HGLOBAL;
|
|
pmedium->hGlobal = (HGLOBAL)pfgd;
|
|
pmedium->pUnkForRelease = NULL;
|
|
|
|
return(NOERROR);
|
|
}
|
|
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
if (!InitFileContents())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uFileContents)
|
|
{
|
|
if (pfmt->ptd==NULL && (pfmt->dwAspect&DVASPECT_CONTENT))
|
|
{
|
|
if (pfmt->tymed & TYMED_ISTREAM)
|
|
{
|
|
int cItems = m_lSel.GetCount();
|
|
|
|
if ((-1 == pfmt->lindex) || (pfmt->lindex >= cItems))
|
|
{
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
HRESULT hRes = InitContents();
|
|
if (FAILED(hRes))
|
|
{
|
|
return(hRes);
|
|
}
|
|
|
|
if (!m_lContents[pfmt->lindex])
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
CCabStream *stream = new CCabStream(m_lContents[pfmt->lindex],m_lSel[pfmt->lindex]->dwFileSize);
|
|
if (!stream)
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
pmedium->tymed = TYMED_ISTREAM;
|
|
pmedium->pstm = stream;
|
|
pmedium->pUnkForRelease = NULL;
|
|
|
|
m_lContents[pfmt->lindex] = NULL;
|
|
|
|
return(NOERROR);
|
|
}
|
|
|
|
if (pfmt->tymed&TYMED_HGLOBAL)
|
|
{
|
|
int cItems = m_lSel.GetCount();
|
|
if (pfmt->lindex >= cItems)
|
|
{
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
HRESULT hRes = InitContents();
|
|
if (FAILED(hRes))
|
|
{
|
|
return(hRes);
|
|
}
|
|
|
|
if (!m_lContents[pfmt->lindex])
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
pmedium->tymed = TYMED_HGLOBAL;
|
|
pmedium->hGlobal = m_lContents[pfmt->lindex];
|
|
pmedium->pUnkForRelease = NULL;
|
|
|
|
m_lContents[pfmt->lindex] = NULL;
|
|
|
|
return(NOERROR);
|
|
}
|
|
}
|
|
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
if (!InitPersistedDataObject())
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uPersistedDataObject)
|
|
{
|
|
if ((pfmt->ptd == NULL) &&
|
|
(pfmt->dwAspect & DVASPECT_CONTENT) &&
|
|
(pfmt->lindex == -1) &&
|
|
(pfmt->tymed & TYMED_HGLOBAL))
|
|
{
|
|
return _RenderOlePersist(pmedium);
|
|
}
|
|
else
|
|
{
|
|
return E_INVALIDARG;
|
|
}
|
|
}
|
|
|
|
if (!InitHIDA())
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uHIDA)
|
|
{
|
|
HRESULT hr = E_INVALIDARG;
|
|
if ((pfmt->ptd == NULL) &&
|
|
(pfmt->dwAspect & DVASPECT_CONTENT) &&
|
|
(pfmt->lindex == -1) &&
|
|
(pfmt->tymed & TYMED_HGLOBAL))
|
|
{
|
|
LPITEMIDLIST pidl;
|
|
hr = m_pcfHere->GetCurFolder(&pidl);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
HIDA hida = HIDA_Create(pidl, m_lSel.GetCount(), (LPCITEMIDLIST*)m_lSel.GetArray());
|
|
if (hida)
|
|
{
|
|
pmedium->tymed = TYMED_HGLOBAL;
|
|
pmedium->hGlobal = hida;
|
|
hr = S_OK;
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
ILFree(pidl);
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::GetDataHere(FORMATETC *pformatetc, STGMEDIUM *pmedium)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::QueryGetData(FORMATETC *pfmt)
|
|
{
|
|
if (!InitFileGroupDesc())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uFileGroupDesc)
|
|
{
|
|
if (pfmt->ptd==NULL && (pfmt->dwAspect&DVASPECT_CONTENT) && pfmt->lindex==-1
|
|
&& (pfmt->tymed&TYMED_HGLOBAL))
|
|
{
|
|
return(S_OK);
|
|
}
|
|
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
if (!InitFileContents())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uFileContents)
|
|
{
|
|
if (pfmt->ptd==NULL && (pfmt->dwAspect&DVASPECT_CONTENT)
|
|
&& (pfmt->tymed & (TYMED_ISTREAM | TYMED_HGLOBAL)))
|
|
{
|
|
return(S_OK);
|
|
}
|
|
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
if (!InitPersistedDataObject())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uPersistedDataObject)
|
|
{
|
|
if (pfmt->ptd == NULL && (pfmt->dwAspect & DVASPECT_CONTENT)
|
|
&& (pfmt->tymed & TYMED_HGLOBAL))
|
|
{
|
|
return(S_OK);
|
|
}
|
|
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
if (!InitHIDA())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
if (pfmt->cfFormat == s_uHIDA)
|
|
{
|
|
if (pfmt->ptd == NULL && (pfmt->dwAspect & DVASPECT_CONTENT)
|
|
&& (pfmt->tymed & TYMED_HGLOBAL))
|
|
{
|
|
return(S_OK);
|
|
}
|
|
|
|
return(E_INVALIDARG);
|
|
}
|
|
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::GetCanonicalFormatEtc(FORMATETC *pformatetcIn, FORMATETC *pformatetcOut)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::SetData(FORMATETC *pformatetc, STGMEDIUM *pmedium, BOOL fRelease)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC **ppenumFormatEtc)
|
|
{
|
|
if (!InitFileGroupDesc() || !InitFileContents() || !InitPersistedDataObject() || !InitHIDA())
|
|
{
|
|
return(E_UNEXPECTED);
|
|
}
|
|
|
|
FORMATETC fmte[] = {
|
|
{(USHORT)s_uFileContents, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL|TYMED_ISTREAM },
|
|
{(USHORT)s_uFileGroupDesc, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL },
|
|
{(USHORT)s_uPersistedDataObject, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL },
|
|
{(USHORT)s_uHIDA, NULL, DVASPECT_CONTENT, -1, TYMED_HGLOBAL },
|
|
};
|
|
|
|
CObjFormats *pFmts = new CObjFormats(ARRAYSIZE(fmte), fmte);
|
|
if (!pFmts)
|
|
{
|
|
return(E_OUTOFMEMORY);
|
|
}
|
|
|
|
pFmts->AddRef();
|
|
HRESULT hRes = pFmts->QueryInterface(IID_IEnumFORMATETC, (LPVOID *)ppenumFormatEtc);
|
|
pFmts->Release();
|
|
|
|
return(hRes);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::DAdvise(FORMATETC *pformatetc, DWORD advf, IAdviseSink *pAdvSink,
|
|
DWORD *pdwConnection)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::DUnadvise(DWORD dwConnection)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
STDMETHODIMP CCabObj::EnumDAdvise(IEnumSTATDATA **ppenumAdvise)
|
|
{
|
|
return(E_NOTIMPL);
|
|
}
|
|
|
|
|
|
BOOL CCabObj::InitFileGroupDesc()
|
|
{
|
|
if (s_uFileGroupDesc)
|
|
{
|
|
return(TRUE);
|
|
}
|
|
|
|
s_uFileGroupDesc = RegisterClipboardFormat(CFSTR_FILEDESCRIPTOR);
|
|
return(s_uFileGroupDesc != 0);
|
|
}
|
|
|
|
|
|
BOOL CCabObj::InitFileContents()
|
|
{
|
|
if (s_uFileContents)
|
|
{
|
|
return(TRUE);
|
|
}
|
|
|
|
s_uFileContents = RegisterClipboardFormat(CFSTR_FILECONTENTS);
|
|
return(s_uFileContents != 0);
|
|
}
|
|
|
|
#define CFSTR_OLECLIPBOARDPERSISTONFLUSH TEXT("OleClipboardPersistOnFlush")
|
|
BOOL CCabObj::InitPersistedDataObject()
|
|
{
|
|
if (s_uPersistedDataObject)
|
|
{
|
|
return(TRUE);
|
|
}
|
|
|
|
s_uPersistedDataObject = RegisterClipboardFormat(CFSTR_OLECLIPBOARDPERSISTONFLUSH);
|
|
return(s_uPersistedDataObject != 0);
|
|
}
|
|
|
|
BOOL CCabObj::InitHIDA()
|
|
{
|
|
if (s_uHIDA)
|
|
{
|
|
return(TRUE);
|
|
}
|
|
|
|
s_uHIDA = RegisterClipboardFormat(CFSTR_SHELLIDLIST);
|
|
return(s_uHIDA != 0);
|
|
}
|
|
|
|
HGLOBAL * CALLBACK CCabObj::ShouldExtract(LPCTSTR pszFile, DWORD dwSize, UINT date,
|
|
UINT time, UINT attribs, LPARAM lParam)
|
|
{
|
|
CCabObj *pThis = (CCabObj*)lParam;
|
|
|
|
int iItem = pThis->m_lSel.FindInList(pszFile, dwSize, date, time, attribs);
|
|
if (iItem < 0)
|
|
{
|
|
return(EXTRACT_FALSE);
|
|
}
|
|
|
|
// Copy nothing for now
|
|
return(&(pThis->m_lContents[iItem]));
|
|
}
|
|
|
|
|
|
HRESULT CCabObj::InitContents()
|
|
{
|
|
if (m_lContents)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
int iCount = m_lSel.GetCount();
|
|
if (iCount < 1)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
m_lContents = (HGLOBAL *)GlobalAlloc(GMEM_FIXED|GMEM_ZEROINIT,
|
|
sizeof(HGLOBAL)*m_lSel.GetCount());
|
|
if (!m_lContents)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
TCHAR szHere[MAX_PATH];
|
|
if ((NULL == m_pcfHere) || (!m_pcfHere->GetPath(szHere)))
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
CCabExtract ceHere(szHere);
|
|
|
|
ceHere.ExtractItems(m_hwndOwner, DIR_MEM, ShouldExtract, (LPARAM)this);
|
|
|
|
return S_OK;
|
|
}
|