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

#pragma once
#include <assert.h>
#include "utils.h"
#include "barrier.h"
#include "execute_function.h"

struct ExceptionStackFrame;
typedef struct CoreCtx {
    struct ExceptionStackFrame *userFrame;      // @0x00
    u64 scratch;                                // @0x08
    u8 *crashStack;                             // @0x10
    u64 kernelArgument;                         // @0x18
    uintptr_t kernelEntrypoint;                 // @0x20
    u32 coreId;                                 // @0x28
    u8 gicInterfaceMask;                        // @0x2C. Equal to BIT(coreId) anyway
    bool isBootCore;                            // @0x2D
    bool warmboot;                              // @0x2E

    // Timer stuff
    u64 totalTimeInHypervisor;                  // @0x30. cntvoff_el2 is updated to that value.
    u64 emulPtimerCval;                         // @0x38. When setting cntp_cval_el0 and on interrupt

    // "Execute function"
    ExecutedFunction executedFunction;          // @0x40
    void *executedFunctionArgs;                 // @0x48
    Barrier executedFunctionBarrier;            // @0x50
    bool executedFunctionSync;                  // @0x54
} CoreCtx;

static_assert(offsetof(CoreCtx, warmboot) == 0x2E, "Wrong definition for CoreCtx");
static_assert(offsetof(CoreCtx, emulPtimerCval) == 0x38, "Wrong definition for CoreCtx");
static_assert(offsetof(CoreCtx, executedFunctionSync) == 0x54, "Wrong definition for CoreCtx");

extern CoreCtx g_coreCtxs[4];
register CoreCtx *currentCoreCtx asm("x18");

void coreCtxInit(u32 coreId, bool isBootCore, u64 argument);

void setCurrentCoreActive(void);
u32 getActiveCoreMask(void);