/* SPDX-License-Identifier: MIT
 *
 * Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 */
#include "priv.h"

#include <nvhw/drf.h>
#include <nvhw/ref/gb202/dev_therm.h>

static int
gb202_fsp_wait_secure_boot(struct nvkm_fsp *fsp)
{
        struct nvkm_device *device = fsp->subdev.device;
        unsigned timeout_ms = 4000;

        do {
                u32 status = NVKM_RD32(device, NV_THERM, I2CS_SCRATCH, FSP_BOOT_COMPLETE_STATUS);

                if (status == NV_THERM_I2CS_SCRATCH_FSP_BOOT_COMPLETE_STATUS_SUCCESS)
                        return 0;

                usleep_range(1000, 2000);
        } while (timeout_ms--);

        return -ETIMEDOUT;
}

static const struct nvkm_fsp_func
gb202_fsp = {
        .wait_secure_boot = gb202_fsp_wait_secure_boot,
        .cot = {
                .version = 2,
                .size_hash = 48,
                .size_pkey = 97,
                .size_sig = 96,
                .boot_gsp_fmc = gh100_fsp_boot_gsp_fmc,
        },
};

int
gb202_fsp_new(struct nvkm_device *device,
              enum nvkm_subdev_type type, int inst, struct nvkm_fsp **pfsp)
{
        return nvkm_fsp_new_(&gb202_fsp, device, type, inst, pfsp);
}