blob: 26e1079b605166582dfa0eb193c7e9fe1b8a61db (
plain)
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
|
from __future__ import absolute_import
"""M2Crypto wrapper for OpenSSL RC4 API.
Copyright (c) 1999-2003 Ng Pheng Siong. All rights reserved."""
from M2Crypto.m2 import rc4_free, rc4_new, rc4_set_key, rc4_update
class RC4(object):
"""Object interface to the stream cipher RC4."""
rc4_free = rc4_free
def __init__(self, key=None):
# type: (bytes) -> None
self.cipher = rc4_new()
if key:
rc4_set_key(self.cipher, key)
def __del__(self):
# type: () -> None
if getattr(self, 'cipher', None):
self.rc4_free(self.cipher)
def set_key(self, key):
# type: (bytes) -> None
rc4_set_key(self.cipher, key)
def update(self, data):
# type: (bytes) -> bytes
return rc4_update(self.cipher, data)
def final(self):
# type: () -> str
return ''
|