0
0
mirror of https://gitlab.nic.cz/labs/bird.git synced 2024-12-23 02:01:55 +00:00

Flowspec links don't peruse the all-hooks list for lookup

This commit is contained in:
Maria Matejka 2024-02-11 22:58:29 +01:00
parent 77a0310f87
commit 6bbe9dea3a
2 changed files with 32 additions and 21 deletions

View File

@ -11,6 +11,7 @@
#define _BIRD_ROUTE_H_ #define _BIRD_ROUTE_H_
#include "lib/lists.h" #include "lib/lists.h"
#include "lib/tlists.h"
#include "lib/bitmap.h" #include "lib/bitmap.h"
#include "lib/resource.h" #include "lib/resource.h"
#include "lib/net.h" #include "lib/net.h"
@ -129,6 +130,7 @@ struct rtable_private {
list imports; /* Registered route importers */ list imports; /* Registered route importers */
struct rt_exporter exporter; /* Exporter API structure */ struct rt_exporter exporter; /* Exporter API structure */
TLIST_STRUCT_DEF(rt_flowspec_link, struct rt_flowspec_link) flowspec_links; /* Links serving flowspec reload */
struct hmap id_map; struct hmap id_map;
struct hostcache *hostcache; struct hostcache *hostcache;
@ -542,13 +544,6 @@ struct hostcache {
struct rt_export_request req; /* Notifier */ struct rt_export_request req; /* Notifier */
}; };
struct rt_flowspec_link {
rtable *src;
rtable *dst;
u32 uc;
struct rt_export_request req;
};
#define rte_update channel_rte_import #define rte_update channel_rte_import
/** /**
* rte_update - enter a new update to a routing table * rte_update - enter a new update to a routing table

View File

@ -2564,6 +2564,23 @@ rt_kick_prune_timer(struct rtable_private *tab)
tm_start_in(tab->prune_timer, gc_period, tab->loop); tm_start_in(tab->prune_timer, gc_period, tab->loop);
} }
#define TLIST_PREFIX rt_flowspec_link
#define TLIST_TYPE struct rt_flowspec_link
#define TLIST_ITEM n
#define TLIST_WANT_WALK
#define TLIST_WANT_ADD_TAIL
#define TLIST_DEFINED_BEFORE
struct rt_flowspec_link {
TLIST_DEFAULT_NODE;
rtable *src;
rtable *dst;
u32 uc;
struct rt_export_request req;
};
#include "lib/tlists.h"
static void static void
rt_flowspec_export_one(struct rt_export_request *req, const net_addr *net, struct rt_pending_export *first) rt_flowspec_export_one(struct rt_export_request *req, const net_addr *net, struct rt_pending_export *first)
@ -2618,20 +2635,15 @@ rt_flowspec_log_state_change(struct rt_export_request *req, u8 state)
static struct rt_flowspec_link * static struct rt_flowspec_link *
rt_flowspec_find_link(struct rtable_private *src, rtable *dst) rt_flowspec_find_link(struct rtable_private *src, rtable *dst)
{ {
struct rt_export_hook *hook; node *n; WALK_TLIST(rt_flowspec_link, ln, &src->flowspec_links)
WALK_LIST2(hook, n, src->exporter.hooks, n) if (ln->dst == dst && ln->req.hook)
switch (atomic_load_explicit(&hook->export_state, memory_order_acquire)) switch (atomic_load_explicit(&ln->req.hook->export_state, memory_order_acquire))
{ {
case TES_HUNGRY: case TES_HUNGRY:
case TES_FEEDING: case TES_FEEDING:
case TES_READY: case TES_READY:
if (hook->req->export_one == rt_flowspec_export_one)
{
struct rt_flowspec_link *ln = SKIP_BACK(struct rt_flowspec_link, req, hook->req);
if (ln->dst == dst)
return ln; return ln;
} }
}
return NULL; return NULL;
} }
@ -2665,6 +2677,7 @@ rt_flowspec_link(rtable *src_pub, rtable *dst_pub)
.log_state_change = rt_flowspec_log_state_change, .log_state_change = rt_flowspec_log_state_change,
.export_one = rt_flowspec_export_one, .export_one = rt_flowspec_export_one,
}; };
rt_flowspec_link_add_tail(&src->flowspec_links, ln);
rt_table_export_start_locked(src, &ln->req); rt_table_export_start_locked(src, &ln->req);
@ -2703,8 +2716,11 @@ rt_flowspec_unlink(rtable *src, rtable *dst)
ASSERT(ln && (ln->uc > 0)); ASSERT(ln && (ln->uc > 0));
if (!--ln->uc) if (!--ln->uc)
{
rt_flowspec_link_rem_node(&t->flowspec_links, ln);
rt_stop_export(&ln->req, rt_flowspec_link_stopped); rt_stop_export(&ln->req, rt_flowspec_link_stopped);
} }
}
birdloop_leave(dst->loop); birdloop_leave(dst->loop);
} }