Refs #21120. Fix discovery server tests.

Signed-off-by: Miguel Company <miguelcompany@eprosima.com>
eProsima · Jun 6, 2024 · 258078d · 258078d
1 parent 7901acd
commit 258078d
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 3 deletions.
diff --git a/src/cpp/rtps/builtin/discovery/participant/PDPClient.cpp b/src/cpp/rtps/builtin/discovery/participant/PDPClient.cpp
@@ -44,6 +44,7 @@
 #include <rtps/builtin/discovery/participant/timedevent/DSClientEvent.h>
 #include <rtps/builtin/liveliness/WLP.hpp>
 #include <rtps/participant/RTPSParticipantImpl.h>
+#include <rtps/reader/BaseReader.hpp>
 #include <rtps/reader/StatefulReader.hpp>
 #include <rtps/writer/StatefulWriter.hpp>
 #include <utils/shared_mutex.hpp>
@@ -348,7 +349,6 @@ bool PDPClient::create_ds_pdp_reliable_endpoints(
     ratt.endpoint.reliabilityKind = RELIABLE;
     ratt.times.heartbeatResponseDelay = pdp_heartbeat_response_delay;
 #if HAVE_SECURITY
-    ratt.accept_messages_from_unkown_writers = !is_discovery_protected && mp_RTPSParticipant->is_secure();
     if (is_discovery_protected)
     {
         ratt.endpoint.security_attributes().is_submessage_protected = true;
@@ -460,6 +460,10 @@ bool PDPClient::create_ds_pdp_reliable_endpoints(
                 match_pdp_writer_nts_(it);
                 match_pdp_reader_nts_(it);
             }
+            else if (!is_discovery_protected)
+            {
+                BaseReader::downcast(endpoints.reader.reader_)->allow_unknown_writers();
+            }
 #else
             if (!is_discovery_protected)
             {

diff --git a/src/cpp/rtps/builtin/discovery/participant/PDPServer.cpp b/src/cpp/rtps/builtin/discovery/participant/PDPServer.cpp
@@ -45,6 +45,7 @@
 #include <rtps/builtin/discovery/participant/timedevent/DServerEvent.hpp>
 #include <rtps/builtin/liveliness/WLP.hpp>
 #include <rtps/participant/RTPSParticipantImpl.h>
+#include <rtps/reader/BaseReader.hpp>
 #include <rtps/reader/StatefulReader.hpp>
 #include <rtps/writer/StatefulWriter.hpp>
 #include <utils/TimeConversion.hpp>
@@ -401,13 +402,14 @@ bool PDPServer::create_ds_pdp_reliable_endpoints(
 #else
     EntityId_t reader_entity = c_EntityId_SPDPReader;
 #endif // if HAVE_SECURITY
-    // Enable unknown clients to reach this reader
-    ratt.accept_messages_from_unkown_writers = true;
     if (mp_RTPSParticipant->createReader(&reader, ratt, endpoints.reader.history_.get(),
             endpoints.reader.listener_.get(), reader_entity, true, false))
     {
         endpoints.reader.reader_ = dynamic_cast<fastrtps::rtps::StatefulReader*>(reader);
 
+        // Enable unknown clients to reach this reader
+        BaseReader::downcast(endpoints.reader.reader_)->allow_unknown_writers();
+
 #if HAVE_SECURITY
         mp_RTPSParticipant->set_endpoint_rtps_protection_supports(reader, false);
 #endif // if HAVE_SECURITY

diff --git a/src/cpp/rtps/reader/BaseReader.hpp b/src/cpp/rtps/reader/BaseReader.hpp
@@ -126,6 +126,11 @@ class BaseReader
         trusted_writer_entity_id_ = writer;
     }
 
+    void allow_unknown_writers()
+    {
+        accept_messages_from_unkown_writers_ = true;
+    }
+
     /**
      * Whether the reader accepts messages directed to unknown readers.
      *