/* * COM+ Services * * Copyright 2013 Alistair Leslie-Hughes * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #define COBJMACROS #include <stdarg.h> #include "windef.h" #include "winbase.h" #include "ole2.h" #include "rpcproxy.h" #include "comsvcs.h" #include "wine/heap.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(comsvcs); static HINSTANCE COMSVCS_hInstance; typedef struct dispensermanager { IDispenserManager IDispenserManager_iface; LONG ref; HANDLE mta_thread, mta_stop_event; } dispensermanager; typedef struct holder { IHolder IHolder_iface; LONG ref; IDispenserDriver *driver; } holder; static inline dispensermanager *impl_from_IDispenserManager(IDispenserManager *iface) { return CONTAINING_RECORD(iface, dispensermanager, IDispenserManager_iface); } static inline holder *impl_from_IHolder(IHolder *iface) { return CONTAINING_RECORD(iface, holder, IHolder_iface); } static HRESULT WINAPI holder_QueryInterface(IHolder *iface, REFIID riid, void **object) { holder *This = impl_from_IHolder(iface); TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), object); *object = NULL; if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IHolder)) { *object = &This->IHolder_iface; IUnknown_AddRef( (IUnknown*)*object); return S_OK; } WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),object); return E_NOINTERFACE; } static ULONG WINAPI holder_AddRef(IHolder *iface) { holder *This = impl_from_IHolder(iface); ULONG ref = InterlockedIncrement(&This->ref); TRACE("(%p)->(%d)\n", This, ref); return ref; } static ULONG WINAPI holder_Release(IHolder *iface) { holder *This = impl_from_IHolder(iface); ULONG ref = InterlockedDecrement(&This->ref); TRACE("(%p)->(%d)\n", This, ref); if (!ref) { heap_free(This); } return ref; } static HRESULT WINAPI holder_AllocResource(IHolder *iface, const RESTYPID typeid, RESID *resid) { holder *This = impl_from_IHolder(iface); HRESULT hr; TIMEINSECS secs; TRACE("(%p)->(%08lx, %p) stub\n", This, typeid, resid); hr = IDispenserDriver_CreateResource(This->driver, typeid, resid, &secs); TRACE("<- 0x%08x\n", hr); return hr; } static HRESULT WINAPI holder_FreeResource(IHolder *iface, const RESID resid) { holder *This = impl_from_IHolder(iface); HRESULT hr; TRACE("(%p)->(%08lx) stub\n", This, resid); hr = IDispenserDriver_DestroyResource(This->driver, resid); TRACE("<- 0x%08x\n", hr); return hr; } static HRESULT WINAPI holder_TrackResource(IHolder *iface, const RESID resid) { holder *This = impl_from_IHolder(iface); FIXME("(%p)->(%08lx) stub\n", This, resid); return E_NOTIMPL; } static HRESULT WINAPI holder_TrackResourceS(IHolder *iface, const SRESID resid) { holder *This = impl_from_IHolder(iface); FIXME("(%p)->(%s) stub\n", This, debugstr_w(resid)); return E_NOTIMPL; } static HRESULT WINAPI holder_UntrackResource(IHolder *iface, const RESID resid, const BOOL value) { holder *This = impl_from_IHolder(iface); FIXME("(%p)->(%08lx, %d) stub\n", This, resid, value); return E_NOTIMPL; } static HRESULT WINAPI holder_UntrackResourceS(IHolder *iface, const SRESID resid, const BOOL value) { holder *This = impl_from_IHolder(iface); FIXME("(%p)->(%s, %d) stub\n", This, debugstr_w(resid), value); return E_NOTIMPL; } static HRESULT WINAPI holder_Close(IHolder *iface) { holder *This = impl_from_IHolder(iface); FIXME("(%p) stub\n", This); IDispenserDriver_Release(This->driver); return S_OK; } static HRESULT WINAPI holder_RequestDestroyResource(IHolder *iface, const RESID resid) { holder *This = impl_from_IHolder(iface); FIXME("(%p)->(%08lx) stub\n", This, resid); return E_NOTIMPL; } struct IHolderVtbl holder_vtbl = { holder_QueryInterface, holder_AddRef, holder_Release, holder_AllocResource, holder_FreeResource, holder_TrackResource, holder_TrackResourceS, holder_UntrackResource, holder_UntrackResourceS, holder_Close, holder_RequestDestroyResource }; static HRESULT create_holder(IDispenserDriver *driver, IHolder **object) { holder *hold; HRESULT ret; TRACE("(%p)\n", object); hold = heap_alloc(sizeof(*hold)); if (!hold) { *object = NULL; return E_OUTOFMEMORY; } hold->IHolder_iface.lpVtbl = &holder_vtbl; hold->ref = 1; hold->driver = driver; ret = holder_QueryInterface(&hold->IHolder_iface, &IID_IHolder, (void**)object); holder_Release(&hold->IHolder_iface); return ret; } static HRESULT WINAPI dismanager_QueryInterface(IDispenserManager *iface, REFIID riid, void **object) { dispensermanager *This = impl_from_IDispenserManager(iface); TRACE("(%p)->(%s %p)\n", This, debugstr_guid(riid), object); *object = NULL; if (IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_IDispenserManager)) { *object = &This->IDispenserManager_iface; IUnknown_AddRef( (IUnknown*)*object); return S_OK; } WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),object); return E_NOINTERFACE; } static ULONG WINAPI dismanager_AddRef(IDispenserManager *iface) { dispensermanager *This = impl_from_IDispenserManager(iface); ULONG ref = InterlockedIncrement(&This->ref); TRACE("(%p)->(%d)\n", This, ref); return ref; } static ULONG WINAPI dismanager_Release(IDispenserManager *iface) { dispensermanager *This = impl_from_IDispenserManager(iface); ULONG ref = InterlockedDecrement(&This->ref); TRACE("(%p)->(%d)\n", This, ref); if (!ref) { if (This->mta_thread) { SetEvent(This->mta_stop_event); WaitForSingleObject(This->mta_thread, INFINITE); CloseHandle(This->mta_stop_event); CloseHandle(This->mta_thread); } heap_free(This); } return ref; } static DWORD WINAPI mta_thread_proc(void *arg) { CoInitializeEx(NULL, COINIT_MULTITHREADED); WaitForSingleObject(arg, INFINITE); CoUninitialize(); return 0; } static HRESULT WINAPI dismanager_RegisterDispenser(IDispenserManager *iface, IDispenserDriver *driver, LPCOLESTR name, IHolder **dispenser) { dispensermanager *This = impl_from_IDispenserManager(iface); HRESULT hr; TRACE("(%p)->(%p, %s, %p)\n", This, driver, debugstr_w(name), dispenser); if(!dispenser) return E_INVALIDARG; hr = create_holder(driver, dispenser); if (!This->mta_thread) { This->mta_stop_event = CreateEventA(NULL, TRUE, FALSE, NULL); This->mta_thread = CreateThread(NULL, 0, mta_thread_proc, This->mta_stop_event, 0, NULL); } TRACE("<-- 0x%08x, %p\n", hr, *dispenser); return hr; } static HRESULT WINAPI dismanager_GetContext(IDispenserManager *iface, INSTID *id, TRANSID *transid) { dispensermanager *This = impl_from_IDispenserManager(iface); FIXME("(%p)->(%p, %p) stub\n", This, id, transid); return E_NOTIMPL; } struct IDispenserManagerVtbl dismanager_vtbl = { dismanager_QueryInterface, dismanager_AddRef, dismanager_Release, dismanager_RegisterDispenser, dismanager_GetContext }; static HRESULT WINAPI comsvcscf_CreateInstance(IClassFactory *cf,IUnknown* outer, REFIID riid,void **object) { dispensermanager *dismanager; HRESULT ret; TRACE("(%p %s %p)\n", outer, debugstr_guid(riid), object); dismanager = heap_alloc_zero(sizeof(*dismanager)); if (!dismanager) { *object = NULL; return E_OUTOFMEMORY; } dismanager->IDispenserManager_iface.lpVtbl = &dismanager_vtbl; dismanager->ref = 1; ret = dismanager_QueryInterface(&dismanager->IDispenserManager_iface, riid, object); dismanager_Release(&dismanager->IDispenserManager_iface); return ret; } BOOL WINAPI DllMain(HINSTANCE hinst, DWORD reason, LPVOID lpv) { switch(reason) { case DLL_WINE_PREATTACH: return FALSE; /* prefer native version */ case DLL_PROCESS_ATTACH: COMSVCS_hInstance = hinst; DisableThreadLibraryCalls(hinst); break; } return TRUE; } static HRESULT WINAPI comsvcscf_QueryInterface(IClassFactory *iface, REFIID riid, void **ppv ) { *ppv = NULL; if(IsEqualGUID(&IID_IUnknown, riid)) { TRACE("(%p)->(IID_IUnknown %p)\n", iface, ppv); *ppv = iface; }else if(IsEqualGUID(&IID_IClassFactory, riid)) { TRACE("(%p)->(IID_IClassFactory %p)\n", iface, ppv); *ppv = iface; } if(*ppv) { IUnknown_AddRef((IUnknown*)*ppv); return S_OK; } WARN("(%p)->(%s %p)\n", iface, debugstr_guid(riid), ppv); return E_NOINTERFACE; } static ULONG WINAPI comsvcscf_AddRef(IClassFactory *iface ) { TRACE("(%p)\n", iface); return 2; } static ULONG WINAPI comsvcscf_Release(IClassFactory *iface ) { TRACE("(%p)\n", iface); return 1; } static HRESULT WINAPI comsvcscf_LockServer(IClassFactory *iface, BOOL fLock) { TRACE("(%p)->(%x)\n", iface, fLock); return S_OK; } static const struct IClassFactoryVtbl comsvcscf_vtbl = { comsvcscf_QueryInterface, comsvcscf_AddRef, comsvcscf_Release, comsvcscf_CreateInstance, comsvcscf_LockServer }; static IClassFactory DispenserManageFactory = { &comsvcscf_vtbl }; /****************************************************************** * DllGetClassObject */ HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, void **ppv) { if(IsEqualGUID(&CLSID_DispenserManager, rclsid)) { TRACE("(CLSID_DispenserManager %s %p)\n", debugstr_guid(riid), ppv); return IClassFactory_QueryInterface(&DispenserManageFactory, riid, ppv); } FIXME("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(riid), ppv); return CLASS_E_CLASSNOTAVAILABLE; } /****************************************************************** * DllCanUnloadNow */ HRESULT WINAPI DllCanUnloadNow(void) { return S_FALSE; } /*********************************************************************** * DllRegisterServer (comsvcs.@) */ HRESULT WINAPI DllRegisterServer(void) { return __wine_register_resources( COMSVCS_hInstance ); } /*********************************************************************** * DllUnregisterServer (comsvcs.@) */ HRESULT WINAPI DllUnregisterServer(void) { return __wine_unregister_resources( COMSVCS_hInstance ); }