// SPDX-License-Identifier: GPL-2.0
// Copyright (C) 2020 ARM Limited

#include <stdio.h>
#include <stdlib.h>
#include <sys/auxv.h>

#include "helper.h"

int main(void)
{
        struct signatures signed_vals;
        unsigned long hwcaps;
        size_t val;

        size_t size = fread(&val, sizeof(size_t), 1, stdin);

        if (size != 1) {
                fprintf(stderr, "Could not read input from stdin\n");
                return EXIT_FAILURE;
        }

        /* don't try to execute illegal (unimplemented) instructions) caller
         * should have checked this and keep worker simple
         */
        hwcaps = getauxval(AT_HWCAP);

        if (hwcaps & HWCAP_PACA) {
                signed_vals.keyia = keyia_sign(val);
                signed_vals.keyib = keyib_sign(val);
                signed_vals.keyda = keyda_sign(val);
                signed_vals.keydb = keydb_sign(val);
        }
        signed_vals.keyg = (hwcaps & HWCAP_PACG) ?  keyg_sign(val) : 0;

        fwrite(&signed_vals, sizeof(struct signatures), 1, stdout);

        return 0;
}