// SPDX-License-Identifier: GPL-2.0-only

#include "../hwprobe/hwprobe.h"
#include <asm/vendor/thead.h>
#include <stdbool.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/wait.h>

bool is_xtheadvector_supported(void)
{
        struct riscv_hwprobe pair;

        pair.key = RISCV_HWPROBE_KEY_VENDOR_EXT_THEAD_0;
        riscv_hwprobe(&pair, 1, 0, NULL, 0);
        return pair.value & RISCV_HWPROBE_VENDOR_EXT_XTHEADVECTOR;
}

bool is_vector_supported(void)
{
        struct riscv_hwprobe pair;

        pair.key = RISCV_HWPROBE_KEY_IMA_EXT_0;
        riscv_hwprobe(&pair, 1, 0, NULL, 0);
        return pair.value & RISCV_HWPROBE_EXT_ZVE32X;
}

unsigned long get_vr_len(void)
{
        unsigned long vlenb;

        if (is_vector_supported()) {
                asm volatile("csrr %[vlenb], vlenb" : [vlenb] "=r"(vlenb));
                return vlenb;
        }

        if (is_xtheadvector_supported()) {
                asm volatile (
                        // 0 | zimm[10:0] | rs1 | 1 1 1 | rd | 1010111 | vsetvli
                        // vsetvli      t4, x0, e8, m1, d1
                        ".4byte         0b00000000000000000111111011010111\n\t"
                        "mv             %[vlenb], t4\n\t"
                        : [vlenb] "=r"(vlenb) : : "memory", "t4");
                return vlenb;
        }

        printf("WARNING: vector not supported\n");
        return 0;
}

int launch_test(char *next_program, int test_inherit, int xtheadvector)
{
        char *exec_argv[4], *exec_envp[1];
        int rc, pid, status;

        pid = fork();
        if (pid < 0) {
                printf("fork failed %d", pid);
                return -1;
        }

        if (!pid) {
                exec_argv[0] = next_program;
                exec_argv[1] = test_inherit != 0 ? "x" : NULL;
                exec_argv[2] = xtheadvector != 0 ? "x" : NULL;
                exec_argv[3] = NULL;
                exec_envp[0] = NULL;
                /* launch the program again to check inherit */
                rc = execve(next_program, exec_argv, exec_envp);
                if (rc) {
                        perror("execve");
                        printf("child execve failed %d\n", rc);
                        exit(-1);
                }
        }

        rc = waitpid(-1, &status, 0);
        if (rc < 0) {
                printf("waitpid failed\n");
                return -3;
        }

        if ((WIFEXITED(status) && WEXITSTATUS(status) == -1) ||
            WIFSIGNALED(status)) {
                printf("child exited abnormally\n");
                return -4;
        }

        return WEXITSTATUS(status);
}