netwerk/base/nsAsyncStreamCopier.h
author Edwin Gao <egao@mozilla.com>
Wed, 13 Mar 2019 19:47:25 +0000
changeset 463893 15c16889e3b7f6e871a518b5f3254459ed5e0213
parent 448947 6f3709b3878117466168c40affa7bca0b60cf75b
child 472056 e1993a1f09ac53cd1a04fdf6a87f8cad8e44f73e
permissions -rw-r--r--
Bug 1531590, 1531598, 1534811, 1336075, 1531571, 1531572, 1531572, 1531574, 1534855, 1534857, 1535082 - skip tests in mochitest-dev-tools and mochitest-browser-chrome suites for windows10-aarch64 r=jmaher Bug 1531598 - disable browser_markup_copy_image_data.js Bug 1531598 - disable browser_markup_links_04.js Bug 1531598 - disable browser_inspector_menu-01-sensitivity.js Bug 1534811 - disable accessible/tests/browser/general Bug 1336075 - disable browser_largeAllocation_non_win32.js Bug 1531571 - disable browser_jsterm_context_menu_labels.js Bug 1531572 - disable browser_jsterm_helper_dollar_x.js Bug 1531573 - disable browser_jsterm_no_input_and_tab_key_pressed.js Bug 1531574 - disable browser_jsterm_syntax_highlight_output.js Bug 1534855 - disable accessible/tests/browser/ Bug 1534857 - disable browser_ext_slow_script.js Bug 1535082 - disable browser_jsterm_helper_dollar_dollar.js Differential Revision: https://phabricator.services.mozilla.com/D23369

/* 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__