f4bd388765
This patch does the following: * It adds nsITransferable::Init(nsILoadContext*). The load context might be null, which means that the transferable is non-private, but if it's non-null, we extract the boolean value for the privacy mode and store it in the transferable. * It adds checks in debug builds to make sure that Init is always called, in form of fatal assertions. * It adds nsIDOMDocument* agruments to nsIClipboardHelper methods which represent the document that the string is coming from. nsIClipboardHelper implementation internally gets the nsILoadContext from that and passes it on to the transferable upon creation. The reason that I did this was that nsIClipboardHelper is supposed to be a high-level helper, and in most of its call sites, we have easy access to a document object. * It modifies all of the call sites of the above interfaces according to this change. * It adds a GetLoadContext helper to nsIDocument to help with changing the call sites.
136 lines
4.1 KiB
C++
136 lines
4.1 KiB
C++
/* -*- 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/. */
|
|
|
|
#include "mozilla/Util.h"
|
|
|
|
#include "nsClipboardPrivacyHandler.h"
|
|
#include "nsITransferable.h"
|
|
#include "nsISupportsPrimitives.h"
|
|
#include "nsIObserverService.h"
|
|
#include "nsIClipboard.h"
|
|
#include "nsComponentManagerUtils.h"
|
|
#include "nsServiceManagerUtils.h"
|
|
#include "nsLiteralString.h"
|
|
#include "nsNetCID.h"
|
|
#include "nsXPCOM.h"
|
|
#include "mozilla/Services.h"
|
|
|
|
#if defined(XP_WIN)
|
|
#include <ole2.h>
|
|
#endif
|
|
|
|
using namespace mozilla;
|
|
|
|
#define NS_MOZ_DATA_FROM_PRIVATEBROWSING "application/x-moz-private-browsing"
|
|
|
|
NS_IMPL_ISUPPORTS2(nsClipboardPrivacyHandler, nsIObserver, nsISupportsWeakReference)
|
|
|
|
nsresult
|
|
nsClipboardPrivacyHandler::Init()
|
|
{
|
|
nsCOMPtr<nsIObserverService> observerService =
|
|
mozilla::services::GetObserverService();
|
|
if (!observerService)
|
|
return NS_ERROR_FAILURE;
|
|
return observerService->AddObserver(this, NS_PRIVATE_BROWSING_SWITCH_TOPIC,
|
|
true);
|
|
}
|
|
|
|
/**
|
|
* Prepare the transferable object to be inserted into the clipboard
|
|
*
|
|
*/
|
|
nsresult
|
|
nsClipboardPrivacyHandler::PrepareDataForClipboard(nsITransferable * aTransferable)
|
|
{
|
|
NS_ASSERTION(aTransferable, "clipboard given a null transferable");
|
|
|
|
nsresult rv = NS_OK;
|
|
if (InPrivateBrowsing()) {
|
|
nsCOMPtr<nsISupportsPRBool> data = do_CreateInstance(NS_SUPPORTS_PRBOOL_CONTRACTID);
|
|
if (data) {
|
|
rv = data->SetData(true);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = aTransferable->AddDataFlavor(NS_MOZ_DATA_FROM_PRIVATEBROWSING);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
rv = aTransferable->SetTransferData(NS_MOZ_DATA_FROM_PRIVATEBROWSING, data, sizeof(bool));
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
}
|
|
}
|
|
|
|
return rv;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
nsClipboardPrivacyHandler::Observe(nsISupports *aSubject, char const *aTopic, PRUnichar const *aData)
|
|
{
|
|
if (NS_LITERAL_STRING(NS_PRIVATE_BROWSING_LEAVE).Equals(aData)) {
|
|
nsresult rv;
|
|
nsCOMPtr<nsIClipboard> clipboard =
|
|
do_GetService("@mozilla.org/widget/clipboard;1", &rv);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
const char * flavors[] = { NS_MOZ_DATA_FROM_PRIVATEBROWSING };
|
|
bool haveFlavors;
|
|
rv = clipboard->HasDataMatchingFlavors(flavors,
|
|
ArrayLength(flavors),
|
|
nsIClipboard::kGlobalClipboard,
|
|
&haveFlavors);
|
|
if (NS_SUCCEEDED(rv) && haveFlavors) {
|
|
#if defined(XP_WIN)
|
|
// Workaround for bug 518412. On Windows 7 x64, there is a bug
|
|
// in handling clipboard data without any formats between
|
|
// 32-bit/64-bit boundaries, which could lead Explorer to crash.
|
|
// We work around the problem by clearing the clipboard using
|
|
// the usual Win32 API.
|
|
NS_ENSURE_TRUE(SUCCEEDED(::OleSetClipboard(NULL)), NS_ERROR_FAILURE);
|
|
#else
|
|
// Empty the native clipboard by copying an empty transferable
|
|
nsCOMPtr<nsITransferable> nullData =
|
|
do_CreateInstance("@mozilla.org/widget/transferable;1", &rv);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
nullData->Init(nsnull);
|
|
rv = clipboard->SetData(nullData, nsnull,
|
|
nsIClipboard::kGlobalClipboard);
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
bool
|
|
nsClipboardPrivacyHandler::InPrivateBrowsing()
|
|
{
|
|
bool inPrivateBrowsingMode = false;
|
|
if (!mPBService)
|
|
mPBService = do_GetService(NS_PRIVATE_BROWSING_SERVICE_CONTRACTID);
|
|
if (mPBService)
|
|
mPBService->GetPrivateBrowsingEnabled(&inPrivateBrowsingMode);
|
|
return inPrivateBrowsingMode;
|
|
}
|
|
|
|
nsresult
|
|
NS_NewClipboardPrivacyHandler(nsClipboardPrivacyHandler ** aHandler)
|
|
{
|
|
NS_PRECONDITION(aHandler != nsnull, "null ptr");
|
|
if (!aHandler)
|
|
return NS_ERROR_NULL_POINTER;
|
|
|
|
*aHandler = new nsClipboardPrivacyHandler();
|
|
if (!*aHandler)
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
|
|
NS_ADDREF(*aHandler);
|
|
nsresult rv = (*aHandler)->Init();
|
|
if (NS_FAILED(rv))
|
|
NS_RELEASE(*aHandler);
|
|
|
|
return rv;
|
|
}
|