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.
 
 
 
 
 
 

190 lines
5.2 KiB

//***************************************************************************
//
// MAINDLL.CPP
//
// Module: WMI Instance provider code for boot parameters
//
// Purpose: Contains DLL entry points. Also has code that controls
// when the DLL can be unloaded by tracking the number of
// objects and locks as well as routines that support
// self registration.
//
// Copyright (c) 1997-1999 Microsoft Corporation
//
//***************************************************************************
#include <initguid.h>
#include <objbase.h>
#include "bootini.h"
HMODULE ghModule;
// TODO, GuidGen should be used to generate a unique number for any
// providers that are going to be used for anything more extensive
// than just testing.
DEFINE_GUID(CLSID_instprovider,0x22cb8761, 0x914a, 0x11cf, 0xb7, 0x5, 0x0, 0xaa, 0x0, 0x62, 0xcb, 0xb8);
// {22CB8761-914A-11cf-B705-00AA0062CBB8}
//Count number of objects and number of locks.
long g_cObj=0;
long g_cLock=0;
//***************************************************************************
//
// LibMain32
//
// Purpose: Entry point for DLL.
//
// Return: TRUE if OK.
//
//***************************************************************************
BOOL WINAPI LibMain32(HINSTANCE hInstance, ULONG ulReason
, LPVOID pvReserved)
{
if (DLL_PROCESS_ATTACH==ulReason)
ghModule = hInstance;
return TRUE;
}
//***************************************************************************
//
// DllGetClassObject
//
// Purpose: Called by Ole when some client wants a class factory. Return
// one only if it is the sort of class this DLL supports.
//
//***************************************************************************
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, PPVOID ppv)
{
HRESULT hr;
CBootProvFactory *pObj;
if (CLSID_instprovider!=rclsid)
return E_FAIL;
pObj=new CBootProvFactory();
if (NULL==pObj)
return E_OUTOFMEMORY;
hr=pObj->QueryInterface(riid, ppv);
if (FAILED(hr))
delete pObj;
return hr;
}
//***************************************************************************
//
// DllCanUnloadNow
//
// Purpose: Called periodically by Ole in order to determine if the
// DLL can be freed.
//
// Return: S_OK if there are no objects in use and the class factory
// isn't locked.
//
//***************************************************************************
STDAPI DllCanUnloadNow(void)
{
SCODE sc;
//It is OK to unload if there are no objects or locks on the
// class factory.
sc=(0L==g_cObj && 0L==g_cLock) ? S_OK : S_FALSE;
return sc;
}
//***************************************************************************
//
// DllRegisterServer
//
// Purpose: Called during setup or by regsvr32.
//
// Return: NOERROR if registration successful, error otherwise.
//***************************************************************************
STDAPI DllRegisterServer(void)
{
char szID[128];
WCHAR wcID[128];
char szCLSID[128];
char szModule[MAX_PATH];
char * pName = "WMI Boot Instance Provider";
char * pModel = "Both";
HKEY hKey1, hKey2;
// Create the path.
StringFromGUID2(CLSID_instprovider, wcID, 128);
wcstombs(szID, wcID, 128);
lstrcpy(szCLSID, TEXT("Software\\classes\\CLSID\\"));
lstrcat(szCLSID, szID);
// Create entries under CLSID
RegCreateKey(HKEY_LOCAL_MACHINE, szCLSID, &hKey1);
RegSetValueEx(hKey1, NULL, 0, REG_SZ, (BYTE *)pName, lstrlen(pName)+1);
RegCreateKey(hKey1,"InprocServer32",&hKey2);
GetModuleFileName(ghModule, szModule, MAX_PATH);
RegSetValueEx(hKey2, NULL, 0, REG_SZ, (BYTE *)szModule,
lstrlen(szModule)+1);
RegSetValueEx(hKey2, "ThreadingModel", 0, REG_SZ,
(BYTE *)pModel, lstrlen(pModel)+1);
RegCloseKey(hKey1);
RegCloseKey(hKey2);
return NOERROR;
}
//***************************************************************************
//
// DllUnregisterServer
//
// Purpose: Called when it is time to remove the registry entries.
//
// Return: NOERROR if registration successful, error otherwise.
//***************************************************************************
STDAPI DllUnregisterServer(void)
{
char szID[128];
WCHAR wcID[128];
char szCLSID[128];
HKEY hKey;
// Create the path using the CLSID
StringFromGUID2(CLSID_instprovider, wcID, 128);
wcstombs(szID, wcID, 128);
lstrcpy(szCLSID, TEXT("Software\\classes\\CLSID\\"));
lstrcat(szCLSID, szID);
// First delete the InProcServer subkey.
DWORD dwRet = RegOpenKey(HKEY_LOCAL_MACHINE, szCLSID, &hKey);
if(dwRet == NO_ERROR)
{
RegDeleteKey(hKey, "InProcServer32");
RegCloseKey(hKey);
}
dwRet = RegOpenKey(HKEY_LOCAL_MACHINE, "Software\\classes\\CLSID", &hKey);
if(dwRet == NO_ERROR)
{
RegDeleteKey(hKey,szID);
RegCloseKey(hKey);
}
return NOERROR;
}