First pass

This commit is contained in:
Aaron Kimbre
2025-01-08 14:01:09 -06:00
parent 23551d4ed8
commit 6cd1310460
4 changed files with 336 additions and 98 deletions

View File

@@ -2,11 +2,18 @@
#define __BITSTREAMUTILS__H__
#include "GeneralUtils.h"
#include "MessageIdentifiers.h"
#include "BitStream.h"
#include <string>
#include <algorithm>
namespace MessageType {
enum class Auth : uint32_t;
enum class Client : uint32_t;
enum class Server : uint32_t;
enum class World : uint32_t;
enum class Master : uint32_t;
}
enum class eConnectionType : uint16_t;
struct LUString {
@@ -45,6 +52,19 @@ struct LUWString {
};
};
template <typename T>
struct LUHeader {
MessageID messageID;
eConnectionType connectionType;
T internalPacketID;
LUHeader(eConnectionType connectionType, T internalPacketID, MessageID messageID = ID_USER_PACKET_ENUM) {
this->messageID = messageID;
this->connectionType = connectionType;
this->internalPacketID = internalPacketID;
};
};
namespace BitStreamUtils {
template<typename T>
void WriteHeader(RakNet::BitStream& bitStream, eConnectionType connectionType, T internalPacketID) {
@@ -53,7 +73,6 @@ namespace BitStreamUtils {
bitStream.Write(static_cast<uint32_t>(internalPacketID));
bitStream.Write<uint8_t>(0);
}
}
namespace RakNet {
@@ -100,6 +119,46 @@ namespace RakNet {
value.string.resize(value.size);
this->Write(value.string);
}
template <>
inline void RakNet::BitStream::Write<LUHeader<MessageType::Auth>>(LUHeader<MessageType::Auth> value) {
this->Write<MessageID>(value.messageID);
this->Write<eConnectionType>(value.connectionType);
this->Write<uint32_t>(static_cast<uint32_t>(value.internalPacketID));
this->Write<uint8_t>(0);
}
template <>
inline void RakNet::BitStream::Write<LUHeader<MessageType::Client>>(LUHeader<MessageType::Client> value) {
this->Write<MessageID>(value.messageID);
this->Write<eConnectionType>(value.connectionType);
this->Write<uint32_t>(static_cast<uint32_t>(value.internalPacketID));
this->Write<uint8_t>(0);
}
template <>
inline void RakNet::BitStream::Write<LUHeader<MessageType::Server>>(LUHeader<MessageType::Server> value) {
this->Write<MessageID>(value.messageID);
this->Write<eConnectionType>(value.connectionType);
this->Write<uint32_t>(static_cast<uint32_t>(value.internalPacketID));
this->Write<uint8_t>(0);
}
template <>
inline void RakNet::BitStream::Write<LUHeader<MessageType::World>>(LUHeader<MessageType::World> value) {
this->Write<MessageID>(value.messageID);
this->Write<eConnectionType>(value.connectionType);
this->Write<uint32_t>(static_cast<uint32_t>(value.internalPacketID));
this->Write<uint8_t>(0);
}
template <>
inline void RakNet::BitStream::Write<LUHeader<MessageType::Master>>(LUHeader<MessageType::Master> value) {
this->Write<MessageID>(value.messageID);
this->Write<eConnectionType>(value.connectionType);
this->Write<uint32_t>(static_cast<uint32_t>(value.internalPacketID));
this->Write<uint8_t>(0);
}
};
#endif //!__BITSTREAMUTILS__H__