2013-10-17 14:05:27 +00:00
|
|
|
#!/usr/bin/env python
|
2008-04-26 06:47:56 +00:00
|
|
|
|
2009-11-04 10:35:16 +00:00
|
|
|
""" Python implementation of Arcfour encryption algorithm.
|
|
|
|
|
|
|
|
This code is in the public domain.
|
|
|
|
|
|
|
|
"""
|
2008-08-30 07:40:52 +00:00
|
|
|
|
2013-11-07 08:35:04 +00:00
|
|
|
|
2008-08-30 07:40:52 +00:00
|
|
|
## Arcfour
|
|
|
|
##
|
2008-07-09 15:15:32 +00:00
|
|
|
class Arcfour(object):
|
2008-04-26 06:47:56 +00:00
|
|
|
|
2010-01-04 12:41:23 +00:00
|
|
|
"""
|
|
|
|
>>> Arcfour('Key').process('Plaintext').encode('hex')
|
|
|
|
'bbf316e8d940af0ad3'
|
|
|
|
>>> Arcfour('Wiki').process('pedia').encode('hex')
|
|
|
|
'1021bf0420'
|
|
|
|
>>> Arcfour('Secret').process('Attack at dawn').encode('hex')
|
|
|
|
'45a01f645fc35b383552544b9bf5'
|
|
|
|
"""
|
|
|
|
|
2009-10-24 04:41:59 +00:00
|
|
|
def __init__(self, key):
|
|
|
|
s = range(256)
|
|
|
|
j = 0
|
|
|
|
klen = len(key)
|
|
|
|
for i in xrange(256):
|
|
|
|
j = (j + s[i] + ord(key[i % klen])) % 256
|
|
|
|
(s[i], s[j]) = (s[j], s[i])
|
|
|
|
self.s = s
|
|
|
|
(self.i, self.j) = (0, 0)
|
|
|
|
return
|
|
|
|
|
|
|
|
def process(self, data):
|
|
|
|
(i, j) = (self.i, self.j)
|
|
|
|
s = self.s
|
|
|
|
r = ''
|
|
|
|
for c in data:
|
|
|
|
i = (i+1) % 256
|
|
|
|
j = (j+s[i]) % 256
|
|
|
|
(s[i], s[j]) = (s[j], s[i])
|
|
|
|
k = s[(s[i]+s[j]) % 256]
|
|
|
|
r += chr(ord(c) ^ k)
|
|
|
|
(self.i, self.j) = (i, j)
|
|
|
|
return r
|
2014-05-19 14:27:43 +00:00
|
|
|
|
|
|
|
encrypt = decrypt = process
|
|
|
|
|
|
|
|
new = Arcfour
|
2008-04-26 06:47:56 +00:00
|
|
|
|
2008-08-30 07:40:52 +00:00
|
|
|
# test
|
2008-04-26 06:47:56 +00:00
|
|
|
if __name__ == '__main__':
|
2010-01-04 12:41:23 +00:00
|
|
|
import doctest
|
|
|
|
doctest.testmod()
|