MPLS: Handle label allocation failures

3 files changed, 63 insertions, 18 deletions

diff --git a/nest/mpls.c b/nest/mpls.c
index c9ae78f8..31deb91d 100644
--- a/nest/mpls.c
+++ b/nest/mpls.c
@@ -78,7 +78,6 @@
  * TODO:
  *  - protocols should do route refresh instead of restart when reconfiguration
  *    requires changing labels (e.g. different label range)
- *  - handle label allocation failures
  *  - special handling of reserved labels
  */
 
@@ -720,17 +719,21 @@ struct mpls_fec *
 mpls_get_fec_by_label(struct mpls_fec_map *m, u32 label)
 {
   struct mpls_fec *fec = HASH_FIND(m->label_hash, LABEL, label);
-  /* FIXME: check if (fec->policy == MPLS_POLICY_STATIC) */
+
+  if (!m->static_handle)
+    m->static_handle = mpls_new_handle(m->domain, m->domain->cf->static_range->range);
 
   if (fec)
-    return fec;
+    return (fec->policy == MPLS_POLICY_STATIC) ? fec : NULL;
 
-  fec = sl_allocz(mpls_slab(m, 0));
+  label = mpls_new_label(m->domain, m->static_handle, label);
 
-  if (!m->static_handle)
-    m->static_handle = mpls_new_handle(m->domain, m->domain->cf->static_range->range);
+  if (!label)
+    return NULL;
+
+  fec = sl_allocz(mpls_slab(m, 0));
 
-  fec->label = mpls_new_label(m->domain, m->static_handle, label);
+  fec->label = label;
   fec->policy = MPLS_POLICY_STATIC;
 
   DBG("New FEC lab %u\n", fec->label);
@@ -752,13 +755,18 @@ mpls_get_fec_by_net(struct mpls_fec_map *m, const net_addr *net, u32 path_id)
   if (fec)
     return fec;
 
+  u32 label = mpls_new_label(m->domain, m->handle, 0);
+
+  if (!label)
+    return NULL;
+
   fec = sl_allocz(mpls_slab(m, net->type));
 
   fec->hash = hash;
   fec->path_id = path_id;
   net_copy(fec->net, net);
 
-  fec->label = mpls_new_label(m->domain, m->handle, 0);
+  fec->label = label;
   fec->policy = MPLS_POLICY_PREFIX;
 
   DBG("New FEC net %u\n", fec->label);
@@ -785,13 +793,21 @@ mpls_get_fec_by_rta(struct mpls_fec_map *m, const rta *src, u32 class_id)
     return fec;
   }
 
+  u32 label = mpls_new_label(m->domain, m->handle, 0);
+
+  if (!label)
+  {
+    rta_free(rta);
+    return NULL;
+  }
+
   fec = sl_allocz(mpls_slab(m, 0));
 
   fec->hash = hash;
   fec->class_id = class_id;
   fec->rta = rta;
 
-  fec->label = mpls_new_label(m->domain, m->handle, 0);
+  fec->label = label;
   fec->policy = MPLS_POLICY_AGGREGATE;
 
   DBG("New FEC rta %u\n", fec->label);
@@ -810,9 +826,14 @@ mpls_get_fec_for_vrf(struct mpls_fec_map *m)
   if (fec)
     return fec;
 
+  u32 label = mpls_new_label(m->domain, m->handle, 0);
+
+  if (!label)
+    return NULL;
+
   fec = sl_allocz(mpls_slab(m, 0));
 
-  fec->label = mpls_new_label(m->domain, m->handle, 0);
+  fec->label = label;
   fec->policy = MPLS_POLICY_VRF;
   fec->iface = m->vrf_iface;
 
@@ -872,7 +893,7 @@ static inline void mpls_unlock_fec(struct mpls_fec_map *x, struct mpls_fec *fec)
 static inline void
 mpls_damage_fec(struct mpls_fec_map *m UNUSED, struct mpls_fec *fec)
 {
-  if (fec->state == MPLS_FEC_CLEAN)
+  if (fec && (fec->state == MPLS_FEC_CLEAN))
     fec->state = MPLS_FEC_DIRTY;
 }
 
@@ -992,7 +1013,7 @@ mpls_apply_fec(rte *r, struct mpls_fec *fec, linpool *lp)
 }
 
 
-void
+int
 mpls_handle_rte(struct mpls_fec_map *m, const net_addr *n, rte *r, linpool *lp, struct mpls_fec **locked_fec)
 {
   ASSERT(!(r->flags & REF_COW));
@@ -1004,15 +1025,22 @@ mpls_handle_rte(struct mpls_fec_map *m, const net_addr *n, rte *r, linpool *lp,
   switch (policy)
   {
   case MPLS_POLICY_NONE:
-    return;
+    return 0;
 
   case MPLS_POLICY_STATIC:;
     uint label = ea_get_int(r->attrs->eattrs, EA_MPLS_LABEL, 0);
 
     if (label < 16)
-      return;
+      return 0;
 
     fec = mpls_get_fec_by_label(m, label);
+    if (!fec)
+    {
+      log(L_WARN "Static label %u failed for %N from %s",
+	  label, n, r->sender->proto->name);
+      return -1;
+    }
+
     mpls_damage_fec(m, fec);
     break;
 
@@ -1028,14 +1056,22 @@ mpls_handle_rte(struct mpls_fec_map *m, const net_addr *n, rte *r, linpool *lp,
 
   case MPLS_POLICY_VRF:
     if (!m->vrf_iface)
-      return;
+      return 0;
 
     fec = mpls_get_fec_for_vrf(m);
     break;
 
   default:
     log(L_WARN "Route %N has invalid MPLS policy %u", n, policy);
-    return;
+    return -1;
+  }
+
+  /* Label allocation failure */
+  if (!fec)
+  {
+    log(L_WARN "Label allocation in range %s failed for %N from %s",
+	m->handle->range->name, n, r->sender->proto->name);
+    return -1;
   }
 
   /* Temporarily lock FEC */
@@ -1048,6 +1084,8 @@ mpls_handle_rte(struct mpls_fec_map *m, const net_addr *n, rte *r, linpool *lp,
   /* Announce MPLS rule for new/updated FEC */
   if (fec->state != MPLS_FEC_CLEAN)
     mpls_announce_fec(m, fec, r->attrs);
+
+  return 0;
 }
 
 void
diff --git a/nest/mpls.h b/nest/mpls.h
index 52865f1c..1f3d02dc 100644
--- a/nest/mpls.h
+++ b/nest/mpls.h
@@ -168,7 +168,7 @@ struct mpls_fec *mpls_get_fec_by_label(struct mpls_fec_map *m, u32 label);
 struct mpls_fec *mpls_get_fec_by_net(struct mpls_fec_map *m, const net_addr *net, u32 path_id);
 struct mpls_fec *mpls_get_fec_by_rta(struct mpls_fec_map *m, const rta *src, u32 class_id);
 void mpls_free_fec(struct mpls_fec_map *x, struct mpls_fec *fec);
-void mpls_handle_rte(struct mpls_fec_map *m, const net_addr *n, rte *r, linpool *lp, struct mpls_fec **locked_fec);
+int  mpls_handle_rte(struct mpls_fec_map *m, const net_addr *n, rte *r, linpool *lp, struct mpls_fec **locked_fec);
 void mpls_handle_rte_cleanup(struct mpls_fec_map *m, struct mpls_fec **locked_fec);
 void mpls_rte_insert(net *n UNUSED, rte *r);
 void mpls_rte_remove(net *n UNUSED, rte *r);
diff --git a/nest/rt-table.c b/nest/rt-table.c
index 6eaee069..e497524f 100644
--- a/nest/rt-table.c
+++ b/nest/rt-table.c
@@ -1614,7 +1614,14 @@ rte_update2(struct channel *c, const net_addr *n, rte *new, struct rte_src *src)
 	}
 
       if (p->mpls_map)
-	mpls_handle_rte(p->mpls_map, n, new, rte_update_pool, &fec);
+        {
+	  if (mpls_handle_rte(p->mpls_map, n, new, rte_update_pool, &fec) < 0)
+	    {
+	      rte_trace_in(D_FILTERS, c, new, "invalid");
+	      stats->imp_updates_invalid++;
+	      goto drop;
+	    }
+	}
 
       if (!rta_is_cached(new->attrs)) /* Need to copy attributes */
 	new->attrs = rta_lookup(new->attrs);