// SPDX-License-Identifier: GPL-2.0-only
/*
 * AMD Secure Processor Dynamic Boost Control sample library
 *
 * Copyright (C) 2023 Advanced Micro Devices, Inc.
 *
 * Author: Mario Limonciello <mario.limonciello@amd.com>
 */

#include <assert.h>
#include <errno.h>
#include <string.h>
#include <sys/ioctl.h>

/* if uapi header isn't installed, this might not yet exist */
#ifndef __packed
#define __packed __attribute__((packed))
#endif
#include <linux/psp-dbc.h>

int get_nonce(int fd, void *nonce_out, void *signature)
{
        struct dbc_user_nonce tmp = {
                .auth_needed = !!signature,
        };

        assert(nonce_out);

        if (signature)
                memcpy(tmp.signature, signature, sizeof(tmp.signature));

        if (ioctl(fd, DBCIOCNONCE, &tmp))
                return errno;
        memcpy(nonce_out, tmp.nonce, sizeof(tmp.nonce));

        return 0;
}

int set_uid(int fd, __u8 *uid, __u8 *signature)
{
        struct dbc_user_setuid tmp;

        assert(uid);
        assert(signature);

        memcpy(tmp.uid, uid, sizeof(tmp.uid));
        memcpy(tmp.signature, signature, sizeof(tmp.signature));

        if (ioctl(fd, DBCIOCUID, &tmp))
                return errno;
        return 0;
}

int process_param(int fd, int msg_index, __u8 *signature, int *data)
{
        struct dbc_user_param tmp = {
                .msg_index = msg_index,
                .param = *data,
        };

        assert(signature);
        assert(data);

        memcpy(tmp.signature, signature, sizeof(tmp.signature));

        if (ioctl(fd, DBCIOCPARAM, &tmp))
                return errno;

        *data = tmp.param;
        memcpy(signature, tmp.signature, sizeof(tmp.signature));
        return 0;
}