10f97f3bfc
This reverts commit 09feb0f3d9.
(In order to make WebRTC happy this also needs to be reverted.)
123 lines
4.7 KiB
C++
123 lines
4.7 KiB
C++
/* 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 "../test/scoped_types.h"
|
|
#include "internal.h"
|
|
|
|
|
|
static bool TestNewhope(FileTest *t, void *arg) {
|
|
ScopedNEWHOPE_POLY a(NEWHOPE_POLY_new());
|
|
ScopedNEWHOPE_POLY s(NEWHOPE_POLY_new()), sp(NEWHOPE_POLY_new());
|
|
ScopedNEWHOPE_POLY e(NEWHOPE_POLY_new()), ep(NEWHOPE_POLY_new()),
|
|
epp(NEWHOPE_POLY_new());
|
|
ScopedNEWHOPE_POLY in_pk(NEWHOPE_POLY_new());
|
|
ScopedNEWHOPE_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]);
|
|
}
|