tipc: first draft object allocation logic

libraries/libstratosphere/include/stratosphere/sf/sf_object_factory.hpp

@@ -43,7 +43,7 @@ namespace ams::sf {
             using Type = Impl;
         };
 
-        template<typename Impl> requires (std::is_abstract<Impl>::value )
+        template<typename Impl> requires (std::is_abstract<Impl>::value)
         struct UnmanagedEmplaceImplHolderBaseGetter<Impl> {
             class Impl2 : public Impl {
                 public:

libraries/libstratosphere/include/stratosphere/tipc.hpp

@@ -15,4 +15,7 @@
  */
 #pragma once
 
+#include <stratosphere/tipc/tipc_service_object.hpp>
+#include <stratosphere/tipc/tipc_allocators.hpp>
+
 #include <stratosphere/tipc/impl/tipc_impl_command_serialization.hpp>

libraries/libstratosphere/include/stratosphere/tipc/impl/tipc_impl_template_base.hpp

@@ -0,0 +1,32 @@
+/*
+ * Copyright (c) 2018-2020 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/>.
+ */
+#pragma once
+#include <vapours.hpp>
+
+namespace ams::tipc::impl {
+
+    template<typename Interface, typename Base, typename ImplHolder, typename ImplGetter, typename Root>
+    class ImplTemplateBaseT;
+
+    template<typename Interface, typename Base, typename ImplHolder, typename ImplGetter>
+    class ImplTemplateBase : public ImplTemplateBaseT<Interface, Base, ImplHolder, ImplGetter, Interface> {
+        private:
+            using BaseImpl = ImplTemplateBaseT<Interface, Base, ImplHolder, ImplGetter, Interface>;
+        public:
+            using BaseImpl::BaseImpl;
+    };
+
+}

libraries/libstratosphere/include/stratosphere/tipc/tipc_allocators.hpp

@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2018-2020 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/>.
+ */
+#pragma once
+#include <vapours.hpp>
+#include <stratosphere/tipc/tipc_common.hpp>
+#include <stratosphere/tipc/tipc_service_object_base.hpp>
+
+namespace ams::tipc {
+
+    template<typename T>
+    concept IsServiceObjectAllocator = requires (T &t) {
+        { t.Allocate() } -> std::convertible_to<ServiceObjectBase *>;
+    };
+
+    template<typename T> requires IsServiceObject<T>
+    class SingletonAllocator final {
+        private:
+            T m_singleton;
+        public:
+            constexpr ALWAYS_INLINE SingletonAllocator() : m_singleton() { /* ... */ }
+
+            ALWAYS_INLINE ServiceObjectBase *Allocate() { return std::addressof(m_singleton); }
+    };
+
+    template<typename T, size_t N> requires IsServiceObject<T>
+    class SlabAllocator final : public ServiceObjectDeleter {
+        private:
+            struct Entry {
+                bool used;
+                util::TypedStorage<T> storage;
+            };
+        private:
+            os::SdkMutex m_mutex;
+            Entry m_entries[N];
+        public:
+            constexpr ALWAYS_INLINE SlabAllocator() : m_entries() { /* ... */ }
+
+            ServiceObjectBase *Allocate() {
+                std::scoped_lock lk(m_mutex);
+
+                for (size_t i = 0; i < N; ++i) {
+                    if (!m_entries[i].used) {
+                        m_entries[i].used = true;
+                        return util::ConstructAt(m_entries[i].storage);
+                    }
+                }
+
+                AMS_ABORT("Failed to allocate entry in SlabAllocator<T, N>");
+            }
+
+            void Deallocate(ServiceObjectBase *object) {
+                std::scoped_lock lk(m_mutex);
+
+                for (size_t i = 0; i < N; ++i) {
+                    if (m_entries[i].used && GetPointer(m_entries[i].storage) == object) {
+                        util::DestroyAt(m_entries[i].storage);
+                        m_entries[i].used = false;
+                        return;
+                    }
+                }
+
+                AMS_ABORT("Failed to deallocate entry in SlabAllocator<T, N>");
+            }
+        public:
+            virtual void DeleteServiceObject(ServiceObjectBase *object) override {
+                return this->Deallocate(object);
+            }
+    };
+    static_assert(IsServiceObjectAllocator<SlabAllocator<ServiceObjectBase, 1>>);
+    static_assert(IsServiceObjectDeleter<SlabAllocator<ServiceObjectBase, 1>>);
+
+}
+

libraries/libstratosphere/include/stratosphere/tipc/tipc_service_object.hpp

@@ -0,0 +1,61 @@
+/*
+ * Copyright (c) 2018-2020 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/>.
+ */
+#pragma once
+#include <vapours.hpp>
+#include <stratosphere/tipc/tipc_service_object_base.hpp>
+#include <stratosphere/tipc/impl/tipc_impl_template_base.hpp>
+
+namespace ams::tipc {
+
+    namespace impl {
+
+        template<typename Impl>
+        class EmplacedImplHolderBaseGetter {
+            using Type = Impl;
+        };
+
+        template<typename Impl>
+        class EmplacedImplHolder {
+            template<typename, typename, typename, typename>
+            friend class impl::ImplTemplateBaseT;
+            private:
+                using Impl2 = typename EmplacedImplHolderBaseGetter<Impl>::Type;
+                static_assert(!std::is_abstract<Impl2>::value);
+            private:
+                Impl2 m_impl;
+            private:
+                template<typename... Args>
+                constexpr explicit EmplacedImplHolder(Args &&... args) : m_impl(std::forward<Args>(args)...) { /* ... */ }
+            public:
+                static constexpr Impl *GetImplPointer(EmplacedImplHolder *holder) {
+                    return std::addressof(holder->m_impl);
+                }
+        };
+
+    }
+
+    template<typename Interface, typename Impl>
+    class ServiceObject final : public impl::ImplTemplateBase<Interface, Interface, impl::EmplacedImplHolder<Impl>, impl::EmplacedImplHolder<Impl>> {
+        private:
+            using ImplBase = impl::ImplTemplateBase<Interface, Interface, impl::EmplacedImplHolder<Impl>, impl::EmplacedImplHolder<Impl>>;
+        public:
+            using ImplBase::ImplBase;
+
+            constexpr Impl &GetImpl() { return *impl::EmplacedImplHolder<Impl>::GetImplPointer(this); }
+    };
+
+}
+

libraries/libstratosphere/include/stratosphere/tipc/tipc_service_object_base.hpp

@@ -0,0 +1,53 @@
+/*
+ * Copyright (c) 2018-2020 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/>.
+ */
+#pragma once
+#include <vapours.hpp>
+
+namespace ams::tipc {
+
+    class ServiceObjectBase;
+
+    class ServiceObjectDeleter {
+        public:
+            virtual void DeleteServiceObject(ServiceObjectBase *object) = 0;
+    };
+
+    template<typename T>
+    concept IsServiceObjectDeleter = std::derived_from<T, ServiceObjectDeleter>;
+
+    class ServiceObjectBase {
+        private:
+            ServiceObjectDeleter *m_deleter;
+        public:
+            constexpr ALWAYS_INLINE ServiceObjectBase() : m_deleter(nullptr) { /* ... */ }
+
+            ALWAYS_INLINE void SetDeleter(ServiceObjectDeleter *deleter) {
+                m_deleter = deleter;
+            }
+
+            ALWAYS_INLINE ServiceObjectDeleter *GetDeleter() const {
+                return m_deleter;
+            }
+
+            virtual ~ServiceObjectBase() { /* ... */ }
+            virtual Result ProcessRequest() = 0;
+    };
+
+    template<typename T>
+    concept IsServiceObject = std::derived_from<T, ServiceObjectBase>;
+
+}
+