Subversion Repositories aes-128-ecb-encoder

if __name__ == '__main__':

    import os

    import time

    import aes128

    print('Step 1:')

    while True:

        print('Press 1 for encription smth and 2 for decription')

        way = input()

        if way not in ['1', '2']:

            print('Action denied')

            continue

        else:

            break

    print()

    print('Step 2:')

    while True:

        print('Enter full name of file')

        input_path = os.path.abspath(input())

        if os.path.isfile(input_path):

            break

        else:

            print('This is not a file')

            continue

    print()

    print('Step 3:')

    while True:

        print('Enter your Key for encription/decription. The Key must be less than 16 symbols. Please, don\'t forget it!')

        key = input()

        if len(key) > 16:

            print('Too long Key. Imagine another one')

            continue

        for symbol in key:

            if ord(symbol) > 0xff:

                print('That key won\'t work. Try another using only latin alphabet and numbers')

                continue

        break

    print('\r\nPlease, wait...')

    time_before = time.time()

    # Input data

    with open(input_path, 'rb') as f:

        data = f.read()

    if way == '1':

        crypted_data = []

        temp = []

        for byte in data:

            temp.append(byte)

            if len(temp) == 16:

                crypted_part = aes128.encrypt(temp, key)

                crypted_data.extend(crypted_part)

                del temp[:]

        else:

            #padding v1

            # crypted_data.extend(temp)

            # padding v2

            if 0 < len(temp) < 16:

                empty_spaces = 16 - len(temp)

                for i in range(empty_spaces - 1):

                    temp.append(0)

                temp.append(1)

                crypted_part = aes128.encrypt(temp, key)

                crypted_data.extend(crypted_part)

        out_path = os.path.join(os.path.dirname(input_path) , 'crypted_' + os.path.basename(input_path))

        # Ounput data

        with open(out_path, 'xb') as ff:

            ff.write(bytes(crypted_data))

    else: # if way == '2'

        decrypted_data = []

        temp = []

        for byte in data:

            temp.append(byte)

            if len(temp) == 16:

                decrypted_part = aes128.decrypt(temp, key)

                decrypted_data.extend(decrypted_part)

                del temp[:]

        else:

            #padding v1

            # decrypted_data.extend(temp)

            # padding v2

            if 0 < len(temp) < 16:

                empty_spaces = 16 - len(temp)

                for i in range(empty_spaces - 1):

                    temp.append(0)

                temp.append(1)

                decrypted_part = aes128.encrypt(temp, key)

                decrypted_data.extend(crypted_part)

        out_path = os.path.join(os.path.dirname(input_path) , 'decrypted_' + os.path.basename(input_path))

        # Ounput data

        with open(out_path, 'xb') as ff:

            ff.write(bytes(decrypted_data))

    time_after = time.time()

print('New file here:', out_path, '--', time_after - time_before, ' seconds')

print('If smth wrong check the key you entered')