a9549c33e3
TcpClient: added close_connection to shutdown the connection on both send and receive TcpServer: changed clients list initialization from constructors to using the reserve function
79 lines
1.8 KiB
C++
79 lines
1.8 KiB
C++
#ifndef TCP_CLIENT_HPP
|
|
#define TCP_CLIENT_HPP
|
|
|
|
#include "Defs.hpp"
|
|
#include "NetworkMessage.hpp"
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <functional>
|
|
#include <future>
|
|
|
|
#ifdef _MSC_VER
|
|
#pragma once
|
|
#endif
|
|
|
|
struct TcpClient
|
|
{
|
|
TcpClient(const SOCKET &socket);
|
|
TcpClient(const std::string &ip);
|
|
TcpClient(const std::string &ip, uint16 port = default_client_port);
|
|
~TcpClient();
|
|
|
|
void Shutdown();
|
|
|
|
const std::string &GetIP();
|
|
void SetIP(const std::string &ip);
|
|
|
|
uint16 GetPort();
|
|
void SetPort(uint16 port);
|
|
|
|
uint16 GetID();
|
|
void SetID(uint16 id);
|
|
|
|
bool Connect();
|
|
|
|
bool DataAvailable(int32 &size);
|
|
|
|
//this method will receive the messages automaticaly and use the callback methods
|
|
void ReceiveMessages();
|
|
|
|
//this is a more manual method with no callbacks
|
|
const NetworkMessage &ReceiveMessage();
|
|
|
|
std::future<bool> SendMessage(const NetworkMessage &message);
|
|
bool SendBytes(const std::vector<byte> &bytes);
|
|
bool SendBytes(byte *bytes, uint32 lenght);
|
|
|
|
void SetOnDisconnectCallback(void (*func)(uint16));
|
|
void SetOnConnectCallback(void (*func)(uint16));
|
|
void SetOnMessageCallback(void (*func)(uint16, byte, byte, void*));
|
|
|
|
static const TcpClient &DefaultTcpClient();
|
|
|
|
private:
|
|
TcpClient();
|
|
|
|
const NetworkBuffer &receive_data_array();
|
|
static void receive_data(TcpClient *client);
|
|
static bool send_network_message(const NetworkMessage &message, TcpClient *client);
|
|
static void close_connection(TcpClient *client);
|
|
|
|
bool initialize(const std::string &ip, uint16 port = default_client_port);
|
|
|
|
uint16 id = -2;
|
|
std::string ip;
|
|
uint16 port = 0;
|
|
bool initialized = false;
|
|
bool receive = false;
|
|
|
|
std::function<void(uint16)> OnDisconnect;
|
|
std::function<void(uint16)> OnConnect;
|
|
std::function<void(uint16, byte, byte, void*)> OnMessage;
|
|
|
|
SOCKET tcp_socket = INVALID_SOCKET;
|
|
struct addrinfo *result = nullptr;
|
|
struct addrinfo hints;
|
|
};
|
|
|
|
#endif |