2018-09-07 16:00:13 +01:00
|
|
|
/*
|
|
|
|
* Copyright (c) 2018 Atmosphère-NX
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
|
|
* under the terms and conditions of the GNU General Public License,
|
|
|
|
* version 2, as published by the Free Software Foundation.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope it will be useful, but WITHOUT
|
|
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
|
|
* more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
2018-06-10 02:33:22 +01:00
|
|
|
#include <switch.h>
|
|
|
|
#include "fsmitm_service.hpp"
|
2018-06-10 18:11:05 +01:00
|
|
|
#include "fs_shim.h"
|
|
|
|
|
|
|
|
#include "fsmitm_utils.hpp"
|
|
|
|
#include "fsmitm_romstorage.hpp"
|
2018-06-15 00:50:01 +01:00
|
|
|
#include "fsmitm_layeredrom.hpp"
|
2018-06-10 18:11:05 +01:00
|
|
|
|
|
|
|
#include "debug.hpp"
|
2018-06-10 02:33:22 +01:00
|
|
|
|
2018-10-30 00:30:39 +00:00
|
|
|
void FsMitmService::PostProcess(IMitmServiceObject *obj, IpcResponseContext *ctx) {
|
|
|
|
auto this_ptr = static_cast<FsMitmService *>(obj);
|
|
|
|
switch ((FspSrvCmd)ctx->cmd_id) {
|
|
|
|
case FspSrvCmd_SetCurrentProcess:
|
|
|
|
if (R_SUCCEEDED(ctx->rc)) {
|
|
|
|
this_ptr->has_initialized = true;
|
|
|
|
this_ptr->process_id = ctx->request.Pid;
|
|
|
|
this_ptr->title_id = this_ptr->process_id;
|
|
|
|
if (R_FAILED(MitmQueryUtils::GetAssociatedTidForPid(this_ptr->process_id, &this_ptr->title_id))) {
|
|
|
|
/* Log here, if desired. */
|
|
|
|
}
|
2018-06-15 00:50:01 +01:00
|
|
|
break;
|
2018-06-10 10:06:50 +01:00
|
|
|
}
|
2018-06-19 19:07:31 +01:00
|
|
|
break;
|
|
|
|
default:
|
2018-06-10 10:06:50 +01:00
|
|
|
break;
|
|
|
|
}
|
2018-06-10 18:11:05 +01:00
|
|
|
}
|
|
|
|
|
2018-06-15 00:50:01 +01:00
|
|
|
/* Add redirection for RomFS to the SD card. */
|
2018-10-30 00:30:39 +00:00
|
|
|
Result FsMitmService::OpenDataStorageByCurrentProcess(Out<std::shared_ptr<IStorageInterface>> out_storage) {
|
|
|
|
std::shared_ptr<IStorageInterface> storage = nullptr;
|
2018-06-15 00:50:01 +01:00
|
|
|
u32 out_domain_id = 0;
|
2018-10-30 00:30:39 +00:00
|
|
|
Result rc = 0;
|
2018-10-30 05:14:34 +00:00
|
|
|
|
|
|
|
ON_SCOPE_EXIT {
|
|
|
|
if (R_SUCCEEDED(rc)) {
|
|
|
|
out_storage.SetValue(std::move(storage));
|
|
|
|
if (out_storage.IsDomain()) {
|
|
|
|
out_storage.ChangeObjectId(out_domain_id);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
2018-10-30 00:30:39 +00:00
|
|
|
|
2018-06-28 06:02:06 +01:00
|
|
|
if (this->romfs_storage != nullptr) {
|
2018-10-30 00:30:39 +00:00
|
|
|
if (out_storage.IsDomain()) {
|
2018-10-16 21:33:45 +01:00
|
|
|
FsStorage s = {0};
|
2018-10-30 00:30:39 +00:00
|
|
|
rc = fsOpenDataStorageByCurrentProcessFwd(this->forward_service.get(), &s);
|
2018-10-16 21:33:45 +01:00
|
|
|
if (R_SUCCEEDED(rc)) {
|
|
|
|
out_domain_id = s.s.object_id;
|
|
|
|
}
|
2018-06-28 06:02:06 +01:00
|
|
|
} else {
|
|
|
|
rc = 0;
|
|
|
|
}
|
2018-06-15 00:50:01 +01:00
|
|
|
if (R_SUCCEEDED(rc)) {
|
2018-10-30 00:30:39 +00:00
|
|
|
storage = this->romfs_storage;
|
2018-06-15 00:50:01 +01:00
|
|
|
}
|
2018-06-28 06:02:06 +01:00
|
|
|
} else {
|
|
|
|
FsStorage data_storage;
|
|
|
|
FsFile data_file;
|
2018-10-16 21:33:45 +01:00
|
|
|
|
2018-10-30 00:30:39 +00:00
|
|
|
rc = fsOpenDataStorageByCurrentProcessFwd(this->forward_service.get(), &data_storage);
|
2018-06-28 06:02:06 +01:00
|
|
|
|
|
|
|
Log(armGetTls(), 0x100);
|
|
|
|
if (R_SUCCEEDED(rc)) {
|
2018-11-04 20:45:29 +00:00
|
|
|
if (Utils::HasSdRomfsContent(this->title_id)) {
|
|
|
|
/* TODO: Is there a sensible path that ends in ".romfs" we can use?" */
|
|
|
|
if (R_SUCCEEDED(Utils::OpenSdFileForAtmosphere(this->title_id, "romfs.bin", FS_OPEN_READ, &data_file))) {
|
|
|
|
storage = std::make_shared<IStorageInterface>(new LayeredRomFS(std::make_shared<RomInterfaceStorage>(data_storage), std::make_shared<RomFileStorage>(data_file), this->title_id));
|
|
|
|
} else {
|
|
|
|
storage = std::make_shared<IStorageInterface>(new LayeredRomFS(std::make_shared<RomInterfaceStorage>(data_storage), nullptr, this->title_id));
|
|
|
|
}
|
|
|
|
this->romfs_storage = storage;
|
|
|
|
if (out_storage.IsDomain()) {
|
|
|
|
out_domain_id = data_storage.s.object_id;
|
|
|
|
}
|
2018-06-28 06:02:06 +01:00
|
|
|
} else {
|
2018-11-04 20:45:29 +00:00
|
|
|
/* If we don't have anything to modify, there's no sense in maintaining a copy of the metadata tables. */
|
|
|
|
fsStorageClose(&data_storage);
|
|
|
|
rc = RESULT_FORWARD_TO_SESSION;
|
2018-06-28 06:02:06 +01:00
|
|
|
}
|
2018-06-15 00:50:01 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-10-30 00:30:39 +00:00
|
|
|
return rc;
|
2018-06-15 00:50:01 +01:00
|
|
|
}
|
|
|
|
|
2018-06-10 18:11:05 +01:00
|
|
|
/* Add redirection for System Data Archives to the SD card. */
|
2018-10-30 13:29:30 +00:00
|
|
|
Result FsMitmService::OpenDataStorageByDataId(Out<std::shared_ptr<IStorageInterface>> out_storage, u64 data_id, u8 sid) {
|
2018-06-12 23:00:09 +01:00
|
|
|
FsStorageId storage_id = (FsStorageId)sid;
|
2018-06-10 18:11:05 +01:00
|
|
|
FsStorage data_storage;
|
|
|
|
FsFile data_file;
|
2018-10-30 00:30:39 +00:00
|
|
|
|
|
|
|
std::shared_ptr<IStorageInterface> storage = nullptr;
|
2018-06-15 00:50:01 +01:00
|
|
|
u32 out_domain_id = 0;
|
2018-10-30 00:30:39 +00:00
|
|
|
Result rc = 0;
|
2018-10-16 21:33:45 +01:00
|
|
|
|
2018-10-30 00:30:39 +00:00
|
|
|
ON_SCOPE_EXIT {
|
|
|
|
if (R_SUCCEEDED(rc)) {
|
|
|
|
out_storage.SetValue(std::move(storage));
|
|
|
|
if (out_storage.IsDomain()) {
|
|
|
|
out_storage.ChangeObjectId(out_domain_id);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
rc = fsOpenDataStorageByDataIdFwd(this->forward_service.get(), storage_id, data_id, &data_storage);
|
2018-10-16 21:33:45 +01:00
|
|
|
|
2018-06-10 18:11:05 +01:00
|
|
|
if (R_SUCCEEDED(rc)) {
|
2018-11-04 20:45:29 +00:00
|
|
|
if (Utils::HasSdRomfsContent(data_id)) {
|
|
|
|
/* TODO: Is there a sensible path that ends in ".romfs" we can use?" */
|
|
|
|
if (R_SUCCEEDED(Utils::OpenSdFileForAtmosphere(data_id, "romfs.bin", FS_OPEN_READ, &data_file))) {
|
|
|
|
storage = std::make_shared<IStorageInterface>(new LayeredRomFS(std::make_shared<RomInterfaceStorage>(data_storage), std::make_shared<RomFileStorage>(data_file), data_id));
|
|
|
|
} else {
|
|
|
|
storage = std::make_shared<IStorageInterface>(new LayeredRomFS(std::make_shared<RomInterfaceStorage>(data_storage), nullptr, data_id));
|
|
|
|
}
|
|
|
|
if (out_storage.IsDomain()) {
|
|
|
|
out_domain_id = data_storage.s.object_id;
|
|
|
|
}
|
2018-06-15 00:50:01 +01:00
|
|
|
} else {
|
2018-11-04 20:45:29 +00:00
|
|
|
/* If we don't have anything to modify, there's no sense in maintaining a copy of the metadata tables. */
|
|
|
|
fsStorageClose(&data_storage);
|
|
|
|
rc = RESULT_FORWARD_TO_SESSION;
|
2018-06-12 23:00:09 +01:00
|
|
|
}
|
2018-06-10 18:11:05 +01:00
|
|
|
}
|
2018-06-12 23:00:09 +01:00
|
|
|
|
2018-10-30 00:30:39 +00:00
|
|
|
return rc;
|
2018-06-10 02:33:22 +01:00
|
|
|
}
|