Use different keys based on process elevation

Author: OneBlue

Handle.h

@@ -18,6 +18,7 @@ namespace libreg
         const Handle<T>& operator=(const Handle&) = delete;
 
         T Get() const;
+        T* Addr();
         void Detach();
         void Close();
         operator T() const;

Handle.hxx

@@ -77,4 +77,10 @@ namespace libreg
     {
         return _value != nullptr;
     }
+
+    template <typename T>
+    inline T* Handle<T>::Addr()
+    {
+      return &_value;
+    }
 }

Tests/TestRegistryOperations.cpp

@@ -16,6 +16,22 @@ TestRegistryOperations::TestRegistryOperations() : Tester("Registry operations")
 
 void TestRegistryOperations::RunImpl()
 {
+
+  bool isElevated = false;
+
+  libreg::Handle<HANDLE> token{ nullptr };
+
+  if (OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, token.Addr()))
+  {
+    TOKEN_ELEVATION elevation{};
+    DWORD size = sizeof(elevation);
+
+    if (GetTokenInformation(token.Get(), TokenElevation, &elevation, sizeof(elevation), &size))
+    {
+      isElevated = elevation.TokenIsElevated;
+    }
+  }
+
   auto dummy_key = "Software";
 
   auto key = Key::OpenOrCreate(Hive::CurrentUser, dummy_key, Access::AllAccess);
@@ -98,7 +114,7 @@ void TestRegistryOperations::RunImpl()
 
   TestThrow<KeyNotFoundException>([&](){subkey.Delete();}, "Keys can't be deleted twice");
 
-  auto read_only_key = Key::Open(Hive::LocalMachine, "SAM\\SAM", Access::QueryValue);
+  auto read_only_key = Key::Open(Hive::LocalMachine, isElevated ? "BCD00000000" : "Software", Access::QueryValue);
 
   TestThrow<AccessDeniedException>([&]() {read_only_key.CreateSubKey("denied"); },
     "AccessDeniedException is thrown");

Tests/Tester.cpp

@@ -15,7 +15,17 @@ bool Tester::Run()
   cout << Colors::Blue() << Colors::Bold() << "\n\t\t\t======= Tests for \""
         << _category << "\" =======" << endl
         << Colors::Reset();
-  RunImpl();
+
+  try
+
+  {
+    RunImpl();
+  }
+  catch(const std::exception & e)
+  {
+    std::cout << Colors::Red() << "Tests caught unexpected exception: " << e.what() << std::endl;
+    return false;
+  }
 
   return _failed == 0;
 }