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
|
#!/usr/bin/env python
"""Unit tests for M2Crypto.RC4.
Copyright (c) 2009 Heikki Toivonen. All rights reserved."""
try:
import unittest2 as unittest
except ImportError:
import unittest
from M2Crypto import RC4, Rand
from binascii import hexlify
from fips import fips_mode
class RC4TestCase(unittest.TestCase):
@unittest.skipIf(fips_mode, "Can't be run in FIPS mode")
def test_vectors(self):
"""
Test with test vectors from Wikipedia: http://en.wikipedia.org/wiki/Rc4
"""
vectors = (('Key', 'Plaintext', 'BBF316E8D940AF0AD3'),
('Wiki', 'pedia', '1021BF0420'),
('Secret', 'Attack at dawn',
'45A01F645FC35B383552544B9BF5'))
rc4 = RC4.RC4()
for key, plaintext, ciphertext in vectors:
rc4.set_key(key)
self.assertEqual(hexlify(rc4.update(plaintext)).upper(),
ciphertext)
self.assertEqual(rc4.final(), '')
@unittest.skipIf(fips_mode, "Can't be run in FIPS mode")
def test_bad(self):
rc4 = RC4.RC4('foo')
self.assertNotEqual(hexlify(rc4.update('bar')).upper(), '45678')
def suite():
return unittest.makeSuite(RC4TestCase)
if __name__ == '__main__':
Rand.load_file('randpool.dat', -1)
unittest.TextTestRunner().run(suite())
Rand.save_file('randpool.dat')
|
