Add an equality operator for PublicMessage

cisco · Sep 5, 2023 · 78d5957 · 78d5957
1 parent 949c447
commit 78d5957
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/include/mls/messages.h b/include/mls/messages.h
@@ -592,6 +592,7 @@ 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);
 
 private:
   GroupContent content;

diff --git a/src/messages.cpp b/src/messages.cpp
@@ -630,6 +630,14 @@ 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;
+}
+
 static bytes
 marshal_ciphertext_content(const GroupContent& content,
                            const GroupContentAuthData& auth,