def test_binary(dir='binary', odir='encrypt'):
"""test for image file"""
segment_greeting(greet="SEF File Format Test")
if not os.path.isdir(dir):
raise ValueError('not valid dir')
print Colors.colorize('Tester for image file, directory: ', 'blue',
'white', 'bold'), dir
print '-' * 80
key = Key.random_key(64)
print Colors.colorize('Keys: ', 'cyan', 'black', 'bold'), key
mask = 3
print Colors.colorize('Mask: ', 'cyan', 'black', 'bold'), mask
for file in os.listdir(dir):
print Colors.colorize('current test image: ', 'cyan', 'black',
'bold'), file
print '-' * 80
data = open(os.path.join(dir, file), 'rb').read()
fname = os.path.splitext(file)[0] + '.dat'
out = os.path.join(odir, fname)
start = time.clock()
ret = SEF_celler(data, key, mask, out=out)
stop = time.clock()
delta = stop - start
print Colors.colorize('time: ', 'cyan', 'black', 'bold'), delta
status = 'success' if ret else 'fail'
print Colors.colorize('status: ', 'cyan', 'black', 'bold'), status
print ''
def test_general():
"""run message length test for AES, DES, DES3"""
segment_greeting(greet="General Encryption Test")
files = 'DES.dat', 'DES3.dat', 'AES.dat'
cellers = DES_celler, DES3_celler, AES_celler
print("Key Size: 128\n" "Cellers: {0}").format(('DES', 'DES3', 'AES'))
key = Key.random_key(128)[:16]
print Colors.colorize('key: ', 'cyan', 'black', 'bold'), key
turns = 13
print Colors.colorize('test turns: ', 'cyan', 'black', 'bold'), turns
for index, celler in enumerate(cellers):
lens = []
deltas = []
key_ = key[:8] if index == 0 else key
cell = files[index].split('.')[0]
print Colors.colorize('run test for ' + cell, 'cyan', 'black', 'bold')
print '-' * 80
print Colors.colorize('Tester Result: ', 'cyan', 'black', 'bold')
print('{:>15}' * 3).format('message len', 'status', 'runtime(s)')
print ''
for length in length_generator(turns):
lens.append(length)
message = generate_message(length)
start = time.clock()
ret = celler(message, key_)
stop = time.clock()
delta = stop - start
deltas.append(delta)
status = 'success' if ret else 'fail'
print('{:>15}' * 3).format(humanize_bytes(length), status, delta)
print '-' * 80
print Colors.colorize('Time Consuming Growing: ', 'cyan', 'black',
'bold')
gnuplot(
'message growing',
lens,
deltas,
lens[-1],
deltas[-1],
'message/bytes',
'time/s',
)
curve.fit(lens, deltas)
print Colors.colorize('Scipy Curve Fit: ', 'cyan', 'black',
'bold'), curve
_, ys = curve.sample()
save_data(lens, deltas, ys, files[index])
print ''
def test_keys():
"""test same message for different length key
"""
segment_greeting(greet="SEF Key Test")
message = (
"Python was conceived in the late 1980s,[30] and its implementation "
"began in December 1989[31] by Guido van Rossum at Centrum Wiskunde & "
"Informatica (CWI) in the Netherlands as a successor to the ABC language "
"(itself inspired by SETL)[32] capable of exception handling and interfacing "
"with the operating system Amoeba.[7] Van Rossum is Python's principal author, "
"and his continuing central role in deciding the direction of Python is reflected "
"in the title given to him by the Python community, benevolent dictator for life (BDFL)."
"About the origin of Python, Van Rossum wrote in 1996:[33]")
print Colors.colorize('message source: ', 'cyan', 'black', 'bold')
pprint(message)
mask = 0x03
print Colors.colorize('key mask: ', 'cyan', 'black', 'bold'), mask
key_types = [i * 8 for i in range(1, 33)]
keys = [(type, Key.random_key(type)) for type in key_types]
print Colors.colorize('keys: ', 'cyan', 'black', 'bold')
pprint(keys)
print Colors.colorize('Tester Result: ', 'cyan', 'black', 'bold')
print('{:>15}' * 3).format('key', 'status', 'runtime(s)')
print '-' * 80
print ''
deltas = []
for type, key in keys:
start = time.clock()
ret = SEF_celler(message, key, mask)
stop = time.clock()
delta = stop - start
deltas.append(delta)
status = 'success' if ret else 'fail'
print('{:>15}' * 3).format(type, status, delta)
print '-' * 80
print Colors.colorize('Time Consuming Growing: ', 'cyan', 'black', 'bold')
gnuplot('key bits growing', key_types, deltas, key_types[-1], deltas[-1],
'key/bits', 'time/s')
curve.fit(key_types, deltas)
print Colors.colorize('Scipy Curve Fit: ', 'cyan', 'black', 'bold'), curve
_, ys = curve.sample()
save_data(key_types, deltas, ys, 'KEY.dat')
print ''
def test_message():
"""test for the message length
"""
segment_greeting(greet="SEF Message Length Test")
key = Key.random_key(64)
print Colors.colorize('encrypt key: ', 'cyan', 'black', 'bold'), key
mask = 5
print Colors.colorize('mask: ', 'cyan', 'black', 'bold'), mask
turns = 13
print Colors.colorize('test turns: ', 'cyan', 'black', 'bold'), turns
print Colors.colorize('Tester Result: ', 'cyan', 'black', 'bold')
print('{:>15}' * 3).format('message len', 'status', 'runtime(s)')
print '-' * 80
print ''
lens = []
deltas = []
for length in length_generator(turns):
# get the message
lens.append(length)
message = generate_message(length)
# measure time
start = time.clock()
ret = SEF_celler(message, key, mask)
stop = time.clock()
delta = stop - start
deltas.append(delta)
status = 'success' if ret else 'fail'
print('{:>15}' * 3).format(humanize_bytes(length), status, delta)
print '-' * 80
print Colors.colorize('Time Consuming Growing: ', 'cyan', 'black', 'bold')
gnuplot(
'message growing',
lens,
deltas,
lens[-1],
deltas[-1],
'message/bytes',
'time/s',
)
curve.fit(lens, deltas)
print Colors.colorize('Scipy Curve Fit: ', 'cyan', 'black', 'bold'), curve
_, ys = curve.sample()
save_data(lens, deltas, ys, 'SEF.dat')
print ''
def test_zip(file='binary/test-video.mp4', odir='zip'):
"""test for compressing the file"""
segment_greeting(greet="SEF Zip Test")
if not os.path.isfile(file):
raise ValueError('not valid file')
print Colors.colorize('Compressing Test', 'blue', 'white', 'bold')
fsize = os.stat(file).st_size
print Colors.colorize('Source File: ', 'blue', 'white', 'bold'), file
print Colors.colorize('File Size(bytes): ', 'blue', 'white', 'bold'), fsize
names = 'SEF', 'DES', '3DES', 'AES'
cellers = SEF_celler, DES_celler, DES3_celler, AES_celler
mask = 3
key = Key.random_key(128)[:16]
message = open(file, 'rb').read()
bytebuf = io.BytesIO()
cmprbuf = io.BytesIO()
#zipfile = gzip.GzipFile(fileobj=cmprbuf, mode='wb')
#zipfile.write(message)
#csize = len(cmprbuf.getvalue())
#rate = float(csize) / fsize * 100
#print Colors.colorize('Source Compress Rate: ', 'cyan', 'black', 'bold'), rate
#print '-' * 80
for index, cell in enumerate(cellers):
print Colors.colorize('Run Zip Test for ', 'cyan', 'black',
'bold'), names[index]
print '-' * 80
key_ = key[:8] if names[index] == 'DES' else key
ret = cell(message, key_, mask, out=bytebuf)
if ret:
zipfile = gzip.GzipFile(fileobj=cmprbuf, mode='wb')
zipfile.write(bytebuf.getvalue())
csize = len(cmprbuf.getvalue())
rate = float(csize) / fsize * 100
print Colors.colorize('rate: ', 'cyan', 'black', 'bold'), rate
else:
print Colors.colorize('test failed.', 'cyan', 'black', 'bold')
# reset the byte buffer
bytebuf.truncate(0)
bytebuf.seek(0, 0)
cmprbuf.truncate(0)
cmprbuf.seek(0, 0)
def test_key_relation():
"""test the key relation with cipher
test times : 3
"""
segment_greeting(greet='SEF Key & Cipher')
mask = 5
print Colors.colorize('Mask: ', 'cyan', 'black', 'bold'), mask
times = 3
key = Key.random_key(256)
keys = [key[16 * time:16 * (time + 1)] for time in range(times)]
print Colors.colorize('Keys: ', 'cyan', 'black', 'bold')
pprint(keys)
message = generate_message(1024)
msize = len(message)
print Colors.colorize('Message Size: ', 'cyan', 'black', 'bold'), msize
print Colors.colorize('Message: ', 'cyan', 'black', 'bold')
pprint(message)
print ''
def unsimilara_rate(item1, item2):
"""return rate"""
counter = 0
csize = len(item1)
for index, letter in enumerate(item1):
if item1[index] != item2[index]:
counter += 1
rate = counter * 0.1 / csize * 100
return rate
def key_relation(key, message):
"""test key relation"""
key_ = random_key_bits(key)
print Colors.colorize('Key: ', 'cyan', 'black', 'bold'), key
print Colors.colorize('ALT: ', 'cyan', 'black', 'bold'), key_
message1 = sef_encrypt(key, mask, message)
message2 = sef_encrypt(key_, mask, message)
if len(message2) != len(message1):
raise Exception('encryption error')
rate = unsimilara_rate(message1, message2)
print Colors.colorize('Change Rate: ', 'cyan', 'black',
'bold'), '%.3f %%' % rate
print ''
def message_relation(key, message):
"""test the message relation"""
byte = random.randrange(msize)
letter = message[byte]
message_ = message[:byte] + chr(ord(letter) + 1) + message[byte:]
print Colors.colorize('MSG: ', 'cyan', 'black', 'bold')
pprint(message_)
print Colors.colorize('Index: ', 'cyan', 'black', 'bold'), byte
message1 = sef_encrypt(key, mask, message)
message2 = sef_encrypt(key, mask, message_)
if len(message2) != len(message1):
raise Exception('encryption error')
rate = unsimilara_rate(message1, message2)
print Colors.colorize('Change Rate: ', 'cyan', 'black',
'bold'), '%.3f %%' % rate
print ''
for key in keys:
print '*' * 80
key_relation(key, message)
print '-' * 80
message_relation(key, message)