1import unittest
2from util import *
3from ctypes import create_string_buffer
4
5# HMAC vectors from https://tools.ietf.org/html/rfc4231
6hmac_cases = [
7    ['0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b', '4869205468657265',
8
9     'b0344c61d8db38535ca8afceaf0bf12b 881dc200c9833da726e9376c2e32cff7',
10     '87aa7cdea5ef619d4ff0b4241a1d6cb0 2379f4e2ce4ec2787ad0b30545e17cde'
11     'daa833b7d6b8a702038b274eaea3f4e4 be9d914eeb61f1702e696c203a126854'],
12
13    ['4a656665', '7768617420646f2079612077616e7420666f72206e6f7468696e673f',
14
15     '5bdcc146bf60754e6a042426089575c7 5a003f089d2739839dec58b964ec3843',
16     '164b7a7bfcf819e2e395fbe73b56e0a3 87bd64222e831fd610270cd7ea250554'
17     '9758bf75c05a994a6d034f65f8f0e6fd caeab1a34d4a6b4b636e070a38bce737'],
18
19    ['aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa',
20     'dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd'
21     'dddddddddddddddddddddddddddddddddddd',
22
23     '773ea91e36800e46854db8ebd09181a7 2959098b3ef8c122d9635514ced565fe',
24     'fa73b0089d56a284efb0f0756c890be9 b1b5dbdd8ee81a3655f83e33b2279d39'
25     'bf3e848279a722c806b485a47e67c807 b946a337bee8942674278859e13292fb'],
26
27    ['0102030405060708090a0b0c0d0e0f10111213141516171819',
28     'cdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcd'
29     'cdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcd',
30
31     '82558a389a443c0ea4cc819899f2083a 85f0faa3e578f8077a2e3ff46729665b',
32     'b0ba465637458c6990e5a8c5f61d4af7 e576d97ff94b872de76f8050361ee3db'
33     'a91ca5c11aa25eb4d679275cc5788063 a5f19741120c4f2de2adebeb10a298dd'],
34
35    ['0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c0c',
36     '546573742057697468205472756e636174696f6e',
37
38     'a3b6167473100ee06e0c796c2955552b',
39     '415fad6271580a531d4179bc891d87a6'],
40
41    ['aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
42     'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
43     'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
44     'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
45     'aaaaaa',
46     '54657374205573696e67204c6172676572205468616e20426c6f636b2d53697a'
47     '65204b6579202d2048617368204b6579204669727374',
48
49     '60e431591ee0b67f0d8a26aacbf5b77f 8e0bc6213728c5140546040f0ee37f54',
50     '80b24263c7c1a3ebb71493c1dd7be8b4 9b46d1f41b4aeec1121b013783f8f352'
51     '6b56d037e05f2598bd0fd2215d6a1e52 95e64f73f63f0aec8b915a985d786598'],
52
53    ['aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
54     'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
55     'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
56     'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa'
57     'aaaaaa',
58     '5468697320697320612074657374207573696e672061206c6172676572207468'
59     '616e20626c6f636b2d73697a65206b657920616e642061206c61726765722074'
60     '68616e20626c6f636b2d73697a6520646174612e20546865206b6579206e6565'
61     '647320746f20626520686173686564206265666f7265206265696e6720757365'
62     '642062792074686520484d414320616c676f726974686d2e',
63
64     '9b09ffa71b942fcb27635fbcd5b0e944 bfdc63644f0713938a7f51535c3a35e2',
65     'e37b6a775dc87dbaa4dfa9f96e5e3ffd debd71f8867289865df5a32d20cdc944'
66     'b6022cac3c4982b10d5eeb55c3e4de15 134676fb6de0446065c97440fa8c6a58']
67]
68
69class HMACTests(unittest.TestCase):
70
71    def doHMAC(self, fn, key_in, msg_in):
72        key, key_len = make_cbuffer(key_in)
73        msg, msg_len = make_cbuffer(msg_in)
74        buf_len = 64 if fn == wally_hmac_sha512 else 32
75        buf = create_string_buffer(buf_len)
76        ret = fn(key, key_len, msg, msg_len, buf, buf_len)
77        return ret, h(buf)
78
79
80    def test_vectors(self):
81
82        for test in hmac_cases:
83            k, msg = test[0], test[1]
84            for fn, expected in [(wally_hmac_sha256, test[2]),
85                                 (wally_hmac_sha512, test[3])]:
86                ret, result = self.doHMAC(fn, k, msg)
87                self.assertEqual(ret, 0)
88                expected = utf8(expected.replace(' ', ''))
89                # Note we truncate the result as one of the test vectors has
90                # a truncated result in the RFC
91                self.assertEqual(result[0:len(expected)], expected)
92
93
94if __name__ == '__main__':
95    unittest.main()
96