nold/Atmosphere
1
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere synced 2024-12-23 04:41:12 +00:00
Code Issues Projects Releases Packages Wiki Activity

tma.tio: Improve read/write bounding.

Browse source
This commit is contained in:
Michael Scire 2018-12-06 23:25:16 -08:00
parent f79f4d175b
commit 29153af2bc
1 changed files with 27 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
stratosphere/tma/source/target_io/tio_task.cpp
View file

@ -27,12 +27,22 @@ void TIOFileReadTask::OnStart(TmaPacket *packet) {
packet->Read<u64>(this->size_remaining); packet->Read<u64>(this->size_remaining);
packet->Read<u64>(this->cur_offset); packet->Read<u64>(this->cur_offset);
Result rc = 0;
if (strlen(path) == 0) { if (strlen(path) == 0) {
this->SendResult(0x202); rc = 0x202;
return;
} }
Result rc = dmntTargetIOFileOpen(&this->handle, path, FS_OPEN_READ, DmntTIOCreateOption_OpenExisting); if (R_SUCCEEDED(rc)) {
u64 file_size;
rc = dmntTargetIOFileGetSize(path, &file_size);
if (R_SUCCEEDED(rc)) {
if (file_size < this->cur_offset + this->size_remaining) {
this->size_remaining = file_size - this->cur_offset;
}
}
}
rc = dmntTargetIOFileOpen(&this->handle, path, FS_OPEN_READ, DmntTIOCreateOption_OpenExisting);
if (R_FAILED(rc)) { if (R_FAILED(rc)) {
this->SendResult(rc); this->SendResult(rc);
return; return;
@ -101,12 +111,23 @@ void TIOFileWriteTask::OnStart(TmaPacket *packet) {
packet->Read<u64>(this->size_remaining); packet->Read<u64>(this->size_remaining);
packet->Read<u64>(this->cur_offset); packet->Read<u64>(this->cur_offset);
Result rc = 0;
if (strlen(path) == 0) { if (strlen(path) == 0) {
this->SendResult(0x202); rc = 0x202;
return;
} }
Result rc = dmntTargetIOFileOpen(&this->handle, path, FS_OPEN_READ, DmntTIOCreateOption_OpenExisting); if (R_SUCCEEDED(rc)) {
u64 file_size;
rc = dmntTargetIOFileGetSize(path, &file_size);
if (R_SUCCEEDED(rc)) {
if (file_size < this->cur_offset + this->size_remaining) {
this->size_remaining = file_size - this->cur_offset;
}
}
}
rc = dmntTargetIOFileOpen(&this->handle, path, FS_OPEN_READ, DmntTIOCreateOption_OpenExisting);
if (R_FAILED(rc)) { if (R_FAILED(rc)) {
this->SendResult(rc); this->SendResult(rc);
} }