takashi kadono / Mbed OS Nucleo_446

Important changes to repositories hosted on mbed.com

Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.

To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.

It is also possible to export all your personal repositories from the account settings page.

Dependencies:   ssd1331

mbed-os/features/mbedtls/src/chachapoly.c@0:8fdf9a60065b, 2018-10-10 (annotated)

Committer:
kadonotakashi
Date:
Wed Oct 10 00:33:53 2018 +0000
Revision:
0:8fdf9a60065b
how to make mbed librry

Who changed what in which revision?

UserRevisionLine numberNew contents of line
kadonotakashi 0:8fdf9a60065b 1 /**
kadonotakashi 0:8fdf9a60065b 2 * \file chachapoly.c
kadonotakashi 0:8fdf9a60065b 3 *
kadonotakashi 0:8fdf9a60065b 4 * \brief ChaCha20-Poly1305 AEAD construction based on RFC 7539.
kadonotakashi 0:8fdf9a60065b 5 *
kadonotakashi 0:8fdf9a60065b 6 * Copyright (C) 2006-2016, ARM Limited, All Rights Reserved
kadonotakashi 0:8fdf9a60065b 7 * SPDX-License-Identifier: Apache-2.0
kadonotakashi 0:8fdf9a60065b 8 *
kadonotakashi 0:8fdf9a60065b 9 * Licensed under the Apache License, Version 2.0 (the "License"); you may
kadonotakashi 0:8fdf9a60065b 10 * not use this file except in compliance with the License.
kadonotakashi 0:8fdf9a60065b 11 * You may obtain a copy of the License at
kadonotakashi 0:8fdf9a60065b 12 *
kadonotakashi 0:8fdf9a60065b 13 * http://www.apache.org/licenses/LICENSE-2.0
kadonotakashi 0:8fdf9a60065b 14 *
kadonotakashi 0:8fdf9a60065b 15 * Unless required by applicable law or agreed to in writing, software
kadonotakashi 0:8fdf9a60065b 16 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
kadonotakashi 0:8fdf9a60065b 17 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
kadonotakashi 0:8fdf9a60065b 18 * See the License for the specific language governing permissions and
kadonotakashi 0:8fdf9a60065b 19 * limitations under the License.
kadonotakashi 0:8fdf9a60065b 20 *
kadonotakashi 0:8fdf9a60065b 21 * This file is part of mbed TLS (https://tls.mbed.org)
kadonotakashi 0:8fdf9a60065b 22 */
kadonotakashi 0:8fdf9a60065b 23 #if !defined(MBEDTLS_CONFIG_FILE)
kadonotakashi 0:8fdf9a60065b 24 #include "mbedtls/config.h"
kadonotakashi 0:8fdf9a60065b 25 #else
kadonotakashi 0:8fdf9a60065b 26 #include MBEDTLS_CONFIG_FILE
kadonotakashi 0:8fdf9a60065b 27 #endif
kadonotakashi 0:8fdf9a60065b 28
kadonotakashi 0:8fdf9a60065b 29 #if defined(MBEDTLS_CHACHAPOLY_C)
kadonotakashi 0:8fdf9a60065b 30
kadonotakashi 0:8fdf9a60065b 31 #include "mbedtls/chachapoly.h"
kadonotakashi 0:8fdf9a60065b 32 #include "mbedtls/platform_util.h"
kadonotakashi 0:8fdf9a60065b 33
kadonotakashi 0:8fdf9a60065b 34 #include <string.h>
kadonotakashi 0:8fdf9a60065b 35
kadonotakashi 0:8fdf9a60065b 36 #if defined(MBEDTLS_SELF_TEST)
kadonotakashi 0:8fdf9a60065b 37 #if defined(MBEDTLS_PLATFORM_C)
kadonotakashi 0:8fdf9a60065b 38 #include "mbedtls/platform.h"
kadonotakashi 0:8fdf9a60065b 39 #else
kadonotakashi 0:8fdf9a60065b 40 #include <stdio.h>
kadonotakashi 0:8fdf9a60065b 41 #define mbedtls_printf printf
kadonotakashi 0:8fdf9a60065b 42 #endif /* MBEDTLS_PLATFORM_C */
kadonotakashi 0:8fdf9a60065b 43 #endif /* MBEDTLS_SELF_TEST */
kadonotakashi 0:8fdf9a60065b 44
kadonotakashi 0:8fdf9a60065b 45 #if !defined(MBEDTLS_CHACHAPOLY_ALT)
kadonotakashi 0:8fdf9a60065b 46
kadonotakashi 0:8fdf9a60065b 47 #define CHACHAPOLY_STATE_INIT ( 0 )
kadonotakashi 0:8fdf9a60065b 48 #define CHACHAPOLY_STATE_AAD ( 1 )
kadonotakashi 0:8fdf9a60065b 49 #define CHACHAPOLY_STATE_CIPHERTEXT ( 2 ) /* Encrypting or decrypting */
kadonotakashi 0:8fdf9a60065b 50 #define CHACHAPOLY_STATE_FINISHED ( 3 )
kadonotakashi 0:8fdf9a60065b 51
kadonotakashi 0:8fdf9a60065b 52 /**
kadonotakashi 0:8fdf9a60065b 53 * \brief Adds nul bytes to pad the AAD for Poly1305.
kadonotakashi 0:8fdf9a60065b 54 *
kadonotakashi 0:8fdf9a60065b 55 * \param ctx The ChaCha20-Poly1305 context.
kadonotakashi 0:8fdf9a60065b 56 */
kadonotakashi 0:8fdf9a60065b 57 static int chachapoly_pad_aad( mbedtls_chachapoly_context *ctx )
kadonotakashi 0:8fdf9a60065b 58 {
kadonotakashi 0:8fdf9a60065b 59 uint32_t partial_block_len = (uint32_t) ( ctx->aad_len % 16U );
kadonotakashi 0:8fdf9a60065b 60 unsigned char zeroes[15];
kadonotakashi 0:8fdf9a60065b 61
kadonotakashi 0:8fdf9a60065b 62 if( partial_block_len == 0U )
kadonotakashi 0:8fdf9a60065b 63 return( 0 );
kadonotakashi 0:8fdf9a60065b 64
kadonotakashi 0:8fdf9a60065b 65 memset( zeroes, 0, sizeof( zeroes ) );
kadonotakashi 0:8fdf9a60065b 66
kadonotakashi 0:8fdf9a60065b 67 return( mbedtls_poly1305_update( &ctx->poly1305_ctx,
kadonotakashi 0:8fdf9a60065b 68 zeroes,
kadonotakashi 0:8fdf9a60065b 69 16U - partial_block_len ) );
kadonotakashi 0:8fdf9a60065b 70 }
kadonotakashi 0:8fdf9a60065b 71
kadonotakashi 0:8fdf9a60065b 72 /**
kadonotakashi 0:8fdf9a60065b 73 * \brief Adds nul bytes to pad the ciphertext for Poly1305.
kadonotakashi 0:8fdf9a60065b 74 *
kadonotakashi 0:8fdf9a60065b 75 * \param ctx The ChaCha20-Poly1305 context.
kadonotakashi 0:8fdf9a60065b 76 */
kadonotakashi 0:8fdf9a60065b 77 static int chachapoly_pad_ciphertext( mbedtls_chachapoly_context *ctx )
kadonotakashi 0:8fdf9a60065b 78 {
kadonotakashi 0:8fdf9a60065b 79 uint32_t partial_block_len = (uint32_t) ( ctx->ciphertext_len % 16U );
kadonotakashi 0:8fdf9a60065b 80 unsigned char zeroes[15];
kadonotakashi 0:8fdf9a60065b 81
kadonotakashi 0:8fdf9a60065b 82 if( partial_block_len == 0U )
kadonotakashi 0:8fdf9a60065b 83 return( 0 );
kadonotakashi 0:8fdf9a60065b 84
kadonotakashi 0:8fdf9a60065b 85 memset( zeroes, 0, sizeof( zeroes ) );
kadonotakashi 0:8fdf9a60065b 86 return( mbedtls_poly1305_update( &ctx->poly1305_ctx,
kadonotakashi 0:8fdf9a60065b 87 zeroes,
kadonotakashi 0:8fdf9a60065b 88 16U - partial_block_len ) );
kadonotakashi 0:8fdf9a60065b 89 }
kadonotakashi 0:8fdf9a60065b 90
kadonotakashi 0:8fdf9a60065b 91 void mbedtls_chachapoly_init( mbedtls_chachapoly_context *ctx )
kadonotakashi 0:8fdf9a60065b 92 {
kadonotakashi 0:8fdf9a60065b 93 if( ctx != NULL )
kadonotakashi 0:8fdf9a60065b 94 {
kadonotakashi 0:8fdf9a60065b 95 mbedtls_chacha20_init( &ctx->chacha20_ctx );
kadonotakashi 0:8fdf9a60065b 96 mbedtls_poly1305_init( &ctx->poly1305_ctx );
kadonotakashi 0:8fdf9a60065b 97 ctx->aad_len = 0U;
kadonotakashi 0:8fdf9a60065b 98 ctx->ciphertext_len = 0U;
kadonotakashi 0:8fdf9a60065b 99 ctx->state = CHACHAPOLY_STATE_INIT;
kadonotakashi 0:8fdf9a60065b 100 ctx->mode = MBEDTLS_CHACHAPOLY_ENCRYPT;
kadonotakashi 0:8fdf9a60065b 101 }
kadonotakashi 0:8fdf9a60065b 102 }
kadonotakashi 0:8fdf9a60065b 103
kadonotakashi 0:8fdf9a60065b 104 void mbedtls_chachapoly_free( mbedtls_chachapoly_context *ctx )
kadonotakashi 0:8fdf9a60065b 105 {
kadonotakashi 0:8fdf9a60065b 106 if( ctx != NULL )
kadonotakashi 0:8fdf9a60065b 107 {
kadonotakashi 0:8fdf9a60065b 108 mbedtls_chacha20_free( &ctx->chacha20_ctx );
kadonotakashi 0:8fdf9a60065b 109 mbedtls_poly1305_free( &ctx->poly1305_ctx );
kadonotakashi 0:8fdf9a60065b 110 ctx->aad_len = 0U;
kadonotakashi 0:8fdf9a60065b 111 ctx->ciphertext_len = 0U;
kadonotakashi 0:8fdf9a60065b 112 ctx->state = CHACHAPOLY_STATE_INIT;
kadonotakashi 0:8fdf9a60065b 113 ctx->mode = MBEDTLS_CHACHAPOLY_ENCRYPT;
kadonotakashi 0:8fdf9a60065b 114 }
kadonotakashi 0:8fdf9a60065b 115 }
kadonotakashi 0:8fdf9a60065b 116
kadonotakashi 0:8fdf9a60065b 117 int mbedtls_chachapoly_setkey( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 118 const unsigned char key[32] )
kadonotakashi 0:8fdf9a60065b 119 {
kadonotakashi 0:8fdf9a60065b 120 int ret;
kadonotakashi 0:8fdf9a60065b 121
kadonotakashi 0:8fdf9a60065b 122 if( ( ctx == NULL ) || ( key == NULL ) )
kadonotakashi 0:8fdf9a60065b 123 {
kadonotakashi 0:8fdf9a60065b 124 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 125 }
kadonotakashi 0:8fdf9a60065b 126
kadonotakashi 0:8fdf9a60065b 127 ret = mbedtls_chacha20_setkey( &ctx->chacha20_ctx, key );
kadonotakashi 0:8fdf9a60065b 128
kadonotakashi 0:8fdf9a60065b 129 return( ret );
kadonotakashi 0:8fdf9a60065b 130 }
kadonotakashi 0:8fdf9a60065b 131
kadonotakashi 0:8fdf9a60065b 132 int mbedtls_chachapoly_starts( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 133 const unsigned char nonce[12],
kadonotakashi 0:8fdf9a60065b 134 mbedtls_chachapoly_mode_t mode )
kadonotakashi 0:8fdf9a60065b 135 {
kadonotakashi 0:8fdf9a60065b 136 int ret;
kadonotakashi 0:8fdf9a60065b 137 unsigned char poly1305_key[64];
kadonotakashi 0:8fdf9a60065b 138
kadonotakashi 0:8fdf9a60065b 139 if( ( ctx == NULL ) || ( nonce == NULL ) )
kadonotakashi 0:8fdf9a60065b 140 {
kadonotakashi 0:8fdf9a60065b 141 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 142 }
kadonotakashi 0:8fdf9a60065b 143
kadonotakashi 0:8fdf9a60065b 144 /* Set counter = 0, will be update to 1 when generating Poly1305 key */
kadonotakashi 0:8fdf9a60065b 145 ret = mbedtls_chacha20_starts( &ctx->chacha20_ctx, nonce, 0U );
kadonotakashi 0:8fdf9a60065b 146 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 147 goto cleanup;
kadonotakashi 0:8fdf9a60065b 148
kadonotakashi 0:8fdf9a60065b 149 /* Generate the Poly1305 key by getting the ChaCha20 keystream output with
kadonotakashi 0:8fdf9a60065b 150 * counter = 0. This is the same as encrypting a buffer of zeroes.
kadonotakashi 0:8fdf9a60065b 151 * Only the first 256-bits (32 bytes) of the key is used for Poly1305.
kadonotakashi 0:8fdf9a60065b 152 * The other 256 bits are discarded.
kadonotakashi 0:8fdf9a60065b 153 */
kadonotakashi 0:8fdf9a60065b 154 memset( poly1305_key, 0, sizeof( poly1305_key ) );
kadonotakashi 0:8fdf9a60065b 155 ret = mbedtls_chacha20_update( &ctx->chacha20_ctx, sizeof( poly1305_key ),
kadonotakashi 0:8fdf9a60065b 156 poly1305_key, poly1305_key );
kadonotakashi 0:8fdf9a60065b 157 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 158 goto cleanup;
kadonotakashi 0:8fdf9a60065b 159
kadonotakashi 0:8fdf9a60065b 160 ret = mbedtls_poly1305_starts( &ctx->poly1305_ctx, poly1305_key );
kadonotakashi 0:8fdf9a60065b 161
kadonotakashi 0:8fdf9a60065b 162 if( ret == 0 )
kadonotakashi 0:8fdf9a60065b 163 {
kadonotakashi 0:8fdf9a60065b 164 ctx->aad_len = 0U;
kadonotakashi 0:8fdf9a60065b 165 ctx->ciphertext_len = 0U;
kadonotakashi 0:8fdf9a60065b 166 ctx->state = CHACHAPOLY_STATE_AAD;
kadonotakashi 0:8fdf9a60065b 167 ctx->mode = mode;
kadonotakashi 0:8fdf9a60065b 168 }
kadonotakashi 0:8fdf9a60065b 169
kadonotakashi 0:8fdf9a60065b 170 cleanup:
kadonotakashi 0:8fdf9a60065b 171 mbedtls_platform_zeroize( poly1305_key, 64U );
kadonotakashi 0:8fdf9a60065b 172 return( ret );
kadonotakashi 0:8fdf9a60065b 173 }
kadonotakashi 0:8fdf9a60065b 174
kadonotakashi 0:8fdf9a60065b 175 int mbedtls_chachapoly_update_aad( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 176 const unsigned char *aad,
kadonotakashi 0:8fdf9a60065b 177 size_t aad_len )
kadonotakashi 0:8fdf9a60065b 178 {
kadonotakashi 0:8fdf9a60065b 179 if( ctx == NULL )
kadonotakashi 0:8fdf9a60065b 180 {
kadonotakashi 0:8fdf9a60065b 181 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 182 }
kadonotakashi 0:8fdf9a60065b 183 else if( ( aad_len > 0U ) && ( aad == NULL ) )
kadonotakashi 0:8fdf9a60065b 184 {
kadonotakashi 0:8fdf9a60065b 185 /* aad pointer is allowed to be NULL if aad_len == 0 */
kadonotakashi 0:8fdf9a60065b 186 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 187 }
kadonotakashi 0:8fdf9a60065b 188 else if( ctx->state != CHACHAPOLY_STATE_AAD )
kadonotakashi 0:8fdf9a60065b 189 {
kadonotakashi 0:8fdf9a60065b 190 return( MBEDTLS_ERR_CHACHAPOLY_BAD_STATE );
kadonotakashi 0:8fdf9a60065b 191 }
kadonotakashi 0:8fdf9a60065b 192
kadonotakashi 0:8fdf9a60065b 193 ctx->aad_len += aad_len;
kadonotakashi 0:8fdf9a60065b 194
kadonotakashi 0:8fdf9a60065b 195 return( mbedtls_poly1305_update( &ctx->poly1305_ctx, aad, aad_len ) );
kadonotakashi 0:8fdf9a60065b 196 }
kadonotakashi 0:8fdf9a60065b 197
kadonotakashi 0:8fdf9a60065b 198 int mbedtls_chachapoly_update( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 199 size_t len,
kadonotakashi 0:8fdf9a60065b 200 const unsigned char *input,
kadonotakashi 0:8fdf9a60065b 201 unsigned char *output )
kadonotakashi 0:8fdf9a60065b 202 {
kadonotakashi 0:8fdf9a60065b 203 int ret;
kadonotakashi 0:8fdf9a60065b 204
kadonotakashi 0:8fdf9a60065b 205 if( ctx == NULL )
kadonotakashi 0:8fdf9a60065b 206 {
kadonotakashi 0:8fdf9a60065b 207 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 208 }
kadonotakashi 0:8fdf9a60065b 209 else if( ( len > 0U ) && ( ( input == NULL ) || ( output == NULL ) ) )
kadonotakashi 0:8fdf9a60065b 210 {
kadonotakashi 0:8fdf9a60065b 211 /* input and output pointers are allowed to be NULL if len == 0 */
kadonotakashi 0:8fdf9a60065b 212 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 213 }
kadonotakashi 0:8fdf9a60065b 214 else if( ( ctx->state != CHACHAPOLY_STATE_AAD ) &&
kadonotakashi 0:8fdf9a60065b 215 ( ctx->state != CHACHAPOLY_STATE_CIPHERTEXT ) )
kadonotakashi 0:8fdf9a60065b 216 {
kadonotakashi 0:8fdf9a60065b 217 return( MBEDTLS_ERR_CHACHAPOLY_BAD_STATE );
kadonotakashi 0:8fdf9a60065b 218 }
kadonotakashi 0:8fdf9a60065b 219
kadonotakashi 0:8fdf9a60065b 220 if( ctx->state == CHACHAPOLY_STATE_AAD )
kadonotakashi 0:8fdf9a60065b 221 {
kadonotakashi 0:8fdf9a60065b 222 ctx->state = CHACHAPOLY_STATE_CIPHERTEXT;
kadonotakashi 0:8fdf9a60065b 223
kadonotakashi 0:8fdf9a60065b 224 ret = chachapoly_pad_aad( ctx );
kadonotakashi 0:8fdf9a60065b 225 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 226 return( ret );
kadonotakashi 0:8fdf9a60065b 227 }
kadonotakashi 0:8fdf9a60065b 228
kadonotakashi 0:8fdf9a60065b 229 ctx->ciphertext_len += len;
kadonotakashi 0:8fdf9a60065b 230
kadonotakashi 0:8fdf9a60065b 231 if( ctx->mode == MBEDTLS_CHACHAPOLY_ENCRYPT )
kadonotakashi 0:8fdf9a60065b 232 {
kadonotakashi 0:8fdf9a60065b 233 ret = mbedtls_chacha20_update( &ctx->chacha20_ctx, len, input, output );
kadonotakashi 0:8fdf9a60065b 234 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 235 return( ret );
kadonotakashi 0:8fdf9a60065b 236
kadonotakashi 0:8fdf9a60065b 237 ret = mbedtls_poly1305_update( &ctx->poly1305_ctx, output, len );
kadonotakashi 0:8fdf9a60065b 238 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 239 return( ret );
kadonotakashi 0:8fdf9a60065b 240 }
kadonotakashi 0:8fdf9a60065b 241 else /* DECRYPT */
kadonotakashi 0:8fdf9a60065b 242 {
kadonotakashi 0:8fdf9a60065b 243 ret = mbedtls_poly1305_update( &ctx->poly1305_ctx, input, len );
kadonotakashi 0:8fdf9a60065b 244 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 245 return( ret );
kadonotakashi 0:8fdf9a60065b 246
kadonotakashi 0:8fdf9a60065b 247 ret = mbedtls_chacha20_update( &ctx->chacha20_ctx, len, input, output );
kadonotakashi 0:8fdf9a60065b 248 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 249 return( ret );
kadonotakashi 0:8fdf9a60065b 250 }
kadonotakashi 0:8fdf9a60065b 251
kadonotakashi 0:8fdf9a60065b 252 return( 0 );
kadonotakashi 0:8fdf9a60065b 253 }
kadonotakashi 0:8fdf9a60065b 254
kadonotakashi 0:8fdf9a60065b 255 int mbedtls_chachapoly_finish( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 256 unsigned char mac[16] )
kadonotakashi 0:8fdf9a60065b 257 {
kadonotakashi 0:8fdf9a60065b 258 int ret;
kadonotakashi 0:8fdf9a60065b 259 unsigned char len_block[16];
kadonotakashi 0:8fdf9a60065b 260
kadonotakashi 0:8fdf9a60065b 261 if( ( ctx == NULL ) || ( mac == NULL ) )
kadonotakashi 0:8fdf9a60065b 262 {
kadonotakashi 0:8fdf9a60065b 263 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 264 }
kadonotakashi 0:8fdf9a60065b 265 else if( ctx->state == CHACHAPOLY_STATE_INIT )
kadonotakashi 0:8fdf9a60065b 266 {
kadonotakashi 0:8fdf9a60065b 267 return( MBEDTLS_ERR_CHACHAPOLY_BAD_STATE );
kadonotakashi 0:8fdf9a60065b 268 }
kadonotakashi 0:8fdf9a60065b 269
kadonotakashi 0:8fdf9a60065b 270 if( ctx->state == CHACHAPOLY_STATE_AAD )
kadonotakashi 0:8fdf9a60065b 271 {
kadonotakashi 0:8fdf9a60065b 272 ret = chachapoly_pad_aad( ctx );
kadonotakashi 0:8fdf9a60065b 273 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 274 return( ret );
kadonotakashi 0:8fdf9a60065b 275 }
kadonotakashi 0:8fdf9a60065b 276 else if( ctx->state == CHACHAPOLY_STATE_CIPHERTEXT )
kadonotakashi 0:8fdf9a60065b 277 {
kadonotakashi 0:8fdf9a60065b 278 ret = chachapoly_pad_ciphertext( ctx );
kadonotakashi 0:8fdf9a60065b 279 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 280 return( ret );
kadonotakashi 0:8fdf9a60065b 281 }
kadonotakashi 0:8fdf9a60065b 282
kadonotakashi 0:8fdf9a60065b 283 ctx->state = CHACHAPOLY_STATE_FINISHED;
kadonotakashi 0:8fdf9a60065b 284
kadonotakashi 0:8fdf9a60065b 285 /* The lengths of the AAD and ciphertext are processed by
kadonotakashi 0:8fdf9a60065b 286 * Poly1305 as the final 128-bit block, encoded as little-endian integers.
kadonotakashi 0:8fdf9a60065b 287 */
kadonotakashi 0:8fdf9a60065b 288 len_block[ 0] = (unsigned char)( ctx->aad_len );
kadonotakashi 0:8fdf9a60065b 289 len_block[ 1] = (unsigned char)( ctx->aad_len >> 8 );
kadonotakashi 0:8fdf9a60065b 290 len_block[ 2] = (unsigned char)( ctx->aad_len >> 16 );
kadonotakashi 0:8fdf9a60065b 291 len_block[ 3] = (unsigned char)( ctx->aad_len >> 24 );
kadonotakashi 0:8fdf9a60065b 292 len_block[ 4] = (unsigned char)( ctx->aad_len >> 32 );
kadonotakashi 0:8fdf9a60065b 293 len_block[ 5] = (unsigned char)( ctx->aad_len >> 40 );
kadonotakashi 0:8fdf9a60065b 294 len_block[ 6] = (unsigned char)( ctx->aad_len >> 48 );
kadonotakashi 0:8fdf9a60065b 295 len_block[ 7] = (unsigned char)( ctx->aad_len >> 56 );
kadonotakashi 0:8fdf9a60065b 296 len_block[ 8] = (unsigned char)( ctx->ciphertext_len );
kadonotakashi 0:8fdf9a60065b 297 len_block[ 9] = (unsigned char)( ctx->ciphertext_len >> 8 );
kadonotakashi 0:8fdf9a60065b 298 len_block[10] = (unsigned char)( ctx->ciphertext_len >> 16 );
kadonotakashi 0:8fdf9a60065b 299 len_block[11] = (unsigned char)( ctx->ciphertext_len >> 24 );
kadonotakashi 0:8fdf9a60065b 300 len_block[12] = (unsigned char)( ctx->ciphertext_len >> 32 );
kadonotakashi 0:8fdf9a60065b 301 len_block[13] = (unsigned char)( ctx->ciphertext_len >> 40 );
kadonotakashi 0:8fdf9a60065b 302 len_block[14] = (unsigned char)( ctx->ciphertext_len >> 48 );
kadonotakashi 0:8fdf9a60065b 303 len_block[15] = (unsigned char)( ctx->ciphertext_len >> 56 );
kadonotakashi 0:8fdf9a60065b 304
kadonotakashi 0:8fdf9a60065b 305 ret = mbedtls_poly1305_update( &ctx->poly1305_ctx, len_block, 16U );
kadonotakashi 0:8fdf9a60065b 306 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 307 return( ret );
kadonotakashi 0:8fdf9a60065b 308
kadonotakashi 0:8fdf9a60065b 309 ret = mbedtls_poly1305_finish( &ctx->poly1305_ctx, mac );
kadonotakashi 0:8fdf9a60065b 310
kadonotakashi 0:8fdf9a60065b 311 return( ret );
kadonotakashi 0:8fdf9a60065b 312 }
kadonotakashi 0:8fdf9a60065b 313
kadonotakashi 0:8fdf9a60065b 314 static int chachapoly_crypt_and_tag( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 315 mbedtls_chachapoly_mode_t mode,
kadonotakashi 0:8fdf9a60065b 316 size_t length,
kadonotakashi 0:8fdf9a60065b 317 const unsigned char nonce[12],
kadonotakashi 0:8fdf9a60065b 318 const unsigned char *aad,
kadonotakashi 0:8fdf9a60065b 319 size_t aad_len,
kadonotakashi 0:8fdf9a60065b 320 const unsigned char *input,
kadonotakashi 0:8fdf9a60065b 321 unsigned char *output,
kadonotakashi 0:8fdf9a60065b 322 unsigned char tag[16] )
kadonotakashi 0:8fdf9a60065b 323 {
kadonotakashi 0:8fdf9a60065b 324 int ret;
kadonotakashi 0:8fdf9a60065b 325
kadonotakashi 0:8fdf9a60065b 326 ret = mbedtls_chachapoly_starts( ctx, nonce, mode );
kadonotakashi 0:8fdf9a60065b 327 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 328 goto cleanup;
kadonotakashi 0:8fdf9a60065b 329
kadonotakashi 0:8fdf9a60065b 330 ret = mbedtls_chachapoly_update_aad( ctx, aad, aad_len );
kadonotakashi 0:8fdf9a60065b 331 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 332 goto cleanup;
kadonotakashi 0:8fdf9a60065b 333
kadonotakashi 0:8fdf9a60065b 334 ret = mbedtls_chachapoly_update( ctx, length, input, output );
kadonotakashi 0:8fdf9a60065b 335 if( ret != 0 )
kadonotakashi 0:8fdf9a60065b 336 goto cleanup;
kadonotakashi 0:8fdf9a60065b 337
kadonotakashi 0:8fdf9a60065b 338 ret = mbedtls_chachapoly_finish( ctx, tag );
kadonotakashi 0:8fdf9a60065b 339
kadonotakashi 0:8fdf9a60065b 340 cleanup:
kadonotakashi 0:8fdf9a60065b 341 return( ret );
kadonotakashi 0:8fdf9a60065b 342 }
kadonotakashi 0:8fdf9a60065b 343
kadonotakashi 0:8fdf9a60065b 344 int mbedtls_chachapoly_encrypt_and_tag( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 345 size_t length,
kadonotakashi 0:8fdf9a60065b 346 const unsigned char nonce[12],
kadonotakashi 0:8fdf9a60065b 347 const unsigned char *aad,
kadonotakashi 0:8fdf9a60065b 348 size_t aad_len,
kadonotakashi 0:8fdf9a60065b 349 const unsigned char *input,
kadonotakashi 0:8fdf9a60065b 350 unsigned char *output,
kadonotakashi 0:8fdf9a60065b 351 unsigned char tag[16] )
kadonotakashi 0:8fdf9a60065b 352 {
kadonotakashi 0:8fdf9a60065b 353 return( chachapoly_crypt_and_tag( ctx, MBEDTLS_CHACHAPOLY_ENCRYPT,
kadonotakashi 0:8fdf9a60065b 354 length, nonce, aad, aad_len,
kadonotakashi 0:8fdf9a60065b 355 input, output, tag ) );
kadonotakashi 0:8fdf9a60065b 356 }
kadonotakashi 0:8fdf9a60065b 357
kadonotakashi 0:8fdf9a60065b 358 int mbedtls_chachapoly_auth_decrypt( mbedtls_chachapoly_context *ctx,
kadonotakashi 0:8fdf9a60065b 359 size_t length,
kadonotakashi 0:8fdf9a60065b 360 const unsigned char nonce[12],
kadonotakashi 0:8fdf9a60065b 361 const unsigned char *aad,
kadonotakashi 0:8fdf9a60065b 362 size_t aad_len,
kadonotakashi 0:8fdf9a60065b 363 const unsigned char tag[16],
kadonotakashi 0:8fdf9a60065b 364 const unsigned char *input,
kadonotakashi 0:8fdf9a60065b 365 unsigned char *output )
kadonotakashi 0:8fdf9a60065b 366 {
kadonotakashi 0:8fdf9a60065b 367 int ret;
kadonotakashi 0:8fdf9a60065b 368 unsigned char check_tag[16];
kadonotakashi 0:8fdf9a60065b 369 size_t i;
kadonotakashi 0:8fdf9a60065b 370 int diff;
kadonotakashi 0:8fdf9a60065b 371
kadonotakashi 0:8fdf9a60065b 372 if( tag == NULL )
kadonotakashi 0:8fdf9a60065b 373 return( MBEDTLS_ERR_POLY1305_BAD_INPUT_DATA );
kadonotakashi 0:8fdf9a60065b 374
kadonotakashi 0:8fdf9a60065b 375 if( ( ret = chachapoly_crypt_and_tag( ctx,
kadonotakashi 0:8fdf9a60065b 376 MBEDTLS_CHACHAPOLY_DECRYPT, length, nonce,
kadonotakashi 0:8fdf9a60065b 377 aad, aad_len, input, output, check_tag ) ) != 0 )
kadonotakashi 0:8fdf9a60065b 378 {
kadonotakashi 0:8fdf9a60065b 379 return( ret );
kadonotakashi 0:8fdf9a60065b 380 }
kadonotakashi 0:8fdf9a60065b 381
kadonotakashi 0:8fdf9a60065b 382 /* Check tag in "constant-time" */
kadonotakashi 0:8fdf9a60065b 383 for( diff = 0, i = 0; i < sizeof( check_tag ); i++ )
kadonotakashi 0:8fdf9a60065b 384 diff |= tag[i] ^ check_tag[i];
kadonotakashi 0:8fdf9a60065b 385
kadonotakashi 0:8fdf9a60065b 386 if( diff != 0 )
kadonotakashi 0:8fdf9a60065b 387 {
kadonotakashi 0:8fdf9a60065b 388 mbedtls_platform_zeroize( output, length );
kadonotakashi 0:8fdf9a60065b 389 return( MBEDTLS_ERR_CHACHAPOLY_AUTH_FAILED );
kadonotakashi 0:8fdf9a60065b 390 }
kadonotakashi 0:8fdf9a60065b 391
kadonotakashi 0:8fdf9a60065b 392 return( 0 );
kadonotakashi 0:8fdf9a60065b 393 }
kadonotakashi 0:8fdf9a60065b 394
kadonotakashi 0:8fdf9a60065b 395 #endif /* MBEDTLS_CHACHAPOLY_ALT */
kadonotakashi 0:8fdf9a60065b 396
kadonotakashi 0:8fdf9a60065b 397 #if defined(MBEDTLS_SELF_TEST)
kadonotakashi 0:8fdf9a60065b 398
kadonotakashi 0:8fdf9a60065b 399 static const unsigned char test_key[1][32] =
kadonotakashi 0:8fdf9a60065b 400 {
kadonotakashi 0:8fdf9a60065b 401 {
kadonotakashi 0:8fdf9a60065b 402 0x80, 0x81, 0x82, 0x83, 0x84, 0x85, 0x86, 0x87,
kadonotakashi 0:8fdf9a60065b 403 0x88, 0x89, 0x8a, 0x8b, 0x8c, 0x8d, 0x8e, 0x8f,
kadonotakashi 0:8fdf9a60065b 404 0x90, 0x91, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97,
kadonotakashi 0:8fdf9a60065b 405 0x98, 0x99, 0x9a, 0x9b, 0x9c, 0x9d, 0x9e, 0x9f
kadonotakashi 0:8fdf9a60065b 406 }
kadonotakashi 0:8fdf9a60065b 407 };
kadonotakashi 0:8fdf9a60065b 408
kadonotakashi 0:8fdf9a60065b 409 static const unsigned char test_nonce[1][12] =
kadonotakashi 0:8fdf9a60065b 410 {
kadonotakashi 0:8fdf9a60065b 411 {
kadonotakashi 0:8fdf9a60065b 412 0x07, 0x00, 0x00, 0x00, /* 32-bit common part */
kadonotakashi 0:8fdf9a60065b 413 0x40, 0x41, 0x42, 0x43, 0x44, 0x45, 0x46, 0x47 /* 64-bit IV */
kadonotakashi 0:8fdf9a60065b 414 }
kadonotakashi 0:8fdf9a60065b 415 };
kadonotakashi 0:8fdf9a60065b 416
kadonotakashi 0:8fdf9a60065b 417 static const unsigned char test_aad[1][12] =
kadonotakashi 0:8fdf9a60065b 418 {
kadonotakashi 0:8fdf9a60065b 419 {
kadonotakashi 0:8fdf9a60065b 420 0x50, 0x51, 0x52, 0x53, 0xc0, 0xc1, 0xc2, 0xc3,
kadonotakashi 0:8fdf9a60065b 421 0xc4, 0xc5, 0xc6, 0xc7
kadonotakashi 0:8fdf9a60065b 422 }
kadonotakashi 0:8fdf9a60065b 423 };
kadonotakashi 0:8fdf9a60065b 424
kadonotakashi 0:8fdf9a60065b 425 static const size_t test_aad_len[1] =
kadonotakashi 0:8fdf9a60065b 426 {
kadonotakashi 0:8fdf9a60065b 427 12U
kadonotakashi 0:8fdf9a60065b 428 };
kadonotakashi 0:8fdf9a60065b 429
kadonotakashi 0:8fdf9a60065b 430 static const unsigned char test_input[1][114] =
kadonotakashi 0:8fdf9a60065b 431 {
kadonotakashi 0:8fdf9a60065b 432 {
kadonotakashi 0:8fdf9a60065b 433 0x4c, 0x61, 0x64, 0x69, 0x65, 0x73, 0x20, 0x61,
kadonotakashi 0:8fdf9a60065b 434 0x6e, 0x64, 0x20, 0x47, 0x65, 0x6e, 0x74, 0x6c,
kadonotakashi 0:8fdf9a60065b 435 0x65, 0x6d, 0x65, 0x6e, 0x20, 0x6f, 0x66, 0x20,
kadonotakashi 0:8fdf9a60065b 436 0x74, 0x68, 0x65, 0x20, 0x63, 0x6c, 0x61, 0x73,
kadonotakashi 0:8fdf9a60065b 437 0x73, 0x20, 0x6f, 0x66, 0x20, 0x27, 0x39, 0x39,
kadonotakashi 0:8fdf9a60065b 438 0x3a, 0x20, 0x49, 0x66, 0x20, 0x49, 0x20, 0x63,
kadonotakashi 0:8fdf9a60065b 439 0x6f, 0x75, 0x6c, 0x64, 0x20, 0x6f, 0x66, 0x66,
kadonotakashi 0:8fdf9a60065b 440 0x65, 0x72, 0x20, 0x79, 0x6f, 0x75, 0x20, 0x6f,
kadonotakashi 0:8fdf9a60065b 441 0x6e, 0x6c, 0x79, 0x20, 0x6f, 0x6e, 0x65, 0x20,
kadonotakashi 0:8fdf9a60065b 442 0x74, 0x69, 0x70, 0x20, 0x66, 0x6f, 0x72, 0x20,
kadonotakashi 0:8fdf9a60065b 443 0x74, 0x68, 0x65, 0x20, 0x66, 0x75, 0x74, 0x75,
kadonotakashi 0:8fdf9a60065b 444 0x72, 0x65, 0x2c, 0x20, 0x73, 0x75, 0x6e, 0x73,
kadonotakashi 0:8fdf9a60065b 445 0x63, 0x72, 0x65, 0x65, 0x6e, 0x20, 0x77, 0x6f,
kadonotakashi 0:8fdf9a60065b 446 0x75, 0x6c, 0x64, 0x20, 0x62, 0x65, 0x20, 0x69,
kadonotakashi 0:8fdf9a60065b 447 0x74, 0x2e
kadonotakashi 0:8fdf9a60065b 448 }
kadonotakashi 0:8fdf9a60065b 449 };
kadonotakashi 0:8fdf9a60065b 450
kadonotakashi 0:8fdf9a60065b 451 static const unsigned char test_output[1][114] =
kadonotakashi 0:8fdf9a60065b 452 {
kadonotakashi 0:8fdf9a60065b 453 {
kadonotakashi 0:8fdf9a60065b 454 0xd3, 0x1a, 0x8d, 0x34, 0x64, 0x8e, 0x60, 0xdb,
kadonotakashi 0:8fdf9a60065b 455 0x7b, 0x86, 0xaf, 0xbc, 0x53, 0xef, 0x7e, 0xc2,
kadonotakashi 0:8fdf9a60065b 456 0xa4, 0xad, 0xed, 0x51, 0x29, 0x6e, 0x08, 0xfe,
kadonotakashi 0:8fdf9a60065b 457 0xa9, 0xe2, 0xb5, 0xa7, 0x36, 0xee, 0x62, 0xd6,
kadonotakashi 0:8fdf9a60065b 458 0x3d, 0xbe, 0xa4, 0x5e, 0x8c, 0xa9, 0x67, 0x12,
kadonotakashi 0:8fdf9a60065b 459 0x82, 0xfa, 0xfb, 0x69, 0xda, 0x92, 0x72, 0x8b,
kadonotakashi 0:8fdf9a60065b 460 0x1a, 0x71, 0xde, 0x0a, 0x9e, 0x06, 0x0b, 0x29,
kadonotakashi 0:8fdf9a60065b 461 0x05, 0xd6, 0xa5, 0xb6, 0x7e, 0xcd, 0x3b, 0x36,
kadonotakashi 0:8fdf9a60065b 462 0x92, 0xdd, 0xbd, 0x7f, 0x2d, 0x77, 0x8b, 0x8c,
kadonotakashi 0:8fdf9a60065b 463 0x98, 0x03, 0xae, 0xe3, 0x28, 0x09, 0x1b, 0x58,
kadonotakashi 0:8fdf9a60065b 464 0xfa, 0xb3, 0x24, 0xe4, 0xfa, 0xd6, 0x75, 0x94,
kadonotakashi 0:8fdf9a60065b 465 0x55, 0x85, 0x80, 0x8b, 0x48, 0x31, 0xd7, 0xbc,
kadonotakashi 0:8fdf9a60065b 466 0x3f, 0xf4, 0xde, 0xf0, 0x8e, 0x4b, 0x7a, 0x9d,
kadonotakashi 0:8fdf9a60065b 467 0xe5, 0x76, 0xd2, 0x65, 0x86, 0xce, 0xc6, 0x4b,
kadonotakashi 0:8fdf9a60065b 468 0x61, 0x16
kadonotakashi 0:8fdf9a60065b 469 }
kadonotakashi 0:8fdf9a60065b 470 };
kadonotakashi 0:8fdf9a60065b 471
kadonotakashi 0:8fdf9a60065b 472 static const size_t test_input_len[1] =
kadonotakashi 0:8fdf9a60065b 473 {
kadonotakashi 0:8fdf9a60065b 474 114U
kadonotakashi 0:8fdf9a60065b 475 };
kadonotakashi 0:8fdf9a60065b 476
kadonotakashi 0:8fdf9a60065b 477 static const unsigned char test_mac[1][16] =
kadonotakashi 0:8fdf9a60065b 478 {
kadonotakashi 0:8fdf9a60065b 479 {
kadonotakashi 0:8fdf9a60065b 480 0x1a, 0xe1, 0x0b, 0x59, 0x4f, 0x09, 0xe2, 0x6a,
kadonotakashi 0:8fdf9a60065b 481 0x7e, 0x90, 0x2e, 0xcb, 0xd0, 0x60, 0x06, 0x91
kadonotakashi 0:8fdf9a60065b 482 }
kadonotakashi 0:8fdf9a60065b 483 };
kadonotakashi 0:8fdf9a60065b 484
kadonotakashi 0:8fdf9a60065b 485 #define ASSERT( cond, args ) \
kadonotakashi 0:8fdf9a60065b 486 do \
kadonotakashi 0:8fdf9a60065b 487 { \
kadonotakashi 0:8fdf9a60065b 488 if( ! ( cond ) ) \
kadonotakashi 0:8fdf9a60065b 489 { \
kadonotakashi 0:8fdf9a60065b 490 if( verbose != 0 ) \
kadonotakashi 0:8fdf9a60065b 491 mbedtls_printf args; \
kadonotakashi 0:8fdf9a60065b 492 \
kadonotakashi 0:8fdf9a60065b 493 return( -1 ); \
kadonotakashi 0:8fdf9a60065b 494 } \
kadonotakashi 0:8fdf9a60065b 495 } \
kadonotakashi 0:8fdf9a60065b 496 while( 0 )
kadonotakashi 0:8fdf9a60065b 497
kadonotakashi 0:8fdf9a60065b 498 int mbedtls_chachapoly_self_test( int verbose )
kadonotakashi 0:8fdf9a60065b 499 {
kadonotakashi 0:8fdf9a60065b 500 mbedtls_chachapoly_context ctx;
kadonotakashi 0:8fdf9a60065b 501 unsigned i;
kadonotakashi 0:8fdf9a60065b 502 int ret;
kadonotakashi 0:8fdf9a60065b 503 unsigned char output[200];
kadonotakashi 0:8fdf9a60065b 504 unsigned char mac[16];
kadonotakashi 0:8fdf9a60065b 505
kadonotakashi 0:8fdf9a60065b 506 for( i = 0U; i < 1U; i++ )
kadonotakashi 0:8fdf9a60065b 507 {
kadonotakashi 0:8fdf9a60065b 508 if( verbose != 0 )
kadonotakashi 0:8fdf9a60065b 509 mbedtls_printf( " ChaCha20-Poly1305 test %u ", i );
kadonotakashi 0:8fdf9a60065b 510
kadonotakashi 0:8fdf9a60065b 511 mbedtls_chachapoly_init( &ctx );
kadonotakashi 0:8fdf9a60065b 512
kadonotakashi 0:8fdf9a60065b 513 ret = mbedtls_chachapoly_setkey( &ctx, test_key[i] );
kadonotakashi 0:8fdf9a60065b 514 ASSERT( 0 == ret, ( "setkey() error code: %i\n", ret ) );
kadonotakashi 0:8fdf9a60065b 515
kadonotakashi 0:8fdf9a60065b 516 ret = mbedtls_chachapoly_encrypt_and_tag( &ctx,
kadonotakashi 0:8fdf9a60065b 517 test_input_len[i],
kadonotakashi 0:8fdf9a60065b 518 test_nonce[i],
kadonotakashi 0:8fdf9a60065b 519 test_aad[i],
kadonotakashi 0:8fdf9a60065b 520 test_aad_len[i],
kadonotakashi 0:8fdf9a60065b 521 test_input[i],
kadonotakashi 0:8fdf9a60065b 522 output,
kadonotakashi 0:8fdf9a60065b 523 mac );
kadonotakashi 0:8fdf9a60065b 524
kadonotakashi 0:8fdf9a60065b 525 ASSERT( 0 == ret, ( "crypt_and_tag() error code: %i\n", ret ) );
kadonotakashi 0:8fdf9a60065b 526
kadonotakashi 0:8fdf9a60065b 527 ASSERT( 0 == memcmp( output, test_output[i], test_input_len[i] ),
kadonotakashi 0:8fdf9a60065b 528 ( "failure (wrong output)\n" ) );
kadonotakashi 0:8fdf9a60065b 529
kadonotakashi 0:8fdf9a60065b 530 ASSERT( 0 == memcmp( mac, test_mac[i], 16U ),
kadonotakashi 0:8fdf9a60065b 531 ( "failure (wrong MAC)\n" ) );
kadonotakashi 0:8fdf9a60065b 532
kadonotakashi 0:8fdf9a60065b 533 mbedtls_chachapoly_free( &ctx );
kadonotakashi 0:8fdf9a60065b 534
kadonotakashi 0:8fdf9a60065b 535 if( verbose != 0 )
kadonotakashi 0:8fdf9a60065b 536 mbedtls_printf( "passed\n" );
kadonotakashi 0:8fdf9a60065b 537 }
kadonotakashi 0:8fdf9a60065b 538
kadonotakashi 0:8fdf9a60065b 539 if( verbose != 0 )
kadonotakashi 0:8fdf9a60065b 540 mbedtls_printf( "\n" );
kadonotakashi 0:8fdf9a60065b 541
kadonotakashi 0:8fdf9a60065b 542 return( 0 );
kadonotakashi 0:8fdf9a60065b 543 }
kadonotakashi 0:8fdf9a60065b 544
kadonotakashi 0:8fdf9a60065b 545 #endif /* MBEDTLS_SELF_TEST */
kadonotakashi 0:8fdf9a60065b 546
kadonotakashi 0:8fdf9a60065b 547 #endif /* MBEDTLS_CHACHAPOLY_C */