mirror of
https://gitlab.winehq.org/wine/wine-gecko.git
synced 2024-09-13 09:24:08 -07:00
33bb32f549
This patch was automatically generated using the following script: function convert() { echo "Converting $1 to $2..." find . \ ! -wholename "*/.git*" \ ! -wholename "obj-ff-dbg*" \ -type f \ \( -iname "*.cpp" \ -o -iname "*.h" \ -o -iname "*.c" \ -o -iname "*.cc" \ -o -iname "*.idl" \ -o -iname "*.ipdl" \ -o -iname "*.ipdlh" \ -o -iname "*.mm" \) | \ xargs -n 1 sed -i -e "s/\b$1\b/$2/g" } convert MOZ_OVERRIDE override convert MOZ_FINAL final
112 lines
3.2 KiB
C++
112 lines
3.2 KiB
C++
/* 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 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);
|
|
|
|
// lock protects access to mListener;
|
|
// so mListener is not cleared while being used/locked.
|
|
mozilla::Mutex mLock;
|
|
PRFileDesc *mFD;
|
|
mozilla::net::NetAddr mAddr;
|
|
nsCOMPtr<nsIUDPSocketListener> mListener;
|
|
nsCOMPtr<nsIEventTarget> mListenerTarget;
|
|
bool mAttached;
|
|
nsRefPtr<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(mozilla::net::NetAddr* aAddr,
|
|
nsIOutputStream* aOutputStream,
|
|
FallibleTArray<uint8_t>& aData);
|
|
|
|
private:
|
|
virtual ~nsUDPMessage();
|
|
|
|
mozilla::net::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();
|
|
|
|
nsRefPtr<nsUDPSocket> mSocket;
|
|
PRFileDesc *mFD;
|
|
PRNetAddr mPrClientAddr;
|
|
bool mIsClosed;
|
|
};
|
|
|
|
#endif // nsUDPSocket_h__
|