Added the ability to use random source address in route tunnel

A similar feature was available in tun64 and balancer. The ipv6 src address from 65 to 96 bits is written to the ipv4 src of the original packet.
yanet-platform · Dec 15, 2024 · c9f0cbb · c9f0cbb
1 parent 20ad724
commit c9f0cbb
Show file tree

Hide file tree

Showing 7 changed files with 20 additions and 3 deletions.
diff --git a/common/controlplaneconfig.h b/common/controlplaneconfig.h
@@ -128,6 +128,7 @@ class config_t
 	std::set<common::ip_prefix_t> local_prefixes; ///< for fallback to default
 	std::map<uint32_t, std::string> peers;
 	std::map<std::string, interface_t> interfaces;
+	bool random_source{};
 };
 
 }

diff --git a/common/idp.h b/common/idp.h
@@ -199,7 +199,8 @@ namespace update_route
 {
 using tunnel = std::tuple<ipv4_address_t, ///< ipv4AddressSource
                           ipv6_address_t, ///< ipv6AddressSource
-                          uint16_t>; ///< udpDestinationPort
+                          uint16_t, ///< udpDestinationPort
+                          bool>; ///< srcRndEnabled
 
 using request = std::tuple<tRouteId,
                            std::optional<tunnel>>;

diff --git a/controlplane/configconverter.cpp b/controlplane/configconverter.cpp
@@ -432,7 +432,8 @@ void config_converter_t::serializeRoutes()
 		{
 			tunnel = {route.ipv4_source_address,
 			          route.ipv6_source_address,
-			          route.udp_destination_port};
+			          route.udp_destination_port,
+			          route.random_source};
 		}
 
 		globalbase.emplace_back(common::idp::updateGlobalBase::requestType::update_route,

diff --git a/controlplane/configparser.cpp b/controlplane/configparser.cpp
@@ -367,6 +367,11 @@ void config_parser_t::loadConfig_route(controlplane::base_t& baseNext,
 		route.tunnel_enabled = false;
 	}
 
+	if (exist(moduleJson, "random_source"))
+	{
+		route.random_source = moduleJson["random_source"].get<bool>();
+	}
+
 	//
 
 	route.routeId = routeId;

diff --git a/dataplane/globalbase.cpp b/dataplane/globalbase.cpp
@@ -1042,11 +1042,12 @@ eResult generation::update_route(const common::idp::updateGlobalBase::update_rou
 
 	if (tunnel)
 	{
-		const auto& [ipv4AddressSource, ipv6AddressSource, udpDestinationPort] = *tunnel;
+		const auto& [ipv4AddressSource, ipv6AddressSource, udpDestinationPort, randomSource] = *tunnel;
 
 		route.ipv4AddressSource = ipv4_address_t::convert(ipv4AddressSource);
 		route.ipv6AddressSource = ipv6_address_t::convert(ipv6AddressSource);
 		route.udpDestinationPort = rte_cpu_to_be_16(udpDestinationPort);
+		route.randomSource = randomSource;
 	}
 
 	return eResult::success;

diff --git a/dataplane/type.h b/dataplane/type.h
@@ -287,6 +287,7 @@ struct route_t
 	ipv4_address_t ipv4AddressSource;
 	ipv6_address_t ipv6AddressSource;
 	uint16_t udpDestinationPort;
+	bool randomSource;
 };
 
 struct tInterface

diff --git a/dataplane/worker.cpp b/dataplane/worker.cpp
@@ -2761,12 +2761,14 @@ inline void cWorker::route_tunnel_nexthop(rte_mbuf* mbuf,
 	{
 		uint32_t vtc_flow = 0;
 		uint16_t payload_len = 0;
+		uint32_t ipv4_src_addr;
 		if (is_ipv4)
 		{
 			rte_ipv4_hdr* ipv4HeaderInner = rte_pktmbuf_mtod_offset(mbuf, rte_ipv4_hdr*, metadata->network_headerOffset);
 			vtc_flow = rte_cpu_to_be_32((0x6 << 28) | (ipv4HeaderInner->type_of_service << 20)); ///< @todo: flow label
 			payload_len = rte_cpu_to_be_16((is_ipip_tunnel ? 0 : sizeof(rte_udp_hdr) + YADECAP_MPLS_HEADER_SIZE) + rte_be_to_cpu_16(ipv4HeaderInner->total_length));
 			payload_length = rte_be_to_cpu_16(ipv4HeaderInner->total_length);
+			ipv4_src_addr = ipv4HeaderInner->src_addr;
 		}
 		else
 		{
@@ -2797,6 +2799,11 @@ inline void cWorker::route_tunnel_nexthop(rte_mbuf* mbuf,
 		rte_memcpy(ipv6Header->src_addr, route.ipv6AddressSource.bytes, 16);
 		rte_memcpy(ipv6Header->dst_addr, nexthop.nexthop_address.bytes, 16);
 
+		if (route.randomSource && is_ipv4)
+		{
+			((uint32_t*)ipv6Header->src_addr)[2] = ipv4_src_addr;
+		}
+
 		metadata->transport_headerOffset = metadata->network_headerOffset + sizeof(rte_ipv6_hdr);
 	}