109 lines
3.4 KiB
C++
109 lines
3.4 KiB
C++
#ifndef STEAM_NETWORKING_MANAGER_H
|
|
#define STEAM_NETWORKING_MANAGER_H
|
|
|
|
#include <vector>
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <memory>
|
|
#include <steam_api.h>
|
|
#include <isteamnetworkingsockets.h>
|
|
#include <isteamnetworkingutils.h>
|
|
#include <steamnetworkingtypes.h>
|
|
#include "steam_message_handler.h"
|
|
|
|
// Forward declarations
|
|
class TCPClient;
|
|
class TCPServer;
|
|
class SteamNetworkingManager;
|
|
|
|
// User info structure
|
|
struct UserInfo {
|
|
CSteamID steamID;
|
|
std::string name;
|
|
int ping;
|
|
bool isRelay;
|
|
};
|
|
|
|
class SteamNetworkingManager {
|
|
public:
|
|
static SteamNetworkingManager* instance;
|
|
SteamNetworkingManager();
|
|
~SteamNetworkingManager();
|
|
|
|
bool initialize();
|
|
void shutdown();
|
|
|
|
// Joining
|
|
bool joinHost(uint64 hostID);
|
|
void disconnect();
|
|
|
|
// Getters
|
|
bool isHost() const { return g_isHost; }
|
|
bool isClient() const { return g_isClient; }
|
|
bool isConnected() const { return g_isConnected; }
|
|
const std::map<HSteamNetConnection, UserInfo>& getUserMap() const { return userMap; }
|
|
const std::vector<HSteamNetConnection>& getConnections() const { return connections; }
|
|
HSteamNetConnection getConnection() const { return g_hConnection; }
|
|
ISteamNetworkingSockets* getInterface() const { return m_pInterface; }
|
|
|
|
// For SteamRoomManager access
|
|
std::unique_ptr<TCPServer>*& getServer() { return server_; }
|
|
int*& getLocalPort() { return localPort_; }
|
|
boost::asio::io_context*& getIOContext() { return io_context_; }
|
|
std::map<HSteamNetConnection, std::shared_ptr<TCPClient>>*& getClientMap() { return clientMap_; }
|
|
std::mutex*& getClientMutex() { return clientMutex_; }
|
|
HSteamListenSocket& getListenSock() { return hListenSock; }
|
|
ISteamNetworkingSockets* getInterface() { return m_pInterface; }
|
|
bool& getIsHost() { return g_isHost; }
|
|
|
|
void setMessageHandlerDependencies(boost::asio::io_context& io_context, std::map<HSteamNetConnection, std::shared_ptr<TCPClient>>& clientMap, std::mutex& clientMutex, std::unique_ptr<TCPServer>& server, int& localPort);
|
|
|
|
// Message handler
|
|
void startMessageHandler();
|
|
void stopMessageHandler();
|
|
|
|
// Update user info (ping, relay status)
|
|
void update();
|
|
|
|
// For callbacks
|
|
void setHostSteamID(CSteamID id) { g_hostSteamID = id; }
|
|
CSteamID getHostSteamID() const { return g_hostSteamID; }
|
|
|
|
private:
|
|
// Steam API
|
|
ISteamNetworkingSockets* m_pInterface;
|
|
|
|
// Hosting
|
|
HSteamListenSocket hListenSock;
|
|
bool g_isHost;
|
|
bool g_isClient;
|
|
bool g_isConnected;
|
|
HSteamNetConnection g_hConnection;
|
|
CSteamID g_hostSteamID;
|
|
|
|
// Connections
|
|
std::vector<HSteamNetConnection> connections;
|
|
std::map<HSteamNetConnection, UserInfo> userMap;
|
|
std::mutex connectionsMutex;
|
|
|
|
// Connection config
|
|
int g_retryCount;
|
|
const int MAX_RETRIES = 3;
|
|
int g_currentVirtualPort;
|
|
|
|
// Message handler dependencies
|
|
boost::asio::io_context* io_context_;
|
|
std::map<HSteamNetConnection, std::shared_ptr<TCPClient>>* clientMap_;
|
|
std::mutex* clientMutex_;
|
|
std::unique_ptr<TCPServer>* server_;
|
|
int* localPort_;
|
|
SteamMessageHandler* messageHandler_;
|
|
|
|
// Callback
|
|
static void OnSteamNetConnectionStatusChanged(SteamNetConnectionStatusChangedCallback_t *pInfo);
|
|
void handleConnectionStatusChanged(SteamNetConnectionStatusChangedCallback_t *pInfo);
|
|
|
|
friend class SteamRoomManager;
|
|
};
|
|
|
|
#endif // STEAM_NETWORKING_MANAGER_H
|