dom/base/nsContentPolicy.h
author Ehsan Akhgari <ehsan@mozilla.com>
Sat, 15 Apr 2017 16:23:37 -0400
changeset 403849 938824ac57496e64e634e4a658b5b531de9e69ac
parent 281327 1eefebfb86eb72ac66b6a5707bcdff3fd1043974
child 407876 7c9e467c10b6c47659b86b03409c9a6f7f159bb4
permissions -rw-r--r--
Bug 1356835 - Reduce the hashtable lookup cost in nsContentPolicy::CheckPolicy(); r=qdot

/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
// vim: ft=cpp ts=8 sw=4 et tw=78
/* 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 __nsContentPolicy_h__
#define __nsContentPolicy_h__

#include "nsIContentPolicy.h"
#include "nsISimpleContentPolicy.h"
#include "nsCategoryCache.h"

/* 
 * Implementation of the "@mozilla.org/layout/content-policy;1" contract.
 */

class nsContentPolicy : public nsIContentPolicy
{
 public:
    NS_DECL_ISUPPORTS
    NS_DECL_NSICONTENTPOLICY

    nsContentPolicy();

 protected:
    virtual ~nsContentPolicy();

 private:
    //Array of policies
    nsCategoryCache<nsIContentPolicy> mPolicies;
    nsCategoryCache<nsISimpleContentPolicy> mSimplePolicies;

    nsCOMPtr<nsIContentPolicy> mMixedContentBlocker;
    nsCOMPtr<nsIContentPolicy> mCSPService;

    //Helper type for CheckPolicy
    typedef
    NS_STDCALL_FUNCPROTO(nsresult, CPMethod, nsIContentPolicy,
                         ShouldProcess,
                         (uint32_t, nsIURI*, nsIURI*, nsISupports*,
                           const nsACString &, nsISupports*, nsIPrincipal*,
                           int16_t*));

    typedef
    NS_STDCALL_FUNCPROTO(nsresult, SCPMethod, nsISimpleContentPolicy,
                         ShouldProcess,
                         (uint32_t, nsIURI*, nsIURI*, nsIDOMElement*, bool,
                           const nsACString &, nsISupports*, nsIPrincipal*,
                           int16_t*));

    //Helper method that applies policyMethod across all policies in mPolicies
    // with the given parameters
    nsresult CheckPolicy(CPMethod policyMethod, SCPMethod simplePolicyMethod,
                         nsContentPolicyType contentType,
                         nsIURI *aURI, nsIURI *origURI,
                         nsISupports *requestingContext,
                         const nsACString &mimeGuess, nsISupports *extra,
                         nsIPrincipal *requestPrincipal,
                         int16_t *decision);
};

nsresult
NS_NewContentPolicy(nsIContentPolicy **aResult);

#endif /* __nsContentPolicy_h__ */