mirror of https://github.com/OpenRCT2/OpenRCT2.git
112 lines
2.4 KiB
C++
112 lines
2.4 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.
|
|
*****************************************************************************/
|
|
|
|
#ifndef DISABLE_NETWORK
|
|
|
|
# include "NetworkPacket.h"
|
|
|
|
# include "NetworkTypes.h"
|
|
|
|
# include <memory>
|
|
|
|
NetworkPacket::NetworkPacket(NetworkCommand id) noexcept
|
|
: Header{ 0, id }
|
|
{
|
|
}
|
|
|
|
uint8_t* NetworkPacket::GetData() noexcept
|
|
{
|
|
return Data.data();
|
|
}
|
|
|
|
const uint8_t* NetworkPacket::GetData() const noexcept
|
|
{
|
|
return Data.data();
|
|
}
|
|
|
|
NetworkCommand NetworkPacket::GetCommand() const noexcept
|
|
{
|
|
return Header.Id;
|
|
}
|
|
|
|
void NetworkPacket::Clear() noexcept
|
|
{
|
|
BytesTransferred = 0;
|
|
BytesRead = 0;
|
|
Data.clear();
|
|
}
|
|
|
|
bool NetworkPacket::CommandRequiresAuth() const noexcept
|
|
{
|
|
switch (GetCommand())
|
|
{
|
|
case NetworkCommand::Ping:
|
|
case NetworkCommand::Auth:
|
|
case NetworkCommand::Token:
|
|
case NetworkCommand::GameInfo:
|
|
case NetworkCommand::ObjectsList:
|
|
case NetworkCommand::ScriptsHeader:
|
|
case NetworkCommand::ScriptsData:
|
|
case NetworkCommand::MapRequest:
|
|
case NetworkCommand::Heartbeat:
|
|
return false;
|
|
default:
|
|
return true;
|
|
}
|
|
}
|
|
|
|
void NetworkPacket::Write(const void* bytes, size_t size)
|
|
{
|
|
const uint8_t* src = reinterpret_cast<const uint8_t*>(bytes);
|
|
Data.insert(Data.end(), src, src + size);
|
|
}
|
|
|
|
void NetworkPacket::WriteString(std::string_view s)
|
|
{
|
|
Write(s.data(), s.size());
|
|
Data.push_back(0);
|
|
}
|
|
|
|
const uint8_t* NetworkPacket::Read(size_t size)
|
|
{
|
|
if (BytesRead + size > Data.size())
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
const uint8_t* data = Data.data() + BytesRead;
|
|
BytesRead += size;
|
|
return data;
|
|
}
|
|
|
|
std::string_view NetworkPacket::ReadString()
|
|
{
|
|
if (BytesRead >= Data.size())
|
|
return {};
|
|
|
|
const char* str = reinterpret_cast<const char*>(Data.data() + BytesRead);
|
|
|
|
size_t stringLen = 0;
|
|
while (BytesRead < Data.size() && str[stringLen] != '\0')
|
|
{
|
|
BytesRead++;
|
|
stringLen++;
|
|
}
|
|
|
|
if (str[stringLen] != '\0')
|
|
return {};
|
|
|
|
// Skip null terminator.
|
|
BytesRead++;
|
|
|
|
return std::string_view(str, stringLen);
|
|
}
|
|
|
|
#endif
|