mirror of
https://github.com/yuzu-emu/mbedtls
synced 2024-11-24 21:48:30 +00:00
503 lines
14 KiB
Text
503 lines
14 KiB
Text
/* BEGIN_HEADER */
|
|
#include <polarssl/ecp.h>
|
|
|
|
#define POLARSSL_ECP_PF_UNKNOWN -1
|
|
/* END_HEADER */
|
|
|
|
/* BEGIN_DEPENDENCIES
|
|
* depends_on:POLARSSL_ECP_C
|
|
* END_DEPENDENCIES
|
|
*/
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_small_add( int a_zero, char *x_a, char *y_a, int b_zero, char *x_b,
|
|
char *y_b, int c_zero, int x_c, int y_c )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point A, B, C;
|
|
|
|
ecp_group_init( &grp );
|
|
ecp_point_init( &A ); ecp_point_init( &B ); ecp_point_init( &C );
|
|
|
|
TEST_ASSERT( ecp_group_read_string( &grp, 10,
|
|
"47", "4", "17", "42", "13" ) == 0 );
|
|
|
|
if( a_zero )
|
|
ecp_set_zero( &A );
|
|
else
|
|
TEST_ASSERT( ecp_point_read_string( &A, 10, x_a, y_a ) == 0 );
|
|
|
|
if( b_zero )
|
|
ecp_set_zero( &B );
|
|
else
|
|
TEST_ASSERT( ecp_point_read_string( &B, 10, x_b, y_b ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_add( &grp, &C, &A, &B ) == 0 );
|
|
|
|
if( c_zero )
|
|
TEST_ASSERT( mpi_cmp_int( &C.Z, 0 ) == 0 );
|
|
else
|
|
{
|
|
TEST_ASSERT( mpi_cmp_int( &C.X, x_c ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_int( &C.Y, y_c ) == 0 );
|
|
}
|
|
|
|
TEST_ASSERT( ecp_add( &grp, &C, &B, &A ) == 0 );
|
|
|
|
if( c_zero )
|
|
TEST_ASSERT( mpi_cmp_int( &C.Z, 0 ) == 0 );
|
|
else
|
|
{
|
|
TEST_ASSERT( mpi_cmp_int( &C.X, x_c ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_int( &C.Y, y_c ) == 0 );
|
|
}
|
|
|
|
ecp_group_free( &grp );
|
|
ecp_point_free( &A ); ecp_point_free( &B ); ecp_point_free( &C );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_small_sub( int a_zero, char *x_a, char *y_a, int b_zero, char *x_b,
|
|
char *y_b, int c_zero, int x_c, int y_c )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point A, B, C;
|
|
|
|
ecp_group_init( &grp );
|
|
ecp_point_init( &A ); ecp_point_init( &B ); ecp_point_init( &C );
|
|
|
|
TEST_ASSERT( ecp_group_read_string( &grp, 10,
|
|
"47", "4", "17", "42", "13" ) == 0 );
|
|
|
|
if( a_zero )
|
|
ecp_set_zero( &A );
|
|
else
|
|
TEST_ASSERT( ecp_point_read_string( &A, 10, x_a, y_a ) == 0 );
|
|
|
|
if( b_zero )
|
|
ecp_set_zero( &B );
|
|
else
|
|
TEST_ASSERT( ecp_point_read_string( &B, 10, x_b, y_b ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_sub( &grp, &C, &A, &B ) == 0 );
|
|
|
|
if( c_zero )
|
|
TEST_ASSERT( mpi_cmp_int( &C.Z, 0 ) == 0 );
|
|
else
|
|
{
|
|
TEST_ASSERT( mpi_cmp_int( &C.X, x_c ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_int( &C.Y, y_c ) == 0 );
|
|
}
|
|
|
|
ecp_group_free( &grp );
|
|
ecp_point_free( &A ); ecp_point_free( &B ); ecp_point_free( &C );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_small_mul( int m_str, int r_zero, int x_r, int y_r, int ret )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point R;
|
|
mpi m;
|
|
|
|
ecp_group_init( &grp );
|
|
ecp_point_init( &R );
|
|
mpi_init( &m );
|
|
|
|
TEST_ASSERT( ecp_group_read_string( &grp, 10,
|
|
"47", "4", "17", "42", "13" ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_lset( &m, m_str ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_mul( &grp, &R, &m, &grp.G ) == ret );
|
|
|
|
if( r_zero )
|
|
TEST_ASSERT( mpi_cmp_int( &R.Z, 0 ) == 0 );
|
|
else
|
|
{
|
|
TEST_ASSERT( mpi_cmp_int( &R.X, x_r ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_int( &R.Y, y_r ) == 0 );
|
|
}
|
|
|
|
ecp_group_free( &grp );
|
|
ecp_point_free( &R );
|
|
mpi_free( &m );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_small_check_pub( int x, int y, int z, int ret )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point P;
|
|
|
|
ecp_group_init( &grp );
|
|
ecp_point_init( &P );
|
|
|
|
TEST_ASSERT( ecp_group_read_string( &grp, 10,
|
|
"47", "4", "17", "42", "13" ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_lset( &P.X, x ) == 0 );
|
|
TEST_ASSERT( mpi_lset( &P.Y, y ) == 0 );
|
|
TEST_ASSERT( mpi_lset( &P.Z, z ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &P ) == ret );
|
|
|
|
ecp_group_free( &grp );
|
|
ecp_point_free( &P );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_test_vect( int id, char *dA_str, char *xA_str, char *yA_str,
|
|
char *dB_str, char *xB_str, char *yB_str, char *xZ_str,
|
|
char *yZ_str )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point R;
|
|
mpi dA, xA, yA, dB, xB, yB, xZ, yZ;
|
|
|
|
ecp_group_init( &grp ); ecp_point_init( &R );
|
|
mpi_init( &dA ); mpi_init( &xA ); mpi_init( &yA ); mpi_init( &dB );
|
|
mpi_init( &xB ); mpi_init( &yB ); mpi_init( &xZ ); mpi_init( &yZ );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &grp.G ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_read_string( &dA, 16, dA_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &xA, 16, xA_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &yA, 16, yA_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &dB, 16, dB_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &xB, 16, xB_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &yB, 16, yB_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &xZ, 16, xZ_str ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &yZ, 16, yZ_str ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_mul( &grp, &R, &dA, &grp.G ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.X, &xA ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.Y, &yA ) == 0 );
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &R ) == 0 );
|
|
TEST_ASSERT( ecp_mul( &grp, &R, &dB, &R ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.X, &xZ ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.Y, &yZ ) == 0 );
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &R ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_mul( &grp, &R, &dB, &grp.G ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.X, &xB ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.Y, &yB ) == 0 );
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &R ) == 0 );
|
|
TEST_ASSERT( ecp_mul( &grp, &R, &dA, &R ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.X, &xZ ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &R.Y, &yZ ) == 0 );
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &R ) == 0 );
|
|
|
|
ecp_group_free( &grp ); ecp_point_free( &R );
|
|
mpi_free( &dA ); mpi_free( &xA ); mpi_free( &yA ); mpi_free( &dB );
|
|
mpi_free( &xB ); mpi_free( &yB ); mpi_free( &xZ ); mpi_free( &yZ );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_fast_mod( int id, char *N_str )
|
|
{
|
|
ecp_group grp;
|
|
mpi N, R;
|
|
|
|
mpi_init( &N ); mpi_init( &R );
|
|
ecp_group_init( &grp );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &N, 16, N_str ) == 0 );
|
|
|
|
/*
|
|
* Store correct result before we touch N
|
|
*/
|
|
TEST_ASSERT( mpi_mod_mpi( &R, &N, &grp.P ) == 0 );
|
|
|
|
TEST_ASSERT( grp.modp( &N ) == 0 );
|
|
TEST_ASSERT( mpi_msb( &N ) <= grp.pbits + 3 );
|
|
|
|
/*
|
|
* Use mod rather than addition/substraction in case previous test fails
|
|
*/
|
|
TEST_ASSERT( mpi_mod_mpi( &N, &N, &grp.P ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &N, &R ) == 0 );
|
|
|
|
mpi_free( &N ); mpi_free( &R );
|
|
ecp_group_free( &grp );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_write_binary( int id, char *x, char *y, char *z, int format,
|
|
char *out, int blen, int ret )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point P;
|
|
unsigned char buf[256], str[512];
|
|
size_t olen;
|
|
|
|
memset( buf, 0, sizeof( buf ) );
|
|
memset( str, 0, sizeof( str ) );
|
|
|
|
ecp_group_init( &grp ); ecp_point_init( &P );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_read_string( &P.X, 16, x ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &P.Y, 16, y ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &P.Z, 16, z ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_point_write_binary( &grp, &P, format,
|
|
&olen, buf, blen ) == ret );
|
|
|
|
if( ret == 0 )
|
|
{
|
|
hexify( str, buf, olen );
|
|
TEST_ASSERT( strcasecmp( (char *) str, out ) == 0 );
|
|
}
|
|
|
|
ecp_group_free( &grp ); ecp_point_free( &P );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_read_binary( int id, char *input, char *x, char *y, char *z,
|
|
int ret )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point P;
|
|
mpi X, Y, Z;
|
|
int ilen;
|
|
unsigned char buf[256];
|
|
|
|
memset( buf, 0, sizeof( buf ) );
|
|
|
|
ecp_group_init( &grp ); ecp_point_init( &P );
|
|
mpi_init( &X ); mpi_init( &Y ); mpi_init( &Z );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_read_string( &X, 16, x ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &Y, 16, y ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &Z, 16, z ) == 0 );
|
|
|
|
ilen = unhexify( buf, input );
|
|
|
|
TEST_ASSERT( ecp_point_read_binary( &grp, &P, buf, ilen ) == ret );
|
|
|
|
if( ret == 0 )
|
|
{
|
|
TEST_ASSERT( mpi_cmp_mpi( &P.X, &X ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &P.Y, &Y ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &P.Z, &Z ) == 0 );
|
|
}
|
|
|
|
ecp_group_free( &grp ); ecp_point_free( &P );
|
|
mpi_free( &X ); mpi_free( &Y ); mpi_free( &Z );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_tls_read_point( int id, char *input, char *x, char *y, char *z,
|
|
int ret )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point P;
|
|
mpi X, Y, Z;
|
|
size_t ilen;
|
|
unsigned char buf[256];
|
|
const unsigned char *vbuf = buf;
|
|
|
|
memset( buf, 0, sizeof( buf ) );
|
|
|
|
ecp_group_init( &grp ); ecp_point_init( &P );
|
|
mpi_init( &X ); mpi_init( &Y ); mpi_init( &Z );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_read_string( &X, 16, x ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &Y, 16, y ) == 0 );
|
|
TEST_ASSERT( mpi_read_string( &Z, 16, z ) == 0 );
|
|
|
|
ilen = unhexify( buf, input );
|
|
|
|
TEST_ASSERT( ecp_tls_read_point( &grp, &P, &vbuf, ilen ) == ret );
|
|
|
|
if( ret == 0 )
|
|
{
|
|
TEST_ASSERT( mpi_cmp_mpi( &P.X, &X ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &P.Y, &Y ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &P.Z, &Z ) == 0 );
|
|
TEST_ASSERT( *vbuf == 0x00 );
|
|
}
|
|
|
|
ecp_group_free( &grp ); ecp_point_free( &P );
|
|
mpi_free( &X ); mpi_free( &Y ); mpi_free( &Z );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_tls_write_read_point( int id )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point pt;
|
|
unsigned char buf[256];
|
|
const unsigned char *vbuf;
|
|
size_t olen;
|
|
|
|
ecp_group_init( &grp );
|
|
ecp_point_init( &pt );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
memset( buf, 0x00, sizeof( buf ) ); vbuf = buf;
|
|
TEST_ASSERT( ecp_tls_write_point( &grp, &grp.G,
|
|
POLARSSL_ECP_PF_COMPRESSED, &olen, buf, 256 ) == 0 );
|
|
TEST_ASSERT( ecp_tls_read_point( &grp, &pt, &vbuf, olen )
|
|
== POLARSSL_ERR_ECP_BAD_INPUT_DATA );
|
|
TEST_ASSERT( vbuf == buf + olen );
|
|
|
|
memset( buf, 0x00, sizeof( buf ) ); vbuf = buf;
|
|
TEST_ASSERT( ecp_tls_write_point( &grp, &grp.G,
|
|
POLARSSL_ECP_PF_UNCOMPRESSED, &olen, buf, 256 ) == 0 );
|
|
TEST_ASSERT( ecp_tls_read_point( &grp, &pt, &vbuf, olen ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &grp.G.X, &pt.X ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &grp.G.Y, &pt.Y ) == 0 );
|
|
TEST_ASSERT( mpi_cmp_mpi( &grp.G.Z, &pt.Z ) == 0 );
|
|
TEST_ASSERT( vbuf == buf + olen );
|
|
|
|
memset( buf, 0x00, sizeof( buf ) ); vbuf = buf;
|
|
TEST_ASSERT( ecp_set_zero( &pt ) == 0 );
|
|
TEST_ASSERT( ecp_tls_write_point( &grp, &pt,
|
|
POLARSSL_ECP_PF_COMPRESSED, &olen, buf, 256 ) == 0 );
|
|
TEST_ASSERT( ecp_tls_read_point( &grp, &pt, &vbuf, olen ) == 0 );
|
|
TEST_ASSERT( ecp_is_zero( &pt ) );
|
|
TEST_ASSERT( vbuf == buf + olen );
|
|
|
|
memset( buf, 0x00, sizeof( buf ) ); vbuf = buf;
|
|
TEST_ASSERT( ecp_set_zero( &pt ) == 0 );
|
|
TEST_ASSERT( ecp_tls_write_point( &grp, &pt,
|
|
POLARSSL_ECP_PF_UNCOMPRESSED, &olen, buf, 256 ) == 0 );
|
|
TEST_ASSERT( ecp_tls_read_point( &grp, &pt, &vbuf, olen ) == 0 );
|
|
TEST_ASSERT( ecp_is_zero( &pt ) );
|
|
TEST_ASSERT( vbuf == buf + olen );
|
|
|
|
ecp_group_free( &grp );
|
|
ecp_point_free( &pt );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_tls_read_group( char *record, int result, int bits )
|
|
{
|
|
ecp_group grp;
|
|
unsigned char buf[10];
|
|
const unsigned char *vbuf = buf;
|
|
int len, ret;
|
|
|
|
ecp_group_init( &grp );
|
|
memset( buf, 0x00, sizeof( buf ) );
|
|
|
|
len = unhexify( buf, record );
|
|
|
|
ret = ecp_tls_read_group( &grp, &vbuf, len );
|
|
|
|
TEST_ASSERT( ret == result );
|
|
if( ret == 0)
|
|
{
|
|
TEST_ASSERT( mpi_msb( &grp.P ) == (size_t) bits );
|
|
TEST_ASSERT( *vbuf == 0x00 );
|
|
}
|
|
|
|
ecp_group_free( &grp );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_tls_write_read_group( int id )
|
|
{
|
|
ecp_group grp1, grp2;
|
|
unsigned char buf[10];
|
|
const unsigned char *vbuf = buf;
|
|
size_t len;
|
|
int ret;
|
|
|
|
ecp_group_init( &grp1 );
|
|
ecp_group_init( &grp2 );
|
|
memset( buf, 0x00, sizeof( buf ) );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp1, id ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_tls_write_group( &grp1, &len, buf, 10 ) == 0 );
|
|
TEST_ASSERT( ( ret = ecp_tls_read_group( &grp2, &vbuf, len ) ) == 0 );
|
|
|
|
if( ret == 0 )
|
|
{
|
|
TEST_ASSERT( mpi_cmp_mpi( &grp1.N, &grp2.N ) == 0 );
|
|
TEST_ASSERT( grp1.id == grp2.id );
|
|
}
|
|
|
|
ecp_group_free( &grp1 );
|
|
ecp_group_free( &grp2 );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_check_privkey( int id )
|
|
{
|
|
ecp_group grp;
|
|
mpi d;
|
|
|
|
ecp_group_init( &grp );
|
|
mpi_init( &d );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
TEST_ASSERT( mpi_lset( &d, 0 ) == 0 );
|
|
TEST_ASSERT( ecp_check_privkey( &grp, &d ) == POLARSSL_ERR_ECP_GENERIC );
|
|
|
|
TEST_ASSERT( mpi_copy( &d, &grp.N ) == 0 );
|
|
TEST_ASSERT( ecp_check_privkey( &grp, &d ) == POLARSSL_ERR_ECP_GENERIC );
|
|
|
|
ecp_group_free( &grp );
|
|
mpi_free( &d );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_gen_keypair( int id )
|
|
{
|
|
ecp_group grp;
|
|
ecp_point Q;
|
|
mpi d;
|
|
rnd_pseudo_info rnd_info;
|
|
|
|
ecp_group_init( &grp );
|
|
ecp_point_init( &Q );
|
|
mpi_init( &d );
|
|
memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
|
|
|
|
TEST_ASSERT( ecp_use_known_dp( &grp, id ) == 0 );
|
|
|
|
TEST_ASSERT( ecp_gen_keypair( &grp, &d, &Q, &rnd_pseudo_rand, &rnd_info )
|
|
== 0 );
|
|
|
|
TEST_ASSERT( ecp_check_pubkey( &grp, &Q ) == 0 );
|
|
TEST_ASSERT( ecp_check_privkey( &grp, &d ) == 0 );
|
|
|
|
ecp_group_free( &grp );
|
|
ecp_point_free( &Q );
|
|
mpi_free( &d );
|
|
}
|
|
/* END_CASE */
|
|
|
|
/* BEGIN_CASE */
|
|
void ecp_selftest()
|
|
{
|
|
TEST_ASSERT( ecp_self_test( 0 ) == 0 );
|
|
}
|
|
/* END_CASE */
|