Compare commits
11 Commits
v1.0.4
...
3bc5b50905
| Author | SHA1 | Date | |
|---|---|---|---|
| 3bc5b50905 | |||
| 66835554f1 | |||
| e16ad84865 | |||
| c1e6409c18 | |||
| 3ad18cacf6 | |||
| 2eab50932f | |||
| baa52d3baa | |||
| d924685e0c | |||
| 044b12cdec | |||
| e39f8de898 | |||
| 0b28bde25b |
31
.github/workflows/ubuntu.yml
vendored
31
.github/workflows/ubuntu.yml
vendored
@@ -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
|
|
||||||
10
README.md
10
README.md
@@ -1,10 +0,0 @@
|
|||||||
# SimpleProtocolLib
|
|
||||||
|
|
||||||
Network engine used to (mainly) communicate with packets
|
|
||||||
|
|
||||||
# Integrate with xmake
|
|
||||||
|
|
||||||
```lua
|
|
||||||
add_repositories("persson-repo https://git.ale-pri.com/Persson-dev/xmake-repo.git")
|
|
||||||
add_requires("splib", { debug = is_mode("debug") })
|
|
||||||
```
|
|
||||||
@@ -6,13 +6,12 @@ enum PacketId {
|
|||||||
UpgradeTower,
|
UpgradeTower,
|
||||||
};
|
};
|
||||||
|
|
||||||
#include <examples/DisconnectPacket.h>
|
|
||||||
#include <examples/KeepAlivePacket.h>
|
#include <examples/KeepAlivePacket.h>
|
||||||
|
#include <examples/DisconnectPacket.h>
|
||||||
#include <examples/UpgradeTowerPacket.h>
|
#include <examples/UpgradeTowerPacket.h>
|
||||||
|
|
||||||
// they must be in the same order as in the enum !
|
// they must be in the same order as in the enum !
|
||||||
using AllPackets = std::tuple<KeepAlivePacket, DisconnectPacket, UpgradeTowerPacket>;
|
using AllPackets = std::tuple<KeepAlivePacket, DisconnectPacket, UpgradeTowerPacket>;
|
||||||
|
|
||||||
#include <sp/default/DefaultPacketDispatcher.h>
|
#include <sp/default/DefaultPacketHandler.h>
|
||||||
#include <sp/default/DefaultPacketFactory.h>
|
#include <sp/default/DefaultPacketFactory.h>
|
||||||
#include <sp/default/DefaultPacketHandler.h>
|
|
||||||
@@ -14,8 +14,7 @@ using UpgradeTowerFields = std::tuple<
|
|||||||
sp::BitField<std::uint16_t,
|
sp::BitField<std::uint16_t,
|
||||||
sp::Field<std::uint16_t, 12>, //<- m_Tower
|
sp::Field<std::uint16_t, 12>, //<- m_Tower
|
||||||
sp::Field<std::uint8_t, 4> //<- m_Upgrade
|
sp::Field<std::uint8_t, 4> //<- m_Upgrade
|
||||||
>,
|
>
|
||||||
sp::VarInt //<- just for testing
|
|
||||||
>;
|
>;
|
||||||
|
|
||||||
DeclarePacket(UpgradeTower){
|
DeclarePacket(UpgradeTower){
|
||||||
|
|||||||
@@ -1,10 +1,26 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
|
#ifdef _WIN32
|
||||||
|
#include <winsock2.h>
|
||||||
|
#else
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
bool IsSystemBigEndian();
|
template <typename T>
|
||||||
|
void SwapBytes(T& value) {
|
||||||
|
char* ptr = reinterpret_cast<char*>(&value);
|
||||||
|
std::reverse(ptr, ptr + sizeof(T));
|
||||||
|
}
|
||||||
|
|
||||||
|
bool IsSystemBigEndian() {
|
||||||
|
static constexpr std::uint16_t test = 10;
|
||||||
|
static const bool isBigEndian = reinterpret_cast<const std::uint8_t*>(&test)[1] == 10;
|
||||||
|
return isBigEndian;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \brief Serialize value to (network byte order) big endian
|
* \brief Serialize value to (network byte order) big endian
|
||||||
@@ -13,13 +29,21 @@ template <typename T>
|
|||||||
void ToNetwork(T& value) {}
|
void ToNetwork(T& value) {}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void ToNetwork<std::uint16_t>(std::uint16_t& value);
|
void ToNetwork<std::uint16_t>(std::uint16_t& value) {
|
||||||
|
value = htons(value);
|
||||||
|
}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void ToNetwork<std::uint32_t>(std::uint32_t& value);
|
void ToNetwork<std::uint32_t>(std::uint32_t& value) {
|
||||||
|
value = htonl(value);
|
||||||
|
}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void ToNetwork<std::uint64_t>(std::uint64_t& value);
|
void ToNetwork<std::uint64_t>(std::uint64_t& value) {
|
||||||
|
if (IsSystemBigEndian())
|
||||||
|
return;
|
||||||
|
SwapBytes(value);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \brief Deserialize value from (network byte order) big endian
|
* \brief Deserialize value from (network byte order) big endian
|
||||||
@@ -28,13 +52,21 @@ template <typename T>
|
|||||||
void FromNetwork(T& value) {}
|
void FromNetwork(T& value) {}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void FromNetwork<std::uint16_t>(std::uint16_t& value);
|
void FromNetwork<std::uint16_t>(std::uint16_t& value) {
|
||||||
|
value = ntohs(value);
|
||||||
|
}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void FromNetwork<std::uint32_t>(std::uint32_t& value);
|
void FromNetwork<std::uint32_t>(std::uint32_t& value) {
|
||||||
|
value = ntohl(value);
|
||||||
|
}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void FromNetwork<std::uint64_t>(std::uint64_t& value);
|
void FromNetwork<std::uint64_t>(std::uint64_t& value) {
|
||||||
|
if (IsSystemBigEndian())
|
||||||
|
return;
|
||||||
|
SwapBytes(value);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \brief Swap bytes if the value is any kind of integer
|
* \brief Swap bytes if the value is any kind of integer
|
||||||
@@ -43,12 +75,18 @@ template <typename T>
|
|||||||
void TrySwapBytes(T& value) {}
|
void TrySwapBytes(T& value) {}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void TrySwapBytes<std::uint16_t>(std::uint16_t& value);
|
void TrySwapBytes<std::uint16_t>(std::uint16_t& value) {
|
||||||
|
SwapBytes(value);
|
||||||
|
}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void TrySwapBytes<std::uint32_t>(std::uint32_t& value);
|
void TrySwapBytes<std::uint32_t>(std::uint32_t& value) {
|
||||||
|
SwapBytes(value);
|
||||||
|
}
|
||||||
|
|
||||||
template <>
|
template <>
|
||||||
void TrySwapBytes<std::uint64_t>(std::uint64_t& value);
|
void TrySwapBytes<std::uint64_t>(std::uint64_t& value) {
|
||||||
|
SwapBytes(value);
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -13,7 +13,6 @@
|
|||||||
#include <string>
|
#include <string>
|
||||||
#include <sp/common/VarInt.h>
|
#include <sp/common/VarInt.h>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <map>
|
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
@@ -35,7 +34,6 @@ class DataBuffer {
|
|||||||
typedef Data::difference_type difference_type;
|
typedef Data::difference_type difference_type;
|
||||||
|
|
||||||
DataBuffer();
|
DataBuffer();
|
||||||
DataBuffer(std::size_t a_InitialSize);
|
|
||||||
DataBuffer(const DataBuffer& other);
|
DataBuffer(const DataBuffer& other);
|
||||||
DataBuffer(const DataBuffer& other, difference_type offset);
|
DataBuffer(const DataBuffer& other, difference_type offset);
|
||||||
DataBuffer(DataBuffer&& other);
|
DataBuffer(DataBuffer&& other);
|
||||||
@@ -90,19 +88,6 @@ class DataBuffer {
|
|||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Append a map to the buffer by first writing the size
|
|
||||||
* \param data The map to append
|
|
||||||
*/
|
|
||||||
template <typename K, typename V>
|
|
||||||
DataBuffer& operator<<(const std::map<K, V>& data) {
|
|
||||||
*this << VarInt{data.size()};
|
|
||||||
for (const auto& element : data) {
|
|
||||||
*this << element.first << element.second;
|
|
||||||
}
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \brief Append an array to the buffer by first writing the size
|
* \brief Append an array to the buffer by first writing the size
|
||||||
* \param data The buffer to append
|
* \param data The buffer to append
|
||||||
@@ -155,23 +140,6 @@ class DataBuffer {
|
|||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Read a map (size + data) from the buffer
|
|
||||||
* \pre The map is assumed to be empty
|
|
||||||
*/
|
|
||||||
template <typename K, typename V>
|
|
||||||
DataBuffer& operator>>(std::map<K, V>& data) {
|
|
||||||
VarInt mapSize;
|
|
||||||
*this >> mapSize;
|
|
||||||
for (std::size_t i = 0; i < mapSize.GetValue(); i++) {
|
|
||||||
K newKey;
|
|
||||||
V newValue;
|
|
||||||
*this >> newKey >> newValue;
|
|
||||||
data.emplace(newKey, newValue);
|
|
||||||
}
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \brief Read an array from the buffer
|
* \brief Read an array from the buffer
|
||||||
*/
|
*/
|
||||||
|
|||||||
@@ -1,25 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \file NonCopyable.h
|
|
||||||
* \brief File containing the sp::NonCopyable class
|
|
||||||
*/
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \class NonCopyable
|
|
||||||
* \brief Class used to make a class non copyable
|
|
||||||
* \note Inherit from this class privately to make a class non copyable
|
|
||||||
*/
|
|
||||||
class NonCopyable {
|
|
||||||
public:
|
|
||||||
NonCopyable(const NonCopyable&) = delete;
|
|
||||||
NonCopyable& operator=(const NonCopyable&) = delete;
|
|
||||||
|
|
||||||
protected:
|
|
||||||
NonCopyable() {}
|
|
||||||
~NonCopyable() {}
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,60 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <cxxabi.h>
|
|
||||||
#include <string>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::string GetBasicClassName(const T& a_Value) {
|
|
||||||
int status;
|
|
||||||
char* demangled = abi::__cxa_demangle(typeid(a_Value).name(), 0, 0, &status);
|
|
||||||
if (status != 0)
|
|
||||||
return "";
|
|
||||||
return std::string(demangled);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::string GetBasicClassName() {
|
|
||||||
int status;
|
|
||||||
char* demangled = abi::__cxa_demangle(typeid(T).name(), 0, 0, &status);
|
|
||||||
if (status != 0)
|
|
||||||
return "";
|
|
||||||
return std::string(demangled);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
struct Reflector {
|
|
||||||
static std::string GetClassName() {
|
|
||||||
return GetBasicClassName<T>();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
struct Reflector<std::string> {
|
|
||||||
static std::string GetClassName() {
|
|
||||||
return "std::string";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename K, typename V>
|
|
||||||
struct Reflector<std::map<K, V>> {
|
|
||||||
static std::string GetClassName();
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
struct Reflector<std::vector<T>> {
|
|
||||||
static std::string GetClassName();
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::string Reflector<std::vector<T>>::GetClassName() {
|
|
||||||
return "std::vector<" + Reflector<T>::GetClassName() + ">";
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename K, typename V>
|
|
||||||
std::string Reflector<std::map<K, V>>::GetClassName() {
|
|
||||||
return "std::map<" + Reflector<K>::GetClassName() + ", " + Reflector<V>::GetClassName() + ">";
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,7 +1,6 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <string>
|
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
@@ -51,48 +50,4 @@ void TupleForEach(TFunc&& func, TTuple&& tuple) {
|
|||||||
details::TupleForEachHelper<TupleSize>::Exec(std::forward<TTuple>(tuple), std::forward<TFunc>(func));
|
details::TupleForEachHelper<TupleSize>::Exec(std::forward<TTuple>(tuple), std::forward<TFunc>(func));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
namespace details {
|
|
||||||
|
|
||||||
template <typename T, typename U = void>
|
|
||||||
struct is_mappish_impl : std::false_type {};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
struct is_mappish_impl<T, std::void_t<typename T::key_type, typename T::mapped_type,
|
|
||||||
decltype(std::declval<T&>()[std::declval<const typename T::key_type&>()])>> : std::true_type {};
|
|
||||||
|
|
||||||
template <typename T, typename U = void>
|
|
||||||
struct is_vectish_impl : std::false_type {};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
struct is_vectish_impl<T,
|
|
||||||
std::void_t<typename T::value_type, decltype(std::declval<T&>()[std::declval<const typename T::value_type&>()])>>
|
|
||||||
: std::true_type {};
|
|
||||||
|
|
||||||
template <typename T, typename U = void>
|
|
||||||
struct is_pairish_impl : std::false_type {};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
struct is_pairish_impl<T,
|
|
||||||
std::void_t<typename T::first_type, decltype(std::declval<T&>()[std::declval<const typename T::first_type&>()])>>
|
|
||||||
: std::true_type {};
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
using is_general_t = std::integral_constant<bool,
|
|
||||||
(std::is_same_v<T, std::string> || (!std::is_same_v<T, char> && !std::is_same_v<T, unsigned char> && !std::is_arithmetic_v<T> &&
|
|
||||||
!is_pairish_impl<T>::value && !is_mappish_impl<T>::value && !is_vectish_impl<T>::value))>;
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
using is_primitive =
|
|
||||||
std::integral_constant<bool, std::is_same_v<T, char> || std::is_same_v<T, unsigned char> || std::is_arithmetic_v<T>>;
|
|
||||||
|
|
||||||
} // namespace details
|
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -7,7 +7,6 @@
|
|||||||
|
|
||||||
#include <cstddef>
|
#include <cstddef>
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
#include <sp/protocol/MessagePrinter.h>
|
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
@@ -22,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
|
||||||
@@ -58,9 +55,4 @@ class VarInt {
|
|||||||
friend DataBuffer& operator>>(DataBuffer& in, VarInt& var);
|
friend DataBuffer& operator>>(DataBuffer& in, VarInt& var);
|
||||||
};
|
};
|
||||||
|
|
||||||
template<>
|
|
||||||
inline std::string PrintData(const VarInt& a_VarInt) {
|
|
||||||
return PrintData(a_VarInt.GetValue());
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -6,13 +6,11 @@
|
|||||||
namespace sp {
|
namespace sp {
|
||||||
class PacketHandler;
|
class PacketHandler;
|
||||||
|
|
||||||
using PacketMessage = Message<
|
using PacketMessage = Message<option::MsgIdType<std::uint8_t>, // add id() operation
|
||||||
option::MsgIdType<std::uint8_t>, // add id() operation
|
|
||||||
option::ReadOperations, // add read() operation
|
option::ReadOperations, // add read() operation
|
||||||
option::WriteOperations, // add write() operation
|
option::WriteOperations, // add write() operation
|
||||||
option::WriteId, // write id before data
|
option::WriteId, // write id before data
|
||||||
option::Handler<PacketHandler>, // add dispatch() operation
|
option::Handler<PacketHandler> // add dispatch() operation
|
||||||
option::DebugPrint // add ToString() operator
|
|
||||||
>;
|
>;
|
||||||
|
|
||||||
#define PacketConstructor(packetName) \
|
#define PacketConstructor(packetName) \
|
||||||
@@ -24,8 +22,7 @@ using PacketMessage = Message<
|
|||||||
|
|
||||||
#define DeclarePacket(packetName) \
|
#define DeclarePacket(packetName) \
|
||||||
class packetName##Packet : public sp::MessageBase<sp::PacketMessage, sp::option::StaticNumIdImpl<packetName>, \
|
class packetName##Packet : public sp::MessageBase<sp::PacketMessage, sp::option::StaticNumIdImpl<packetName>, \
|
||||||
sp::option::DispatchImpl<packetName##Packet>, sp::option::FieldsImpl<packetName##Fields>>, \
|
sp::option::DispatchImpl<packetName##Packet>, sp::option::FieldsImpl<packetName##Fields>>
|
||||||
sp::option::ToStringImpl<packetName##Packet>
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,15 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/default/DefaultPacket.h>
|
|
||||||
#include <sp/default/DefaultPacketHandler.h>
|
|
||||||
#include <sp/protocol/MessageDispatcher.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
using PacketDispatcher = MessageDispatcher<
|
|
||||||
PacketMessage::ParsedOptions::MsgIdType,
|
|
||||||
PacketMessage,
|
|
||||||
PacketMessage::ParsedOptions::HandlerType::HandlerT
|
|
||||||
>;
|
|
||||||
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,54 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \file Compression.h
|
|
||||||
* \brief File containing compress utilities
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <cstdint>
|
|
||||||
#include <sp/common/DataBuffer.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace option {
|
|
||||||
|
|
||||||
struct ZlibCompress {
|
|
||||||
bool m_Enabled = true;
|
|
||||||
std::size_t m_CompressionThreshold = 64;
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace option
|
|
||||||
} // namespace sp
|
|
||||||
|
|
||||||
#include <sp/io/IOInterface.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace zlib {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Compress some data
|
|
||||||
* \param buffer the data to compress
|
|
||||||
* \return the compressed data
|
|
||||||
*/
|
|
||||||
DataBuffer Compress(const DataBuffer& buffer, std::size_t a_CompressionThreshold = 64);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Uncompress some data
|
|
||||||
* \param buffer the data to uncompress
|
|
||||||
* \param packetLength lenght of data
|
|
||||||
* \return the uncompressed data
|
|
||||||
*/
|
|
||||||
DataBuffer Decompress(DataBuffer& buffer, std::uint64_t packetLength);
|
|
||||||
|
|
||||||
} // namespace zlib
|
|
||||||
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
template <>
|
|
||||||
class MessageEncapsulator<option::ZlibCompress> {
|
|
||||||
public:
|
|
||||||
static DataBuffer Encapsulate(const DataBuffer& a_Data, const option::ZlibCompress& a_Option);
|
|
||||||
static DataBuffer Decapsulate(DataBuffer& a_Data, const option::ZlibCompress& a_Option);
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,9 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#if __has_include(<sp/extensions/Compress.h>)
|
|
||||||
#include <sp/extensions/Compress.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if __has_include(<sp/extensions/Tcp.h>)
|
|
||||||
#include <sp/extensions/Tcp.h>
|
|
||||||
#endif
|
|
||||||
@@ -1,4 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/extensions/tcp/TcpListener.h>
|
|
||||||
#include <sp/extensions/tcp/TcpSocket.h>
|
|
||||||
@@ -1,65 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/extensions/tcp/TcpSocket.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \class TcpListener
|
|
||||||
*/
|
|
||||||
class TcpListener : private NonCopyable {
|
|
||||||
public:
|
|
||||||
/**
|
|
||||||
* \brief Starts listening for guests to connect
|
|
||||||
* \param port The port to listen to
|
|
||||||
* \param maxConnexions The maximum amount of connexion that can happen at the same time. \n
|
|
||||||
* Every other guests will be kicked if this amount is reached.
|
|
||||||
* \return Whether this action was succesfull
|
|
||||||
*/
|
|
||||||
TcpListener(std::uint16_t a_Port, int a_MaxConnexions);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Default destructor
|
|
||||||
*/
|
|
||||||
~TcpListener();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Tries to accept an incoming request to connect
|
|
||||||
* \return the new socket if a new connexion was accepted or nullptr
|
|
||||||
*/
|
|
||||||
std::unique_ptr<TcpSocket> Accept();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Closes the socket
|
|
||||||
*/
|
|
||||||
void Close();
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Allows to set the socket in non blocking/blocking mode
|
|
||||||
* \param a_Blocking If set to true, every call to Read will wait until the socket receives something
|
|
||||||
* \return true if the operation was successful
|
|
||||||
*/
|
|
||||||
bool SetBlocking(bool a_Blocking);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Getter of the m_Port member
|
|
||||||
* \return The port which the socket listen to
|
|
||||||
*/
|
|
||||||
std::uint16_t GetListeningPort() const;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Getter of the m_MaxConnections member
|
|
||||||
* \return The maximum amount of connexions that can happen at the same time.
|
|
||||||
*/
|
|
||||||
int GetMaximumConnections() const;
|
|
||||||
|
|
||||||
private:
|
|
||||||
SocketHandle m_Handle;
|
|
||||||
std::uint16_t m_Port;
|
|
||||||
int m_MaxConnections;
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,86 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/common/NonCopyable.h>
|
|
||||||
#include <sp/io/IOInterface.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
using SocketHandle = int;
|
|
||||||
|
|
||||||
struct TcpTag {};
|
|
||||||
|
|
||||||
class SocketError : public std::exception {
|
|
||||||
private:
|
|
||||||
std::string m_Error;
|
|
||||||
|
|
||||||
public:
|
|
||||||
SocketError(std::string&& a_Msg) : m_Error(std::move(a_Msg)) {}
|
|
||||||
|
|
||||||
virtual const char* what() const noexcept override {
|
|
||||||
return m_Error.c_str();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
class IOInterface<TcpTag> : private NonCopyable {
|
|
||||||
public:
|
|
||||||
/**
|
|
||||||
* \enum Status
|
|
||||||
* \brief Describes the state of a socket
|
|
||||||
*/
|
|
||||||
enum class Status {
|
|
||||||
/** The socket is connected */
|
|
||||||
Connected,
|
|
||||||
/** The socket is not connected */
|
|
||||||
Disconnected,
|
|
||||||
/** Something bad happened */
|
|
||||||
Error,
|
|
||||||
};
|
|
||||||
|
|
||||||
IOInterface();
|
|
||||||
IOInterface(const std::string& a_Host, std::uint16_t a_Port);
|
|
||||||
IOInterface(IOInterface&& a_Other);
|
|
||||||
IOInterface& operator=(IOInterface&& a_Other);
|
|
||||||
virtual ~IOInterface();
|
|
||||||
|
|
||||||
DataBuffer Read(std::size_t a_Amount);
|
|
||||||
void Write(const sp::DataBuffer& a_Data);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Allows to set the socket in non blocking/blocking mode
|
|
||||||
* \param a_Block If set to true, every call to Read will wait until the socket receives something
|
|
||||||
* \return true if the operation was successful
|
|
||||||
*/
|
|
||||||
bool SetBlocking(bool a_Block);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Getter of the m_Status member
|
|
||||||
* \return The TcpSocket::Status of this socket
|
|
||||||
*/
|
|
||||||
Status GetStatus() const;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Disconnects the socket from the remote
|
|
||||||
* \note Does nothing if the socket is not connected. \n
|
|
||||||
* This function is also called by the destructor.
|
|
||||||
*/
|
|
||||||
void Disconnect();
|
|
||||||
|
|
||||||
|
|
||||||
private:
|
|
||||||
SocketHandle m_Handle;
|
|
||||||
Status m_Status;
|
|
||||||
|
|
||||||
void Connect(const std::string& a_Host, std::uint16_t a_Port);
|
|
||||||
|
|
||||||
friend class TcpListener;
|
|
||||||
};
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \typedef TcpSocket
|
|
||||||
*/
|
|
||||||
using TcpSocket = IOInterface<TcpTag>;
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,33 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <fstream>
|
|
||||||
#include <sp/io/IOInterface.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
struct FileTag {
|
|
||||||
enum OpenMode {
|
|
||||||
In = 1,
|
|
||||||
Out = 1 << 1,
|
|
||||||
};
|
|
||||||
};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
class IOInterface<FileTag> {
|
|
||||||
private:
|
|
||||||
std::unique_ptr<std::ifstream> m_FileInput;
|
|
||||||
std::unique_ptr<std::ofstream> m_FileOutput;
|
|
||||||
|
|
||||||
public:
|
|
||||||
IOInterface(const std::string& a_FilePath, unsigned int a_OpenMode);
|
|
||||||
IOInterface(IOInterface&& other);
|
|
||||||
|
|
||||||
DataBuffer Read(std::size_t a_Amount);
|
|
||||||
void Write(const sp::DataBuffer& a_Data);
|
|
||||||
};
|
|
||||||
|
|
||||||
using File = IOInterface<FileTag>;
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,58 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <memory>
|
|
||||||
#include <sp/common/DataBuffer.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
template <typename IOTag>
|
|
||||||
class IOInterface {
|
|
||||||
public:
|
|
||||||
DataBuffer Read(std::size_t a_Amount);
|
|
||||||
void Write(const DataBuffer& a_Data);
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TOption>
|
|
||||||
class MessageEncapsulator {
|
|
||||||
public:
|
|
||||||
static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOption& a_Option);
|
|
||||||
static DataBuffer Decapsulate(DataBuffer& a_Data, const TOption& a_Option);
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
|
||||||
class Stream {
|
|
||||||
protected:
|
|
||||||
MessageDispatcher m_Dispatcher;
|
|
||||||
IOInterface<IOTag> m_Interface;
|
|
||||||
std::tuple<TOptions...> m_Options;
|
|
||||||
|
|
||||||
using MessageBase = typename MessageDispatcher::MessageBaseType;
|
|
||||||
|
|
||||||
public:
|
|
||||||
Stream() {}
|
|
||||||
Stream(IOInterface<IOTag>&& a_Interface, TOptions&&... a_Options);
|
|
||||||
Stream(Stream&& a_Stream);
|
|
||||||
|
|
||||||
void RecieveMessages();
|
|
||||||
void SendMessage(const MessageBase& a_Message);
|
|
||||||
|
|
||||||
|
|
||||||
template <typename TOption>
|
|
||||||
TOption& GetOption() {
|
|
||||||
return std::get<TOption>(m_Options);
|
|
||||||
}
|
|
||||||
|
|
||||||
MessageDispatcher& GetDispatcher() {
|
|
||||||
return m_Dispatcher;
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOptions&... a_Options);
|
|
||||||
static DataBuffer Decapsulate(DataBuffer& a_Data, const TOptions&... a_Options);
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
|
|
||||||
#include <sp/io/IOInterfaceImpl.inl>
|
|
||||||
@@ -1,129 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/extensions/Compress.h>
|
|
||||||
#include <stdexcept>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
namespace details {
|
|
||||||
|
|
||||||
template <typename... TOptions>
|
|
||||||
struct MessageEncapsulatorPack {};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
struct MessageEncapsulatorPack<> {
|
|
||||||
static DataBuffer Encapsulate(const DataBuffer& a_Data) {
|
|
||||||
return a_Data;
|
|
||||||
}
|
|
||||||
static DataBuffer Decapsulate(DataBuffer& a_Data) {
|
|
||||||
return a_Data;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TOption, typename... TOptions>
|
|
||||||
struct MessageEncapsulatorPack<TOption, TOptions...> {
|
|
||||||
static DataBuffer Encapsulate(const DataBuffer& a_Data, const TOption& a_Option, const TOptions&... a_Options) {
|
|
||||||
DataBuffer data = io::MessageEncapsulator<TOption>::Encapsulate(a_Data, a_Option);
|
|
||||||
return MessageEncapsulatorPack<TOptions...>::Encapsulate(data, a_Options...);
|
|
||||||
}
|
|
||||||
static DataBuffer Decapsulate(DataBuffer& a_Data, const TOption& a_Option, const TOptions&... a_Options) {
|
|
||||||
DataBuffer data = io::MessageEncapsulator<TOption>::Decapsulate(a_Data, a_Option);
|
|
||||||
return MessageEncapsulatorPack<TOptions...>::Decapsulate(data, a_Options...);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace details
|
|
||||||
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
|
||||||
Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(IOInterface<IOTag>&& a_Interface, TOptions&&... a_Options) :
|
|
||||||
m_Interface(std::move(a_Interface)), m_Options(std::make_tuple<TOptions...>(std::move(a_Options)...)) {}
|
|
||||||
|
|
||||||
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
|
||||||
Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(
|
|
||||||
Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>&& 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, typename... TOptions>
|
|
||||||
void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::SendMessage(const MessageBase& a_Message) {
|
|
||||||
DataBuffer data = a_Message.Write();
|
|
||||||
TupleForEach([&data](const auto&... a_Options) {
|
|
||||||
data = Encapsulate(data, a_Options...);
|
|
||||||
}, m_Options);
|
|
||||||
DataBuffer finalData;
|
|
||||||
finalData.Reserve(data.GetSize() + sizeof(VarInt::MAX_VALUE));
|
|
||||||
finalData << VarInt{data.GetSize()} << data;
|
|
||||||
m_Interface.Write(finalData);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
|
||||||
void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::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);
|
|
||||||
|
|
||||||
TupleForEach([&buffer, lenghtValue](const auto&... a_Options) {
|
|
||||||
buffer = Decapsulate(buffer, a_Options...);
|
|
||||||
}, m_Options);
|
|
||||||
|
|
||||||
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, typename... TOptions>
|
|
||||||
DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Encapsulate(
|
|
||||||
const DataBuffer& a_Data, const TOptions&... a_Options) {
|
|
||||||
return details::MessageEncapsulatorPack<TOptions...>::Encapsulate(a_Data, a_Options...);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
|
||||||
DataBuffer Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Decapsulate(
|
|
||||||
DataBuffer& a_Data, const TOptions&... a_Options) {
|
|
||||||
return details::MessageEncapsulatorPack<TOptions...>::Decapsulate(a_Data, a_Options...);
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,23 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/io/IOInterface.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
struct MemoryTag {};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
class IOInterface<MemoryTag> {
|
|
||||||
private:
|
|
||||||
sp::DataBuffer m_VirtualIO;
|
|
||||||
|
|
||||||
public:
|
|
||||||
sp::DataBuffer Read(std::size_t a_Amount);
|
|
||||||
void Write(const sp::DataBuffer& a_Data);
|
|
||||||
};
|
|
||||||
|
|
||||||
using Memory = IOInterface<MemoryTag>;
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -70,32 +70,29 @@ class BitField {
|
|||||||
* \tparam ValueType the type of the value to store
|
* \tparam ValueType the type of the value to store
|
||||||
* \tparam IAlignment 0 means no alignment
|
* \tparam IAlignment 0 means no alignment
|
||||||
*/
|
*/
|
||||||
template <typename ValueType, std::size_t IAlignment>
|
template <typename ValueType, int IAlignment>
|
||||||
class Field {
|
class Field {
|
||||||
public:
|
public:
|
||||||
using StorageType = ValueType;
|
|
||||||
static constexpr std::size_t AlignmentValue = IAlignment;
|
|
||||||
|
|
||||||
// Provide an access to the stored value
|
// Provide an access to the stored value
|
||||||
StorageType& GetValue() {
|
ValueType& GetValue() {
|
||||||
return m_Value;
|
return m_Value;
|
||||||
}
|
}
|
||||||
|
|
||||||
const StorageType& GetValue() const {
|
const ValueType& GetValue() const {
|
||||||
return m_Value;
|
return m_Value;
|
||||||
}
|
}
|
||||||
|
|
||||||
Field& operator=(const StorageType& value) {
|
Field& operator=(const ValueType& value) {
|
||||||
m_Value = value;
|
m_Value = value;
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
constexpr std::size_t GetAlignment() const {
|
constexpr int GetAlignment() const {
|
||||||
return IAlignment;
|
return IAlignment;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
StorageType m_Value;
|
ValueType m_Value;
|
||||||
};
|
};
|
||||||
|
|
||||||
namespace details {
|
namespace details {
|
||||||
@@ -117,6 +114,6 @@ template <typename TField, typename... TFields>
|
|||||||
struct FieldsBuilder<TField, TFields...> {
|
struct FieldsBuilder<TField, TFields...> {
|
||||||
using Type = sp::tuple_cat_t<std::tuple<Field<TField, 0>>, typename FieldsBuilder<TFields...>::Type>;
|
using Type = sp::tuple_cat_t<std::tuple<Field<TField, 0>>, typename FieldsBuilder<TFields...>::Type>;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -13,6 +13,4 @@ namespace sp {
|
|||||||
template <typename TBase, typename... TOptions>
|
template <typename TBase, typename... TOptions>
|
||||||
class MessageBase : public details::MessageImplBuilder<TBase, TOptions...>::Type {};
|
class MessageBase : public details::MessageImplBuilder<TBase, TOptions...>::Type {};
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|
||||||
#include <sp/protocol/message/MessagePrinterImpl.h>
|
|
||||||
@@ -1,59 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \file MessageDispatcher.h
|
|
||||||
* \brief File containing the sp::MessageDispatcher class
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <map>
|
|
||||||
#include <memory>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \class MessageDispatcher
|
|
||||||
* \brief Class used to dispatch messages
|
|
||||||
*/
|
|
||||||
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
||||||
class MessageDispatcher {
|
|
||||||
private:
|
|
||||||
std::map<MessageIdType, std::vector<MessageHandler*>> m_Handlers;
|
|
||||||
|
|
||||||
public:
|
|
||||||
using MessageBaseType = MessageBase;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Constructor
|
|
||||||
*/
|
|
||||||
MessageDispatcher() {}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Dispatch a packet
|
|
||||||
* \param packet The packet to dispatch
|
|
||||||
*/
|
|
||||||
void Dispatch(const MessageBase& a_Message);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Register a packet handler
|
|
||||||
* \param type The packet type
|
|
||||||
* \param handler The packet handler
|
|
||||||
*/
|
|
||||||
void RegisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Unregister a packet handler
|
|
||||||
* \param type The packet type
|
|
||||||
* \param handler The packet handler
|
|
||||||
*/
|
|
||||||
void UnregisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* \brief Unregister a packet handler
|
|
||||||
* \param handler The packet handler
|
|
||||||
*/
|
|
||||||
void UnregisterHandler(MessageHandler* a_Handler);
|
|
||||||
};
|
|
||||||
|
|
||||||
#include <sp/protocol/message/MessageDispatcherImpl.inl>
|
|
||||||
|
|
||||||
} // namespace sp
|
|
||||||
@@ -5,10 +5,45 @@
|
|||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
|
||||||
#include <sp/protocol/message/ArrayFillerImpl.h>
|
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
|
namespace details {
|
||||||
|
|
||||||
|
template <typename TBase>
|
||||||
|
using ArrayType = std::vector<std::function<std::unique_ptr<TBase>(void)>>;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
template <typename TBase, typename... TMessages>
|
||||||
|
struct ArrayFiller {};
|
||||||
|
|
||||||
|
template <typename TBase, typename... TMessages>
|
||||||
|
struct ArrayFiller<TBase, std::tuple<TMessages...>> {
|
||||||
|
static ArrayType<TBase> ArrayCreate() {
|
||||||
|
ArrayType<TBase> array;
|
||||||
|
array.reserve(sizeof...(TMessages));
|
||||||
|
ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
|
||||||
|
return array;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename TBase, typename TMessage, typename... TMessages>
|
||||||
|
struct ArrayFiller<TBase, TMessage, TMessages...> {
|
||||||
|
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
||||||
|
ArrayFiller<TBase, TMessage>::ArrayAppend(array);
|
||||||
|
ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename TBase, typename TMessage>
|
||||||
|
struct ArrayFiller<TBase, TMessage> {
|
||||||
|
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
||||||
|
array.push_back([]() -> std::unique_ptr<TBase> { return std::make_unique<TMessage>(); });
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace details
|
||||||
|
|
||||||
template <typename TBase, typename TTMessages>
|
template <typename TBase, typename TTMessages>
|
||||||
class MessageFactory {
|
class MessageFactory {
|
||||||
public:
|
public:
|
||||||
@@ -16,7 +51,7 @@ class MessageFactory {
|
|||||||
|
|
||||||
MessageFactory() : m_Factory(details::ArrayFiller<TBase, TTMessages>::ArrayCreate()) {}
|
MessageFactory() : m_Factory(details::ArrayFiller<TBase, TTMessages>::ArrayCreate()) {}
|
||||||
|
|
||||||
std::unique_ptr<TBase> CreateMessage(IdType id) const {
|
std::unique_ptr<TBase> CreateMessage(IdType id) {
|
||||||
if (id >= m_Factory.size())
|
if (id >= m_Factory.size())
|
||||||
return nullptr;
|
return nullptr;
|
||||||
return m_Factory.at(id)();
|
return m_Factory.at(id)();
|
||||||
|
|||||||
@@ -1,55 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <map>
|
|
||||||
#include <sp/common/Templates.h>
|
|
||||||
#include <string>
|
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
template <typename T, std::enable_if_t<details::is_general_t<T>::value, bool> = true>
|
|
||||||
inline std::string PrintData(const T& a_Data);
|
|
||||||
|
|
||||||
template <typename T, std::enable_if_t<details::is_primitive<T>::value, bool> = true>
|
|
||||||
inline std::string PrintData(T a_Data) {
|
|
||||||
return std::to_string(a_Data);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
inline std::string PrintData(const std::string& a_Data) {
|
|
||||||
return "\"" + a_Data + "\"";
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename K, typename V>
|
|
||||||
std::string PrintData(const std::pair<K, V>& a_Data);
|
|
||||||
|
|
||||||
template <typename K, typename V>
|
|
||||||
std::string PrintData(const std::map<K, V>& a_Data);
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::string PrintData(const std::vector<T>& a_Data);
|
|
||||||
|
|
||||||
template <typename K, typename V>
|
|
||||||
std::string PrintData(const std::pair<K, V>& a_Data) {
|
|
||||||
return "{" + PrintData(a_Data.first) + " => " + PrintData(a_Data.second) + "}";
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename K, typename V>
|
|
||||||
std::string PrintData(const std::map<K, V>& a_Data) {
|
|
||||||
std::string result = "{";
|
|
||||||
for (const auto& pair : a_Data) {
|
|
||||||
result += PrintData(pair) + ", ";
|
|
||||||
}
|
|
||||||
return result.substr(0, result.size() - 2) + "}";
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
std::string PrintData(const std::vector<T>& a_Data) {
|
|
||||||
std::string result = "{";
|
|
||||||
for (const T& value : a_Data) {
|
|
||||||
result += PrintData(value) + ", ";
|
|
||||||
}
|
|
||||||
return result.substr(0, result.size() - 2) + "}";
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,39 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace details {
|
|
||||||
|
|
||||||
template <typename TBase>
|
|
||||||
using ArrayType = std::vector<std::function<std::unique_ptr<TBase>(void)>>;
|
|
||||||
|
|
||||||
|
|
||||||
template <typename TBase, typename... TMessages>
|
|
||||||
struct ArrayFiller {};
|
|
||||||
|
|
||||||
template <typename TBase, typename... TMessages>
|
|
||||||
struct ArrayFiller<TBase, std::tuple<TMessages...>> {
|
|
||||||
static ArrayType<TBase> ArrayCreate() {
|
|
||||||
ArrayType<TBase> array;
|
|
||||||
array.reserve(sizeof...(TMessages));
|
|
||||||
ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
|
|
||||||
return array;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TBase, typename TMessage, typename... TMessages>
|
|
||||||
struct ArrayFiller<TBase, TMessage, TMessages...> {
|
|
||||||
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
|
||||||
ArrayFiller<TBase, TMessage>::ArrayAppend(array);
|
|
||||||
ArrayFiller<TBase, TMessages...>::ArrayAppend(array);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TBase, typename TMessage>
|
|
||||||
struct ArrayFiller<TBase, TMessage> {
|
|
||||||
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
|
||||||
array.emplace_back([]() -> std::unique_ptr<TBase> { return std::make_unique<TMessage>(); });
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace details
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,36 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
||||||
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::RegisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler) {
|
|
||||||
assert(a_Handler);
|
|
||||||
auto found = std::find(m_Handlers[a_MessageType].begin(), m_Handlers[a_MessageType].end(), a_Handler);
|
|
||||||
if (found == m_Handlers[a_MessageType].end())
|
|
||||||
m_Handlers[a_MessageType].push_back(a_Handler);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
||||||
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::UnregisterHandler(MessageIdType a_MessageType, MessageHandler* a_Handler) {
|
|
||||||
auto found = std::find(m_Handlers[a_MessageType].begin(), m_Handlers[a_MessageType].end(), a_Handler);
|
|
||||||
if (found != m_Handlers[a_MessageType].end())
|
|
||||||
m_Handlers[a_MessageType].erase(found);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
||||||
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::UnregisterHandler(MessageHandler* a_Handler) {
|
|
||||||
for (auto& pair : m_Handlers) {
|
|
||||||
if (pair.second.empty())
|
|
||||||
continue;
|
|
||||||
|
|
||||||
MessageIdType type = pair.first;
|
|
||||||
|
|
||||||
pair.second.erase(std::remove(pair.second.begin(), pair.second.end(), a_Handler), pair.second.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename MessageIdType, typename MessageBase, typename MessageHandler>
|
|
||||||
void MessageDispatcher<MessageIdType, MessageBase, MessageHandler>::Dispatch(const MessageBase& a_Message) {
|
|
||||||
MessageIdType type = a_Message.GetId();
|
|
||||||
for (auto& handler : m_Handlers[type]) {
|
|
||||||
a_Message.Dispatch(*handler);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -37,12 +37,8 @@ struct MessageImplBuilder {
|
|||||||
static const bool HasValidImpl = InterfaceOptions::HasValid && ImplOptions::HasFieldsImpl;
|
static const bool HasValidImpl = InterfaceOptions::HasValid && ImplOptions::HasFieldsImpl;
|
||||||
using Base6 = typename MessageImplProcessValidFields<Base5, HasValidImpl>::Type;
|
using Base6 = typename MessageImplProcessValidFields<Base5, HasValidImpl>::Type;
|
||||||
|
|
||||||
// Provide ToStringImpl() if possible
|
|
||||||
static const bool HasToStringImpl = InterfaceOptions::HasToString;
|
|
||||||
using Base7 = typename MessageImplProcessToString<Base6, ImplOptions, HasToStringImpl>::Type;
|
|
||||||
|
|
||||||
// The last BaseN must be taken as final type.
|
// The last BaseN must be taken as final type.
|
||||||
using Type = Base7;
|
using Type = Base6;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ namespace sp {
|
|||||||
namespace option {
|
namespace option {
|
||||||
|
|
||||||
// Provide static numeric ID, to facilitate implementation of GetIdImpl()
|
// Provide static numeric ID, to facilitate implementation of GetIdImpl()
|
||||||
template <std::uintmax_t TId>
|
template <std::intmax_t TId>
|
||||||
struct StaticNumIdImpl {};
|
struct StaticNumIdImpl {};
|
||||||
|
|
||||||
// Facilitate implementation of DispatchImpl()
|
// Facilitate implementation of DispatchImpl()
|
||||||
@@ -16,11 +16,6 @@ struct DispatchImpl {};
|
|||||||
template <typename TFields>
|
template <typename TFields>
|
||||||
struct FieldsImpl {};
|
struct FieldsImpl {};
|
||||||
|
|
||||||
// Print fields of the message, facilitate implementation of
|
|
||||||
// ToStringImpl()
|
|
||||||
template <typename TActual>
|
|
||||||
struct ToStringImpl {};
|
|
||||||
|
|
||||||
} // namespace option
|
} // namespace option
|
||||||
|
|
||||||
|
|
||||||
@@ -44,10 +39,10 @@ struct MessageImplParsedOptions<> {
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
template <std::uintmax_t TId, typename... TOptions>
|
template <std::intmax_t TId, typename... TOptions>
|
||||||
struct MessageImplParsedOptions<option::StaticNumIdImpl<TId>, TOptions...> : public MessageImplParsedOptions<TOptions...> {
|
struct MessageImplParsedOptions<option::StaticNumIdImpl<TId>, TOptions...> : public MessageImplParsedOptions<TOptions...> {
|
||||||
static const bool HasStaticNumIdImpl = true;
|
static const bool HasStaticNumIdImpl = true;
|
||||||
static const std::uintmax_t MsgId = TId;
|
static const std::intmax_t MsgId = TId;
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename TActual, typename... TOptions>
|
template <typename TActual, typename... TOptions>
|
||||||
|
|||||||
@@ -4,20 +4,6 @@ namespace sp {
|
|||||||
namespace details {
|
namespace details {
|
||||||
|
|
||||||
|
|
||||||
// ToString impl
|
|
||||||
template <typename TBase, typename ParsedImplOptions, bool TImplement>
|
|
||||||
struct MessageImplProcessToString;
|
|
||||||
|
|
||||||
template <typename TBase, typename ParsedImplOptions>
|
|
||||||
struct MessageImplProcessToString<TBase, ParsedImplOptions, true> {
|
|
||||||
using Type = MessageImplToStringBase<TBase, typename ParsedImplOptions::ActualMessage>;
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TBase, typename ParsedImplOptions>
|
|
||||||
struct MessageImplProcessToString<TBase, ParsedImplOptions, false> {
|
|
||||||
using Type = TBase;
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
// id impl
|
// id impl
|
||||||
template <typename TBase, typename ParsedImplOptions, bool TImplement>
|
template <typename TBase, typename ParsedImplOptions, bool TImplement>
|
||||||
|
|||||||
@@ -33,11 +33,8 @@ struct MessageInterfaceBuilder {
|
|||||||
// add write id functionality if write id and write was provided
|
// add write id functionality if write id and write was provided
|
||||||
using Base7 = typename MessageInterfaceProcessWriteId<Base6, ParsedOptions::HasWriteId && ParsedOptions::HasWriteOperations>::Type;
|
using Base7 = typename MessageInterfaceProcessWriteId<Base6, ParsedOptions::HasWriteId && ParsedOptions::HasWriteOperations>::Type;
|
||||||
|
|
||||||
// add ToString() if HasToString was provided
|
// The last Base7 must be taken as final type.
|
||||||
using Base8 = typename MessageInterfaceProcessToString<Base7, ParsedOptions::HasToString>::Type;
|
using Type = Base7;
|
||||||
|
|
||||||
// The last Base8 must be taken as final type.
|
|
||||||
using Type = Base8;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
|
|||||||
@@ -101,19 +101,5 @@ struct MessageInterfaceProcessWriteId<TBase, false> {
|
|||||||
using Type = TBase;
|
using Type = TBase;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Build to string
|
|
||||||
template <typename TBase, bool THasToString>
|
|
||||||
struct MessageInterfaceProcessToString;
|
|
||||||
|
|
||||||
template <typename TBase>
|
|
||||||
struct MessageInterfaceProcessToString<TBase, true> {
|
|
||||||
using Type = MessageInterfaceToStringBase<TBase>;
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TBase>
|
|
||||||
struct MessageInterfaceProcessToString<TBase, false> {
|
|
||||||
using Type = TBase;
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
@@ -30,7 +30,7 @@ class MessageInterfaceBigEndian : public TBase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void WriteData(T value, DataBuffer& buffer) const {
|
void WriteData(T value, DataBuffer& buffer) {
|
||||||
ToNetwork(value);
|
ToNetwork(value);
|
||||||
buffer << value;
|
buffer << value;
|
||||||
}
|
}
|
||||||
@@ -71,19 +71,12 @@ class MessageInterfaceReadBase : public TBase {
|
|||||||
template <typename TBase>
|
template <typename TBase>
|
||||||
class MessageInterfaceWriteBase : public TBase {
|
class MessageInterfaceWriteBase : public TBase {
|
||||||
public:
|
public:
|
||||||
void Write(DataBuffer& buffer) const {
|
void Write(DataBuffer& buffer) {
|
||||||
WriteImpl(buffer);
|
WriteImpl(buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
// helper
|
|
||||||
DataBuffer Write() const {
|
|
||||||
DataBuffer buffer;
|
|
||||||
this->Write(buffer);
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
virtual void WriteImpl(DataBuffer& buffer) const = 0;
|
virtual void WriteImpl(DataBuffer& buffer) = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Handler functionality chunk
|
// Handler functionality chunk
|
||||||
@@ -92,12 +85,12 @@ class MessageInterfaceHandlerBase : public TBase {
|
|||||||
public:
|
public:
|
||||||
using HandlerType = typename THandler::HandlerT;
|
using HandlerType = typename THandler::HandlerT;
|
||||||
|
|
||||||
void Dispatch(HandlerType& handler) const {
|
void Dispatch(HandlerType& handler) {
|
||||||
DispatchImpl(handler);
|
DispatchImpl(handler);
|
||||||
}
|
}
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
virtual void DispatchImpl(HandlerType& handler) const = 0;
|
virtual void DispatchImpl(HandlerType& handler) = 0;
|
||||||
};
|
};
|
||||||
|
|
||||||
// Validity functionality chunk
|
// Validity functionality chunk
|
||||||
@@ -116,29 +109,10 @@ class MessageInterfaceValidityBase : public TBase {
|
|||||||
template <typename TBase>
|
template <typename TBase>
|
||||||
class MessageInterfaceWriteIdBase : public TBase {
|
class MessageInterfaceWriteIdBase : public TBase {
|
||||||
public:
|
public:
|
||||||
void Write(DataBuffer& buffer) const {
|
void Write(DataBuffer& buffer) {
|
||||||
buffer << VarInt{this->GetId()};
|
this->WriteData(this->GetId(), buffer);
|
||||||
this->WriteImpl(buffer);
|
this->WriteImpl(buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
// helper
|
|
||||||
DataBuffer Write() const {
|
|
||||||
DataBuffer buffer;
|
|
||||||
this->Write(buffer);
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Debug print functionality chunk
|
|
||||||
template <typename TBase>
|
|
||||||
class MessageInterfaceToStringBase : public TBase {
|
|
||||||
public:
|
|
||||||
std::string ToString() const {
|
|
||||||
return ToStringImpl();
|
|
||||||
}
|
|
||||||
|
|
||||||
protected:
|
|
||||||
virtual std::string ToStringImpl() const = 0;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
|
|||||||
@@ -16,7 +16,6 @@ struct MessageInterfaceParsedOptions<> {
|
|||||||
static const bool HasWriteId = false;
|
static const bool HasWriteId = false;
|
||||||
static const bool HasHandler = false;
|
static const bool HasHandler = false;
|
||||||
static const bool HasValid = false;
|
static const bool HasValid = false;
|
||||||
static const bool HasToString = false;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@@ -60,10 +59,5 @@ struct MessageInterfaceParsedOptions<option::ValidCheckInterface, TOptions...> :
|
|||||||
static const bool HasValid = true;
|
static const bool HasValid = true;
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename... TOptions>
|
|
||||||
struct MessageInterfaceParsedOptions<option::DebugPrint, TOptions...> : public MessageInterfaceParsedOptions<TOptions...> {
|
|
||||||
static const bool HasToString = true;
|
|
||||||
};
|
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -24,9 +24,6 @@ struct LittleEndian {};
|
|||||||
// Include validity check in public interface
|
// Include validity check in public interface
|
||||||
struct ValidCheckInterface {};
|
struct ValidCheckInterface {};
|
||||||
|
|
||||||
// Add a ToString() method containing fields
|
|
||||||
struct DebugPrint {};
|
|
||||||
|
|
||||||
// Define handler class
|
// Define handler class
|
||||||
template <typename T>
|
template <typename T>
|
||||||
struct Handler {
|
struct Handler {
|
||||||
|
|||||||
@@ -1,71 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <sp/common/Reflection.h>
|
|
||||||
#include <sp/common/Templates.h>
|
|
||||||
#include <sp/protocol/MessagePrinter.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace details {
|
|
||||||
|
|
||||||
template <typename... TOptions>
|
|
||||||
struct IdPrinter {};
|
|
||||||
|
|
||||||
template <>
|
|
||||||
struct IdPrinter<> {
|
|
||||||
static std::string PrintMessageId() {
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename TOption, typename... TOptions>
|
|
||||||
struct IdPrinter<TOption, TOptions...> {
|
|
||||||
static std::string PrintMessageId() {
|
|
||||||
return IdPrinter<TOptions...>::PrintMessageId();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename... TOptions, std::uintmax_t TId>
|
|
||||||
struct IdPrinter<option::StaticNumIdImpl<TId>, TOptions...> {
|
|
||||||
static std::string PrintMessageId() {
|
|
||||||
return "(Id: " + std::to_string(TId) + ")";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename... TFields>
|
|
||||||
std::string PrintFields(const std::tuple<TFields...>& a_Fields);
|
|
||||||
|
|
||||||
template <typename TField, unsigned int IAlignment>
|
|
||||||
struct FieldPrinter {
|
|
||||||
static std::string PrintField(const sp::Field<TField, IAlignment>& a_Field) {
|
|
||||||
return Reflector<TField>::GetClassName() + "=" + PrintData(a_Field.GetValue());
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <unsigned int IAlignment, typename TContainer, typename... TFields>
|
|
||||||
struct FieldPrinter<BitField<TContainer, TFields...>, IAlignment> {
|
|
||||||
static std::string PrintField(const Field<BitField<TContainer, TFields...>, IAlignment>& a_Field) {
|
|
||||||
return "BitField<" + Reflector<TContainer>::GetClassName() + ">[" + PrintFields(a_Field.GetValue().GetFields()) + "]";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
template <typename... TFields>
|
|
||||||
std::string PrintFields(const std::tuple<TFields...>& a_Fields) {
|
|
||||||
std::string concat;
|
|
||||||
TupleForEach(
|
|
||||||
[&concat](const auto& a_Field) {
|
|
||||||
using TField = typename std::decay<decltype(a_Field)>::type;
|
|
||||||
constexpr std::size_t alignment = TField::AlignmentValue;
|
|
||||||
concat += FieldPrinter<typename TField::StorageType, alignment>::PrintField(a_Field) + ", ";
|
|
||||||
},
|
|
||||||
a_Fields);
|
|
||||||
return concat.substr(0, concat.size() - 2);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <typename TBase, typename... TOptions>
|
|
||||||
std::string PrintMessage(const MessageBase<TBase, TOptions...>& a_Message) {
|
|
||||||
return sp::GetBasicClassName(a_Message) + sp::details::IdPrinter<TOptions...>::PrintMessageId() + "[" +
|
|
||||||
sp::details::PrintFields(a_Message.GetFields()) + "]";
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace details
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,24 +1,8 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
template <typename TBase, typename... TOptions>
|
|
||||||
class MessageBase;
|
|
||||||
|
|
||||||
namespace details {
|
namespace details {
|
||||||
|
|
||||||
template <typename TBase, typename... TOptions>
|
|
||||||
std::string PrintMessage(const MessageBase<TBase, TOptions...>& a_Message);
|
|
||||||
|
|
||||||
// ID information chunk
|
|
||||||
template <typename TBase, typename TActual>
|
|
||||||
class MessageImplToStringBase : public TBase {
|
|
||||||
protected:
|
|
||||||
virtual std::string ToStringImpl() const override {
|
|
||||||
return PrintMessage(static_cast<const TActual&>(*this));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// ID information chunk
|
// ID information chunk
|
||||||
@@ -34,8 +18,6 @@ class MessageImplStaticNumIdBase : public TBase {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// Dispatch implementation chunk
|
// Dispatch implementation chunk
|
||||||
template <typename TBase, typename TActual>
|
template <typename TBase, typename TActual>
|
||||||
class MessageImplDispatchBase : public TBase {
|
class MessageImplDispatchBase : public TBase {
|
||||||
@@ -44,8 +26,8 @@ class MessageImplDispatchBase : public TBase {
|
|||||||
using Handler = typename TBase::HandlerType;
|
using Handler = typename TBase::HandlerType;
|
||||||
|
|
||||||
protected:
|
protected:
|
||||||
virtual void DispatchImpl(Handler& handler) const override {
|
virtual void DispatchImpl(Handler& handler) override {
|
||||||
handler.Handle(static_cast<const TActual&>(*this));
|
handler.Handle(static_cast<TActual&>(*this));
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -99,7 +81,7 @@ class MessageImplFieldsReadBase : public TBase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// reading field in bitfield
|
// reading field in bitfield
|
||||||
template <typename TFieldType, typename TField, std::size_t IAlignment>
|
template <typename TFieldType, typename TField, int IAlignment>
|
||||||
void ReadField(Field<TField, IAlignment>& field, TFieldType data, std::size_t offset) {
|
void ReadField(Field<TField, IAlignment>& field, TFieldType data, std::size_t offset) {
|
||||||
static constexpr std::size_t TotalBitCount = sizeof(TFieldType) * 8;
|
static constexpr std::size_t TotalBitCount = sizeof(TFieldType) * 8;
|
||||||
// we suppose that the first element is at the highest bits
|
// we suppose that the first element is at the highest bits
|
||||||
@@ -131,13 +113,13 @@ class MessageImplFieldsWriteBase : public TBase {
|
|||||||
private:
|
private:
|
||||||
// normal writing
|
// normal writing
|
||||||
template <typename TField>
|
template <typename TField>
|
||||||
void WriteField(const Field<TField, 0>& field, DataBuffer& buffer) const {
|
void WriteField(Field<TField, 0>& field, DataBuffer& buffer) {
|
||||||
this->WriteData(field.GetValue(), buffer);
|
this->WriteData(field.GetValue(), buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
// writing field in bitfield
|
// writing field in bitfield
|
||||||
template <typename TFieldType, typename TField, std::size_t IAlignment>
|
template <typename TFieldType, typename TField, int IAlignment>
|
||||||
void WriteField(const Field<TField, IAlignment>& field, TFieldType& data, std::size_t offset) const {
|
void WriteField(Field<TField, IAlignment>& field, TFieldType& data, std::size_t offset) {
|
||||||
static constexpr std::size_t TotalBitCount = sizeof(TFieldType) * 8;
|
static constexpr std::size_t TotalBitCount = sizeof(TFieldType) * 8;
|
||||||
// we suppose that the first element is at the highest bits
|
// we suppose that the first element is at the highest bits
|
||||||
data |= (field.GetValue() & ((1 << IAlignment) - 1)) << TotalBitCount - IAlignment - offset;
|
data |= (field.GetValue() & ((1 << IAlignment) - 1)) << TotalBitCount - IAlignment - offset;
|
||||||
@@ -145,7 +127,7 @@ class MessageImplFieldsWriteBase : public TBase {
|
|||||||
|
|
||||||
// writing bitfield
|
// writing bitfield
|
||||||
template <typename TContainer, typename TFirst, typename... TFields>
|
template <typename TContainer, typename TFirst, typename... TFields>
|
||||||
void WriteField(const Field<BitField<TContainer, TFirst, TFields...>, 0>& field, DataBuffer& buffer) const {
|
void WriteField(Field<BitField<TContainer, TFirst, TFields...>, 0>& field, DataBuffer& buffer) {
|
||||||
TContainer data = 0;
|
TContainer data = 0;
|
||||||
std::size_t offset = 0;
|
std::size_t offset = 0;
|
||||||
TupleForEach(
|
TupleForEach(
|
||||||
@@ -157,9 +139,9 @@ class MessageImplFieldsWriteBase : public TBase {
|
|||||||
this->WriteData(data, buffer);
|
this->WriteData(data, buffer);
|
||||||
}
|
}
|
||||||
|
|
||||||
void WriteImpl(DataBuffer& buffer) const override {
|
void WriteImpl(DataBuffer& buffer) override {
|
||||||
auto& allFields = this->GetFields();
|
auto& allFields = this->GetFields();
|
||||||
TupleForEach([&buffer, this](const auto& field) { this->WriteField(field, buffer); }, allFields);
|
TupleForEach([&buffer, this](auto& field) { this->WriteField(field, buffer); }, allFields);
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -3,13 +3,9 @@
|
|||||||
#include <examples/PacketExample.h>
|
#include <examples/PacketExample.h>
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
|
||||||
#include <sp/default/DefaultPacketDispatcher.h>
|
|
||||||
#include <sp/extensions/Extensions.h>
|
|
||||||
|
|
||||||
class KeepAliveHandler : public sp::PacketHandler {
|
class KeepAliveHandler : public sp::PacketHandler {
|
||||||
void Handle(const KeepAlivePacket& packet) {
|
void Handle(const KeepAlivePacket& packet) {
|
||||||
std::cout << "KeepAlive handled !!\n";
|
std::cout << "KeepAlive handled !\n";
|
||||||
std::cout << packet.ToString() << std::endl;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void Handle(const DisconnectPacket& packet) {
|
void Handle(const DisconnectPacket& packet) {
|
||||||
@@ -22,16 +18,15 @@ class KeepAliveHandler : public sp::PacketHandler {
|
|||||||
};
|
};
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
auto upgradeTower = std::make_unique<UpgradeTowerPacket>(std::make_tuple(666, 9), 789);
|
auto upgradeTower = std::make_unique<UpgradeTowerPacket>(std::make_tuple(666, 9));
|
||||||
auto keepAlive = std::make_unique<KeepAlivePacket>(6969);
|
|
||||||
|
|
||||||
sp::PacketMessage* msg = upgradeTower.get();
|
sp::PacketMessage* msg = upgradeTower.get();
|
||||||
|
|
||||||
auto handler = std::make_shared<KeepAliveHandler>();
|
KeepAliveHandler handler;
|
||||||
msg->Dispatch(*handler);
|
msg->Dispatch(handler);
|
||||||
keepAlive->Dispatch(*handler);
|
|
||||||
|
|
||||||
sp::DataBuffer buffer = msg->Write();
|
sp::DataBuffer buffer;
|
||||||
|
msg->Write(buffer);
|
||||||
|
|
||||||
std::uint8_t msgId;
|
std::uint8_t msgId;
|
||||||
buffer >> msgId;
|
buffer >> msgId;
|
||||||
@@ -39,7 +34,7 @@ int main() {
|
|||||||
auto upgradeTower2 = std::make_unique<UpgradeTowerPacket>();
|
auto upgradeTower2 = std::make_unique<UpgradeTowerPacket>();
|
||||||
upgradeTower2->Read(buffer);
|
upgradeTower2->Read(buffer);
|
||||||
|
|
||||||
std::cout << "Test : " << msg->ToString() << "\n";
|
std::cout << "Test : " << (unsigned) upgradeTower2->GetTowerId() << "\n";
|
||||||
|
|
||||||
sp::PacketFactory factory;
|
sp::PacketFactory factory;
|
||||||
auto packet = factory.CreateMessage(msgId);
|
auto packet = factory.CreateMessage(msgId);
|
||||||
@@ -48,13 +43,7 @@ int main() {
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
std::cout << (unsigned)packet->GetId() << std::endl;
|
std::cout << (unsigned)packet->GetId() << std::endl;
|
||||||
packet->Dispatch(*handler);
|
packet->Dispatch(handler);
|
||||||
|
|
||||||
sp::PacketDispatcher dispatcher;
|
|
||||||
dispatcher.RegisterHandler(PacketId::KeepAlive, handler.get());
|
|
||||||
dispatcher.Dispatch(*packet);
|
|
||||||
dispatcher.UnregisterHandler(PacketId::KeepAlive, handler.get());
|
|
||||||
dispatcher.UnregisterHandler(handler.get());
|
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@@ -1,74 +0,0 @@
|
|||||||
#include <sp/common/ByteSwapping.h>
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
#include <winsock2.h>
|
|
||||||
#else
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <algorithm>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
void SwapBytes(T& value) {
|
|
||||||
char* ptr = reinterpret_cast<char*>(&value);
|
|
||||||
std::reverse(ptr, ptr + sizeof(T));
|
|
||||||
}
|
|
||||||
|
|
||||||
bool IsSystemBigEndian() {
|
|
||||||
static constexpr std::uint16_t test = 10;
|
|
||||||
static const bool isBigEndian = reinterpret_cast<const std::uint8_t*>(&test)[1] == 10;
|
|
||||||
return isBigEndian;
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void ToNetwork<std::uint16_t>(std::uint16_t& value) {
|
|
||||||
value = htons(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void ToNetwork<std::uint32_t>(std::uint32_t& value) {
|
|
||||||
value = htonl(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void ToNetwork<std::uint64_t>(std::uint64_t& value) {
|
|
||||||
if (IsSystemBigEndian())
|
|
||||||
return;
|
|
||||||
SwapBytes(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void FromNetwork<std::uint16_t>(std::uint16_t& value) {
|
|
||||||
value = ntohs(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void FromNetwork<std::uint32_t>(std::uint32_t& value) {
|
|
||||||
value = ntohl(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void FromNetwork<std::uint64_t>(std::uint64_t& value) {
|
|
||||||
if (IsSystemBigEndian())
|
|
||||||
return;
|
|
||||||
SwapBytes(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void TrySwapBytes<std::uint16_t>(std::uint16_t& value) {
|
|
||||||
SwapBytes(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void TrySwapBytes<std::uint32_t>(std::uint32_t& value) {
|
|
||||||
SwapBytes(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
template <>
|
|
||||||
void TrySwapBytes<std::uint64_t>(std::uint64_t& value) {
|
|
||||||
SwapBytes(value);
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace sp
|
|
||||||
@@ -8,8 +8,6 @@ namespace sp {
|
|||||||
|
|
||||||
DataBuffer::DataBuffer() : m_ReadOffset(0) {}
|
DataBuffer::DataBuffer() : m_ReadOffset(0) {}
|
||||||
|
|
||||||
DataBuffer::DataBuffer(std::size_t a_InitialSize) : m_Buffer(a_InitialSize), m_ReadOffset(0) {}
|
|
||||||
|
|
||||||
DataBuffer::DataBuffer(const DataBuffer& other) : m_Buffer(other.m_Buffer), m_ReadOffset(other.m_ReadOffset) {}
|
DataBuffer::DataBuffer(const DataBuffer& other) : m_Buffer(other.m_Buffer), m_ReadOffset(other.m_ReadOffset) {}
|
||||||
|
|
||||||
DataBuffer::DataBuffer(DataBuffer&& other) : m_Buffer(std::move(other.m_Buffer)), m_ReadOffset(std::move(other.m_ReadOffset)) {}
|
DataBuffer::DataBuffer(DataBuffer&& other) : m_Buffer(std::move(other.m_Buffer)), m_ReadOffset(std::move(other.m_ReadOffset)) {}
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
#include <sp/common/VarInt.h>
|
#include <sp/common/VarInt.h>
|
||||||
|
|
||||||
#include <sp/common/DataBuffer.h>
|
|
||||||
#include <stdexcept>
|
#include <stdexcept>
|
||||||
|
#include <sp/common/DataBuffer.h>
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
|
|||||||
@@ -1,89 +0,0 @@
|
|||||||
#include <sp/extensions/Compress.h>
|
|
||||||
|
|
||||||
#include <cassert>
|
|
||||||
#include <sp/common/VarInt.h>
|
|
||||||
#include <zlib.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace zlib {
|
|
||||||
|
|
||||||
static DataBuffer Inflate(const std::uint8_t* source, std::size_t size, std::size_t uncompressedSize) {
|
|
||||||
DataBuffer result;
|
|
||||||
result.Resize(uncompressedSize);
|
|
||||||
|
|
||||||
uncompress(static_cast<Bytef*>(result.data()), reinterpret_cast<uLongf*>(&uncompressedSize), static_cast<const Bytef*>(source),
|
|
||||||
static_cast<uLong>(size));
|
|
||||||
|
|
||||||
assert(result.GetSize() == uncompressedSize);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
static DataBuffer Deflate(const std::uint8_t* source, std::size_t size) {
|
|
||||||
DataBuffer result;
|
|
||||||
uLongf compressedSize = size;
|
|
||||||
|
|
||||||
result.Resize(size); // Resize for the compressed data to fit into
|
|
||||||
compress(static_cast<Bytef*>(result.data()), &compressedSize, static_cast<const Bytef*>(source), static_cast<uLong>(size));
|
|
||||||
result.Resize(compressedSize); // Resize to cut useless data
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
DataBuffer Compress(const DataBuffer& buffer, std::size_t a_CompressionThreshold) {
|
|
||||||
DataBuffer packet;
|
|
||||||
|
|
||||||
if (buffer.GetSize() < a_CompressionThreshold) {
|
|
||||||
// Don't compress since it's a small packet
|
|
||||||
packet << VarInt{0};
|
|
||||||
packet << buffer;
|
|
||||||
return packet;
|
|
||||||
}
|
|
||||||
|
|
||||||
DataBuffer compressedData = Deflate(buffer.data(), buffer.GetSize());
|
|
||||||
VarInt uncompressedDataLength = buffer.GetSize();
|
|
||||||
|
|
||||||
if (compressedData.GetSize() >= buffer.GetSize()) {
|
|
||||||
// the compression is overkill so we don't send the compressed buffer
|
|
||||||
packet << VarInt{0};
|
|
||||||
packet << buffer;
|
|
||||||
} else {
|
|
||||||
packet << uncompressedDataLength;
|
|
||||||
packet << compressedData;
|
|
||||||
}
|
|
||||||
|
|
||||||
return packet;
|
|
||||||
}
|
|
||||||
|
|
||||||
DataBuffer Decompress(DataBuffer& buffer, std::uint64_t packetLength) {
|
|
||||||
VarInt uncompressedLength;
|
|
||||||
buffer >> uncompressedLength;
|
|
||||||
|
|
||||||
std::uint64_t compressedLength = packetLength - uncompressedLength.GetSerializedLength();
|
|
||||||
|
|
||||||
if (uncompressedLength.GetValue() == 0) {
|
|
||||||
// Data already uncompressed. Nothing to do
|
|
||||||
DataBuffer ret;
|
|
||||||
buffer.ReadSome(ret, compressedLength);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
assert(buffer.GetReadOffset() + compressedLength <= buffer.GetSize());
|
|
||||||
|
|
||||||
return Inflate(buffer.data() + buffer.GetReadOffset(), compressedLength, uncompressedLength.GetValue());
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace zlib
|
|
||||||
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
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 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) {
|
|
||||||
return zlib::Decompress(a_Data, a_Data.GetSize());
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,114 +0,0 @@
|
|||||||
#include <sp/extensions/tcp/TcpListener.h>
|
|
||||||
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
// Windows
|
|
||||||
|
|
||||||
#include <winsock2.h>
|
|
||||||
#include <ws2tcpip.h>
|
|
||||||
|
|
||||||
#define ioctl ioctlsocket
|
|
||||||
#define WOULDBLOCK WSAEWOULDBLOCK
|
|
||||||
#define MSG_DONTWAIT 0
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
// Linux/Unix
|
|
||||||
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
#include <netdb.h>
|
|
||||||
#include <netinet/in.h>
|
|
||||||
#include <sys/ioctl.h>
|
|
||||||
#include <sys/socket.h>
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
|
|
||||||
#define closesocket close
|
|
||||||
#define WOULDBLOCK EWOULDBLOCK
|
|
||||||
#define SD_BOTH SHUT_RDWR
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef INVALID_SOCKET
|
|
||||||
#define INVALID_SOCKET -1
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
TcpListener::TcpListener(std::uint16_t a_Port, int a_MaxConnexions) {
|
|
||||||
if ((m_Handle = static_cast<SocketHandle>(socket(AF_INET, SOCK_STREAM, IPPROTO_TCP))) < 0) {
|
|
||||||
throw SocketError("Failed to create server socket");
|
|
||||||
}
|
|
||||||
|
|
||||||
struct sockaddr_in address;
|
|
||||||
address.sin_family = AF_INET;
|
|
||||||
address.sin_addr.s_addr = INADDR_ANY;
|
|
||||||
address.sin_port = htons(a_Port);
|
|
||||||
|
|
||||||
if (bind(m_Handle, reinterpret_cast<sockaddr*>(&address), sizeof(address)) < 0)
|
|
||||||
throw SocketError("Failed to create server socket");
|
|
||||||
|
|
||||||
if (listen(m_Handle, a_MaxConnexions) < 0)
|
|
||||||
throw SocketError("Failed to create server socket");
|
|
||||||
|
|
||||||
socklen_t len = sizeof(address);
|
|
||||||
if (getsockname(m_Handle, reinterpret_cast<sockaddr*>(&address), &len) < 0)
|
|
||||||
throw SocketError("Failed to create server socket");
|
|
||||||
|
|
||||||
m_Port = ntohs(address.sin_port);
|
|
||||||
m_MaxConnections = a_MaxConnexions;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
TcpListener::~TcpListener() {
|
|
||||||
Close();
|
|
||||||
}
|
|
||||||
|
|
||||||
std::unique_ptr<TcpSocket> TcpListener::Accept() {
|
|
||||||
sockaddr remoteAddress;
|
|
||||||
int addrlen = sizeof(remoteAddress);
|
|
||||||
|
|
||||||
auto newSocket = std::make_unique<TcpSocket>();
|
|
||||||
|
|
||||||
newSocket->m_Handle = static_cast<SocketHandle>(
|
|
||||||
accept(m_Handle, reinterpret_cast<sockaddr*>(&remoteAddress), reinterpret_cast<socklen_t*>(&addrlen)));
|
|
||||||
|
|
||||||
if (newSocket->m_Handle < 0)
|
|
||||||
return nullptr;
|
|
||||||
|
|
||||||
newSocket->m_Status = TcpSocket::Status::Connected;
|
|
||||||
return newSocket;
|
|
||||||
}
|
|
||||||
|
|
||||||
void TcpListener::Close() {
|
|
||||||
if (m_Handle > 0) {
|
|
||||||
closesocket(m_Handle);
|
|
||||||
shutdown(m_Handle, SD_BOTH);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool TcpListener::SetBlocking(bool a_Blocking) {
|
|
||||||
unsigned long mode = !a_Blocking;
|
|
||||||
|
|
||||||
if (ioctl(m_Handle, FIONBIO, &mode) < 0) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::uint16_t TcpListener::GetListeningPort() const {
|
|
||||||
return m_Port;
|
|
||||||
}
|
|
||||||
|
|
||||||
int TcpListener::GetMaximumConnections() const {
|
|
||||||
return m_MaxConnections;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,164 +0,0 @@
|
|||||||
#include <sp/extensions/tcp/TcpSocket.h>
|
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
|
|
||||||
// Windows
|
|
||||||
|
|
||||||
#include <winsock2.h>
|
|
||||||
#include <ws2tcpip.h>
|
|
||||||
|
|
||||||
#define ioctl ioctlsocket
|
|
||||||
#define WOULDBLOCK WSAEWOULDBLOCK
|
|
||||||
#define MSG_DONTWAIT 0
|
|
||||||
|
|
||||||
#else
|
|
||||||
|
|
||||||
// Linux/Unix
|
|
||||||
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
#include <netdb.h>
|
|
||||||
#include <netinet/in.h>
|
|
||||||
#include <sys/ioctl.h>
|
|
||||||
#include <sys/socket.h>
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
|
|
||||||
#define closesocket close
|
|
||||||
#define WOULDBLOCK EWOULDBLOCK
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#ifndef INVALID_SOCKET
|
|
||||||
#define INVALID_SOCKET -1
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
TcpSocket::IOInterface() : m_Handle(static_cast<SocketHandle>(INVALID_SOCKET)), m_Status(Status::Disconnected) {}
|
|
||||||
|
|
||||||
TcpSocket::IOInterface(const std::string& a_Host, std::uint16_t a_Port) : IOInterface() {
|
|
||||||
Connect(a_Host, a_Port);
|
|
||||||
}
|
|
||||||
|
|
||||||
TcpSocket::IOInterface(IOInterface&& a_Other) {
|
|
||||||
std::swap(m_Handle, a_Other.m_Handle);
|
|
||||||
std::swap(m_Status, a_Other.m_Status);
|
|
||||||
}
|
|
||||||
|
|
||||||
TcpSocket::~IOInterface() {}
|
|
||||||
|
|
||||||
void TcpSocket::Connect(const std::string& a_Host, std::uint16_t a_Port) {
|
|
||||||
struct addrinfo hints {};
|
|
||||||
|
|
||||||
struct addrinfo* result = nullptr;
|
|
||||||
|
|
||||||
hints.ai_family = AF_INET;
|
|
||||||
hints.ai_socktype = SOCK_STREAM;
|
|
||||||
hints.ai_protocol = IPPROTO_TCP;
|
|
||||||
|
|
||||||
m_Status = Status::Error;
|
|
||||||
|
|
||||||
if (getaddrinfo(a_Host.c_str(), std::to_string(static_cast<int>(a_Port)).c_str(), &hints, &result) != 0) {
|
|
||||||
throw SocketError("Failed to get address info");
|
|
||||||
}
|
|
||||||
|
|
||||||
m_Handle = static_cast<SocketHandle>(socket(AF_INET, SOCK_STREAM, IPPROTO_TCP));
|
|
||||||
if (m_Handle < 0) {
|
|
||||||
throw SocketError("Failed to create socket");
|
|
||||||
}
|
|
||||||
|
|
||||||
struct addrinfo* ptr = nullptr;
|
|
||||||
for (ptr = result; ptr != nullptr; ptr = ptr->ai_next) {
|
|
||||||
struct sockaddr* sockaddr = ptr->ai_addr;
|
|
||||||
if (connect(m_Handle, sockaddr, sizeof(sockaddr_in)) == 0) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
freeaddrinfo(result);
|
|
||||||
|
|
||||||
if (!ptr) {
|
|
||||||
throw SocketError("Could not find a suitable interface for connecting");
|
|
||||||
}
|
|
||||||
|
|
||||||
m_Status = Status::Connected;
|
|
||||||
}
|
|
||||||
|
|
||||||
DataBuffer TcpSocket::Read(std::size_t a_Amount) {
|
|
||||||
DataBuffer buffer(a_Amount);
|
|
||||||
|
|
||||||
std::size_t totalRecieved = 0;
|
|
||||||
|
|
||||||
while (totalRecieved < a_Amount) {
|
|
||||||
int recvAmount =
|
|
||||||
recv(m_Handle, reinterpret_cast<char*>(buffer.data() + totalRecieved), static_cast<int>(a_Amount - totalRecieved), 0);
|
|
||||||
if (recvAmount <= 0) {
|
|
||||||
#if defined(_WIN32) || defined(WIN32)
|
|
||||||
int err = WSAGetLastError();
|
|
||||||
#else
|
|
||||||
int err = errno;
|
|
||||||
#endif
|
|
||||||
if (err == WOULDBLOCK) {
|
|
||||||
// we are in non blocking mode and nothing is available
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
Disconnect();
|
|
||||||
m_Status = Status::Error;
|
|
||||||
throw SocketError("Error while reading");
|
|
||||||
}
|
|
||||||
totalRecieved += recvAmount;
|
|
||||||
}
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
void TcpSocket::Write(const sp::DataBuffer& a_Data) {
|
|
||||||
if (GetStatus() != Status::Connected)
|
|
||||||
return;
|
|
||||||
|
|
||||||
std::size_t sent = 0;
|
|
||||||
|
|
||||||
while (sent < a_Data.GetSize()) {
|
|
||||||
int cur = send(m_Handle, reinterpret_cast<const char*>(a_Data.data() + sent), static_cast<int>(a_Data.GetSize() - sent), 0);
|
|
||||||
|
|
||||||
if (cur <= 0) {
|
|
||||||
Disconnect();
|
|
||||||
m_Status = Status::Error;
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
sent += static_cast<std::size_t>(cur);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool TcpSocket::SetBlocking(bool a_Block) {
|
|
||||||
unsigned long mode = !a_Block;
|
|
||||||
|
|
||||||
if (ioctl(m_Handle, FIONBIO, &mode) < 0) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
TcpSocket::Status TcpSocket::GetStatus() const {
|
|
||||||
return m_Status;
|
|
||||||
}
|
|
||||||
|
|
||||||
void TcpSocket::Disconnect() {
|
|
||||||
if (m_Handle > 0)
|
|
||||||
closesocket(m_Handle);
|
|
||||||
m_Status = Status::Disconnected;
|
|
||||||
}
|
|
||||||
|
|
||||||
TcpSocket& TcpSocket::operator=(IOInterface&& a_Other) {
|
|
||||||
std::swap(m_Handle, a_Other.m_Handle);
|
|
||||||
std::swap(m_Status, a_Other.m_Status);
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,31 +0,0 @@
|
|||||||
#include <sp/io/File.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
File::IOInterface(const std::string& a_FilePath, unsigned int a_OpenMode) {
|
|
||||||
if (a_OpenMode & FileTag::OpenMode::In)
|
|
||||||
m_FileInput = std::make_unique<std::ifstream>(a_FilePath, std::ios::binary);
|
|
||||||
if (a_OpenMode & FileTag::OpenMode::Out)
|
|
||||||
m_FileOutput = std::make_unique<std::ofstream>(a_FilePath, std::ios::binary);
|
|
||||||
}
|
|
||||||
|
|
||||||
File::IOInterface(File&& other) :
|
|
||||||
m_FileOutput(std::move(other.m_FileOutput)), m_FileInput(std::move(other.m_FileInput)) {}
|
|
||||||
|
|
||||||
DataBuffer File::Read(std::size_t a_Amount) {
|
|
||||||
DataBuffer buffer;
|
|
||||||
buffer.Resize(a_Amount);
|
|
||||||
assert(m_FileInput != nullptr);
|
|
||||||
m_FileInput->read(reinterpret_cast<char*>(buffer.data()), a_Amount);
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
void File::Write(const sp::DataBuffer& a_Data) {
|
|
||||||
assert(m_FileOutput != nullptr);
|
|
||||||
m_FileOutput->write(reinterpret_cast<const char*>(a_Data.data()), a_Data.GetSize());
|
|
||||||
m_FileOutput->flush();
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,16 +0,0 @@
|
|||||||
#include <sp/io/Memory.h>
|
|
||||||
|
|
||||||
namespace sp {
|
|
||||||
namespace io {
|
|
||||||
|
|
||||||
sp::DataBuffer Memory::Read(std::size_t a_Amount) {
|
|
||||||
DataBuffer data;
|
|
||||||
m_VirtualIO.ReadSome(data, a_Amount > m_VirtualIO.GetRemaining() ? m_VirtualIO.GetRemaining() : a_Amount);
|
|
||||||
return data;
|
|
||||||
}
|
|
||||||
void Memory::Write(const sp::DataBuffer& a_Data) {
|
|
||||||
m_VirtualIO << a_Data;
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace io
|
|
||||||
} // namespace sp
|
|
||||||
@@ -1,42 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
|
|
||||||
#include <examples/PacketExample.h>
|
|
||||||
#include <sp/io/File.h>
|
|
||||||
|
|
||||||
class CustomPacketHandler : public sp::PacketHandler {
|
|
||||||
void Handle(const KeepAlivePacket& packet) {
|
|
||||||
std::cout << "KeepAlive handled ! " << packet.GetKeepAliveId() << "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
void Handle(const DisconnectPacket& packet) {
|
|
||||||
std::cout << "Disconnect handled ! " << packet.GetReason() << "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
void Handle(const UpgradeTowerPacket& packet) {
|
|
||||||
std::cout << "UpgradeTower handled !\n";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
using FileStream = sp::io::Stream<sp::io::FileTag, sp::PacketDispatcher, sp::PacketFactory, sp::option::ZlibCompress>;
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
auto handler = std::make_shared<CustomPacketHandler>();
|
|
||||||
|
|
||||||
FileStream stream(sp::io::File{"test.txt", sp::io::FileTag::In | sp::io::FileTag::Out}, {});
|
|
||||||
stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler.get());
|
|
||||||
stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler.get());
|
|
||||||
|
|
||||||
stream.SendMessage(KeepAlivePacket{96});
|
|
||||||
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();
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
@@ -1,40 +0,0 @@
|
|||||||
#include <iostream>
|
|
||||||
|
|
||||||
#include <examples/PacketExample.h>
|
|
||||||
#include <sp/io/Memory.h>
|
|
||||||
|
|
||||||
using DataBufferStream = sp::io::Stream<sp::io::MemoryTag, sp::PacketDispatcher, sp::PacketFactory>;
|
|
||||||
|
|
||||||
class CustomPacketHandler : public sp::PacketHandler {
|
|
||||||
void Handle(const KeepAlivePacket& packet) {
|
|
||||||
std::cout << "KeepAlive handled ! " << packet.GetKeepAliveId() << "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
void Handle(const DisconnectPacket& packet) {
|
|
||||||
std::cout << "Disconnect handled ! " << packet.GetReason() << "\n";
|
|
||||||
}
|
|
||||||
|
|
||||||
void Handle(const UpgradeTowerPacket& packet) {
|
|
||||||
std::cout << "UpgradeTower handled !\n";
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
int main() {
|
|
||||||
auto handler = std::make_shared<CustomPacketHandler>();
|
|
||||||
|
|
||||||
DataBufferStream stream;
|
|
||||||
stream.GetDispatcher().RegisterHandler(PacketId::Disconnect, handler.get());
|
|
||||||
|
|
||||||
// this should not be dispatched
|
|
||||||
stream.SendMessage(KeepAlivePacket{96});
|
|
||||||
stream.RecieveMessages();
|
|
||||||
|
|
||||||
stream.GetDispatcher().RegisterHandler(PacketId::KeepAlive, handler.get());
|
|
||||||
|
|
||||||
stream.SendMessage(KeepAlivePacket{69});
|
|
||||||
stream.RecieveMessages();
|
|
||||||
stream.SendMessage(DisconnectPacket{"A valid reason"});
|
|
||||||
stream.RecieveMessages();
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
166
xmake.lua
166
xmake.lua
@@ -2,93 +2,13 @@ add_rules("mode.debug", "mode.release")
|
|||||||
|
|
||||||
set_languages("c++17")
|
set_languages("c++17")
|
||||||
|
|
||||||
local modules = {
|
add_requires("enet6")
|
||||||
Compression = {
|
|
||||||
Option = "zlib",
|
|
||||||
Deps = {"zlib"},
|
|
||||||
Includes = {"include/(sp/extensions/Compress.h)"},
|
|
||||||
Sources = {"src/sp/extensions/Compress.cpp"}
|
|
||||||
},
|
|
||||||
TcpSocket = {
|
|
||||||
Option = "tcp",
|
|
||||||
Deps = {},
|
|
||||||
Includes = {"include/(sp/extensions/Tcp.h)", "include/(sp/extensions/tcp/*.h)"},
|
|
||||||
Sources = {"src/sp/extensions/Tcp*.cpp"}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
target("SimpleProtocolLib")
|
||||||
|
add_includedirs("include", {public = true})
|
||||||
|
set_kind("binary")
|
||||||
|
add_files("src/**.cpp")
|
||||||
-- Map modules to options
|
add_packages("enet6", {public = true})
|
||||||
for name, module in table.orderpairs(modules) do
|
|
||||||
if module.Option then
|
|
||||||
option(module.Option, { description = "Enables the " .. name .. " module", default = true, category = "Modules" })
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
-- Add modules requirements
|
|
||||||
for name, module in table.orderpairs(modules) do
|
|
||||||
if module.Deps then
|
|
||||||
add_requires(module.Deps)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
-- Add modules targets
|
|
||||||
for name, module in table.orderpairs(modules) do
|
|
||||||
if module.Deps and has_config(module.Option) then
|
|
||||||
target("SimpleProtocol-" .. name)
|
|
||||||
add_includedirs("include")
|
|
||||||
for _, include in table.orderpairs(module.Includes) do
|
|
||||||
add_headerfiles(include)
|
|
||||||
end
|
|
||||||
for _, source in table.orderpairs(module.Sources) do
|
|
||||||
add_files(source)
|
|
||||||
end
|
|
||||||
for _, package in table.orderpairs(module.Deps) do
|
|
||||||
add_packages(package)
|
|
||||||
end
|
|
||||||
set_group("Library")
|
|
||||||
set_kind("$(kind)")
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
target("SimpleProtocol")
|
|
||||||
add_includedirs("include")
|
|
||||||
add_files("src/sp/**.cpp")
|
|
||||||
set_group("Library")
|
|
||||||
set_kind("$(kind)")
|
|
||||||
|
|
||||||
add_headerfiles("include/(sp/**.h)", "include/(sp/**.inl)")
|
|
||||||
|
|
||||||
-- adding extensions
|
|
||||||
for name, module in table.orderpairs(modules) do
|
|
||||||
if module.Deps and has_config(module.Option) then
|
|
||||||
add_deps("SimpleProtocol-" .. name)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
-- we don't want extensions
|
|
||||||
remove_files("src/sp/extensions/**.cpp")
|
|
||||||
remove_headerfiles("include/(sp/extension/**.h)")
|
|
||||||
|
|
||||||
-- we need this for endian functions
|
|
||||||
if is_os("windows") then
|
|
||||||
add_links("ws2_32")
|
|
||||||
end
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@@ -99,9 +19,79 @@ for _, file in ipairs(os.files("test/**.cpp")) do
|
|||||||
set_kind("binary")
|
set_kind("binary")
|
||||||
|
|
||||||
add_files(file)
|
add_files(file)
|
||||||
add_includedirs("include")
|
|
||||||
|
|
||||||
add_deps("SimpleProtocol")
|
set_default(false)
|
||||||
|
|
||||||
|
add_deps("SimpleProtocolLib")
|
||||||
|
|
||||||
add_tests("compile_and_run")
|
add_tests("compile_and_run")
|
||||||
end
|
end
|
||||||
|
--
|
||||||
|
-- If you want to known more usage about xmake, please see https://xmake.io
|
||||||
|
--
|
||||||
|
-- ## FAQ
|
||||||
|
--
|
||||||
|
-- You can enter the project directory firstly before building project.
|
||||||
|
--
|
||||||
|
-- $ cd projectdir
|
||||||
|
--
|
||||||
|
-- 1. How to build project?
|
||||||
|
--
|
||||||
|
-- $ xmake
|
||||||
|
--
|
||||||
|
-- 2. How to configure project?
|
||||||
|
--
|
||||||
|
-- $ xmake f -p [macosx|linux|iphoneos ..] -a [x86_64|i386|arm64 ..] -m [debug|release]
|
||||||
|
--
|
||||||
|
-- 3. Where is the build output directory?
|
||||||
|
--
|
||||||
|
-- The default output directory is `./build` and you can configure the output directory.
|
||||||
|
--
|
||||||
|
-- $ xmake f -o outputdir
|
||||||
|
-- $ xmake
|
||||||
|
--
|
||||||
|
-- 4. How to run and debug target after building project?
|
||||||
|
--
|
||||||
|
-- $ xmake run [targetname]
|
||||||
|
-- $ xmake run -d [targetname]
|
||||||
|
--
|
||||||
|
-- 5. How to install target to the system directory or other output directory?
|
||||||
|
--
|
||||||
|
-- $ xmake install
|
||||||
|
-- $ xmake install -o installdir
|
||||||
|
--
|
||||||
|
-- 6. Add some frequently-used compilation flags in xmake.lua
|
||||||
|
--
|
||||||
|
-- @code
|
||||||
|
-- -- add debug and release modes
|
||||||
|
-- add_rules("mode.debug", "mode.release")
|
||||||
|
--
|
||||||
|
-- -- add macro definition
|
||||||
|
-- add_defines("NDEBUG", "_GNU_SOURCE=1")
|
||||||
|
--
|
||||||
|
-- -- set warning all as error
|
||||||
|
-- set_warnings("all", "error")
|
||||||
|
--
|
||||||
|
-- -- set language: c99, c++11
|
||||||
|
-- set_languages("c99", "c++11")
|
||||||
|
--
|
||||||
|
-- -- set optimization: none, faster, fastest, smallest
|
||||||
|
-- set_optimize("fastest")
|
||||||
|
--
|
||||||
|
-- -- add include search directories
|
||||||
|
-- add_includedirs("/usr/include", "/usr/local/include")
|
||||||
|
--
|
||||||
|
-- -- add link libraries and search directories
|
||||||
|
-- add_links("tbox")
|
||||||
|
-- add_linkdirs("/usr/local/lib", "/usr/lib")
|
||||||
|
--
|
||||||
|
-- -- add system link libraries
|
||||||
|
-- add_syslinks("z", "pthread")
|
||||||
|
--
|
||||||
|
-- -- add compilation and link flags
|
||||||
|
-- add_cxflags("-stdnolib", "-fno-strict-aliasing")
|
||||||
|
-- add_ldflags("-L/usr/local/lib", "-lpthread", {force = true})
|
||||||
|
--
|
||||||
|
-- @endcode
|
||||||
|
--
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user