blob: 9f616b8bb3acb3b9e4a887503dc6b8d10981c20d [file] [log] [blame]
/*
* Copyright (C) 2011 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "ReadBuffer.h"
#include "ErrorLog.h"
#include <algorithm>
#include <assert.h>
#include <string.h>
#include <limits.h>
ReadBuffer::ReadBuffer(size_t bufsize) {
m_size = bufsize;
m_buf = (unsigned char*)malloc(m_size);
m_validData = 0;
m_readPtr = m_buf;
}
ReadBuffer::~ReadBuffer() {
free(m_buf);
}
int ReadBuffer::getData(IOStream* stream, int minSize) {
assert(stream);
assert(minSize > (int)m_validData);
const int minSizeToRead = minSize - m_validData;
int maxSizeToRead;
const int freeTailSize = m_buf + m_size - (m_readPtr + m_validData);
if (freeTailSize >= minSizeToRead) {
maxSizeToRead = freeTailSize;
} else {
if (freeTailSize + (m_readPtr - m_buf) >= minSizeToRead) {
// There's some gap in the beginning, if we move the data over it
// that's going to be enough.
memmove(m_buf, m_readPtr, m_validData);
} else {
// Not enough space even with moving, reallocate.
// Note: make sure we can fit at least two of the requested packets
// into the new buffer to minimize the reallocations and
// memmove()-ing stuff around.
size_t new_size = std::max<size_t>(
2 * minSizeToRead + m_validData,
2 * m_size);
if (new_size < m_size) { // overflow check
new_size = INT_MAX;
}
const auto new_buf = (unsigned char*)malloc(new_size);
if (!new_buf) {
ERR("Failed to alloc %zu bytes for ReadBuffer\n", new_size);
return -1;
}
memcpy(new_buf, m_readPtr, m_validData);
free(m_buf);
m_buf = new_buf;
m_size = new_size;
}
// We can read more now, let's request it in case all data is ready
// for reading.
maxSizeToRead = m_size - m_validData;
m_readPtr = m_buf;
}
// get fresh data into the buffer;
int readTotal = 0;
do {
const size_t readNow = stream->read(m_readPtr + m_validData,
maxSizeToRead - readTotal);
if (!readNow) {
return -1;
}
readTotal += readNow;
m_validData += readNow;
} while (readTotal < minSizeToRead);
return readTotal;
}
void ReadBuffer::consume(size_t amount) {
assert(amount <= m_validData);
m_validData -= amount;
m_readPtr += amount;
}