전에 올린 글의 주제였던 AES를 파이썬으로 암호화 해보기로 했다...

1. pycryptodome

파이썬을 이용하여 암호화를 하기 위해서 난 pycryptodome을 사용할 것이다.

pip install PyCryptodome

위에 코드를 입력했더니

Raquirement already satisfied: PyCryptodome in /home/bassel/anaconda3/lib/python3.8/site-packages (3.9.9)

라고 나왔다. 그럼 이제 파이썬으로 코드를 입력하면 된다.

2. Python으로 코드 입력

from Crypto.Cipher import AES
from secrets import token_bytes

key = token_bytes(16)

def encrypt(msg):
    cipher = AES.new(key, AES.MODE_EAX)
    nonce = cipher.nonce
    ciphertext, tag = cipher.encrypt_and_digest(msg.encode('ascii'))
    return nonce, ciphertext, tag

def decrypt(nonce, ciphertext, tag):
    cipher = AES.new(key, AES.MODE_EAX, nonce=nonce)
    plaintext = cipher.decrypt(ciphertext)
    try:
        cipher.verify(tag)
        return plaintext.decode('ascii')
    except:
        return False

nonce, ciphertext, tag = encrypt(input('Enter a message: '))
plaintext = decrypt(nonce, ciphertext, tag)
print(f'Cipher text: {ciphertext}')
if not plaintext:
    print('Message is corrupted')
else:
    print(f'Plain text: {plaintext}')

이렇게 코드를 입력했다.

3. 결과

터미널에 파일명인

python aes_encryption.py

를 입력했더니

Enter a message:

라고 뜬다. 그리고 입력창에 아무 메세지나 입력해서 enter를 누르면 AES 알고리즘을 통해 Cipher text와 Plain text가 차례대로 나타난다.

Reference
https://pycryptodome.readthedocs.io/en/latest/src/cipher/aes.html?highlight=AES