gecko-dev/netwerk/base/nsUDPSocket.h

121 строка
3.3 KiB
C
Исходник Обычный вид История

2013-03-03 12:04:38 +04:00
/* 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__
2013-03-03 12:04:38 +04:00
#include "nsIUDPSocket.h"
2013-03-03 12:04:38 +04:00
#include "mozilla/Mutex.h"
#include "nsIOutputStream.h"
#include "nsAutoPtr.h"
#include "nsCycleCollectionParticipant.h"
2013-03-03 12:04:38 +04:00
//-----------------------------------------------------------------------------
namespace mozilla {
namespace net {
class nsUDPSocket final : public nsASocketHandler
, public nsIUDPSocket
2013-03-03 12:04:38 +04:00
{
public:
NS_DECL_THREADSAFE_ISUPPORTS
NS_DECL_NSIUDPSOCKET
2013-03-03 12:04:38 +04:00
// nsASocketHandler methods:
virtual void OnSocketReady(PRFileDesc* fd, int16_t outFlags) override;
virtual void OnSocketDetached(PRFileDesc* fd) override;
virtual void IsLocal(bool* aIsLocal) override;
2013-03-03 12:04:38 +04:00
uint64_t ByteCountSent() override { return mByteWriteCount; }
uint64_t ByteCountReceived() override { return mByteReadCount; }
2013-03-03 12:04:38 +04:00
void AddOutputBytes(uint64_t aBytes);
nsUDPSocket();
2013-03-03 12:04:38 +04:00
private:
virtual ~nsUDPSocket();
2013-03-03 12:04:38 +04:00
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();
2013-03-03 12:04:38 +04:00
// lock protects access to mListener;
// so mListener is not cleared while being used/locked.
Mutex mLock;
PRFileDesc *mFD;
NetAddr mAddr;
OriginAttributes mOriginAttributes;
nsCOMPtr<nsIUDPSocketListener> mListener;
2013-03-03 12:04:38 +04:00
nsCOMPtr<nsIEventTarget> mListenerTarget;
bool mAttached;
RefPtr<nsSocketTransportService> mSts;
2013-03-03 12:04:38 +04:00
uint64_t mByteReadCount;
uint64_t mByteWriteCount;
};
//-----------------------------------------------------------------------------
class nsUDPMessage : public nsIUDPMessage
{
public:
NS_DECL_CYCLE_COLLECTING_ISUPPORTS
NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS(nsUDPMessage)
2013-03-03 12:04:38 +04:00
NS_DECL_NSIUDPMESSAGE
nsUDPMessage(NetAddr* aAddr,
2013-03-03 12:04:38 +04:00
nsIOutputStream* aOutputStream,
FallibleTArray<uint8_t>& aData);
2013-03-03 12:04:38 +04:00
private:
virtual ~nsUDPMessage();
NetAddr mAddr;
2013-03-03 12:04:38 +04:00
nsCOMPtr<nsIOutputStream> mOutputStream;
FallibleTArray<uint8_t> mData;
JS::Heap<JSObject*> mJsobj;
2013-03-03 12:04:38 +04:00
};
//-----------------------------------------------------------------------------
class nsUDPOutputStream : public nsIOutputStream
{
public:
NS_DECL_THREADSAFE_ISUPPORTS
2013-03-03 12:04:38 +04:00
NS_DECL_NSIOUTPUTSTREAM
nsUDPOutputStream(nsUDPSocket* aSocket,
2013-03-03 12:04:38 +04:00
PRFileDesc* aFD,
PRNetAddr& aPrClientAddr);
private:
virtual ~nsUDPOutputStream();
Bug 1207245 - part 6 - rename nsRefPtr<T> to RefPtr<T>; r=ehsan; a=Tomcat The bulk of this commit was generated with a script, executed at the top level of a typical source code checkout. The only non-machine-generated part was modifying MFBT's moz.build to reflect the new naming. CLOSED TREE makes big refactorings like this a piece of cake. # The main substitution. find . -name '*.cpp' -o -name '*.cc' -o -name '*.h' -o -name '*.mm' -o -name '*.idl'| \ xargs perl -p -i -e ' s/nsRefPtr\.h/RefPtr\.h/g; # handle includes s/nsRefPtr ?</RefPtr</g; # handle declarations and variables ' # Handle a special friend declaration in gfx/layers/AtomicRefCountedWithFinalize.h. perl -p -i -e 's/::nsRefPtr;/::RefPtr;/' gfx/layers/AtomicRefCountedWithFinalize.h # Handle nsRefPtr.h itself, a couple places that define constructors # from nsRefPtr, and code generators specially. We do this here, rather # than indiscriminantly s/nsRefPtr/RefPtr/, because that would rename # things like nsRefPtrHashtable. perl -p -i -e 's/nsRefPtr/RefPtr/g' \ mfbt/nsRefPtr.h \ xpcom/glue/nsCOMPtr.h \ xpcom/base/OwningNonNull.h \ ipc/ipdl/ipdl/lower.py \ ipc/ipdl/ipdl/builtin.py \ dom/bindings/Codegen.py \ python/lldbutils/lldbutils/utils.py # In our indiscriminate substitution above, we renamed # nsRefPtrGetterAddRefs, the class behind getter_AddRefs. Fix that up. find . -name '*.cpp' -o -name '*.h' -o -name '*.idl' | \ xargs perl -p -i -e 's/nsRefPtrGetterAddRefs/RefPtrGetterAddRefs/g' if [ -d .git ]; then git mv mfbt/nsRefPtr.h mfbt/RefPtr.h else hg mv mfbt/nsRefPtr.h mfbt/RefPtr.h fi --HG-- rename : mfbt/nsRefPtr.h => mfbt/RefPtr.h
2015-10-18 08:24:48 +03:00
RefPtr<nsUDPSocket> mSocket;
2013-03-03 12:04:38 +04:00
PRFileDesc *mFD;
PRNetAddr mPrClientAddr;
bool mIsClosed;
};
} // namespace net
} // namespace mozilla
#endif // nsUDPSocket_h__