diff options
author | Yu Watanabe <watanabe.yu+github@gmail.com> | 2019-07-17 02:14:27 +0900 |
---|---|---|
committer | Yu Watanabe <watanabe.yu+github@gmail.com> | 2019-07-19 01:44:44 +0900 |
commit | d4c52ee5b546670b6f16c88a389df0428cc9d1dd (patch) | |
tree | 1c86d273b8cef454ef869ea8965a3ca040292272 /src/network/networkd-dhcp4.c | |
parent | 01aaa3df1678b1fe3e531126e28a54df3ca173a7 (diff) | |
download | systemd-d4c52ee5b546670b6f16c88a389df0428cc9d1dd.tar.gz systemd-d4c52ee5b546670b6f16c88a389df0428cc9d1dd.tar.bz2 systemd-d4c52ee5b546670b6f16c88a389df0428cc9d1dd.zip |
network: store routes provided by DHCPv4 in Set
This re-writes d03073ddcde6dd2d5604b70ff4184acbe0a7961a.
Diffstat (limited to 'src/network/networkd-dhcp4.c')
-rw-r--r-- | src/network/networkd-dhcp4.c | 123 |
1 files changed, 61 insertions, 62 deletions
diff --git a/src/network/networkd-dhcp4.c b/src/network/networkd-dhcp4.c index 1bb1df32e9..6ebc19db1a 100644 --- a/src/network/networkd-dhcp4.c +++ b/src/network/networkd-dhcp4.c @@ -15,12 +15,12 @@ #include "string-util.h" #include "sysctl-util.h" -static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, sd_dhcp_lease *new_lease, struct in_addr *address); -static int dhcp_remove_router(Link *link, sd_dhcp_lease *lease, struct in_addr *address); -static int dhcp_remove_address(Link *link, sd_dhcp_lease *lease, struct in_addr *address); +static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, const struct in_addr *address, bool remove_all); +static int dhcp_remove_router(Link *link, sd_dhcp_lease *lease, const struct in_addr *address, bool remove_all); +static int dhcp_remove_address(Link *link, sd_dhcp_lease *lease, const struct in_addr *address); void dhcp4_release_old_lease(Link *link) { - union in_addr_union address = IN_ADDR_NULL, address_old = IN_ADDR_NULL; + struct in_addr address = {}, address_old = {}; assert(link); @@ -29,15 +29,14 @@ void dhcp4_release_old_lease(Link *link) { assert(link->dhcp_lease); - (void) sd_dhcp_lease_get_address(link->dhcp_lease_old, &address_old.in); - (void) sd_dhcp_lease_get_address(link->dhcp_lease, &address.in); + (void) sd_dhcp_lease_get_address(link->dhcp_lease_old, &address_old); + (void) sd_dhcp_lease_get_address(link->dhcp_lease, &address); - (void) dhcp_remove_routes(link, link->dhcp_lease_old, link->dhcp_lease, &address_old.in); + (void) dhcp_remove_routes(link, link->dhcp_lease_old, &address_old, false); + (void) dhcp_remove_router(link, link->dhcp_lease_old, &address_old, false); - if (!in_addr_equal(AF_INET, &address_old, &address)) { - (void) dhcp_remove_router(link, link->dhcp_lease_old, &address_old.in); - (void) dhcp_remove_address(link, link->dhcp_lease_old, &address_old.in); - } + if (!in4_addr_equal(&address_old, &address)) + (void) dhcp_remove_address(link, link->dhcp_lease_old, &address_old); link->dhcp_lease_old = sd_dhcp_lease_unref(link->dhcp_lease_old); link_dirty(link); @@ -84,6 +83,30 @@ static int route_scope_from_address(const Route *route, const struct in_addr *se return RT_SCOPE_UNIVERSE; } +static int dhcp_route_configure(Route **route, Link *link) { + int r; + + assert(route); + assert(*route); + assert(link); + + if (set_contains(link->dhcp_routes, *route)) + return 0; + + r = route_configure(*route, link, dhcp4_route_handler); + if (r <= 0) + return r; + + link->dhcp4_messages++; + + r = set_put(link->dhcp_routes, *route); + if (r < 0) + return r; + + TAKE_PTR(*route); + return 0; +} + static int link_set_dhcp_routes(Link *link) { _cleanup_free_ sd_dhcp_route **static_routes = NULL; bool classless_route = false, static_route = false; @@ -108,6 +131,13 @@ static int link_set_dhcp_routes(Link *link) { * the addresses now, let's not configure the routes either. */ return 0; + r = set_ensure_allocated(&link->dhcp_routes, &route_full_hash_ops); + if (r < 0) + return log_oom(); + + /* Clear old entries in case the set was already allocated */ + set_clear(link->dhcp_routes); + table = link_get_dhcp_route_table(link); r = sd_dhcp_lease_get_address(link->dhcp_lease, &address); @@ -155,11 +185,12 @@ static int link_set_dhcp_routes(Link *link) { if (IN_SET(route->scope, RT_SCOPE_LINK, RT_SCOPE_UNIVERSE)) route->prefsrc.in = address; - r = route_configure(route, link, dhcp4_route_handler); + if (set_contains(link->dhcp_routes, route)) + continue; + + r = dhcp_route_configure(&route, link); if (r < 0) - return log_link_error_errno(link, r, "Could not set host route: %m"); - if (r > 0) - link->dhcp4_messages++; + return log_link_error_errno(link, r, "Could not set route: %m"); } r = sd_dhcp_lease_get_router(link->dhcp_lease, &router); @@ -194,11 +225,9 @@ static int link_set_dhcp_routes(Link *link) { route_gw->priority = link->network->dhcp_route_metric; route_gw->table = table; - r = route_configure(route_gw, link, dhcp4_route_handler); + r = dhcp_route_configure(&route_gw, link); if (r < 0) return log_link_error_errno(link, r, "Could not set host route: %m"); - if (r > 0) - link->dhcp4_messages++; r = route_new(&route); if (r < 0) @@ -211,42 +240,18 @@ static int link_set_dhcp_routes(Link *link) { route->priority = link->network->dhcp_route_metric; route->table = table; - r = route_configure(route, link, dhcp4_route_handler); + r = dhcp_route_configure(&route, link); if (r < 0) - return log_link_error_errno(link, r, "Could not set routes: %m"); - if (r > 0) - link->dhcp4_messages++; + return log_link_error_errno(link, r, "Could not set router: %m"); } return 0; } -static bool route_present_in_routes(const Route *route, sd_dhcp_route **routes, unsigned n_routes) { - assert(n_routes == 0 || routes); - - for (unsigned j = 0; j < n_routes; j++) { - union in_addr_union a; - unsigned char l; - - assert_se(sd_dhcp_route_get_gateway(routes[j], &a.in) >= 0); - if (!in_addr_equal(AF_INET, &a, &route->gw)) - continue; - assert_se(sd_dhcp_route_get_destination(routes[j], &a.in) >= 0); - if (!in_addr_equal(AF_INET, &a, &route->dst)) - continue; - assert_se(sd_dhcp_route_get_destination_prefix_length(routes[j], &l) >= 0); - if (l != route->dst_prefixlen) - continue; - return true; - } - - return false; -} - -static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, sd_dhcp_lease *new_lease, struct in_addr *address) { - _cleanup_free_ sd_dhcp_route **routes = NULL, **new_routes = NULL; +static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, const struct in_addr *address, bool remove_all) { + _cleanup_free_ sd_dhcp_route **routes = NULL; uint32_t table; - int m = 0, n, i, r; + int n, i, r; assert(link); assert(address); @@ -260,14 +265,6 @@ static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, sd_dhcp_lease *n else if (n < 0) return log_link_error_errno(link, n, "DHCP error: Failed to get routes: %m"); - if (new_lease) { - m = sd_dhcp_lease_get_routes(new_lease, &new_routes); - if (m == -ENODATA) - m = 0; - else if (m < 0) - return log_link_error_errno(link, m, "DHCP error: Failed to get routes: %m"); - } - table = link_get_dhcp_route_table(link); for (i = 0; i < n; i++) { @@ -287,7 +284,7 @@ static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, sd_dhcp_lease *n if (IN_SET(route->scope, RT_SCOPE_LINK, RT_SCOPE_UNIVERSE)) route->prefsrc.in = *address; - if (route_present_in_routes(route, new_routes, m)) + if (!remove_all && set_contains(link->dhcp_routes, route)) continue; (void) route_remove(route, link, NULL); @@ -296,7 +293,7 @@ static int dhcp_remove_routes(Link *link, sd_dhcp_lease *lease, sd_dhcp_lease *n return n; } -static int dhcp_remove_router(Link *link, sd_dhcp_lease *lease, struct in_addr *address) { +static int dhcp_remove_router(Link *link, sd_dhcp_lease *lease, const struct in_addr *address, bool remove_all) { _cleanup_(route_freep) Route *route_gw = NULL, *route = NULL; const struct in_addr *router; uint32_t table; @@ -334,7 +331,8 @@ static int dhcp_remove_router(Link *link, sd_dhcp_lease *lease, struct in_addr * route_gw->priority = link->network->dhcp_route_metric; route_gw->table = table; - (void) route_remove(route_gw, link, NULL); + if (remove_all || !set_contains(link->dhcp_routes, route_gw)) + (void) route_remove(route_gw, link, NULL); r = route_new(&route); if (r < 0) @@ -347,12 +345,13 @@ static int dhcp_remove_router(Link *link, sd_dhcp_lease *lease, struct in_addr * route->priority = link->network->dhcp_route_metric; route->table = table; - (void) route_remove(route, link, NULL); + if (remove_all || !set_contains(link->dhcp_routes, route)) + (void) route_remove(route, link, NULL); return 0; } -static int dhcp_remove_address(Link *link, sd_dhcp_lease *lease, struct in_addr *address) { +static int dhcp_remove_address(Link *link, sd_dhcp_lease *lease, const struct in_addr *address) { _cleanup_(address_freep) Address *a = NULL; struct in_addr netmask; int r; @@ -439,8 +438,8 @@ static int dhcp_lease_lost(Link *link) { link->dhcp4_configured = false; (void) sd_dhcp_lease_get_address(link->dhcp_lease, &address); - (void) dhcp_remove_routes(link, link->dhcp_lease, NULL, &address); - (void) dhcp_remove_router(link, link->dhcp_lease, &address); + (void) dhcp_remove_routes(link, link->dhcp_lease, &address, true); + (void) dhcp_remove_router(link, link->dhcp_lease, &address, true); (void) dhcp_remove_address(link, link->dhcp_lease, &address); (void) dhcp_reset_mtu(link); (void) dhcp_reset_hostname(link); |