2018-06-10 02:33:22 +01:00
|
|
|
#pragma once
|
|
|
|
#include <switch.h>
|
|
|
|
#include <stratosphere/iserviceobject.hpp>
|
2018-06-10 08:17:00 +01:00
|
|
|
#include "imitmserviceobject.hpp"
|
2018-06-12 23:00:09 +01:00
|
|
|
#include "fs_istorage.hpp"
|
2018-06-10 02:33:22 +01:00
|
|
|
|
2018-06-19 19:07:31 +01:00
|
|
|
enum class FspSrvCmd {
|
|
|
|
SetCurrentProcess = 1,
|
|
|
|
OpenDataStorageByCurrentProcess = 200,
|
|
|
|
OpenDataStorageByDataId = 202,
|
2018-06-10 10:06:50 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
class FsMitMService : public IMitMServiceObject {
|
|
|
|
private:
|
|
|
|
bool has_initialized;
|
2018-06-15 00:50:01 +01:00
|
|
|
u64 init_pid;
|
2018-06-28 06:02:06 +01:00
|
|
|
std::shared_ptr<IStorageInterface> romfs_storage;
|
2018-06-10 02:33:22 +01:00
|
|
|
public:
|
2018-06-15 00:50:01 +01:00
|
|
|
FsMitMService(Service *s) : IMitMServiceObject(s), has_initialized(false), init_pid(0) {
|
2018-06-10 10:12:34 +01:00
|
|
|
/* ... */
|
|
|
|
}
|
2018-06-12 23:00:09 +01:00
|
|
|
|
2018-06-15 00:50:01 +01:00
|
|
|
static bool should_mitm(u64 pid, u64 tid) {
|
|
|
|
return tid >= 0x0100000000010000ULL;
|
|
|
|
}
|
|
|
|
|
2018-06-12 23:00:09 +01:00
|
|
|
FsMitMService *clone() override {
|
|
|
|
auto new_srv = new FsMitMService((Service *)&this->forward_service);
|
|
|
|
this->clone_to(new_srv);
|
|
|
|
return new_srv;
|
|
|
|
}
|
|
|
|
|
|
|
|
void clone_to(void *o) override {
|
|
|
|
FsMitMService *other = (FsMitMService *)o;
|
|
|
|
other->has_initialized = has_initialized;
|
2018-06-15 00:50:01 +01:00
|
|
|
other->init_pid = init_pid;
|
2018-06-12 23:00:09 +01:00
|
|
|
}
|
|
|
|
|
2018-06-10 02:33:22 +01:00
|
|
|
virtual Result dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size);
|
2018-06-12 23:00:09 +01:00
|
|
|
virtual void postprocess(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size);
|
2018-06-10 02:33:22 +01:00
|
|
|
virtual Result handle_deferred();
|
2018-06-10 18:11:05 +01:00
|
|
|
|
|
|
|
protected:
|
|
|
|
/* Overridden commands. */
|
2018-06-15 00:50:01 +01:00
|
|
|
std::tuple<Result, OutSession<IStorageInterface>> open_data_storage_by_current_process();
|
2018-06-12 23:00:09 +01:00
|
|
|
std::tuple<Result, OutSession<IStorageInterface>> open_data_storage_by_data_id(u64 storage_id, u64 data_id);
|
2018-06-19 19:07:31 +01:00
|
|
|
};
|