Skip to content

Instantly share code, notes, and snippets.

@5S
Created Jan 31, 2019
Embed
What would you like to do?
PyCryptodome を利用して AES 形式による暗号化/復号を行うクラス
# -*- coding: utf-8 -*-
import base64
import hashlib
from Crypto import Random
from Crypto.Cipher import AES
from Crypto.Util import Padding
class AESCipher(object):
def __init__(self, key):
self.key = (hashlib.md5(key.encode('utf-8')).hexdigest()).encode('utf-8')
def encrypt(self, raw):
iv = Random.get_random_bytes(AES.block_size)
cipher = AES.new(self.key, AES.MODE_CBC, iv)
data = Padding.pad(raw.encode('utf-8'), AES.block_size, 'pkcs7')
return base64.b64encode(iv + cipher.encrypt(data))
def decrypt(self, enc):
enc = base64.b64decode(enc)
iv = enc[:AES.block_size]
cipher = AES.new(self.key, AES.MODE_CBC, iv)
data = Padding.unpad(cipher.decrypt(enc[AES.block_size:]), AES.block_size, 'pkcs7')
return data.decode('utf-8')
# -*- coding: utf-8 -*-
import random
import string
from crypt import AESCipher
text = 'plain text'
key = ''.join([random.choice(string.ascii_letters + string.digits) for i in range(32)]) # -> '4yKxQ5hMcUJixcG4Z8Lc5ZPBr5McS65X'
cipher = AESCipher(key)
encrypted = cipher.encrypt(text)
print(encrypted) # -> b'MLXpzLheE1383lHyVkGzoppMmO78otn3d0BOgh7WGdw='
decrypted = cipher.decrypt(encrypted)
print(decrypted) # -> 'plain text'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment