using NewLife.Collections;

namespace NewLife.Security;

/// <summary>RC4对称加密算法</summary>
/// <remarks>
/// RC4于1987年提出，和DES算法一样，是一种对称加密算法，也就是说使用的密钥为单钥（或称为私钥）。
/// 但不同于DES的是，RC4不是对明文进行分组处理，而是字节流的方式依次加密明文中的每一个字节，解密的时候也是依次对密文中的每一个字节进行解密。
/// 
/// RC4算法的特点是算法简单，运行速度快，而且密钥长度是可变的，可变范围为1-256字节(8-2048比特)，
/// 在如今技术支持的前提下，当密钥长度为128比特时，用暴力法搜索密钥已经不太可行，所以可以预见RC4的密钥范围任然可以在今后相当长的时间里抵御暴力搜索密钥的攻击。
/// 实际上，如今也没有找到对于128bit密钥长度的RC4加密算法的有效攻击方法。
/// </remarks>
class RC4
{
    /// <summary>加密</summary>
    /// <param name="data">数据</param>
    /// <param name="pass">密码</param>
    /// <returns></returns>
    public static Byte[] Encrypt(Byte[] data, Byte[] pass)
    {
        if (data == null || data.Length == 0) return [];
        if (pass == null || pass.Length == 0) return data;

        var output = new Byte[data.Length];
        var i = 0;
        var j = 0;
        var box = GetKey(pass, 256);
        // 加密  
        for (var k = 0; k < data.Length; k++)
        {
            i = (i + 1) % box.Length;
            j = (j + box[i]) % box.Length;
            var temp = box[i];
            box[i] = box[j];
            box[j] = temp;
            var a = data[k];
            var b = box[(box[i] + box[j]) % box.Length];
            output[k] = (Byte)(a ^ b);
        }

        return output;
    }

    /// <summary>打乱密码</summary>  
    /// <param name="pass">密码</param>  
    /// <param name="len">密码箱长度</param>  
    /// <returns>打乱后的密码</returns>  
    private static Byte[] GetKey(Byte[] pass, Int32 len)
    {
        var box = new Byte[len];
        for (var i = 0; i < len; i++)
        {
            box[i] = (Byte)i;
        }
        var j = 0;
        for (var i = 0; i < len; i++)
        {
            j = (j + box[i] + pass[i % pass.Length]) % len;
            var temp = box[i];
            box[i] = box[j];
            box[j] = temp;
        }
        return box;
    }
}