Added class Package to help refactor the code. The code got easier to read.

This commit is contained in:
stiffly
2015-12-10 17:07:18 +01:00
parent d22e59f28f
commit 58952725e9
7 changed files with 138 additions and 129 deletions
+1
View File
@@ -30,6 +30,7 @@ private:
void SendToServer(); void SendToServer();
int Receive(char* data, size_t length); int Receive(char* data, size_t length);
void Send(Package& message);
int CreateMessage(MessageType type, std::string message, char* data); int CreateMessage(MessageType type, std::string message, char* data);
void Connect(); void Connect();
void Disconnect(); void Disconnect();
@@ -3,6 +3,8 @@
#include <boost/asio/ip/udp.hpp> #include <boost/asio/ip/udp.hpp>
#include <queue> #include <queue>
#include "Network/Package.h"
#define BOARDSIZE 16 #define BOARDSIZE 16
#define MAXCONNECTIONS 8 #define MAXCONNECTIONS 8
+32
View File
@@ -0,0 +1,32 @@
#ifndef Package_h__
#define Package_h__
#include <string>
#include "Network/MessageType.h"
// Defines the
class Package
{
public:
// arg1: Type of message (Connect, Disconnect...)
// arg2: PackageID for identifying packet loss.
Package(MessageType type);
~Package();
// Add primitive types like int, float, char...
template<typename T>
void AddPrimitive(T val)
{
memcpy(m_Data + m_Offset, &val, sizeof(T));
m_Offset += sizeof(T);
}
void AddString(std::string str);
int Size() { return m_Offset; };
char* Data() { return m_Data; };
private:
char* m_Data = new char[128];
int m_Offset = 0;
};
#endif
+3 -1
View File
@@ -47,10 +47,12 @@ private:
int Receive(char* data, size_t length); int Receive(char* data, size_t length);
void Send(Package& package, int playerID);
void Send(Package& package);
int CreateMessage(MessageType type, std::string message, char * data); int CreateMessage(MessageType type, std::string message, char * data);
void MoveMessageHead(char*& data, size_t& length, size_t stepSize); void MoveMessageHead(char*& data, size_t& length, size_t stepSize);
void Broadcast(std::string message); void Broadcast(std::string message);
void Broadcast(char* data, size_t length); void Broadcast(Package& package);
void SendSnapshot(); void SendSnapshot();
void SendPing(); void SendPing();
void CheckForTimeOuts(); void CheckForTimeOuts();
+25 -64
View File
@@ -75,22 +75,14 @@ void Client::ReadFromServer()
void Client::SendToServer() void Client::SendToServer()
{ {
if (m_NextSnapshot.inputForward != "" && m_NextSnapshot.inputForward[0] != '\0') { if (m_NextSnapshot.inputForward != "" && m_NextSnapshot.inputForward[0] != '\0') {
char* dataPackage = new char[INPUTSIZE]; // The package that will be sent to the server, when filled Package message(MessageType::Event);
int len = CreateMessage(MessageType::Event, m_NextSnapshot.inputForward, dataPackage); message.AddString(m_NextSnapshot.inputForward);
m_Socket.send_to(boost::asio::buffer( Send(message);
dataPackage,
len),
m_ReceiverEndpoint, 0);
delete[] dataPackage;
} }
if (m_NextSnapshot.inputRight != "" && m_NextSnapshot.inputRight[0] != '\0') { if (m_NextSnapshot.inputRight != "" && m_NextSnapshot.inputRight[0] != '\0') {
char* dataPackage = new char[INPUTSIZE]; // The package that will be sent to the server, when filled Package message(MessageType::Event);
int len = CreateMessage(MessageType::Event, m_NextSnapshot.inputRight, dataPackage); message.AddString(m_NextSnapshot.inputRight);
m_Socket.send_to(boost::asio::buffer( Send(message);
dataPackage,
len),
m_ReceiverEndpoint, 0);
delete[] dataPackage;
} }
} }
@@ -146,16 +138,9 @@ void Client::ParsePing()
void Client::ParseServerPing() void Client::ParseServerPing()
{ {
char* testMessage = new char[128]; Package message(MessageType::ServerPing);
int testOffset = CreateMessage(MessageType::ServerPing, "Ping recieved", testMessage); message.AddString("Ping recieved");
Send(message);
//std::cout << "Parsing ping." << std::endl;
m_Socket.send_to(boost::asio::buffer(
testMessage,
testOffset),
m_ReceiverEndpoint, 0);
delete[] testMessage;
} }
void Client::ParseEventMessage(char* data, size_t length) void Client::ParseEventMessage(char* data, size_t length)
@@ -223,59 +208,35 @@ int Client::Receive(char* data, size_t length)
return bytesReceived; return bytesReceived;
} }
int Client::CreateMessage(MessageType type, std::string message, char* data) void Client::Send(Package& package)
{ {
int lengthOfMessage = 0; m_Socket.send_to(boost::asio::buffer(
int messageType = static_cast<int>(type); package.Data(),
lengthOfMessage = message.size(); package.Size()),
m_ReceiverEndpoint, 0);
int offset = 0;
// Message type
memcpy(data + offset, &messageType, sizeof(int));
offset += sizeof(int);
// Message, add one extra byte for null terminator
memcpy(data + offset, message.data(), (lengthOfMessage + 1) * sizeof(char));
offset += (lengthOfMessage + 1) * sizeof(char);
return offset;
} }
void Client::Connect() void Client::Connect()
{ {
char* dataPackage = new char[INPUTSIZE]; // The package that will be sent to the server, when filled Package message(MessageType::Connect);
int length = CreateMessage(MessageType::Connect, m_PlayerName, dataPackage); message.AddString(m_PlayerName);
m_StartPingTime = std::clock(); m_StartPingTime = std::clock();
m_Socket.send_to(boost::asio::buffer( Send(message);
dataPackage,
length),
m_ReceiverEndpoint, 0);
delete[] dataPackage;
} }
void Client::Disconnect() void Client::Disconnect()
{ {
char* dataPackage = new char[INPUTSIZE]; // The package that will be sent to the server, when filled Package message(MessageType::Connect);
int len = CreateMessage(MessageType::Disconnect, "+Disconnect", dataPackage); message.AddString("+Disconnect");
m_Socket.send_to(boost::asio::buffer( Send(message);
dataPackage,
len),
m_ReceiverEndpoint, 0);
delete[] dataPackage;
} }
void Client::Ping() void Client::Ping()
{ {
char* dataPackage = new char[INPUTSIZE]; Package message(MessageType::Connect);
if (GetAsyncKeyState('P')) { // Maybe use previous key here message.AddString("Ping");
int length = CreateMessage(MessageType::ClientPing, "Ping", dataPackage); m_StartPingTime = std::clock();
m_StartPingTime = std::clock(); Send(message);
m_Socket.send_to(boost::asio::buffer(
dataPackage,
length),
m_ReceiverEndpoint, 0);
}
memset(dataPackage, 0, INPUTSIZE);
delete[] dataPackage;
} }
void Client::MoveMessageHead(char*& data, size_t& length, size_t stepSize) void Client::MoveMessageHead(char*& data, size_t& length, size_t stepSize)
+24
View File
@@ -0,0 +1,24 @@
#include "Network/Package.h"
Package::Package(MessageType type)
{
// Create message header
// Add message type
int messageType = static_cast<int>(type);
memcpy(m_Data, &messageType, sizeof(int));
m_Offset += sizeof(int);
}
Package::~Package()
{
delete[] m_Data;
}
void Package::AddString(std::string str)
{
// Message, add one extra byte for null terminator
memcpy(m_Data + m_Offset, str.data(), (str.size() + 1) * sizeof(char));
m_Offset += (str.size() + 1) * sizeof(char);
}
+51 -64
View File
@@ -154,6 +154,24 @@ int Server::Receive(char * data, size_t length)
return length; return length;
} }
void Server::Send(Package& message, int playerID)
{
m_Socket.send_to(
boost::asio::buffer(message.Data(), message.Size()),
m_PlayerDefinitions[playerID].Endpoint,
0);
}
void Server::Send(Package & package)
{
m_Socket.send_to(
boost::asio::buffer(
package.Data(),
package.Size()),
m_ReceiverEndpoint,
0);
}
int Server::CreateMessage(MessageType type, std::string message, char * data) int Server::CreateMessage(MessageType type, std::string message, char * data)
{ {
int lengthOfMessage = 0; int lengthOfMessage = 0;
@@ -183,55 +201,42 @@ void Server::MoveMessageHead(char *& data, size_t & length, size_t stepSize)
void Server::Broadcast(std::string message) void Server::Broadcast(std::string message)
{ {
std::cout << m_PacketID << ": Broadcast: " << message << std::endl; Package package(MessageType::Event);
char* data = new char[128]; package.AddPrimitive<int>(12); // Input PackageID here
int offset = CreateMessage(MessageType::Event, message, data); package.AddString(message);
for (int i = 0; i < MAXCONNECTIONS; i++) { for (int i = 0; i < MAXCONNECTIONS; i++) {
if (m_PlayerDefinitions[i].Endpoint.address() != boost::asio::ip::address()) { if (m_PlayerDefinitions[i].Endpoint.address() != boost::asio::ip::address()) {
m_Socket.send_to( Send(package, i);
boost::asio::buffer(data, offset),
m_PlayerDefinitions[i].Endpoint,
0);
} }
} }
delete[] data;
} }
void Server::Broadcast(char * data, size_t length) void Server::Broadcast(Package& package)
{ {
for (int i = 0; i < MAXCONNECTIONS; ++i) { for (int i = 0; i < MAXCONNECTIONS; ++i) {
if (m_PlayerDefinitions[i].Endpoint.address() != boost::asio::ip::address()) { if (m_PlayerDefinitions[i].Endpoint.address() != boost::asio::ip::address()) {
m_Socket.send_to( Send(package, i);
boost::asio::buffer(data, length),
m_PlayerDefinitions[i].Endpoint,
0);
} }
} }
} }
void Server::SendSnapshot() void Server::SendSnapshot()
{ {
char* data = new char[INPUTSIZE]; Package package(MessageType::Snapshot);
int offset = CreateHeader(MessageType::Snapshot, data); package.AddPrimitive<int>(12); // Input PackageID here
for (size_t i = 0; i < MAXCONNECTIONS; i++) { for (size_t i = 0; i < MAXCONNECTIONS; i++) {
if (m_PlayerDefinitions[i].EntityID == -1) { if (m_PlayerDefinitions[i].EntityID == -1) {
continue; continue;
} }
// Pack player pos into data package // Pack player pos into data package
glm::vec3 playerPos = m_World->GetComponent(m_PlayerDefinitions[i].EntityID, "Transform")["Position"]; glm::vec3 playerPos = m_World->GetComponent(m_PlayerDefinitions[i].EntityID, "Transform")["Position"];
memcpy(data + offset, &playerPos.x, sizeof(float)); package.AddPrimitive<float>(playerPos.x);
offset += sizeof(float); package.AddPrimitive<float>(playerPos.y);
memcpy(data + offset, &playerPos.y, sizeof(float)); package.AddPrimitive<float>(playerPos.z);
offset += sizeof(float);
memcpy(data + offset, &playerPos.z, sizeof(float)); package.AddString(m_PlayerDefinitions[i].Name);
offset += sizeof(float);
// +1 for null terminator
// Pack player name into data package
memcpy(data + offset, m_PlayerDefinitions[i].Name.data(), m_PlayerDefinitions[i].Name.size() + 1);
offset += (m_PlayerDefinitions[i].Name.size() + 1) * sizeof(char);
} }
Broadcast(data, offset); Broadcast(package);
delete[] data;
} }
void Server::SendPing() void Server::SendPing()
@@ -244,14 +249,13 @@ void Server::SendPing()
} }
// Create ping message // Create ping message
char* data = new char[128]; Package package(MessageType::ServerPing);
int len = CreateMessage(MessageType::ServerPing, "Ping from server", data); package.AddPrimitive<int>(12); // Input PackageID here
// Time message package.AddString("Ping from server");
// Time message
m_StartPingTime = std::clock(); m_StartPingTime = std::clock();
// Send message // Send message
Broadcast(data, len); Broadcast(package);
delete[] data;
} }
void Server::CheckForTimeOuts() void Server::CheckForTimeOuts()
@@ -358,35 +362,23 @@ void Server::ParseConnect(char * data, size_t length)
m_PlayerDefinitions[i].Endpoint = m_ReceiverEndpoint; m_PlayerDefinitions[i].Endpoint = m_ReceiverEndpoint;
m_PlayerDefinitions[i].Name = std::string(data); m_PlayerDefinitions[i].Name = std::string(data);
// +1 is the null terminator
MoveMessageHead(data, length, m_PlayerDefinitions[i].Name.size() + 1);
m_StopTimes[i] = std::clock(); m_StopTimes[i] = std::clock();
std::cout << m_PacketID << ": Player \"" << m_PlayerDefinitions[i].Name << "\" connected on IP: " << std::cout << m_PacketID << ": Player \"" << m_PlayerDefinitions[i].Name << "\" connected on IP: " <<
m_PlayerDefinitions[i].Endpoint.address().to_string() << std::endl; m_PlayerDefinitions[i].Endpoint.address().to_string() << std::endl;
int offset = 0; Package package(MessageType::Connect);
char* temp = new char[sizeof(int) * 2]; package.AddPrimitive<int>(12); // Input PackageID here
int messagType = 0; package.AddPrimitive<int>(i); // Player ID
memcpy(temp, &messagType, sizeof(int)); Send(package, i);
offset += sizeof(int);
memcpy(temp + offset, &i, sizeof(int));
memcpy(temp, &m_PacketID, sizeof(int));
offset += sizeof(int);
m_PacketCounter++;
m_Socket.send_to(
boost::asio::buffer(temp, sizeof(int) * 2),
m_PlayerDefinitions[i].Endpoint,
0);
// Send notification that a player has connected // Send notification that a player has connected
std::string str = m_PacketID + "Player " + m_PlayerDefinitions[i].Name + " connected on: " std::string str = m_PacketID + "Player " + m_PlayerDefinitions[i].Name + " connected on: "
+ m_PlayerDefinitions[i].Endpoint.address().to_string(); + m_PlayerDefinitions[i].Endpoint.address().to_string();
Broadcast(str); Broadcast(str);
// +1 is the null terminator
MoveMessageHead(data, length, m_PlayerDefinitions[i].Name.size() + 1);
delete[] temp;
break; break;
} }
} }
@@ -406,18 +398,12 @@ void Server::ParseDisconnect()
void Server::ParseClientPing() void Server::ParseClientPing()
{ {
char* testMesssage = new char[128]; std::cout << m_PacketID << ":Parsing ping." << std::endl;
int testOffset = CreateMessage(MessageType::ClientPing, "Ping recieved", testMesssage); // Return ping
Package package(MessageType::ClientPing);
std::cout << m_PacketID << ":Parsing ping." << std::endl; package.AddPrimitive<int>(12); // Insert packet ID here
// Return ping package.AddString("Ping received");
m_Socket.send_to( Send(package);
boost::asio::buffer(
testMesssage,
testOffset),
m_ReceiverEndpoint,
0);
delete[] testMesssage;
} }
void Server::ParseServerPing() void Server::ParseServerPing()
@@ -430,6 +416,7 @@ void Server::ParseServerPing()
} }
} }
// NOT USED
void Server::ParseSnapshot(char * data, size_t length) void Server::ParseSnapshot(char * data, size_t length)
{ {
// Does no logic. Returns snapshot if client request one // Does no logic. Returns snapshot if client request one