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 "alloc-util.h"
  33 #include "event-util.h"
  34 #include "in-addr-util.h"
  35 #include "netlink-util.h"
  36 #include "parse-util.h"
  37 #include "string-util.h"
  38 #include "util.h"
  39
  40 static void ll_handler(sd_ipv4ll *ll, int event, void *userdata) {
  41         _cleanup_free_ char *address = NULL;
  42         struct in_addr addr = {};
  43
  44         assert_se(ll);
  45
  46         if (sd_ipv4ll_get_address(ll, &addr) >= 0)
  47                 assert_se(in_addr_to_string(AF_INET, (const union in_addr_union*) &addr, &address) >= 0);
  48
  49         switch (event) {
  50         case SD_IPV4LL_EVENT_BIND:
  51                 log_info("bound %s", strna(address));
  52                 break;
  53         case SD_IPV4LL_EVENT_CONFLICT:
  54                 log_info("conflict on %s", strna(address));
  55                 break;
  56         case SD_IPV4LL_EVENT_STOP:
  57                 log_error("the client was stopped with address %s", strna(address));
  58                 break;
  59         default:
  60                 assert_not_reached("invalid LL event");
  61         }
  62 }
  63
  64 static int client_run(int ifindex, const char *seed_str, const struct ether_addr *ha, sd_event *e) {
  65         sd_ipv4ll *ll;
  66
  67         assert_se(sd_ipv4ll_new(&ll) >= 0);
  68         assert_se(sd_ipv4ll_attach_event(ll, e, 0) >= 0);
  69
  70         assert_se(sd_ipv4ll_set_index(ll, ifindex) >= 0);
  71         assert_se(sd_ipv4ll_set_mac(ll, ha) >= 0);
  72         assert_se(sd_ipv4ll_set_callback(ll, ll_handler, NULL) >= 0);
  73
  74         if (seed_str) {
  75                 unsigned seed;
  76
  77                 assert_se(safe_atou(seed_str, &seed) >= 0);
  78
  79                 assert_se(sd_ipv4ll_set_address_seed(ll, seed) >= 0);
  80         }
  81
  82         log_info("starting IPv4LL client");
  83
  84         assert_se(sd_ipv4ll_start(ll) >= 0);
  85
  86         assert_se(sd_event_loop(e) >= 0);
  87
  88         assert_se(!sd_ipv4ll_unref(ll));
  89
  90         return EXIT_SUCCESS;
  91 }
  92
  93 static int test_ll(const char *ifname, const char *seed) {
  94         _cleanup_event_unref_ sd_event *e = NULL;
  95         _cleanup_netlink_unref_ sd_netlink *rtnl = NULL;
  96         _cleanup_netlink_message_unref_ sd_netlink_message *m = NULL, *reply = NULL;
  97         struct ether_addr ha;
  98         int ifindex;
  99
 100         assert_se(sd_event_new(&e) >= 0);
 101
 102         assert_se(sd_netlink_open(&rtnl) >= 0);
 103         assert_se(sd_netlink_attach_event(rtnl, e, 0) >= 0);
 104
 105         assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, 0) >= 0);
 106         assert_se(sd_netlink_message_append_string(m, IFLA_IFNAME, ifname) >= 0);
 107         assert_se(sd_netlink_call(rtnl, m, 0, &reply) >= 0);
 108
 109         assert_se(sd_rtnl_message_link_get_ifindex(reply, &ifindex) >= 0);
 110         assert_se(sd_netlink_message_read_ether_addr(reply, IFLA_ADDRESS, &ha) >= 0);
 111
 112         client_run(ifindex, seed, &ha, e);
 113
 114         return EXIT_SUCCESS;
 115 }
 116
 117 int main(int argc, char *argv[]) {
 118         log_set_max_level(LOG_DEBUG);
 119         log_parse_environment();
 120         log_open();
 121
 122         if (argc == 2)
 123                 return test_ll(argv[1], NULL);
 124         else if (argc == 3)
 125                 return test_ll(argv[1], argv[2]);
 126         else {
 127                 log_error("This program takes one or two arguments.\n"
 128                           "\t %s <ifname> [<seed>]", program_invocation_short_name);
 129                 return EXIT_FAILURE;
 130         }
 131 }