unit-tests: Move ntru_test_rng_t to a utility class in libtest

This commit is contained in:
Tobias Brunner
2013-12-04 20:32:59 +01:00
committed by Andreas Steffen
parent 6354466a5b
commit a24eec4649
6 changed files with 37 additions and 48 deletions
@@ -14,7 +14,6 @@ libstrongswan_ntru_la_SOURCES = \
ntru_plugin.h ntru_plugin.c \
ntru_drbg.h ntru_drbg.c \
ntru_ke.h ntru_ke.c \
ntru_test_rng.h ntru_test_rng.c \
ntru_crypto/ntru_crypto.h ntru_crypto/ntru_crypto_error.h \
ntru_crypto/ntru_crypto_ntru_convert.h \
ntru_crypto/ntru_crypto_ntru_convert.c \
@@ -1,89 +0,0 @@
/*
* Copyright (C) 2013 Andreas Steffen
* HSR Hochschule fuer Technik Rapperswil
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2 of the License, or (at your
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* for more details.
*/
#include "ntru_test_rng.h"
typedef struct private_ntru_test_rng_t private_ntru_test_rng_t;
/**
* Private data of an ntru_test_rng_t object.
*/
struct private_ntru_test_rng_t {
/**
* Public ntru_test_rng_t interface.
*/
ntru_test_rng_t public;
/**
* entropy string
*/
chunk_t entropy;
};
METHOD(rng_t, get_bytes, bool,
private_ntru_test_rng_t *this, size_t bytes, u_int8_t *buffer)
{
if (bytes > this->entropy.len)
{
return FALSE;
}
memcpy(buffer, this->entropy.ptr, bytes);
this->entropy = chunk_skip(this->entropy, bytes);
return TRUE;
}
METHOD(rng_t, allocate_bytes, bool,
private_ntru_test_rng_t *this, size_t bytes, chunk_t *chunk)
{
if (bytes > this->entropy.len)
{
*chunk = chunk_empty;
return FALSE;
}
*chunk = chunk_alloc(bytes);
memcpy(chunk->ptr, this->entropy.ptr, bytes);
this->entropy = chunk_skip(this->entropy, bytes);
return TRUE;
}
METHOD(rng_t, destroy, void,
private_ntru_test_rng_t *this)
{
free(this);
}
/*
* Described in header.
*/
rng_t *ntru_test_rng_create(chunk_t entropy)
{
private_ntru_test_rng_t *this;
INIT(this,
.public = {
.rng = {
.get_bytes = _get_bytes,
.allocate_bytes = _allocate_bytes,
.destroy = _destroy,
},
},
.entropy = entropy,
);
return &this->public.rng;
}
@@ -1,47 +0,0 @@
/*
* Copyright (C) 2013 Andreas Steffen
* HSR Hochschule fuer Technik Rapperswil
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the
* Free Software Foundation; either version 2 of the License, or (at your
* option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
* for more details.
*/
/**
* @defgroup ntru_test_rng ntru_test_rng
* @{ @ingroup ntru_p
*/
#ifndef NTRU_TEST_RNG_H_
#define NTRU_TEST_RNG_H_
typedef struct ntru_test_rng_t ntru_test_rng_t;
#include <library.h>
/**
* rng_t providing NIST SP 800-90A entropy test vectors
*/
struct ntru_test_rng_t {
/**
* Implements rng_t.
*/
rng_t rng;
};
/**
* Creates an ntru_test_rng_t instance.
*
* @param entropy entropy test vector
* @return created ntru_test_rng_t
*/
rng_t *ntru_test_rng_create(chunk_t entropy);
#endif /** NTRU_TEST_RNG_H_ @} */