261 lines
5.7 KiB
C++
261 lines
5.7 KiB
C++
|
#include <iostream>
|
||
|
#include <ctime>
|
||
|
#include <iostream>
|
||
|
#include <string>
|
||
|
#include <set>
|
||
|
#include <boost/array.hpp>
|
||
|
#include <boost/bind.hpp>
|
||
|
#include <boost/shared_ptr.hpp>
|
||
|
#include <boost/asio.hpp>
|
||
|
|
||
|
namespace net
|
||
|
{
|
||
|
#pragma pack(push, 1)
|
||
|
struct ReceivedDatagram
|
||
|
{
|
||
|
uint32_t number;
|
||
|
uint32_t localTime;
|
||
|
float posX;
|
||
|
float posY;
|
||
|
float posZ;
|
||
|
float angle;
|
||
|
uint8_t pressedButtons;
|
||
|
|
||
|
};
|
||
|
|
||
|
struct PlayerDatagramToSend
|
||
|
{
|
||
|
uint16_t type;
|
||
|
uint16_t state;
|
||
|
float posX;
|
||
|
float posY;
|
||
|
float posZ;
|
||
|
float angle;
|
||
|
float velX;
|
||
|
float velY;
|
||
|
float velZ;
|
||
|
};
|
||
|
|
||
|
struct ServerDatagramToSend
|
||
|
{
|
||
|
uint32_t number;
|
||
|
uint32_t localTime;
|
||
|
uint16_t yourType;
|
||
|
uint16_t numberOfPlayers;
|
||
|
|
||
|
std::array<PlayerDatagramToSend, 4> players;
|
||
|
|
||
|
|
||
|
};
|
||
|
#pragma pack(pop)
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
|
||
|
struct ServerPlayerStruct
|
||
|
{
|
||
|
uint32_t lastReceivedDatagramNumber = 0;
|
||
|
uint32_t lastReceivedLocalTime = 0;
|
||
|
|
||
|
net::PlayerDatagramToSend clientPlayer;
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
using boost::asio::ip::udp;
|
||
|
|
||
|
std::map<boost::asio::ip::address, ServerPlayerStruct> clientMap;
|
||
|
|
||
|
std::mutex clientMapMutex;
|
||
|
|
||
|
volatile bool quit = false;
|
||
|
|
||
|
uint16_t currentTypeIndex = 1;
|
||
|
|
||
|
uint32_t currentNumber = 1;
|
||
|
|
||
|
ServerPlayerStruct CreateNewPlayer()
|
||
|
{
|
||
|
ServerPlayerStruct result;
|
||
|
|
||
|
result.clientPlayer.type = 0;
|
||
|
result.clientPlayer.state = 0;
|
||
|
result.clientPlayer.posX = 0;
|
||
|
result.clientPlayer.posY = 0;
|
||
|
result.clientPlayer.posZ = 0;
|
||
|
result.clientPlayer.angle = 0;
|
||
|
result.clientPlayer.velX = 0;
|
||
|
result.clientPlayer.velY = 0;
|
||
|
result.clientPlayer.velZ = 0;
|
||
|
|
||
|
result.clientPlayer.type = currentTypeIndex++;
|
||
|
|
||
|
return result;
|
||
|
}
|
||
|
|
||
|
void HandleReceivedData(udp::endpoint& sender, const std::array<char, 255>& datagram)
|
||
|
{
|
||
|
std::lock_guard<std::mutex> lock(clientMapMutex);
|
||
|
|
||
|
net::ReceivedDatagram receivedDatagram;
|
||
|
std::copy(&datagram[0], &datagram[0] + sizeof(net::ReceivedDatagram), reinterpret_cast<char*>(&receivedDatagram));
|
||
|
|
||
|
|
||
|
if (clientMap.count(sender.address()) == 0)
|
||
|
{
|
||
|
clientMap[sender.address()] = CreateNewPlayer();
|
||
|
}
|
||
|
|
||
|
ServerPlayerStruct& currentPlayer = clientMap[sender.address()];
|
||
|
|
||
|
|
||
|
if (receivedDatagram.number <= currentPlayer.lastReceivedDatagramNumber)
|
||
|
{
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
currentPlayer.lastReceivedDatagramNumber = receivedDatagram.number;
|
||
|
currentPlayer.lastReceivedLocalTime = receivedDatagram.localTime;
|
||
|
|
||
|
currentPlayer.clientPlayer.posX = receivedDatagram.posX;
|
||
|
currentPlayer.clientPlayer.posY = receivedDatagram.posY;
|
||
|
currentPlayer.clientPlayer.posZ = receivedDatagram.posZ;
|
||
|
|
||
|
currentPlayer.clientPlayer.angle = currentPlayer.clientPlayer.angle;
|
||
|
|
||
|
|
||
|
}
|
||
|
|
||
|
void UdpSendTimerHandler(const boost::system::error_code& /*e*/, boost::asio::steady_timer& t, boost::asio::io_context& udpSenderIoContext)
|
||
|
{
|
||
|
std::lock_guard<std::mutex> lock(clientMapMutex);
|
||
|
|
||
|
std::array<char, 256> arrayBuffer;
|
||
|
|
||
|
arrayBuffer.fill(0);
|
||
|
|
||
|
net::ServerDatagramToSend data;
|
||
|
|
||
|
data.localTime = 0;
|
||
|
data.yourType = 0;
|
||
|
data.number = currentNumber++;
|
||
|
data.numberOfPlayers = clientMap.size();
|
||
|
|
||
|
uint32_t count = clientMap.size() > 4 ? 4 : clientMap.size();
|
||
|
|
||
|
auto clientMapIterator = clientMap.begin();
|
||
|
|
||
|
for (size_t i = 0; i < count; i++)
|
||
|
{
|
||
|
data.players[i] = clientMapIterator->second.clientPlayer;
|
||
|
|
||
|
clientMapIterator++;
|
||
|
}
|
||
|
|
||
|
|
||
|
std::cout << "Timer run!!!\n";
|
||
|
|
||
|
for (auto& c : clientMap)
|
||
|
{
|
||
|
data.yourType = c.second.clientPlayer.type;
|
||
|
|
||
|
//std::copy(&arrayBuffer[0], &arrayBuffer[0] + sizeof(net::ServerDatagramToSend), reinterpret_cast<char*>(&data));
|
||
|
std::copy(reinterpret_cast<char*>(&data), reinterpret_cast<char*>(&data) + sizeof(net::ServerDatagramToSend), &arrayBuffer[0]);
|
||
|
|
||
|
udp::socket socket(udpSenderIoContext);
|
||
|
|
||
|
udp::endpoint remote_endpoint = udp::endpoint(boost::asio::ip::make_address("127.0.0.1"), 14);
|
||
|
|
||
|
socket.open(udp::v4());
|
||
|
|
||
|
boost::system::error_code err;
|
||
|
|
||
|
auto sent = socket.send_to(boost::asio::buffer(arrayBuffer), remote_endpoint, 0, err);
|
||
|
|
||
|
socket.close();
|
||
|
|
||
|
std::cout << "Sent Payload --- " << sent << "\n";
|
||
|
}
|
||
|
|
||
|
|
||
|
if (!quit)
|
||
|
{
|
||
|
t.expires_at(t.expiry() + boost::asio::chrono::milliseconds(50));
|
||
|
t.async_wait(boost::bind(&UdpSendTimerHandler, boost::asio::placeholders::error, boost::ref(t), boost::ref(udpSenderIoContext)));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
class UdpReceiver
|
||
|
{
|
||
|
public:
|
||
|
UdpReceiver(boost::asio::io_context& io_context)
|
||
|
: socket_(io_context, udp::endpoint(udp::v4(), 13))
|
||
|
{
|
||
|
start_receive();
|
||
|
}
|
||
|
|
||
|
|
||
|
private:
|
||
|
void start_receive()
|
||
|
{
|
||
|
|
||
|
socket_.async_receive_from(
|
||
|
boost::asio::buffer(recv_buffer_), remote_endpoint_,
|
||
|
boost::bind(&UdpReceiver::handle_receive, this,
|
||
|
boost::asio::placeholders::error,
|
||
|
boost::asio::placeholders::bytes_transferred));
|
||
|
}
|
||
|
|
||
|
void handle_receive(const boost::system::error_code& error,
|
||
|
std::size_t /*bytes_transferred*/)
|
||
|
{
|
||
|
if (!error)
|
||
|
{
|
||
|
|
||
|
HandleReceivedData(remote_endpoint_, recv_buffer_);
|
||
|
|
||
|
start_receive();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void handle_send(boost::shared_ptr<std::string> /*message*/,
|
||
|
const boost::system::error_code& /*error*/,
|
||
|
std::size_t /*bytes_transferred*/)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
udp::socket socket_;
|
||
|
|
||
|
udp::endpoint remote_endpoint_;
|
||
|
std::array<char, 255> recv_buffer_;
|
||
|
|
||
|
};
|
||
|
|
||
|
int main()
|
||
|
{
|
||
|
try
|
||
|
{
|
||
|
boost::asio::io_context udpSenderIoContext;
|
||
|
|
||
|
boost::asio::steady_timer sendTimer(udpSenderIoContext, boost::asio::chrono::milliseconds(50));
|
||
|
|
||
|
sendTimer.async_wait(boost::bind(&UdpSendTimerHandler, boost::asio::placeholders::error, boost::ref(sendTimer), boost::ref(udpSenderIoContext)));
|
||
|
|
||
|
std::thread timerThread(boost::bind(&boost::asio::io_context::run, &udpSenderIoContext));
|
||
|
|
||
|
boost::asio::io_context udpReceiverIoContext;
|
||
|
|
||
|
UdpReceiver server(udpReceiverIoContext);
|
||
|
|
||
|
udpReceiverIoContext.run();
|
||
|
}
|
||
|
catch (std::exception& e)
|
||
|
{
|
||
|
std::cerr << e.what() << std::endl;
|
||
|
}
|
||
|
|
||
|
return 0;
|
||
|
}
|