blob: 64aa0bbc3b42bca87b8d969b4d692d2eb6d4d76f [file] [log] [blame]
/* Copyright (c) 2016, Google Inc.
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <openssl/crypto.h>
#include <openssl/rand.h>
#include "../test/file_test.h"
#include "internal.h"
static bool TestNewhope(FileTest *t, void *arg) {
bssl::UniquePtr<NEWHOPE_POLY> a(NEWHOPE_POLY_new());
bssl::UniquePtr<NEWHOPE_POLY> s(NEWHOPE_POLY_new()), sp(NEWHOPE_POLY_new());
bssl::UniquePtr<NEWHOPE_POLY> e(NEWHOPE_POLY_new()), ep(NEWHOPE_POLY_new()),
epp(NEWHOPE_POLY_new());
bssl::UniquePtr<NEWHOPE_POLY> in_pk(NEWHOPE_POLY_new());
bssl::UniquePtr<NEWHOPE_POLY> in_rec(NEWHOPE_POLY_new());
if (t->GetType() == "InRandA") {
std::vector<uint8_t> a_bytes, s_bytes, e_bytes, expected_pk;
if (!t->GetBytes(&a_bytes, "InRandA") ||
!t->GetBytes(&s_bytes, "InNoiseS") ||
!t->GetBytes(&e_bytes, "InNoiseE") ||
!t->GetBytes(&expected_pk, "OutPK")) {
return false;
}
NEWHOPE_POLY_frombytes(a.get(), a_bytes.data());
NEWHOPE_POLY_frombytes(s.get(), s_bytes.data());
NEWHOPE_POLY_frombytes(e.get(), e_bytes.data());
NEWHOPE_POLY pk;
NEWHOPE_offer_computation(&pk, s.get(), e.get(), a.get());
uint8_t pk_bytes[NEWHOPE_POLY_LENGTH];
NEWHOPE_POLY_tobytes(pk_bytes, &pk);
return t->ExpectBytesEqual(expected_pk.data(), expected_pk.size(),
pk_bytes, NEWHOPE_POLY_LENGTH);
} else if (t->GetType() == "InPK") {
std::vector<uint8_t> rand, in_pk_bytes, a_bytes, sp_bytes, ep_bytes,
epp_bytes, expected_pk, expected_rec, expected_key;
if (!t->GetBytes(&in_pk_bytes, "InPK") ||
!t->GetBytes(&rand, "InRand") ||
!t->GetBytes(&a_bytes, "InA") ||
!t->GetBytes(&sp_bytes, "InNoiseSP") ||
!t->GetBytes(&ep_bytes, "InNoiseEP") ||
!t->GetBytes(&epp_bytes, "InNoiseEPP") ||
!t->GetBytes(&expected_pk, "OutPK") ||
!t->GetBytes(&expected_rec, "OutRec") ||
!t->GetBytes(&expected_key, "Key")) {
return false;
}
NEWHOPE_POLY_frombytes(in_pk.get(), in_pk_bytes.data());
NEWHOPE_POLY_frombytes(a.get(), a_bytes.data());
NEWHOPE_POLY_frombytes(sp.get(), sp_bytes.data());
NEWHOPE_POLY_frombytes(ep.get(), ep_bytes.data());
NEWHOPE_POLY_frombytes(epp.get(), epp_bytes.data());
uint8_t key[NEWHOPE_KEY_LENGTH];
NEWHOPE_POLY pk, rec;
NEWHOPE_accept_computation(key, &pk, &rec,
sp.get(), ep.get(), epp.get(),
rand.data(), in_pk.get(), a.get());
uint8_t pk_bytes[NEWHOPE_POLY_LENGTH], rec_bytes[NEWHOPE_POLY_LENGTH];
NEWHOPE_POLY_tobytes(pk_bytes, &pk);
NEWHOPE_POLY_tobytes(rec_bytes, &rec);
return
t->ExpectBytesEqual(expected_key.data(), expected_key.size(),
key, NEWHOPE_KEY_LENGTH) &&
t->ExpectBytesEqual(expected_pk.data(), expected_pk.size(),
pk_bytes, NEWHOPE_POLY_LENGTH) &&
t->ExpectBytesEqual(expected_rec.data(), expected_rec.size(),
rec_bytes, NEWHOPE_POLY_LENGTH);
} else if (t->GetType() == "InNoiseS") {
std::vector<uint8_t> s_bytes, in_pk_bytes, in_rec_bytes, expected_key;
if (!t->GetBytes(&s_bytes, "InNoiseS") ||
!t->GetBytes(&in_pk_bytes, "InPK") ||
!t->GetBytes(&in_rec_bytes, "InRec") ||
!t->GetBytes(&expected_key, "Key")) {
return false;
}
NEWHOPE_POLY_frombytes(s.get(), s_bytes.data());
NEWHOPE_POLY_frombytes(in_pk.get(), in_pk_bytes.data());
NEWHOPE_POLY_frombytes(in_rec.get(), in_rec_bytes.data());
uint8_t key[NEWHOPE_KEY_LENGTH];
NEWHOPE_finish_computation(key, s.get(), in_pk.get(), in_rec.get());
return t->ExpectBytesEqual(expected_key.data(), expected_key.size(), key,
NEWHOPE_KEY_LENGTH);
} else {
t->PrintLine("Unknown test '%s'", t->GetType().c_str());
return false;
}
}
int main(int argc, char **argv) {
CRYPTO_library_init();
if (argc != 2) {
fprintf(stderr, "%s <test file>\n", argv[0]);
return 1;
}
return FileTestMain(TestNewhope, nullptr, argv[1]);
}