This commit is contained in:
2025-11-24 14:19:51 +05:30
commit f5c1412b28
6734 changed files with 1527575 additions and 0 deletions

View 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

View 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

View 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

View 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

View 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