def testRFC2104_2(self):
""" RFC2104 test 2 """
keyedHashAlg = HMAC(SHA1)
key = 'Jefe'
keyedHashAlg.setKey(key)
data = 'what do ya want for nothing?'
digest = a2b_hex('effcdf6ae5eb2fa2d27416d5f184df9c259a7c79')
cd = keyedHashAlg(data)
assert (cd == digest), 'RFC2104 test 2 failed'
def testRFC2104_2(self):
""" RFC2104 test 2 """
keyedHashAlg = HMAC(SHA1)
key = 'Jefe'
keyedHashAlg.setKey(key)
data = 'what do ya want for nothing?'
digest = a2b_hex('effcdf6ae5eb2fa2d27416d5f184df9c259a7c79')
cd = keyedHashAlg(data)
assert( cd == digest ), 'RFC2104 test 2 failed'
def testSHA1_NullKey(self):
""" HMAC_SHA1 testNullKey """
ki = ''.join([chr(0x36) for i in range(64)])
ko = ''.join([chr(0x5C) for i in range(64)])
h = SHA1()
keyedHashAlg = HMAC(SHA1,key='')
assert ( keyedHashAlg('') == h(ko+h(ki)) ), 'Null key, Null data test'
assert ( keyedHashAlg('a') == h(ko+h(ki+'a')) ), 'Null key, a data test'
assert ( keyedHashAlg('ab') == h(ko+h(ki+'ab')) ), 'Null key, ab data test'
assert ( keyedHashAlg(50*'a') == h(ko+h(ki+50*'a')) ), 'Null key, 50*a data test'
# try hmac in two steps of 25 chrs
manual_hmac = h(ko+h(ki+50*'a'))
keyedHashAlg.update(25*'a')
keyedHashAlg.update(25*'a')
hm = keyedHashAlg.digest()
assert (hm == manual_hmac), 'HMAC as update, update and digest'
def testRFC2104_1(self):
""" RFC2104 test 1 and various calling methods """
key = chr(0x0b) * 20
keyedHashAlg = HMAC(SHA1, key)
data = "Hi There"
digest = a2b_hex('b617318655057264e28bc0b6fb378c8ef146be00')
cd = keyedHashAlg(data)
assert (cd == digest), 'RFC2104 test 1 failed'
hmac_sha1 = HMAC_SHA1(key)
cd = hmac_sha1.hash(data)
assert (cd == digest), 'RFC2104 test 1 failed, HMAC_SHA1 called'
cd = hmac_sha1.hash(data[:3], more=1)
cd = hmac_sha1.hash(data[3:])
assert (cd == digest), 'RFC2104 test 1 failed, HMAC_SHA1 called twice'
hmac_sha1.update(data[:3])
hmac_sha1.update(data[3:])
cd = hmac_sha1.digest()
print b2a_hex(cd)
assert (cd == digest
), 'RFC2104 test 1 failed, HMAC_SHA1 called with update'
hmac_sha1.reset(data)
cd1 = hmac_sha1.hash(data)
cd2 = hmac_sha1.hash(data)
print b2a_hex(cd1)
print b2a_hex(cd2)
assert (cd1 == cd2), 'hash method should default to reseting state'
def testSHA1_NullKey(self):
""" HMAC_SHA1 testNullKey """
ki = ''.join([chr(0x36) for i in range(64)])
ko = ''.join([chr(0x5C) for i in range(64)])
h = SHA1()
keyedHashAlg = HMAC(SHA1, key='')
assert (keyedHashAlg('') == h(ko + h(ki))), 'Null key, Null data test'
assert (keyedHashAlg('a') == h(ko +
h(ki + 'a'))), 'Null key, a data test'
assert (
keyedHashAlg('ab') == h(ko +
h(ki + 'ab'))), 'Null key, ab data test'
assert (keyedHashAlg(
50 * 'a') == h(ko + h(ki + 50 * 'a'))), 'Null key, 50*a data test'
# try hmac in two steps of 25 chrs
manual_hmac = h(ko + h(ki + 50 * 'a'))
keyedHashAlg.update(25 * 'a')
keyedHashAlg.update(25 * 'a')
hm = keyedHashAlg.digest()
assert (hm == manual_hmac), 'HMAC as update, update and digest'
def testMD5_NullKey(self):
""" HMAC_MD5 testNullKey """
ki = ''.join([chr(0x36) for i in range(64)])
ko = ''.join([chr(0x5C) for i in range(64)])
h = MD5()
keyedHashAlg = HMAC(MD5, key='')
assert (keyedHashAlg('') == h(ko + h(ki))), 'Null key, Null data test'
assert (keyedHashAlg('a') == h(ko +
h(ki + 'a'))), 'Null key, a data test'
assert (
keyedHashAlg('ab') == h(ko +
h(ki + 'ab'))), 'Null key, ab data test'
assert (keyedHashAlg(
50 * 'a') == h(ko + h(ki + 50 * 'a'))), 'Null key, 50*a data test'
def testSHA1_oneByteKey(self):
""" HMAC_SHA1 oneByteKey of 0xFF"""
ki = ''.join([chr(0x36) for i in range(64)])
ko = ''.join([chr(0x5C) for i in range(64)])
ki = chr(ord(ki[0]) ^ 0xFF) + ki[1:]
ko = chr(ord(ko[0]) ^ 0xFF) + ko[1:]
h = SHA1()
keyedHashAlg = HMAC(SHA1, chr(0xff))
assert (keyedHashAlg('') == h(ko +
h(ki))), 'one byte key, Null data test'
assert (
keyedHashAlg('a') == h(ko +
h(ki + 'a'))), 'one byte key, a data test'
assert (keyedHashAlg('ab') == h(ko + h(ki + 'ab'))
), 'one byte key, ab data test'
assert (keyedHashAlg(
50 * 'a') == h(ko +
h(ki + 50 * 'a'))), 'one byte key, 50*a data test'
