commit 3ac6a2b9a769d451cd7bcf5eb138839dbfca1f16
author Manuel Pégourié-Gonnard <mpg@elzevir.fr> Wed May 28 22:04:25 2014 +0200
committer Paul Bakker <p.j.bakker@polarssl.org> Thu Jun 12 21:16:02 2014 +0200
tree e52ba898d055c0ca2ed2307b483ab5a3177a8d6f
parent afd5a08e33b51b8f86d4ecf980a92238daeecce7

Same as previous commit with Camellia

library/camellia.c

diff --git a/library/camellia.c b/library/camellia.c
index 524e6ff..306be61 100644
--- a/library/camellia.c
+++ b/library/camellia.c
@@ -435,20 +435,14 @@
     uint32_t *SK;
     int ret;
 
-    switch( keysize )
-    {
-        case 128: ctx->nr = 3; idx = 0; break;
-        case 192:
-        case 256: ctx->nr = 4; idx = 1; break;
-        default : return( POLARSSL_ERR_CAMELLIA_INVALID_KEY_LENGTH );
-    }
-
-    RK = ctx->rk;
-
-    ret = camellia_setkey_enc(&cty, key, keysize);
-    if( ret != 0 )
+    /* Also checks keysize */
+    if( ( ret = camellia_setkey_enc(&cty, key, keysize) ) )
         return( ret );
 
+    ctx->nr = cty.nr;
+    idx = ( ctx->nr == 4 );
+
+    RK = ctx->rk;
     SK = cty.rk + 24 * 2 + 8 * idx * 2;
 
     *RK++ = *SK++;