diff -r 000000000000 -r 6474c204b198 netwerk/base/src/nsUDPSocket.h --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/netwerk/base/src/nsUDPSocket.h Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,103 @@ +/* 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" + +//----------------------------------------------------------------------------- + +class nsUDPSocket : public nsASocketHandler + , public nsIUDPSocket +{ +public: + NS_DECL_THREADSAFE_ISUPPORTS + NS_DECL_NSIUDPSOCKET + + // nsASocketHandler methods: + virtual void OnSocketReady(PRFileDesc* fd, int16_t outFlags); + virtual void OnSocketDetached(PRFileDesc* fd); + virtual void IsLocal(bool* aIsLocal); + + uint64_t ByteCountSent() { return mByteWriteCount; } + uint64_t ByteCountReceived() { return mByteReadCount; } + + void AddOutputBytes(uint64_t aBytes); + + nsUDPSocket(); + + // This must be public to support older compilers (xlC_r on AIX) + virtual ~nsUDPSocket(); + +private: + void OnMsgClose(); + void OnMsgAttach(); + + // try attaching our socket (mFD) to the STS's poll list. + nsresult TryAttach(); + + // lock protects access to mListener; + // so mListener is not cleared while being used/locked. + mozilla::Mutex mLock; + PRFileDesc *mFD; + mozilla::net::NetAddr mAddr; + nsCOMPtr mListener; + nsCOMPtr mListenerTarget; + bool mAttached; + nsRefPtr 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(mozilla::net::NetAddr* aAddr, + nsIOutputStream* aOutputStream, + FallibleTArray& aData); + +private: + virtual ~nsUDPMessage(); + + mozilla::net::NetAddr mAddr; + nsCOMPtr mOutputStream; + FallibleTArray mData; + JS::Heap mJsobj; +}; + + +//----------------------------------------------------------------------------- + +class nsUDPOutputStream : public nsIOutputStream +{ +public: + NS_DECL_THREADSAFE_ISUPPORTS + NS_DECL_NSIOUTPUTSTREAM + + nsUDPOutputStream(nsUDPSocket* aSocket, + PRFileDesc* aFD, + PRNetAddr& aPrClientAddr); + virtual ~nsUDPOutputStream(); + +private: + nsRefPtr mSocket; + PRFileDesc *mFD; + PRNetAddr mPrClientAddr; + bool mIsClosed; +}; + +#endif // nsUDPSocket_h__