updated build system

support/src/utils/windows.cpp

@@ -0,0 +1,105 @@
+/*
+  Copyright <2018-2024> <scott.e.graves@protonmail.com>
+
+  Permission is hereby granted, free of charge, to any person obtaining a copy
+  of this software and associated documentation files (the "Software"), to deal
+  in the Software without restriction, including without limitation the rights
+  to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+  copies of the Software, and to permit persons to whom the Software is
+  furnished to do so, subject to the following conditions:
+
+  The above copyright notice and this permission notice shall be included in all
+  copies or substantial portions of the Software.
+
+  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+  IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+  AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+  SOFTWARE.
+*/
+#if defined(_WIN32)
+#include "utils/windows.hpp"
+
+#include "utils/com_init_wrapper.hpp"
+#include "utils/string.hpp"
+
+namespace repertory::utils {
+auto get_last_error_code() -> DWORD { return ::GetLastError(); }
+
+auto get_local_app_data_directory() -> const std::string & {
+  static std::string app_data = ([]() -> std::string {
+    com_init_wrapper cw;
+    PWSTR local_app_data{};
+    if (SUCCEEDED(::SHGetKnownFolderPath(FOLDERID_LocalAppData, 0, nullptr,
+                                         &local_app_data))) {
+      auto ret = utils::string::to_utf8(local_app_data);
+      ::CoTaskMemFree(local_app_data);
+      return ret;
+    }
+
+    throw std::runtime_error("unable to detect local application data folder");
+  })();
+
+  return app_data;
+}
+
+auto get_thread_id() -> std::uint64_t {
+  return static_cast<std::uint64_t>(::GetCurrentThreadId());
+}
+
+auto is_process_elevated() -> bool {
+  auto ret{false};
+  HANDLE token{INVALID_HANDLE_VALUE};
+  if (::OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY, &token)) {
+    TOKEN_ELEVATION te{};
+    DWORD sz = sizeof(te);
+    if (::GetTokenInformation(token, TokenElevation, &te, sizeof(te), &sz)) {
+      ret = (te.TokenIsElevated != 0);
+    }
+  }
+
+  if (token != INVALID_HANDLE_VALUE) {
+    ::CloseHandle(token);
+  }
+
+  return ret;
+}
+
+auto run_process_elevated(std::vector<const char *> args) -> int {
+  std::cout << "Elevating Process" << std::endl;
+  std::string parameters{"-hidden"};
+  for (std::size_t idx = 1U; idx < args.size(); idx++) {
+    parameters +=
+        (parameters.empty() ? args.at(idx) : " " + std::string(args.at(idx)));
+  }
+
+  std::string full_path;
+  full_path.resize(MAX_PATH + 1);
+
+  if (::GetModuleFileNameA(nullptr, full_path.data(), MAX_PATH)) {
+    SHELLEXECUTEINFOA sei{};
+    sei.fMask = SEE_MASK_NOCLOSEPROCESS;
+    sei.cbSize = sizeof(sei);
+    sei.lpVerb = "runas";
+    sei.lpFile = full_path.c_str();
+    sei.lpParameters = parameters.c_str();
+    sei.hwnd = nullptr;
+    sei.nShow = SW_NORMAL;
+    if (::ShellExecuteExA(&sei)) {
+      ::WaitForSingleObject(sei.hProcess, INFINITE);
+      DWORD exit_code{};
+      ::GetExitCodeProcess(sei.hProcess, &exit_code);
+      ::CloseHandle(sei.hProcess);
+      return static_cast<int>(exit_code);
+    }
+  }
+
+  return static_cast<int>(::GetLastError());
+}
+
+void set_last_error_code(DWORD error_code) { ::SetLastError(error_code); }
+} // namespace repertory::utils
+
+#endif // defined(_WIN32)