Quelle  lwt_seg6local.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0-only

/* Connects 6 network namespaces through veths.
 * Each NS may have different IPv6 global scope addresses :
 *
 *          NS1            NS2             NS3              NS4               NS5             NS6
 *      lo  veth1 <-> veth2 veth3 <-> veth4 veth5 <-> veth6 lo veth7 <-> veth8 veth9 <-> veth10 lo
 * fb00 ::1  ::12      ::21  ::34      ::43  ::56      ::65     ::78      ::87  ::910     ::109  ::6
 * fd00                                                                                          ::4
 * fc42                                                     ::1
 *
 * All IPv6 packets going to fb00::/16 through NS2 will be encapsulated in a
 * IPv6 header with a Segment Routing Header, with segments :
 * fd00::1 -> fd00::2 -> fd00::3 -> fd00::4
 *
 * 3 fd00::/16 IPv6 addresses are binded to seg6local End.BPF actions :
 * - fd00::1 : add a TLV, change the flags and apply a End.X action to fc42::1
 * - fd00::2 : remove the TLV, change the flags, add a tag
 * - fd00::3 : apply an End.T action to fd00::4, through routing table 117
 *
 * fd00::4 is a simple Segment Routing node decapsulating the inner IPv6 packet.
 * Each End.BPF action will validate the operations applied on the SRH by the
 * previous BPF program in the chain, otherwise the packet is dropped.
 *
 * An UDP datagram is sent from fb00::1 to fb00::6. The test succeeds if this
 * datagram can be read on NS6 when binding to fb00::6.
 */

#include "network_helpers.h"
#include "test_progs.h"

#define NETNS_BASE "lwt-seg6local-"
#define BPF_FILE "test_lwt_seg6local.bpf.o"

static void cleanup(void)
{
 int ns;

 for (ns = 1; ns < 7; ns++)
  SYS_NOFAIL("ip netns del %s%d", NETNS_BASE, ns);
}

static int setup(void)
{
 int ns;

 for (ns = 1; ns < 7; ns++)
  SYS(fail, "ip netns add %s%d", NETNS_BASE, ns);

 SYS(fail, "ip -n %s6 link set dev lo up", NETNS_BASE);

 for (ns = 1; ns < 6; ns++) {
  int local_id = ns * 2 - 1;
  int peer_id = ns * 2;
  int next_ns = ns + 1;

  SYS(fail, "ip -n %s%d link add veth%d type veth peer name veth%d netns %s%d",
      NETNS_BASE, ns, local_id, peer_id, NETNS_BASE, next_ns);

  SYS(fail, "ip -n %s%d link set dev veth%d up", NETNS_BASE, ns, local_id);
  SYS(fail, "ip -n %s%d link set dev veth%d up", NETNS_BASE, next_ns, peer_id);

  /* All link scope addresses to veths */
  SYS(fail, "ip -n %s%d -6 addr add fb00::%d%d/16 dev veth%d scope link",
      NETNS_BASE, ns, local_id, peer_id, local_id);
  SYS(fail, "ip -n %s%d -6 addr add fb00::%d%d/16 dev veth%d scope link",
      NETNS_BASE, next_ns, peer_id, local_id, peer_id);
 }


 SYS(fail, "ip -n %s5 -6 route add fb00::109 table 117 dev veth9 scope link", NETNS_BASE);

 SYS(fail, "ip -n %s1 -6 addr add fb00::1/16 dev lo", NETNS_BASE);
 SYS(fail, "ip -n %s1 -6 route add fb00::6 dev veth1 via fb00::21", NETNS_BASE);

 SYS(fail, "ip -n %s2 -6 route add fb00::6 encap bpf in obj %s sec encap_srh dev veth2",
     NETNS_BASE, BPF_FILE);
 SYS(fail, "ip -n %s2 -6 route add fd00::1 dev veth3 via fb00::43 scope link", NETNS_BASE);

 SYS(fail, "ip -n %s3 -6 route add fc42::1 dev veth5 via fb00::65", NETNS_BASE);
 SYS(fail,
     "ip -n %s3 -6 route add fd00::1 encap seg6local action End.BPF endpoint obj %s sec add_egr_x dev veth4",
     NETNS_BASE, BPF_FILE);

 SYS(fail,
     "ip -n %s4 -6 route add fd00::2 encap seg6local action End.BPF endpoint obj %s sec pop_egr dev veth6",
     NETNS_BASE, BPF_FILE);
 SYS(fail, "ip -n %s4 -6 addr add fc42::1 dev lo", NETNS_BASE);
 SYS(fail, "ip -n %s4 -6 route add fd00::3 dev veth7 via fb00::87", NETNS_BASE);

 SYS(fail, "ip -n %s5 -6 route add fd00::4 table 117 dev veth9 via fb00::109", NETNS_BASE);
 SYS(fail,
     "ip -n %s5 -6 route add fd00::3 encap seg6local action End.BPF endpoint obj %s sec inspect_t dev veth8",
     NETNS_BASE, BPF_FILE);

 SYS(fail, "ip -n %s6 -6 addr add fb00::6/16 dev lo", NETNS_BASE);
 SYS(fail, "ip -n %s6 -6 addr add fd00::4/16 dev lo", NETNS_BASE);

 for (ns = 1; ns < 6; ns++)
  SYS(fail, "ip netns exec %s%d sysctl -wq net.ipv6.conf.all.forwarding=1",
      NETNS_BASE, ns);

 SYS(fail, "ip netns exec %s6 sysctl -wq net.ipv6.conf.all.seg6_enabled=1", NETNS_BASE);
 SYS(fail, "ip netns exec %s6 sysctl -wq net.ipv6.conf.lo.seg6_enabled=1", NETNS_BASE);
 SYS(fail, "ip netns exec %s6 sysctl -wq net.ipv6.conf.veth10.seg6_enabled=1", NETNS_BASE);

 return 0;
fail:
 return -1;
}

#define SERVER_PORT 7330
#define CLIENT_PORT 2121
void test_lwt_seg6local(void)
{
 struct sockaddr_in6 server_addr = {};
 const char *ns1 = NETNS_BASE "1";
 const char *ns6 = NETNS_BASE "6";
 struct nstoken *nstoken = NULL;
 const char *foobar = "foobar";
 ssize_t bytes;
 int sfd, cfd;
 char buf[7];

 if (!ASSERT_OK(setup(), "setup"))
  goto out;

 nstoken = open_netns(ns6);
 if (!ASSERT_OK_PTR(nstoken, "open ns6"))
  goto out;

 sfd = start_server_str(AF_INET6, SOCK_DGRAM, "fb00::6", SERVER_PORT, NULL);
 if (!ASSERT_OK_FD(sfd, "start server"))
  goto close_netns;

 close_netns(nstoken);

 nstoken = open_netns(ns1);
 if (!ASSERT_OK_PTR(nstoken, "open ns1"))
  goto close_server;

 cfd = start_server_str(AF_INET6, SOCK_DGRAM, "fb00::1", CLIENT_PORT, NULL);
 if (!ASSERT_OK_FD(cfd, "start client"))
  goto close_server;

 close_netns(nstoken);
 nstoken = NULL;

 /* Send a packet larger than MTU */
 server_addr.sin6_family = AF_INET6;
 server_addr.sin6_port = htons(SERVER_PORT);
 if (!ASSERT_EQ(inet_pton(AF_INET6, "fb00::6", &server_addr.sin6_addr), 1,
         "build target addr"))
  goto close_client;

 bytes = sendto(cfd, foobar, sizeof(foobar), 0,
         (struct sockaddr *)&server_addr, sizeof(server_addr));
 if (!ASSERT_EQ(bytes, sizeof(foobar), "send packet"))
  goto close_client;

 /* Verify we received all expected bytes */
 bytes = read(sfd, buf, sizeof(buf));
 if (!ASSERT_EQ(bytes, sizeof(buf), "receive packet"))
  goto close_client;
 ASSERT_STREQ(buf, foobar, "check udp packet");

close_client:
 close(cfd);
close_server:
 close(sfd);
close_netns:
 close_netns(nstoken);

out:
 cleanup();
}