// Copyright 2015 Citra Emulator Project // Licensed under GPLv2 or any later version // Refer to the license.txt file included. #include #include #include #include #include #include "common/file_util.h" #include "common/logging/log.h" #include "common/string_util.h" #include "core/file_sys/errors.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/kernel/client_port.h" #include "core/hle/kernel/client_session.h" #include "core/hle/kernel/errors.h" #include "core/hle/kernel/handle_table.h" #include "core/hle/kernel/server_session.h" #include "core/hle/kernel/session.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" #include "core/loader/smdh.h" namespace Service { namespace AM { constexpr u16 PLATFORM_CTR = 0x0004; constexpr u16 CATEGORY_SYSTEM = 0x0010; constexpr u16 CATEGORY_DLP = 0x0001; constexpr u8 VARIATION_SYSTEM = 0x02; constexpr u32 TID_HIGH_UPDATE = 0x0004000E; constexpr u32 TID_HIGH_DLC = 0x0004008C; // CIA installation static context variables static bool cia_installing = false; static u64 cia_installing_tid; static Service::FS::MediaType cia_installing_media_type; 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"); ResultVal CIAFile::Read(u64 offset, size_t length, u8* buffer) const { UNIMPLEMENTED(); return MakeResult(length); } ResultVal CIAFile::WriteTitleMetadata(u64 offset, size_t length, const u8* buffer) { container.LoadTitleMetadata(data, container.GetTitleMetadataOffset()); FileSys::TitleMetadata tmd = container.GetTitleMetadata(); cia_installing_tid = tmd.GetTitleID(); tmd.Print(); // If a TMD already exists for this app (ie 00000000.tmd), the incoming TMD // will be the same plus one, (ie 00000001.tmd), both will be kept until // the install is finalized and old contents can be discarded. if (FileUtil::Exists(GetTitleMetadataPath(media_type, tmd.GetTitleID()))) is_update = true; std::string tmd_path = GetTitleMetadataPath(media_type, tmd.GetTitleID(), is_update); // Create content/ folder if it doesn't exist std::string tmd_folder; Common::SplitPath(tmd_path, &tmd_folder, nullptr, nullptr); FileUtil::CreateFullPath(tmd_folder); // Save TMD so that we can start getting new .app paths if (tmd.Save(tmd_path) != Loader::ResultStatus::Success) return FileSys::ERROR_INSUFFICIENT_SPACE; // Create any other .app folders which may not exist yet std::string app_folder; Common::SplitPath(GetTitleContentPath(media_type, tmd.GetTitleID(), FileSys::TMDContentIndex::Main, is_update), &app_folder, nullptr, nullptr); FileUtil::CreateFullPath(app_folder); content_written.resize(container.GetTitleMetadata().GetContentCount()); install_state = CIAInstallState::TMDLoaded; return MakeResult(length); } ResultVal CIAFile::WriteContentData(u64 offset, size_t length, const u8* buffer) { // Data is not being buffered, so we have to keep track of how much of each .app // has been written since we might get a written buffer which contains multiple .app // contents or only part of a larger .app's contents. u64 offset_max = offset + length; for (int i = 0; i < container.GetTitleMetadata().GetContentCount(); i++) { if (content_written[i] < container.GetContentSize(i)) { // The size, minimum unwritten offset, and maximum unwritten offset of this content u64 size = container.GetContentSize(i); u64 range_min = container.GetContentOffset(i) + content_written[i]; u64 range_max = container.GetContentOffset(i) + size; // The unwritten range for this content is beyond the buffered data we have // or comes before the buffered data we have, so skip this content ID. if (range_min > offset_max || range_max < offset) continue; // Figure out how much of this content ID we have just recieved/can write out u64 available_to_write = std::min(offset_max, range_max) - range_min; // Since the incoming TMD has already been written, we can use GetTitleContentPath // to get the content paths to write to. FileSys::TitleMetadata tmd = container.GetTitleMetadata(); FileUtil::IOFile file(GetTitleContentPath(media_type, tmd.GetTitleID(), i, is_update), content_written[i] ? "ab" : "wb"); if (!file.IsOpen()) return FileSys::ERROR_INSUFFICIENT_SPACE; file.WriteBytes(buffer + (range_min - offset), available_to_write); // Keep tabs on how much of this content ID has been written so new range_min // values can be calculated. content_written[i] += available_to_write; LOG_DEBUG(Service_AM, "Wrote %" PRIx64 " to content %u, total %" PRIx64, available_to_write, i, content_written[i]); } } return MakeResult(length); } ResultVal CIAFile::Write(u64 offset, size_t length, bool flush, const u8* buffer) { written += length; // TODO(shinyquagsire23): Can we assume that things will only be written in sequence? // Does AM send an error if we write to things out of order? // Or does it just ignore offsets and assume a set sequence of incoming data? // The data in CIAs is always stored CIA Header > Cert > Ticket > TMD > Content > Meta. // The CIA Header describes Cert, Ticket, TMD, total content sizes, and TMD is needed for // content sizes so it ends up becoming a problem of keeping track of how much has been // written and what we have been able to pick up. if (install_state == CIAInstallState::InstallStarted) { size_t buf_copy_size = std::min(length, FileSys::CIA_HEADER_SIZE); size_t buf_max_size = std::min(static_cast(offset + length), FileSys::CIA_HEADER_SIZE); data.resize(buf_max_size); memcpy(data.data() + offset, buffer, buf_copy_size); // We have enough data to load a CIA header and parse it. if (written >= FileSys::CIA_HEADER_SIZE) { container.LoadHeader(data); container.Print(); install_state = CIAInstallState::HeaderLoaded; } } // If we don't have a header yet, we can't pull offsets of other sections if (install_state == CIAInstallState::InstallStarted) return MakeResult(length); // If we have been given data before (or including) .app content, pull it into // our buffer, but only pull *up to* the content offset, no further. if (offset < container.GetContentOffset()) { size_t buf_loaded = data.size(); size_t copy_offset = std::max(static_cast(offset), buf_loaded); size_t buf_offset = buf_loaded - offset; size_t buf_copy_size = std::min(length, static_cast(container.GetContentOffset() - offset)) - buf_loaded; size_t buf_max_size = std::min(offset + length, container.GetContentOffset()); data.resize(buf_max_size); memcpy(data.data() + copy_offset, buffer + buf_offset, buf_copy_size); } // TODO(shinyquagsire23): Write out .tik files to nand? // The end of our TMD is at the beginning of Content data, so ensure we have that much // buffered before trying to parse. if (written >= container.GetContentOffset() && install_state != CIAInstallState::TMDLoaded) { auto result = WriteTitleMetadata(offset, length, buffer); if (result.Failed()) return result; } // Content data sizes can only be retrieved from TMD data if (install_state != CIAInstallState::TMDLoaded) return MakeResult(length); // From this point forward, data will no longer be buffered in data auto result = WriteContentData(offset, length, buffer); if (result.Failed()) return result; return MakeResult(length); } u64 CIAFile::GetSize() const { return written; } bool CIAFile::SetSize(u64 size) const { return false; } bool CIAFile::Close() const { return true; } void CIAFile::Flush() const {} Service::FS::MediaType GetTitleMediaType(u64 titleId) { u16 platform = static_cast(titleId >> 48); u16 category = static_cast((titleId >> 32) & 0xFFFF); u8 variation = static_cast(titleId & 0xFF); if (platform != PLATFORM_CTR) return Service::FS::MediaType::NAND; if (category & CATEGORY_SYSTEM || category & CATEGORY_DLP || variation & VARIATION_SYSTEM) return Service::FS::MediaType::NAND; return Service::FS::MediaType::SDMC; } std::string GetTitleMetadataPath(Service::FS::MediaType media_type, u64 tid, bool update) { 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, the smallest will be the // base ID and the largest will be the (currently installing) update ID. constexpr u32 MAX_TMD_ID = 0xFFFFFFFF; u32 base_id = MAX_TMD_ID; u32 update_id = 0; 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") { u32 id = std::stoul(filename_filename.c_str(), nullptr, 16); base_id = std::min(base_id, id); update_id = std::max(update_id, id); } } // If we didn't find anything, default to 00000000.tmd for it to be created. if (base_id == MAX_TMD_ID) base_id = 0; // Update ID should be one more than the last, if it hasn't been created yet. if (base_id == update_id) update_id++; return content_path + Common::StringFromFormat("%08x.tmd", (update ? update_id : base_id)); } std::string GetTitleContentPath(Service::FS::MediaType media_type, u64 tid, u16 index, bool update) { 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, update); u32 content_id = 0; FileSys::TitleMetadata tmd; if (tmd.Load(tmd_path) == 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) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1, 1, 0); // 0x00010040 u32 media_type = rp.Pop(); IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); rb.Push(RESULT_SUCCESS); rb.Push(am_title_list[media_type].size()); } void FindDLCContentInfos(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1002, 4, 4); // 0x10020104 auto media_type = static_cast(rp.Pop()); u64 title_id = rp.Pop(); u32 content_count = rp.Pop(); size_t input_buffer_size, output_buffer_size; IPC::MappedBufferPermissions input_buffer_perms, output_buffer_perms; VAddr content_requested_in = rp.PopMappedBuffer(&input_buffer_size, &input_buffer_perms); VAddr content_info_out = rp.PopMappedBuffer(&output_buffer_size, &output_buffer_perms); // Validate that only DLC TIDs are passed in u32 tid_high = static_cast(title_id >> 32); if (tid_high != TID_HIGH_DLC) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 4); rb.Push(ResultCode(ErrCodes::InvalidTIDInList, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Usage)); rb.PushMappedBuffer(content_requested_in, input_buffer_size, input_buffer_perms); rb.PushMappedBuffer(content_info_out, output_buffer_size, output_buffer_perms); return; } std::vector content_requested(content_count); Memory::ReadBlock(content_requested_in, content_requested.data(), content_count * sizeof(u16)); std::string tmd_path = GetTitleMetadataPath(media_type, title_id); u32 content_read = 0; FileSys::TitleMetadata tmd; if (tmd.Load(tmd_path) == 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, 4); rb.Push(RESULT_SUCCESS); rb.PushMappedBuffer(content_requested_in, input_buffer_size, input_buffer_perms); rb.PushMappedBuffer(content_info_out, output_buffer_size, output_buffer_perms); } void ListDLCContentInfos(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1003, 5, 2); // 0x10030142 u32 content_count = rp.Pop(); auto media_type = static_cast(rp.Pop()); u64 title_id = rp.Pop(); u32 start_index = rp.Pop(); size_t output_buffer_size; IPC::MappedBufferPermissions output_buffer_perms; VAddr content_info_out = rp.PopMappedBuffer(&output_buffer_size, &output_buffer_perms); // Validate that only DLC TIDs are passed in u32 tid_high = static_cast(title_id >> 32); if (tid_high != TID_HIGH_DLC) { IPC::RequestBuilder rb = rp.MakeBuilder(2, 2); rb.Push(ResultCode(ErrCodes::InvalidTIDInList, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Usage)); rb.Push(0); rb.PushMappedBuffer(content_info_out, output_buffer_size, output_buffer_perms); return; } std::string tmd_path = GetTitleMetadataPath(media_type, title_id); u32 copied = 0; FileSys::TitleMetadata tmd; if (tmd.Load(tmd_path) == 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; 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, 2); rb.Push(RESULT_SUCCESS); rb.Push(copied); rb.PushMappedBuffer(content_info_out, output_buffer_size, output_buffer_perms); } void DeleteContents(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1004, 4, 2); // 0x10040102 u8 media_type = rp.Pop(); u64 title_id = rp.Pop(); u32 content_count = rp.Pop(); VAddr content_ids_in = rp.PopMappedBuffer(); 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_in=0x%08x", media_type, title_id, content_count, content_ids_in); } void GetProgramList(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 2, 2, 2); // 0x00020082 u32 count = rp.Pop(); u8 media_type = rp.Pop(); VAddr title_ids_output_pointer = rp.PopMappedBuffer(); 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; } 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); } ResultCode GetTitleInfoFromList(const std::vector& title_id_list, Service::FS::MediaType media_type, VAddr title_info_out) { for (u32 i = 0; i < title_id_list.size(); 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; if (tmd.Load(tmd_path) == 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(); } else { return ResultCode(ErrorDescription::NotFound, ErrorModule::AM, ErrorSummary::InvalidState, ErrorLevel::Permanent); } Memory::WriteBlock(title_info_out, &title_info, sizeof(TitleInfo)); title_info_out += sizeof(TitleInfo); } return RESULT_SUCCESS; } void GetProgramInfos(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 3, 2, 4); // 0x00030084 auto media_type = static_cast(rp.Pop()); u32 title_count = rp.Pop(); size_t title_id_list_size, title_info_size; IPC::MappedBufferPermissions title_id_list_perms, title_info_perms; VAddr title_id_list_pointer = rp.PopMappedBuffer(&title_id_list_size, &title_id_list_perms); VAddr title_info_out = rp.PopMappedBuffer(&title_info_size, &title_info_perms); std::vector title_id_list(title_count); Memory::ReadBlock(title_id_list_pointer, title_id_list.data(), title_count * sizeof(u64)); ResultCode result = GetTitleInfoFromList(title_id_list, media_type, title_info_out); IPC::RequestBuilder rb = rp.MakeBuilder(1, 4); rb.Push(result); rb.PushMappedBuffer(title_id_list_pointer, title_id_list_size, title_id_list_perms); rb.PushMappedBuffer(title_info_out, title_info_size, title_info_perms); } void GetDLCTitleInfos(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1005, 2, 4); // 0x10050084 auto media_type = static_cast(rp.Pop()); u32 title_count = rp.Pop(); size_t title_id_list_size, title_info_size; IPC::MappedBufferPermissions title_id_list_perms, title_info_perms; VAddr title_id_list_pointer = rp.PopMappedBuffer(&title_id_list_size, &title_id_list_perms); VAddr title_info_out = rp.PopMappedBuffer(&title_info_size, &title_info_perms); std::vector title_id_list(title_count); Memory::ReadBlock(title_id_list_pointer, title_id_list.data(), title_count * sizeof(u64)); ResultCode result = RESULT_SUCCESS; // Validate that DLC TIDs were passed in for (u32 i = 0; i < title_count; i++) { u32 tid_high = static_cast(title_id_list[i] >> 32); if (tid_high != TID_HIGH_DLC) { result = ResultCode(ErrCodes::InvalidTIDInList, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Usage); break; } } if (result.IsSuccess()) { result = GetTitleInfoFromList(title_id_list, media_type, title_info_out); } IPC::RequestBuilder rb = rp.MakeBuilder(1, 4); rb.Push(result); rb.PushMappedBuffer(title_id_list_pointer, title_id_list_size, title_id_list_perms); rb.PushMappedBuffer(title_info_out, title_info_size, title_info_perms); } void GetPatchTitleInfos(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x100D, 2, 4); // 0x100D0084 auto media_type = static_cast(rp.Pop()); u32 title_count = rp.Pop(); size_t title_id_list_size, title_info_size; IPC::MappedBufferPermissions title_id_list_perms, title_info_perms; VAddr title_id_list_pointer = rp.PopMappedBuffer(&title_id_list_size, &title_id_list_perms); VAddr title_info_out = rp.PopMappedBuffer(&title_info_size, &title_info_perms); std::vector title_id_list(title_count); Memory::ReadBlock(title_id_list_pointer, title_id_list.data(), title_count * sizeof(u64)); ResultCode result = RESULT_SUCCESS; // Validate that update TIDs were passed in for (u32 i = 0; i < title_count; i++) { u32 tid_high = static_cast(title_id_list[i] >> 32); if (tid_high != TID_HIGH_UPDATE) { result = ResultCode(ErrCodes::InvalidTIDInList, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Usage); break; } } if (result.IsSuccess()) { result = GetTitleInfoFromList(title_id_list, media_type, title_info_out); } IPC::RequestBuilder rb = rp.MakeBuilder(1, 4); rb.Push(result); rb.PushMappedBuffer(title_id_list_pointer, title_id_list_size, title_id_list_perms); rb.PushMappedBuffer(title_info_out, title_info_size, title_info_perms); } void ListDataTitleTicketInfos(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1007, 4, 4); // 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; 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 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); LOG_WARNING(Service_AM, "(STUBBED) ticket_count=0x%08X, title_id=0x%016" PRIx64 ", start_index=0x%08X, ticket_info_out=0x%08X", ticket_count, title_id, start_index, ticket_info_out); } void GetDLCContentInfoCount(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x1001, 3, 0); // 0x100100C0 auto media_type = static_cast(rp.Pop()); u64 title_id = rp.Pop(); // Validate that only DLC TIDs are passed in u32 tid_high = static_cast(title_id >> 32); if (tid_high != TID_HIGH_DLC) { IPC::RequestBuilder rb = rp.MakeBuilder(2, 2); rb.Push(ResultCode(ErrCodes::InvalidTID, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Usage)); rb.Push(0); return; } 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; if (tmd.Load(tmd_path) == 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, static_cast(media_type), title_id); } } void DeleteTicket(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 7, 2, 0); // 0x00070080 u64 title_id = rp.Pop(); 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) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 8, 0, 0); // 0x00080000 u32 ticket_count = 0; 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) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 9, 2, 2); // 0x00090082 u32 ticket_list_count = rp.Pop(); u32 ticket_index = rp.Pop(); VAddr ticket_tids_out = rp.PopMappedBuffer(); 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); } void QueryAvailableTitleDatabase(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x19, 1, 0); // 0x190040 u8 media_type = rp.Pop(); IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); rb.Push(RESULT_SUCCESS); // No error rb.Push(true); LOG_WARNING(Service_AM, "(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_AM, "(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_AM, "(STUBBED) tid=%016" PRIx64 ", content_index=%u", tid, content_index); } void BeginImportProgram(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x0402, 1, 0); // 0x04020040 auto media_type = static_cast(rp.Pop()); if (cia_installing) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::CIACurrentlyInstalling, ErrorModule::AM, ErrorSummary::InvalidState, ErrorLevel::Permanent)); return; } // Create our CIAFile handle for the app to write to, and while the app writes // Citra will store contents out to sdmc/nand const FileSys::Path cia_path = {}; auto file = std::make_shared(std::make_unique(media_type), cia_path); auto sessions = Kernel::ServerSession::CreateSessionPair(file->GetName()); file->ClientConnected(std::get>(sessions)); cia_installing = true; cia_installing_media_type = media_type; IPC::RequestBuilder rb = rp.MakeBuilder(1, 2); rb.Push(RESULT_SUCCESS); // No error rb.PushCopyHandles( Kernel::g_handle_table.Create(std::get>(sessions)) .Unwrap()); LOG_WARNING(Service_AM, "(STUBBED) media_type=%u", static_cast(media_type)); } void EndImportProgram(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x0405, 0, 2); // 0x04050002 auto cia_handle = rp.PopHandle(); // Clean up older content data if we installed newer content on top std::string old_tmd_path = GetTitleMetadataPath(cia_installing_media_type, cia_installing_tid, false); std::string new_tmd_path = GetTitleMetadataPath(cia_installing_media_type, cia_installing_tid, true); if (FileUtil::Exists(new_tmd_path) && old_tmd_path != new_tmd_path) { FileSys::TitleMetadata old_tmd; FileSys::TitleMetadata new_tmd; old_tmd.Load(old_tmd_path); new_tmd.Load(new_tmd_path); // For each content ID in the old TMD, check if there is a matching ID in the new // TMD. If a CIA contains (and wrote to) an identical ID, it should be kept while // IDs which only existed for the old TMD should be deleted. for (u16 old_index = 0; old_index < old_tmd.GetContentCount(); old_index++) { bool abort = false; for (u16 new_index = 0; new_index < new_tmd.GetContentCount(); new_index++) { if (old_tmd.GetContentIDByIndex(old_index) == new_tmd.GetContentIDByIndex(new_index)) { abort = true; } } if (abort) break; FileUtil::Delete( GetTitleContentPath(cia_installing_media_type, old_tmd.GetTitleID(), old_index)); } FileUtil::Delete(old_tmd_path); } ScanForAllTitles(); cia_installing = false; IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(RESULT_SUCCESS); } ResultVal> GetFileFromHandle(Kernel::Handle handle) { // Step up the chain from Handle->ClientSession->ServerSession and then // cast to File. For AM on 3DS, invalid handles actually hang the system. auto file_session = Kernel::g_handle_table.Get(handle); if (file_session == nullptr || file_session->parent == nullptr) { LOG_WARNING(Service_AM, "Invalid file handle!"); return Kernel::ERR_INVALID_HANDLE; } Kernel::SharedPtr server = file_session->parent->server; if (server == nullptr) { LOG_WARNING(Service_AM, "File handle ServerSession disconnected!"); return Kernel::ERR_SESSION_CLOSED_BY_REMOTE; } if (server->hle_handler != nullptr) { auto file = std::dynamic_pointer_cast(server->hle_handler); // TODO(shinyquagsire23): This requires RTTI, use service calls directly instead? if (file != nullptr) return MakeResult>(file); LOG_ERROR(Service_AM, "Failed to cast handle to FSFile!"); return Kernel::ERR_INVALID_HANDLE; } // Probably the best bet if someone is LLEing the fs service is to just have them LLE AM // while they're at it, so not implemented. LOG_ERROR(Service_AM, "Given file handle does not have an HLE handler!"); return Kernel::ERR_NOT_IMPLEMENTED; } void GetProgramInfoFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x0408, 1, 2); // 0x04080042 auto media_type = static_cast(rp.Pop()); // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } FileSys::TitleMetadata tmd = container.GetTitleMetadata(); TitleInfo title_info = {}; container.Print(); // TODO(shinyquagsire23): Sizes allegedly depend on the mediatype, and will double // on some mediatypes. Since this is more of a required install size we'll report // what Citra needs, but it would be good to be more accurate here. title_info.tid = tmd.GetTitleID(); title_info.size = tmd.GetContentSizeByIndex(FileSys::TMDContentIndex::Main); title_info.version = tmd.GetTitleVersion(); title_info.type = tmd.GetTitleType(); IPC::RequestBuilder rb = rp.MakeBuilder(8, 0); rb.Push(RESULT_SUCCESS); rb.PushRaw(title_info); } void GetSystemMenuDataFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x0409, 0, 4); // 0x04090004 // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } size_t output_buffer_size; IPC::MappedBufferPermissions output_buffer_perms; VAddr output_buffer = rp.PopMappedBuffer(&output_buffer_size, &output_buffer_perms); output_buffer_size = std::min(output_buffer_size, sizeof(Loader::SMDH)); auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } std::vector temp(output_buffer_size); // Read from the Meta offset + 0x400 for the 0x36C0-large SMDH auto read_result = file->backend->Read(container.GetMetadataOffset() + FileSys::CIA_METADATA_SIZE, output_buffer_size, temp.data()); if (read_result.Failed() || *read_result != output_buffer_size) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } Memory::WriteBlock(output_buffer, temp.data(), output_buffer_size); IPC::RequestBuilder rb = rp.MakeBuilder(1, 2); rb.PushMappedBuffer(output_buffer, output_buffer_size, output_buffer_perms); rb.Push(RESULT_SUCCESS); } void GetDependencyListFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x040A, 0, 2); // 0x040A0002 // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } size_t output_buffer_size; VAddr output_buffer = rp.PeekStaticBuffer(0, &output_buffer_size); output_buffer_size = std::min(output_buffer_size, FileSys::CIA_DEPENDENCY_SIZE); auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } Memory::WriteBlock(output_buffer, container.GetDependencies().data(), output_buffer_size); IPC::RequestBuilder rb = rp.MakeBuilder(1, 2); rb.Push(RESULT_SUCCESS); rb.PushStaticBuffer(output_buffer, output_buffer_size, 0); } void GetTransferSizeFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x040B, 0, 2); // 0x040B0002 // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } IPC::RequestBuilder rb = rp.MakeBuilder(3, 0); rb.Push(RESULT_SUCCESS); rb.Push(container.GetMetadataOffset()); } void GetCoreVersionFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x040C, 0, 2); // 0x040C0002 // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); rb.Push(RESULT_SUCCESS); rb.Push(container.GetCoreVersion()); } void GetRequiredSizeFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x040D, 1, 2); // 0x040D0042 auto media_type = static_cast(rp.Pop()); // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } // TODO(shinyquagsire23): Sizes allegedly depend on the mediatype, and will double // on some mediatypes. Since this is more of a required install size we'll report // what Citra needs, but it would be good to be more accurate here. IPC::RequestBuilder rb = rp.MakeBuilder(3, 0); rb.Push(RESULT_SUCCESS); rb.Push(container.GetTitleMetadata().GetContentSizeByIndex(FileSys::TMDContentIndex::Main)); } void GetMetaSizeFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x0413, 0, 2); // 0x04130002 // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } IPC::RequestBuilder rb = rp.MakeBuilder(2, 0); rb.Push(RESULT_SUCCESS); rb.Push(container.GetMetadataSize()); } void GetMetaDataFromCia(Service::Interface* self) { IPC::RequestParser rp(Kernel::GetCommandBuffer(), 0x0414, 0, 2); // 0x04140044 u32 output_size = rp.Pop(); // Get a File from our Handle auto file_res = GetFileFromHandle(rp.PopHandle()); if (!file_res.Succeeded()) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(file_res.Code()); return; } size_t output_buffer_size; VAddr output_buffer = rp.PeekStaticBuffer(0, &output_buffer_size); // Don't write beyond the actual static buffer size. output_size = std::min(static_cast(output_buffer_size), output_size); auto file = file_res.Unwrap(); FileSys::CIAContainer container; if (container.Load(*file->backend) != Loader::ResultStatus::Success) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } // Read from the Meta offset for the specified size std::vector temp(output_size); auto read_result = file->backend->Read(container.GetMetadataOffset(), output_size, temp.data()); if (read_result.Failed() || *read_result != output_size) { IPC::RequestBuilder rb = rp.MakeBuilder(1, 0); rb.Push(ResultCode(ErrCodes::InvalidCIAHeader, ErrorModule::AM, ErrorSummary::InvalidArgument, ErrorLevel::Permanent)); return; } Memory::WriteBlock(output_buffer, temp.data(), output_size); IPC::RequestBuilder rb = rp.MakeBuilder(1, 2); rb.Push(RESULT_SUCCESS); rb.PushStaticBuffer(output_buffer, output_buffer_size, 0); } void Init() { AddService(new AM_APP_Interface); AddService(new AM_NET_Interface); AddService(new AM_SYS_Interface); AddService(new AM_U_Interface); ScanForAllTitles(); } void Shutdown() { cia_installing = false; } } // namespace AM } // namespace Service