forked from mirrors/gecko-dev
		
	Depends on D172974 Differential Revision: https://phabricator.services.mozilla.com/D172975
		
			
				
	
	
		
			109 lines
		
	
	
	
		
			3.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			109 lines
		
	
	
	
		
			3.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
 | 
						|
/* 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 "ExtensionPort.h"
 | 
						|
#include "ExtensionBrowser.h"
 | 
						|
#include "ExtensionEventManager.h"
 | 
						|
 | 
						|
#include "mozilla/dom/BindingUtils.h"  // SequenceRooter
 | 
						|
#include "mozilla/dom/ExtensionPortBinding.h"
 | 
						|
#include "mozilla/dom/ScriptSettings.h"  // AutoEntryScript
 | 
						|
#include "nsIGlobalObject.h"
 | 
						|
 | 
						|
namespace mozilla {
 | 
						|
namespace extensions {
 | 
						|
 | 
						|
NS_IMPL_CYCLE_COLLECTING_ADDREF(ExtensionPort);
 | 
						|
NS_IMPL_CYCLE_COLLECTING_RELEASE(ExtensionPort)
 | 
						|
 | 
						|
NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE_CLASS(ExtensionPort)
 | 
						|
 | 
						|
NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(ExtensionPort)
 | 
						|
  // Clean the entry for this instance from the ports lookup map
 | 
						|
  // stored in the related ExtensionBrowser instance.
 | 
						|
  tmp->ForgetReleasedPort();
 | 
						|
  NS_IMPL_CYCLE_COLLECTION_UNLINK(mExtensionBrowser, mOnDisconnectEventMgr,
 | 
						|
                                  mOnMessageEventMgr)
 | 
						|
  NS_IMPL_CYCLE_COLLECTION_UNLINK_PRESERVED_WRAPPER
 | 
						|
  NS_IMPL_CYCLE_COLLECTION_UNLINK_WEAK_PTR
 | 
						|
NS_IMPL_CYCLE_COLLECTION_UNLINK_END
 | 
						|
 | 
						|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(ExtensionPort)
 | 
						|
  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mExtensionBrowser, mOnDisconnectEventMgr,
 | 
						|
                                    mOnMessageEventMgr)
 | 
						|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
 | 
						|
 | 
						|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(ExtensionPort)
 | 
						|
  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
 | 
						|
  NS_INTERFACE_MAP_ENTRY(nsISupports)
 | 
						|
NS_INTERFACE_MAP_END
 | 
						|
 | 
						|
NS_IMPL_WEBEXT_EVENTMGR(ExtensionPort, u"onMessage"_ns, OnMessage)
 | 
						|
NS_IMPL_WEBEXT_EVENTMGR(ExtensionPort, u"onDisconnect"_ns, OnDisconnect)
 | 
						|
 | 
						|
// static
 | 
						|
already_AddRefed<ExtensionPort> ExtensionPort::Create(
 | 
						|
    nsIGlobalObject* aGlobal, ExtensionBrowser* aExtensionBrowser,
 | 
						|
    UniquePtr<dom::ExtensionPortDescriptor>&& aPortDescriptor) {
 | 
						|
  RefPtr<ExtensionPort> port =
 | 
						|
      new ExtensionPort(aGlobal, aExtensionBrowser, std::move(aPortDescriptor));
 | 
						|
  return port.forget();
 | 
						|
}
 | 
						|
 | 
						|
// static
 | 
						|
UniquePtr<dom::ExtensionPortDescriptor> ExtensionPort::ToPortDescriptor(
 | 
						|
    JS::Handle<JS::Value> aDescriptorValue, ErrorResult& aRv) {
 | 
						|
  if (!aDescriptorValue.isObject()) {
 | 
						|
    aRv.Throw(NS_ERROR_UNEXPECTED);
 | 
						|
    return nullptr;
 | 
						|
  }
 | 
						|
 | 
						|
  dom::AutoEntryScript aes(&aDescriptorValue.toObject(), __func__);
 | 
						|
  JSContext* acx = aes.cx();
 | 
						|
  auto portDescriptor = MakeUnique<dom::ExtensionPortDescriptor>();
 | 
						|
  if (!portDescriptor->Init(acx, aDescriptorValue, __func__)) {
 | 
						|
    aRv.Throw(NS_ERROR_UNEXPECTED);
 | 
						|
    return nullptr;
 | 
						|
  }
 | 
						|
 | 
						|
  return portDescriptor;
 | 
						|
}
 | 
						|
 | 
						|
ExtensionPort::ExtensionPort(
 | 
						|
    nsIGlobalObject* aGlobal, ExtensionBrowser* aExtensionBrowser,
 | 
						|
    UniquePtr<dom::ExtensionPortDescriptor>&& aPortDescriptor)
 | 
						|
    : mGlobal(aGlobal),
 | 
						|
      mExtensionBrowser(aExtensionBrowser),
 | 
						|
      mPortDescriptor(std::move(aPortDescriptor)) {
 | 
						|
  MOZ_DIAGNOSTIC_ASSERT(mGlobal);
 | 
						|
  MOZ_DIAGNOSTIC_ASSERT(mExtensionBrowser);
 | 
						|
}
 | 
						|
 | 
						|
void ExtensionPort::ForgetReleasedPort() {
 | 
						|
  if (mExtensionBrowser) {
 | 
						|
    mExtensionBrowser->ForgetReleasedPort(mPortDescriptor->mPortId);
 | 
						|
    mExtensionBrowser = nullptr;
 | 
						|
  }
 | 
						|
  mPortDescriptor = nullptr;
 | 
						|
  mGlobal = nullptr;
 | 
						|
}
 | 
						|
 | 
						|
nsString ExtensionPort::GetAPIObjectId() const {
 | 
						|
  return mPortDescriptor->mPortId;
 | 
						|
}
 | 
						|
 | 
						|
JSObject* ExtensionPort::WrapObject(JSContext* aCx,
 | 
						|
                                    JS::Handle<JSObject*> aGivenProto) {
 | 
						|
  return dom::ExtensionPort_Binding::Wrap(aCx, this, aGivenProto);
 | 
						|
}
 | 
						|
 | 
						|
nsIGlobalObject* ExtensionPort::GetParentObject() const { return mGlobal; }
 | 
						|
 | 
						|
void ExtensionPort::GetName(nsAString& aString) {
 | 
						|
  aString.Assign(mPortDescriptor->mName);
 | 
						|
}
 | 
						|
 | 
						|
}  // namespace extensions
 | 
						|
}  // namespace mozilla
 |