Swift中的Checksum和XOR

nfg76nw0  于 12个月前  发布在  Swift
关注(0)|答案(5)|浏览(96)

我把这些方法写在C语言中。他们只是校验和和异或一些NSData

- (void)XOR:(NSMutableData *)inputData withKey:(NSData *)key
{
    unsigned char* inputByteData = (unsigned char*)[inputData mutableBytes];
    unsigned char* keyByteData   = (unsigned char*)[key bytes];
    for (int i = 0; i < [inputData length]; i++)
    {
        inputByteData[i] = inputByteData[i] ^ keyByteData[i % [key length]];
    }
}

- (Byte)checkSum:(NSMutableData *)data withLength:(Byte)dataLength
{
    Byte * dataByte = (Byte *)malloc(dataLength);
    memcpy(dataByte, [data bytes], dataLength);

    Byte result = 0;
    int count = 0;
    while (dataLength>0) {
        result += dataByte[count];
        dataLength--;
        count++;
    };
    result = result&0xff;
    return result&0xff;
}

然而,我不熟悉位运算符,特别是在Swift中,这些UnsafeMutablePointer<Void> ...东西
有人能帮我转换吗?(基本上,我需要校验和和异或功能)
还有一件事,它们应该放在NSData/NSMutableData扩展中吗?
谢谢

46scxncf

46scxncf1#

UnsafeBufferPointer/UnsafeMutableBufferPointer可能是您现在需要的。我试着把你的代码翻译成Swift。(但代码没有经过很好的测试。

func XOR(inputData: NSMutableData, withKey key: NSData) {
    let b = UnsafeMutableBufferPointer<UInt8>(start:
        UnsafeMutablePointer(inputData.mutableBytes), count: inputData.length)

    let k = UnsafeBufferPointer<UInt8>(start:
        UnsafePointer(key.bytes), count: key.length)

    for i in 0..<inputData.length {
        b[i] ^= k[i % key.length]
    }
}

func checkSum(data: NSData) -> Int {
    let b = UnsafeBufferPointer<UInt8>(start:
        UnsafePointer(data.bytes), count: data.length)

    var sum = 0
    for i in 0..<data.length {
        sum += Int(b[i])
    }
    return sum & 0xff
}
dfddblmv

dfddblmv2#

Swift 3更新:

public extension Data {

    public mutating func xor(key: Data) {
        for i in 0..<self.count {
            self[i] ^= key[i % key.count]
        }
    }

    public func checkSum() -> Int {
        return self.map { Int($0) }.reduce(0, +) & 0xff
    }
}

您还可以创建另一个函数:xored(key: Data) -> Data
然后你可以链接这些操作符:xored(key).checksum()

ghhkc1vu

ghhkc1vu3#

Swift支持运算符重载,让你可以轻松做let xorData = data1 ^ data2。我写了一个扩展,用于非相似大小的数据到xor。

extension Data {
    static func ^ (left: Data, right: Data) -> Data {
        if left.count != right.count {
            NSLog("Warning! XOR operands are not equal. left = \(left), right = \(right)")
        }

        var result: Data = Data()
        var smaller: Data, bigger: Data
        if left.count <= right.count {
            smaller = left
            bigger = right
        } else {
            smaller = right
            bigger = left
        }

        let bs:[UInt8] = Array(smaller)
        let bb:[UInt8] = Array (bigger)
        var br = [UInt8] ()
        for i in 0..<bs.count {
            br.append(bs[i] ^ bb[i])
        }
        for j in bs.count..<bb.count {
            br.append(bb[j])
        }
        result = Data(br)
        return result
    }
}
laik7k3q

laik7k3q4#

Swift 3更新:

func xor(data: Data, with key: Data) -> Data {
    var xorData = data

    xorData.withUnsafeMutableBytes { (start: UnsafeMutablePointer<UInt8>) -> Void in
        key.withUnsafeBytes { (keyStart: UnsafePointer<UInt8>) -> Void in
            let b = UnsafeMutableBufferPointer<UInt8>(start: start, count: xorData.count)

            let k = UnsafeBufferPointer<UInt8>(start: keyStart, count: data.count)
            let length = data.count

            for i in 0..<xorData.count {
                b[i] ^= k[i % length]
            }
        }
    }

    return xorData
}
lqfhib0f

lqfhib0f5#

Swift可以很容易地将Data对象视为字节数组,您可以使用zipmap的组合进行操作,从而避免了for循环的需要。这导致了更简洁和更可读的代码:

extension Data {
    static func ^ (lhs: Data, rhs: Data) -> Data {
        Data(zip(Array(lhs), Array(rhs)).map { $0 ^ $1 })
    }
}

相关问题