File:  [ELWIX - Embedded LightWeight unIX -] / embedaddon / quagga / zebra / test_main.c
Revision 1.1.1.1 (vendor branch): download - view: text, annotated - select for diffs - revision graph
Tue Feb 21 17:26:11 2012 UTC (12 years, 4 months ago) by misho
Branches: quagga, MAIN
CVS tags: v0_99_21, v0_99_20_1, v0_99_20, HEAD
quagga

    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: 
  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: 
  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: 
  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: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>