Encoder.cpp 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226
  1. /*
  2. * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
  3. * Copyright (c) 2021, kleines Filmröllchen <filmroellchen@serenityos.org>
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #include <AK/BitCast.h>
  8. #include <AK/ByteBuffer.h>
  9. #include <AK/JsonObject.h>
  10. #include <AK/JsonValue.h>
  11. #include <AK/String.h>
  12. #include <AK/URL.h>
  13. #include <LibCore/AnonymousBuffer.h>
  14. #include <LibCore/DateTime.h>
  15. #include <LibCore/Proxy.h>
  16. #include <LibIPC/Dictionary.h>
  17. #include <LibIPC/Encoder.h>
  18. #include <LibIPC/File.h>
  19. namespace IPC {
  20. Encoder& Encoder::operator<<(bool value)
  21. {
  22. return *this << (u8)value;
  23. }
  24. Encoder& Encoder::operator<<(u8 value)
  25. {
  26. m_buffer.data.append(value);
  27. return *this;
  28. }
  29. Encoder& Encoder::operator<<(u16 value)
  30. {
  31. m_buffer.data.ensure_capacity(m_buffer.data.size() + 2);
  32. m_buffer.data.unchecked_append((u8)value);
  33. m_buffer.data.unchecked_append((u8)(value >> 8));
  34. return *this;
  35. }
  36. void Encoder::encode_u32(u32 value)
  37. {
  38. m_buffer.data.ensure_capacity(m_buffer.data.size() + 4);
  39. m_buffer.data.unchecked_append((u8)value);
  40. m_buffer.data.unchecked_append((u8)(value >> 8));
  41. m_buffer.data.unchecked_append((u8)(value >> 16));
  42. m_buffer.data.unchecked_append((u8)(value >> 24));
  43. }
  44. void Encoder::encode_u64(u64 value)
  45. {
  46. m_buffer.data.ensure_capacity(m_buffer.data.size() + 8);
  47. m_buffer.data.unchecked_append((u8)value);
  48. m_buffer.data.unchecked_append((u8)(value >> 8));
  49. m_buffer.data.unchecked_append((u8)(value >> 16));
  50. m_buffer.data.unchecked_append((u8)(value >> 24));
  51. m_buffer.data.unchecked_append((u8)(value >> 32));
  52. m_buffer.data.unchecked_append((u8)(value >> 40));
  53. m_buffer.data.unchecked_append((u8)(value >> 48));
  54. m_buffer.data.unchecked_append((u8)(value >> 56));
  55. }
  56. Encoder& Encoder::operator<<(unsigned value)
  57. {
  58. encode_u32(value);
  59. return *this;
  60. }
  61. Encoder& Encoder::operator<<(unsigned long value)
  62. {
  63. if constexpr (sizeof(value) == 4)
  64. encode_u32(value);
  65. else
  66. encode_u64(value);
  67. return *this;
  68. }
  69. Encoder& Encoder::operator<<(unsigned long long value)
  70. {
  71. if constexpr (sizeof(value) == 4)
  72. encode_u32(value);
  73. else
  74. encode_u64(value);
  75. return *this;
  76. }
  77. Encoder& Encoder::operator<<(i8 value)
  78. {
  79. m_buffer.data.append((u8)value);
  80. return *this;
  81. }
  82. Encoder& Encoder::operator<<(i16 value)
  83. {
  84. m_buffer.data.ensure_capacity(m_buffer.data.size() + 2);
  85. m_buffer.data.unchecked_append((u8)value);
  86. m_buffer.data.unchecked_append((u8)(value >> 8));
  87. return *this;
  88. }
  89. Encoder& Encoder::operator<<(i32 value)
  90. {
  91. m_buffer.data.ensure_capacity(m_buffer.data.size() + 4);
  92. m_buffer.data.unchecked_append((u8)value);
  93. m_buffer.data.unchecked_append((u8)(value >> 8));
  94. m_buffer.data.unchecked_append((u8)(value >> 16));
  95. m_buffer.data.unchecked_append((u8)(value >> 24));
  96. return *this;
  97. }
  98. Encoder& Encoder::operator<<(i64 value)
  99. {
  100. m_buffer.data.ensure_capacity(m_buffer.data.size() + 8);
  101. m_buffer.data.unchecked_append((u8)value);
  102. m_buffer.data.unchecked_append((u8)(value >> 8));
  103. m_buffer.data.unchecked_append((u8)(value >> 16));
  104. m_buffer.data.unchecked_append((u8)(value >> 24));
  105. m_buffer.data.unchecked_append((u8)(value >> 32));
  106. m_buffer.data.unchecked_append((u8)(value >> 40));
  107. m_buffer.data.unchecked_append((u8)(value >> 48));
  108. m_buffer.data.unchecked_append((u8)(value >> 56));
  109. return *this;
  110. }
  111. Encoder& Encoder::operator<<(float value)
  112. {
  113. u32 as_u32 = bit_cast<u32>(value);
  114. return *this << as_u32;
  115. }
  116. Encoder& Encoder::operator<<(double value)
  117. {
  118. u64 as_u64 = bit_cast<u64>(value);
  119. return *this << as_u64;
  120. }
  121. Encoder& Encoder::operator<<(char const* value)
  122. {
  123. return *this << StringView { value, strlen(value) };
  124. }
  125. Encoder& Encoder::operator<<(StringView value)
  126. {
  127. m_buffer.data.append((u8 const*)value.characters_without_null_termination(), value.length());
  128. return *this;
  129. }
  130. Encoder& Encoder::operator<<(String const& value)
  131. {
  132. if (value.is_null())
  133. return *this << (i32)-1;
  134. *this << static_cast<i32>(value.length());
  135. return *this << value.view();
  136. }
  137. Encoder& Encoder::operator<<(ByteBuffer const& value)
  138. {
  139. *this << static_cast<i32>(value.size());
  140. m_buffer.data.append(value.data(), value.size());
  141. return *this;
  142. }
  143. Encoder& Encoder::operator<<(JsonValue const& value)
  144. {
  145. *this << value.serialized<StringBuilder>();
  146. return *this;
  147. }
  148. Encoder& Encoder::operator<<(URL const& value)
  149. {
  150. return *this << value.to_string();
  151. }
  152. Encoder& Encoder::operator<<(Dictionary const& dictionary)
  153. {
  154. *this << (u64)dictionary.size();
  155. dictionary.for_each_entry([this](auto& key, auto& value) {
  156. *this << key << value;
  157. });
  158. return *this;
  159. }
  160. Encoder& Encoder::operator<<(File const& file)
  161. {
  162. int fd = file.fd();
  163. if (fd != -1) {
  164. auto result = dup(fd);
  165. if (result < 0) {
  166. perror("dup");
  167. VERIFY_NOT_REACHED();
  168. }
  169. fd = result;
  170. }
  171. m_buffer.fds.append(adopt_ref(*new AutoCloseFileDescriptor(fd)));
  172. return *this;
  173. }
  174. template<>
  175. bool encode(Encoder& encoder, Core::AnonymousBuffer const& buffer)
  176. {
  177. encoder << buffer.is_valid();
  178. if (buffer.is_valid()) {
  179. encoder << (u32)buffer.size();
  180. encoder << IPC::File(buffer.fd());
  181. }
  182. return true;
  183. }
  184. template<>
  185. bool encode(Encoder& encoder, Core::DateTime const& datetime)
  186. {
  187. encoder << static_cast<i64>(datetime.timestamp());
  188. return true;
  189. }
  190. template<>
  191. bool encode(Encoder& encoder, Core::ProxyData const& proxy)
  192. {
  193. encoder << to_underlying(proxy.type);
  194. encoder << proxy.host_ipv4;
  195. encoder << proxy.port;
  196. return true;
  197. }
  198. }