Hannes Tschofenig
/
aes-gcm-test-program
Example program to test AES-GCM functionality. Used for a workshop
SSL/library/pkcs5.c@0:796d0f61a05b, 2018-09-27 (annotated)
- Committer:
- HannesTschofenig
- Date:
- Thu Sep 27 06:34:22 2018 +0000
- Revision:
- 0:796d0f61a05b
Example AES-GCM test program
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
HannesTschofenig | 0:796d0f61a05b | 1 | /** |
HannesTschofenig | 0:796d0f61a05b | 2 | * \file pkcs5.c |
HannesTschofenig | 0:796d0f61a05b | 3 | * |
HannesTschofenig | 0:796d0f61a05b | 4 | * \brief PKCS#5 functions |
HannesTschofenig | 0:796d0f61a05b | 5 | * |
HannesTschofenig | 0:796d0f61a05b | 6 | * \author Mathias Olsson <mathias@kompetensum.com> |
HannesTschofenig | 0:796d0f61a05b | 7 | * |
HannesTschofenig | 0:796d0f61a05b | 8 | * Copyright (C) 2006-2014, Brainspark B.V. |
HannesTschofenig | 0:796d0f61a05b | 9 | * |
HannesTschofenig | 0:796d0f61a05b | 10 | * This file is part of PolarSSL (http://www.polarssl.org) |
HannesTschofenig | 0:796d0f61a05b | 11 | * Lead Maintainer: Paul Bakker <polarssl_maintainer at polarssl.org> |
HannesTschofenig | 0:796d0f61a05b | 12 | * |
HannesTschofenig | 0:796d0f61a05b | 13 | * All rights reserved. |
HannesTschofenig | 0:796d0f61a05b | 14 | * |
HannesTschofenig | 0:796d0f61a05b | 15 | * This program is free software; you can redistribute it and/or modify |
HannesTschofenig | 0:796d0f61a05b | 16 | * it under the terms of the GNU General Public License as published by |
HannesTschofenig | 0:796d0f61a05b | 17 | * the Free Software Foundation; either version 2 of the License, or |
HannesTschofenig | 0:796d0f61a05b | 18 | * (at your option) any later version. |
HannesTschofenig | 0:796d0f61a05b | 19 | * |
HannesTschofenig | 0:796d0f61a05b | 20 | * This program is distributed in the hope that it will be useful, |
HannesTschofenig | 0:796d0f61a05b | 21 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
HannesTschofenig | 0:796d0f61a05b | 22 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
HannesTschofenig | 0:796d0f61a05b | 23 | * GNU General Public License for more details. |
HannesTschofenig | 0:796d0f61a05b | 24 | * |
HannesTschofenig | 0:796d0f61a05b | 25 | * You should have received a copy of the GNU General Public License along |
HannesTschofenig | 0:796d0f61a05b | 26 | * with this program; if not, write to the Free Software Foundation, Inc., |
HannesTschofenig | 0:796d0f61a05b | 27 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. |
HannesTschofenig | 0:796d0f61a05b | 28 | */ |
HannesTschofenig | 0:796d0f61a05b | 29 | /* |
HannesTschofenig | 0:796d0f61a05b | 30 | * PKCS#5 includes PBKDF2 and more |
HannesTschofenig | 0:796d0f61a05b | 31 | * |
HannesTschofenig | 0:796d0f61a05b | 32 | * http://tools.ietf.org/html/rfc2898 (Specification) |
HannesTschofenig | 0:796d0f61a05b | 33 | * http://tools.ietf.org/html/rfc6070 (Test vectors) |
HannesTschofenig | 0:796d0f61a05b | 34 | */ |
HannesTschofenig | 0:796d0f61a05b | 35 | |
HannesTschofenig | 0:796d0f61a05b | 36 | #if !defined(POLARSSL_CONFIG_FILE) |
HannesTschofenig | 0:796d0f61a05b | 37 | #include "polarssl/config.h" |
HannesTschofenig | 0:796d0f61a05b | 38 | #else |
HannesTschofenig | 0:796d0f61a05b | 39 | #include POLARSSL_CONFIG_FILE |
HannesTschofenig | 0:796d0f61a05b | 40 | #endif |
HannesTschofenig | 0:796d0f61a05b | 41 | |
HannesTschofenig | 0:796d0f61a05b | 42 | #if defined(POLARSSL_PKCS5_C) |
HannesTschofenig | 0:796d0f61a05b | 43 | |
HannesTschofenig | 0:796d0f61a05b | 44 | #include "polarssl/pkcs5.h" |
HannesTschofenig | 0:796d0f61a05b | 45 | #include "polarssl/asn1.h" |
HannesTschofenig | 0:796d0f61a05b | 46 | #include "polarssl/cipher.h" |
HannesTschofenig | 0:796d0f61a05b | 47 | #include "polarssl/oid.h" |
HannesTschofenig | 0:796d0f61a05b | 48 | |
HannesTschofenig | 0:796d0f61a05b | 49 | #if defined(POLARSSL_PLATFORM_C) |
HannesTschofenig | 0:796d0f61a05b | 50 | #include "polarssl/platform.h" |
HannesTschofenig | 0:796d0f61a05b | 51 | #else |
HannesTschofenig | 0:796d0f61a05b | 52 | #define polarssl_printf printf |
HannesTschofenig | 0:796d0f61a05b | 53 | #endif |
HannesTschofenig | 0:796d0f61a05b | 54 | |
HannesTschofenig | 0:796d0f61a05b | 55 | static int pkcs5_parse_pbkdf2_params( asn1_buf *params, |
HannesTschofenig | 0:796d0f61a05b | 56 | asn1_buf *salt, int *iterations, |
HannesTschofenig | 0:796d0f61a05b | 57 | int *keylen, md_type_t *md_type ) |
HannesTschofenig | 0:796d0f61a05b | 58 | { |
HannesTschofenig | 0:796d0f61a05b | 59 | int ret; |
HannesTschofenig | 0:796d0f61a05b | 60 | asn1_buf prf_alg_oid; |
HannesTschofenig | 0:796d0f61a05b | 61 | unsigned char **p = ¶ms->p; |
HannesTschofenig | 0:796d0f61a05b | 62 | const unsigned char *end = params->p + params->len; |
HannesTschofenig | 0:796d0f61a05b | 63 | |
HannesTschofenig | 0:796d0f61a05b | 64 | if( params->tag != ( ASN1_CONSTRUCTED | ASN1_SEQUENCE ) ) |
HannesTschofenig | 0:796d0f61a05b | 65 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + |
HannesTschofenig | 0:796d0f61a05b | 66 | POLARSSL_ERR_ASN1_UNEXPECTED_TAG ); |
HannesTschofenig | 0:796d0f61a05b | 67 | /* |
HannesTschofenig | 0:796d0f61a05b | 68 | * PBKDF2-params ::= SEQUENCE { |
HannesTschofenig | 0:796d0f61a05b | 69 | * salt OCTET STRING, |
HannesTschofenig | 0:796d0f61a05b | 70 | * iterationCount INTEGER, |
HannesTschofenig | 0:796d0f61a05b | 71 | * keyLength INTEGER OPTIONAL |
HannesTschofenig | 0:796d0f61a05b | 72 | * prf AlgorithmIdentifier DEFAULT algid-hmacWithSHA1 |
HannesTschofenig | 0:796d0f61a05b | 73 | * } |
HannesTschofenig | 0:796d0f61a05b | 74 | * |
HannesTschofenig | 0:796d0f61a05b | 75 | */ |
HannesTschofenig | 0:796d0f61a05b | 76 | if( ( ret = asn1_get_tag( p, end, &salt->len, ASN1_OCTET_STRING ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 77 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + ret ); |
HannesTschofenig | 0:796d0f61a05b | 78 | |
HannesTschofenig | 0:796d0f61a05b | 79 | salt->p = *p; |
HannesTschofenig | 0:796d0f61a05b | 80 | *p += salt->len; |
HannesTschofenig | 0:796d0f61a05b | 81 | |
HannesTschofenig | 0:796d0f61a05b | 82 | if( ( ret = asn1_get_int( p, end, iterations ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 83 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + ret ); |
HannesTschofenig | 0:796d0f61a05b | 84 | |
HannesTschofenig | 0:796d0f61a05b | 85 | if( *p == end ) |
HannesTschofenig | 0:796d0f61a05b | 86 | return( 0 ); |
HannesTschofenig | 0:796d0f61a05b | 87 | |
HannesTschofenig | 0:796d0f61a05b | 88 | if( ( ret = asn1_get_int( p, end, keylen ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 89 | { |
HannesTschofenig | 0:796d0f61a05b | 90 | if( ret != POLARSSL_ERR_ASN1_UNEXPECTED_TAG ) |
HannesTschofenig | 0:796d0f61a05b | 91 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + ret ); |
HannesTschofenig | 0:796d0f61a05b | 92 | } |
HannesTschofenig | 0:796d0f61a05b | 93 | |
HannesTschofenig | 0:796d0f61a05b | 94 | if( *p == end ) |
HannesTschofenig | 0:796d0f61a05b | 95 | return( 0 ); |
HannesTschofenig | 0:796d0f61a05b | 96 | |
HannesTschofenig | 0:796d0f61a05b | 97 | if( ( ret = asn1_get_alg_null( p, end, &prf_alg_oid ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 98 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + ret ); |
HannesTschofenig | 0:796d0f61a05b | 99 | |
HannesTschofenig | 0:796d0f61a05b | 100 | if( !OID_CMP( OID_HMAC_SHA1, &prf_alg_oid ) ) |
HannesTschofenig | 0:796d0f61a05b | 101 | return( POLARSSL_ERR_PKCS5_FEATURE_UNAVAILABLE ); |
HannesTschofenig | 0:796d0f61a05b | 102 | |
HannesTschofenig | 0:796d0f61a05b | 103 | *md_type = POLARSSL_MD_SHA1; |
HannesTschofenig | 0:796d0f61a05b | 104 | |
HannesTschofenig | 0:796d0f61a05b | 105 | if( *p != end ) |
HannesTschofenig | 0:796d0f61a05b | 106 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + |
HannesTschofenig | 0:796d0f61a05b | 107 | POLARSSL_ERR_ASN1_LENGTH_MISMATCH ); |
HannesTschofenig | 0:796d0f61a05b | 108 | |
HannesTschofenig | 0:796d0f61a05b | 109 | return( 0 ); |
HannesTschofenig | 0:796d0f61a05b | 110 | } |
HannesTschofenig | 0:796d0f61a05b | 111 | |
HannesTschofenig | 0:796d0f61a05b | 112 | int pkcs5_pbes2( asn1_buf *pbe_params, int mode, |
HannesTschofenig | 0:796d0f61a05b | 113 | const unsigned char *pwd, size_t pwdlen, |
HannesTschofenig | 0:796d0f61a05b | 114 | const unsigned char *data, size_t datalen, |
HannesTschofenig | 0:796d0f61a05b | 115 | unsigned char *output ) |
HannesTschofenig | 0:796d0f61a05b | 116 | { |
HannesTschofenig | 0:796d0f61a05b | 117 | int ret, iterations = 0, keylen = 0; |
HannesTschofenig | 0:796d0f61a05b | 118 | unsigned char *p, *end; |
HannesTschofenig | 0:796d0f61a05b | 119 | asn1_buf kdf_alg_oid, enc_scheme_oid, kdf_alg_params, enc_scheme_params; |
HannesTschofenig | 0:796d0f61a05b | 120 | asn1_buf salt; |
HannesTschofenig | 0:796d0f61a05b | 121 | md_type_t md_type = POLARSSL_MD_SHA1; |
HannesTschofenig | 0:796d0f61a05b | 122 | unsigned char key[32], iv[32]; |
HannesTschofenig | 0:796d0f61a05b | 123 | size_t olen = 0; |
HannesTschofenig | 0:796d0f61a05b | 124 | const md_info_t *md_info; |
HannesTschofenig | 0:796d0f61a05b | 125 | const cipher_info_t *cipher_info; |
HannesTschofenig | 0:796d0f61a05b | 126 | md_context_t md_ctx; |
HannesTschofenig | 0:796d0f61a05b | 127 | cipher_type_t cipher_alg; |
HannesTschofenig | 0:796d0f61a05b | 128 | cipher_context_t cipher_ctx; |
HannesTschofenig | 0:796d0f61a05b | 129 | |
HannesTschofenig | 0:796d0f61a05b | 130 | p = pbe_params->p; |
HannesTschofenig | 0:796d0f61a05b | 131 | end = p + pbe_params->len; |
HannesTschofenig | 0:796d0f61a05b | 132 | |
HannesTschofenig | 0:796d0f61a05b | 133 | memset( &md_ctx, 0, sizeof(md_context_t) ); |
HannesTschofenig | 0:796d0f61a05b | 134 | memset( &cipher_ctx, 0, sizeof(cipher_context_t) ); |
HannesTschofenig | 0:796d0f61a05b | 135 | |
HannesTschofenig | 0:796d0f61a05b | 136 | /* |
HannesTschofenig | 0:796d0f61a05b | 137 | * PBES2-params ::= SEQUENCE { |
HannesTschofenig | 0:796d0f61a05b | 138 | * keyDerivationFunc AlgorithmIdentifier {{PBES2-KDFs}}, |
HannesTschofenig | 0:796d0f61a05b | 139 | * encryptionScheme AlgorithmIdentifier {{PBES2-Encs}} |
HannesTschofenig | 0:796d0f61a05b | 140 | * } |
HannesTschofenig | 0:796d0f61a05b | 141 | */ |
HannesTschofenig | 0:796d0f61a05b | 142 | if( pbe_params->tag != ( ASN1_CONSTRUCTED | ASN1_SEQUENCE ) ) |
HannesTschofenig | 0:796d0f61a05b | 143 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + |
HannesTschofenig | 0:796d0f61a05b | 144 | POLARSSL_ERR_ASN1_UNEXPECTED_TAG ); |
HannesTschofenig | 0:796d0f61a05b | 145 | |
HannesTschofenig | 0:796d0f61a05b | 146 | if( ( ret = asn1_get_alg( &p, end, &kdf_alg_oid, &kdf_alg_params ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 147 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + ret ); |
HannesTschofenig | 0:796d0f61a05b | 148 | |
HannesTschofenig | 0:796d0f61a05b | 149 | // Only PBKDF2 supported at the moment |
HannesTschofenig | 0:796d0f61a05b | 150 | // |
HannesTschofenig | 0:796d0f61a05b | 151 | if( !OID_CMP( OID_PKCS5_PBKDF2, &kdf_alg_oid ) ) |
HannesTschofenig | 0:796d0f61a05b | 152 | return( POLARSSL_ERR_PKCS5_FEATURE_UNAVAILABLE ); |
HannesTschofenig | 0:796d0f61a05b | 153 | |
HannesTschofenig | 0:796d0f61a05b | 154 | if( ( ret = pkcs5_parse_pbkdf2_params( &kdf_alg_params, |
HannesTschofenig | 0:796d0f61a05b | 155 | &salt, &iterations, &keylen, |
HannesTschofenig | 0:796d0f61a05b | 156 | &md_type ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 157 | { |
HannesTschofenig | 0:796d0f61a05b | 158 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 159 | } |
HannesTschofenig | 0:796d0f61a05b | 160 | |
HannesTschofenig | 0:796d0f61a05b | 161 | md_info = md_info_from_type( md_type ); |
HannesTschofenig | 0:796d0f61a05b | 162 | if( md_info == NULL ) |
HannesTschofenig | 0:796d0f61a05b | 163 | return( POLARSSL_ERR_PKCS5_FEATURE_UNAVAILABLE ); |
HannesTschofenig | 0:796d0f61a05b | 164 | |
HannesTschofenig | 0:796d0f61a05b | 165 | if( ( ret = asn1_get_alg( &p, end, &enc_scheme_oid, |
HannesTschofenig | 0:796d0f61a05b | 166 | &enc_scheme_params ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 167 | { |
HannesTschofenig | 0:796d0f61a05b | 168 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT + ret ); |
HannesTschofenig | 0:796d0f61a05b | 169 | } |
HannesTschofenig | 0:796d0f61a05b | 170 | |
HannesTschofenig | 0:796d0f61a05b | 171 | if ( oid_get_cipher_alg( &enc_scheme_oid, &cipher_alg ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 172 | return( POLARSSL_ERR_PKCS5_FEATURE_UNAVAILABLE ); |
HannesTschofenig | 0:796d0f61a05b | 173 | |
HannesTschofenig | 0:796d0f61a05b | 174 | cipher_info = cipher_info_from_type( cipher_alg ); |
HannesTschofenig | 0:796d0f61a05b | 175 | if( cipher_info == NULL ) |
HannesTschofenig | 0:796d0f61a05b | 176 | return( POLARSSL_ERR_PKCS5_FEATURE_UNAVAILABLE ); |
HannesTschofenig | 0:796d0f61a05b | 177 | |
HannesTschofenig | 0:796d0f61a05b | 178 | keylen = cipher_info->key_length / 8; |
HannesTschofenig | 0:796d0f61a05b | 179 | |
HannesTschofenig | 0:796d0f61a05b | 180 | if( enc_scheme_params.tag != ASN1_OCTET_STRING || |
HannesTschofenig | 0:796d0f61a05b | 181 | enc_scheme_params.len != cipher_info->iv_size ) |
HannesTschofenig | 0:796d0f61a05b | 182 | { |
HannesTschofenig | 0:796d0f61a05b | 183 | return( POLARSSL_ERR_PKCS5_INVALID_FORMAT ); |
HannesTschofenig | 0:796d0f61a05b | 184 | } |
HannesTschofenig | 0:796d0f61a05b | 185 | |
HannesTschofenig | 0:796d0f61a05b | 186 | memcpy( iv, enc_scheme_params.p, enc_scheme_params.len ); |
HannesTschofenig | 0:796d0f61a05b | 187 | |
HannesTschofenig | 0:796d0f61a05b | 188 | if( ( ret = md_init_ctx( &md_ctx, md_info ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 189 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 190 | |
HannesTschofenig | 0:796d0f61a05b | 191 | if ( ( ret = pkcs5_pbkdf2_hmac( &md_ctx, pwd, pwdlen, salt.p, salt.len, |
HannesTschofenig | 0:796d0f61a05b | 192 | iterations, keylen, key ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 193 | { |
HannesTschofenig | 0:796d0f61a05b | 194 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 195 | } |
HannesTschofenig | 0:796d0f61a05b | 196 | |
HannesTschofenig | 0:796d0f61a05b | 197 | if( ( ret = cipher_init_ctx( &cipher_ctx, cipher_info ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 198 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 199 | |
HannesTschofenig | 0:796d0f61a05b | 200 | if( ( ret = cipher_setkey( &cipher_ctx, key, 8 * keylen, mode ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 201 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 202 | |
HannesTschofenig | 0:796d0f61a05b | 203 | if( ( ret = cipher_set_iv( &cipher_ctx, iv, enc_scheme_params.len ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 204 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 205 | |
HannesTschofenig | 0:796d0f61a05b | 206 | if( ( ret = cipher_reset( &cipher_ctx ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 207 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 208 | |
HannesTschofenig | 0:796d0f61a05b | 209 | if( ( ret = cipher_update( &cipher_ctx, data, datalen, |
HannesTschofenig | 0:796d0f61a05b | 210 | output, &olen ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 211 | { |
HannesTschofenig | 0:796d0f61a05b | 212 | goto exit; |
HannesTschofenig | 0:796d0f61a05b | 213 | } |
HannesTschofenig | 0:796d0f61a05b | 214 | |
HannesTschofenig | 0:796d0f61a05b | 215 | if( ( ret = cipher_finish( &cipher_ctx, output + olen, &olen ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 216 | ret = POLARSSL_ERR_PKCS5_PASSWORD_MISMATCH; |
HannesTschofenig | 0:796d0f61a05b | 217 | |
HannesTschofenig | 0:796d0f61a05b | 218 | exit: |
HannesTschofenig | 0:796d0f61a05b | 219 | md_free_ctx( &md_ctx ); |
HannesTschofenig | 0:796d0f61a05b | 220 | cipher_free_ctx( &cipher_ctx ); |
HannesTschofenig | 0:796d0f61a05b | 221 | |
HannesTschofenig | 0:796d0f61a05b | 222 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 223 | } |
HannesTschofenig | 0:796d0f61a05b | 224 | |
HannesTschofenig | 0:796d0f61a05b | 225 | int pkcs5_pbkdf2_hmac( md_context_t *ctx, const unsigned char *password, |
HannesTschofenig | 0:796d0f61a05b | 226 | size_t plen, const unsigned char *salt, size_t slen, |
HannesTschofenig | 0:796d0f61a05b | 227 | unsigned int iteration_count, |
HannesTschofenig | 0:796d0f61a05b | 228 | uint32_t key_length, unsigned char *output ) |
HannesTschofenig | 0:796d0f61a05b | 229 | { |
HannesTschofenig | 0:796d0f61a05b | 230 | int ret, j; |
HannesTschofenig | 0:796d0f61a05b | 231 | unsigned int i; |
HannesTschofenig | 0:796d0f61a05b | 232 | unsigned char md1[POLARSSL_MD_MAX_SIZE]; |
HannesTschofenig | 0:796d0f61a05b | 233 | unsigned char work[POLARSSL_MD_MAX_SIZE]; |
HannesTschofenig | 0:796d0f61a05b | 234 | unsigned char md_size = md_get_size( ctx->md_info ); |
HannesTschofenig | 0:796d0f61a05b | 235 | size_t use_len; |
HannesTschofenig | 0:796d0f61a05b | 236 | unsigned char *out_p = output; |
HannesTschofenig | 0:796d0f61a05b | 237 | unsigned char counter[4]; |
HannesTschofenig | 0:796d0f61a05b | 238 | |
HannesTschofenig | 0:796d0f61a05b | 239 | memset( counter, 0, 4 ); |
HannesTschofenig | 0:796d0f61a05b | 240 | counter[3] = 1; |
HannesTschofenig | 0:796d0f61a05b | 241 | |
HannesTschofenig | 0:796d0f61a05b | 242 | if( iteration_count > 0xFFFFFFFF ) |
HannesTschofenig | 0:796d0f61a05b | 243 | return( POLARSSL_ERR_PKCS5_BAD_INPUT_DATA ); |
HannesTschofenig | 0:796d0f61a05b | 244 | |
HannesTschofenig | 0:796d0f61a05b | 245 | while( key_length ) |
HannesTschofenig | 0:796d0f61a05b | 246 | { |
HannesTschofenig | 0:796d0f61a05b | 247 | // U1 ends up in work |
HannesTschofenig | 0:796d0f61a05b | 248 | // |
HannesTschofenig | 0:796d0f61a05b | 249 | if( ( ret = md_hmac_starts( ctx, password, plen ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 250 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 251 | |
HannesTschofenig | 0:796d0f61a05b | 252 | if( ( ret = md_hmac_update( ctx, salt, slen ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 253 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 254 | |
HannesTschofenig | 0:796d0f61a05b | 255 | if( ( ret = md_hmac_update( ctx, counter, 4 ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 256 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 257 | |
HannesTschofenig | 0:796d0f61a05b | 258 | if( ( ret = md_hmac_finish( ctx, work ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 259 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 260 | |
HannesTschofenig | 0:796d0f61a05b | 261 | memcpy( md1, work, md_size ); |
HannesTschofenig | 0:796d0f61a05b | 262 | |
HannesTschofenig | 0:796d0f61a05b | 263 | for ( i = 1; i < iteration_count; i++ ) |
HannesTschofenig | 0:796d0f61a05b | 264 | { |
HannesTschofenig | 0:796d0f61a05b | 265 | // U2 ends up in md1 |
HannesTschofenig | 0:796d0f61a05b | 266 | // |
HannesTschofenig | 0:796d0f61a05b | 267 | if( ( ret = md_hmac_starts( ctx, password, plen ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 268 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 269 | |
HannesTschofenig | 0:796d0f61a05b | 270 | if( ( ret = md_hmac_update( ctx, md1, md_size ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 271 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 272 | |
HannesTschofenig | 0:796d0f61a05b | 273 | if( ( ret = md_hmac_finish( ctx, md1 ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 274 | return( ret ); |
HannesTschofenig | 0:796d0f61a05b | 275 | |
HannesTschofenig | 0:796d0f61a05b | 276 | // U1 xor U2 |
HannesTschofenig | 0:796d0f61a05b | 277 | // |
HannesTschofenig | 0:796d0f61a05b | 278 | for( j = 0; j < md_size; j++ ) |
HannesTschofenig | 0:796d0f61a05b | 279 | work[j] ^= md1[j]; |
HannesTschofenig | 0:796d0f61a05b | 280 | } |
HannesTschofenig | 0:796d0f61a05b | 281 | |
HannesTschofenig | 0:796d0f61a05b | 282 | use_len = ( key_length < md_size ) ? key_length : md_size; |
HannesTschofenig | 0:796d0f61a05b | 283 | memcpy( out_p, work, use_len ); |
HannesTschofenig | 0:796d0f61a05b | 284 | |
HannesTschofenig | 0:796d0f61a05b | 285 | key_length -= (uint32_t) use_len; |
HannesTschofenig | 0:796d0f61a05b | 286 | out_p += use_len; |
HannesTschofenig | 0:796d0f61a05b | 287 | |
HannesTschofenig | 0:796d0f61a05b | 288 | for( i = 4; i > 0; i-- ) |
HannesTschofenig | 0:796d0f61a05b | 289 | if( ++counter[i - 1] != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 290 | break; |
HannesTschofenig | 0:796d0f61a05b | 291 | } |
HannesTschofenig | 0:796d0f61a05b | 292 | |
HannesTschofenig | 0:796d0f61a05b | 293 | return( 0 ); |
HannesTschofenig | 0:796d0f61a05b | 294 | } |
HannesTschofenig | 0:796d0f61a05b | 295 | |
HannesTschofenig | 0:796d0f61a05b | 296 | #if defined(POLARSSL_SELF_TEST) |
HannesTschofenig | 0:796d0f61a05b | 297 | |
HannesTschofenig | 0:796d0f61a05b | 298 | #include <stdio.h> |
HannesTschofenig | 0:796d0f61a05b | 299 | |
HannesTschofenig | 0:796d0f61a05b | 300 | #define MAX_TESTS 6 |
HannesTschofenig | 0:796d0f61a05b | 301 | |
HannesTschofenig | 0:796d0f61a05b | 302 | size_t plen[MAX_TESTS] = |
HannesTschofenig | 0:796d0f61a05b | 303 | { 8, 8, 8, 8, 24, 9 }; |
HannesTschofenig | 0:796d0f61a05b | 304 | |
HannesTschofenig | 0:796d0f61a05b | 305 | unsigned char password[MAX_TESTS][32] = |
HannesTschofenig | 0:796d0f61a05b | 306 | { |
HannesTschofenig | 0:796d0f61a05b | 307 | "password", |
HannesTschofenig | 0:796d0f61a05b | 308 | "password", |
HannesTschofenig | 0:796d0f61a05b | 309 | "password", |
HannesTschofenig | 0:796d0f61a05b | 310 | "password", |
HannesTschofenig | 0:796d0f61a05b | 311 | "passwordPASSWORDpassword", |
HannesTschofenig | 0:796d0f61a05b | 312 | "pass\0word", |
HannesTschofenig | 0:796d0f61a05b | 313 | }; |
HannesTschofenig | 0:796d0f61a05b | 314 | |
HannesTschofenig | 0:796d0f61a05b | 315 | size_t slen[MAX_TESTS] = |
HannesTschofenig | 0:796d0f61a05b | 316 | { 4, 4, 4, 4, 36, 5 }; |
HannesTschofenig | 0:796d0f61a05b | 317 | |
HannesTschofenig | 0:796d0f61a05b | 318 | unsigned char salt[MAX_TESTS][40] = |
HannesTschofenig | 0:796d0f61a05b | 319 | { |
HannesTschofenig | 0:796d0f61a05b | 320 | "salt", |
HannesTschofenig | 0:796d0f61a05b | 321 | "salt", |
HannesTschofenig | 0:796d0f61a05b | 322 | "salt", |
HannesTschofenig | 0:796d0f61a05b | 323 | "salt", |
HannesTschofenig | 0:796d0f61a05b | 324 | "saltSALTsaltSALTsaltSALTsaltSALTsalt", |
HannesTschofenig | 0:796d0f61a05b | 325 | "sa\0lt", |
HannesTschofenig | 0:796d0f61a05b | 326 | }; |
HannesTschofenig | 0:796d0f61a05b | 327 | |
HannesTschofenig | 0:796d0f61a05b | 328 | uint32_t it_cnt[MAX_TESTS] = |
HannesTschofenig | 0:796d0f61a05b | 329 | { 1, 2, 4096, 16777216, 4096, 4096 }; |
HannesTschofenig | 0:796d0f61a05b | 330 | |
HannesTschofenig | 0:796d0f61a05b | 331 | uint32_t key_len[MAX_TESTS] = |
HannesTschofenig | 0:796d0f61a05b | 332 | { 20, 20, 20, 20, 25, 16 }; |
HannesTschofenig | 0:796d0f61a05b | 333 | |
HannesTschofenig | 0:796d0f61a05b | 334 | |
HannesTschofenig | 0:796d0f61a05b | 335 | unsigned char result_key[MAX_TESTS][32] = |
HannesTschofenig | 0:796d0f61a05b | 336 | { |
HannesTschofenig | 0:796d0f61a05b | 337 | { 0x0c, 0x60, 0xc8, 0x0f, 0x96, 0x1f, 0x0e, 0x71, |
HannesTschofenig | 0:796d0f61a05b | 338 | 0xf3, 0xa9, 0xb5, 0x24, 0xaf, 0x60, 0x12, 0x06, |
HannesTschofenig | 0:796d0f61a05b | 339 | 0x2f, 0xe0, 0x37, 0xa6 }, |
HannesTschofenig | 0:796d0f61a05b | 340 | { 0xea, 0x6c, 0x01, 0x4d, 0xc7, 0x2d, 0x6f, 0x8c, |
HannesTschofenig | 0:796d0f61a05b | 341 | 0xcd, 0x1e, 0xd9, 0x2a, 0xce, 0x1d, 0x41, 0xf0, |
HannesTschofenig | 0:796d0f61a05b | 342 | 0xd8, 0xde, 0x89, 0x57 }, |
HannesTschofenig | 0:796d0f61a05b | 343 | { 0x4b, 0x00, 0x79, 0x01, 0xb7, 0x65, 0x48, 0x9a, |
HannesTschofenig | 0:796d0f61a05b | 344 | 0xbe, 0xad, 0x49, 0xd9, 0x26, 0xf7, 0x21, 0xd0, |
HannesTschofenig | 0:796d0f61a05b | 345 | 0x65, 0xa4, 0x29, 0xc1 }, |
HannesTschofenig | 0:796d0f61a05b | 346 | { 0xee, 0xfe, 0x3d, 0x61, 0xcd, 0x4d, 0xa4, 0xe4, |
HannesTschofenig | 0:796d0f61a05b | 347 | 0xe9, 0x94, 0x5b, 0x3d, 0x6b, 0xa2, 0x15, 0x8c, |
HannesTschofenig | 0:796d0f61a05b | 348 | 0x26, 0x34, 0xe9, 0x84 }, |
HannesTschofenig | 0:796d0f61a05b | 349 | { 0x3d, 0x2e, 0xec, 0x4f, 0xe4, 0x1c, 0x84, 0x9b, |
HannesTschofenig | 0:796d0f61a05b | 350 | 0x80, 0xc8, 0xd8, 0x36, 0x62, 0xc0, 0xe4, 0x4a, |
HannesTschofenig | 0:796d0f61a05b | 351 | 0x8b, 0x29, 0x1a, 0x96, 0x4c, 0xf2, 0xf0, 0x70, |
HannesTschofenig | 0:796d0f61a05b | 352 | 0x38 }, |
HannesTschofenig | 0:796d0f61a05b | 353 | { 0x56, 0xfa, 0x6a, 0xa7, 0x55, 0x48, 0x09, 0x9d, |
HannesTschofenig | 0:796d0f61a05b | 354 | 0xcc, 0x37, 0xd7, 0xf0, 0x34, 0x25, 0xe0, 0xc3 }, |
HannesTschofenig | 0:796d0f61a05b | 355 | }; |
HannesTschofenig | 0:796d0f61a05b | 356 | |
HannesTschofenig | 0:796d0f61a05b | 357 | int pkcs5_self_test( int verbose ) |
HannesTschofenig | 0:796d0f61a05b | 358 | { |
HannesTschofenig | 0:796d0f61a05b | 359 | md_context_t sha1_ctx; |
HannesTschofenig | 0:796d0f61a05b | 360 | const md_info_t *info_sha1; |
HannesTschofenig | 0:796d0f61a05b | 361 | int ret, i; |
HannesTschofenig | 0:796d0f61a05b | 362 | unsigned char key[64]; |
HannesTschofenig | 0:796d0f61a05b | 363 | |
HannesTschofenig | 0:796d0f61a05b | 364 | info_sha1 = md_info_from_type( POLARSSL_MD_SHA1 ); |
HannesTschofenig | 0:796d0f61a05b | 365 | if( info_sha1 == NULL ) |
HannesTschofenig | 0:796d0f61a05b | 366 | return( 1 ); |
HannesTschofenig | 0:796d0f61a05b | 367 | |
HannesTschofenig | 0:796d0f61a05b | 368 | if( ( ret = md_init_ctx( &sha1_ctx, info_sha1 ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 369 | return( 1 ); |
HannesTschofenig | 0:796d0f61a05b | 370 | |
HannesTschofenig | 0:796d0f61a05b | 371 | if( verbose != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 372 | polarssl_printf( " PBKDF2 note: test #3 may be slow!\n" ); |
HannesTschofenig | 0:796d0f61a05b | 373 | |
HannesTschofenig | 0:796d0f61a05b | 374 | for( i = 0; i < MAX_TESTS; i++ ) |
HannesTschofenig | 0:796d0f61a05b | 375 | { |
HannesTschofenig | 0:796d0f61a05b | 376 | if( verbose != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 377 | polarssl_printf( " PBKDF2 (SHA1) #%d: ", i ); |
HannesTschofenig | 0:796d0f61a05b | 378 | |
HannesTschofenig | 0:796d0f61a05b | 379 | ret = pkcs5_pbkdf2_hmac( &sha1_ctx, password[i], plen[i], salt[i], |
HannesTschofenig | 0:796d0f61a05b | 380 | slen[i], it_cnt[i], key_len[i], key ); |
HannesTschofenig | 0:796d0f61a05b | 381 | if( ret != 0 || |
HannesTschofenig | 0:796d0f61a05b | 382 | memcmp( result_key[i], key, key_len[i] ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 383 | { |
HannesTschofenig | 0:796d0f61a05b | 384 | if( verbose != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 385 | polarssl_printf( "failed\n" ); |
HannesTschofenig | 0:796d0f61a05b | 386 | |
HannesTschofenig | 0:796d0f61a05b | 387 | return( 1 ); |
HannesTschofenig | 0:796d0f61a05b | 388 | } |
HannesTschofenig | 0:796d0f61a05b | 389 | |
HannesTschofenig | 0:796d0f61a05b | 390 | if( verbose != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 391 | polarssl_printf( "passed\n" ); |
HannesTschofenig | 0:796d0f61a05b | 392 | } |
HannesTschofenig | 0:796d0f61a05b | 393 | |
HannesTschofenig | 0:796d0f61a05b | 394 | polarssl_printf( "\n" ); |
HannesTschofenig | 0:796d0f61a05b | 395 | |
HannesTschofenig | 0:796d0f61a05b | 396 | if( ( ret = md_free_ctx( &sha1_ctx ) ) != 0 ) |
HannesTschofenig | 0:796d0f61a05b | 397 | return( 1 ); |
HannesTschofenig | 0:796d0f61a05b | 398 | |
HannesTschofenig | 0:796d0f61a05b | 399 | return( 0 ); |
HannesTschofenig | 0:796d0f61a05b | 400 | } |
HannesTschofenig | 0:796d0f61a05b | 401 | |
HannesTschofenig | 0:796d0f61a05b | 402 | #endif /* POLARSSL_SELF_TEST */ |
HannesTschofenig | 0:796d0f61a05b | 403 | |
HannesTschofenig | 0:796d0f61a05b | 404 | #endif /* POLARSSL_PKCS5_C */ |
HannesTschofenig | 0:796d0f61a05b | 405 | |
HannesTschofenig | 0:796d0f61a05b | 406 |