Annotation of embedaddon/quagga/zebra/test_main.c, revision 1.1.1.1

1.1       misho       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>