pbrd/pbr_main.c

   1 /*
   2  * PBR - main code
   3  * Copyright (C) 2018 Cumulus Networks, Inc.
   4  *               Donald Sharp
   5  *
   6  * FRR 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
   8  * Free Software Foundation; either version 2, or (at your option) any
   9  * later version.
  10  *
  11  * FRR is distributed in the hope that it will be useful, but
  12  * WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * General Public License for 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 <lib/version.h>
  23 #include "getopt.h"
  24 #include "thread.h"
  25 #include "prefix.h"
  26 #include "linklist.h"
  27 #include "if.h"
  28 #include "vector.h"
  29 #include "vty.h"
  30 #include "command.h"
  31 #include "filter.h"
  32 #include "plist.h"
  33 #include "stream.h"
  34 #include "log.h"
  35 #include "memory.h"
  36 #include "privs.h"
  37 #include "sigevent.h"
  38 #include "zclient.h"
  39 #include "keychain.h"
  40 #include "distribute.h"
  41 #include "libfrr.h"
  42 #include "routemap.h"
  43 #include "nexthop.h"
  44 #include "nexthop_group.h"
  45
  46 #include "pbr_nht.h"
  47 #include "pbr_map.h"
  48 #include "pbr_zebra.h"
  49 #include "pbr_vty.h"
  50 #include "pbr_debug.h"
  51 #include "pbr_vrf.h"
  52
  53 zebra_capabilities_t _caps_p[] = {
  54         ZCAP_NET_RAW, ZCAP_BIND, ZCAP_NET_ADMIN,
  55 };
  56
  57 struct zebra_privs_t pbr_privs = {
  58 #if defined(FRR_USER) && defined(FRR_GROUP)
  59         .user = FRR_USER,
  60         .group = FRR_GROUP,
  61 #endif
  62 #if defined(VTY_GROUP)
  63         .vty_group = VTY_GROUP,
  64 #endif
  65         .caps_p = _caps_p,
  66         .cap_num_p = array_size(_caps_p),
  67         .cap_num_i = 0};
  68
  69 struct option longopts[] = { { 0 } };
  70
  71 /* Master of threads. */
  72 struct thread_master *master;
  73
  74 /* SIGHUP handler. */
  75 static void sighup(void)
  76 {
  77         zlog_info("SIGHUP received");
  78 }
  79
  80 /* SIGINT / SIGTERM handler. */
  81 static void sigint(void)
  82 {
  83         zlog_notice("Terminating on signal");
  84
  85         exit(0);
  86 }
  87
  88 /* SIGUSR1 handler. */
  89 static void sigusr1(void)
  90 {
  91         zlog_rotate();
  92 }
  93
  94 struct quagga_signal_t pbr_signals[] = {
  95         {
  96                 .signal = SIGHUP,
  97                 .handler = &sighup,
  98         },
  99         {
 100                 .signal = SIGUSR1,
 101                 .handler = &sigusr1,
 102         },
 103         {
 104                 .signal = SIGINT,
 105                 .handler = &sigint,
 106         },
 107         {
 108                 .signal = SIGTERM,
 109                 .handler = &sigint,
 110         },
 111 };
 112
 113 #define PBR_VTY_PORT 2615
 114
 115 static const struct frr_yang_module_info *const pbrd_yang_modules[] = {
 116         &frr_interface_info,
 117 };
 118
 119 FRR_DAEMON_INFO(pbrd, PBR, .vty_port = PBR_VTY_PORT,
 120
 121                 .proghelp = "Implementation of PBR.",
 122
 123                 .signals = pbr_signals,
 124                 .n_signals = array_size(pbr_signals),
 125
 126                 .privs = &pbr_privs,
 127
 128                 .yang_modules = pbrd_yang_modules,
 129                 .n_yang_modules = array_size(pbrd_yang_modules), )
 130
 131 int main(int argc, char **argv, char **envp)
 132 {
 133         frr_preinit(&pbrd_di, argc, argv);
 134         frr_opt_add("", longopts, "");
 135
 136         while (1) {
 137                 int opt;
 138
 139                 opt = frr_getopt(argc, argv, NULL);
 140
 141                 if (opt == EOF)
 142                         break;
 143
 144                 switch (opt) {
 145                 case 0:
 146                         break;
 147                 default:
 148                         frr_help_exit(1);
 149                         break;
 150                 }
 151         }
 152
 153         master = frr_init();
 154
 155         pbr_debug_init();
 156
 157         nexthop_group_init(pbr_nhgroup_add_cb,
 158                            pbr_nhgroup_add_nexthop_cb,
 159                            pbr_nhgroup_del_nexthop_cb,
 160                            pbr_nhgroup_delete_cb);
 161
 162         /*
 163          * So we safely ignore these commands since
 164          * we are getting them at this point in time
 165          */
 166         access_list_init();
 167         pbr_nht_init();
 168         pbr_map_init();
 169         if_zapi_callbacks(pbr_ifp_create, pbr_ifp_up,
 170                           pbr_ifp_down, pbr_ifp_destroy);
 171         pbr_zebra_init();
 172         pbr_vrf_init();
 173         pbr_vty_init();
 174
 175         frr_config_fork();
 176         frr_run(master);
 177
 178         /* Not reached. */
 179         return 0;
 180 }