/* vim:set ts=2 sw=2 et cindent: */ /* 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/. */ #ifndef nsUDPSocket_h__ #define nsUDPSocket_h__ #include "nsIUDPSocket.h" #include "mozilla/Mutex.h" #include "nsIOutputStream.h" #include "nsAutoPtr.h" #include "nsCycleCollectionParticipant.h" //----------------------------------------------------------------------------- namespace mozilla { namespace net { class nsUDPSocket final : public nsASocketHandler , public nsIUDPSocket { public: NS_DECL_THREADSAFE_ISUPPORTS NS_DECL_NSIUDPSOCKET // nsASocketHandler methods: virtual void OnSocketReady(PRFileDesc* fd, int16_t outFlags) override; virtual void OnSocketDetached(PRFileDesc* fd) override; virtual void IsLocal(bool* aIsLocal) override; uint64_t ByteCountSent() override { return mByteWriteCount; } uint64_t ByteCountReceived() override { return mByteReadCount; } void AddOutputBytes(uint64_t aBytes); nsUDPSocket(); private: virtual ~nsUDPSocket(); void OnMsgClose(); void OnMsgAttach(); // try attaching our socket (mFD) to the STS's poll list. nsresult TryAttach(); friend class SetSocketOptionRunnable; nsresult SetSocketOption(const PRSocketOptionData& aOpt); nsresult JoinMulticastInternal(const PRNetAddr& aAddr, const PRNetAddr& aIface); nsresult LeaveMulticastInternal(const PRNetAddr& aAddr, const PRNetAddr& aIface); nsresult SetMulticastInterfaceInternal(const PRNetAddr& aIface); void CloseSocket(); // lock protects access to mListener; // so mListener is not cleared while being used/locked. Mutex mLock; PRFileDesc *mFD; NetAddr mAddr; uint32_t mAppId; bool mIsInIsolatedMozBrowserElement; nsCOMPtr<nsIUDPSocketListener> mListener; nsCOMPtr<nsIEventTarget> mListenerTarget; bool mAttached; RefPtr<nsSocketTransportService> mSts; uint64_t mByteReadCount; uint64_t mByteWriteCount; }; //----------------------------------------------------------------------------- class nsUDPMessage : public nsIUDPMessage { public: NS_DECL_CYCLE_COLLECTING_ISUPPORTS NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS(nsUDPMessage) NS_DECL_NSIUDPMESSAGE nsUDPMessage(NetAddr* aAddr, nsIOutputStream* aOutputStream, FallibleTArray<uint8_t>& aData); private: virtual ~nsUDPMessage(); NetAddr mAddr; nsCOMPtr<nsIOutputStream> mOutputStream; FallibleTArray<uint8_t> mData; JS::Heap<JSObject*> mJsobj; }; //----------------------------------------------------------------------------- class nsUDPOutputStream : public nsIOutputStream { public: NS_DECL_THREADSAFE_ISUPPORTS NS_DECL_NSIOUTPUTSTREAM nsUDPOutputStream(nsUDPSocket* aSocket, PRFileDesc* aFD, PRNetAddr& aPrClientAddr); private: virtual ~nsUDPOutputStream(); RefPtr<nsUDPSocket> mSocket; PRFileDesc *mFD; PRNetAddr mPrClientAddr; bool mIsClosed; }; } // namespace net } // namespace mozilla #endif // nsUDPSocket_h__