nsMsgContentPolicy.h

/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */

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

/******************************************************************************

 * nsMsgContentPolicy enforces the specified content policy on images, js,

 * plugins, etc. This is the class used to determine what elements in a message

 * should be loaded.

 * nsMsgCookiePolicy enforces our cookie policy for mail and RSS messages.

 ******************************************************************************/

#ifndef _nsMsgContentPolicy_H_

#define _nsMsgContentPolicy_H_

#include "nsIContentPolicy.h"

#include "nsIObserver.h"

#include "nsWeakReference.h"

#include "nsString.h"

#include "nsIMsgMailNewsUrl.h"

#include "nsIMsgCompose.h"

#include "nsIDocShell.h"

#include "nsIPermissionManager.h"

#include "nsIMsgContentPolicy.h"

#include "nsTArray.h"

/* DBFCFDF0-4489-4faa-8122-190FD1EFA16C */

#define NS_MSGCONTENTPOLICY_CID                     \

  {                                                 \

    0xdbfcfdf0, 0x4489, 0x4faa, {                   \

      0x81, 0x22, 0x19, 0xf, 0xd1, 0xef, 0xa1, 0x6c \

    }                                               \

#define NS_MSGCONTENTPOLICY_CONTRACTID "@mozilla.org/messenger/content-policy;1"

class nsIMsgDBHdr;

class nsIDocShell;

class nsMsgContentPolicy : public nsIContentPolicy,

                           public nsIObserver,

                           public nsIMsgContentPolicy,

                           public nsSupportsWeakReference {

 public:

  nsMsgContentPolicy();

  nsresult Init();

  NS_DECL_ISUPPORTS

  NS_DECL_NSICONTENTPOLICY

  NS_DECL_NSIOBSERVER

  NS_DECL_NSIMSGCONTENTPOLICY

 protected:

  virtual ~nsMsgContentPolicy();

  bool mBlockRemoteImages;

  nsCString mTrustedMailDomains;

  nsCOMPtr<nsIPermissionManager> mPermissionManager;

  bool IsTrustedDomain(nsIURI* aContentLocation);

  bool IsSafeRequestingLocation(nsIURI* aRequestingLocation);

  bool IsExposedProtocol(nsIURI* aContentLocation);

  bool IsExposedChromeProtocol(nsIURI* aContentLocation);

  bool ShouldBlockUnexposedProtocol(nsIURI* aContentLocation);

  bool ShouldAcceptRemoteContentForSender(nsIMsgDBHdr* aMsgHdr);

  int16_t ShouldAcceptRemoteContentForMsgHdr(nsIMsgDBHdr* aMsgHdr,

                                             nsIURI* aRequestingLocation,

                                             nsIURI* aContentLocation);

  void NotifyContentWasBlocked(uint64_t aBrowsingContextId,

                               nsIURI* aContentLocation);

  void ShouldAcceptContentForPotentialMsg(uint64_t aBrowsingContextId,

                                          nsIURI* aOriginatorLocation,

                                          nsIURI* aContentLocation,

                                          int16_t* aDecision);

  void ComposeShouldLoad(nsIMsgCompose* aMsgCompose,

                         nsISupports* aRequestingContext,

                         nsIURI* aOriginatorLocation, nsIURI* aContentLocation,

                         int16_t* aDecision);

  already_AddRefed<nsIMsgCompose> GetMsgComposeForBrowsingContext(

      mozilla::dom::BrowsingContext* aRequestingContext);

  nsresult GetRootDocShellForContext(nsISupports* aRequestingContext,

                                     nsIDocShell** aDocShell);

  nsresult GetOriginatingURIForContext(nsISupports* aRequestingContext,

                                       nsIURI** aURI);

  nsresult SetDisableItemsOnMailNewsUrlDocshells(nsIURI* aContentLocation,

                                                 nsILoadInfo* aLoadInfo);

  nsTArray<nsCString> mCustomExposedProtocols;

};

#endif  // _nsMsgContentPolicy_H_