StreamLoader.h - mozsearch

Enable keyboard shortcuts

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

#define mozilla_css_StreamLoader_h

#include "nsIStreamListener.h"

#include "nsIThreadRetargetableStreamListener.h"

#include "nsIChannelEventSink.h"

#include "nsIInterfaceRequestor.h"

#include "nsIURI.h"

#include "nsString.h"

#include "mozilla/css/SheetLoadData.h"

#include "mozilla/Assertions.h"

class nsIInputStream;

namespace mozilla::css {

class StreamLoader : public nsIThreadRetargetableStreamListener,

                     public nsIChannelEventSink,

                     public nsIInterfaceRequestor {

 public:

  NS_DECL_THREADSAFE_ISUPPORTS

  NS_DECL_NSIREQUESTOBSERVER

  NS_DECL_NSISTREAMLISTENER

  NS_DECL_NSICHANNELEVENTSINK

  NS_DECL_NSIINTERFACEREQUESTOR

  NS_DECL_NSITHREADRETARGETABLESTREAMLISTENER

  explicit StreamLoader(SheetLoadData&);

  void ChannelOpenFailed(nsresult rv) {

#ifdef NIGHTLY_BUILD

    mChannelOpenFailed = true;

#endif

 private:

  virtual ~StreamLoader();

/**

   * callback method used for ReadSegments

*/

  static nsresult WriteSegmentFun(nsIInputStream*, void*, const char*, uint32_t,

                                  uint32_t, uint32_t*);

  void HandleBOM();

  RefPtr<SheetLoadData> mSheetLoadData;

  nsresult mStatus;

  Maybe<const Encoding*> mEncodingFromBOM;

  // We store the initial three bytes of the stream into mBOMBytes, and then

  // use that buffer to detect a BOM. We then shift any non-BOM bytes into

  // mBytes, and store all subsequent data in that buffer.

  nsCString mBytes;

  nsAutoCStringN<3> mBOMBytes;

  nsCOMPtr<nsIRequest> mRequest;

  nsCOMPtr<nsIURI> mFinalChannelURI;

  nsCOMPtr<nsIPrincipal> mChannelResultPrincipal;

  // flag to indicate that we can skip processing of data in OnStopRequest

  bool mOnStopProcessingDone{false};

  RefPtr<SheetLoadDataHolder> mMainThreadSheetLoadData;

  mozilla::TimeStamp mOnDataFinishedTime;

#ifdef NIGHTLY_BUILD

  bool mChannelOpenFailed = false;

#endif

};

}  // namespace mozilla::css

#endif  // mozilla_css_StreamLoader_h