sharpd: Allow the registration of import checks to zebra

[mirror_frr.git] / sharpd / sharp_zebra.c

diff --git a/sharpd/sharp_zebra.c b/sharpd/sharp_zebra.c
index 30e616a05775a6b127e5f4b9254314abf47e4b03..4682dbc73a4fd3e4e73602d880b9ef7765eb92b2 100644 (file)
--- a/sharpd/sharp_zebra.c
+++ b/sharpd/sharp_zebra.c
@@ -320,12 +320,22 @@ void route_delete(struct prefix *p, uint8_t instance)
        return;
 }
 
-void sharp_zebra_nexthop_watch(struct prefix *p, bool watch, bool connected)
+void sharp_zebra_nexthop_watch(struct prefix *p, bool import,
+                              bool watch, bool connected)
 {
-       int command = ZEBRA_NEXTHOP_REGISTER;
+       int command;
 
-       if (!watch)
-               command = ZEBRA_NEXTHOP_UNREGISTER;
+       if (!import) {
+               command = ZEBRA_NEXTHOP_REGISTER;
+
+               if (!watch)
+                       command = ZEBRA_NEXTHOP_UNREGISTER;
+       } else {
+               command = ZEBRA_IMPORT_ROUTE_REGISTER;
+
+               if (!watch)
+                       command = ZEBRA_IMPORT_ROUTE_UNREGISTER;
+       }
 
        if (zclient_send_rnh(zclient, command, p, connected, VRF_DEFAULT) < 0)
                zlog_warn("%s: Failure to send nexthop to zebra",
@@ -405,4 +415,5 @@ void sharp_zebra_init(void)
        zclient->interface_address_delete = interface_address_delete;
        zclient->route_notify_owner = route_notify_owner;
        zclient->nexthop_update = sharp_nexthop_update;
+       zclient->import_check_update = sharp_nexthop_update;
 }