babeld/babel_main.c

   1 /*
   2 Copyright 2011 by Matthieu Boutier and Juliusz Chroboczek
   3
   4 Permission is hereby granted, free of charge, to any person obtaining a copy
   5 of this software and associated documentation files (the "Software"), to deal
   6 in the Software without restriction, including without limitation the rights
   7 to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   8 copies of the Software, and to permit persons to whom the Software is
   9 furnished to do so, subject to the following conditions:
  10
  11 The above copyright notice and this permission notice shall be included in
  12 all copies or substantial portions of the Software.
  13
  14 THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15 IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16 FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
  17 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20 THE SOFTWARE.
  21 */
  22
  23 /* include zebra library */
  24 #include <zebra.h>
  25 #include "getopt.h"
  26 #include "if.h"
  27 #include "log.h"
  28 #include "thread.h"
  29 #include "privs.h"
  30 #include "sigevent.h"
  31 #include "version.h"
  32 #include "command.h"
  33 #include "vty.h"
  34 #include "memory.h"
  35 #include "libfrr.h"
  36
  37 #include "babel_main.h"
  38 #include "babeld.h"
  39 #include "util.h"
  40 #include "kernel.h"
  41 #include "babel_interface.h"
  42 #include "neighbour.h"
  43 #include "route.h"
  44 #include "xroute.h"
  45 #include "message.h"
  46 #include "resend.h"
  47 #include "babel_zebra.h"
  48
  49 static void babel_fail(void);
  50 static void babel_init_random(void);
  51 static void babel_replace_by_null(int fd);
  52 static void babel_exit_properly(void);
  53 static void babel_save_state_file(void);
  54
  55
  56 struct thread_master *master;     /* quagga's threads handler */
  57 struct timeval babel_now;         /* current time             */
  58
  59 unsigned char myid[8];            /* unique id (mac address of an interface) */
  60 int debug = 0;
  61
  62 int resend_delay = -1;
  63
  64 const unsigned char zeroes[16] = {0};
  65 const unsigned char ones[16] =
  66     {0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF,
  67      0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF};
  68
  69 static const char *state_file = DAEMON_VTY_DIR "/babel-state";
  70
  71 unsigned char protocol_group[16]; /* babel's link-local multicast address */
  72 int protocol_port;                /* babel's port */
  73 int protocol_socket = -1;         /* socket: communicate with others babeld */
  74
  75 static char babel_config_default[] = SYSCONFDIR BABEL_DEFAULT_CONFIG;
  76 static char *babel_vty_addr = NULL;
  77 static int babel_vty_port = BABEL_VTY_PORT;
  78
  79 /* babeld privileges */
  80 static zebra_capabilities_t _caps_p [] =
  81 {
  82     ZCAP_NET_RAW,
  83     ZCAP_BIND
  84 };
  85
  86 struct zebra_privs_t babeld_privs =
  87 {
  88 #if defined(FRR_USER)
  89     .user = FRR_USER,
  90 #endif
  91 #if defined FRR_GROUP
  92     .group = FRR_GROUP,
  93 #endif
  94 #ifdef VTY_GROUP
  95     .vty_group = VTY_GROUP,
  96 #endif
  97     .caps_p = _caps_p,
  98     .cap_num_p = array_size(_caps_p),
  99     .cap_num_i = 0
 100 };
 101
 102 static void
 103 babel_sigexit(void)
 104 {
 105     zlog_notice("Terminating on signal");
 106
 107     babel_exit_properly();
 108 }
 109
 110 static void
 111 babel_sigusr1 (void)
 112 {
 113     zlog_rotate ();
 114 }
 115
 116 static struct quagga_signal_t babel_signals[] =
 117   {
 118     {
 119       .signal = SIGUSR1,
 120       .handler = &babel_sigusr1,
 121     },
 122     {
 123       .signal = SIGINT,
 124       .handler = &babel_sigexit,
 125     },
 126     {
 127       .signal = SIGTERM,
 128       .handler = &babel_sigexit,
 129     },
 130   };
 131
 132 struct option longopts[] =
 133   {
 134     { 0 }
 135   };
 136
 137 FRR_DAEMON_INFO(babeld, BABELD,
 138                 .vty_port = BABEL_VTY_PORT,
 139                 .proghelp = "Implementation of the BABEL routing protocol.",
 140
 141                 .signals = babel_signals,
 142                 .n_signals = array_size(babel_signals),
 143
 144                 .privs = &babeld_privs,
 145                 )
 146
 147 int
 148 main(int argc, char **argv)
 149 {
 150     int rc;
 151
 152     frr_preinit (&babeld_di, argc, argv);
 153     frr_opt_add ("", longopts, "");
 154
 155     babel_init_random();
 156
 157     /* set the Babel's default link-local multicast address and Babel's port */
 158     parse_address("ff02:0:0:0:0:0:1:6", protocol_group, NULL);
 159     protocol_port = 6696;
 160
 161     /* get options */
 162     while(1) {
 163         int opt;
 164
 165         opt = frr_getopt (argc, argv, NULL);
 166
 167         if (opt == EOF)
 168           break;
 169
 170         switch (opt)
 171           {
 172           case 0:
 173             break;
 174           default:
 175             frr_help_exit (1);
 176             break;
 177           }
 178     }
 179
 180     /* create the threads handler */
 181     master = frr_init ();
 182
 183     /* Library inits. */
 184     zprivs_init (&babeld_privs);
 185     cmd_init (1);
 186     vty_init (master);
 187
 188     resend_delay = BABEL_DEFAULT_RESEND_DELAY;
 189     change_smoothing_half_life(BABEL_DEFAULT_SMOOTHING_HALF_LIFE);
 190
 191     babel_replace_by_null(STDIN_FILENO);
 192
 193     /* init some quagga's dependencies, and babeld's commands */
 194     babeld_quagga_init();
 195     /* init zebra client's structure and it's commands */
 196     /* this replace kernel_setup && kernel_setup_socket */
 197     babelz_zebra_init ();
 198
 199     /* init buffer */
 200     rc = resize_receive_buffer(1500);
 201     if(rc < 0)
 202         babel_fail();
 203
 204     schedule_neighbours_check(5000, 1);
 205
 206     frr_config_fork();
 207     frr_run(master);
 208
 209     return 0;
 210 }
 211
 212 static void
 213 babel_fail(void)
 214 {
 215     exit(1);
 216 }
 217
 218 /* initialize random value, and set 'babel_now' by the way. */
 219 static void
 220 babel_init_random(void)
 221 {
 222     gettime(&babel_now);
 223     int rc;
 224     unsigned int seed;
 225
 226     rc = read_random_bytes(&seed, sizeof(seed));
 227     if(rc < 0) {
 228         zlog_err("read(random): %s", safe_strerror(errno));
 229         seed = 42;
 230     }
 231
 232     seed ^= (babel_now.tv_sec ^ babel_now.tv_usec);
 233     srandom(seed);
 234 }
 235
 236 /*
 237  close fd, and replace it by "/dev/null"
 238  exit if error
 239  */
 240 static void
 241 babel_replace_by_null(int fd)
 242 {
 243     int fd_null;
 244     int rc;
 245
 246     fd_null = open("/dev/null", O_RDONLY);
 247     if(fd_null < 0) {
 248         zlog_err("open(null): %s", safe_strerror(errno));
 249         exit(1);
 250     }
 251
 252     rc = dup2(fd_null, fd);
 253     if(rc < 0) {
 254         zlog_err("dup2(null, 0): %s", safe_strerror(errno));
 255         exit(1);
 256     }
 257
 258     close(fd_null);
 259 }
 260
 261 /*
 262  Load the state file: check last babeld's running state, usefull in case of
 263  "/etc/init.d/babeld restart"
 264  */
 265 void
 266 babel_load_state_file(void)
 267 {
 268     int fd;
 269     int rc;
 270
 271     fd = open(state_file, O_RDONLY);
 272     if(fd < 0 && errno != ENOENT)
 273         zlog_err("open(babel-state: %s)", safe_strerror(errno));
 274     rc = unlink(state_file);
 275     if(fd >= 0 && rc < 0) {
 276         zlog_err("unlink(babel-state): %s", safe_strerror(errno));
 277         /* If we couldn't unlink it, it's probably stale. */
 278         goto fini;
 279     }
 280     if(fd >= 0) {
 281         char buf[100];
 282         char buf2[100];
 283         int s;
 284         long t;
 285         rc = read(fd, buf, 99);
 286         if(rc < 0) {
 287             zlog_err("read(babel-state): %s", safe_strerror(errno));
 288         } else {
 289             buf[rc] = '\0';
 290             rc = sscanf(buf, "%99s %d %ld\n", buf2, &s, &t);
 291             if(rc == 3 && s >= 0 && s <= 0xFFFF) {
 292                 unsigned char sid[8];
 293                 rc = parse_eui64(buf2, sid);
 294                 if(rc < 0) {
 295                     zlog_err("Couldn't parse babel-state.");
 296                 } else {
 297                     struct timeval realnow;
 298                     debugf(BABEL_DEBUG_COMMON,
 299                            "Got %s %d %ld from babel-state.",
 300                            format_eui64(sid), s, t);
 301                     gettimeofday(&realnow, NULL);
 302                     if(memcmp(sid, myid, 8) == 0)
 303                         myseqno = seqno_plus(s, 1);
 304                     else
 305                         zlog_err("ID mismatch in babel-state. id=%s; old=%s",
 306                                  format_eui64(myid),
 307                                  format_eui64(sid));
 308                 }
 309             } else {
 310                 zlog_err("Couldn't parse babel-state.");
 311             }
 312         }
 313         goto fini;
 314     }
 315 fini:
 316     if (fd >= 0)
 317         close(fd);
 318     return ;
 319 }
 320
 321 static void
 322 babel_exit_properly(void)
 323 {
 324     debugf(BABEL_DEBUG_COMMON, "Exiting...");
 325     usleep(roughly(10000));
 326     gettime(&babel_now);
 327
 328     /* Uninstall and flush all routes. */
 329     debugf(BABEL_DEBUG_COMMON, "Uninstall routes.");
 330     flush_all_routes();
 331     babel_interface_close_all();
 332     babel_zebra_close_connexion();
 333     babel_save_state_file();
 334     debugf(BABEL_DEBUG_COMMON, "Remove pid file.");
 335     debugf(BABEL_DEBUG_COMMON, "Done.");
 336     frr_fini();
 337
 338     exit(0);
 339 }
 340
 341 static void
 342 babel_save_state_file(void)
 343 {
 344     int fd;
 345     int rc;
 346
 347     debugf(BABEL_DEBUG_COMMON, "Save state file.");
 348     fd = open(state_file, O_WRONLY | O_TRUNC | O_CREAT, 0644);
 349     if(fd < 0) {
 350         zlog_err("creat(babel-state): %s", safe_strerror(errno));
 351         unlink(state_file);
 352     } else {
 353         struct timeval realnow;
 354         char buf[100];
 355         gettimeofday(&realnow, NULL);
 356         rc = snprintf(buf, 100, "%s %d %ld\n",
 357                       format_eui64(myid), (int)myseqno,
 358                       (long)realnow.tv_sec);
 359         if(rc < 0 || rc >= 100) {
 360             zlog_err("write(babel-state): overflow.");
 361             unlink(state_file);
 362         } else {
 363             rc = write(fd, buf, rc);
 364             if(rc < 0) {
 365                 zlog_err("write(babel-state): %s", safe_strerror(errno));
 366                 unlink(state_file);
 367             }
 368             fsync(fd);
 369         }
 370         close(fd);
 371     }
 372 }
 373
 374 void
 375 show_babel_main_configuration (struct vty *vty)
 376 {
 377     vty_out (vty,
 378             "state file              = %s\n"
 379             "configuration file      = %s\n"
 380             "protocol informations:\n"
 381             "  multicast address     = %s\n"
 382             "  port                  = %d\n"
 383             "vty address             = %s\n"
 384             "vty port                = %d\n"
 385             "id                      = %s\n"
 386             "kernel_metric           = %d\n",
 387             state_file,
 388             babeld_di.config_file ? babeld_di.config_file : babel_config_default,
 389             format_address(protocol_group),
 390             protocol_port,
 391             babel_vty_addr ? babel_vty_addr : "None",
 392             babel_vty_port,
 393             format_eui64(myid),
 394             kernel_metric);
 395 }