Compare commits
10 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 471b5b56f8 | |||
| 392fcb3d17 | |||
| 205c09a338 | |||
| 77356ce749 | |||
| 7f8d9e3f96 | |||
| 81c9dbadd6 | |||
| e8367cd91c | |||
| fa6ba74068 | |||
| d501c0181d | |||
| 680f72f881 |
@@ -14,7 +14,9 @@ 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
|
||||||
|
std::map<std::string, std::vector<int>>
|
||||||
>;
|
>;
|
||||||
|
|
||||||
DeclarePacket(UpgradeTower){
|
DeclarePacket(UpgradeTower){
|
||||||
|
|||||||
60
include/sp/common/Reflection.h
Normal file
60
include/sp/common/Reflection.h
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
#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,6 +1,8 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
|
#include <string>
|
||||||
|
#include <tuple>
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
@@ -50,4 +52,48 @@ 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,6 +7,8 @@
|
|||||||
|
|
||||||
#include <cstddef>
|
#include <cstddef>
|
||||||
#include <cstdint>
|
#include <cstdint>
|
||||||
|
#include <ostream>
|
||||||
|
#include <sp/protocol/Field.h>
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
@@ -55,6 +57,8 @@ class VarInt {
|
|||||||
* \param var The variable integer to deserialize
|
* \param var The variable integer to deserialize
|
||||||
*/
|
*/
|
||||||
friend DataBuffer& operator>>(DataBuffer& in, VarInt& var);
|
friend DataBuffer& operator>>(DataBuffer& in, VarInt& var);
|
||||||
|
|
||||||
|
friend std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<VarInt>& a_VarInt);
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -11,8 +11,9 @@ using PacketMessage = Message<
|
|||||||
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) \
|
||||||
packetName##Packet() {} \
|
packetName##Packet() {} \
|
||||||
@@ -23,7 +24,8 @@ 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>
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -10,6 +10,8 @@ namespace io {
|
|||||||
*/
|
*/
|
||||||
class TcpListener : private NonCopyable {
|
class TcpListener : private NonCopyable {
|
||||||
public:
|
public:
|
||||||
|
using SocketHandle = TcpTag::SocketHandle;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \brief Starts listening for guests to connect
|
* \brief Starts listening for guests to connect
|
||||||
* \param port The port to listen to
|
* \param port The port to listen to
|
||||||
|
|||||||
@@ -6,9 +6,9 @@
|
|||||||
namespace sp {
|
namespace sp {
|
||||||
namespace io {
|
namespace io {
|
||||||
|
|
||||||
using SocketHandle = int;
|
struct TcpTag {
|
||||||
|
using SocketHandle = int;
|
||||||
struct TcpTag {};
|
};
|
||||||
|
|
||||||
class SocketError : public std::exception {
|
class SocketError : public std::exception {
|
||||||
private:
|
private:
|
||||||
@@ -25,6 +25,8 @@ class SocketError : public std::exception {
|
|||||||
template <>
|
template <>
|
||||||
class IOInterface<TcpTag> : private NonCopyable {
|
class IOInterface<TcpTag> : private NonCopyable {
|
||||||
public:
|
public:
|
||||||
|
using SocketHandle = TcpTag::SocketHandle;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* \enum Status
|
* \enum Status
|
||||||
* \brief Describes the state of a socket
|
* \brief Describes the state of a socket
|
||||||
|
|||||||
@@ -48,11 +48,12 @@ Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::Stream(
|
|||||||
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
template <typename IOTag, typename MessageDispatcher, typename MessageFactory, typename... TOptions>
|
||||||
void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::SendMessage(const MessageBase& a_Message) {
|
void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::SendMessage(const MessageBase& a_Message) {
|
||||||
DataBuffer data = a_Message.Write();
|
DataBuffer data = a_Message.Write();
|
||||||
DataBuffer encapsulated = std::apply([&data](const auto&... a_Options) {
|
TupleForEach([&data](const auto&... a_Options) {
|
||||||
return Encapsulate(data, a_Options...);
|
data = Encapsulate(data, a_Options...);
|
||||||
}, m_Options);
|
}, m_Options);
|
||||||
DataBuffer finalData;
|
DataBuffer finalData;
|
||||||
finalData << VarInt{encapsulated.GetSize()} << encapsulated;
|
finalData.Reserve(data.GetSize() + sizeof(VarInt::MAX_VALUE));
|
||||||
|
finalData << VarInt{data.GetSize()} << data;
|
||||||
m_Interface.Write(finalData);
|
m_Interface.Write(finalData);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -93,8 +94,8 @@ void Stream<IOTag, MessageDispatcher, MessageFactory, TOptions...>::RecieveMessa
|
|||||||
DataBuffer buffer;
|
DataBuffer buffer;
|
||||||
buffer = m_Interface.Read(lenghtValue);
|
buffer = m_Interface.Read(lenghtValue);
|
||||||
|
|
||||||
buffer = std::apply([&buffer, lenghtValue](const auto&... a_Options) {
|
TupleForEach([&buffer, lenghtValue](const auto&... a_Options) {
|
||||||
return Decapsulate(buffer, a_Options...);
|
buffer = Decapsulate(buffer, a_Options...);
|
||||||
}, m_Options);
|
}, m_Options);
|
||||||
|
|
||||||
VarInt packetType;
|
VarInt packetType;
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <sp/common/DataBuffer.h>
|
|
||||||
#include <sp/common/Templates.h>
|
#include <sp/common/Templates.h>
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
@@ -70,29 +69,45 @@ 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, int IAlignment>
|
template <typename ValueType, std::size_t 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
|
||||||
ValueType& GetValue() {
|
StorageType& GetValue() {
|
||||||
return m_Value;
|
return m_Value;
|
||||||
}
|
}
|
||||||
|
|
||||||
const ValueType& GetValue() const {
|
const StorageType& GetValue() const {
|
||||||
return m_Value;
|
return m_Value;
|
||||||
}
|
}
|
||||||
|
|
||||||
Field& operator=(const ValueType& value) {
|
Field& operator=(const StorageType& value) {
|
||||||
m_Value = value;
|
m_Value = value;
|
||||||
return *this;
|
return *this;
|
||||||
}
|
}
|
||||||
|
|
||||||
constexpr int GetAlignment() const {
|
constexpr std::size_t GetAlignment() const {
|
||||||
return IAlignment;
|
return IAlignment;
|
||||||
}
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
ValueType m_Value;
|
StorageType m_Value;
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
class PrintableField {
|
||||||
|
public:
|
||||||
|
PrintableField(const T& a_Value) : m_Value(a_Value) {}
|
||||||
|
|
||||||
|
const T& GetValue() const {
|
||||||
|
return m_Value;
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
const T& m_Value;
|
||||||
};
|
};
|
||||||
|
|
||||||
namespace details {
|
namespace details {
|
||||||
|
|||||||
@@ -14,3 +14,5 @@ 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>
|
||||||
61
include/sp/protocol/MessagePrinter.h
Normal file
61
include/sp/protocol/MessagePrinter.h
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
#include <ostream>
|
||||||
|
#include <sp/protocol/Field.h>
|
||||||
|
#include <sp/protocol/message/OstreamFieldIterator.h>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
namespace sp {
|
||||||
|
|
||||||
|
template <typename T, std::enable_if_t<details::is_primitive<T>::value, bool> = true>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<T>& a_Field) {
|
||||||
|
return a_Stream << std::to_string(a_Field.GetValue());
|
||||||
|
}
|
||||||
|
|
||||||
|
inline std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<long unsigned int>& a_Field) {
|
||||||
|
return a_Stream << a_Field.GetValue();
|
||||||
|
}
|
||||||
|
|
||||||
|
inline std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::string>& a_Field) {
|
||||||
|
return a_Stream << "\"" << a_Field.GetValue() << "\"";
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
template <typename K, typename V>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::pair<K, V>>& a_Data);
|
||||||
|
|
||||||
|
template <typename K, typename V>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::map<K, V>>& a_Data);
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::vector<T>>& a_Data);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
template <typename K, typename V>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::pair<K, V>>& a_Data) {
|
||||||
|
return a_Stream << PrintableField<K>(a_Data.GetValue().first) << " => " << PrintableField<V>(a_Data.GetValue().second);
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename K, typename V>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::map<K, V>>& a_Data) {
|
||||||
|
a_Stream << "{";
|
||||||
|
std::copy(a_Data.GetValue().begin(), a_Data.GetValue().end(), OstreamFieldIterator<std::pair<K, V>>(std::cout, ", "));
|
||||||
|
return a_Stream << "}";
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<std::vector<T>>& a_Data) {
|
||||||
|
a_Stream << "{";
|
||||||
|
std::copy(a_Data.GetValue().begin(), a_Data.GetValue().end(), OstreamFieldIterator<T>(std::cout, ", "));
|
||||||
|
return a_Stream << "}";
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace sp
|
||||||
@@ -31,7 +31,7 @@ struct ArrayFiller<TBase, TMessage, TMessages...> {
|
|||||||
template <typename TBase, typename TMessage>
|
template <typename TBase, typename TMessage>
|
||||||
struct ArrayFiller<TBase, TMessage> {
|
struct ArrayFiller<TBase, TMessage> {
|
||||||
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
static void ArrayAppend(details::ArrayType<TBase>& array) {
|
||||||
array.push_back([]() -> std::unique_ptr<TBase> { return std::make_unique<TMessage>(); });
|
array.emplace_back([]() -> std::unique_ptr<TBase> { return std::make_unique<TMessage>(); });
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -37,8 +37,12 @@ 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 = Base6;
|
using Type = Base7;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // 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::intmax_t TId>
|
template <std::uintmax_t TId>
|
||||||
struct StaticNumIdImpl {};
|
struct StaticNumIdImpl {};
|
||||||
|
|
||||||
// Facilitate implementation of DispatchImpl()
|
// Facilitate implementation of DispatchImpl()
|
||||||
@@ -16,6 +16,11 @@ 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
|
||||||
|
|
||||||
|
|
||||||
@@ -39,10 +44,10 @@ struct MessageImplParsedOptions<> {
|
|||||||
|
|
||||||
|
|
||||||
|
|
||||||
template <std::intmax_t TId, typename... TOptions>
|
template <std::uintmax_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::intmax_t MsgId = TId;
|
static const std::uintmax_t MsgId = TId;
|
||||||
};
|
};
|
||||||
|
|
||||||
template <typename TActual, typename... TOptions>
|
template <typename TActual, typename... TOptions>
|
||||||
|
|||||||
@@ -4,6 +4,20 @@ 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,8 +33,11 @@ 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;
|
||||||
|
|
||||||
// The last Base7 must be taken as final type.
|
// add ToString() if HasToString was provided
|
||||||
using Type = Base7;
|
using Base8 = typename MessageInterfaceProcessToString<Base7, ParsedOptions::HasToString>::Type;
|
||||||
|
|
||||||
|
// The last Base8 must be taken as final type.
|
||||||
|
using Type = Base8;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
|
|||||||
@@ -101,5 +101,19 @@ 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
|
||||||
@@ -129,5 +129,17 @@ class MessageInterfaceWriteIdBase : public TBase {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Debug print functionality chunk
|
||||||
|
template <typename TBase>
|
||||||
|
class MessageInterfaceToStringBase : public TBase {
|
||||||
|
public:
|
||||||
|
friend std::ostream& operator<<(std::ostream& a_Stream, const MessageInterfaceToStringBase& a_Message) {
|
||||||
|
return a_Message.OpOutImpl(a_Stream);
|
||||||
|
}
|
||||||
|
|
||||||
|
protected:
|
||||||
|
virtual std::ostream& OpOutImpl(std::ostream& a_Stream) const = 0;
|
||||||
|
};
|
||||||
|
|
||||||
} // namespace details
|
} // namespace details
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
@@ -16,6 +16,7 @@ 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;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
@@ -59,5 +60,10 @@ 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,6 +24,9 @@ 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 {
|
||||||
|
|||||||
79
include/sp/protocol/message/MessagePrinterImpl.h
Normal file
79
include/sp/protocol/message/MessagePrinterImpl.h
Normal file
@@ -0,0 +1,79 @@
|
|||||||
|
#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) + ")";
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace details
|
||||||
|
|
||||||
|
template <typename... TFields>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const std::tuple<TFields...>& a_Fields);
|
||||||
|
|
||||||
|
template <typename TField, unsigned int IAlignment>
|
||||||
|
struct FieldPrinter {
|
||||||
|
static std::ostream& PrintField(std::ostream& a_Stream, const sp::Field<TField, IAlignment>& a_Field) {
|
||||||
|
return a_Stream << sp::Reflector<TField>::GetClassName() << "=" << PrintableField<TField>(a_Field.GetValue());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <unsigned int IAlignment, typename TContainer, typename... TFields>
|
||||||
|
struct FieldPrinter<sp::BitField<TContainer, TFields...>, IAlignment> {
|
||||||
|
static std::ostream& PrintField(
|
||||||
|
std::ostream& a_Stream, const sp::Field<sp::BitField<TContainer, TFields...>, IAlignment>& a_Field) {
|
||||||
|
a_Stream << "BitField<" << sp::Reflector<TContainer>::GetClassName() << ">[";
|
||||||
|
a_Stream << a_Field.GetValue().GetFields() << "]";
|
||||||
|
return a_Stream;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
template <typename... TFields>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const std::tuple<TFields...>& a_Fields) {
|
||||||
|
bool first = true;
|
||||||
|
TupleForEach(
|
||||||
|
[&a_Stream, &first](const auto& a_Field) {
|
||||||
|
if (!first)
|
||||||
|
a_Stream << ", ";
|
||||||
|
using TField = typename std::decay<decltype(a_Field)>::type;
|
||||||
|
constexpr std::size_t alignment = TField::AlignmentValue;
|
||||||
|
FieldPrinter<typename TField::StorageType, alignment>::PrintField(a_Stream, a_Field);
|
||||||
|
first = false;
|
||||||
|
},
|
||||||
|
a_Fields);
|
||||||
|
return a_Stream;
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename TBase, typename... TOptions>
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const sp::MessageBase<TBase, TOptions...>& a_Message) {
|
||||||
|
a_Stream << sp::GetBasicClassName(a_Message) << sp::details::IdPrinter<TOptions...>::PrintMessageId() << "["
|
||||||
|
<< a_Message.GetFields() << "]";
|
||||||
|
return a_Stream;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace sp
|
||||||
@@ -1,8 +1,24 @@
|
|||||||
#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::ostream& OpOutImpl(std::ostream& a_Stream) const override{
|
||||||
|
return a_Stream << static_cast<const TActual&>(*this);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// ID information chunk
|
// ID information chunk
|
||||||
@@ -18,6 +34,8 @@ 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 {
|
||||||
@@ -81,7 +99,7 @@ class MessageImplFieldsReadBase : public TBase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// reading field in bitfield
|
// reading field in bitfield
|
||||||
template <typename TFieldType, typename TField, int IAlignment>
|
template <typename TFieldType, typename TField, std::size_t 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
|
||||||
@@ -118,7 +136,7 @@ class MessageImplFieldsWriteBase : public TBase {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// writing field in bitfield
|
// writing field in bitfield
|
||||||
template <typename TFieldType, typename TField, int IAlignment>
|
template <typename TFieldType, typename TField, std::size_t IAlignment>
|
||||||
void WriteField(const Field<TField, IAlignment>& field, TFieldType& data, std::size_t offset) const {
|
void WriteField(const Field<TField, IAlignment>& field, TFieldType& data, std::size_t offset) const {
|
||||||
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
|
||||||
|
|||||||
56
include/sp/protocol/message/OstreamFieldIterator.h
Normal file
56
include/sp/protocol/message/OstreamFieldIterator.h
Normal file
@@ -0,0 +1,56 @@
|
|||||||
|
// infix_iterator.h
|
||||||
|
//
|
||||||
|
// Lifted from Jerry Coffin's 's prefix_ostream_iterator
|
||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include <ostream>
|
||||||
|
#include <sp/protocol/Field.h>
|
||||||
|
|
||||||
|
namespace sp {
|
||||||
|
|
||||||
|
template <class T, class charT = char, class traits = std::char_traits<charT>>
|
||||||
|
class OstreamFieldIterator {
|
||||||
|
private:
|
||||||
|
std::basic_ostream<charT, traits>* m_Os;
|
||||||
|
std::string m_Delimiter;
|
||||||
|
bool m_FirstElem;
|
||||||
|
|
||||||
|
public:
|
||||||
|
using iterator_category = std::output_iterator_tag;
|
||||||
|
using value_type = void;
|
||||||
|
using difference_type = void;
|
||||||
|
using pointer = void;
|
||||||
|
using reference = void;
|
||||||
|
|
||||||
|
using char_type = charT;
|
||||||
|
using traits_type = traits;
|
||||||
|
using ostream_type = std::basic_ostream<charT, traits>;
|
||||||
|
|
||||||
|
OstreamFieldIterator(ostream_type& a_Stream) : m_Os(&a_Stream), m_Delimiter(0), m_FirstElem(true) {}
|
||||||
|
OstreamFieldIterator(ostream_type& a_Stream, std::string&& a_Delimiter) :
|
||||||
|
m_Os(&a_Stream), m_Delimiter(std::move(a_Delimiter)), m_FirstElem(true) {}
|
||||||
|
|
||||||
|
auto& operator=(const T& item) {
|
||||||
|
// Here's the only real change from ostream_iterator:
|
||||||
|
// Normally, the '*m_Os << item;' would come before the 'if'.
|
||||||
|
if (!m_FirstElem && !m_Delimiter.empty())
|
||||||
|
*m_Os << m_Delimiter;
|
||||||
|
*m_Os << sp::PrintableField<T>(item);
|
||||||
|
m_FirstElem = false;
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto& operator*() {
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto& operator++() {
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto& operator++(int) {
|
||||||
|
return *this;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace sp
|
||||||
@@ -1,7 +1,7 @@
|
|||||||
#include <sp/common/VarInt.h>
|
#include <sp/common/VarInt.h>
|
||||||
|
|
||||||
#include <stdexcept>
|
|
||||||
#include <sp/common/DataBuffer.h>
|
#include <sp/common/DataBuffer.h>
|
||||||
|
#include <stdexcept>
|
||||||
|
|
||||||
namespace sp {
|
namespace sp {
|
||||||
|
|
||||||
@@ -49,4 +49,8 @@ DataBuffer& operator>>(DataBuffer& in, VarInt& var) {
|
|||||||
return in;
|
return in;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
std::ostream& operator<<(std::ostream& a_Stream, const PrintableField<VarInt>& a_VarInt) {
|
||||||
|
return a_Stream << a_VarInt.GetValue().GetValue();
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace sp
|
} // namespace sp
|
||||||
|
|||||||
52
src/sp/extensions/SslContext.cpp
Normal file
52
src/sp/extensions/SslContext.cpp
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
#include <cstring>
|
||||||
|
#include <iostream>
|
||||||
|
#include <mbedtls/ctr_drbg.h>
|
||||||
|
#include <mbedtls/entropy.h>
|
||||||
|
#include <mbedtls/x509_crt.h>
|
||||||
|
#include <sp/common/DataBuffer.h>
|
||||||
|
#include <sp/common/NonCopyable.h>
|
||||||
|
|
||||||
|
namespace sp {
|
||||||
|
|
||||||
|
class SslContext : private NonCopyable {
|
||||||
|
public:
|
||||||
|
SslContext(unsigned int a_KeySizeBits) {
|
||||||
|
mbedtls_rsa_context rsaContext;
|
||||||
|
mbedtls_rsa_init(&rsaContext);
|
||||||
|
mbedtls_rsa_gen_key(&rsaContext, mbedtls_ctr_drbg_random, &m_CtrCrbg, a_KeySizeBits, 65537);
|
||||||
|
mbedtls_rsa_free(&rsaContext);
|
||||||
|
}
|
||||||
|
|
||||||
|
SslContext(const DataBuffer& a_Data) {
|
||||||
|
mbedtls_x509_crt_parse(&m_CaCert, (const unsigned char*)a_Data.data(), a_Data.GetSize()) == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
SslContext(const std::string& a_CertFilePath) {
|
||||||
|
mbedtls_x509_crt_parse_file(&m_CaCert, a_CertFilePath.c_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
~SslContext() {
|
||||||
|
mbedtls_ctr_drbg_free(&m_CtrCrbg);
|
||||||
|
mbedtls_entropy_free(&m_Entropy);
|
||||||
|
mbedtls_x509_crt_free(&m_CaCert);
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
void InitContext() {
|
||||||
|
int error = 0;
|
||||||
|
|
||||||
|
mbedtls_x509_crt_init(&m_CaCert);
|
||||||
|
mbedtls_ctr_drbg_init(&m_CtrCrbg);
|
||||||
|
|
||||||
|
mbedtls_entropy_init(&m_Entropy);
|
||||||
|
if ((error = mbedtls_ctr_drbg_seed(&m_CtrCrbg, mbedtls_entropy_func, &m_Entropy, nullptr, 0)) != 0) {
|
||||||
|
throw std::runtime_error("Failed to initialise random number generator. Returned error: " + std::to_string(error));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
mbedtls_entropy_context m_Entropy;
|
||||||
|
mbedtls_ctr_drbg_context m_CtrCrbg;
|
||||||
|
mbedtls_x509_crt m_CaCert;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace sp
|
||||||
@@ -8,7 +8,8 @@
|
|||||||
|
|
||||||
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 << std::endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
void Handle(const DisconnectPacket& packet) {
|
void Handle(const DisconnectPacket& packet) {
|
||||||
@@ -21,12 +22,15 @@ class KeepAliveHandler : public sp::PacketHandler {
|
|||||||
};
|
};
|
||||||
|
|
||||||
int main() {
|
int main() {
|
||||||
auto upgradeTower = std::make_unique<UpgradeTowerPacket>(std::make_tuple(666, 9));
|
std::map<std::string, std::vector<int>> yes = {{"woa", {5, 8}}, {"insane", {6, 9}}};
|
||||||
|
auto upgradeTower = std::make_unique<UpgradeTowerPacket>(std::make_tuple(666, 9), 789, yes);
|
||||||
|
auto keepAlive = std::make_unique<KeepAlivePacket>(6969);
|
||||||
|
|
||||||
sp::PacketMessage* msg = upgradeTower.get();
|
sp::PacketMessage* msg = upgradeTower.get();
|
||||||
|
|
||||||
auto handler = std::make_shared<KeepAliveHandler>();
|
auto handler = std::make_shared<KeepAliveHandler>();
|
||||||
msg->Dispatch(*handler);
|
msg->Dispatch(*handler);
|
||||||
|
keepAlive->Dispatch(*handler);
|
||||||
|
|
||||||
sp::DataBuffer buffer = msg->Write();
|
sp::DataBuffer buffer = msg->Write();
|
||||||
|
|
||||||
@@ -36,7 +40,7 @@ int main() {
|
|||||||
auto upgradeTower2 = std::make_unique<UpgradeTowerPacket>();
|
auto upgradeTower2 = std::make_unique<UpgradeTowerPacket>();
|
||||||
upgradeTower2->Read(buffer);
|
upgradeTower2->Read(buffer);
|
||||||
|
|
||||||
std::cout << "Test : " << (unsigned)upgradeTower2->GetTowerId() << "\n";
|
std::cout << "Test : " << *msg << "\n";
|
||||||
|
|
||||||
sp::PacketFactory factory;
|
sp::PacketFactory factory;
|
||||||
auto packet = factory.CreateMessage(msgId);
|
auto packet = factory.CreateMessage(msgId);
|
||||||
|
|||||||
@@ -14,6 +14,12 @@ local modules = {
|
|||||||
Deps = {},
|
Deps = {},
|
||||||
Includes = {"include/(sp/extensions/Tcp.h)", "include/(sp/extensions/tcp/*.h)"},
|
Includes = {"include/(sp/extensions/Tcp.h)", "include/(sp/extensions/tcp/*.h)"},
|
||||||
Sources = {"src/sp/extensions/Tcp*.cpp"}
|
Sources = {"src/sp/extensions/Tcp*.cpp"}
|
||||||
|
},
|
||||||
|
MbedTls = {
|
||||||
|
Option = "tls",
|
||||||
|
Deps = {"mbedtls"},
|
||||||
|
Includes = {"include/(sp/extensions/Ssl.h)", "include/(sp/extensions/Ssl/*.h)"},
|
||||||
|
Sources = {"src/sp/extensions/Ssl*.cpp"}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user