ole32: Use GetWindowLongPtr/SetWindowLongPtr for pointers.
[wine] / dlls / ole32 / oleobj.c
1 /*
2  *      OLE2 COM objects
3  *
4  *      Copyright 1998 Eric Kohl
5  *      Copyright 1999 Francis Beaudet
6  *
7  * This library is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * This library is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20  */
21
22
23 #include <stdarg.h>
24 #include <string.h>
25
26 #define COBJMACROS
27
28 #include "windef.h"
29 #include "winbase.h"
30 #include "winuser.h"
31 #include "winerror.h"
32 #include "wine/debug.h"
33 #include "ole2.h"
34
35 #include "compobj_private.h"
36
37 WINE_DEFAULT_DEBUG_CHANNEL(ole);
38
39 #define INITIAL_SINKS 10
40
41 /**************************************************************************
42  *  OleAdviseHolderImpl Implementation
43  */
44 typedef struct OleAdviseHolderImpl
45 {
46   const IOleAdviseHolderVtbl *lpVtbl;
47
48   LONG ref;
49
50   DWORD         maxSinks;
51   IAdviseSink** arrayOfSinks;
52
53 } OleAdviseHolderImpl;
54
55 static HRESULT EnumOleSTATDATA_Construct(OleAdviseHolderImpl *pOleAdviseHolder, ULONG index, IEnumSTATDATA **ppenum);
56
57 typedef struct
58 {
59     const IEnumSTATDATAVtbl *lpvtbl;
60     LONG ref;
61
62     ULONG index;
63     OleAdviseHolderImpl *pOleAdviseHolder;
64 } EnumOleSTATDATA;
65
66 static HRESULT WINAPI EnumOleSTATDATA_QueryInterface(
67     IEnumSTATDATA *iface, REFIID riid, void **ppv)
68 {
69     TRACE("(%s, %p)\n", debugstr_guid(riid), ppv);
70     if (IsEqualIID(riid, &IID_IUnknown) ||
71         IsEqualIID(riid, &IID_IEnumSTATDATA))
72     {
73         IUnknown_AddRef(iface);
74         *ppv = iface;
75         return S_OK;
76     }
77     return E_NOINTERFACE;
78 }
79
80 static ULONG WINAPI EnumOleSTATDATA_AddRef(
81     IEnumSTATDATA *iface)
82 {
83     EnumOleSTATDATA *This = (EnumOleSTATDATA *)iface;
84     TRACE("()\n");
85     return InterlockedIncrement(&This->ref);
86 }
87
88 static ULONG WINAPI EnumOleSTATDATA_Release(
89     IEnumSTATDATA *iface)
90 {
91     EnumOleSTATDATA *This = (EnumOleSTATDATA *)iface;
92     LONG refs = InterlockedDecrement(&This->ref);
93     TRACE("()\n");
94     if (!refs)
95     {
96         IOleAdviseHolder_Release((IOleAdviseHolder *)This->pOleAdviseHolder);
97         HeapFree(GetProcessHeap(), 0, This);
98     }
99     return refs;
100 }
101
102 static HRESULT WINAPI EnumOleSTATDATA_Next(
103     IEnumSTATDATA *iface, ULONG celt, LPSTATDATA rgelt,
104     ULONG *pceltFetched)
105 {
106     EnumOleSTATDATA *This = (EnumOleSTATDATA *)iface;
107     HRESULT hr = S_OK;
108
109     TRACE("(%d, %p, %p)\n", celt, rgelt, pceltFetched);
110
111     if (pceltFetched)
112         *pceltFetched = 0;
113
114     for (; celt; celt--, rgelt++)
115     {
116         while ((This->index < This->pOleAdviseHolder->maxSinks) && 
117                !This->pOleAdviseHolder->arrayOfSinks[This->index])
118         {
119             This->index++;
120         }
121         if (This->index >= This->pOleAdviseHolder->maxSinks)
122         {
123             hr = S_FALSE;
124             break;
125         }
126
127         memset(&rgelt->formatetc, 0, sizeof(rgelt->formatetc));
128         rgelt->advf = 0;
129         rgelt->pAdvSink = This->pOleAdviseHolder->arrayOfSinks[This->index];
130         IAdviseSink_AddRef(rgelt->pAdvSink);
131         rgelt->dwConnection = This->index;
132
133         if (pceltFetched)
134             (*pceltFetched)++;
135         This->index++;
136     }
137     return hr;
138 }
139
140 static HRESULT WINAPI EnumOleSTATDATA_Skip(
141     IEnumSTATDATA *iface, ULONG celt)
142 {
143     EnumOleSTATDATA *This = (EnumOleSTATDATA *)iface;
144
145     TRACE("(%d)\n", celt);
146
147     for (; celt; celt--)
148     {
149         while ((This->index < This->pOleAdviseHolder->maxSinks) && 
150                !This->pOleAdviseHolder->arrayOfSinks[This->index])
151         {
152             This->index++;
153         }
154         if (This->index >= This->pOleAdviseHolder->maxSinks)
155             return S_FALSE;
156         This->index++;
157     }
158     return S_OK;
159 }
160
161 static HRESULT WINAPI EnumOleSTATDATA_Reset(
162     IEnumSTATDATA *iface)
163 {
164     EnumOleSTATDATA *This = (EnumOleSTATDATA *)iface;
165
166     TRACE("()\n");
167
168     This->index = 0;
169     return S_OK;
170 }
171
172 static HRESULT WINAPI EnumOleSTATDATA_Clone(
173     IEnumSTATDATA *iface,
174     IEnumSTATDATA **ppenum)
175 {
176     EnumOleSTATDATA *This = (EnumOleSTATDATA *)iface;
177     return EnumOleSTATDATA_Construct(This->pOleAdviseHolder, This->index, ppenum);
178 }
179
180 static const IEnumSTATDATAVtbl EnumOleSTATDATA_VTable =
181 {
182     EnumOleSTATDATA_QueryInterface,
183     EnumOleSTATDATA_AddRef,
184     EnumOleSTATDATA_Release,
185     EnumOleSTATDATA_Next,
186     EnumOleSTATDATA_Skip,
187     EnumOleSTATDATA_Reset,
188     EnumOleSTATDATA_Clone
189 };
190
191 static HRESULT EnumOleSTATDATA_Construct(OleAdviseHolderImpl *pOleAdviseHolder, ULONG index, IEnumSTATDATA **ppenum)
192 {
193     EnumOleSTATDATA *This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
194     if (!This)
195         return E_OUTOFMEMORY;
196     This->lpvtbl = &EnumOleSTATDATA_VTable;
197     This->ref = 1;
198     This->index = index;
199     This->pOleAdviseHolder = pOleAdviseHolder;
200     IOleAdviseHolder_AddRef((IOleAdviseHolder *)pOleAdviseHolder);
201     *ppenum = (IEnumSTATDATA *)&This->lpvtbl;
202     return S_OK;    
203 }
204
205 /**************************************************************************
206  *  OleAdviseHolderImpl_Destructor
207  */
208 static void OleAdviseHolderImpl_Destructor(
209   OleAdviseHolderImpl* ptrToDestroy)
210 {
211   DWORD index;
212   TRACE("%p\n", ptrToDestroy);
213
214   for (index = 0; index < ptrToDestroy->maxSinks; index++)
215   {
216     if (ptrToDestroy->arrayOfSinks[index]!=0)
217     {
218       IAdviseSink_Release(ptrToDestroy->arrayOfSinks[index]);
219       ptrToDestroy->arrayOfSinks[index] = NULL;
220     }
221   }
222
223   HeapFree(GetProcessHeap(),
224            0,
225            ptrToDestroy->arrayOfSinks);
226
227
228   HeapFree(GetProcessHeap(),
229            0,
230            ptrToDestroy);
231 }
232
233 /**************************************************************************
234  *  OleAdviseHolderImpl_QueryInterface
235  */
236 static HRESULT WINAPI OleAdviseHolderImpl_QueryInterface(
237   LPOLEADVISEHOLDER iface,
238   REFIID            riid,
239   LPVOID*           ppvObj)
240 {
241   OleAdviseHolderImpl *This = (OleAdviseHolderImpl *)iface;
242   TRACE("(%p)->(%s,%p)\n",This,debugstr_guid(riid),ppvObj);
243   /*
244    * Sanity check
245    */
246   if (ppvObj==NULL)
247     return E_POINTER;
248
249   *ppvObj = NULL;
250
251   if (IsEqualIID(riid, &IID_IUnknown))
252   {
253     /* IUnknown */
254     *ppvObj = This;
255   }
256   else if(IsEqualIID(riid, &IID_IOleAdviseHolder))
257   {
258     /* IOleAdviseHolder */
259     *ppvObj = (IOleAdviseHolder*) This;
260   }
261
262   if(*ppvObj == NULL)
263     return E_NOINTERFACE;
264
265   /*
266    * A successful QI always increments the reference count.
267    */
268   IUnknown_AddRef((IUnknown*)*ppvObj);
269
270   return S_OK;
271 }
272
273 /******************************************************************************
274  * OleAdviseHolderImpl_AddRef
275  */
276 static ULONG WINAPI OleAdviseHolderImpl_AddRef(
277   LPOLEADVISEHOLDER iface)
278 {
279   OleAdviseHolderImpl *This = (OleAdviseHolderImpl *)iface;
280   ULONG ref = InterlockedIncrement(&This->ref);
281
282   TRACE("(%p)->(ref=%d)\n", This, ref - 1);
283
284   return ref;
285 }
286
287 /******************************************************************************
288  * OleAdviseHolderImpl_Release
289  */
290 static ULONG WINAPI OleAdviseHolderImpl_Release(
291   LPOLEADVISEHOLDER iface)
292 {
293   OleAdviseHolderImpl *This = (OleAdviseHolderImpl *)iface;
294   ULONG ref;
295   TRACE("(%p)->(ref=%d)\n", This, This->ref);
296   ref = InterlockedDecrement(&This->ref);
297
298   if (ref == 0) OleAdviseHolderImpl_Destructor(This);
299
300   return ref;
301 }
302
303 /******************************************************************************
304  * OleAdviseHolderImpl_Advise
305  */
306 static HRESULT WINAPI OleAdviseHolderImpl_Advise(
307   LPOLEADVISEHOLDER iface,
308   IAdviseSink*      pAdvise,
309   DWORD*            pdwConnection)
310 {
311   DWORD index;
312
313   OleAdviseHolderImpl *This = (OleAdviseHolderImpl *)iface;
314
315   TRACE("(%p)->(%p, %p)\n", This, pAdvise, pdwConnection);
316
317   /*
318    * Sanity check
319    */
320   if (pdwConnection==NULL)
321     return E_POINTER;
322
323   *pdwConnection = 0;
324
325   /*
326    * Find a free spot in the array.
327    */
328   for (index = 0; index < This->maxSinks; index++)
329   {
330     if (This->arrayOfSinks[index]==NULL)
331       break;
332   }
333
334   /*
335    * If the array is full, we need to grow it.
336    */
337   if (index == This->maxSinks)
338   {
339     DWORD i;
340
341     This->maxSinks+=INITIAL_SINKS;
342
343     This->arrayOfSinks = HeapReAlloc(GetProcessHeap(),
344                                      0,
345                                      This->arrayOfSinks,
346                                      This->maxSinks*sizeof(IAdviseSink*));
347
348     for (i=index;i < This->maxSinks; i++)
349       This->arrayOfSinks[i]=0;
350   }
351
352   /*
353    * Store the new sink
354    */
355   This->arrayOfSinks[index] = pAdvise;
356
357   if (This->arrayOfSinks[index]!=NULL)
358     IAdviseSink_AddRef(This->arrayOfSinks[index]);
359
360   /*
361    * Return the index as the cookie.
362    * Since 0 is not a valid cookie, we will increment by
363    * 1 the index in the table.
364    */
365   *pdwConnection = index+1;
366
367   return S_OK;
368 }
369
370 /******************************************************************************
371  * OleAdviseHolderImpl_Unadvise
372  */
373 static HRESULT WINAPI OleAdviseHolderImpl_Unadvise(
374   LPOLEADVISEHOLDER iface,
375   DWORD             dwConnection)
376 {
377   OleAdviseHolderImpl *This = (OleAdviseHolderImpl *)iface;
378
379   TRACE("(%p)->(%u)\n", This, dwConnection);
380
381   /*
382    * So we don't return 0 as a cookie, the index was
383    * incremented by 1 in OleAdviseHolderImpl_Advise
384    * we have to compensate.
385    */
386   dwConnection--;
387
388   /*
389    * Check for invalid cookies.
390    */
391   if (dwConnection >= This->maxSinks)
392     return OLE_E_NOCONNECTION;
393
394   if (This->arrayOfSinks[dwConnection] == NULL)
395     return OLE_E_NOCONNECTION;
396
397   /*
398    * Release the sink and mark the spot in the list as free.
399    */
400   IAdviseSink_Release(This->arrayOfSinks[dwConnection]);
401   This->arrayOfSinks[dwConnection] = NULL;
402
403   return S_OK;
404 }
405
406 /******************************************************************************
407  * OleAdviseHolderImpl_EnumAdvise
408  */
409 static HRESULT WINAPI
410 OleAdviseHolderImpl_EnumAdvise (LPOLEADVISEHOLDER iface, IEnumSTATDATA **ppenumAdvise)
411 {
412     OleAdviseHolderImpl *This = (OleAdviseHolderImpl *)iface;
413
414     TRACE("(%p)->(%p)\n", This, ppenumAdvise);
415
416     *ppenumAdvise = NULL;
417
418     return EnumOleSTATDATA_Construct(This, 0, ppenumAdvise);
419 }
420
421 /******************************************************************************
422  * OleAdviseHolderImpl_SendOnRename
423  */
424 static HRESULT WINAPI
425 OleAdviseHolderImpl_SendOnRename (LPOLEADVISEHOLDER iface, IMoniker *pmk)
426 {
427     IEnumSTATDATA *pEnum;
428     HRESULT hr;
429
430     TRACE("(%p)->(%p)\n", iface, pmk);
431
432     hr = IOleAdviseHolder_EnumAdvise(iface, &pEnum);
433     if (SUCCEEDED(hr))
434     {
435         STATDATA statdata;
436         while (IEnumSTATDATA_Next(pEnum, 1, &statdata, NULL) == S_OK)
437         {
438             IAdviseSink_OnRename(statdata.pAdvSink, pmk);
439
440             IAdviseSink_Release(statdata.pAdvSink);
441         }
442         IEnumSTATDATA_Release(pEnum);
443     }
444
445     return hr;
446 }
447
448 /******************************************************************************
449  * OleAdviseHolderImpl_SendOnSave
450  */
451 static HRESULT WINAPI
452 OleAdviseHolderImpl_SendOnSave (LPOLEADVISEHOLDER iface)
453 {
454     IEnumSTATDATA *pEnum;
455     HRESULT hr;
456
457     TRACE("(%p)->()\n", iface);
458
459     hr = IOleAdviseHolder_EnumAdvise(iface, &pEnum);
460     if (SUCCEEDED(hr))
461     {
462         STATDATA statdata;
463         while (IEnumSTATDATA_Next(pEnum, 1, &statdata, NULL) == S_OK)
464         {
465             IAdviseSink_OnSave(statdata.pAdvSink);
466
467             IAdviseSink_Release(statdata.pAdvSink);
468         }
469         IEnumSTATDATA_Release(pEnum);
470     }
471
472     return hr;
473 }
474
475 /******************************************************************************
476  * OleAdviseHolderImpl_SendOnClose
477  */
478 static HRESULT WINAPI
479 OleAdviseHolderImpl_SendOnClose (LPOLEADVISEHOLDER iface)
480 {
481     IEnumSTATDATA *pEnum;
482     HRESULT hr;
483
484     TRACE("(%p)->()\n", iface);
485
486     hr = IOleAdviseHolder_EnumAdvise(iface, &pEnum);
487     if (SUCCEEDED(hr))
488     {
489         STATDATA statdata;
490         while (IEnumSTATDATA_Next(pEnum, 1, &statdata, NULL) == S_OK)
491         {
492             IAdviseSink_OnClose(statdata.pAdvSink);
493
494             IAdviseSink_Release(statdata.pAdvSink);
495         }
496         IEnumSTATDATA_Release(pEnum);
497     }
498
499     return hr;
500 }
501
502 /**************************************************************************
503  *  OleAdviseHolderImpl_VTable
504  */
505 static const IOleAdviseHolderVtbl oahvt =
506 {
507     OleAdviseHolderImpl_QueryInterface,
508     OleAdviseHolderImpl_AddRef,
509     OleAdviseHolderImpl_Release,
510     OleAdviseHolderImpl_Advise,
511     OleAdviseHolderImpl_Unadvise,
512     OleAdviseHolderImpl_EnumAdvise,
513     OleAdviseHolderImpl_SendOnRename,
514     OleAdviseHolderImpl_SendOnSave,
515     OleAdviseHolderImpl_SendOnClose
516 };
517
518 /**************************************************************************
519  *  OleAdviseHolderImpl_Constructor
520  */
521
522 static LPOLEADVISEHOLDER OleAdviseHolderImpl_Constructor(void)
523 {
524   OleAdviseHolderImpl* lpoah;
525   DWORD                index;
526
527   lpoah = HeapAlloc(GetProcessHeap(), 0, sizeof(OleAdviseHolderImpl));
528
529   lpoah->lpVtbl = &oahvt;
530   lpoah->ref = 1;
531   lpoah->maxSinks = INITIAL_SINKS;
532   lpoah->arrayOfSinks = HeapAlloc(GetProcessHeap(),
533                                   0,
534                                   lpoah->maxSinks * sizeof(IAdviseSink*));
535
536   for (index = 0; index < lpoah->maxSinks; index++)
537     lpoah->arrayOfSinks[index]=0;
538
539   TRACE("returning %p\n", lpoah);
540   return (LPOLEADVISEHOLDER)lpoah;
541 }
542
543 /**************************************************************************
544  *  DataAdviseHolder Implementation
545  */
546 typedef struct DataAdviseConnection {
547   IAdviseSink *sink;
548   FORMATETC fmat;
549   DWORD advf;
550   DWORD remote_connection;
551 } DataAdviseConnection;
552
553 typedef struct DataAdviseHolder
554 {
555   const IDataAdviseHolderVtbl *lpVtbl;
556
557   LONG                  ref;
558   DWORD                 maxCons;
559   DataAdviseConnection* Connections;
560   IDataObject*          delegate;
561 } DataAdviseHolder;
562
563 /* this connection has also has been advised to the delegate data object */
564 #define WINE_ADVF_REMOTE 0x80000000
565
566 /******************************************************************************
567  * DataAdviseHolder_Destructor
568  */
569 static void DataAdviseHolder_Destructor(DataAdviseHolder* ptrToDestroy)
570 {
571   DWORD index;
572   TRACE("%p\n", ptrToDestroy);
573
574   for (index = 0; index < ptrToDestroy->maxCons; index++)
575   {
576     if (ptrToDestroy->Connections[index].sink != NULL)
577     {
578       if (ptrToDestroy->delegate && 
579           (ptrToDestroy->Connections[index].advf & WINE_ADVF_REMOTE))
580         IDataObject_DUnadvise(ptrToDestroy->delegate,
581           ptrToDestroy->Connections[index].remote_connection);
582
583       IAdviseSink_Release(ptrToDestroy->Connections[index].sink);
584       ptrToDestroy->Connections[index].sink = NULL;
585     }
586   }
587
588   HeapFree(GetProcessHeap(), 0, ptrToDestroy->Connections);
589   HeapFree(GetProcessHeap(), 0, ptrToDestroy);
590 }
591
592 /************************************************************************
593  * DataAdviseHolder_QueryInterface (IUnknown)
594  *
595  * See Windows documentation for more details on IUnknown methods.
596  */
597 static HRESULT WINAPI DataAdviseHolder_QueryInterface(
598   IDataAdviseHolder*      iface,
599   REFIID                  riid,
600   void**                  ppvObject)
601 {
602   DataAdviseHolder *This = (DataAdviseHolder *)iface;
603   TRACE("(%p)->(%s,%p)\n",This,debugstr_guid(riid),ppvObject);
604   /*
605    * Perform a sanity check on the parameters.
606    */
607   if ( (This==0) || (ppvObject==0) )
608     return E_INVALIDARG;
609
610   /*
611    * Initialize the return parameter.
612    */
613   *ppvObject = 0;
614
615   /*
616    * Compare the riid with the interface IDs implemented by this object.
617    */
618   if ( (memcmp(&IID_IUnknown, riid, sizeof(IID_IUnknown)) == 0) ||
619        (memcmp(&IID_IDataAdviseHolder, riid, sizeof(IID_IDataAdviseHolder)) == 0)  )
620   {
621     *ppvObject = iface;
622   }
623
624   /*
625    * Check that we obtained an interface.
626    */
627   if ((*ppvObject)==0)
628   {
629     return E_NOINTERFACE;
630   }
631
632   /*
633    * Query Interface always increases the reference count by one when it is
634    * successful.
635    */
636   IUnknown_AddRef((IUnknown*)*ppvObject);
637
638   return S_OK;
639 }
640
641 /************************************************************************
642  * DataAdviseHolder_AddRef (IUnknown)
643  *
644  * See Windows documentation for more details on IUnknown methods.
645  */
646 static ULONG WINAPI       DataAdviseHolder_AddRef(
647   IDataAdviseHolder*      iface)
648 {
649   DataAdviseHolder *This = (DataAdviseHolder *)iface;
650   TRACE("(%p) (ref=%d)\n", This, This->ref);
651   return InterlockedIncrement(&This->ref);
652 }
653
654 /************************************************************************
655  * DataAdviseHolder_Release (IUnknown)
656  *
657  * See Windows documentation for more details on IUnknown methods.
658  */
659 static ULONG WINAPI DataAdviseHolder_Release(
660   IDataAdviseHolder*      iface)
661 {
662   DataAdviseHolder *This = (DataAdviseHolder *)iface;
663   ULONG ref;
664   TRACE("(%p) (ref=%d)\n", This, This->ref);
665
666   /*
667    * Decrease the reference count on this object.
668    */
669   ref = InterlockedDecrement(&This->ref);
670
671   /*
672    * If the reference count goes down to 0, perform suicide.
673    */
674   if (ref==0) DataAdviseHolder_Destructor(This);
675
676   return ref;
677 }
678
679 /************************************************************************
680  * DataAdviseHolder_Advise
681  *
682  */
683 static HRESULT WINAPI DataAdviseHolder_Advise(
684   IDataAdviseHolder*      iface,
685   IDataObject*            pDataObject,
686   FORMATETC*              pFetc,
687   DWORD                   advf,
688   IAdviseSink*            pAdvise,
689   DWORD*                  pdwConnection)
690 {
691   DWORD index;
692
693   DataAdviseHolder *This = (DataAdviseHolder *)iface;
694
695   TRACE("(%p)->(%p, %p, %08x, %p, %p)\n", This, pDataObject, pFetc, advf,
696         pAdvise, pdwConnection);
697   /*
698    * Sanity check
699    */
700   if (pdwConnection==NULL)
701     return E_POINTER;
702
703   *pdwConnection = 0;
704
705   /*
706    * Find a free spot in the array.
707    */
708   for (index = 0; index < This->maxCons; index++)
709   {
710     if (This->Connections[index].sink == NULL)
711       break;
712   }
713
714   /*
715    * If the array is full, we need to grow it.
716    */
717   if (index == This->maxCons)
718   {
719     This->maxCons+=INITIAL_SINKS;
720     This->Connections = HeapReAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY,
721                                     This->Connections,
722                                     This->maxCons*sizeof(DataAdviseConnection));
723   }
724   /*
725    * Store the new sink
726    */
727   This->Connections[index].sink = pAdvise;
728   This->Connections[index].advf = advf & ~WINE_ADVF_REMOTE;
729   This->Connections[index].fmat = *pFetc;
730   if (pFetc->ptd)
731   {
732     This->Connections[index].fmat.ptd = CoTaskMemAlloc(pFetc->ptd->tdSize);
733     if (!This->Connections[index].fmat.ptd)
734     {
735       IDataAdviseHolder_Unadvise(iface, index + 1);
736       return E_OUTOFMEMORY;
737     }
738     memcpy(This->Connections[index].fmat.ptd, pFetc->ptd, pFetc->ptd->tdSize);
739   }
740
741   if (This->Connections[index].sink != NULL) {
742     IAdviseSink_AddRef(This->Connections[index].sink);
743
744     /* if we are already connected advise the remote object */
745     if (This->delegate)
746     {
747         HRESULT hr;
748
749         hr = IDataObject_DAdvise(This->delegate, &This->Connections[index].fmat,
750                                  This->Connections[index].advf,
751                                  This->Connections[index].sink,
752                                  &This->Connections[index].remote_connection);
753         if (FAILED(hr))
754         {
755             IDataAdviseHolder_Unadvise(iface, index + 1);
756             return hr;
757         }
758         This->Connections[index].advf |= WINE_ADVF_REMOTE;
759     }
760     else if(advf & ADVF_PRIMEFIRST)
761       /* only do this if we have no delegate, since in the above case the
762        * delegate will do the priming for us */
763       IDataAdviseHolder_SendOnDataChange(iface, pDataObject, 0, advf);
764   }
765   /*
766    * Return the index as the cookie.
767    * Since 0 is not a valid cookie, we will increment by
768    * 1 the index in the table.
769    */
770   *pdwConnection = index+1;
771
772   return S_OK;
773 }
774
775 /******************************************************************************
776  * DataAdviseHolder_Unadvise
777  */
778 static HRESULT WINAPI     DataAdviseHolder_Unadvise(
779   IDataAdviseHolder*      iface,
780   DWORD                   dwConnection)
781 {
782   DataAdviseHolder *This = (DataAdviseHolder *)iface;
783
784   TRACE("(%p)->(%u)\n", This, dwConnection);
785
786   /*
787    * So we don't return 0 as a cookie, the index was
788    * incremented by 1 in OleAdviseHolderImpl_Advise
789    * we have to compensate.
790    */
791   dwConnection--;
792
793   /*
794    * Check for invalid cookies.
795    */
796   if (dwConnection >= This->maxCons)
797     return OLE_E_NOCONNECTION;
798
799   if (This->Connections[dwConnection].sink == NULL)
800     return OLE_E_NOCONNECTION;
801
802   if (This->delegate && This->Connections[dwConnection].advf & WINE_ADVF_REMOTE)
803     IDataObject_DUnadvise(This->delegate,
804       This->Connections[dwConnection].remote_connection);
805
806   /*
807    * Release the sink and mark the spot in the list as free.
808    */
809   IAdviseSink_Release(This->Connections[dwConnection].sink);
810   memset(&(This->Connections[dwConnection]), 0, sizeof(DataAdviseConnection));
811
812   return S_OK;
813 }
814
815 static HRESULT WINAPI     DataAdviseHolder_EnumAdvise(
816   IDataAdviseHolder*      iface,
817   IEnumSTATDATA**         ppenumAdvise)
818 {
819   DataAdviseHolder *This = (DataAdviseHolder *)iface;
820
821   FIXME("(%p)->(%p)\n", This, ppenumAdvise);
822   return E_NOTIMPL;
823 }
824
825 /******************************************************************************
826  * DataAdviseHolder_SendOnDataChange
827  */
828 static HRESULT WINAPI     DataAdviseHolder_SendOnDataChange(
829   IDataAdviseHolder*      iface,
830   IDataObject*            pDataObject,
831   DWORD                   dwReserved,
832   DWORD                   advf)
833 {
834   DataAdviseHolder *This = (DataAdviseHolder *)iface;
835   DWORD index;
836   STGMEDIUM stg;
837   HRESULT res;
838
839   TRACE("(%p)->(%p,%08x,%08x)\n", This, pDataObject, dwReserved, advf);
840
841   for(index = 0; index < This->maxCons; index++) {
842     if(This->Connections[index].sink != NULL) {
843       memset(&stg, 0, sizeof(stg));
844       if(!(This->Connections[index].advf & ADVF_NODATA)) {
845         TRACE("Calling IDataObject_GetData\n");
846         res = IDataObject_GetData(pDataObject,
847                                   &(This->Connections[index].fmat),
848                                   &stg);
849         TRACE("returns %08x\n", res);
850       }
851       TRACE("Calling IAdviseSink_OnDataChange\n");
852       IAdviseSink_OnDataChange(This->Connections[index].sink,
853                                      &(This->Connections[index].fmat),
854                                      &stg);
855       TRACE("Done IAdviseSink_OnDataChange\n");
856       if(This->Connections[index].advf & ADVF_ONLYONCE) {
857         TRACE("Removing connection\n");
858         DataAdviseHolder_Unadvise(iface, index+1);
859       }
860     }
861   }
862   return S_OK;
863 }
864
865 /**************************************************************************
866  *  DataAdviseHolderImpl_VTable
867  */
868 static const IDataAdviseHolderVtbl DataAdviseHolderImpl_VTable =
869 {
870   DataAdviseHolder_QueryInterface,
871   DataAdviseHolder_AddRef,
872   DataAdviseHolder_Release,
873   DataAdviseHolder_Advise,
874   DataAdviseHolder_Unadvise,
875   DataAdviseHolder_EnumAdvise,
876   DataAdviseHolder_SendOnDataChange
877 };
878
879 HRESULT DataAdviseHolder_OnConnect(IDataAdviseHolder *iface, IDataObject *pDelegate)
880 {
881   DataAdviseHolder *This = (DataAdviseHolder *)iface;
882   DWORD index;
883   HRESULT hr = S_OK;
884
885   for(index = 0; index < This->maxCons; index++)
886   {
887     if(This->Connections[index].sink != NULL)
888     {
889       hr = IDataObject_DAdvise(pDelegate, &This->Connections[index].fmat,
890                                This->Connections[index].advf,
891                                This->Connections[index].sink,
892                                &This->Connections[index].remote_connection);
893       if (FAILED(hr)) break;
894       This->Connections[index].advf |= WINE_ADVF_REMOTE;
895     }
896   }
897   This->delegate = pDelegate;
898   return hr;
899 }
900
901 void DataAdviseHolder_OnDisconnect(IDataAdviseHolder *iface)
902 {
903   DataAdviseHolder *This = (DataAdviseHolder *)iface;
904   DWORD index;
905
906   for(index = 0; index < This->maxCons; index++)
907   {
908     if((This->Connections[index].sink != NULL) &&
909        (This->Connections[index].advf & WINE_ADVF_REMOTE))
910     {
911       IDataObject_DUnadvise(This->delegate,
912         This->Connections[index].remote_connection);
913       This->Connections[index].advf &= ~WINE_ADVF_REMOTE;
914     }
915   }
916   This->delegate = NULL;
917 }
918
919 /******************************************************************************
920  * DataAdviseHolder_Constructor
921  */
922 static IDataAdviseHolder* DataAdviseHolder_Constructor(void)
923 {
924   DataAdviseHolder* newHolder;
925
926   newHolder = HeapAlloc(GetProcessHeap(), 0, sizeof(DataAdviseHolder));
927
928   newHolder->lpVtbl = &DataAdviseHolderImpl_VTable;
929   newHolder->ref = 1;
930   newHolder->maxCons = INITIAL_SINKS;
931   newHolder->Connections = HeapAlloc(GetProcessHeap(),
932                                      HEAP_ZERO_MEMORY,
933                                      newHolder->maxCons *
934                                      sizeof(DataAdviseConnection));
935   newHolder->delegate = NULL;
936
937   TRACE("returning %p\n", newHolder);
938   return (IDataAdviseHolder*)newHolder;
939 }
940
941 /***********************************************************************
942  * API functions
943  */
944
945 /***********************************************************************
946  * CreateOleAdviseHolder [OLE32.@]
947  */
948 HRESULT WINAPI CreateOleAdviseHolder(
949   LPOLEADVISEHOLDER *ppOAHolder)
950 {
951   TRACE("(%p)\n", ppOAHolder);
952
953   /*
954    * Sanity check,
955    */
956   if (ppOAHolder==NULL)
957     return E_POINTER;
958
959   *ppOAHolder = OleAdviseHolderImpl_Constructor ();
960
961   if (*ppOAHolder != NULL)
962     return S_OK;
963
964   return E_OUTOFMEMORY;
965 }
966
967 /******************************************************************************
968  *              CreateDataAdviseHolder        [OLE32.@]
969  */
970 HRESULT WINAPI CreateDataAdviseHolder(
971   LPDATAADVISEHOLDER* ppDAHolder)
972 {
973   TRACE("(%p)\n", ppDAHolder);
974
975   /*
976    * Sanity check,
977    */
978   if (ppDAHolder==NULL)
979     return E_POINTER;
980
981   *ppDAHolder = DataAdviseHolder_Constructor();
982
983   if (*ppDAHolder != NULL)
984     return S_OK;
985
986   return E_OUTOFMEMORY;
987 }