1/* BEGIN_HEADER */
2#include "mbedtls/ecdsa.h"
3/* END_HEADER */
4
5/* BEGIN_DEPENDENCIES
6 * depends_on:MBEDTLS_ECDSA_C
7 * END_DEPENDENCIES
8 */
9
10/* BEGIN_CASE */
11void ecdsa_prim_random( int id )
12{
13    mbedtls_ecp_group grp;
14    mbedtls_ecp_point Q;
15    mbedtls_mpi d, r, s;
16    rnd_pseudo_info rnd_info;
17    unsigned char buf[66];
18
19    mbedtls_ecp_group_init( &grp );
20    mbedtls_ecp_point_init( &Q );
21    mbedtls_mpi_init( &d ); mbedtls_mpi_init( &r ); mbedtls_mpi_init( &s );
22    memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
23    memset( buf, 0, sizeof( buf ) );
24
25    /* prepare material for signature */
26    TEST_ASSERT( rnd_pseudo_rand( &rnd_info, buf, sizeof( buf ) ) == 0 );
27    TEST_ASSERT( mbedtls_ecp_group_load( &grp, id ) == 0 );
28    TEST_ASSERT( mbedtls_ecp_gen_keypair( &grp, &d, &Q, &rnd_pseudo_rand, &rnd_info )
29                 == 0 );
30
31    TEST_ASSERT( mbedtls_ecdsa_sign( &grp, &r, &s, &d, buf, sizeof( buf ),
32                             &rnd_pseudo_rand, &rnd_info ) == 0 );
33    TEST_ASSERT( mbedtls_ecdsa_verify( &grp, buf, sizeof( buf ), &Q, &r, &s ) == 0 );
34
35exit:
36    mbedtls_ecp_group_free( &grp );
37    mbedtls_ecp_point_free( &Q );
38    mbedtls_mpi_free( &d ); mbedtls_mpi_free( &r ); mbedtls_mpi_free( &s );
39}
40/* END_CASE */
41
42/* BEGIN_CASE */
43void ecdsa_prim_test_vectors( int id, char *d_str, char *xQ_str, char *yQ_str,
44                              char *k_str, char *hash_str, char *r_str,
45                              char *s_str )
46{
47    mbedtls_ecp_group grp;
48    mbedtls_ecp_point Q;
49    mbedtls_mpi d, r, s, r_check, s_check;
50    unsigned char hash[66], rnd_buf[66];
51    size_t hlen;
52    rnd_buf_info rnd_info;
53
54    mbedtls_ecp_group_init( &grp );
55    mbedtls_ecp_point_init( &Q );
56    mbedtls_mpi_init( &d ); mbedtls_mpi_init( &r ); mbedtls_mpi_init( &s );
57    mbedtls_mpi_init( &r_check ); mbedtls_mpi_init( &s_check );
58    memset( hash, 0, sizeof( hash ) );
59    memset( rnd_buf, 0, sizeof( rnd_buf ) );
60
61    TEST_ASSERT( mbedtls_ecp_group_load( &grp, id ) == 0 );
62    TEST_ASSERT( mbedtls_ecp_point_read_string( &Q, 16, xQ_str, yQ_str ) == 0 );
63    TEST_ASSERT( mbedtls_mpi_read_string( &d, 16, d_str ) == 0 );
64    TEST_ASSERT( mbedtls_mpi_read_string( &r_check, 16, r_str ) == 0 );
65    TEST_ASSERT( mbedtls_mpi_read_string( &s_check, 16, s_str ) == 0 );
66    hlen = unhexify(hash, hash_str);
67    rnd_info.buf = rnd_buf;
68    rnd_info.length = unhexify( rnd_buf, k_str );
69
70    /* Fix rnd_buf by shifting it left if necessary */
71    if( grp.nbits % 8 != 0 )
72    {
73        unsigned char shift = 8 - ( grp.nbits % 8 );
74        size_t i;
75
76        for( i = 0; i < rnd_info.length - 1; i++ )
77            rnd_buf[i] = rnd_buf[i] << shift | rnd_buf[i+1] >> ( 8 - shift );
78
79        rnd_buf[rnd_info.length-1] <<= shift;
80    }
81
82    TEST_ASSERT( mbedtls_ecdsa_sign( &grp, &r, &s, &d, hash, hlen,
83                 rnd_buffer_rand, &rnd_info ) == 0 );
84
85    TEST_ASSERT( mbedtls_mpi_cmp_mpi( &r, &r_check ) == 0 );
86    TEST_ASSERT( mbedtls_mpi_cmp_mpi( &s, &s_check ) == 0 );
87
88    TEST_ASSERT( mbedtls_ecdsa_verify( &grp, hash, hlen, &Q, &r_check, &s_check ) == 0 );
89
90exit:
91    mbedtls_ecp_group_free( &grp );
92    mbedtls_ecp_point_free( &Q );
93    mbedtls_mpi_free( &d ); mbedtls_mpi_free( &r ); mbedtls_mpi_free( &s );
94    mbedtls_mpi_free( &r_check ); mbedtls_mpi_free( &s_check );
95}
96/* END_CASE */
97
98/* BEGIN_CASE depends_on:MBEDTLS_ECDSA_DETERMINISTIC */
99void ecdsa_det_test_vectors( int id, char *d_str, int md_alg,
100                             char *msg, char *r_str, char *s_str )
101{
102    mbedtls_ecp_group grp;
103    mbedtls_mpi d, r, s, r_check, s_check;
104    unsigned char hash[MBEDTLS_MD_MAX_SIZE];
105    size_t hlen;
106    const mbedtls_md_info_t *md_info;
107
108    mbedtls_ecp_group_init( &grp );
109    mbedtls_mpi_init( &d ); mbedtls_mpi_init( &r ); mbedtls_mpi_init( &s );
110    mbedtls_mpi_init( &r_check ); mbedtls_mpi_init( &s_check );
111    memset( hash, 0, sizeof( hash ) );
112
113    TEST_ASSERT( mbedtls_ecp_group_load( &grp, id ) == 0 );
114    TEST_ASSERT( mbedtls_mpi_read_string( &d, 16, d_str ) == 0 );
115    TEST_ASSERT( mbedtls_mpi_read_string( &r_check, 16, r_str ) == 0 );
116    TEST_ASSERT( mbedtls_mpi_read_string( &s_check, 16, s_str ) == 0 );
117
118    md_info = mbedtls_md_info_from_type( md_alg );
119    TEST_ASSERT( md_info != NULL );
120    hlen = mbedtls_md_get_size( md_info );
121    mbedtls_md( md_info, (const unsigned char *) msg, strlen( msg ), hash );
122
123    TEST_ASSERT( mbedtls_ecdsa_sign_det( &grp, &r, &s, &d, hash, hlen, md_alg ) == 0 );
124
125    TEST_ASSERT( mbedtls_mpi_cmp_mpi( &r, &r_check ) == 0 );
126    TEST_ASSERT( mbedtls_mpi_cmp_mpi( &s, &s_check ) == 0 );
127
128exit:
129    mbedtls_ecp_group_free( &grp );
130    mbedtls_mpi_free( &d ); mbedtls_mpi_free( &r ); mbedtls_mpi_free( &s );
131    mbedtls_mpi_free( &r_check ); mbedtls_mpi_free( &s_check );
132}
133/* END_CASE */
134
135/* BEGIN_CASE depends_on:MBEDTLS_SHA256_C */
136void ecdsa_write_read_random( int id )
137{
138    mbedtls_ecdsa_context ctx;
139    rnd_pseudo_info rnd_info;
140    unsigned char hash[32];
141    unsigned char sig[200];
142    size_t sig_len, i;
143
144    mbedtls_ecdsa_init( &ctx );
145    memset( &rnd_info, 0x00, sizeof( rnd_pseudo_info ) );
146    memset( hash, 0, sizeof( hash ) );
147    memset( sig, 0x2a, sizeof( sig ) );
148
149    /* prepare material for signature */
150    TEST_ASSERT( rnd_pseudo_rand( &rnd_info, hash, sizeof( hash ) ) == 0 );
151
152    /* generate signing key */
153    TEST_ASSERT( mbedtls_ecdsa_genkey( &ctx, id, &rnd_pseudo_rand, &rnd_info ) == 0 );
154
155    /* generate and write signature, then read and verify it */
156    TEST_ASSERT( mbedtls_ecdsa_write_signature( &ctx, MBEDTLS_MD_SHA256,
157                 hash, sizeof( hash ),
158                 sig, &sig_len, &rnd_pseudo_rand, &rnd_info ) == 0 );
159    TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
160                 sig, sig_len ) == 0 );
161
162    /* check we didn't write past the announced length */
163    for( i = sig_len; i < sizeof( sig ); i++ )
164        TEST_ASSERT( sig[i] == 0x2a );
165
166    /* try verification with invalid length */
167    TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
168                 sig, sig_len - 1 ) != 0 );
169    TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
170                 sig, sig_len + 1 ) != 0 );
171
172    /* try invalid sequence tag */
173    sig[0]++;
174    TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
175                 sig, sig_len ) != 0 );
176    sig[0]--;
177
178    /* try modifying r */
179    sig[10]++;
180    TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
181                 sig, sig_len ) != 0 );
182    sig[10]--;
183
184    /* try modifying s */
185    sig[sig_len - 1]++;
186    TEST_ASSERT( mbedtls_ecdsa_read_signature( &ctx, hash, sizeof( hash ),
187                 sig, sig_len ) != 0 );
188    sig[sig_len - 1]--;
189
190exit:
191    mbedtls_ecdsa_free( &ctx );
192}
193/* END_CASE */
194
195