staticd: merge NHT register & update, reorganize

nh_update is only called in two places and both precede a matching
follow-up nht_register call.  Fold the update into register, and make
register do the right thing™ for all cases (i.e. update refcounts as
needed, and retry zebra NHT registration if it failed before).

Signed-off-by: David Lamparter <equinox@opensourcerouting.org>
This commit is contained in:
David Lamparter 2022-04-24 17:10:52 +02:00
parent ecb490350e
commit f75d39259c
3 changed files with 39 additions and 56 deletions

View File

@ -369,13 +369,11 @@ void static_install_nexthop(struct static_nexthop *nh)
switch (nh->type) {
case STATIC_IPV4_GATEWAY:
case STATIC_IPV6_GATEWAY:
if (!static_zebra_nh_update(nh))
static_zebra_nht_register(nh, true);
static_zebra_nht_register(nh, true);
break;
case STATIC_IPV4_GATEWAY_IFNAME:
case STATIC_IPV6_GATEWAY_IFNAME:
if (!static_zebra_nh_update(nh))
static_zebra_nht_register(nh, true);
static_zebra_nht_register(nh, true);
break;
case STATIC_BLACKHOLE:
static_install_path(pn);

View File

@ -57,6 +57,7 @@ struct static_nht_data {
uint32_t refcount;
uint8_t nh_num;
bool registered;
};
static int static_nht_data_cmp(const struct static_nht_data *nhtd1,
@ -316,7 +317,7 @@ void static_zebra_nht_register(struct static_nexthop *nh, bool reg)
struct static_path *pn = nh->pn;
struct route_node *rn = pn->rn;
struct static_route_info *si = static_route_info_from_rnode(rn);
struct static_nht_data lookup = {};
struct static_nht_data *nhtd, lookup = {};
uint32_t cmd;
if (!static_zebra_nht_get_prefix(nh, &lookup.nh))
@ -324,41 +325,57 @@ void static_zebra_nht_register(struct static_nexthop *nh, bool reg)
lookup.nh_vrf_id = nh->nh_vrf_id;
lookup.safi = si->safi;
if (reg) {
struct static_nht_data *nhtd;
if (nh->nh_registered) {
/* nh->nh_registered means we own a reference on the nhtd */
nhtd = static_nht_hash_find(static_nht_hash, &lookup);
assertf(nhtd, "BUG: NH %pFX registered but not in hashtable",
&lookup.nh);
} else if (reg) {
nhtd = static_nht_hash_getref(&lookup);
if (nhtd->refcount > 1) {
if (nhtd->refcount > 1)
DEBUGD(&static_dbg_route,
"Already registered nexthop(%pFX) for %pRN %d",
"Reusing registered nexthop(%pFX) for %pRN %d",
&lookup.nh, rn, nhtd->nh_num);
if (nhtd->nh_num) {
afi_t afi = prefix_afi(&lookup.nh);
} else {
/* !reg && !nh->nh_registered */
zlog_warn("trying to unregister nexthop %pFX twice",
&lookup.nh);
return;
}
static_nht_update(&rn->p, &nhtd->nh,
nhtd->nh_num, afi, si->safi,
nh->nh_vrf_id);
}
nh->nh_registered = reg;
if (nhtd->nh_registered)
return;
if (reg) {
if (nhtd->nh_num) {
/* refresh with existing data */
afi_t afi = prefix_afi(&lookup.nh);
if (nh->state == STATIC_NOT_INSTALLED)
nh->state = STATIC_START;
static_nht_update(&rn->p, &nhtd->nh, nhtd->nh_num, afi,
si->safi, nh->nh_vrf_id);
return;
}
if (nhtd->registered)
/* have no data, but did send register */
return;
cmd = ZEBRA_NEXTHOP_REGISTER;
DEBUGD(&static_dbg_route, "Registering nexthop(%pFX) for %pRN",
&lookup.nh, rn);
} else {
struct static_nht_data *nhtd;
bool was_zebra_registered;
nhtd = static_nht_hash_find(static_nht_hash, &lookup);
if (!nhtd)
return;
was_zebra_registered = nhtd->registered;
if (static_nht_hash_decref(&nhtd))
/* still got references alive */
return;
if (!nhtd->nh_registered)
/* NB: nhtd is now NULL. */
if (!was_zebra_registered)
return;
cmd = ZEBRA_NEXTHOP_UNREGISTER;
@ -370,39 +387,8 @@ void static_zebra_nht_register(struct static_nexthop *nh, bool reg)
nh->nh_vrf_id) == ZCLIENT_SEND_FAILURE)
zlog_warn("%s: Failure to send nexthop %pFX for %pRN to zebra",
__func__, &lookup.nh, rn);
else
nh->nh_registered = reg;
}
/*
* When nexthop gets updated via configuration then use the
* already registered NH and resend the route to zebra
*/
int static_zebra_nh_update(struct static_nexthop *nh)
{
struct static_path *pn = nh->pn;
struct route_node *rn = pn->rn;
struct static_route_info *si = static_route_info_from_rnode(rn);
struct static_nht_data *nhtd, lookup = {};
if (!nh->nh_registered)
return 0;
if (!static_zebra_nht_get_prefix(nh, &lookup.nh))
return 0;
lookup.nh_vrf_id = nh->nh_vrf_id;
lookup.safi = si->safi;
nhtd = static_nht_hash_find(static_nht_hash, &lookup);
if (nhtd && nhtd->nh_num) {
afi_t afi = prefix_afi(&lookup.nh);
nh->state = STATIC_START;
static_nht_update(&rn->p, &nhtd->nh, nhtd->nh_num, afi,
si->safi, nh->nh_vrf_id);
return 1;
}
return 0;
else if (reg)
nhtd->registered = true;
}
extern void static_zebra_route_add(struct static_path *pn, bool install)

View File

@ -33,7 +33,6 @@ extern void static_zebra_init(void);
extern void static_zebra_stop(void);
extern void static_zebra_vrf_register(struct vrf *vrf);
extern void static_zebra_vrf_unregister(struct vrf *vrf);
extern int static_zebra_nh_update(struct static_nexthop *nh);
#ifdef __cplusplus
}