dllregisterserverimpl.cpp

来自「firtext搜索引擎源码」· C++ 代码 · 共 196 行

#include "com/Com.h"
#include "com/comerror.h"

#undef FX_STDAPI
#undef FX_STDAPIV
#undef FX_STDMETHOD
#undef FX_STDMETHODCALLTYPE
#undef FX_STDMETHODIMP
#undef FX_WINOLEAPI
#undef FX_WINOLEAPI_
#undef FX_WINOLEAUTAPI
#undef FX_WINOLEAUTAPI_

#define STRICT
#include <windows.h>

extern HINSTANCE hDllInstance;

static firtex::com::FX_HRESULT RegSetValueW( HKEY hKey, LPCWSTR lpValueName, LPCWSTR lpValue )
{
	return ::RegSetValueW( hKey, lpValueName, REG_SZ, lpValue, (DWORD)wcslen(lpValue) );
}

static firtex::com::FX_HRESULT SetThreadingModel( HKEY hKey, firtex::com::THREADINGMODEL tm )
{
	// look up the value
	wchar_t const* text = 0;
	switch (tm) {
	case firtex::com::ThreadingModelApartment: text = L"Apartment"; break;
	case firtex::com::ThreadingModelFree:  text = L"Free"; break;
	case firtex::com::ThreadingModelBoth:  text = L"Both"; break;
	case firtex::com::ThreadingModelNeutral:text = L"Neutral"; break;
	};
	if ( !text ) return ERROR_SUCCESS;

	// open up the sub-key
	HKEY hSubKey = 0;
	firtex::com::FX_HRESULT hr = RegOpenKeyExW( hKey, 
		L"InprocServer32", 
		0, 
		KEY_ALL_ACCESS, 
		&hSubKey );
	if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);

	// set the subvalue
	hr = ::RegSetValueExW( hSubKey, L"ThreadingModel", 0, REG_SZ, 
		reinterpret_cast<CONST byte*>(text), (DWORD)wcslen(text)*sizeof(*text) );
	RegCloseKey( hSubKey );
	return hr;
}

firtex::com::FX_HRESULT FX_STDAPICALLTYPE firtex::com::DllRegisterServerImpl(firtex::com::FX_REFCLSID clsid_, wchar_t const* name, THREADINGMODEL tm, wchar_t const* VersionIndependentProgId, wchar_t const* ProgId )
{
	// First, we are going to need the DLL filename
	wchar_t filename[MAX_PATH+1];
	firtex::com::FX_HRESULT hr = GetModuleFileNameW(hDllInstance,filename,sizeof(filename)/sizeof(*filename));
	if ( hr == 0 ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,GetLastError());

	// need string from clsid
	wchar_t clsid[64];
	int clsid_len = FX_StringFromGUID2( clsid_, clsid, sizeof(clsid)/sizeof(*clsid) );
	clsid_len -= 1;

	// setup version independent prog id key
	if ( VersionIndependentProgId )
	{
		HKEY hKey = 0;
		DWORD dw;
		firtex::com::FX_HRESULT hr = RegCreateKeyExW( HKEY_CLASSES_ROOT, 
			VersionIndependentProgId, 
			0, 
			NULL, 
			REG_OPTION_NON_VOLATILE,
			KEY_ALL_ACCESS,
			NULL,
			&hKey,
			&dw );
		if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);
		RegSetValueW(	hKey, NULL, REG_SZ, name, (DWORD)wcslen(name) );
		RegSetValueW( hKey, L"CLSID", REG_SZ, clsid, clsid_len );
		if ( ProgId )
			RegSetValueW( hKey, L"CurVer", REG_SZ, ProgId, (DWORD)wcslen(ProgId) );
		RegCloseKey( hKey );
	}
	// setup prog id key
	if ( ProgId )
	{
		HKEY hKey = 0;
		DWORD dw;
		firtex::com::FX_HRESULT hr = RegCreateKeyExW( HKEY_CLASSES_ROOT, 
			ProgId, 
			0, 
			NULL, 
			REG_OPTION_NON_VOLATILE,
			KEY_ALL_ACCESS,
			NULL,
			&hKey,
			&dw );
		if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);
		RegSetValueW( hKey, NULL, REG_SZ, name, (DWORD)wcslen(name) );
		RegSetValueW( hKey, L"CLSID", REG_SZ, clsid, clsid_len );
		RegCloseKey( hKey );
	}

	// construct key path
	wchar_t subkey[6+sizeof(clsid)] = L"CLSID\\";
	wcscat( subkey, clsid );
	// setup CLSID key
	HKEY hKey = 0;
	DWORD dw;
	hr = RegCreateKeyExW( HKEY_CLASSES_ROOT, 
		subkey, 
		0, 
		NULL, 
		REG_OPTION_NON_VOLATILE,
		KEY_ALL_ACCESS,
		NULL,
		&hKey,
		&dw );
	if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);
	RegSetValueW( hKey, NULL, REG_SZ, name, (DWORD)wcslen(name) );
	RegSetValueW( hKey, L"InprocServer32", filename );
	SetThreadingModel( hKey, tm );
	if ( ProgId )
		RegSetValueW( hKey, L"ProgID", ProgId );
	if ( VersionIndependentProgId )
		RegSetValueW( hKey, L"VersionIndependentProgID", VersionIndependentProgId );
	RegCloseKey( hKey );

	return FX_S_OK;
}

static firtex::com::FX_HRESULT FX_STDAPICALLTYPE MyRegDeleteKey( HKEY hKey, wchar_t const* lpSubKey )
{
	// open sub key, to check for children
	HKEY hSubKey = 0;
	firtex::com::FX_HRESULT hr = RegOpenKeyExW(hKey,lpSubKey,0,KEY_ALL_ACCESS,&hSubKey);
	if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);

	// first, delete all sub keys
	wchar_t name[MAX_PATH];
	DWORD   name_len = sizeof(name)/sizeof(*name);
	hr = RegEnumKeyExW( hSubKey, 0, name, &name_len, 0, NULL, NULL, NULL );
	while ( hr == ERROR_SUCCESS )
	{
		// delete the subkey
		hr = MyRegDeleteKey( hSubKey, name );
		if ( FX_FAILED(hr) ) return hr;
		// get next subkey
		DWORD   name_len = sizeof(name)/sizeof(*name);
		hr = RegEnumKeyExW( hSubKey, 0, name, &name_len, 0, NULL, NULL, NULL );
	}
	if ( hr != ERROR_NO_MORE_ITEMS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);;

	// close the key, and then delete
	RegCloseKey( hSubKey );
	hr = RegDeleteKeyW( hKey, lpSubKey );
	if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);

	return FX_S_OK;
}

firtex::com::FX_HRESULT FX_STDAPICALLTYPE firtex::com::DllUnregisterServerImpl(firtex::com::FX_REFCLSID clsid_)
{
	// first extract prog ids
	wchar_t	progid[MAX_PATH] = L"";
	wchar_t versionindependentprogid[MAX_PATH] = L"";

	// need string from clsid
	wchar_t clsid[64];
	FX_StringFromGUID2( clsid_, clsid, sizeof(clsid)/sizeof(*clsid) );
	// construct key path
	wchar_t subkey[6+sizeof(clsid)] = L"CLSID\\";
	wcscat( subkey, clsid );

	HKEY hKey = 0;
	firtex::com::FX_HRESULT hr = RegOpenKeyExW(HKEY_CLASSES_ROOT,subkey,0,KEY_ALL_ACCESS,&hKey);
	if ( hr != ERROR_SUCCESS ) return MAKE_HRESULT(SEVERITY_ERROR,FACILITY_WIN32,hr);

	// extract prog ids
	LONG lng = sizeof(progid)/sizeof(*progid);
	hr = RegQueryValueW(hKey,L"ProgID",progid,&lng);
	if ( hr != ERROR_SUCCESS ) progid[0] = 0;
	lng = sizeof(versionindependentprogid)/sizeof(*versionindependentprogid);
	hr = RegQueryValueW(hKey,L"VersionIndependentProgID",versionindependentprogid,&lng);
	if ( hr != ERROR_SUCCESS ) progid[0] = 0;

	// delete the extra info
	if ( progid[0] ) 
		MyRegDeleteKey( HKEY_CLASSES_ROOT, progid );
	if ( versionindependentprogid[0] ) 
		MyRegDeleteKey( HKEY_CLASSES_ROOT, versionindependentprogid );

	return MyRegDeleteKey( HKEY_CLASSES_ROOT, subkey );
}