tested OK

git-svn-id: http://svn.eurecom.fr/openair4G/trunk@5513 818b1a75-f10b-46b9-bf7c-635c3b92a50f

openair-cn/SECU/nas_stream_eia1.c

+#include <stdlib.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <string.h>
+#include <math.h> // double ceil(double x);
+
+#include "secu_defs.h"
+
+#include "assertions.h"
+#include "conversions.h"
+#include "snow3g.h"
+
+#define SECU_DEBUG
+
+uint64_t MUL64x(uint64_t V, uint64_t c);
+uint64_t MUL64xPOW(uint64_t V, uint32_t i, uint64_t c);
+uint64_t MUL64(uint64_t V, uint64_t P, uint64_t c);
+int      nas_stream_encrypt_eia1(nas_stream_cipher_t *stream_cipher, uint8_t out[4]);
+
+
+// see spec 3GPP Confidentiality and Integrity Algorithms UEA2&UIA2. Document 1: UEA2 and UIA2 Specification. Version 1.1
+
+/* MUL64x.
+ * Input V: a 64-bit input.
+ * Input c: a 64-bit input.
+ * Output : a 64-bit output.
+ * A 64-bit memory is allocated which is to be freed by the calling
+ * function.
+ * See section 4.3.2 for details.
+ */
+uint64_t MUL64x(uint64_t V, uint64_t c)
+{
+	if ( V & 0x8000000000000000 )
+		return (V << 1) ^ c;
+	else
+		return V << 1;
+}
+/* MUL64xPOW.
+ * Input V: a 64-bit input.
+ * Input i: a positive integer.
+ * Input c: a 64-bit input.
+ * Output : a 64-bit output.
+ * A 64-bit memory is allocated which is to be freed by the calling
+function.
+ * See section 4.3.3 for details.
+ */
+uint64_t MUL64xPOW(uint64_t V, uint32_t i, uint64_t c)
+{
+	if ( i == 0)
+		return V;
+	else
+		return MUL64x( MUL64xPOW(V,i-1,c) , c);
+}
+/* MUL64.
+ * Input V: a 64-bit input.
+ * Input P: a 64-bit input.
+ * Input c: a 64-bit input.
+ * Output : a 64-bit output.
+ * A 64-bit memory is allocated which is to be freed by the calling
+ * function.
+ * See section 4.3.4 for details.
+ */
+uint64_t MUL64(uint64_t V, uint64_t P, uint64_t c)
+{
+	uint64_t result = 0;
+	int i = 0;
+	for ( i=0; i<64; i++)
+	{
+		if( ( P>>i ) & 0x1 )
+			result ^= MUL64xPOW(V,i,c);
+	}
+	return result;
+}
+
+/* mask32bit.
+* Input n: an integer in 1-32.
+* Output : a 32 bit mask.
+* Prepares a 32 bit mask with required number of 1 bits on the MSB side.
+*/
+uint32_t mask32bit(int n)
+{
+	uint32_t mask=0x0;
+	if ( n%32 == 0 )
+		return 0xffffffff;
+	while (n--)
+		mask = (mask>>1) ^ 0x80000000;
+	return mask;
+}
+
+
+/*!
+ * @brief Create integrity cmac t for a given message.
+ * @param[in] stream_cipher Structure containing various variables to setup encoding
+ * @param[out] out For EIA1 the output string is 32 bits long
+ */
+int nas_stream_encrypt_eia1(nas_stream_cipher_t *stream_cipher, uint8_t out[4])
+{
+    snow_3g_context_t snow_3g_context;
+    uint32_t        K[4],IV[4], z[5];
+    int             i=0,D;
+    uint32_t        MAC_I = 0;
+    uint64_t        EVAL;
+    uint64_t        V;
+    uint64_t        P;
+    uint64_t        Q;
+    uint64_t        c;
+    uint64_t        M_D_2;
+    int             rem_bits;
+    uint32_t        mask = 0;
+    uint32_t       *message;
+
+    message = (uint32_t*)stream_cipher->message; /* To operate 32 bit message internally. */
+    /* Load the Integrity Key for SNOW3G initialization as in section 4.4. */
+    memcpy(K+3,stream_cipher->key+0,4); /*K[3] = key[0]; we assume
+    K[3]=key[0]||key[1]||...||key[31] , with key[0] the
+    * most important bit of key*/
+    memcpy(K+2,stream_cipher->key+4,4); /*K[2] = key[1];*/
+    memcpy(K+1,stream_cipher->key+8,4); /*K[1] = key[2];*/
+    memcpy(K+0,stream_cipher->key+12,4); /*K[0] = key[3]; we assume
+    K[0]=key[96]||key[97]||...||key[127] , with key[127] the
+    * least important bit of key*/
+	K[3] = hton_int32(K[3]);
+	K[2] = hton_int32(K[2]);
+	K[1] = hton_int32(K[1]);
+	K[0] = hton_int32(K[0]);
+	/* Prepare the Initialization Vector (IV) for SNOW3G initialization as in
+    section 4.4. */
+    IV[3] = (uint32_t)stream_cipher->count;
+    IV[2] = ((((uint32_t)stream_cipher->bearer) & 0x0000001F) << 27);
+    IV[1] = (uint32_t)(stream_cipher->count) ^ ( (uint32_t)(stream_cipher->direction) << 31 ) ;
+    IV[0] = ((((uint32_t)stream_cipher->bearer) & 0x0000001F) << 27) ^ ((uint32_t)(stream_cipher->direction & 0x00000001) << 15);
+    //printf ("K:\n");
+	//hexprint(K, 16);
+    //printf ("K[0]:%08X\n",K[0]);
+    //printf ("K[1]:%08X\n",K[1]);
+    //printf ("K[2]:%08X\n",K[2]);
+    //printf ("K[3]:%08X\n",K[3]);
+
+    //printf ("IV:\n");
+	//hexprint(IV, 16);
+    //printf ("IV[0]:%08X\n",IV[0]);
+    //printf ("IV[1]:%08X\n",IV[1]);
+    //printf ("IV[2]:%08X\n",IV[2]);
+    //printf ("IV[3]:%08X\n",IV[3]);
+    z[0] = z[1] = z[2] = z[3] = z[4] = 0;
+    /* Run SNOW 3G to produce 5 keystream words z_1, z_2, z_3, z_4 and z_5. */
+    snow3g_initialize(K, IV, &snow_3g_context);
+    snow3g_generate_key_stream(5, z, &snow_3g_context);
+    //printf ("z[0]:%08X\n",z[0]);
+    //printf ("z[1]:%08X\n",z[1]);
+    //printf ("z[2]:%08X\n",z[2]);
+    //printf ("z[3]:%08X\n",z[3]);
+    //printf ("z[4]:%08X\n",z[4]);
+    P = ((uint64_t)z[0] << 32) | (uint64_t)z[1];
+    Q = ((uint64_t)z[2] << 32) | (uint64_t)z[3];
+    //printf ("P:%16lX\n",P);
+    //printf ("Q:%16lX\n",Q);
+    /* Calculation */
+    D = ceil( stream_cipher->blength / 64.0 ) + 1;
+    //printf ("D:%d\n",D);
+    EVAL = 0;
+    c = 0x1b;
+    /* for 0 <= i <= D-3 */
+    for (i=0;i<D-2;i++) {
+    	V = EVAL ^ ( (uint64_t)hton_int32(message[2*i]) << 32 | (uint64_t)hton_int32(message[2*i+1]) );
+    	EVAL = MUL64(V,P,c);
+        //printf ("Mi: %16X %16X\tEVAL: %16lX\n",hton_int32(message[2*i]),hton_int32(message[2*i+1]), EVAL);
+    }
+    /* for D-2 */
+    rem_bits = stream_cipher->blength % 64;
+    if (rem_bits == 0)
+    	rem_bits = 64;
+    mask = mask32bit(rem_bits%32);
+    if (rem_bits > 32) {
+    	M_D_2 = ( (uint64_t) hton_int32(message[2*(D-2)]) << 32 ) |
+    			(uint64_t) (hton_int32(message[2*(D-2)+1]) &  mask);
+    } else {
+    	M_D_2 = ( (uint64_t) hton_int32(message[2*(D-2)]) & mask) << 32 ;
+    }
+    V = EVAL ^ M_D_2;
+    EVAL = MUL64(V,P,c);
+    /* for D-1 */
+    EVAL ^= stream_cipher->blength;
+    /* Multiply by Q */
+    EVAL = MUL64(EVAL,Q,c);
+    MAC_I = (uint32_t)(EVAL >> 32) ^ z[4];
+    //printf ("MAC_I:%16X\n",MAC_I);
+    MAC_I = hton_int32(MAC_I);
+    memcpy(out, &MAC_I, 4);
+    return 0;
+}

openair-cn/SECU/nas_stream_eia2.c

@@ -76,17 +76,8 @@ int nas_stream_encrypt_eia2(nas_stream_cipher_t *stream_cipher, uint8_t out[4])
 
     cmac_ctx = CMAC_CTX_new();
     ret = CMAC_Init(cmac_ctx, stream_cipher->key, stream_cipher->key_length, cipher, NULL);
-#if defined(SECU_DEBUG)
-    printf("CMAC_Init returned %d\n", ret);
-#endif
     ret = CMAC_Update(cmac_ctx, m, m_length + 8);
-#if defined(SECU_DEBUG)
-    printf("CMAC_Update returned %d\n", ret);
-#endif
     CMAC_Final(cmac_ctx, data, &size);
-#if defined(SECU_DEBUG)
-    printf("CMAC_Final returned %d, size = %u\n", ret, size);
-#endif
     CMAC_CTX_free(cmac_ctx);
 
 #if defined(SECU_DEBUG)