staticd/static_zebra.c

   1 /*
   2  * Zebra connect code.
   3  * Copyright (C) 2018 Cumulus Networks, Inc.
   4  *               Donald Sharp
   5  *
   6  * This program is free software; you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License as published by the Free
   8  * Software Foundation; either version 2 of the License, or (at your option)
   9  * any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  14  * more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with this program; see the file COPYING; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20 #include <zebra.h>
  21
  22 #include "thread.h"
  23 #include "command.h"
  24 #include "network.h"
  25 #include "prefix.h"
  26 #include "routemap.h"
  27 #include "table.h"
  28 #include "srcdest_table.h"
  29 #include "stream.h"
  30 #include "memory.h"
  31 #include "zclient.h"
  32 #include "filter.h"
  33 #include "plist.h"
  34 #include "log.h"
  35 #include "nexthop.h"
  36 #include "nexthop_group.h"
  37 #include "hash.h"
  38 #include "jhash.h"
  39
  40 #include "static_vrf.h"
  41 #include "static_routes.h"
  42 #include "static_zebra.h"
  43 #include "static_nht.h"
  44 #include "static_vty.h"
  45
  46 bool debug;
  47
  48 /* Zebra structure to hold current status. */
  49 struct zclient *zclient;
  50 static struct hash *static_nht_hash;
  51
  52 static struct interface *zebra_interface_if_lookup(struct stream *s)
  53 {
  54         char ifname_tmp[INTERFACE_NAMSIZ];
  55
  56         /* Read interface name. */
  57         stream_get(ifname_tmp, s, INTERFACE_NAMSIZ);
  58
  59         /* And look it up. */
  60         return if_lookup_by_name(ifname_tmp, VRF_DEFAULT);
  61 }
  62
  63 /* Inteface addition message from zebra. */
  64 static int interface_add(ZAPI_CALLBACK_ARGS)
  65 {
  66         struct interface *ifp;
  67
  68         ifp = zebra_interface_add_read(zclient->ibuf, vrf_id);
  69
  70         if (!ifp)
  71                 return 0;
  72
  73         static_ifindex_update(ifp, true);
  74         return 0;
  75 }
  76
  77 static int interface_delete(ZAPI_CALLBACK_ARGS)
  78 {
  79         struct interface *ifp;
  80         struct stream *s;
  81
  82         s = zclient->ibuf;
  83         /* zebra_interface_state_read () updates interface structure in iflist
  84          */
  85         ifp = zebra_interface_state_read(s, vrf_id);
  86
  87         if (ifp == NULL)
  88                 return 0;
  89
  90         if_set_index(ifp, IFINDEX_INTERNAL);
  91
  92         static_ifindex_update(ifp, false);
  93         return 0;
  94 }
  95
  96 static int interface_address_add(ZAPI_CALLBACK_ARGS)
  97 {
  98         zebra_interface_address_read(cmd, zclient->ibuf, vrf_id);
  99
 100         return 0;
 101 }
 102
 103 static int interface_address_delete(ZAPI_CALLBACK_ARGS)
 104 {
 105         struct connected *c;
 106
 107         c = zebra_interface_address_read(cmd, zclient->ibuf, vrf_id);
 108
 109         if (!c)
 110                 return 0;
 111
 112         connected_free(c);
 113         return 0;
 114 }
 115
 116 static int interface_state_up(ZAPI_CALLBACK_ARGS)
 117 {
 118         struct interface *ifp;
 119
 120         ifp = zebra_interface_if_lookup(zclient->ibuf);
 121
 122         if (ifp) {
 123                 if (if_is_vrf(ifp)) {
 124                         struct static_vrf *svrf =
 125                                         static_vrf_lookup_by_id(vrf_id);
 126
 127                         static_fixup_vrf_ids(svrf);
 128                         static_config_install_delayed_routes(svrf);
 129                 }
 130
 131                 /* Install any static reliant on this interface coming up */
 132                 static_install_intf_nh(ifp);
 133                 static_ifindex_update(ifp, true);
 134         }
 135
 136         return 0;
 137 }
 138
 139 static int interface_state_down(ZAPI_CALLBACK_ARGS)
 140 {
 141         struct interface *ifp;
 142
 143         ifp = zebra_interface_state_read(zclient->ibuf, vrf_id);
 144
 145         if (ifp)
 146                 static_ifindex_update(ifp, false);
 147
 148         return 0;
 149 }
 150
 151 static int route_notify_owner(ZAPI_CALLBACK_ARGS)
 152 {
 153         struct prefix p;
 154         enum zapi_route_notify_owner note;
 155         uint32_t table_id;
 156         char buf[PREFIX_STRLEN];
 157
 158         if (!zapi_route_notify_decode(zclient->ibuf, &p, &table_id, &note))
 159                 return -1;
 160
 161         prefix2str(&p, buf, sizeof(buf));
 162
 163         switch (note) {
 164         case ZAPI_ROUTE_FAIL_INSTALL:
 165                 static_nht_mark_state(&p, vrf_id, STATIC_NOT_INSTALLED);
 166                 zlog_warn("%s: Route %s failed to install for table: %u",
 167                           __PRETTY_FUNCTION__, buf, table_id);
 168                 break;
 169         case ZAPI_ROUTE_BETTER_ADMIN_WON:
 170                 static_nht_mark_state(&p, vrf_id, STATIC_NOT_INSTALLED);
 171                 zlog_warn("%s: Route %s over-ridden by better route for table: %u",
 172                           __PRETTY_FUNCTION__, buf, table_id);
 173                 break;
 174         case ZAPI_ROUTE_INSTALLED:
 175                 static_nht_mark_state(&p, vrf_id, STATIC_INSTALLED);
 176                 break;
 177         case ZAPI_ROUTE_REMOVED:
 178                 static_nht_mark_state(&p, vrf_id, STATIC_NOT_INSTALLED);
 179                 break;
 180         case ZAPI_ROUTE_REMOVE_FAIL:
 181                 static_nht_mark_state(&p, vrf_id, STATIC_INSTALLED);
 182                 zlog_warn("%s: Route %s failure to remove for table: %u",
 183                           __PRETTY_FUNCTION__, buf, table_id);
 184                 break;
 185         }
 186
 187         return 0;
 188 }
 189 static void zebra_connected(struct zclient *zclient)
 190 {
 191         zclient_send_reg_requests(zclient, VRF_DEFAULT);
 192 }
 193
 194 struct static_nht_data {
 195         struct prefix *nh;
 196
 197         vrf_id_t nh_vrf_id;
 198
 199         uint32_t refcount;
 200         uint8_t nh_num;
 201 };
 202
 203 static int static_zebra_nexthop_update(ZAPI_CALLBACK_ARGS)
 204 {
 205         struct static_nht_data *nhtd, lookup;
 206         struct zapi_route nhr;
 207         afi_t afi = AFI_IP;
 208
 209         if (!zapi_nexthop_update_decode(zclient->ibuf, &nhr)) {
 210                 zlog_warn("Failure to decode nexthop update message");
 211                 return 1;
 212         }
 213
 214         if (nhr.prefix.family == AF_INET6)
 215                 afi = AFI_IP6;
 216
 217         memset(&lookup, 0, sizeof(lookup));
 218         lookup.nh = &nhr.prefix;
 219         lookup.nh_vrf_id = vrf_id;
 220
 221         nhtd = hash_lookup(static_nht_hash, &lookup);
 222
 223         if (nhtd) {
 224                 nhtd->nh_num = nhr.nexthop_num;
 225
 226                 static_nht_reset_start(&nhr.prefix, afi, nhtd->nh_vrf_id);
 227                 static_nht_update(NULL, &nhr.prefix, nhr.nexthop_num, afi,
 228                                   nhtd->nh_vrf_id);
 229         } else
 230                 zlog_err("No nhtd?");
 231
 232         return 1;
 233 }
 234
 235 static void static_zebra_capabilities(struct zclient_capabilities *cap)
 236 {
 237         mpls_enabled = cap->mpls_enabled;
 238 }
 239
 240 static unsigned int static_nht_hash_key(const void *data)
 241 {
 242         const struct static_nht_data *nhtd = data;
 243         unsigned int key = 0;
 244
 245         key = prefix_hash_key(nhtd->nh);
 246         return jhash_1word(nhtd->nh_vrf_id, key);
 247 }
 248
 249 static bool static_nht_hash_cmp(const void *d1, const void *d2)
 250 {
 251         const struct static_nht_data *nhtd1 = d1;
 252         const struct static_nht_data *nhtd2 = d2;
 253
 254         if (nhtd1->nh_vrf_id != nhtd2->nh_vrf_id)
 255                 return false;
 256
 257         return prefix_same(nhtd1->nh, nhtd2->nh);
 258 }
 259
 260 static void *static_nht_hash_alloc(void *data)
 261 {
 262         struct static_nht_data *copy = data;
 263         struct static_nht_data *new;
 264
 265         new = XMALLOC(MTYPE_TMP, sizeof(*new));
 266
 267         new->nh = prefix_new();
 268         prefix_copy(new->nh, copy->nh);
 269         new->refcount = 0;
 270         new->nh_num = 0;
 271         new->nh_vrf_id = copy->nh_vrf_id;
 272
 273         return new;
 274 }
 275
 276 static void static_nht_hash_free(void *data)
 277 {
 278         struct static_nht_data *nhtd = data;
 279
 280         prefix_free(nhtd->nh);
 281         XFREE(MTYPE_TMP, nhtd);
 282 }
 283
 284 void static_zebra_nht_register(struct route_node *rn,
 285                                struct static_route *si, bool reg)
 286 {
 287         struct static_nht_data *nhtd, lookup;
 288         uint32_t cmd;
 289         struct prefix p;
 290         afi_t afi = AFI_IP;
 291
 292         cmd = (reg) ?
 293                 ZEBRA_NEXTHOP_REGISTER : ZEBRA_NEXTHOP_UNREGISTER;
 294
 295         if (si->nh_registered && reg)
 296                 return;
 297
 298         if (!si->nh_registered && !reg)
 299                 return;
 300
 301         memset(&p, 0, sizeof(p));
 302         switch (si->type) {
 303         case STATIC_IFNAME:
 304         case STATIC_BLACKHOLE:
 305                 return;
 306         case STATIC_IPV4_GATEWAY:
 307         case STATIC_IPV4_GATEWAY_IFNAME:
 308                 p.family = AF_INET;
 309                 p.prefixlen = IPV4_MAX_BITLEN;
 310                 p.u.prefix4 = si->addr.ipv4;
 311                 afi = AFI_IP;
 312                 break;
 313         case STATIC_IPV6_GATEWAY:
 314         case STATIC_IPV6_GATEWAY_IFNAME:
 315                 p.family = AF_INET6;
 316                 p.prefixlen = IPV6_MAX_BITLEN;
 317                 p.u.prefix6 = si->addr.ipv6;
 318                 afi = AFI_IP6;
 319                 break;
 320         }
 321
 322         memset(&lookup, 0, sizeof(lookup));
 323         lookup.nh = &p;
 324         lookup.nh_vrf_id = si->nh_vrf_id;
 325
 326         si->nh_registered = reg;
 327
 328         if (reg) {
 329                 nhtd = hash_get(static_nht_hash, &lookup,
 330                                 static_nht_hash_alloc);
 331                 nhtd->refcount++;
 332
 333                 if (debug)
 334                         zlog_debug("Registered nexthop(%pFX) for %pRN %d", &p,
 335                                    rn, nhtd->nh_num);
 336                 if (nhtd->refcount > 1 && nhtd->nh_num) {
 337                         static_nht_update(&rn->p, nhtd->nh, nhtd->nh_num,
 338                                           afi, si->nh_vrf_id);
 339                         return;
 340                 }
 341         } else {
 342                 nhtd = hash_lookup(static_nht_hash, &lookup);
 343                 if (!nhtd)
 344                         return;
 345
 346                 nhtd->refcount--;
 347                 if (nhtd->refcount >= 1)
 348                         return;
 349
 350                 hash_release(static_nht_hash, nhtd);
 351                 static_nht_hash_free(nhtd);
 352         }
 353
 354         if (zclient_send_rnh(zclient, cmd, &p, false, si->nh_vrf_id) < 0)
 355                 zlog_warn("%s: Failure to send nexthop to zebra",
 356                           __PRETTY_FUNCTION__);
 357 }
 358
 359 extern void static_zebra_route_add(struct route_node *rn,
 360                                    struct static_route *si_changed,
 361                                    vrf_id_t vrf_id, safi_t safi, bool install)
 362 {
 363         struct static_route *si = rn->info;
 364         const struct prefix *p, *src_pp;
 365         struct zapi_nexthop *api_nh;
 366         struct zapi_route api;
 367         uint32_t nh_num = 0;
 368
 369         p = src_pp = NULL;
 370         srcdest_rnode_prefixes(rn, &p, &src_pp);
 371
 372         memset(&api, 0, sizeof(api));
 373         api.vrf_id = vrf_id;
 374         api.type = ZEBRA_ROUTE_STATIC;
 375         api.safi = safi;
 376         memcpy(&api.prefix, p, sizeof(api.prefix));
 377
 378         if (src_pp) {
 379                 SET_FLAG(api.message, ZAPI_MESSAGE_SRCPFX);
 380                 memcpy(&api.src_prefix, src_pp, sizeof(api.src_prefix));
 381         }
 382         SET_FLAG(api.flags, ZEBRA_FLAG_RR_USE_DISTANCE);
 383         SET_FLAG(api.message, ZAPI_MESSAGE_NEXTHOP);
 384         if (si_changed->distance) {
 385                 SET_FLAG(api.message, ZAPI_MESSAGE_DISTANCE);
 386                 api.distance = si_changed->distance;
 387         }
 388         if (si_changed->tag) {
 389                 SET_FLAG(api.message, ZAPI_MESSAGE_TAG);
 390                 api.tag = si_changed->tag;
 391         }
 392         if (si_changed->table_id != 0) {
 393                 SET_FLAG(api.message, ZAPI_MESSAGE_TABLEID);
 394                 api.tableid = si_changed->table_id;
 395         }
 396         for (/*loaded above*/; si; si = si->next) {
 397                 api_nh = &api.nexthops[nh_num];
 398                 if (si->nh_vrf_id == VRF_UNKNOWN)
 399                         continue;
 400
 401                 if (si->distance != si_changed->distance)
 402                         continue;
 403
 404                 if (si->table_id != si_changed->table_id)
 405                         continue;
 406
 407                 api_nh->vrf_id = si->nh_vrf_id;
 408                 api_nh->onlink = si->onlink;
 409
 410                 si->state = STATIC_SENT_TO_ZEBRA;
 411
 412                 switch (si->type) {
 413                 case STATIC_IFNAME:
 414                         if (si->ifindex == IFINDEX_INTERNAL)
 415                                 continue;
 416                         api_nh->ifindex = si->ifindex;
 417                         api_nh->type = NEXTHOP_TYPE_IFINDEX;
 418                         break;
 419                 case STATIC_IPV4_GATEWAY:
 420                         if (!si->nh_valid)
 421                                 continue;
 422                         api_nh->type = NEXTHOP_TYPE_IPV4;
 423                         api_nh->gate = si->addr;
 424                         break;
 425                 case STATIC_IPV4_GATEWAY_IFNAME:
 426                         if (si->ifindex == IFINDEX_INTERNAL)
 427                                 continue;
 428                         api_nh->ifindex = si->ifindex;
 429                         api_nh->type = NEXTHOP_TYPE_IPV4_IFINDEX;
 430                         api_nh->gate = si->addr;
 431                         break;
 432                 case STATIC_IPV6_GATEWAY:
 433                         if (!si->nh_valid)
 434                                 continue;
 435                         api_nh->type = NEXTHOP_TYPE_IPV6;
 436                         api_nh->gate = si->addr;
 437                         break;
 438                 case STATIC_IPV6_GATEWAY_IFNAME:
 439                         if (si->ifindex == IFINDEX_INTERNAL)
 440                                 continue;
 441                         api_nh->type = NEXTHOP_TYPE_IPV6_IFINDEX;
 442                         api_nh->ifindex = si->ifindex;
 443                         api_nh->gate = si->addr;
 444                         break;
 445                 case STATIC_BLACKHOLE:
 446                         api_nh->type = NEXTHOP_TYPE_BLACKHOLE;
 447                         switch (si->bh_type) {
 448                         case STATIC_BLACKHOLE_DROP:
 449                         case STATIC_BLACKHOLE_NULL:
 450                                 api_nh->bh_type = BLACKHOLE_NULL;
 451                                 break;
 452                         case STATIC_BLACKHOLE_REJECT:
 453                                 api_nh->bh_type = BLACKHOLE_REJECT;
 454                         }
 455                         break;
 456                 }
 457
 458                 if (si->snh_label.num_labels) {
 459                         int i;
 460
 461                         SET_FLAG(api.message, ZAPI_MESSAGE_LABEL);
 462                         api_nh->label_num = si->snh_label.num_labels;
 463                         for (i = 0; i < api_nh->label_num; i++)
 464                                 api_nh->labels[i] = si->snh_label.label[i];
 465                 }
 466                 nh_num++;
 467         }
 468
 469         api.nexthop_num = nh_num;
 470
 471         /*
 472          * If we have been given an install but nothing is valid
 473          * go ahead and delete the route for double plus fun
 474          */
 475         if (!nh_num && install)
 476                 install = false;
 477
 478         zclient_route_send(install ?
 479                            ZEBRA_ROUTE_ADD : ZEBRA_ROUTE_DELETE,
 480                            zclient, &api);
 481 }
 482 void static_zebra_init(void)
 483 {
 484         struct zclient_options opt = { .receive_notify = true };
 485
 486         zclient = zclient_new(master, &opt);
 487
 488         zclient_init(zclient, ZEBRA_ROUTE_STATIC, 0, &static_privs);
 489         zclient->zebra_capabilities = static_zebra_capabilities;
 490         zclient->zebra_connected = zebra_connected;
 491         zclient->interface_add = interface_add;
 492         zclient->interface_delete = interface_delete;
 493         zclient->interface_up = interface_state_up;
 494         zclient->interface_down = interface_state_down;
 495         zclient->interface_address_add = interface_address_add;
 496         zclient->interface_address_delete = interface_address_delete;
 497         zclient->route_notify_owner = route_notify_owner;
 498         zclient->nexthop_update = static_zebra_nexthop_update;
 499
 500         static_nht_hash = hash_create(static_nht_hash_key,
 501                                       static_nht_hash_cmp,
 502                                       "Static Nexthop Tracking hash");
 503 }