5 Commits

Author SHA1 Message Date
9fa023f716 generic encapsulator 2025-02-27 22:50:51 +01:00
f6620dc522 compress refactor 2025-02-27 21:45:09 +01:00
6a52b7fe2a fix compression 2025-02-26 15:09:23 +01:00
68fcd514a3 xmake: rename targets 2025-02-26 12:34:58 +01:00
5f9cc86ad2 xmake: add target extensions deps 2025-02-26 12:33:19 +01:00
11 changed files with 37 additions and 129 deletions

View File

@@ -1,31 +0,0 @@
name: Linux arm64
run-name: Build And Test
on: [push]
env:
XMAKE_ROOT: y
jobs:
Build:
runs-on: ubuntu-latest
steps:
- name: Check out repository code
uses: actions/checkout@v3
- name: Prepare XMake
uses: xmake-io/github-action-setup-xmake@v1
with:
xmake-version: latest
actions-cache-folder: '.xmake-cache'
actions-cache-key: 'ubuntu'
- name: XMake config
run: xmake f -p linux -y
- name: Build
run: xmake
- name: Test
run: xmake test

View File

@@ -166,7 +166,7 @@ class DataBuffer {
K newKey; K newKey;
V newValue; V newValue;
*this >> newKey >> newValue; *this >> newKey >> newValue;
data.emplace(newKey, newValue); data.insert({newKey, newValue});
} }
return *this; return *this;
} }

View File

@@ -21,8 +21,6 @@ class VarInt {
std::uint64_t m_Value; std::uint64_t m_Value;
public: public:
static const std::uint64_t MAX_VALUE = static_cast<std::uint64_t>(-1) >> 8;
VarInt() : m_Value(0) {} VarInt() : m_Value(0) {}
/** /**
* \brief Construct a variable integer from a value * \brief Construct a variable integer from a value

View File

@@ -47,7 +47,7 @@ template <>
class MessageEncapsulator<option::ZlibCompress> { class MessageEncapsulator<option::ZlibCompress> {
public: public:
static DataBuffer Encapsulate(const DataBuffer& a_Data, const option::ZlibCompress& a_Option); static DataBuffer Encapsulate(const DataBuffer& a_Data, const option::ZlibCompress& a_Option);
static DataBuffer Decapsulate(DataBuffer& a_Data, const option::ZlibCompress& a_Option); static DataBuffer Decapsulate(DataBuffer& a_Data, std::size_t a_Length, const option::ZlibCompress& a_Option);
}; };
} // namespace io } // namespace io

View File

@@ -1,41 +0,0 @@
#pragma once
/**
* \file Encrypt.h
* \brief File containing encrypt utilities
*/
#include <sp/common/DataBuffer.h>
namespace sp {
namespace option {
struct TlsEncrypt {
bool m_Enabled = true;
};
} // namespace option
} // namespace sp
#include <sp/io/IOInterface.h>
namespace sp {
namespace ssl {
// encrypt
// decrypt
} // namespace ssl
namespace io {
template <>
class MessageEncapsulator<option::TlsEncrypt> {
public:
static DataBuffer Encapsulate(const DataBuffer& a_Data, const option::TlsEncrypt& a_Option);
static DataBuffer Decapsulate(DataBuffer& a_Data, const option::TlsEncrypt& a_Option);
};
} // namespace io
} // namespace sp

View File

@@ -17,7 +17,7 @@ template <typename TOption>
class MessageEncapsulator { class MessageEncapsulator {
public: public:
static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOption& a_Option); static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOption& a_Option);
static DataBuffer Decapsulate(DataBuffer& a_Data, const TOption& a_Option); static DataBuffer Decapsulate(DataBuffer& a_Data, std::size_t a_Length, const TOption& a_Option);
}; };
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions> template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
@@ -25,23 +25,16 @@ class Stream {
protected: protected:
MessageDispatcher m_Dispatcher; MessageDispatcher m_Dispatcher;
IOInterface<IOTag> m_Interface; IOInterface<IOTag> m_Interface;
std::tuple<TOptions...> m_Options;
using MessageBase = typename MessageDispatcher::MessageBaseType; using MessageBase = typename MessageDispatcher::MessageBaseType;
public: public:
Stream() {} Stream() {}
Stream(IOInterface<IOTag>&& a_Interface, TOptions&&... a_Options); Stream(IOInterface<IOTag>&& a_Interface);
Stream(Stream&& a_Stream); Stream(Stream&& a_Stream);
void RecieveMessages(); void RecieveMessages(const TOptions&... a_Options);
void SendMessage(const MessageBase& a_Message); void SendMessage(const MessageBase& a_Message, const TOptions&... a_Options);
template <typename TOption>
TOption& GetOption() {
return std::get<TOption>(m_Options);
}
MessageDispatcher& GetDispatcher() { MessageDispatcher& GetDispatcher() {
return m_Dispatcher; return m_Dispatcher;
@@ -49,7 +42,7 @@ class Stream {
private: private:
static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOptions&... a_Options); static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOptions&... a_Options);
static DataBuffer Decapsulate(DataBuffer& a_Data, const TOptions&... a_Options); static DataBuffer Decapsulate(DataBuffer& a_Data, std::size_t a_Length, const TOptions&... a_Options);
}; };
} // namespace io } // namespace io

View File

@@ -15,7 +15,7 @@ struct MessageEncapsulatorPack<> {
static DataBuffer Encapsulate(const DataBuffer& a_Data) { static DataBuffer Encapsulate(const DataBuffer& a_Data) {
return a_Data; return a_Data;
} }
static DataBuffer Decapsulate(DataBuffer& a_Data) { static DataBuffer Decapsulate(DataBuffer& a_Data, std::size_t a_Length) {
return a_Data; return a_Data;
} }
}; };
@@ -26,9 +26,9 @@ struct MessageEncapsulatorPack<TOption, TOptions...> {
DataBuffer data = io::MessageEncapsulator<TOption>::Encapsulate(a_Data, a_Option); DataBuffer data = io::MessageEncapsulator<TOption>::Encapsulate(a_Data, a_Option);
return MessageEncapsulatorPack<TOptions...>::Encapsulate(data, a_Options...); return MessageEncapsulatorPack<TOptions...>::Encapsulate(data, a_Options...);
} }
static DataBuffer Decapsulate(DataBuffer& a_Data, const TOption& a_Option, const TOptions&... a_Options) { static DataBuffer Decapsulate(DataBuffer& a_Data, std::size_t a_Length, const TOption& a_Option, const TOptions&... a_Options) {
DataBuffer data = io::MessageEncapsulator<TOption>::Decapsulate(a_Data, a_Option); DataBuffer data = io::MessageEncapsulator<TOption>::Decapsulate(a_Data, a_Length, a_Option);
return MessageEncapsulatorPack<TOptions...>::Decapsulate(data, a_Options...); return MessageEncapsulatorPack<TOptions...>::Decapsulate(data, data.GetSize(), a_Options...);
} }
}; };
@@ -37,8 +37,8 @@ struct MessageEncapsulatorPack<TOption, TOptions...> {
namespace io { namespace io {
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions> template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(IOInterface<IOTag>&& a_Interface, TOptions&&... a_Options) : Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(IOInterface<IOTag>&& a_Interface) :
m_Interface(std::move(a_Interface)), m_Options(std::make_tuple<TOptions...>(std::move(a_Options)...)) {} m_Interface(std::move(a_Interface)) {}
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions> template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream( Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(
@@ -46,18 +46,17 @@ Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(
m_Dispatcher(std::move(a_Stream.m_Dispatcher)), m_Interface(std::move(a_Stream.m_Interface)) {} m_Dispatcher(std::move(a_Stream.m_Dispatcher)), m_Interface(std::move(a_Stream.m_Interface)) {}
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions> template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::SendMessage(const MessageBase& a_Message) { void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::SendMessage(
const MessageBase& a_Message, const TOptions&... a_Options) {
DataBuffer data = a_Message.Write(); DataBuffer data = a_Message.Write();
DataBuffer encapsulated = std::apply([&data](const auto&... a_Options) { DataBuffer encapsulated = Encapsulate(data, a_Options...);
return Encapsulate(data, a_Options...);
}, m_Options);
DataBuffer finalData; DataBuffer finalData;
finalData << VarInt{encapsulated.GetSize()} << encapsulated; finalData << VarInt{encapsulated.GetSize()} << encapsulated;
m_Interface.Write(finalData); m_Interface.Write(finalData);
} }
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions> template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::RecieveMessages() { void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::RecieveMessages(const TOptions&... a_Options) {
while (true) { while (true) {
// reading the first VarInt part byte by byte // reading the first VarInt part byte by byte
std::uint64_t lenghtValue = 0; std::uint64_t lenghtValue = 0;
@@ -93,9 +92,7 @@ void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::RecieveMessa
DataBuffer buffer; DataBuffer buffer;
buffer = m_Interface.Read(lenghtValue); buffer = m_Interface.Read(lenghtValue);
buffer = std::apply([&buffer, lenghtValue](const auto&... a_Options) { buffer = Decapsulate(buffer, lenghtValue, a_Options...);
return Decapsulate(buffer, a_Options...);
}, m_Options);
VarInt packetType; VarInt packetType;
buffer >> packetType; buffer >> packetType;
@@ -120,8 +117,8 @@ DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Encaps
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions> template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Decapsulate( DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Decapsulate(
DataBuffer& a_Data, const TOptions&... a_Options) { DataBuffer& a_Data, std::size_t a_Length, const TOptions&... a_Options) {
return details::MessageEncapsulatorPack<TOptions...>::Decapsulate(a_Data, a_Options...); return details::MessageEncapsulatorPack<TOptions...>::Decapsulate(a_Data, a_Length, a_Options...);
} }
} // namespace io } // namespace io

View File

@@ -77,12 +77,12 @@ DataBuffer Decompress(DataBuffer& buffer, std::uint64_t packetLength) {
namespace io { namespace io {
DataBuffer MessageEncapsulator<option::ZlibCompress>::Encapsulate(const DataBuffer& a_Data, const option::ZlibCompress& a_Option) { DataBuffer MessageEncapsulator<option::ZlibCompress>::Encapsulate(const DataBuffer& a_Data, const option::ZlibCompress& a_Option) {
static constexpr std::size_t MAX_COMPRESS_THRESHOLD = VarInt::MAX_VALUE; return a_Option.m_Enabled ? zlib::Compress(a_Data, a_Option.m_CompressionThreshold) : a_Data;
return zlib::Compress(a_Data, a_Option.m_Enabled ? a_Option.m_CompressionThreshold : MAX_COMPRESS_THRESHOLD);
} }
DataBuffer MessageEncapsulator<option::ZlibCompress>::Decapsulate(DataBuffer& a_Data, const option::ZlibCompress& a_Option) { DataBuffer MessageEncapsulator<option::ZlibCompress>::Decapsulate(
return zlib::Decompress(a_Data, a_Data.GetSize()); DataBuffer& a_Data, std::size_t a_Length, const option::ZlibCompress& a_Option) {
return a_Option.m_Enabled ? zlib::Decompress(a_Data, a_Length) : a_Data;
} }
} // namespace io } // namespace io

View File

@@ -1,3 +0,0 @@
#include <sp/extensions/Encrypt.h>
#include <openssl/ssl.h>

View File

@@ -22,21 +22,21 @@ using FileStream = sp::io::Stream<sp::io::FileTag, sp::PacketDispatcher, sp::Pac
int main() { int main() {
auto handler = std::make_shared<CustomPacketHandler>(); auto handler = std::make_shared<CustomPacketHandler>();
FileStream stream(sp::io::File{"test.txt", sp::io::FileTag::In | sp::io::FileTag::Out}, {}); FileStream stream(sp::io::File{"test.txt", sp::io::FileTag::In | sp::io::FileTag::Out});
stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler); stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler);
stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler); stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler);
stream.SendMessage(KeepAlivePacket{96}); auto zlibOptions = sp::option::ZlibCompress{};
stream.SendMessage(KeepAlivePacket{69});
stream.SendMessage(DisconnectPacket{
"This is in the "
"fiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiile !"});
stream.GetOption<sp::option::ZlibCompress>().m_Enabled = false;
stream.SendMessage(DisconnectPacket{
"This is in the "
"fiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiile !"});
stream.RecieveMessages(); stream.SendMessage(KeepAlivePacket{96}, zlibOptions);
stream.SendMessage(KeepAlivePacket{69}, zlibOptions);
stream.SendMessage(
DisconnectPacket{
"This is in the "
"fiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiiile !"},
zlibOptions);
stream.RecieveMessages(zlibOptions);
return 0; return 0;
} }

View File

@@ -6,14 +6,9 @@ local modules = {
Compression = { Compression = {
Option = "zlib", Option = "zlib",
Deps = {"zlib"}, Deps = {"zlib"},
Packages = {"zlib"},
Includes = {"include/(sp/extensions/Compress.h)"}, Includes = {"include/(sp/extensions/Compress.h)"},
Sources = {"src/sp/extensions/Compress.cpp"} Sources = {"src/sp/extensions/Compress.cpp"}
},
Encryption = {
Option = "ssl",
Deps = {"openssl"},
Includes = {"include/(sp/extensions/Encrypt.h)"},
Sources = {"src/sp/extensions/Encrypt.cpp"}
} }
} }
@@ -42,7 +37,7 @@ for name, module in table.orderpairs(modules) do
for _, source in table.orderpairs(module.Sources) do for _, source in table.orderpairs(module.Sources) do
add_files(source) add_files(source)
end end
for _, package in table.orderpairs(module.Deps) do for _, package in table.orderpairs(module.Packages) do
add_packages(package) add_packages(package)
end end
set_group("Library") set_group("Library")