python实现rc4加解密算法

下面是 Python 实现 RC4 加解密算法的示例代码：

def KSA(key):
    """
    Key-scheduling algorithm (KSA)
    """
    key_length = len(key)
    # 初始化 S 数组
    S = list(range(256))
    j = 0
    for i in range(256):
        j = (j + S[i] + key[i % key_length]) % 256
        S[i], S[j] = S[j], S[i]  # 交换 S[i] 和 S[j]
    return S

def PRGA(S):
    """
    Pseudo-random generation algorithm (PRGA)
    """
    i = 0
    j = 0
    while True:
        i = (i + 1) % 256
        j = (j + S[i]) % 256
        S[i], S[j] = S[j], S[i]  # 交换 S[i] 和 S[j]
        K = S[(S[i] + S[j]) % 256]
        yield K

def RC4(key, plaintext):
    """
    RC4 加密/解密
    """
    S = KSA(key)
    keystream = PRGA(S)
    ciphertext = bytearray()
    for b in plaintext:
        # 生成密钥流
        keystream_byte = next(keystream)
        # 将密钥流和明文数据异或
        ciphertext_byte = b ^ keystream_byte
        ciphertext.append(ciphertext_byte)
    return ciphertext

在上面的代码中，`KSA` 函数用于进行密钥调度，生成初始的 S 数组。`PRGA` 函数用于生成伪随机数流，实现加密/解密过程。`RC4` 函数用于调用 `KSA` 和 `PRGA` 函数，完成 RC4 加密/解密操作。需要注意的是，RC4 加密算法是一种对称加密算法，也就是加密和解密使用的是同一个密钥。

下面是使用示例：

# 加密示例
key = b'secret_key'
plaintext = b'Hello, world!'
ciphertext = RC4(key, plaintext)
print(ciphertext)

# 解密示例
decrypted_plaintext = RC4(key, ciphertext)
print(decrypted_plaintext)

在上面的示例中，我们分别使用 `RC4` 函数对明文数据进行加密和解密，需要传入一个密钥和要加密/解密的数据。在示例中，我们使用了一个字节数组作为密钥和明文数据，但实际上你可以使用任何类型的数据作为密钥和明文数据。