6d7c403324
The advantages are that we save a few lines, and that we can override logging using environment variables in more test executables.
111 lines
3.3 KiB
C
111 lines
3.3 KiB
C
/* SPDX-License-Identifier: LGPL-2.1+ */
|
|
|
|
#include <errno.h>
|
|
#include <net/if.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <linux/veth.h>
|
|
|
|
#include "sd-event.h"
|
|
#include "sd-ipv4ll.h"
|
|
#include "sd-netlink.h"
|
|
|
|
#include "alloc-util.h"
|
|
#include "in-addr-util.h"
|
|
#include "netlink-util.h"
|
|
#include "parse-util.h"
|
|
#include "string-util.h"
|
|
#include "tests.h"
|
|
#include "util.h"
|
|
|
|
static void ll_handler(sd_ipv4ll *ll, int event, void *userdata) {
|
|
_cleanup_free_ char *address = NULL;
|
|
struct in_addr addr = {};
|
|
|
|
assert_se(ll);
|
|
|
|
if (sd_ipv4ll_get_address(ll, &addr) >= 0)
|
|
assert_se(in_addr_to_string(AF_INET, (const union in_addr_union*) &addr, &address) >= 0);
|
|
|
|
switch (event) {
|
|
case SD_IPV4LL_EVENT_BIND:
|
|
log_info("bound %s", strna(address));
|
|
break;
|
|
case SD_IPV4LL_EVENT_CONFLICT:
|
|
log_info("conflict on %s", strna(address));
|
|
break;
|
|
case SD_IPV4LL_EVENT_STOP:
|
|
log_error("the client was stopped with address %s", strna(address));
|
|
break;
|
|
default:
|
|
assert_not_reached("invalid LL event");
|
|
}
|
|
}
|
|
|
|
static int client_run(int ifindex, const char *seed_str, const struct ether_addr *ha, sd_event *e) {
|
|
sd_ipv4ll *ll;
|
|
|
|
assert_se(sd_ipv4ll_new(&ll) >= 0);
|
|
assert_se(sd_ipv4ll_attach_event(ll, e, 0) >= 0);
|
|
|
|
assert_se(sd_ipv4ll_set_ifindex(ll, ifindex) >= 0);
|
|
assert_se(sd_ipv4ll_set_mac(ll, ha) >= 0);
|
|
assert_se(sd_ipv4ll_set_callback(ll, ll_handler, NULL) >= 0);
|
|
|
|
if (seed_str) {
|
|
unsigned seed;
|
|
|
|
assert_se(safe_atou(seed_str, &seed) >= 0);
|
|
|
|
assert_se(sd_ipv4ll_set_address_seed(ll, seed) >= 0);
|
|
}
|
|
|
|
log_info("starting IPv4LL client");
|
|
|
|
assert_se(sd_ipv4ll_start(ll) >= 0);
|
|
|
|
assert_se(sd_event_loop(e) >= 0);
|
|
|
|
assert_se(!sd_ipv4ll_unref(ll));
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
static int test_ll(const char *ifname, const char *seed) {
|
|
_cleanup_(sd_event_unrefp) sd_event *e = NULL;
|
|
_cleanup_(sd_netlink_unrefp) sd_netlink *rtnl = NULL;
|
|
_cleanup_(sd_netlink_message_unrefp) sd_netlink_message *m = NULL, *reply = NULL;
|
|
struct ether_addr ha;
|
|
int ifindex;
|
|
|
|
assert_se(sd_event_new(&e) >= 0);
|
|
|
|
assert_se(sd_netlink_open(&rtnl) >= 0);
|
|
assert_se(sd_netlink_attach_event(rtnl, e, 0) >= 0);
|
|
|
|
assert_se(sd_rtnl_message_new_link(rtnl, &m, RTM_GETLINK, 0) >= 0);
|
|
assert_se(sd_netlink_message_append_string(m, IFLA_IFNAME, ifname) >= 0);
|
|
assert_se(sd_netlink_call(rtnl, m, 0, &reply) >= 0);
|
|
|
|
assert_se(sd_rtnl_message_link_get_ifindex(reply, &ifindex) >= 0);
|
|
assert_se(sd_netlink_message_read_ether_addr(reply, IFLA_ADDRESS, &ha) >= 0);
|
|
|
|
client_run(ifindex, seed, &ha, e);
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
int main(int argc, char *argv[]) {
|
|
test_setup_logging(LOG_DEBUG);
|
|
|
|
if (argc == 2)
|
|
return test_ll(argv[1], NULL);
|
|
else if (argc == 3)
|
|
return test_ll(argv[1], argv[2]);
|
|
else {
|
|
log_error("This program takes one or two arguments.\n"
|
|
"\t %s <ifname> [<seed>]", program_invocation_short_name);
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|