LibIPC: Add encoder and decoder for AK::OrderedHashMap

Seems like a useful thing to have.
cubiclove · Apr 3, 2022 · f2b4c04 · f2b4c04
1 parent a8a50a9
commit f2b4c04
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/Userland/Libraries/LibIPC/Decoder.h b/Userland/Libraries/LibIPC/Decoder.h
@@ -67,6 +67,24 @@ class Decoder {
  return {};
  }
 
+ template<typename K, typename V>
+ ErrorOr<void> decode(OrderedHashMap<K, V>& hashmap)
+ {
+ u32 size;
+ TRY(decode(size));
+ if (size > NumericLimits<i32>::max())
+ return Error::from_string_literal("IPC: Invalid HashMap size"sv);
+
+ for (size_t i = 0; i < size; ++i) {
+ K key;
+ TRY(decode(key));
+ V value;
+ TRY(decode(value));
+ TRY(hashmap.try_set(move(key), move(value)));
+ }
+ return {};
+ }
+
  template<Enum T>
  ErrorOr<void> decode(T& enum_value)
  {

diff --git a/Userland/Libraries/LibIPC/Encoder.h b/Userland/Libraries/LibIPC/Encoder.h
@@ -57,6 +57,17 @@ class Encoder {
  return *this;
  }
 
+ template<typename K, typename V>
+ Encoder& operator<<(OrderedHashMap<K, V> const& hashmap)
+ {
+ *this << (u32)hashmap.size();
+ for (auto it : hashmap) {
+ *this << it.key;
+ *this << it.value;
+ }
+ return *this;
+ }
+
  template<typename T>
  Encoder& operator<<(Vector<T> const& vector)
  {