50 lines
1.3 KiB
C++
50 lines
1.3 KiB
C++
|
#include <algorithm>
|
||
|
|
#include <vector>
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
#include "capi/pqapi.h"
|
||
|
|
|
||
|
|
extern const kem_params_t kems[];
|
||
|
|
extern const sig_params_t sigs[];
|
||
|
|
|
||
|
|
TEST(Kyber,KEMOneOff) {
|
||
|
|
|
||
|
|
for (int i=0; i<3; i++) {
|
||
|
|
const params_t *p = (params_t*)&kems[i];
|
||
|
|
|
||
|
|
std::vector<uint8_t> ct(ciphertext_bsz(p));
|
||
|
|
std::vector<uint8_t> ss1(shared_secret_bsz(p));
|
||
|
|
std::vector<uint8_t> ss2(shared_secret_bsz(p));
|
||
|
|
std::vector<uint8_t> sk(private_key_bsz(p));
|
||
|
|
std::vector<uint8_t> pk(public_key_bsz(p));
|
||
|
|
|
||
|
|
ASSERT_TRUE(
|
||
|
|
pqc_keygen(p, pk.data(), sk.data()));
|
||
|
|
ASSERT_TRUE(
|
||
|
|
pqc_kem_encapsulate(p, ct.data(), ss1.data(), pk.data()));
|
||
|
|
ASSERT_TRUE(
|
||
|
|
pqc_kem_decapsulate(p, ss2.data(), ct.data(), sk.data()));
|
||
|
|
ASSERT_TRUE(
|
||
|
|
std::equal(ss1.begin(), ss1.end(), ss2.begin()));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(Kyber,SIGNOneOff) {
|
||
|
|
|
||
|
|
for (int i=0; i<2; i++) {
|
||
|
|
const params_t *p = (params_t*)&sigs[i];
|
||
|
|
|
||
|
|
uint8_t msg[1234];
|
||
|
|
std::vector<uint8_t> sig(signature_bsz(p));
|
||
|
|
std::vector<uint8_t> sk(private_key_bsz(p));
|
||
|
|
std::vector<uint8_t> pk(public_key_bsz(p));
|
||
|
|
|
||
|
|
ASSERT_TRUE(
|
||
|
|
pqc_keygen(p, pk.data(), sk.data()));
|
||
|
|
uint64_t sigsz = sig.size();
|
||
|
|
ASSERT_TRUE(
|
||
|
|
pqc_sig_create(p, sig.data(), &sigsz, msg, 1234, sk.data()));
|
||
|
|
ASSERT_TRUE(
|
||
|
|
pqc_sig_verify(p, sig.data(), sigsz, msg, 1234, pk.data()));
|
||
|
|
}
|
||
|
|
}
|