OLD | NEW |
1 """PyCrypto RC4 implementation.""" | 1 """PyCrypto RC4 implementation.""" |
2 | 2 |
3 from cryptomath import * | 3 from cryptomath import * |
4 from RC4 import * | 4 from rc4 import * |
5 | 5 |
6 if pycryptoLoaded: | 6 if pycryptoLoaded: |
7 import Crypto.Cipher.ARC4 | 7 import Crypto.Cipher.ARC4 |
8 | 8 |
9 def new(key): | 9 def new(key): |
10 return PyCrypto_RC4(key) | 10 return PyCrypto_RC4(key) |
11 | 11 |
12 class PyCrypto_RC4(RC4): | 12 class PyCrypto_RC4(RC4): |
13 | 13 |
14 def __init__(self, key): | 14 def __init__(self, key): |
15 RC4.__init__(self, key, "pycrypto") | 15 RC4.__init__(self, key, "pycrypto") |
16 self.context = Crypto.Cipher.ARC4.new(key) | 16 self.context = Crypto.Cipher.ARC4.new(key) |
17 | 17 |
18 def encrypt(self, plaintext): | 18 def encrypt(self, plaintext): |
19 return self.context.encrypt(plaintext) | 19 return self.context.encrypt(plaintext) |
20 | 20 |
21 def decrypt(self, ciphertext): | 21 def decrypt(self, ciphertext): |
22 return self.context.decrypt(ciphertext) | 22 return self.context.decrypt(ciphertext) |
OLD | NEW |