File: RC4.swift

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (111 lines) | stat: -rw-r--r-- 2,782 bytes parent folder | download
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
//===--- RC4.swift --------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2021 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

// This test is based on util/benchmarks/RC4, with modifications
// for performance measuring.
import TestsUtils

public let benchmarks =
  BenchmarkInfo(
    name: "RC4",
    runFunction: run_RC4,
    tags: [.validation, .algorithm])

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()
    }
  }
}

let refResults: [UInt8] = [
  245, 62, 245, 202, 138, 120, 186, 107, 255, 189,
  184, 223, 65, 77, 112, 201, 238, 161, 74, 192, 145,
  21, 43, 41, 91, 136, 182, 176, 237, 155, 208, 16,
  17, 139, 33, 195, 24, 136, 79, 183, 211, 21, 56,
  202, 235, 65, 201, 184, 68, 29, 110, 218, 112, 122,
  194, 77, 41, 230, 147, 84, 0, 233, 168, 6, 55, 131,
  70, 119, 41, 119, 234, 131, 87, 24, 51, 130, 28,
  66, 172, 105, 33, 97, 179, 48, 81, 229, 114, 216,
  208, 119, 39, 31, 47, 109, 172, 215, 246, 210, 48,
  203]


@inline(never)
public func run_RC4(_ n: Int) {
  let messageLen = 100
  let iterations = 500
  let secret = "This is my secret message"
  let key    = "This is my key"
  let secretData : [UInt8] = Array(secret.utf8)
  let keyData    : [UInt8] = Array(key.utf8)

  var longData : [UInt8] = [UInt8](repeating: 0, count: messageLen)

  for _ in 1...n {
    // Generate a long message.
    for i in 0..<messageLen {
      longData[i] = secretData[i % secretData.count]
    }

    var enc = RC4()
    enc.initialize(keyData)

    for _ in 1...iterations {
      enc.encrypt(&longData)
    }

    check(longData == refResults)
  }
}