# Copyright 2015 Egor Tensin <Egor.Tensin@gmail.com>
# This file is licensed under the terms of the MIT License.
# See LICENSE.txt for details.
if __name__ == '__main__':
import argparse, json, toolkit, sys
parser = argparse.ArgumentParser()
parser.add_argument('--root', '-r', required=True,
help='set path to *.exe files')
parser.add_argument('--sde', '-e', action='store_true',
help='use Intel SDE to run *.exe files')
args = parser.parse_args()
tools = toolkit.Tools(args.root, use_sde=args.sde)
vectors = json.load(open('800-38a.json'))
success = True
for prefix in vectors:
if 'iv' in vectors[prefix]:
iv = vectors[prefix]['iv']
encrypt = lambda key, plaintexts: tools.encrypt_with_iv(prefix, key, iv, plaintexts)
decrypt = lambda key, ciphers: tools.decrypt_with_iv(prefix, key, iv, ciphers)
else:
encrypt = lambda key, plaintexts: tools.encrypt(prefix, key, plaintexts)
decrypt = lambda key, ciphers: tools.decrypt(prefix, key, ciphers)
key = vectors[prefix]['key']
plaintexts = vectors[prefix]['plaintexts']
ciphers = vectors[prefix]['ciphers']
success = success and tools.detect_mismatches(plaintexts, encrypt(key, plaintexts), ciphers)
success = success and tools.detect_mismatches(ciphers, decrypt(key, ciphers), plaintexts)
if success:
print('No mismatches detected!')
else:
print('Detected mismatches!')
sys.exit(1)
