// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2024 Rivos Inc.
 */

#include <linux/cpu.h>
#include <linux/device.h>
#include <linux/sprintf.h>

#include <asm/bugs.h>
#include <asm/vendor_extensions/thead.h>

static enum mitigation_state ghostwrite_state;

void ghostwrite_set_vulnerable(void)
{
        ghostwrite_state = VULNERABLE;
}

/*
 * Vendor extension alternatives will use the value set at the time of boot
 * alternative patching, thus this must be called before boot alternatives are
 * patched (and after extension probing) to be effective.
 *
 * Returns true if mitgated, false otherwise.
 */
bool ghostwrite_enable_mitigation(void)
{
        if (IS_ENABLED(CONFIG_RISCV_ISA_XTHEADVECTOR) &&
            ghostwrite_state == VULNERABLE && !cpu_mitigations_off()) {
                disable_xtheadvector();
                ghostwrite_state = MITIGATED;
                return true;
        }

        return false;
}

enum mitigation_state ghostwrite_get_state(void)
{
        return ghostwrite_state;
}

ssize_t cpu_show_ghostwrite(struct device *dev, struct device_attribute *attr, char *buf)
{
        if (IS_ENABLED(CONFIG_RISCV_ISA_XTHEADVECTOR)) {
                switch (ghostwrite_state) {
                case UNAFFECTED:
                        return sprintf(buf, "Not affected\n");
                case MITIGATED:
                        return sprintf(buf, "Mitigation: xtheadvector disabled\n");
                case VULNERABLE:
                        fallthrough;
                default:
                        return sprintf(buf, "Vulnerable\n");
                }
        } else {
                return sprintf(buf, "Not affected\n");
        }
}