aes_test.c

嵌入式linux系统下hi3510平台的osd开发源码

#include <stdio.h>
#include "hi_aes_api.h"

/******************************************below is cbc**************************************/
  static  unsigned char key_cbc[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static  unsigned char pplaintext_cbc[2][16]={
                               {0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a},
                               {0x76,0x49,0xab,0xac,0x81,0x19,0xb2,0x46,0xce,0xe9,0x8e,0x9b,0x12,0xe9,0x19,0x7d}
                               };

  static  unsigned char ivin_cbc[2][16]={
                               {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f},
                               {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}     
                               };

   static  unsigned char dout_cbc[2][16]={
                               {0x76,0x49,0xab,0xac,0x81,0x19,0xb2,0x46,0xce,0xe9,0x8e,0x9b,0x12,0xe9,0x19,0x7d},
                               {0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a}
                              };
  static  unsigned char mydout_cbc[2][16];

/*****************************************************below is cfb_8*********************************/
  static  unsigned char key_cfb_8[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static   unsigned char pplaintext_cfb_8[2][16]={
                               {0x6b,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
                               {0x3b,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00} 
                             };
 static  unsigned char ivin_cfb_8[2][16]={
                               {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f},
                               {0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
                               };                                                    
static  unsigned char dout_cfb_8[2][16]={
                               {0x3b,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
                               {0x6b,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}
                              };  
 static  unsigned char mydout_cfb_8[2][16];

/***************************************************below is cfb_1***************************************/
  static  unsigned char key_cfb_1[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static   unsigned char pplaintext_cfb_1[2][16]={
{0x6b,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
{0x68,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}
                             };
 static  unsigned char ivin_cfb_1[2][16]={
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
                               };                                                    
static  unsigned char dout_cfb_1[2][16]={
   {0x68,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00},
   {0x6b,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00}
                              };  
 static  unsigned char mydout_cfb_1[2][16];

/****************************************************below is cfb_128************************************/
 static  unsigned char key_cfb_128[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static   unsigned char pplaintext_cfb_128[2][16]={
 {0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a},
 {0x3b,0x3f,0xd9,0x2e,0xb7,0x2d,0xad,0x20,0x33,0x34,0x49,0xf8,0xe8,0x3c,0xfb,0x4a}
                             };
 static  unsigned char ivin_cfb_128[2][16]={
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
                               };                                                    
static  unsigned char dout_cfb_128[2][16]={
{0x3b,0x3f,0xd9,0x2e,0xb7,0x2d,0xad,0x20,0x33,0x34,0x49,0xf8,0xe8,0x3c,0xfb,0x4a}, 
{0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a} 
                              };  
 static  unsigned char mydout_cfb_128[2][16];


/*****************************************************below is ofb************************************/
static  unsigned char key_ofb[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static   unsigned char pplaintext_ofb[2][16]={
 {0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a},
 {0x3b,0x3f,0xd9,0x2e,0xb7,0x2d,0xad,0x20,0x33,0x34,0x49,0xf8,0xe8,0x3c,0xfb,0x4a}
                             };
 static  unsigned char ivin_ofb[2][16]={
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
                               };                                                    
static  unsigned char dout_ofb[2][16]={
{0x3b,0x3f,0xd9,0x2e,0xb7,0x2d,0xad,0x20,0x33,0x34,0x49,0xf8,0xe8,0x3c,0xfb,0x4a}, 
{0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a} 
                              };  
 static  unsigned char mydout_ofb[2][16];


/*************************************************below is ecb**************************************/
static  unsigned char key_ecb[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static   unsigned char pplaintext_ecb[2][16]={
 {0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a},
 {0x3a,0xd7,0x7b,0xb4,0x0d,0x7a,0x36,0x60,0xa8,0x9e,0xca,0xf3,0x24,0x66,0xef,0x97}
                             };
 static  unsigned char ivin_ecb[2][16]={
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
                               };                                                    
static  unsigned char dout_ecb[2][16]={
{0x3a,0xd7,0x7b,0xb4,0x0d,0x7a,0x36,0x60,0xa8,0x9e,0xca,0xf3,0x24,0x66,0xef,0x97}, 
{0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a} 
 };  
 static  unsigned char mydout_ecb[2][16];

/*************************************************below is ctr**************************************/

static  unsigned char key_ctr[2][16]={
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c},
                               {0x2b,0x7e,0x15,0x16,0x28,0xae,0xd2,0xa6,0xab,0xf7,0x15,0x88,0x09,0xcf,0x4f,0x3c}
                              };
  static   unsigned char pplaintext_ctr[2][16]={
 {0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a},
 {0x87,0x4d,0x61,0x91,0xb6,0x20,0xe3,0x26,0x1b,0xef,0x68,0x64,0x99,0x0d,0xb6,0xce}
                             };
 static  unsigned char ivin_ctr[2][16]={
{0xf0,0xf1,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,0xf9,0xfa,0xfb,0xfc,0xfd,0xfe,0xff},
{0xf0,0xf1,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,0xf9,0xfa,0xfb,0xfc,0xfd,0xfe,0xff}
                               };                                                    
static  unsigned char dout_ctr[2][16]={
{0x87,0x4d,0x61,0x91,0xb6,0x20,0xe3,0x26,0x1b,0xef,0x68,0x64,0x99,0x0d,0xb6,0xce},
{0x6b,0xc1,0xbe,0xe2,0x2e,0x40,0x9f,0x96,0xe9,0x3d,0x7e,0x11,0x73,0x93,0x17,0x2a}
 };  
 static  unsigned char mydout_ctr[2][16];

/***************************************************that's all**************************/




int ctr_test_aes();
int ecb_test_aes();
int ofb_test_aes();
int cfb_128_test_aes();
int cfb_1_test_aes();
int cfb_8_test_aes();
int cbc_test_aes();
struct aes_encrypt_ctrl  ctrl;
//int ofb_test_aes();
    
int main(void)
{ 
    hi_aes_init();
   ofb_test_aes();

   ctr_test_aes();
   ecb_test_aes();
   cfb_128_test_aes();
   cfb_8_test_aes();
   cfb_1_test_aes();
   cbc_test_aes();
   hi_aes_exit();
   return 0;
}

int ctr_test_aes()
{
     unsigned int i,j;
     ctrl.aes_type = 6; //ctr;

     printf(" aes_ctr_test begin!\n"); 
     for(i=0;i<2;i++)
     {
      
          // memcpy(ctrl.iv,&ivin_ctr[i][0],16);

           hi_aes_config(&ctrl);
           //printf("keys1:%llx",*((unsigned long long *)&key_ctr[i][0]));
           if(i<1)
           { 
                aes_ctr_crypt(&pplaintext_ctr[i][0],&mydout_ctr[i][0],16,&key_ctr[i][0],&ivin_ctr[i][0]);
           }
           else
           {
               aes_ctr_decrypt(&pplaintext_ctr[i][0],&mydout_ctr[i][0],16,&key_ctr[i][0],&ivin_ctr[i][0]);
           }
          // printf("output:%llx",*((unsigned long long *)&mydout_ctr[i][0]));
           for(j=0;j<16;j++)
           {
             // printf("%d___:%x\n",i,mydout_ctr[i][j]);
               if((mydout_ctr[i][j])!=(dout_ctr[i][j]))
               {
                   printf("aes_ctr test failed at: %d row!",i);
                   return -1;
               }
              // printf(" pplaintext[i]:%x\n",pplaintext_ctr[i][j]);
           } 
     }    

     printf(" all aes_ctr tests passed  !\n");
     return 0;
}



int ecb_test_aes()
{
     unsigned int i,j;
     ctrl.aes_type = 7; //ecb;

     printf(" aes_ecb_test begin!\n"); 
     for(i=0;i<2;i++)
     {
      
          // memcpy(ctrl.iv,&ivin_ecb[i][0],16);