197 lines
6.9 KiB
Python
197 lines
6.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# SelfTest/Hash/common.py: Common code for Crypto.SelfTest.Hash
|
|
#
|
|
# Written in 2008 by Dwayne C. Litzenberger <dlitz@dlitz.net>
|
|
#
|
|
# ===================================================================
|
|
# The contents of this file are dedicated to the public domain. To
|
|
# the extent that dedication to the public domain is not available,
|
|
# everyone is granted a worldwide, perpetual, royalty-free,
|
|
# non-exclusive license to exercise all rights associated with the
|
|
# contents of this file for any purpose whatsoever.
|
|
# No rights are reserved.
|
|
#
|
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
|
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
# MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
|
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
|
# BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
|
# ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
# SOFTWARE.
|
|
# ===================================================================
|
|
|
|
"""Self-testing for PyCrypto hash modules"""
|
|
|
|
__revision__ = "$Id$"
|
|
|
|
import sys
|
|
import unittest
|
|
import binascii
|
|
from Crypto.Util.py3compat import *
|
|
|
|
# For compatibility with Python 2.1 and Python 2.2
|
|
if sys.hexversion < 0x02030000:
|
|
# Python 2.1 doesn't have a dict() function
|
|
# Python 2.2 dict() function raises TypeError if you do dict(MD5='blah')
|
|
def dict(**kwargs):
|
|
return kwargs.copy()
|
|
else:
|
|
dict = dict
|
|
|
|
|
|
class HashDigestSizeSelfTest(unittest.TestCase):
|
|
|
|
def __init__(self, hashmod, description, expected):
|
|
unittest.TestCase.__init__(self)
|
|
self.hashmod = hashmod
|
|
self.expected = expected
|
|
self.description = description
|
|
|
|
def shortDescription(self):
|
|
return self.description
|
|
|
|
def runTest(self):
|
|
self.assertTrue(hasattr(self.hashmod, "digest_size"))
|
|
self.assertEqual(self.hashmod.digest_size, self.expected)
|
|
h = self.hashmod.new()
|
|
self.assertTrue(hasattr(h, "digest_size"))
|
|
self.assertEqual(h.digest_size, self.expected)
|
|
|
|
|
|
class HashSelfTest(unittest.TestCase):
|
|
|
|
def __init__(self, hashmod, description, expected, input):
|
|
unittest.TestCase.__init__(self)
|
|
self.hashmod = hashmod
|
|
self.expected = expected
|
|
self.input = input
|
|
self.description = description
|
|
|
|
def shortDescription(self):
|
|
return self.description
|
|
|
|
def runTest(self):
|
|
h = self.hashmod.new()
|
|
h.update(self.input)
|
|
|
|
out1 = binascii.b2a_hex(h.digest())
|
|
out2 = h.hexdigest()
|
|
|
|
h = self.hashmod.new(self.input)
|
|
|
|
out3 = h.hexdigest()
|
|
out4 = binascii.b2a_hex(h.digest())
|
|
|
|
# PY3K: hexdigest() should return str(), and digest() bytes
|
|
self.assertEqual(self.expected, out1) # h = .new(); h.update(data); h.digest()
|
|
if sys.version_info[0] == 2:
|
|
self.assertEqual(self.expected, out2) # h = .new(); h.update(data); h.hexdigest()
|
|
self.assertEqual(self.expected, out3) # h = .new(data); h.hexdigest()
|
|
else:
|
|
self.assertEqual(self.expected.decode(), out2) # h = .new(); h.update(data); h.hexdigest()
|
|
self.assertEqual(self.expected.decode(), out3) # h = .new(data); h.hexdigest()
|
|
self.assertEqual(self.expected, out4) # h = .new(data); h.digest()
|
|
|
|
# Verify that new() object method produces a fresh hash object
|
|
h2 = h.new()
|
|
h2.update(self.input)
|
|
out5 = binascii.b2a_hex(h2.digest())
|
|
self.assertEqual(self.expected, out5)
|
|
|
|
class HashTestOID(unittest.TestCase):
|
|
def __init__(self, hashmod, oid):
|
|
unittest.TestCase.__init__(self)
|
|
self.hashmod = hashmod
|
|
self.oid = oid
|
|
|
|
def runTest(self):
|
|
h = self.hashmod.new()
|
|
if self.oid==None:
|
|
try:
|
|
raised = 0
|
|
a = h.oid
|
|
except AttributeError:
|
|
raised = 1
|
|
self.assertEqual(raised,1)
|
|
else:
|
|
self.assertEqual(h.oid, self.oid)
|
|
|
|
class MACSelfTest(unittest.TestCase):
|
|
|
|
def __init__(self, hashmod, description, expected_dict, input, key, hashmods):
|
|
unittest.TestCase.__init__(self)
|
|
self.hashmod = hashmod
|
|
self.expected_dict = expected_dict
|
|
self.input = input
|
|
self.key = key
|
|
self.hashmods = hashmods
|
|
self.description = description
|
|
|
|
def shortDescription(self):
|
|
return self.description
|
|
|
|
def runTest(self):
|
|
for hashname in list(self.expected_dict.keys()):
|
|
hashmod = self.hashmods[hashname]
|
|
key = binascii.a2b_hex(b(self.key))
|
|
data = binascii.a2b_hex(b(self.input))
|
|
|
|
# Strip whitespace from the expected string (which should be in lowercase-hex)
|
|
expected = b("".join(self.expected_dict[hashname].split()))
|
|
|
|
h = self.hashmod.new(key, digestmod=hashmod)
|
|
h.update(data)
|
|
out1 = binascii.b2a_hex(h.digest())
|
|
out2 = h.hexdigest()
|
|
|
|
h = self.hashmod.new(key, data, hashmod)
|
|
|
|
out3 = h.hexdigest()
|
|
out4 = binascii.b2a_hex(h.digest())
|
|
|
|
# Test .copy()
|
|
h2 = h.copy()
|
|
h.update(b("blah blah blah")) # Corrupt the original hash object
|
|
out5 = binascii.b2a_hex(h2.digest()) # The copied hash object should return the correct result
|
|
|
|
# PY3K: hexdigest() should return str(), and digest() bytes
|
|
self.assertEqual(expected, out1)
|
|
if sys.version_info[0] == 2:
|
|
self.assertEqual(expected, out2)
|
|
self.assertEqual(expected, out3)
|
|
else:
|
|
self.assertEqual(expected.decode(), out2)
|
|
self.assertEqual(expected.decode(), out3)
|
|
self.assertEqual(expected, out4)
|
|
self.assertEqual(expected, out5)
|
|
|
|
def make_hash_tests(module, module_name, test_data, digest_size, oid=None):
|
|
tests = []
|
|
for i in range(len(test_data)):
|
|
row = test_data[i]
|
|
(expected, input) = list(map(b,row[0:2]))
|
|
if len(row) < 3:
|
|
description = repr(input)
|
|
else:
|
|
description = row[2].encode('latin-1')
|
|
name = "%s #%d: %s" % (module_name, i+1, description)
|
|
tests.append(HashSelfTest(module, name, expected, input))
|
|
if oid is not None:
|
|
oid = b(oid)
|
|
name = "%s #%d: digest_size" % (module_name, i+1)
|
|
tests.append(HashDigestSizeSelfTest(module, name, digest_size))
|
|
tests.append(HashTestOID(module, oid))
|
|
return tests
|
|
|
|
def make_mac_tests(module, module_name, test_data, hashmods):
|
|
tests = []
|
|
for i in range(len(test_data)):
|
|
row = test_data[i]
|
|
(key, data, results, description) = row
|
|
name = "%s #%d: %s" % (module_name, i+1, description)
|
|
tests.append(MACSelfTest(module, name, results, data, key, hashmods))
|
|
return tests
|
|
|
|
# vim:set ts=4 sw=4 sts=4 expandtab:
|