Merge pull request #5703 from ton31337/feature/limit_outgoing_prefixes

[mirror_frr.git] / bgpd / bgp_labelpool.c

diff --git a/bgpd/bgp_labelpool.c b/bgpd/bgp_labelpool.c
index e0e3d7ff2e99afddd76765786f9e9445b42c143c..feda0328bd5af13228421d60914f716df65f9d71 100644 (file)
--- a/bgpd/bgp_labelpool.c
+++ b/bgpd/bgp_labelpool.c
@@ -29,11 +29,13 @@
 #include "skiplist.h"
 #include "workqueue.h"
 #include "zclient.h"
+#include "mpls.h"
 
 #include "bgpd/bgpd.h"
 #include "bgpd/bgp_labelpool.h"
 #include "bgpd/bgp_debug.h"
 #include "bgpd/bgp_errors.h"
+#include "bgpd/bgp_route.h"
 
 /*
  * Definitions and external declarations.
@@ -390,7 +392,8 @@ void bgp_lp_get(
        if (lp_fifo_count(&lp->requests) > lp->pending_count) {
                if (!zclient || zclient->sock < 0)
                        return;
-               if (!zclient_send_get_label_chunk(zclient, 0, LP_CHUNK_SIZE))
+               if (!zclient_send_get_label_chunk(zclient, 0, LP_CHUNK_SIZE,
+                                                 MPLS_LABEL_BASE_ANY))
                        lp->pending_count += LP_CHUNK_SIZE;
        }
 }
@@ -551,7 +554,8 @@ void bgp_lp_event_zebra_up(void)
                return;
        }
 
-       zclient_send_get_label_chunk(zclient, 0, labels_needed);
+       zclient_send_get_label_chunk(zclient, 0, labels_needed,
+                                    MPLS_LABEL_BASE_ANY);
        lp->pending_count = labels_needed;
 
        /*