fix for 2743

VirtusLab · Feb 20, 2024 · 3f1faeb · 3f1faeb
1 parent 2c61e87
commit 3f1faeb
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/modules/cli/src/main/scala/scala/cli/packaging/NativeImage.scala b/modules/cli/src/main/scala/scala/cli/packaging/NativeImage.scala
@@ -93,14 +93,24 @@ object NativeImage {
     res.exitCode
   }
 
+  private lazy val mountedDrives: String = {
+    val str = "HKEY_LOCAL_MACHINE/SYSTEM/MountedDevices".replace('/', '\\')
+    val queryDrives = s"reg query $str"
+    val lines = os.proc("cmd", "/c", queryDrives).call().out.lines()
+    val dosDevices = lines.filter { s =>
+      s.contains("DosDevices")
+    }.map { s =>
+      s.replaceAll(".DosDevices.", "").replaceAll(":.*", "")
+    }
+    dosDevices.mkString
+  }
   private def availableDriveLetter(): Char = {
-
+    
     @tailrec
     def helper(from: Char): Char =
       if (from > 'Z') sys.error("Cannot find free drive letter")
       else {
-        val p = os.Path(s"$from:" + "\\")
-        if (os.exists(p)) helper((from + 1).toChar)
+        if (mountedDrives.contains(from)) helper((from + 1).toChar)
         else from
       }