diff --git a/nest/proto.c b/nest/proto.c index c7c21ddf..76422e31 100644 --- a/nest/proto.c +++ b/nest/proto.c @@ -26,23 +26,40 @@ list proto_list; static list inactive_proto_list; static list initial_proto_list; +static list flush_proto_list; + +static event *proto_flush_event; static char *p_states[] = { "DOWN", "START", "UP", "STOP" }; static char *c_states[] = { "HUNGRY", "FEEDING", "HAPPY", "FLUSHING" }; +static void proto_flush_all(void *); + static void proto_relink(struct proto *p) { + list *l; + rem_node(&p->n); - add_tail(p->core_state == FS_HAPPY ? &proto_list : &inactive_proto_list, &p->n); + switch (p->core_state) + { + case FS_HAPPY: + l = &proto_list; + break; + case FS_FLUSHING: + l = &flush_proto_list; + break; + default: + l = &inactive_proto_list; + } + add_tail(l, &p->n); } void * proto_new(struct proto_config *c, unsigned size) { struct protocol *pr = c->proto; - pool *r = rp_new(proto_pool, c->name); - struct proto *p = mb_alloc(r, size); + struct proto *p = mb_alloc(proto_pool, size); p->cf = c; p->debug = c->debug; @@ -50,12 +67,19 @@ proto_new(struct proto_config *c, unsigned size) p->preference = c->preference; p->disabled = c->disabled; p->proto = pr; - p->pool = r; - p->attn = ev_new(r); - p->attn->data = p; return p; } +static void +proto_init_instance(struct proto *p) +{ + struct proto_config *c = p->cf; + + p->pool = rp_new(proto_pool, c->name); + p->attn = ev_new(p->pool); + p->attn->data = p; +} + void * proto_config_new(struct protocol *pr, unsigned size) { @@ -77,6 +101,7 @@ protos_preconfig(struct config *c) init_list(&proto_list); init_list(&inactive_proto_list); init_list(&initial_proto_list); + init_list(&flush_proto_list); debug("Protocol preconfig:"); WALK_LIST(p, protocol_list) { @@ -136,6 +161,7 @@ proto_rethink_goal(struct proto *p) if (p->core_state == FS_HUNGRY && p->proto_state == PS_DOWN) { DBG("Kicking %s up\n", p->name); + proto_init_instance(p); proto_notify_state(p, (q->start ? q->start(p) : PS_UP)); } } @@ -201,6 +227,8 @@ protos_build(void) add_tail(&protocol_list, &proto_static.n); #endif proto_pool = rp_new(&root_pool, "Protocols"); + proto_flush_event = ev_new(proto_pool); + proto_flush_event->hook = proto_flush_all; } static void @@ -223,15 +251,6 @@ proto_feed(void *P) DBG("Protocol %s up and running\n", p->name); } -static void -proto_flush(void *P) -{ - struct proto *p = P; - - DBG("Flushing protocol %s\n", p->name); - bug("Protocol flushing not supported yet!"); /* FIXME */ -} - void proto_notify_state(struct proto *p, unsigned ps) { @@ -270,8 +289,7 @@ proto_notify_state(struct proto *p, unsigned ps) schedule_flush: DBG("%s: Scheduling flush\n", p->name); cs = FS_FLUSHING; - p->attn->hook = proto_flush; - ev_schedule(p->attn); + ev_schedule(proto_flush_event); } default: error: @@ -281,3 +299,20 @@ proto_notify_state(struct proto *p, unsigned ps) p->core_state = cs; proto_relink(p); } + +static void +proto_flush_all(void *unused) +{ + struct proto *p; + + rt_prune(&master_table); + while ((p = HEAD(flush_proto_list))->n.next) + { + DBG("Flushing protocol %s\n", p->name); + rfree(p->pool); + p->pool = NULL; + p->core_state = FS_HUNGRY; + proto_relink(p); + proto_rethink_goal(p); + } +} diff --git a/nest/rt-table.c b/nest/rt-table.c index 0ccb8da6..cc735e32 100644 --- a/nest/rt-table.c +++ b/nest/rt-table.c @@ -8,6 +8,8 @@ #include +#define LOCAL_DEBUG + #include "nest/bird.h" #include "nest/route.h" #include "nest/protocol.h" @@ -282,3 +284,33 @@ rt_init(void) rt_setup(&master_table, "master"); rte_slab = sl_new(&root_pool, sizeof(rte)); } + +void +rt_prune(rtable *tab) +{ + struct fib_iterator fit; + int cnt = 0; + + DBG("Pruning route table %s\n", tab->name); + while (tab) + { + FIB_ITERATE_INIT(&fit, &tab->fib); + again: + FIB_ITERATE_START(&tab->fib, &fit, f) + { + net *n = (net *) f; + rte *e; + for (e=n->routes; e; e=e->next) + if (e->attrs->proto->core_state != FS_HAPPY) + { + FIB_ITERATE_PUT(&fit, f); + rte_discard(e); + cnt++; + goto again; + } + } + FIB_ITERATE_END(f); + tab = tab->sibling; + } + DBG("Pruned %d routes\n", cnt); +}