zebra/test_main.c

   1 /* main routine.
   2  * Copyright (C) 1997, 98 Kunihiro Ishiguro
   3  *
   4  * GNU Zebra is free software; you can redistribute it and/or modify it
   5  * under the terms of the GNU General Public License as published by the
   6  * Free Software Foundation; either version 2, or (at your option) any
   7  * later version.
   8  *
   9  * GNU Zebra is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with GNU Zebra; see the file COPYING.  If not, write to the Free
  16  * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  17  * 02111-1307, USA.
  18  */
  19
  20 #include <zebra.h>
  21
  22 #include <lib/version.h>
  23 #include "getopt.h"
  24 #include "command.h"
  25 #include "thread.h"
  26 #include "filter.h"
  27 #include "memory.h"
  28 #include "prefix.h"
  29 #include "log.h"
  30 #include "privs.h"
  31 #include "sigevent.h"
  32
  33 #include "zebra/rib.h"
  34 #include "zebra/zserv.h"
  35 #include "zebra/debug.h"
  36 #include "zebra/router-id.h"
  37 #include "zebra/interface.h"
  38
  39 /* Zebra instance */
  40 struct zebra_t zebrad =
  41 {
  42   .rtm_table_default = 0,
  43 };
  44
  45 /* process id. */
  46 pid_t pid;
  47
  48 /* zebra_rib's workqueue hold time. Private export for use by test code only */
  49 extern int rib_process_hold_time;
  50
  51 /* Pacify zclient.o in libzebra, which expects this variable. */
  52 struct thread_master *master;
  53
  54 /* Command line options. */
  55 struct option longopts[] =
  56 {
  57   { "batch",       no_argument,       NULL, 'b'},
  58   { "daemon",      no_argument,       NULL, 'd'},
  59   { "config_file", required_argument, NULL, 'f'},
  60   { "help",        no_argument,       NULL, 'h'},
  61   { "vty_addr",    required_argument, NULL, 'A'},
  62   { "vty_port",    required_argument, NULL, 'P'},
  63   { "version",     no_argument,       NULL, 'v'},
  64   { "rib_hold",    required_argument, NULL, 'r'},
  65   { 0 }
  66 };
  67
  68 zebra_capabilities_t _caps_p [] =
  69 {
  70   ZCAP_NET_ADMIN,
  71   ZCAP_SYS_ADMIN,
  72   ZCAP_NET_RAW,
  73 };
  74
  75 /* Default configuration file path. */
  76 char config_default[] = SYSCONFDIR DEFAULT_CONFIG_FILE;
  77
  78 /* Process ID saved for use by init system */
  79 const char *pid_file = PATH_ZEBRA_PID;
  80
  81 /* Help information display. */
  82 static void
  83 usage (char *progname, int status)
  84 {
  85   if (status != 0)
  86     fprintf (stderr, "Try `%s --help' for more information.\n", progname);
  87   else
  88     {
  89       printf ("Usage : %s [OPTION...]\n\n"\
  90               "Daemon which manages kernel routing table management and "\
  91               "redistribution between different routing protocols.\n\n"\
  92               "-b, --batch        Runs in batch mode\n"\
  93               "-d, --daemon       Runs in daemon mode\n"\
  94               "-f, --config_file  Set configuration file name\n"\
  95               "-A, --vty_addr     Set vty's bind address\n"\
  96               "-P, --vty_port     Set vty's port number\n"\
  97               "-r, --rib_hold     Set rib-queue hold time\n"\
  98               "-v, --version      Print program version\n"\
  99               "-h, --help         Display this help and exit\n"\
 100               "\n"\
 101               "Report bugs to %s\n", progname, ZEBRA_BUG_ADDRESS);
 102     }
 103
 104   exit (status);
 105 }
 106 \f
 107 static unsigned int test_ifindex = 0;
 108
 109 /* testrib commands */
 110 DEFUN (test_interface_state,
 111        test_interface_state_cmd,
 112        "state (up|down)",
 113        "configure interface\n"
 114        "up\n"
 115        "down\n")
 116 {
 117   struct interface *ifp;
 118   if (argc < 1)
 119     return CMD_WARNING;
 120
 121   ifp = vty->index;
 122   if (ifp->ifindex == IFINDEX_INTERNAL)
 123     {
 124       ifp->ifindex = ++test_ifindex;
 125       ifp->mtu = 1500;
 126       ifp->flags = IFF_BROADCAST|IFF_MULTICAST;
 127     }
 128
 129   switch (argv[0][0])
 130     {
 131       case 'u':
 132         SET_FLAG (ifp->flags, IFF_UP);
 133         if_add_update (ifp);
 134         printf ("up\n");
 135         break;
 136       case 'd':
 137         UNSET_FLAG (ifp->flags, IFF_UP);
 138         if_delete_update (ifp);
 139         printf ("down\n");
 140         break;
 141       default:
 142         return CMD_WARNING;
 143     }
 144   return CMD_SUCCESS;
 145 }
 146
 147 static void
 148 test_cmd_init (void)
 149 {
 150   install_element (INTERFACE_NODE, &test_interface_state_cmd);
 151 }
 152 \f
 153 /* SIGHUP handler. */
 154 static void
 155 sighup (void)
 156 {
 157   zlog_info ("SIGHUP received");
 158
 159   /* Reload of config file. */
 160   ;
 161 }
 162
 163 /* SIGINT handler. */
 164 static void
 165 sigint (void)
 166 {
 167   zlog_notice ("Terminating on signal");
 168
 169   exit (0);
 170 }
 171
 172 /* SIGUSR1 handler. */
 173 static void
 174 sigusr1 (void)
 175 {
 176   zlog_rotate (NULL);
 177 }
 178
 179 struct quagga_signal_t zebra_signals[] =
 180 {
 181   {
 182     .signal = SIGHUP,
 183     .handler = &sighup,
 184   },
 185   {
 186     .signal = SIGUSR1,
 187     .handler = &sigusr1,
 188   },
 189   {
 190     .signal = SIGINT,
 191     .handler = &sigint,
 192   },
 193   {
 194     .signal = SIGTERM,
 195     .handler = &sigint,
 196   },
 197 };
 198 \f
 199 /* Main startup routine. */
 200 int
 201 main (int argc, char **argv)
 202 {
 203   char *p;
 204   char *vty_addr = NULL;
 205   int vty_port = 0;
 206   int batch_mode = 0;
 207   int daemon_mode = 0;
 208   char *config_file = NULL;
 209   char *progname;
 210   struct thread thread;
 211
 212   /* Set umask before anything for security */
 213   umask (0027);
 214
 215   /* preserve my name */
 216   progname = ((p = strrchr (argv[0], '/')) ? ++p : argv[0]);
 217
 218   zlog_default = openzlog (progname, ZLOG_ZEBRA,
 219                            LOG_CONS|LOG_NDELAY|LOG_PID, LOG_DAEMON);
 220
 221   while (1)
 222     {
 223       int opt;
 224
 225       opt = getopt_long (argc, argv, "bdf:hA:P:r:v", longopts, 0);
 226
 227       if (opt == EOF)
 228         break;
 229
 230       switch (opt)
 231         {
 232         case 0:
 233           break;
 234         case 'b':
 235           batch_mode = 1;
 236         case 'd':
 237           daemon_mode = 1;
 238           break;
 239         case 'f':
 240           config_file = optarg;
 241           break;
 242         case 'A':
 243           vty_addr = optarg;
 244           break;
 245         case 'P':
 246           /* Deal with atoi() returning 0 on failure, and zebra not
 247              listening on zebra port... */
 248           if (strcmp(optarg, "0") == 0)
 249             {
 250               vty_port = 0;
 251               break;
 252             }
 253           vty_port = atoi (optarg);
 254           break;
 255         case 'r':
 256           rib_process_hold_time = atoi(optarg);
 257           break;
 258         case 'v':
 259           print_version (progname);
 260           exit (0);
 261           break;
 262         case 'h':
 263           usage (progname, 0);
 264           break;
 265         default:
 266           usage (progname, 1);
 267           break;
 268         }
 269     }
 270
 271   /* port and conf file mandatory */
 272   if (!vty_port || !config_file)
 273     {
 274       fprintf (stderr, "Error: --vty_port and --config_file arguments"
 275                        " are both required\n");
 276       usage (progname, 1);
 277     }
 278
 279   /* Make master thread emulator. */
 280   zebrad.master = thread_master_create ();
 281
 282   /* Vty related initialize. */
 283   signal_init (zebrad.master, Q_SIGC(zebra_signals), zebra_signals);
 284   cmd_init (1);
 285   vty_init (zebrad.master);
 286   memory_init ();
 287   if_init();
 288   zebra_debug_init ();
 289   zebra_if_init ();
 290   test_cmd_init ();
 291
 292   /* Zebra related initialize. */
 293   rib_init ();
 294   access_list_init ();
 295
 296   /* Make kernel routing socket. */
 297   kernel_init ();
 298   route_read ();
 299   zebra_vty_init();
 300
 301   /* Sort VTY commands. */
 302   sort_node ();
 303
 304   /* Configuration file read*/
 305   vty_read_config (config_file, config_default);
 306
 307   /* Clean up rib. */
 308   rib_weed_tables ();
 309
 310   /* Exit when zebra is working in batch mode. */
 311   if (batch_mode)
 312     exit (0);
 313
 314   /* Daemonize. */
 315   if (daemon_mode && daemon (0, 0) < 0)
 316     {
 317       perror("daemon start failed");
 318       exit (1);
 319     }
 320
 321   /* Needed for BSD routing socket. */
 322   pid = getpid ();
 323
 324   /* Make vty server socket. */
 325   vty_serv_sock (vty_addr, vty_port, "/tmp/test_zebra");
 326
 327   /* Print banner. */
 328   zlog_notice ("Zebra %s starting: vty@%d", QUAGGA_VERSION, vty_port);
 329
 330   while (thread_fetch (zebrad.master, &thread))
 331     thread_call (&thread);
 332
 333   /* Not reached... */
 334   return 0;
 335 }