Decoder.cpp 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211
  1. /*
  2. * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/JsonValue.h>
  7. #include <AK/MemoryStream.h>
  8. #include <AK/URL.h>
  9. #include <LibCore/AnonymousBuffer.h>
  10. #include <LibCore/DateTime.h>
  11. #include <LibCore/Proxy.h>
  12. #include <LibIPC/Decoder.h>
  13. #include <LibIPC/Dictionary.h>
  14. #include <LibIPC/File.h>
  15. #include <fcntl.h>
  16. namespace IPC {
  17. ErrorOr<void> Decoder::decode(bool& value)
  18. {
  19. m_stream >> value;
  20. return m_stream.try_handle_any_error();
  21. }
  22. ErrorOr<void> Decoder::decode(u8& value)
  23. {
  24. m_stream >> value;
  25. return m_stream.try_handle_any_error();
  26. }
  27. ErrorOr<void> Decoder::decode(u16& value)
  28. {
  29. m_stream >> value;
  30. return m_stream.try_handle_any_error();
  31. }
  32. ErrorOr<void> Decoder::decode(unsigned& value)
  33. {
  34. m_stream >> value;
  35. return m_stream.try_handle_any_error();
  36. }
  37. ErrorOr<void> Decoder::decode(unsigned long& value)
  38. {
  39. m_stream >> value;
  40. return m_stream.try_handle_any_error();
  41. }
  42. ErrorOr<void> Decoder::decode(unsigned long long& value)
  43. {
  44. m_stream >> value;
  45. return m_stream.try_handle_any_error();
  46. }
  47. ErrorOr<void> Decoder::decode(i8& value)
  48. {
  49. m_stream >> value;
  50. return m_stream.try_handle_any_error();
  51. }
  52. ErrorOr<void> Decoder::decode(i16& value)
  53. {
  54. m_stream >> value;
  55. return m_stream.try_handle_any_error();
  56. }
  57. ErrorOr<void> Decoder::decode(i32& value)
  58. {
  59. m_stream >> value;
  60. return m_stream.try_handle_any_error();
  61. }
  62. ErrorOr<void> Decoder::decode(i64& value)
  63. {
  64. m_stream >> value;
  65. return m_stream.try_handle_any_error();
  66. }
  67. ErrorOr<void> Decoder::decode(float& value)
  68. {
  69. m_stream >> value;
  70. return m_stream.try_handle_any_error();
  71. }
  72. ErrorOr<void> Decoder::decode(double& value)
  73. {
  74. m_stream >> value;
  75. return m_stream.try_handle_any_error();
  76. }
  77. ErrorOr<void> Decoder::decode(String& value)
  78. {
  79. i32 length;
  80. TRY(decode(length));
  81. if (length < 0) {
  82. value = {};
  83. return {};
  84. }
  85. if (length == 0) {
  86. value = String::empty();
  87. return {};
  88. }
  89. char* text_buffer = nullptr;
  90. auto text_impl = StringImpl::create_uninitialized(static_cast<size_t>(length), text_buffer);
  91. m_stream >> Bytes { text_buffer, static_cast<size_t>(length) };
  92. value = *text_impl;
  93. return m_stream.try_handle_any_error();
  94. }
  95. ErrorOr<void> Decoder::decode(ByteBuffer& value)
  96. {
  97. i32 length;
  98. TRY(decode(length));
  99. if (length < 0) {
  100. value = {};
  101. return {};
  102. }
  103. if (length == 0) {
  104. value = {};
  105. return {};
  106. }
  107. value = TRY(ByteBuffer::create_uninitialized(length));
  108. m_stream >> value.bytes();
  109. return m_stream.try_handle_any_error();
  110. }
  111. ErrorOr<void> Decoder::decode(JsonValue& value)
  112. {
  113. String string;
  114. TRY(decode(string));
  115. value = TRY(JsonValue::from_string(string));
  116. return {};
  117. }
  118. ErrorOr<void> Decoder::decode(URL& value)
  119. {
  120. String string;
  121. TRY(decode(string));
  122. value = URL(string);
  123. return {};
  124. }
  125. ErrorOr<void> Decoder::decode(Dictionary& dictionary)
  126. {
  127. u64 size;
  128. TRY(decode(size));
  129. if (size >= (size_t)NumericLimits<i32>::max())
  130. VERIFY_NOT_REACHED();
  131. for (size_t i = 0; i < size; ++i) {
  132. String key;
  133. TRY(decode(key));
  134. String value;
  135. TRY(decode(value));
  136. dictionary.add(move(key), move(value));
  137. }
  138. return {};
  139. }
  140. ErrorOr<void> Decoder::decode([[maybe_unused]] File& file)
  141. {
  142. int fd = TRY(m_socket.receive_fd(O_CLOEXEC));
  143. file = File(fd, File::ConstructWithReceivedFileDescriptor);
  144. return {};
  145. }
  146. template<>
  147. ErrorOr<void> decode(Decoder& decoder, Core::AnonymousBuffer& buffer)
  148. {
  149. bool valid;
  150. TRY(decoder.decode(valid));
  151. if (!valid) {
  152. buffer = {};
  153. return {};
  154. }
  155. u32 size;
  156. TRY(decoder.decode(size));
  157. IPC::File anon_file;
  158. TRY(decoder.decode(anon_file));
  159. buffer = TRY(Core::AnonymousBuffer::create_from_anon_fd(anon_file.take_fd(), size));
  160. return {};
  161. }
  162. template<>
  163. ErrorOr<void> decode(Decoder& decoder, Core::DateTime& datetime)
  164. {
  165. i64 timestamp;
  166. TRY(decoder.decode(timestamp));
  167. datetime = Core::DateTime::from_timestamp(static_cast<time_t>(timestamp));
  168. return {};
  169. }
  170. template<>
  171. ErrorOr<void> decode(Decoder& decoder, Core::ProxyData& data)
  172. {
  173. UnderlyingType<decltype(data.type)> type;
  174. TRY(decoder.decode(type));
  175. data.type = static_cast<Core::ProxyData::Type>(type);
  176. TRY(decoder.decode(data.host_ipv4));
  177. TRY(decoder.decode(data.port));
  178. return {};
  179. }
  180. }