-
Notifications
You must be signed in to change notification settings - Fork 30
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Generation reworked: Each test vector is generated entirely ahead of time then cut into bits instead of generated segment by segment * Added new TVGenerator, `PatternGenerator`, that is not random and uses the following pattern(s) depending on the iteration count: * All 0 * All 1 * Half 0, Half 1 * Half 1, Half 0 * Alternating 0s and 1s * Alternating 1s and 0s * Moving `max(1, min(32, bits/4))` window of 1s (even) and the complement of that (odd) * TVGenerators now take a seed, fixed default seed but can be changed over CLI * Fixed test * Incorporated `ARC4RandomNumberGenerator`from Swift TensorFlow sources (takes a seed)
- Loading branch information
Showing
6 changed files
with
313 additions
and
195 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,112 @@ | ||
| //===-- Random.swift ------------------------------------------*- swift -*-===// | ||
| // | ||
| // This source file is part of the Swift.org open source project | ||
| // | ||
| // Copyright (c) 2014 - 2018 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 | ||
| // | ||
| //===----------------------------------------------------------------------===// | ||
| // | ||
| // A (pseudo-)random number generation library. The library separates concerns | ||
| // into engines which generate random bytes and distributions which use an | ||
| // engine to generate values from some statistical distribution. | ||
| // | ||
| //===----------------------------------------------------------------------===// | ||
|
|
||
| #if os(macOS) || os(iOS) || os(watchOS) || os(tvOS) | ||
| import Darwin | ||
| #else | ||
| import Glibc | ||
| #endif | ||
|
|
||
| /// A type that provides seedable deterministic pseudo-random data. | ||
| /// | ||
| /// A SeedableRandomNumberGenerator can be used anywhere where a | ||
| /// RandomNumberGenerator would be used. It is useful when the pseudo-random | ||
| /// data needs to be reproducible across runs. | ||
| /// | ||
| /// Conforming to the SeedableRandomNumberGenerator Protocol | ||
| /// ======================================================== | ||
| /// | ||
| /// To make a custom type conform to the `SeedableRandomNumberGenerator` | ||
| /// protocol, implement the `init(seed: [UInt8])` initializer, as well as the | ||
| /// requirements for `RandomNumberGenerator`. The values returned by `next()` | ||
| /// must form a deterministic sequence that depends only on the seed provided | ||
| /// upon initialization. | ||
| public protocol SeedableRandomNumberGenerator: RandomNumberGenerator { | ||
| init(seed: [UInt8]) | ||
| init<T: BinaryInteger>(seed: T) | ||
| } | ||
|
|
||
| extension SeedableRandomNumberGenerator { | ||
| public init<T: BinaryInteger>(seed: T) { | ||
| var newSeed: [UInt8] = [] | ||
| for i in 0..<seed.bitWidth / UInt8.bitWidth { | ||
| newSeed.append(UInt8(truncatingIfNeeded: seed >> (UInt8.bitWidth * i))) | ||
| } | ||
| self.init(seed: newSeed) | ||
| } | ||
| } | ||
|
|
||
| /// An implementation of `SeedableRandomNumberGenerator` using ARC4. | ||
| /// | ||
| /// ARC4 is a stream cipher that generates a pseudo-random stream of bytes. This | ||
| /// PRNG uses the seed as its key. | ||
| /// | ||
| /// ARC4 is described in Schneier, B., "Applied Cryptography: Protocols, | ||
| /// Algorithms, and Source Code in C", 2nd Edition, 1996. | ||
| /// | ||
| /// An individual generator is not thread-safe, but distinct generators do not | ||
| /// share state. The random data generated is of high-quality, but is not | ||
| /// suitable for cryptographic applications. | ||
| @frozen | ||
| public struct ARC4RandomNumberGenerator: SeedableRandomNumberGenerator { | ||
| public static var global = ARC4RandomNumberGenerator(seed: UInt32(time(nil))) | ||
| var state: [UInt8] = Array(0...255) | ||
| var iPos: UInt8 = 0 | ||
| var jPos: UInt8 = 0 | ||
|
|
||
| /// Initialize ARC4RandomNumberGenerator using an array of UInt8. The array | ||
| /// must have length between 1 and 256 inclusive. | ||
| public init(seed: [UInt8]) { | ||
| precondition(seed.count > 0, "Length of seed must be positive") | ||
| precondition(seed.count <= 256, "Length of seed must be at most 256") | ||
| var j: UInt8 = 0 | ||
| for i: UInt8 in 0...255 { | ||
| j &+= S(i) &+ seed[Int(i) % seed.count] | ||
| swapAt(i, j) | ||
| } | ||
| } | ||
|
|
||
| // Produce the next random UInt64 from the stream, and advance the internal | ||
| // state. | ||
| public mutating func next() -> UInt64 { | ||
| var result: UInt64 = 0 | ||
| for _ in 0..<UInt64.bitWidth / UInt8.bitWidth { | ||
| result <<= UInt8.bitWidth | ||
| result += UInt64(nextByte()) | ||
| } | ||
| return result | ||
| } | ||
|
|
||
| // Helper to access the state. | ||
| private func S(_ index: UInt8) -> UInt8 { | ||
| return state[Int(index)] | ||
| } | ||
|
|
||
| // Helper to swap elements of the state. | ||
| private mutating func swapAt(_ i: UInt8, _ j: UInt8) { | ||
| state.swapAt(Int(i), Int(j)) | ||
| } | ||
|
|
||
| // Generates the next byte in the keystream. | ||
| private mutating func nextByte() -> UInt8 { | ||
| iPos &+= 1 | ||
| jPos &+= S(iPos) | ||
| swapAt(iPos, jPos) | ||
| return S(S(iPos) &+ S(jPos)) | ||
| } | ||
| } |
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.