0ac8b4f6e8
Fixed NetworkBuffer include Fixed some data types TcpClient: Added SendBytes, SetOnDisconnectCallback, SetOnConnectCallback, SetOnMessageCallback methods and made callback std::function's private and fixed some method implementations Added AcceptConnection on TcpServer Fixed TcpServer AddToClientsList
66 lines
1.3 KiB
C++
66 lines
1.3 KiB
C++
#ifndef TCP_SERVER_HPP
|
|
#define TCP_SERVER_HPP
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma once
|
|
#endif
|
|
|
|
#include "Defs.hpp"
|
|
#include "TcpClient.hpp"
|
|
|
|
#include <functional>
|
|
|
|
#ifdef _MSC_VER
|
|
#undef SendMessage
|
|
#endif
|
|
|
|
class TcpServer
|
|
{
|
|
public:
|
|
TcpServer();
|
|
TcpServer(uint16 port = default_server_port);
|
|
~TcpServer();
|
|
|
|
void Shutdown();
|
|
|
|
uint16 AllocateID();
|
|
void AddToClientsList(TcpClient &client);
|
|
bool StartServer(bool accept_connections);
|
|
void AcceptConnections();
|
|
void SendMessage(const NetworkMessage &message);
|
|
|
|
void RejectConnection(TcpClient &client);
|
|
void AcceptConnection(TcpClient &client);
|
|
|
|
std::function<void(const NetworkMessage &message)> OnMessage;
|
|
|
|
private:
|
|
static void process_message(TcpServer *server, const NetworkMessage &message);
|
|
static void process_client_received_data(TcpServer *server);
|
|
static void accept_connections(TcpServer *server);
|
|
bool initialize(uint16 port = default_server_port);
|
|
|
|
bool initialized = false;
|
|
bool running = false;
|
|
|
|
uint16 max_connections = 0;
|
|
|
|
std::vector<TcpClient> clients;
|
|
|
|
#ifdef _MSC_VER
|
|
SOCKET server_tcp_socket = INVALID_SOCKET;
|
|
struct addrinfo *result = nullptr;
|
|
struct addrinfo hints;
|
|
#endif
|
|
};
|
|
|
|
#ifdef _MSC_VER
|
|
#ifdef UNICODE
|
|
#define SendMessage SendMessageW
|
|
#else
|
|
#define SendMessage SendMessageA
|
|
#endif // !UNICODE
|
|
#endif
|
|
|
|
#endif
|