Encoder.cpp 4.7 KB

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