diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-19 00:47:55 +0000 |
commit | 26a029d407be480d791972afb5975cf62c9360a6 (patch) | |
tree | f435a8308119effd964b339f76abb83a57c29483 /xpcom/io/StreamBufferSinkImpl.h | |
parent | Initial commit. (diff) | |
download | firefox-upstream/124.0.1.tar.xz firefox-upstream/124.0.1.zip |
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'xpcom/io/StreamBufferSinkImpl.h')
-rw-r--r-- | xpcom/io/StreamBufferSinkImpl.h | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/xpcom/io/StreamBufferSinkImpl.h b/xpcom/io/StreamBufferSinkImpl.h new file mode 100644 index 0000000000..9fba413ef2 --- /dev/null +++ b/xpcom/io/StreamBufferSinkImpl.h @@ -0,0 +1,49 @@ +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set ts=8 sts=2 et sw=2 tw=80: */ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this file, + * You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#ifndef mozilla_StreamBufferSinkImpl_h +#define mozilla_StreamBufferSinkImpl_h + +#include "mozilla/Buffer.h" +#include "mozilla/StreamBufferSink.h" +#include "mozilla/UniquePtr.h" + +namespace mozilla { + +class BufferSink final : public StreamBufferSink { + public: + explicit BufferSink(Buffer<char>&& aBuffer) : mBuffer(std::move(aBuffer)) {} + + explicit BufferSink(size_t aLength) : mBuffer(aLength) {} + + static UniquePtr<BufferSink> Alloc(size_t aLength) { + auto maybeBuffer = Buffer<char>::Alloc(aLength); + if (!maybeBuffer) { + return nullptr; + } + + return MakeUnique<BufferSink>(maybeBuffer.extract()); + } + + mozilla::Span<char> Data() override { return mBuffer.AsWritableSpan(); } + + private: + Buffer<char> mBuffer; +}; + +class nsBorrowedSink final : public StreamBufferSink { + public: + explicit nsBorrowedSink(mozilla::Span<char> aBuffer) : mBuffer(aBuffer) {} + + mozilla::Span<char> Data() override { return mBuffer; } + + private: + mozilla::Span<char> mBuffer; +}; + +} // namespace mozilla + +#endif // mozilla_StreamBufferSinkImpl_h |