Correct crypt context init in CryptProtectData/CryptUnprotectData.
[wine] / dlls / ole32 / stubmanager.c
1 /*
2  * A stub manager is an object that controls interface stubs. It is
3  * identified by an OID (object identifier) and acts as the network
4  * identity of the object. There can be many stub managers in a
5  * process or apartment.
6  *
7  * Copyright 2002 Marcus Meissner
8  * Copyright 2004 Mike Hearn for CodeWeavers
9  * Copyright 2004 Robert Shearman (for CodeWeavers)
10  *
11  * This library is free software; you can redistribute it and/or
12  * modify it under the terms of the GNU Lesser General Public
13  * License as published by the Free Software Foundation; either
14  * version 2.1 of the License, or (at your option) any later version.
15  *
16  * This library is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
19  * Lesser General Public License for more details.
20  *
21  * You should have received a copy of the GNU Lesser General Public
22  * License along with this library; if not, write to the Free Software
23  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
24  */
25
26 #define COBJMACROS
27 #define NONAMELESSUNION
28 #define NONAMELESSSTRUCT
29
30 #include <assert.h>
31 #include <stdarg.h>
32 #include <limits.h>
33
34 #include "windef.h"
35 #include "winbase.h"
36 #include "winuser.h"
37 #include "objbase.h"
38 #include "ole2.h"
39 #include "ole2ver.h"
40 #include "rpc.h"
41 #include "wine/debug.h"
42 #include "compobj_private.h"
43
44 WINE_DEFAULT_DEBUG_CHANNEL(ole);
45
46 static void stub_manager_delete_ifstub(struct stub_manager *m, struct ifstub *ifstub);
47 static struct ifstub *stub_manager_ipid_to_ifstub(struct stub_manager *m, const IPID *ipid);
48
49 /* creates a new stub manager and adds it into the apartment. caller must
50  * release stub manager when it is no longer required. the apartment and
51  * external refs together take one implicit ref */
52 struct stub_manager *new_stub_manager(APARTMENT *apt, IUnknown *object, MSHLFLAGS mshlflags)
53 {
54     struct stub_manager *sm;
55
56     assert( apt );
57     
58     sm = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(struct stub_manager));
59     if (!sm) return NULL;
60
61     list_init(&sm->ifstubs);
62
63     InitializeCriticalSection(&sm->lock);
64     DEBUG_SET_CRITSEC_NAME(&sm->lock, "stub_manager");
65
66     IUnknown_AddRef(object);
67     sm->object = object;
68     sm->apt    = apt;
69
70     /* start off with 2 references because the stub is in the apartment
71      * and the caller will also hold a reference */
72     sm->refs   = 2;
73
74     /* yes, that's right, this starts at zero. that's zero EXTERNAL
75      * refs, ie nobody has unmarshalled anything yet. we can't have
76      * negative refs because the stub manager cannot be explicitly
77      * killed, it has to die by somebody unmarshalling then releasing
78      * the marshalled ifptr.
79      */
80     sm->extrefs = 0;
81
82     if (mshlflags & MSHLFLAGS_TABLESTRONG)
83         sm->state = STUBSTATE_TABLE_STRONG;
84     else if (mshlflags & MSHLFLAGS_TABLEWEAK)
85         sm->state = STUBSTATE_TABLE_WEAK_UNMARSHALED;
86     else
87         sm->state = STUBSTATE_NORMAL_MARSHALED;
88     
89     EnterCriticalSection(&apt->cs);
90     sm->oid    = apt->oidc++;
91     list_add_head(&apt->stubmgrs, &sm->entry);
92     LeaveCriticalSection(&apt->cs);
93
94     TRACE("Created new stub manager (oid=%s) at %p for object with IUnknown %p\n", wine_dbgstr_longlong(sm->oid), sm, object);
95     
96     return sm;
97 }
98
99 /* m->apt->cs must be held on entry to this function */
100 static void stub_manager_delete(struct stub_manager *m)
101 {
102     struct list *cursor;
103
104     TRACE("destroying %p (oid=%s)\n", m, wine_dbgstr_longlong(m->oid));
105
106     list_remove(&m->entry);
107
108     /* release every ifstub */
109     while ((cursor = list_head(&m->ifstubs)))
110     {
111         struct ifstub *ifstub = LIST_ENTRY(cursor, struct ifstub, entry);
112         stub_manager_delete_ifstub(m, ifstub);
113     }
114
115     IUnknown_Release(m->object);
116
117     DEBUG_CLEAR_CRITSEC_NAME(&m->lock);
118     DeleteCriticalSection(&m->lock);
119
120     HeapFree(GetProcessHeap(), 0, m);
121 }
122
123 /* gets the stub manager associated with an object - caller must have
124  * a reference to the apartment while a reference to the stub manager is held.
125  * it must also call release on the stub manager when it is no longer needed */
126 struct stub_manager *get_stub_manager_from_object(APARTMENT *apt, void *object)
127 {
128     struct stub_manager *result = NULL;
129     struct list         *cursor;
130
131     EnterCriticalSection(&apt->cs);
132     LIST_FOR_EACH( cursor, &apt->stubmgrs )
133     {
134         struct stub_manager *m = LIST_ENTRY( cursor, struct stub_manager, entry );
135
136         if (m->object == object)
137         {
138             result = m;
139             stub_manager_int_addref(result);
140             break;
141         }
142     }
143     LeaveCriticalSection(&apt->cs);
144
145     if (result)
146         TRACE("found %p for object %p\n", result, object);
147     else
148         TRACE("not found for object %p\n", object);
149
150     return result;    
151 }
152
153 /* removes the apartment reference to an object, destroying it when no other
154  * threads have a reference to it */
155 void apartment_disconnectobject(struct apartment *apt, void *object)
156 {
157     int found = FALSE;
158     struct stub_manager *stubmgr;
159
160     EnterCriticalSection(&apt->cs);
161     LIST_FOR_EACH_ENTRY( stubmgr, &apt->stubmgrs, struct stub_manager, entry )
162     {
163         if (stubmgr->object == object)
164         {
165             found = TRUE;
166             stub_manager_int_release(stubmgr);
167             break;
168         }
169     }
170     LeaveCriticalSection(&apt->cs);
171
172     if (found)
173         TRACE("disconnect object %p\n", object);
174     else
175         WARN("couldn't find object %p\n", object);
176 }
177
178 /* gets the stub manager associated with an object id - caller must have
179  * a reference to the apartment while a reference to the stub manager is held.
180  * it must also call release on the stub manager when it is no longer needed */
181 struct stub_manager *get_stub_manager(APARTMENT *apt, OID oid)
182 {
183     struct stub_manager *result = NULL;
184     struct list         *cursor;
185
186     EnterCriticalSection(&apt->cs);
187     LIST_FOR_EACH( cursor, &apt->stubmgrs )
188     {
189         struct stub_manager *m = LIST_ENTRY( cursor, struct stub_manager, entry );
190
191         if (m->oid == oid)
192         {
193             result = m;
194             stub_manager_int_addref(result);
195             break;
196         }
197     }
198     LeaveCriticalSection(&apt->cs);
199
200     if (result)
201         TRACE("found %p for oid %s\n", result, wine_dbgstr_longlong(oid));
202     else
203         TRACE("not found for oid %s\n", wine_dbgstr_longlong(oid));
204
205     return result;
206 }
207
208 /* increments the internal refcount */
209 ULONG stub_manager_int_addref(struct stub_manager *This)
210 {
211     ULONG refs;
212
213     EnterCriticalSection(&This->apt->cs);
214     refs = ++This->refs;
215     LeaveCriticalSection(&This->apt->cs);
216
217     TRACE("before %ld\n", refs - 1);
218
219     return refs;
220 }
221
222 /* decrements the internal refcount */
223 ULONG stub_manager_int_release(struct stub_manager *This)
224 {
225     ULONG refs;
226     APARTMENT *apt = This->apt;
227
228     EnterCriticalSection(&apt->cs);
229     refs = --This->refs;
230
231     TRACE("after %ld\n", refs);
232
233     if (!refs)
234         stub_manager_delete(This);
235     LeaveCriticalSection(&apt->cs);
236
237     return refs;
238 }
239
240 /* add some external references (ie from a client that unmarshaled an ifptr) */
241 ULONG stub_manager_ext_addref(struct stub_manager *m, ULONG refs)
242 {
243     ULONG rc;
244
245     EnterCriticalSection(&m->lock);
246     
247     /* make sure we don't overflow extrefs */
248     refs = min(refs, (ULONG_MAX-1 - m->extrefs));
249     rc = (m->extrefs += refs);
250
251     LeaveCriticalSection(&m->lock);
252     
253     TRACE("added %lu refs to %p (oid %s), rc is now %lu\n", refs, m, wine_dbgstr_longlong(m->oid), rc);
254
255     return rc;
256 }
257
258 /* remove some external references */
259 ULONG stub_manager_ext_release(struct stub_manager *m, ULONG refs)
260 {
261     ULONG rc;
262
263     EnterCriticalSection(&m->lock);
264
265     /* make sure we don't underflow extrefs */
266     refs = min(refs, m->extrefs);
267     rc = (m->extrefs -= refs);
268
269     LeaveCriticalSection(&m->lock);
270     
271     TRACE("removed %lu refs from %p (oid %s), rc is now %lu\n", refs, m, wine_dbgstr_longlong(m->oid), rc);
272
273     if (rc == 0)
274         stub_manager_int_release(m);
275
276     return rc;
277 }
278
279 static struct ifstub *stub_manager_ipid_to_ifstub(struct stub_manager *m, const IPID *ipid)
280 {
281     struct list    *cursor;
282     struct ifstub  *result = NULL;
283     
284     EnterCriticalSection(&m->lock);
285     LIST_FOR_EACH( cursor, &m->ifstubs )
286     {
287         struct ifstub *ifstub = LIST_ENTRY( cursor, struct ifstub, entry );
288
289         if (IsEqualGUID(ipid, &ifstub->ipid))
290         {
291             result = ifstub;
292             break;
293         }
294     }
295     LeaveCriticalSection(&m->lock);
296
297     return result;
298 }
299
300 /* gets the stub manager associated with an ipid - caller must have
301  * a reference to the apartment while a reference to the stub manager is held.
302  * it must also call release on the stub manager when it is no longer needed */
303 static struct stub_manager *get_stub_manager_from_ipid(APARTMENT *apt, const IPID *ipid)
304 {
305     struct stub_manager *result = NULL;
306     struct list         *cursor;
307
308     EnterCriticalSection(&apt->cs);
309     LIST_FOR_EACH( cursor, &apt->stubmgrs )
310     {
311         struct stub_manager *m = LIST_ENTRY( cursor, struct stub_manager, entry );
312
313         if (stub_manager_ipid_to_ifstub(m, ipid))
314         {
315             result = m;
316             stub_manager_int_addref(result);
317             break;
318         }
319     }
320     LeaveCriticalSection(&apt->cs);
321
322     if (result)
323         TRACE("found %p for ipid %s\n", result, debugstr_guid(ipid));
324     else
325         ERR("not found for ipid %s\n", debugstr_guid(ipid));
326
327     return result;
328 }
329
330 HRESULT ipid_to_stub_manager(const IPID *ipid, APARTMENT **stub_apt, struct stub_manager **stubmgr_ret)
331 {
332     /* FIXME: hack for IRemUnknown */
333     if (ipid->Data2 == 0xffff)
334         *stub_apt = apartment_findfromoxid(*(OXID *)ipid->Data4, TRUE);
335     else
336         *stub_apt = apartment_findfromtid(ipid->Data2);
337     if (!*stub_apt)
338     {
339         ERR("Couldn't find apartment corresponding to TID 0x%04x\n", ipid->Data2);
340         return RPC_E_INVALID_OBJECT;
341     }
342     *stubmgr_ret = get_stub_manager_from_ipid(*stub_apt, ipid);
343     if (!*stubmgr_ret)
344     {
345         apartment_release(*stub_apt);
346         *stub_apt = NULL;
347         return RPC_E_INVALID_OBJECT;
348     }
349     return S_OK;
350 }
351
352 /* gets the apartment and IRpcStubBuffer from an object. the caller must
353  * release the references to both objects */
354 IRpcStubBuffer *ipid_to_apt_and_stubbuffer(const IPID *ipid, APARTMENT **stub_apt)
355 {
356     IRpcStubBuffer *ret = NULL;
357     struct stub_manager *stubmgr;
358     struct ifstub *ifstub;
359     HRESULT hr;
360
361     *stub_apt = NULL;
362
363     hr = ipid_to_stub_manager(ipid, stub_apt, &stubmgr);
364     if (hr != S_OK) return NULL;
365
366     ifstub = stub_manager_ipid_to_ifstub(stubmgr, ipid);
367     if (ifstub)
368         ret = ifstub->stubbuffer;
369
370     if (ret) IRpcStubBuffer_AddRef(ret);
371
372     stub_manager_int_release(stubmgr);
373
374     return ret;
375 }
376
377 /* generates an ipid in the following format (similar to native version):
378  * Data1 = apartment-local ipid counter
379  * Data2 = apartment creator thread ID
380  * Data3 = process ID
381  * Data4 = random value
382  */
383 static inline HRESULT generate_ipid(struct stub_manager *m, IPID *ipid)
384 {
385     HRESULT hr;
386     hr = UuidCreate(ipid);
387     if (FAILED(hr))
388     {
389         ERR("couldn't create IPID for stub manager %p\n", m);
390         UuidCreateNil(ipid);
391         return hr;
392     }
393
394     ipid->Data1 = InterlockedIncrement(&m->apt->ipidc);
395     ipid->Data2 = (USHORT)m->apt->tid;
396     ipid->Data3 = (USHORT)GetCurrentProcessId();
397     return S_OK;
398 }
399
400 /* registers a new interface stub COM object with the stub manager and returns registration record */
401 struct ifstub *stub_manager_new_ifstub(struct stub_manager *m, IRpcStubBuffer *sb, IUnknown *iptr, REFIID iid)
402 {
403     struct ifstub *stub;
404
405     TRACE("oid=%s, stubbuffer=%p, iptr=%p, iid=%s\n",
406           wine_dbgstr_longlong(m->oid), sb, iptr, debugstr_guid(iid));
407
408     stub = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(struct ifstub));
409     if (!stub) return NULL;
410
411     stub->stubbuffer = sb;
412     IUnknown_AddRef(sb);
413
414     /* no need to ref this, same object as sb */
415     stub->iface = iptr;
416
417     stub->iid = *iid;
418
419     /* FIXME: hack for IRemUnknown because we don't notify SCM of our IPID
420      * yet, so we need to use a well-known one */
421     if (IsEqualIID(iid, &IID_IRemUnknown))
422     {
423         stub->ipid.Data1 = 0xffffffff;
424         stub->ipid.Data2 = 0xffff;
425         stub->ipid.Data3 = 0xffff;
426         assert(sizeof(stub->ipid.Data4) == sizeof(m->apt->oxid));
427         memcpy(&stub->ipid.Data4, &m->apt->oxid, sizeof(OXID));
428     }
429     else
430         generate_ipid(m, &stub->ipid);
431
432     EnterCriticalSection(&m->lock);
433     list_add_head(&m->ifstubs, &stub->entry);
434     LeaveCriticalSection(&m->lock);
435
436     TRACE("ifstub %p created with ipid %s\n", stub, debugstr_guid(&stub->ipid));
437
438     return stub;
439 }
440
441 static void stub_manager_delete_ifstub(struct stub_manager *m, struct ifstub *ifstub)
442 {
443     TRACE("m=%p, m->oid=%s, ipid=%s\n", m, wine_dbgstr_longlong(m->oid), debugstr_guid(&ifstub->ipid));
444
445     list_remove(&ifstub->entry);
446
447     RPC_UnregisterInterface(&ifstub->iid);
448         
449     IUnknown_Release(ifstub->stubbuffer);
450     IUnknown_Release(ifstub->iface);
451
452     HeapFree(GetProcessHeap(), 0, ifstub);
453 }
454
455 /* returns TRUE if it is possible to unmarshal, FALSE otherwise. */
456 BOOL stub_manager_notify_unmarshal(struct stub_manager *m)
457 {
458     BOOL ret;
459
460     EnterCriticalSection(&m->lock);
461
462     switch (m->state)
463     {
464     case STUBSTATE_TABLE_STRONG:
465     case STUBSTATE_TABLE_WEAK_MARSHALED:
466         /* no transition */
467         ret = TRUE;
468         break;
469     case STUBSTATE_TABLE_WEAK_UNMARSHALED:
470         m->state = STUBSTATE_TABLE_WEAK_MARSHALED;
471         ret = TRUE;
472         break;
473     case STUBSTATE_NORMAL_MARSHALED:
474         m->state = STUBSTATE_NORMAL_UNMARSHALED;
475         ret = TRUE;
476         break;
477     default:
478         WARN("object OID %s already unmarshaled\n",
479             wine_dbgstr_longlong(m->oid));
480         ret = FALSE;
481         break;
482     }
483
484     LeaveCriticalSection(&m->lock);
485
486     return ret;
487 }
488
489 void stub_manager_release_marshal_data(struct stub_manager *m, ULONG refs)
490 {
491     EnterCriticalSection(&m->lock);
492
493     switch (m->state)
494     {
495     case STUBSTATE_NORMAL_MARSHALED:
496     case STUBSTATE_NORMAL_UNMARSHALED: /* FIXME: check this */
497         /* nothing to change */
498         break;
499     case STUBSTATE_TABLE_WEAK_UNMARSHALED:
500     case STUBSTATE_TABLE_STRONG:
501         refs = 1;
502         break;
503     case STUBSTATE_TABLE_WEAK_MARSHALED:
504         refs = 0; /* like native */
505         break;
506     }
507
508     stub_manager_ext_release(m, refs);
509
510     LeaveCriticalSection(&m->lock);
511 }
512
513 /* is an ifstub table marshaled? */
514 BOOL stub_manager_is_table_marshaled(struct stub_manager *m)
515 {
516     BOOL ret;
517
518     EnterCriticalSection(&m->lock);
519     ret = ((m->state == STUBSTATE_TABLE_STRONG) ||
520            (m->state == STUBSTATE_TABLE_WEAK_MARSHALED) ||
521            (m->state == STUBSTATE_TABLE_WEAK_UNMARSHALED));
522     LeaveCriticalSection(&m->lock);
523
524     return ret;
525 }
526
527
528 /*****************************************************************************
529  *
530  * IRemUnknown implementation
531  *
532  *
533  * Note: this object is not related to the lifetime of a stub_manager, but it
534  * interacts with stub managers.
535  */
536
537 const IID IID_IRemUnknown = { 0x00000131, 0, 0, {0xc0, 0, 0, 0, 0, 0, 0, 0x46} };
538
539 typedef struct rem_unknown
540 {
541     const IRemUnknownVtbl *lpVtbl;
542     ULONG refs;
543 } RemUnknown;
544
545 static const IRemUnknownVtbl RemUnknown_Vtbl;
546
547
548 /* construct an IRemUnknown object with one outstanding reference */
549 static HRESULT RemUnknown_Construct(IRemUnknown **ppRemUnknown)
550 {
551     RemUnknown *This = HeapAlloc(GetProcessHeap(), 0, sizeof(*This));
552
553     if (!This) return E_OUTOFMEMORY;
554
555     This->lpVtbl = &RemUnknown_Vtbl;
556     This->refs = 1;
557
558     *ppRemUnknown = (IRemUnknown *)This;
559     return S_OK;
560 }
561
562 static HRESULT WINAPI RemUnknown_QueryInterface(IRemUnknown *iface, REFIID riid, void **ppv)
563 {
564     TRACE("(%p)->(%s, %p)\n", iface, debugstr_guid(riid), ppv);
565
566     if (IsEqualIID(riid, &IID_IUnknown) ||
567         IsEqualIID(riid, &IID_IRemUnknown))
568     {
569         *ppv = (LPVOID)iface;
570         IRemUnknown_AddRef(iface);
571         return S_OK;
572     }
573
574     FIXME("No interface for iid %s\n", debugstr_guid(riid));
575
576     *ppv = NULL;
577     return E_NOINTERFACE;
578 }
579
580 static ULONG WINAPI RemUnknown_AddRef(IRemUnknown *iface)
581 {
582     ULONG refs;
583     RemUnknown *This = (RemUnknown *)iface;
584
585     refs = InterlockedIncrement(&This->refs);
586
587     TRACE("%p before: %ld\n", iface, refs-1);
588     return refs;
589 }
590
591 static ULONG WINAPI RemUnknown_Release(IRemUnknown *iface)
592 {
593     ULONG refs;
594     RemUnknown *This = (RemUnknown *)iface;
595
596     refs = InterlockedDecrement(&This->refs);
597     if (!refs)
598         HeapFree(GetProcessHeap(), 0, This);
599
600     TRACE("%p after: %ld\n", iface, refs);
601     return refs;
602 }
603
604 static HRESULT WINAPI RemUnknown_RemQueryInterface(IRemUnknown *iface,
605     REFIPID ripid, ULONG cRefs, USHORT cIids, IID *iids /* [size_is(cIids)] */,
606     REMQIRESULT **ppQIResults /* [size_is(,cIids)] */)
607 {
608     HRESULT hr;
609     USHORT i;
610     USHORT successful_qis = 0;
611     APARTMENT *apt;
612     struct stub_manager *stubmgr;
613
614     TRACE("(%p)->(%s, %ld, %d, %p, %p)\n", iface, debugstr_guid(ripid), cRefs, cIids, iids, ppQIResults);
615
616     hr = ipid_to_stub_manager(ripid, &apt, &stubmgr);
617     if (hr != S_OK) return hr;
618
619     *ppQIResults = CoTaskMemAlloc(sizeof(REMQIRESULT) * cIids);
620
621     for (i = 0; i < cIids; i++)
622     {
623         HRESULT hrobj = marshal_object(apt, &(*ppQIResults)[i].std, &iids[i],
624                                        stubmgr->object, MSHLFLAGS_NORMAL);
625         if (hrobj == S_OK)
626             successful_qis++;
627         (*ppQIResults)[i].hResult = hrobj;
628     }
629
630     stub_manager_int_release(stubmgr);
631     apartment_release(apt);
632
633     if (successful_qis == cIids)
634         return S_OK; /* we got all requested interfaces */
635     else if (successful_qis == 0)
636         return E_NOINTERFACE; /* we didn't get any interfaces */
637     else
638         return S_FALSE; /* we got some interfaces */
639 }
640
641 static HRESULT WINAPI RemUnknown_RemAddRef(IRemUnknown *iface,
642     USHORT cInterfaceRefs,
643     REMINTERFACEREF* InterfaceRefs /* [size_is(cInterfaceRefs)] */,
644     HRESULT *pResults /* [size_is(cInterfaceRefs)] */)
645 {
646     HRESULT hr = S_OK;
647     USHORT i;
648
649     TRACE("(%p)->(%d, %p, %p)\n", iface, cInterfaceRefs, InterfaceRefs, pResults);
650
651     for (i = 0; i < cInterfaceRefs; i++)
652     {
653         APARTMENT *apt;
654         struct stub_manager *stubmgr;
655
656         pResults[i] = ipid_to_stub_manager(&InterfaceRefs[i].ipid, &apt, &stubmgr);
657         if (pResults[i] != S_OK)
658         {
659             hr = S_FALSE;
660             continue;
661         }
662
663         stub_manager_ext_addref(stubmgr, InterfaceRefs[i].cPublicRefs);
664         if (InterfaceRefs[i].cPrivateRefs)
665             FIXME("Adding %ld refs securely not implemented\n", InterfaceRefs[i].cPrivateRefs);
666
667         stub_manager_int_release(stubmgr);
668         apartment_release(apt);
669     }
670
671     return hr;
672 }
673
674 static HRESULT WINAPI RemUnknown_RemRelease(IRemUnknown *iface,
675     USHORT cInterfaceRefs,
676     REMINTERFACEREF* InterfaceRefs /* [size_is(cInterfaceRefs)] */)
677 {
678     HRESULT hr = S_OK;
679     USHORT i;
680
681     TRACE("(%p)->(%d, %p)\n", iface, cInterfaceRefs, InterfaceRefs);
682
683     for (i = 0; i < cInterfaceRefs; i++)
684     {
685         APARTMENT *apt;
686         struct stub_manager *stubmgr;
687
688         hr = ipid_to_stub_manager(&InterfaceRefs[i].ipid, &apt, &stubmgr);
689         if (hr != S_OK)
690         {
691             hr = E_INVALIDARG;
692             /* FIXME: we should undo any changes already made in this function */
693             break;
694         }
695
696         stub_manager_ext_release(stubmgr, InterfaceRefs[i].cPublicRefs);
697         if (InterfaceRefs[i].cPrivateRefs)
698             FIXME("Releasing %ld refs securely not implemented\n", InterfaceRefs[i].cPrivateRefs);
699
700         stub_manager_int_release(stubmgr);
701         apartment_release(apt);
702     }
703
704     return hr;
705 }
706
707 static const IRemUnknownVtbl RemUnknown_Vtbl =
708 {
709     RemUnknown_QueryInterface,
710     RemUnknown_AddRef,
711     RemUnknown_Release,
712     RemUnknown_RemQueryInterface,
713     RemUnknown_RemAddRef,
714     RemUnknown_RemRelease
715 };
716
717 /* starts the IRemUnknown listener for the current apartment */
718 HRESULT start_apartment_remote_unknown()
719 {
720     IRemUnknown *pRemUnknown;
721     HRESULT hr = S_OK;
722     APARTMENT *apt = COM_CurrentApt();
723
724     EnterCriticalSection(&apt->cs);
725     if (!apt->remunk_exported)
726     {
727         /* create the IRemUnknown object */
728         hr = RemUnknown_Construct(&pRemUnknown);
729         if (hr == S_OK)
730         {
731             STDOBJREF stdobjref; /* dummy - not used */
732             /* register it with the stub manager */
733             hr = marshal_object(apt, &stdobjref, &IID_IRemUnknown, (IUnknown *)pRemUnknown, MSHLFLAGS_NORMAL);
734             /* release our reference to the object as the stub manager will manage the life cycle for us */
735             IRemUnknown_Release(pRemUnknown);
736             if (hr == S_OK)
737                 apt->remunk_exported = TRUE;
738         }
739     }
740     LeaveCriticalSection(&apt->cs);
741     return hr;
742 }