From 7e23d873dc9efc48318c2d4137024bd7f2dc5a62 Mon Sep 17 00:00:00 2001
From: Bartosz Taudul <wolf@nereid.pl>
Date: Sun, 27 Nov 2022 21:48:20 +0100
Subject: [PATCH] Check elevation status on Windows.

---
 profiler/build/win32/Tracy.vcxproj         |  2 ++
 profiler/build/win32/Tracy.vcxproj.filters |  6 ++++
 profiler/src/IsElevated.cpp                | 32 ++++++++++++++++++++++
 profiler/src/IsElevated.hpp                |  6 ++++
 4 files changed, 46 insertions(+)
 create mode 100644 profiler/src/IsElevated.cpp
 create mode 100644 profiler/src/IsElevated.hpp

diff --git a/profiler/build/win32/Tracy.vcxproj b/profiler/build/win32/Tracy.vcxproj
index 834e38c5..83351c3d 100644
--- a/profiler/build/win32/Tracy.vcxproj
+++ b/profiler/build/win32/Tracy.vcxproj
@@ -212,6 +212,7 @@
     <ClCompile Include="..\..\src\ImGuiContext.cpp" />
     <ClCompile Include="..\..\src\imgui\imgui_impl_glfw.cpp" />
     <ClCompile Include="..\..\src\imgui\imgui_impl_opengl3.cpp" />
+    <ClCompile Include="..\..\src\IsElevated.cpp" />
     <ClCompile Include="..\..\src\main.cpp" />
     <ClCompile Include="..\..\src\ResolvService.cpp" />
     <ClCompile Include="..\..\src\RunQueue.cpp" />
@@ -347,6 +348,7 @@
     <ClInclude Include="..\..\src\imgui\imgui_impl_glfw.h" />
     <ClInclude Include="..\..\src\imgui\imgui_impl_opengl3.h" />
     <ClInclude Include="..\..\src\imgui\imgui_impl_opengl3_loader.h" />
+    <ClInclude Include="..\..\src\IsElevated.hpp" />
     <ClInclude Include="..\..\src\ResolvService.hpp" />
     <ClInclude Include="..\..\src\RunQueue.hpp" />
     <ClInclude Include="..\..\src\stb_image.h" />
diff --git a/profiler/build/win32/Tracy.vcxproj.filters b/profiler/build/win32/Tracy.vcxproj.filters
index a97ea0b0..31415380 100644
--- a/profiler/build/win32/Tracy.vcxproj.filters
+++ b/profiler/build/win32/Tracy.vcxproj.filters
@@ -378,6 +378,9 @@
     <ClCompile Include="..\..\..\server\TracyFileselector.cpp">
       <Filter>server</Filter>
     </ClCompile>
+    <ClCompile Include="..\..\src\IsElevated.cpp">
+      <Filter>src</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <ClInclude Include="..\..\..\server\TracyEvent.hpp">
@@ -764,6 +767,9 @@
     <ClInclude Include="..\..\..\server\TracyFileselector.hpp">
       <Filter>server</Filter>
     </ClInclude>
+    <ClInclude Include="..\..\src\IsElevated.hpp">
+      <Filter>src</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <Natvis Include="DebugVis.natvis" />
diff --git a/profiler/src/IsElevated.cpp b/profiler/src/IsElevated.cpp
new file mode 100644
index 00000000..2f257b72
--- /dev/null
+++ b/profiler/src/IsElevated.cpp
@@ -0,0 +1,32 @@
+#include "IsElevated.hpp"
+
+#ifdef _WIN32
+
+#include <windows.h>
+
+bool IsElevated()
+{
+    HANDLE token;
+    if( OpenProcessToken( GetCurrentProcess(), TOKEN_QUERY, &token ) == 0 ) return false;
+
+    TOKEN_ELEVATION te;
+    DWORD sz;
+    if( GetTokenInformation( token, TokenElevation, &te, sizeof( te ), &sz ) == 0 )
+    {
+        CloseHandle( token );
+        return false;
+    }
+
+    bool ret = te.TokenIsElevated;
+    CloseHandle( token );
+    return ret;
+}
+
+#else
+
+bool IsElevated()
+{
+    return false;
+}
+
+#endif
diff --git a/profiler/src/IsElevated.hpp b/profiler/src/IsElevated.hpp
new file mode 100644
index 00000000..35c73984
--- /dev/null
+++ b/profiler/src/IsElevated.hpp
@@ -0,0 +1,6 @@
+#ifndef __ISELEVATED_HPP__
+#define __ISELEVATED_HPP__
+
+bool IsElevated();
+
+#endif