IOS/ES: Move GetStoredContentsFromTMD to NandUtils

This commit is contained in:
Léo Lam 2017-03-12 21:12:19 +01:00
parent 0ed8d68192
commit 8984112501
3 changed files with 35 additions and 30 deletions

View File

@ -4,6 +4,7 @@
#include <algorithm> #include <algorithm>
#include <cctype> #include <cctype>
#include <iterator>
#include <string> #include <string>
#include <vector> #include <vector>
@ -11,8 +12,10 @@
#include "Common/FileUtil.h" #include "Common/FileUtil.h"
#include "Common/Logging/Log.h" #include "Common/Logging/Log.h"
#include "Common/NandPaths.h" #include "Common/NandPaths.h"
#include "Common/StringUtil.h"
#include "Core/IOS/ES/Formats.h" #include "Core/IOS/ES/Formats.h"
#include "Core/IOS/ES/NandUtils.h" #include "Core/IOS/ES/NandUtils.h"
#include "DiscIO/NANDContentLoader.h"
namespace IOS namespace IOS
{ {
@ -133,5 +136,30 @@ std::vector<u64> GetTitlesWithTickets()
return title_ids; return title_ids;
} }
std::vector<Content> GetStoredContentsFromTMD(const TMDReader& tmd)
{
if (!tmd.IsValid())
return {};
const DiscIO::CSharedContent shared{Common::FROM_SESSION_ROOT};
const std::vector<Content> contents = tmd.GetContents();
std::vector<Content> stored_contents;
std::copy_if(contents.begin(), contents.end(), std::back_inserter(stored_contents),
[&tmd, &shared](const auto& content) {
if (content.IsShared())
{
const std::string path = shared.GetFilenameFromSHA1(content.sha1.data());
return path != "unk" && File::Exists(path);
}
return File::Exists(
Common::GetTitleContentPath(tmd.GetTitleId(), Common::FROM_SESSION_ROOT) +
StringFromFormat("%08x.app", content.id));
});
return stored_contents;
}
} // namespace ES } // namespace ES
} // namespace IOS } // namespace IOS

View File

@ -7,12 +7,14 @@
#include <vector> #include <vector>
#include "Common/CommonTypes.h" #include "Common/CommonTypes.h"
#include "Core/IOS/ES/Formats.h"
namespace IOS namespace IOS
{ {
namespace ES namespace ES
{ {
struct Content;
class TMDReader;
TMDReader FindImportTMD(u64 title_id); TMDReader FindImportTMD(u64 title_id);
TMDReader FindInstalledTMD(u64 title_id); TMDReader FindInstalledTMD(u64 title_id);
@ -22,5 +24,7 @@ std::vector<u64> GetInstalledTitles();
std::vector<u64> GetTitleImports(); std::vector<u64> GetTitleImports();
// Get titles for which there is a ticket (in /ticket). // Get titles for which there is a ticket (in /ticket).
std::vector<u64> GetTitlesWithTickets(); std::vector<u64> GetTitlesWithTickets();
std::vector<Content> GetStoredContentsFromTMD(const TMDReader& tmd);
} // namespace ES } // namespace ES
} // namespace IOS } // namespace IOS

View File

@ -4,10 +4,8 @@
#include "Core/IOS/ES/ES.h" #include "Core/IOS/ES/ES.h"
#include <algorithm>
#include <cinttypes> #include <cinttypes>
#include <cstdio> #include <cstdio>
#include <iterator>
#include <string> #include <string>
#include <vector> #include <vector>
@ -26,31 +24,6 @@ namespace HLE
{ {
namespace Device namespace Device
{ {
static std::vector<IOS::ES::Content> GetStoredContentsFromTMD(const IOS::ES::TMDReader& tmd)
{
if (!tmd.IsValid())
return {};
const DiscIO::CSharedContent shared{Common::FROM_SESSION_ROOT};
const std::vector<IOS::ES::Content> contents = tmd.GetContents();
std::vector<IOS::ES::Content> stored_contents;
std::copy_if(contents.begin(), contents.end(), std::back_inserter(stored_contents),
[&tmd, &shared](const auto& content) {
if (content.IsShared())
{
const std::string path = shared.GetFilenameFromSHA1(content.sha1.data());
return path != "unk" && File::Exists(path);
}
return File::Exists(
Common::GetTitleContentPath(tmd.GetTitleId(), Common::FROM_SESSION_ROOT) +
StringFromFormat("%08x.app", content.id));
});
return stored_contents;
}
// Used by the GetStoredContents ioctlvs. This assumes that the first output vector // Used by the GetStoredContents ioctlvs. This assumes that the first output vector
// is used for the content count (u32). // is used for the content count (u32).
IPCCommandResult ES::GetStoredContentsCount(const IOS::ES::TMDReader& tmd, IPCCommandResult ES::GetStoredContentsCount(const IOS::ES::TMDReader& tmd,
@ -59,7 +32,7 @@ IPCCommandResult ES::GetStoredContentsCount(const IOS::ES::TMDReader& tmd,
if (request.io_vectors[0].size != sizeof(u32) || !tmd.IsValid()) if (request.io_vectors[0].size != sizeof(u32) || !tmd.IsValid())
return GetDefaultReply(ES_PARAMETER_SIZE_OR_ALIGNMENT); return GetDefaultReply(ES_PARAMETER_SIZE_OR_ALIGNMENT);
const u16 num_contents = static_cast<u16>(GetStoredContentsFromTMD(tmd).size()); const u16 num_contents = static_cast<u16>(IOS::ES::GetStoredContentsFromTMD(tmd).size());
Memory::Write_U32(num_contents, request.io_vectors[0].address); Memory::Write_U32(num_contents, request.io_vectors[0].address);
INFO_LOG(IOS_ES, "GetStoredContentsCount (0x%x): %u content(s) for %016" PRIx64, request.request, INFO_LOG(IOS_ES, "GetStoredContentsCount (0x%x): %u content(s) for %016" PRIx64, request.request,
@ -80,7 +53,7 @@ IPCCommandResult ES::GetStoredContents(const IOS::ES::TMDReader& tmd, const IOCt
return GetDefaultReply(ES_PARAMETER_SIZE_OR_ALIGNMENT); return GetDefaultReply(ES_PARAMETER_SIZE_OR_ALIGNMENT);
} }
const auto contents = GetStoredContentsFromTMD(tmd); const auto contents = IOS::ES::GetStoredContentsFromTMD(tmd);
const u32 max_content_count = Memory::Read_U32(request.in_vectors[1].address); const u32 max_content_count = Memory::Read_U32(request.in_vectors[1].address);
for (u32 i = 0; i < std::min(static_cast<u32>(contents.size()), max_content_count); ++i) for (u32 i = 0; i < std::min(static_cast<u32>(contents.size()), max_content_count); ++i)
Memory::Write_U32(contents[i].id, request.io_vectors[0].address + i * sizeof(u32)); Memory::Write_U32(contents[i].id, request.io_vectors[0].address + i * sizeof(u32));