1import unittest
2from util import *
3
4# Test cases from
5# https://tools.ietf.org/html/draft-josefsson-scrypt-kdf-04
6cases = [
7    ["", "",
8     16, 1, 1, 64,
9     '77 d6 57 62 38 65 7b 20 3b 19 ca 42 c1 8a 04 97'
10     'f1 6b 48 44 e3 07 4a e8 df df fa 3f ed e2 14 42'
11     'fc d0 06 9d ed 09 48 f8 32 6a 75 3a 0f c8 1f 17'
12     'e8 d3 e0 fb 2e 0d 36 28 cf 35 e2 0c 38 d1 89 06'],
13
14    ['password', 'NaCl',
15     1024, 8, 16, 64,
16     'fd ba be 1c 9d 34 72 00 78 56 e7 19 0d 01 e9 fe'
17     '7c 6a d7 cb c8 23 78 30 e7 73 76 63 4b 37 31 62'
18     '2e af 30 d9 2e 22 a3 88 6f f1 09 27 9d 98 30 da'
19     'c7 27 af b9 4a 83 ee 6d 83 60 cb df a2 cc 06 40'],
20
21    ['pleaseletmein', 'SodiumChloride',
22     16384, 8, 1, 64,
23     '70 23 bd cb 3a fd 73 48 46 1c 06 cd 81 fd 38 eb'
24     'fd a8 fb ba 90 4f 8e 3e a9 b5 43 f6 54 5d a1 f2'
25     'd5 43 29 55 61 3f 0f cf 62 d4 97 05 24 2a 9a f9'
26     'e6 1e 85 dc 0d 65 1e 40 df cf 01 7b 45 57 58 87'],
27
28    ['pleaseletmein', 'SodiumChloride',
29     1048576, 8, 1, 64,
30     '21 01 cb 9b 6a 51 1a ae ad db be 09 cf 70 f8 81'
31     'ec 56 8d 57 4a 2f fd 4d ab e5 ee 98 20 ad aa 47'
32     '8e 56 fd 8f 4b a5 d0 9f fa 1c 6d 92 7c 40 f4 c3'
33     '37 30 40 49 e8 a9 52 fb cb f4 5c 6f a7 7a 41 a4']
34]
35
36class ScryptTests(unittest.TestCase):
37
38    def test_scrypt(self):
39
40        for c in cases:
41            passwd, salt, cost, block, parallel, length, expected = c
42            passwd = utf8(passwd)
43            salt = utf8(salt)
44            expected = expected.replace(' ', '')
45            assert len(expected) == length * 2
46            out_buf, out_len = make_cbuffer('0' * len(expected))
47
48            ret = wally_scrypt(passwd, len(passwd), salt, len(salt),
49                         cost, block, parallel, out_buf, out_len)
50            self.assertEqual(ret, 0)
51            self.assertEqual(h(out_buf), utf8(expected))
52
53
54if __name__ == '__main__':
55    unittest.main()
56