fixes for recent integration

cisco · Sep 7, 2023 · c7d0e40 · c7d0e40
1 parent 0e6ce41
commit c7d0e40
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/include/mls/messages.h b/include/mls/messages.h
@@ -592,6 +592,8 @@ struct PublicMessage
 
   friend tls::ostream& operator<<(tls::ostream& str, const PublicMessage& obj);
   friend tls::istream& operator>>(tls::istream& str, PublicMessage& obj);
+  friend bool operator==(const PublicMessage& lhs, const PublicMessage& rhs);
+  friend bool operator!=(const PublicMessage& lhs, const PublicMessage& rhs);
 
 private:
   GroupContent content;

diff --git a/lib/bytes/include/bytes/bytes.h b/lib/bytes/include/bytes/bytes.h
@@ -79,6 +79,8 @@ struct bytes
   auto& at(size_t pos) { return _data.at(pos); }
 
   void resize(size_t count) { _data.resize(count); }
+  void reserve(size_t len) { _data.reserve(len); }
+  void push_back(uint8_t byte) { _data.push_back(byte); }
 
   // Equality operators
   bool operator==(const bytes& other) const;

diff --git a/src/messages.cpp b/src/messages.cpp
@@ -630,6 +630,19 @@ operator>>(tls::istream& str, PublicMessage& obj)
   return str;
 }
 
+bool
+operator==(const PublicMessage& lhs, const PublicMessage& rhs)
+{
+  return lhs.content == rhs.content && lhs.auth == rhs.auth &&
+         lhs.membership_tag == rhs.membership_tag;
+}
+
+bool
+operator!=(const PublicMessage& lhs, const PublicMessage& rhs)
+{
+  return !(lhs == rhs);
+}
+
 static bytes
 marshal_ciphertext_content(const GroupContent& content,
                            const GroupContentAuthData& auth,