// SPDX-License-Identifier: MIT
/*
 * Copyright 2025 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 */
#include "ras_sys.h"
#include "amdgpu_ras_mgr.h"
#include "amdgpu_ras.h"
#include "amdgpu_reset.h"

static int amdgpu_ras_sys_detect_fatal_event(struct ras_core_context *ras_core, void *data)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;
        int ret;
        uint64_t seq_no;

        ret = amdgpu_ras_global_ras_isr(adev);
        if (ret)
                return ret;

        seq_no = amdgpu_ras_mgr_gen_ras_event_seqno(adev, RAS_SEQNO_TYPE_UE);
        RAS_DEV_INFO(adev,
                "{%llu} Uncorrectable hardware error(ERREVENT_ATHUB_INTERRUPT) detected!\n",
                seq_no);

        return amdgpu_ras_process_handle_unexpected_interrupt(adev, data);
}

static int amdgpu_ras_sys_poison_consumption_event(struct ras_core_context *ras_core,
                                void *data)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;
        struct ras_event_req *req = (struct ras_event_req *)data;
        pasid_notify pasid_fn;

        if (!req)
                return -EINVAL;

        if (req->pasid_fn) {
                pasid_fn = (pasid_notify)req->pasid_fn;
                pasid_fn(adev, req->pasid, req->data);
        }

        return 0;
}

static int amdgpu_ras_sys_gen_seqno(struct ras_core_context *ras_core,
                        enum ras_seqno_type seqno_type, uint64_t *seqno)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;
        struct amdgpu_ras_mgr *ras_mgr = amdgpu_ras_mgr_get_context(adev);
        struct ras_event_manager *event_mgr;
        struct ras_event_state *event_state;
        struct amdgpu_hive_info *hive;
        enum ras_event_type event_type;
        uint64_t seq_no;

        if (!ras_mgr || !seqno ||
                (seqno_type >= RAS_SEQNO_TYPE_COUNT_MAX))
                return -EINVAL;

        switch (seqno_type) {
        case RAS_SEQNO_TYPE_UE:
                event_type = RAS_EVENT_TYPE_FATAL;
                break;
        case RAS_SEQNO_TYPE_CE:
        case RAS_SEQNO_TYPE_DE:
                event_type = RAS_EVENT_TYPE_POISON_CREATION;
                break;
        case RAS_SEQNO_TYPE_POISON_CONSUMPTION:
                event_type = RAS_EVENT_TYPE_POISON_CONSUMPTION;
                break;
        default:
                event_type = RAS_EVENT_TYPE_INVALID;
                break;
        }

        hive = amdgpu_get_xgmi_hive(adev);
        event_mgr = hive ? &hive->event_mgr : &ras_mgr->ras_event_mgr;
        event_state = &event_mgr->event_state[event_type];
        if ((event_type == RAS_EVENT_TYPE_FATAL) && amdgpu_ras_in_recovery(adev)) {
                seq_no = event_state->last_seqno;
        } else {
                seq_no = atomic64_inc_return(&event_mgr->seqno);
                event_state->last_seqno = seq_no;
                atomic64_inc(&event_state->count);
        }
        amdgpu_put_xgmi_hive(hive);

        *seqno = seq_no;
        return 0;

}

static int amdgpu_ras_sys_event_notifier(struct ras_core_context *ras_core,
                                   enum ras_notify_event event_id, void *data)
{
        struct amdgpu_ras_mgr *ras_mgr = amdgpu_ras_mgr_get_context(ras_core->dev);
        int ret = 0;

        switch (event_id) {
        case RAS_EVENT_ID__BAD_PAGE_DETECTED:
                schedule_delayed_work(&ras_mgr->retire_page_dwork, 0);
                break;
        case RAS_EVENT_ID__POISON_CONSUMPTION:
                amdgpu_ras_sys_poison_consumption_event(ras_core, data);
                break;
        case RAS_EVENT_ID__RESERVE_BAD_PAGE:
                ret = amdgpu_ras_reserve_page(ras_core->dev, *(uint64_t *)data);
                break;
        case RAS_EVENT_ID__FATAL_ERROR_DETECTED:
                ret = amdgpu_ras_sys_detect_fatal_event(ras_core, data);
                break;
        case RAS_EVENT_ID__UPDATE_BAD_PAGE_NUM:
                ret = amdgpu_dpm_send_hbm_bad_pages_num(ras_core->dev, *(uint32_t *)data);
                break;
        case RAS_EVENT_ID__UPDATE_BAD_CHANNEL_BITMAP:
                ret = amdgpu_dpm_send_hbm_bad_channel_flag(ras_core->dev, *(uint32_t *)data);
                break;
        case RAS_EVENT_ID__DEVICE_RMA:
                ras_log_ring_add_log_event(ras_core, RAS_LOG_EVENT_RMA, NULL, NULL);
                ret = amdgpu_dpm_send_rma_reason(ras_core->dev);
                break;
        case RAS_EVENT_ID__RESET_GPU:
                ret = amdgpu_ras_mgr_reset_gpu(ras_core->dev, *(uint32_t *)data);
                break;
        case RAS_EVENT_ID__RAS_EVENT_PROC_BEGIN:
                ret = amdgpu_ras_process_begin(ras_core->dev);
                break;
        case RAS_EVENT_ID__RAS_EVENT_PROC_END:
                ret = amdgpu_ras_process_end(ras_core->dev);
                break;
        default:
                RAS_DEV_WARN(ras_core->dev, "Invalid ras notify event:%d\n", event_id);
                break;
        }

        return ret;
}

static u64 amdgpu_ras_sys_get_utc_second_timestamp(struct ras_core_context *ras_core)
{
        return ktime_get_real_seconds();
}

static int amdgpu_ras_sys_check_gpu_status(struct ras_core_context *ras_core,
                                uint32_t *status)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;
        uint32_t gpu_status = 0;

        if (amdgpu_in_reset(adev) || amdgpu_ras_in_recovery(adev))
                gpu_status |= RAS_GPU_STATUS__IN_RESET;

        if (amdgpu_sriov_vf(adev))
                gpu_status |= RAS_GPU_STATUS__IS_VF;

        *status = gpu_status;

        return 0;
}

static int amdgpu_ras_sys_get_device_system_info(struct ras_core_context *ras_core,
                        struct device_system_info *dev_info)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;

        dev_info->device_id = adev->pdev->device;
        dev_info->vendor_id = adev->pdev->vendor;
        dev_info->socket_id = adev->smuio.funcs->get_socket_id(adev);

        return 0;
}

static int amdgpu_ras_sys_gpu_reset_lock(struct ras_core_context *ras_core,
                        bool down, bool try)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;
        int ret = 0;

        if (down && try)
                ret = down_read_trylock(&adev->reset_domain->sem);
        else if (down)
                down_read(&adev->reset_domain->sem);
        else
                up_read(&adev->reset_domain->sem);

        return ret;
}

static bool amdgpu_ras_sys_detect_ras_interrupt(struct ras_core_context *ras_core)
{
        return !!atomic_read(&amdgpu_ras_in_intr);
}

static int amdgpu_ras_sys_get_gpu_mem(struct ras_core_context *ras_core,
        enum gpu_mem_type mem_type, struct gpu_mem_block *gpu_mem)
{
        struct amdgpu_device *adev = (struct amdgpu_device *)ras_core->dev;
        struct psp_context *psp = &adev->psp;
        struct psp_ring *psp_ring;
        struct ta_mem_context *mem_ctx;

        if (mem_type == GPU_MEM_TYPE_RAS_PSP_RING) {
                psp_ring = &psp->km_ring;
                gpu_mem->mem_bo = adev->firmware.rbuf;
                gpu_mem->mem_size = psp_ring->ring_size;
                gpu_mem->mem_mc_addr = psp_ring->ring_mem_mc_addr;
                gpu_mem->mem_cpu_addr = psp_ring->ring_mem;
        } else if (mem_type == GPU_MEM_TYPE_RAS_PSP_CMD) {
                gpu_mem->mem_bo = psp->cmd_buf_bo;
                gpu_mem->mem_size = PSP_CMD_BUFFER_SIZE;
                gpu_mem->mem_mc_addr = psp->cmd_buf_mc_addr;
                gpu_mem->mem_cpu_addr = psp->cmd_buf_mem;
        } else if (mem_type == GPU_MEM_TYPE_RAS_PSP_FENCE) {
                gpu_mem->mem_bo = psp->fence_buf_bo;
                gpu_mem->mem_size = PSP_FENCE_BUFFER_SIZE;
                gpu_mem->mem_mc_addr = psp->fence_buf_mc_addr;
                gpu_mem->mem_cpu_addr = psp->fence_buf;
        } else if (mem_type == GPU_MEM_TYPE_RAS_TA_FW) {
                gpu_mem->mem_bo = psp->fw_pri_bo;
                gpu_mem->mem_size = PSP_1_MEG;
                gpu_mem->mem_mc_addr = psp->fw_pri_mc_addr;
                gpu_mem->mem_cpu_addr = psp->fw_pri_buf;
        } else if (mem_type == GPU_MEM_TYPE_RAS_TA_CMD) {
                mem_ctx = &psp->ras_context.context.mem_context;
                gpu_mem->mem_bo = mem_ctx->shared_bo;
                gpu_mem->mem_size = mem_ctx->shared_mem_size;
                gpu_mem->mem_mc_addr = mem_ctx->shared_mc_addr;
                gpu_mem->mem_cpu_addr = mem_ctx->shared_buf;
        } else {
                return -EINVAL;
        }

        if (!gpu_mem->mem_bo || !gpu_mem->mem_size ||
                !gpu_mem->mem_mc_addr || !gpu_mem->mem_cpu_addr) {
                RAS_DEV_ERR(ras_core->dev, "The ras psp gpu memory is invalid!\n");
                return -ENOMEM;
        }

        return 0;
}

static int amdgpu_ras_sys_put_gpu_mem(struct ras_core_context *ras_core,
        enum gpu_mem_type mem_type, struct gpu_mem_block *gpu_mem)
{

        return 0;
}

const struct ras_sys_func amdgpu_ras_sys_fn = {
        .ras_notifier = amdgpu_ras_sys_event_notifier,
        .get_utc_second_timestamp = amdgpu_ras_sys_get_utc_second_timestamp,
        .gen_seqno = amdgpu_ras_sys_gen_seqno,
        .check_gpu_status = amdgpu_ras_sys_check_gpu_status,
        .get_device_system_info = amdgpu_ras_sys_get_device_system_info,
        .gpu_reset_lock = amdgpu_ras_sys_gpu_reset_lock,
        .detect_ras_interrupt = amdgpu_ras_sys_detect_ras_interrupt,
        .get_gpu_mem = amdgpu_ras_sys_get_gpu_mem,
        .put_gpu_mem = amdgpu_ras_sys_put_gpu_mem,
};