#include "client_test_helpers.h"
#include "sa.h"
#include "gtest/gtest.h"
using namespace client_test_helpers;
namespace {
TEST(SaCryptoSign, nominal2048RsaPkcs1v15Sha256EmptyIn) {
auto clear_key = sample_rsa_2048_pkcs8();
sa_digest_algorithm const digest_algorithm = SA_DIGEST_ALGORITHM_SHA256;
sa_rights rights;
sa_rights_set_allow_all(&rights);
auto key = create_sa_key_rsa(&rights, clear_key);
ASSERT_NE(key, nullptr);
if (*key == UNSUPPORTED_KEY)
GTEST_SKIP() << "key type, key size, or curve not supported";
size_t out_length = 0;
sa_sign_parameters_rsa_pkcs1v15 parameters = {digest_algorithm, false};
sa_status status = sa_crypto_sign(nullptr, &out_length, SA_SIGNATURE_ALGORITHM_RSA_PKCS1V15, *key,
nullptr, 0, ¶meters);
if (status == SA_STATUS_OPERATION_NOT_SUPPORTED)
GTEST_SKIP() << "Unsupported signature algorithm";
ASSERT_EQ(status, SA_STATUS_OK);
ASSERT_EQ(out_length, 256);
auto out = std::vector<uint8_t>(out_length);
auto in = std::vector<uint8_t>(0);
status = sa_crypto_sign(out.data(), &out_length, SA_SIGNATURE_ALGORITHM_RSA_PKCS1V15, *key, in.data(),
in.size(), ¶meters);
if (status == SA_STATUS_OPERATION_NOT_SUPPORTED)
GTEST_SKIP() << "Unsupported signature algorithm";
ASSERT_EQ(status, SA_STATUS_OK);
out.resize(out_length);
auto rsa_key = rsa_import_pkcs8(clear_key);
ASSERT_NE(rsa_key, nullptr);
ASSERT_TRUE(verify_rsa_pkcs1v15_openssl(rsa_key, digest_algorithm, in, out));
}
TEST(SaCryptoSign, failsRsaPkcs1v15InvalidOutLength) {
auto clear_key = sample_rsa_2048_pkcs8();
sa_digest_algorithm const digest_algorithm = SA_DIGEST_ALGORITHM_SHA256;
sa_rights rights;
sa_rights_set_allow_all(&rights);
auto key = create_sa_key_rsa(&rights, clear_key);
ASSERT_NE(key, nullptr);
if (*key == UNSUPPORTED_KEY)
GTEST_SKIP() << "key type, key size, or curve not supported";
size_t out_length = 0;
sa_sign_parameters_rsa_pkcs1v15 parameters = {digest_algorithm, false};
sa_status status = sa_crypto_sign(nullptr, &out_length, SA_SIGNATURE_ALGORITHM_RSA_PKCS1V15, *key,
nullptr, 0, ¶meters);
if (status == SA_STATUS_OPERATION_NOT_SUPPORTED)
GTEST_SKIP() << "Unsupported signature algorithm";
ASSERT_EQ(status, SA_STATUS_OK);
ASSERT_EQ(out_length, 256);
auto out = std::vector<uint8_t>(out_length);
out_length -= 1;
auto in = random(25);
status = sa_crypto_sign(out.data(), &out_length, SA_SIGNATURE_ALGORITHM_RSA_PKCS1V15, *key, in.data(),
in.size(), ¶meters);
if (status == SA_STATUS_OPERATION_NOT_SUPPORTED)
GTEST_SKIP() << "Unsupported signature algorithm";
ASSERT_EQ(status, SA_STATUS_INVALID_PARAMETER);
}
TEST(SaCryptoSign, failsRsaPkcs1v15NullParameters) {
auto clear_key = sample_rsa_2048_pkcs8();
sa_rights rights;
sa_rights_set_allow_all(&rights);
auto key = create_sa_key_rsa(&rights, clear_key);
ASSERT_NE(key, nullptr);
if (*key == UNSUPPORTED_KEY)
GTEST_SKIP() << "key type, key size, or curve not supported";
auto out = std::vector<uint8_t>(512);
size_t out_length = out.size();
auto in = random(25);
sa_status const status = sa_crypto_sign(out.data(), &out_length, SA_SIGNATURE_ALGORITHM_RSA_PKCS1V15, *key,
in.data(), in.size(), nullptr);
if (status == SA_STATUS_OPERATION_NOT_SUPPORTED)
GTEST_SKIP() << "Unsupported signature algorithm";
ASSERT_EQ(status, SA_STATUS_NULL_PARAMETER);
}
}