46 lines
1.2 KiB
C++
46 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <boost/asio.hpp>
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <isteamnetworkingsockets.h>
|
|
#include <isteamnetworkingutils.h>
|
|
#include <steamnetworkingtypes.h>
|
|
|
|
using boost::asio::ip::tcp;
|
|
|
|
// Extern declarations for global variables used in TCPServer
|
|
extern HSteamNetConnection g_hConnection;
|
|
extern bool g_isConnected;
|
|
extern ISteamNetworkingSockets* m_pInterface;
|
|
extern bool forwarding;
|
|
|
|
// TCP Server class
|
|
class TCPServer {
|
|
public:
|
|
TCPServer(int port);
|
|
~TCPServer();
|
|
|
|
bool start();
|
|
void stop();
|
|
void sendToAll(const std::string& message, std::shared_ptr<tcp::socket> excludeSocket = nullptr);
|
|
void sendToAll(const char* data, size_t size, std::shared_ptr<tcp::socket> excludeSocket = nullptr);
|
|
int getClientCount();
|
|
|
|
private:
|
|
void start_accept();
|
|
void start_read(std::shared_ptr<tcp::socket> socket);
|
|
|
|
int port_;
|
|
bool running_;
|
|
boost::asio::io_context io_context_;
|
|
boost::asio::executor_work_guard<boost::asio::io_context::executor_type> work_;
|
|
tcp::acceptor acceptor_;
|
|
std::vector<std::shared_ptr<tcp::socket>> clients_;
|
|
std::mutex clientsMutex_;
|
|
std::thread serverThread_;
|
|
bool hasAcceptedConnection_;
|
|
}; |