bertos/sec/cipher/aes.c

   1 /**
   2  * \file
   3  * <!--
   4  * This file is part of BeRTOS.
   5  *
   6  * Bertos is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  * As a special exception, you may use this file as part of a free software
  21  * library without restriction.  Specifically, if other files instantiate
  22  * templates or use macros or inline functions from this file, or you compile
  23  * this file and link it with other files to produce an executable, this
  24  * file does not by itself cause the resulting executable to be covered by
  25  * the GNU General Public License.  This exception does not however
  26  * invalidate any other reasons why the executable file might be covered by
  27  * the GNU General Public License.
  28  *
  29  * Copyright 2006 Develer S.r.l. (http://www.develer.com/)
  30  *
  31  * -->
  32  *
  33  * \brief AES Advanced Encryption Standard implementation
  34  *
  35  * \author Giovanni Bajo <rasky@develer.com>
  36  *
  37  * $WIZ$ module_name = "aes"
  38  */
  39
  40
  41 #include "aes.h"
  42 #include <cfg/compiler.h>
  43 #include <cpu/byteorder.h>
  44
  45 #include <string.h>
  46
  47 // AES only supports Nb=4
  48 #define Nb 4                    // number of columns in the state & expanded key
  49
  50 typedef struct
  51 {
  52         BlockCipher c;
  53         uint8_t num_rounds;
  54         int8_t key_status;
  55         uint8_t _dummy1;
  56         uint8_t _dummy2;
  57         uint8_t expkey[0];
  58 } AES_Context;
  59
  60
  61 #if CPU_REG_BITS == 32
  62
  63 // 32-bit optimized implementation
  64 #include "aes_f32.h"
  65
  66 #else
  67
  68 // Full 8-bit implementation
  69 #include "aes_f8.h"
  70
  71 #endif
  72
  73
  74 /******************************************************************************/
  75
  76 void AES128_init(AES128_Context *aes_)
  77 {
  78         AES_Context *aes = (AES_Context *)aes_;
  79         aes->c.set_key = AES_expandKey;
  80         aes->c.enc_block = AES_encrypt;
  81         aes->c.dec_block = AES_decrypt;
  82         aes->c.block_len = Nb*4;
  83         aes->c.key_len = 16;
  84         aes->num_rounds = 10;
  85 }
  86
  87 void AES192_init(AES192_Context *aes_)
  88 {
  89         AES_Context *aes = (AES_Context *)aes_;
  90         aes->c.set_key = AES_expandKey;
  91         aes->c.enc_block = AES_encrypt;
  92         aes->c.dec_block = AES_decrypt;
  93         aes->c.block_len = Nb*4;
  94         aes->c.key_len = 24;
  95         aes->num_rounds = 12;
  96 }
  97
  98 void AES256_init(AES256_Context *aes_)
  99 {
 100         AES_Context *aes = (AES_Context *)aes_;
 101         aes->c.set_key = AES_expandKey;
 102         aes->c.enc_block = AES_encrypt;
 103         aes->c.dec_block = AES_decrypt;
 104         aes->c.block_len = Nb*4;
 105         aes->c.key_len = 32;
 106         aes->num_rounds = 14;
 107 }