今日pycryptoを見つけたばかりで、AES暗号化クラスに取り組んでいます。残念ながら、半分しか機能しません。 self.h.md5はmd5ハッシュを16進形式で出力し、32バイトです。これが出力です。メッセージを復号化するようですが、復号化後にランダムな文字を挿入します。この場合、\ n\n\n ... self.dataのブロックサイズに問題があると思います。誰でもこれを修正する方法を知っていますか?
Jans-MacBook-Pro:test2 jan $ ../../bin/python3 data.py b'RLfGmn5jf5WTJphnmW0hXG7IaIYcCRpjaTTqwXR6yiJCUytnDib + GQYlFORm + jIctest 1 2 3 5\n\n '
from Crypto.Cipher import AES
from base64 import b64encode, b64decode
from os import urandom
class Encryption():
def __init__(self):
self.h = Hash()
def values(self, data, key):
self.data = data
self.key = key
self.mode = AES.MODE_CBC
self.iv = urandom(16)
if not self.key:
self.key = Cfg_Encrypt_Key
self.key = self.h.md5(self.key, True)
def encrypt(self, data, key):
self.values(data, key)
return b64encode(self.iv + AES.new(self.key, self.mode, self.iv).encrypt(self.data))
def decrypt(self, data, key):
self.values(data, key)
self.iv = b64decode(self.data)[:16]
return AES.new(self.key, self.mode, self.iv).decrypt(b64decode(self.data)[16:])
正直に言うと、文字"\ n\n\n\n\n\n\n\n\n\n"は私にはランダムに見えません。 ;-)
CESモードでAESを使用しています。そのためには、プレーンテキストと暗号化テキストの長さが常に16バイトの倍数である必要があります。表示するコードを使用すると、encrypt()
に渡されるdata
がそのような条件を満たさない場合に、実際に例外が発生することがわかります。プレーンテキストが整列されるまで、入力に何でも十分な改行文字('\ n'を追加したようです。
それとは別に、アライメントの問題を解決する2つの一般的な方法があります。
CBCからの切り替え(AES.MODE_CBC
)からCFB(AES.MODE_CFB
)。デフォルトのsegment_size
PyCryptoで使用されているため、プレーンテキストと暗号化テキストの長さに制限はありません。
CBCを保持し、PKCS#7のような埋め込みスキームを使用します。
X
バイトのプレーンテキストを暗号化する前に、次の16バイト境界に到達するのに必要なバイト数だけ後ろに追加します。すべてのパディングバイトの値は同じです:追加するバイト数:
length = 16 - (len(data) % 16)
data += bytes([length])*length
Python 3スタイル。Python 2では、次のようになります。
length = 16 - (len(data) % 16)
data += chr(length)*length
復号化後、パディングで示されたバイト数だけプレーンテキストの後ろから削除します。
data = data[:-data[-1]]
あなたの場合、それは単なるクラスの演習であると理解していますが、何らかの形式の認証(MACなど)なしでデータを送信することは安全ではないことを指摘したいと思います。
from hashlib import md5
from Crypto.Cipher import AES
from Crypto import Random
import base64
def derive_key_and_iv(password, salt, key_length, iv_length):
d = d_i = ''
while len(d) < key_length + iv_length:
d_i = md5(d_i + password + salt).digest()
d += d_i
return d[:key_length], d[key_length:key_length+iv_length]
def encrypt(in_file, out_file, password, key_length=32):
bs = AES.block_size
salt = Random.new().read(bs - len('Salted__'))
key, iv = derive_key_and_iv(password, salt, key_length, bs)
cipher = AES.new(key, AES.MODE_CBC, iv)
#print in_file
in_file = file(in_file, 'rb')
out_file = file(out_file, 'wb')
out_file.write('Salted__' + salt)
finished = False
while not finished:
chunk = in_file.read(1024 * bs)
if len(chunk) == 0 or len(chunk) % bs != 0:
padding_length = bs - (len(chunk) % bs)
chunk += padding_length * chr(padding_length)
finished = True
out_file.write(cipher.encrypt(chunk))
in_file.close()
out_file.close()
def decrypt(in_file, out_file, password, key_length=32):
bs = AES.block_size
in_file = file(in_file, 'rb')
out_file = file(out_file, 'wb')
salt = in_file.read(bs)[len('Salted__'):]
key, iv = derive_key_and_iv(password, salt, key_length, bs)
cipher = AES.new(key, AES.MODE_CBC, iv)
next_chunk = ''
finished = False
while not finished:
chunk, next_chunk = next_chunk, cipher.decrypt(in_file.read(1024 * bs))
if len(next_chunk) == 0:
padding_length = ord(chunk[-1])
if padding_length < 1 or padding_length > bs:
raise ValueError("bad decrypt pad (%d)" % padding_length)
# all the pad-bytes must be the same
if chunk[-padding_length:] != (padding_length * chr(padding_length)):
# this is similar to the bad decrypt:evp_enc.c from openssl program
raise ValueError("bad decrypt")
chunk = chunk[:-padding_length]
finished = True
out_file.write(chunk)
in_file.close()
out_file.close()
def encode(in_file, out_file):
in_file = file(in_file, 'rb')
out_file = file(out_file, 'wb')
data = in_file.read()
out_file.write(base64.b64encode(data))
in_file.close()
out_file.close()
def decode(in_file, out_file):
in_file = file(in_file, 'rb')
out_file = file(out_file, 'wb')
data = in_file.read()
out_file.write(base64.b64decode(data))
in_file.close()
out_file.close()
初期ペイロードの長さを覚えている限り、修正文字を使用できるため、有用な終了バイトを「捨てる」ことはありません。これを試して:
import base64
from Crypto.Cipher import AES
def encrypt(payload, salt, key):
return AES.new(key, AES.MODE_CBC, salt).encrypt(r_pad(payload))
def decrypt(payload, salt, key, length):
return AES.new(key, AES.MODE_CBC, salt).decrypt(payload)[:length]
def r_pad(payload, block_size=16):
length = block_size - (len(payload) % block_size)
return payload + chr(length) * length
print(decrypt(encrypt("some cyphertext", "b" * 16, "b" * 16), "b" * 16, "b" * 16, len("some cyphertext")))