mirror of https://github.com/OpenRCT2/OpenRCT2.git
82 lines
2.1 KiB
C++
82 lines
2.1 KiB
C++
/*****************************************************************************
|
|
* Copyright (c) 2014-2024 OpenRCT2 developers
|
|
*
|
|
* For a complete list of all authors, please refer to contributors.md
|
|
* Interested in contributing? Visit https://github.com/OpenRCT2/OpenRCT2
|
|
*
|
|
* OpenRCT2 is licensed under the GNU General Public License version 3.
|
|
*****************************************************************************/
|
|
|
|
#pragma once
|
|
|
|
#include "../common.h"
|
|
#include "../core/DataSerialiser.h"
|
|
#include "NetworkTypes.h"
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
|
|
#pragma pack(push, 1)
|
|
struct PacketHeader
|
|
{
|
|
uint16_t Size = 0;
|
|
NetworkCommand Id = NetworkCommand::Invalid;
|
|
};
|
|
static_assert(sizeof(PacketHeader) == 6);
|
|
#pragma pack(pop)
|
|
|
|
struct NetworkPacket final
|
|
{
|
|
NetworkPacket() noexcept = default;
|
|
NetworkPacket(NetworkCommand id) noexcept;
|
|
|
|
uint8_t* GetData() noexcept;
|
|
const uint8_t* GetData() const noexcept;
|
|
|
|
NetworkCommand GetCommand() const noexcept;
|
|
|
|
void Clear() noexcept;
|
|
bool CommandRequiresAuth() const noexcept;
|
|
|
|
const uint8_t* Read(size_t size);
|
|
std::string_view ReadString();
|
|
|
|
void Write(const void* bytes, size_t size);
|
|
void WriteString(std::string_view s);
|
|
|
|
template<typename T> NetworkPacket& operator>>(T& value)
|
|
{
|
|
if (BytesRead + sizeof(value) > Header.Size)
|
|
{
|
|
value = T{};
|
|
}
|
|
else
|
|
{
|
|
T local;
|
|
std::memcpy(&local, &GetData()[BytesRead], sizeof(local));
|
|
value = ByteSwapBE(local);
|
|
BytesRead += sizeof(value);
|
|
}
|
|
return *this;
|
|
}
|
|
|
|
template<typename T> NetworkPacket& operator<<(T value)
|
|
{
|
|
T swapped = ByteSwapBE(value);
|
|
Write(&swapped, sizeof(T));
|
|
return *this;
|
|
}
|
|
|
|
NetworkPacket& operator<<(DataSerialiser& data)
|
|
{
|
|
Write(static_cast<const uint8_t*>(data.GetStream().GetData()), data.GetStream().GetLength());
|
|
return *this;
|
|
}
|
|
|
|
public:
|
|
PacketHeader Header{};
|
|
std::vector<uint8_t> Data;
|
|
size_t BytesTransferred = 0;
|
|
size_t BytesRead = 0;
|
|
};
|