sid
/
NazaraEngine
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Core/PoolByteStream: Fix pool returning on Reset

This commit is contained in:
Lynix 2022-08-28 17:54:49 +02:00
parent 50ed8b4028
commit 45c947faf1
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/Nazara/Core/PoolByteStream.hpp
View File

@ -18,8 +18,6 @@ namespace Nz
class NAZARA_CORE_API PoolByteStream : public ByteStream class NAZARA_CORE_API PoolByteStream : public ByteStream
{ {
friend class Network;
public: public:
inline PoolByteStream(ByteArrayPool& pool); inline PoolByteStream(ByteArrayPool& pool);
inline PoolByteStream(ByteArrayPool& pool, std::size_t capacity); inline PoolByteStream(ByteArrayPool& pool, std::size_t capacity);

8
src/Nazara/Core/PoolByteStream.cpp
View File

@ -28,9 +28,13 @@ namespace Nz
void PoolByteStream::Reset(std::size_t capacity) void PoolByteStream::Reset(std::size_t capacity)
{ {
if (m_buffer.GetCapacity() < capacity) if (m_buffer.GetCapacity() < capacity)
m_buffer = m_pool.GetByteArray(capacity); {
if (m_buffer.GetCapacity() > 0)
m_pool.ReturnByteArray(std::move(m_buffer));
SetStream(&m_buffer, Nz::OpenMode_ReadWrite); m_buffer = m_pool.GetByteArray(capacity);
SetStream(&m_buffer, Nz::OpenMode_ReadWrite);
}
} }
void PoolByteStream::OnEmptyStream() void PoolByteStream::OnEmptyStream()