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
|
#!/usr/bin/env python
"""Unit tests for M2Crypto.RC4.
Copyright (c) 2009 Heikki Toivonen. All rights reserved."""
from M2Crypto import RC4, Rand
from binascii import hexlify
from tests import unittest
from tests.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
"""
if fips_mode:
return
vectors = (
(b"Key", b"Plaintext", b"BBF316E8D940AF0AD3"),
(b"Wiki", b"pedia", b"1021BF0420"),
(
b"Secret",
b"Attack at dawn",
b"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):
if fips_mode:
return
rc4 = RC4.RC4(b"foo")
self.assertNotEqual(hexlify(rc4.update(b"bar")).upper(), b"45678")
def suite():
return unittest.TestLoader().loadTestsFromTestCase(RC4TestCase)
if __name__ == "__main__":
Rand.load_file("randpool.dat", -1)
unittest.TextTestRunner().run(suite())
Rand.save_file("randpool.dat")
|
