diff --git a/src/core/util/byte_stream.cc b/src/core/util/byte_stream.cc index 2e5bf665..04f55b1f 100644 --- a/src/core/util/byte_stream.cc +++ b/src/core/util/byte_stream.cc @@ -30,8 +30,8 @@ #include "core/util/byte_stream.h" -#include #include +#include #include #include #include @@ -46,11 +46,10 @@ namespace kovri { namespace core { - // Input ByteStream::ByteStream(const std::uint8_t* data, std::size_t len) - : m_DataPtr(const_cast(data)), m_Size(len), m_Length(len), m_Counter{} - // TODO(anonimal): remove const cast! + : m_DataPtr(const_cast(data)), m_Length(len), m_Counter{} +// TODO(anonimal): remove const cast! { assert(data || len); @@ -62,7 +61,7 @@ ByteStream::ByteStream(const std::uint8_t* data, std::size_t len) // Output ByteStream::ByteStream(std::uint8_t* data, std::size_t len) - : m_DataPtr(data), m_Size(len), m_Length(len), m_Counter{} + : m_DataPtr(data), m_Length(len), m_Counter{} { assert(data || len); @@ -72,8 +71,7 @@ ByteStream::ByteStream(std::uint8_t* data, std::size_t len) throw std::length_error("ByteStream: null length"); } -ByteStream::ByteStream(std::size_t len) - : m_Size(len), m_Length(len), m_Counter{} +ByteStream::ByteStream(std::size_t len) : m_Length(len), m_Counter{} { assert(len); diff --git a/src/core/util/byte_stream.h b/src/core/util/byte_stream.h index 7c380dac..d350afd0 100644 --- a/src/core/util/byte_stream.h +++ b/src/core/util/byte_stream.h @@ -80,7 +80,7 @@ class ByteStream /// @return Total size std::size_t Size() const noexcept { - return m_Size; + return m_Length + m_Counter; } /// @brief Get the current position in the stream