serialization.hpp 8.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268
  1. //
  2. // Copyright (c) 2019-2023 Ruben Perez Hidalgo (rubenperez038 at gmail dot com)
  3. //
  4. // Distributed under the Boost Software License, Version 1.0. (See accompanying
  5. // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  6. //
  7. #ifndef BOOST_MYSQL_DETAIL_PROTOCOL_SERIALIZATION_HPP
  8. #define BOOST_MYSQL_DETAIL_PROTOCOL_SERIALIZATION_HPP
  9. #include <boost/mysql/detail/auxiliar/bytestring.hpp>
  10. #include <boost/mysql/detail/protocol/deserialization_context.hpp>
  11. #include <boost/mysql/detail/protocol/deserialize_errc.hpp>
  12. #include <boost/mysql/detail/protocol/protocol_types.hpp>
  13. #include <boost/mysql/detail/protocol/serialization_context.hpp>
  14. #include <boost/endian/conversion.hpp>
  15. #include <algorithm>
  16. #include <type_traits>
  17. namespace boost {
  18. namespace mysql {
  19. namespace detail {
  20. enum class serialization_tag
  21. {
  22. none,
  23. plain_int,
  24. enumeration,
  25. struct_with_fields
  26. };
  27. template <class T>
  28. constexpr serialization_tag get_serialization_tag();
  29. template <class T, serialization_tag = get_serialization_tag<T>()>
  30. struct serialization_traits;
  31. template <class... Types>
  32. deserialize_errc deserialize(deserialization_context& ctx, Types&... output) noexcept;
  33. template <class... Types>
  34. void serialize(serialization_context& ctx, const Types&... input) noexcept;
  35. template <class... Types>
  36. std::size_t get_size(const serialization_context& ctx, const Types&... input) noexcept;
  37. // Plain integers
  38. template <class T>
  39. struct serialization_traits<T, serialization_tag::plain_int>
  40. {
  41. static deserialize_errc deserialize_(deserialization_context& ctx, T& output) noexcept;
  42. static void serialize_(serialization_context& ctx, T input) noexcept;
  43. static constexpr std::size_t get_size_(const serialization_context&, T) noexcept
  44. {
  45. return sizeof(T);
  46. }
  47. };
  48. // int3
  49. template <>
  50. struct serialization_traits<int3, serialization_tag::none>
  51. {
  52. static inline deserialize_errc deserialize_(deserialization_context& ctx, int3& output) noexcept
  53. {
  54. if (!ctx.enough_size(3))
  55. return deserialize_errc::incomplete_message;
  56. output.value = boost::endian::load_little_u24(ctx.first());
  57. ctx.advance(3);
  58. return deserialize_errc::ok;
  59. }
  60. static inline void serialize_(serialization_context& ctx, int3 input) noexcept
  61. {
  62. boost::endian::store_little_u24(ctx.first(), input.value);
  63. ctx.advance(3);
  64. }
  65. static inline std::size_t get_size_(const serialization_context&, int3) noexcept { return 3; }
  66. };
  67. // int_lenenc
  68. template <>
  69. struct serialization_traits<int_lenenc, serialization_tag::none>
  70. {
  71. static inline deserialize_errc deserialize_(
  72. deserialization_context& ctx,
  73. int_lenenc& output
  74. ) noexcept;
  75. static inline void serialize_(serialization_context& ctx, int_lenenc input) noexcept;
  76. static inline std::size_t get_size_(const serialization_context&, int_lenenc input) noexcept;
  77. };
  78. // string_fixed
  79. template <std::size_t N>
  80. struct serialization_traits<string_fixed<N>, serialization_tag::none>
  81. {
  82. static deserialize_errc deserialize_(
  83. deserialization_context& ctx,
  84. string_fixed<N>& output
  85. ) noexcept
  86. {
  87. if (!ctx.enough_size(N))
  88. return deserialize_errc::incomplete_message;
  89. memcpy(output.data(), ctx.first(), N);
  90. ctx.advance(N);
  91. return deserialize_errc::ok;
  92. }
  93. static void serialize_(serialization_context& ctx, const string_fixed<N>& input) noexcept
  94. {
  95. ctx.write(input.data(), N);
  96. }
  97. static constexpr std::size_t get_size_(const serialization_context&, const string_fixed<N>&) noexcept
  98. {
  99. return N;
  100. }
  101. };
  102. // string_null
  103. template <>
  104. struct serialization_traits<string_null, serialization_tag::none>
  105. {
  106. static inline deserialize_errc deserialize_(
  107. deserialization_context& ctx,
  108. string_null& output
  109. ) noexcept;
  110. static inline void serialize_(serialization_context& ctx, string_null input) noexcept
  111. {
  112. ctx.write(input.value.data(), input.value.size());
  113. ctx.write(0); // null terminator
  114. }
  115. static inline std::size_t get_size_(const serialization_context&, string_null input) noexcept
  116. {
  117. return input.value.size() + 1;
  118. }
  119. };
  120. // string_eof
  121. template <>
  122. struct serialization_traits<string_eof, serialization_tag::none>
  123. {
  124. static inline deserialize_errc deserialize_(
  125. deserialization_context& ctx,
  126. string_eof& output
  127. ) noexcept;
  128. static inline void serialize_(serialization_context& ctx, string_eof input) noexcept
  129. {
  130. ctx.write(input.value.data(), input.value.size());
  131. }
  132. static inline std::size_t get_size_(const serialization_context&, string_eof input) noexcept
  133. {
  134. return input.value.size();
  135. }
  136. };
  137. // string_lenenc
  138. template <>
  139. struct serialization_traits<string_lenenc, serialization_tag::none>
  140. {
  141. static inline deserialize_errc deserialize_(
  142. deserialization_context& ctx,
  143. string_lenenc& output
  144. ) noexcept;
  145. static inline void serialize_(serialization_context& ctx, string_lenenc input) noexcept
  146. {
  147. serialize(ctx, int_lenenc(input.value.size()));
  148. ctx.write(input.value.data(), input.value.size());
  149. }
  150. static inline std::size_t get_size_(
  151. const serialization_context& ctx,
  152. string_lenenc input
  153. ) noexcept
  154. {
  155. return get_size(ctx, int_lenenc(input.value.size())) + input.value.size();
  156. }
  157. };
  158. // Enums
  159. template <class T>
  160. struct serialization_traits<T, serialization_tag::enumeration>
  161. {
  162. using underlying_type = typename std::underlying_type<T>::type;
  163. static deserialize_errc deserialize_(deserialization_context& ctx, T& output) noexcept
  164. {
  165. underlying_type value = 0;
  166. auto err = deserialize(ctx, value);
  167. output = static_cast<T>(value);
  168. return err;
  169. }
  170. static void serialize_(serialization_context& ctx, T input) noexcept
  171. {
  172. serialize(ctx, static_cast<underlying_type>(input));
  173. }
  174. static std::size_t get_size_(const serialization_context&, T) noexcept
  175. {
  176. return sizeof(underlying_type);
  177. }
  178. };
  179. // Structs and commands
  180. // To allow a limited way of reflection, structs should
  181. // provide a static member function with signature:
  182. // void apply(Self&, Callable&&), which will invoke Callable
  183. // with each field in the struct as argument (similar to std::apply).
  184. // Making the member function static with a Self template parameter
  185. // allows calling apply with const and non-const objects as self.
  186. // Types fulfilling this requirement will have the below function
  187. // return true
  188. template <class T>
  189. constexpr bool is_struct_with_fields();
  190. template <class T>
  191. struct serialization_traits<T, serialization_tag::struct_with_fields>
  192. {
  193. static deserialize_errc deserialize_(deserialization_context& ctx, T& output) noexcept;
  194. static void serialize_(serialization_context& ctx, const T& input) noexcept;
  195. static std::size_t get_size_(const serialization_context& ctx, const T& input) noexcept;
  196. };
  197. // Use these to make all messages implement all methods, leaving the not required
  198. // ones with a default implementation. While this is not strictly required,
  199. // it simplifies the testing infrastructure
  200. template <class T>
  201. struct noop_serialize
  202. {
  203. static inline std::size_t get_size_(const serialization_context&, const T&) noexcept
  204. {
  205. return 0;
  206. }
  207. static inline void serialize_(serialization_context&, const T&) noexcept {}
  208. };
  209. template <class T>
  210. struct noop_deserialize
  211. {
  212. static inline deserialize_errc deserialize_(deserialization_context&, T&) noexcept
  213. {
  214. return deserialize_errc::ok;
  215. }
  216. };
  217. // Helper to serialize top-level messages
  218. template <class Serializable, class Allocator>
  219. void serialize_message(
  220. const Serializable& input,
  221. capabilities caps,
  222. basic_bytestring<Allocator>& buffer
  223. );
  224. template <class Deserializable>
  225. error_code deserialize_message(deserialization_context& ctx, Deserializable& output);
  226. template <class Deserializable>
  227. error_code deserialize_message_part(deserialization_context& ctx, Deserializable& output);
  228. // Helpers for (de) serializing a set of fields
  229. template <class... Types>
  230. void serialize_fields(serialization_context& ctx, const Types&... fields) noexcept;
  231. } // namespace detail
  232. } // namespace mysql
  233. } // namespace boost
  234. #include <boost/mysql/detail/protocol/impl/serialization.hpp>
  235. #include <boost/mysql/detail/protocol/impl/serialization.ipp>
  236. #endif