202 lines
4.5 KiB
C++
202 lines
4.5 KiB
C++
#include "privcpp.h"
|
|
#include "ids.h"
|
|
#include <advpub.h>
|
|
|
|
#define INITGUID
|
|
#include <initguid.h>
|
|
#include "packguid.h"
|
|
|
|
#define DECL_CRTFREE
|
|
#include <crtfree.h>
|
|
|
|
LONG g_cRefThisDll = 0;
|
|
HINSTANCE g_hinst = NULL;
|
|
HINSTANCE g_hinstResDLL = NULL;
|
|
|
|
STDAPI_(BOOL) DllMain(HINSTANCE hDll, DWORD dwReason, LPVOID lpReserved)
|
|
{
|
|
switch(dwReason) {
|
|
case DLL_PROCESS_ATTACH:
|
|
g_hinst = hDll;
|
|
g_hinstResDLL = hDll;
|
|
#ifdef DEBUG
|
|
CcshellGetDebugFlags();
|
|
#endif // DEBUG
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
class CMyClassFactory : public IClassFactory
|
|
{
|
|
public:
|
|
CMyClassFactory(REFCLSID rclsid);
|
|
~CMyClassFactory() { g_cRefThisDll--; }
|
|
|
|
// IUnKnown
|
|
STDMETHODIMP QueryInterface(REFIID,void **);
|
|
STDMETHODIMP_(ULONG) AddRef(void);
|
|
STDMETHODIMP_(ULONG) Release(void);
|
|
|
|
// IClassFactory
|
|
STDMETHODIMP CreateInstance(IUnknown* pUnkOuter, REFIID riid, void **ppvObject);
|
|
STDMETHODIMP LockServer(BOOL fLock);
|
|
|
|
protected:
|
|
LONG _cRef;
|
|
CLSID _clsid;
|
|
};
|
|
|
|
CMyClassFactory::CMyClassFactory(REFCLSID rclsid) : _cRef(1), _clsid(rclsid)
|
|
{
|
|
g_cRefThisDll++;
|
|
}
|
|
|
|
HRESULT CMyClassFactory::QueryInterface(REFIID riid, void **ppvObject)
|
|
{
|
|
HRESULT hres;
|
|
if (IsEqualGUID(riid, IID_IClassFactory) || IsEqualGUID(riid, IID_IUnknown))
|
|
{
|
|
DebugMsg(DM_TRACE, "pack cf - QueryInterface called");
|
|
|
|
_cRef++;
|
|
*ppvObject = (IClassFactory *)this;
|
|
hres = NOERROR;
|
|
}
|
|
else
|
|
{
|
|
*ppvObject = NULL;
|
|
hres = E_NOINTERFACE;
|
|
}
|
|
|
|
return hres;
|
|
}
|
|
|
|
ULONG CMyClassFactory::AddRef(void)
|
|
{
|
|
DebugMsg(DM_TRACE, "pack cf - AddRef called");
|
|
return ++_cRef;
|
|
}
|
|
|
|
ULONG CMyClassFactory::Release(void)
|
|
{
|
|
DebugMsg(DM_TRACE, "pack cf - Release called");
|
|
|
|
LONG cRef = InterlockedDecrement( &_cRef );
|
|
if ( 0 == cRef )
|
|
{
|
|
delete this;
|
|
}
|
|
return cRef;
|
|
}
|
|
|
|
HRESULT CMyClassFactory::CreateInstance(IUnknown* pUnkOuter, REFIID riid, void **ppvObject)
|
|
{
|
|
DebugMsg(DM_TRACE, "CMyClassFactory::CreateInstance called");
|
|
|
|
if (pUnkOuter)
|
|
return CLASS_E_NOAGGREGATION;
|
|
|
|
IUnknown* punk;
|
|
HRESULT hres;
|
|
if (IsEqualGUID(_clsid, CLSID_CPackage))
|
|
{
|
|
hres = CPackage_CreateInstance(&punk);
|
|
}
|
|
else
|
|
{
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
if (SUCCEEDED(hres))
|
|
{
|
|
hres = punk->QueryInterface(riid, ppvObject);
|
|
punk->Release();
|
|
}
|
|
return hres;
|
|
}
|
|
|
|
HRESULT CMyClassFactory::LockServer(BOOL fLock)
|
|
{
|
|
DebugMsg(DM_TRACE, "pack cf - LockServer called");
|
|
return S_OK;
|
|
}
|
|
|
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID FAR* ppvOut)
|
|
{
|
|
DebugMsg(DM_TRACE, "pack - DllGetClassObject called");
|
|
|
|
if (IsEqualGUID(rclsid,CLSID_CPackage))
|
|
{
|
|
CMyClassFactory *pmycls = new CMyClassFactory(rclsid);
|
|
if (pmycls)
|
|
{
|
|
HRESULT hres = pmycls->QueryInterface(riid, ppvOut);
|
|
pmycls->Release();
|
|
return hres;
|
|
}
|
|
return E_OUTOFMEMORY;
|
|
}
|
|
return CLASS_E_CLASSNOTAVAILABLE;
|
|
}
|
|
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
DebugMsg(DM_TRACE, "pack - DllCanUnloadNow called");
|
|
if (g_cRefThisDll)
|
|
{
|
|
return S_FALSE;
|
|
}
|
|
|
|
DebugMsg(DM_TRACE, "pack - DllCanUnloadNow returning S_OK (bye, bye...)");
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
// Call ADVPACK for the given section of our resource based INF>
|
|
// hInstance = resource instance to get REGINST section from
|
|
// szSection = section name to invoke
|
|
HRESULT CallRegInstall(HINSTANCE hInstance, LPCSTR szSection)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
HINSTANCE hinstAdvPack = LoadLibrary(TEXT("ADVPACK.DLL"));
|
|
if (hinstAdvPack)
|
|
{
|
|
REGINSTALL pfnri = (REGINSTALL)GetProcAddress(hinstAdvPack, "RegInstall");
|
|
if ( pfnri )
|
|
{
|
|
#ifdef WINNT
|
|
STRENTRY seReg[] =
|
|
{
|
|
// These two NT-specific entries must be at the end
|
|
{ "25", "%SystemRoot%" },
|
|
{ "11", "%SystemRoot%\\system32" },
|
|
};
|
|
STRTABLE stReg = { ARRAYSIZE(seReg), seReg };
|
|
hr = pfnri(hInstance, szSection, &stReg);
|
|
#else
|
|
hr = pfnri(hInstance, szSection, NULL);
|
|
#endif
|
|
}
|
|
FreeLibrary(hinstAdvPack);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
CallRegInstall(g_hinst, "RegDll");
|
|
return S_OK;
|
|
}
|
|
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
CallRegInstall(g_hinst, "UnregDll");
|
|
return S_OK;
|
|
}
|
|
|