1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
#!/usr/bin/env python
"""Unit tests for M2Crypto.DSA.
Copyright (c) 2000 Ng Pheng Siong. All rights reserved."""
import hashlib
from M2Crypto import DSA, Rand
from tests import unittest
class DSATestCase(unittest.TestCase):
errkey = 'tests/rsa.priv.pem'
privkey = 'tests/dsa.priv.pem'
pubkey = 'tests/dsa.pub.pem'
param = 'tests/dsa.param.pem'
data = hashlib.sha1(b'Can you spell subliminal channel?').digest()
different_data = hashlib.sha1(b'I can spell.').digest()
def callback(self, *args):
pass
def test_loadkey_junk(self):
with self.assertRaises(DSA.DSAError):
DSA.load_key(self.errkey)
def test_loadkey(self):
dsa = DSA.load_key(self.privkey)
self.assertEqual(len(dsa), 1024)
with self.assertRaises(AttributeError):
getattr(dsa, 'foobar')
for k in ('p', 'q', 'g', 'priv', 'pub'):
with self.assertRaises(DSA.DSAError):
setattr(dsa, k, 1)
def test_loadparam(self):
with self.assertRaises(DSA.DSAError):
DSA.load_key(self.param)
dsa = DSA.load_params(self.param)
assert not dsa.check_key()
self.assertEqual(len(dsa), 1024)
def test_sign(self):
dsa = DSA.load_key(self.privkey)
assert dsa.check_key()
r, s = dsa.sign(self.data)
assert dsa.verify(self.data, r, s)
assert not dsa.verify(self.data, s, r)
def test_sign_asn1(self):
dsa = DSA.load_key(self.privkey)
blob = dsa.sign_asn1(self.data)
assert dsa.verify_asn1(self.data, blob)
def test_sign_with_params_only(self):
dsa = DSA.load_params(self.param)
with self.assertRaises(AssertionError):
dsa.sign(self.data)
with self.assertRaises(AssertionError):
dsa.sign_asn1(self.data)
def test_pub_verify(self):
dsa = DSA.load_key(self.privkey)
r, s = dsa.sign(self.data)
dsapub = DSA.load_pub_key(self.pubkey)
assert dsapub.check_key()
assert dsapub.verify(self.data, r, s)
with self.assertRaises(DSA.DSAError):
dsapub.sign()
def test_verify_fail(self):
dsa = DSA.load_key(self.privkey)
r, s = dsa.sign(self.data)
assert not dsa.verify(self.different_data, r, s)
def test_verify_fail2(self):
dsa = DSA.load_key(self.privkey)
r, s = dsa.sign(self.data)
dsa2 = DSA.load_params(self.param)
assert not dsa2.check_key()
with self.assertRaises(AssertionError):
dsa2.verify(self.data, r, s)
def test_genparam_setparam_genkey(self):
dsa = DSA.gen_params(1024, self.callback)
self.assertEqual(len(dsa), 1024)
p = dsa.p
q = dsa.q
g = dsa.g
dsa2 = DSA.set_params(p, q, g)
assert not dsa2.check_key()
dsa2.gen_key()
assert dsa2.check_key()
r, s = dsa2.sign(self.data)
assert dsa2.verify(self.data, r, s)
def test_pub_key_from_params(self):
dsa = DSA.gen_params(1024, self.callback)
dsa.gen_key()
assert len(dsa) == 1024
p = dsa.p
q = dsa.q
g = dsa.g
pub = dsa.pub
dsa2 = DSA.pub_key_from_params(p, q, g, pub)
assert dsa2.check_key()
r, s = dsa.sign(self.data)
assert dsa2.verify(self.data, r, s)
def suite():
return unittest.TestLoader().loadTestsFromTestCase(DSATestCase)
if __name__ == '__main__':
Rand.load_file('randpool.dat', -1)
unittest.TextTestRunner().run(suite())
Rand.save_file('randpool.dat')
|
