kern: SvcGetLastThreadInfo, SvcGetDebugFutureThreadInfo

libraries/libmesosphere/source/arch/arm64/kern_k_thread_context.cpp

@@ -40,15 +40,6 @@ namespace ams::kern::arch::arm64 {
 
     namespace {
 
-        /* TODO: Should these be elsewhere? (In a header)? */
-        ALWAYS_INLINE KExceptionContext *GetExceptionContext(KThread *thread) {
-            return reinterpret_cast<KExceptionContext *>(reinterpret_cast<uintptr_t>(thread->GetKernelStackTop()) - sizeof(KThread::StackParameters) - sizeof(KExceptionContext));
-        }
-
-        ALWAYS_INLINE const KExceptionContext *GetExceptionContext(const KThread *thread) {
-            return reinterpret_cast<const KExceptionContext *>(reinterpret_cast<uintptr_t>(thread->GetKernelStackTop()) - sizeof(KThread::StackParameters) - sizeof(KExceptionContext));
-        }
-
         ALWAYS_INLINE bool IsFpuEnabled() {
             return cpu::ArchitecturalFeatureAccessControlRegisterAccessor().IsFpEnabled();
         }

libraries/libmesosphere/source/kern_k_debug_base.cpp

@@ -234,6 +234,36 @@ namespace ams::kern {
         return ResultSuccess();
     }
 
+    Result KDebugBase::GetRunningThreadInfo(ams::svc::LastThreadContext *out_context, u64 *out_thread_id) {
+        /* Get the attached process. */
+        KScopedAutoObject process = this->GetProcess();
+        R_UNLESS(process.IsNotNull(), svc::ResultProcessTerminated());
+
+        /* Get the thread info. */
+        {
+            KScopedSchedulerLock sl;
+
+            /* Get the running thread. */
+            const s32 core_id = GetCurrentCoreId();
+            KThread *thread = process->GetRunningThread(core_id);
+
+            /* Check that the thread's idle count is correct. */
+            R_UNLESS(process->GetRunningThreadIdleCount(core_id) == Kernel::GetScheduler(core_id).GetIdleCount(), svc::ResultNoThread());
+
+            /* Check that the thread is running on the current core. */
+            R_UNLESS(thread != nullptr,                  svc::ResultUnknownThread());
+            R_UNLESS(thread->GetActiveCore() == core_id, svc::ResultUnknownThread());
+
+            /* Get the thread's exception context. */
+            GetExceptionContext(thread)->GetSvcThreadContext(out_context);
+
+            /* Get the thread's id. */
+            *out_thread_id = thread->GetId();
+        }
+
+        return ResultSuccess();
+    }
+
     Result KDebugBase::Attach(KProcess *target) {
         /* Check that the process isn't null. */
         MESOSPHERE_ASSERT(target != nullptr);

libraries/libmesosphere/source/kern_k_wait_object.cpp

@@ -18,7 +18,86 @@
 namespace ams::kern {
 
     void KWaitObject::OnTimer() {
-        MESOSPHERE_UNIMPLEMENTED();
+        MESOSPHERE_ASSERT(KScheduler::IsSchedulerLockedByCurrentThread());
+
+        /* Wake up all the waiting threads. */
+        Entry *entry = std::addressof(this->root);
+        while (true) {
+            /* Get the next thread. */
+            KThread *thread = entry->GetNext();
+            if (thread == nullptr) {
+                break;
+            }
+
+            /* Wake it up. */
+            thread->Wakeup();
+
+            /* Advance. */
+            entry = std::addressof(thread->GetSleepingQueueEntry());
+        }
+    }
+
+    Result KWaitObject::Synchronize(s64 timeout) {
+        /* Perform the wait. */
+        KHardwareTimer *timer = nullptr;
+        KThread *cur_thread   = GetCurrentThreadPointer();
+        {
+            KScopedSchedulerLock sl;
+
+            /* Check that the thread isn't terminating. */
+            R_UNLESS(!cur_thread->IsTerminationRequested(), svc::ResultTerminationRequested());
+
+            /* Verify that nothing else is already waiting on the object. */
+            if (timeout > 0) {
+                R_UNLESS(!this->timer_used, svc::ResultBusy());
+            }
+
+            /* Check that we're not already in use. */
+            if (timeout >= 0) {
+                /* Verify the timer isn't already in use. */
+                R_UNLESS(!this->timer_used, svc::ResultBusy());
+            }
+
+            /* If we need to, register our timeout. */
+            if (timeout > 0) {
+                /* Mark that we're using the timer. */
+                this->timer_used = true;
+
+                /* Use the timer. */
+                timer = std::addressof(Kernel::GetHardwareTimer());
+                timer->RegisterAbsoluteTask(this, timeout);
+            }
+
+            if (timeout == 0) {
+                /* If we're timed out immediately, just wake up the thread. */
+                this->OnTimer();
+            } else {
+                /* Otherwise, sleep until the timeout occurs. */
+                this->Enqueue(cur_thread);
+                cur_thread->SetState(KThread::ThreadState_Waiting);
+                cur_thread->SetSyncedObject(nullptr, svc::ResultTimedOut());
+            }
+        }
+
+        /* Cleanup as necessary. */
+        {
+            KScopedSchedulerLock sl;
+
+            /* Remove from the timer. */
+            if (timeout > 0) {
+                MESOSPHERE_ASSERT(this->timer_used);
+                MESOSPHERE_ASSERT(timer != nullptr);
+                timer->CancelTask(this);
+                this->timer_used = false;
+            }
+
+            /* Remove the thread from our queue. */
+            if (timeout != 0) {
+                this->Remove(cur_thread);
+            }
+        }
+
+        return ResultSuccess();
     }
 
 }

libraries/libmesosphere/source/svc/kern_svc_thread_profiler.cpp

@@ -0,0 +1,135 @@
+/*
+ * 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/>.
+ */
+#include <mesosphere.hpp>
+
+namespace ams::kern::svc {
+
+    /* =============================    Common    ============================= */
+
+    namespace {
+
+        Result GetDebugFutureThreadInfo(ams::svc::LastThreadContext *out_context, uint64_t *out_thread_id, ams::svc::Handle debug_handle, int64_t ns) {
+            /* Only allow invoking the svc on development hardware. */
+            R_UNLESS(KTargetSystem::IsDebugMode(), svc::ResultNoThread());
+
+            /* Get the debug object. */
+            KScopedAutoObject debug = GetCurrentProcess().GetHandleTable().GetObject<KDebug>(debug_handle);
+            R_UNLESS(debug.IsNotNull(), svc::ResultInvalidHandle());
+
+            /* Synchronize the current process to the desired time. */
+            {
+                /* Get the wait object. */
+                KWaitObject *wait_object = GetCurrentProcess().GetWaitObjectPointer();
+
+                /* Convert the timeout from nanoseconds to ticks. */
+                s64 timeout;
+                if (ns > 0) {
+                    u64 ticks = KHardwareTimer::GetTick();
+                    ticks += ams::svc::Tick(TimeSpan::FromNanoSeconds(ns));
+                    ticks += 2;
+
+                    timeout = ticks;
+                } else {
+                    timeout = ns;
+                }
+
+                /* Synchronize to the desired time. */
+                R_TRY(wait_object->Synchronize(timeout));
+            }
+
+            /* Get the running thread info. */
+            R_TRY(debug->GetRunningThreadInfo(out_context, out_thread_id));
+
+            return ResultSuccess();
+        }
+
+        Result GetLastThreadInfo(ams::svc::LastThreadContext *out_context, uintptr_t *out_tls_address, uint32_t *out_flags) {
+            /* Only allow invoking the svc on development hardware. */
+            R_UNLESS(KTargetSystem::IsDebugMode(), svc::ResultNoThread());
+
+            /* Get the thread info. */
+            {
+                KScopedInterruptDisable di;
+
+                /* Get the previous thread. */
+                KThread *prev_thread = Kernel::GetScheduler().GetPreviousThread();
+                R_UNLESS(prev_thread != nullptr, svc::ResultNoThread());
+
+                /* Verify the last thread was owned by the current process. */
+                R_UNLESS(prev_thread->GetOwnerProcess() == GetCurrentProcessPointer(), svc::ResultUnknownThread());
+
+                /* Clear the output flags. */
+                *out_flags = 0;
+
+                /* Get the thread's exception context. */
+                GetExceptionContext(prev_thread)->GetSvcThreadContext(out_context);
+
+                /* Get the tls address. */
+                *out_tls_address = GetInteger(prev_thread->GetThreadLocalRegionAddress());
+
+                /* Set the syscall flag if appropriate. */
+                if (prev_thread->IsCallingSvc()) {
+                    *out_flags |= ams::svc::LastThreadInfoFlag_ThreadInSystemCall;
+                }
+            }
+
+            return ResultSuccess();
+        }
+
+    }
+
+    /* =============================    64 ABI    ============================= */
+
+    Result GetDebugFutureThreadInfo64(ams::svc::lp64::LastThreadContext *out_context, uint64_t *out_thread_id, ams::svc::Handle debug_handle, int64_t ns) {
+        return GetDebugFutureThreadInfo(out_context, out_thread_id, debug_handle, ns);
+    }
+
+    Result GetLastThreadInfo64(ams::svc::lp64::LastThreadContext *out_context, ams::svc::Address *out_tls_address, uint32_t *out_flags) {
+        static_assert(sizeof(*out_tls_address) == sizeof(uintptr_t));
+        return GetLastThreadInfo(out_context, reinterpret_cast<uintptr_t *>(out_tls_address), out_flags);
+    }
+
+    /* ============================= 64From32 ABI ============================= */
+
+    Result GetDebugFutureThreadInfo64From32(ams::svc::ilp32::LastThreadContext *out_context, uint64_t *out_thread_id, ams::svc::Handle debug_handle, int64_t ns) {
+        ams::svc::LastThreadContext context = {};
+        R_TRY(GetDebugFutureThreadInfo(std::addressof(context), out_thread_id, debug_handle, ns));
+
+        *out_context = {
+            .fp = static_cast<u32>(context.fp),
+            .sp = static_cast<u32>(context.sp),
+            .lr = static_cast<u32>(context.lr),
+            .pc = static_cast<u32>(context.pc),
+        };
+        return ResultSuccess();
+    }
+
+    Result GetLastThreadInfo64From32(ams::svc::ilp32::LastThreadContext *out_context, ams::svc::Address *out_tls_address, uint32_t *out_flags) {
+        static_assert(sizeof(*out_tls_address) == sizeof(uintptr_t));
+
+        ams::svc::LastThreadContext context = {};
+        R_TRY(GetLastThreadInfo(std::addressof(context), reinterpret_cast<uintptr_t *>(out_tls_address), out_flags));
+
+        *out_context = {
+            .fp = static_cast<u32>(context.fp),
+            .sp = static_cast<u32>(context.sp),
+            .lr = static_cast<u32>(context.lr),
+            .pc = static_cast<u32>(context.pc),
+        };
+        return ResultSuccess();
+    }
+
+}