michael@0: /* -*- Mode: C++; c-basic-offset: 2; indent-tabs-mode: nil; tab-width: 8; -*- */
michael@0: /* vim: set sw=4 ts=8 et tw=80 : */
michael@0: /* This Source Code Form is subject to the terms of the Mozilla Public
michael@0:  * License, v. 2.0. If a copy of the MPL was not distributed with this
michael@0:  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
michael@0: 
michael@0: #include "nsInProcessTabChildGlobal.h"
michael@0: #include "nsContentUtils.h"
michael@0: #include "nsIScriptSecurityManager.h"
michael@0: #include "nsIInterfaceRequestorUtils.h"
michael@0: #include "nsIComponentManager.h"
michael@0: #include "nsIServiceManager.h"
michael@0: #include "nsIJSRuntimeService.h"
michael@0: #include "nsComponentManagerUtils.h"
michael@0: #include "nsNetUtil.h"
michael@0: #include "nsScriptLoader.h"
michael@0: #include "nsFrameLoader.h"
michael@0: #include "xpcpublic.h"
michael@0: #include "nsIMozBrowserFrame.h"
michael@0: #include "nsDOMClassInfoID.h"
michael@0: #include "mozilla/EventDispatcher.h"
michael@0: #include "mozilla/dom/StructuredCloneUtils.h"
michael@0: #include "js/StructuredClone.h"
michael@0: 
michael@0: using mozilla::dom::StructuredCloneData;
michael@0: using mozilla::dom::StructuredCloneClosure;
michael@0: using namespace mozilla;
michael@0: 
michael@0: bool
michael@0: nsInProcessTabChildGlobal::DoSendBlockingMessage(JSContext* aCx,
michael@0:                                                  const nsAString& aMessage,
michael@0:                                                  const dom::StructuredCloneData& aData,
michael@0:                                                  JS::Handle<JSObject *> aCpows,
michael@0:                                                  nsIPrincipal* aPrincipal,
michael@0:                                                  InfallibleTArray<nsString>* aJSONRetVal,
michael@0:                                                  bool aIsSync)
michael@0: {
michael@0:   nsTArray<nsCOMPtr<nsIRunnable> > asyncMessages;
michael@0:   asyncMessages.SwapElements(mASyncMessages);
michael@0:   uint32_t len = asyncMessages.Length();
michael@0:   for (uint32_t i = 0; i < len; ++i) {
michael@0:     nsCOMPtr<nsIRunnable> async = asyncMessages[i];
michael@0:     async->Run();
michael@0:   }
michael@0:   if (mChromeMessageManager) {
michael@0:     SameProcessCpowHolder cpows(js::GetRuntime(aCx), aCpows);
michael@0:     nsRefPtr<nsFrameMessageManager> mm = mChromeMessageManager;
michael@0:     mm->ReceiveMessage(mOwner, aMessage, true, &aData, &cpows, aPrincipal,
michael@0:                        aJSONRetVal);
michael@0:   }
michael@0:   return true;
michael@0: }
michael@0: 
michael@0: class nsAsyncMessageToParent : public nsSameProcessAsyncMessageBase,
michael@0:                                public nsRunnable
michael@0: {
michael@0: public:
michael@0:   nsAsyncMessageToParent(JSContext* aCx,
michael@0:                          nsInProcessTabChildGlobal* aTabChild,
michael@0:                          const nsAString& aMessage,
michael@0:                          const StructuredCloneData& aData,
michael@0:                          JS::Handle<JSObject *> aCpows,
michael@0:                          nsIPrincipal* aPrincipal)
michael@0:     : nsSameProcessAsyncMessageBase(aCx, aMessage, aData, aCpows, aPrincipal),
michael@0:       mTabChild(aTabChild), mRun(false)
michael@0:   {
michael@0:   }
michael@0: 
michael@0:   NS_IMETHOD Run()
michael@0:   {
michael@0:     if (mRun) {
michael@0:       return NS_OK;
michael@0:     }
michael@0: 
michael@0:     mRun = true;
michael@0:     mTabChild->mASyncMessages.RemoveElement(this);
michael@0:     ReceiveMessage(mTabChild->mOwner, mTabChild->mChromeMessageManager);
michael@0:     return NS_OK;
michael@0:   }
michael@0:   nsRefPtr<nsInProcessTabChildGlobal> mTabChild;
michael@0:   // True if this runnable has already been called. This can happen if DoSendSyncMessage
michael@0:   // is called while waiting for an asynchronous message send.
michael@0:   bool mRun;
michael@0: };
michael@0: 
michael@0: bool
michael@0: nsInProcessTabChildGlobal::DoSendAsyncMessage(JSContext* aCx,
michael@0:                                               const nsAString& aMessage,
michael@0:                                               const StructuredCloneData& aData,
michael@0:                                               JS::Handle<JSObject *> aCpows,
michael@0:                                               nsIPrincipal* aPrincipal)
michael@0: {
michael@0:   nsCOMPtr<nsIRunnable> ev =
michael@0:     new nsAsyncMessageToParent(aCx, this, aMessage, aData, aCpows, aPrincipal);
michael@0:   mASyncMessages.AppendElement(ev);
michael@0:   NS_DispatchToCurrentThread(ev);
michael@0:   return true;
michael@0: }
michael@0: 
michael@0: nsInProcessTabChildGlobal::nsInProcessTabChildGlobal(nsIDocShell* aShell,
michael@0:                                                      nsIContent* aOwner,
michael@0:                                                      nsFrameMessageManager* aChrome)
michael@0: : mDocShell(aShell), mInitialized(false), mLoadingScript(false),
michael@0:   mOwner(aOwner), mChromeMessageManager(aChrome)
michael@0: {
michael@0: 
michael@0:   // If owner corresponds to an <iframe mozbrowser> or <iframe mozapp>, we'll
michael@0:   // have to tweak our PreHandleEvent implementation.
michael@0:   nsCOMPtr<nsIMozBrowserFrame> browserFrame = do_QueryInterface(mOwner);
michael@0:   if (browserFrame) {
michael@0:     mIsBrowserOrAppFrame = browserFrame->GetReallyIsBrowserOrApp();
michael@0:   }
michael@0:   else {
michael@0:     mIsBrowserOrAppFrame = false;
michael@0:   }
michael@0: }
michael@0: 
michael@0: nsInProcessTabChildGlobal::~nsInProcessTabChildGlobal()
michael@0: {
michael@0: }
michael@0: 
michael@0: /* [notxpcom] boolean markForCC (); */
michael@0: // This method isn't automatically forwarded safely because it's notxpcom, so
michael@0: // the IDL binding doesn't know what value to return.
michael@0: NS_IMETHODIMP_(bool)
michael@0: nsInProcessTabChildGlobal::MarkForCC()
michael@0: {
michael@0:   return mMessageManager ? mMessageManager->MarkForCC() : false;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: nsInProcessTabChildGlobal::Init()
michael@0: {
michael@0: #ifdef DEBUG
michael@0:   nsresult rv =
michael@0: #endif
michael@0:   InitTabChildGlobal();
michael@0:   NS_WARN_IF_FALSE(NS_SUCCEEDED(rv),
michael@0:                    "Couldn't initialize nsInProcessTabChildGlobal");
michael@0:   mMessageManager = new nsFrameMessageManager(this,
michael@0:                                               nullptr,
michael@0:                                               dom::ipc::MM_CHILD);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMPL_CYCLE_COLLECTION_INHERITED(nsInProcessTabChildGlobal,
michael@0:                                    DOMEventTargetHelper,
michael@0:                                    mMessageManager,
michael@0:                                    mGlobal)
michael@0: 
michael@0: NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION_INHERITED(nsInProcessTabChildGlobal)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsIMessageListenerManager)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsIMessageSender)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsISyncMessageSender)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsIContentFrameMessageManager)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsIInProcessContentFrameMessageManager)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsIScriptObjectPrincipal)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsIGlobalObject)
michael@0:   NS_INTERFACE_MAP_ENTRY(nsISupportsWeakReference)
michael@0:   NS_DOM_INTERFACE_MAP_ENTRY_CLASSINFO(ContentFrameMessageManager)
michael@0: NS_INTERFACE_MAP_END_INHERITING(DOMEventTargetHelper)
michael@0: 
michael@0: NS_IMPL_ADDREF_INHERITED(nsInProcessTabChildGlobal, DOMEventTargetHelper)
michael@0: NS_IMPL_RELEASE_INHERITED(nsInProcessTabChildGlobal, DOMEventTargetHelper)
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInProcessTabChildGlobal::GetContent(nsIDOMWindow** aContent)
michael@0: {
michael@0:   *aContent = nullptr;
michael@0:   nsCOMPtr<nsIDOMWindow> window = do_GetInterface(mDocShell);
michael@0:   window.swap(*aContent);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInProcessTabChildGlobal::GetDocShell(nsIDocShell** aDocShell)
michael@0: {
michael@0:   NS_IF_ADDREF(*aDocShell = mDocShell);
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInProcessTabChildGlobal::Btoa(const nsAString& aBinaryData,
michael@0:                             nsAString& aAsciiBase64String)
michael@0: {
michael@0:   return nsContentUtils::Btoa(aBinaryData, aAsciiBase64String);
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInProcessTabChildGlobal::Atob(const nsAString& aAsciiString,
michael@0:                             nsAString& aBinaryData)
michael@0: {
michael@0:   return nsContentUtils::Atob(aAsciiString, aBinaryData);
michael@0: }
michael@0: 
michael@0: 
michael@0: NS_IMETHODIMP
michael@0: nsInProcessTabChildGlobal::PrivateNoteIntentionalCrash()
michael@0: {
michael@0:     return NS_ERROR_NOT_IMPLEMENTED;
michael@0: }
michael@0: 
michael@0: void
michael@0: nsInProcessTabChildGlobal::Disconnect()
michael@0: {
michael@0:   // Let the frame scripts know the child is being closed. We do any other
michael@0:   // cleanup after the event has been fired. See DelayedDisconnect
michael@0:   nsContentUtils::AddScriptRunner(
michael@0:      NS_NewRunnableMethod(this, &nsInProcessTabChildGlobal::DelayedDisconnect)
michael@0:   );
michael@0: }
michael@0: 
michael@0: void
michael@0: nsInProcessTabChildGlobal::DelayedDisconnect()
michael@0: {
michael@0:   // Don't let the event escape
michael@0:   mOwner = nullptr;
michael@0: 
michael@0:   // Fire the "unload" event
michael@0:   DOMEventTargetHelper::DispatchTrustedEvent(NS_LITERAL_STRING("unload"));
michael@0: 
michael@0:   // Continue with the Disconnect cleanup
michael@0:   nsCOMPtr<nsPIDOMWindow> win = do_GetInterface(mDocShell);
michael@0:   if (win) {
michael@0:     MOZ_ASSERT(win->IsOuterWindow());
michael@0:     win->SetChromeEventHandler(win->GetChromeEventHandler());
michael@0:   }
michael@0:   mDocShell = nullptr;
michael@0:   mChromeMessageManager = nullptr;
michael@0:   if (mMessageManager) {
michael@0:     static_cast<nsFrameMessageManager*>(mMessageManager.get())->Disconnect();
michael@0:     mMessageManager = nullptr;
michael@0:   }
michael@0:   if (mListenerManager) {
michael@0:     mListenerManager->Disconnect();
michael@0:   }
michael@0: }
michael@0: 
michael@0: NS_IMETHODIMP_(nsIContent *)
michael@0: nsInProcessTabChildGlobal::GetOwnerContent()
michael@0: {
michael@0:   return mOwner;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: nsInProcessTabChildGlobal::PreHandleEvent(EventChainPreVisitor& aVisitor)
michael@0: {
michael@0:   aVisitor.mCanHandle = true;
michael@0: 
michael@0:   if (mIsBrowserOrAppFrame &&
michael@0:       (!mOwner || !nsContentUtils::IsInChromeDocshell(mOwner->OwnerDoc()))) {
michael@0:     if (mOwner) {
michael@0:       nsPIDOMWindow* innerWindow = mOwner->OwnerDoc()->GetInnerWindow();
michael@0:       if (innerWindow) {
michael@0:         aVisitor.mParentTarget = innerWindow->GetParentTarget();
michael@0:       }
michael@0:     }
michael@0:   } else {
michael@0:     aVisitor.mParentTarget = mOwner;
michael@0:   }
michael@0: 
michael@0: #ifdef DEBUG
michael@0:   if (mOwner) {
michael@0:     nsCOMPtr<nsIFrameLoaderOwner> owner = do_QueryInterface(mOwner);
michael@0:     nsRefPtr<nsFrameLoader> fl = owner->GetFrameLoader();
michael@0:     if (fl) {
michael@0:       NS_ASSERTION(this == fl->GetTabChildGlobalAsEventTarget(),
michael@0:                    "Wrong event target!");
michael@0:       NS_ASSERTION(fl->mMessageManager == mChromeMessageManager,
michael@0:                    "Wrong message manager!");
michael@0:     }
michael@0:   }
michael@0: #endif
michael@0: 
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: nsresult
michael@0: nsInProcessTabChildGlobal::InitTabChildGlobal()
michael@0: {
michael@0:   nsAutoCString id;
michael@0:   id.AssignLiteral("inProcessTabChildGlobal");
michael@0:   nsIURI* uri = mOwner->OwnerDoc()->GetDocumentURI();
michael@0:   if (uri) {
michael@0:     nsAutoCString u;
michael@0:     uri->GetSpec(u);
michael@0:     id.AppendLiteral("?ownedBy=");
michael@0:     id.Append(u);
michael@0:   }
michael@0:   nsISupports* scopeSupports = NS_ISUPPORTS_CAST(EventTarget*, this);
michael@0:   NS_ENSURE_STATE(InitTabChildGlobalInternal(scopeSupports, id));
michael@0:   return NS_OK;
michael@0: }
michael@0: 
michael@0: class nsAsyncScriptLoad : public nsRunnable
michael@0: {
michael@0: public:
michael@0:     nsAsyncScriptLoad(nsInProcessTabChildGlobal* aTabChild, const nsAString& aURL,
michael@0:                       bool aRunInGlobalScope)
michael@0:       : mTabChild(aTabChild), mURL(aURL), mRunInGlobalScope(aRunInGlobalScope) {}
michael@0: 
michael@0:   NS_IMETHOD Run()
michael@0:   {
michael@0:     mTabChild->LoadFrameScript(mURL, mRunInGlobalScope);
michael@0:     return NS_OK;
michael@0:   }
michael@0:   nsRefPtr<nsInProcessTabChildGlobal> mTabChild;
michael@0:   nsString mURL;
michael@0:   bool mRunInGlobalScope;
michael@0: };
michael@0: 
michael@0: void
michael@0: nsInProcessTabChildGlobal::LoadFrameScript(const nsAString& aURL, bool aRunInGlobalScope)
michael@0: {
michael@0:   if (!nsContentUtils::IsSafeToRunScript()) {
michael@0:     nsContentUtils::AddScriptRunner(new nsAsyncScriptLoad(this, aURL, aRunInGlobalScope));
michael@0:     return;
michael@0:   }
michael@0:   if (!mInitialized) {
michael@0:     mInitialized = true;
michael@0:     Init();
michael@0:   }
michael@0:   bool tmp = mLoadingScript;
michael@0:   mLoadingScript = true;
michael@0:   LoadFrameScriptInternal(aURL, aRunInGlobalScope);
michael@0:   mLoadingScript = tmp;
michael@0: }