kep-1144 Dropping messages that do not have the correct swarm id

bluzelle · Apr 17, 2019 · 9e846a4 · 9e846a4
1 parent dfbc949
commit 9e846a4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/node/node.cpp b/node/node.cpp
@@ -134,6 +134,12 @@ node::priv_protobuf_handler(const bzn_envelope& msg, std::shared_ptr<bzn::sessio
 {
     std::shared_lock<std::shared_mutex> lock(this->message_map_mutex); // lock for read access
 
+    if (msg.swarm_id() != this->options->get_swarm_id())
+    {
+        LOG(error) << "Dropping message with invalid swarm id: " << msg.ShortDebugString().substr(0, MAX_MESSAGE_SIZE);
+        return;
+    }
+
     if ((!msg.sender().empty()) && (!this->crypto->verify(msg)))
     {
         LOG(error) << "Dropping message with invalid signature: " << msg.ShortDebugString().substr(0, MAX_MESSAGE_SIZE);
@@ -148,7 +154,6 @@ node::priv_protobuf_handler(const bzn_envelope& msg, std::shared_ptr<bzn::sessio
     {
         LOG(debug) << "no handler for message type " << msg.payload_case();
     }
-
 }
 
 void