HMODULE LoadCoreClr()
{
errno_t errno = 0;
TCHAR szCoreClrDirectory[MAX_PATH];
DWORD dwCoreClrDirectory = GetEnvironmentVariableW(L"CORECLR_DIR", szCoreClrDirectory, MAX_PATH);
HMODULE hCoreCLRModule = nullptr;
if (dwCoreClrDirectory != 0)
{
WCHAR wszClrPath[MAX_PATH];
wszClrPath[0] = L'\0';
errno = wcscpy_s(wszClrPath, _countof(wszClrPath), szCoreClrDirectory);
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
if (wszClrPath[wcslen(wszClrPath) - 1] != L'\\')
{
errno = wcscat_s(wszClrPath, _countof(wszClrPath), L"\\");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
}
errno = wcscat_s(wszClrPath, _countof(wszClrPath), L"coreclr.dll");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
// Add the core clr directory to the list of dll search paths
AddDllDirectory(szCoreClrDirectory);
// Modify the default dll flags so that dependencies can be found in this path
SetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_DEFAULT_DIRS | LOAD_LIBRARY_SEARCH_USER_DIRS);
// Continue loading as usual
hCoreCLRModule = ::LoadLibraryExW(wszClrPath, NULL, 0);
}
if (hCoreCLRModule == nullptr)
{
// This is used when developing
#if AMD64
hCoreCLRModule = ::LoadLibraryExW(L"..\\..\\..\\artifacts\\build\\ProjectK\\Runtime\\amd64\\coreclr.dll", NULL, 0);
#else
hCoreCLRModule = ::LoadLibraryExW(L"..\\..\\..\\artifacts\\build\\ProjectK\\Runtime\\x86\\coreclr.dll", NULL, 0);
#endif
}
if (hCoreCLRModule == nullptr)
{
// Try the relative location based in install
hCoreCLRModule = ::LoadLibraryExW(L"coreclr.dll", NULL, 0);
}
Finished:
return hCoreCLRModule;
}
extern "C" __declspec(dllexport) HRESULT __stdcall CallApplicationMain(PCALL_APPLICATION_MAIN_DATA data)
{
HRESULT hr = S_OK;
errno_t errno = 0;
FnGetCLRRuntimeHost pfnGetCLRRuntimeHost = nullptr;
ICLRRuntimeHost2* pCLRRuntimeHost = nullptr;
TCHAR szCurrentDirectory[MAX_PATH];
TCHAR szCoreClrDirectory[MAX_PATH];
TCHAR lpCoreClrModulePath[MAX_PATH];
size_t cchTrustedPlatformAssemblies = 0;
LPWSTR pwszTrustedPlatformAssemblies = nullptr;
Win32KDisable();
if (data->runtimeDirectory) {
errno = wcscpy_s(szCurrentDirectory, data->runtimeDirectory);
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
}
else {
GetModuleDirectory(NULL, szCurrentDirectory);
}
HMODULE hCoreCLRModule = LoadCoreClr();
if (!hCoreCLRModule)
{
printf_s("Failed to locate coreclr.dll.\n");
return E_FAIL;
}
// Get the path to the module
DWORD dwCoreClrModulePathSize = GetModuleFileName(hCoreCLRModule, lpCoreClrModulePath, MAX_PATH);
lpCoreClrModulePath[dwCoreClrModulePathSize] = '\0';
GetModuleDirectory(hCoreCLRModule, szCoreClrDirectory);
HMODULE ignoreModule;
// Pin the module - CoreCLR.dll does not support being unloaded.
if (!::GetModuleHandleExW(GET_MODULE_HANDLE_EX_FLAG_PIN, lpCoreClrModulePath, &ignoreModule))
{
printf_s("Failed to pin coreclr.dll.\n");
return E_FAIL;
}
pfnGetCLRRuntimeHost = (FnGetCLRRuntimeHost)::GetProcAddress(hCoreCLRModule, "GetCLRRuntimeHost");
if (!pfnGetCLRRuntimeHost)
{
printf_s("Failed to find export GetCLRRuntimeHost.\n");
return E_FAIL;
}
hr = pfnGetCLRRuntimeHost(IID_ICLRRuntimeHost2, (IUnknown**)&pCLRRuntimeHost);
if (FAILED(hr))
{
printf_s("Failed to get IID_ICLRRuntimeHost2.\n");
return hr;
}
STARTUP_FLAGS dwStartupFlags = (STARTUP_FLAGS)(
STARTUP_FLAGS::STARTUP_LOADER_OPTIMIZATION_SINGLE_DOMAIN |
STARTUP_FLAGS::STARTUP_SINGLE_APPDOMAIN |
STARTUP_FLAGS::STARTUP_SERVER_GC
);
pCLRRuntimeHost->SetStartupFlags(dwStartupFlags);
// Authenticate with either CORECLR_HOST_AUTHENTICATION_KEY or CORECLR_HOST_AUTHENTICATION_KEY_NONGEN
hr = pCLRRuntimeHost->Authenticate(CORECLR_HOST_AUTHENTICATION_KEY);
if (FAILED(hr))
{
printf_s("Failed to Authenticate().\n");
return hr;
}
hr = pCLRRuntimeHost->Start();
if (FAILED(hr))
{
printf_s("Failed to Start().\n");
return hr;
}
const wchar_t* property_keys[] =
{
// Allowed property names:
// APPBASE
// - The base path of the application from which the exe and other assemblies will be loaded
L"APPBASE",
//
// TRUSTED_PLATFORM_ASSEMBLIES
// - The list of complete paths to each of the fully trusted assemblies
L"TRUSTED_PLATFORM_ASSEMBLIES",
//
// APP_PATHS
// - The list of paths which will be probed by the assembly loader
L"APP_PATHS",
//
// APP_NI_PATHS
// - The list of additional paths that the assembly loader will probe for ngen images
//
// NATIVE_DLL_SEARCH_DIRECTORIES
// - The list of paths that will be probed for native DLLs called by PInvoke
//
};
cchTrustedPlatformAssemblies = TRUSTED_PLATFORM_ASSEMBLIES_STRING_BUFFER_SIZE_CCH;
pwszTrustedPlatformAssemblies = (LPWSTR)calloc(cchTrustedPlatformAssemblies+1, sizeof(WCHAR));
if (pwszTrustedPlatformAssemblies == NULL)
{
goto Finished;
}
pwszTrustedPlatformAssemblies[0] = L'\0';
// Try native images first
if (!GetTrustedPlatformAssembliesList(szCoreClrDirectory, true, pwszTrustedPlatformAssemblies, cchTrustedPlatformAssemblies))
{
if (!GetTrustedPlatformAssembliesList(szCoreClrDirectory, false, pwszTrustedPlatformAssemblies, cchTrustedPlatformAssemblies))
{
printf_s("Failed to find files in the coreclr directory\n");
return E_FAIL;
}
}
// Add the assembly containing the app domain manager to the trusted list
errno = wcscat_s(pwszTrustedPlatformAssemblies, cchTrustedPlatformAssemblies, szCurrentDirectory);
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
errno = wcscat_s(pwszTrustedPlatformAssemblies, cchTrustedPlatformAssemblies, L"kre.coreclr.managed.dll");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
//wstring appPaths(szCurrentDirectory);
WCHAR wszAppPaths[MAX_PATH];
wszAppPaths[0] = L'\0';
errno = wcscat_s(wszAppPaths, _countof(wszAppPaths), szCurrentDirectory);
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
errno = wcscat_s(wszAppPaths, _countof(wszAppPaths), L";");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
errno = wcscat_s(wszAppPaths, _countof(wszAppPaths), szCoreClrDirectory);
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
errno = wcscat_s(wszAppPaths, _countof(wszAppPaths), L";");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
const wchar_t* property_values[] = {
// APPBASE
data->applicationBase,
// TRUSTED_PLATFORM_ASSEMBLIES
pwszTrustedPlatformAssemblies,
// APP_PATHS
wszAppPaths,
};
DWORD domainId;
DWORD dwFlagsAppDomain =
APPDOMAIN_ENABLE_PLATFORM_SPECIFIC_APPS |
APPDOMAIN_ENABLE_PINVOKE_AND_CLASSIC_COMINTEROP;
LPCWSTR szAssemblyName = L"kre.coreclr.managed, Version=0.1.0.0";
LPCWSTR szEntryPointTypeName = L"DomainManager";
LPCWSTR szMainMethodName = L"Execute";
int nprops = sizeof(property_keys) / sizeof(wchar_t*);
hr = pCLRRuntimeHost->CreateAppDomainWithManager(
L"kre.coreclr.managed",
dwFlagsAppDomain,
NULL,
NULL,
nprops,
property_keys,
property_values,
&domainId);
if (FAILED(hr))
{
wprintf_s(L"TPA %d %S\n", wcslen(pwszTrustedPlatformAssemblies), pwszTrustedPlatformAssemblies);
wprintf_s(L"AppPaths %S\n", wszAppPaths);
printf_s("Failed to create app domain (%x).\n", hr);
return hr;
}
HostMain pHostMain;
hr = pCLRRuntimeHost->CreateDelegate(
domainId,
szAssemblyName,
szEntryPointTypeName,
szMainMethodName,
(INT_PTR*)&pHostMain);
if (FAILED(hr))
{
printf_s("Failed to create main delegate (%x).\n", hr);
return hr;
}
SetEnvironmentVariable(L"KRE_FRAMEWORK", L"aspnetcore50");
// Call main
data->exitcode = pHostMain(data->argc, data->argv);
pCLRRuntimeHost->UnloadAppDomain(domainId, true);
pCLRRuntimeHost->Stop();
Finished:
if (pwszTrustedPlatformAssemblies != NULL)
{
free(pwszTrustedPlatformAssemblies);
pwszTrustedPlatformAssemblies = NULL;
}
if (FAILED(hr))
{
return hr;
}
else
{
return S_OK;
}
}
HMODULE LoadCoreClr()
{
errno_t errno = 0;
bool fSuccess = true;
TCHAR szTrace[1] = {};
DWORD dwRet = GetEnvironmentVariableW(L"KRE_TRACE", szTrace, 1);
bool m_fVerboseTrace = dwRet > 0;
LPWSTR rgwzOSLoaderModuleNames[] = {
L"api-ms-win-core-libraryloader-l1-1-1.dll",
L"kernel32.dll",
NULL
};
LPWSTR rgwszModuleFileName = NULL;
DWORD dwModuleFileName = 0;
HMODULE hOSLoaderModule = nullptr;
// Note: need to keep as ASCII as GetProcAddress function takes ASCII params
LPCSTR pszAddDllDirectoryName = "AddDllDirectory";
FnAddDllDirectory pFnAddDllDirectory = nullptr;
LPCSTR pszSetDefaultDllDirectoriesName = "SetDefaultDllDirectories";
FnSetDefaultDllDirectories pFnSetDefaultDllDirectories = nullptr;
TCHAR szCoreClrDirectory[MAX_PATH];
DWORD dwCoreClrDirectory = GetEnvironmentVariableW(L"CORECLR_DIR", szCoreClrDirectory, MAX_PATH);
HMODULE hCoreCLRModule = nullptr;
if (dwCoreClrDirectory != 0)
{
WCHAR wszClrPath[MAX_PATH];
wszClrPath[0] = L'\0';
errno = wcscpy_s(wszClrPath, _countof(wszClrPath), szCoreClrDirectory);
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
if (wszClrPath[wcslen(wszClrPath) - 1] != L'\\')
{
errno = wcscat_s(wszClrPath, _countof(wszClrPath), L"\\");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
}
errno = wcscat_s(wszClrPath, _countof(wszClrPath), L"coreclr.dll");
CHECK_RETURN_VALUE_FAIL_EXIT_VIA_FINISHED(errno);
//Scan through module name list looking for a valid module that has first DLL export
rgwszModuleFileName = rgwzOSLoaderModuleNames[dwModuleFileName];
while (rgwszModuleFileName != NULL)
{
fSuccess = KlrLoadLibraryExWAndGetProcAddress(
rgwszModuleFileName,
pszAddDllDirectoryName,
&hOSLoaderModule,
(FARPROC*)&pFnAddDllDirectory);
if (fSuccess)
break;
dwModuleFileName++;
rgwszModuleFileName = rgwzOSLoaderModuleNames[dwModuleFileName];
}
if (!hOSLoaderModule || !pFnAddDllDirectory)
{
fSuccess = false;
goto Finished;
}
//Find the second DLL export
pFnSetDefaultDllDirectories = (FnSetDefaultDllDirectories)::GetProcAddress(hOSLoaderModule, pszSetDefaultDllDirectoriesName);
if (!pFnSetDefaultDllDirectories)
{
if (m_fVerboseTrace)
::wprintf_s(L"Failed to find function %S in %s\n", pszSetDefaultDllDirectoriesName, rgwszModuleFileName);
fSuccess = false;
goto Finished;
}
//Verify HANDLE and two DLL exports are valid before proceeding
if (!hOSLoaderModule || !pFnAddDllDirectory || !pFnSetDefaultDllDirectories)
{
fSuccess = false;
goto Finished;
}
pFnAddDllDirectory(szCoreClrDirectory);
// Modify the default dll flags so that dependencies can be found in this path
pFnSetDefaultDllDirectories(LOAD_LIBRARY_SEARCH_DEFAULT_DIRS | LOAD_LIBRARY_SEARCH_USER_DIRS);
fSuccess = true;
// Continue loading as usual
hCoreCLRModule = ::LoadLibraryExW(wszClrPath, NULL, 0);
}
if (hCoreCLRModule == nullptr)
{
// This is used when developing
#if AMD64
hCoreCLRModule = ::LoadLibraryExW(L"..\\..\\..\\artifacts\\build\\ProjectK\\Runtime\\amd64\\coreclr.dll", NULL, 0);
#else
hCoreCLRModule = ::LoadLibraryExW(L"..\\..\\..\\artifacts\\build\\ProjectK\\Runtime\\x86\\coreclr.dll", NULL, 0);
#endif
}
if (hCoreCLRModule == nullptr)
{
// Try the relative location based in install
hCoreCLRModule = ::LoadLibraryExW(L"coreclr.dll", NULL, 0);
}
Finished:
return hCoreCLRModule;
}
