hipc: begin implementing domains. fixes ro + sm together

2019-10-14 09:20:21 -07:00
parent c8ed190e5c
commit 0c7827104f
10 changed files with 557 additions and 17 deletions
--- a/stratosphere/libstratosphere/source/sf/hipc/sf_hipc_server_domain_session_manager.cpp
+++ b/stratosphere/libstratosphere/source/sf/hipc/sf_hipc_server_domain_session_manager.cpp
@@ -0,0 +1,95 @@
+/*
+ * Copyright (c) 2018-2019 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/>.
+ */
+#include <stratosphere.hpp>
+
+namespace sts::sf::hipc {
+
+    namespace impl {
+
+        class HipcManager : public IServiceObject {
+            private:
+                enum class CommandId {
+                    ConvertCurrentObjectToDomain = 0,
+                    CopyFromCurrentDomain        = 1,
+                    CloneCurrentObject           = 2,
+                    QueryPointerBufferSize       = 3,
+                    CloneCurrentObjectEx         = 4,
+                };
+            private:
+                ServerDomainSessionManager *manager;
+                ServerSession *session;
+                bool is_mitm_session;
+            private:
+                Result CloneCurrentObjectImpl(Handle *out_client_handle, ServerSessionManager *tagged_manager) {
+                    /* Clone the object. */
+                    cmif::ServiceObjectHolder &&clone = this->session->srv_obj_holder.Clone();
+                    R_UNLESS(clone, ResultHipcDomainObjectNotFound);
+
+                    /* Create new session handles. */
+                    Handle server_handle;
+                    R_ASSERT(hipc::CreateSession(&server_handle, out_client_handle));
+
+                    /* Register with manager. */
+                    if (!is_mitm_session) {
+                        R_ASSERT(tagged_manager->RegisterSession(server_handle, std::move(clone)));
+                    } else {
+                        std::shared_ptr<::Service> forward_service = this->session->forward_service;
+                        R_ASSERT(tagged_manager->RegisterMitmSession(server_handle, std::move(clone), std::move(forward_service)));
+                    }
+
+                    return ResultSuccess;
+                }
+            public:
+                explicit HipcManager(ServerDomainSessionManager *m, ServerSession *s) : manager(m), session(s), is_mitm_session(s->forward_service != nullptr) {
+                    /* ... */
+                }
+
+                /* TODO: ConvertCurrentObjectToDomain */
+                /* TODO: CopyFromCurrentDomain */
+
+                Result CloneCurrentObject(sf::OutMoveHandle out) {
+                    return this->CloneCurrentObjectImpl(out.GetHandlePointer(), this->manager);
+                }
+
+                void QueryPointerBufferSize(sf::Out<u16> out) {
+                    out.SetValue(this->session->pointer_buffer.GetSize());
+                }
+
+                Result CloneCurrentObjectEx(sf::OutMoveHandle out, u32 tag) {
+                    return this->CloneCurrentObjectImpl(out.GetHandlePointer(), this->manager->GetSessionManagerByTag(tag));
+                }
+
+            public:
+                DEFINE_SERVICE_DISPATCH_TABLE {
+                    /* TODO: MAKE_SERVICE_COMMAND_META(ConvertCurrentObjectToDomain), */
+                    /* TODO: MAKE_SERVICE_COMMAND_META(CopyFromCurrentDomain), */
+                    MAKE_SERVICE_COMMAND_META(CloneCurrentObject),
+                    MAKE_SERVICE_COMMAND_META(QueryPointerBufferSize),
+                    MAKE_SERVICE_COMMAND_META(CloneCurrentObjectEx),
+                };
+        };
+
+    }
+
+    Result ServerDomainSessionManager::DispatchManagerRequest(ServerSession *session, const cmif::PointerAndSize &in_message, const cmif::PointerAndSize &out_message) {
+        /* Make a stack object, and pass a shared pointer to it to DispatchRequest. */
+        /* Note: This is safe, as no additional references to the hipc manager can ever be stored. */
+        /* The shared pointer to stack object is definitely gross, though. */
+        impl::HipcManager hipc_manager(this, session);
+        return this->DispatchRequest(cmif::ServiceObjectHolder(std::shared_ptr<impl::HipcManager>(&hipc_manager, [](impl::HipcManager *) { /* Empty deleter. */ })), session, in_message, out_message);
+    }
+
+}