Reland Change to module directory when loading swiftshader ICD.

BUILD.gn

@@ -107,6 +107,9 @@ config("internal_config") {
 
   if (is_win) {
     defines += [ "ANGLE_IS_WIN" ]
+    if (angle_is_winuwp) {
+      defines += [ "ANGLE_IS_WINUWP" ]
+    }
   } else if (is_linux || is_chromeos) {
     defines += [ "ANGLE_IS_LINUX" ]
   }

src/common/system_utils.h

@@ -19,7 +19,7 @@ namespace angle
 std::string GetExecutableName();
 std::string GetExecutablePath();
 std::string GetExecutableDirectory();
-std::string GetHelperExecutableDir();
+std::string GetModuleDirectory();
 const char *GetSharedLibraryExtension();
 const char *GetExecutableExtension();
 char GetPathSeparator();

src/common/system_utils_posix.cpp

@@ -56,7 +56,7 @@ const char *GetPathSeparatorForEnvironmentVar()
     return ":";
 }
 
-std::string GetHelperExecutableDir()
+std::string GetModuleDirectory()
 {
     std::string directory;
     static int placeholderSymbol = 0;
@@ -66,6 +66,12 @@ std::string GetHelperExecutableDir()
         std::string moduleName = dlInfo.dli_fname;
         directory              = moduleName.substr(0, moduleName.find_last_of('/') + 1);
     }
+    // Ensure we return the full path to the module, not the relative path
+    Optional<std::string> cwd = GetCWD();
+    if (directory.at(0) != '/' && cwd.valid())
+    {
+        directory = cwd.value() + GetPathSeparator() + directory;
+    }
     return directory;
 }
 
@@ -107,7 +113,7 @@ Library *OpenSharedLibrary(const char *libraryName, SearchType searchType)
         // On iOS, shared libraries must be loaded from within the app bundle.
         directory = GetExecutableDirectory() + "/Frameworks/";
 #else
-        directory = GetHelperExecutableDir();
+        directory = GetModuleDirectory();
 #endif
     }
 

src/common/system_utils_win.cpp

@@ -15,21 +15,37 @@
 
 namespace angle
 {
-std::string GetExecutablePath()
+
+namespace
+{
+
+std::string GetPath(HMODULE module)
 {
     std::array<char, MAX_PATH> executableFileBuf;
-    DWORD executablePathLen = GetModuleFileNameA(nullptr, executableFileBuf.data(),
+    DWORD executablePathLen = GetModuleFileNameA(module, executableFileBuf.data(),
                                                  static_cast<DWORD>(executableFileBuf.size()));
     return (executablePathLen > 0 ? std::string(executableFileBuf.data()) : "");
 }
 
-std::string GetExecutableDirectory()
+std::string GetDirectory(HMODULE module)
 {
-    std::string executablePath = GetExecutablePath();
+    std::string executablePath = GetPath(module);
     size_t lastPathSepLoc      = executablePath.find_last_of("\\/");
     return (lastPathSepLoc != std::string::npos) ? executablePath.substr(0, lastPathSepLoc) : "";
 }
 
+}  // anonymous namespace
+
+std::string GetExecutablePath()
+{
+    return GetPath(nullptr);
+}
+
+std::string GetExecutableDirectory()
+{
+    return GetDirectory(nullptr);
+}
+
 const char *GetSharedLibraryExtension()
 {
     return "dll";
@@ -101,8 +117,20 @@ char GetPathSeparator()
     return '\\';
 }
 
-std::string GetHelperExecutableDir()
+std::string GetModuleDirectory()
 {
-    return "";
+// GetModuleHandleEx is unavailable on UWP
+#if !defined(ANGLE_IS_WINUWP)
+    static int placeholderSymbol = 0;
+    HMODULE module               = nullptr;
+    if (GetModuleHandleExA(
+            GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT,
+            reinterpret_cast<LPCSTR>(&placeholderSymbol), &module))
+    {
+        return GetDirectory(module);
+    }
+#endif
+    return GetDirectory(nullptr);
 }
+
 }  // namespace angle

src/common/vulkan/vulkan_icd.cpp

@@ -51,12 +51,9 @@ namespace
     !defined(ANGLE_PLATFORM_GGP)
 const std::string WrapICDEnvironment(const char *icdEnvironment)
 {
-#    if defined(ANGLE_PLATFORM_APPLE)
-    // On MacOS the libraries are bundled into the application directory
-    std::string ret = angle::GetHelperExecutableDir() + icdEnvironment;
+    // The libraries are bundled into the module directory
+    std::string ret = angle::GetModuleDirectory() + GetPathSeparator() + icdEnvironment;
     return ret;
-#    endif  // defined(ANGLE_PLATFORM_APPLE)
-    return icdEnvironment;
 }
 
 constexpr char kLoaderLayersPathEnv[] = "VK_LAYER_PATH";
@@ -146,9 +143,9 @@ ScopedVkLoaderEnvironment::ScopedVkLoaderEnvironment(bool enableValidationLayers
         }
         else
         {
-            mPreviousCWD       = cwd.value();
-            std::string exeDir = angle::GetExecutableDirectory();
-            mChangedCWD        = angle::SetCWD(exeDir.c_str());
+            mPreviousCWD          = cwd.value();
+            std::string moduleDir = angle::GetModuleDirectory();
+            mChangedCWD           = angle::SetCWD(moduleDir.c_str());
             if (!mChangedCWD)
             {
                 ERR() << "Error setting CWD for Vulkan layers init.";