os: implement waitable management.

This implements waitable management for Events (and implements Events). It also refactors PM to use new Event/Waitable semantics, and also adds STS_ASSERT as a macro for asserting a boolean expression. The rest of stratosphere has been refactored to use STS_ASSERT whenever possible.
2019-09-27 18:04:58 -07:00
parent e07011be32
commit 609a302e16
108 changed files with 2752 additions and 1223 deletions
--- a/stratosphere/libstratosphere/source/os/os_waitable_holder.cpp
+++ b/stratosphere/libstratosphere/source/os/os_waitable_holder.cpp
@@ -0,0 +1,111 @@
+/*
+ * 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>
+#include "impl/os_waitable_holder_impl.hpp"
+#include "impl/os_waitable_manager_impl.hpp"
+
+namespace sts::os {
+
+    WaitableHolder::WaitableHolder(Handle handle) {
+        /* Don't allow invalid handles. */
+        STS_ASSERT(handle != INVALID_HANDLE);
+
+        /* Initialize appropriate holder. */
+        new (GetPointer(this->impl_storage)) impl::WaitableHolderOfHandle(handle);
+
+        /* Set user-data. */
+        this->user_data = 0;
+    }
+
+    WaitableHolder::WaitableHolder(Event *event) {
+        /* Initialize appropriate holder. */
+        new (GetPointer(this->impl_storage)) impl::WaitableHolderOfEvent(event);
+
+        /* Set user-data. */
+        this->user_data = 0;
+    }
+
+    WaitableHolder::WaitableHolder(SystemEvent *event) {
+        /* Initialize appropriate holder. */
+        switch (event->GetState()) {
+            case SystemEventState::Event:
+                new (GetPointer(this->impl_storage)) impl::WaitableHolderOfEvent(&event->GetEvent());
+                break;
+            case SystemEventState::InterProcessEvent:
+                new (GetPointer(this->impl_storage)) impl::WaitableHolderOfInterProcessEvent(&event->GetInterProcessEvent());
+                break;
+            case SystemEventState::Uninitialized:
+            default:
+                std::abort();
+        }
+
+        /* Set user-data. */
+        this->user_data = 0;
+    }
+
+    WaitableHolder::WaitableHolder(InterruptEvent *event) {
+        /* Initialize appropriate holder. */
+        new (GetPointer(this->impl_storage)) impl::WaitableHolderOfInterruptEvent(event);
+
+        /* Set user-data. */
+        this->user_data = 0;
+    }
+
+    WaitableHolder::WaitableHolder(Thread *thread) {
+        /* Initialize appropriate holder. */
+        new (GetPointer(this->impl_storage)) impl::WaitableHolderOfThread(thread);
+
+        /* Set user-data. */
+        this->user_data = 0;
+    }
+
+    WaitableHolder::WaitableHolder(MessageQueue *message_queue, MessageQueueWaitKind wait_kind) {
+        /* Initialize appropriate holder. */
+        switch (wait_kind) {
+            case MessageQueueWaitKind::ForNotFull:
+                new (GetPointer(this->impl_storage)) impl::WaitableHolderOfMessageQueueForNotFull(message_queue);
+                break;
+            case MessageQueueWaitKind::ForNotEmpty:
+                new (GetPointer(this->impl_storage)) impl::WaitableHolderOfMessageQueueForNotEmpty(message_queue);
+                break;
+            default:
+                std::abort();
+        }
+
+        /* Set user-data. */
+        this->user_data = 0;
+    }
+
+    WaitableHolder::~WaitableHolder() {
+        auto holder_base = reinterpret_cast<impl::WaitableHolderBase *>(GetPointer(this->impl_storage));
+
+        /* Don't allow destruction of a linked waitable holder. */
+        STS_ASSERT(!holder_base->IsLinkedToManager());
+
+        holder_base->~WaitableHolderBase();
+    }
+
+    void WaitableHolder::UnlinkFromWaitableManager() {
+        auto holder_base = reinterpret_cast<impl::WaitableHolderBase *>(GetPointer(this->impl_storage));
+
+        /* Don't allow unlinking of an unlinked holder. */
+        STS_ASSERT(holder_base->IsLinkedToManager());
+
+        holder_base->GetManager()->UnlinkWaitableHolder(*holder_base);
+        holder_base->SetManager(nullptr);
+    }
+
+}