msi: Don't crash if record has no fields.
[wine] / dlls / quartz / enummoniker.c
1 /*
2  * IEnumMoniker implementation
3  *
4  * Copyright 2003 Robert Shearman
5  *
6  * This file contains the (internal) driver registration functions,
7  * driver enumeration APIs and DirectDraw creation functions.
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with this library; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22  */
23
24 #define COBJMACROS
25
26 #include "quartz_private.h"
27
28 #include "wine/debug.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(quartz);
31
32 typedef struct EnumMonikerImpl
33 {
34     const IEnumMonikerVtbl *lpVtbl;
35     LONG ref;
36     IMoniker ** ppMoniker;
37     ULONG nMonikerCount;
38     ULONG index;
39 } EnumMonikerImpl;
40
41 static const IEnumMonikerVtbl EnumMonikerImpl_Vtbl;
42
43 static ULONG WINAPI EnumMonikerImpl_AddRef(LPENUMMONIKER iface);
44
45 HRESULT EnumMonikerImpl_Create(IMoniker ** ppMoniker, ULONG nMonikerCount, IEnumMoniker ** ppEnum)
46 {
47     /* NOTE: assumes that array of IMonikers has already been AddRef'd
48      * I.e. this function does not AddRef the array of incoming
49      * IMonikers */
50     EnumMonikerImpl * pemi = CoTaskMemAlloc(sizeof(EnumMonikerImpl));
51
52     TRACE("(%p, %d, %p)\n", ppMoniker, nMonikerCount, ppEnum);
53
54     *ppEnum = NULL;
55
56     if (!pemi)
57         return E_OUTOFMEMORY;
58
59     pemi->lpVtbl = &EnumMonikerImpl_Vtbl;
60     pemi->ref = 1;
61     pemi->ppMoniker = CoTaskMemAlloc(nMonikerCount * sizeof(IMoniker*));
62     memcpy(pemi->ppMoniker, ppMoniker, nMonikerCount*sizeof(IMoniker*));
63     pemi->nMonikerCount = nMonikerCount;
64     pemi->index = 0;
65
66     *ppEnum = (IEnumMoniker *)pemi;
67
68     return S_OK;
69 }
70
71 /**********************************************************************
72  * IEnumMoniker_QueryInterface (also IUnknown)
73  */
74 static HRESULT WINAPI EnumMonikerImpl_QueryInterface(
75     LPENUMMONIKER iface,
76     REFIID riid,
77     LPVOID *ppvObj)
78 {
79     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
80     TRACE("\n\tIID:\t%s\n",debugstr_guid(riid));
81
82     if (This == NULL || ppvObj == NULL) return E_POINTER;
83
84     if (IsEqualGUID(riid, &IID_IUnknown) ||
85         IsEqualGUID(riid, &IID_IEnumMoniker))
86     {
87         *ppvObj = (LPVOID)iface;
88         EnumMonikerImpl_AddRef(iface);
89         return S_OK;
90     }
91
92     FIXME("- no interface\n\tIID:\t%s\n", debugstr_guid(riid));
93     return E_NOINTERFACE;
94 }
95
96 /**********************************************************************
97  * IEnumMoniker_AddRef (also IUnknown)
98  */
99 static ULONG WINAPI EnumMonikerImpl_AddRef(LPENUMMONIKER iface)
100 {
101     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
102     ULONG ref;
103
104     if (This == NULL) return E_POINTER;
105
106     ref = InterlockedIncrement(&This->ref);
107
108     TRACE("(%p)->() AddRef from %d\n", iface, ref - 1);
109
110     return ref;
111 }
112
113 /**********************************************************************
114  * IEnumMoniker_Release (also IUnknown)
115  */
116 static ULONG WINAPI EnumMonikerImpl_Release(LPENUMMONIKER iface)
117 {
118     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
119     ULONG ref = InterlockedDecrement(&This->ref);
120
121     TRACE("(%p)->() Release from %d\n", iface, ref + 1);
122
123     if (!ref)
124     {
125         CoTaskMemFree(This->ppMoniker);
126         This->ppMoniker = NULL;
127         CoTaskMemFree(This);
128         return 0;
129     }
130     return ref;
131 }
132
133 static HRESULT WINAPI EnumMonikerImpl_Next(LPENUMMONIKER iface, ULONG celt, IMoniker ** rgelt, ULONG * pceltFetched)
134 {
135     ULONG fetched;
136     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
137
138     TRACE("(%p)->(%d, %p, %p)\n", iface, celt, rgelt, pceltFetched);
139
140     for (fetched = 0; (This->index + fetched < This->nMonikerCount) && (fetched < celt); fetched++)
141     {
142         rgelt[fetched] = This->ppMoniker[This->index + fetched];
143         IMoniker_AddRef(rgelt[fetched]);
144     }
145
146     This->index += fetched;
147
148     TRACE("-- fetched %d\n", fetched);
149
150     if (pceltFetched)
151         *pceltFetched = fetched;
152
153     if (fetched != celt)
154         return S_FALSE;
155     else
156         return S_OK;
157 }
158
159 static HRESULT WINAPI EnumMonikerImpl_Skip(LPENUMMONIKER iface, ULONG celt)
160 {
161     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
162
163     TRACE("(%p)->(%d)\n", iface, celt);
164
165     This->index += celt;
166
167     return S_OK;
168 }
169
170 static HRESULT WINAPI EnumMonikerImpl_Reset(LPENUMMONIKER iface)
171 {
172     EnumMonikerImpl *This = (EnumMonikerImpl *)iface;
173
174     TRACE("(%p)->()\n", iface);
175
176     This->index = 0;
177
178     return S_OK;
179 }
180
181 static HRESULT WINAPI EnumMonikerImpl_Clone(LPENUMMONIKER iface, IEnumMoniker ** ppenum)
182 {
183     FIXME("(%p)->(%p): stub\n", iface, ppenum);
184
185     return E_NOTIMPL;
186 }
187
188 /**********************************************************************
189  * IEnumMoniker_Vtbl
190  */
191 static const IEnumMonikerVtbl EnumMonikerImpl_Vtbl =
192 {
193     EnumMonikerImpl_QueryInterface,
194     EnumMonikerImpl_AddRef,
195     EnumMonikerImpl_Release,
196     EnumMonikerImpl_Next,
197     EnumMonikerImpl_Skip,
198     EnumMonikerImpl_Reset,
199     EnumMonikerImpl_Clone
200 };