-
-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
25 changed files
with
398 additions
and
241 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
Large diffs are not rendered by default.
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
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,84 @@ | ||
| // CryptoSwift | ||
| // | ||
| // Copyright (C) 2014-__YEAR__ Marcin Krzyżanowski <[email protected]> | ||
| // 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. | ||
| // | ||
|
|
||
| public class BlockDecryptor: Cryptor, Updatable { | ||
| private let blockSize: Int | ||
| private let padding: Padding | ||
| private var worker: CipherModeWorker | ||
| private var accumulated = Array<UInt8>() | ||
|
|
||
| init(blockSize: Int, padding: Padding, _ worker: CipherModeWorker) throws { | ||
| self.blockSize = blockSize | ||
| self.padding = padding | ||
| self.worker = worker | ||
| } | ||
|
|
||
| public func update(withBytes bytes: ArraySlice<UInt8>, isLast: Bool = false) throws -> Array<UInt8> { | ||
| accumulated += bytes | ||
|
|
||
| // If a worker (eg GCM) can combine ciphertext + tag | ||
| // we need to remove tag from the ciphertext. | ||
| if !isLast && accumulated.count < blockSize + worker.additionalBufferSize { | ||
| return [] | ||
| } | ||
|
|
||
| let accumulatedWithoutSuffix: Array<UInt8> | ||
| if worker.additionalBufferSize > 0 { | ||
| // FIXME: how slow is that? | ||
| accumulatedWithoutSuffix = Array(accumulated.prefix(accumulated.count - worker.additionalBufferSize)) | ||
| } else { | ||
| accumulatedWithoutSuffix = accumulated | ||
| } | ||
|
|
||
| var processedBytesCount = 0 | ||
| var plaintext = Array<UInt8>(reserveCapacity: accumulatedWithoutSuffix.count) | ||
| // Processing in a block-size manner. It's good for block modes, but bad for stream modes. | ||
| for var chunk in accumulatedWithoutSuffix.batched(by: blockSize) { | ||
| if isLast || (accumulatedWithoutSuffix.count - processedBytesCount) >= blockSize { | ||
|
|
||
| if isLast, var finalizingWorker = worker as? BlockModeWorkerFinalizing { | ||
| chunk = try finalizingWorker.willDecryptLast(block: chunk + accumulated.suffix(worker.additionalBufferSize)) // tag size | ||
| } | ||
|
|
||
| if !chunk.isEmpty { | ||
| plaintext += worker.decrypt(block: chunk) | ||
| } | ||
|
|
||
| if var finalizingWorker = worker as? BlockModeWorkerFinalizing, isLast == true { | ||
| plaintext = try finalizingWorker.didDecryptLast(block: plaintext.slice) | ||
| } | ||
|
|
||
| processedBytesCount += chunk.count | ||
| } | ||
| } | ||
| accumulated.removeFirst(processedBytesCount) // super-slow | ||
|
|
||
| if isLast { | ||
| plaintext = padding.remove(from: plaintext, blockSize: blockSize) | ||
| } | ||
|
|
||
| return plaintext | ||
| } | ||
|
|
||
| public func seek(to position: Int) throws { | ||
| guard var worker = self.worker as? StreamModeWorker else { | ||
| fatalError("Not supported") | ||
| } | ||
|
|
||
| try worker.seek(to: position) | ||
| self.worker = worker | ||
|
|
||
| accumulated = [] | ||
| } | ||
| } |
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,57 @@ | ||
| // CryptoSwift | ||
| // | ||
| // Copyright (C) 2014-__YEAR__ Marcin Krzyżanowski <[email protected]> | ||
| // 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. | ||
| // | ||
| final class BlockEncryptor: Cryptor, Updatable { | ||
| private let blockSize: Int | ||
| private var worker: CipherModeWorker | ||
| private let padding: Padding | ||
| // Accumulated bytes. Not all processed bytes. | ||
| private var accumulated = Array<UInt8>(reserveCapacity: 16) | ||
|
|
||
| private var lastBlockRemainder = 0 | ||
|
|
||
| init(blockSize: Int, padding: Padding, _ worker: CipherModeWorker) throws { | ||
| self.blockSize = blockSize | ||
| self.padding = padding | ||
| self.worker = worker | ||
| } | ||
|
|
||
| // MARK: Updatable | ||
| public func update(withBytes bytes: ArraySlice<UInt8>, isLast: Bool) throws -> Array<UInt8> { | ||
| accumulated += bytes | ||
|
|
||
| if isLast { | ||
| accumulated = padding.add(to: accumulated, blockSize: blockSize) | ||
| } | ||
|
|
||
| var encrypted = Array<UInt8>(reserveCapacity: accumulated.count) | ||
| for chunk in accumulated.batched(by: blockSize) { | ||
| if isLast || chunk.count == blockSize { | ||
| encrypted += worker.encrypt(block: chunk) | ||
| } | ||
| } | ||
|
|
||
| // Stream encrypts all, so it removes all elements | ||
| accumulated.removeFirst(encrypted.count) | ||
|
|
||
| if var finalizingWorker = worker as? BlockModeWorkerFinalizing, isLast == true { | ||
| encrypted = try finalizingWorker.finalize(encrypt: encrypted.slice) | ||
| } | ||
|
|
||
| return encrypted | ||
| } | ||
|
|
||
| func seek(to: Int) throws { | ||
| fatalError("Not supported") | ||
| } | ||
| } |
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
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.