# Author: Trevor Perrin
# See the LICENSE file for legal information regarding use of this file.
"""OpenSSL/M2Crypto RC4 implementation."""
from .cryptomath import *
from .rc4 import RC4
if m2cryptoLoaded:
[docs] def new(key):
return OpenSSL_RC4(key)
[docs] class OpenSSL_RC4(RC4):
[docs] def __init__(self, key):
RC4.__init__(self, key, "openssl")
self.rc4 = m2.rc4_new()
m2.rc4_set_key(self.rc4, key)
[docs] def __del__(self):
m2.rc4_free(self.rc4)
[docs] def encrypt(self, plaintext):
return bytearray(m2.rc4_update(self.rc4, plaintext))
[docs] def decrypt(self, ciphertext):
return bytearray(self.encrypt(ciphertext))