From 3678a22c90f9543ca60de8ca395ccc6c01ba52a6 Mon Sep 17 00:00:00 2001
From: "Dr. Chat" <arkolbed@gmail.com>
Date: Wed, 2 Dec 2015 20:11:19 -0600
Subject: [PATCH] Use TLS to store the current thread.

---
 src/xenia/base/threading.h      |  2 +-
 src/xenia/base/threading_win.cc | 15 ++++++++++++---
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/xenia/base/threading.h b/src/xenia/base/threading.h
index 3278f9fcb..62fb5e815 100644
--- a/src/xenia/base/threading.h
+++ b/src/xenia/base/threading.h
@@ -345,7 +345,7 @@ class Thread : public WaitHandle {
   // within that thread.
   static std::unique_ptr<Thread> Create(CreationParameters params,
                                         std::function<void()> start_routine);
-  static std::unique_ptr<Thread> GetCurrentThread();
+  static Thread* GetCurrentThread();
 
   // Ends the calling thread.
   // No destructors are called, and this function does not return.
diff --git a/src/xenia/base/threading_win.cc b/src/xenia/base/threading_win.cc
index f6b472cd6..d36ee76e1 100644
--- a/src/xenia/base/threading_win.cc
+++ b/src/xenia/base/threading_win.cc
@@ -426,10 +426,14 @@ class Win32Thread : public Win32Handle<Thread> {
   }
 };
 
+thread_local std::unique_ptr<Win32Thread> current_thread_ = nullptr;
+
 struct ThreadStartData {
   std::function<void()> start_routine;
 };
 DWORD WINAPI ThreadStartRoutine(LPVOID parameter) {
+  current_thread_ = std::make_unique<Win32Thread>(::GetCurrentThread());
+
   auto start_data = reinterpret_cast<ThreadStartData*>(parameter);
   start_data->start_routine();
   delete start_data;
@@ -449,17 +453,22 @@ std::unique_ptr<Thread> Thread::Create(CreationParameters params,
     delete start_data;
     return nullptr;
   }
-  GetThreadId(handle);
+
   return std::make_unique<Win32Thread>(handle);
 }
 
-std::unique_ptr<Thread> Thread::GetCurrentThread() {
+Thread* Thread::GetCurrentThread() {
+  if (current_thread_) {
+    return current_thread_.get();
+  }
+
   HANDLE handle = ::GetCurrentThread();
   if (handle == INVALID_HANDLE_VALUE) {
     return nullptr;
   }
 
-  return std::make_unique<Win32Thread>(handle);
+  current_thread_ = std::make_unique<Win32Thread>(handle);
+  return current_thread_.get();
 }
 
 void Thread::Exit(int exit_code) { ExitThread(exit_code); }