4 files changed, 637 insertions, 0 deletions

diff --git a/src/core/core.vcxproj b/src/core/core.vcxproj
index 7d7dceee3..c771f2e2d 100644
--- a/src/core/core.vcxproj
+++ b/src/core/core.vcxproj
@@ -138,6 +138,7 @@
     <ClCompile Include="src\arm\disassembler\arm_disasm.cpp" />
     <ClCompile Include="src\core.cpp" />
     <ClCompile Include="src\file_sys\directory_file_system.cpp" />
+    <ClCompile Include="src\file_sys\meta_file_system.cpp" />
     <ClCompile Include="src\loader.cpp" />
     <ClCompile Include="src\mem_map.cpp" />
     <ClCompile Include="src\mem_map_funcs.cpp" />
@@ -159,6 +160,7 @@
     <ClInclude Include="src\core.h" />
     <ClInclude Include="src\file_sys\directory_file_system.h" />
     <ClInclude Include="src\file_sys\file_sys.h" />
+    <ClInclude Include="src\file_sys\meta_file_system.h" />
     <ClInclude Include="src\loader.h" />
     <ClInclude Include="src\mem_map.h" />
   </ItemGroup>
diff --git a/src/core/core.vcxproj.filters b/src/core/core.vcxproj.filters
index 93becd9d7..f75e6305c 100644
--- a/src/core/core.vcxproj.filters
+++ b/src/core/core.vcxproj.filters
@@ -17,6 +17,9 @@
     <ClCompile Include="src\file_sys\directory_file_system.cpp">
       <Filter>file_sys</Filter>
     </ClCompile>
+    <ClCompile Include="src\file_sys\meta_file_system.cpp">
+      <Filter>file_sys</Filter>
+    </ClCompile>
   </ItemGroup>
   <ItemGroup>
     <Filter Include="arm">
@@ -81,6 +84,9 @@
     <ClInclude Include="src\file_sys\directory_file_system.h">
       <Filter>file_sys</Filter>
     </ClInclude>
+    <ClInclude Include="src\file_sys\meta_file_system.h">
+      <Filter>file_sys</Filter>
+    </ClInclude>
   </ItemGroup>
   <ItemGroup>
     <None Include="CMakeLists.txt" />
diff --git a/src/core/src/file_sys/meta_file_system.cpp b/src/core/src/file_sys/meta_file_system.cpp
new file mode 100644
index 000000000..f86c3cb18
--- /dev/null
+++ b/src/core/src/file_sys/meta_file_system.cpp
@@ -0,0 +1,520 @@
+// Copyright (c) 2012- PPSSPP Project.
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 2.0 or later versions.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License 2.0 for more details.
+
+// A copy of the GPL 2.0 should have been included with the program.
+// If not, see http://www.gnu.org/licenses/
+
+// Official git repository and contact information can be found at
+// https://github.com/hrydgard/ppsspp and http://www.ppsspp.org/.
+
+#include <set>
+#include <algorithm>
+#include "string_util.h"
+#include "file_sys/meta_file_system.h"
+//#include "Core/HLE/sceKernelThread.h"
+//#include "Core/Reporting.h"
+
+static bool ApplyPathStringToComponentsVector(std::vector<std::string> &vector, const std::string &pathString)
+{
+	size_t len = pathString.length();
+	size_t start = 0;
+
+	while (start < len)
+	{
+		size_t i = pathString.find('/', start);
+		if (i == std::string::npos)
+			i = len;
+
+		if (i > start)
+		{
+			std::string component = pathString.substr(start, i - start);
+			if (component != ".")
+			{
+				if (component == "..")
+				{
+					if (vector.size() != 0)
+					{
+						vector.pop_back();
+					}
+					else
+					{
+						// The PSP silently ignores attempts to .. to parent of root directory
+						WARN_LOG(FILESYS, "RealPath: ignoring .. beyond root - root directory is its own parent: \"%s\"", pathString.c_str());
+					}
+				}
+				else
+				{
+					vector.push_back(component);
+				}
+			}
+		}
+
+		start = i + 1;
+	}
+
+	return true;
+}
+
+/*
+ * Changes relative paths to absolute, removes ".", "..", and trailing "/"
+ * "drive:./blah" is absolute (ignore the dot) and "/blah" is relative (because it's missing "drive:")
+ * babel (and possibly other games) use "/directoryThatDoesNotExist/../directoryThatExists/filename"
+ */
+static bool RealPath(const std::string &currentDirectory, const std::string &inPath, std::string &outPath)
+{
+	size_t inLen = inPath.length();
+	if (inLen == 0)
+	{
+		WARN_LOG(FILESYS, "RealPath: inPath is empty");
+		outPath = currentDirectory;
+		return true;
+	}
+
+	size_t inColon = inPath.find(':');
+	if (inColon + 1 == inLen)
+	{
+		// There's nothing after the colon, e.g. umd0: - this is perfectly valid.
+		outPath = inPath;
+		return true;
+	}
+
+	bool relative = (inColon == std::string::npos);
+	
+	std::string prefix, inAfterColon;
+	std::vector<std::string> cmpnts;  // path components
+	size_t outPathCapacityGuess = inPath.length();
+
+	if (relative)
+	{
+		size_t curDirLen = currentDirectory.length();
+		if (curDirLen == 0)
+		{
+			ERROR_LOG(FILESYS, "RealPath: inPath \"%s\" is relative, but current directory is empty", inPath.c_str());
+			return false;
+		}
+		
+		size_t curDirColon = currentDirectory.find(':');
+		if (curDirColon == std::string::npos)
+		{
+			ERROR_LOG(FILESYS, "RealPath: inPath \"%s\" is relative, but current directory \"%s\" has no prefix", inPath.c_str(), currentDirectory.c_str());
+			return false;
+		}
+		if (curDirColon + 1 == curDirLen)
+		{
+			ERROR_LOG(FILESYS, "RealPath: inPath \"%s\" is relative, but current directory \"%s\" is all prefix and no path. Using \"/\" as path for current directory.", inPath.c_str(), currentDirectory.c_str());
+		}
+		else
+		{
+			const std::string curDirAfter = currentDirectory.substr(curDirColon + 1);
+			if (! ApplyPathStringToComponentsVector(cmpnts, curDirAfter) )
+			{
+				ERROR_LOG(FILESYS,"RealPath: currentDirectory is not a valid path: \"%s\"", currentDirectory.c_str());
+				return false;
+			}
+
+			outPathCapacityGuess += curDirLen;
+		}
+
+		prefix = currentDirectory.substr(0, curDirColon + 1);
+		inAfterColon = inPath;
+	}
+	else
+	{
+		prefix = inPath.substr(0, inColon + 1);
+		inAfterColon = inPath.substr(inColon + 1);
+	}
+
+	// Special case: "disc0:" is different from "disc0:/", so keep track of the single slash.
+	if (inAfterColon == "/")
+	{
+		outPath = prefix + inAfterColon;
+		return true;
+	}
+
+	if (! ApplyPathStringToComponentsVector(cmpnts, inAfterColon) )
+	{
+		WARN_LOG(FILESYS, "RealPath: inPath is not a valid path: \"%s\"", inPath.c_str());
+		return false;
+	}
+
+	outPath.clear();
+	outPath.reserve(outPathCapacityGuess);
+
+	outPath.append(prefix);
+
+	size_t numCmpnts = cmpnts.size();
+	for (size_t i = 0; i < numCmpnts; i++)
+	{
+		outPath.append(1, '/');
+		outPath.append(cmpnts[i]);
+	}
+
+	return true;
+}
+
+IFileSystem *MetaFileSystem::GetHandleOwner(u32 handle)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	for (size_t i = 0; i < fileSystems.size(); i++)
+	{
+		if (fileSystems[i].system->OwnsHandle(handle))
+			return fileSystems[i].system; //got it!
+	}
+	//none found?
+	return 0;
+}
+
+bool MetaFileSystem::MapFilePath(const std::string &_inpath, std::string &outpath, MountPoint **system)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string realpath;
+
+	// Special handling: host0:command.txt (as seen in Super Monkey Ball Adventures, for example)
+	// appears to mean the current directory on the UMD. Let's just assume the current directory.
+	std::string inpath = _inpath;
+	if (strncasecmp(inpath.c_str(), "host0:", strlen("host0:")) == 0) {
+		INFO_LOG(FILESYS, "Host0 path detected, stripping: %s", inpath.c_str());
+		inpath = inpath.substr(strlen("host0:"));
+	}
+
+	const std::string *currentDirectory = &startingDirectory;
+
+	_assert_msg_(FILESYS, false, "must implement equiv of __KernelGetCurThread");
+
+	int currentThread = 0;//__KernelGetCurThread();
+	currentDir_t::iterator it = currentDir.find(currentThread);
+	if (it == currentDir.end()) 
+	{
+		//Attempt to emulate SCE_KERNEL_ERROR_NOCWD / 8002032C: may break things requiring fixes elsewhere
+		if (inpath.find(':') == std::string::npos /* means path is relative */) 
+		{
+			lastOpenError = -1;//SCE_KERNEL_ERROR_NOCWD;
+			WARN_LOG(FILESYS, "Path is relative, but current directory not set for thread %i. returning 8002032C(SCE_KERNEL_ERROR_NOCWD) instead.", currentThread);
+		}
+	}
+	else
+	{
+		currentDirectory = &(it->second);
+	}
+
+	if ( RealPath(*currentDirectory, inpath, realpath) )
+	{
+		for (size_t i = 0; i < fileSystems.size(); i++)
+		{
+			size_t prefLen = fileSystems[i].prefix.size();
+			if (strncasecmp(fileSystems[i].prefix.c_str(), realpath.c_str(), prefLen) == 0)
+			{
+				outpath = realpath.substr(prefLen);
+				*system = &(fileSystems[i]);
+
+				INFO_LOG(FILESYS, "MapFilePath: mapped \"%s\" to prefix: \"%s\", path: \"%s\"", inpath.c_str(), fileSystems[i].prefix.c_str(), outpath.c_str());
+
+				return true;
+			}
+		}
+	}
+
+	DEBUG_LOG(FILESYS, "MapFilePath: failed mapping \"%s\", returning false", inpath.c_str());
+	return false;
+}
+
+void MetaFileSystem::Mount(std::string prefix, IFileSystem *system)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	MountPoint x;
+	x.prefix = prefix;
+	x.system = system;
+	fileSystems.push_back(x);
+}
+
+void MetaFileSystem::Unmount(std::string prefix, IFileSystem *system)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	MountPoint x;
+	x.prefix = prefix;
+	x.system = system;
+	fileSystems.erase(std::remove(fileSystems.begin(), fileSystems.end(), x), fileSystems.end());
+}
+
+void MetaFileSystem::Shutdown()
+{
+	std::lock_guard<std::mutex> guard(lock);
+	current = 6;
+
+	// Ownership is a bit convoluted. Let's just delete everything once.
+
+	std::set<IFileSystem *> toDelete;
+	for (size_t i = 0; i < fileSystems.size(); i++) {
+		toDelete.insert(fileSystems[i].system);
+	}
+
+	for (auto iter = toDelete.begin(); iter != toDelete.end(); ++iter)
+	{
+		delete *iter;
+	}
+
+	fileSystems.clear();
+	currentDir.clear();
+	startingDirectory = "";
+}
+
+u32 MetaFileSystem::OpenWithError(int &error, std::string filename, FileAccess access, const char *devicename)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	u32 h = OpenFile(filename, access, devicename);
+	error = lastOpenError;
+	return h;
+}
+
+u32 MetaFileSystem::OpenFile(std::string filename, FileAccess access, const char *devicename)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	lastOpenError = 0;
+	std::string of;
+	MountPoint *mount;
+	if (MapFilePath(filename, of, &mount))
+	{
+		return mount->system->OpenFile(of, access, mount->prefix.c_str());
+	}
+	else
+	{
+		return 0;
+	}
+}
+
+FileInfo MetaFileSystem::GetFileInfo(std::string filename)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	IFileSystem *system;
+	if (MapFilePath(filename, of, &system))
+	{
+		return system->GetFileInfo(of);
+	}
+	else
+	{
+		FileInfo bogus; // TODO
+		return bogus; 
+	}
+}
+
+bool MetaFileSystem::GetHostPath(const std::string &inpath, std::string &outpath)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	IFileSystem *system;
+	if (MapFilePath(inpath, of, &system)) {
+		return system->GetHostPath(of, outpath);
+	} else {
+		return false;
+	}
+}
+
+std::vector<FileInfo> MetaFileSystem::GetDirListing(std::string path)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	IFileSystem *system;
+	if (MapFilePath(path, of, &system))
+	{
+		return system->GetDirListing(of);
+	}
+	else
+	{
+		std::vector<FileInfo> empty;
+		return empty;
+	}
+}
+
+void MetaFileSystem::ThreadEnded(int threadID)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	currentDir.erase(threadID);
+}
+
+int MetaFileSystem::ChDir(const std::string &dir)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	// Retain the old path and fail if the arg is 1023 bytes or longer.
+	if (dir.size() >= 1023)
+		return -1;//SCE_KERNEL_ERROR_NAMETOOLONG;
+
+	_assert_msg_(FILESYS, false, "must implement equiv of __KernelGetCurThread");
+
+	int curThread = 0; //__KernelGetCurThread();
+	
+	std::string of;
+	MountPoint *mountPoint;
+	if (MapFilePath(dir, of, &mountPoint))
+	{
+		currentDir[curThread] = mountPoint->prefix + of;
+		return 0;
+	}
+	else
+	{
+		for (size_t i = 0; i < fileSystems.size(); i++)
+		{
+			const std::string &prefix = fileSystems[i].prefix;
+			if (strncasecmp(prefix.c_str(), dir.c_str(), prefix.size()) == 0)
+			{
+				// The PSP is completely happy with invalid current dirs as long as they have a valid device.
+				WARN_LOG(FILESYS, "ChDir failed to map path \"%s\", saving as current directory anyway", dir.c_str());
+				currentDir[curThread] = dir;
+				return 0;
+			}
+		}
+
+		WARN_LOG(FILESYS, "ChDir failed to map device for \"%s\", failing", dir.c_str());
+		return -1;//SCE_KERNEL_ERROR_NODEV;
+	}
+}
+
+bool MetaFileSystem::MkDir(const std::string &dirname)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	IFileSystem *system;
+	if (MapFilePath(dirname, of, &system))
+	{
+		return system->MkDir(of);
+	}
+	else
+	{
+		return false;
+	}
+}
+
+bool MetaFileSystem::RmDir(const std::string &dirname)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	IFileSystem *system;
+	if (MapFilePath(dirname, of, &system))
+	{
+		return system->RmDir(of);
+	}
+	else
+	{
+		return false;
+	}
+}
+
+int MetaFileSystem::RenameFile(const std::string &from, const std::string &to)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	std::string rf;
+	IFileSystem *osystem;
+	IFileSystem *rsystem = NULL;
+	if (MapFilePath(from, of, &osystem))
+	{
+		// If it's a relative path, it seems to always use from's filesystem.
+		if (to.find(":/") != to.npos)
+		{
+			if (!MapFilePath(to, rf, &rsystem))
+				return -1;
+		}
+		else
+		{
+			rf = to;
+			rsystem = osystem;
+		}
+
+		if (osystem != rsystem)
+			return -1;//SCE_KERNEL_ERROR_XDEV;
+
+		return osystem->RenameFile(of, rf);
+	}
+	else
+	{
+		return -1;
+	}
+}
+
+bool MetaFileSystem::RemoveFile(const std::string &filename)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	std::string of;
+	IFileSystem *system;
+	if (MapFilePath(filename, of, &system))
+	{
+		return system->RemoveFile(of);
+	}
+	else
+	{
+		return false;
+	}
+}
+
+void MetaFileSystem::CloseFile(u32 handle)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	IFileSystem *sys = GetHandleOwner(handle);
+	if (sys)
+		sys->CloseFile(handle);
+}
+
+size_t MetaFileSystem::ReadFile(u32 handle, u8 *pointer, s64 size)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	IFileSystem *sys = GetHandleOwner(handle);
+	if (sys)
+		return sys->ReadFile(handle,pointer,size);
+	else
+		return 0;
+}
+
+size_t MetaFileSystem::WriteFile(u32 handle, const u8 *pointer, s64 size)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	IFileSystem *sys = GetHandleOwner(handle);
+	if (sys)
+		return sys->WriteFile(handle,pointer,size);
+	else
+		return 0;
+}
+
+size_t MetaFileSystem::SeekFile(u32 handle, s32 position, FileMove type)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	IFileSystem *sys = GetHandleOwner(handle);
+	if (sys)
+		return sys->SeekFile(handle,position,type);
+	else
+		return 0;
+}
+
+void MetaFileSystem::DoState(PointerWrap &p)
+{
+	std::lock_guard<std::mutex> guard(lock);
+	
+	auto s = p.Section("MetaFileSystem", 1);
+	if (!s)
+		return;
+
+	p.Do(current);
+
+	// Save/load per-thread current directory map
+	p.Do(currentDir);
+
+	u32 n = (u32) fileSystems.size();
+	p.Do(n);
+	if (n != (u32) fileSystems.size())
+	{
+		p.SetError(p.ERROR_FAILURE);
+		ERROR_LOG(FILESYS, "Savestate failure: number of filesystems doesn't match.");
+		return;
+	}
+
+	for (u32 i = 0; i < n; ++i)
+		fileSystems[i].system->DoState(p);
+}
+
diff --git a/src/core/src/file_sys/meta_file_system.h b/src/core/src/file_sys/meta_file_system.h
new file mode 100644
index 000000000..0de23d49c
--- /dev/null
+++ b/src/core/src/file_sys/meta_file_system.h
@@ -0,0 +1,109 @@
+// Copyright (c) 2012- PPSSPP Project.
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, version 2.0 or later versions.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License 2.0 for more details.
+
+// A copy of the GPL 2.0 should have been included with the program.
+// If not, see http://www.gnu.org/licenses/
+
+// Official git repository and contact information can be found at
+// https://github.com/hrydgard/ppsspp and http://www.ppsspp.org/.
+
+#pragma once
+
+#include "std_mutex.h"
+#include "file_sys.h"
+
+class MetaFileSystem : public IHandleAllocator, public IFileSystem
+{
+private:
+	u32 current;
+	struct MountPoint
+	{
+		std::string prefix;
+		IFileSystem *system;
+
+		bool operator == (const MountPoint &other) const
+		{
+			return prefix == other.prefix && system == other.system;
+		}
+	};
+	std::vector<MountPoint> fileSystems;
+
+	typedef std::map<int, std::string> currentDir_t;
+	currentDir_t currentDir;
+
+	std::string startingDirectory;
+	int lastOpenError;
+	std::recursive_mutex lock;
+
+public:
+	MetaFileSystem()
+	{
+		current = 6;  // what?
+	}
+
+	void Mount(std::string prefix, IFileSystem *system);
+	void Unmount(std::string prefix, IFileSystem *system);
+
+	void ThreadEnded(int threadID);
+
+	void Shutdown();
+
+	u32 GetNewHandle() {return current++;}
+	void FreeHandle(u32 handle) {}
+
+	virtual void DoState(PointerWrap &p);
+
+	IFileSystem *GetHandleOwner(u32 handle);
+	bool MapFilePath(const std::string &inpath, std::string &outpath, MountPoint **system);
+
+	inline bool MapFilePath(const std::string &_inpath, std::string &outpath, IFileSystem **system)
+	{
+		MountPoint *mountPoint;
+		if (MapFilePath(_inpath, outpath, &mountPoint))
+		{
+			*system = mountPoint->system;
+			return true;
+		}
+
+		return false;
+	}
+
+	// Only possible if a file system is a DirectoryFileSystem or similar.
+	bool GetHostPath(const std::string &inpath, std::string &outpath);
+	
+	std::vector<FileInfo> GetDirListing(std::string path);
+	u32      OpenFile(std::string filename, FileAccess access, const char *devicename = NULL);
+	u32      OpenWithError(int &error, std::string filename, FileAccess access, const char *devicename = NULL);
+	void     CloseFile(u32 handle);
+	size_t   ReadFile(u32 handle, u8 *pointer, s64 size);
+	size_t   WriteFile(u32 handle, const u8 *pointer, s64 size);
+	size_t   SeekFile(u32 handle, s32 position, FileMove type);
+	FileInfo GetFileInfo(std::string filename);
+	bool     OwnsHandle(u32 handle) {return false;}
+	inline size_t GetSeekPos(u32 handle)
+	{
+		return SeekFile(handle, 0, FILEMOVE_CURRENT);
+	}
+
+	virtual int ChDir(const std::string &dir);
+
+	virtual bool MkDir(const std::string &dirname);
+	virtual bool RmDir(const std::string &dirname);
+	virtual int  RenameFile(const std::string &from, const std::string &to);
+	virtual bool RemoveFile(const std::string &filename);
+
+	// TODO: void IoCtl(...)
+
+	void SetStartingDirectory(const std::string &dir) {
+		std::lock_guard<std::mutex> guard(lock);
+		startingDirectory = dir;
+	}
+};