The Tor Browser: netwerk/base/src/PrivateBrowsingChannel.h@b8a032363ba2

Incorporate requested changes from Mozilla in review:
https://bugzilla.mozilla.org/show_bug.cgi?id=1123480#c6

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

     2 /* vim:set ts=4 sts=4 sw=4 et cin: */

     3 /* This Source Code Form is subject to the terms of the Mozilla Public

     4  * License, v. 2.0. If a copy of the MPL was not distributed with this

     5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

     7 #ifndef mozilla_net_PrivateBrowsingChannel_h__

     8 #define mozilla_net_PrivateBrowsingChannel_h__

    10 #include "nsIPrivateBrowsingChannel.h"

    11 #include "nsCOMPtr.h"

    12 #include "nsILoadGroup.h"

    13 #include "nsILoadContext.h"

    14 #include "nsIInterfaceRequestorUtils.h"

    15 #include "nsIInterfaceRequestor.h"

    17 namespace mozilla {

    18 namespace net {

    20 template <class Channel>

    21 class PrivateBrowsingChannel : public nsIPrivateBrowsingChannel

    22 {

    23 public:

    24   PrivateBrowsingChannel() :

    25     mPrivateBrowsingOverriden(false),

    26     mPrivateBrowsing(false)

    27   {

    28   }

    30   NS_IMETHOD SetPrivate(bool aPrivate)

    31   {

    32       // Make sure that we don't have a load context

    33       // This is a fatal error in debug builds, and a runtime error in release

    34       // builds.

    35       nsCOMPtr<nsILoadContext> loadContext;

    36       NS_QueryNotificationCallbacks(static_cast<Channel*>(this), loadContext);

    37       MOZ_ASSERT(!loadContext);

    38       if (loadContext) {

    39           return NS_ERROR_FAILURE;

    40       }

    42       mPrivateBrowsingOverriden = true;

    43       mPrivateBrowsing = aPrivate;

    44       return NS_OK;

    45   }

    47   NS_IMETHOD GetIsChannelPrivate(bool *aResult)

    48   {

    49       NS_ENSURE_ARG_POINTER(aResult);

    50       *aResult = NS_UsePrivateBrowsing(static_cast<Channel*>(this));

    51       return NS_OK;

    52   }

    54   NS_IMETHOD IsPrivateModeOverriden(bool* aValue, bool *aResult)

    55   {

    56       NS_ENSURE_ARG_POINTER(aValue);

    57       NS_ENSURE_ARG_POINTER(aResult);

    58       *aResult = mPrivateBrowsingOverriden;

    59       if (mPrivateBrowsingOverriden) {

    60           *aValue = mPrivateBrowsing;

    61       }

    62       return NS_OK;

    63   }

    65   bool CanSetCallbacks(nsIInterfaceRequestor* aCallbacks) const

    66   {

    67       // Make sure that the private bit override flag is not set.

    68       // This is a fatal error in debug builds, and a runtime error in release

    69       // builds.

    70       if (!aCallbacks) {

    71           return true;

    72       }

    73       nsCOMPtr<nsILoadContext> loadContext = do_GetInterface(aCallbacks);

    74       if (!loadContext) {

    75           return true;

    76       }

    77       MOZ_ASSERT(!mPrivateBrowsingOverriden);

    78       return !mPrivateBrowsingOverriden;

    79   }

    81   bool CanSetLoadGroup(nsILoadGroup* aLoadGroup) const

    82   {

    83       // Make sure that the private bit override flag is not set.

    84       // This is a fatal error in debug builds, and a runtime error in release

    85       // builds.

    86       if (!aLoadGroup) {

    87           return true;

    88       }

    89       nsCOMPtr<nsIInterfaceRequestor> callbacks;

    90       aLoadGroup->GetNotificationCallbacks(getter_AddRefs(callbacks));

    91       // From this point on, we just hand off the work to CanSetCallbacks,

    92       // because the logic is exactly the same.

    93       return CanSetCallbacks(callbacks);

    94   }

    96 protected:

    97   bool mPrivateBrowsingOverriden;

    98   bool mPrivateBrowsing;

    99 };

   101 }

   102 }

   104 #endif

The Tor Browser / file revision