import os
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad

flag = os.environ.get('FLAG', 'CTFMN{fake_flag_for_testing}')
key = os.urandom(16)

def encode(text):
    result = text
    while len(result) % 16 != 0:
        result += 'X'
    return result


def encrypt(pt):
    cipher = AES.new(key, AES.MODE_ECB)
    ciphertext = cipher.encrypt(pad(pt.encode(), 16))
    return ciphertext.hex()


while True:
    pt = input('plaintext = ')
    pt = encode(pt) + encode(flag)
    print('ciphertext =', encrypt(pt))
