次のロジックで実装する最速の方法は何ですか。
def xor(data, key):
l = len(key)
buff = ""
for i in range(0, len(data)):
buff += chr(ord(data[i]) ^ ord(key[i % l]))
return buff
私の場合、キーは20バイトのsha1ダイジェストであり、データは20バイトから数メガバイト(1、2、3)メガバイトの長さのバイナリデータです。
アップデート:
OKみんな。これは3.5倍高速な実装で、データとキーを4、2、または1バイトのチャンクで分割します(私の場合、ほとんどの場合、4バイトの長整数です)。
def xor(data, key):
index = len(data) % 4
size = (4, 1, 2, 1)[index]
type = ('L', 'B', 'H', 'B')[index]
key_len = len(key)/size
data_len = len(data)/size
key_fmt = "<" + str(key_len) + type;
data_fmt = "<" + str(data_len) + type;
key_list = struct.unpack(key_fmt, key)
data_list = struct.unpack(data_fmt, data)
result = []
for i in range(data_len):
result.append (key_list[i % key_len] ^ data_list[i])
return struct.pack(data_fmt, *result)
大量のメモリを使用しますが、私の場合は大したことではありません。
速度をさらに数回上げる方法はありますか?:-)
最終更新:
OK、OK...numpyがその仕事をしました。それはただの速さです:
def xor(data, key):
import numpy, math
# key multiplication in order to match the data length
key = (key*int(math.ceil(float(len(data))/float(len(key)))))[:len(data)]
# Select the type size in bytes
for i in (8,4,2,1):
if not len(data) % i: break
if i == 8: dt = numpy.dtype('<Q8');
elif i == 4: dt = numpy.dtype('<L4');
elif i == 2: dt = numpy.dtype('<H2');
else: dt = numpy.dtype('B');
return numpy.bitwise_xor(numpy.fromstring(key, dtype=dt), numpy.fromstring(data, dtype=dt)).tostring()
最初の実装ではギガバイトを処理するのに8分50秒かかり、2番目の実装では約2分30秒、最後の実装ではちょうど....0分10秒でした。
アイデアとコードを提供してくれた人に感謝します。あなたは素晴らしい人です!