Shell Extensions for .NET Assemblies

// Modules.h: Module class wrappers for dynamic runtime linking.

#pragma once

#include <windows.h>

struct ENTRYPOINT
{
	BOOL fLoaded;
	BOOL fFound;
	LPVOID pfn;
} EMPTY_ENTRYPOINT = { FALSE, FALSE, NULL };
typedef ENTRYPOINT *LPENTRYPOINT;

class CModule
{
protected:

	CModule(LPCTSTR pszModuleFileName) throw()
	{
		m_hModule = LoadLibrary(pszModuleFileName);
	}

	~CModule() throw()
	{
		if (m_hModule)
			FreeLibrary(m_hModule);
	}

	template<typename T>
	T GetProc(LPCSTR pszProcName, LPENTRYPOINT proc)
	{
		// If not already loaded, get the proc address.
		if (m_hModule && !proc->fLoaded)
		{
			proc->pfn = GetProcAddress(m_hModule, pszProcName);
			if (proc->pfn)
				proc->fFound = TRUE;

			proc->fLoaded = TRUE;
		}
		else if (!m_hModule)
		{
			// mscoree.dll was not found.
			throw ERROR_FILE_NOT_FOUND;
		}
		
		if (!proc->fFound)
		{
			// The function isn't defined/exported.
			throw ERROR_INVALID_FUNCTION;
		}

		return static_cast<T>(proc->pfn);
	}

private:

	HMODULE m_hModule;
};

class CMscoreeModule :
	public CModule
{
public:

	CMscoreeModule() throw() :
		m_fnStrongNameFreeBuffer(EMPTY_ENTRYPOINT),
		m_fnStrongNameTokenFromAssembly(EMPTY_ENTRYPOINT),
		m_fnStrongNameTokenFromAssemblyEx(EMPTY_ENTRYPOINT),
		CModule(TEXT("mscoree.dll"))
	{
	}

	VOID StrongNameFreeBuffer(LPBYTE pbMemory)
	{
		STRONGNAMEFREEBUFFER pfn = GetProc<STRONGNAMEFREEBUFFER>("StrongNameFreeBuffer", &m_fnStrongNameFreeBuffer);
		return (pfn)(pbMemory);
	}

	BOOLEAN StrongNameTokenFromAssembly(LPCWSTR wszFilePath, LPBYTE* ppbStrongNameToken, ULONG* pcbStrongNameToken)
	{
		STRONGNAMETOKENFROMASSEMBLY pfn = GetProc<STRONGNAMETOKENFROMASSEMBLY>("StrongNameTokenFromAssembly", &m_fnStrongNameTokenFromAssembly);
		return (pfn)(wszFilePath, ppbStrongNameToken, pcbStrongNameToken);
	}

	BOOLEAN StrongNameTokenFromAssemblyEx(LPCWSTR wszFilePath, LPBYTE* ppbStrongNameToken, ULONG* pcbStrongNameToken,
		LPBYTE* ppbPublicKeyBlob, ULONG* pcbPublicKeyBlob)
	{
		STRONGNAMETOKENFROMASSEMBLYEX pfn = GetProc<STRONGNAMETOKENFROMASSEMBLYEX>("StrongNameTokenFromAssemblyEx", &m_fnStrongNameTokenFromAssemblyEx);
		return (pfn)(wszFilePath, ppbStrongNameToken, pcbStrongNameToken, ppbPublicKeyBlob, pcbPublicKeyBlob);
	}

private:

	typedef VOID (WINAPI *STRONGNAMEFREEBUFFER)(LPBYTE);
	typedef BOOLEAN (WINAPI *STRONGNAMETOKENFROMASSEMBLY)(LPCWSTR, LPBYTE*, ULONG*);
	typedef BOOLEAN (WINAPI *STRONGNAMETOKENFROMASSEMBLYEX)(LPCWSTR, LPBYTE*, ULONG*, LPBYTE*, ULONG*);

	ENTRYPOINT m_fnStrongNameFreeBuffer;
	ENTRYPOINT m_fnStrongNameTokenFromAssembly;
	ENTRYPOINT m_fnStrongNameTokenFromAssemblyEx;

};
CMscoreeModule g_mscoree;

//class CKernel32Module :
//	public CModule
//{
//public:
//
//	CKernel32Module() :
//	  m_fnGetUserDefaultUILanguage(EMPTY_ENTRYPOINT),
//	  CModule(TEXT("kernel32.dll"))
//	{
//	}
//
//	LANGID GetUserDefaultUILanguage(void)
//	{
//		GETUSERDEFAULTUILANGUAGE pfn = GetProc<GETUSERDEFAULTUILANGUAGE>("GetUserDefaultUILanguage", &m_fnGetUserDefaultUILanguage);
//		return (pfn)();
//	}
//
//private:
//
//	typedef LANGID (WINAPI *GETUSERDEFAULTUILANGUAGE)(void);
//
//	ENTRYPOINT m_fnGetUserDefaultUILanguage;
//};
//CKernel32Module g_kernel32;