xpcom/io/StreamBufferSourceImpl.h

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #ifndef mozilla_StreamBufferSourceImpl_h
   8 #define mozilla_StreamBufferSourceImpl_h
   9
  10 #include "mozilla/StreamBufferSource.h"
  11
  12 #include "nsTArray.h"
  13
  14 namespace mozilla {
  15
  16 class nsTArraySource final : public StreamBufferSource {
  17  public:
  18   explicit nsTArraySource(nsTArray<uint8_t>&& aArray)
  19       : mArray(std::move(aArray)) {}
  20
  21   Span<const char> Data() override {
  22     return Span{reinterpret_cast<const char*>(mArray.Elements()),
  23                 mArray.Length()};
  24   }
  25
  26   bool Owning() override { return true; }
  27
  28   size_t SizeOfExcludingThisEvenIfShared(MallocSizeOf aMallocSizeOf) override {
  29     return mArray.ShallowSizeOfExcludingThis(aMallocSizeOf);
  30   }
  31
  32  private:
  33   const nsTArray<uint8_t> mArray;
  34 };
  35
  36 class nsCStringSource final : public StreamBufferSource {
  37  public:
  38   explicit nsCStringSource(nsACString&& aString)
  39       : mString(std::move(aString)) {}
  40
  41   Span<const char> Data() override { return mString; }
  42
  43   nsresult GetData(nsACString& aString) override {
  44     if (!aString.Assign(mString, fallible)) {
  45       return NS_ERROR_OUT_OF_MEMORY;
  46     }
  47     return NS_OK;
  48   }
  49
  50   bool Owning() override { return true; }
  51
  52   size_t SizeOfExcludingThisIfUnshared(MallocSizeOf aMallocSizeOf) override {
  53     return mString.SizeOfExcludingThisIfUnshared(aMallocSizeOf);
  54   }
  55
  56   size_t SizeOfExcludingThisEvenIfShared(MallocSizeOf aMallocSizeOf) override {
  57     return mString.SizeOfExcludingThisEvenIfShared(aMallocSizeOf);
  58   }
  59
  60  private:
  61   const nsCString mString;
  62 };
  63
  64 class nsBorrowedSource final : public StreamBufferSource {
  65  public:
  66   explicit nsBorrowedSource(Span<const char> aBuffer) : mBuffer(aBuffer) {}
  67
  68   Span<const char> Data() override { return mBuffer; }
  69
  70   bool Owning() override { return false; }
  71
  72   size_t SizeOfExcludingThisEvenIfShared(MallocSizeOf aMallocSizeOf) override {
  73     return 0;
  74   }
  75
  76  private:
  77   const Span<const char> mBuffer;
  78 };
  79
  80 }  // namespace mozilla
  81
  82 #endif  // mozilla_StreamBufferSourceImpl_h