From b9fc359e7e5507e48fcc5e511e43ca1f7765f173 Mon Sep 17 00:00:00 2001 From: shinyquagsire23 Date: Thu, 5 Oct 2017 16:22:15 -0600 Subject: [PATCH] Services/AM: Add title scanning, move to ipc_helper.h, implement most stubbed service calls --- src/core/hle/service/am/am.cpp | 479 +++++++++++++++++++++++------ src/core/hle/service/am/am.h | 84 +++++ src/core/hle/service/am/am_sys.cpp | 6 +- 3 files changed, 470 insertions(+), 99 deletions(-) diff --git a/src/core/hle/service/am/am.cpp b/src/core/hle/service/am/am.cpp index 961305e9f..1d1b57df5 100644 --- a/src/core/hle/service/am/am.cpp +++ b/src/core/hle/service/am/am.cpp @@ -4,116 +4,335 @@ #include #include -#include "common/common_types.h" +#include "common/file_util.h" #include "common/logging/log.h" +#include "common/string_util.h" +#include "core/file_sys/ncch_container.h" +#include "core/file_sys/title_metadata.h" #include "core/hle/ipc.h" +#include "core/hle/ipc_helpers.h" #include "core/hle/result.h" #include "core/hle/service/am/am.h" #include "core/hle/service/am/am_app.h" #include "core/hle/service/am/am_net.h" #include "core/hle/service/am/am_sys.h" #include "core/hle/service/am/am_u.h" +#include "core/hle/service/fs/archive.h" #include "core/hle/service/service.h" +#include "core/loader/loader.h" namespace Service { namespace AM { -static std::array am_content_count = {0, 0, 0}; -static std::array am_titles_count = {0, 0, 0}; -static std::array am_titles_list_count = {0, 0, 0}; -static u32 am_ticket_count = 0; -static u32 am_ticket_list_count = 0; +static bool lists_initialized = false; +static std::array, 3> am_title_list; + +struct TitleInfo { + u64_le tid; + u64_le size; + u16_le version; + u16_le unused; + u32_le type; +}; + +static_assert(sizeof(TitleInfo) == 0x18, "Title info structure size is wrong"); + +struct ContentInfo { + u16_le index; + u16_le type; + u32_le content_id; + u64_le size; + u64_le romfs_size; +}; + +static_assert(sizeof(ContentInfo) == 0x18, "Content info structure size is wrong"); + +struct TicketInfo { + u64_le title_id; + u64_le ticket_id; + u16_le version; + u16_le unused; + u32_le size; +}; + +static_assert(sizeof(TicketInfo) == 0x18, "Ticket info structure size is wrong"); + +std::string GetTitleMetadataPath(Service::FS::MediaType media_type, u64 tid) { + std::string content_path = GetTitlePath(media_type, tid) + "content/"; + + if (media_type == Service::FS::MediaType::GameCard) { + LOG_ERROR(Service_AM, "Invalid request for nonexistent gamecard title metadata!"); + return ""; + } + + // The TMD ID is usually held in the title databases, which we don't implement. + // For now, just scan for any .tmd files which exist and use the first .tmd + // found (there should only really be one unless the directories are meddled with) + FileUtil::FSTEntry entries; + FileUtil::ScanDirectoryTree(content_path, entries); + for (const FileUtil::FSTEntry& entry : entries.children) { + std::string filename_filename, filename_extension; + Common::SplitPath(entry.virtualName, nullptr, &filename_filename, &filename_extension); + + if (filename_extension == ".tmd") + return content_path + entry.virtualName; + } + + // If we can't find an existing .tmd, return a path for one to be created. + return content_path + "00000000.tmd"; +} + +std::string GetTitleContentPath(Service::FS::MediaType media_type, u64 tid, u16 index) { + std::string content_path = GetTitlePath(media_type, tid) + "content/"; + + if (media_type == Service::FS::MediaType::GameCard) { + // TODO(shinyquagsire23): get current app file if TID matches? + LOG_ERROR(Service_AM, "Request for gamecard partition %u content path unimplemented!", + static_cast(index)); + return ""; + } + + std::string tmd_path = GetTitleMetadataPath(media_type, tid); + + u32 content_id = 0; + FileSys::TitleMetadata tmd(tmd_path); + if (tmd.Load() == Loader::ResultStatus::Success) { + content_id = tmd.GetContentIDByIndex(index); + + // TODO(shinyquagsire23): how does DLC actually get this folder on hardware? + // For now, check if the second (index 1) content has the optional flag set, for most + // apps this is usually the manual and not set optional, DLC has it set optional. + // All .apps (including index 0) will be in the 00000000/ folder for DLC. + if (tmd.GetContentCount() > 1 && + tmd.GetContentTypeByIndex(1) & FileSys::TMDContentTypeFlag::Optional) { + content_path += "00000000/"; + } + } + + return Common::StringFromFormat("%s%08x.app", content_path.c_str(), content_id); +} + +std::string GetTitlePath(Service::FS::MediaType media_type, u64 tid) { + u32 high = static_cast(tid >> 32); + u32 low = static_cast(tid & 0xFFFFFFFF); + + if (media_type == Service::FS::MediaType::NAND || media_type == Service::FS::MediaType::SDMC) + return Common::StringFromFormat("%s%08x/%08x/", GetMediaTitlePath(media_type).c_str(), high, + low); + + if (media_type == Service::FS::MediaType::GameCard) { + // TODO(shinyquagsire23): get current app path if TID matches? + LOG_ERROR(Service_AM, "Request for gamecard title path unimplemented!"); + return ""; + } + + return ""; +} + +std::string GetMediaTitlePath(Service::FS::MediaType media_type) { + if (media_type == Service::FS::MediaType::NAND) + return Common::StringFromFormat("%s%s/title/", FileUtil::GetUserPath(D_NAND_IDX).c_str(), + SYSTEM_ID); + + if (media_type == Service::FS::MediaType::SDMC) + return Common::StringFromFormat("%sNintendo 3DS/%s/%s/title/", + FileUtil::GetUserPath(D_SDMC_IDX).c_str(), SYSTEM_ID, + SDCARD_ID); + + if (media_type == Service::FS::MediaType::GameCard) { + // TODO(shinyquagsire23): get current app parent folder if TID matches? + LOG_ERROR(Service_AM, "Request for gamecard parent path unimplemented!"); + return ""; + } + + return ""; +} + +void ScanForTitles(Service::FS::MediaType media_type) { + am_title_list[static_cast(media_type)].clear(); + + std::string title_path = GetMediaTitlePath(media_type); + + FileUtil::FSTEntry entries; + FileUtil::ScanDirectoryTree(title_path, entries, 1); + for (const FileUtil::FSTEntry& tid_high : entries.children) { + for (const FileUtil::FSTEntry& tid_low : tid_high.children) { + std::string tid_string = tid_high.virtualName + tid_low.virtualName; + u64 tid = std::stoull(tid_string.c_str(), nullptr, 16); + + FileSys::NCCHContainer container(GetTitleContentPath(media_type, tid)); + if (container.Load() == Loader::ResultStatus::Success) + am_title_list[static_cast(media_type)].push_back(tid); + } + } +} + +void ScanForAllTitles() { + ScanForTitles(Service::FS::MediaType::NAND); + ScanForTitles(Service::FS::MediaType::SDMC); +} void GetNumPrograms(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1, 1, 0); // 0x00010040 + u32 media_type = rp.Pop(); - u32 media_type = cmd_buff[1] & 0xFF; - - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = am_titles_count[media_type]; - LOG_WARNING(Service_AM, "(STUBBED) media_type=%u, title_count=0x%08x", media_type, - am_titles_count[media_type]); + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); + rb.Push(am_title_list[media_type].size()); } void FindContentInfos(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1002, 4, 2); // 0x10020104 - u32 media_type = cmd_buff[1] & 0xFF; - u64 title_id = (static_cast(cmd_buff[3]) << 32) | cmd_buff[2]; - u32 content_ids_pointer = cmd_buff[6]; - u32 content_info_pointer = cmd_buff[8]; + auto media_type = static_cast(rp.Pop()); + u64 title_id = rp.Pop(); + u32 content_count = rp.Pop(); + VAddr content_requested_in = rp.PopMappedBuffer(); + VAddr content_info_out = rp.PopMappedBuffer(); - am_content_count[media_type] = cmd_buff[4]; + std::vector content_requested(content_count); + Memory::ReadBlock(content_requested_in, content_requested.data(), content_count * sizeof(u16)); - cmd_buff[1] = RESULT_SUCCESS.raw; - LOG_WARNING(Service_AM, "(STUBBED) media_type=%u, title_id=0x%016llx, content_cound=%u, " - "content_ids_pointer=0x%08x, content_info_pointer=0x%08x", - media_type, title_id, am_content_count[media_type], content_ids_pointer, - content_info_pointer); + std::string tmd_path = GetTitleMetadataPath(media_type, title_id); + + u32 content_read = 0; + FileSys::TitleMetadata tmd(tmd_path); + if (tmd.Load() == Loader::ResultStatus::Success) { + // Get info for each content index requested + for (size_t i = 0; i < content_count; i++) { + std::shared_ptr romfs_file; + u64 romfs_offset = 0; + u64 romfs_size = 0; + + FileSys::NCCHContainer ncch_container(GetTitleContentPath(media_type, title_id, i)); + ncch_container.ReadRomFS(romfs_file, romfs_offset, romfs_size); + + ContentInfo content_info = {}; + content_info.index = static_cast(i); + content_info.type = tmd.GetContentTypeByIndex(content_requested[i]); + content_info.content_id = tmd.GetContentIDByIndex(content_requested[i]); + content_info.size = tmd.GetContentSizeByIndex(content_requested[i]); + content_info.romfs_size = romfs_size; + + Memory::WriteBlock(content_info_out, &content_info, sizeof(ContentInfo)); + content_info_out += sizeof(ContentInfo); + content_read++; + } + } + + IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); + rb.Push(RESULT_SUCCESS); } void ListContentInfos(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1003, 5, 1); // 0x10030142 + u32 content_count = rp.Pop(); + auto media_type = static_cast(rp.Pop()); + u64 title_id = rp.Pop(); + u32 start_index = rp.Pop(); + VAddr content_info_out = rp.PopMappedBuffer(); - u32 media_type = cmd_buff[2] & 0xFF; - u64 title_id = (static_cast(cmd_buff[4]) << 32) | cmd_buff[3]; - u32 start_index = cmd_buff[5]; - u32 content_info_pointer = cmd_buff[7]; + std::string tmd_path = GetTitleMetadataPath(media_type, title_id); - am_content_count[media_type] = cmd_buff[1]; + u32 copied = 0; + FileSys::TitleMetadata tmd(tmd_path); + if (tmd.Load() == Loader::ResultStatus::Success) { + copied = std::min(content_count, static_cast(tmd.GetContentCount())); + for (u32 i = start_index; i < copied; i++) { + std::shared_ptr romfs_file; + u64 romfs_offset = 0; + u64 romfs_size = 0; - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = am_content_count[media_type]; - LOG_WARNING(Service_AM, "(STUBBED) media_type=%u, content_count=%u, title_id=0x%016" PRIx64 - ", start_index=0x%08x, content_info_pointer=0x%08X", - media_type, am_content_count[media_type], title_id, start_index, - content_info_pointer); + FileSys::NCCHContainer ncch_container(GetTitleContentPath(media_type, title_id, i)); + ncch_container.ReadRomFS(romfs_file, romfs_offset, romfs_size); + + ContentInfo content_info = {}; + content_info.index = static_cast(i); + content_info.type = tmd.GetContentTypeByIndex(i); + content_info.content_id = tmd.GetContentIDByIndex(i); + content_info.size = tmd.GetContentSizeByIndex(i); + content_info.romfs_size = romfs_size; + + Memory::WriteBlock(content_info_out, &content_info, sizeof(ContentInfo)); + content_info_out += sizeof(ContentInfo); + } + } + + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); + rb.Push(copied); } void DeleteContents(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1004, 4, 1); // 0x10040102 + u8 media_type = rp.Pop(); + u64 title_id = rp.Pop(); + u32 content_count = rp.Pop(); + VAddr content_ids_in = rp.PopMappedBuffer(); - u32 media_type = cmd_buff[1] & 0xFF; - u64 title_id = (static_cast(cmd_buff[3]) << 32) | cmd_buff[2]; - u32 content_ids_pointer = cmd_buff[6]; - - am_content_count[media_type] = cmd_buff[4]; - - cmd_buff[1] = RESULT_SUCCESS.raw; + IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); + rb.Push(RESULT_SUCCESS); LOG_WARNING(Service_AM, "(STUBBED) media_type=%u, title_id=0x%016" PRIx64 - ", content_count=%u, content_ids_pointer=0x%08x", - media_type, title_id, am_content_count[media_type], content_ids_pointer); + ", content_count=%u, content_ids_in=0x%08x", + media_type, title_id, content_count, content_ids_in); } void GetProgramList(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 2, 2, 1); // 0x00020082 - u32 media_type = cmd_buff[2] & 0xFF; - u32 title_ids_output_pointer = cmd_buff[4]; + u32 count = rp.Pop(); + u8 media_type = rp.Pop(); + VAddr title_ids_output_pointer = rp.PopMappedBuffer(); - am_titles_list_count[media_type] = cmd_buff[1]; + if (!Memory::IsValidVirtualAddress(title_ids_output_pointer) || media_type > 2) { + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(-1); // TODO(shinyquagsire23): Find the right error code + rb.Push(0); + return; + } - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = am_titles_list_count[media_type]; - LOG_WARNING( - Service_AM, - "(STUBBED) media_type=%u, titles_list_count=0x%08X, title_ids_output_pointer=0x%08X", - media_type, am_titles_list_count[media_type], title_ids_output_pointer); + u32 media_count = static_cast(am_title_list[media_type].size()); + u32 copied = std::min(media_count, count); + + Memory::WriteBlock(title_ids_output_pointer, am_title_list[media_type].data(), + copied * sizeof(u64)); + + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); + rb.Push(copied); } void GetProgramInfos(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 3, 2, 2); // 0x00030084 - u32 media_type = cmd_buff[1] & 0xFF; - u32 title_id_list_pointer = cmd_buff[4]; - u32 title_list_pointer = cmd_buff[6]; + auto media_type = static_cast(rp.Pop()); + u32 title_count = rp.Pop(); + VAddr title_id_list_pointer = rp.PopMappedBuffer(); + VAddr title_info_out = rp.PopMappedBuffer(); - am_titles_count[media_type] = cmd_buff[2]; + std::vector title_id_list(title_count); + Memory::ReadBlock(title_id_list_pointer, title_id_list.data(), title_count * sizeof(u64)); - cmd_buff[1] = RESULT_SUCCESS.raw; - LOG_WARNING(Service_AM, "(STUBBED) media_type=%u, total_titles=0x%08X, " - "title_id_list_pointer=0x%08X, title_list_pointer=0x%08X", - media_type, am_titles_count[media_type], title_id_list_pointer, title_list_pointer); + for (u32 i = 0; i < title_count; i++) { + std::string tmd_path = GetTitleMetadataPath(media_type, title_id_list[i]); + + TitleInfo title_info = {}; + title_info.tid = title_id_list[i]; + + FileSys::TitleMetadata tmd(tmd_path); + if (tmd.Load() == Loader::ResultStatus::Success) { + // TODO(shinyquagsire23): This is the total size of all files this process owns, + // including savefiles and other content. This comes close but is off. + title_info.size = tmd.GetContentSizeByIndex(FileSys::TMDContentIndex::Main); + title_info.version = tmd.GetTitleVersion(); + title_info.type = tmd.GetTitleType(); + } + Memory::WriteBlock(title_info_out, &title_info, sizeof(TitleInfo)); + title_info_out += sizeof(TitleInfo); + } + + IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); + rb.Push(RESULT_SUCCESS); } void GetDataTitleInfos(Service::Interface* self) { @@ -123,60 +342,126 @@ void GetDataTitleInfos(Service::Interface* self) { } void ListDataTitleTicketInfos(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1007, 4, 1); // 0x10070102 + u32 ticket_count = rp.Pop(); + u64 title_id = rp.Pop(); + u32 start_index = rp.Pop(); + VAddr ticket_info_out = rp.PopMappedBuffer(); + VAddr ticket_info_write = ticket_info_out; - u64 title_id = (static_cast(cmd_buff[3]) << 32) | cmd_buff[2]; - u32 start_index = cmd_buff[4]; - u32 ticket_info_pointer = cmd_buff[6]; + for (u32 i = 0; i < ticket_count; i++) { + TicketInfo ticket_info = {}; + ticket_info.title_id = title_id; + ticket_info.version = 0; // TODO + ticket_info.size = 0; // TODO - am_ticket_count = cmd_buff[1]; + Memory::WriteBlock(ticket_info_write, &ticket_info, sizeof(TicketInfo)); + ticket_info_write += sizeof(TicketInfo); + } + + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); + rb.Push(ticket_count); - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = am_ticket_count; LOG_WARNING(Service_AM, "(STUBBED) ticket_count=0x%08X, title_id=0x%016" PRIx64 - ", start_index=0x%08X, ticket_info_pointer=0x%08X", - am_ticket_count, title_id, start_index, ticket_info_pointer); + ", start_index=0x%08X, ticket_info_out=0x%08X", + ticket_count, title_id, start_index, ticket_info_out); } void GetNumContentInfos(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1001, 3, 0); // 0x100100C0 + auto media_type = static_cast(rp.Pop()); + u64 title_id = rp.Pop(); - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = 1; // Number of content infos plus one - LOG_WARNING(Service_AM, "(STUBBED) called"); + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); // No error + + std::string tmd_path = GetTitleMetadataPath(media_type, title_id); + + FileSys::TitleMetadata tmd(tmd_path); + if (tmd.Load() == Loader::ResultStatus::Success) { + rb.Push(tmd.GetContentCount()); + } else { + rb.Push(1); // Number of content infos plus one + LOG_WARNING(Service_AM, "(STUBBED) called media_type=%u, title_id=0x%016" PRIx64, + media_type, title_id); + } } void DeleteTicket(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 7, 2, 0); // 0x00070080 + u64 title_id = rp.Pop(); - u64 title_id = (static_cast(cmd_buff[2]) << 32) | cmd_buff[1]; - - cmd_buff[1] = RESULT_SUCCESS.raw; + IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); + rb.Push(RESULT_SUCCESS); LOG_WARNING(Service_AM, "(STUBBED) called title_id=0x%016" PRIx64 "", title_id); } void GetNumTickets(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 8, 0, 0); // 0x00080000 + u32 ticket_count = 0; - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = am_ticket_count; - LOG_WARNING(Service_AM, "(STUBBED) called ticket_count=0x%08x", am_ticket_count); + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); + rb.Push(ticket_count); + LOG_WARNING(Service_AM, "(STUBBED) called ticket_count=0x%08x", ticket_count); } void GetTicketList(Service::Interface* self) { - u32* cmd_buff = Kernel::GetCommandBuffer(); + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 9, 2, 1); // 0x00090082 + u32 ticket_list_count = rp.Pop(); + u32 ticket_index = rp.Pop(); + VAddr ticket_tids_out = rp.PopMappedBuffer(); - u32 num_of_skip = cmd_buff[2]; - u32 ticket_list_pointer = cmd_buff[4]; + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); + rb.Push(ticket_list_count); + LOG_WARNING(Service_AM, + "(STUBBED) ticket_list_count=0x%08x, ticket_index=0x%08x, ticket_tids_out=0x%08x", + ticket_list_count, ticket_index, ticket_tids_out); +} - am_ticket_list_count = cmd_buff[1]; +void QueryAvailableTitleDatabase(Service::Interface* self) { + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x19, 1, 0); // 0x190040 + u8 media_type = rp.Pop(); - cmd_buff[1] = RESULT_SUCCESS.raw; - cmd_buff[2] = am_ticket_list_count; - LOG_WARNING( - Service_AM, - "(STUBBED) ticket_list_count=0x%08x, num_of_skip=0x%08x, ticket_list_pointer=0x%08x", - am_ticket_list_count, num_of_skip, ticket_list_pointer); + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); // No error + rb.Push(true); + + LOG_WARNING(Service_APT, "(STUBBED) media_type=%u", media_type); +} + +void CheckContentRights(Service::Interface* self) { + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x25, 3, 0); // 0x2500C0 + u64 tid = rp.Pop(); + u16 content_index = rp.Pop(); + + // TODO(shinyquagsire23): Read tickets for this instead? + bool has_rights = + FileUtil::Exists(GetTitleContentPath(Service::FS::MediaType::SDMC, tid, content_index)); + + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); // No error + rb.Push(has_rights); + + LOG_WARNING(Service_APT, "(STUBBED) tid=%016" PRIx64 ", content_index=%u", tid, content_index); +} + +void CheckContentRightsIgnorePlatform(Service::Interface* self) { + IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x2D, 3, 0); // 0x2D00C0 + u64 tid = rp.Pop(); + u16 content_index = rp.Pop(); + + // TODO(shinyquagsire23): Read tickets for this instead? + bool has_rights = + FileUtil::Exists(GetTitleContentPath(Service::FS::MediaType::SDMC, tid, content_index)); + + IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); + rb.Push(RESULT_SUCCESS); // No error + rb.Push(has_rights); + + LOG_WARNING(Service_APT, "(STUBBED) tid=%016" PRIx64 ", content_index=%u", tid, content_index); } void Init() { @@ -184,6 +469,8 @@ void Init() { AddService(new AM_NET_Interface); AddService(new AM_SYS_Interface); AddService(new AM_U_Interface); + + ScanForAllTitles(); } void Shutdown() {} diff --git a/src/core/hle/service/am/am.h b/src/core/hle/service/am/am.h index 9bc2ca305..6ee4908b3 100644 --- a/src/core/hle/service/am/am.h +++ b/src/core/hle/service/am/am.h @@ -4,12 +4,64 @@ #pragma once +#include +#include "common/common_types.h" + +namespace Service { +namespace FS { +enum class MediaType : u32; +} +} + namespace Service { class Interface; namespace AM { +/** + * Get the .tmd path for a title + * @param media_type the media the title exists on + * @param tid the title ID to get + * @returns string path to the .tmd file if it exists, otherwise a path to create one is given. + */ +std::string GetTitleMetadataPath(Service::FS::MediaType media_type, u64 tid); + +/** + * Get the .app path for a title's installed content index. + * @param media_type the media the title exists on + * @param tid the title ID to get + * @param index the content index to get + * @returns string path to the .app file + */ +std::string GetTitleContentPath(Service::FS::MediaType media_type, u64 tid, u16 index = 0); + +/** + * Get the folder for a title's installed content. + * @param media_type the media the title exists on + * @param tid the title ID to get + * @returns string path to the title folder + */ +std::string GetTitlePath(Service::FS::MediaType media_type, u64 tid); + +/** + * Get the title/ folder for a storage medium. + * @param media_type the storage medium to get the path for + * @returns string path to the folder + */ +std::string GetMediaTitlePath(Service::FS::MediaType media_type); + +/** + * Scans the for titles in a storage medium for listing. + * @param media_type the storage medium to scan + */ +void ScanForTitles(Service::FS::MediaType media_type); + +/** + * Scans all storage mediums for titles for listing. + */ +void ScanForAllTitles(); + /** * AM::GetNumPrograms service function * Gets the number of installed titles in the requested media type @@ -154,6 +206,38 @@ void GetNumTickets(Service::Interface* self); */ void GetTicketList(Service::Interface* self); +/** + * AM::QueryAvailableTitleDatabase service function + * Inputs: + * 1 : Media Type + * Outputs: + * 1 : Result, 0 on success, otherwise error code + * 2 : Boolean, database availability + */ +void QueryAvailableTitleDatabase(Service::Interface* self); + +/** + * AM::CheckContentRights service function + * Inputs: + * 1-2 : Title ID + * 3 : Content Index + * Outputs: + * 1 : Result, 0 on success, otherwise error code + * 2 : Boolean, whether we have rights to this content + */ +void CheckContentRights(Service::Interface* self); + +/** + * AM::CheckContentRightsIgnorePlatform service function + * Inputs: + * 1-2 : Title ID + * 3 : Content Index + * Outputs: + * 1 : Result, 0 on success, otherwise error code + * 2 : Boolean, whether we have rights to this content + */ +void CheckContentRightsIgnorePlatform(Service::Interface* self); + /// Initialize AM service void Init(); diff --git a/src/core/hle/service/am/am_sys.cpp b/src/core/hle/service/am/am_sys.cpp index 949b3591d..5112a6a0d 100644 --- a/src/core/hle/service/am/am_sys.cpp +++ b/src/core/hle/service/am/am_sys.cpp @@ -33,7 +33,7 @@ const Interface::FunctionInfo FunctionTable[] = { {0x00160000, nullptr, "DeleteAllTemporaryPrograms"}, {0x00170044, nullptr, "ImportTwlBackupLegacy"}, {0x00180080, nullptr, "InitializeTitleDatabase"}, - {0x00190040, nullptr, "QueryAvailableTitleDatabase"}, + {0x00190040, QueryAvailableTitleDatabase, "QueryAvailableTitleDatabase"}, {0x001A00C0, nullptr, "CalcTwlBackupSize"}, {0x001B0144, nullptr, "ExportTwlBackup"}, {0x001C0084, nullptr, "ImportTwlBackup"}, @@ -45,7 +45,7 @@ const Interface::FunctionInfo FunctionTable[] = { {0x00220080, nullptr, "DeleteAllImportContextsFiltered"}, {0x00230080, nullptr, "GetNumImportTitleContextsFiltered"}, {0x002400C2, nullptr, "GetImportTitleContextListFiltered"}, - {0x002500C0, nullptr, "CheckContentRights"}, + {0x002500C0, CheckContentRights, "CheckContentRights"}, {0x00260044, nullptr, "GetTicketLimitInfos"}, {0x00270044, nullptr, "GetDemoLaunchInfos"}, {0x00280108, nullptr, "ReadTwlBackupInfoEx"}, @@ -53,7 +53,7 @@ const Interface::FunctionInfo FunctionTable[] = { {0x002A00C0, nullptr, "GetNumExistingContentInfosSystem"}, {0x002B0142, nullptr, "ListExistingContentInfosSystem"}, {0x002C0084, nullptr, "GetProgramInfosIgnorePlatform"}, - {0x002D00C0, nullptr, "CheckContentRightsIgnorePlatform"}, + {0x002D00C0, CheckContentRightsIgnorePlatform, "CheckContentRightsIgnorePlatform"}, {0x100100C0, GetNumContentInfos, "GetNumContentInfos"}, {0x10020104, FindContentInfos, "FindContentInfos"}, {0x10030142, ListContentInfos, "ListContentInfos"},