mirror of
https://github.com/RGBCube/serenity
synced 2025-07-22 19:27:35 +00:00
LibCore: Move the MemoryStream implementation into a separate file
This commit is contained in:
parent
c6d71ca727
commit
8dbc7aa63f
3 changed files with 145 additions and 87 deletions
|
@ -14,6 +14,7 @@ set(SOURCES
|
||||||
IODevice.cpp
|
IODevice.cpp
|
||||||
LockFile.cpp
|
LockFile.cpp
|
||||||
MappedFile.cpp
|
MappedFile.cpp
|
||||||
|
MemoryStream.cpp
|
||||||
MimeData.cpp
|
MimeData.cpp
|
||||||
NetworkJob.cpp
|
NetworkJob.cpp
|
||||||
Notifier.cpp
|
Notifier.cpp
|
||||||
|
|
127
Userland/Libraries/LibCore/MemoryStream.cpp
Normal file
127
Userland/Libraries/LibCore/MemoryStream.cpp
Normal file
|
@ -0,0 +1,127 @@
|
||||||
|
/*
|
||||||
|
* Copyright (c) 2021, kleines Filmröllchen <filmroellchen@serenityos.org>.
|
||||||
|
*
|
||||||
|
* SPDX-License-Identifier: BSD-2-Clause
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <LibCore/MemoryStream.h>
|
||||||
|
|
||||||
|
namespace Core::Stream {
|
||||||
|
|
||||||
|
FixedMemoryStream::FixedMemoryStream(Bytes bytes)
|
||||||
|
: m_bytes(bytes)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
FixedMemoryStream::FixedMemoryStream(ReadonlyBytes bytes)
|
||||||
|
: m_bytes({ const_cast<u8*>(bytes.data()), bytes.size() })
|
||||||
|
, m_writing_enabled(false)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<NonnullOwnPtr<FixedMemoryStream>> FixedMemoryStream::construct(Bytes bytes)
|
||||||
|
{
|
||||||
|
return adopt_nonnull_own_or_enomem<FixedMemoryStream>(new (nothrow) FixedMemoryStream(bytes));
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<NonnullOwnPtr<FixedMemoryStream>> FixedMemoryStream::construct(ReadonlyBytes bytes)
|
||||||
|
{
|
||||||
|
return adopt_nonnull_own_or_enomem<FixedMemoryStream>(new (nothrow) FixedMemoryStream(bytes));
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FixedMemoryStream::is_eof() const
|
||||||
|
{
|
||||||
|
return m_offset >= m_bytes.size();
|
||||||
|
}
|
||||||
|
|
||||||
|
bool FixedMemoryStream::is_open() const
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void FixedMemoryStream::close()
|
||||||
|
{
|
||||||
|
// FIXME: It doesn't make sense to close a memory stream. Therefore, we don't do anything here. Is that fine?
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<void> FixedMemoryStream::truncate(off_t)
|
||||||
|
{
|
||||||
|
return Error::from_errno(ENOTSUP);
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<Bytes> FixedMemoryStream::read(Bytes bytes)
|
||||||
|
{
|
||||||
|
auto to_read = min(remaining(), bytes.size());
|
||||||
|
if (to_read == 0)
|
||||||
|
return Bytes {};
|
||||||
|
|
||||||
|
m_bytes.slice(m_offset, to_read).copy_to(bytes);
|
||||||
|
m_offset += to_read;
|
||||||
|
return bytes.trim(to_read);
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<off_t> FixedMemoryStream::seek(i64 offset, SeekMode seek_mode)
|
||||||
|
{
|
||||||
|
switch (seek_mode) {
|
||||||
|
case SeekMode::SetPosition:
|
||||||
|
if (offset > static_cast<i64>(m_bytes.size()))
|
||||||
|
return Error::from_string_literal("Offset past the end of the stream memory");
|
||||||
|
|
||||||
|
m_offset = offset;
|
||||||
|
break;
|
||||||
|
case SeekMode::FromCurrentPosition:
|
||||||
|
if (offset + static_cast<i64>(m_offset) > static_cast<i64>(m_bytes.size()))
|
||||||
|
return Error::from_string_literal("Offset past the end of the stream memory");
|
||||||
|
|
||||||
|
m_offset += offset;
|
||||||
|
break;
|
||||||
|
case SeekMode::FromEndPosition:
|
||||||
|
if (offset > static_cast<i64>(m_bytes.size()))
|
||||||
|
return Error::from_string_literal("Offset past the start of the stream memory");
|
||||||
|
|
||||||
|
m_offset = m_bytes.size() - offset;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return static_cast<off_t>(m_offset);
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<size_t> FixedMemoryStream::write(ReadonlyBytes bytes)
|
||||||
|
{
|
||||||
|
VERIFY(m_writing_enabled);
|
||||||
|
|
||||||
|
// FIXME: Can this not error?
|
||||||
|
auto const nwritten = bytes.copy_trimmed_to(m_bytes.slice(m_offset));
|
||||||
|
m_offset += nwritten;
|
||||||
|
return nwritten;
|
||||||
|
}
|
||||||
|
|
||||||
|
ErrorOr<void> FixedMemoryStream::write_entire_buffer(ReadonlyBytes bytes)
|
||||||
|
{
|
||||||
|
if (remaining() < bytes.size())
|
||||||
|
return Error::from_string_literal("Write of entire buffer ends past the memory area");
|
||||||
|
|
||||||
|
TRY(write(bytes));
|
||||||
|
return {};
|
||||||
|
}
|
||||||
|
|
||||||
|
Bytes FixedMemoryStream::bytes()
|
||||||
|
{
|
||||||
|
VERIFY(m_writing_enabled);
|
||||||
|
return m_bytes;
|
||||||
|
}
|
||||||
|
ReadonlyBytes FixedMemoryStream::bytes() const
|
||||||
|
{
|
||||||
|
return m_bytes;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t FixedMemoryStream::offset() const
|
||||||
|
{
|
||||||
|
return m_offset;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t FixedMemoryStream::remaining() const
|
||||||
|
{
|
||||||
|
return m_bytes.size() - m_offset;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -19,99 +19,29 @@ namespace Core::Stream {
|
||||||
/// using a single read/write head.
|
/// using a single read/write head.
|
||||||
class FixedMemoryStream final : public SeekableStream {
|
class FixedMemoryStream final : public SeekableStream {
|
||||||
public:
|
public:
|
||||||
static ErrorOr<NonnullOwnPtr<FixedMemoryStream>> construct(Bytes bytes)
|
static ErrorOr<NonnullOwnPtr<FixedMemoryStream>> construct(Bytes bytes);
|
||||||
{
|
static ErrorOr<NonnullOwnPtr<FixedMemoryStream>> construct(ReadonlyBytes bytes);
|
||||||
return adopt_nonnull_own_or_enomem<FixedMemoryStream>(new (nothrow) FixedMemoryStream(bytes));
|
|
||||||
}
|
|
||||||
|
|
||||||
static ErrorOr<NonnullOwnPtr<FixedMemoryStream>> construct(ReadonlyBytes bytes)
|
virtual bool is_eof() const override;
|
||||||
{
|
virtual bool is_open() const override;
|
||||||
return adopt_nonnull_own_or_enomem<FixedMemoryStream>(new (nothrow) FixedMemoryStream(bytes));
|
virtual void close() override;
|
||||||
}
|
virtual ErrorOr<void> truncate(off_t) override;
|
||||||
|
virtual ErrorOr<Bytes> read(Bytes bytes) override;
|
||||||
|
|
||||||
virtual bool is_eof() const override { return m_offset >= m_bytes.size(); }
|
virtual ErrorOr<off_t> seek(i64 offset, SeekMode seek_mode = SeekMode::SetPosition) override;
|
||||||
virtual bool is_open() const override { return true; }
|
|
||||||
// FIXME: It doesn't make sense to close an memory stream. Therefore, we don't do anything here. Is that fine?
|
|
||||||
virtual void close() override { }
|
|
||||||
// FIXME: It doesn't make sense to truncate a memory stream. Therefore, we don't do anything here. Is that fine?
|
|
||||||
virtual ErrorOr<void> truncate(off_t) override { return Error::from_errno(ENOTSUP); }
|
|
||||||
|
|
||||||
virtual ErrorOr<Bytes> read(Bytes bytes) override
|
virtual ErrorOr<size_t> write(ReadonlyBytes bytes) override;
|
||||||
{
|
virtual ErrorOr<void> write_entire_buffer(ReadonlyBytes bytes) override;
|
||||||
auto to_read = min(remaining(), bytes.size());
|
|
||||||
if (to_read == 0)
|
|
||||||
return Bytes {};
|
|
||||||
|
|
||||||
m_bytes.slice(m_offset, to_read).copy_to(bytes);
|
Bytes bytes();
|
||||||
m_offset += to_read;
|
ReadonlyBytes bytes() const;
|
||||||
return bytes.trim(to_read);
|
size_t offset() const;
|
||||||
}
|
size_t remaining() const;
|
||||||
|
|
||||||
virtual ErrorOr<off_t> seek(i64 offset, SeekMode seek_mode = SeekMode::SetPosition) override
|
|
||||||
{
|
|
||||||
switch (seek_mode) {
|
|
||||||
case SeekMode::SetPosition:
|
|
||||||
if (offset > static_cast<i64>(m_bytes.size()))
|
|
||||||
return Error::from_string_literal("Offset past the end of the stream memory");
|
|
||||||
|
|
||||||
m_offset = offset;
|
|
||||||
break;
|
|
||||||
case SeekMode::FromCurrentPosition:
|
|
||||||
if (offset + static_cast<i64>(m_offset) > static_cast<i64>(m_bytes.size()))
|
|
||||||
return Error::from_string_literal("Offset past the end of the stream memory");
|
|
||||||
|
|
||||||
m_offset += offset;
|
|
||||||
break;
|
|
||||||
case SeekMode::FromEndPosition:
|
|
||||||
if (offset > static_cast<i64>(m_bytes.size()))
|
|
||||||
return Error::from_string_literal("Offset past the start of the stream memory");
|
|
||||||
|
|
||||||
m_offset = m_bytes.size() - offset;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
return static_cast<off_t>(m_offset);
|
|
||||||
}
|
|
||||||
|
|
||||||
virtual ErrorOr<size_t> write(ReadonlyBytes bytes) override
|
|
||||||
{
|
|
||||||
VERIFY(m_writing_enabled);
|
|
||||||
|
|
||||||
// FIXME: Can this not error?
|
|
||||||
auto const nwritten = bytes.copy_trimmed_to(m_bytes.slice(m_offset));
|
|
||||||
m_offset += nwritten;
|
|
||||||
return nwritten;
|
|
||||||
}
|
|
||||||
virtual ErrorOr<void> write_entire_buffer(ReadonlyBytes bytes) override
|
|
||||||
{
|
|
||||||
if (remaining() < bytes.size())
|
|
||||||
return Error::from_string_literal("Write of entire buffer ends past the memory area");
|
|
||||||
|
|
||||||
TRY(write(bytes));
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
Bytes bytes()
|
|
||||||
{
|
|
||||||
VERIFY(m_writing_enabled);
|
|
||||||
return m_bytes;
|
|
||||||
}
|
|
||||||
ReadonlyBytes bytes() const { return m_bytes; }
|
|
||||||
size_t offset() const { return m_offset; }
|
|
||||||
size_t remaining() const { return m_bytes.size() - m_offset; }
|
|
||||||
|
|
||||||
protected:
|
|
||||||
explicit FixedMemoryStream(Bytes bytes)
|
|
||||||
: m_bytes(bytes)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
explicit FixedMemoryStream(ReadonlyBytes bytes)
|
|
||||||
: m_bytes({ const_cast<u8*>(bytes.data()), bytes.size() })
|
|
||||||
, m_writing_enabled(false)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
explicit FixedMemoryStream(Bytes bytes);
|
||||||
|
explicit FixedMemoryStream(ReadonlyBytes bytes);
|
||||||
|
|
||||||
Bytes m_bytes;
|
Bytes m_bytes;
|
||||||
size_t m_offset { 0 };
|
size_t m_offset { 0 };
|
||||||
bool m_writing_enabled { true };
|
bool m_writing_enabled { true };
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue