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

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 /***
   3   This file is part of systemd.
   4
   5   Copyright (C) 2014 Tom Gundersen <teg@jklm.no>
   6
   7   systemd is free software; you can redistribute it and/or modify it
   8   under the terms of the GNU Lesser General Public License as published by
   9   the Free Software Foundation; either version 2.1 of the License, or
  10   (at your option) any later version.
  11
  12   systemd is distributed in the hope that it will be useful, but
  13   WITHOUT ANY WARRANTY; without even the implied warranty of
  14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15   Lesser General Public License for more details.
  16
  17   You should have received a copy of the GNU Lesser General Public License
  18   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  19 ***/
  20
  21 #include <errno.h>
  22 #include <net/if.h>
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25 #include <linux/veth.h>
  26
  27 #include "sd-event.h"
  28 #include "sd-ipv4ll.h"
  29 #include "sd-netlink.h"
  30
  31 #include "alloc-util.h"
  32 #include "in-addr-util.h"
  33 #include "netlink-util.h"
  34 #include "parse-util.h"
  35 #include "string-util.h"
  36 #include "util.h"
  37
  38 static void ll_handler(sd_ipv4ll *ll, int event, void *userdata) {
  39         _cleanup_free_ char *address = NULL;
  40         struct in_addr addr = {};
  41
  42         assert_se(ll);
  43
  44         if (sd_ipv4ll_get_address(ll, &addr) >= 0)
  45                 assert_se(in_addr_to_string(AF_INET, (const union in_addr_union*) &addr, &address) >= 0);
  46
  47         switch (event) {
  48         case SD_IPV4LL_EVENT_BIND:
  49                 log_info("bound %s", strna(address));
  50                 break;
  51         case SD_IPV4LL_EVENT_CONFLICT:
  52                 log_info("conflict on %s", strna(address));
  53                 break;
  54         case SD_IPV4LL_EVENT_STOP:
  55                 log_error("the client was stopped with address %s", strna(address));
  56                 break;
  57         default:
  58                 assert_not_reached("invalid LL event");
  59         }
  60 }
  61
  62 static int client_run(int ifindex, const char *seed_str, const struct ether_addr *ha, sd_event *e) {
  63         sd_ipv4ll *ll;
  64
  65         assert_se(sd_ipv4ll_new(&ll) >= 0);
  66         assert_se(sd_ipv4ll_attach_event(ll, e, 0) >= 0);
  67
  68         assert_se(sd_ipv4ll_set_ifindex(ll, ifindex) >= 0);
  69         assert_se(sd_ipv4ll_set_mac(ll, ha) >= 0);
  70         assert_se(sd_ipv4ll_set_callback(ll, ll_handler, NULL) >= 0);
  71
  72         if (seed_str) {
  73                 unsigned seed;
  74
  75                 assert_se(safe_atou(seed_str, &seed) >= 0);
  76
  77                 assert_se(sd_ipv4ll_set_address_seed(ll, seed) >= 0);
  78         }
  79
  80         log_info("starting IPv4LL client");
  81
  82         assert_se(sd_ipv4ll_start(ll) >= 0);
  83
  84         assert_se(sd_event_loop(e) >= 0);
  85
  86         assert_se(!sd_ipv4ll_unref(ll));
  87
  88         return EXIT_SUCCESS;
  89 }
  90
  91 static int test_ll(const char *ifname, const char *seed) {
  92         _cleanup_(sd_event_unrefp) sd_event *e = NULL;
  93         _cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL;
  94         _cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL, *reply = NULL;
  95         struct ether_addr ha;
  96         int ifindex;
  97
  98         assert_se(sd_event_new(&e) >= 0);
  99
 100         assert_se(sd_netlink_open(&rtnl) >= 0);
 101         assert_se(sd_netlink_attach_event(rtnl, e, 0) >= 0);
 102
 103         assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, 0) >= 0);
 104         assert_se(sd_netlink_message_append_string(m, IFLA_IFNAME, ifname) >= 0);
 105         assert_se(sd_netlink_call(rtnl, m, 0, &reply) >= 0);
 106
 107         assert_se(sd_rtnl_message_link_get_ifindex(reply, &ifindex) >= 0);
 108         assert_se(sd_netlink_message_read_ether_addr(reply, IFLA_ADDRESS, &ha) >= 0);
 109
 110         client_run(ifindex, seed, &ha, e);
 111
 112         return EXIT_SUCCESS;
 113 }
 114
 115 int main(int argc, char *argv[]) {
 116         log_set_max_level(LOG_DEBUG);
 117         log_parse_environment();
 118         log_open();
 119
 120         if (argc == 2)
 121                 return test_ll(argv[1], NULL);
 122         else if (argc == 3)
 123                 return test_ll(argv[1], argv[2]);
 124         else {
 125                 log_error("This program takes one or two arguments.\n"
 126                           "\t %s <ifname> [<seed>]", program_invocation_short_name);
 127                 return EXIT_FAILURE;
 128         }
 129 }