416 lines
12 KiB
C++
416 lines
12 KiB
C++
|
//+--------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Microsoft Windows
|
||
|
|
// Copyright (C) Microsoft Corporation, 1994 - 1997.
|
||
|
|
//
|
||
|
|
// File: snapin.cpp
|
||
|
|
//
|
||
|
|
// Contents: DLL support routines, class factory and registration
|
||
|
|
// functions.
|
||
|
|
//
|
||
|
|
// Classes:
|
||
|
|
//
|
||
|
|
// Functions:
|
||
|
|
//
|
||
|
|
// History: 2-12-1998 stevebl comment header added
|
||
|
|
// 4-22-1998 rahulth added CSection objects for the ini files
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#include "precomp.hxx"
|
||
|
|
#include "initguid.h"
|
||
|
|
#include "gpedit.h"
|
||
|
|
|
||
|
|
extern const CLSID CLSID_Snapin = {0x88E729D6,0xBDC1,0x11D1,{0xBD,0x2A,0x00,0xC0,0x4F,0xB9,0x60,0x3F}};
|
||
|
|
extern const wchar_t * szCLSID_Snapin = L"{88E729D6-BDC1-11D1-BD2A-00C04FB9603F}";
|
||
|
|
|
||
|
|
// Main NodeType GUID on numeric format
|
||
|
|
extern const GUID cNodeType = {0xE0494114,0xBDC1,0x11D1,{0xBD,0x2A,0x00,0xC0,0x4F,0xB9,0x60,0x3F}};
|
||
|
|
|
||
|
|
// Main NodeType GUID on string format
|
||
|
|
extern const wchar_t* cszNodeType = L"{E0494114-BDC1-11D1-BD2A-00C04FB9603F}";
|
||
|
|
|
||
|
|
// RSOP GUIDs
|
||
|
|
extern const CLSID CLSID_RSOP_Snapin = {0XC40D66A0,0XE90C,0X46C6,{0XAA,0X3B,0X47,0X3E,0X38,0XC7,0X2B,0XF2}};
|
||
|
|
extern const wchar_t * szCLSID_RSOP_Snapin = L"{C40D66A0-E90C-46C6-AA3B-473E38C72BF2}";
|
||
|
|
|
||
|
|
IMalloc * g_pIMalloc;
|
||
|
|
|
||
|
|
#include "safereg.hxx"
|
||
|
|
|
||
|
|
#define BREAK_ON_FAIL_HRESULT(hr) if (FAILED(hr)) break
|
||
|
|
|
||
|
|
#define THREADING_STR L"Apartment"
|
||
|
|
|
||
|
|
//+--------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: RegDeleteTree
|
||
|
|
//
|
||
|
|
// Synopsis: deletes a registry key and all of its children
|
||
|
|
//
|
||
|
|
// Arguments: [hKey] - handle to the key's parent
|
||
|
|
// [szSubKey] - name of the key to be deleted
|
||
|
|
//
|
||
|
|
// Returns: ERROR_SUCCESS
|
||
|
|
//
|
||
|
|
// History: 3-17-1998 stevebl Copied from ADE
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
LONG RegDeleteTree(HKEY hKey, TCHAR * szSubKey)
|
||
|
|
{
|
||
|
|
HKEY hKeyNew;
|
||
|
|
LONG lResult = RegOpenKey(hKey, szSubKey, &hKeyNew);
|
||
|
|
if (lResult != ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
return lResult;
|
||
|
|
}
|
||
|
|
TCHAR szName[256];
|
||
|
|
while (ERROR_SUCCESS == RegEnumKey(hKeyNew, 0, szName, 256))
|
||
|
|
{
|
||
|
|
RegDeleteTree(hKeyNew, szName);
|
||
|
|
}
|
||
|
|
RegCloseKey(hKeyNew);
|
||
|
|
return RegDeleteKey(hKey, szSubKey);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
RegisterInterface(
|
||
|
|
CSafeReg *pshkInterface,
|
||
|
|
LPWSTR wszInterfaceGUID,
|
||
|
|
LPWSTR wszInterfaceName,
|
||
|
|
LPWSTR wszNumMethods,
|
||
|
|
LPWSTR wszProxyCLSID);
|
||
|
|
|
||
|
|
CComModule _Module;
|
||
|
|
|
||
|
|
BEGIN_OBJECT_MAP(ObjectMap)
|
||
|
|
OBJECT_ENTRY(CLSID_Snapin, CUserComponentDataImpl)
|
||
|
|
OBJECT_ENTRY(CLSID_RSOP_Snapin, CRSOPUserComponentDataImpl)
|
||
|
|
END_OBJECT_MAP()
|
||
|
|
|
||
|
|
#ifdef _DEBUG
|
||
|
|
#define new DEBUG_NEW
|
||
|
|
#undef THIS_FILE
|
||
|
|
static char THIS_FILE[] = __FILE__;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
class CSnapinApp : public CWinApp
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
virtual BOOL InitInstance();
|
||
|
|
virtual int ExitInstance();
|
||
|
|
};
|
||
|
|
|
||
|
|
CSnapinApp theApp;
|
||
|
|
|
||
|
|
HINSTANCE ghInstance;
|
||
|
|
|
||
|
|
BOOL CSnapinApp::InitInstance()
|
||
|
|
{
|
||
|
|
ghInstance = m_hInstance;
|
||
|
|
_Module.Init(ObjectMap, m_hInstance);
|
||
|
|
if (FAILED(CoGetMalloc(1, &g_pIMalloc)))
|
||
|
|
return FALSE;
|
||
|
|
return CWinApp::InitInstance();
|
||
|
|
}
|
||
|
|
|
||
|
|
int CSnapinApp::ExitInstance()
|
||
|
|
{
|
||
|
|
_Module.Term();
|
||
|
|
|
||
|
|
DEBUG_VERIFY_INSTANCE_COUNT(CResultPane);
|
||
|
|
DEBUG_VERIFY_INSTANCE_COUNT(CScopePane);
|
||
|
|
|
||
|
|
g_pIMalloc->Release();
|
||
|
|
return CWinApp::ExitInstance();
|
||
|
|
}
|
||
|
|
|
||
|
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
// Used to determine whether the DLL can be unloaded by OLE
|
||
|
|
|
||
|
|
STDAPI DllCanUnloadNow(void)
|
||
|
|
{
|
||
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
||
|
|
return (AfxDllCanUnloadNow()==S_OK && _Module.GetLockCount()==0) ? S_OK : S_FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
// Returns a class factory to create an object of the requested type
|
||
|
|
|
||
|
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
|
||
|
|
{
|
||
|
|
return _Module.GetClassObject(rclsid, riid, ppv);
|
||
|
|
}
|
||
|
|
|
||
|
|
const wchar_t * szUser_Namespace = L"{08114B47-BDC2-11D1-BD2A-00C04FB9603F}";
|
||
|
|
const wchar_t * szUserAppName = L"Folder Redirection Editor (Users)";
|
||
|
|
const wchar_t * szUser_RSOP_Namespace = L"{8CDE1CC8-0D3A-4B60-99EA-27EF3D7C0174}";
|
||
|
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
// DllRegisterServer - Adds entries to the system registry
|
||
|
|
|
||
|
|
STDAPI DllRegisterServer(void)
|
||
|
|
{
|
||
|
|
AFX_MANAGE_STATE (AfxGetStaticModuleState());
|
||
|
|
|
||
|
|
CSafeReg shk;
|
||
|
|
CSafeReg shkCLSID;
|
||
|
|
CSafeReg shkServer;
|
||
|
|
CSafeReg shkTemp;
|
||
|
|
WCHAR szMUIUserAppName[MAX_PATH + 50];
|
||
|
|
WCHAR szModule[MAX_PATH];
|
||
|
|
|
||
|
|
// Get the path to the module.
|
||
|
|
if (0 != ::GetModuleFileName(AfxGetInstanceHandle(), szModule, MAX_PATH))
|
||
|
|
{
|
||
|
|
// MUI: MMC: Use the new NameStringIndirect value of MMC to display
|
||
|
|
// a localized name for the snap-in while using MUI.
|
||
|
|
wsprintf (szMUIUserAppName, L"@%s,-%d", szModule, IDS_SNAPIN_NAME);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
szMUIUserAppName[0] = L'\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
do
|
||
|
|
{
|
||
|
|
hr = _Module.RegisterServer(FALSE);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
// register extension
|
||
|
|
hr = shkCLSID.Open(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\SnapIns", KEY_WRITE);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
|
||
|
|
hr = shkCLSID.Create(szCLSID_Snapin, &shk);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shk.SetValue(L"NameString",
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) szUserAppName,
|
||
|
|
sizeof(WCHAR) * (lstrlen(szUserAppName)+ 1));
|
||
|
|
|
||
|
|
// MUI: MMC: Use localized snap-in name for MUI.
|
||
|
|
if (L'\0' != szMUIUserAppName[0])
|
||
|
|
{
|
||
|
|
hr = shk.SetValue(L"NameStringIndirect",
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) szMUIUserAppName,
|
||
|
|
sizeof(WCHAR) * (lstrlen(szMUIUserAppName) + 1));
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = shk.Create(L"NodeTypes", &shkTemp);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkTemp.Create(szUser_Namespace, &shkServer);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
shkServer.Close();
|
||
|
|
shkTemp.Close();
|
||
|
|
shk.Close();
|
||
|
|
shkCLSID.Close();
|
||
|
|
|
||
|
|
hr = shkCLSID.Open(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\NodeTypes", KEY_WRITE);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkCLSID.Create(szUser_Namespace, &shk);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
shk.Close();
|
||
|
|
|
||
|
|
WCHAR szGUID[50];
|
||
|
|
StringFromGUID2 (NODEID_User, szGUID, 50);
|
||
|
|
|
||
|
|
hr = shkCLSID.Create(szGUID, &shk);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shk.Create(L"Extensions", &shkServer);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkServer.Create(L"NameSpace", &shkTemp);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkTemp.SetValue(szCLSID_Snapin,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) szUserAppName,
|
||
|
|
sizeof(WCHAR) * (lstrlen(szUserAppName)+ 1));
|
||
|
|
shkTemp.Close();
|
||
|
|
shkServer.Close();
|
||
|
|
shk.Close();
|
||
|
|
shkCLSID.Close();
|
||
|
|
|
||
|
|
// register RSOP extension
|
||
|
|
hr = shkCLSID.Open(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\SnapIns", KEY_WRITE);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
|
||
|
|
hr = shkCLSID.Create(szCLSID_RSOP_Snapin, &shk);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shk.SetValue(L"NameString",
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) szUserAppName,
|
||
|
|
sizeof(WCHAR) * (lstrlen(szUserAppName)+ 1));
|
||
|
|
|
||
|
|
// MUI: MMC: Use localized snap-in name for MUI.
|
||
|
|
if (L'\0' != szMUIUserAppName[0])
|
||
|
|
{
|
||
|
|
hr = shk.SetValue(L"NameStringIndirect",
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) szMUIUserAppName,
|
||
|
|
sizeof (WCHAR) * (lstrlen(szMUIUserAppName) + 1));
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = shk.Create(L"NodeTypes", &shkTemp);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkTemp.Create(szUser_RSOP_Namespace, &shkServer);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
shkServer.Close();
|
||
|
|
shkTemp.Close();
|
||
|
|
shk.Close();
|
||
|
|
shkCLSID.Close();
|
||
|
|
|
||
|
|
hr = shkCLSID.Open(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\NodeTypes", KEY_WRITE);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkCLSID.Create(szUser_RSOP_Namespace, &shk);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
shk.Close();
|
||
|
|
|
||
|
|
StringFromGUID2 (NODEID_RSOPUser, szGUID, 50);
|
||
|
|
|
||
|
|
hr = shkCLSID.Create(szGUID, &shk);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shk.Create(L"Extensions", &shkServer);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkServer.Create(L"NameSpace", &shkTemp);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkTemp.SetValue(szCLSID_RSOP_Snapin,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) szUserAppName,
|
||
|
|
sizeof(WCHAR) * (lstrlen(szUserAppName)+ 1));
|
||
|
|
shkTemp.Close();
|
||
|
|
shkServer.Close();
|
||
|
|
shk.Close();
|
||
|
|
shkCLSID.Close();
|
||
|
|
} while (0);
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
/////////////////////////////////////////////////////////////////////////////
|
||
|
|
// DllUnregisterServer - Removes entries from the system registry
|
||
|
|
|
||
|
|
STDAPI DllUnregisterServer(void)
|
||
|
|
{
|
||
|
|
_Module.UnregisterServer();
|
||
|
|
|
||
|
|
HKEY hkey;
|
||
|
|
CString sz;
|
||
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\SnapIns\\", 0, KEY_WRITE, &hkey);
|
||
|
|
RegDeleteTree(hkey, (LPOLESTR)((LPCOLESTR)szCLSID_Snapin));
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\NodeTypes\\", 0, KEY_WRITE, &hkey);
|
||
|
|
RegDeleteTree(HKEY_LOCAL_MACHINE, (LPOLESTR)((LPCOLESTR)szUser_Namespace));
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
WCHAR szGUID[50];
|
||
|
|
sz = L"Software\\Microsoft\\MMC\\NodeTypes\\";
|
||
|
|
StringFromGUID2 (NODEID_User, szGUID, 50);
|
||
|
|
sz += szGUID;
|
||
|
|
sz += L"\\Extensions\\NameSpace";
|
||
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, sz, 0, KEY_WRITE, &hkey);
|
||
|
|
RegDeleteValue(hkey, szCLSID_Snapin);
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
|
||
|
|
// unregister RSOP nodes
|
||
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\SnapIns\\", 0, KEY_WRITE, &hkey);
|
||
|
|
RegDeleteTree(hkey, (LPOLESTR)((LPCOLESTR)szCLSID_RSOP_Snapin));
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, L"Software\\Microsoft\\MMC\\NodeTypes\\", 0, KEY_WRITE, &hkey);
|
||
|
|
RegDeleteTree(HKEY_LOCAL_MACHINE, (LPOLESTR)((LPCOLESTR)szUser_RSOP_Namespace));
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
sz = L"Software\\Microsoft\\MMC\\NodeTypes\\";
|
||
|
|
StringFromGUID2 (NODEID_RSOPUser, szGUID, 50);
|
||
|
|
sz += szGUID;
|
||
|
|
sz += L"\\Extensions\\NameSpace";
|
||
|
|
RegOpenKeyEx(HKEY_LOCAL_MACHINE, sz, 0, KEY_WRITE, &hkey);
|
||
|
|
RegDeleteValue(hkey, szCLSID_RSOP_Snapin);
|
||
|
|
RegCloseKey(hkey);
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
//+--------------------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: RegisterInterface
|
||
|
|
//
|
||
|
|
// Synopsis: Add the registry entries required for an interface.
|
||
|
|
//
|
||
|
|
// Arguments: [pshkInterface] - handle to CLSID\Interface key
|
||
|
|
// [wszInterfaceGUID] - GUID of interface to add
|
||
|
|
// [wszInterfaceName] - human-readable name of interface
|
||
|
|
// [wszNumMethods] - number of methods (including inherited)
|
||
|
|
// [wszProxyCLSID] - GUID of dll containing proxy/stubs
|
||
|
|
//
|
||
|
|
// Returns: HRESULT
|
||
|
|
//
|
||
|
|
// History: 3-31-1997 DavidMun Created
|
||
|
|
// 5-09-1997 SteveBl Modified for use with AppMgr
|
||
|
|
//
|
||
|
|
//---------------------------------------------------------------------------
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
RegisterInterface(
|
||
|
|
CSafeReg *pshkInterface,
|
||
|
|
LPWSTR wszInterfaceGUID,
|
||
|
|
LPWSTR wszInterfaceName,
|
||
|
|
LPWSTR wszNumMethods,
|
||
|
|
LPWSTR wszProxyCLSID)
|
||
|
|
{
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
CSafeReg shkIID;
|
||
|
|
CSafeReg shkNumMethods;
|
||
|
|
CSafeReg shkProxy;
|
||
|
|
|
||
|
|
do
|
||
|
|
{
|
||
|
|
hr = pshkInterface->Create(wszInterfaceGUID, &shkIID);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkIID.SetValue(NULL,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *) wszInterfaceName,
|
||
|
|
sizeof(WCHAR) * (lstrlen(wszInterfaceName) + 1));
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkIID.Create(L"NumMethods", &shkNumMethods);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkNumMethods.SetValue(NULL,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *)wszNumMethods,
|
||
|
|
sizeof(WCHAR) * (lstrlen(wszNumMethods) + 1));
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkIID.Create(L"ProxyStubClsid32", &shkProxy);
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
|
||
|
|
hr = shkProxy.SetValue(NULL,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *)wszProxyCLSID,
|
||
|
|
sizeof(WCHAR) * (lstrlen(wszProxyCLSID) + 1));
|
||
|
|
BREAK_ON_FAIL_HRESULT(hr);
|
||
|
|
} while (0);
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|