Thread registry (#4518)

kotlin-native/runtime/src/mm/cpp/Memory.cpp

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2010-2020 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the LICENSE file.
+ */
+
+#include "Memory.h"
+
+#include "ThreadData.hpp"
+#include "ThreadRegistry.hpp"
+#include "Utils.hpp"
+
+using namespace kotlin;
+
+extern "C" struct MemoryState {
+    mm::ThreadRegistry::Node data;
+    // Do not add any other fields: this struct is just a wrapper around ThreadDataNode.
+};
+
+namespace {
+
+ALWAYS_INLINE MemoryState* ToMemoryState(mm::ThreadRegistry::Node* data) {
+    return wrapper_cast(MemoryState, data, data);
+}
+
+ALWAYS_INLINE mm::ThreadRegistry::Node* FromMemoryState(MemoryState* state) {
+    return &state->data;
+}
+
+} // namespace
+
+extern "C" MemoryState* InitMemory(bool firstRuntime) {
+    return ToMemoryState(mm::ThreadRegistry::Instance().RegisterCurrentThread());
+}
+
+extern "C" void DeinitMemory(MemoryState* state, bool destroyRuntime) {
+    mm::ThreadRegistry::Instance().Unregister(FromMemoryState(state));
+}