1 // Copyright 2020 The Pigweed Authors
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not
4 // use this file except in compliance with the License. You may obtain a copy of
5 // the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
12 // License for the specific language governing permissions and limitations under
13 // the License.
14
15 #include "pw_bytes/byte_builder.h"
16
17 namespace pw {
18
append(size_t count,std::byte b)19 ByteBuilder& ByteBuilder::append(size_t count, std::byte b) {
20 std::byte* const append_destination = buffer_.data() + size_;
21 std::fill_n(append_destination, ResizeForAppend(count), b);
22 return *this;
23 }
24
append(const void * bytes,size_t count)25 ByteBuilder& ByteBuilder::append(const void* bytes, size_t count) {
26 std::byte* const append_destination = buffer_.data() + size_;
27 std::copy_n(static_cast<const std::byte*>(bytes),
28 ResizeForAppend(count),
29 append_destination);
30 return *this;
31 }
32
ResizeForAppend(size_t bytes_to_append)33 size_t ByteBuilder::ResizeForAppend(size_t bytes_to_append) {
34 if (!status_.ok()) {
35 return 0;
36 }
37
38 if (bytes_to_append > max_size() - size()) {
39 status_ = Status::ResourceExhausted();
40 return 0;
41 }
42
43 size_ += bytes_to_append;
44 status_ = OkStatus();
45 return bytes_to_append;
46 }
47
resize(size_t new_size)48 void ByteBuilder::resize(size_t new_size) {
49 if (new_size <= size_) {
50 size_ = new_size;
51 status_ = OkStatus();
52 } else {
53 status_ = Status::OutOfRange();
54 }
55 }
56
57 } // namespace pw
58