src/libsystemd-network/test-ipv4ll-manual.c

   1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
   2
   3 /* Make sure the net/if.h header is included before any linux/ one */
   4 #include <net/if.h>
   5 #include <errno.h>
   6 #include <linux/veth.h>
   7 #include <stdlib.h>
   8 #include <unistd.h>
   9
  10 #include "sd-event.h"
  11 #include "sd-ipv4ll.h"
  12 #include "sd-netlink.h"
  13
  14 #include "alloc-util.h"
  15 #include "in-addr-util.h"
  16 #include "parse-util.h"
  17 #include "string-util.h"
  18 #include "tests.h"
  19
  20 static void ll_handler(sd_ipv4ll *ll, int event, void *userdata) {
  21         assert_se(ll);
  22
  23         struct in_addr addr;
  24         const char *pretty = sd_ipv4ll_get_address(ll, &addr) >= 0 ? IN4_ADDR_TO_STRING(&addr) : NULL;
  25
  26         switch (event) {
  27         case SD_IPV4LL_EVENT_BIND:
  28                 log_info("bound %s", strna(pretty));
  29                 break;
  30         case SD_IPV4LL_EVENT_CONFLICT:
  31                 log_info("conflict on %s", strna(pretty));
  32                 break;
  33         case SD_IPV4LL_EVENT_STOP:
  34                 log_error("the client was stopped with address %s", strna(pretty));
  35                 break;
  36         default:
  37                 assert_not_reached();
  38         }
  39 }
  40
  41 static int client_run(int ifindex, const char *seed_str, const struct in_addr *start_address, const struct ether_addr *ha, sd_event *e) {
  42         sd_ipv4ll *ll;
  43
  44         assert_se(sd_ipv4ll_new(&ll) >= 0);
  45         assert_se(sd_ipv4ll_attach_event(ll, e, 0) >= 0);
  46
  47         assert_se(sd_ipv4ll_set_ifindex(ll, ifindex) >= 0);
  48         assert_se(sd_ipv4ll_set_mac(ll, ha) >= 0);
  49         assert_se(sd_ipv4ll_set_callback(ll, ll_handler, NULL) >= 0);
  50
  51         if (seed_str) {
  52                 unsigned seed;
  53
  54                 assert_se(safe_atou(seed_str, &seed) >= 0);
  55
  56                 assert_se(sd_ipv4ll_set_address_seed(ll, seed) >= 0);
  57         }
  58
  59         if (start_address && in4_addr_is_set(start_address))
  60                 assert_se(sd_ipv4ll_set_address(ll, start_address) >= 0);
  61
  62         log_info("starting IPv4LL client");
  63
  64         assert_se(sd_ipv4ll_start(ll) >= 0);
  65
  66         assert_se(sd_event_loop(e) >= 0);
  67
  68         assert_se(!sd_ipv4ll_unref(ll));
  69
  70         return EXIT_SUCCESS;
  71 }
  72
  73 static int test_ll(const char *ifname, const char *seed, const struct in_addr *start_address) {
  74         _cleanup_(sd_event_unrefp) sd_event *e = NULL;
  75         _cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL;
  76         _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL, *reply = NULL;
  77         struct ether_addr ha;
  78         int ifindex;
  79
  80         assert_se(sd_event_new(&e) >= 0);
  81
  82         assert_se(sd_netlink_open(&rtnl) >= 0);
  83         assert_se(sd_netlink_attach_event(rtnl, e, 0) >= 0);
  84
  85         assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, 0) >= 0);
  86         assert_se(sd_netlink_message_append_string(m, IFLA_IFNAME, ifname) >= 0);
  87         assert_se(sd_netlink_call(rtnl, m, 0, &reply) >= 0);
  88
  89         assert_se(sd_rtnl_message_link_get_ifindex(reply, &ifindex) >= 0);
  90         assert_se(sd_netlink_message_read_ether_addr(reply, IFLA_ADDRESS, &ha) >= 0);
  91
  92         client_run(ifindex, seed, start_address, &ha, e);
  93
  94         return EXIT_SUCCESS;
  95 }
  96
  97 int main(int argc, char *argv[]) {
  98         test_setup_logging(LOG_DEBUG);
  99
 100         if (argc == 2)
 101                 return test_ll(argv[1], NULL, NULL);
 102         else if (argc == 3) {
 103                 int r;
 104                 union in_addr_union a;
 105
 106                 r = in_addr_from_string(AF_INET, argv[2], &a);
 107                 if (r < 0)
 108                         return test_ll(argv[1], argv[2], NULL);
 109                 else
 110                         return test_ll(argv[1], NULL, &a.in);
 111         } else {
 112                 log_error("This program takes one or two arguments.\n"
 113                           "\t %s <ifname> [<seed>|<start_address>]", program_invocation_short_name);
 114                 return EXIT_FAILURE;
 115         }
 116 }