Merge pull request #19 from yescallop/master

Fix Steam compatibility and directory creation
arcusmaximus · Apr 12, 2023 · 9590c3c · 9590c3c
2 parents b114e8a + 47b4741
commit 9590c3c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/Common/Directory.cpp b/Common/Directory.cpp
@@ -3,9 +3,9 @@
 
 using namespace std;
 
-void Directory::Create(const std::wstring& folderPath)
+void Directory::Create(const wstring& folderPath)
 {
-    CreateDirectory(folderPath.c_str(), nullptr);
+    filesystem::create_directories(folderPath);
 }
 
 vector<wstring> Directory::GetFiles(const wstring& folderPath, const wstring& pattern)

diff --git a/KirikiriUnencryptedArchive/Patcher.cpp b/KirikiriUnencryptedArchive/Patcher.cpp
@@ -90,11 +90,14 @@ void Patcher::CustomTVPRegisterStorageMedia(iTVPStorageMedia* pMedia)
 {
     ttstr mediaName;
     pMedia->GetName(mediaName);
-    Debugger::Log(L"Hooking storage media \"%s\"", mediaName.c_str());
 
-    void** pVtable = *(void***)pMedia;
-    OriginalStorageMediaOpen[pMedia] = (decltype(CustomStorageMediaOpen)*)pVtable[6];
-    MemoryUtil::WritePointer(&pVtable[6], CustomStorageMediaOpen);
+    if (mediaName != ttstr(L"steam")) {
+        Debugger::Log(L"Hooking storage media \"%s\"", mediaName.c_str());
+
+        void** pVtable = *(void***)pMedia;
+        OriginalStorageMediaOpen[pMedia] = (decltype(CustomStorageMediaOpen)*)pVtable[6];
+        MemoryUtil::WritePointer(&pVtable[6], CustomStorageMediaOpen);
+    }
 
     OriginalTVPRegisterStorageMedia(pMedia);
 }

diff --git a/KirikiriUnencryptedArchive/stdafx.h b/KirikiriUnencryptedArchive/stdafx.h
@@ -8,6 +8,7 @@
 #include <algorithm>
 #include <cassert>
 #include <exception>
+#include <filesystem>
 #include <functional>
 #include <map>
 #include <ranges>