104 lines
3.1 KiB
C++
104 lines
3.1 KiB
C++
#pragma once
|
|
|
|
#include <sp/extensions/Compress.h>
|
|
#include <stdexcept>
|
|
|
|
namespace sp {
|
|
namespace io {
|
|
|
|
template <>
|
|
class MessageEncapsulator<option::ZlibCompress> {
|
|
public:
|
|
static DataBuffer Encapsulate(const DataBuffer& a_Data) {
|
|
return zlib::Compress(a_Data);
|
|
}
|
|
static DataBuffer Decapsulate(DataBuffer& a_Data, std::size_t a_Length) {
|
|
return zlib::Decompress(a_Data, a_Length);
|
|
}
|
|
};
|
|
|
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory>
|
|
Stream<IOTag, MessageDispatcher, MessageFactory>::Stream(IOInterface<IOTag>&& a_Interface) : m_Interface(std::move(a_Interface)) {}
|
|
|
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory>
|
|
Stream<IOTag, MessageDispatcher, MessageFactory>::Stream(Stream<IOTag, MessageDispatcher, MessageFactory>&& a_Stream) :
|
|
m_Dispatcher(std::move(a_Stream.m_Dispatcher)), m_Interface(std::move(a_Stream.m_Interface)) {}
|
|
|
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory>
|
|
void Stream<IOTag, MessageDispatcher, MessageFactory>::SendMessage(const MessageBase& a_Message) {
|
|
DataBuffer data = a_Message.Write();
|
|
DataBuffer encapsulated = Encapsulate(data);
|
|
DataBuffer finalData;
|
|
finalData << VarInt{encapsulated.GetSize()} << encapsulated;
|
|
m_Interface.Write(finalData);
|
|
}
|
|
|
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory>
|
|
void Stream<IOTag, MessageDispatcher, MessageFactory>::RecieveMessages() {
|
|
while (true) {
|
|
|
|
// reading the first VarInt part byte by byte
|
|
std::uint64_t lenghtValue = 0;
|
|
unsigned int readPos = 0;
|
|
|
|
while (true) {
|
|
static constexpr int SEGMENT_BITS = (1 << 7) - 1;
|
|
static constexpr int CONTINUE_BIT = 1 << 7;
|
|
|
|
DataBuffer buffer = m_Interface.Read(sizeof(std::uint8_t));
|
|
|
|
// eof
|
|
if (buffer.GetSize() == 0)
|
|
return;
|
|
|
|
std::uint8_t part;
|
|
buffer >> part;
|
|
lenghtValue |= static_cast<std::uint64_t>(part & SEGMENT_BITS) << readPos;
|
|
|
|
if ((part & CONTINUE_BIT) == 0)
|
|
break;
|
|
|
|
readPos += 7;
|
|
|
|
if (readPos >= 8 * sizeof(lenghtValue))
|
|
throw std::runtime_error("VarInt is too big");
|
|
}
|
|
|
|
// nothing to read
|
|
if (lenghtValue == 0)
|
|
return;
|
|
|
|
DataBuffer buffer;
|
|
buffer = m_Interface.Read(lenghtValue);
|
|
|
|
buffer = Decapsulate(buffer, lenghtValue);
|
|
|
|
VarInt packetType;
|
|
buffer >> packetType;
|
|
|
|
static const MessageFactory messageFactory;
|
|
|
|
std::unique_ptr<MessageBase> message = messageFactory.CreateMessage(packetType.GetValue());
|
|
|
|
assert(message != nullptr);
|
|
|
|
message->Read(buffer);
|
|
|
|
GetDispatcher().Dispatch(*message);
|
|
}
|
|
}
|
|
|
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory>
|
|
DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory>::Encapsulate(const DataBuffer& a_Data) {
|
|
// TODO: process compress + encryption
|
|
return MessageEncapsulator<option::ZlibCompress>::Encapsulate(a_Data);
|
|
}
|
|
|
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory>
|
|
DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory>::Decapsulate(DataBuffer& a_Data, std::size_t a_Length) {
|
|
// TODO: process compress + encryption
|
|
return MessageEncapsulator<option::ZlibCompress>::Decapsulate(a_Data, a_Length);
|
|
}
|
|
|
|
} // namespace io
|
|
} // namespace sp
|