/*
 明文分組長度為192  狀態烈數為6

*/

#include<stdio.h>
#include<math.h>
#define Nb  6
#define Nk  6
#define Nr  12
unsigned char Rc[12]={0x01,0x02,0x04,0x08,0x10,0x20,0x40,0x80,0x1b,0x36,0x6c,0xd8};
unsigned char Rcon[12][4]={
                            0x01,0x00,0x00,0x00,
                            0x02,0x00,0x00,0x00,
                            0x04,0x00,0x00,0x00,
                            0x08,0x00,0x00,0x00,
                            0x10,0x00,0x00,0x00,
                            0x20,0x00,0x00,0x00,
                            0x40,0x00,0x00,0x00,
                            0x80,0x00,0x00,0x00,
                            0x1b,0x00,0x00,0x00,
                            0x36,0x00,0x00,0x00,
                            0x6c,0x00,0x00,0x00,
                            0xd8,0x00,0x00,0x00
                          };
unsigned char S_Encipher[0x10][0x10]=
{
    {0x63,0x7c,0x77,0x7b,0xf2,0x6b,0x6f,0xc5,0x30,0x01,0x67,0x2b,0xfe,0xd7,0xab,0x76},
    {0xca,0x82,0xc9,0x7d,0xfa,0x59,0x47,0xf0,0xad,0xd4,0xa2,0xaf,0x9c,0xa4,0x72,0xc0},
    {0xb7,0xfd,0x93,0x26,0x36,0x3f,0xf7,0xcc,0x34,0xa5,0xe5,0xf1,0x71,0xd8,0x31,0x15},
    {0x04,0xc7,0x23,0xc3,0x18,0x96,0x05,0x9a,0x07,0x12,0x80,0xe2,0xeb,0x27,0xb2,0x75},
    {0x09,0x83,0x2c,0x1a,0x1b,0x6e,0x5a,0xa0,0x52,0x3b,0xd6,0xb3,0x29,0xe3,0x2f,0x84},
    {0x53,0xd1,0x00,0xed,0x20,0xfc,0xb1,0x5b,0x6a,0xcb,0xbe,0x39,0x4a,0x4c,0x58,0xcf},
    {0xd0,0xef,0xaa,0xfb,0x43,0x4d,0x33,0x85,0x45,0xf9,0x02,0x7f,0x50,0x3c,0x9f,0xa8},
    {0x51,0xa3,0x40,0x8f,0x92,0x9d,0x38,0xf5,0xbc,0xb6,0xda,0x21,0x10,0xff,0xf3,0xd2},
    {0xcd,0x0c,0x13,0xec,0x5f,0x97,0x44,0x17,0xc4,0xa7,0x7e,0x3d,0x64,0x5d,0x19,0x73},
    {0x60,0x81,0x4f,0xdc,0x22,0x2a,0x90,0x88,0x46,0xee,0xb9,0x14,0xde,0x5e,0x0b,0xdb},
    {0xe0,0x32,0x3a,0x0a,0x49,0x06,0x24,0x5c,0xc2,0xd3,0xac,0x62,0x91,0x95,0xe4,0x79},
    {0xe7,0xc8,0x37,0x6d,0x8d,0xd5,0x4e,0xa9,0x6c,0x56,0xf4,0xea,0x65,0x7a,0xae,0x08},
    {0xba,0x78,0x25,0x2e,0x1c,0xa6,0xb4,0xc6,0xe8,0xdd,0x74,0x1f,0x4b,0xbd,0x8b,0x8a},
    {0x70,0x3e,0xb5,0x66,0x48,0x03,0xf6,0x0e,0x61,0x35,0x57,0xb9,0x86,0xc1,0x1d,0x9e},
    {0xe1,0xf8,0x98,0x11,0x69,0xd9,0x8e,0x94,0x9b,0x1e,0x87,0xe9,0xce,0x55,0x28,0xdf},
    {0x8c,0xa1,0x89,0x0d,0xbf,0xe6,0x42,0x68,0x41,0x99,0x2d,0x0f,0xb0,0x54,0xbb,0x16}

};
unsigned char S_InvEncipher[0x10][0x10]=
{
        {0x52,0x09,0x6a,0xd5,0x30,0x36,0xa5,0x38,0xbf,0x40,0xa3,0x9e,0x81,0xf3,0xd7,0xfb},
        {0x7c,0xe3,0x39,0x82,0x9b,0x2f,0xff,0x87,0x34,0x8e,0x43,0x44,0xc4,0xde,0xe9,0xcb},
        {0x54,0x7b,0x94,0x32,0xa6,0xc2,0x23,0x3d,0xee,0x4c,0x95,0x0b,0x42,0xfa,0xc3,0x4e},
        {0x08,0x2e,0xa1,0x66,0x28,0xd9,0x24,0xb2,0x76,0x5b,0xa2,0x49,0x6d,0x8b,0xd1,0x25},
        {0x72,0xf8,0xf6,0x64,0x86,0x68,0x98,0x16,0xd4,0xa4,0x5c,0xcc,0x5d,0x65,0xb6,0x92},
        {0x6c,0x70,0x48,0x50,0xfd,0xed,0xb9,0xda,0x5e,0x15,0x46,0x57,0xa7,0x8d,0x9d,0x84},
        {0x90,0xd8,0xab,0x00,0x8c,0xbc,0xd3,0x0a,0xf7,0xe4,0x58,0x05,0xb8,0xb3,0x45,0x06},
        {0xd0,0x2c,0x1e,0x8f,0xca,0x3f,0x0f,0x02,0xc1,0xaf,0xbd,0x03,0x01,0x13,0x8a,0x6b},
        {0x3a,0x91,0x11,0x41,0x4f,0x67,0xdc,0xea,0x97,0xf2,0xcf,0xce,0xf0,0xb4,0xe6,0x73},
        {0x96,0xac,0x74,0x22,0xe7,0xad,0x35,0x85,0xe2,0xf9,0x37,0xe8,0x1c,0x75,0xdf,0x6e},
        {0x47,0xf1,0x1a,0x71,0x1d,0x29,0xc5,0x89,0x6f,0xb7,0x62,0x0e,0xaa,0x18,0xbe,0x1b},
        {0xfc,0x56,0x3e,0x4b,0xc6,0xd2,0x79,0x20,0x9a,0xdb,0xc0,0xfe,0x78,0xcd,0x5a,0xf4},
        {0x1f,0xdd,0xa8,0x33,0x88,0x07,0xc7,0x31,0xb1,0x12,0x10,0x59,0x27,0x80,0xec,0x5f},
        {0x60,0x51,0x7f,0xa9,0x19,0xb5,0x4a,0x0d,0x2d,0xe5,0x7a,0x9f,0x93,0xc9,0x9c,0xef},
        {0xa0,0xe0,0x3b,0x4d,0xae,0x2a,0xf5,0xb0,0xc8,0xeb,0xbb,0x3c,0x83,0x53,0x99,0x61},
        {0x17,0x2b,0x04,0x7e,0xba,0x77,0xd6,0x26,0xe1,0x69,0x14,0x63,0x55,0x21,0x0c,0x7d}
};
unsigned char MixColumn_Max[4][4]={ 0x02,0x03,0x01,0x01,
                                    0x01,0x02,0x03,0x01,
                                    0x01,0x01,0x02,0x03,
                                    0x03,0x01,0x01,0x02 };//列混合固定矩陣
unsigned char InvMixColumn_Max[4][4]={
                                       0x0e,0x0b,0x0d,0x09,
                                       0x09,0x0e,0x0b,0x0d,
                                       0x0d,0x09,0x0e,0x0b,
                                       0x0b,0x0d,0x09,0x0e};
unsigned char  Key[24];//密鑰
unsigned char  w[312];//擴展密鑰
unsigned char  Invw[312];

////////////////////////////////////
/*
 加密算法
*/

void ByteSub(unsigned char state[4][6])//二維數組字節替換
{
  int i=0,j=0;
  int m=0,n=0;
  for(i=0;i<4;i++)
      for(j=0;j<6;j++)
      {
            n=state[i][j]>>4;//高四位
            m=state[i][j]&0x0f;//低四位
            state[i][j]=S_Encipher[n][m];
      }

}

void ShiftRow(unsigned char state[4][6])//行移位變換  左移
{
  int i,j;
  int n;
  int temp[6];
  
  for(i=1;i<4;i++)
  {
      for(j=i,n=0;n<6;j++,n++)
      {
        temp[n]=state[i][j%6];    
      }
     for(j=0;j<6;j++)
         state[i][j]=temp[j];
  }
}

void sub(unsigned char a[4])//數組乘法
{ 
  int i,j;
  int b[4];
  for(i=0;i<4;i++)
  {
      for(j=0;j<4;j++)
          b[j]^=MixColumn_Max[i][j]*a[j];
  }
  for(i=0;i<4;i++)
      a[i]=b[i];
}

void MixColumn(unsigned char state[4][6])//列混合變換
{
   unsigned char a[4];
   int i,j;
   for(j=0;j<6;j++)
   {
    
       for(i=0;i<4;i++)
       {
           a[i]=state[i][j];
       }
       sub(a);
       for(i=0;i<4;i++)
       {
           state[i][j]=a[i];
       }
       
   } 

} 
//輪密鑰的生成


void RotByte(unsigned char temp[4])
{
   int i;
   unsigned char a[4];
   for(i=0;i<4;i++)
   {
      a[i]=temp[(i+1)%4];
   }
   for(i=0;i<4;i++)
   {
      temp[i]=a[i];
   }
}

void SubByte(unsigned char a[4])
{
    int j,n,m;
   for(j=0;j<4;j++)
      {
            n=a[j]>>4;//高四位
            m=a[j]&0x0f;//低四位
            a[j]=S_Encipher[n][m];
      }


}
//void KeyExpansion(unsigned char Key[24],unsigned char w[312])//密鑰擴展  key[i]是8bit w[32]是32bit
void KeyExpansion(unsigned char state[4][6])
{
   int i,j,m,n;                                              //Nk=6
   unsigned char temp[4];

   for(i=0;i<24;i++)
     w[i]=Key[i];//密鑰數組
   for(i=24;i<312;i+=4)                                 //Nb=6  Nr=12
   { 
      for(j=0;j<4;j++)
          temp[j]=w[i-4+j];
      if(i%24==0)
      {  RotByte(temp);//循環左移一個字節
         SubByte(temp);//對每個字節作代換
         n=i/24;
         for(m=0;m<4;m++)
            temp[m]=temp[m]^Rcon[n][m];
         
      }
      for(j=i;j<i+4;j++)
          w[j]=w[j]^temp[j];   
   }
}

void AddRoundKey(unsigned char state[4][6],int t)//第t輪密鑰加變換  
{
   unsigned char K[4][6];
   int i,j,n=0;
   for(i=t*24;i<4;i++)
       for(j=0;j<6;j++)
       {
           K[i][j]=Key[n];
           n++;
       }
    for(i=0;i<4;i++)
        for(j=0;j<6;j++)
            state[i][j]^=K[i][j];

}

void  Round(unsigned char state[4][6],int i)
{
        ByteSub(state);
        ShiftRow(state);
        MixColumn(state);
        AddRoundKey(state,i);

}
void FinalRound(unsigned char state[4][6],int i)
{
        ByteSub(state);
        ShiftRow(state);
        AddRoundKey(state,i);
}

void  Rijndael(unsigned char state[4][6],unsigned char Cipherkey[4][6])//加密算法
{
    int i;
   KeyExpansion(Cipherkey);
   AddRoundKey(state,0);
   for(i=1;i<12;i++)
       Round(state,i);
   FinalRound(state,12);
}
/////////////////////////////////////////
/*
解密算法
*/
void InvByteSub(unsigned char state[4][6])////字節逆變換
{
    int i=0,j=0;
    int m=0,n=0;
    for(i=0;i<4;i++)
      for(j=0;j<6;j++)
      {
            n=state[i][j]>>4;//高四位
            m=state[i][j]&0x0f;//低四位
            state[i][j]=S_InvEncipher[n][m];
      }
      
}
void InvShiftRow(unsigned char state[4][6])//行移位逆變換   循環右移
{
   int i,j;
   int n;
   int temp[6];
 for(i=1;i<4;i++)
  {
      for(j=0,n=i;j<6;j++,n++)
      {
        temp[n%6]=state[i][j];    
      }
     for(j=0;j<6;j++)
         state[i][j]=temp[j];
  }
}
void Invsub(unsigned char a[4])//數組乘法
{ 
  int i,j;
  int b[4];
  for(i=0;i<4;i++)
  {
      for(j=0;j<4;j++)
          b[j]^=InvMixColumn_Max[i][j]*a[j];
  }
  for(i=0;i<4;i++)
      a[i]=b[i];
}
void InvMixColumn(unsigned char  state[4][6])//列混合逆變換
{
   unsigned char a[4];
   int i,j;
   for(j=0;j<6;j++)
   {
     for(i=0;i<4;i++)
       {
           a[i]=state[i][j];
       }
       Invsub(a);
       for(i=0;i<4;i++)
       {
           state[i][j]=a[i];
       }
   } 
}

void InvRound(unsigned char state[4][6],int i)
{
    InvByteSub(state);
    InvShiftRow(state);
    InvMixColumn(state);
    AddRoundKey(state,i);
}


void InvFinalRound(unsigned char state[4][6],int i)
{
    InvByteSub(state);
    InvShiftRow(state);
    AddRoundKey(state,i);
}



void InvKeyExpansion()
{
    int i,j;
    int n;
    int m;
    unsigned char state[4][6];
    for(i=0;i<312;i++)
       Invw[i]=w[311-i];
    for(m=24;m<312-24;m=m+24)
    {
        for(i=0;i<6;i++)
        for(j=0,n=m;j<4;j++,n++)
            state[j][i]=Invw[n];
        InvMixColumn(state);
        for(i=0;i<6;i++)
        for(j=0,n=m;j<4;j++,n++)
            Invw[n]=state[j][i];
    }


}

//////
void InvRijndael(unsigned char state[4][6])
{
  int i;
  InvKeyExpansion();
  AddRoundKey(state,0);
  for(i=1;i<12;i++)
      InvRound(state,i);
  InvFinalRound(state,i);
}

void main()
{
    int i,j,t=0;
    unsigned char message[4][6];//192bit
    unsigned char K0[4][6];//初始密鑰
    printf("Input your Message:\n");
    for(j=0;j<6;j++)
            for(i=0;i<4;i++)
            	{
                 scanf("%x",&message[i][j]);
                 //printf("input your message!");
            }	
        
        printf("input your cipherkey!");
      for(i=0;i<6;i++)
        for(j=0;j<4;j++)
        {
             scanf("%x",&K0[j][i]);
    //  printf("input your message!\n");
        }
        Rijndael(message,K0);//加密
    printf("\nYour code:\n");

    for(i=0;i<4;i++)
        for(j=0;j<6;j++)
        {
           printf("%x    ",message[i][j]);  
           if(++t%6==0)printf("\n");
        }
    InvRijndael(message);//解密
    printf("\nYour Invcode:\n");
        for(i=0;i<4;i++)
        for(j=0;j<6;j++)
        {
           printf("%x    ",message[i][j]);  
           if(++t%6==0)printf("\n");
        }
}