Quelle  ScriptResponseHeaderProcessor.h   Sprache: C

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

#include "mozilla/dom/WorkerCommon.h"

#include "nsIHttpChannel.h"
#include "nsIHttpChannelInternal.h"
#include "nsIStreamLoader.h"
#include "nsStreamUtils.h"
#include "js/Modules.h"
#include "mozilla/StaticPrefs_browser.h"
#include "mozilla/StaticPrefs_dom.h"

namespace mozilla::dom {

class ThreadSafeWorkerRef;

namespace workerinternals::loader {

/* ScriptResponseHeaderProcessor
 *
 * This class handles Policy headers. It can be used as a RequestObserver in a
 * Tee, as it is for NetworkLoadHandler in WorkerScriptLoader, or the static
 * method can be called directly, as it is in CacheLoadHandler.
 *
 */

class ScriptResponseHeaderProcessor final : public nsIRequestObserver {
 public:
  NS_DECL_ISUPPORTS

  ScriptResponseHeaderProcessor(RefPtr<ThreadSafeWorkerRef>& aWorkerRef,
                                bool aIsMainScript,
                                bool aRequiresStrictMimeCheck,
                                JS::ModuleType aModuleType)
      : mWorkerRef(aWorkerRef),
        mIsMainScript(aIsMainScript),
        mRequiresStrictMimeCheck(aRequiresStrictMimeCheck),
        mModuleType(aModuleType) {
    AssertIsOnMainThread();
  }

  NS_IMETHOD OnStartRequest(nsIRequest* aRequest) override {
    nsresult rv = NS_OK;
    if (mRequiresStrictMimeCheck &&
        StaticPrefs::dom_workers_importScripts_enforceStrictMimeType()) {
      rv = EnsureExpectedModuleType(aRequest);
      if (NS_WARN_IF(NS_FAILED(rv))) {
        aRequest->Cancel(rv);
        return NS_OK;
      }
    }

    if (!StaticPrefs::browser_tabs_remote_useCrossOriginEmbedderPolicy()) {
      return NS_OK;
    }

    rv = ProcessCrossOriginEmbedderPolicyHeader(aRequest);

    if (NS_WARN_IF(NS_FAILED(rv))) {
      aRequest->Cancel(rv);
    }

    return rv;
  }

  NS_IMETHOD OnStopRequest(nsIRequest* aRequest,
                           nsresult aStatusCode) override {
    return NS_OK;
  }

  static nsresult ProcessCrossOriginEmbedderPolicyHeader(
      WorkerPrivate* aWorkerPrivate,
      nsILoadInfo::CrossOriginEmbedderPolicy aPolicy, bool aIsMainScript);

 private:
  ~ScriptResponseHeaderProcessor() = default;

  nsresult EnsureExpectedModuleType(nsIRequest* aRequest);

  nsresult ProcessCrossOriginEmbedderPolicyHeader(nsIRequest* aRequest);

  // The owner of ScriptResponseHeaderProcessor should give the WorkerRef to
  // ensure ScriptResponseHeaderProcessor works with an valid WorkerPrivate.
  RefPtr<ThreadSafeWorkerRef> mWorkerRef;
  const bool mIsMainScript;
  const bool mRequiresStrictMimeCheck;
  const JS::ModuleType mModuleType;
};

}  // namespace workerinternals::loader

}  // namespace mozilla::dom

#endif /* mozilla_dom_workers_ScriptResponseHeaderProcessor_h__ */