vsort/win32.cpp: fix crash when vsmlrt-cuda does not exist, and also …

…skip non-DLL files Signed-off-by: akarin <[email protected]>
AmusementClub · Dec 8, 2021 · 86aace1 · 86aace1
1 parent 53dcaaa
commit 86aace1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/vsort/win32.cpp b/vsort/win32.cpp
@@ -102,10 +102,14 @@ void preloadCudaDlls() {
     std::map<std::wstring, std::filesystem::path> dllmap;
 
     auto findDllIn = [&](const std::filesystem::path &dir) {
+        if (!std::filesystem::is_directory(dir))
+            return;
         for (const auto &ent: std::filesystem::directory_iterator{dir}) {
             if (!ent.is_regular_file())
                 continue;
             const auto path = ent.path();
+            if (path.extension() != ".dll")
+                continue;
             const std::wstring filename = path.filename().wstring();
             for (const auto &dll: cudaDlls) {
                 if (dllmap.count(dll) > 0)