1 files changed, 64 insertions, 0 deletions
diff --git a/src/ToolBox/superpmi/superpmi-shim-collector/jithost.cpp b/src/ToolBox/superpmi/superpmi-shim-collector/jithost.cpp
new file mode 100644
index 0000000000..5e63a76be7
--- /dev/null
+++ b/src/ToolBox/superpmi/superpmi-shim-collector/jithost.cpp
@@ -0,0 +1,64 @@
+//
+// Copyright (c) Microsoft. All rights reserved.
+// Licensed under the MIT license. See LICENSE file in the project root for full license information.
+//
+
+#include "standardpch.h"
+#include "runtimedetails.h"
+#include "spmiutil.h"
+#include "jithost.h"
+
+JitHost* g_ourJitHost;
+
+JitHost::JitHost(ICorJitHost* wrappedHost, MethodContext* methodContext) : wrappedHost(wrappedHost), mc(methodContext)
+{
+}
+
+void JitHost::setMethodContext(MethodContext* methodContext)
+{
+    this->mc = methodContext;
+}
+
+void* JitHost::allocateMemory(size_t size, bool usePageAllocator)
+{
+    return wrappedHost->allocateMemory(size, usePageAllocator);
+}
+
+void JitHost::freeMemory(void* block, bool usePageAllocator)
+{
+    return wrappedHost->freeMemory(block, usePageAllocator);
+}
+
+int JitHost::getIntConfigValue(const wchar_t* key, int defaultValue)
+{
+    mc->cr->AddCall("getIntConfigValue");
+    int result = wrappedHost->getIntConfigValue(key, defaultValue);
+
+    // The JIT eagerly asks about every config value. If we store all these
+    // queries, it takes almost half the MC file space. So only store the
+    // non-default answers.
+    if (result != defaultValue)
+    {
+        mc->recGetIntConfigValue(key, defaultValue, result);
+    }
+    return result;
+}
+
+const wchar_t* JitHost::getStringConfigValue(const wchar_t* key)
+{
+    mc->cr->AddCall("getStringConfigValue");
+    const wchar_t *result = wrappedHost->getStringConfigValue(key);
+
+    // Don't store null returns, which is the default
+    if (result != nullptr)
+    {
+        mc->recGetStringConfigValue(key, result);
+    }
+    return result;
+}
+
+void JitHost::freeStringConfigValue(const wchar_t* value)
+{
+    mc->cr->AddCall("freeStringConfigValue");
+    wrappedHost->freeStringConfigValue(value);
+}