Swift实现RC4加解密

最新推荐文章于 2022-11-16 17:30:19 发布

土豆吞噬者

最新推荐文章于 2022-11-16 17:30:19 发布

阅读量379

点赞数

分类专栏： Swift

本文链接：https://blog.csdn.net/xiongya8888/article/details/91869380

版权

Swift 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

struct RC4 {
    var State : [UInt8]
    var I: UInt8 = 0
    var J: UInt8 = 0
    
    init() {
        State = [UInt8](repeating: 0, count: 256)
    }
    
    mutating
    func initialize(_ Key: [UInt8]) {
        for i in 0..<256 {
            State[i] = UInt8(i)
        }
        
        var j: UInt8 = 0
        for i in 0..<256 {
            let K : UInt8 = Key[i % Key.count]
            let S : UInt8 = State[i]
            j = j &+ S &+ K
            swapByIndex(i, y: Int(j))
        }
    }
    
    mutating
    func swapByIndex(_ x: Int, y: Int) {
        let T1 : UInt8 = State[x]
        let T2 : UInt8 = State[y]
        State[x] = T2
        State[y] = T1
    }
    
    mutating
    func next() -> UInt8 {
        I = I &+ 1
        J = J &+ State[Int(I)]
        swapByIndex(Int(I), y: Int(J))
        return State[Int(State[Int(I)] &+ State[Int(J)]) & 0xFF]
    }
    
    mutating
    func encrypt(_ Data: inout [UInt8]) {
        let cnt = Data.count
        for i in 0..<cnt {
            Data[i] = Data[i] ^ next()
        }
    }
    
    static func encrypt(data:[UInt8],key:[UInt8])->[UInt8]{
        var ret=data
        var enc=RC4()
        enc.initialize(key)
        enc.encrypt(&ret)
        return ret
    }
}