/*
 * Copyright (c) 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>

extern "C" void _start();

namespace ams::result::impl {

    NORETURN void OnResultAssertion(Result result) {
        MESOSPHERE_PANIC("OnResultAssertion(2%03d-%04d)", result.GetModule(), result.GetDescription());
    }

}

namespace ams::kern {

    /* NOTE: This is not exposed via a header, but is referenced via assembly. */
    /* NOTE: Nintendo does not save register contents on panic; we use this */
    /*       to generate an atmosphere fatal report on panic. */
    constinit KExceptionContext g_panic_exception_contexts[cpu::NumCores];

    namespace {

        constexpr std::array<s32, cpu::NumCores> NegativeArray = [] {
            std::array<s32, cpu::NumCores> arr = {};
            for (size_t i = 0; i < arr.size(); i++) {
                arr[i] = -1;
            }
            return arr;
        }();

        constinit util::Atomic<s32> g_next_ticket    = 0;
        constinit util::Atomic<s32> g_current_ticket = 0;

        constinit std::array<s32, cpu::NumCores> g_core_tickets = NegativeArray;

        s32 GetCoreTicket() {
            const s32 core_id = GetCurrentCoreId();
            if (g_core_tickets[core_id] == -1) {
                g_core_tickets[core_id] = 2 * (g_next_ticket++);
            }
            return g_core_tickets[core_id];
        }

        void WaitCoreTicket() {
            const s32 expected = GetCoreTicket();
            const s32 desired  = expected + 1;
            s32 compare = g_current_ticket.Load<std::memory_order_relaxed>();
            do {
                if (compare == desired) {
                    break;
                }
                compare = expected;
            } while (!g_current_ticket.CompareExchangeWeak(compare, desired));
        }

        void ReleaseCoreTicket() {
            const s32 expected = GetCoreTicket() + 1;
            const s32 desired  = expected + 1;

            s32 compare = expected;
            g_current_ticket.CompareExchangeStrong(compare, desired);
        }

        ALWAYS_INLINE KExceptionContext *GetPanicExceptionContext(int core_id) {
            #if defined(MESOSPHERE_ENABLE_PANIC_REGISTER_DUMP)
                return std::addressof(g_panic_exception_contexts[core_id]);
            #else
                return nullptr;
            #endif
        }

        [[gnu::unused]] void PrintCurrentState() {
            /* Wait for it to be our turn to print. */
            WaitCoreTicket();

            /* Get the current exception context. */
            const s32 core_id    = GetCurrentCoreId();
            const auto *core_ctx = GetPanicExceptionContext(core_id);

            /* Print the state. */
            MESOSPHERE_RELEASE_LOG("Core[%d] Current State:\n", core_id);

            /* Print kernel state. */
            #ifdef ATMOSPHERE_ARCH_ARM64
                MESOSPHERE_RELEASE_LOG("Kernel Registers:\n");
                for (size_t i = 0; i < 31; i++) {
                    MESOSPHERE_RELEASE_LOG("   X[%02zu] = %016lx\n", i, core_ctx->x[i]);
                }
                MESOSPHERE_RELEASE_LOG("   SP    = %016lx\n", core_ctx->sp);

                /* Print kernel backtrace. */
                MESOSPHERE_RELEASE_LOG("Kernel Backtrace:\n");
                uintptr_t fp = core_ctx != nullptr ? core_ctx->x[29] : reinterpret_cast<uintptr_t>(__builtin_frame_address(0));
                for (size_t i = 0; i < 32 && fp && util::IsAligned(fp, 0x10) && cpu::GetPhysicalAddressWritable(nullptr, fp, true); i++) {
                    struct {
                        uintptr_t fp;
                        uintptr_t lr;
                    } *stack_frame = reinterpret_cast<decltype(stack_frame)>(fp);
                    MESOSPHERE_RELEASE_LOG("   [%02zx]: %p\n", i, reinterpret_cast<void *>(stack_frame->lr));
                    fp = stack_frame->fp;
                }
            #endif

            /* Print registers and user backtrace. */
            KDebug::PrintRegister();
            KDebug::PrintBacktrace();

            MESOSPHERE_RELEASE_LOG("\n");

            /* Allow the next core to print. */
            ReleaseCoreTicket();
        }

        NORETURN void StopSystem() {
            #ifdef MESOSPHERE_BUILD_FOR_DEBUGGING
                /* Print the current core. */
                PrintCurrentState();
            #endif

            KSystemControl::StopSystem(GetPanicExceptionContext(GetCurrentCoreId()));
        }

    }

    NORETURN WEAK_SYMBOL void PanicImpl(const char *file, int line, const char *format, ...) {
        #ifdef MESOSPHERE_BUILD_FOR_DEBUGGING
            /* Wait for it to be our turn to print. */
            WaitCoreTicket();

            ::std::va_list vl;
            va_start(vl, format);
            MESOSPHERE_RELEASE_LOG("Core[%d]: Kernel Panic at %s:%d\n", GetCurrentCoreId(), file, line);
            if (KProcess *cur_process = GetCurrentProcessPointer(); cur_process != nullptr) {
                MESOSPHERE_RELEASE_LOG("Core[%d]: Current Process: %s\n", GetCurrentCoreId(), cur_process->GetName());
            }
            MESOSPHERE_RELEASE_VLOG(format, vl);
            MESOSPHERE_RELEASE_LOG("\n");
            va_end(vl);
        #else
            MESOSPHERE_UNUSED(file, line, format);
        #endif

        StopSystem();
    }

    NORETURN WEAK_SYMBOL void PanicImpl() {
        StopSystem();
    }

}