netwerk/base/nsAsyncStreamCopier.h
author Sylvestre Ledru <sledru@mozilla.com>
Fri, 30 Nov 2018 11:46:48 +0100
changeset 448947 6f3709b3878117466168c40affa7bca0b60cf75b
parent 236283 8bc7561d75577b08ae0c5449104378144a6fc9e0
child 472056 e1993a1f09ac53cd1a04fdf6a87f8cad8e44f73e
permissions -rw-r--r--
Bug 1511181 - Reformat everything to the Google coding style r=ehsan a=clang-format # ignore-this-changeset

/* 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 nsAsyncStreamCopier_h__
#define nsAsyncStreamCopier_h__

#include "nsIAsyncStreamCopier.h"
#include "nsIAsyncStreamCopier2.h"
#include "mozilla/Mutex.h"
#include "nsStreamUtils.h"
#include "nsCOMPtr.h"

class nsIRequestObserver;

//-----------------------------------------------------------------------------

class nsAsyncStreamCopier final : public nsIAsyncStreamCopier,
                                  nsIAsyncStreamCopier2 {
 public:
  NS_DECL_THREADSAFE_ISUPPORTS
  NS_DECL_NSIREQUEST
  NS_DECL_NSIASYNCSTREAMCOPIER

  // nsIAsyncStreamCopier2
  // We declare it by hand instead of NS_DECL_NSIASYNCSTREAMCOPIER2
  // as nsIAsyncStreamCopier2 duplicates methods of nsIAsyncStreamCopier
  NS_IMETHOD Init(nsIInputStream *aSource, nsIOutputStream *aSink,
                  nsIEventTarget *aTarget, uint32_t aChunkSize,
                  bool aCloseSource, bool aCloseSink) override;

  nsAsyncStreamCopier();

  //-------------------------------------------------------------------------
  // these methods may be called on any thread

  bool IsComplete(nsresult *status = nullptr);
  void Complete(nsresult status);

 private:
  virtual ~nsAsyncStreamCopier();

  nsresult InitInternal(nsIInputStream *source, nsIOutputStream *sink,
                        nsIEventTarget *target, uint32_t chunkSize,
                        bool closeSource, bool closeSink);

  static void OnAsyncCopyComplete(void *, nsresult);

  void AsyncCopyInternal();
  nsresult ApplyBufferingPolicy();
  nsIRequest *AsRequest();

  nsCOMPtr<nsIInputStream> mSource;
  nsCOMPtr<nsIOutputStream> mSink;

  nsCOMPtr<nsIRequestObserver> mObserver;

  nsCOMPtr<nsIEventTarget> mTarget;

  nsCOMPtr<nsISupports> mCopierCtx;

  mozilla::Mutex mLock;

  nsAsyncCopyMode mMode;
  uint32_t mChunkSize;
  nsresult mStatus;
  bool mIsPending;
  bool mCloseSource;
  bool mCloseSink;
  bool mShouldSniffBuffering;

  friend class ProceedWithAsyncCopy;
  friend class AsyncApplyBufferingPolicyEvent;
};

#endif  // !nsAsyncStreamCopier_h__