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

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