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

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