Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

108 lines
2.6 KiB

// regc.h : Declaration of the Cregc
#ifndef __REGC_H_
#define __REGC_H_
#include "resource.h" // main symbols
/////////////////////////////////////////////////////////////////////////////
// Cregc
class ATL_NO_VTABLE Cregc :
public CComObjectRootEx<CComSingleThreadModel>,
public CComCoClass<Cregc, &CLSID_regc>,
public IDispatchImpl<IComponentRegistrar, &IID_IComponentRegistrar, &LIBID_MDSYNCLib>
{
public:
Cregc()
{
}
DECLARE_NO_REGISTRY()
BEGIN_COM_MAP(Cregc)
COM_INTERFACE_ENTRY(IComponentRegistrar)
COM_INTERFACE_ENTRY(IDispatch)
END_COM_MAP()
// IComponentRegistrar
public:
STDMETHOD(Attach)(BSTR bstrPath)
{
return E_NOTIMPL;
}
STDMETHOD(RegisterAll)()
{
_ATL_OBJMAP_ENTRY* pEntry = _Module.m_pObjMap;
HRESULT hr = S_OK;
while (pEntry->pclsid != NULL)
{
if (pEntry->pfnGetObjectDescription() != NULL)
{
hr = _Module.RegisterServer(TRUE, pEntry->pclsid);
if (FAILED(hr))
break;
}
pEntry++;
}
return hr;
}
STDMETHOD(UnregisterAll)()
{
_ATL_OBJMAP_ENTRY* pEntry = _Module.m_pObjMap;
while (pEntry->pclsid != NULL)
{
if (pEntry->pfnGetObjectDescription() != NULL)
_Module.UnregisterServer(pEntry->pclsid);
pEntry++;
}
return S_OK;
}
STDMETHOD(GetComponents)(SAFEARRAY **ppCLSIDs, SAFEARRAY **ppDescriptions)
{
_ATL_OBJMAP_ENTRY* pEntry = _Module.m_pObjMap;
int nComponents = 0;
while (pEntry->pclsid != NULL)
{
LPCTSTR pszDescription = pEntry->pfnGetObjectDescription();
if (pszDescription)
nComponents++;
pEntry++;
}
SAFEARRAYBOUND rgBound[1];
rgBound[0].lLbound = 0;
rgBound[0].cElements = nComponents;
*ppCLSIDs = SafeArrayCreate(VT_BSTR, 1, rgBound);
*ppDescriptions = SafeArrayCreate(VT_BSTR, 1, rgBound);
pEntry = _Module.m_pObjMap;
for (long i=0; pEntry->pclsid != NULL; pEntry++)
{
LPCTSTR pszDescription = pEntry->pfnGetObjectDescription();
if (pszDescription)
{
LPOLESTR pszCLSID;
StringFromCLSID(*pEntry->pclsid, &pszCLSID);
SafeArrayPutElement(*ppCLSIDs, &i, OLE2BSTR(pszCLSID));
CoTaskMemFree(pszCLSID);
SafeArrayPutElement(*ppCLSIDs, &i, T2BSTR(pszDescription));
i++;
}
}
return S_OK;
}
STDMETHOD(RegisterComponent)(BSTR bstrCLSID)
{
CLSID clsid;
CLSIDFromString(bstrCLSID, &clsid);
_Module.RegisterServer(TRUE, &clsid);
return S_OK;
}
STDMETHOD(UnregisterComponent)(BSTR bstrCLSID)
{
CLSID clsid;
CLSIDFromString(bstrCLSID, &clsid);
_Module.UnregisterServer(&clsid);
return S_OK;
}
};
#endif //__REGC_H_