杨锴
2024-08-14 909e20941e45f8712c012db602034b47da0bfdb0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
//
//  CryptoSwift
//
//  Copyright (C) 2014-2022 Marcin Krzyżanowski <marcin@krzyzanowskim.com>
//  This software is provided 'as-is', without any express or implied warranty.
//
//  In no event will the authors be held liable for any damages arising from the use of this software.
//
//  Permission is granted to anyone to use this software for any purpose,including commercial applications, and to alter it and redistribute it freely, subject to the following restrictions:
//
//  - The origin of this software must not be misrepresented; you must not claim that you wrote the original software. If you use this software in a product, an acknowledgment in the product documentation is required.
//  - Altered source versions must be plainly marked as such, and must not be misrepresented as being the original software.
//  - This notice may not be removed or altered from any source or binary distribution.
//
 
import Foundation
 
extension Data {
  /// Two octet checksum as defined in RFC-4880. Sum of all octets, mod 65536
  public func checksum() -> UInt16 {
    let s = self.withUnsafeBytes { buf in
        return buf.lazy.map(UInt32.init).reduce(UInt32(0), +)
    }
    return UInt16(s % 65535)
  }
 
  public func md5() -> Data {
    Data( Digest.md5(bytes))
  }
 
  public func sha1() -> Data {
    Data( Digest.sha1(bytes))
  }
 
  public func sha224() -> Data {
    Data( Digest.sha224(bytes))
  }
 
  public func sha256() -> Data {
    Data( Digest.sha256(bytes))
  }
 
  public func sha384() -> Data {
    Data( Digest.sha384(bytes))
  }
 
  public func sha512() -> Data {
    Data( Digest.sha512(bytes))
  }
 
  public func sha3(_ variant: SHA3.Variant) -> Data {
    Data( Digest.sha3(bytes, variant: variant))
  }
 
  public func crc32(seed: UInt32? = nil, reflect: Bool = true) -> Data {
    Data( Checksum.crc32(bytes, seed: seed, reflect: reflect).bytes())
  }
 
  public func crc32c(seed: UInt32? = nil, reflect: Bool = true) -> Data {
    Data( Checksum.crc32c(bytes, seed: seed, reflect: reflect).bytes())
  }
 
  public func crc16(seed: UInt16? = nil) -> Data {
    Data( Checksum.crc16(bytes, seed: seed).bytes())
  }
 
  public func encrypt(cipher: Cipher) throws -> Data {
    Data( try cipher.encrypt(bytes.slice))
  }
 
  public func decrypt(cipher: Cipher) throws -> Data {
    Data( try cipher.decrypt(bytes.slice))
  }
 
  public func authenticate(with authenticator: Authenticator) throws -> Data {
    Data( try authenticator.authenticate(bytes))
  }
}
 
extension Data {
  public init(hex: String) {
    self.init(Array<UInt8>(hex: hex))
  }
 
  public var bytes: Array<UInt8> {
    Array(self)
  }
 
  public func toHexString() -> String {
    self.bytes.toHexString()
  }
}