Add a way to restrict ISO reading to a single path.

This is for GameInfoCache.

Core/FileSystems/ISOFileSystem.cpp

@@ -133,8 +133,21 @@ struct VolDescriptor
 
 std::list<ISOFileSystem::TreeEntry *> ISOFileSystem::entryFreeList;
 
-ISOFileSystem::ISOFileSystem(IHandleAllocator *_hAlloc, BlockDevice *_blockDevice) 
+ISOFileSystem::ISOFileSystem(IHandleAllocator *_hAlloc, BlockDevice *_blockDevice, std::string _restrictPath) 
 {
+	if (!_restrictPath.empty())
+	{
+		size_t pos = _restrictPath.find_first_not_of('/');
+		while (!_restrictPath.empty())
+		{
+			size_t endPos = _restrictPath.find_first_of('/', pos);
+			if (pos != endPos)
+				restrictTree.push_back(_restrictPath.substr(pos, endPos));
+			pos = _restrictPath.find_first_not_of('/', endPos);
+			_restrictPath.erase(0, pos);
+		}
+	}
+
 	blockDevice = _blockDevice;
 	hAlloc = _hAlloc;
 
@@ -169,7 +182,7 @@ ISOFileSystem::ISOFileSystem(IHandleAllocator *_hAlloc, BlockDevice *_blockDevic
 	u32 rootSector = desc.root.firstDataSectorLE;
 	u32 rootSize = desc.root.dataLengthLE;
 
-	ReadDirectory(rootSector, rootSize, treeroot);
+	ReadDirectory(rootSector, rootSize, treeroot, 0);
 }
 
 ISOFileSystem::~ISOFileSystem()
@@ -178,7 +191,7 @@ ISOFileSystem::~ISOFileSystem()
 	ReleaseTreeEntry(treeroot);
 }
 
-void ISOFileSystem::ReadDirectory(u32 startsector, u32 dirsize, TreeEntry *root)
+void ISOFileSystem::ReadDirectory(u32 startsector, u32 dirsize, TreeEntry *root, size_t level)
 {
 	for (u32 secnum = startsector, endsector = dirsize/2048 + startsector; secnum < endsector; ++secnum)
 	{
@@ -198,7 +211,7 @@ void ISOFileSystem::ReadDirectory(u32 startsector, u32 dirsize, TreeEntry *root)
 			if (offset + IDENTIFIER_OFFSET + dir.identifierLength > 2048)
 			{
 				ERROR_LOG(FILESYS, "Directory entry crosses sectors, corrupt iso?");
-				break;
+				return;
 			}
 
 			offset += dir.size;
@@ -241,7 +254,14 @@ void ISOFileSystem::ReadDirectory(u32 startsector, u32 dirsize, TreeEntry *root)
 				}
 				else
 				{
-					ReadDirectory(dir.firstDataSectorLE, dir.dataLengthLE, e);
+					bool doRecurse = true;
+					if (!restrictTree.empty())
+						doRecurse = level < restrictTree.size() && restrictTree[level] == e->name;
+
+					if (doRecurse)
+						ReadDirectory(dir.firstDataSectorLE, dir.dataLengthLE, e, level + 1);
+					else
+						continue;
 				}
 			}
 			root->children.push_back(e);

Core/FileSystems/ISOFileSystem.h

@@ -20,6 +20,7 @@
 #include <map>
 #include <string>
 #include <list>
+#include <vector>
 
 #include "FileSystem.h"
 
@@ -29,7 +30,7 @@
 class ISOFileSystem : public IFileSystem
 {
 public:
-	ISOFileSystem(IHandleAllocator *_hAlloc, BlockDevice *_blockDevice);
+	ISOFileSystem(IHandleAllocator *_hAlloc, BlockDevice *_blockDevice, std::string _restrictPath = "");
 	~ISOFileSystem();
 	void DoState(PointerWrap &p);
 	std::vector<PSPFileInfo> GetDirListing(std::string path);
@@ -87,7 +88,10 @@ private:
 
 	TreeEntry entireISO;
 
-	void ReadDirectory(u32 startsector, u32 dirsize, TreeEntry *root);
+	// Don't use this in the emu, not savestated.
+	std::vector<std::string> restrictTree;
+
+	void ReadDirectory(u32 startsector, u32 dirsize, TreeEntry *root, size_t level);
 	TreeEntry *GetFromPath(std::string path, bool catchError=true);
 	std::string EntryFullPath(TreeEntry *e);
 

UI/GameInfoCache.cpp

@@ -120,7 +120,7 @@ again:
 		BlockDevice *bd = constructBlockDevice(gamePath.c_str());
 		if (!bd)
 			return 0;  // nothing to do here..
-		ISOFileSystem umd(&handles, bd);
+		ISOFileSystem umd(&handles, bd, "/PSP_GAME");
 
 		GameInfo *info = new GameInfo();
 		info->wantBG = wantBG;