//==---------------- windows_ur.cpp ----------------------------------------==//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include <sycl/detail/defines.hpp>

#include <cassert>
#include <string>
#include <windows.h>
#include <winreg.h>

#include "detail/windows_os_utils.hpp"
#include "ur_win_proxy_loader.hpp"

namespace sycl {
inline namespace _V1 {
namespace detail {

void *GetWinProcAddress(void *module, const char *funcName) {
  return (void *)GetProcAddress((HMODULE)module, funcName);
}

namespace ur {

void *loadOsLibrary(const std::string &LibraryPath) {
  // Tells the system to not display the critical-error-handler message box.
  // Instead, the system sends the error to the calling process.
  // This is crucial for graceful handling of shared libs that can't be
  // loaded, e.g. due to missing native run-times.

  UINT SavedMode = SetErrorMode(SEM_FAILCRITICALERRORS);
  // Exclude current directory from DLL search path
  if (!SetDllDirectoryA("")) {
    assert(false && "Failed to update DLL search path");
  }

  auto Result = (void *)LoadLibraryExA(LibraryPath.c_str(), NULL, NULL);
  (void)SetErrorMode(SavedMode);
  if (!SetDllDirectoryA(nullptr)) {
    assert(false && "Failed to restore DLL search path");
  }

  return Result;
}

int unloadOsLibrary(void *Library) {
  return (int)FreeLibrary((HMODULE)Library);
}

void *getOsLibraryFuncAddress(void *Library, const std::string &FunctionName) {
  return reinterpret_cast<void *>(
      GetProcAddress((HMODULE)Library, FunctionName.c_str()));
}

void *getURLoaderLibrary() { return getPreloadedURLib(); }

} // namespace ur
} // namespace detail
} // namespace _V1
} // namespace sycl