eapol/eapol_framework/wapi_common/src/wapi_am_crypto_sms4.cpp
changeset 18 7aac0b9e8906
parent 17 8840d3e38314
child 20 8b3129ac4c0f
equal deleted inserted replaced
17:8840d3e38314 18:7aac0b9e8906
     1 /*
       
     2 * ============================================================================
       
     3 *  Name        : ./accesssec/eapol/eapol_framework/wapi_common/src/wapi_am_crypto_sms4.cpp
       
     4 *  Part of     : WAPI / WAPI       *** Info from the SWAD
       
     5 *  Description : WAPI authentication
       
     6 *  Version     : %version: 4 % << Don't touch! Updated by Synergy at check-out.
       
     7 *
       
     8 *  Copyright © 2001-2009 Nokia.  All rights reserved.
       
     9 *  This material, including documentation and any related computer
       
    10 *  programs, is protected by copyright controlled by Nokia.  All
       
    11 *  rights are reserved.  Copying, including reproducing, storing,
       
    12 *  adapting or translating, any or all of this material requires the
       
    13 *  prior written consent of Nokia.  This material also contains
       
    14 *  confidential information which may not be disclosed to others
       
    15 *  without the prior written consent of Nokia.
       
    16 * ============================================================================
       
    17 * Template version: 4.1.1
       
    18 */
       
    19 
       
    20 // This is enumeration of WAPI source code.
       
    21 #if defined(USE_EAP_MINIMUM_RELEASE_TRACES)
       
    22 	#undef EAP_FILE_NUMBER_ENUM
       
    23 	#define EAP_FILE_NUMBER_ENUM 20010 
       
    24 	#undef EAP_FILE_NUMBER_DATE 
       
    25 	#define EAP_FILE_NUMBER_DATE 1127594498 
       
    26 #endif //#if defined(USE_EAP_MINIMUM_RELEASE_TRACES)
       
    27 
       
    28 
       
    29 #include "eap_am_memory.h"
       
    30 #include "wapi_am_crypto_sms4.h"
       
    31 
       
    32 
       
    33 //------------------------------------------------------------
       
    34 // SMS4 constants 
       
    35 
       
    36 // CK
       
    37 const u32_t wapi_am_crypto_sms4_c::m_CK[wapi_am_crypto_sms4_c::WAPI_AM_CRYPTO_SMS4_CK_u32_COUNT] =
       
    38 {
       
    39 	0x00070e15, 0x1c232a31, 0x383f464d, 0x545b6269,
       
    40 	0x70777e85, 0x8c939aa1, 0xa8afb6bd, 0xc4cbd2d9,
       
    41 	0xe0e7eef5, 0xfc030a11, 0x181f262d, 0x343b4249,
       
    42 	0x50575e65, 0x6c737a81, 0x888f969d, 0xa4abb2b9,
       
    43 	0xc0c7ced5, 0xdce3eaf1, 0xf8ff060d, 0x141b2229,
       
    44 	0x30373e45, 0x4c535a61, 0x686f767d, 0x848b9299,
       
    45 	0xa0a7aeb5, 0xbcc3cad1, 0xd8dfe6ed, 0xf4fb0209,
       
    46 	0x10171e25, 0x2c333a41, 0x484f565d, 0x646b7279
       
    47 };
       
    48 
       
    49 // FK
       
    50 const u32_t wapi_am_crypto_sms4_c::m_FK[wapi_am_crypto_sms4_c::WAPI_AM_CRYPTO_SMS4_FK_u32_COUNT] =
       
    51 {
       
    52 	0xa3b1bac6, 0x56aa3350, 0x677d9197, 0xb27022dc
       
    53 };
       
    54 
       
    55 
       
    56 // S-box
       
    57 const u8_t wapi_am_crypto_sms4_c::m_SBOX[wapi_am_crypto_sms4_c::WAPI_AM_CRYPTO_SMS4_SBOX_u8_SIZE] =
       
    58 {
       
    59 	0xd6, 0x90, 0xe9, 0xfe, 0xcc, 0xe1, 0x3d, 0xb7, 0x16, 0xb6, 0x14, 0xc2, 0x28, 0xfb, 0x2c, 0x05,
       
    60 	0x2b, 0x67, 0x9a, 0x76, 0x2a, 0xbe, 0x04, 0xc3, 0xaa, 0x44, 0x13, 0x26, 0x49, 0x86, 0x06, 0x99,
       
    61 	0x9c, 0x42, 0x50, 0xf4, 0x91, 0xef, 0x98, 0x7a, 0x33, 0x54, 0x0b, 0x43, 0xed, 0xcf, 0xac, 0x62,
       
    62 	0xe4, 0xb3, 0x1c, 0xa9, 0xc9, 0x08, 0xe8, 0x95, 0x80, 0xdf, 0x94, 0xfa, 0x75, 0x8f, 0x3f, 0xa6,
       
    63 	0x47, 0x07, 0xa7, 0xfc, 0xf3, 0x73, 0x17, 0xba, 0x83, 0x59, 0x3c, 0x19, 0xe6, 0x85, 0x4f, 0xa8,
       
    64 	0x68, 0x6b, 0x81, 0xb2, 0x71, 0x64, 0xda, 0x8b, 0xf8, 0xeb, 0x0f, 0x4b, 0x70, 0x56, 0x9d, 0x35,
       
    65 	0x1e, 0x24, 0x0e, 0x5e, 0x63, 0x58, 0xd1, 0xa2, 0x25, 0x22, 0x7c, 0x3b, 0x01, 0x21, 0x78, 0x87,
       
    66 	0xd4, 0x00, 0x46, 0x57, 0x9f, 0xd3, 0x27, 0x52, 0x4c, 0x36, 0x02, 0xe7, 0xa0, 0xc4, 0xc8, 0x9e,
       
    67 	0xea, 0xbf, 0x8a, 0xd2, 0x40, 0xc7, 0x38, 0xb5, 0xa3, 0xf7, 0xf2, 0xce, 0xf9, 0x61, 0x15, 0xa1,
       
    68 	0xe0, 0xae, 0x5d, 0xa4, 0x9b, 0x34, 0x1a, 0x55, 0xad, 0x93, 0x32, 0x30, 0xf5, 0x8c, 0xb1, 0xe3,
       
    69 	0x1d, 0xf6, 0xe2, 0x2e, 0x82, 0x66, 0xca, 0x60, 0xc0, 0x29, 0x23, 0xab, 0x0d, 0x53, 0x4e, 0x6f,
       
    70 	0xd5, 0xdb, 0x37, 0x45, 0xde, 0xfd, 0x8e, 0x2f, 0x03, 0xff, 0x6a, 0x72, 0x6d, 0x6c, 0x5b, 0x51,
       
    71 	0x8d, 0x1b, 0xaf, 0x92, 0xbb, 0xdd, 0xbc, 0x7f, 0x11, 0xd9, 0x5c, 0x41, 0x1f, 0x10, 0x5a, 0xd8,
       
    72 	0x0a, 0xc1, 0x31, 0x88, 0xa5, 0xcd, 0x7b, 0xbd, 0x2d, 0x74, 0xd0, 0x12, 0xb8, 0xe5, 0xb4, 0xb0,
       
    73 	0x89, 0x69, 0x97, 0x4a, 0x0c, 0x96, 0x77, 0x7e, 0x65, 0xb9, 0xf1, 0x09, 0xc5, 0x6e, 0xc6, 0x84,
       
    74 	0x18, 0xf0, 0x7d, 0xec, 0x3a, 0xdc, 0x4d, 0x20, 0x79, 0xee, 0x5f, 0x3e, 0xd7, 0xcb, 0x39, 0x48
       
    75 };
       
    76 
       
    77 //------------------------------------------------------------
       
    78 
       
    79 
       
    80 
       
    81 EAP_FUNC_EXPORT wapi_am_crypto_sms4_c::~wapi_am_crypto_sms4_c()
       
    82 {
       
    83 	m_is_valid = false;
       
    84 }
       
    85 
       
    86 //------------------------------------------------------------
       
    87 
       
    88 EAP_FUNC_EXPORT wapi_am_crypto_sms4_c::wapi_am_crypto_sms4_c(
       
    89 	abs_eap_am_tools_c * const tools)
       
    90 	: m_am_tools(tools)
       
    91 	, m_is_valid(false)
       
    92 {
       
    93 	set_is_valid();
       
    94 }
       
    95 
       
    96 //------------------------------------------------------------
       
    97 
       
    98 EAP_FUNC_EXPORT void wapi_am_crypto_sms4_c::set_is_invalid()
       
    99 {
       
   100 	m_is_valid = false;
       
   101 }
       
   102 
       
   103 //------------------------------------------------------------
       
   104 
       
   105 EAP_FUNC_EXPORT void wapi_am_crypto_sms4_c::set_is_valid()
       
   106 {
       
   107 	m_is_valid = true;
       
   108 }
       
   109 
       
   110 //------------------------------------------------------------
       
   111 
       
   112 EAP_FUNC_EXPORT bool wapi_am_crypto_sms4_c::get_is_valid()
       
   113 {
       
   114 	return m_is_valid;
       
   115 }
       
   116 
       
   117 //------------------------------------------------------------
       
   118 
       
   119 EAP_FUNC_EXPORT u32_t wapi_am_crypto_sms4_c::get_key_size()
       
   120 {
       
   121 	return WAPI_AM_CRYPTO_SMS4_KEY_u8_SIZE;
       
   122 }
       
   123 
       
   124 //------------------------------------------------------------
       
   125 
       
   126 EAP_FUNC_EXPORT u32_t wapi_am_crypto_sms4_c::get_block_size()
       
   127 {
       
   128 	return WAPI_AM_CRYPTO_SMS4_BLOCK_u8_SIZE;
       
   129 }
       
   130 
       
   131 //------------------------------------------------------------
       
   132 
       
   133 EAP_FUNC_EXPORT eap_status_e wapi_am_crypto_sms4_c::set_key(
       
   134 	const eap_variable_data_c * const p_key)
       
   135 {
       
   136 
       
   137 	// key in network order
       
   138 
       
   139 	// check if key is ok
       
   140 	if( p_key == 0 ||
       
   141 		p_key->get_is_valid() == false ||
       
   142 		p_key->get_data_length() != WAPI_AM_CRYPTO_SMS4_KEY_u8_SIZE )
       
   143 	{
       
   144 		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_encryption_parameter_size);
       
   145 	}
       
   146 
       
   147 	// temporary table to hold the currently
       
   148 	// needed key data words
       
   149 	u32_t key_table[WAPI_AM_CRYPTO_SMS4_KEY_u8_SIZE/4];
       
   150 	u32_t ind = 0;
       
   151 
       
   152 	for( ind = 0; ind < WAPI_AM_CRYPTO_SMS4_KEY_u8_SIZE/4; ++ind )
       
   153 	{
       
   154 		// read key from variable data to a u32_t table for faster processing
       
   155 		// TODO: Is this ok for endianness on other platforms?
       
   156 		key_table[ind] = eap_read_u32_t_network_order(p_key->get_data() + ind*4, 4);
       
   157 	}
       
   158 		
       
   159 	// temporary variable to hold the current
       
   160 	// expansion result (one word)
       
   161 	u32_t tmp_K;
       
   162 
       
   163 	// XOR key words with FKs
       
   164 	for( ind = 0; ind < WAPI_AM_CRYPTO_SMS4_KEY_u8_SIZE/4; ++ind )
       
   165 	{
       
   166 		key_table[ind] = key_table[ind] ^ m_FK[ind];
       
   167 	}
       
   168 
       
   169 	// compute the key expansion
       
   170 	for( ind = 0; ind < WAPI_AM_CRYPTO_SMS4_KEY_SCHEDULE_u32_SIZE; ++ind )
       
   171 	{
       
   172 
       
   173 		// See SMS4 spec for these
       
   174 		tmp_K = key_table[1] ^ key_table[2] ^ key_table[3] ^ m_CK[ind];
       
   175 		sms4_substitute( &tmp_K );
       
   176 		L_key( &tmp_K );
       
   177 		tmp_K ^= key_table[0];
       
   178 
       
   179 		// store the result for the next round
       
   180 		key_table[0] = key_table[1];
       
   181 		key_table[1] = key_table[2];
       
   182 		key_table[2] = key_table[3];
       
   183 		key_table[3] = tmp_K;
       
   184 
       
   185 		// store the expansion result
       
   186 		m_key_schedule[ind] = tmp_K;
       
   187 
       
   188 	} // for()
       
   189 
       
   190 
       
   191 	return EAP_STATUS_RETURN(m_am_tools, eap_status_ok);
       
   192 }
       
   193 
       
   194 //------------------------------------------------------------
       
   195 
       
   196 /*
       
   197  * Performs L' function for a key word.
       
   198  */
       
   199 
       
   200 EAP_FUNC_EXPORT void wapi_am_crypto_sms4_c::L_key( u32_t* data )
       
   201 {
       
   202 	*data ^= sms4_rotate_left(*data, 13) ^ sms4_rotate_left(*data, 23);
       
   203 	return;
       
   204 }
       
   205 
       
   206 //------------------------------------------------------------
       
   207 
       
   208 /*
       
   209  * Performs S-box substitution for a data word,
       
   210  * i.e. four S-box substitutions.
       
   211  */
       
   212 
       
   213 EAP_FUNC_EXPORT void wapi_am_crypto_sms4_c::sms4_substitute( u32_t* data )
       
   214 {	
       
   215 	u8_t* tmp = reinterpret_cast<u8_t*>(data);
       
   216 
       
   217 	// S-box substitution to the bytes of the word
       
   218 	*tmp = m_SBOX[*tmp];
       
   219 	*(tmp+1) = m_SBOX[*(tmp+1)];
       
   220 	*(tmp+2) = m_SBOX[*(tmp+2)];
       
   221 	*(tmp+3) = m_SBOX[*(tmp+3)];
       
   222 
       
   223 	return;
       
   224 }
       
   225 
       
   226 //------------------------------------------------------------
       
   227 
       
   228 /*
       
   229  * Performs L function for a data word.
       
   230  */
       
   231 
       
   232 EAP_FUNC_EXPORT void wapi_am_crypto_sms4_c::L_data( u32_t* data )
       
   233 {
       
   234 	*data ^= 
       
   235 		sms4_rotate_left(*data, 2) ^ 
       
   236 		sms4_rotate_left(*data, 10) ^ 
       
   237 		sms4_rotate_left(*data, 18) ^
       
   238 		sms4_rotate_left(*data, 24);
       
   239 	return;
       
   240 }
       
   241 
       
   242 //------------------------------------------------------------
       
   243 
       
   244 EAP_FUNC_EXPORT eap_status_e wapi_am_crypto_sms4_c::ecb_process_data(
       
   245 	const void * const data_in, 
       
   246 	void * const data_out,
       
   247 	const u32_t data_blocks,
       
   248 	bool encrypt)
       
   249 {
       
   250 
       
   251 	// data in network order
       
   252 
       
   253 	// check if data is ok
       
   254 	if( data_in == 0 ||
       
   255 		data_out == 0 ||
       
   256 		data_blocks <= 0 )
       
   257 	{
       
   258 		return EAP_STATUS_RETURN(m_am_tools, eap_status_illegal_encryption_parameter_size);
       
   259 	}
       
   260 
       
   261 	// temporary pointers to data_in and data_out
       
   262 	const u8_t* p_data_in = reinterpret_cast<const u8_t*>(data_in);
       
   263 	u8_t* p_data_out = reinterpret_cast<u8_t*>(data_out);
       
   264 
       
   265 	// count of the processed blocks
       
   266 	u32_t blocks_processed = 0;
       
   267 
       
   268 	// temporary table to hold the currently needed data words
       
   269 	u32_t data_table[WAPI_AM_CRYPTO_SMS4_BLOCK_u8_SIZE/4];
       
   270 	u32_t ind = 0;
       
   271 	eap_status_e status(eap_status_ok);
       
   272 
       
   273 	// temporary variable to hold the current result (one word)
       
   274 	u32_t tmp_X;
       
   275 
       
   276 	// ecrypt data in ECB mode
       
   277 	while( blocks_processed < data_blocks ) 
       
   278 	{
       
   279 
       
   280 		for( ind = 0; ind < 4; ++ind )
       
   281 		{
       
   282 			// read network order data to a u32_t table for faster processing
       
   283 			// TODO: Is this ok for endianness on other platforms?
       
   284 			data_table[ind] = eap_read_u32_t_network_order(p_data_in + ind*4, 4);
       
   285 		}
       
   286 		
       
   287 		// execute the SMS4 rounds
       
   288 		for( ind = 0; ind < 32; ++ind )
       
   289 		{
       
   290 
       
   291 			// See SMS4 spec for these
       
   292 			tmp_X = data_table[1] ^ data_table[2] ^ data_table[3];
       
   293 
       
   294 			if( encrypt == true )
       
   295 			{
       
   296 				tmp_X ^= m_key_schedule[ind];
       
   297 			}
       
   298 			else
       
   299 			{
       
   300 				// in decryption the key schedule is reversed
       
   301 				tmp_X ^= m_key_schedule[WAPI_AM_CRYPTO_SMS4_KEY_SCHEDULE_u32_SIZE-1-ind];
       
   302 			}
       
   303 
       
   304 			sms4_substitute( &tmp_X );
       
   305 			L_data( &tmp_X );
       
   306 			tmp_X ^= data_table[0];
       
   307 
       
   308 			// store the result for the next round
       
   309 			data_table[0] = data_table[1];
       
   310 			data_table[1] = data_table[2];
       
   311 			data_table[2] = data_table[3];
       
   312 			data_table[3] = tmp_X;
       
   313 
       
   314 		} // for()
       
   315 
       
   316 		// the result of the final round is the output,
       
   317 		// except that the order of the words is reversed 
       
   318 		// (R function in the SMS4 spec)
       
   319 		for( ind = 0; ind < 4; ++ind )
       
   320 		{
       
   321 			status = eap_write_u32_t_network_order( p_data_out+4*ind, 4, data_table[3-ind] );
       
   322 			if( status != eap_status_ok )
       
   323 			{
       
   324 				return EAP_STATUS_RETURN(m_am_tools, status);
       
   325 			}
       
   326 		}
       
   327 
       
   328 		++blocks_processed;
       
   329 		// take the next data block for processing
       
   330 		p_data_in += 16;
       
   331 		p_data_out += 16;
       
   332 
       
   333 	} // for()
       
   334 
       
   335 	return EAP_STATUS_RETURN(m_am_tools, eap_status_ok);
       
   336 }
       
   337 
       
   338 //------------------------------------------------------------
       
   339 
       
   340 EAP_FUNC_EXPORT eap_status_e wapi_am_crypto_sms4_c::ecb_encrypt(
       
   341 	const void * const data_in, 
       
   342 	void * const data_out,
       
   343 	const u32_t data_blocks)
       
   344 {
       
   345 	eap_status_e status = ecb_process_data( data_in, data_out, data_blocks, true );
       
   346 	return EAP_STATUS_RETURN(m_am_tools, status);
       
   347 }
       
   348 
       
   349 //------------------------------------------------------------
       
   350 
       
   351 EAP_FUNC_EXPORT eap_status_e wapi_am_crypto_sms4_c::ecb_decrypt(
       
   352 	const void * const data_in, 
       
   353 	void * const data_out,
       
   354 	const u32_t data_blocks)
       
   355 {
       
   356 	eap_status_e status = ecb_process_data( data_in, data_out, data_blocks, false );
       
   357 	return EAP_STATUS_RETURN(m_am_tools, status);
       
   358 }
       
   359 
       
   360 //------------------------------------------------------------
       
   361 
       
   362 // End.