// SPDX-License-Identifier: GPL-2.0-or-later
/* Self-testing for signature checking.
 *
 * Copyright (C) 2022 Red Hat, Inc. All Rights Reserved.
 * Written by David Howells (dhowells@redhat.com)
 */

#include <crypto/pkcs7.h>
#include <linux/cred.h>
#include <linux/kernel.h>
#include <linux/key.h>
#include <linux/module.h>
#include "selftest.h"
#include "x509_parser.h"

void fips_signature_selftest(const char *name,
        const u8 *keys, size_t keys_len,
        const u8 *data, size_t data_len,
        const u8 *sig, size_t sig_len)
{
 struct key *keyring;
 int ret;

 pr_notice("Running certificate verification %s selftest\n", name);

 keyring = keyring_alloc(".certs_selftest",
    GLOBAL_ROOT_UID, GLOBAL_ROOT_GID, current_cred(),
    (KEY_POS_ALL & ~KEY_POS_SETATTR) |
    KEY_USR_VIEW | KEY_USR_READ |
    KEY_USR_SEARCH,
    KEY_ALLOC_NOT_IN_QUOTA,
    NULL, NULL);
 if (IS_ERR(keyring))
  panic("Can't allocate certs %s selftest keyring: %ld\n", name, PTR_ERR(keyring));

 ret = x509_load_certificate_list(keys, keys_len, keyring);
 if (ret < 0)
  panic("Can't allocate certs %s selftest keyring: %d\n", name, ret);

 struct pkcs7_message *pkcs7;

 pkcs7 = pkcs7_parse_message(sig, sig_len);
 if (IS_ERR(pkcs7))
  panic("Certs %s selftest: pkcs7_parse_message() = %d\n", name, ret);

 pkcs7_supply_detached_data(pkcs7, data, data_len);

 ret = pkcs7_verify(pkcs7, VERIFYING_MODULE_SIGNATURE);
 if (ret < 0)
  panic("Certs %s selftest: pkcs7_verify() = %d\n", name, ret);

 ret = pkcs7_validate_trust(pkcs7, keyring);
 if (ret < 0)
  panic("Certs %s selftest: pkcs7_validate_trust() = %d\n", name, ret);

 pkcs7_free_message(pkcs7);

 key_put(keyring);
}

static int __init fips_signature_selftest_init(void)
{
 fips_signature_selftest_rsa();
 fips_signature_selftest_ecdsa();
 return 0;
}

late_initcall(fips_signature_selftest_init);

MODULE_DESCRIPTION("X.509 self tests");
MODULE_AUTHOR("Red Hat, Inc.");
MODULE_LICENSE("GPL");