Skip to content

Commit

Permalink
Use interface whitelist for localhost only (ros2#476)
Browse files Browse the repository at this point in the history
* Use interface whitelist for localhost only

Signed-off-by: Miguel Company <MiguelCompany@eprosima.com>
  • Loading branch information
MiguelCompany authored and fujitatomoya committed Mar 4, 2021
1 parent 684b10b commit b65cdb6
Showing 1 changed file with 21 additions and 8 deletions.
29 changes: 21 additions & 8 deletions rmw_fastrtps_shared_cpp/src/participant.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@

#include <limits.h>
#include <string>
#include <memory>

#include "fastrtps/config.h"
#include "fastrtps/Domain.h"
Expand All @@ -33,6 +34,7 @@
#include "fastrtps/subscriber/Subscriber.h"
#include "fastrtps/subscriber/SubscriberListener.h"
#include "fastrtps/subscriber/SampleInfo.h"
#include "fastrtps/transport/UDPv4TransportDescriptor.h"

#include "rcutils/filesystem.h"
#include "rcutils/get_env.h"
Expand All @@ -50,6 +52,12 @@ using Locator_t = eprosima::fastrtps::rtps::Locator_t;
using Participant = eprosima::fastrtps::Participant;
using ParticipantAttributes = eprosima::fastrtps::ParticipantAttributes;
using StatefulReader = eprosima::fastrtps::rtps::StatefulReader;
using UDPv4TransportDescriptor = eprosima::fastrtps::rtps::UDPv4TransportDescriptor;

#if FASTRTPS_VERSION_MAJOR >= 2
#include "fastdds/rtps/transport/shared_mem/SharedMemTransportDescriptor.h"
using SharedMemTransportDescriptor = eprosima::fastdds::rtps::SharedMemTransportDescriptor;
#endif

#if HAVE_SECURITY
static
Expand Down Expand Up @@ -154,14 +162,19 @@ rmw_fastrtps_shared_cpp::create_participant(
Domain::getDefaultParticipantAttributes(participantAttrs);

if (localhost_only) {
Locator_t local_network_interface_locator;
static const std::string local_ip_name("127.0.0.1");
local_network_interface_locator.kind = 1;
local_network_interface_locator.port = 0;
IPLocator::setIPv4(local_network_interface_locator, local_ip_name);
participantAttrs.rtps.builtin.metatrafficUnicastLocatorList.push_back(
local_network_interface_locator);
participantAttrs.rtps.builtin.initialPeersList.push_back(local_network_interface_locator);
// In order to use the interface white list, we need to disable the default transport config
participantAttrs.rtps.useBuiltinTransports = false;

// Add a UDPv4 transport with only localhost enabled
auto udp_transport = std::make_shared<UDPv4TransportDescriptor>();
udp_transport->interfaceWhiteList.emplace_back("127.0.0.1");
participantAttrs.rtps.userTransports.push_back(udp_transport);

// Add SHM transport if available
#if FASTRTPS_VERSION_MAJOR >= 2
auto shm_transport = std::make_shared<SharedMemTransportDescriptor>();
participantAttrs.rtps.userTransports.push_back(shm_transport);
#endif
}

// No custom handling of RMW_DEFAULT_DOMAIN_ID. Simply use a reasonable domain id.
Expand Down

0 comments on commit b65cdb6

Please sign in to comment.