forked from mirrors/gecko-dev
		
	 8f6f9181e7
			
		
	
	
		8f6f9181e7
		
	
	
	
	
		
			
			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
 |