49 lines
1.3 KiB
Swift
49 lines
1.3 KiB
Swift
//
|
|
// Copyright 2019 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
//
|
|
|
|
// Some parts Copyright (c) 2014 - 2016 Apple Inc. and the project authors
|
|
// Those parts Licensed under Apache License v2.0 with Runtime Library Exception
|
|
|
|
import Foundation
|
|
|
|
enum ChunkedInputStreamError: Error {
|
|
case truncated
|
|
case malformed
|
|
}
|
|
|
|
struct ChunkedInputStream {
|
|
private var remainingData: Data
|
|
|
|
init(forReadingFrom dataValue: Data) {
|
|
self.remainingData = dataValue
|
|
}
|
|
|
|
mutating func decodeData(count: Int) throws -> Data {
|
|
let result = self.remainingData.prefix(count)
|
|
try consume(length: count)
|
|
return result
|
|
}
|
|
|
|
/// Based on SwiftProtobuf.BinaryDecoder.decodeSingularUInt32Field
|
|
mutating func decodeSingularUInt32Field() throws -> UInt32 {
|
|
guard let result = UInt32(exactly: try remainingData.removeFirstVarint()) else {
|
|
throw ChunkedInputStreamError.malformed
|
|
}
|
|
return result
|
|
}
|
|
|
|
var isEmpty: Bool {
|
|
return self.remainingData.isEmpty
|
|
}
|
|
|
|
/// Advance the current position.
|
|
private mutating func consume(length: Int) throws {
|
|
guard self.remainingData.count >= length else {
|
|
throw ChunkedInputStreamError.truncated
|
|
}
|
|
self.remainingData = self.remainingData.dropFirst(length)
|
|
}
|
|
}
|