1f8e2c85aSAndrew Rist /**************************************************************
248fbb544Smseidel *
3f8e2c85aSAndrew Rist * Licensed to the Apache Software Foundation (ASF) under one
4f8e2c85aSAndrew Rist * or more contributor license agreements. See the NOTICE file
5f8e2c85aSAndrew Rist * distributed with this work for additional information
6f8e2c85aSAndrew Rist * regarding copyright ownership. The ASF licenses this file
7f8e2c85aSAndrew Rist * to you under the Apache License, Version 2.0 (the
8f8e2c85aSAndrew Rist * "License"); you may not use this file except in compliance
9f8e2c85aSAndrew Rist * with the License. You may obtain a copy of the License at
1048fbb544Smseidel *
11f8e2c85aSAndrew Rist * http://www.apache.org/licenses/LICENSE-2.0
1248fbb544Smseidel *
13f8e2c85aSAndrew Rist * Unless required by applicable law or agreed to in writing,
14f8e2c85aSAndrew Rist * software distributed under the License is distributed on an
15f8e2c85aSAndrew Rist * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16f8e2c85aSAndrew Rist * KIND, either express or implied. See the License for the
17f8e2c85aSAndrew Rist * specific language governing permissions and limitations
18f8e2c85aSAndrew Rist * under the License.
1948fbb544Smseidel *
20f8e2c85aSAndrew Rist *************************************************************/
21f8e2c85aSAndrew Rist
22f8e2c85aSAndrew Rist
23cdf0e10cSrcweir
24cdf0e10cSrcweir // MARKER(update_precomp.py): autogen include statement, do not remove
25cdf0e10cSrcweir #include "precompiled_shell.hxx"
26cdf0e10cSrcweir
27cdf0e10cSrcweir #include "internal/global.hxx"
28cdf0e10cSrcweir
29cdf0e10cSrcweir #ifndef INFOTIPS_HXX_INCLUDED
30cdf0e10cSrcweir #include "internal/thumbviewer.hxx"
31cdf0e10cSrcweir #endif
32cdf0e10cSrcweir #include "internal/shlxthdl.hxx"
33cdf0e10cSrcweir #include "internal/registry.hxx"
34cdf0e10cSrcweir #include "internal/fileextensions.hxx"
35cdf0e10cSrcweir #include "internal/config.hxx"
36cdf0e10cSrcweir #include "internal/zipfile.hxx"
37cdf0e10cSrcweir #include "internal/utilities.hxx"
38cdf0e10cSrcweir
39cdf0e10cSrcweir #include "internal/resource.h"
40cdf0e10cSrcweir
41cdf0e10cSrcweir #include <stdio.h>
42cdf0e10cSrcweir #include <utility>
43cdf0e10cSrcweir #include <stdlib.h>
44cdf0e10cSrcweir
45cdf0e10cSrcweir #if defined _MSC_VER
46cdf0e10cSrcweir #pragma warning(push, 1)
47cdf0e10cSrcweir #endif
48cdf0e10cSrcweir #include <shellapi.h>
49cdf0e10cSrcweir #if defined _MSC_VER
50cdf0e10cSrcweir #pragma warning(pop)
5148fbb544Smseidel #endif
52cdf0e10cSrcweir #include <memory>
53cdf0e10cSrcweir
54cdf0e10cSrcweir extern HINSTANCE g_hModule;
55cdf0e10cSrcweir
56cdf0e10cSrcweir namespace internal
57cdf0e10cSrcweir {
58d9fc6b3aSmseidel /* The signet.png used for thumbnails of signed documents
59d9fc6b3aSmseidel is contained as resource in this module, the resource
60d9fc6b3aSmseidel id is 2000 */
LoadSignetImageFromResource(ZipFile::ZipContentBuffer_t & buffer)61d9fc6b3aSmseidel void LoadSignetImageFromResource(ZipFile::ZipContentBuffer_t& buffer)
62d9fc6b3aSmseidel {
63d9fc6b3aSmseidel HRSRC hrc = FindResource(g_hModule, TEXT("#2000"), RT_RCDATA);
64d9fc6b3aSmseidel DWORD size = SizeofResource(g_hModule, hrc);
65d9fc6b3aSmseidel HGLOBAL hglob = LoadResource(g_hModule, hrc);
66d9fc6b3aSmseidel char* data = reinterpret_cast<char*>(LockResource(hglob));
67d9fc6b3aSmseidel buffer = ZipFile::ZipContentBuffer_t(data, data + size);
68d9fc6b3aSmseidel }
69d9fc6b3aSmseidel
IsSignedDocument(const ZipFile * zipfile)70d9fc6b3aSmseidel bool IsSignedDocument(const ZipFile* zipfile)
71d9fc6b3aSmseidel {
72d9fc6b3aSmseidel return zipfile->HasContent("META-INF/documentsignatures.xml");
73d9fc6b3aSmseidel }
74d9fc6b3aSmseidel
IsWindowsXP()75d9fc6b3aSmseidel bool IsWindowsXP()
76d9fc6b3aSmseidel {
77d9fc6b3aSmseidel OSVERSIONINFO osvi;
78d9fc6b3aSmseidel ZeroMemory(&osvi, sizeof(osvi));
79d9fc6b3aSmseidel osvi.dwOSVersionInfoSize = sizeof(osvi);
80d9fc6b3aSmseidel GetVersionEx(&osvi);
81d9fc6b3aSmseidel
82d9fc6b3aSmseidel return ((osvi.dwPlatformId == VER_PLATFORM_WIN32_NT) &&
83d9fc6b3aSmseidel ((osvi.dwMajorVersion >= 5) && (osvi.dwMinorVersion >= 1)));
84d9fc6b3aSmseidel }
85d9fc6b3aSmseidel
86d9fc6b3aSmseidel /* Calculate where to position the signet image.
87d9fc6b3aSmseidel On Windows ME we need to shift the signet a
88d9fc6b3aSmseidel little bit to the left because Windows ME
89d9fc6b3aSmseidel puts an overlay icon to the lower right
90d9fc6b3aSmseidel corner of a thumbnail image so that our signet
91d9fc6b3aSmseidel will be hidden. */
CalcSignetPosition(const Gdiplus::Rect & canvas,const Gdiplus::Rect & thumbnail_border,const Gdiplus::Rect & signet)92d9fc6b3aSmseidel Gdiplus::Point CalcSignetPosition(
93d9fc6b3aSmseidel const Gdiplus::Rect& canvas, const Gdiplus::Rect& thumbnail_border, const Gdiplus::Rect& signet)
94d9fc6b3aSmseidel {
95d9fc6b3aSmseidel int x = 0;
96d9fc6b3aSmseidel int y = 0;
97d9fc6b3aSmseidel int hoffset = canvas.GetRight() - thumbnail_border.GetRight();
98d9fc6b3aSmseidel int voffset = canvas.GetBottom() - thumbnail_border.GetBottom();
99d9fc6b3aSmseidel
100d9fc6b3aSmseidel if (hoffset > voffset)
101d9fc6b3aSmseidel {
102d9fc6b3aSmseidel x = thumbnail_border.GetRight() - signet.GetRight() + min(signet.GetRight() / 2, hoffset);
103d9fc6b3aSmseidel y = thumbnail_border.GetBottom() - signet.GetBottom();
104d9fc6b3aSmseidel }
105d9fc6b3aSmseidel else
106d9fc6b3aSmseidel {
107d9fc6b3aSmseidel x = thumbnail_border.GetRight() - signet.GetRight();
108d9fc6b3aSmseidel y = thumbnail_border.GetBottom() - signet.GetBottom() + min(signet.GetBottom() / 2, voffset);
109d9fc6b3aSmseidel }
110d9fc6b3aSmseidel
111d9fc6b3aSmseidel if (!IsWindowsXP())
112d9fc6b3aSmseidel x -= 15;
113d9fc6b3aSmseidel
114d9fc6b3aSmseidel return Gdiplus::Point(x,y);
115d9fc6b3aSmseidel }
116cdf0e10cSrcweir }
117cdf0e10cSrcweir
118cdf0e10cSrcweir class StreamOnZipBuffer : public IStream
119cdf0e10cSrcweir {
120cdf0e10cSrcweir public:
121d9fc6b3aSmseidel StreamOnZipBuffer(const ZipFile::ZipContentBuffer_t& zip_buffer);
122d9fc6b3aSmseidel
123d9fc6b3aSmseidel // IUnknown
124d9fc6b3aSmseidel virtual ULONG STDMETHODCALLTYPE AddRef();
125d9fc6b3aSmseidel virtual ULONG STDMETHODCALLTYPE Release( void);
126d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject);
127d9fc6b3aSmseidel
128d9fc6b3aSmseidel // IStream
129d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Read(void *pv, ULONG cb, ULONG *pcbRead);
130d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Write(void const *pv, ULONG cb, ULONG *pcbWritten);
131d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Seek(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *plibNewPosition);
132d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE SetSize(ULARGE_INTEGER libNewSize);
133d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE CopyTo(IStream *pstm, ULARGE_INTEGER cb, ULARGE_INTEGER *pcbRead, ULARGE_INTEGER *pcbWritten);
134d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Commit(DWORD grfCommitFlags);
135d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Revert(void);
136d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE LockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
137d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE UnlockRegion(ULARGE_INTEGER libOffset, ULARGE_INTEGER cb, DWORD dwLockType);
138d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Stat(STATSTG *pstatstg, DWORD grfStatFlag);
139d9fc6b3aSmseidel virtual HRESULT STDMETHODCALLTYPE Clone(IStream **ppstm);
14048fbb544Smseidel
141cdf0e10cSrcweir private:
142d9fc6b3aSmseidel LONG ref_count_;
143d9fc6b3aSmseidel const ZipFile::ZipContentBuffer_t& ref_zip_buffer_;
144d9fc6b3aSmseidel size_t pos_;
145cdf0e10cSrcweir };
146cdf0e10cSrcweir
StreamOnZipBuffer(const ZipFile::ZipContentBuffer_t & zip_buffer)147cdf0e10cSrcweir StreamOnZipBuffer::StreamOnZipBuffer(const ZipFile::ZipContentBuffer_t& zip_buffer) :
148d9fc6b3aSmseidel ref_count_(1),
149d9fc6b3aSmseidel ref_zip_buffer_(zip_buffer),
150d9fc6b3aSmseidel pos_(0)
151cdf0e10cSrcweir {
152cdf0e10cSrcweir }
153cdf0e10cSrcweir
154cdf0e10cSrcweir // IUnknown methods
155cdf0e10cSrcweir
AddRef(void)156cdf0e10cSrcweir ULONG STDMETHODCALLTYPE StreamOnZipBuffer::AddRef(void)
157cdf0e10cSrcweir {
158d9fc6b3aSmseidel return InterlockedIncrement(&ref_count_);
159cdf0e10cSrcweir }
160d9fc6b3aSmseidel
Release(void)161cdf0e10cSrcweir ULONG STDMETHODCALLTYPE StreamOnZipBuffer::Release( void)
162cdf0e10cSrcweir {
163d9fc6b3aSmseidel long refcnt = InterlockedDecrement(&ref_count_);
164cdf0e10cSrcweir
165d9fc6b3aSmseidel if (0 == ref_count_)
166d9fc6b3aSmseidel delete this;
167cdf0e10cSrcweir
168d9fc6b3aSmseidel return refcnt;
169cdf0e10cSrcweir }
170cdf0e10cSrcweir
QueryInterface(REFIID riid,void __RPC_FAR * __RPC_FAR * ppvObject)171cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::QueryInterface(REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject)
172cdf0e10cSrcweir {
173d9fc6b3aSmseidel *ppvObject = 0;
174d9fc6b3aSmseidel IUnknown* pUnk = 0;
175d9fc6b3aSmseidel
176d9fc6b3aSmseidel if ((IID_IUnknown == riid) || (IID_IStream == riid))
177d9fc6b3aSmseidel {
178d9fc6b3aSmseidel pUnk = static_cast<IStream*>(this);
179d9fc6b3aSmseidel pUnk->AddRef();
180d9fc6b3aSmseidel *ppvObject = pUnk;
181d9fc6b3aSmseidel return S_OK;
182d9fc6b3aSmseidel }
183d9fc6b3aSmseidel return E_NOINTERFACE;
184cdf0e10cSrcweir }
185cdf0e10cSrcweir
Read(void * pv,ULONG cb,ULONG * pcbRead)186cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Read(void *pv, ULONG cb, ULONG *pcbRead)
187cdf0e10cSrcweir {
188d9fc6b3aSmseidel if (pv == NULL)
189d9fc6b3aSmseidel return STG_E_INVALIDPOINTER;
190d9fc6b3aSmseidel
191d9fc6b3aSmseidel size_t size = ref_zip_buffer_.size();
192d9fc6b3aSmseidel
193d9fc6b3aSmseidel if (pos_ > size)
194d9fc6b3aSmseidel return S_FALSE;
195d9fc6b3aSmseidel
196d9fc6b3aSmseidel char* p = reinterpret_cast<char*>(pv);
197d9fc6b3aSmseidel ULONG read = 0;
198d9fc6b3aSmseidel
199d9fc6b3aSmseidel for ( ;(pos_ < size) && (cb > 0); pos_++, cb--, read++)
200d9fc6b3aSmseidel *p++ = ref_zip_buffer_[pos_];
201d9fc6b3aSmseidel
202d9fc6b3aSmseidel if (pcbRead)
203d9fc6b3aSmseidel *pcbRead = read;
204d9fc6b3aSmseidel
205d9fc6b3aSmseidel return S_OK;
206cdf0e10cSrcweir }
207cdf0e10cSrcweir
Seek(LARGE_INTEGER dlibMove,DWORD dwOrigin,ULARGE_INTEGER *)208cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Seek(LARGE_INTEGER dlibMove, DWORD dwOrigin, ULARGE_INTEGER *)
209d9fc6b3aSmseidel {
210d9fc6b3aSmseidel __int64 size = (__int64) ref_zip_buffer_.size();
211d9fc6b3aSmseidel __int64 p = 0;
212d9fc6b3aSmseidel
213d9fc6b3aSmseidel switch (dwOrigin)
214d9fc6b3aSmseidel {
215d9fc6b3aSmseidel case STREAM_SEEK_SET:
216d9fc6b3aSmseidel break;
217d9fc6b3aSmseidel case STREAM_SEEK_CUR:
218d9fc6b3aSmseidel p = (__int64) pos_;
219d9fc6b3aSmseidel break;
220d9fc6b3aSmseidel case STREAM_SEEK_END:
221d9fc6b3aSmseidel p = size - 1;
222d9fc6b3aSmseidel break;
223d9fc6b3aSmseidel }
224d9fc6b3aSmseidel
225d9fc6b3aSmseidel HRESULT hr = STG_E_INVALIDFUNCTION;
226d9fc6b3aSmseidel
227d9fc6b3aSmseidel p += dlibMove.QuadPart;
228d9fc6b3aSmseidel
229d9fc6b3aSmseidel if ( ( p >= 0 ) && (p < size) )
230d9fc6b3aSmseidel {
231d9fc6b3aSmseidel pos_ = (size_t) p;
232d9fc6b3aSmseidel hr = S_OK;
233d9fc6b3aSmseidel }
234d9fc6b3aSmseidel return hr;
235cdf0e10cSrcweir }
236cdf0e10cSrcweir
Stat(STATSTG * pstatstg,DWORD grfStatFlag)237cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Stat(STATSTG *pstatstg, DWORD grfStatFlag)
238d9fc6b3aSmseidel {
239d9fc6b3aSmseidel if (pstatstg == NULL)
240d9fc6b3aSmseidel return STG_E_INVALIDPOINTER;
241d9fc6b3aSmseidel
242d9fc6b3aSmseidel ZeroMemory(pstatstg, sizeof(STATSTG));
243d9fc6b3aSmseidel
244d9fc6b3aSmseidel if (grfStatFlag == STATFLAG_DEFAULT)
245d9fc6b3aSmseidel {
246d9fc6b3aSmseidel size_t sz = 4 * sizeof(wchar_t);
247d9fc6b3aSmseidel wchar_t* name = reinterpret_cast<wchar_t*>(CoTaskMemAlloc(sz));
248d9fc6b3aSmseidel ZeroMemory(name, sz);
249d9fc6b3aSmseidel memcpy(name, L"png", 3 * sizeof(wchar_t));
250d9fc6b3aSmseidel pstatstg->pwcsName = name;
251d9fc6b3aSmseidel }
252d9fc6b3aSmseidel
253d9fc6b3aSmseidel pstatstg->type = STGTY_LOCKBYTES;
254d9fc6b3aSmseidel
255d9fc6b3aSmseidel ULARGE_INTEGER uli;
256d9fc6b3aSmseidel uli.LowPart = ref_zip_buffer_.size();
257d9fc6b3aSmseidel uli.HighPart = 0;
258d9fc6b3aSmseidel
259d9fc6b3aSmseidel pstatstg->cbSize = uli;
260d9fc6b3aSmseidel
261d9fc6b3aSmseidel return S_OK;
262cdf0e10cSrcweir }
263cdf0e10cSrcweir
Write(void const *,ULONG,ULONG *)264cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Write(void const *, ULONG, ULONG *)
265cdf0e10cSrcweir { return E_NOTIMPL; }
266cdf0e10cSrcweir
SetSize(ULARGE_INTEGER)267cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::SetSize(ULARGE_INTEGER)
268cdf0e10cSrcweir { return E_NOTIMPL; }
269cdf0e10cSrcweir
CopyTo(IStream *,ULARGE_INTEGER,ULARGE_INTEGER *,ULARGE_INTEGER *)270cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::CopyTo(IStream *, ULARGE_INTEGER, ULARGE_INTEGER *, ULARGE_INTEGER *)
271cdf0e10cSrcweir { return E_NOTIMPL; }
272cdf0e10cSrcweir
Commit(DWORD)273cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Commit(DWORD)
274cdf0e10cSrcweir { return E_NOTIMPL; }
275cdf0e10cSrcweir
Revert(void)276cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Revert(void)
277cdf0e10cSrcweir { return E_NOTIMPL; }
278cdf0e10cSrcweir
LockRegion(ULARGE_INTEGER,ULARGE_INTEGER,DWORD)279cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::LockRegion(ULARGE_INTEGER, ULARGE_INTEGER, DWORD)
280cdf0e10cSrcweir { return E_NOTIMPL; }
281cdf0e10cSrcweir
UnlockRegion(ULARGE_INTEGER,ULARGE_INTEGER,DWORD)282cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::UnlockRegion(ULARGE_INTEGER, ULARGE_INTEGER, DWORD)
283cdf0e10cSrcweir { return E_NOTIMPL; }
284cdf0e10cSrcweir
Clone(IStream **)285cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE StreamOnZipBuffer::Clone(IStream **)
286d9fc6b3aSmseidel { return E_NOTIMPL; }
287cdf0e10cSrcweir
288cdf0e10cSrcweir
289cdf0e10cSrcweir //#########################################
290cdf0e10cSrcweir
291cdf0e10cSrcweir
CThumbviewer(long RefCnt)292d9fc6b3aSmseidel CThumbviewer::CThumbviewer(long RefCnt) :
293d9fc6b3aSmseidel ref_count_(RefCnt)
294cdf0e10cSrcweir {
295d9fc6b3aSmseidel InterlockedIncrement(&g_DllRefCnt);
296d9fc6b3aSmseidel
297d9fc6b3aSmseidel thumbnail_size_.cx = 0;
298d9fc6b3aSmseidel thumbnail_size_.cy = 0;
299d9fc6b3aSmseidel
300d9fc6b3aSmseidel Gdiplus::GdiplusStartupInput gdiplusStartupInput;
301d9fc6b3aSmseidel Gdiplus::GdiplusStartup(&gdiplus_token_, &gdiplusStartupInput, NULL);
302d9fc6b3aSmseidel
303d9fc6b3aSmseidel ZipFile::ZipContentBuffer_t img_data;
304d9fc6b3aSmseidel internal::LoadSignetImageFromResource(img_data);
305d9fc6b3aSmseidel IStream* stream = new StreamOnZipBuffer(img_data);
306d9fc6b3aSmseidel signet_ = new Gdiplus::Bitmap(stream, TRUE);
307d9fc6b3aSmseidel stream->Release();
308cdf0e10cSrcweir }
309cdf0e10cSrcweir
~CThumbviewer()310cdf0e10cSrcweir CThumbviewer::~CThumbviewer()
311cdf0e10cSrcweir {
312d9fc6b3aSmseidel delete signet_;
313d9fc6b3aSmseidel Gdiplus::GdiplusShutdown(gdiplus_token_);
314d9fc6b3aSmseidel InterlockedDecrement(&g_DllRefCnt);
315cdf0e10cSrcweir }
316cdf0e10cSrcweir
317cdf0e10cSrcweir // IUnknown methods
318cdf0e10cSrcweir
QueryInterface(REFIID riid,void __RPC_FAR * __RPC_FAR * ppvObject)319cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::QueryInterface(REFIID riid, void __RPC_FAR *__RPC_FAR *ppvObject)
320cdf0e10cSrcweir {
321d9fc6b3aSmseidel *ppvObject = 0;
322d9fc6b3aSmseidel IUnknown* pUnk = 0;
323d9fc6b3aSmseidel
324d9fc6b3aSmseidel if ((IID_IUnknown == riid) || (IID_IPersistFile == riid))
325d9fc6b3aSmseidel {
326d9fc6b3aSmseidel pUnk = static_cast<IPersistFile*>(this);
327d9fc6b3aSmseidel pUnk->AddRef();
328d9fc6b3aSmseidel *ppvObject = pUnk;
329d9fc6b3aSmseidel return S_OK;
330d9fc6b3aSmseidel }
331d9fc6b3aSmseidel else if (IID_IExtractImage == riid)
332d9fc6b3aSmseidel {
333d9fc6b3aSmseidel pUnk = static_cast<IExtractImage*>(this);
334d9fc6b3aSmseidel pUnk->AddRef();
335d9fc6b3aSmseidel *ppvObject = pUnk;
336d9fc6b3aSmseidel return S_OK;
337d9fc6b3aSmseidel }
338d9fc6b3aSmseidel return E_NOINTERFACE;
339cdf0e10cSrcweir }
340cdf0e10cSrcweir
AddRef(void)341cdf0e10cSrcweir ULONG STDMETHODCALLTYPE CThumbviewer::AddRef(void)
342cdf0e10cSrcweir {
343d9fc6b3aSmseidel return InterlockedIncrement(&ref_count_);
344cdf0e10cSrcweir }
34548fbb544Smseidel
Release(void)346cdf0e10cSrcweir ULONG STDMETHODCALLTYPE CThumbviewer::Release( void)
347cdf0e10cSrcweir {
348d9fc6b3aSmseidel long refcnt = InterlockedDecrement(&ref_count_);
349cdf0e10cSrcweir
350d9fc6b3aSmseidel if (0 == ref_count_)
351d9fc6b3aSmseidel delete this;
352cdf0e10cSrcweir
353d9fc6b3aSmseidel return refcnt;
354cdf0e10cSrcweir }
355cdf0e10cSrcweir
356cdf0e10cSrcweir // IExtractImage2 methods
357cdf0e10cSrcweir
358cdf0e10cSrcweir const std::string THUMBNAIL_CONTENT = "Thumbnails/thumbnail.png";
359cdf0e10cSrcweir
Extract(HBITMAP * phBmpImage)360cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::Extract(HBITMAP *phBmpImage)
36148fbb544Smseidel {
362d9fc6b3aSmseidel HRESULT hr = E_FAIL;
363d9fc6b3aSmseidel
364d9fc6b3aSmseidel try
365d9fc6b3aSmseidel {
366d9fc6b3aSmseidel std::wstring fname = getShortPathName( filename_ );
367d9fc6b3aSmseidel std::auto_ptr<ZipFile> zipfile( new ZipFile( WStringToString( fname ) ) );
368d9fc6b3aSmseidel
369d9fc6b3aSmseidel if (zipfile->HasContent(THUMBNAIL_CONTENT))
370d9fc6b3aSmseidel {
371d9fc6b3aSmseidel ZipFile::ZipContentBuffer_t thumbnail;
372d9fc6b3aSmseidel zipfile->GetUncompressedContent(THUMBNAIL_CONTENT, thumbnail);
373d9fc6b3aSmseidel IStream* stream = new StreamOnZipBuffer(thumbnail);
374d9fc6b3aSmseidel
375d9fc6b3aSmseidel Gdiplus::Bitmap thumbnail_png(stream, TRUE);
376d9fc6b3aSmseidel
377d9fc6b3aSmseidel if ((thumbnail_png.GetHeight() == 0) || (thumbnail_png.GetWidth() == 0))
378d9fc6b3aSmseidel {
379d9fc6b3aSmseidel stream->Release();
380d9fc6b3aSmseidel return E_FAIL;
381d9fc6b3aSmseidel }
382d9fc6b3aSmseidel
383d9fc6b3aSmseidel HWND hwnd = GetDesktopWindow();
384d9fc6b3aSmseidel HDC hdc = GetDC(hwnd);
385d9fc6b3aSmseidel HDC memDC = CreateCompatibleDC(hdc);
386d9fc6b3aSmseidel
387d9fc6b3aSmseidel if (memDC)
388d9fc6b3aSmseidel {
389d9fc6b3aSmseidel UINT offset = 1; // reserve a little border space
390d9fc6b3aSmseidel
391d9fc6b3aSmseidel Gdiplus::Rect canvas(0, 0, thumbnail_size_.cx, thumbnail_size_.cy);
392*ef262aceSmseidel Gdiplus::Rect canvas_thumbnail(offset, offset, thumbnail_size_.cx - (2 * offset) - 1, thumbnail_size_.cy - (2 * offset) - 1);
393d9fc6b3aSmseidel
394d9fc6b3aSmseidel Gdiplus::Rect scaledRect = CalcScaledAspectRatio(
395d9fc6b3aSmseidel Gdiplus::Rect(0, 0, thumbnail_png.GetWidth(), thumbnail_png.GetHeight()), canvas_thumbnail);
396d9fc6b3aSmseidel
397d9fc6b3aSmseidel struct {
398d9fc6b3aSmseidel BITMAPINFOHEADER bi;
399d9fc6b3aSmseidel DWORD ct[256];
400d9fc6b3aSmseidel } dib;
401d9fc6b3aSmseidel
402d9fc6b3aSmseidel ZeroMemory(&dib, sizeof(dib));
403d9fc6b3aSmseidel
404d9fc6b3aSmseidel dib.bi.biSize = sizeof(BITMAPINFOHEADER);
405d9fc6b3aSmseidel dib.bi.biWidth = thumbnail_size_.cx;
406d9fc6b3aSmseidel dib.bi.biHeight = thumbnail_size_.cy;
407d9fc6b3aSmseidel dib.bi.biPlanes = 1;
408d9fc6b3aSmseidel dib.bi.biBitCount = static_cast<WORD>(color_depth_);
409d9fc6b3aSmseidel dib.bi.biCompression = BI_RGB;
410d9fc6b3aSmseidel
411d9fc6b3aSmseidel LPVOID lpBits;
412d9fc6b3aSmseidel HBITMAP hMemBmp = CreateDIBSection(memDC, (LPBITMAPINFO)&dib, DIB_RGB_COLORS, &lpBits, NULL, 0);
413d9fc6b3aSmseidel HGDIOBJ hOldObj = SelectObject(memDC, hMemBmp);
414d9fc6b3aSmseidel
415d9fc6b3aSmseidel Gdiplus::Graphics graphics(memDC);
416*ef262aceSmseidel Gdiplus::Pen grayPen(Gdiplus::Color(255, 127, 127, 127), 1);
417d9fc6b3aSmseidel
418d9fc6b3aSmseidel Gdiplus::SolidBrush whiteBrush(Gdiplus::Color(255, 255, 255, 255));
419d9fc6b3aSmseidel graphics.FillRectangle(&whiteBrush, canvas);
420d9fc6b3aSmseidel
421d9fc6b3aSmseidel scaledRect.X = (canvas.Width - scaledRect.Width) / 2;
422d9fc6b3aSmseidel scaledRect.Y = (canvas.Height - scaledRect.Height) / 2;
423d9fc6b3aSmseidel
424d9fc6b3aSmseidel Gdiplus::Rect border_rect(scaledRect.X, scaledRect.Y, scaledRect.Width, scaledRect.Height);
425*ef262aceSmseidel graphics.DrawRectangle(&grayPen, border_rect);
426d9fc6b3aSmseidel
427d9fc6b3aSmseidel scaledRect.X += 1;
428d9fc6b3aSmseidel scaledRect.Y += 1;
429d9fc6b3aSmseidel scaledRect.Width -= 1;
430d9fc6b3aSmseidel scaledRect.Height -= 1;
431d9fc6b3aSmseidel
432d9fc6b3aSmseidel graphics.SetInterpolationMode(Gdiplus::InterpolationModeHighQualityBicubic);
433d9fc6b3aSmseidel Gdiplus::Status stat = graphics.DrawImage(
434d9fc6b3aSmseidel &thumbnail_png, scaledRect, 0 , 0,
435d9fc6b3aSmseidel thumbnail_png.GetWidth(), thumbnail_png.GetHeight(), Gdiplus::UnitPixel);
436d9fc6b3aSmseidel
437d9fc6b3aSmseidel /* Add a signet sign to the thumbnail of signed documents */
438d9fc6b3aSmseidel if (internal::IsSignedDocument(zipfile.get()))
439d9fc6b3aSmseidel {
440d9fc6b3aSmseidel double SCALING_FACTOR = 0.6;
441d9fc6b3aSmseidel Gdiplus::Rect signet_scaled(
442d9fc6b3aSmseidel 0, 0, static_cast<INT>(signet_->GetWidth() * SCALING_FACTOR), static_cast<INT>(signet_->GetHeight() * SCALING_FACTOR));
443d9fc6b3aSmseidel Gdiplus::Point pos_signet = internal::CalcSignetPosition(canvas_thumbnail, border_rect, signet_scaled);
444d9fc6b3aSmseidel Gdiplus::Rect dest(pos_signet.X, pos_signet.Y, signet_scaled.GetRight(), signet_scaled.GetBottom());
445d9fc6b3aSmseidel
446d9fc6b3aSmseidel stat = graphics.DrawImage(
447d9fc6b3aSmseidel signet_, dest,
448d9fc6b3aSmseidel 0, 0, signet_->GetWidth(), signet_->GetHeight(),
449d9fc6b3aSmseidel Gdiplus::UnitPixel);
450d9fc6b3aSmseidel }
451d9fc6b3aSmseidel
452d9fc6b3aSmseidel if (stat == Gdiplus::Ok)
453d9fc6b3aSmseidel {
454d9fc6b3aSmseidel *phBmpImage = hMemBmp;
455d9fc6b3aSmseidel hr = NOERROR;
456d9fc6b3aSmseidel }
457d9fc6b3aSmseidel
458d9fc6b3aSmseidel SelectObject(memDC, hOldObj);
459d9fc6b3aSmseidel DeleteDC(memDC);
46048fbb544Smseidel }
46148fbb544Smseidel
46248fbb544Smseidel ReleaseDC(hwnd, hdc);
46348fbb544Smseidel stream->Release();
46448fbb544Smseidel }
46548fbb544Smseidel }
46648fbb544Smseidel catch(std::exception&)
46748fbb544Smseidel {
46848fbb544Smseidel OutputDebugStringFormat( "CThumbviewer Extract ERROR!\n" );
46948fbb544Smseidel hr = E_FAIL;
47048fbb544Smseidel }
47148fbb544Smseidel return hr;
472cdf0e10cSrcweir }
473cdf0e10cSrcweir
GetLocation(LPWSTR pszPathBuffer,DWORD cchMax,DWORD * pdwPriority,const SIZE * prgSize,DWORD dwRecClrDepth,DWORD * pdwFlags)474cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::GetLocation(
47548fbb544Smseidel LPWSTR pszPathBuffer, DWORD cchMax, DWORD *pdwPriority, const SIZE *prgSize, DWORD dwRecClrDepth, DWORD *pdwFlags)
476cdf0e10cSrcweir {
47748fbb544Smseidel if ((prgSize == NULL) || (pdwFlags == NULL) || ((*pdwFlags & IEIFLAG_ASYNC) && (pdwPriority == NULL)))
47848fbb544Smseidel return E_INVALIDARG;
47948fbb544Smseidel
48048fbb544Smseidel thumbnail_size_ = *prgSize;
48148fbb544Smseidel color_depth_ = dwRecClrDepth;
48248fbb544Smseidel
48348fbb544Smseidel *pdwFlags = IEIFLAG_CACHE; // we don't cache the image
48448fbb544Smseidel
48548fbb544Smseidel wcsncpy(pszPathBuffer, filename_.c_str(), cchMax);
48648fbb544Smseidel
48748fbb544Smseidel return NOERROR;
488cdf0e10cSrcweir }
48948fbb544Smseidel
490cdf0e10cSrcweir // IPersist methods
491cdf0e10cSrcweir
GetClassID(CLSID * pClassID)492cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::GetClassID(CLSID* pClassID)
493cdf0e10cSrcweir {
49448fbb544Smseidel pClassID = const_cast<CLSID*>(&CLSID_THUMBVIEWER_HANDLER);
49548fbb544Smseidel return S_OK;
496cdf0e10cSrcweir }
497cdf0e10cSrcweir
498cdf0e10cSrcweir // IPersistFile methods
499cdf0e10cSrcweir
Load(LPCOLESTR pszFileName,DWORD)500cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::Load(LPCOLESTR pszFileName, DWORD)
50148fbb544Smseidel {
50248fbb544Smseidel filename_ = pszFileName;
50348fbb544Smseidel return S_OK;
504cdf0e10cSrcweir }
505cdf0e10cSrcweir
IsDirty()506cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::IsDirty()
507cdf0e10cSrcweir { return E_NOTIMPL; }
50848fbb544Smseidel
Save(LPCOLESTR,BOOL)509cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::Save(LPCOLESTR, BOOL)
510cdf0e10cSrcweir { return E_NOTIMPL; }
51148fbb544Smseidel
SaveCompleted(LPCOLESTR)512cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::SaveCompleted(LPCOLESTR)
513cdf0e10cSrcweir { return E_NOTIMPL; }
51448fbb544Smseidel
GetCurFile(LPOLESTR __RPC_FAR *)515cdf0e10cSrcweir HRESULT STDMETHODCALLTYPE CThumbviewer::GetCurFile(LPOLESTR __RPC_FAR*)
516cdf0e10cSrcweir { return E_NOTIMPL; }
517cdf0e10cSrcweir
518cdf0e10cSrcweir
CalcScaledAspectRatio(Gdiplus::Rect src,Gdiplus::Rect dest)519cdf0e10cSrcweir Gdiplus::Rect CThumbviewer::CalcScaledAspectRatio(Gdiplus::Rect src, Gdiplus::Rect dest)
520cdf0e10cSrcweir {
52148fbb544Smseidel Gdiplus::Rect result;
52248fbb544Smseidel if (src.Width >= src.Height)
52348fbb544Smseidel result = Gdiplus::Rect(0, 0, dest.Width, src.Height * dest.Width / src.Width);
52448fbb544Smseidel else
52548fbb544Smseidel result = Gdiplus::Rect(0, 0, src.Width * dest.Height / src.Height, dest.Height);
526cdf0e10cSrcweir
52748fbb544Smseidel return result;
528cdf0e10cSrcweir }
529cdf0e10cSrcweir
53048fbb544Smseidel /* vim: set noet sw=4 ts=4: */
531