blob: 066bb39425ae3e63233509240fc98eed8f0ecfbf [file] [log] [blame]
Daniel Kingadc32c02016-05-16 18:25:45 -03001/* BEGIN_HEADER */
2#include "mbedtls/poly1305.h"
3#include <stddef.h>
4/* END_HEADER */
5
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +02006/* BEGIN_DEPENDENCIES
7 * depends_on:MBEDTLS_POLY1305_C
8 * END_DEPENDENCIES
9 */
10
11/* BEGIN_CASE */
Daniel Kingadc32c02016-05-16 18:25:45 -030012void mbedtls_poly1305( char *hex_key_string, char *hex_mac_string, char *hex_src_string )
13{
Manuel Pégourié-Gonnard528524b2018-05-09 11:21:21 +020014 unsigned char src_str[375]; /* max size of binary input */
15 unsigned char key[32]; /* size set by the standard */
16 unsigned char mac[16]; /* size set by the standard */
17 unsigned char mac_str[33]; /* hex expansion of the above */
Daniel Kingadc32c02016-05-16 18:25:45 -030018 size_t src_len;
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020019 mbedtls_poly1305_context ctx;
Daniel Kingadc32c02016-05-16 18:25:45 -030020
Manuel Pégourié-Gonnard528524b2018-05-09 11:21:21 +020021 memset( src_str, 0x00, sizeof( src_str ) );
22 memset( mac_str, 0x00, sizeof( mac_str ) );
23 memset( key, 0x00, sizeof( key ) );
24 memset( mac, 0x00, sizeof( mac ) );
Daniel Kingadc32c02016-05-16 18:25:45 -030025
26 src_len = unhexify( src_str, hex_src_string );
27 unhexify( key, hex_key_string );
28
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020029 /*
30 * Test the integrated API
31 */
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +020032 TEST_ASSERT( mbedtls_poly1305_mac( key, src_str, src_len, mac ) == 0 );
Daniel Kingadc32c02016-05-16 18:25:45 -030033
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020034 hexify( mac_str, mac, 16 );
Daniel Kingadc32c02016-05-16 18:25:45 -030035 TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020036
37 /*
38 * Test the streaming API
39 */
40 mbedtls_poly1305_init( &ctx );
41
42 TEST_ASSERT( mbedtls_poly1305_starts( &ctx, key ) == 0 );
43
44 TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str, src_len ) == 0 );
45
46 TEST_ASSERT( mbedtls_poly1305_finish( &ctx, mac ) == 0 );
47
48 hexify( mac_str, mac, 16 );
49 TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
50
51 /*
52 * Test the streaming API again, piecewise
53 */
54
55 /* Don't free/init the context, in order to test that starts() does the
56 * right thing. */
Manuel Pégourié-Gonnard444f7112018-05-10 11:06:46 +020057 if( src_len >= 1 )
58 {
59 TEST_ASSERT( mbedtls_poly1305_starts( &ctx, key ) == 0 );
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020060
Manuel Pégourié-Gonnard444f7112018-05-10 11:06:46 +020061 TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str, 1 ) == 0 );
62 TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str + 1, src_len - 1 ) == 0 );
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020063
Manuel Pégourié-Gonnard444f7112018-05-10 11:06:46 +020064 TEST_ASSERT( mbedtls_poly1305_finish( &ctx, mac ) == 0 );
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020065
Manuel Pégourié-Gonnard444f7112018-05-10 11:06:46 +020066 hexify( mac_str, mac, 16 );
67 TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
68 }
69
70 /*
71 * Again with more pieces
72 */
73 if( src_len >= 2 )
74 {
75 TEST_ASSERT( mbedtls_poly1305_starts( &ctx, key ) == 0 );
76
77 TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str, 1 ) == 0 );
78 TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str + 1, 1 ) == 0 );
79 TEST_ASSERT( mbedtls_poly1305_update( &ctx, src_str + 2, src_len - 2 ) == 0 );
80
81 TEST_ASSERT( mbedtls_poly1305_finish( &ctx, mac ) == 0 );
82
83 hexify( mac_str, mac, 16 );
84 TEST_ASSERT( strcmp( (char *) mac_str, hex_mac_string ) == 0 );
85 }
Manuel Pégourié-Gonnard14656022018-05-09 12:51:54 +020086
87 mbedtls_poly1305_free( &ctx );
Daniel Kingadc32c02016-05-16 18:25:45 -030088}
89/* END_CASE */
90
Hanno Beckerae2ff022018-12-11 15:14:02 +000091/* BEGIN_CASE depends_on:MBEDTLS_CHECK_PARAMS:!MBEDTLS_PARAM_FAILED_ALT */
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +020092void poly1305_bad_params()
93{
94 unsigned char src[1];
95 unsigned char key[32];
96 unsigned char mac[16];
97 size_t src_len = sizeof( src );
98 mbedtls_poly1305_context ctx;
99
Hanno Beckerae2ff022018-12-11 15:14:02 +0000100 TEST_INVALID_PARAM( mbedtls_poly1305_init( NULL ) );
Hanno Beckera994b232018-12-12 16:44:41 +0000101 TEST_VALID_PARAM( mbedtls_poly1305_free( NULL ) );
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +0200102
Hanno Beckerae2ff022018-12-11 15:14:02 +0000103 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
104 mbedtls_poly1305_starts( NULL, key ) );
105 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
106 mbedtls_poly1305_starts( &ctx, NULL ) );
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +0200107
Hanno Beckerae2ff022018-12-11 15:14:02 +0000108 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
109 mbedtls_poly1305_update( NULL, src, 0 ) );
110 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
111 mbedtls_poly1305_update( &ctx, NULL, src_len ) );
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +0200112
Hanno Beckerae2ff022018-12-11 15:14:02 +0000113 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
114 mbedtls_poly1305_finish( NULL, mac ) );
115 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
116 mbedtls_poly1305_finish( &ctx, NULL ) );
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +0200117
Hanno Beckerae2ff022018-12-11 15:14:02 +0000118 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
119 mbedtls_poly1305_mac( NULL, src, 0, mac ) );
120 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
121 mbedtls_poly1305_mac( key, NULL, src_len, mac ) );
122 TEST_INVALID_PARAM_RET( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA,
123 mbedtls_poly1305_mac( key, src, 0, NULL ) );
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +0200124
Hanno Beckerae2ff022018-12-11 15:14:02 +0000125exit:
126 return;
Manuel Pégourié-Gonnarda8fa8b82018-05-10 10:12:36 +0200127}
128/* END_CASE */
129
130/* BEGIN_CASE depends_on:MBEDTLS_SELF_TEST */
Daniel Kingadc32c02016-05-16 18:25:45 -0300131void poly1305_selftest()
132{
Manuel Pégourié-Gonnardfce88b22018-05-09 13:06:12 +0200133 TEST_ASSERT( mbedtls_poly1305_self_test( 1 ) == 0 );
Daniel Kingadc32c02016-05-16 18:25:45 -0300134}
135/* END_CASE */