Init
This commit is contained in:
126
DependentExtensions/cat/net/win/TCPClient.hpp
Normal file
126
DependentExtensions/cat/net/win/TCPClient.hpp
Normal file
@ -0,0 +1,126 @@
|
||||
/*
|
||||
Copyright (c) 2009-2010 Christopher A. Taylor. All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
* Neither the name of LibCat nor the names of its contributors may be used
|
||||
to endorse or promote products derived from this software without
|
||||
specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef CAT_TCP_CLIENT_HPP
|
||||
#define CAT_TCP_CLIENT_HPP
|
||||
|
||||
/*
|
||||
Windows version of thread pool sockets with IO Completion Ports
|
||||
|
||||
Included from <cat/net/ThreadPoolSockets.hpp>
|
||||
Do not include directly
|
||||
*/
|
||||
|
||||
namespace cat {
|
||||
|
||||
|
||||
/*
|
||||
class TCPClient
|
||||
|
||||
Object that represents a TCPClient bound to a single port
|
||||
|
||||
ValidClient() : Returns true if the client socket is valid
|
||||
|
||||
Connect() : Connects to the given address
|
||||
DisconnectServer() : Disconnects from the server
|
||||
PostToServer() : Send a message to the server (will fail if not connected)
|
||||
|
||||
OnConnectToServer() : Called when connection is accepted
|
||||
OnReadFromServer() : Return false to disconnect the server in response to data
|
||||
OnWriteToServer() : Informs the derived class that data has been sent
|
||||
OnDisconnectFromServer() : Informs the derived class that the server has disconnected
|
||||
*/
|
||||
class TCPClient : public ThreadRefObject
|
||||
{
|
||||
// Remembers if socket is IPv6 so that user-provided
|
||||
// addresses can be promoted if necessary.
|
||||
bool _ipv6;
|
||||
Socket _socket;
|
||||
volatile u32 _disconnecting; // Disconnect flag
|
||||
|
||||
public:
|
||||
TCPClient(int priorityLevel);
|
||||
virtual ~TCPClient();
|
||||
|
||||
CAT_INLINE bool Valid() { return _socket != SOCKET_ERROR; }
|
||||
|
||||
void Disconnect();
|
||||
bool Connect(bool onlySupportIPv4, const NetAddr &remoteServerAddress);
|
||||
|
||||
public:
|
||||
bool Post(u8 *data, u32 data_bytes, u32 skip_bytes = 0);
|
||||
|
||||
private:
|
||||
bool ConnectEx(const NetAddr &remoteServerAddress);
|
||||
bool Read(AsyncBuffer *buffer = 0);
|
||||
bool Disco(AsyncBuffer *buffer = 0);
|
||||
|
||||
private:
|
||||
bool OnConnectEx(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
bool OnRead(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
bool OnWrite(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
bool OnDisco(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
|
||||
protected:
|
||||
virtual void OnConnectToServer(ThreadPoolLocalStorage *tls) = 0;
|
||||
virtual bool OnReadFromServer(ThreadPoolLocalStorage *tls, u8 *data, u32 bytes) = 0; // false = disconnect
|
||||
virtual bool OnWriteToServer(ThreadPoolLocalStorage *tls, AsyncBuffer *buffer, u32 bytes) = 0; // true = delete AsyncBuffer object
|
||||
virtual void OnDisconnectFromServer() = 0;
|
||||
};
|
||||
|
||||
|
||||
/*
|
||||
class TCPClientQueued
|
||||
|
||||
Base class for a TCP client that needs to queue up data for sending before
|
||||
a connection has been established. e.g. Uplink for a proxy server.
|
||||
|
||||
PostQueuedToServer() : Call in OnConnectToServer() to post the queued messages.
|
||||
*/
|
||||
class TCPClientQueued : public TCPClient
|
||||
{
|
||||
private:
|
||||
volatile bool _queuing;
|
||||
|
||||
Mutex _queueLock;
|
||||
AsyncBuffer *_queueBuffer;
|
||||
|
||||
protected:
|
||||
void PostQueuedToServer();
|
||||
|
||||
public:
|
||||
TCPClientQueued(int priorityLevel);
|
||||
virtual ~TCPClientQueued();
|
||||
|
||||
bool Post(u8 *data, u32 data_bytes, u32 skip_bytes = 0);
|
||||
};
|
||||
|
||||
|
||||
} // namespace cat
|
||||
|
||||
#endif // CAT_TCP_CLIENT_HPP
|
||||
102
DependentExtensions/cat/net/win/TCPConnection.hpp
Normal file
102
DependentExtensions/cat/net/win/TCPConnection.hpp
Normal file
@ -0,0 +1,102 @@
|
||||
/*
|
||||
Copyright (c) 2009-2010 Christopher A. Taylor. All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
* Neither the name of LibCat nor the names of its contributors may be used
|
||||
to endorse or promote products derived from this software without
|
||||
specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef CAT_TCP_CONNECTION_HPP
|
||||
#define CAT_TCP_CONNECTION_HPP
|
||||
|
||||
/*
|
||||
Windows version of thread pool sockets with IO Completion Ports
|
||||
|
||||
Included from <cat/net/ThreadPoolSockets.hpp>
|
||||
Do not include directly
|
||||
*/
|
||||
|
||||
namespace cat {
|
||||
|
||||
|
||||
/*
|
||||
class TCPConnection
|
||||
|
||||
Object that represents a TCPServer's connection from a TCPClient
|
||||
|
||||
Object is instantiated just before accepting a connection
|
||||
|
||||
DisconnectClient() : Disconnect the client
|
||||
PostToClient() : Send a message to the client
|
||||
ValidServerConnection() : Returns true iff the connection is valid
|
||||
|
||||
OnConnectFromClient() : Return false to deny this connection
|
||||
OnReadFromClient() : Return false to disconnect the client in response to a message
|
||||
OnWriteToClient() : Informs the derived class that data has been sent
|
||||
OnDisconectFromClient() : Informs the derived class that the client has disconnected
|
||||
*/
|
||||
class TCPConnection : public ThreadRefObject
|
||||
{
|
||||
friend class TCPServer;
|
||||
friend class ThreadPool;
|
||||
|
||||
public:
|
||||
TCPConnection(int priorityLevel);
|
||||
virtual ~TCPConnection();
|
||||
|
||||
bool ValidServerConnection();
|
||||
|
||||
void DisconnectClient();
|
||||
bool PostToClient(void *buffer, u32 bytes);
|
||||
|
||||
protected:
|
||||
virtual bool OnConnectFromClient(const NetAddr &remoteClientAddress) = 0; // false = disconnect
|
||||
virtual bool OnReadFromClient(u8 *data, u32 bytes) = 0; // false = disconnect
|
||||
virtual void OnWriteToClient(u32 bytes) = 0;
|
||||
virtual void OnDisconnectFromClient() = 0;
|
||||
|
||||
private:
|
||||
Socket _socket;
|
||||
LPFN_DISCONNECTEX _lpfnDisconnectEx;
|
||||
TypedOverlapped *_recvOv;
|
||||
volatile u32 _disconnecting;
|
||||
|
||||
private:
|
||||
bool AcceptConnection(Socket listenSocket, Socket acceptSocket,
|
||||
LPFN_DISCONNECTEX lpfnDisconnectEx, const NetAddr &acceptAddress,
|
||||
const NetAddr &remoteClientAddress);
|
||||
|
||||
bool QueueWSARecv();
|
||||
void OnWSARecvComplete(int error, u32 bytes);
|
||||
|
||||
bool QueueWSASend(TypedOverlapped *sendOv, u32 bytes);
|
||||
void OnWSASendComplete(int error, u32 bytes);
|
||||
|
||||
bool QueueDisconnectEx();
|
||||
void OnDisconnectExComplete(int error);
|
||||
};
|
||||
|
||||
|
||||
} // namespace cat
|
||||
|
||||
#endif // CAT_TCP_CONNECTION_HPP
|
||||
98
DependentExtensions/cat/net/win/TCPConnexion.hpp
Normal file
98
DependentExtensions/cat/net/win/TCPConnexion.hpp
Normal file
@ -0,0 +1,98 @@
|
||||
/*
|
||||
Copyright (c) 2009-2010 Christopher A. Taylor. All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
* Neither the name of LibCat nor the names of its contributors may be used
|
||||
to endorse or promote products derived from this software without
|
||||
specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef CAT_TCP_CONNEXION_HPP
|
||||
#define CAT_TCP_CONNEXION_HPP
|
||||
|
||||
/*
|
||||
Windows version of thread pool sockets with IO Completion Ports
|
||||
|
||||
Included from <cat/net/ThreadPoolSockets.hpp>
|
||||
Do not include directly
|
||||
*/
|
||||
|
||||
namespace cat {
|
||||
|
||||
|
||||
/*
|
||||
class TCPConnexion
|
||||
|
||||
Object that represents a TCPServer's connection from a TCPClient
|
||||
|
||||
Object is instantiated just before accepting a connection
|
||||
|
||||
Disconnect() : Disconnect the client
|
||||
Post() : Send a message to the client
|
||||
Valid() : Returns true if the connection is valid
|
||||
|
||||
OnConnectFromClient() : Return false to deny this connection
|
||||
OnReadFromClient() : Return false to disconnect the client in response to a message
|
||||
OnWriteToClient() : Informs the derived class that data has been sent
|
||||
OnDisconectFromClient() : Informs the derived class that the client has disconnected
|
||||
*/
|
||||
class TCPConnexion : public ThreadRefObject
|
||||
{
|
||||
friend class TCPServer;
|
||||
|
||||
Socket _socket;
|
||||
LPFN_DISCONNECTEX _lpfnDisconnectEx;
|
||||
volatile u32 _disconnecting;
|
||||
|
||||
bool Accept(ThreadPoolLocalStorage *tls, Socket listenSocket, Socket acceptSocket,
|
||||
LPFN_DISCONNECTEX lpfnDisconnectEx, const NetAddr &acceptAddress,
|
||||
const NetAddr &remoteClientAddress);
|
||||
|
||||
public:
|
||||
TCPConnexion(int priorityLevel);
|
||||
virtual ~TCPConnexion();
|
||||
|
||||
CAT_INLINE bool Valid() { return _socket != SOCKET_ERROR; }
|
||||
|
||||
void Disconnect();
|
||||
bool Post(u8 *data, u32 data_bytes, u32 skip_bytes = 0);
|
||||
|
||||
private:
|
||||
bool Read(AsyncBuffer *buffer = 0);
|
||||
bool Disco(AsyncBuffer *buffer = 0);
|
||||
|
||||
private:
|
||||
bool OnRead(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
bool OnWrite(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
bool OnDisco(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
|
||||
protected:
|
||||
virtual bool OnConnectFromClient(ThreadPoolLocalStorage *tls, const NetAddr &remoteClientAddress) = 0; // false = disconnect
|
||||
virtual bool OnReadFromClient(ThreadPoolLocalStorage *tls, u8 *data, u32 bytes) = 0; // false = disconnect
|
||||
virtual bool OnWriteToClient(ThreadPoolLocalStorage *tls, AsyncBuffer *buffer, u32 bytes) = 0; // true = delete AsyncBuffer
|
||||
virtual void OnDisconnectFromClient() = 0;
|
||||
};
|
||||
|
||||
|
||||
} // namespace cat
|
||||
|
||||
#endif // CAT_TCP_CONNEXION_HPP
|
||||
97
DependentExtensions/cat/net/win/TCPServer.hpp
Normal file
97
DependentExtensions/cat/net/win/TCPServer.hpp
Normal file
@ -0,0 +1,97 @@
|
||||
/*
|
||||
Copyright (c) 2009-2010 Christopher A. Taylor. All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
* Neither the name of LibCat nor the names of its contributors may be used
|
||||
to endorse or promote products derived from this software without
|
||||
specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef CAT_TCP_SERVER_HPP
|
||||
#define CAT_TCP_SERVER_HPP
|
||||
|
||||
/*
|
||||
Windows version of thread pool sockets with IO Completion Ports
|
||||
|
||||
Included from <cat/net/ThreadPoolSockets.hpp>
|
||||
Do not include directly
|
||||
*/
|
||||
|
||||
namespace cat {
|
||||
|
||||
|
||||
/*
|
||||
class TCPServer
|
||||
|
||||
Object that represents a TCP server bound to a single port
|
||||
|
||||
Overload InstantiateServerConnexion() to subclass connections with the server
|
||||
*/
|
||||
class TCPServer : public ThreadRefObject
|
||||
{
|
||||
friend class TCPConnexion;
|
||||
|
||||
Socket _socket;
|
||||
LPFN_ACCEPTEX _lpfnAcceptEx;
|
||||
LPFN_GETACCEPTEXSOCKADDRS _lpfnGetAcceptExSockAddrs;
|
||||
LPFN_DISCONNECTEX _lpfnDisconnectEx;
|
||||
Port _port;
|
||||
|
||||
struct AcceptTag
|
||||
{
|
||||
Socket acceptSocket;
|
||||
|
||||
// Space pre-allocated to receive addresses
|
||||
// NOTE: This is not necessarily how the addresses are organized in memory
|
||||
struct
|
||||
{
|
||||
// Not necessarily an IPv6 address
|
||||
sockaddr_in6 addr[2];
|
||||
u8 padding[2*16];
|
||||
} addresses;
|
||||
};
|
||||
|
||||
bool PostAccept(AsyncBuffer *buffer = 0);
|
||||
bool QueueAccepts();
|
||||
|
||||
public:
|
||||
TCPServer(int priorityLevel);
|
||||
virtual ~TCPServer();
|
||||
|
||||
bool ValidServer();
|
||||
Port GetPort();
|
||||
|
||||
bool Bind(bool onlySupportIPv4, Port port = 0);
|
||||
void Close();
|
||||
|
||||
protected:
|
||||
// Return true to release overlapped object memory, or return false to keep it
|
||||
virtual bool OnAccept(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
|
||||
protected:
|
||||
virtual TCPConnexion *InstantiateServerConnexion() = 0;
|
||||
};
|
||||
|
||||
|
||||
} // namespace cat
|
||||
|
||||
#endif // CAT_TCP_SERVER_HPP
|
||||
105
DependentExtensions/cat/net/win/UDPEndpoint.hpp
Normal file
105
DependentExtensions/cat/net/win/UDPEndpoint.hpp
Normal file
@ -0,0 +1,105 @@
|
||||
/*
|
||||
Copyright (c) 2009-2010 Christopher A. Taylor. All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
|
||||
* Redistributions of source code must retain the above copyright notice,
|
||||
this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright notice,
|
||||
this list of conditions and the following disclaimer in the documentation
|
||||
and/or other materials provided with the distribution.
|
||||
* Neither the name of LibCat nor the names of its contributors may be used
|
||||
to endorse or promote products derived from this software without
|
||||
specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
||||
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
||||
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
||||
ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
|
||||
LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
||||
CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
||||
SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
||||
INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
||||
CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
||||
ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||||
POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#ifndef CAT_UDP_ENDPOINT_HPP
|
||||
#define CAT_UDP_ENDPOINT_HPP
|
||||
|
||||
/*
|
||||
Windows version of thread pool sockets with IO Completion Ports
|
||||
|
||||
Included from <cat/net/ThreadPoolSockets.hpp>
|
||||
Do not include directly
|
||||
*/
|
||||
|
||||
namespace cat {
|
||||
|
||||
|
||||
/*
|
||||
class UDPEndpoint
|
||||
|
||||
Object that represents a UDP endpoint bound to a single port
|
||||
*/
|
||||
class UDPEndpoint : public ThreadRefObject
|
||||
{
|
||||
Socket _socket;
|
||||
Port _port;
|
||||
volatile u32 _closing;
|
||||
bool _ipv6;
|
||||
|
||||
struct RecvFromTag
|
||||
{
|
||||
// Not necessarily an IPv6 address, but we allocate enough space for one
|
||||
int addrLen;
|
||||
sockaddr_in6 addr;
|
||||
};
|
||||
|
||||
public:
|
||||
UDPEndpoint(int priorityLevel);
|
||||
virtual ~UDPEndpoint();
|
||||
|
||||
CAT_INLINE bool Valid() { return _socket != SOCKET_ERROR; }
|
||||
Port GetPort();
|
||||
|
||||
// Is6() result is only valid AFTER Bind()
|
||||
CAT_INLINE bool Is6() { return _ipv6; }
|
||||
CAT_INLINE bool IsClosed() { return _closing != 0; }
|
||||
|
||||
// For servers: Bind() with ignoreUnreachable = true ((default))
|
||||
// For clients: Bind() with ignoreUnreachable = false and call this
|
||||
// after the first packet from the server is received.
|
||||
bool IgnoreUnreachable();
|
||||
|
||||
// Disabled by default; useful for MTU discovery
|
||||
bool DontFragment(bool df = true);
|
||||
|
||||
void Close(); // Invalidates this object
|
||||
bool Bind(bool onlySupportIPv4, Port port = 0, bool ignoreUnreachable = true, int kernelReceiveBufferBytes = 0);
|
||||
bool QueueWSARecvFrom();
|
||||
|
||||
// If Is6() == true, the address must be promoted to IPv6
|
||||
// before calling PostWrite() with addr.PromoteTo6()
|
||||
// skip_bytes: Number of bytes to skip at the start of the post buffer
|
||||
bool Post(const NetAddr &addr, u8 *data, u32 data_bytes, u32 skip_bytes = 0);
|
||||
|
||||
private:
|
||||
bool Read(AsyncBuffer *buffer = 0);
|
||||
|
||||
bool OnReadComplete(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
bool OnWriteComplete(ThreadPoolLocalStorage *tls, int error, AsyncBuffer *buffer, u32 bytes);
|
||||
|
||||
protected:
|
||||
virtual void OnRead(ThreadPoolLocalStorage *tls, const NetAddr &addr, u8 *data, u32 bytes) = 0; // false = close
|
||||
virtual bool OnWrite(ThreadPoolLocalStorage *tls, AsyncBuffer *buffer, u32 bytes) { return true; } // false = do not delete AsyncBase object
|
||||
virtual void OnClose() = 0;
|
||||
virtual void OnUnreachable(const NetAddr &addr) {} // Only IP is valid
|
||||
};
|
||||
|
||||
|
||||
} // namespace cat
|
||||
|
||||
#endif // CAT_UDP_ENDPOINT_HPP
|
||||
Reference in New Issue
Block a user