kern: Implement KAutoObject, KSlabHeap, KLightLock

libraries/libmesosphere/include/mesosphere/kern_k_auto_object_container.hpp

@@ -0,0 +1,65 @@
+/*
+ * 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 <mesosphere/kern_k_auto_object.hpp>
+#include <mesosphere/kern_k_light_lock.hpp>
+
+namespace ams::kern {
+
+    class KAutoObjectWithListContainer {
+        NON_COPYABLE(KAutoObjectWithListContainer);
+        NON_MOVEABLE(KAutoObjectWithListContainer);
+        private:
+            using ListType = util::IntrusiveRedBlackTreeMemberTraits<&KAutoObjectWithList::list_node>::TreeType<KAutoObjectWithList>;
+        public:
+            class ListAccessor : public KScopedLightLock {
+                private:
+                    ListType &list;
+                public:
+                    explicit ListAccessor(KAutoObjectWithListContainer *container) : KScopedLightLock(container->lock), list(container->object_list) { /* ... */ }
+                    explicit ListAccessor(KAutoObjectWithListContainer &container) : KScopedLightLock(container.lock), list(container.object_list) { /* ... */ }
+
+                    typename ListType::iterator begin() const {
+                        return this->list.begin();
+                    }
+
+                    typename ListType::iterator end() const {
+                        return this->list.end();
+                    }
+
+                    typename ListType::iterator find(typename ListType::const_reference ref) const {
+                        return this->list.find(ref);
+                    }
+            };
+
+            friend class ListAccessor;
+        private:
+            KLightLock lock;
+            ListType object_list;
+        public:
+            constexpr KAutoObjectWithListContainer() : lock(), object_list() { /* ... */ }
+
+            void Initialize() { /* Nothing to do. */ }
+            void Finalize() { /* Nothing to do. */ }
+
+            Result Register(KAutoObjectWithList *obj);
+            Result Unregister(KAutoObjectWithList *obj);
+            size_t GetOwnedCount(KProcess *owner);
+    };
+
+
+}