448 lines
11 KiB
C
448 lines
11 KiB
C
/*
|
|
* 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 "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;
|
|
case DLL_PROCESS_DETACH:
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
/******************************************************************************
|
|
* ITSS ClassFactory
|
|
*/
|
|
typedef struct {
|
|
const IClassFactoryVtbl *lpVtbl;
|
|
HRESULT (*pfnCreateInstance)(IUnknown *pUnkOuter, LPVOID *ppObj);
|
|
} IClassFactoryImpl;
|
|
|
|
static HRESULT WINAPI
|
|
ITSSCF_QueryInterface(LPCLASSFACTORY iface,REFIID riid,LPVOID *ppobj)
|
|
{
|
|
IClassFactoryImpl *This = (IClassFactoryImpl *)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 = (IClassFactoryImpl *)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 {
|
|
const IITStorageVtbl *vtbl_IITStorage;
|
|
LONG ref;
|
|
} ITStorageImpl;
|
|
|
|
|
|
static HRESULT WINAPI ITStorageImpl_QueryInterface(
|
|
IITStorage* iface,
|
|
REFIID riid,
|
|
void** ppvObject)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)iface;
|
|
if (IsEqualGUID(riid, &IID_IUnknown)
|
|
|| IsEqualGUID(riid, &IID_IITStorage))
|
|
{
|
|
IClassFactory_AddRef(iface);
|
|
*ppvObject = This;
|
|
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 = (ITStorageImpl *)iface;
|
|
TRACE("%p\n", This);
|
|
return InterlockedIncrement(&This->ref);
|
|
}
|
|
|
|
static ULONG WINAPI ITStorageImpl_Release(
|
|
IITStorage* iface)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)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 = (ITStorageImpl *)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 = (ITStorageImpl *)iface;
|
|
FIXME("%p\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI ITStorageImpl_StgIsStorageFile(
|
|
IITStorage* iface,
|
|
const WCHAR* pwcsName)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)iface;
|
|
FIXME("%p\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI ITStorageImpl_StgIsStorageILockBytes(
|
|
IITStorage* iface,
|
|
ILockBytes* plkbyt)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)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 = (ITStorageImpl *)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 = (ITStorageImpl *)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 = (ITStorageImpl *)iface;
|
|
FIXME("%p\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI ITStorageImpl_SetControlData(
|
|
IITStorage* iface,
|
|
PITS_Control_Data pControlData)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)iface;
|
|
FIXME("%p\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI ITStorageImpl_DefaultControlData(
|
|
IITStorage* iface,
|
|
PITS_Control_Data* ppControlData)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)iface;
|
|
FIXME("%p\n", This);
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static HRESULT WINAPI ITStorageImpl_Compact(
|
|
IITStorage* iface,
|
|
const WCHAR* pwcsName,
|
|
ECompactionLev iLev)
|
|
{
|
|
ITStorageImpl *This = (ITStorageImpl *)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->vtbl_IITStorage = &ITStorageImpl_Vtbl;
|
|
its->ref = 1;
|
|
|
|
TRACE("-> %p\n", its);
|
|
*ppObj = (LPVOID) its;
|
|
|
|
ITSS_LockModule();
|
|
return S_OK;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
|
|
HRESULT WINAPI DllCanUnloadNow(void)
|
|
{
|
|
TRACE("dll_count = %u\n", dll_count);
|
|
return dll_count ? S_FALSE : S_OK;
|
|
}
|
|
|
|
#define INF_SET_ID(id) \
|
|
do \
|
|
{ \
|
|
static CHAR name[] = #id; \
|
|
\
|
|
pse[i].pszName = name; \
|
|
clsids[i++] = &id; \
|
|
} while (0)
|
|
|
|
#define INF_SET_CLSID(clsid) INF_SET_ID(CLSID_ ## clsid)
|
|
|
|
static HRESULT register_server(BOOL do_register)
|
|
{
|
|
HRESULT hres;
|
|
HMODULE hAdvpack;
|
|
HRESULT (WINAPI *pRegInstall)(HMODULE hm, LPCSTR pszSection, const STRTABLEA* pstTable);
|
|
STRTABLEA strtable;
|
|
STRENTRYA pse[4];
|
|
static CLSID const *clsids[4];
|
|
DWORD i = 0;
|
|
|
|
static const WCHAR wszAdvpack[] = {'a','d','v','p','a','c','k','.','d','l','l',0};
|
|
|
|
INF_SET_CLSID(ITStorage);
|
|
INF_SET_CLSID(MSFSStore);
|
|
INF_SET_CLSID(MSITStore);
|
|
INF_SET_CLSID(ITSProtocol);
|
|
|
|
strtable.cEntries = sizeof(pse)/sizeof(pse[0]);
|
|
strtable.pse = pse;
|
|
|
|
for(i=0; i < strtable.cEntries; i++) {
|
|
pse[i].pszValue = HeapAlloc(GetProcessHeap(), 0, 39);
|
|
sprintf(pse[i].pszValue, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
|
|
clsids[i]->Data1, clsids[i]->Data2, clsids[i]->Data3, clsids[i]->Data4[0],
|
|
clsids[i]->Data4[1], clsids[i]->Data4[2], clsids[i]->Data4[3], clsids[i]->Data4[4],
|
|
clsids[i]->Data4[5], clsids[i]->Data4[6], clsids[i]->Data4[7]);
|
|
}
|
|
|
|
hAdvpack = LoadLibraryW(wszAdvpack);
|
|
pRegInstall = (void *)GetProcAddress(hAdvpack, "RegInstall");
|
|
|
|
hres = pRegInstall(hInst, do_register ? "RegisterDll" : "UnregisterDll", &strtable);
|
|
|
|
for(i=0; i < sizeof(pse)/sizeof(pse[0]); i++)
|
|
HeapFree(GetProcessHeap(), 0, pse[i].pszValue);
|
|
|
|
return hres;
|
|
}
|
|
|
|
#undef INF_SET_CLSID
|
|
#undef INF_SET_ID
|
|
|
|
/***********************************************************************
|
|
* DllRegisterServer (ITSS.@)
|
|
*/
|
|
HRESULT WINAPI DllRegisterServer(void)
|
|
{
|
|
return register_server(TRUE);
|
|
}
|
|
|
|
/***********************************************************************
|
|
* DllUnregisterServer (ITSS.@)
|
|
*/
|
|
HRESULT WINAPI DllUnregisterServer(void)
|
|
{
|
|
return register_server(FALSE);
|
|
}
|