Jocke
81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
#include "Network/Packet.h"
|
|
|
|
Packet::Packet(MessageType type, unsigned int& packetID)
|
|
{
|
|
m_Data = new char[m_MaxPacketSize];
|
|
Init(type, packetID);
|
|
}
|
|
|
|
// Create message
|
|
Packet::Packet(char* data, const int sizeOfPacket)
|
|
{
|
|
// Resize message
|
|
m_MaxPacketSize = sizeOfPacket;
|
|
// Copy data newly allocated memory
|
|
m_Data = new char[sizeOfPacket];
|
|
memcpy(m_Data, data, sizeOfPacket);
|
|
m_Offset = sizeOfPacket;
|
|
}
|
|
|
|
Packet::~Packet()
|
|
{
|
|
delete[] m_Data;
|
|
}
|
|
|
|
void Packet::Init(MessageType type, unsigned int & packetID)
|
|
{
|
|
m_ReturnDataOffset = 0;
|
|
m_Offset = 0;
|
|
// Create message header
|
|
// Add message type
|
|
int messageType = static_cast<int>(type);
|
|
Packet::WritePrimitive<int>(messageType);
|
|
packetID = packetID % 1000; // Packet id modulos
|
|
Packet::WritePrimitive<int>(packetID);
|
|
packetID++;
|
|
}
|
|
|
|
void Packet::WriteString(std::string str)
|
|
{
|
|
// Message, add one extra byte for null terminator
|
|
int sizeOfString = str.size() + 1;
|
|
if (m_Offset + sizeOfString > m_MaxPacketSize) {
|
|
LOG_WARNING("Package::WriteString(): Data size in packet exceeded maximum package size.\n");
|
|
return;
|
|
}
|
|
memcpy(m_Data + m_Offset, str.data(), sizeOfString * sizeof(char));
|
|
m_Offset += sizeOfString * sizeof(char);
|
|
}
|
|
|
|
void Packet::WriteData(char * data, int sizeOfData)
|
|
{
|
|
if (m_Offset + sizeOfData > m_MaxPacketSize) {
|
|
LOG_WARNING("Packet::WriteData(): Data size in packet exceeded maximum packet size.\n");
|
|
return;
|
|
}
|
|
memcpy(m_Data + m_Offset, data, sizeOfData);
|
|
m_Offset += sizeOfData;
|
|
}
|
|
|
|
std::string Packet::ReadString()
|
|
{
|
|
std::string returnValue(m_Data + m_ReturnDataOffset);
|
|
if (m_Offset < m_ReturnDataOffset + returnValue.size()) {
|
|
LOG_WARNING("packet ReadString(): Oh no! You are trying to remove things outside my memory kingdom");
|
|
return "PopFrontString Failed";
|
|
}
|
|
// +1 for null terminator.
|
|
m_ReturnDataOffset += returnValue.size() + 1;
|
|
return returnValue;
|
|
}
|
|
|
|
char * Packet::ReadData(int SizeOfData)
|
|
{
|
|
if (m_Offset < m_ReturnDataOffset + SizeOfData) {
|
|
LOG_WARNING("packet ReadData(): Oh no! You are trying to remove things outside my memory kingdom");
|
|
return nullptr;
|
|
}
|
|
unsigned int oldReturnDataOffset = m_ReturnDataOffset;
|
|
m_ReturnDataOffset += SizeOfData;
|
|
return (m_Data + oldReturnDataOffset);
|
|
} |