144 lines
3.8 KiB
C++
144 lines
3.8 KiB
C++
/**
|
|
* This DLL hosts CLR4 environment from within a native binary. This way it is possible to
|
|
* call .NET APIs from an unmanaged runtime.
|
|
*
|
|
* Mariusz Banach, mgeeky, 19'
|
|
*
|
|
**/
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <metahost.h>
|
|
#include <stdio.h>
|
|
|
|
#pragma comment(lib, "mscoree.lib")
|
|
|
|
//////////////////////////////////////////////////
|
|
//
|
|
// Specify below .NET assembly, main class to instantiate and parameters to pass there.
|
|
|
|
namespace CustomAssemblyParameters {
|
|
LPCWSTR AssemblyName = L"%TEMP%\\ClmDisableAssembly.dll";
|
|
LPCWSTR TypeName = L"ClmDisableAssembly.ClmDisableAssembly";
|
|
LPCWSTR MethodName = L"Start";
|
|
LPCWSTR Argument = L"(called from native CLR host)";
|
|
}
|
|
|
|
//////////////////////////////////////////////////
|
|
|
|
#ifdef _DEBUG
|
|
# define msg(x) MessageBoxW(nullptr, x, L"LoadCLRFromNativeDLL", 0)
|
|
#else
|
|
# define msg(x) ((void)0)
|
|
#endif
|
|
|
|
void DoProcessAttach()
|
|
{
|
|
ICLRMetaHost *metaHost = nullptr;
|
|
ICLRRuntimeInfo *runtimeInfo = nullptr;
|
|
ICLRRuntimeHost *runtimeHost = nullptr;
|
|
IEnumUnknown *runtime = nullptr;
|
|
IUnknown *enumRuntime = nullptr;
|
|
|
|
LPWSTR frameworkName = nullptr;
|
|
DWORD bytes = 2048;
|
|
DWORD result = 0;
|
|
|
|
if (CLRCreateInstance(
|
|
CLSID_CLRMetaHost,
|
|
IID_ICLRMetaHost,
|
|
reinterpret_cast<LPVOID*>(&metaHost)
|
|
) != S_OK) {
|
|
msg(L"FAIL: Could not create MetaHost CLR instance.");
|
|
return;
|
|
}
|
|
|
|
if (!metaHost || (metaHost->EnumerateInstalledRuntimes(
|
|
&runtime
|
|
) != S_OK)) {
|
|
msg(L"FAIL: Cannot enumerate installed runtimes.");
|
|
return;
|
|
}
|
|
|
|
if (!runtime) {
|
|
msg(L"FAIL: Could not find installed runtimes.");
|
|
return;
|
|
}
|
|
|
|
frameworkName = reinterpret_cast<LPWSTR>(LocalAlloc(
|
|
LPTR,
|
|
bytes
|
|
));
|
|
if (!frameworkName) {
|
|
msg(L"FAIL: could not allocate 2048 bytes for framework name buffer.");
|
|
return;
|
|
}
|
|
|
|
while (runtime->Next(1, &enumRuntime, 0) == S_OK) {
|
|
if (enumRuntime && (enumRuntime->QueryInterface<ICLRRuntimeInfo>(&runtimeInfo) == S_OK)) {
|
|
if (runtimeInfo != nullptr) {
|
|
runtimeInfo->GetVersionString(frameworkName, &bytes);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (runtimeInfo == nullptr || (runtimeInfo->GetInterface(
|
|
CLSID_CLRRuntimeHost,
|
|
IID_ICLRRuntimeHost,
|
|
reinterpret_cast<LPVOID*>(&runtimeHost)
|
|
) != S_OK)) {
|
|
msg(L"FAIL: Could not get CLRRuntimeHost interface's reference.");
|
|
return;
|
|
}
|
|
|
|
if (runtimeHost == nullptr) {
|
|
msg(L"FAIL: Could not obtain reference to CLRRuntimeHost.");
|
|
return;
|
|
}
|
|
|
|
runtimeHost->Start();
|
|
|
|
WCHAR assemblyPath[1024] = L"";
|
|
ExpandEnvironmentStringsW(CustomAssemblyParameters::AssemblyName, assemblyPath, _countof(assemblyPath));
|
|
LPCWSTR assemblyPathPtr = assemblyPath;
|
|
|
|
HRESULT hres = runtimeHost->ExecuteInDefaultAppDomain(
|
|
assemblyPathPtr,
|
|
CustomAssemblyParameters::TypeName,
|
|
CustomAssemblyParameters::MethodName,
|
|
CustomAssemblyParameters::Argument,
|
|
&result
|
|
);
|
|
if (hres != S_OK) {
|
|
wchar_t msgbuf[1024] = L"";
|
|
swprintf_s(msgbuf, L"FAIL: Could not invoke custom .NET assembly, instantiate it's type or invoke a method. HRESULT = 0x%08x . Assembly path: '%s'", hres, assemblyPath);
|
|
msg(msgbuf);
|
|
}
|
|
|
|
//runtimeHost->Stop();
|
|
//runtimeHost->Release();
|
|
runtimeInfo->Release();
|
|
metaHost->Release();
|
|
}
|
|
|
|
BOOLEAN WINAPI DllMain(
|
|
IN HINSTANCE /*hDllHandle*/,
|
|
IN DWORD nReason,
|
|
IN LPVOID /*Reserved*/
|
|
)
|
|
{
|
|
switch (nReason)
|
|
{
|
|
case DLL_PROCESS_ATTACH:
|
|
{
|
|
DoProcessAttach();
|
|
break;
|
|
}
|
|
case DLL_PROCESS_DETACH:
|
|
case DLL_THREAD_ATTACH:
|
|
case DLL_THREAD_DETACH:
|
|
break;
|
|
}
|
|
return TRUE;
|
|
}
|