strat: fix some mitm server management logic

2021-10-14 00:40:43 -07:00
parent 891fa32bf1
commit 7e536f74ae
4 changed files with 19 additions and 4 deletions
--- a/libraries/libstratosphere/source/sf/hipc/sf_hipc_server_manager.cpp
+++ b/libraries/libstratosphere/source/sf/hipc/sf_hipc_server_manager.cpp
@@ -135,6 +135,13 @@ namespace ams::sf::hipc {
            if (!session->m_has_received) {
                R_TRY(this->ReceiveRequest(session, tls_message));
                session->m_has_received = true;
+
+                if (this->CanManageMitmServers()) {
+                    const cmif::PointerAndSize &saved_message = session->m_saved_message;
+                    AMS_ABORT_UNLESS(tls_message.GetSize() == saved_message.GetSize());
+
+                    std::memcpy(saved_message.GetPointer(), tls_message.GetPointer(), tls_message.GetSize());
+                }
            }

            R_TRY_CATCH(this->ProcessRequest(session, tls_message)) {