1 /*
2 * Copyright 2005 Jacek Caban
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 */
18
19 #include "urlmon_main.h"
20 #include "wine/debug.h"
21
22 WINE_DEFAULT_DEBUG_CHANNEL(urlmon);
23
24 static WCHAR wszEnumFORMATETC[] = {'_','E','n','u','m','F','O','R','M','A','T','E','T','C','_',0};
25
26 typedef struct {
27 IEnumFORMATETC IEnumFORMATETC_iface;
28
29 FORMATETC *fetc;
30 UINT fetc_cnt;
31 UINT it;
32
33 LONG ref;
34 } EnumFORMATETC;
35
impl_from_IEnumFORMATETC(IEnumFORMATETC * iface)36 static inline EnumFORMATETC *impl_from_IEnumFORMATETC(IEnumFORMATETC *iface)
37 {
38 return CONTAINING_RECORD(iface, EnumFORMATETC, IEnumFORMATETC_iface);
39 }
40
41 static IEnumFORMATETC *EnumFORMATETC_Create(UINT cfmtetc, const FORMATETC *rgfmtetc, UINT it);
42
EnumFORMATETC_QueryInterface(IEnumFORMATETC * iface,REFIID riid,void ** ppv)43 static HRESULT WINAPI EnumFORMATETC_QueryInterface(IEnumFORMATETC *iface, REFIID riid, void **ppv)
44 {
45 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
46
47 TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), ppv);
48
49 *ppv = NULL;
50
51 if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IEnumFORMATETC, riid)) {
52 IEnumFORMATETC_AddRef(iface);
53 *ppv = iface;
54 return S_OK;
55 }
56
57 WARN("not supported interface %s\n", debugstr_guid(riid));
58 return E_NOINTERFACE;
59 }
60
EnumFORMATETC_AddRef(IEnumFORMATETC * iface)61 static ULONG WINAPI EnumFORMATETC_AddRef(IEnumFORMATETC *iface)
62 {
63 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
64 LONG ref = InterlockedIncrement(&This->ref);
65 TRACE("(%p) ref=%d\n", This, ref);
66 return ref;
67 }
68
EnumFORMATETC_Release(IEnumFORMATETC * iface)69 static ULONG WINAPI EnumFORMATETC_Release(IEnumFORMATETC *iface)
70 {
71 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
72 LONG ref = InterlockedDecrement(&This->ref);
73
74 TRACE("(%p) ref=%d\n", This, ref);
75
76 if(!ref) {
77 heap_free(This->fetc);
78 heap_free(This);
79
80 URLMON_UnlockModule();
81 }
82
83 return ref;
84 }
85
EnumFORMATETC_Next(IEnumFORMATETC * iface,ULONG celt,FORMATETC * rgelt,ULONG * pceltFetched)86 static HRESULT WINAPI EnumFORMATETC_Next(IEnumFORMATETC *iface, ULONG celt,
87 FORMATETC *rgelt, ULONG *pceltFetched)
88 {
89 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
90 ULONG cnt;
91
92 TRACE("(%p)->(%d %p %p)\n", This, celt, rgelt, pceltFetched);
93
94 if(!rgelt)
95 return E_INVALIDARG;
96
97 if(This->it >= This->fetc_cnt || !celt) {
98 if(pceltFetched)
99 *pceltFetched = 0;
100 return celt ? S_FALSE : S_OK;
101 }
102
103 cnt = This->fetc_cnt-This->it > celt ? celt : This->fetc_cnt-This->it;
104
105 memcpy(rgelt, This->fetc+This->it, cnt*sizeof(FORMATETC));
106 This->it += cnt;
107
108 if(pceltFetched)
109 *pceltFetched = cnt;
110
111 return cnt == celt ? S_OK : S_FALSE;
112 }
113
EnumFORMATETC_Skip(IEnumFORMATETC * iface,ULONG celt)114 static HRESULT WINAPI EnumFORMATETC_Skip(IEnumFORMATETC *iface, ULONG celt)
115 {
116 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
117
118 TRACE("(%p)->(%d)\n", This, celt);
119
120 This->it += celt;
121 return This->it > This->fetc_cnt ? S_FALSE : S_OK;
122 }
123
EnumFORMATETC_Reset(IEnumFORMATETC * iface)124 static HRESULT WINAPI EnumFORMATETC_Reset(IEnumFORMATETC *iface)
125 {
126 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
127
128 TRACE("(%p)\n", This);
129
130 This->it = 0;
131 return S_OK;
132 }
133
EnumFORMATETC_Clone(IEnumFORMATETC * iface,IEnumFORMATETC ** ppenum)134 static HRESULT WINAPI EnumFORMATETC_Clone(IEnumFORMATETC *iface, IEnumFORMATETC **ppenum)
135 {
136 EnumFORMATETC *This = impl_from_IEnumFORMATETC(iface);
137
138 TRACE("(%p)->(%p)\n", This, ppenum);
139
140 if(!ppenum)
141 return E_INVALIDARG;
142
143 *ppenum = EnumFORMATETC_Create(This->fetc_cnt, This->fetc, This->it);
144 return S_OK;
145 }
146
147 static const IEnumFORMATETCVtbl EnumFORMATETCVtbl = {
148 EnumFORMATETC_QueryInterface,
149 EnumFORMATETC_AddRef,
150 EnumFORMATETC_Release,
151 EnumFORMATETC_Next,
152 EnumFORMATETC_Skip,
153 EnumFORMATETC_Reset,
154 EnumFORMATETC_Clone
155 };
156
EnumFORMATETC_Create(UINT cfmtetc,const FORMATETC * rgfmtetc,UINT it)157 static IEnumFORMATETC *EnumFORMATETC_Create(UINT cfmtetc, const FORMATETC *rgfmtetc, UINT it)
158 {
159 EnumFORMATETC *ret = heap_alloc(sizeof(EnumFORMATETC));
160
161 URLMON_LockModule();
162
163 ret->IEnumFORMATETC_iface.lpVtbl = &EnumFORMATETCVtbl;
164 ret->ref = 1;
165 ret->it = it;
166 ret->fetc_cnt = cfmtetc;
167
168 ret->fetc = heap_alloc(cfmtetc*sizeof(FORMATETC));
169 memcpy(ret->fetc, rgfmtetc, cfmtetc*sizeof(FORMATETC));
170
171 return &ret->IEnumFORMATETC_iface;
172 }
173
174 /**********************************************************
175 * CreateFormatEnumerator (urlmon.@)
176 */
CreateFormatEnumerator(UINT cfmtetc,FORMATETC * rgfmtetc,IEnumFORMATETC ** ppenumfmtetc)177 HRESULT WINAPI CreateFormatEnumerator(UINT cfmtetc, FORMATETC *rgfmtetc,
178 IEnumFORMATETC** ppenumfmtetc)
179 {
180 TRACE("(%d %p %p)\n", cfmtetc, rgfmtetc, ppenumfmtetc);
181
182 if(!ppenumfmtetc)
183 return E_INVALIDARG;
184 if(!cfmtetc)
185 return E_FAIL;
186
187 *ppenumfmtetc = EnumFORMATETC_Create(cfmtetc, rgfmtetc, 0);
188 return S_OK;
189 }
190
191 /**********************************************************
192 * RegisterFormatEnumerator (urlmon.@)
193 */
RegisterFormatEnumerator(LPBC pBC,IEnumFORMATETC * pEFetc,DWORD reserved)194 HRESULT WINAPI RegisterFormatEnumerator(LPBC pBC, IEnumFORMATETC *pEFetc, DWORD reserved)
195 {
196 TRACE("(%p %p %d)\n", pBC, pEFetc, reserved);
197
198 if(reserved)
199 WARN("reserved != 0\n");
200
201 if(!pBC || !pEFetc)
202 return E_INVALIDARG;
203
204 return IBindCtx_RegisterObjectParam(pBC, wszEnumFORMATETC, (IUnknown*)pEFetc);
205 }
206
207 /**********************************************************
208 * RevokeFormatEnumerator (urlmon.@)
209 */
RevokeFormatEnumerator(LPBC pbc,IEnumFORMATETC * pEFetc)210 HRESULT WINAPI RevokeFormatEnumerator(LPBC pbc, IEnumFORMATETC *pEFetc)
211 {
212 TRACE("(%p %p)\n", pbc, pEFetc);
213
214 if(!pbc)
215 return E_INVALIDARG;
216
217 return IBindCtx_RevokeObjectParam(pbc, wszEnumFORMATETC);
218 }
219