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
|
//===-------- ExponentialGrowthAppendingBinaryByteStream.cpp --------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
#include "swift/Basic/ExponentialGrowthAppendingBinaryByteStream.h"
using namespace llvm;
using namespace swift;
Error ExponentialGrowthAppendingBinaryByteStream::readBytes(
uint64_t Offset, uint64_t Size, ArrayRef<uint8_t> &Buffer) {
if (auto Error = checkOffsetForRead(Offset, Size)) {
return Error;
}
Buffer = ArrayRef<uint8_t>(Data.data() + Offset, Size);
return Error::success();
}
Error ExponentialGrowthAppendingBinaryByteStream::readLongestContiguousChunk(
uint64_t Offset, ArrayRef<uint8_t> &Buffer) {
if (auto Error = checkOffsetForRead(Offset, 0)) {
return Error;
}
Buffer = ArrayRef<uint8_t>(Data.data() + Offset, Data.size() - Offset);
return Error::success();
}
void ExponentialGrowthAppendingBinaryByteStream::reserve(size_t Size) {
Data.reserve(Size);
}
Error ExponentialGrowthAppendingBinaryByteStream::writeBytes(
uint64_t Offset, ArrayRef<uint8_t> Buffer) {
if (Buffer.empty())
return Error::success();
if (auto Error = checkOffsetForWrite(Offset, Buffer.size())) {
return Error;
}
// Resize the internal buffer if needed.
uint64_t RequiredSize = Offset + Buffer.size();
if (RequiredSize > Data.size()) {
Data.resize(RequiredSize);
}
::memcpy(Data.data() + Offset, Buffer.data(), Buffer.size());
return Error::success();
}
|