gecko-dev/dom/webbrowserpersist/WebBrowserPersistDocumentPa...

115 строки
3.9 KiB
C++

/* -*- Mode: C++; tab-width: 4; 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 "WebBrowserPersistDocumentParent.h"
#include "mozilla/ipc/IPCStreamUtils.h"
#include "mozilla/dom/PContentParent.h"
#include "nsIInputStream.h"
#include "nsThreadUtils.h"
#include "WebBrowserPersistResourcesParent.h"
#include "WebBrowserPersistSerializeParent.h"
#include "WebBrowserPersistRemoteDocument.h"
namespace mozilla {
WebBrowserPersistDocumentParent::WebBrowserPersistDocumentParent()
: mReflection(nullptr) {}
void WebBrowserPersistDocumentParent::SetOnReady(
nsIWebBrowserPersistDocumentReceiver* aOnReady) {
MOZ_ASSERT(aOnReady);
MOZ_ASSERT(!mOnReady);
MOZ_ASSERT(!mReflection);
mOnReady = aOnReady;
}
void WebBrowserPersistDocumentParent::ActorDestroy(ActorDestroyReason aWhy) {
if (mReflection) {
mReflection->ActorDestroy();
mReflection = nullptr;
}
if (mOnReady) {
// Bug 1202887: If this is part of a subtree destruction, then
// anything which could cause another actor in that subtree to
// be Send__delete__()ed will cause use-after-free -- such as
// dropping the last reference to another document's
// WebBrowserPersistRemoteDocument. To avoid that, defer the
// callback until after the entire subtree is destroyed.
nsCOMPtr<nsIRunnable> errorLater = NewRunnableMethod<nsresult>(
"nsIWebBrowserPersistDocumentReceiver::OnError", mOnReady,
&nsIWebBrowserPersistDocumentReceiver::OnError, NS_ERROR_FAILURE);
NS_DispatchToCurrentThread(errorLater);
mOnReady = nullptr;
}
}
WebBrowserPersistDocumentParent::~WebBrowserPersistDocumentParent() {
MOZ_RELEASE_ASSERT(!mReflection);
MOZ_ASSERT(!mOnReady);
}
mozilla::ipc::IPCResult WebBrowserPersistDocumentParent::RecvAttributes(
const Attrs& aAttrs, const Maybe<IPCStream>& aPostStream) {
// Deserialize the postData unconditionally so that fds aren't leaked.
nsCOMPtr<nsIInputStream> postData =
mozilla::ipc::DeserializeIPCStream(aPostStream);
if (!mOnReady || mReflection) {
return IPC_FAIL_NO_REASON(this);
}
mReflection = new WebBrowserPersistRemoteDocument(this, aAttrs, postData);
RefPtr<WebBrowserPersistRemoteDocument> reflection = mReflection;
mOnReady->OnDocumentReady(reflection);
mOnReady = nullptr;
return IPC_OK();
}
mozilla::ipc::IPCResult WebBrowserPersistDocumentParent::RecvInitFailure(
const nsresult& aFailure) {
if (!mOnReady || mReflection) {
return IPC_FAIL_NO_REASON(this);
}
mOnReady->OnError(aFailure);
mOnReady = nullptr;
// Warning: Send__delete__ deallocates this object.
IProtocol* mgr = Manager();
if (!Send__delete__(this)) {
return IPC_FAIL_NO_REASON(mgr);
}
return IPC_OK();
}
PWebBrowserPersistResourcesParent*
WebBrowserPersistDocumentParent::AllocPWebBrowserPersistResourcesParent() {
MOZ_CRASH("Don't use this; construct the actor directly and AddRef.");
return nullptr;
}
bool WebBrowserPersistDocumentParent::DeallocPWebBrowserPersistResourcesParent(
PWebBrowserPersistResourcesParent* aActor) {
// Turn the ref held by IPC back into an nsRefPtr.
RefPtr<WebBrowserPersistResourcesParent> actor =
already_AddRefed<WebBrowserPersistResourcesParent>(
static_cast<WebBrowserPersistResourcesParent*>(aActor));
return true;
}
PWebBrowserPersistSerializeParent*
WebBrowserPersistDocumentParent::AllocPWebBrowserPersistSerializeParent(
const WebBrowserPersistURIMap& aMap, const nsCString& aRequestedContentType,
const uint32_t& aEncoderFlags, const uint32_t& aWrapColumn) {
MOZ_CRASH("Don't use this; construct the actor directly.");
return nullptr;
}
bool WebBrowserPersistDocumentParent::DeallocPWebBrowserPersistSerializeParent(
PWebBrowserPersistSerializeParent* aActor) {
delete aActor;
return true;
}
} // namespace mozilla