| /* |
| * ITSS Class Factory |
| * |
| * Copyright 2002 Lionel Ulmer |
| * Copyright 2004 Mike McCormack |
| * |
| * see http://bonedaddy.net/pabs3/hhm/#chmspec |
| * |
| * 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 |
| */ |
| |
| #include "config.h" |
| |
| #include <stdarg.h> |
| #include <stdio.h> |
| |
| #define COBJMACROS |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winuser.h" |
| #include "winreg.h" |
| #include "ole2.h" |
| #include "rpcproxy.h" |
| #include "advpub.h" |
| |
| #include "wine/unicode.h" |
| #include "wine/debug.h" |
| |
| #include "itsstor.h" |
| |
| #include "initguid.h" |
| #include "wine/itss.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(itss); |
| |
| static HRESULT ITSS_create(IUnknown *pUnkOuter, LPVOID *ppObj); |
| |
| LONG dll_count = 0; |
| static HINSTANCE hInst; |
| |
| BOOL WINAPI DllMain(HINSTANCE hInstDLL, DWORD fdwReason, LPVOID lpv) |
| { |
| switch(fdwReason) { |
| case DLL_PROCESS_ATTACH: |
| DisableThreadLibraryCalls(hInstDLL); |
| hInst = hInstDLL; |
| break; |
| } |
| return TRUE; |
| } |
| |
| /****************************************************************************** |
| * ITSS ClassFactory |
| */ |
| typedef struct { |
| IClassFactory IClassFactory_iface; |
| HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj); |
| } IClassFactoryImpl; |
| |
| static inline IClassFactoryImpl *impl_from_IClassFactory(IClassFactory *iface) |
| { |
| return CONTAINING_RECORD(iface, IClassFactoryImpl, IClassFactory_iface); |
| } |
| |
| static HRESULT WINAPI |
| ITSSCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid,LPVOID *ppobj) |
| { |
| IClassFactoryImpl *This = impl_from_IClassFactory(iface); |
| |
| if (IsEqualGUID(riid, &IID_IUnknown) || |
| IsEqualGUID(riid, &IID_IClassFactory)) |
| { |
| IClassFactory_AddRef(iface); |
| *ppobj = This; |
| return S_OK; |
| } |
| |
| WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppobj); |
| return E_NOINTERFACE; |
| } |
| |
| static ULONG WINAPI ITSSCF_AddRef(LPCLASSFACTORY iface) |
| { |
| ITSS_LockModule(); |
| return 2; |
| } |
| |
| static ULONG WINAPI ITSSCF_Release(LPCLASSFACTORY iface) |
| { |
| ITSS_UnlockModule(); |
| return 1; |
| } |
| |
| |
| static HRESULT WINAPI ITSSCF_CreateInstance(LPCLASSFACTORY iface, LPUNKNOWN pOuter, |
| REFIID riid, LPVOID *ppobj) |
| { |
| IClassFactoryImpl *This = impl_from_IClassFactory(iface); |
| HRESULT hres; |
| LPUNKNOWN punk; |
| |
| TRACE("(%p)->(%p,%s,%p)\n", This, pOuter, debugstr_guid(riid), ppobj); |
| |
| *ppobj = NULL; |
| hres = This->pfnCreateInstance(pOuter, (LPVOID *) &punk); |
| if (SUCCEEDED(hres)) { |
| hres = IUnknown_QueryInterface(punk, riid, ppobj); |
| IUnknown_Release(punk); |
| } |
| return hres; |
| } |
| |
| static HRESULT WINAPI ITSSCF_LockServer(LPCLASSFACTORY iface, BOOL dolock) |
| { |
| TRACE("(%p)->(%d)\n", iface, dolock); |
| |
| if (dolock) |
| ITSS_LockModule(); |
| else |
| ITSS_UnlockModule(); |
| |
| return S_OK; |
| } |
| |
| static const IClassFactoryVtbl ITSSCF_Vtbl = |
| { |
| ITSSCF_QueryInterface, |
| ITSSCF_AddRef, |
| ITSSCF_Release, |
| ITSSCF_CreateInstance, |
| ITSSCF_LockServer |
| }; |
| |
| static const IClassFactoryImpl ITStorage_factory = { { &ITSSCF_Vtbl }, ITSS_create }; |
| static const IClassFactoryImpl MSITStore_factory = { { &ITSSCF_Vtbl }, ITS_IParseDisplayName_create }; |
| static const IClassFactoryImpl ITSProtocol_factory = { { &ITSSCF_Vtbl }, ITSProtocol_create }; |
| |
| /*********************************************************************** |
| * DllGetClassObject (ITSS.@) |
| */ |
| HRESULT WINAPI DllGetClassObject(REFCLSID rclsid, REFIID iid, LPVOID *ppv) |
| { |
| const IClassFactoryImpl *factory; |
| |
| TRACE("%s %s %p\n", debugstr_guid(rclsid), debugstr_guid(iid), ppv); |
| |
| if (IsEqualGUID(&CLSID_ITStorage, rclsid)) |
| factory = &ITStorage_factory; |
| else if (IsEqualGUID(&CLSID_MSITStore, rclsid)) |
| factory = &MSITStore_factory; |
| else if (IsEqualGUID(&CLSID_ITSProtocol, rclsid)) |
| factory = &ITSProtocol_factory; |
| else |
| { |
| FIXME("%s: no class found.\n", debugstr_guid(rclsid)); |
| return CLASS_E_CLASSNOTAVAILABLE; |
| } |
| |
| return IUnknown_QueryInterface( (IUnknown*) factory, iid, ppv ); |
| } |
| |
| /*****************************************************************************/ |
| |
| typedef struct { |
| IITStorage IITStorage_iface; |
| LONG ref; |
| } ITStorageImpl; |
| |
| static inline ITStorageImpl *impl_from_IITStorage(IITStorage *iface) |
| { |
| return CONTAINING_RECORD(iface, ITStorageImpl, IITStorage_iface); |
| } |
| |
| |
| static HRESULT WINAPI ITStorageImpl_QueryInterface( |
| IITStorage* iface, |
| REFIID riid, |
| void** ppvObject) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| if (IsEqualGUID(riid, &IID_IUnknown) |
| || IsEqualGUID(riid, &IID_IITStorage)) |
| { |
| IITStorage_AddRef(iface); |
| *ppvObject = iface; |
| return S_OK; |
| } |
| |
| WARN("(%p)->(%s,%p),not found\n",This,debugstr_guid(riid),ppvObject); |
| return E_NOINTERFACE; |
| } |
| |
| static ULONG WINAPI ITStorageImpl_AddRef( |
| IITStorage* iface) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| TRACE("%p\n", This); |
| return InterlockedIncrement(&This->ref); |
| } |
| |
| static ULONG WINAPI ITStorageImpl_Release( |
| IITStorage* iface) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| ULONG ref = InterlockedDecrement(&This->ref); |
| |
| if (ref == 0) { |
| HeapFree(GetProcessHeap(), 0, This); |
| ITSS_UnlockModule(); |
| } |
| |
| return ref; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgCreateDocfile( |
| IITStorage* iface, |
| const WCHAR* pwcsName, |
| DWORD grfMode, |
| DWORD reserved, |
| IStorage** ppstgOpen) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| |
| TRACE("%p %s %u %u %p\n", This, |
| debugstr_w(pwcsName), grfMode, reserved, ppstgOpen ); |
| |
| return ITSS_StgOpenStorage( pwcsName, NULL, grfMode, |
| 0, reserved, ppstgOpen); |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgCreateDocfileOnILockBytes( |
| IITStorage* iface, |
| ILockBytes* plkbyt, |
| DWORD grfMode, |
| DWORD reserved, |
| IStorage** ppstgOpen) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgIsStorageFile( |
| IITStorage* iface, |
| const WCHAR* pwcsName) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgIsStorageILockBytes( |
| IITStorage* iface, |
| ILockBytes* plkbyt) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgOpenStorage( |
| IITStorage* iface, |
| const WCHAR* pwcsName, |
| IStorage* pstgPriority, |
| DWORD grfMode, |
| SNB snbExclude, |
| DWORD reserved, |
| IStorage** ppstgOpen) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| |
| TRACE("%p %s %p %d %p\n", This, debugstr_w( pwcsName ), |
| pstgPriority, grfMode, snbExclude ); |
| |
| return ITSS_StgOpenStorage( pwcsName, pstgPriority, grfMode, |
| snbExclude, reserved, ppstgOpen); |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgOpenStorageOnILockBytes( |
| IITStorage* iface, |
| ILockBytes* plkbyt, |
| IStorage* pStgPriority, |
| DWORD grfMode, |
| SNB snbExclude, |
| DWORD reserved, |
| IStorage** ppstgOpen) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_StgSetTimes( |
| IITStorage* iface, |
| const WCHAR* lpszName, |
| const FILETIME* pctime, |
| const FILETIME* patime, |
| const FILETIME* pmtime) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_SetControlData( |
| IITStorage* iface, |
| PITS_Control_Data pControlData) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_DefaultControlData( |
| IITStorage* iface, |
| PITS_Control_Data* ppControlData) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static HRESULT WINAPI ITStorageImpl_Compact( |
| IITStorage* iface, |
| const WCHAR* pwcsName, |
| ECompactionLev iLev) |
| { |
| ITStorageImpl *This = impl_from_IITStorage(iface); |
| FIXME("%p\n", This); |
| return E_NOTIMPL; |
| } |
| |
| static const IITStorageVtbl ITStorageImpl_Vtbl = |
| { |
| ITStorageImpl_QueryInterface, |
| ITStorageImpl_AddRef, |
| ITStorageImpl_Release, |
| ITStorageImpl_StgCreateDocfile, |
| ITStorageImpl_StgCreateDocfileOnILockBytes, |
| ITStorageImpl_StgIsStorageFile, |
| ITStorageImpl_StgIsStorageILockBytes, |
| ITStorageImpl_StgOpenStorage, |
| ITStorageImpl_StgOpenStorageOnILockBytes, |
| ITStorageImpl_StgSetTimes, |
| ITStorageImpl_SetControlData, |
| ITStorageImpl_DefaultControlData, |
| ITStorageImpl_Compact, |
| }; |
| |
| static HRESULT ITSS_create(IUnknown *pUnkOuter, LPVOID *ppObj) |
| { |
| ITStorageImpl *its; |
| |
| if( pUnkOuter ) |
| return CLASS_E_NOAGGREGATION; |
| |
| its = HeapAlloc( GetProcessHeap(), 0, sizeof(ITStorageImpl) ); |
| its->IITStorage_iface.lpVtbl = &ITStorageImpl_Vtbl; |
| its->ref = 1; |
| |
| TRACE("-> %p\n", its); |
| *ppObj = its; |
| |
| ITSS_LockModule(); |
| return S_OK; |
| } |
| |
| /*****************************************************************************/ |
| |
| HRESULT WINAPI DllCanUnloadNow(void) |
| { |
| TRACE("dll_count = %u\n", dll_count); |
| return dll_count ? S_FALSE : S_OK; |
| } |
| |
| /*********************************************************************** |
| * DllRegisterServer (ITSS.@) |
| */ |
| HRESULT WINAPI DllRegisterServer(void) |
| { |
| return __wine_register_resources( hInst ); |
| } |
| |
| /*********************************************************************** |
| * DllUnregisterServer (ITSS.@) |
| */ |
| HRESULT WINAPI DllUnregisterServer(void) |
| { |
| return __wine_unregister_resources( hInst ); |
| } |