128Bit AES cypher decypher

128 bit block에 대해서만 동작하는 C로 작성된 코드.

Wikipedia와 이만영 교수님의 Internet security 참고.

컴파일 및 테스트 visual studio 2008에서 수행.


 

#include "stdafx.h" typedef unsigned char uint8_t; typedef char int8_t; typedef unsigned short uint16_t; typedef short int16_t; typedef unsigned int uint32_t; typedef int int32_t; typedef unsigned long long uint64_t; typedef long long int64_t; uint8_t AES_SBOX[] = {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,0xb8,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,0xdb,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}; uint8_t AES_INVERSE_SBOX[] = {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 }; uint8_t AES_MULTIPLY_TABLE_BY_9[] = { 0x00,0x09,0x12,0x1b,0x24,0x2d,0x36,0x3f,0x48,0x41,0x5a,0x53,0x6c,0x65,0x7e,0x77, 0x90,0x99,0x82,0x8b,0xb4,0xbd,0xa6,0xaf,0xd8,0xd1,0xca,0xc3,0xfc,0xf5,0xee,0xe7, 0x3b,0x32,0x29,0x20,0x1f,0x16,0x0d,0x04,0x73,0x7a,0x61,0x68,0x57,0x5e,0x45,0x4c, 0xab,0xa2,0xb9,0xb0,0x8f,0x86,0x9d,0x94,0xe3,0xea,0xf1,0xf8,0xc7,0xce,0xd5,0xdc, 0x76,0x7f,0x64,0x6d,0x52,0x5b,0x40,0x49,0x3e,0x37,0x2c,0x25,0x1a,0x13,0x08,0x01, 0xe6,0xef,0xf4,0xfd,0xc2,0xcb,0xd0,0xd9,0xae,0xa7,0xbc,0xb5,0x8a,0x83,0x98,0x91, 0x4d,0x44,0x5f,0x56,0x69,0x60,0x7b,0x72,0x05,0x0c,0x17,0x1e,0x21,0x28,0x33,0x3a, 0xdd,0xd4,0xcf,0xc6,0xf9,0xf0,0xeb,0xe2,0x95,0x9c,0x87,0x8e,0xb1,0xb8,0xa3,0xaa, 0xec,0xe5,0xfe,0xf7,0xc8,0xc1,0xda,0xd3,0xa4,0xad,0xb6,0xbf,0x80,0x89,0x92,0x9b, 0x7c,0x75,0x6e,0x67,0x58,0x51,0x4a,0x43,0x34,0x3d,0x26,0x2f,0x10,0x19,0x02,0x0b, 0xd7,0xde,0xc5,0xcc,0xf3,0xfa,0xe1,0xe8,0x9f,0x96,0x8d,0x84,0xbb,0xb2,0xa9,0xa0, 0x47,0x4e,0x55,0x5c,0x63,0x6a,0x71,0x78,0x0f,0x06,0x1d,0x14,0x2b,0x22,0x39,0x30, 0x9a,0x93,0x88,0x81,0xbe,0xb7,0xac,0xa5,0xd2,0xdb,0xc0,0xc9,0xf6,0xff,0xe4,0xed, 0x0a,0x03,0x18,0x11,0x2e,0x27,0x3c,0x35,0x42,0x4b,0x50,0x59,0x66,0x6f,0x74,0x7d, 0xa1,0xa8,0xb3,0xba,0x85,0x8c,0x97,0x9e,0xe9,0xe0,0xfb,0xf2,0xcd,0xc4,0xdf,0xd6, 0x31,0x38,0x23,0x2a,0x15,0x1c,0x07,0x0e,0x79,0x70,0x6b,0x62,0x5d,0x54,0x4f,0x46 }; uint8_t AES_MULTIPLY_TABLE_BY_11[] = { 0x00,0x0b,0x16,0x1d,0x2c,0x27,0x3a,0x31,0x58,0x53,0x4e,0x45,0x74,0x7f,0x62,0x69, 0xb0,0xbb,0xa6,0xad,0x9c,0x97,0x8a,0x81,0xe8,0xe3,0xfe,0xf5,0xc4,0xcf,0xd2,0xd9, 0x7b,0x70,0x6d,0x66,0x57,0x5c,0x41,0x4a,0x23,0x28,0x35,0x3e,0x0f,0x04,0x19,0x12, 0xcb,0xc0,0xdd,0xd6,0xe7,0xec,0xf1,0xfa,0x93,0x98,0x85,0x8e,0xbf,0xb4,0xa9,0xa2, 0xf6,0xfd,0xe0,0xeb,0xda,0xd1,0xcc,0xc7,0xae,0xa5,0xb8,0xb3,0x82,0x89,0x94,0x9f, 0x46,0x4d,0x50,0x5b,0x6a,0x61,0x7c,0x77,0x1e,0x15,0x08,0x03,0x32,0x39,0x24,0x2f, 0x8d,0x86,0x9b,0x90,0xa1,0xaa,0xb7,0xbc,0xd5,0xde,0xc3,0xc8,0xf9,0xf2,0xef,0xe4, 0x3d,0x36,0x2b,0x20,0x11,0x1a,0x07,0x0c,0x65,0x6e,0x73,0x78,0x49,0x42,0x5f,0x54, 0xf7,0xfc,0xe1,0xea,0xdb,0xd0,0xcd,0xc6,0xaf,0xa4,0xb9,0xb2,0x83,0x88,0x95,0x9e, 0x47,0x4c,0x51,0x5a,0x6b,0x60,0x7d,0x76,0x1f,0x14,0x09,0x02,0x33,0x38,0x25,0x2e, 0x8c,0x87,0x9a,0x91,0xa0,0xab,0xb6,0xbd,0xd4,0xdf,0xc2,0xc9,0xf8,0xf3,0xee,0xe5, 0x3c,0x37,0x2a,0x21,0x10,0x1b,0x06,0x0d,0x64,0x6f,0x72,0x79,0x48,0x43,0x5e,0x55, 0x01,0x0a,0x17,0x1c,0x2d,0x26,0x3b,0x30,0x59,0x52,0x4f,0x44,0x75,0x7e,0x63,0x68, 0xb1,0xba,0xa7,0xac,0x9d,0x96,0x8b,0x80,0xe9,0xe2,0xff,0xf4,0xc5,0xce,0xd3,0xd8, 0x7a,0x71,0x6c,0x67,0x56,0x5d,0x40,0x4b,0x22,0x29,0x34,0x3f,0x0e,0x05,0x18,0x13, 0xca,0xc1,0xdc,0xd7,0xe6,0xed,0xf0,0xfb,0x92,0x99,0x84,0x8f,0xbe,0xb5,0xa8,0xa3 }; uint8_t AES_MULTIPLY_TABLE_BY_13[] = { 0x00,0x0d,0x1a,0x17,0x34,0x39,0x2e,0x23,0x68,0x65,0x72,0x7f,0x5c,0x51,0x46,0x4b, 0xd0,0xdd,0xca,0xc7,0xe4,0xe9,0xfe,0xf3,0xb8,0xb5,0xa2,0xaf,0x8c,0x81,0x96,0x9b, 0xbb,0xb6,0xa1,0xac,0x8f,0x82,0x95,0x98,0xd3,0xde,0xc9,0xc4,0xe7,0xea,0xfd,0xf0, 0x6b,0x66,0x71,0x7c,0x5f,0x52,0x45,0x48,0x03,0x0e,0x19,0x14,0x37,0x3a,0x2d,0x20, 0x6d,0x60,0x77,0x7a,0x59,0x54,0x43,0x4e,0x05,0x08,0x1f,0x12,0x31,0x3c,0x2b,0x26, 0xbd,0xb0,0xa7,0xaa,0x89,0x84,0x93,0x9e,0xd5,0xd8,0xcf,0xc2,0xe1,0xec,0xfb,0xf6, 0xd6,0xdb,0xcc,0xc1,0xe2,0xef,0xf8,0xf5,0xbe,0xb3,0xa4,0xa9,0x8a,0x87,0x90,0x9d, 0x06,0x0b,0x1c,0x11,0x32,0x3f,0x28,0x25,0x6e,0x63,0x74,0x79,0x5a,0x57,0x40,0x4d, 0xda,0xd7,0xc0,0xcd,0xee,0xe3,0xf4,0xf9,0xb2,0xbf,0xa8,0xa5,0x86,0x8b,0x9c,0x91, 0x0a,0x07,0x10,0x1d,0x3e,0x33,0x24,0x29,0x62,0x6f,0x78,0x75,0x56,0x5b,0x4c,0x41, 0x61,0x6c,0x7b,0x76,0x55,0x58,0x4f,0x42,0x09,0x04,0x13,0x1e,0x3d,0x30,0x27,0x2a, 0xb1,0xbc,0xab,0xa6,0x85,0x88,0x9f,0x92,0xd9,0xd4,0xc3,0xce,0xed,0xe0,0xf7,0xfa, 0xb7,0xba,0xad,0xa0,0x83,0x8e,0x99,0x94,0xdf,0xd2,0xc5,0xc8,0xeb,0xe6,0xf1,0xfc, 0x67,0x6a,0x7d,0x70,0x53,0x5e,0x49,0x44,0x0f,0x02,0x15,0x18,0x3b,0x36,0x21,0x2c, 0x0c,0x01,0x16,0x1b,0x38,0x35,0x22,0x2f,0x64,0x69,0x7e,0x73,0x50,0x5d,0x4a,0x47, 0xdc,0xd1,0xc6,0xcb,0xe8,0xe5,0xf2,0xff,0xb4,0xb9,0xae,0xa3,0x80,0x8d,0x9a,0x97 }; uint8_t AES_MULTIPLY_TABLE_BY_14[] = { 0x00,0x0e,0x1c,0x12,0x38,0x36,0x24,0x2a,0x70,0x7e,0x6c,0x62,0x48,0x46,0x54,0x5a, 0xe0,0xee,0xfc,0xf2,0xd8,0xd6,0xc4,0xca,0x90,0x9e,0x8c,0x82,0xa8,0xa6,0xb4,0xba, 0xdb,0xd5,0xc7,0xc9,0xe3,0xed,0xff,0xf1,0xab,0xa5,0xb7,0xb9,0x93,0x9d,0x8f,0x81, 0x3b,0x35,0x27,0x29,0x03,0x0d,0x1f,0x11,0x4b,0x45,0x57,0x59,0x73,0x7d,0x6f,0x61, 0xad,0xa3,0xb1,0xbf,0x95,0x9b,0x89,0x87,0xdd,0xd3,0xc1,0xcf,0xe5,0xeb,0xf9,0xf7, 0x4d,0x43,0x51,0x5f,0x75,0x7b,0x69,0x67,0x3d,0x33,0x21,0x2f,0x05,0x0b,0x19,0x17, 0x76,0x78,0x6a,0x64,0x4e,0x40,0x52,0x5c,0x06,0x08,0x1a,0x14,0x3e,0x30,0x22,0x2c, 0x96,0x98,0x8a,0x84,0xae,0xa0,0xb2,0xbc,0xe6,0xe8,0xfa,0xf4,0xde,0xd0,0xc2,0xcc, 0x41,0x4f,0x5d,0x53,0x79,0x77,0x65,0x6b,0x31,0x3f,0x2d,0x23,0x09,0x07,0x15,0x1b, 0xa1,0xaf,0xbd,0xb3,0x99,0x97,0x85,0x8b,0xd1,0xdf,0xcd,0xc3,0xe9,0xe7,0xf5,0xfb, 0x9a,0x94,0x86,0x88,0xa2,0xac,0xbe,0xb0,0xea,0xe4,0xf6,0xf8,0xd2,0xdc,0xce,0xc0, 0x7a,0x74,0x66,0x68,0x42,0x4c,0x5e,0x50,0x0a,0x04,0x16,0x18,0x32,0x3c,0x2e,0x20, 0xec,0xe2,0xf0,0xfe,0xd4,0xda,0xc8,0xc6,0x9c,0x92,0x80,0x8e,0xa4,0xaa,0xb8,0xb6, 0x0c,0x02,0x10,0x1e,0x34,0x3a,0x28,0x26,0x7c,0x72,0x60,0x6e,0x44,0x4a,0x58,0x56, 0x37,0x39,0x2b,0x25,0x0f,0x01,0x13,0x1d,0x47,0x49,0x5b,0x55,0x7f,0x71,0x63,0x6d, 0xd7,0xd9,0xcb,0xc5,0xef,0xe1,0xf3,0xfd,0xa7,0xa9,0xbb,0xb5,0x9f,0x91,0x83,0x8d }; uint32_t Rcon[13+1] = {0}; uint32_t w[43+1] = {0}; // key schedule words void init_Rcon(void) { Rcon[1] = 0x01000000; Rcon[2] = 0x02000000; Rcon[3] = 0x04000000; Rcon[4] = 0x08000000; Rcon[5] = 0x10000000; Rcon[6] = 0x20000000; Rcon[7] = 0x40000000; Rcon[8] = 0x80000000; Rcon[9] = 0x1b000000; Rcon[10] = 0x36000000; Rcon[11] = 0x6c000000; Rcon[12] = 0xd8000000; Rcon[13] = 0xab000000; } uint32_t RotWord(uint32_t w) { return (w << 8) | (w >> 24); } uint32_t SubWord(uint32_t w) { uint32_t index[8]; uint32_t result; uint8_t i; for(i=0; i < 8; i++) { index[i] = (w >> (28 - 4*i)) & 0xf; } result = (uint32_t)AES_SBOX[index[0]*16 + index[1]] << 24; result |= (uint32_t)AES_SBOX[index[2]*16 + index[3]] << 16; result |= (uint32_t)AES_SBOX[index[4]*16 + index[5]] << 8; result |= (uint32_t)AES_SBOX[index[6]*16 + index[7]]; return result; } #define MAKE_WORD(a,b,c,d) ((uint32_t)a<<24 | (uint32_t)b<<16 | (uint32_t)c << 8 | (uint32_t)d) void Key_Expansion(uint8_t *key, uint32_t *w, uint32_t Nb, uint32_t Nk, uint32_t Nr) { uint32_t i = 0; uint32_t temp = 0; while(i < Nk) { w[i] = MAKE_WORD(key[4*i], key[4*i+1], key[4*i+2], key[4*i+3]); i++; } printf("-----------------------------------------------------------------------------------------------\n"); printf("i Temp After After Rcon[i/Nk] After XOR w[i] = temp\n" " RotWord SubWord with Rcon XOR w[i-Nk]\n"); printf("-----------------------------------------------------------------------------------------------\n"); for(i=0; i < Nk; i++) { printf("%2u %08X\n", i, w[i]); } i=Nk; while(i < Nb*(Nr+1)) { temp = w[i-1]; if((i % Nk) == 0) { temp = SubWord(RotWord(temp)) ^ Rcon[i/Nk]; printf("%2u %08X %08X %08X %08X %08X ", i, w[i-1], RotWord(temp), SubWord(RotWord(temp)), Rcon[i/Nk], SubWord(RotWord(temp)) ^ Rcon[i/Nk]); } else if(Nk == 8 && (i%Nk) == 4) { temp = SubWord(temp); printf("%2u ",i); } else { printf("%2u ",i); } w[i] = w[i-Nk] ^ temp; printf("%08X\n", w[i]); i++; } printf("-----------------------------------------------------------------------------------------------\n"); } void SubBytes(uint8_t state[4][4]) { uint8_t s; uint8_t r, c; uint8_t i; for(i=0; i < 16; i++) { s = *((uint8_t *)state + i); r = s >> 4; c = s & 0xf; *((uint8_t *)state + i) = AES_SBOX[r*16+c]; } } void ShiftRows(uint8_t state[4][4]) { uint32_t s; s = (uint32_t)state[1][0] << 24 | state[1][1] << 16 | state[1][2] << 8 | state[1][3]; s = (s << 8) | (s >> 24); state[1][0] = s >> 24; state[1][1] = (s >> 16)&0xff; state[1][2] = (s >> 8)&0xff; state[1][3] = s&0xff; s = (uint32_t)state[2][0] << 24 | state[2][1] << 16 | state[2][2] << 8 | state[2][3]; s = (s << 16) | (s >> 16); state[2][0] = s >> 24; state[2][1] = (s >> 16)&0xff; state[2][2] = (s >> 8)&0xff; state[2][3] = s&0xff; s = (uint32_t)state[3][0] << 24 | state[3][1] << 16 | state[3][2] << 8 | state[3][3]; s = (s << 24) | (s >> 8); state[3][0] = s >> 24; state[3][1] = (s >> 16)&0xff; state[3][2] = (s >> 8)&0xff; state[3][3] = s&0xff; } void MixColumns(uint8_t state[4][4]) { uint8_t c; uint8_t result1, result2; uint8_t result[4][4]; for(c = 0; c < 4; c++) { if(state[0][c] & 0x80) result1 = (state[0][c] << 1) ^ 0x1b; else result1 = state[0][c] << 1; if(state[1][c] & 0x80) { result2 = (state[1][c] << 1) ^ 0x1b; result2 = result2 ^ state[1][c]; } else { result2 = state[1][c] << 1; result2 = result2 ^ state[1][c]; } result[0][c] = result1 ^ result2 ^ state[2][c] ^ state[3][c]; if(state[1][c] & 0x80) result1 = (state[1][c] << 1) ^ 0x1b; else result1 = state[1][c] << 1; if(state[2][c] & 0x80) { result2 = (state[2][c] << 1) ^ 0x1b; result2 = result2 ^ state[2][c]; } else { result2 = state[2][c] << 1; result2 = result2 ^ state[2][c]; } result[1][c] = state[0][c] ^ result1 ^ result2 ^ state[3][c]; if(state[2][c] & 0x80) result1 = (state[2][c] << 1) ^ 0x1b; else result1 = state[2][c] << 1; if(state[3][c] & 0x80) { result2 = (state[3][c] << 1) ^ 0x1b; result2 = result2 ^ state[3][c]; } else { result2 = state[3][c] << 1; result2 = result2 ^ state[3][c]; } result[2][c] = state[0][c] ^ state[1][c] ^ result1 ^ result2; if(state[3][c] & 0x80) result1 = (state[3][c] << 1) ^ 0x1b; else result1 = state[3][c] << 1; if(state[0][c] & 0x80) { result2 = (state[0][c] << 1) ^ 0x1b; result2 = result2 ^ state[0][c]; } else { result2 = state[0][c] << 1; result2 = result2 ^ state[0][c]; } result[3][c] = result2 ^ state[1][c] ^ state[2][c] ^ result1; } for(c = 0; c < 4; c++) { state[0][c] = result[0][c]; state[1][c] = result[1][c]; state[2][c] = result[2][c]; state[3][c] = result[3][c]; } } void AddRoundKey(uint8_t state[4][4], uint32_t *w, uint32_t round, uint32_t Nb, uint32_t c) { uint32_t i; for(i=0; i < Nb; i++) { state[i][c] = state[i][c] ^ ((w[round*Nb+c] >> (24 - 8*i))&0xff); } } void Cypher(uint8_t *in, uint8_t *out, uint32_t *w, uint32_t Nb, uint32_t Nr) { uint32_t r = 0; // round uint8_t state[4][4]; uint8_t start_of_round[4][4]; uint8_t after_subbytes[4][4]; uint8_t after_shiftrows[4][4]; uint8_t after_mixcolumns[4][4]; uint8_t i,j; for(i=0; i < Nb; i++) { for(j=0; j < Nb; j++) { state[j][i] = in[i*4+j]; } } for(i=0; i < 16; i++) *((uint8_t *)start_of_round + i) = *((uint8_t *)state + i); for(i=0; i < Nb; i++) AddRoundKey(state, w, r, Nb, i); printf("-----------------------------------------------------------------------------\n"); printf("r Start of Round After After After After XOR \n" " SubByte ShiftRows MixColumns with w[]\n"); printf("-----------------------------------------------------------------------------\n"); for(i=0; i < Nb; i++) { if(i == 0) printf("%2u ", r); else printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", start_of_round[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", state[i][j]); } printf("\n"); } for(r = 1; r < Nr; r++) { printf("-----------------------------------------------------------------------------\n"); for(i=0; i < 16; i++) *((uint8_t *)start_of_round + i) = *((uint8_t *)state + i); SubBytes(state); for(i=0; i < 16; i++) *((uint8_t *)after_subbytes + i) = *((uint8_t *)state + i); ShiftRows(state); for(i=0; i < 16; i++) *((uint8_t *)after_shiftrows + i) = *((uint8_t *)state + i); MixColumns(state); for(i=0; i < 16; i++) *((uint8_t *)after_mixcolumns + i) = *((uint8_t *)state + i); for(i=0; i < Nb; i++) AddRoundKey(state, w, r, Nb, i); for(i=0; i < Nb; i++) { if(i == 0) printf("%2u ", r); else printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", start_of_round[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_subbytes[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_shiftrows[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_mixcolumns[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", state[i][j]); } printf("\n"); } } printf("-----------------------------------------------------------------------------\n"); for(i=0; i < 16; i++) *((uint8_t *)start_of_round + i) = *((uint8_t *)state + i); SubBytes(state); for(i=0; i < 16; i++) *((uint8_t *)after_subbytes + i) = *((uint8_t *)state + i); ShiftRows(state); for(i=0; i < 16; i++) *((uint8_t *)after_shiftrows + i) = *((uint8_t *)state + i); for(i=0; i < Nb; i++) AddRoundKey(state, w, r, Nb, i); for(i=0; i < Nb; i++) { if(i == 0) printf("%2u ", r); else printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", start_of_round[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_subbytes[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_shiftrows[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", state[i][j]); } printf("\n"); } printf("-----------------------------------------------------------------------------\n"); for(i=0; i < Nb; i++) { for(j=0; j < Nb; j++) { out[i*4+j] = state[j][i]; } } } void InvShiftRows(uint8_t state[4][4]) { uint32_t s; s = (uint32_t)state[1][0] << 24 | state[1][1] << 16 | state[1][2] << 8 | state[1][3]; s = (s << 24) | (s >> 8); state[1][0] = s >> 24; state[1][1] = (s >> 16)&0xff; state[1][2] = (s >> 8)&0xff; state[1][3] = s&0xff; s = (uint32_t)state[2][0] << 24 | state[2][1] << 16 | state[2][2] << 8 | state[2][3]; s = (s << 16) | (s >> 16); state[2][0] = s >> 24; state[2][1] = (s >> 16)&0xff; state[2][2] = (s >> 8)&0xff; state[2][3] = s&0xff; s = (uint32_t)state[3][0] << 24 | state[3][1] << 16 | state[3][2] << 8 | state[3][3]; s = (s << 8) | (s >> 24); state[3][0] = s >> 24; state[3][1] = (s >> 16)&0xff; state[3][2] = (s >> 8)&0xff; state[3][3] = s&0xff; } void InvSubBytes(uint8_t state[4][4]) { uint8_t s; uint8_t r, c; uint8_t i; for(i=0; i < 16; i++) { s = *((uint8_t *)state + i); } for(i=0; i < 16; i++) { s = *((uint8_t *)state + i); r = s >> 4; c = s & 0xf; *((uint8_t *)state + i) = AES_INVERSE_SBOX[r*16+c]; } } void InvMixColumns(uint8_t state[4][4]) { uint8_t c; uint8_t result1, result2, result3, result4; uint8_t result[4][4]; for(c = 0; c < 4; c++) { result1 = AES_MULTIPLY_TABLE_BY_14[state[0][c]]; result2 = AES_MULTIPLY_TABLE_BY_11[state[1][c]]; result3 = AES_MULTIPLY_TABLE_BY_13[state[2][c]]; result4 = AES_MULTIPLY_TABLE_BY_9 [state[3][c]]; result[0][c] = result1 ^ result2 ^ result3 ^ result4; result1 = AES_MULTIPLY_TABLE_BY_9 [state[0][c]]; result2 = AES_MULTIPLY_TABLE_BY_14[state[1][c]]; result3 = AES_MULTIPLY_TABLE_BY_11[state[2][c]]; result4 = AES_MULTIPLY_TABLE_BY_13[state[3][c]]; result[1][c] = result1 ^ result2 ^ result3 ^ result4; result1 = AES_MULTIPLY_TABLE_BY_13[state[0][c]]; result2 = AES_MULTIPLY_TABLE_BY_9 [state[1][c]]; result3 = AES_MULTIPLY_TABLE_BY_14[state[2][c]]; result4 = AES_MULTIPLY_TABLE_BY_11[state[3][c]]; result[2][c] = result1 ^ result2 ^ result3 ^ result4; result1 = AES_MULTIPLY_TABLE_BY_11[state[0][c]]; result2 = AES_MULTIPLY_TABLE_BY_13[state[1][c]]; result3 = AES_MULTIPLY_TABLE_BY_9 [state[2][c]]; result4 = AES_MULTIPLY_TABLE_BY_14[state[3][c]]; result[3][c] = result1 ^ result2 ^ result3 ^ result4; } for(c = 0; c < 4; c++) { state[0][c] = result[0][c]; state[1][c] = result[1][c]; state[2][c] = result[2][c]; state[3][c] = result[3][c]; } } void DeCypher(uint8_t *in, uint8_t *out, uint32_t *w, uint32_t Nb, uint32_t Nr) { uint32_t r = 0, nr = 0; // round uint8_t state[4][4]; uint8_t start_of_round[4][4]; uint8_t after_subbytes[4][4]; uint8_t after_shiftrows[4][4]; uint8_t after_addroundkeys[4][4]; uint8_t i,j; for(i=0; i < Nb; i++) { for(j=0; j < Nb; j++) { state[j][i] = in[i*4+j]; } } for(i=0; i < 16; i++) *((uint8_t *)start_of_round + i) = *((uint8_t *)state + i); r = 10; nr = 0; for(i=0; i < Nb; i++) AddRoundKey(state, w, r, Nb, i); printf("-----------------------------------------------------------------------------\n"); printf("r Start of Round After After After XOR After \n" " InvShiftRows InvSubBytes with w[] InvMixColumns\n"); printf("-----------------------------------------------------------------------------\n"); for(i=0; i < Nb; i++) { if(i == 0) printf("%2u ", nr); else printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", start_of_round[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", state[i][j]); } printf("\n"); } for(r = 9, nr = 1; r > 0; r--, nr++) { printf("-----------------------------------------------------------------------------\n"); for(i=0; i < 16; i++) *((uint8_t *)start_of_round + i) = *((uint8_t *)state + i); InvShiftRows(state); for(i=0; i < 16; i++) *((uint8_t *)after_shiftrows + i) = *((uint8_t *)state + i); InvSubBytes(state); for(i=0; i < 16; i++) *((uint8_t *)after_subbytes + i) = *((uint8_t *)state + i); for(i=0; i < Nb; i++) AddRoundKey(state, w, r, Nb, i); for(i=0; i < 16; i++) *((uint8_t *)after_addroundkeys + i) = *((uint8_t *)state + i); InvMixColumns(state); for(i=0; i < Nb; i++) { if(i == 0) printf("%2u ", nr); else printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", start_of_round[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_shiftrows[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_subbytes[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_addroundkeys[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", state[i][j]); } printf("\n"); } } printf("-----------------------------------------------------------------------------\n"); for(i=0; i < 16; i++) *((uint8_t *)start_of_round + i) = *((uint8_t *)state + i); InvShiftRows(state); for(i=0; i < 16; i++) *((uint8_t *)after_shiftrows + i) = *((uint8_t *)state + i); InvSubBytes(state); for(i=0; i < 16; i++) *((uint8_t *)after_subbytes + i) = *((uint8_t *)state + i); for(i=0; i < Nb; i++) AddRoundKey(state, w, r, Nb, i); for(i=0; i < 16; i++) *((uint8_t *)after_addroundkeys + i) = *((uint8_t *)state + i); for(i=0; i < Nb; i++) { if(i == 0) printf("%2u ", nr); else printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", start_of_round[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_shiftrows[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_subbytes[i][j]); } printf(" "); for(j=0; j < Nb; j++) { printf("%02X ", after_addroundkeys[i][j]); } printf("\n"); } printf("-----------------------------------------------------------------------------\n"); for(i=0; i < Nb; i++) { for(j=0; j < Nb; j++) { out[i*4+j] = state[j][i]; } } } int _tmain(int argc, _TCHAR* argv[]) { uint8_t i = 0; uint8_t PlainText[16] = {0xa3, 0xc5, 0x08, 0x08, 0x78, 0xa4, 0xff, 0xd3, 0x00, 0xff, 0x36, 0x36, 0x28, 0x5f, 0x01, 0x02}; uint8_t CypherKey[16] = {0x36, 0x8a, 0xc0, 0xf4, 0xed, 0xcf, 0x76, 0xa6, 0x08, 0xa3, 0xb6, 0x78, 0x31, 0x31, 0x27, 0x6e}; uint8_t CypherText[16], DecypherText[16]; uint32_t Nb = 4; uint32_t Nk = 4; uint32_t Nr = 10; printf("PlainText: "); for(i=0; i < 16; i++) printf("%02X ", PlainText[i]); printf("\n"); printf("CypherKey: "); for(i=0; i < 16; i++) printf("%02X ", CypherKey[i]); printf("\n\n"); init_Rcon(); Key_Expansion(CypherKey, w, Nb, Nk, Nr); printf("\n\n"); Cypher(PlainText, CypherText, w, Nb, Nr); printf("\n\n"); printf("CypherText = "); for(i=0; i < 16; i++) printf("%02X ", CypherText[i]); printf("\n\n"); printf("CypherKey = "); for(i=0; i < 16; i++) printf("%02X ", CypherKey[i]); printf("\n\n"); DeCypher(CypherText, DecypherText, w, Nb, Nr); printf("Inverse Cypher Text = "); for(i=0; i < 16; i++) printf("%02X ", DecypherText[i]); printf("\n\n"); printf(" Original PlainText = "); for(i=0; i < 16; i++) printf("%02X ", PlainText[i]); printf("\n"); return 0; }