1 // Copyright 2019 The Android Open Source Project
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://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,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "base/MemStream.h"
16
17 #include "base/StreamSerializing.h"
18
19 #include <algorithm>
20 #include <utility>
21
22 #include <string.h>
23
24 namespace android {
25 namespace base {
26
MemStream(int reserveSize)27 MemStream::MemStream(int reserveSize) {
28 mData.reserve(reserveSize);
29 }
30
MemStream(Buffer && data)31 MemStream::MemStream(Buffer&& data) : mData(std::move(data)) {}
32
read(void * buffer,size_t size)33 ssize_t MemStream::read(void* buffer, size_t size) {
34 const auto sizeToRead = std::min<int>(size, readSize());
35 memcpy(buffer, mData.data() + mReadPos, sizeToRead);
36 mReadPos += sizeToRead;
37 return sizeToRead;
38 }
39
write(const void * buffer,size_t size)40 ssize_t MemStream::write(const void* buffer, size_t size) {
41 mData.insert(mData.end(), (const char*)buffer, (const char*)buffer + size);
42 return size;
43 }
44
writtenSize() const45 int MemStream::writtenSize() const {
46 return (int)mData.size();
47 }
48
readPos() const49 int MemStream::readPos() const {
50 return mReadPos;
51 }
52
readSize() const53 int MemStream::readSize() const {
54 return mData.size() - mReadPos;
55 }
56
save(Stream * stream) const57 void MemStream::save(Stream* stream) const {
58 saveBuffer(stream, mData);
59 }
60
load(Stream * stream)61 void MemStream::load(Stream* stream) {
62 loadBuffer(stream, &mData);
63 mReadPos = 0;
64 }
65
66 } // namespace base
67 } // namespace android
68