Cool_Tools/citra-mk7
1
0
Fork 0
mirror of https://github.com/PabloMK7/citra.git synced 2024-11-28 01:20:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

am: Improve code logic

Browse Source
This commit is contained in:
R-YaTian 2024-10-12 12:40:00 +08:00 committed by GitHub
parent c1e8600e94
commit 6f18e6a39f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/core/hle/service/am/am.cpp
View File

@ -45,6 +45,7 @@ constexpr u16 CATEGORY_DLP = 0x0001;
constexpr u8 VARIATION_SYSTEM = 0x02; constexpr u8 VARIATION_SYSTEM = 0x02;
constexpr u32 TID_HIGH_UPDATE = 0x0004000E; constexpr u32 TID_HIGH_UPDATE = 0x0004000E;
constexpr u32 TID_HIGH_DLC = 0x0004008C; constexpr u32 TID_HIGH_DLC = 0x0004008C;
constexpr u16 MAX_CONTENT_COUNT = 255;
struct TitleInfo { struct TitleInfo {
u64_le tid; u64_le tid;
@ -194,7 +195,7 @@ Result CIAFile::WriteTitleMetadata() {
// TODO: Correct error code. // TODO: Correct error code.
return FileSys::ResultFileNotFound; return FileSys::ResultFileNotFound;
} }
if (content_count > 255) if (content_count > MAX_CONTENT_COUNT)
file.Close(); file.Close();
} }
@ -226,8 +227,6 @@ ResultVal<std::size_t> CIAFile::WriteContentData(u64 offset, std::size_t length,
// has been written since we might get a written buffer which contains multiple .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. // contents or only part of a larger .app's contents.
const u64 offset_max = offset + length; const u64 offset_max = offset + length;
if (content_written.size() > 255)
content_files.clear();
for (std::size_t i = 0; i < content_written.size(); i++) { for (std::size_t i = 0; i < content_written.size(); i++) {
if (content_written[i] < container.GetContentSize(i)) { if (content_written[i] < container.GetContentSize(i)) {
// The size, minimum unwritten offset, and maximum unwritten offset of this content // The size, minimum unwritten offset, and maximum unwritten offset of this content
@ -247,9 +246,12 @@ ResultVal<std::size_t> CIAFile::WriteContentData(u64 offset, std::size_t length,
// Since the incoming TMD has already been written, we can use GetTitleContentPath // Since the incoming TMD has already been written, we can use GetTitleContentPath
// to get the content paths to write to. // to get the content paths to write to.
FileSys::TitleMetadata tmd = container.GetTitleMetadata(); FileSys::TitleMetadata tmd = container.GetTitleMetadata();
auto path = GetTitleContentPath(media_type, tmd.GetTitleID(), i, is_update); auto& file = content_files[i];
auto& file = content_written.size() > 255 ? content_files.emplace_back(path, "ab+") if (content_written.size() > MAX_CONTENT_COUNT) {
: content_files[i]; auto path = GetTitleContentPath(media_type, tmd.GetTitleID(), i, is_update);
FileUtil::IOFile fp(path, "ab+");
file = std::move(fp);
}
std::vector<u8> temp(buffer + (range_min - offset), std::vector<u8> temp(buffer + (range_min - offset),
buffer + (range_min - offset) + available_to_write); buffer + (range_min - offset) + available_to_write);
@ -265,8 +267,9 @@ ResultVal<std::size_t> CIAFile::WriteContentData(u64 offset, std::size_t length,
content_written[i] += available_to_write; content_written[i] += available_to_write;
LOG_DEBUG(Service_AM, "Wrote {:x} to content {}, total {:x}", available_to_write, i, LOG_DEBUG(Service_AM, "Wrote {:x} to content {}, total {:x}", available_to_write, i,
content_written[i]); content_written[i]);
if (content_written.size() > 255) if (content_written.size() > MAX_CONTENT_COUNT) {
file.Close(); file.Close();
}
} }
} }