msvcrt: Implemented setjmp for x86_64.
[wine] / dlls / browseui / aclmulti.c
1 /*
2  *      Multisource AutoComplete list
3  *
4  *      Copyright 2007  Mikolaj Zalewski
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include "config.h"
22
23 #include <stdarg.h>
24
25 #define COBJMACROS
26
27 #include "wine/debug.h"
28 #include "windef.h"
29 #include "winbase.h"
30 #include "winreg.h"
31 #include "winuser.h"
32 #include "shlwapi.h"
33 #include "winerror.h"
34 #include "objbase.h"
35
36 #include "shlguid.h"
37 #include "shlobj.h"
38
39 #include "wine/unicode.h"
40
41 #include "browseui.h"
42
43 WINE_DEFAULT_DEBUG_CHANNEL(browseui);
44
45 struct ACLMultiSublist {
46     IUnknown *punk;
47     IEnumString *pEnum;
48     IACList *pACL;
49 };
50
51 typedef struct tagACLMulti {
52     const IEnumStringVtbl *vtbl;
53     const IACListVtbl *aclVtbl;
54     const IObjMgrVtbl *objmgrVtbl;
55     LONG refCount;
56     INT nObjs;
57     INT currObj;
58     struct ACLMultiSublist *objs;
59 } ACLMulti;
60
61 static inline ACLMulti *impl_from_IACList(IACList *iface)
62 {
63     return (ACLMulti *)((char *)iface - FIELD_OFFSET(ACLMulti, aclVtbl));
64 }
65
66 static inline ACLMulti *impl_from_IObjMgr(IObjMgr *iface)
67 {
68     return (ACLMulti *)((char *)iface - FIELD_OFFSET(ACLMulti, objmgrVtbl));
69 }
70
71 static void release_obj(struct ACLMultiSublist *obj)
72 {
73     IUnknown_Release(obj->punk);
74     if (obj->pEnum)
75         IEnumString_Release(obj->pEnum);
76     if (obj->pACL)
77         IACList_Release(obj->pACL);
78 }
79
80 static void ACLMulti_Destructor(ACLMulti *This)
81 {
82     int i;
83     TRACE("destroying %p\n", This);
84     for (i = 0; i < This->nObjs; i++)
85         release_obj(&This->objs[i]);
86     heap_free(This->objs);
87     heap_free(This);
88     BROWSEUI_refCount--;
89 }
90
91 static HRESULT WINAPI ACLMulti_QueryInterface(IEnumString *iface, REFIID iid, LPVOID *ppvOut)
92 {
93     ACLMulti *This = (ACLMulti *)iface;
94     *ppvOut = NULL;
95
96     if (IsEqualIID(iid, &IID_IUnknown) || IsEqualIID(iid, &IID_IEnumString))
97     {
98         *ppvOut = This;
99     }
100     else if (IsEqualIID(iid, &IID_IACList))
101     {
102         *ppvOut = &This->aclVtbl;
103     }
104     else if (IsEqualIID(iid, &IID_IObjMgr))
105     {
106         *ppvOut = &This->objmgrVtbl;
107     }
108
109     if (*ppvOut)
110     {
111         IUnknown_AddRef(iface);
112         return S_OK;
113     }
114
115     WARN("unsupported interface: %s\n", debugstr_guid(iid));
116     return E_NOINTERFACE;
117 }
118
119 static ULONG WINAPI ACLMulti_AddRef(IEnumString *iface)
120 {
121     ACLMulti *This = (ACLMulti *)iface;
122     return InterlockedIncrement(&This->refCount);
123 }
124
125 static ULONG WINAPI ACLMulti_Release(IEnumString *iface)
126 {
127     ACLMulti *This = (ACLMulti *)iface;
128     ULONG ret;
129
130     ret = InterlockedDecrement(&This->refCount);
131     if (ret == 0)
132         ACLMulti_Destructor(This);
133     return ret;
134 }
135
136 static HRESULT WINAPI ACLMulti_Append(IObjMgr *iface, IUnknown *obj)
137 {
138     ACLMulti *This = impl_from_IObjMgr(iface);
139
140     TRACE("(%p, %p)\n", This, obj);
141     if (obj == NULL)
142         return E_FAIL;
143
144     This->objs = heap_realloc(This->objs, sizeof(This->objs[0]) * (This->nObjs+1));
145     This->objs[This->nObjs].punk = obj;
146     IUnknown_AddRef(obj);
147     if (FAILED(IUnknown_QueryInterface(obj, &IID_IEnumString, (LPVOID *)&This->objs[This->nObjs].pEnum)))
148         This->objs[This->nObjs].pEnum = NULL;
149     if (FAILED(IUnknown_QueryInterface(obj, &IID_IACList, (LPVOID *)&This->objs[This->nObjs].pACL)))
150         This->objs[This->nObjs].pACL = NULL;
151     This->nObjs++;
152     return S_OK;
153 }
154
155 static HRESULT WINAPI ACLMulti_Remove(IObjMgr *iface, IUnknown *obj)
156 {
157     ACLMulti *This = impl_from_IObjMgr(iface);
158     int i;
159
160     TRACE("(%p, %p)\n", This, obj);
161     for (i = 0; i < This->nObjs; i++)
162         if (This->objs[i].punk == obj)
163         {
164             release_obj(&This->objs[i]);
165             memmove(&This->objs[i], &This->objs[i+1], (This->nObjs-i-1)*sizeof(struct ACLMultiSublist));
166             This->nObjs--;
167             This->objs = heap_realloc(This->objs, sizeof(This->objs[0]) * This->nObjs);
168             return S_OK;
169         }
170
171     return E_FAIL;
172 }
173
174 static HRESULT WINAPI ACLMulti_Next(IEnumString *iface, ULONG celt, LPOLESTR *rgelt, ULONG *pceltFetched)
175 {
176     ACLMulti *This = (ACLMulti *)iface;
177
178     TRACE("(%p, %d, %p, %p)\n", iface, celt, rgelt, pceltFetched);
179     while (This->currObj < This->nObjs)
180     {
181         if (This->objs[This->currObj].pEnum)
182         {
183             /* native browseui 6.0 also returns only one element */
184             HRESULT ret = IEnumString_Next(This->objs[This->currObj].pEnum, 1, rgelt, pceltFetched);
185             if (ret != S_FALSE)
186                 return ret;
187         }
188         This->currObj++;
189     }
190
191     if (pceltFetched)
192         *pceltFetched = 0;
193     *rgelt = NULL;
194     return S_FALSE;
195 }
196
197 static HRESULT WINAPI ACLMulti_Reset(IEnumString *iface)
198 {
199     ACLMulti *This = (ACLMulti *)iface;
200     int i;
201
202     This->currObj = 0;
203     for (i = 0; i < This->nObjs; i++)
204     {
205         if (This->objs[i].pEnum)
206             IEnumString_Reset(This->objs[i].pEnum);
207     }
208     return S_OK;
209 }
210
211 static HRESULT WINAPI ACLMulti_Skip(IEnumString *iface, ULONG celt)
212 {
213     /* native browseui 6.0 returns this: */
214     return E_NOTIMPL;
215 }
216
217 static HRESULT WINAPI ACLMulti_Clone(IEnumString *iface, IEnumString **ppOut)
218 {
219     *ppOut = NULL;
220     /* native browseui 6.0 returns this: */
221     return E_OUTOFMEMORY;
222 }
223
224 static HRESULT WINAPI ACLMulti_Expand(IACList *iface, LPCWSTR wstr)
225 {
226     ACLMulti *This = impl_from_IACList(iface);
227     HRESULT res = S_OK;
228     int i;
229
230     for (i = 0; i < This->nObjs; i++)
231     {
232         if (!This->objs[i].pACL)
233             continue;
234         res = IACList_Expand(This->objs[i].pACL, wstr);
235         /* Vista behaviour - XP would break out of the loop if res == S_OK (usually calling Expand only once) */
236     }
237     return res;
238 }
239
240 static const IEnumStringVtbl ACLMultiVtbl =
241 {
242     ACLMulti_QueryInterface,
243     ACLMulti_AddRef,
244     ACLMulti_Release,
245
246     ACLMulti_Next,
247     ACLMulti_Skip,
248     ACLMulti_Reset,
249     ACLMulti_Clone
250 };
251
252 static HRESULT WINAPI ACLMulti_IObjMgr_QueryInterface(IObjMgr *iface, REFIID iid, LPVOID *ppvOut)
253 {
254     ACLMulti *This = impl_from_IObjMgr(iface);
255     return ACLMulti_QueryInterface((IEnumString *)This, iid, ppvOut);
256 }
257
258 static ULONG WINAPI ACLMulti_IObjMgr_AddRef(IObjMgr *iface)
259 {
260     ACLMulti *This = impl_from_IObjMgr(iface);
261     return ACLMulti_AddRef((IEnumString *)This);
262 }
263
264 static ULONG WINAPI ACLMulti_IObjMgr_Release(IObjMgr *iface)
265 {
266     ACLMulti *This = impl_from_IObjMgr(iface);
267     return ACLMulti_Release((IEnumString *)This);
268 }
269
270 static const IObjMgrVtbl ACLMulti_ObjMgrVtbl =
271 {
272     ACLMulti_IObjMgr_QueryInterface,
273     ACLMulti_IObjMgr_AddRef,
274     ACLMulti_IObjMgr_Release,
275
276     ACLMulti_Append,
277     ACLMulti_Remove
278 };
279
280 static HRESULT WINAPI ACLMulti_IACList_QueryInterface(IACList *iface, REFIID iid, LPVOID *ppvOut)
281 {
282     ACLMulti *This = impl_from_IACList(iface);
283     return ACLMulti_QueryInterface((IEnumString *)This, iid, ppvOut);
284 }
285
286 static ULONG WINAPI ACLMulti_IACList_AddRef(IACList *iface)
287 {
288     ACLMulti *This = impl_from_IACList(iface);
289     return ACLMulti_AddRef((IEnumString *)This);
290 }
291
292 static ULONG WINAPI ACLMulti_IACList_Release(IACList *iface)
293 {
294     ACLMulti *This = impl_from_IACList(iface);
295     return ACLMulti_Release((IEnumString *)This);
296 }
297
298 static const IACListVtbl ACLMulti_ACListVtbl =
299 {
300     ACLMulti_IACList_QueryInterface,
301     ACLMulti_IACList_AddRef,
302     ACLMulti_IACList_Release,
303
304     ACLMulti_Expand
305 };
306
307 HRESULT ACLMulti_Constructor(IUnknown *pUnkOuter, IUnknown **ppOut)
308 {
309     ACLMulti *This;
310     if (pUnkOuter)
311         return CLASS_E_NOAGGREGATION;
312
313     This = heap_alloc_zero(sizeof(ACLMulti));
314     if (This == NULL)
315         return E_OUTOFMEMORY;
316
317     This->vtbl = &ACLMultiVtbl;
318     This->aclVtbl = &ACLMulti_ACListVtbl;
319     This->objmgrVtbl = &ACLMulti_ObjMgrVtbl;
320     This->refCount = 1;
321
322     TRACE("returning %p\n", This);
323     *ppOut = (IUnknown *)This;
324     BROWSEUI_refCount++;
325     return S_OK;
326 }