import string
from unittest import result
import numpy as np
from collections import deque


clairMassage = input("Enter you massage here!:")

def getTextReady(clairMassage):
# this functions main role is to take the text and to transform it into chunks of text, meaning to list of 16bytes.
    clairMassage=clairMassage.replace(" ","")
    if len(clairMassage)<16:
        chunks=[]
        tempo=16-len(clairMassage)
        clairMassage=clairMassage+tempo*"0"
        chunks.append(clairMassage)
        return chunks
    elif len(clairMassage)>16:
        n = 16 # chunk length
        chunks = [clairMassage[i:i+n] for i in range(0, len(clairMassage), n)]
        if len(chunks[-1]) < 16:
            tempo=16-len(chunks[-1])
            chunks[-1]=chunks[-1]+tempo*"0"
        return chunks
    
    else :
        return clairMassage
    
clairMassageChunks=getTextReady(clairMassage)
print("_______CLAIR MASSAGE AFTER BREAKING IT TI 16 BYTES : ",clairMassageChunks)
print("--------------------------------Stage zero--------------------------------")
print("----------------------This is the stage of Hexadecimal--------------------")
def convertChunksToHexa(clairMassageChuncks):
    #this functions main rule is to turn clairMassageChunks into hexa format
    #and organise them in matrix orders by raw
    hexaChunks=[] #list of the words after turning them to Hexa
    for element in clairMassageChunks:
        hexaChunks16=[] #words after turning them into hexa
        hexaChunks4=[]
        i=0
        for char in element:
            if i<4:
                hexaChunks4.append(format(ord(char), "x"))
                i+=1
            else:
                i=0
                hexaChunks4=[]
                hexaChunks4.append(format(ord(char), "x"))
                i+=1
            if len(hexaChunks4)==4:
                hexaChunks16.append(hexaChunks4)
            else:
                continue
        hexaChunks.append(hexaChunks16)
    return hexaChunks

hexaChunks1 = convertChunksToHexa(clairMassageChunks)

def transpose(hexaChunks1):
    #this function will switch the matrix to transpose cus i made a mistake in the ordering of the alphabets in the first matrix
    hexaChunks = []
    hexaChunk16=[]
    hexaChunk4=[]
    for i in range(0,len(hexaChunks1)):
        for y in range(0,len(hexaChunks1[i])):
            for t in range(0,len(hexaChunks1[i][y])):
                hexaChunk4.append(hexaChunks1[i][t][y])
            hexaChunk16.append(hexaChunk4)
            hexaChunk4=[]
        hexaChunks.append(hexaChunk16)
        hexaChunk16=[]
    return hexaChunks

hexaChunks = transpose(hexaChunks1)
for index in range(0,len(hexaChunks)):
    print("Matrix number ",index+1)
    for element in range(len(hexaChunks[index])):
        print(hexaChunks[index][element],"\n")

print("--------------------------------stage one--------------------------------")
print("---------------------This is the stage of SuByte Table--------------------")
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, 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]]

def runThroughSBox(hexaChunks):
    #this fucntion will return a list equivlent of hexmatrix in the table of sbox
    matrixs=[]
    list16=[]
    list4=[]
    for element in range(0,len(hexaChunks)):
        i=0
        for element1 in range(0,len(hexaChunks[element])):
            for element2 in range(0,len(hexaChunks[element][element1])):
                tempo = int(hexaChunks[element][element1][element2][0],16)
                tempo1 = int(hexaChunks[element][element1][element2][1],16)
                if i<4:
                    list4.append(hex(int(sBox[tempo][tempo1]))[2:])
                    i+=1
                else:
                    i=0
                    list4=[]
                    list4.append(hex(int(sBox[tempo][tempo1]))[2:])
                    i+=1
                if len(list4)==4:
                    list16.append(list4)
                    list4=[]
                else:
                    continue
        matrixs.append(list16)
        list16=[]
    return matrixs

firstStageMassage=runThroughSBox(hexaChunks)
for v in range(0,len(firstStageMassage)):
    print("Matrix number (after sbox table )",v+1)
    for j in range(0,len(firstStageMassage[v])):
        print(firstStageMassage[v][j])
print("--------------------------------stage two--------------------------------")
print("----------------------This is the stage of shit row----------------------")
def shiftRow(firstStageMassage1):
    #this function's rule is to shift the matrixs 
    firstStageMassage=deque(firstStageMassage1)
    shiftedMatrixs=[]
    for i in range(0,len(firstStageMassage)):
        shifted16List=[]
        shifted16List.append(firstStageMassage[i][0])
        shifted16List.append(firstStageMassage[i][1][-1:]+firstStageMassage[i][1][:-1])
        shifted16List.append(firstStageMassage[i][2][-2:]+firstStageMassage[i][2][:-2])
        shifted16List.append(firstStageMassage[i][3][-3:]+firstStageMassage[i][3][:-3])
        shiftedMatrixs.append(shifted16List)
    return shiftedMatrixs

shiftedMatrix=shiftRow(firstStageMassage)
for v in range(0,len(shiftedMatrix)):
    print("Matrix number (after shift row )",v+1)
    for j in range(0,len(shiftedMatrix[v])):
        print(shiftedMatrix[v][j])

print("--------------------------------stage three--------------------------------")
print("----------------------This is the stage of mixColumn----------------------")
C = [[2,3,1,1],
    [1,2,3,1],
    [1,1,2,3],
    [3,1,1,2]]

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

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


def mixColumn(shiftedMatrix):
    mixColumn4=[]
    mixColumn16=[]
    mixColumn=[]
    for x in range(0,len(shiftedMatrix)): #iterate through all matrix
        for i in range(0,len(C)): #iterate through raws of C
            for y in range(0,len(shiftedMatrix[x])): #iterate through 
                result="0"
                for t in range(0,len(shiftedMatrix[x])): #iterate through raws of shiftedMatrix
                    if len(shiftedMatrix[x][t][y])==1:
                        tempo=int(shiftedMatrix[x][t][y][0],16)
                        if C[i][t] == 1:
                            #print(hex(int(result,base=16)),hex(int(shiftedMatrix[x][t][y],base=16)))
                            result = hex(int(result,base=16) ^ int(shiftedMatrix[x][t][y],base=16))
                        elif C[i][t] == 2:
                            #print(hex(int(result,base=16)),hex(int(hex(int(by2[tempo][tempo1]))[2:].upper(),base=16)))
                            result = hex(int(result,base=16) ^ int(hex(int(by2[0][tempo]))[2:],base=16))
                        elif C[i][t] == 3:
                            #print(hex(int(result,base=16)),hex(int(hex(int(by3[tempo][tempo1]))[2:].upper(),base=16)))
                            result = hex( int(result,base=16) ^ int(hex(int(by3[0][tempo]))[2:],base=16))
                    else:
                        tempo=int(shiftedMatrix[x][t][y][0],16)
                        tempo1=int(shiftedMatrix[x][t][y][1],16)
                        if C[i][t] == 1:
                            #print(hex(int(result,base=16)),hex(int(shiftedMatrix[x][t][y],base=16)))
                            result = hex(int(result,base=16) ^ int(shiftedMatrix[x][t][y],base=16))
                        elif C[i][t] == 2:
                            #print(hex(int(result,base=16)),hex(int(hex(int(by2[tempo][tempo1]))[2:].upper(),base=16)))
                            result = hex(int(result,base=16) ^ int(hex(int(by2[tempo][tempo1]))[2:],base=16))
                        elif C[i][t] == 3:
                            #print(hex(int(result,base=16)),hex(int(hex(int(by3[tempo][tempo1]))[2:].upper(),base=16)))
                            result = hex( int(result,base=16) ^ int(hex(int(by3[tempo][tempo1]))[2:],base=16))
                mixColumn4.append(result)
                print("result",result)
            mixColumn16.append(mixColumn4)
            mixColumn4=[]
        mixColumn.append(mixColumn16)
        mixColumn16=[]
    return mixColumn

mixColumn=mixColumn(shiftedMatrix)
for v in range(0,len(mixColumn)):
    print("Matrix number (after mixColumn)",v+1)
    for j in range(0,len(mixColumn[v])):
        print(mixColumn[v][j])
