重构Steam网络管理,整合房间管理功能,添加多路复用支持,优化TCP服务器和消息处理

This commit is contained in:
Ayndpa
2025-11-19 16:49:36 +08:00
parent 0e65ecb8ca
commit f661b1d369
12 changed files with 217 additions and 192 deletions

View File

@@ -0,0 +1,80 @@
#include "multiplex_manager.h"
#include <iostream>
#include <cstring>
MultiplexManager::MultiplexManager(ISteamNetworkingSockets* steamInterface, HSteamNetConnection steamConn)
: steamInterface_(steamInterface), steamConn_(steamConn), nextId_(1) {}
MultiplexManager::~MultiplexManager() {
// Close all sockets
std::lock_guard<std::mutex> lock(mapMutex_);
for (auto& pair : clientMap_) {
pair.second->close();
}
clientMap_.clear();
}
uint32_t MultiplexManager::addClient(std::shared_ptr<tcp::socket> socket) {
std::lock_guard<std::mutex> lock(mapMutex_);
uint32_t id = nextId_++;
clientMap_[id] = socket;
return id;
}
void MultiplexManager::removeClient(uint32_t id) {
std::lock_guard<std::mutex> lock(mapMutex_);
auto it = clientMap_.find(id);
if (it != clientMap_.end()) {
it->second->close();
clientMap_.erase(it);
}
}
std::shared_ptr<tcp::socket> MultiplexManager::getClient(uint32_t id) {
std::lock_guard<std::mutex> lock(mapMutex_);
auto it = clientMap_.find(id);
if (it != clientMap_.end()) {
return it->second;
}
return nullptr;
}
void MultiplexManager::sendTunnelPacket(uint32_t id, const char* data, size_t len, int type) {
// Packet format: uint32_t id, uint32_t type, then data if type==0
size_t packetSize = sizeof(uint32_t) * 2 + (type == 0 ? len : 0);
std::vector<char> packet(packetSize);
uint32_t* pId = reinterpret_cast<uint32_t*>(&packet[0]);
uint32_t* pType = reinterpret_cast<uint32_t*>(&packet[sizeof(uint32_t)]);
*pId = id;
*pType = type;
if (type == 0 && data) {
std::memcpy(&packet[sizeof(uint32_t) * 2], data, len);
}
steamInterface_->SendMessageToConnection(steamConn_, packet.data(), packet.size(), k_nSteamNetworkingSend_Reliable, nullptr);
}
void MultiplexManager::handleTunnelPacket(const char* data, size_t len) {
if (len < sizeof(uint32_t) * 2) {
std::cerr << "Invalid tunnel packet size" << std::endl;
return;
}
uint32_t id = *reinterpret_cast<const uint32_t*>(data);
uint32_t type = *reinterpret_cast<const uint32_t*>(data + sizeof(uint32_t));
if (type == 0) {
// Data packet
size_t dataLen = len - sizeof(uint32_t) * 2;
const char* packetData = data + sizeof(uint32_t) * 2;
auto socket = getClient(id);
if (socket) {
boost::asio::async_write(*socket, boost::asio::buffer(packetData, dataLen), [](const boost::system::error_code&, std::size_t) {});
} else {
std::cerr << "No client found for id " << id << std::endl;
}
} else if (type == 1) {
// Disconnect packet
removeClient(id);
std::cout << "Client " << id << " disconnected" << std::endl;
} else {
std::cerr << "Unknown packet type " << type << std::endl;
}
}

View File

@@ -0,0 +1,33 @@
#pragma once
#include <unordered_map>
#include <memory>
#include <mutex>
#include <vector>
#include <boost/asio.hpp>
#include <steam_api.h>
#include <isteamnetworkingsockets.h>
#include <steamnetworkingtypes.h>
using boost::asio::ip::tcp;
class MultiplexManager {
public:
MultiplexManager(ISteamNetworkingSockets* steamInterface, HSteamNetConnection steamConn);
~MultiplexManager();
uint32_t addClient(std::shared_ptr<tcp::socket> socket);
void removeClient(uint32_t id);
std::shared_ptr<tcp::socket> getClient(uint32_t id);
void sendTunnelPacket(uint32_t id, const char* data, size_t len, int type);
void handleTunnelPacket(const char* data, size_t len);
private:
ISteamNetworkingSockets* steamInterface_;
HSteamNetConnection steamConn_;
std::unordered_map<uint32_t, std::shared_ptr<tcp::socket>> clientMap_;
std::mutex mapMutex_;
uint32_t nextId_;
};