OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright (C) 2007 Apple Inc. All rights reserved. | |
3 * | |
4 * Redistribution and use in source and binary forms, with or without | |
5 * modification, are permitted provided that the following conditions | |
6 * are met: | |
7 * 1. Redistributions of source code must retain the above copyright | |
8 * notice, this list of conditions and the following disclaimer. | |
9 * 2. Redistributions in binary form must reproduce the above copyright | |
10 * notice, this list of conditions and the following disclaimer in the | |
11 * documentation and/or other materials provided with the distribution. | |
12 * | |
13 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY | |
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
15 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR | |
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR | |
17 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
18 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, | |
19 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR | |
20 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY | |
21 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |
22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
23 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
24 */ | |
25 | |
26 #include "config.h" | |
27 | |
28 #include "WCDataObject.h" | |
29 | |
30 #include "PlatformString.h" | |
31 | |
32 namespace WebCore { | |
33 | |
34 class WCEnumFormatEtc : public IEnumFORMATETC | |
35 { | |
36 public: | |
37 WCEnumFormatEtc(const Vector<FORMATETC>& formats); | |
38 WCEnumFormatEtc(const Vector<FORMATETC*>& formats); | |
39 | |
40 //IUnknown members | |
41 STDMETHOD(QueryInterface)(REFIID, void**); | |
42 STDMETHOD_(ULONG, AddRef)(void); | |
43 STDMETHOD_(ULONG, Release)(void); | |
44 | |
45 //IEnumFORMATETC members | |
46 STDMETHOD(Next)(ULONG, LPFORMATETC, ULONG*); | |
47 STDMETHOD(Skip)(ULONG); | |
48 STDMETHOD(Reset)(void); | |
49 STDMETHOD(Clone)(IEnumFORMATETC**); | |
50 | |
51 private: | |
52 long m_ref; | |
53 Vector<FORMATETC> m_formats; | |
54 size_t m_current; | |
55 }; | |
56 | |
57 | |
58 | |
59 WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC>& formats) | |
60 : m_ref(1) | |
61 , m_current(0) | |
62 { | |
63 for(size_t i = 0; i < formats.size(); ++i) | |
64 m_formats.append(formats[i]); | |
65 } | |
66 | |
67 WCEnumFormatEtc::WCEnumFormatEtc(const Vector<FORMATETC*>& formats) | |
68 : m_ref(1) | |
69 , m_current(0) | |
70 { | |
71 for(size_t i = 0; i < formats.size(); ++i) | |
72 m_formats.append(*formats[i]); | |
73 } | |
74 | |
75 STDMETHODIMP WCEnumFormatEtc::QueryInterface(REFIID riid, void** ppvObject) | |
76 { | |
77 *ppvObject = 0; | |
78 if (IsEqualIID(riid, IID_IUnknown) || | |
79 IsEqualIID(riid, IID_IEnumFORMATETC)) { | |
80 *ppvObject = this; | |
81 AddRef(); | |
82 return S_OK; | |
83 } | |
84 | |
85 return E_NOINTERFACE; | |
86 } | |
87 | |
88 STDMETHODIMP_(ULONG) WCEnumFormatEtc::AddRef(void) | |
89 { | |
90 return InterlockedIncrement(&m_ref); | |
91 } | |
92 | |
93 STDMETHODIMP_(ULONG) WCEnumFormatEtc::Release(void) | |
94 { | |
95 long c = InterlockedDecrement(&m_ref); | |
96 if (c == 0) | |
97 delete this; | |
98 return c; | |
99 } | |
100 | |
101 STDMETHODIMP WCEnumFormatEtc::Next(ULONG celt, LPFORMATETC lpFormatEtc, ULONG* p
celtFetched) | |
102 { | |
103 if(pceltFetched != 0) | |
104 *pceltFetched=0; | |
105 | |
106 ULONG cReturn = celt; | |
107 | |
108 if(celt <= 0 || lpFormatEtc == 0 || m_current >= m_formats.size()) | |
109 return S_FALSE; | |
110 | |
111 if(pceltFetched == 0 && celt != 1) // pceltFetched can be 0 only for 1 item
request | |
112 return S_FALSE; | |
113 | |
114 while (m_current < m_formats.size() && cReturn > 0) { | |
115 *lpFormatEtc++ = m_formats[m_current++]; | |
116 --cReturn; | |
117 } | |
118 if (pceltFetched != 0) | |
119 *pceltFetched = celt - cReturn; | |
120 | |
121 return (cReturn == 0) ? S_OK : S_FALSE; | |
122 } | |
123 | |
124 STDMETHODIMP WCEnumFormatEtc::Skip(ULONG celt) | |
125 { | |
126 if((m_current + int(celt)) >= m_formats.size()) | |
127 return S_FALSE; | |
128 m_current += celt; | |
129 return S_OK; | |
130 } | |
131 | |
132 STDMETHODIMP WCEnumFormatEtc::Reset(void) | |
133 { | |
134 m_current = 0; | |
135 return S_OK; | |
136 } | |
137 | |
138 STDMETHODIMP WCEnumFormatEtc::Clone(IEnumFORMATETC** ppCloneEnumFormatEtc) | |
139 { | |
140 if(!ppCloneEnumFormatEtc) | |
141 return E_POINTER; | |
142 | |
143 WCEnumFormatEtc *newEnum = new WCEnumFormatEtc(m_formats); | |
144 if(!newEnum) | |
145 return E_OUTOFMEMORY; | |
146 | |
147 newEnum->AddRef(); | |
148 newEnum->m_current = m_current; | |
149 *ppCloneEnumFormatEtc = newEnum; | |
150 return S_OK; | |
151 } | |
152 | |
153 | |
154 | |
155 ////////////////////////////////////////////////////////////////////////// | |
156 | |
157 HRESULT WCDataObject::createInstance(WCDataObject** result) | |
158 { | |
159 if (!result) | |
160 return E_POINTER; | |
161 *result = new WCDataObject(); | |
162 return S_OK; | |
163 } | |
164 | |
165 WCDataObject::WCDataObject() | |
166 : m_ref(1) | |
167 { | |
168 } | |
169 | |
170 WCDataObject::~WCDataObject() | |
171 { | |
172 for(size_t i = 0; i < m_medium.size(); ++i) { | |
173 ReleaseStgMedium(m_medium[i]); | |
174 delete m_medium[i]; | |
175 } | |
176 WTF::deleteAllValues(m_formats); | |
177 } | |
178 | |
179 STDMETHODIMP WCDataObject::QueryInterface(REFIID riid,void** ppvObject) | |
180 { | |
181 *ppvObject = 0; | |
182 if (IID_IUnknown==riid || IID_IDataObject==riid) | |
183 *ppvObject=this; | |
184 if (*ppvObject) { | |
185 AddRef(); | |
186 return S_OK; | |
187 } | |
188 return E_NOINTERFACE; | |
189 } | |
190 | |
191 STDMETHODIMP_(ULONG) WCDataObject::AddRef( void) | |
192 { | |
193 return InterlockedIncrement(&m_ref); | |
194 } | |
195 | |
196 STDMETHODIMP_(ULONG) WCDataObject::Release( void) | |
197 { | |
198 long c = InterlockedDecrement(&m_ref); | |
199 if (c == 0) | |
200 delete this; | |
201 return c; | |
202 } | |
203 | |
204 STDMETHODIMP WCDataObject::GetData(FORMATETC* pformatetcIn, STGMEDIUM* pmedium) | |
205 { | |
206 if(!pformatetcIn || !pmedium) | |
207 return E_POINTER; | |
208 pmedium->hGlobal = 0; | |
209 | |
210 for(size_t i=0; i < m_formats.size(); ++i) { | |
211 if(/*pformatetcIn->tymed & m_formats[i]->tymed &&*/ // tymed can be
0 (TYMED_NULL) - but it can have a medium that contains an pUnkForRelease | |
212 pformatetcIn->lindex == m_formats[i]->lindex && | |
213 pformatetcIn->dwAspect == m_formats[i]->dwAspect && | |
214 pformatetcIn->cfFormat == m_formats[i]->cfFormat) { | |
215 CopyMedium(pmedium, m_medium[i], m_formats[i]); | |
216 return S_OK; | |
217 } | |
218 } | |
219 return DV_E_FORMATETC; | |
220 } | |
221 | |
222 STDMETHODIMP WCDataObject::GetDataHere(FORMATETC*, STGMEDIUM*) | |
223 { | |
224 return E_NOTIMPL; | |
225 } | |
226 | |
227 STDMETHODIMP WCDataObject::QueryGetData(FORMATETC* pformatetc) | |
228 { | |
229 if(!pformatetc) | |
230 return E_POINTER; | |
231 | |
232 if (!(DVASPECT_CONTENT & pformatetc->dwAspect)) | |
233 return (DV_E_DVASPECT); | |
234 HRESULT hr = DV_E_TYMED; | |
235 for(size_t i = 0; i < m_formats.size(); ++i) { | |
236 if(pformatetc->tymed & m_formats[i]->tymed) { | |
237 if(pformatetc->cfFormat == m_formats[i]->cfFormat) | |
238 return S_OK; | |
239 else | |
240 hr = DV_E_CLIPFORMAT; | |
241 } | |
242 else | |
243 hr = DV_E_TYMED; | |
244 } | |
245 return hr; | |
246 } | |
247 | |
248 STDMETHODIMP WCDataObject::GetCanonicalFormatEtc(FORMATETC*, FORMATETC*) | |
249 { | |
250 return DATA_S_SAMEFORMATETC; | |
251 } | |
252 | |
253 STDMETHODIMP WCDataObject::SetData(FORMATETC* pformatetc, STGMEDIUM* pmedium, BO
OL fRelease) | |
254 { | |
255 if(!pformatetc || !pmedium) | |
256 return E_POINTER; | |
257 | |
258 FORMATETC* fetc=new FORMATETC; | |
259 if (!fetc) | |
260 return E_OUTOFMEMORY; | |
261 | |
262 STGMEDIUM* pStgMed = new STGMEDIUM; | |
263 | |
264 if(!pStgMed) { | |
265 delete fetc; | |
266 return E_OUTOFMEMORY; | |
267 } | |
268 | |
269 ZeroMemory(fetc,sizeof(FORMATETC)); | |
270 ZeroMemory(pStgMed,sizeof(STGMEDIUM)); | |
271 | |
272 *fetc = *pformatetc; | |
273 m_formats.append(fetc); | |
274 | |
275 if(fRelease) | |
276 *pStgMed = *pmedium; | |
277 else | |
278 CopyMedium(pStgMed, pmedium, pformatetc); | |
279 m_medium.append(pStgMed); | |
280 | |
281 return S_OK; | |
282 } | |
283 | |
284 void WCDataObject::CopyMedium(STGMEDIUM* pMedDest, STGMEDIUM* pMedSrc, FORMATETC
* pFmtSrc) | |
285 { | |
286 switch(pMedSrc->tymed) | |
287 { | |
288 case TYMED_HGLOBAL: | |
289 pMedDest->hGlobal = (HGLOBAL)OleDuplicateData(pMedSrc->hGlobal,pFmtSrc->
cfFormat, 0); | |
290 break; | |
291 case TYMED_GDI: | |
292 pMedDest->hBitmap = (HBITMAP)OleDuplicateData(pMedSrc->hBitmap,pFmtSrc->
cfFormat, 0); | |
293 break; | |
294 case TYMED_MFPICT: | |
295 pMedDest->hMetaFilePict = (HMETAFILEPICT)OleDuplicateData(pMedSrc->hMeta
FilePict,pFmtSrc->cfFormat, 0); | |
296 break; | |
297 case TYMED_ENHMF: | |
298 pMedDest->hEnhMetaFile = (HENHMETAFILE)OleDuplicateData(pMedSrc->hEnhMet
aFile,pFmtSrc->cfFormat, 0); | |
299 break; | |
300 case TYMED_FILE: | |
301 pMedSrc->lpszFileName = (LPOLESTR)OleDuplicateData(pMedSrc->lpszFileName
,pFmtSrc->cfFormat, 0); | |
302 break; | |
303 case TYMED_ISTREAM: | |
304 pMedDest->pstm = pMedSrc->pstm; | |
305 pMedSrc->pstm->AddRef(); | |
306 break; | |
307 case TYMED_ISTORAGE: | |
308 pMedDest->pstg = pMedSrc->pstg; | |
309 pMedSrc->pstg->AddRef(); | |
310 break; | |
311 default: | |
312 break; | |
313 } | |
314 pMedDest->tymed = pMedSrc->tymed; | |
315 pMedDest->pUnkForRelease = 0; | |
316 if(pMedSrc->pUnkForRelease != 0) { | |
317 pMedDest->pUnkForRelease = pMedSrc->pUnkForRelease; | |
318 pMedSrc->pUnkForRelease->AddRef(); | |
319 } | |
320 } | |
321 STDMETHODIMP WCDataObject::EnumFormatEtc(DWORD dwDirection, IEnumFORMATETC** ppe
numFormatEtc) | |
322 { | |
323 if(!ppenumFormatEtc) | |
324 return E_POINTER; | |
325 | |
326 *ppenumFormatEtc=0; | |
327 switch (dwDirection) | |
328 { | |
329 case DATADIR_GET: | |
330 *ppenumFormatEtc= new WCEnumFormatEtc(m_formats); | |
331 if(!(*ppenumFormatEtc)) | |
332 return E_OUTOFMEMORY; | |
333 break; | |
334 | |
335 case DATADIR_SET: | |
336 default: | |
337 return E_NOTIMPL; | |
338 break; | |
339 } | |
340 | |
341 return S_OK; | |
342 } | |
343 | |
344 STDMETHODIMP WCDataObject::DAdvise(FORMATETC*, DWORD, IAdviseSink*,DWORD*) | |
345 { | |
346 return OLE_E_ADVISENOTSUPPORTED; | |
347 } | |
348 | |
349 STDMETHODIMP WCDataObject::DUnadvise(DWORD) | |
350 { | |
351 return E_NOTIMPL; | |
352 } | |
353 | |
354 HRESULT STDMETHODCALLTYPE WCDataObject::EnumDAdvise(IEnumSTATDATA**) | |
355 { | |
356 return OLE_E_ADVISENOTSUPPORTED; | |
357 } | |
358 | |
359 void WCDataObject::clearData(CLIPFORMAT format) | |
360 { | |
361 size_t ptr = 0; | |
362 while (ptr < m_formats.size()) { | |
363 if (m_formats[ptr]->cfFormat == format) { | |
364 FORMATETC* current = m_formats[ptr]; | |
365 m_formats[ptr] = m_formats[m_formats.size() - 1]; | |
366 m_formats[m_formats.size() - 1] = 0; | |
367 m_formats.removeLast(); | |
368 delete current; | |
369 STGMEDIUM* medium = m_medium[ptr]; | |
370 m_medium[ptr] = m_medium[m_medium.size() - 1]; | |
371 m_medium[m_medium.size() - 1] = 0; | |
372 m_medium.removeLast(); | |
373 ReleaseStgMedium(medium); | |
374 delete medium; | |
375 continue; | |
376 } | |
377 ptr++; | |
378 } | |
379 } | |
380 | |
381 | |
382 } | |
OLD | NEW |