1
0
Fork 0
mirror of https://github.com/Atmosphere-NX/Atmosphere.git synced 2024-11-05 19:51:45 +00:00

fs.mitm: Add hook for postprocessing mitm messages.

This commit is contained in:
Michael Scire 2018-06-10 01:17:00 -06:00
parent 40a6b7bb54
commit ff9412feef
4 changed files with 39 additions and 3 deletions

View file

@ -3,6 +3,17 @@
Result FsMitMService::dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) { Result FsMitMService::dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) {
Result rc = 0xF601; Result rc = 0xF601;
while (cmd_id == 18) {
}
return rc;
}
Result FsMitMService::postprocess(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) {
Result rc = 0xF601;
while (cmd_id == 18) {
}
return rc; return rc;
} }

View file

@ -1,9 +1,11 @@
#pragma once #pragma once
#include <switch.h> #include <switch.h>
#include <stratosphere/iserviceobject.hpp> #include <stratosphere/iserviceobject.hpp>
#include "imitmserviceobject.hpp"
class FsMitMService : IServiceObject { class FsMitMService : public IMitMServiceObject {
public: public:
virtual Result dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size); virtual Result dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size);
virtual Result postprocess(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size);
virtual Result handle_deferred(); virtual Result handle_deferred();
}; };

View file

@ -0,0 +1,12 @@
#pragma once
#include <switch.h>
#include <stratosphere.hpp>
class IMitMServiceObject : public IServiceObject {
protected:
virtual ~IMitMServiceObject() { }
virtual Result dispatch(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) = 0;
virtual Result postprocess(IpcParsedCommand &r, IpcCommand &out_c, u64 cmd_id, u8 *pointer_buffer, size_t pointer_buffer_size) = 0;
virtual Result handle_deferred() = 0;
};

View file

@ -1,6 +1,7 @@
#pragma once #pragma once
#include <switch.h> #include <switch.h>
#include <stratosphere.hpp> #include <stratosphere.hpp>
#include "imitmserviceobject.hpp"
#include "mitm_server.hpp" #include "mitm_server.hpp"
@ -12,7 +13,7 @@ class MitMServer;
template <typename T> template <typename T>
class MitMSession final : public IWaitable { class MitMSession final : public IWaitable {
static_assert(std::is_base_of<IServiceObject, T>::value, "Service Objects must derive from IServiceObject"); static_assert(std::is_base_of<IMitMServiceObject, T>::value, "MitM Service Objects must derive from IMitMServiceObject");
T *service_object; T *service_object;
MitMServer<T> *server; MitMServer<T> *server;
@ -102,13 +103,19 @@ class MitMSession final : public IWaitable {
retval = ipcParse(&r); retval = ipcParse(&r);
u64 cmd_id = U64_MAX;
/* TODO: Close input copy handles that we don't need. */ /* TODO: Close input copy handles that we don't need. */
if (R_SUCCEEDED(retval)) { if (R_SUCCEEDED(retval)) {
rawdata_start = (u32 *)r.Raw; rawdata_start = (u32 *)r.Raw;
cmd_id = rawdata_start[2];
retval = 0xF601; retval = 0xF601;
if (r.CommandType == IpcCommandType_Request || r.CommandType == IpcCommandType_RequestWithContext) { if (r.CommandType == IpcCommandType_Request || r.CommandType == IpcCommandType_RequestWithContext) {
retval = this->service_object->dispatch(r, c, rawdata_start[2], (u8 *)this->pointer_buffer, this->pointer_buffer_size); retval = this->service_object->dispatch(r, c, cmd_id, (u8 *)this->pointer_buffer, this->pointer_buffer_size);
if (R_SUCCEEDED(retval)) {
ipcParse(&out_r);
}
} }
/* 0xF601 --> Dispatch onwards. */ /* 0xF601 --> Dispatch onwards. */
@ -137,6 +144,10 @@ class MitMSession final : public IWaitable {
/* Session close. */ /* Session close. */
rc = retval; rc = retval;
} else { } else {
if (R_SUCCEEDED(rc)) {
ipcInitialize(&c);
retval = this->service_object->postprocess(r, c, cmd_id, (u8 *)this->pointer_buffer, this->pointer_buffer_size);
}
Log(armGetTls(), 0x100); Log(armGetTls(), 0x100);
rc = svcReplyAndReceive(&handle_index, &this->server_handle, 0, this->server_handle, 0); rc = svcReplyAndReceive(&handle_index, &this->server_handle, 0, this->server_handle, 0);
/* Clean up copy handles. */ /* Clean up copy handles. */