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.
2151 lines
63 KiB
2151 lines
63 KiB
// need IProgressDialog to be marshaled, def in .idl file (cleanup needed here)
|
|
// get CLSID_MediaDevMgr set to "Both"
|
|
|
|
// issues:
|
|
// can't drop on open folder, Insert() throws an exception
|
|
// Rio600 Next throws exception on cidl > 1
|
|
// Storage2::GetStroage() fails for Rio600, does SP need to implement this?
|
|
|
|
#include "pch.h"
|
|
#include <shlobj.h>
|
|
#include <shsemip.h>
|
|
#include <shlobjp.h> // IDelegateFolder
|
|
#include "thisdll.h"
|
|
#include <windowsx.h>
|
|
#include <varutil.h>
|
|
#include <dobjutil.h>
|
|
#include <malloc.h>
|
|
#include <dpa.h>
|
|
#include <shdguid.h>
|
|
#include "shdup.h"
|
|
#include "ids.h"
|
|
#include <ccstock2.h>
|
|
#pragma hdrstop
|
|
#include <wmsdk.h>
|
|
#include <mswmdm.h>
|
|
#include <scclient.h>
|
|
#include <mswmdm_i.c> // get CLSID/IID defs
|
|
|
|
#include <ntquery.h>
|
|
|
|
#define DEFINE_SCID(name, fmtid, pid) const SHCOLUMNID name = { fmtid, pid }
|
|
#define IsEqualSCID(a, b) (((a).pid == (b).pid) && IsEqualIID((a).fmtid, (b).fmtid) )
|
|
|
|
DEFINE_SCID(SCID_SIZE, PSGUID_STORAGE, PID_STG_SIZE);
|
|
DEFINE_SCID(SCID_DESCRIPTIONID, PSGUID_SHELLDETAILS, PID_DESCRIPTIONID);
|
|
DEFINE_SCID(SCID_CAPACITY, PSGUID_VOLUME, PID_VOLUME_CAPACITY);
|
|
DEFINE_SCID(SCID_FREESPACE, PSGUID_VOLUME, PID_VOLUME_FREE);
|
|
DEFINE_SCID(SCID_DisplayProperties, PSGUID_WEBVIEW, PID_DISPLAY_PROPERTIES);
|
|
DEFINE_SCID(SCID_TYPE, PSGUID_STORAGE, PID_STG_STORAGETYPE);
|
|
DEFINE_SCID(SCID_NAME, PSGUID_STORAGE, PID_STG_NAME);
|
|
DEFINE_SCID(SCID_WRITETIME, PSGUID_STORAGE, PID_STG_WRITETIME);
|
|
|
|
DEFINE_SCID(SCID_AUDIO_Duration, PSGUID_AUDIO, PIDASI_TIMELENGTH); //100ns units, not milliseconds. VT_UI8, not VT_UI4
|
|
DEFINE_SCID(SCID_AUDIO_Bitrate, PSGUID_AUDIO, PIDASI_AVG_DATA_RATE); // bits per second
|
|
|
|
|
|
EXTERN_C const BYTE abPVK[4096];
|
|
EXTERN_C const BYTE abCert[100];
|
|
|
|
#pragma pack(1)
|
|
typedef struct // typedef struct
|
|
{ // {
|
|
// these need to line up -----------------------
|
|
WORD cbSize; // WORD cbSize; // Size of entire item ID
|
|
WORD wOuter; // WORD wOuter; // Private data owned by the outer folder
|
|
WORD cbInner; // WORD cbInner; // Size of delegate's data
|
|
// ---------------------------------------------
|
|
DWORD dwMagic; // BYTE rgb[1]; // Inner folder's data,
|
|
DWORD dwAttributes; // } DELEGATEITEMID;
|
|
ULARGE_INTEGER cbTotal;
|
|
ULARGE_INTEGER cbFree;
|
|
union
|
|
{
|
|
FILETIME ftModified;
|
|
ULARGE_INTEGER ulModified;
|
|
};
|
|
WCHAR szName[1]; // variable size
|
|
} MEDIADEVITEM;
|
|
#pragma pack()
|
|
|
|
typedef UNALIGNED MEDIADEVITEM * LPMEDIADEVITEM;
|
|
typedef const UNALIGNED MEDIADEVITEM * LPCMEDIADEVITEM;
|
|
|
|
#define MEDIADEVITEM_MAGIC 0x08311978
|
|
|
|
class CMediaDeviceFolder;
|
|
class CMediaDeviceEnum;
|
|
class CMediaDeviceDropTarget;
|
|
|
|
class CMediaDeviceFolder : public IShellFolder2, IPersistFolder2, IShellFolderViewCB, IDelegateFolder, IContextMenuCB
|
|
{
|
|
public:
|
|
CMediaDeviceFolder(CMediaDeviceFolder *pParent, IWMDMStorage *pstg);
|
|
|
|
// IUnknown
|
|
STDMETHODIMP QueryInterface(REFIID riid, void **ppv);
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
|
STDMETHODIMP_(ULONG) Release(void);
|
|
|
|
// IPersist
|
|
STDMETHODIMP GetClassID(CLSID *pClassID);
|
|
|
|
// IPersistFolder
|
|
STDMETHODIMP Initialize(LPCITEMIDLIST pidl);
|
|
|
|
// IPersistFolder2
|
|
STDMETHODIMP GetCurFolder(LPITEMIDLIST *ppidl);
|
|
|
|
// IShellFolder
|
|
STDMETHODIMP ParseDisplayName(HWND hwnd, LPBC pbc, LPOLESTR pszName, ULONG * pchEaten, LPITEMIDLIST * ppidl, ULONG *pdwAttributes);
|
|
STDMETHODIMP EnumObjects(HWND hwnd, DWORD grfFlags, IEnumIDList **ppenumIDList);
|
|
STDMETHODIMP BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppvOut);
|
|
STDMETHODIMP BindToStorage(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv)
|
|
{ return BindToObject(pidl, pbc, riid, ppv); };
|
|
STDMETHODIMP CompareIDs(LPARAM lParam, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2);
|
|
STDMETHODIMP CreateViewObject(HWND hwndOwner, REFIID riid, void **ppvOut);
|
|
STDMETHODIMP GetAttributesOf(UINT cidl, LPCITEMIDLIST * apidl, ULONG *rgfInOut);
|
|
STDMETHODIMP GetUIObjectOf(HWND hwndOwner, UINT cidl, LPCITEMIDLIST * apidl, REFIID riid, UINT * prgfInOut, void **ppvOut);
|
|
STDMETHODIMP GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD uFlags, LPSTRRET lpName);
|
|
STDMETHODIMP SetNameOf(HWND hwnd, LPCITEMIDLIST pidl, LPCOLESTR pszName, DWORD uFlags, LPITEMIDLIST * ppidlOut);
|
|
|
|
// IShellFolder2
|
|
STDMETHODIMP GetDefaultSearchGUID(GUID *pGuid)
|
|
{ return E_NOTIMPL; };
|
|
STDMETHODIMP EnumSearches(IEnumExtraSearch **ppenum)
|
|
{ return E_NOTIMPL; };
|
|
STDMETHODIMP GetDefaultColumn(DWORD dwRes, ULONG *pSort, ULONG *pDisplay)
|
|
{ return E_NOTIMPL; };
|
|
STDMETHODIMP GetDefaultColumnState(UINT iColumn, DWORD *pbState)
|
|
{ return E_NOTIMPL; }
|
|
STDMETHODIMP GetDetailsEx(LPCITEMIDLIST pidl, const SHCOLUMNID *pscid, VARIANT *pv);
|
|
STDMETHODIMP GetDetailsOf(LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS *pDetails);
|
|
STDMETHODIMP MapColumnToSCID(UINT iColumn, SHCOLUMNID *pscid);
|
|
|
|
// IShellFolderViewCB
|
|
STDMETHODIMP MessageSFVCB(UINT uMsg, WPARAM wParam, LPARAM lParam);
|
|
|
|
// IDelegateFolder
|
|
STDMETHODIMP SetItemAlloc(IMalloc *pmalloc);
|
|
|
|
// IContextMenuCB
|
|
STDMETHODIMP CallBack(IShellFolder *psf, HWND hwnd, IDataObject *pdtobj, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
|
|
|
// public for friends
|
|
LPCITEMIDLIST GetIDList() { return _pidl; }
|
|
|
|
private:
|
|
~CMediaDeviceFolder();
|
|
|
|
HRESULT _CreateIDList(LPCTSTR pszName, MEDIADEVITEM **ppmditem, BOOL *pfFolder);
|
|
HRESULT _IDListForDevice(IWMDMDevice *pdev, LPITEMIDLIST *ppidl, BOOL *pfFolder);
|
|
HRESULT _IDListForStorage(IWMDMStorage *pstg, LPITEMIDLIST *ppidl, BOOL *pfFolder);
|
|
ULONG _GetAttributesOf(LPCMEDIADEVITEM pmdi, ULONG rgfIn);
|
|
HRESULT _CreateExtractIcon(LPCMEDIADEVITEM pmdi, REFIID riid, void **ppv);
|
|
static HRESULT CALLBACK _ItemsMenuCB(IShellFolder *psf, HWND hwnd, IDataObject *pdtobj, UINT uMsg, WPARAM wParam, LPARAM lParam);
|
|
HRESULT _DeleteItems(HWND hwnd, IDataObject *pdtobj);
|
|
HRESULT _ChangeNotifyItem(LONG lEvent, LPCITEMIDLIST pidl);
|
|
HRESULT _ChangeNotifyObj(LONG lEvent, IUnknown *punkStg);
|
|
|
|
// folder view callback handlers
|
|
HRESULT _OnBackgroundEnum(DWORD pv);
|
|
HRESULT _OnGetNotify(DWORD pv, LPITEMIDLIST *ppidl, LONG *plEvents);
|
|
HRESULT _OnViewMode(DWORD pv, FOLDERVIEWMODE *pvm);
|
|
|
|
LPCMEDIADEVITEM _IsValid(LPCITEMIDLIST pidl);
|
|
DWORD _IsFolder(LPCMEDIADEVITEM pmditem);
|
|
BOOL _FileInfo(LPCMEDIADEVITEM pmdi, DWORD dwFlags, SHFILEINFO *psfi);
|
|
LPCTSTR _DisplayName(LPCMEDIADEVITEM pmdi, LPTSTR pszName, UINT cch);
|
|
LPCTSTR _Name(LPCMEDIADEVITEM pmdi, LPTSTR psz, UINT cch);
|
|
LPCTSTR _RawName(LPCMEDIADEVITEM pmdi, LPTSTR psz, UINT cch);
|
|
LPCTSTR _Type(LPCMEDIADEVITEM pmdi, LPTSTR psz, UINT cch);
|
|
HRESULT _GetMediaDevMgr(REFIID riid, void **ppv);
|
|
HRESULT _Authenticate(IUnknown *punk);
|
|
HRESULT _EnumDevices(IWMDMEnumDevice **ppEnumDevices);
|
|
HRESULT _FindDeviceByName(LPCWSTR pszName, IWMDMDevice **ppdevice);
|
|
HRESULT _StorageByName(LPCWSTR pszName, REFIID riid, void **ppv);
|
|
HRESULT _StroageFromItem(LPCMEDIADEVITEM pmdi, REFIID riid, void **ppv);
|
|
HRESULT _StorageForDevice(IWMDMDevice *pdev, REFIID riid, void **ppv);
|
|
HRESULT _Storage(REFIID riid, void **ppv);
|
|
HRESULT _EnumStorage(IWMDMEnumStorage **ppEnumStorage);
|
|
HRESULT _DeviceFromItem(LPCMEDIADEVITEM pmdi, IWMDMDevice **ppdev);
|
|
BOOL _IsRoot();
|
|
void *_Alloc(SIZE_T cb);
|
|
BOOL _ShouldShowDevice(IWMDMDevice *pdev);
|
|
|
|
friend CMediaDeviceEnum;
|
|
friend CMediaDeviceDropTarget;
|
|
|
|
LONG _cRef;
|
|
IMalloc *_pmalloc;
|
|
LPITEMIDLIST _pidl; // full idlist of this folder
|
|
BOOL _bRoot; // is this the root of the name space
|
|
IGlobalInterfaceTable *_pgit;
|
|
DWORD _dwStorageCookie; // GIT token for IWMDMStorage
|
|
DWORD _dwDevMgrCookie; // GIT token for media device manager
|
|
CSecureChannelClient *_pSAC; // this must outlive any objects that used this
|
|
|
|
BOOL _fDelegate;
|
|
};
|
|
|
|
class CMediaDeviceEnum : public IEnumIDList
|
|
{
|
|
public:
|
|
CMediaDeviceEnum(CMediaDeviceFolder* prf, DWORD grfFlags);
|
|
|
|
// IUnknown
|
|
STDMETHODIMP QueryInterface(REFIID riid, void **ppv);
|
|
STDMETHODIMP_(ULONG) AddRef();
|
|
STDMETHODIMP_(ULONG) Release();
|
|
|
|
// IEnumIDList
|
|
STDMETHODIMP Next(ULONG celt, LPITEMIDLIST *rgelt, ULONG *pceltFetched);
|
|
STDMETHODIMP Skip(ULONG celt) { return E_NOTIMPL; };
|
|
STDMETHODIMP Reset() { _iIndex = 0; return S_OK; };
|
|
STDMETHODIMP Clone(IEnumIDList **ppenum) { return E_NOTIMPL; };
|
|
|
|
private:
|
|
~CMediaDeviceEnum();
|
|
HRESULT _Init();
|
|
BOOL _FilterItem(BOOL bFolder);
|
|
|
|
LONG _cRef;
|
|
CDPA<ITEMIDLIST> _dpaItems;
|
|
int _iIndex;
|
|
CMediaDeviceFolder *_pmdf;
|
|
DWORD _grfFlags;
|
|
};
|
|
|
|
class CMediaDeviceDropTarget : public IDropTarget, INamespaceWalkCB, IWMDMProgress2
|
|
{
|
|
public:
|
|
CMediaDeviceDropTarget(CMediaDeviceFolder *pFolder, HWND hwnd);
|
|
|
|
// IUnknown
|
|
STDMETHODIMP QueryInterface(REFIID riid, void ** ppv);
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
|
STDMETHODIMP_(ULONG) Release(void);
|
|
|
|
// IDropTarget
|
|
STDMETHODIMP DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
|
|
STDMETHODIMP DragOver(DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
|
|
STDMETHODIMP DragLeave();
|
|
STDMETHODIMP Drop(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect);
|
|
|
|
// IWMDMProgress
|
|
STDMETHODIMP Begin(DWORD dwEstimatedTicks);
|
|
STDMETHODIMP Progress(DWORD dwTranspiredTicks);
|
|
STDMETHODIMP End();
|
|
|
|
// IWMDMProgress2
|
|
STDMETHODIMP End2(HRESULT hrCompletionCode);
|
|
|
|
// INamespaceWalkCB
|
|
STDMETHODIMP FoundItem(IShellFolder *psf, LPCITEMIDLIST pidl);
|
|
STDMETHODIMP EnterFolder(IShellFolder *psf, LPCITEMIDLIST pidl);
|
|
STDMETHODIMP LeaveFolder(IShellFolder *psf, LPCITEMIDLIST pidl);
|
|
STDMETHOD(InitializeProgressDialog)(LPWSTR *ppszTitle, LPWSTR *ppszCancel)
|
|
{ *ppszTitle = NULL; *ppszCancel = NULL; return E_NOTIMPL; }
|
|
|
|
private:
|
|
~CMediaDeviceDropTarget();
|
|
DWORD _GetDropEffect(DWORD *pdwEffect, DWORD grfKeyState);
|
|
static DWORD CALLBACK _DoCopyThreadProc(void *pv);
|
|
HRESULT _DoCopy(IDataObject *pdtobj);
|
|
|
|
LONG _cRef;
|
|
|
|
CMediaDeviceFolder *_pmdf;
|
|
HWND _hwnd; // EVIL: used as a site and UI host
|
|
IDataObject *_pdtobj; // used durring DragOver() and DoDrop(), don't use on background thread
|
|
|
|
UINT _idCmd;
|
|
DWORD _grfKeyStateLast; // for previous DragOver/Enter
|
|
DWORD _dwEffectLastReturned; // stashed effect that's returned by base class's dragover
|
|
DWORD _dwEffectPreferred; // if dwData & DTID_PREFERREDEFFECT
|
|
|
|
IProgressDialog *_ppd;
|
|
ULONGLONG _ulProgressTotal;
|
|
ULONGLONG _ulProgressCurrent;
|
|
ULONGLONG _ulThisFile;
|
|
IUnknown *_punkFTM; // make our callback interface calls direct
|
|
};
|
|
|
|
CMediaDeviceFolder::CMediaDeviceFolder(CMediaDeviceFolder *pParent, IWMDMStorage *pstg) :
|
|
_cRef(1), _pidl(NULL), _dwStorageCookie(-1), _dwDevMgrCookie(-1)
|
|
{
|
|
_bRoot = (NULL == pstg);
|
|
|
|
if (SUCCEEDED(CoCreateInstance(CLSID_StdGlobalInterfaceTable, 0, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IGlobalInterfaceTable, &_pgit))))
|
|
{
|
|
if (pstg)
|
|
{
|
|
_pgit->RegisterInterfaceInGlobal(pstg, IID_IWMDMStorage, &_dwStorageCookie);
|
|
}
|
|
}
|
|
DllAddRef();
|
|
}
|
|
|
|
CMediaDeviceFolder::~CMediaDeviceFolder()
|
|
{
|
|
ILFree(_pidl);
|
|
#ifdef _X86_
|
|
if (_pSAC)
|
|
delete _pSAC;
|
|
#endif
|
|
|
|
if (_pgit)
|
|
{
|
|
if (-1 != _dwStorageCookie)
|
|
_pgit->RevokeInterfaceFromGlobal(_dwStorageCookie);
|
|
|
|
if (-1 != _dwDevMgrCookie)
|
|
_pgit->RevokeInterfaceFromGlobal(_dwDevMgrCookie);
|
|
_pgit->Release();
|
|
}
|
|
|
|
ATOMICRELEASE(_pmalloc);
|
|
|
|
DllRelease();
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::QueryInterface(REFIID riid, void **ppv)
|
|
{
|
|
static const QITAB qit[] =
|
|
{
|
|
QITABENTMULTI(CMediaDeviceFolder, IShellFolder, IShellFolder2),
|
|
QITABENT (CMediaDeviceFolder, IShellFolder2),
|
|
QITABENTMULTI(CMediaDeviceFolder, IPersist, IPersistFolder2),
|
|
QITABENTMULTI(CMediaDeviceFolder, IPersistFolder, IPersistFolder2),
|
|
QITABENT (CMediaDeviceFolder, IPersistFolder2),
|
|
QITABENT (CMediaDeviceFolder, IShellFolderViewCB),
|
|
QITABENT (CMediaDeviceFolder, IDelegateFolder),
|
|
QITABENT (CMediaDeviceFolder, IContextMenuCB),
|
|
{ 0 },
|
|
};
|
|
return QISearch(this, qit, riid, ppv);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CMediaDeviceFolder::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CMediaDeviceFolder::Release()
|
|
{
|
|
if (InterlockedDecrement(&_cRef))
|
|
return _cRef;
|
|
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
STDAPI CMediaDeviceFolder_CreateInstance(IUnknown *punkOuter, IUnknown **ppunk, LPCOBJECTINFO poi)
|
|
{
|
|
HRESULT hr;
|
|
CMediaDeviceFolder *pmdf = new CMediaDeviceFolder(NULL, NULL);
|
|
if (pmdf)
|
|
{
|
|
*ppunk = SAFECAST(pmdf, IShellFolder2 *);
|
|
hr = S_OK;
|
|
}
|
|
else
|
|
{
|
|
*ppunk = NULL;
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_Authenticate(IUnknown *punk)
|
|
{
|
|
#ifndef _X86_
|
|
HRESULT hr = E_FAIL;
|
|
#else
|
|
IComponentAuthenticate *pAuth;
|
|
HRESULT hr = punk->QueryInterface(IID_PPV_ARG(IComponentAuthenticate, &pAuth));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (NULL == _pSAC)
|
|
_pSAC = new CSecureChannelClient;
|
|
if (_pSAC)
|
|
{
|
|
hr = _pSAC->SetCertificate(SAC_CERT_V1, (BYTE *)abCert, sizeof(abCert), (BYTE *)abPVK, sizeof(abPVK));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_pSAC->SetInterface(pAuth);
|
|
hr = _pSAC->Authenticate(SAC_PROTOCOL_V1);
|
|
}
|
|
}
|
|
else
|
|
hr = E_OUTOFMEMORY;
|
|
|
|
pAuth->Release();
|
|
}
|
|
#endif
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_GetMediaDevMgr(REFIID riid, void **ppv)
|
|
{
|
|
*ppv = NULL;
|
|
HRESULT hr = E_FAIL;
|
|
if (_IsRoot() && _pgit)
|
|
{
|
|
if (-1 == _dwDevMgrCookie)
|
|
{
|
|
IWMDeviceManager *pdevmgr;
|
|
if (SUCCEEDED(CoCreateInstance(CLSID_MediaDevMgr, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IWMDeviceManager, &pdevmgr))))
|
|
{
|
|
_Authenticate(pdevmgr);
|
|
_pgit->RegisterInterfaceInGlobal(pdevmgr, IID_IWMDeviceManager, &_dwDevMgrCookie);
|
|
pdevmgr->Release();
|
|
}
|
|
}
|
|
|
|
if (-1 != _dwDevMgrCookie)
|
|
hr = _pgit->GetInterfaceFromGlobal(_dwDevMgrCookie, riid, ppv);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_EnumDevices(IWMDMEnumDevice **ppEnumDevices)
|
|
{
|
|
*ppEnumDevices = NULL;
|
|
|
|
IWMDeviceManager *pdevmgr;
|
|
HRESULT hr = _GetMediaDevMgr(IID_PPV_ARG(IWMDeviceManager, &pdevmgr));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pdevmgr->EnumDevices(ppEnumDevices);
|
|
pdevmgr->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
// test to see if this is the root of the media devices name space (where the devices are)
|
|
BOOL CMediaDeviceFolder::_IsRoot()
|
|
{
|
|
return _bRoot;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_Storage(REFIID riid, void **ppv)
|
|
{
|
|
*ppv = NULL;
|
|
HRESULT hr = E_NOINTERFACE;
|
|
if (-1 != _dwStorageCookie)
|
|
{
|
|
ASSERT(!_IsRoot());
|
|
hr = _pgit->GetInterfaceFromGlobal(_dwStorageCookie, riid, ppv);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_EnumStorage(IWMDMEnumStorage **ppEnumStorage)
|
|
{
|
|
*ppEnumStorage = NULL;
|
|
|
|
IWMDMStorage *pStorage;
|
|
HRESULT hr = _Storage(IID_PPV_ARG(IWMDMStorage, &pStorage));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pStorage->EnumStorage(ppEnumStorage);
|
|
pStorage->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_FindDeviceByName(LPCWSTR pszName, IWMDMDevice **ppdevice)
|
|
{
|
|
IWMDMEnumDevice *pEnumDevice;
|
|
HRESULT hr = _EnumDevices(&pEnumDevice);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = E_FAIL;
|
|
ULONG ulFetched;
|
|
IWMDMDevice *pdev;
|
|
while (S_OK == pEnumDevice->Next(1, &pdev, &ulFetched))
|
|
{
|
|
WCHAR szName[MAX_PATH];
|
|
if (SUCCEEDED(pdev->GetName(szName, ARRAYSIZE(szName))) &&
|
|
(0 == StrCmpIC(pszName, szName)))
|
|
{
|
|
*ppdevice = pdev;
|
|
hr = S_OK;
|
|
break;
|
|
}
|
|
pdev->Release();
|
|
}
|
|
pEnumDevice->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_StorageByName(LPCWSTR pszName, REFIID riid, void **ppv)
|
|
{
|
|
IWMDMStorage2 *pstg;
|
|
HRESULT hr = _Storage(IID_PPV_ARG(IWMDMStorage2, &pstg));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IWMDMStorage *pstgFound;
|
|
hr = pstg->GetStorage((LPWSTR)pszName, &pstgFound);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pstgFound->QueryInterface(riid, ppv);
|
|
pstgFound->Release();
|
|
}
|
|
else if (E_NOINTERFACE == hr)
|
|
{
|
|
// Rio600 fails on this now. I asked jhelin to look into this
|
|
IWMDMEnumStorage *penum;
|
|
if (SUCCEEDED(pstg->EnumStorage(&penum)))
|
|
{
|
|
ULONG ulFetched;
|
|
while (S_OK == penum->Next(1, &pstgFound, &ulFetched))
|
|
{
|
|
WCHAR szName[MAX_PATH];
|
|
if (SUCCEEDED(pstgFound->GetName(szName, ARRAYSIZE(szName))) &&
|
|
(0 == StrCmpI(pszName, szName)))
|
|
{
|
|
hr = pstgFound->QueryInterface(riid, ppv);
|
|
}
|
|
pstgFound->Release();
|
|
|
|
if (SUCCEEDED(hr))
|
|
break;
|
|
}
|
|
penum->Release();
|
|
}
|
|
}
|
|
pstg->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
LPCMEDIADEVITEM CMediaDeviceFolder::_IsValid(LPCITEMIDLIST pidl)
|
|
{
|
|
if (pidl && ((LPCMEDIADEVITEM)pidl)->dwMagic == MEDIADEVITEM_MAGIC)
|
|
return (LPCMEDIADEVITEM)pidl;
|
|
return NULL;
|
|
}
|
|
|
|
DWORD CMediaDeviceFolder::_IsFolder(LPCMEDIADEVITEM pmditem)
|
|
{
|
|
return pmditem->dwAttributes & WMDM_FILE_ATTR_FOLDER ? FILE_ATTRIBUTE_DIRECTORY : 0;
|
|
}
|
|
|
|
// helper to support being run as a delegate or a stand alone folder
|
|
//
|
|
// the cbInner is the size of the data needed by the delegate. we need to compute
|
|
// the full size of the pidl for the allocation and init that we the outer folder data
|
|
void *CMediaDeviceFolder::_Alloc(SIZE_T cbInner)
|
|
{
|
|
DELEGATEITEMID *pidl;
|
|
if (_pmalloc)
|
|
{
|
|
pidl = (DELEGATEITEMID *)_pmalloc->Alloc(cbInner);
|
|
}
|
|
else
|
|
{
|
|
SIZE_T cbAlloc =
|
|
sizeof(DELEGATEITEMID) - sizeof(pidl->rgb[0]) + // header
|
|
cbInner + // inner
|
|
sizeof(WORD); // trailing null (pidl terminator)
|
|
|
|
pidl = (DELEGATEITEMID *)SHAlloc(cbAlloc);
|
|
if (pidl)
|
|
{
|
|
ZeroMemory(pidl, cbAlloc); // make it all empty
|
|
pidl->cbSize = (WORD)cbAlloc - sizeof(WORD);
|
|
pidl->cbInner = (WORD)cbInner;
|
|
}
|
|
}
|
|
return (void *)pidl;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_CreateIDList(LPCTSTR pszName, MEDIADEVITEM **ppmditem, BOOL *pfFolder)
|
|
{
|
|
HRESULT hr;
|
|
UINT cbInner = sizeof(MEDIADEVITEM) - (sizeof(DELEGATEITEMID) - 1) +
|
|
(sizeof(WCHAR) * lstrlen(pszName)); // null is accounted for in szName[1]
|
|
*ppmditem = (MEDIADEVITEM *)_Alloc(cbInner);
|
|
if (*ppmditem)
|
|
{
|
|
(*ppmditem)->dwMagic = MEDIADEVITEM_MAGIC;
|
|
StrCpyW((*ppmditem)->szName, pszName);
|
|
if (pfFolder)
|
|
*pfFolder = _IsFolder(*ppmditem);
|
|
hr = S_OK;
|
|
}
|
|
else
|
|
hr = E_OUTOFMEMORY;
|
|
return hr;
|
|
}
|
|
|
|
// Creates an item identifier list for the objects in the namespace
|
|
|
|
HRESULT CMediaDeviceFolder::_IDListForDevice(IWMDMDevice *pdev, LPITEMIDLIST *ppidl, BOOL *pfFolder)
|
|
{
|
|
WCHAR szName[MAX_PATH];
|
|
HRESULT hr = pdev->GetName(szName, ARRAYSIZE(szName));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// WCHAR szTemp[128];
|
|
// pdev->GetManufacturer(szTemp, ARRAYSIZE(szTemp));
|
|
|
|
MEDIADEVITEM *pmditem;
|
|
hr = _CreateIDList(szName, &pmditem, pfFolder);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IWMDMStorage *pstg;
|
|
|
|
if (SUCCEEDED(_StorageForDevice(pdev, IID_PPV_ARG(IWMDMStorage, &pstg))))
|
|
{
|
|
IWMDMStorageGlobals *pstgGlobals;
|
|
if (SUCCEEDED(pstg->GetStorageGlobals(&pstgGlobals)))
|
|
{
|
|
pstgGlobals->GetTotalSize(&pmditem->cbTotal.LowPart, &pmditem->cbTotal.HighPart);
|
|
pstgGlobals->GetTotalFree(&pmditem->cbFree.LowPart, &pmditem->cbFree.HighPart);
|
|
|
|
pstgGlobals->Release();
|
|
}
|
|
pstg->Release();
|
|
}
|
|
|
|
pmditem->dwAttributes = WMDM_FILE_ATTR_FOLDER | WMDM_STORAGE_ATTR_HAS_FOLDERS;
|
|
// pdev->GetType(&pmditem->dwType); // this is an expensive call, hits the disk
|
|
*ppidl = (LPITEMIDLIST)pmditem;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
void WMDMDateTimeToSystemTime(const WMDMDATETIME *pWMDM, FILETIME *pft)
|
|
{
|
|
SYSTEMTIME st = {0};
|
|
st.wYear = pWMDM->wYear;
|
|
st.wMonth = pWMDM->wMonth;
|
|
st.wDay = pWMDM->wDay;
|
|
st.wHour = pWMDM->wHour;
|
|
st.wMinute = pWMDM->wMinute;
|
|
st.wSecond = pWMDM->wSecond;
|
|
|
|
SystemTimeToFileTime(&st, pft);
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_IDListForStorage(IWMDMStorage *pstg, LPITEMIDLIST *ppidl, BOOL *pfFolder)
|
|
{
|
|
WCHAR szName[MAX_PATH];
|
|
HRESULT hr = pstg->GetName(szName, ARRAYSIZE(szName));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
MEDIADEVITEM *pmditem;
|
|
hr = _CreateIDList(szName, &pmditem, pfFolder);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_WAVEFORMATEX fmt;
|
|
pstg->GetAttributes(&pmditem->dwAttributes, &fmt);
|
|
|
|
WMDMDATETIME datetime;
|
|
if (SUCCEEDED(pstg->GetDate(&datetime)))
|
|
{
|
|
WMDMDateTimeToSystemTime(&datetime, &pmditem->ftModified);
|
|
}
|
|
|
|
if (!(pmditem->dwAttributes & WMDM_FILE_ATTR_FOLDER))
|
|
{
|
|
pstg->GetSize(&pmditem->cbTotal.LowPart, &pmditem->cbTotal.HighPart);
|
|
}
|
|
*ppidl = (LPITEMIDLIST)pmditem;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
// IPersist
|
|
|
|
STDMETHODIMP CMediaDeviceFolder::GetClassID(CLSID *pClassID)
|
|
{
|
|
*pClassID = CLSID_MediaDeviceFolder;
|
|
return S_OK;
|
|
}
|
|
|
|
// IPersistFolder
|
|
|
|
STDMETHODIMP CMediaDeviceFolder::Initialize(LPCITEMIDLIST pidl)
|
|
{
|
|
ILFree(_pidl);
|
|
return SHILClone(pidl, &_pidl);
|
|
}
|
|
|
|
// IPersistFolder2
|
|
|
|
HRESULT CMediaDeviceFolder::GetCurFolder(LPITEMIDLIST *ppidl)
|
|
{
|
|
if (_pidl)
|
|
return SHILClone(_pidl, ppidl);
|
|
|
|
*ppidl = NULL;
|
|
return S_FALSE;
|
|
}
|
|
|
|
|
|
// IShellFolder(2)
|
|
|
|
HRESULT CMediaDeviceFolder::ParseDisplayName(HWND hwnd, LPBC pbc, LPOLESTR pszName, ULONG *pchEaten, LPITEMIDLIST *ppidl, ULONG *pdwAttributes)
|
|
{
|
|
HRESULT hr;
|
|
|
|
if (!pszName || !ppidl)
|
|
return E_INVALIDARG;
|
|
|
|
*ppidl = NULL;
|
|
|
|
#if 1
|
|
hr = E_NOTIMPL;
|
|
#else
|
|
TCHAR szName[MAX_PATH];
|
|
hr = _NextSegment((LPCWSTR*)&pszName, szName, ARRAYSIZE(szName), TRUE);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = _IDListForDevice(szName, ppidl, NULL);
|
|
if (SUCCEEDED(hr) && pszName)
|
|
{
|
|
IShellFolder *psf;
|
|
hr = BindToObject(*ppidl, pbc, IID_PPV_ARG(IShellFolder, &psf));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ULONG chEaten;
|
|
LPITEMIDLIST pidlNext;
|
|
hr = psf->ParseDisplayName(hwnd, pbc, pszName, &chEaten, &pidlNext, pdwAttributes);
|
|
if (SUCCEEDED(hr))
|
|
hr = SHILAppend(pidlNext, ppidl);
|
|
psf->Release();
|
|
}
|
|
}
|
|
else if (SUCCEEDED(hr) && pdwAttributes && *pdwAttributes)
|
|
{
|
|
GetAttributesOf(1, (LPCITEMIDLIST *)ppidl, pdwAttributes);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
// clean up if the parse failed.
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
ILFree(*ppidl);
|
|
*ppidl = NULL;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::EnumObjects(HWND hwnd, DWORD grfFlags, IEnumIDList **ppenum)
|
|
{
|
|
HRESULT hr;
|
|
CMediaDeviceEnum *penum = new CMediaDeviceEnum(this, grfFlags);
|
|
if (penum)
|
|
{
|
|
hr = penum->QueryInterface(IID_PPV_ARG(IEnumIDList, ppenum));
|
|
penum->Release();
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
*ppenum = NULL;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
|
|
STDAPI_(LPITEMIDLIST) ILCombineParentAndFirst(LPCITEMIDLIST pidlParent, LPCITEMIDLIST pidl, LPCITEMIDLIST pidlNext)
|
|
{
|
|
ULONG cbParent = ILGetSize(pidlParent);
|
|
ULONG cbRest = (ULONG)((ULONG_PTR)pidlNext - (ULONG_PTR)pidl);
|
|
LPITEMIDLIST pidlNew = _ILCreate(cbParent + cbRest);
|
|
if (pidlNew)
|
|
{
|
|
cbParent -= sizeof(pidlParent->mkid.cb);
|
|
memcpy(pidlNew, pidlParent, cbParent);
|
|
memcpy((BYTE *)pidlNew + cbParent, pidl, cbRest);
|
|
ASSERT(_ILSkip(pidlNew, cbParent + cbRest)->mkid.cb == 0);
|
|
}
|
|
return pidlNew;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_StorageForDevice(IWMDMDevice *pdev, REFIID riid, void **ppv)
|
|
{
|
|
*ppv = NULL;
|
|
IWMDMEnumStorage *penum;
|
|
HRESULT hr = pdev->EnumStorage(&penum);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ULONG ulFetched;
|
|
IWMDMStorage *pstg;
|
|
if (S_OK == penum->Next(1, &pstg, &ulFetched))
|
|
{
|
|
hr= pstg->QueryInterface(riid, ppv);
|
|
pstg->Release();
|
|
}
|
|
else
|
|
{
|
|
hr = E_NOINTERFACE;
|
|
}
|
|
penum->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_StroageFromItem(LPCMEDIADEVITEM pmdi, REFIID riid, void **ppv)
|
|
{
|
|
WCHAR szName[MAX_PATH];
|
|
_Name(pmdi, szName, ARRAYSIZE(szName));
|
|
|
|
*ppv = NULL;
|
|
HRESULT hr;
|
|
if (_IsRoot())
|
|
{
|
|
// root case
|
|
IWMDMDevice *pdev;
|
|
hr = _FindDeviceByName(szName, &pdev);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = _StorageForDevice(pdev, riid, ppv);
|
|
pdev->Release();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = _StorageByName(szName, riid, ppv);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_ChangeNotifyItem(LONG lEvent, LPCITEMIDLIST pidl)
|
|
{
|
|
ITEMIDLIST *pidlFull;
|
|
HRESULT hr = SHILCombine(_pidl, pidl, &pidlFull);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
SHChangeNotify(lEvent, 0, (LPCTSTR)pidlFull, NULL);
|
|
ILFree(pidlFull);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_ChangeNotifyObj(LONG lEvent, IUnknown *punkStg)
|
|
{
|
|
IWMDMStorage *pstg;
|
|
HRESULT hr = punkStg->QueryInterface(IID_PPV_ARG(IWMDMStorage, &pstg));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ITEMIDLIST *pidl;
|
|
BOOL bFolder;
|
|
hr = _IDListForStorage(pstg, &pidl, &bFolder);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = _ChangeNotifyItem(lEvent, pidl);
|
|
ILFree(pidl);
|
|
}
|
|
pstg->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_DeviceFromItem(LPCMEDIADEVITEM pmdi, IWMDMDevice **ppdev)
|
|
{
|
|
*ppdev = NULL;
|
|
|
|
TCHAR szName[MAX_PATH];
|
|
return _IsRoot() ? _FindDeviceByName(_RawName(pmdi, szName, ARRAYSIZE(szName)), ppdev) : E_NOINTERFACE;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::BindToObject(LPCITEMIDLIST pidl, LPBC pbc, REFIID riid, void **ppv)
|
|
{
|
|
*ppv = NULL;
|
|
|
|
HRESULT hr = E_NOINTERFACE;
|
|
LPCMEDIADEVITEM pmdi = _IsValid(pidl);
|
|
if (pmdi && _IsFolder(pmdi))
|
|
{
|
|
if (IID_IShellFolder == riid ||
|
|
IID_IShellFolder2 == riid)
|
|
{
|
|
IWMDMStorage *pstg;
|
|
hr = _StroageFromItem(pmdi, IID_PPV_ARG(IWMDMStorage, &pstg));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
LPCITEMIDLIST pidlNext = _ILNext(pidl);
|
|
LPITEMIDLIST pidlSubFolder = ILCombineParentAndFirst(_pidl, pidl, pidlNext);
|
|
if (pidlSubFolder)
|
|
{
|
|
CMediaDeviceFolder *pmdf = new CMediaDeviceFolder(this, pstg);
|
|
if (pmdf)
|
|
{
|
|
hr = pmdf->Initialize(pidlSubFolder);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// if there's nothing left in the pidl, get the interface on this one.
|
|
if (ILIsEmpty(pidlNext))
|
|
hr = pmdf->QueryInterface(riid, ppv);
|
|
else
|
|
{
|
|
// otherwise, hand the rest of it off to the new shellfolder.
|
|
hr = pmdf->BindToObject(pidlNext, pbc, riid, ppv);
|
|
}
|
|
}
|
|
pmdf->Release();
|
|
}
|
|
else
|
|
hr = E_OUTOFMEMORY;
|
|
|
|
ILFree(pidlSubFolder);
|
|
}
|
|
else
|
|
hr = E_OUTOFMEMORY;
|
|
|
|
pstg->Release();
|
|
}
|
|
}
|
|
}
|
|
else
|
|
hr = E_FAIL;
|
|
|
|
ASSERT((SUCCEEDED(hr) && *ppv) || (FAILED(hr) && (NULL == *ppv))); // Assert hr is consistent w/out param.
|
|
return hr;
|
|
}
|
|
|
|
enum
|
|
{
|
|
DEV_COL_NAME = 0,
|
|
DEV_COL_SIZE,
|
|
DEV_COL_TYPE,
|
|
DEV_COL_MODIFIED,
|
|
};
|
|
|
|
HRESULT CMediaDeviceFolder::CompareIDs(LPARAM lParam, LPCITEMIDLIST pidl1, LPCITEMIDLIST pidl2)
|
|
{
|
|
LPCMEDIADEVITEM pmdi1 = _IsValid(pidl1);
|
|
LPCMEDIADEVITEM pmdi2 = _IsValid(pidl2);
|
|
int nCmp = 0;
|
|
|
|
if (!pmdi1 || !pmdi2)
|
|
return E_INVALIDARG;
|
|
|
|
// folders always sort to the top of the list, if either of these items
|
|
// are folders then compare on the folderness
|
|
|
|
if (_IsFolder(pmdi1) || _IsFolder(pmdi2))
|
|
{
|
|
if (_IsFolder(pmdi1) && !_IsFolder(pmdi2))
|
|
nCmp = -1;
|
|
else if (!_IsFolder(pmdi1) && _IsFolder(pmdi2))
|
|
nCmp = 1;
|
|
}
|
|
|
|
// if they match (or are not folders, then lets compare based on the column ID.
|
|
|
|
if (nCmp == 0)
|
|
{
|
|
switch (lParam & SHCIDS_COLUMNMASK)
|
|
{
|
|
case DEV_COL_NAME: // caught later on
|
|
break;
|
|
|
|
case DEV_COL_SIZE:
|
|
if (pmdi1->cbTotal.QuadPart < pmdi2->cbTotal.QuadPart)
|
|
nCmp = -1;
|
|
else if (pmdi1->cbTotal.QuadPart > pmdi2->cbTotal.QuadPart)
|
|
nCmp = 1;
|
|
break;
|
|
|
|
case DEV_COL_TYPE:
|
|
{
|
|
TCHAR szType1[MAX_PATH], szType2[MAX_PATH];
|
|
_Type(pmdi1, szType1, ARRAYSIZE(szType1));
|
|
_Type(pmdi2, szType2, ARRAYSIZE(szType2));
|
|
nCmp = StrCmp(szType1, szType2);
|
|
break;
|
|
}
|
|
|
|
case DEV_COL_MODIFIED:
|
|
if (pmdi1->ulModified.QuadPart < pmdi2->ulModified.QuadPart)
|
|
nCmp = -1;
|
|
else if (pmdi1->ulModified.QuadPart > pmdi2->ulModified.QuadPart)
|
|
nCmp = 1;
|
|
break;
|
|
}
|
|
|
|
if (nCmp == 0)
|
|
{
|
|
TCHAR szName1[MAX_PATH], szName2[MAX_PATH];
|
|
nCmp = StrCmpLogicalW(_Name(pmdi1, szName1, ARRAYSIZE(szName1)), _Name(pmdi2, szName2, ARRAYSIZE(szName2)));
|
|
}
|
|
}
|
|
|
|
return ResultFromShort(nCmp);
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_OnBackgroundEnum(DWORD pv)
|
|
{
|
|
return S_OK; // do enum on backgound always
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_OnGetNotify(DWORD pv, LPITEMIDLIST *ppidl, LONG *plEvents)
|
|
{
|
|
*ppidl = _pidl; // evil alais
|
|
*plEvents = SHCNE_RENAMEITEM | SHCNE_RENAMEFOLDER | \
|
|
SHCNE_CREATE | SHCNE_DELETE | SHCNE_UPDATEDIR | SHCNE_UPDATEITEM | \
|
|
SHCNE_MKDIR | SHCNE_RMDIR;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_OnViewMode(DWORD pv, FOLDERVIEWMODE *pvm)
|
|
{
|
|
*pvm = FVM_TILE;
|
|
return S_OK;
|
|
}
|
|
|
|
// IShellFolderViewCB
|
|
STDMETHODIMP CMediaDeviceFolder::MessageSFVCB(UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
switch (uMsg)
|
|
{
|
|
HANDLE_MSG(0, SFVM_BACKGROUNDENUM, _OnBackgroundEnum);
|
|
HANDLE_MSG(0, SFVM_GETNOTIFY, _OnGetNotify);
|
|
HANDLE_MSG(0, SFVM_DEFVIEWMODE, _OnViewMode);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::CreateViewObject(HWND hwnd, REFIID riid, void **ppv)
|
|
{
|
|
HRESULT hr = E_NOINTERFACE;
|
|
*ppv = NULL;
|
|
|
|
if (IsEqualIID(riid, IID_IShellView))
|
|
{
|
|
SFV_CREATE sSFV = { 0 };
|
|
sSFV.cbSize = sizeof(sSFV);
|
|
sSFV.psfvcb = SAFECAST(this, IShellFolderViewCB *);
|
|
sSFV.pshf = SAFECAST(this, IShellFolder *);
|
|
hr = SHCreateShellFolderView(&sSFV, (IShellView**)ppv);
|
|
}
|
|
else if (IsEqualIID(riid, IID_IContextMenu))
|
|
{
|
|
HKEY hkNoFiles = NULL;
|
|
|
|
RegOpenKey(HKEY_CLASSES_ROOT, TEXT("Directory\\Background"), &hkNoFiles);
|
|
|
|
hr = CDefFolderMenu_Create2(_pidl, hwnd, 0, NULL, this, NULL,
|
|
1, &hkNoFiles, (IContextMenu **)ppv);
|
|
if (hkNoFiles)
|
|
RegCloseKey(hkNoFiles);
|
|
}
|
|
else if (IsEqualIID(riid, IID_IDropTarget) && !_IsRoot())
|
|
{
|
|
CMediaDeviceDropTarget *psdt = new CMediaDeviceDropTarget(this, hwnd);
|
|
if (psdt)
|
|
{
|
|
hr = psdt->QueryInterface(riid, ppv);
|
|
psdt->Release();
|
|
}
|
|
else
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
DWORD WMDMAttributesSFGAO(DWORD dwWMDMAttributes)
|
|
{
|
|
DWORD dwShell = SFGAO_DROPTARGET; // defaults
|
|
|
|
if (WMDM_FILE_ATTR_FOLDER & dwWMDMAttributes)
|
|
dwShell |= SFGAO_FOLDER;
|
|
if (WMDM_FILE_ATTR_LINK & dwWMDMAttributes)
|
|
dwShell |= SFGAO_LINK;
|
|
if (WMDM_FILE_ATTR_CANDELETE & dwWMDMAttributes)
|
|
dwShell |= SFGAO_CANDELETE;
|
|
if (WMDM_FILE_ATTR_CANMOVE & dwWMDMAttributes)
|
|
dwShell |= SFGAO_CANMOVE;
|
|
if (WMDM_FILE_ATTR_CANRENAME & dwWMDMAttributes)
|
|
dwShell |= SFGAO_CANRENAME;
|
|
if (WMDM_FILE_ATTR_HIDDEN & dwWMDMAttributes)
|
|
dwShell |= SFGAO_HIDDEN;
|
|
if (WMDM_FILE_ATTR_READONLY & dwWMDMAttributes)
|
|
dwShell |= SFGAO_READONLY;
|
|
if (WMDM_STORAGE_ATTR_HAS_FOLDERS & dwWMDMAttributes)
|
|
dwShell |= SFGAO_HASSUBFOLDER;
|
|
return dwShell;
|
|
}
|
|
|
|
ULONG CMediaDeviceFolder::_GetAttributesOf(LPCMEDIADEVITEM pmdi, ULONG rgfIn)
|
|
{
|
|
ULONG lResult = rgfIn & WMDMAttributesSFGAO(pmdi->dwAttributes);
|
|
if (_IsRoot())
|
|
{
|
|
lResult |= SFGAO_CANLINK | SFGAO_REMOVABLE;
|
|
}
|
|
return lResult;
|
|
}
|
|
|
|
BOOL IsSelf(UINT cidl, LPCITEMIDLIST *apidl)
|
|
{
|
|
return cidl == 0 || (cidl == 1 && (apidl == NULL || apidl[0] == NULL || ILIsEmpty(apidl[0])));
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::GetAttributesOf(UINT cidl, LPCITEMIDLIST *apidl, ULONG *prgfInOut)
|
|
{
|
|
UINT rgfOut = *prgfInOut;
|
|
|
|
// return attributes of the namespace root?
|
|
|
|
if (IsSelf(cidl, apidl))
|
|
{
|
|
rgfOut &= SFGAO_FOLDER |
|
|
SFGAO_LINK | SFGAO_DROPTARGET |
|
|
SFGAO_CANRENAME | SFGAO_CANDELETE |
|
|
SFGAO_CANLINK | SFGAO_CANCOPY |
|
|
SFGAO_CANMOVE | SFGAO_HASSUBFOLDER;
|
|
}
|
|
else
|
|
{
|
|
for (UINT i = 0; i < cidl; i++)
|
|
{
|
|
LPCMEDIADEVITEM pmdi = _IsValid(apidl[i]);
|
|
if (pmdi)
|
|
rgfOut &= _GetAttributesOf(pmdi, *prgfInOut);
|
|
}
|
|
}
|
|
|
|
*prgfInOut = rgfOut;
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT _CreateProgress(HWND hwnd, IProgressDialog **pppd)
|
|
{
|
|
HRESULT hr = CoCreateInstance(CLSID_ProgressDialog, NULL, CLSCTX_INPROC_SERVER, IID_PPV_ARG(IProgressDialog, pppd));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
(*pppd)->StartProgressDialog(hwnd, NULL, PROGDLG_AUTOTIME, NULL);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_DeleteItems(HWND hwnd, IDataObject *pdtobj)
|
|
{
|
|
IProgressDialog *ppd;
|
|
HRESULT hr = _CreateProgress(hwnd, &ppd);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ppd->SetTitle(L"Deleting Files...");
|
|
|
|
STGMEDIUM medium;
|
|
LPIDA pida = DataObj_GetHIDA(pdtobj, &medium);
|
|
if (pida)
|
|
{
|
|
for (UINT i = 0; i < pida->cidl; i++)
|
|
{
|
|
LPCMEDIADEVITEM pmdi = _IsValid(HIDA_GetPIDLItem(pida, i));
|
|
if (pmdi)
|
|
{
|
|
TCHAR szName[MAX_PATH];
|
|
|
|
ppd->SetLine(2, _DisplayName(pmdi, szName, ARRAYSIZE(szName)), FALSE, NULL);
|
|
ppd->SetProgress64(i, pida->cidl);
|
|
|
|
IWMDMStorageControl *pstgCtrl;
|
|
hr = _StroageFromItem(pmdi, IID_PPV_ARG(IWMDMStorageControl, &pstgCtrl));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pstgCtrl->Delete(WMDM_MODE_BLOCK, NULL);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_ChangeNotifyItem(SHCNE_DELETE, (LPCITEMIDLIST)pmdi);
|
|
}
|
|
pstgCtrl->Release();
|
|
|
|
ppd->SetProgress64(i + 1, pida->cidl);
|
|
|
|
if (ppd->HasUserCancelled())
|
|
{
|
|
hr = HRESULT_FROM_WIN32(ERROR_CANCELLED);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
// free space changed on device, notify
|
|
// SHChangeNotify(SHCNE_UPDATEITEM, 0, (LPCTSTR)_pidl, NULL);
|
|
}
|
|
HIDA_ReleaseStgMedium(pida, &medium);
|
|
}
|
|
ppd->StopProgressDialog();
|
|
ppd->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::CallBack(IShellFolder *psf, HWND hwnd, IDataObject *pdtobj, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
switch (uMsg)
|
|
{
|
|
case DFM_MERGECONTEXTMENU:
|
|
break;
|
|
|
|
case DFM_INVOKECOMMANDEX:
|
|
{
|
|
DFMICS *pdfmics = (DFMICS *)lParam;
|
|
switch (wParam)
|
|
{
|
|
case DFM_CMD_DELETE:
|
|
hr = _DeleteItems(hwnd, pdtobj);
|
|
break;
|
|
|
|
case DFM_CMD_PROPERTIES:
|
|
break;
|
|
|
|
default:
|
|
// This is common menu items, use the default code.
|
|
hr = S_FALSE;
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
|
|
default:
|
|
hr = E_NOTIMPL;
|
|
break;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CALLBACK CMediaDeviceFolder::_ItemsMenuCB(IShellFolder *psf, HWND hwnd, IDataObject *pdtobj, UINT uMsg, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
IContextMenuCB *pcmcb;
|
|
HRESULT hr = psf->QueryInterface(IID_PPV_ARG(IContextMenuCB, &pcmcb));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pcmcb->CallBack(psf, hwnd, pdtobj, uMsg, wParam, lParam);
|
|
pcmcb->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
class CExtractConstIcon : public IExtractIconW
|
|
{
|
|
public:
|
|
CExtractConstIcon(LPCTSTR psz, UINT uID);
|
|
|
|
// IUnknown
|
|
STDMETHODIMP QueryInterface(REFIID riid, void** ppv);
|
|
STDMETHODIMP_(ULONG) AddRef();
|
|
STDMETHODIMP_(ULONG) Release();
|
|
|
|
// IExtractIconW
|
|
STDMETHODIMP GetIconLocation(UINT uFlags, LPWSTR pszIconFile, UINT cchMax, int* piIndex, UINT* pwFlags);
|
|
STDMETHODIMP Extract(LPCWSTR pszFile, UINT nIconIndex, HICON* phiconLarge, HICON* phiconSmall, UINT nIconSize);
|
|
|
|
private:
|
|
LONG _cRef;
|
|
TCHAR _szPath[MAX_PATH];
|
|
UINT _uID;
|
|
};
|
|
|
|
CExtractConstIcon::CExtractConstIcon(LPCTSTR psz, UINT uID) : _cRef(1), _uID(uID)
|
|
{
|
|
StrCpyN(_szPath, psz, ARRAYSIZE(_szPath));
|
|
}
|
|
|
|
HRESULT CExtractConstIcon::QueryInterface(REFIID riid, void **ppv)
|
|
{
|
|
static const QITAB qit[] =
|
|
{
|
|
QITABENT(CExtractConstIcon, IExtractIconW),
|
|
{ 0 },
|
|
};
|
|
return QISearch(this, qit, riid, ppv);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CExtractConstIcon::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CExtractConstIcon::Release()
|
|
{
|
|
if (InterlockedDecrement(&_cRef))
|
|
return _cRef;
|
|
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
STDMETHODIMP CExtractConstIcon::GetIconLocation(UINT uFlags, LPWSTR pszIconFile, UINT cchMax, int *piIndex, UINT *pwFlags)
|
|
{
|
|
StrCpyN(pszIconFile, _szPath, cchMax);
|
|
*piIndex = _uID;
|
|
*pwFlags = GIL_PERINSTANCE;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CExtractConstIcon::Extract(LPCWSTR pszFile, UINT nIconIndex, HICON* phiconLarge, HICON* phiconSmall, UINT nIconSize)
|
|
{
|
|
return S_FALSE; // do the extract for me
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::_CreateExtractIcon(LPCMEDIADEVITEM pmdi, REFIID riid, void **ppv)
|
|
{
|
|
*ppv = NULL;
|
|
HRESULT hr = E_FAIL;
|
|
|
|
if (_IsRoot())
|
|
{
|
|
#define IDI_AUDIOPLAYER 299
|
|
|
|
CExtractConstIcon *pei = new CExtractConstIcon(TEXT("shell32.dll"), -IDI_AUDIOPLAYER);
|
|
if (pei)
|
|
{
|
|
hr = pei->QueryInterface(riid, ppv);
|
|
pei->Release();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
WCHAR szName[MAX_PATH];
|
|
hr = SHCreateFileExtractIconW(_Name(pmdi, szName, ARRAYSIZE(szName)), _IsFolder(pmdi), riid, ppv);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::GetUIObjectOf(HWND hwnd, UINT cidl, LPCITEMIDLIST *apidl,
|
|
REFIID riid, UINT *prgfInOut, void **ppv)
|
|
{
|
|
HRESULT hr = E_INVALIDARG;
|
|
LPCMEDIADEVITEM pmdi = cidl ? _IsValid(apidl[0]) : NULL;
|
|
|
|
if (pmdi &&
|
|
(IsEqualIID(riid, IID_IExtractIconA) || IsEqualIID(riid, IID_IExtractIconW)))
|
|
{
|
|
hr = _CreateExtractIcon(pmdi, riid, ppv);
|
|
}
|
|
else if (IsEqualIID(riid, IID_IDataObject) && cidl)
|
|
{
|
|
hr = CIDLData_CreateFromIDArray(_pidl, cidl, apidl, (IDataObject**) ppv);
|
|
}
|
|
else if (IsEqualIID(riid, IID_IContextMenu) && pmdi)
|
|
{
|
|
// get the association for these files and lets attempt to
|
|
// build the context menu for the selection.
|
|
|
|
IQueryAssociations *pqa;
|
|
hr = GetUIObjectOf(hwnd, 1, apidl, IID_PPV_ARG_NULL(IQueryAssociations, &pqa));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
HKEY ahk[3];
|
|
DWORD cKeys = SHGetAssocKeys(pqa, ahk, ARRAYSIZE(ahk));
|
|
hr = CDefFolderMenu_Create2(_pidl, hwnd, cidl, apidl,
|
|
this, _ItemsMenuCB, cKeys, ahk,
|
|
(IContextMenu **)ppv);
|
|
|
|
SHRegCloseKeys(ahk, cKeys);
|
|
pqa->Release();
|
|
}
|
|
}
|
|
else if (IsEqualIID(riid, IID_IQueryAssociations) && pmdi)
|
|
{
|
|
TCHAR szName[MAX_PATH];
|
|
hr = SHAssocCreateForFile(_Name(pmdi, szName, ARRAYSIZE(szName)), _IsFolder(pmdi), NULL, riid, ppv);
|
|
}
|
|
else if (IsEqualIID(riid, IID_IDropTarget) && pmdi && _IsFolder(pmdi))
|
|
{
|
|
// If a directory is selected in the view, the drop is going to a folder,
|
|
// so we need to bind to that folder and ask it to create a drop target
|
|
|
|
IShellFolder *psf;
|
|
hr = BindToObject(apidl[0], NULL, IID_PPV_ARG(IShellFolder, &psf));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = psf->CreateViewObject(hwnd, IID_IDropTarget, ppv);
|
|
psf->Release();
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
BOOL CMediaDeviceFolder::_FileInfo(LPCMEDIADEVITEM pmdi, DWORD dwFlags, SHFILEINFO *psfi)
|
|
{
|
|
TCHAR szName[MAX_PATH];
|
|
ASSERT(!(dwFlags & SHGFI_SYSICONINDEX));
|
|
|
|
ZeroMemory(psfi, sizeof(*psfi));
|
|
|
|
return (BOOL)SHGetFileInfo(_Name(pmdi, szName, ARRAYSIZE(szName)), _IsFolder(pmdi), psfi, sizeof(*psfi), SHGFI_USEFILEATTRIBUTES | dwFlags);
|
|
}
|
|
|
|
LPCTSTR CMediaDeviceFolder::_RawName(LPCMEDIADEVITEM pmdi, LPTSTR psz, UINT cch)
|
|
{
|
|
ualstrcpyn(psz, pmdi->szName, cch);
|
|
return psz;
|
|
}
|
|
|
|
LPCTSTR CMediaDeviceFolder::_Name(LPCMEDIADEVITEM pmdi, LPTSTR psz, UINT cch)
|
|
{
|
|
ualstrcpyn(psz, pmdi->szName, cch);
|
|
return PathFindFileName(psz); // some devices use path components in the names of the items
|
|
}
|
|
|
|
LPCTSTR CMediaDeviceFolder::_DisplayName(LPCMEDIADEVITEM pmdi, LPTSTR pszResult, UINT cch)
|
|
{
|
|
SHFILEINFO sfi;
|
|
if (_FileInfo(pmdi, SHGFI_DISPLAYNAME, &sfi))
|
|
{
|
|
StrCpyN(pszResult, sfi.szDisplayName, cch);
|
|
}
|
|
else
|
|
{
|
|
*pszResult = 0;
|
|
}
|
|
return pszResult;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::GetDisplayNameOf(LPCITEMIDLIST pidl, DWORD dwFlags, STRRET *pStrRet)
|
|
{
|
|
HRESULT hr;
|
|
LPCMEDIADEVITEM pmdi = _IsValid(pidl);
|
|
if (pmdi)
|
|
{
|
|
TCHAR szTemp[MAX_PATH];
|
|
if (dwFlags & SHGDN_FORPARSING)
|
|
{
|
|
if (dwFlags & SHGDN_INFOLDER)
|
|
{
|
|
_RawName(pmdi, szTemp, ARRAYSIZE(szTemp)); // relative parse name
|
|
}
|
|
else
|
|
{
|
|
SHGetNameAndFlags(_pidl, dwFlags, szTemp, ARRAYSIZE(szTemp), NULL);
|
|
|
|
TCHAR szName[MAX_PATH];
|
|
PathAppend(szTemp, _RawName(pmdi, szName, ARRAYSIZE(szName)));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
_DisplayName(pmdi, szTemp, ARRAYSIZE(szTemp));
|
|
}
|
|
hr = StringToStrRetW(szTemp, pStrRet);
|
|
}
|
|
else
|
|
hr = E_INVALIDARG;
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceFolder::SetNameOf(HWND hwnd, LPCITEMIDLIST pidl, LPCOLESTR pszName, DWORD dwFlags, LPITEMIDLIST *ppidlOut)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceFolder::GetDetailsEx(LPCITEMIDLIST pidl, const SHCOLUMNID *pscid, VARIANT *pv)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
LPCMEDIADEVITEM pmdi = _IsValid(pidl);
|
|
if (pmdi)
|
|
{
|
|
if (_IsRoot())
|
|
{
|
|
if (IsEqualSCID(*pscid, SCID_DESCRIPTIONID))
|
|
{
|
|
SHDESCRIPTIONID did = {0};
|
|
did.dwDescriptionId = SHDID_COMPUTER_AUDIO;
|
|
hr = InitVariantFromBuffer(pv, &did, sizeof(did));
|
|
}
|
|
else if (IsEqualSCID(*pscid, SCID_CAPACITY))
|
|
{
|
|
pv->vt = VT_UI8;
|
|
pv->ullVal = pmdi->cbTotal.QuadPart;
|
|
hr = S_OK;
|
|
}
|
|
else if (IsEqualSCID(*pscid, SCID_FREESPACE))
|
|
{
|
|
pv->vt = VT_UI8;
|
|
pv->ullVal = pmdi->cbFree.QuadPart;
|
|
hr = S_OK;
|
|
}
|
|
else if (IsEqualSCID(*pscid, SCID_DisplayProperties))
|
|
{
|
|
hr = InitVariantFromStr(pv, TEXT("prop:Name;Type;FreeSpace;Capacity"));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (IsEqualSCID(*pscid, SCID_TYPE))
|
|
{
|
|
TCHAR sz[64];
|
|
hr = InitVariantFromStr(pv, _Type(pmdi, sz, ARRAYSIZE(sz)));
|
|
}
|
|
else if (IsEqualSCID(*pscid, SCID_SIZE))
|
|
{
|
|
pv->vt = VT_UI8;
|
|
pv->ullVal = pmdi->cbTotal.QuadPart;
|
|
hr = S_OK;
|
|
}
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
LPCTSTR CMediaDeviceFolder::_Type(LPCMEDIADEVITEM pmdi, LPTSTR psz, UINT cch)
|
|
{
|
|
*psz = 0;
|
|
SHFILEINFO sfi;
|
|
if (_FileInfo(pmdi, SHGFI_TYPENAME, &sfi))
|
|
StrCpyN(psz, sfi.szTypeName, cch);
|
|
return psz;
|
|
}
|
|
|
|
static const struct
|
|
{
|
|
UINT iTitle;
|
|
UINT cchCol;
|
|
UINT iFmt;
|
|
const SHCOLUMNID *pscid;
|
|
}
|
|
c_aMediaDeviceColumns[] =
|
|
{
|
|
{IDS_NAME_COL, 40, LVCFMT_LEFT, &SCID_NAME},
|
|
{IDS_SIZE_COL, 10, LVCFMT_RIGHT, &SCID_SIZE},
|
|
{IDS_TYPE_COL, 30, LVCFMT_LEFT, &SCID_TYPE},
|
|
{IDS_MODIFIED_COL, 20, LVCFMT_LEFT, &SCID_WRITETIME},
|
|
};
|
|
|
|
HRESULT CMediaDeviceFolder::GetDetailsOf(LPCITEMIDLIST pidl, UINT iColumn, SHELLDETAILS *pDetail)
|
|
{
|
|
if (iColumn >= ARRAYSIZE(c_aMediaDeviceColumns))
|
|
return E_NOTIMPL;
|
|
|
|
HRESULT hr = S_OK;
|
|
TCHAR szTemp[MAX_PATH];
|
|
szTemp[0] = 0;
|
|
|
|
if (NULL == pidl)
|
|
{
|
|
pDetail->fmt = c_aMediaDeviceColumns[iColumn].iFmt;
|
|
pDetail->cxChar = c_aMediaDeviceColumns[iColumn].cchCol;
|
|
LoadString(m_hInst, c_aMediaDeviceColumns[iColumn].iTitle, szTemp, ARRAYSIZE(szTemp));
|
|
}
|
|
else
|
|
{
|
|
LPCMEDIADEVITEM pmdi = _IsValid(pidl);
|
|
if (pmdi)
|
|
{
|
|
// return the property to the caller that is being requested, this is based on the
|
|
// list of columns we gave out when the view was created.
|
|
switch (iColumn)
|
|
{
|
|
case DEV_COL_NAME:
|
|
_DisplayName(pmdi, szTemp, ARRAYSIZE(szTemp));
|
|
break;
|
|
|
|
case DEV_COL_SIZE:
|
|
if (!_IsFolder(pmdi))
|
|
{
|
|
ULARGE_INTEGER ullSize = pmdi->cbTotal;
|
|
StrFormatKBSize(ullSize.QuadPart, szTemp, ARRAYSIZE(szTemp));
|
|
}
|
|
break;
|
|
|
|
case DEV_COL_TYPE:
|
|
_Type(pmdi, szTemp, ARRAYSIZE(szTemp));
|
|
break;
|
|
|
|
case DEV_COL_MODIFIED:
|
|
SHFormatDateTime(&pmdi->ftModified, NULL, szTemp, ARRAYSIZE(szTemp));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return StringToStrRetW(szTemp, &(pDetail->str));
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceFolder::MapColumnToSCID(UINT iColumn, SHCOLUMNID *pscid)
|
|
{
|
|
HRESULT hr = E_INVALIDARG;
|
|
ZeroMemory(pscid, sizeof(*pscid));
|
|
|
|
if (!_IsRoot())
|
|
{
|
|
if (iColumn < ARRAYSIZE(c_aMediaDeviceColumns))
|
|
{
|
|
*pscid = *c_aMediaDeviceColumns[iColumn].pscid;
|
|
hr = S_OK;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
// IDelegateFolder
|
|
HRESULT CMediaDeviceFolder::SetItemAlloc(IMalloc *pmalloc)
|
|
{
|
|
_fDelegate = TRUE;
|
|
|
|
IUnknown_Set((IUnknown**)&_pmalloc, pmalloc);
|
|
return S_OK;
|
|
}
|
|
|
|
CMediaDeviceEnum::CMediaDeviceEnum(CMediaDeviceFolder *pmdf, DWORD grfFlags) : _cRef(1), _grfFlags(grfFlags), _iIndex(0)
|
|
{
|
|
_pmdf = pmdf;
|
|
_pmdf->AddRef();
|
|
|
|
// force full enum on this thread. avoid problem where calls made on the
|
|
// marshaled thread fail due to authentication
|
|
_Init();
|
|
|
|
DllAddRef();
|
|
}
|
|
|
|
int CALLBACK _FreeItems(ITEMIDLIST *pidl, IShellFolder *psf)
|
|
{
|
|
ILFree(pidl);
|
|
return 1;
|
|
}
|
|
|
|
CMediaDeviceEnum::~CMediaDeviceEnum()
|
|
{
|
|
if ((HDPA)_dpaItems)
|
|
_dpaItems.DestroyCallbackEx(_FreeItems, (IShellFolder *)NULL);
|
|
|
|
_pmdf->Release();
|
|
DllRelease();
|
|
}
|
|
|
|
HRESULT CMediaDeviceEnum::QueryInterface(REFIID riid, void **ppv)
|
|
{
|
|
static const QITAB qit[] =
|
|
{
|
|
QITABENT(CMediaDeviceEnum, IEnumIDList), // IID_IEnumIDList
|
|
{ 0 },
|
|
};
|
|
return QISearch(this, qit, riid, ppv);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CMediaDeviceEnum::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CMediaDeviceEnum::Release()
|
|
{
|
|
if (InterlockedDecrement(&_cRef))
|
|
return _cRef;
|
|
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
BOOL CMediaDeviceEnum::_FilterItem(BOOL bFolder)
|
|
{
|
|
if (bFolder)
|
|
{
|
|
if (!(_grfFlags & SHCONTF_FOLDERS))
|
|
{
|
|
return TRUE;
|
|
}
|
|
}
|
|
else if (!(_grfFlags & SHCONTF_NONFOLDERS))
|
|
{
|
|
return TRUE;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
BOOL CMediaDeviceFolder::_ShouldShowDevice(IWMDMDevice *pdev)
|
|
{
|
|
BOOL bShow = TRUE;
|
|
if (_fDelegate)
|
|
{
|
|
DWORD dwPowerSource, dwPercentRemaining;
|
|
pdev->GetPowerSource(&dwPowerSource, &dwPercentRemaining);
|
|
|
|
// heuristic to detect file system driver supported devices.
|
|
// if it runs on batteries assume it is not a file system device
|
|
|
|
bShow = (WMDM_POWER_CAP_BATTERY & dwPowerSource);
|
|
}
|
|
return bShow;
|
|
}
|
|
|
|
|
|
HRESULT CMediaDeviceEnum::_Init()
|
|
{
|
|
if ((HDPA)_dpaItems)
|
|
return S_OK;
|
|
|
|
HRESULT hr = _dpaItems.Create(10) ? S_OK : E_OUTOFMEMORY;
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (_pmdf->_IsRoot())
|
|
{
|
|
IWMDMEnumDevice *pEnumDevice;
|
|
hr = _pmdf->_EnumDevices(&pEnumDevice);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IWMDMDevice *pdev;
|
|
ULONG ulFetched;
|
|
while (S_OK == pEnumDevice->Next(1, &pdev, &ulFetched))
|
|
{
|
|
if (_pmdf->_ShouldShowDevice(pdev))
|
|
{
|
|
ITEMIDLIST *pidl;
|
|
BOOL bFolder;
|
|
hr = _pmdf->_IDListForDevice(pdev, &pidl, &bFolder);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (_FilterItem(bFolder) ||
|
|
(-1 == _dpaItems.AppendPtr(pidl)))
|
|
{
|
|
ILFree(pidl);
|
|
}
|
|
}
|
|
}
|
|
pdev->Release();
|
|
}
|
|
pEnumDevice->Release();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
IWMDMEnumStorage *penum;
|
|
hr = _pmdf->_EnumStorage(&penum);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ULONG ulFetched;
|
|
IWMDMStorage *rgpstg[1]; // 1 is to WORKAROUND Rio600 faults on > 1
|
|
while (SUCCEEDED(penum->Next(ARRAYSIZE(rgpstg), rgpstg, &ulFetched)))
|
|
{
|
|
for (UINT i = 0; i < ulFetched; i++)
|
|
{
|
|
ITEMIDLIST *pidl;
|
|
BOOL bFolder;
|
|
hr = _pmdf->_IDListForStorage(rgpstg[i], &pidl, &bFolder);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (_FilterItem(bFolder) ||
|
|
(-1 == _dpaItems.AppendPtr(pidl)))
|
|
{
|
|
ILFree(pidl);
|
|
}
|
|
}
|
|
rgpstg[i]->Release();
|
|
}
|
|
|
|
if (ulFetched != ARRAYSIZE(rgpstg))
|
|
break;
|
|
}
|
|
penum->Release();
|
|
}
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CMediaDeviceEnum::Next(ULONG celt, LPITEMIDLIST *rgelt, ULONG *pceltFetched)
|
|
{
|
|
HRESULT hr = _Init();
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = S_FALSE;
|
|
if (_iIndex < _dpaItems.GetPtrCount())
|
|
{
|
|
hr = SHILClone(_dpaItems.GetPtr(_iIndex++), rgelt);
|
|
}
|
|
if (pceltFetched)
|
|
*pceltFetched = (hr == S_OK) ? 1 : 0;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
CMediaDeviceDropTarget::CMediaDeviceDropTarget(CMediaDeviceFolder *pmdf, HWND hwnd) :
|
|
_cRef(1), _pmdf(pmdf), _hwnd(hwnd), _grfKeyStateLast(-1)
|
|
{
|
|
_pmdf->AddRef();
|
|
// use the FTM to make the call back interface calls unmarshaled
|
|
CoCreateFreeThreadedMarshaler(SAFECAST(this, IWMDMProgress2 *), &_punkFTM);
|
|
DllAddRef();
|
|
}
|
|
|
|
CMediaDeviceDropTarget::~CMediaDeviceDropTarget()
|
|
{
|
|
DragLeave();
|
|
_pmdf->Release();
|
|
if (_punkFTM)
|
|
_punkFTM->Release();
|
|
DllRelease();
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CMediaDeviceDropTarget::AddRef()
|
|
{
|
|
return InterlockedIncrement(&_cRef);
|
|
}
|
|
|
|
STDMETHODIMP_(ULONG) CMediaDeviceDropTarget::Release()
|
|
{
|
|
if (InterlockedDecrement(&_cRef))
|
|
return _cRef;
|
|
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
HRESULT CMediaDeviceDropTarget::QueryInterface(REFIID riid, void** ppv)
|
|
{
|
|
static const QITAB qit[] = {
|
|
QITABENT(CMediaDeviceDropTarget, IDropTarget),
|
|
QITABENTMULTI(CMediaDeviceDropTarget, IWMDMProgress, IWMDMProgress2),
|
|
QITABENT (CMediaDeviceDropTarget, IWMDMProgress2),
|
|
QITABENT(CMediaDeviceDropTarget, INamespaceWalkCB),
|
|
{ 0 },
|
|
};
|
|
HRESULT hr = QISearch(this, qit, riid, ppv);
|
|
if (FAILED(hr) && (IID_IMarshal == riid) && _punkFTM)
|
|
hr = _punkFTM->QueryInterface(riid, ppv);
|
|
return hr;
|
|
}
|
|
|
|
|
|
CLIPFORMAT g_cfPreferredDropEffect = 0; // IMPLEMENT
|
|
CLIPFORMAT g_cfLogicalPerformedDropEffect = 0;
|
|
CLIPFORMAT g_cfPerformedDropEffect = 0;
|
|
|
|
#define POPUP_NONDEFAULTDD 1
|
|
#define DDIDM_COPY 2
|
|
#define DDIDM_MOVE 3
|
|
#define DDIDM_LINK 4
|
|
|
|
DWORD CMediaDeviceDropTarget::_GetDropEffect(DWORD *pdwEffect, DWORD grfKeyState)
|
|
{
|
|
DWORD dwEffectReturned = DROPEFFECT_NONE;
|
|
switch (grfKeyState & (MK_CONTROL | MK_SHIFT | MK_ALT))
|
|
{
|
|
case MK_CONTROL: dwEffectReturned = DROPEFFECT_COPY; break;
|
|
case MK_SHIFT: dwEffectReturned = DROPEFFECT_MOVE; break;
|
|
case MK_SHIFT | MK_CONTROL: dwEffectReturned = DROPEFFECT_LINK; break;
|
|
case MK_ALT: dwEffectReturned = DROPEFFECT_LINK; break;
|
|
|
|
default:
|
|
{
|
|
// no modifier keys:
|
|
// if the data object contains a preferred drop effect, try to use it
|
|
DWORD dwPreferred = DataObj_GetDWORD(_pdtobj, g_cfPreferredDropEffect, DROPEFFECT_NONE) & *pdwEffect;
|
|
if (dwPreferred)
|
|
{
|
|
if (dwPreferred & DROPEFFECT_MOVE)
|
|
{
|
|
dwEffectReturned = DROPEFFECT_MOVE;
|
|
}
|
|
else if (dwPreferred & DROPEFFECT_COPY)
|
|
{
|
|
dwEffectReturned = DROPEFFECT_COPY;
|
|
}
|
|
else if (dwPreferred & DROPEFFECT_LINK)
|
|
{
|
|
dwEffectReturned = DROPEFFECT_LINK;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwEffectReturned = DROPEFFECT_COPY;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
return dwEffectReturned;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::DragEnter(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
|
|
{
|
|
IUnknown_Set((IUnknown **)&_pdtobj, pdtobj);
|
|
|
|
_grfKeyStateLast = grfKeyState;
|
|
|
|
if (pdwEffect)
|
|
*pdwEffect = _dwEffectLastReturned = _GetDropEffect(pdwEffect, grfKeyState);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::DragOver(DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
|
|
{
|
|
// has the key state changed? if not then lets return the previously cached
|
|
// version, otherwise recompute.
|
|
|
|
if (_grfKeyStateLast == grfKeyState)
|
|
{
|
|
if (*pdwEffect)
|
|
*pdwEffect = _dwEffectLastReturned;
|
|
}
|
|
else if (*pdwEffect)
|
|
{
|
|
*pdwEffect = _GetDropEffect(pdwEffect, grfKeyState);
|
|
}
|
|
|
|
_dwEffectLastReturned = *pdwEffect;
|
|
_grfKeyStateLast = grfKeyState;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::DragLeave()
|
|
{
|
|
ATOMICRELEASE(_pdtobj);
|
|
return S_OK;
|
|
}
|
|
|
|
class COPY_THREAD_DATA
|
|
{
|
|
public:
|
|
COPY_THREAD_DATA(CMediaDeviceDropTarget *pmddt) : _pmddt(pmddt), _pstm(NULL)
|
|
{
|
|
_pmddt->AddRef();
|
|
}
|
|
|
|
~COPY_THREAD_DATA()
|
|
{
|
|
_pmddt->Release();
|
|
if (_pstm)
|
|
_pstm->Release();
|
|
}
|
|
|
|
HRESULT Marshal(IDataObject *pdtobj)
|
|
{
|
|
return CoMarshalInterThreadInterfaceInStream(IID_IDataObject, pdtobj, &_pstm);
|
|
}
|
|
|
|
HRESULT UnMarshal(IDataObject **ppdtobj)
|
|
{
|
|
HRESULT hr = CoGetInterfaceAndReleaseStream(_pstm, IID_PPV_ARG(IDataObject, ppdtobj));
|
|
_pstm = NULL;
|
|
return hr;
|
|
}
|
|
|
|
CMediaDeviceDropTarget *_pmddt;
|
|
|
|
private:
|
|
IStream *_pstm; // IStream for marshaling the IDataObject
|
|
};
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::Drop(IDataObject *pdtobj, DWORD grfKeyState, POINTL pt, DWORD *pdwEffect)
|
|
{
|
|
*pdwEffect = DROPEFFECT_NONE; // incase of failure
|
|
|
|
// determine the type of operation to performed, if the right button is down
|
|
// then lets display the menu, otherwise base it on the drop effect
|
|
|
|
UINT idCmd = 0; // Choice from drop popup menu
|
|
|
|
#if 0
|
|
if (!(_grfKeyStateLast & MK_LBUTTON))
|
|
{
|
|
HMENU hMenu = SHLoadPopupMenu(m_hInst, POPUP_NONDEFAULTDD);
|
|
if (!hMenu)
|
|
{
|
|
DragLeave();
|
|
return E_FAIL;
|
|
}
|
|
|
|
SetMenuDefaultItem(hMenu, POPUP_NONDEFAULTDD, FALSE);
|
|
idCmd = TrackPopupMenu(hMenu,
|
|
TPM_RETURNCMD|TPM_RIGHTBUTTON|TPM_LEFTALIGN,
|
|
pt.x, pt.y, 0, _hwnd, NULL);
|
|
DestroyMenu(hMenu);
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
switch (_GetDropEffect(pdwEffect, grfKeyState))
|
|
{
|
|
case DROPEFFECT_COPY: idCmd = DDIDM_COPY; break;
|
|
case DROPEFFECT_MOVE: idCmd = DDIDM_MOVE; break;
|
|
}
|
|
}
|
|
|
|
// now perform the operation, based on the command ID we have.
|
|
|
|
HRESULT hr = E_FAIL;
|
|
switch (idCmd)
|
|
{
|
|
case DDIDM_COPY:
|
|
COPY_THREAD_DATA *pctd = new COPY_THREAD_DATA(this);
|
|
if (pctd)
|
|
{
|
|
if (SUCCEEDED(pctd->Marshal(pdtobj)))
|
|
{
|
|
if (!SHCreateThread(_DoCopyThreadProc, pctd, CTF_COINIT, NULL))
|
|
{
|
|
delete pctd;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
|
|
#if 0
|
|
// success so lets populate the new changes to the effect
|
|
|
|
if (SUCCEEDED(hr) && *pdwEffect)
|
|
{
|
|
DataObj_SetDWORD(pdtobj, g_cfLogicalPerformedDropEffect, *pdwEffect);
|
|
DataObj_SetDWORD(pdtobj, g_cfPerformedDropEffect, *pdwEffect);
|
|
}
|
|
#endif
|
|
|
|
DragLeave();
|
|
return hr;
|
|
}
|
|
|
|
DWORD CMediaDeviceDropTarget::_DoCopyThreadProc(void *pv)
|
|
{
|
|
COPY_THREAD_DATA *pctd = (COPY_THREAD_DATA *)pv;
|
|
IDataObject *pdtobj;
|
|
HRESULT hr = pctd->UnMarshal(&pdtobj);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
pctd->_pmddt->_DoCopy(pdtobj);
|
|
pdtobj->Release();
|
|
}
|
|
delete pctd;
|
|
return 0;
|
|
}
|
|
|
|
HRESULT CMediaDeviceDropTarget::_DoCopy(IDataObject *pdtobj)
|
|
{
|
|
IProgressDialog *ppd;
|
|
HRESULT hr = _CreateProgress(_hwnd, &ppd);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ppd->SetTitle(L"Copying Files...");
|
|
|
|
IWMDMStorageControl *pstgCtrl;
|
|
hr = _pmdf->_Storage(IID_PPV_ARG(IWMDMStorageControl, &pstgCtrl));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
INamespaceWalk *pnsw;
|
|
hr = CoCreateInstance(CLSID_NamespaceWalker, NULL, CLSCTX_INPROC, IID_PPV_ARG(INamespaceWalk, &pnsw));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pnsw->Walk(pdtobj, 0, 4, SAFECAST(this, INamespaceWalkCB *));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
UINT cItems;
|
|
LPITEMIDLIST *ppidls;
|
|
hr = pnsw->GetIDArrayResult(&cItems, &ppidls);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
for (UINT i = 0; SUCCEEDED(hr) && (i < cItems); i++)
|
|
{
|
|
TCHAR szPath[MAX_PATH];
|
|
if (SHGetPathFromIDList(ppidls[i], szPath))
|
|
{
|
|
ppd->SetLine(2, PathFindFileName(szPath), FALSE, NULL);
|
|
ppd->SetProgress64(_ulProgressCurrent, _ulProgressTotal);
|
|
|
|
_ppd = ppd; // used by the callback interface
|
|
|
|
IWMDMStorage *pstgNew;
|
|
hr = pstgCtrl->Insert(WMDM_MODE_BLOCK | WMDM_CONTENT_FILE,
|
|
szPath, NULL, SAFECAST(this, IWMDMProgress2 *), &pstgNew);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_pmdf->_ChangeNotifyObj(SHCNE_CREATE, pstgNew);
|
|
pstgNew->Release();
|
|
|
|
ppd->SetProgress64(_ulProgressCurrent, _ulProgressTotal);
|
|
|
|
if (ppd->HasUserCancelled())
|
|
{
|
|
hr = HRESULT_FROM_WIN32(ERROR_CANCELLED);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
FreeIDListArray(ppidls, cItems);
|
|
|
|
// free space changed on device, notify
|
|
// SHChangeNotify(SHCNE_UPDATEITEM, 0, (LPCTSTR)_pmdf->GetIDList(), NULL);
|
|
}
|
|
}
|
|
pnsw->Release();
|
|
}
|
|
pstgCtrl->Release();
|
|
}
|
|
ppd->StopProgressDialog();
|
|
ppd->Release();
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
// IWMDMProgress
|
|
// since we use the FTM we get called here on an MTA thread
|
|
STDMETHODIMP CMediaDeviceDropTarget::Begin(DWORD dwEstimatedTicks)
|
|
{
|
|
_ulThisFile = dwEstimatedTicks;
|
|
return S_OK; // dwEstimatedTicks size in bytes of file
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::Progress(DWORD dwTranspiredTicks)
|
|
{
|
|
_ppd->SetProgress64(_ulProgressCurrent + dwTranspiredTicks, _ulProgressTotal);
|
|
|
|
if (_ulThisFile == dwTranspiredTicks)
|
|
_ulProgressCurrent += _ulThisFile; // end of this file, advance total
|
|
|
|
return _ppd->HasUserCancelled() ? HRESULT_FROM_WIN32(ERROR_CANCELLED) : S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::End()
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
// IWMDMProgress2
|
|
STDMETHODIMP CMediaDeviceDropTarget::End2(HRESULT hrCompletionCode)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
// INamespaceWalkCB
|
|
STDMETHODIMP CMediaDeviceDropTarget::FoundItem(IShellFolder *psf, LPCITEMIDLIST pidl)
|
|
{
|
|
IShellFolder2 *psf2;
|
|
if (SUCCEEDED(psf->QueryInterface(IID_PPV_ARG(IShellFolder2, &psf2))))
|
|
{
|
|
ULONGLONG ul;
|
|
if (SUCCEEDED(GetLongProperty(psf2, pidl, &SCID_SIZE, &ul)))
|
|
{
|
|
// TCHAR szName[MAX_PATH];
|
|
// DisplayNameOf(psf, pidl, SHGDN_FORPARSING | SHGDN_INFOLDER, szName, ARRAYSIZE(szName));
|
|
_ulProgressTotal += ul;
|
|
}
|
|
psf2->Release();
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::EnterFolder(IShellFolder *psf, LPCITEMIDLIST pidl)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CMediaDeviceDropTarget::LeaveFolder(IShellFolder *psf, LPCITEMIDLIST pidl)
|
|
{
|
|
return S_OK;
|
|
}
|
|
|