From 8fdb67f055ae12ff186e01675f67b06886726778 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Thu, 31 Oct 2024 13:04:49 +0000 Subject: [PATCH] [Test] Add some specific unit tests for Rspamd x25519 DH --- test/rspamd_cxx_unit_cryptobox.hxx | 52 ++++++++++++++++++++++++++++++ 1 file changed, 52 insertions(+) diff --git a/test/rspamd_cxx_unit_cryptobox.hxx b/test/rspamd_cxx_unit_cryptobox.hxx index 5829b1e43..18ccd98ce 100644 --- a/test/rspamd_cxx_unit_cryptobox.hxx +++ b/test/rspamd_cxx_unit_cryptobox.hxx @@ -21,6 +21,7 @@ #include "libcryptobox/cryptobox.h" #include #include +#include TEST_SUITE("rspamd_cryptobox") { @@ -177,6 +178,57 @@ TEST_SUITE("rspamd_cryptobox") g_free(out); g_free(decrypted); } + + TEST_CASE("rspamd x25519 scalarmult") + { + rspamd_sk_t sk; + + // Use a fixed zero secret key + memset(sk, 0, sizeof(sk)); + + // Use a well known public key + const char *pk = "k4nz984k36xmcynm1hr9kdbn6jhcxf4ggbrb1quay7f88rpm9kay"; + gsize outlen; + auto *pk_decoded = rspamd_decode_base32(pk, strlen(pk), &outlen, RSPAMD_BASE32_DEFAULT); + const unsigned char expected[32] = {95, 76, 225, 188, 0, 26, 146, 94, 70, 249, + 90, 189, 35, 51, 1, 42, 9, 37, 94, 254, 204, 55, 198, 91, 180, 90, + 46, 217, 140, 226, 211, 90}; + const auto expected_arr = std::to_array(expected); + + CHECK(outlen == 32); + unsigned char out[32]; + /* Clamp integer */ + sk[0] &= 248; + sk[31] &= 127; + sk[31] |= 64; + CHECK(crypto_scalarmult(out, sk, pk_decoded) != -1); + auto out_arr = std::to_array(out); + CHECK(out_arr == expected_arr); + } + + TEST_CASE("rspamd x25519 ecdh") + { + rspamd_sk_t sk; + + // Use a fixed zero secret key + memset(sk, 0, sizeof(sk)); + + // Use a well known public key + const char *pk = "k4nz984k36xmcynm1hr9kdbn6jhcxf4ggbrb1quay7f88rpm9kay"; + gsize outlen; + auto *pk_decoded = rspamd_decode_base32(pk, strlen(pk), &outlen, RSPAMD_BASE32_DEFAULT); + const unsigned char expected[32] = {61, 109, 220, 195, 100, 174, 127, 237, 148, + 122, 154, 61, 165, 83, 93, 105, 127, 166, 153, 112, 103, 224, 2, 200, + 136, 243, 73, 51, 8, 163, 150, 7}; + const auto expected_arr = std::to_array(expected); + + CHECK(outlen == 32); + unsigned char out[32]; + + rspamd_cryptobox_nm(out, pk_decoded, sk); + auto out_arr = std::to_array(out); + CHECK(out_arr == expected_arr); + } } #endif -- 2.39.5