- Instead of adjusted len, buffer read() / write() returns exception in case of length mismatch
git-svn-id: http://moon:8086/svn/software/trunk/libsrc/cpp@1027 b431acfa-c32f-4a4a-93f1-934dc6c82436
This commit is contained in:
@@ -61,8 +61,8 @@ class Buffer : public IBuffer
|
|||||||
T* data_w();
|
T* data_w();
|
||||||
T* data_r();
|
T* data_r();
|
||||||
|
|
||||||
size_t write(const T* pData, size_t len);
|
Result write(const T* pData, size_t len);
|
||||||
size_t read(T* pData, size_t len);
|
Result read(T* pData, size_t len);
|
||||||
T readAt(size_t offset);
|
T readAt(size_t offset);
|
||||||
Result consume(size_t len);
|
Result consume(size_t len);
|
||||||
Result produce(size_t len);
|
Result produce(size_t len);
|
||||||
|
|||||||
@@ -94,35 +94,35 @@ T* Buffer<T>::data_r()
|
|||||||
}
|
}
|
||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
size_t Buffer<T>::write(const T *pData, size_t len)
|
typename Buffer<T>::Result Buffer<T>::write(const T *pData, size_t len)
|
||||||
{
|
{
|
||||||
len = std::min(free(), len);
|
if (len > free())
|
||||||
size_t numWritten = 0;
|
|
||||||
|
|
||||||
if (len == 0)
|
|
||||||
{
|
{
|
||||||
throw Buffer::Exception(Result::Err_InvalidSize);
|
RETURN_OR_EXCEPT(Result::Err_InvalidSize);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
size_t numWritten = 0;
|
||||||
while(len)
|
while(len)
|
||||||
{
|
{
|
||||||
size_t num_remain = m_capacity - m_wi;
|
size_t num_remain = m_capacity - m_wi;
|
||||||
size_t actual_len = std::min(num_remain, len);
|
size_t actual_len = std::min(num_remain, len);
|
||||||
|
|
||||||
memcpy(data_w(), pData, actual_len*sizeof(T));
|
memcpy(data_w(), pData, actual_len*sizeof(T));
|
||||||
produce(actual_len);
|
|
||||||
len -= actual_len;
|
len -= actual_len;
|
||||||
pData += actual_len;
|
pData += actual_len;
|
||||||
numWritten += actual_len;
|
numWritten += actual_len;
|
||||||
}
|
}
|
||||||
return numWritten;
|
return produce(numWritten);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
size_t Buffer<T>::read(T* pData, size_t len)
|
typename Buffer<T>::Result Buffer<T>::read(T* pData, size_t len)
|
||||||
{
|
{
|
||||||
len = std::min(m_len, len);
|
if (len > m_len)
|
||||||
|
{
|
||||||
|
RETURN_OR_EXCEPT(Result::Err_InvalidSize);
|
||||||
|
}
|
||||||
|
|
||||||
size_t numRead = 0;
|
size_t numRead = 0;
|
||||||
while(len)
|
while(len)
|
||||||
{
|
{
|
||||||
@@ -134,15 +134,13 @@ size_t Buffer<T>::read(T* pData, size_t len)
|
|||||||
pData += actual_len;
|
pData += actual_len;
|
||||||
numRead += actual_len;
|
numRead += actual_len;
|
||||||
}
|
}
|
||||||
consume(numRead);
|
return consume(numRead);
|
||||||
return numRead;
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
template<typename T>
|
template<typename T>
|
||||||
T Buffer<T>::readAt(size_t offset)
|
T Buffer<T>::readAt(size_t offset)
|
||||||
{
|
{
|
||||||
if (m_len < offset)
|
if (m_len < (1+offset))
|
||||||
{
|
{
|
||||||
throw Buffer::Exception(Result::Err_InvalidSize);
|
throw Buffer::Exception(Result::Err_InvalidSize);
|
||||||
}
|
}
|
||||||
@@ -168,7 +166,7 @@ typename Buffer<T>::Result Buffer<T>::consume(size_t len)
|
|||||||
template<typename T>
|
template<typename T>
|
||||||
typename Buffer<T>::Result Buffer<T>::produce(size_t len)
|
typename Buffer<T>::Result Buffer<T>::produce(size_t len)
|
||||||
{
|
{
|
||||||
if (len > (m_capacity - m_len))
|
if (len > free())
|
||||||
{
|
{
|
||||||
RETURN_OR_EXCEPT(Result::Err_InvalidSize);
|
RETURN_OR_EXCEPT(Result::Err_InvalidSize);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user