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.
1470 lines
37 KiB
1470 lines
37 KiB
#include "shellprv.h"
|
|
#pragma hdrstop
|
|
|
|
#include "debug.h"
|
|
#include "stgutil.h"
|
|
#include "ids.h"
|
|
#include "tngen\ctngen.h"
|
|
#include "tlist.h"
|
|
#include "thumbutil.h"
|
|
|
|
void SHGetThumbnailSizeForThumbsDB(SIZE *psize);
|
|
|
|
STDAPI CThumbStore_CreateInstance(IUnknown* punkOuter, REFIID riid, void **ppv);
|
|
|
|
class CThumbStore : public IShellImageStore,
|
|
public IPersistFolder,
|
|
public IPersistFile,
|
|
public CComObjectRootEx<CComMultiThreadModel>,
|
|
public CComCoClass< CThumbStore,&CLSID_ShellThumbnailDiskCache >
|
|
{
|
|
struct CATALOG_ENTRY
|
|
{
|
|
DWORD cbSize;
|
|
DWORD dwIndex;
|
|
FILETIME ftTimeStamp;
|
|
WCHAR szName[1];
|
|
};
|
|
|
|
struct CATALOG_HEADER
|
|
{
|
|
WORD cbSize;
|
|
WORD wVersion;
|
|
DWORD dwEntryCount;
|
|
SIZE szThumbnailExtent;
|
|
};
|
|
|
|
public:
|
|
BEGIN_COM_MAP(CThumbStore)
|
|
COM_INTERFACE_ENTRY_IID(IID_IShellImageStore,IShellImageStore)
|
|
COM_INTERFACE_ENTRY(IPersistFolder)
|
|
COM_INTERFACE_ENTRY(IPersistFile)
|
|
END_COM_MAP()
|
|
|
|
DECLARE_NOT_AGGREGATABLE(CThumbStore)
|
|
|
|
CThumbStore();
|
|
~CThumbStore();
|
|
|
|
// IPersist
|
|
STDMETHOD(GetClassID)(CLSID *pClassID);
|
|
|
|
// IPersistFolder
|
|
STDMETHOD(Initialize)(LPCITEMIDLIST pidl);
|
|
|
|
// IPersistFile
|
|
STDMETHOD (IsDirty)(void);
|
|
STDMETHOD (Load)(LPCWSTR pszFileName, DWORD dwMode);
|
|
STDMETHOD (Save)(LPCWSTR pszFileName, BOOL fRemember);
|
|
STDMETHOD (SaveCompleted)(LPCWSTR pszFileName);
|
|
STDMETHOD (GetCurFile)(LPWSTR *ppszFileName);
|
|
|
|
// IImageCache
|
|
STDMETHOD (Open)(DWORD dwMode, DWORD *pdwLock);
|
|
STDMETHOD (Create)(DWORD dwMode, DWORD *pdwLock);
|
|
STDMETHOD (Close)(DWORD const *pdwLock);
|
|
STDMETHOD (Commit)(DWORD const *pdwLock);
|
|
STDMETHOD (ReleaseLock)(DWORD const *pdwLock);
|
|
STDMETHOD (IsLocked)(THIS);
|
|
|
|
STDMETHOD (GetMode)(DWORD *pdwMode);
|
|
STDMETHOD (GetCapabilities)(DWORD *pdwCapMask);
|
|
|
|
STDMETHOD (AddEntry)(LPCWSTR pszName, const FILETIME *pftTimeStamp, DWORD dwMode, HBITMAP hImage);
|
|
STDMETHOD (GetEntry)(LPCWSTR pszName, DWORD dwMode, HBITMAP *phImage);
|
|
STDMETHOD (DeleteEntry)(LPCWSTR pszName);
|
|
STDMETHOD (IsEntryInStore)(LPCWSTR pszName, FILETIME *pftTimeStamp);
|
|
|
|
STDMETHOD (Enum)(IEnumShellImageStore ** ppEnum);
|
|
|
|
protected:
|
|
friend class CEnumThumbStore;
|
|
|
|
HRESULT LoadCatalog(void);
|
|
HRESULT SaveCatalog(void);
|
|
|
|
HRESULT FindStreamID(LPCWSTR pszName, DWORD *pdwStream, CATALOG_ENTRY **ppEntry);
|
|
HRESULT GetEntryStream(DWORD dwStream, DWORD dwMode, IStream **ppStream);
|
|
DWORD GetAccessMode(DWORD dwMode, BOOL fStream);
|
|
|
|
DWORD AcquireLock(void);
|
|
void ReleaseLock(DWORD dwLock);
|
|
|
|
BOOL InitCodec(void);
|
|
|
|
HRESULT PrepImage(HBITMAP *phBmp, SIZE * prgSize, void ** ppBits);
|
|
BOOL DecompressImage(void * pvInBuffer, ULONG ulBufferSize, HBITMAP *phBmp);
|
|
BOOL CompressImage(HBITMAP hBmp, void ** ppvOutBuffer, ULONG * plBufSize);
|
|
|
|
HRESULT WriteImage(IStream *pStream, HBITMAP hBmp);
|
|
HRESULT ReadImage(IStream *pStream, HBITMAP *phBmp);
|
|
BOOL _MatchNodeName(CATALOG_ENTRY *pNode, LPCWSTR pszName);
|
|
|
|
HRESULT _InitFromPath(LPCTSTR pszPath, DWORD dwMode);
|
|
void _SetAttribs(BOOL bForce);
|
|
|
|
CATALOG_HEADER m_rgHeader;
|
|
CList<CATALOG_ENTRY> m_rgCatalog;
|
|
IStorage *_pStorageThumb;
|
|
DWORD _dwModeStorage;
|
|
|
|
DWORD m_dwModeAllow;
|
|
WCHAR m_szPath[MAX_PATH];
|
|
DWORD m_dwMaxIndex;
|
|
DWORD m_dwCatalogChange;
|
|
|
|
// Crit section used to protect the internals
|
|
CRITICAL_SECTION m_csInternals;
|
|
|
|
// needed for this object to be free-threaded... so that
|
|
// we can query the catalog from the main thread whilst icons are
|
|
// being read and written from the main thread.
|
|
CRITICAL_SECTION m_csLock;
|
|
|
|
DWORD m_dwLock;
|
|
int m_fLocked;
|
|
CThumbnailFCNContainer * m_pJPEGCodec;
|
|
};
|
|
|
|
HRESULT CEnumThumbStore_Create(CThumbStore * pThis, IEnumShellImageStore ** ppEnum);
|
|
|
|
class CEnumThumbStore : public IEnumShellImageStore,
|
|
public CComObjectRoot
|
|
{
|
|
public:
|
|
BEGIN_COM_MAP(CEnumThumbStore)
|
|
COM_INTERFACE_ENTRY_IID(IID_IEnumShellImageStore,IEnumShellImageStore)
|
|
END_COM_MAP()
|
|
|
|
CEnumThumbStore();
|
|
~CEnumThumbStore();
|
|
|
|
STDMETHOD (Reset)(void);
|
|
STDMETHOD (Next)(ULONG celt, PENUMSHELLIMAGESTOREDATA *prgElt, ULONG *pceltFetched);
|
|
STDMETHOD (Skip)(ULONG celt);
|
|
STDMETHOD (Clone)(IEnumShellImageStore ** pEnum);
|
|
|
|
protected:
|
|
friend HRESULT CEnumThumbStore_Create(CThumbStore *pThis, IEnumShellImageStore **ppEnum);
|
|
|
|
CThumbStore * m_pStore;
|
|
CLISTPOS m_pPos;
|
|
DWORD m_dwCatalogChange;
|
|
};
|
|
|
|
|
|
#define THUMB_FILENAME L"Thumbs.db"
|
|
#define CATALOG_STREAM L"Catalog"
|
|
|
|
#define CATALOG_VERSION 0x0006
|
|
#define CATALOG_VERSION_XPGOLD 0x0005
|
|
#define STREAMFLAGS_JPEG 0x0001
|
|
#define STREAMFLAGS_DIB 0x0002
|
|
|
|
struct STREAM_HEADER
|
|
{
|
|
DWORD cbSize;
|
|
DWORD dwFlags;
|
|
ULONG ulSize;
|
|
};
|
|
|
|
void GenerateStreamName(LPWSTR pszBuffer, DWORD cbSize, DWORD dwNumber);
|
|
HRESULT ReadImage(IStream *pStream, HBITMAP *phImage);
|
|
HRESULT WriteImage(IStream *pStream, HBITMAP hImage);
|
|
BITMAPINFO *BitmapToDIB(HBITMAP hBmp);
|
|
|
|
CThumbStore::CThumbStore()
|
|
{
|
|
m_szPath[0] = 0;
|
|
m_rgHeader.dwEntryCount = 0;
|
|
m_rgHeader.wVersion = CATALOG_VERSION;
|
|
m_rgHeader.cbSize = sizeof(m_rgHeader);
|
|
SHGetThumbnailSizeForThumbsDB(&m_rgHeader.szThumbnailExtent);
|
|
|
|
m_dwMaxIndex = 0;
|
|
m_dwModeAllow = STGM_READWRITE;
|
|
|
|
// this counter is inc'd everytime the catalog changes so that we know when it
|
|
// must be committed and so enumerators can detect the list has changed...
|
|
m_dwCatalogChange = 0;
|
|
|
|
m_fLocked = 0;
|
|
InitializeCriticalSection(&m_csLock);
|
|
InitializeCriticalSection(&m_csInternals);
|
|
}
|
|
|
|
CThumbStore::~CThumbStore()
|
|
{
|
|
CLISTPOS pCur = m_rgCatalog.GetHeadPosition();
|
|
while (pCur != NULL)
|
|
{
|
|
CATALOG_ENTRY *pNode = m_rgCatalog.GetNext(pCur);
|
|
ASSERT(pNode != NULL);
|
|
|
|
LocalFree((void *) pNode);
|
|
}
|
|
|
|
m_rgCatalog.RemoveAll();
|
|
|
|
if (_pStorageThumb)
|
|
{
|
|
_pStorageThumb->Release();
|
|
}
|
|
|
|
if (m_pJPEGCodec)
|
|
{
|
|
delete m_pJPEGCodec;
|
|
}
|
|
|
|
// assume these are free, we are at ref count zero, no one should still be calling us...
|
|
DeleteCriticalSection(&m_csLock);
|
|
DeleteCriticalSection(&m_csInternals);
|
|
}
|
|
|
|
STDAPI CThumbStore_CreateInstance(IUnknown* punkOuter, REFIID riid, void **ppv)
|
|
{
|
|
return CComCreator< CComObject< CThumbStore > >::CreateInstance((void *)punkOuter, riid, (void **)ppv);
|
|
}
|
|
|
|
DWORD CThumbStore::AcquireLock(void)
|
|
{
|
|
EnterCriticalSection(&m_csLock);
|
|
|
|
// inc the lock (we use a counter because we may reenter this on the same thread)
|
|
m_fLocked++;
|
|
|
|
// Never return a lock signature of zero, because that means "not locked".
|
|
if (++m_dwLock == 0)
|
|
++m_dwLock;
|
|
return m_dwLock;
|
|
}
|
|
|
|
void CThumbStore::ReleaseLock(DWORD dwLock)
|
|
{
|
|
if (dwLock)
|
|
{
|
|
ASSERT(m_fLocked);
|
|
m_fLocked--;
|
|
LeaveCriticalSection(&m_csLock);
|
|
}
|
|
}
|
|
|
|
// the structure of the catalog is simple, it is a just a header stream
|
|
HRESULT CThumbStore::LoadCatalog()
|
|
{
|
|
HRESULT hr;
|
|
if (_pStorageThumb == NULL)
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
}
|
|
else if (m_rgHeader.dwEntryCount != 0)
|
|
{
|
|
// it is already loaded....
|
|
hr = S_OK;
|
|
}
|
|
else
|
|
{
|
|
// open the catalog stream...
|
|
IStream *pCatalog;
|
|
hr = _pStorageThumb->OpenStream(CATALOG_STREAM, NULL, GetAccessMode(STGM_READ, TRUE), NULL, &pCatalog);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
EnterCriticalSection(&m_csInternals);
|
|
|
|
// now read in the catalog from the stream ...
|
|
hr = IStream_Read(pCatalog, &m_rgHeader, sizeof(m_rgHeader));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
SIZE szCurrentSize;
|
|
SHGetThumbnailSizeForThumbsDB(&szCurrentSize);
|
|
if ((m_rgHeader.cbSize != sizeof(m_rgHeader)) || (m_rgHeader.wVersion != CATALOG_VERSION) ||
|
|
(m_rgHeader.szThumbnailExtent.cx != szCurrentSize.cx) || (m_rgHeader.szThumbnailExtent.cy != szCurrentSize.cy))
|
|
{
|
|
if (m_rgHeader.wVersion == CATALOG_VERSION_XPGOLD)
|
|
{
|
|
hr = STG_E_DOCFILECORRUPT; // SECURITY: Many issues encrypting XPGOLD thumbnail databases, just delete it
|
|
_pStorageThumb->Release();
|
|
_pStorageThumb = NULL;
|
|
}
|
|
else
|
|
{
|
|
_SetAttribs(TRUE); // SECURITY: Old formats can't be encrypted
|
|
hr = STG_E_OLDFORMAT;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for (UINT iEntry = 0; (iEntry < m_rgHeader.dwEntryCount) && SUCCEEDED(hr); iEntry++)
|
|
{
|
|
DWORD cbSize;
|
|
hr = IStream_Read(pCatalog, &cbSize, sizeof(cbSize));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ASSERT(cbSize <= sizeof(CATALOG_ENTRY) + sizeof(WCHAR) * MAX_PATH);
|
|
|
|
CATALOG_ENTRY *pEntry = (CATALOG_ENTRY *)LocalAlloc(LPTR, cbSize);
|
|
if (pEntry)
|
|
{
|
|
pEntry->cbSize = cbSize;
|
|
|
|
// read the rest with out the size on the front...
|
|
hr = IStream_Read(pCatalog, ((BYTE *)pEntry + sizeof(cbSize)), cbSize - sizeof(cbSize));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
CLISTPOS pCur = m_rgCatalog.AddTail(pEntry);
|
|
if (pCur)
|
|
{
|
|
if (m_dwMaxIndex < pEntry->dwIndex)
|
|
{
|
|
m_dwMaxIndex = pEntry->dwIndex;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
// reset the catalog header...
|
|
m_rgHeader.wVersion = CATALOG_VERSION;
|
|
m_rgHeader.cbSize = sizeof(m_rgHeader);
|
|
SHGetThumbnailSizeForThumbsDB(&m_rgHeader.szThumbnailExtent);
|
|
m_rgHeader.dwEntryCount = 0;
|
|
}
|
|
|
|
m_dwCatalogChange = 0;
|
|
LeaveCriticalSection(&m_csInternals);
|
|
|
|
pCatalog->Release();
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CThumbStore::SaveCatalog()
|
|
{
|
|
_pStorageThumb->DestroyElement(CATALOG_STREAM);
|
|
|
|
IStream *pCatalog;
|
|
HRESULT hr = _pStorageThumb->CreateStream(CATALOG_STREAM, GetAccessMode(STGM_WRITE, TRUE), NULL, NULL, &pCatalog);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
EnterCriticalSection(&m_csInternals);
|
|
|
|
// now write the catalog to the stream ...
|
|
hr = IStream_Write(pCatalog, &m_rgHeader, sizeof(m_rgHeader));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
CLISTPOS pCur = m_rgCatalog.GetHeadPosition();
|
|
while (pCur && SUCCEEDED(hr))
|
|
{
|
|
CATALOG_ENTRY *pEntry = m_rgCatalog.GetNext(pCur);
|
|
if (pEntry)
|
|
{
|
|
hr = IStream_Write(pCatalog, pEntry, pEntry->cbSize);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
m_dwCatalogChange = 0;
|
|
|
|
LeaveCriticalSection(&m_csInternals);
|
|
pCatalog->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
void GenerateStreamName(LPWSTR pszBuffer, DWORD cbSize, DWORD dwNumber)
|
|
{
|
|
UINT cPos = 0;
|
|
while (dwNumber > 0)
|
|
{
|
|
DWORD dwRem = dwNumber % 10;
|
|
|
|
// based the fact that UNICODE chars 0-9 are the same as the ANSI chars 0 - 9
|
|
pszBuffer[cPos++] = (WCHAR)(dwRem + '0');
|
|
dwNumber /= 10;
|
|
}
|
|
pszBuffer[cPos] = 0;
|
|
}
|
|
|
|
// IPersist methods
|
|
|
|
STDMETHODIMP CThumbStore::GetClassID(CLSID *pClsid)
|
|
{
|
|
*pClsid = CLSID_ShellThumbnailDiskCache;
|
|
return S_OK;
|
|
}
|
|
|
|
// IPersistFolder
|
|
|
|
STDMETHODIMP CThumbStore::Initialize(LPCITEMIDLIST pidl)
|
|
{
|
|
WCHAR szPath[MAX_PATH];
|
|
|
|
HRESULT hr = SHGetPathFromIDList(pidl, szPath) ? S_OK : E_FAIL;
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (PathAppend(szPath, THUMB_FILENAME))
|
|
hr = _InitFromPath(szPath, STGM_READWRITE);
|
|
else
|
|
hr = E_INVALIDARG;
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
// IPersistFile
|
|
|
|
STDMETHODIMP CThumbStore::IsDirty(void)
|
|
{
|
|
return m_dwCatalogChange ? S_OK : S_FALSE;
|
|
}
|
|
|
|
HRESULT CThumbStore::_InitFromPath(LPCTSTR pszPath, DWORD dwMode)
|
|
{
|
|
if (PathIsRemovable(pszPath))
|
|
dwMode = STGM_READ;
|
|
|
|
m_dwModeAllow = dwMode;
|
|
StrCpyN(m_szPath, pszPath, ARRAYSIZE(m_szPath));
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::Load(LPCWSTR pszFileName, DWORD dwMode)
|
|
{
|
|
TCHAR szPath[MAX_PATH];
|
|
PathCombine(szPath, pszFileName, THUMB_FILENAME);
|
|
return _InitFromPath(szPath, dwMode);
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::Save(LPCWSTR pszFileName, BOOL fRemember)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::SaveCompleted(LPCWSTR pszFileName)
|
|
{
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::GetCurFile(LPWSTR *ppszFileName)
|
|
{
|
|
return SHStrDupW(m_szPath, ppszFileName);
|
|
}
|
|
|
|
// IShellImageStore methods
|
|
void CThumbStore::_SetAttribs(BOOL bForce)
|
|
{
|
|
// reduce spurious changenotifies by checking file attribs first
|
|
DWORD dwAttrib = GetFileAttributes(m_szPath);
|
|
if (bForce ||
|
|
((dwAttrib != 0xFFFFFFFF) &&
|
|
(dwAttrib & (FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM)) != (FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM)))
|
|
{
|
|
SetFileAttributes(m_szPath, FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM);
|
|
|
|
WCHAR szStream[MAX_PATH + 13];
|
|
|
|
StrCpyNW(szStream, m_szPath, ARRAYSIZE(szStream));
|
|
StrCatW(szStream, TEXT(":encryptable"));
|
|
|
|
HANDLE hStream = CreateFile(szStream, GENERIC_WRITE, NULL, NULL, CREATE_NEW, NULL, NULL);
|
|
if (hStream != INVALID_HANDLE_VALUE)
|
|
{
|
|
CloseHandle(hStream);
|
|
}
|
|
|
|
SHChangeNotify(SHCNE_UPDATEITEM, SHCNF_PATH, m_szPath, NULL); // suppress the update dir
|
|
}
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::Open(DWORD dwMode, DWORD *pdwLock)
|
|
{
|
|
if (m_szPath[0] == 0)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if ((m_dwModeAllow == STGM_READ) && (dwMode != STGM_READ))
|
|
return STG_E_ACCESSDENIED;
|
|
|
|
// at this point we have the lock if we need it, so we can close and reopen if we
|
|
// don't have it open with the right permissions...
|
|
if (_pStorageThumb)
|
|
{
|
|
if (_dwModeStorage == dwMode)
|
|
{
|
|
// we already have it open in this mode...
|
|
*pdwLock = AcquireLock();
|
|
return S_FALSE;
|
|
}
|
|
else
|
|
{
|
|
// we are open and the mode is different, so close it. Note, no lock is passed, we already
|
|
// have it
|
|
HRESULT hr = Close(NULL);
|
|
if (FAILED(hr))
|
|
{
|
|
return hr;
|
|
}
|
|
}
|
|
}
|
|
|
|
DWORD dwLock = AcquireLock();
|
|
|
|
DWORD dwFlags = GetAccessMode(dwMode, FALSE);
|
|
|
|
// now open the DocFile
|
|
HRESULT hr = StgOpenStorage(m_szPath, NULL, dwFlags, NULL, NULL, &_pStorageThumb);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_dwModeStorage = dwMode & (STGM_READ | STGM_WRITE | STGM_READWRITE);
|
|
_SetAttribs(FALSE);
|
|
hr = LoadCatalog();
|
|
*pdwLock = dwLock;
|
|
}
|
|
|
|
if (STG_E_DOCFILECORRUPT == hr)
|
|
{
|
|
DeleteFile(m_szPath);
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
ReleaseLock(dwLock);
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::Create(DWORD dwMode, DWORD *pdwLock)
|
|
{
|
|
if (m_szPath[0] == 0)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (_pStorageThumb)
|
|
{
|
|
// we already have it open, so we can't create it ...
|
|
return STG_E_ACCESSDENIED;
|
|
}
|
|
|
|
if ((m_dwModeAllow == STGM_READ) && (dwMode != STGM_READ))
|
|
return STG_E_ACCESSDENIED;
|
|
|
|
DWORD dwLock = AcquireLock();
|
|
|
|
DWORD dwFlags = GetAccessMode(dwMode, FALSE);
|
|
|
|
HRESULT hr = StgCreateDocfile(m_szPath, dwFlags, NULL, &_pStorageThumb);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
_dwModeStorage = dwMode & (STGM_READ | STGM_WRITE | STGM_READWRITE);
|
|
_SetAttribs(FALSE);
|
|
*pdwLock = dwLock;
|
|
}
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
ReleaseLock(dwLock);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::ReleaseLock(DWORD const *pdwLock)
|
|
{
|
|
ReleaseLock(*pdwLock);
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::IsLocked()
|
|
{
|
|
return (m_fLocked > 0 ? S_OK : S_FALSE);
|
|
}
|
|
|
|
// pdwLock can be NULL indicating close the last opened lock
|
|
|
|
STDMETHODIMP CThumbStore::Close(DWORD const *pdwLock)
|
|
{
|
|
DWORD dwLock;
|
|
DWORD const *pdwRel = pdwLock;
|
|
|
|
if (!pdwLock)
|
|
{
|
|
dwLock = AcquireLock();
|
|
pdwRel = &dwLock;
|
|
}
|
|
|
|
HRESULT hr = S_FALSE;
|
|
if (_pStorageThumb)
|
|
{
|
|
if (_dwModeStorage != STGM_READ)
|
|
{
|
|
// write out the new catalog...
|
|
hr = Commit(pdwLock);
|
|
_pStorageThumb->Commit(0);
|
|
|
|
SHChangeNotify(SHCNE_UPDATEITEM, SHCNF_PATH, m_szPath, NULL); // suppress the update dir
|
|
}
|
|
|
|
_pStorageThumb->Release();
|
|
_pStorageThumb = NULL;
|
|
}
|
|
|
|
ReleaseLock(*pdwRel);
|
|
|
|
return hr;
|
|
}
|
|
|
|
// pdwLock can be NULL meaning use the current lock
|
|
|
|
STDMETHODIMP CThumbStore::Commit(DWORD const *pdwLock)
|
|
{
|
|
DWORD dwLock;
|
|
if (!pdwLock)
|
|
{
|
|
dwLock = AcquireLock();
|
|
pdwLock = &dwLock;
|
|
}
|
|
|
|
HRESULT hr = S_FALSE;
|
|
|
|
if (_pStorageThumb && _dwModeStorage != STGM_READ)
|
|
{
|
|
if (m_dwCatalogChange)
|
|
{
|
|
SaveCatalog();
|
|
}
|
|
hr = S_OK;
|
|
}
|
|
|
|
ReleaseLock(*pdwLock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::GetMode(DWORD *pdwMode)
|
|
{
|
|
if (!pdwMode)
|
|
{
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (_pStorageThumb)
|
|
{
|
|
*pdwMode = _dwModeStorage;
|
|
return S_OK;
|
|
}
|
|
|
|
*pdwMode = 0;
|
|
return S_FALSE;
|
|
}
|
|
|
|
|
|
STDMETHODIMP CThumbStore::GetCapabilities(DWORD *pdwMode)
|
|
{
|
|
ASSERT(pdwMode);
|
|
|
|
// right now, both are needed/supported for thumbs.db
|
|
*pdwMode = SHIMSTCAPFLAG_LOCKABLE | SHIMSTCAPFLAG_PURGEABLE;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::AddEntry(LPCWSTR pszName, const FILETIME *pftTimeStamp, DWORD dwMode, HBITMAP hImage)
|
|
{
|
|
ASSERT(pszName);
|
|
|
|
if (!_pStorageThumb)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (_dwModeStorage == STGM_READ)
|
|
{
|
|
// can't modify in this mode...
|
|
return E_ACCESSDENIED;
|
|
}
|
|
|
|
// this will block unless we already have the lock on this thread...
|
|
DWORD dwLock = AcquireLock();
|
|
|
|
DWORD dwStream = 0;
|
|
CLISTPOS pCur = NULL;
|
|
CATALOG_ENTRY *pNode = NULL;
|
|
|
|
EnterCriticalSection(&m_csInternals);
|
|
|
|
if (FindStreamID(pszName, &dwStream, &pNode) != S_OK)
|
|
{
|
|
// needs adding to the catalog...
|
|
UINT cbSize = sizeof(*pNode) + lstrlenW(pszName) * sizeof(WCHAR);
|
|
|
|
pNode = (CATALOG_ENTRY *)LocalAlloc(LPTR, cbSize);
|
|
if (pNode == NULL)
|
|
{
|
|
LeaveCriticalSection(&m_csInternals);
|
|
ReleaseLock(dwLock);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
pNode->cbSize = cbSize;
|
|
if (pftTimeStamp)
|
|
{
|
|
pNode->ftTimeStamp = *pftTimeStamp;
|
|
}
|
|
dwStream = pNode->dwIndex = ++m_dwMaxIndex;
|
|
StrCpyW(pNode->szName, pszName);
|
|
|
|
pCur = m_rgCatalog.AddTail(pNode);
|
|
if (pCur == NULL)
|
|
{
|
|
LocalFree(pNode);
|
|
LeaveCriticalSection(&m_csInternals);
|
|
ReleaseLock(dwLock);
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
m_rgHeader.dwEntryCount++;
|
|
}
|
|
else if (pftTimeStamp)
|
|
{
|
|
// update the timestamp .....
|
|
pNode->ftTimeStamp = *pftTimeStamp;
|
|
}
|
|
|
|
LeaveCriticalSection(&m_csInternals);
|
|
|
|
IStream *pStream = NULL;
|
|
HRESULT hr = THR(GetEntryStream(dwStream, dwMode, &pStream));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = THR(WriteImage(pStream, hImage));
|
|
pStream->Release();
|
|
}
|
|
|
|
if (FAILED(hr) && pCur)
|
|
{
|
|
// take it back out of the list if we added it...
|
|
EnterCriticalSection(&m_csInternals);
|
|
m_rgCatalog.RemoveAt(pCur);
|
|
m_rgHeader.dwEntryCount--;
|
|
LeaveCriticalSection(&m_csInternals);
|
|
LocalFree(pNode);
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// catalog change....
|
|
m_dwCatalogChange++;
|
|
}
|
|
|
|
ReleaseLock(dwLock);
|
|
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::GetEntry(LPCWSTR pszName, DWORD dwMode, HBITMAP *phImage)
|
|
{
|
|
if (!_pStorageThumb)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
HRESULT hr;
|
|
DWORD dwStream;
|
|
if (FindStreamID(pszName, &dwStream, NULL) != S_OK)
|
|
{
|
|
hr = E_FAIL;
|
|
}
|
|
else
|
|
{
|
|
IStream *pStream;
|
|
hr = GetEntryStream(dwStream, dwMode, &pStream);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = ReadImage(pStream, phImage);
|
|
pStream->Release();
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
BOOL CThumbStore::_MatchNodeName(CATALOG_ENTRY *pNode, LPCWSTR pszName)
|
|
{
|
|
return (StrCmpIW(pNode->szName, pszName) == 0) ||
|
|
(StrCmpIW(PathFindFileName(pNode->szName), pszName) == 0); // match old thumbs.db files
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::DeleteEntry(LPCWSTR pszName)
|
|
{
|
|
if (!_pStorageThumb)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (_dwModeStorage == STGM_READ)
|
|
{
|
|
// can't modify in this mode...
|
|
return E_ACCESSDENIED;
|
|
}
|
|
|
|
DWORD dwLock = AcquireLock();
|
|
|
|
EnterCriticalSection(&m_csInternals);
|
|
|
|
// check to see if it already exists.....
|
|
CATALOG_ENTRY *pNode = NULL;
|
|
|
|
CLISTPOS pCur = m_rgCatalog.GetHeadPosition();
|
|
while (pCur != NULL)
|
|
{
|
|
CLISTPOS pDel = pCur;
|
|
pNode = m_rgCatalog.GetNext(pCur);
|
|
ASSERT(pNode != NULL);
|
|
|
|
if (_MatchNodeName(pNode, pszName))
|
|
{
|
|
m_rgCatalog.RemoveAt(pDel);
|
|
m_rgHeader.dwEntryCount--;
|
|
m_dwCatalogChange++;
|
|
if (pNode->dwIndex == m_dwMaxIndex)
|
|
{
|
|
m_dwMaxIndex--;
|
|
}
|
|
LeaveCriticalSection(&m_csInternals);
|
|
|
|
WCHAR szStream[30];
|
|
GenerateStreamName(szStream, ARRAYSIZE(szStream), pNode->dwIndex);
|
|
_pStorageThumb->DestroyElement(szStream);
|
|
|
|
LocalFree(pNode);
|
|
ReleaseLock(dwLock);
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
LeaveCriticalSection(&m_csInternals);
|
|
ReleaseLock(dwLock);
|
|
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
|
|
STDMETHODIMP CThumbStore::IsEntryInStore(LPCWSTR pszName, FILETIME *pftTimeStamp)
|
|
{
|
|
if (!_pStorageThumb)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
DWORD dwStream = 0;
|
|
CATALOG_ENTRY *pNode = NULL;
|
|
EnterCriticalSection(&m_csInternals);
|
|
HRESULT hr = FindStreamID(pszName, &dwStream, &pNode);
|
|
if (pftTimeStamp && SUCCEEDED(hr))
|
|
{
|
|
ASSERT(pNode);
|
|
*pftTimeStamp = pNode->ftTimeStamp;
|
|
}
|
|
LeaveCriticalSection(&m_csInternals);
|
|
|
|
return (hr == S_OK) ? S_OK : S_FALSE;
|
|
}
|
|
|
|
STDMETHODIMP CThumbStore::Enum(IEnumShellImageStore **ppEnum)
|
|
{
|
|
return CEnumThumbStore_Create(this, ppEnum);
|
|
}
|
|
|
|
HRESULT CThumbStore::FindStreamID(LPCWSTR pszName, DWORD *pdwStream, CATALOG_ENTRY ** ppNode)
|
|
{
|
|
// check to see if it already exists in the catalog.....
|
|
CATALOG_ENTRY *pNode = NULL;
|
|
|
|
CLISTPOS pCur = m_rgCatalog.GetHeadPosition();
|
|
while (pCur != NULL)
|
|
{
|
|
pNode = m_rgCatalog.GetNext(pCur);
|
|
ASSERT(pNode != NULL);
|
|
|
|
if (_MatchNodeName(pNode, pszName))
|
|
{
|
|
*pdwStream = pNode->dwIndex;
|
|
|
|
if (ppNode != NULL)
|
|
{
|
|
*ppNode = pNode;
|
|
}
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
return E_FAIL;
|
|
}
|
|
|
|
CEnumThumbStore::CEnumThumbStore()
|
|
{
|
|
m_pStore = NULL;
|
|
m_pPos = 0;
|
|
m_dwCatalogChange = 0;
|
|
}
|
|
|
|
CEnumThumbStore::~CEnumThumbStore()
|
|
{
|
|
if (m_pStore)
|
|
{
|
|
SAFECAST(m_pStore, IPersistFile *)->Release();
|
|
}
|
|
}
|
|
|
|
|
|
STDMETHODIMP CEnumThumbStore::Reset(void)
|
|
{
|
|
m_pPos = m_pStore->m_rgCatalog.GetHeadPosition();
|
|
m_dwCatalogChange = m_pStore->m_dwCatalogChange;
|
|
return S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CEnumThumbStore::Next(ULONG celt, PENUMSHELLIMAGESTOREDATA * prgElt, ULONG * pceltFetched)
|
|
{
|
|
if ((celt > 1 && !pceltFetched) || !celt)
|
|
{
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (m_dwCatalogChange != m_pStore->m_dwCatalogChange)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ULONG celtFetched = 0;
|
|
|
|
while (celtFetched < celt &&m_pPos)
|
|
{
|
|
CThumbStore::CATALOG_ENTRY *pNode = m_pStore->m_rgCatalog.GetNext(m_pPos);
|
|
|
|
ASSERT(pNode);
|
|
PENUMSHELLIMAGESTOREDATA pElt = (PENUMSHELLIMAGESTOREDATA) CoTaskMemAlloc(sizeof(ENUMSHELLIMAGESTOREDATA));
|
|
if (!pElt)
|
|
{
|
|
// cleanup others...
|
|
for (ULONG celtCleanup = 0; celtCleanup < celtFetched; celtCleanup++)
|
|
{
|
|
CoTaskMemFree(prgElt[celtCleanup]);
|
|
prgElt[celtCleanup] = NULL;
|
|
}
|
|
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
StrCpyN(pElt->szPath, pNode->szName, MAX_PATH);
|
|
pElt->ftTimeStamp = pNode->ftTimeStamp;
|
|
|
|
ASSERT(!IsBadWritePtr((void *) (prgElt + celtFetched), sizeof(PENUMSHELLIMAGESTOREDATA)));
|
|
prgElt[celtFetched] = pElt;
|
|
|
|
celtFetched++;
|
|
}
|
|
|
|
if (pceltFetched)
|
|
{
|
|
*pceltFetched = celtFetched;
|
|
}
|
|
|
|
if (!celtFetched)
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
return (celtFetched < celt) ? S_FALSE : S_OK;
|
|
}
|
|
|
|
STDMETHODIMP CEnumThumbStore::Skip(ULONG celt)
|
|
{
|
|
if (!celt)
|
|
{
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (m_dwCatalogChange != m_pStore->m_dwCatalogChange)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
ULONG celtSkipped = 0;
|
|
while (celtSkipped < celt &&m_pPos)
|
|
{
|
|
m_pStore->m_rgCatalog.GetNext(m_pPos);
|
|
}
|
|
|
|
if (!celtSkipped)
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
|
|
return (celtSkipped < celt) ? S_FALSE : S_OK;
|
|
}
|
|
|
|
|
|
STDMETHODIMP CEnumThumbStore::Clone(IEnumShellImageStore ** ppEnum)
|
|
{
|
|
CEnumThumbStore * pEnum = new CComObject<CEnumThumbStore>;
|
|
if (!pEnum)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
((IPersistFile *)m_pStore)->AddRef();
|
|
|
|
pEnum->m_pStore = m_pStore;
|
|
pEnum->m_dwCatalogChange = m_dwCatalogChange;
|
|
|
|
// created with zero ref count....
|
|
pEnum->AddRef();
|
|
|
|
*ppEnum = SAFECAST(pEnum, IEnumShellImageStore *);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CEnumThumbStore_Create(CThumbStore * pThis, IEnumShellImageStore ** ppEnum)
|
|
{
|
|
CEnumThumbStore * pEnum = new CComObject<CEnumThumbStore>;
|
|
if (!pEnum)
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
((IPersistFile *)pThis)->AddRef();
|
|
|
|
pEnum->m_pStore = pThis;
|
|
|
|
// created with zero ref count....
|
|
pEnum->AddRef();
|
|
|
|
*ppEnum = SAFECAST(pEnum, IEnumShellImageStore *);
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT Version1ReadImage(IStream *pStream, DWORD cbSize, HBITMAP *phImage)
|
|
{
|
|
*phImage = NULL;
|
|
|
|
HRESULT hr = E_OUTOFMEMORY;
|
|
BITMAPINFO *pbi = (BITMAPINFO *) LocalAlloc(LPTR, cbSize);
|
|
if (pbi)
|
|
{
|
|
hr = IStream_Read(pStream, pbi, cbSize);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
HDC hdc = GetDC(NULL);
|
|
if (hdc)
|
|
{
|
|
*phImage = CreateDIBitmap(hdc, &(pbi->bmiHeader), CBM_INIT, CalcBitsOffsetInDIB(pbi), pbi, DIB_RGB_COLORS);
|
|
ReleaseDC(NULL, hdc);
|
|
hr = S_OK;
|
|
}
|
|
}
|
|
LocalFree(pbi);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CThumbStore::ReadImage(IStream *pStream, HBITMAP *phImage)
|
|
{
|
|
STREAM_HEADER rgHead;
|
|
HRESULT hr = IStream_Read(pStream, &rgHead, sizeof(rgHead));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (rgHead.cbSize == sizeof(rgHead))
|
|
{
|
|
if (rgHead.dwFlags == STREAMFLAGS_DIB)
|
|
{
|
|
hr = Version1ReadImage(pStream, rgHead.ulSize, phImage);
|
|
}
|
|
else if (rgHead.dwFlags == STREAMFLAGS_JPEG)
|
|
{
|
|
void *pBits = LocalAlloc(LPTR, rgHead.ulSize);
|
|
if (pBits)
|
|
{
|
|
hr = IStream_Read(pStream, pBits, rgHead.ulSize);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
if (!DecompressImage(pBits, rgHead.ulSize, phImage))
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
}
|
|
}
|
|
LocalFree(pBits);
|
|
}
|
|
else
|
|
{
|
|
hr = E_OUTOFMEMORY;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return E_FAIL;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = E_FAIL;
|
|
}
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT Version1WriteImage(IStream *pStream, HBITMAP hImage)
|
|
{
|
|
HRESULT hr;
|
|
|
|
BITMAPINFO *pBitmap = BitmapToDIB(hImage);
|
|
if (pBitmap)
|
|
{
|
|
int ncolors = pBitmap->bmiHeader.biClrUsed;
|
|
if (ncolors == 0 && pBitmap->bmiHeader.biBitCount <= 8)
|
|
ncolors = 1 << pBitmap->bmiHeader.biBitCount;
|
|
|
|
if (pBitmap->bmiHeader.biBitCount == 16 ||
|
|
pBitmap->bmiHeader.biBitCount == 32)
|
|
{
|
|
if (pBitmap->bmiHeader.biCompression == BI_BITFIELDS)
|
|
{
|
|
ncolors = 3;
|
|
}
|
|
}
|
|
|
|
int iOffset = ncolors * sizeof(RGBQUAD);
|
|
|
|
STREAM_HEADER rgHead;
|
|
rgHead.cbSize = sizeof(rgHead);
|
|
rgHead.dwFlags = STREAMFLAGS_DIB;
|
|
rgHead.ulSize = pBitmap->bmiHeader.biSize + iOffset + pBitmap->bmiHeader.biSizeImage;
|
|
|
|
hr = IStream_Write(pStream, &rgHead, sizeof(rgHead));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = IStream_Write(pStream, pBitmap, rgHead.ulSize);
|
|
}
|
|
LocalFree(pBitmap);
|
|
}
|
|
else
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CThumbStore::WriteImage(IStream *pStream, HBITMAP hImage)
|
|
{
|
|
void *pBits;
|
|
ULONG ulBuffer;
|
|
|
|
HRESULT hr;
|
|
if (CompressImage(hImage, &pBits, &ulBuffer))
|
|
{
|
|
STREAM_HEADER rgHead;
|
|
rgHead.cbSize = sizeof(rgHead);
|
|
rgHead.dwFlags = STREAMFLAGS_JPEG;
|
|
rgHead.ulSize = ulBuffer;
|
|
|
|
hr = IStream_Write(pStream, &rgHead, sizeof(rgHead));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = IStream_Write(pStream, pBits, ulBuffer);
|
|
}
|
|
CoTaskMemFree(pBits);
|
|
}
|
|
else
|
|
hr = E_FAIL;
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CThumbStore::GetEntryStream(DWORD dwStream, DWORD dwMode, IStream **ppStream)
|
|
{
|
|
WCHAR szStream[30];
|
|
|
|
GenerateStreamName(szStream, ARRAYSIZE(szStream), dwStream);
|
|
|
|
// leave only the STG_READ | STGM_READWRITE | STGM_WRITE modes
|
|
dwMode &= STGM_READ | STGM_WRITE | STGM_READWRITE;
|
|
|
|
if (!_pStorageThumb)
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (_dwModeStorage != STGM_READWRITE && dwMode != _dwModeStorage)
|
|
{
|
|
return E_ACCESSDENIED;
|
|
}
|
|
|
|
DWORD dwFlags = GetAccessMode(dwMode, TRUE);
|
|
if (dwFlags & STGM_WRITE)
|
|
{
|
|
_pStorageThumb->DestroyElement(szStream);
|
|
return _pStorageThumb->CreateStream(szStream, dwFlags, NULL, NULL, ppStream);
|
|
}
|
|
else
|
|
{
|
|
return _pStorageThumb->OpenStream(szStream, NULL, dwFlags, NULL, ppStream);
|
|
}
|
|
}
|
|
|
|
DWORD CThumbStore::GetAccessMode(DWORD dwMode, BOOL fStream)
|
|
{
|
|
dwMode &= STGM_READ | STGM_WRITE | STGM_READWRITE;
|
|
|
|
DWORD dwFlags = dwMode;
|
|
|
|
// the root only needs Deny_Write, streams need exclusive....
|
|
if (dwMode == STGM_READ && !fStream)
|
|
{
|
|
dwFlags |= STGM_SHARE_DENY_WRITE;
|
|
}
|
|
else
|
|
{
|
|
dwFlags |= STGM_SHARE_EXCLUSIVE;
|
|
}
|
|
|
|
return dwFlags;
|
|
}
|
|
|
|
BITMAPINFO *BitmapToDIB(HBITMAP hBmp)
|
|
{
|
|
HDC hdcWnd = GetDC(NULL);
|
|
HDC hMemDC = CreateCompatibleDC(hdcWnd);
|
|
BITMAPINFO bi;
|
|
BITMAP Bitmap;
|
|
|
|
GetObject(hBmp, sizeof(Bitmap), (LPSTR)&Bitmap);
|
|
|
|
bi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
|
|
bi.bmiHeader.biBitCount = 0;
|
|
int iVal = GetDIBits(hMemDC, hBmp, 0, Bitmap.bmHeight, NULL, &bi, DIB_RGB_COLORS);
|
|
|
|
int ncolors = bi.bmiHeader.biClrUsed;
|
|
if (ncolors == 0 && bi.bmiHeader.biBitCount <= 8)
|
|
ncolors = 1 << bi.bmiHeader.biBitCount;
|
|
|
|
if (bi.bmiHeader.biBitCount == 16 ||
|
|
bi.bmiHeader.biBitCount == 32)
|
|
{
|
|
if (bi.bmiHeader.biCompression == BI_BITFIELDS)
|
|
{
|
|
ncolors = 3;
|
|
}
|
|
}
|
|
|
|
int iOffset = ncolors * sizeof(RGBQUAD);
|
|
|
|
void *pBuffer = LocalAlloc(LPTR, sizeof(BITMAPINFOHEADER) + iOffset + bi.bmiHeader.biSizeImage);
|
|
if (pBuffer)
|
|
{
|
|
BITMAPINFO * pbi = (BITMAPINFO *)pBuffer;
|
|
|
|
void *lpBits = (BYTE *)pBuffer + iOffset + sizeof(BITMAPINFOHEADER);
|
|
|
|
// copy members of what was returned in last GetDIBits call.
|
|
CopyMemory(&pbi->bmiHeader, &bi.bmiHeader, sizeof(BITMAPINFOHEADER));
|
|
iVal = GetDIBits(hMemDC, hBmp, 0, Bitmap.bmHeight, lpBits,
|
|
pbi, DIB_RGB_COLORS);
|
|
}
|
|
|
|
DeleteDC(hMemDC);
|
|
ReleaseDC(NULL, hdcWnd);
|
|
return (BITMAPINFO *)pBuffer;
|
|
}
|
|
|
|
BOOL CThumbStore::InitCodec(void)
|
|
{
|
|
if (NULL == m_pJPEGCodec)
|
|
{
|
|
m_pJPEGCodec = new CThumbnailFCNContainer();
|
|
}
|
|
|
|
return (NULL != m_pJPEGCodec);
|
|
}
|
|
|
|
BOOL CThumbStore::CompressImage(HBITMAP hBmp, void **ppvOutBuffer, ULONG *plBufSize)
|
|
{
|
|
// given an HBITMAP, get its data....
|
|
HBITMAP hBmpOut = hBmp;
|
|
void *pBits;
|
|
SIZE rgBmpSize;
|
|
|
|
*ppvOutBuffer = NULL;
|
|
|
|
if (!InitCodec())
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
HRESULT hr = PrepImage(&hBmpOut, &rgBmpSize, &pBits);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
ASSERT(pBits);
|
|
|
|
hr = E_FAIL;
|
|
if (pBits)
|
|
{
|
|
hr = m_pJPEGCodec->EncodeThumbnail(pBits, rgBmpSize.cx, rgBmpSize.cy,
|
|
ppvOutBuffer, plBufSize);
|
|
}
|
|
|
|
if (hBmpOut != hBmp)
|
|
{
|
|
// free the DIBSECTION we were passed back...
|
|
DeleteObject(hBmpOut);
|
|
}
|
|
}
|
|
|
|
return SUCCEEDED(hr);
|
|
}
|
|
|
|
BOOL CThumbStore::DecompressImage(void *pvInBuffer, ULONG ulBufferSize, HBITMAP *phBmp)
|
|
{
|
|
if (!InitCodec())
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
ULONG ulWidth;
|
|
ULONG ulHeight;
|
|
|
|
HRESULT hr = m_pJPEGCodec->DecodeThumbnail(phBmp, &ulWidth, &ulHeight, pvInBuffer, ulBufferSize);
|
|
return SUCCEEDED(hr);
|
|
}
|
|
|
|
|
|
HRESULT CThumbStore::PrepImage(HBITMAP *phBmp, SIZE * prgSize, void ** ppBits)
|
|
{
|
|
ASSERT(phBmp && &phBmp);
|
|
ASSERT(prgSize);
|
|
ASSERT(ppBits);
|
|
|
|
DIBSECTION rgDIB;
|
|
|
|
*ppBits = NULL;
|
|
|
|
HRESULT hr = E_FAIL;
|
|
|
|
// is the image the wrong colour depth or not a DIBSECTION
|
|
if (!GetObject(*phBmp, sizeof(rgDIB), &rgDIB) || (rgDIB.dsBmih.biBitCount != 32))
|
|
{
|
|
HBITMAP hBmp;
|
|
BITMAP rgBmp;
|
|
|
|
GetObject(*phBmp, sizeof(rgBmp), &rgBmp);
|
|
|
|
prgSize->cx = rgBmp.bmWidth;
|
|
prgSize->cy = rgBmp.bmHeight;
|
|
|
|
// generate a 32 bit DIB of the right size
|
|
if (!CreateSizedDIBSECTION(prgSize, 32, NULL, NULL, &hBmp, NULL, ppBits))
|
|
{
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
|
|
HDC hDCMem1 = CreateCompatibleDC(NULL);
|
|
if (hDCMem1)
|
|
{
|
|
HDC hDCMem2 = CreateCompatibleDC(NULL);
|
|
if (hDCMem2)
|
|
{
|
|
HBITMAP hBmpOld1 = (HBITMAP) SelectObject(hDCMem1, *phBmp);
|
|
HBITMAP hBmpOld2 = (HBITMAP) SelectObject(hDCMem2, hBmp);
|
|
|
|
// copy the image accross to generate the right sized DIB
|
|
BitBlt(hDCMem2, 0, 0, prgSize->cx, prgSize->cy, hDCMem1, 0, 0, SRCCOPY);
|
|
|
|
SelectObject(hDCMem1, hBmpOld1);
|
|
SelectObject(hDCMem2, hBmpOld2);
|
|
|
|
DeleteDC(hDCMem2);
|
|
}
|
|
DeleteDC(hDCMem1);
|
|
}
|
|
|
|
// pass back the BMP so it can be destroyed later...
|
|
*phBmp = hBmp;
|
|
return S_OK;
|
|
}
|
|
else
|
|
{
|
|
// HOUSTON, we have a DIBSECTION, this is quicker.....
|
|
*ppBits = rgDIB.dsBm.bmBits;
|
|
|
|
prgSize->cx = rgDIB.dsBm.bmWidth;
|
|
prgSize->cy = rgDIB.dsBm.bmHeight;
|
|
|
|
return S_OK;
|
|
}
|
|
}
|
|
|
|
HRESULT DeleteFileThumbnail(LPCWSTR szFilePath)
|
|
{
|
|
WCHAR szFolder[MAX_PATH];
|
|
WCHAR *szFile;
|
|
HRESULT hr = S_OK;
|
|
|
|
StrCpyNW(szFolder, szFilePath, ARRAYSIZE(szFolder));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
szFile = PathFindFileName(szFolder);
|
|
if (szFile != szFolder)
|
|
{
|
|
*(szFile - 1) = 0; // NULL terminates folder
|
|
|
|
IShellImageStore *pDiskCache = NULL;
|
|
hr = LoadFromFile(CLSID_ShellThumbnailDiskCache, szFolder, IID_PPV_ARG(IShellImageStore, &pDiskCache));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
IPersistFile *pPersist = NULL;
|
|
hr = pDiskCache->QueryInterface(IID_PPV_ARG(IPersistFile, &pPersist));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pPersist->Load(szFolder, STGM_READWRITE);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
DWORD dwLock;
|
|
hr = pDiskCache->Open(STGM_READWRITE, &dwLock);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pDiskCache->DeleteEntry(szFile);
|
|
pDiskCache->ReleaseLock(&dwLock);
|
|
pDiskCache->Close(NULL);
|
|
}
|
|
}
|
|
pPersist->Release();
|
|
}
|
|
pDiskCache->Release();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
hr = E_FAIL;
|
|
}
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|