开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Linux/Unix编程 > 网络 > 查看源码
vtysh.c
资源名称：zebra-0.95a.tar.gz　[点击查看]
上传用户：xiaozhuqw
上传日期：2009-11-15
资源大小：1338k
文件大小：42k
源码类别：

网络

开发平台：

Unix_Linux

vtysh.c：源码内容
  1. /* Virtual terminal interface shell.
  2.  * Copyright (C) 2000 Kunihiro Ishiguro
  3.  *
  4.  * This file is part of GNU Zebra.
  5.  *
  6.  * GNU Zebra 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.  * GNU Zebra 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
  17.  * along with GNU Zebra; see the file COPYING.  If not, write to the Free
  18.  * Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
  19.  * 02111-1307, USA.  
  20.  */
  22. #include <zebra.h>
  24. #include <sys/un.h>
  25. #include <setjmp.h>
  26. #include <sys/wait.h>
  27. #include <sys/resource.h>
  28. #include <sys/stat.h>
  30. #include <readline/readline.h>
  31. #include <readline/history.h>
  33. #include "command.h"
  34. #include "memory.h"
  35. #include "vtysh/vtysh.h"
  37. /* Struct VTY. */
  38. struct vty *vty;
  40. /* VTY shell pager name. */
  41. char *vtysh_pager_name = NULL;
  43. /* VTY shell client structure. */
  44. struct vtysh_client
  45. {
  46.   int fd;
  47. } vtysh_client[VTYSH_INDEX_MAX];
  49. /* When '^Z' is received from vty, move down to the enable mode. */
  50. int
  51. vtysh_end ()
  52. {
  53.   switch (vty->node)
  54.     {
  55.     case VIEW_NODE:
  56.     case ENABLE_NODE:
  57.       /* Nothing to do. */
  58.       break;
  59.     default:
  60.       vty->node = ENABLE_NODE;
  61.       break;
  62.     }
  63.   return CMD_SUCCESS;
  64. }
  66. DEFUNSH (VTYSH_ALL,
  67.  vtysh_end_all,
  68.  vtysh_end_all_cmd,
  69.  "end",
  70.  "End current mode and down to previous moden")
  71. {
  72.   return vtysh_end (vty);
  73. }
  75. DEFUNSH (VTYSH_ALL,
  76.  vtysh_log_stdout,
  77.  vtysh_log_stdout_cmd,
  78.  "log stdout",
  79.  "Logging controln"
  80.  "Logging goes to stdoutn")
  81. {
  82.   return CMD_SUCCESS;
  83. }
  85. DEFUNSH (VTYSH_ALL,
  86.  no_vtysh_log_stdout,
  87.  no_vtysh_log_stdout_cmd,
  88.  "no log stdout",
  89.  NO_STR
  90.  "Logging controln"
  91.  "Logging goes to stdoutn")
  92. {
  93.   return CMD_SUCCESS;
  94. }
  96. DEFUNSH (VTYSH_ALL,
  97.        vtysh_log_file,
  98.        vtysh_log_file_cmd,
  99.        "log file FILENAME",
  100.        "Logging controln"
  101.        "Logging to filen"
  102.        "Logging filenamen")
  103. {
  104.   return CMD_SUCCESS;
  105. }
  107. DEFUNSH (VTYSH_ALL,
  108.        no_vtysh_log_file,
  109.        no_vtysh_log_file_cmd,
  110.        "no log file [FILENAME]",
  111.        NO_STR
  112.        "Logging controln"
  113.        "Cancel logging to filen"
  114.        "Logging file namen")
  115. {
  116.   return CMD_SUCCESS;
  117. }
  119. DEFUNSH (VTYSH_ALL,
  120.  vtysh_log_syslog,
  121.  vtysh_log_syslog_cmd,
  122.  "log syslog",
  123.  "Logging controln"
  124.  "Logging goes to syslogn")
  125. {
  126.   return CMD_SUCCESS;
  127. }
  129. DEFUNSH (VTYSH_ALL,
  130.  no_vtysh_log_syslog,
  131.  no_vtysh_log_syslog_cmd,
  132.  "no log syslog",
  133.  NO_STR
  134.  "Logging controln"
  135.  "Cancel logging to syslogn")
  136. {
  137.   return CMD_SUCCESS;
  138. }
  140. DEFUNSH (VTYSH_ALL,
  141.  vtysh_log_trap,
  142.  vtysh_log_trap_cmd,
  143.  "log trap (emergencies|alerts|critical|errors|warnings|notifications|informational|debugging)",
  144.  "Logging controln"
  145.  "Limit logging to specifed leveln")
  146. {
  147.   return CMD_SUCCESS;
  148. }
  150. DEFUNSH (VTYSH_ALL,
  151.  no_vtysh_log_trap,
  152.  no_vtysh_log_trap_cmd,
  153.  "no log trap",
  154.  NO_STR
  155.  "Logging controln"
  156.  "Permit all logging informationn")
  157. {
  158.   return CMD_SUCCESS;
  159. }
  161. DEFUNSH (VTYSH_ALL,
  162.  vtysh_log_record_priority,
  163.  vtysh_log_record_priority_cmd,
  164.  "log record-priority",
  165.  "Logging controln"
  166.  "Log the priority of the message within the messagen")
  167. {
  168.   return CMD_SUCCESS;
  169. }
  171. DEFUNSH (VTYSH_ALL,
  172.  no_vtysh_log_record_priority,
  173.  no_vtysh_log_record_priority_cmd,
  174.  "no log record-priority",
  175.  NO_STR
  176.  "Logging controln"
  177.  "Do not log the priority of the message within the messagen")
  178. {
  179.   return CMD_SUCCESS;
  180. }
  182. void
  183. vclient_close (struct vtysh_client *vclient)
  184. {
  185.   if (vclient->fd > 0)
  186.     close (vclient->fd);
  187.   vclient->fd = -1;
  188. }
  191. /* Following filled with debug code to trace a problematic condition
  192.    under load - it SHOULD handle it.
  193. */
  194. #define ERR_WHERE_STRING "vtysh(): vtysh_client_config(): "
  195. int
  196. vtysh_client_config (struct vtysh_client *vclient, char *line)
  197. {
  198.   int ret;
  199.   char *buf;
  200.   size_t bufsz;
  201.   char *pbuf;
  202.   size_t left;
  203.   char *eoln;
  204.   int nbytes;
  205.   int i;
  206.   int readln;
  208.   if (vclient->fd < 0)
  209.     return CMD_SUCCESS;
  211.   ret = write (vclient->fd, line, strlen (line) + 1);
  212.   if (ret <= 0)
  213.     {
  214.       vclient_close (vclient);
  215.       return CMD_SUCCESS;
  216.     }
  218.   /* Allow enough room for buffer to read more than a few pages from socket
  219.    */
  220.   bufsz = 5 * sysconf(_SC_PAGESIZE) + 1;
  221.   buf = XMALLOC(MTYPE_TMP, bufsz);
  222.   memset(buf, 0, bufsz);
  223.   pbuf = buf;
  225.   while (1)
  226.     {
  227.       if (pbuf >= ((buf + bufsz) -1))
  228. {
  229.   fprintf (stderr, ERR_WHERE_STRING 
  230.    "warning - pbuf beyond buffer end.n");
  231.   return CMD_WARNING;
  232. }
  234.       readln = (buf + bufsz) - pbuf - 1;
  235.       nbytes = read (vclient->fd, pbuf, readln);
  237.       if (nbytes <= 0)
  238. {
  240.   if (errno == EINTR)
  241.     continue;
  243.   fprintf(stderr, ERR_WHERE_STRING "(%u)", errno);
  244.   perror("");
  246.   if (errno == EAGAIN || errno == EIO)
  247.     continue;
  249.   vclient_close (vclient);
  250.   XFREE(MTYPE_TMP, buf);
  251.   return CMD_SUCCESS;
  252. }
  254.       pbuf[nbytes] = '';
  256.       if (nbytes >= 4)
  257. {
  258.   i = nbytes - 4;
  259.   if (pbuf[i] == '' && pbuf[i + 1] == '' && pbuf[i + 2] == '')
  260.     {
  261.       ret = pbuf[i + 3];
  262.       break;
  263.     }
  264. }
  265.       pbuf += nbytes;
  267.       /* See if a line exists in buffer, if so parse and consume it, and
  268.  reset read position */
  269.       if ((eoln = strrchr(buf, 'n')) == NULL)
  270. continue;
  272.       if (eoln >= ((buf + bufsz) - 1))
  273. {
  274.   fprintf (stderr, ERR_WHERE_STRING 
  275.    "warning - eoln beyond buffer end.n");
  276. }
  277.       vtysh_config_parse(buf);
  279.       eoln++;
  280.       left = (size_t)(buf + bufsz - eoln);
  281.       memmove(buf, eoln, left);
  282.       buf[bufsz-1] = '';
  283.       pbuf = buf + strlen(buf);
  284.     }
  286.   /* parse anything left in the buffer */
  287.   vtysh_config_parse (buf);
  289.   XFREE(MTYPE_TMP, buf);
  290.   return ret;
  291. }
  293. int
  294. vtysh_client_execute (struct vtysh_client *vclient, char *line, FILE *fp)
  295. {
  296.   int ret;
  297.   char buf[1001];
  298.   int nbytes;
  299.   int i;
  301.   if (vclient->fd < 0)
  302.     return CMD_SUCCESS;
  304.   ret = write (vclient->fd, line, strlen (line) + 1);
  305.   if (ret <= 0)
  306.     {
  307.       vclient_close (vclient);
  308.       return CMD_SUCCESS;
  309.     }
  311.   while (1)
  312.     {
  313.       nbytes = read (vclient->fd, buf, sizeof(buf)-1);
  315.       if (nbytes <= 0 && errno != EINTR)
  316. {
  317.   vclient_close (vclient);
  318.   return CMD_SUCCESS;
  319. }
  321.       if (nbytes > 0)
  322. {
  323.   buf[nbytes] = '';
  324.   fprintf (fp, "%s", buf);
  325.   fflush (fp);
  327.   if (nbytes >= 4)
  328.     {
  329.       i = nbytes - 4;
  330.       if (buf[i] == '' && buf[i + 1] == '' && buf[i + 2] == '')
  331. {
  332.   ret = buf[i + 3];
  333.   break;
  334. }
  335.     }
  336. }
  337.     }
  338.   return ret;
  339. }
  341. void
  342. vtysh_exit_ripd_only ()
  343. {
  344.   vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIP], "exit", stdout);
  345. }
  348. void
  349. vtysh_pager_init ()
  350. {
  351.   vtysh_pager_name = getenv ("VTYSH_PAGER");
  352.   if (! vtysh_pager_name)
  353.     vtysh_pager_name = "more";
  354. }
  356. /* Command execution over the vty interface. */
  357. void
  358. vtysh_execute_func (char *line, int pager)
  359. {
  360.   int ret, cmd_stat;
  361.   vector vline;
  362.   struct cmd_element *cmd;
  363.   FILE *fp = NULL;
  365.   /* Split readline string up into the vector */
  366.   vline = cmd_make_strvec (line);
  368.   if (vline == NULL)
  369.     return;
  371.   ret = cmd_execute_command (vline, vty, &cmd);
  373.   cmd_free_strvec (vline);
  375.   switch (ret)
  376.     {
  377.     case CMD_WARNING:
  378.       if (vty->type == VTY_FILE)
  379. printf ("Warning...n");
  380.       break;
  381.     case CMD_ERR_AMBIGUOUS:
  382.       printf ("%% Ambiguous command.n");
  383.       break;
  384.     case CMD_ERR_NO_MATCH:
  385.       printf ("%% Unknown command.n");
  386.       break;
  387.     case CMD_ERR_INCOMPLETE:
  388.       printf ("%% Command incomplete.n");
  389.       break;
  390.     case CMD_SUCCESS_DAEMON:
  391.       {
  392. if (pager && vtysh_pager_name)
  393.   {
  394.     fp = popen ("more", "w");
  395.     if (fp == NULL)
  396.       {
  397. perror ("popen");
  398. exit (1);
  399.       }
  400.   }
  401. else
  402.   fp = stdout;
  404. if (! strcmp(cmd->string,"configure terminal"))
  405.   {
  406.     cmd_stat = vtysh_client_execute (&vtysh_client[VTYSH_INDEX_ZEBRA],
  407.      line, fp);
  408.     if (cmd_stat != CMD_WARNING)
  409.       cmd_stat = vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIP],
  410.        line, fp);
  411.     if (cmd_stat != CMD_WARNING)
  412.       cmd_stat = vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIPNG], line, fp);
  413.     if (cmd_stat != CMD_WARNING)
  414.       cmd_stat = vtysh_client_execute (&vtysh_client[VTYSH_INDEX_OSPF],
  415.        line, fp);
  416.     if (cmd_stat != CMD_WARNING)
  417.       cmd_stat = vtysh_client_execute (&vtysh_client[VTYSH_INDEX_OSPF6], line, fp);
  418.     if (cmd_stat != CMD_WARNING)
  419.       cmd_stat = vtysh_client_execute (&vtysh_client[VTYSH_INDEX_BGP],
  420.        line, fp);
  421.     if (cmd_stat)
  422.       {
  423.                 line = "end";
  424.                 vline = cmd_make_strvec (line);
  426.                 if (vline == NULL)
  427.   {
  428.     if (pager && vtysh_pager_name && fp)
  429.       {
  430. if (pclose (fp) == -1)
  431.   {
  432.     perror ("pclose");
  433.     exit (1);
  434.   }
  435. fp = NULL;
  436.       }
  437.     return;
  438.   }
  440.                 ret = cmd_execute_command (vline, vty, &cmd);
  441.                 cmd_free_strvec (vline);
  442.                 if (ret != CMD_SUCCESS_DAEMON)
  443.                   break;
  444.       }
  445.     else
  446.       if (cmd->func)
  447. {
  448.   (*cmd->func) (cmd, vty, 0, NULL);
  449.   break;
  450.                 }
  451.   }
  453. if (cmd->daemon & VTYSH_ZEBRA)
  454.   if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_ZEBRA], line, fp)
  455.       != CMD_SUCCESS)
  456.     break;
  457. if (cmd->daemon & VTYSH_RIPD)
  458.   if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIP], line, fp)
  459.       != CMD_SUCCESS)
  460.     break;
  461. if (cmd->daemon & VTYSH_RIPNGD)
  462.   if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIPNG], line, fp)
  463.       != CMD_SUCCESS)
  464.     break;
  465. if (cmd->daemon & VTYSH_OSPFD)
  466.   if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_OSPF], line, fp)
  467.       != CMD_SUCCESS)
  468.     break;
  469. if (cmd->daemon & VTYSH_OSPF6D)
  470.   if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_OSPF6], line, fp)
  471.       != CMD_SUCCESS)
  472.     break;
  473. if (cmd->daemon & VTYSH_BGPD)
  474.   if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_BGP], line, fp)
  475.       != CMD_SUCCESS)
  476.     break;
  477. if (cmd->func)
  478.   (*cmd->func) (cmd, vty, 0, NULL);
  479.       }
  480.     }
  481.   if (pager && vtysh_pager_name && fp)
  482.     {
  483.       if (pclose (fp) == -1)
  484. {
  485.   perror ("pclose");
  486.   exit (1);
  487. }
  488.       fp = NULL;
  489.     }
  490. }
  492. void
  493. vtysh_execute_no_pager (char *line)
  494. {
  495.   vtysh_execute_func (line, 0);
  496. }
  498. void
  499. vtysh_execute (char *line)
  500. {
  501.   vtysh_execute_func (line, 1);
  502. }
  504. /* Configration make from file. */
  505. int
  506. vtysh_config_from_file (struct vty *vty, FILE *fp)
  507. {
  508.   int ret;
  509.   vector vline;
  510.   struct cmd_element *cmd;
  512.   while (fgets (vty->buf, VTY_BUFSIZ, fp))
  513.     {
  514.       if (vty->buf[0] == '!' || vty->buf[1] == '#')
  515. continue;
  517.       vline = cmd_make_strvec (vty->buf);
  519.       /* In case of comment line */
  520.       if (vline == NULL)
  521. continue;
  523.       /* Execute configuration command : this is strict match */
  524.       ret = cmd_execute_command_strict (vline, vty, &cmd);
  526.       /* Try again with setting node to CONFIG_NODE */
  527.       if (ret != CMD_SUCCESS 
  528.   && ret != CMD_SUCCESS_DAEMON
  529.   && ret != CMD_WARNING)
  530. {
  531.   if (vty->node == KEYCHAIN_KEY_NODE)
  532.     {
  533.       vty->node = KEYCHAIN_NODE;
  534.       vtysh_exit_ripd_only ();
  535.       ret = cmd_execute_command_strict (vline, vty, &cmd);
  537.       if (ret != CMD_SUCCESS 
  538.   && ret != CMD_SUCCESS_DAEMON 
  539.   && ret != CMD_WARNING)
  540. {
  541.   vtysh_exit_ripd_only ();
  542.   vty->node = CONFIG_NODE;
  543.   ret = cmd_execute_command_strict (vline, vty, &cmd);
  544. }
  545.     }
  546.   else
  547.     {
  548.       vtysh_execute ("end");
  549.       vtysh_execute ("configure terminal");
  550.       vty->node = CONFIG_NODE;
  551.       ret = cmd_execute_command_strict (vline, vty, &cmd);
  552.     }
  553. }   
  555.       cmd_free_strvec (vline);
  557.       switch (ret)
  558. {
  559. case CMD_WARNING:
  560.   if (vty->type == VTY_FILE)
  561.     printf ("Warning...n");
  562.   break;
  563. case CMD_ERR_AMBIGUOUS:
  564.   printf ("%% Ambiguous command.n");
  565.   break;
  566. case CMD_ERR_NO_MATCH:
  567.   printf ("%% Unknown command: %s", vty->buf);
  568.   break;
  569. case CMD_ERR_INCOMPLETE:
  570.   printf ("%% Command incomplete.n");
  571.   break;
  572. case CMD_SUCCESS_DAEMON:
  573.   {
  574.     if (cmd->daemon & VTYSH_ZEBRA)
  575.       if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_ZEBRA],
  576. vty->buf, stdout) != CMD_SUCCESS)
  577. break;
  578.     if (cmd->daemon & VTYSH_RIPD)
  579.       if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIP],
  580. vty->buf, stdout) != CMD_SUCCESS)
  581. break;
  582.     if (cmd->daemon & VTYSH_RIPNGD)
  583.       if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_RIPNG],
  584. vty->buf, stdout) != CMD_SUCCESS)
  585. break;
  586.     if (cmd->daemon & VTYSH_OSPFD)
  587.       if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_OSPF],
  588. vty->buf, stdout) != CMD_SUCCESS)
  589. break;
  590.     if (cmd->daemon & VTYSH_OSPF6D)
  591.       if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_OSPF6],
  592. vty->buf, stdout) != CMD_SUCCESS)
  593. break;
  594.     if (cmd->daemon & VTYSH_BGPD)
  595.       if (vtysh_client_execute (&vtysh_client[VTYSH_INDEX_BGP],
  596. vty->buf, stdout) != CMD_SUCCESS)
  597. break;
  598.     if (cmd->func)
  599.       (*cmd->func) (cmd, vty, 0, NULL);
  600.   }
  601. }
  602.     }
  603.   return CMD_SUCCESS;
  604. }
  606. /* We don't care about the point of the cursor when '?' is typed. */
  607. int
  608. vtysh_rl_describe ()
  609. {
  610.   int ret;
  611.   int i;
  612.   vector vline;
  613.   vector describe;
  614.   int width;
  615.   struct desc *desc;
  617.   vline = cmd_make_strvec (rl_line_buffer);
  619.   /* In case of '> ?'. */
  620.   if (vline == NULL)
  621.     {
  622.       vline = vector_init (1);
  623.       vector_set (vline, '');
  624.     }
  625.   else 
  626.     if (rl_end && isspace ((int) rl_line_buffer[rl_end - 1]))
  627.       vector_set (vline, '');
  629.   describe = cmd_describe_command (vline, vty, &ret);
  631.   printf ("n");
  633.   /* Ambiguous and no match error. */
  634.   switch (ret)
  635.     {
  636.     case CMD_ERR_AMBIGUOUS:
  637.       cmd_free_strvec (vline);
  638.       printf ("%% Ambiguous command.n");
  639.       rl_on_new_line ();
  640.       return 0;
  641.       break;
  642.     case CMD_ERR_NO_MATCH:
  643.       cmd_free_strvec (vline);
  644.       printf ("%% There is no matched command.n");
  645.       rl_on_new_line ();
  646.       return 0;
  647.       break;
  648.     }  
  650.   /* Get width of command string. */
  651.   width = 0;
  652.   for (i = 0; i < vector_max (describe); i++)
  653.     if ((desc = vector_slot (describe, i)) != NULL)
  654.       {
  655. int len;
  657. if (desc->cmd[0] == '')
  658.   continue;
  660. len = strlen (desc->cmd);
  661. if (desc->cmd[0] == '.')
  662.   len--;
  664. if (width < len)
  665.   width = len;
  666.       }
  668.   for (i = 0; i < vector_max (describe); i++)
  669.     if ((desc = vector_slot (describe, i)) != NULL)
  670.       {
  671. if (desc->cmd[0] == '')
  672.   continue;
  674. if (! desc->str)
  675.   printf ("  %-sn",
  676.   desc->cmd[0] == '.' ? desc->cmd + 1 : desc->cmd);
  677. else
  678.   printf ("  %-*s  %sn",
  679.   width,
  680.   desc->cmd[0] == '.' ? desc->cmd + 1 : desc->cmd,
  681.   desc->str);
  682.       }
  684.   cmd_free_strvec (vline);
  685.   vector_free (describe);
  687.   rl_on_new_line();
  689.   return 0;
  690. }
  692. /* result of cmd_complete_command() call will be stored here
  693.    and used in new_completion() in order to put the space in
  694.    correct places only */
  695. int complete_status;
  697. char *
  698. command_generator (char *text, int state)
  699. {
  700.   vector vline;
  701.   static char **matched = NULL;
  702.   static int index = 0;
  704.   /* First call. */
  705.   if (! state)
  706.     {
  707.       index = 0;
  709.       if (vty->node == AUTH_NODE || vty->node == AUTH_ENABLE_NODE)
  710. return NULL;
  712.       vline = cmd_make_strvec (rl_line_buffer);
  713.       if (vline == NULL)
  714. return NULL;
  716.       if (rl_end && isspace ((int) rl_line_buffer[rl_end - 1]))
  717. vector_set (vline, '');
  719.       matched = cmd_complete_command (vline, vty, &complete_status);
  720.     }
  722.   if (matched && matched[index])
  723.     return matched[index++];
  725.   return NULL;
  726. }
  728. char **
  729. new_completion (char *text, int start, int end)
  730. {
  731.   char **matches;
  733.   matches = completion_matches (text, command_generator);
  735.   if (matches)
  736.     {
  737.       rl_point = rl_end;
  738.       if (complete_status == CMD_COMPLETE_FULL_MATCH)
  739. rl_pending_input = ' ';
  740.     }
  742.   return matches;
  743. }
  745. char **
  746. vtysh_completion (char *text, int start, int end)
  747. {
  748.   int ret;
  749.   vector vline;
  750.   char **matched = NULL;
  752.   if (vty->node == AUTH_NODE || vty->node == AUTH_ENABLE_NODE)
  753.     return NULL;
  755.   vline = cmd_make_strvec (rl_line_buffer);
  756.   if (vline == NULL)
  757.     return NULL;
  759.   /* In case of 'help t'. */
  760.   if (rl_end && isspace ((int) rl_line_buffer[rl_end - 1]))
  761.     vector_set (vline, '');
  763.   matched = cmd_complete_command (vline, vty, &ret);
  765.   cmd_free_strvec (vline);
  767.   return (char **) matched;
  768. }
  770. /* BGP node structure. */
  771. struct cmd_node bgp_node =
  772. {
  773.   BGP_NODE,
  774.   "%s(config-router)# ",
  775. };
  777. /* BGP node structure. */
  778. struct cmd_node rip_node =
  779. {
  780.   RIP_NODE,
  781.   "%s(config-router)# ",
  782. };
  784. struct cmd_node interface_node =
  785. {
  786.   INTERFACE_NODE,
  787.   "%s(config-if)# ",
  788. };
  790. DEFUNSH (VTYSH_BGPD,
  791.  router_bgp,
  792.  router_bgp_cmd,
  793.  "router bgp <1-65535>",
  794.  ROUTER_STR
  795.  BGP_STR
  796.  AS_STR)
  797. {
  798.   vty->node = BGP_NODE;
  799.   return CMD_SUCCESS;
  800. }
  802. DEFUNSH (VTYSH_BGPD,
  803.  address_family_vpnv4,
  804.  address_family_vpnv4_cmd,
  805.  "address-family vpnv4",
  806.  "Enter Address Family command moden"
  807.  "Address familyn")
  808. {
  809.   vty->node = BGP_VPNV4_NODE;
  810.   return CMD_SUCCESS;
  811. }
  813. DEFUNSH (VTYSH_BGPD,
  814.  address_family_vpnv4_unicast,
  815.  address_family_vpnv4_unicast_cmd,
  816.  "address-family vpnv4 unicast",
  817.  "Enter Address Family command moden"
  818.  "Address familyn"
  819.  "Address Family Modifiern")
  820. {
  821.   vty->node = BGP_VPNV4_NODE;
  822.   return CMD_SUCCESS;
  823. }
  825. DEFUNSH (VTYSH_BGPD,
  826.  address_family_ipv4_unicast,
  827.  address_family_ipv4_unicast_cmd,
  828.  "address-family ipv4 unicast",
  829.  "Enter Address Family command moden"
  830.  "Address familyn"
  831.  "Address Family Modifiern")
  832. {
  833.   vty->node = BGP_IPV4_NODE;
  834.   return CMD_SUCCESS;
  835. }
  837. DEFUNSH (VTYSH_BGPD,
  838.  address_family_ipv4,
  839.  address_family_ipv4_cmd,
  840.  "address-family ipv4",
  841.  "Enter Address Family command moden"
  842.  "Address familyn")
  843. {
  844.   vty->node = BGP_IPV4_NODE;
  845.   return CMD_SUCCESS;
  846. }
  848. DEFUNSH (VTYSH_BGPD,
  849.  address_family_ipv4_multicast,
  850.  address_family_ipv4_multicast_cmd,
  851.  "address-family ipv4 multicast",
  852.  "Enter Address Family command moden"
  853.  "Address familyn"
  854.  "Address Family Modifiern")
  855. {
  856.   vty->node = BGP_IPV4M_NODE;
  857.   return CMD_SUCCESS;
  858. }
  860. DEFUNSH (VTYSH_BGPD,
  861.  address_family_ipv6,
  862.  address_family_ipv6_cmd,
  863.  "address-family ipv6",
  864.  "Enter Address Family command moden"
  865.  "Address familyn")
  866. {
  867.   vty->node = BGP_IPV6_NODE;
  868.   return CMD_SUCCESS;
  869. }
  871. DEFUNSH (VTYSH_BGPD,
  872.  address_family_ipv6_unicast,
  873.  address_family_ipv6_unicast_cmd,
  874.  "address-family ipv6 unicast",
  875.  "Enter Address Family command moden"
  876.  "Address familyn"
  877.  "Address Family Modifiern")
  878. {
  879.   vty->node = BGP_IPV6_NODE;
  880.   return CMD_SUCCESS;
  881. }
  883. DEFUNSH (VTYSH_RIPD,
  884.  key_chain,
  885.  key_chain_cmd,
  886.  "key chain WORD",
  887.  "Authentication key managementn"
  888.  "Key-chain managementn"
  889.  "Key-chain namen")
  890. {
  891.   vty->node = KEYCHAIN_NODE;
  892.   return CMD_SUCCESS;
  893. }  
  895. DEFUNSH (VTYSH_RIPD,
  896.  key,
  897.  key_cmd,
  898.  "key <0-2147483647>",
  899.  "Configure a keyn"
  900.  "Key identifier numbern")
  901. {
  902.   vty->node = KEYCHAIN_KEY_NODE;
  903.   return CMD_SUCCESS;
  904. }
  906. DEFUNSH (VTYSH_RIPD,
  907.  router_rip,
  908.  router_rip_cmd,
  909.  "router rip",
  910.  ROUTER_STR
  911.  "RIP")
  912. {
  913.   vty->node = RIP_NODE;
  914.   return CMD_SUCCESS;
  915. }
  917. DEFUNSH (VTYSH_RIPNGD,
  918.  router_ripng,
  919.  router_ripng_cmd,
  920.  "router ripng",
  921.  ROUTER_STR
  922.  "RIPng")
  923. {
  924.   vty->node = RIPNG_NODE;
  925.   return CMD_SUCCESS;
  926. }
  928. DEFUNSH (VTYSH_OSPFD,
  929.  router_ospf,
  930.  router_ospf_cmd,
  931.  "router ospf",
  932.  "Enable a routing processn"
  933.  "Start OSPF configurationn")
  934. {
  935.   vty->node = OSPF_NODE;
  936.   return CMD_SUCCESS;
  937. }
  939. DEFUNSH (VTYSH_OSPF6D,
  940.  router_ospf6,
  941.  router_ospf6_cmd,
  942.  "router ospf6",
  943.  OSPF6_ROUTER_STR
  944.  OSPF6_STR)
  945. {
  946.   vty->node = OSPF6_NODE;
  947.   return CMD_SUCCESS;
  948. }
  950. DEFUNSH (VTYSH_RMAP,
  951.  route_map,
  952.  route_map_cmd,
  953.  "route-map WORD (deny|permit) <1-65535>",
  954.  "Create route-map or enter route-map command moden"
  955.  "Route map tagn"
  956.  "Route map denies set operationsn"
  957.  "Route map permits set operationsn"
  958.  "Sequence to insert to/delete from existing route-map entryn")
  959. {
  960.   vty->node = RMAP_NODE;
  961.   return CMD_SUCCESS;
  962. }
  964. /* Enable command */
  965. DEFUNSH (VTYSH_ALL,
  966.  vtysh_enable, 
  967.  vtysh_enable_cmd,
  968.  "enable",
  969.  "Turn on privileged mode commandn")
  970. {
  971.   vty->node = ENABLE_NODE;
  972.   return CMD_SUCCESS;
  973. }
  975. /* Disable command */
  976. DEFUNSH (VTYSH_ALL,
  977.  vtysh_disable, 
  978.  vtysh_disable_cmd,
  979.  "disable",
  980.  "Turn off privileged mode commandn")
  981. {
  982.   if (vty->node == ENABLE_NODE)
  983.     vty->node = VIEW_NODE;
  984.   return CMD_SUCCESS;
  985. }
  987. /* Configration from terminal */
  988. DEFUNSH (VTYSH_ALL,
  989.  vtysh_config_terminal,
  990.  vtysh_config_terminal_cmd,
  991.  "configure terminal",
  992.  "Configuration from vty interfacen"
  993.  "Configuration terminaln")
  994. {
  995.   vty->node = CONFIG_NODE;
  996.   return CMD_SUCCESS;
  997. }
  999. int
  1000. vtysh_exit (struct vty *vty)
  1001. {
  1002.   switch (vty->node)
  1003.     {
  1004.     case VIEW_NODE:
  1005.     case ENABLE_NODE:
  1006.       exit (0);
  1007.       break;
  1008.     case CONFIG_NODE:
  1009.       vty->node = ENABLE_NODE;
  1010.       break;
  1011.     case INTERFACE_NODE:
  1012.     case ZEBRA_NODE:
  1013.     case BGP_NODE:
  1014.     case RIP_NODE:
  1015.     case RIPNG_NODE:
  1016.     case OSPF_NODE:
  1017.     case OSPF6_NODE:
  1018.     case MASC_NODE:
  1019.     case RMAP_NODE:
  1020.     case VTY_NODE:
  1021.     case KEYCHAIN_NODE:
  1022.       vty->node = CONFIG_NODE;
  1023.       break;
  1024.     case BGP_VPNV4_NODE:
  1025.     case BGP_IPV4_NODE:
  1026.     case BGP_IPV4M_NODE:
  1027.     case BGP_IPV6_NODE:
  1028.       vty->node = BGP_NODE;
  1029.       break;
  1030.     case KEYCHAIN_KEY_NODE:
  1031.       vty->node = KEYCHAIN_NODE;
  1032.       break;
  1033.     default:
  1034.       break;
  1035.     }
  1036.   return CMD_SUCCESS;
  1037. }
  1039. DEFUNSH (VTYSH_ALL,
  1040.  vtysh_exit_all,
  1041.  vtysh_exit_all_cmd,
  1042.  "exit",
  1043.  "Exit current mode and down to previous moden")
  1044. {
  1045.   return vtysh_exit (vty);
  1046. }
  1048. ALIAS (vtysh_exit_all,
  1049.        vtysh_quit_all_cmd,
  1050.        "quit",
  1051.        "Exit current mode and down to previous moden");
  1053. DEFUNSH (VTYSH_BGPD,
  1054.  exit_address_family,
  1055.  exit_address_family_cmd,
  1056.  "exit-address-family",
  1057.  "Exit from Address Family configuration moden")
  1058. {
  1059.   if (vty->node == BGP_IPV4_NODE
  1060.       || vty->node == BGP_IPV4M_NODE
  1061.       || vty->node == BGP_VPNV4_NODE
  1062.       || vty->node == BGP_IPV6_NODE)
  1063.     vty->node = BGP_NODE;
  1064.   return CMD_SUCCESS;
  1065. }
  1067. DEFUNSH (VTYSH_ZEBRA,
  1068.  vtysh_exit_zebra,
  1069.  vtysh_exit_zebra_cmd,
  1070.  "exit",
  1071.  "Exit current mode and down to previous moden")
  1072. {
  1073.   return vtysh_exit (vty);
  1074. }
  1076. ALIAS (vtysh_exit_zebra,
  1077.        vtysh_quit_zebra_cmd,
  1078.        "quit",
  1079.        "Exit current mode and down to previous moden");
  1081. DEFUNSH (VTYSH_RIPD,
  1082.  vtysh_exit_ripd,
  1083.  vtysh_exit_ripd_cmd,
  1084.  "exit",
  1085.  "Exit current mode and down to previous moden")
  1086. {
  1087.   return vtysh_exit (vty);
  1088. }
  1090. ALIAS (vtysh_exit_ripd,
  1091.        vtysh_quit_ripd_cmd,
  1092.        "quit",
  1093.        "Exit current mode and down to previous moden");
  1095. DEFUNSH (VTYSH_RIPNGD,
  1096.     vtysh_exit_ripngd,
  1097.     vtysh_exit_ripngd_cmd,
  1098.     "exit",
  1099.     "Exit current mode and down to previous moden")
  1100. {
  1101.   return vtysh_exit (vty);
  1102. }
  1104. ALIAS (vtysh_exit_ripngd,
  1105.        vtysh_quit_ripngd_cmd,
  1106.        "quit",
  1107.        "Exit current mode and down to previous moden");
  1109. DEFUNSH (VTYSH_RMAP,
  1110.  vtysh_exit_rmap,
  1111.  vtysh_exit_rmap_cmd,
  1112.  "exit",
  1113.  "Exit current mode and down to previous moden")
  1114. {
  1115.   return vtysh_exit (vty);
  1116. }
  1118. ALIAS (vtysh_exit_rmap,
  1119.        vtysh_quit_rmap_cmd,
  1120.        "quit",
  1121.        "Exit current mode and down to previous moden");
  1123. DEFUNSH (VTYSH_BGPD,
  1124.  vtysh_exit_bgpd,
  1125.  vtysh_exit_bgpd_cmd,
  1126.  "exit",
  1127.  "Exit current mode and down to previous moden")
  1128. {
  1129.   return vtysh_exit (vty);
  1130. }
  1132. ALIAS (vtysh_exit_bgpd,
  1133.        vtysh_quit_bgpd_cmd,
  1134.        "quit",
  1135.        "Exit current mode and down to previous moden");
  1137. DEFUNSH (VTYSH_OSPFD,
  1138.  vtysh_exit_ospfd,
  1139.  vtysh_exit_ospfd_cmd,
  1140.  "exit",
  1141.  "Exit current mode and down to previous moden")
  1142. {
  1143.   return vtysh_exit (vty);
  1144. }
  1146. ALIAS (vtysh_exit_ospfd,
  1147.        vtysh_quit_ospfd_cmd,
  1148.        "quit",
  1149.        "Exit current mode and down to previous moden");
  1151. DEFUNSH (VTYSH_OSPF6D,
  1152.     vtysh_exit_ospf6d,
  1153.     vtysh_exit_ospf6d_cmd,
  1154.     "exit",
  1155.     "Exit current mode and down to previous moden")
  1156. {
  1157.   return vtysh_exit (vty);
  1158. }
  1160. ALIAS (vtysh_exit_ospf6d,
  1161.        vtysh_quit_ospf6d_cmd,
  1162.        "quit",
  1163.        "Exit current mode and down to previous moden");
  1165. DEFUNSH (VTYSH_ZEBRA|VTYSH_RIPD|VTYSH_OSPFD|VTYSH_OSPF6D,
  1166.  vtysh_interface,
  1167.  vtysh_interface_cmd,
  1168.  "interface IFNAME",
  1169.  "Select an interface to configuren"
  1170.  "Interface's namen")
  1171. {
  1172.   vty->node = INTERFACE_NODE;
  1173.   return CMD_SUCCESS;
  1174. }
  1176. DEFUNSH (VTYSH_ZEBRA|VTYSH_RIPD|VTYSH_OSPFD|VTYSH_OSPF6D,
  1177.  vtysh_exit_interface,
  1178.  vtysh_exit_interface_cmd,
  1179.  "exit",
  1180.  "Exit current mode and down to previous moden")
  1181. {
  1182.   return vtysh_exit (vty);
  1183. }
  1185. ALIAS (vtysh_exit_interface,
  1186.        vtysh_quit_interface_cmd,
  1187.        "quit",
  1188.        "Exit current mode and down to previous moden");
  1190. DEFUN (vtysh_write_terminal,
  1191.        vtysh_write_terminal_cmd,
  1192.        "write terminal",
  1193.        "Write running configuration to memory, network, or terminaln"
  1194.        "Write to terminaln")
  1195. {
  1196.   int ret;
  1197.   char line[] = "write terminaln";
  1198.   FILE *fp = NULL;
  1200.   if (vtysh_pager_name)
  1201.     {
  1202.       fp = popen ("more", "w");
  1203.       if (fp == NULL)
  1204. {
  1205.   perror ("popen");
  1206.   exit (1);
  1207. }
  1208.     }
  1209.   else
  1210.     fp = stdout;
  1212.   vty_out (vty, "Building configuration...%s", VTY_NEWLINE);
  1213.   vty_out (vty, "%sCurrent configuration:%s", VTY_NEWLINE,
  1214.    VTY_NEWLINE);
  1216.   vtysh_config_write (fp);
  1218.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_ZEBRA], line);
  1219.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_RIP], line);
  1220.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_RIPNG], line);
  1221.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_OSPF], line);
  1222.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_OSPF6], line);
  1223.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_BGP], line);
  1225.   vtysh_config_dump (fp);
  1227.   if (vtysh_pager_name && fp)
  1228.     {
  1229.       fflush (fp);
  1230.       if (pclose (fp) == -1)
  1231. {
  1232.   perror ("pclose");
  1233.   exit (1);
  1234. }
  1235.       fp = NULL;
  1236.     }
  1238.   return CMD_SUCCESS;
  1239. }
  1241. DEFUN (vtysh_write_memory,
  1242.        vtysh_write_memory_cmd,
  1243.        "write memory",
  1244.        "Write running configuration to memory, network, or terminaln"
  1245.        "Write configuration to the file (same as write file)n")
  1246. {
  1247.   int ret;
  1248.   mode_t old_umask;
  1249.   char line[] = "write terminaln";
  1250.   FILE *fp;
  1251.   char *integrate_sav = NULL;
  1253.   /* config files have 0600 perms... */ 
  1254.   old_umask = umask (0077);
  1256.   integrate_sav = malloc (strlen (integrate_default) 
  1257.     + strlen (CONF_BACKUP_EXT) + 1);
  1258.   strcpy (integrate_sav, integrate_default);
  1259.   strcat (integrate_sav, CONF_BACKUP_EXT);
  1262.   printf ("Building Configuration...n");
  1264.   /* Move current configuration file to backup config file */
  1265.   unlink (integrate_sav);
  1266.   rename (integrate_default, integrate_sav);
  1268.   fp = fopen (integrate_default, "w");
  1269.   if (fp == NULL)
  1270.     {
  1271.       printf ("%% Can't open configuration file %s.n", integrate_default);
  1272.       umask (old_umask);
  1273.       return CMD_SUCCESS;
  1274.     }
  1275.   else
  1276.     printf ("[OK]n");
  1277.   
  1279.   vtysh_config_write (fp);
  1281.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_ZEBRA], line);
  1282.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_RIP], line);
  1283.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_RIPNG], line);
  1284.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_OSPF], line);
  1285.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_OSPF6], line);
  1286.   ret = vtysh_client_config (&vtysh_client[VTYSH_INDEX_BGP], line);
  1288.   vtysh_config_dump (fp);
  1290.   fclose (fp);
  1292.   umask (old_umask);
  1293.   return CMD_SUCCESS;
  1294. }
  1296. ALIAS (vtysh_write_memory,
  1297.        vtysh_copy_runningconfig_startupconfig_cmd,
  1298.        "copy running-config startup-config",  
  1299.        "Copy from one file to anothern"
  1300.        "Copy from current system configurationn"
  1301.        "Copy to startup configurationn");
  1303. ALIAS (vtysh_write_memory,
  1304.        vtysh_write_file_cmd,
  1305.        "write file",
  1306.        "Write running configuration to memory, network, or terminaln"
  1307.        "Write configuration to the file (same as write memory)n");
  1309. ALIAS (vtysh_write_terminal,
  1310.        vtysh_show_running_config_cmd,
  1311.        "show running-config",
  1312.        SHOW_STR
  1313.        "Current operating configurationn");
  1315. /* Execute command in child process. */
  1316. int
  1317. execute_command (char *command, int argc, char *arg1, char *arg2)
  1318. {
  1319.   int ret;
  1320.   pid_t pid;
  1321.   int status;
  1323.   /* Call fork(). */
  1324.   pid = fork ();
  1326.   if (pid < 0)
  1327.     {
  1328.       /* Failure of fork(). */
  1329.       fprintf (stderr, "Can't fork: %sn", strerror (errno));
  1330.       exit (1);
  1331.     }
  1332.   else if (pid == 0)
  1333.     {
  1334.       /* This is child process. */
  1335.       switch (argc)
  1336. {
  1337. case 0:
  1338.   ret = execlp (command, command, NULL);
  1339.   break;
  1340. case 1:
  1341.   ret = execlp (command, command, arg1, NULL);
  1342.   break;
  1343. case 2:
  1344.   ret = execlp (command, command, arg1, arg2, NULL);
  1345.   break;
  1346. }
  1348.       /* When execlp suceed, this part is not executed. */
  1349.       fprintf (stderr, "Can't execute %s: %sn", command, strerror (errno));
  1350.       exit (1);
  1351.     }
  1352.   else
  1353.     {
  1354.       /* This is parent. */
  1355.       execute_flag = 1;
  1356.       ret = wait4 (pid, &status, 0, NULL);
  1357.       execute_flag = 0;
  1358.     }
  1359.   return 0;
  1360. }
  1362. DEFUN (vtysh_ping,
  1363.        vtysh_ping_cmd,
  1364.        "ping WORD",
  1365.        "send echo messagesn"
  1366.        "Ping destination address or hostnamen")
  1367. {
  1368.   execute_command ("ping", 1, argv[0], NULL);
  1369.   return CMD_SUCCESS;
  1370. }
  1372. DEFUN (vtysh_traceroute,
  1373.        vtysh_traceroute_cmd,
  1374.        "traceroute WORD",
  1375.        "Trace route to destinationn"
  1376.        "Trace route to destination address or hostnamen")
  1377. {
  1378.   execute_command ("traceroute", 1, argv[0], NULL);
  1379.   return CMD_SUCCESS;
  1380. }
  1382. DEFUN (vtysh_telnet,
  1383.        vtysh_telnet_cmd,
  1384.        "telnet WORD",
  1385.        "Open a telnet connectionn"
  1386.        "IP address or hostname of a remote systemn")
  1387. {
  1388.   execute_command ("telnet", 1, argv[0], NULL);
  1389.   return CMD_SUCCESS;
  1390. }
  1392. DEFUN (vtysh_telnet_port,
  1393.        vtysh_telnet_port_cmd,
  1394.        "telnet WORD PORT",
  1395.        "Open a telnet connectionn"
  1396.        "IP address or hostname of a remote systemn"
  1397.        "TCP Port numbern")
  1398. {
  1399.   execute_command ("telnet", 2, argv[0], argv[1]);
  1400.   return CMD_SUCCESS;
  1401. }
  1403. DEFUN (vtysh_start_shell,
  1404.        vtysh_start_shell_cmd,
  1405.        "start-shell",
  1406.        "Start UNIX shelln")
  1407. {
  1408.   execute_command ("sh", 0, NULL, NULL);
  1409.   return CMD_SUCCESS;
  1410. }
  1412. DEFUN (vtysh_start_bash,
  1413.        vtysh_start_bash_cmd,
  1414.        "start-shell bash",
  1415.        "Start UNIX shelln"
  1416.        "Start bashn")
  1417. {
  1418.   execute_command ("bash", 0, NULL, NULL);
  1419.   return CMD_SUCCESS;
  1420. }
  1422. DEFUN (vtysh_start_zsh,
  1423.        vtysh_start_zsh_cmd,
  1424.        "start-shell zsh",
  1425.        "Start UNIX shelln"
  1426.        "Start Z shelln")
  1427. {
  1428.   execute_command ("zsh", 0, NULL, NULL);
  1429.   return CMD_SUCCESS;
  1430. }
  1432. /* Route map node structure. */
  1433. struct cmd_node rmap_node =
  1434. {
  1435.   RMAP_NODE,
  1436.   "%s(config-route-map)# "
  1437. };
  1439. /* Zebra node structure. */
  1440. struct cmd_node zebra_node =
  1441. {
  1442.   ZEBRA_NODE,
  1443.   "%s(config-router)# "
  1444. };
  1446. struct cmd_node bgp_vpnv4_node =
  1447. {
  1448.   BGP_VPNV4_NODE,
  1449.   "%s(config-router-af)# "
  1450. };
  1452. struct cmd_node bgp_ipv4_node =
  1453. {
  1454.   BGP_IPV4_NODE,
  1455.   "%s(config-router-af)# "
  1456. };
  1458. struct cmd_node bgp_ipv4m_node =
  1459. {
  1460.   BGP_IPV4M_NODE,
  1461.   "%s(config-router-af)# "
  1462. };
  1464. struct cmd_node bgp_ipv6_node =
  1465. {
  1466.   BGP_IPV6_NODE,
  1467.   "%s(config-router-af)# "
  1468. };
  1470. struct cmd_node ospf_node =
  1471. {
  1472.   OSPF_NODE,
  1473.   "%s(config-router)# "
  1474. };
  1476. /* RIPng node structure. */
  1477. struct cmd_node ripng_node =
  1478. {
  1479.   RIPNG_NODE,
  1480.   "%s(config-router)# "
  1481. };
  1483. /* OSPF6 node structure. */
  1484. struct cmd_node ospf6_node =
  1485. {
  1486.   OSPF6_NODE,
  1487.   "%s(config-ospf6)# "
  1488. };
  1490. struct cmd_node keychain_node =
  1491. {
  1492.   KEYCHAIN_NODE,
  1493.   "%s(config-keychain)# "
  1494. };
  1496. struct cmd_node keychain_key_node =
  1497. {
  1498.   KEYCHAIN_KEY_NODE,
  1499.   "%s(config-keychain-key)# "
  1500. };
  1502. void
  1503. vtysh_install_default (enum node_type node)
  1504. {
  1505.   install_element (node, &config_list_cmd);
  1506. }
  1508. /* Making connection to protocol daemon. */
  1509. int
  1510. vtysh_connect (struct vtysh_client *vclient, char *path)
  1511. {
  1512.   int ret;
  1513.   int sock, len;
  1514.   struct sockaddr_un addr;
  1515.   struct stat s_stat;
  1516.   uid_t euid;
  1517.   gid_t egid;
  1519.   memset (vclient, 0, sizeof (struct vtysh_client));
  1520.   vclient->fd = -1;
  1522.   /* Stat socket to see if we have permission to access it. */
  1523.   euid = geteuid();
  1524.   egid = getegid();
  1525.   ret = stat (path, &s_stat);
  1526.   if (ret < 0 && errno != ENOENT)
  1527.     {
  1528.       fprintf  (stderr, "vtysh_connect(%s): stat = %sn", 
  1529. path, strerror(errno)); 
  1530.       exit(1);
  1531.     }
  1532.   
  1533.   if (ret >= 0)
  1534.     {
  1535.       if (! S_ISSOCK(s_stat.st_mode))
  1536. {
  1537.   fprintf (stderr, "vtysh_connect(%s): Not a socketn",
  1538.    path);
  1539.   exit (1);
  1540. }
  1541.       
  1542.       if (euid != s_stat.st_uid 
  1543.   || !(s_stat.st_mode & S_IWUSR)
  1544.   || !(s_stat.st_mode & S_IRUSR))
  1545. {
  1546.   fprintf (stderr, "vtysh_connect(%s): No permission to access socketn",
  1547.    path);
  1548.   exit (1);
  1549. }
  1550.     }
  1552.   sock = socket (AF_UNIX, SOCK_STREAM, 0);
  1553.   if (sock < 0)
  1554.     {
  1555. #ifdef DEBUG
  1556.       fprintf(stderr, "vtysh_connect(%s): socket = %sn", path, strerror(errno));
  1557. #endif /* DEBUG */
  1558.       return -1;
  1559.     }
  1561.   memset (&addr, 0, sizeof (struct sockaddr_un));
  1562.   addr.sun_family = AF_UNIX;
  1563.   strncpy (addr.sun_path, path, strlen (path));
  1564. #ifdef HAVE_SUN_LEN
  1565.   len = addr.sun_len = SUN_LEN(&addr);
  1566. #else
  1567.   len = sizeof (addr.sun_family) + strlen (addr.sun_path);
  1568. #endif /* HAVE_SUN_LEN */
  1570.   ret = connect (sock, (struct sockaddr *) &addr, len);
  1571.   if (ret < 0)
  1572.     {
  1573. #ifdef DEBUG
  1574.       fprintf(stderr, "vtysh_connect(%s): connect = %sn", path, strerror(errno));
  1575. #endif /* DEBUG */
  1576.       close (sock);
  1577.       return -1;
  1578.     }
  1579.   vclient->fd = sock;
  1581.   return 0;
  1582. }
  1584. void
  1585. vtysh_connect_all()
  1586. {
  1587.   /* Clear each daemons client structure. */
  1588.   vtysh_connect (&vtysh_client[VTYSH_INDEX_ZEBRA], ZEBRA_PATH);
  1589.   vtysh_connect (&vtysh_client[VTYSH_INDEX_RIP], RIP_PATH);
  1590.   vtysh_connect (&vtysh_client[VTYSH_INDEX_RIPNG], RIPNG_PATH);
  1591.   vtysh_connect (&vtysh_client[VTYSH_INDEX_OSPF], OSPF_PATH);
  1592.   vtysh_connect (&vtysh_client[VTYSH_INDEX_OSPF6], OSPF6_PATH);
  1593.   vtysh_connect (&vtysh_client[VTYSH_INDEX_BGP], BGP_PATH);
  1594. }
  1597. /* To disable readline's filename completion */
  1598. int
  1599. vtysh_completion_entry_function (int ignore, int invoking_key)
  1600. {
  1601.   return 0;
  1602. }
  1604. void
  1605. vtysh_readline_init ()
  1606. {
  1607.   /* readline related settings. */
  1608.   rl_bind_key ('?', vtysh_rl_describe);
  1609.   rl_completion_entry_function = vtysh_completion_entry_function;
  1610.   rl_attempted_completion_function = (CPPFunction *)new_completion;
  1611.   /* do not append space after completion. It will be appended
  1612.      in new_completion() function explicitly */
  1613.   rl_completion_append_character = '';
  1614. }
  1616. char *
  1617. vtysh_prompt ()
  1618. {
  1619.   struct utsname names;
  1620.   static char buf[100];
  1621.   const char*hostname;
  1622.   extern struct host host;
  1624.   hostname = host.name;
  1626.   if (!hostname)
  1627.     {
  1628.       uname (&names);
  1629.       hostname = names.nodename;
  1630.     }
  1632.   snprintf (buf, sizeof buf, cmd_prompt (vty->node), hostname);
  1634.   return buf;
  1635. }
  1637. void
  1638. vtysh_init_vty ()
  1639. {
  1640.   /* Make vty structure. */
  1641.   vty = vty_new ();
  1642.   vty->type = VTY_SHELL;
  1643.   vty->node = VIEW_NODE;
  1645.   /* Initialize commands. */
  1646.   cmd_init (0);
  1648.   /* Install nodes. */
  1649.   install_node (&bgp_node, NULL);
  1650.   install_node (&rip_node, NULL);
  1651.   install_node (&interface_node, NULL);
  1652.   install_node (&rmap_node, NULL);
  1653.   install_node (&zebra_node, NULL);
  1654.   install_node (&bgp_vpnv4_node, NULL);
  1655.   install_node (&bgp_ipv4_node, NULL);
  1656.   install_node (&bgp_ipv4m_node, NULL);
  1657. /* #ifdef HAVE_IPV6 */
  1658.   install_node (&bgp_ipv6_node, NULL);
  1659. /* #endif */
  1660.   install_node (&ospf_node, NULL);
  1661. /* #ifdef HAVE_IPV6 */
  1662.   install_node (&ripng_node, NULL);
  1663.   install_node (&ospf6_node, NULL);
  1664. /* #endif */
  1665.   install_node (&keychain_node, NULL);
  1666.   install_node (&keychain_key_node, NULL);
  1668.   vtysh_install_default (VIEW_NODE);
  1669.   vtysh_install_default (ENABLE_NODE);
  1670.   vtysh_install_default (CONFIG_NODE);
  1671.   vtysh_install_default (BGP_NODE);
  1672.   vtysh_install_default (RIP_NODE);
  1673.   vtysh_install_default (INTERFACE_NODE);
  1674.   vtysh_install_default (RMAP_NODE);
  1675.   vtysh_install_default (ZEBRA_NODE);
  1676.   vtysh_install_default (BGP_VPNV4_NODE);
  1677.   vtysh_install_default (BGP_IPV4_NODE);
  1678.   vtysh_install_default (BGP_IPV4M_NODE);
  1679.   vtysh_install_default (BGP_IPV6_NODE);
  1680.   vtysh_install_default (OSPF_NODE);
  1681.   vtysh_install_default (RIPNG_NODE);
  1682.   vtysh_install_default (OSPF6_NODE);
  1683.   vtysh_install_default (KEYCHAIN_NODE);
  1684.   vtysh_install_default (KEYCHAIN_KEY_NODE);
  1686.   install_element (VIEW_NODE, &vtysh_enable_cmd);
  1687.   install_element (ENABLE_NODE, &vtysh_config_terminal_cmd);
  1688.   install_element (ENABLE_NODE, &vtysh_disable_cmd);
  1690.   /* "exit" command. */
  1691.   install_element (VIEW_NODE, &vtysh_exit_all_cmd);
  1692.   install_element (VIEW_NODE, &vtysh_quit_all_cmd);
  1693.   install_element (CONFIG_NODE, &vtysh_exit_all_cmd);
  1694.   /* install_element (CONFIG_NODE, &vtysh_quit_all_cmd); */
  1695.   install_element (ENABLE_NODE, &vtysh_exit_all_cmd);
  1696.   install_element (ENABLE_NODE, &vtysh_quit_all_cmd);
  1697.   install_element (RIP_NODE, &vtysh_exit_ripd_cmd);
  1698.   install_element (RIP_NODE, &vtysh_quit_ripd_cmd);
  1699.   install_element (RIPNG_NODE, &vtysh_exit_ripngd_cmd);
  1700.   install_element (RIPNG_NODE, &vtysh_quit_ripngd_cmd);
  1701.   install_element (OSPF_NODE, &vtysh_exit_ospfd_cmd);
  1702.   install_element (OSPF_NODE, &vtysh_quit_ospfd_cmd);
  1703.   install_element (OSPF6_NODE, &vtysh_exit_ospf6d_cmd);
  1704.   install_element (OSPF6_NODE, &vtysh_quit_ospf6d_cmd);
  1705.   install_element (BGP_NODE, &vtysh_exit_bgpd_cmd);
  1706.   install_element (BGP_NODE, &vtysh_quit_bgpd_cmd);
  1707.   install_element (BGP_VPNV4_NODE, &vtysh_exit_bgpd_cmd);
  1708.   install_element (BGP_VPNV4_NODE, &vtysh_quit_bgpd_cmd);
  1709.   install_element (BGP_IPV4_NODE, &vtysh_exit_bgpd_cmd);
  1710.   install_element (BGP_IPV4_NODE, &vtysh_quit_bgpd_cmd);
  1711.   install_element (BGP_IPV4M_NODE, &vtysh_exit_bgpd_cmd);
  1712.   install_element (BGP_IPV4M_NODE, &vtysh_quit_bgpd_cmd);
  1713.   install_element (BGP_IPV6_NODE, &vtysh_exit_bgpd_cmd);
  1714.   install_element (BGP_IPV6_NODE, &vtysh_quit_bgpd_cmd);
  1715.   install_element (KEYCHAIN_NODE, &vtysh_exit_ripd_cmd);
  1716.   install_element (KEYCHAIN_NODE, &vtysh_quit_ripd_cmd);
  1717.   install_element (KEYCHAIN_KEY_NODE, &vtysh_exit_ripd_cmd);
  1718.   install_element (KEYCHAIN_KEY_NODE, &vtysh_quit_ripd_cmd);
  1719.   install_element (RMAP_NODE, &vtysh_exit_rmap_cmd);
  1720.   install_element (RMAP_NODE, &vtysh_quit_rmap_cmd);
  1722.   /* "end" command. */
  1723.   install_element (CONFIG_NODE, &vtysh_end_all_cmd);
  1724.   install_element (ENABLE_NODE, &vtysh_end_all_cmd);
  1725.   install_element (RIP_NODE, &vtysh_end_all_cmd);
  1726.   install_element (RIPNG_NODE, &vtysh_end_all_cmd);
  1727.   install_element (OSPF_NODE, &vtysh_end_all_cmd);
  1728.   install_element (OSPF6_NODE, &vtysh_end_all_cmd);
  1729.   install_element (BGP_NODE, &vtysh_end_all_cmd);
  1730.   install_element (BGP_IPV4_NODE, &vtysh_end_all_cmd);
  1731.   install_element (BGP_IPV4M_NODE, &vtysh_end_all_cmd);
  1732.   install_element (BGP_VPNV4_NODE, &vtysh_end_all_cmd);
  1733.   install_element (BGP_IPV6_NODE, &vtysh_end_all_cmd);
  1734.   install_element (KEYCHAIN_NODE, &vtysh_end_all_cmd);
  1735.   install_element (KEYCHAIN_KEY_NODE, &vtysh_end_all_cmd);
  1736.   install_element (RMAP_NODE, &vtysh_end_all_cmd);
  1738.   install_element (INTERFACE_NODE, &vtysh_end_all_cmd);
  1739.   install_element (INTERFACE_NODE, &vtysh_exit_interface_cmd);
  1740.   install_element (INTERFACE_NODE, &vtysh_quit_interface_cmd);
  1741.   install_element (CONFIG_NODE, &router_rip_cmd);
  1742. #ifdef HAVE_IPV6
  1743.   install_element (CONFIG_NODE, &router_ripng_cmd);
  1744. #endif
  1745.   install_element (CONFIG_NODE, &router_ospf_cmd);
  1746. #ifdef HAVE_IPV6
  1747.   install_element (CONFIG_NODE, &router_ospf6_cmd);
  1748. #endif
  1749.   install_element (CONFIG_NODE, &router_bgp_cmd);
  1750.   install_element (BGP_NODE, &address_family_vpnv4_cmd);
  1751.   install_element (BGP_NODE, &address_family_vpnv4_unicast_cmd);
  1752.   install_element (BGP_NODE, &address_family_ipv4_cmd);
  1753.   install_element (BGP_NODE, &address_family_ipv4_unicast_cmd);
  1754.   install_element (BGP_NODE, &address_family_ipv4_multicast_cmd);
  1755. #ifdef HAVE_IPV6
  1756.   install_element (BGP_NODE, &address_family_ipv6_cmd);
  1757.   install_element (BGP_NODE, &address_family_ipv6_unicast_cmd);
  1758. #endif
  1759.   install_element (BGP_VPNV4_NODE, &exit_address_family_cmd);
  1760.   install_element (BGP_IPV4_NODE, &exit_address_family_cmd);
  1761.   install_element (BGP_IPV4M_NODE, &exit_address_family_cmd);
  1762.   install_element (BGP_IPV6_NODE, &exit_address_family_cmd);
  1763.   install_element (CONFIG_NODE, &key_chain_cmd);
  1764.   install_element (CONFIG_NODE, &route_map_cmd);
  1765.   install_element (KEYCHAIN_NODE, &key_cmd);
  1766.   install_element (KEYCHAIN_NODE, &key_chain_cmd);
  1767.   install_element (KEYCHAIN_KEY_NODE, &key_chain_cmd);
  1768.   install_element (CONFIG_NODE, &vtysh_interface_cmd);
  1769.   install_element (ENABLE_NODE, &vtysh_show_running_config_cmd);
  1770.   install_element (ENABLE_NODE, &vtysh_copy_runningconfig_startupconfig_cmd);
  1771.   install_element (ENABLE_NODE, &vtysh_write_file_cmd);
  1773.   /* write terminal command */
  1774.   install_element (ENABLE_NODE, &vtysh_write_terminal_cmd);
  1775.   install_element (CONFIG_NODE, &vtysh_write_terminal_cmd);
  1776.   install_element (BGP_NODE, &vtysh_write_terminal_cmd);
  1777.   install_element (BGP_VPNV4_NODE, &vtysh_write_terminal_cmd);
  1778.   install_element (BGP_IPV4_NODE, &vtysh_write_terminal_cmd);
  1779.   install_element (BGP_IPV4M_NODE, &vtysh_write_terminal_cmd);
  1780.   install_element (BGP_IPV6_NODE, &vtysh_write_terminal_cmd);
  1781.   install_element (RIP_NODE, &vtysh_write_terminal_cmd);
  1782.   install_element (RIPNG_NODE, &vtysh_write_terminal_cmd);
  1783.   install_element (OSPF_NODE, &vtysh_write_terminal_cmd);
  1784.   install_element (OSPF6_NODE, &vtysh_write_terminal_cmd);
  1785.   install_element (INTERFACE_NODE, &vtysh_write_terminal_cmd);
  1786.   install_element (RMAP_NODE, &vtysh_write_terminal_cmd);
  1787.   install_element (KEYCHAIN_NODE, &vtysh_write_terminal_cmd);
  1788.   install_element (KEYCHAIN_KEY_NODE, &vtysh_write_terminal_cmd);
  1790.   /* write memory command */
  1791.   install_element (ENABLE_NODE, &vtysh_write_memory_cmd);
  1792.   install_element (CONFIG_NODE, &vtysh_write_memory_cmd);
  1793.   install_element (BGP_NODE, &vtysh_write_memory_cmd);
  1794.   install_element (BGP_VPNV4_NODE, &vtysh_write_memory_cmd);
  1795.   install_element (BGP_IPV4_NODE, &vtysh_write_memory_cmd);
  1796.   install_element (BGP_IPV4M_NODE, &vtysh_write_memory_cmd);
  1797.   install_element (BGP_IPV6_NODE, &vtysh_write_memory_cmd);
  1798.   install_element (RIP_NODE, &vtysh_write_memory_cmd);
  1799.   install_element (RIPNG_NODE, &vtysh_write_memory_cmd);
  1800.   install_element (OSPF_NODE, &vtysh_write_memory_cmd);
  1801.   install_element (OSPF6_NODE, &vtysh_write_memory_cmd);
  1802.   install_element (INTERFACE_NODE, &vtysh_write_memory_cmd);
  1803.   install_element (RMAP_NODE, &vtysh_write_memory_cmd);
  1804.   install_element (KEYCHAIN_NODE, &vtysh_write_memory_cmd);
  1805.   install_element (KEYCHAIN_KEY_NODE, &vtysh_write_memory_cmd);
  1807.   install_element (VIEW_NODE, &vtysh_ping_cmd);
  1808.   install_element (VIEW_NODE, &vtysh_traceroute_cmd);
  1809.   install_element (VIEW_NODE, &vtysh_telnet_cmd);
  1810.   install_element (VIEW_NODE, &vtysh_telnet_port_cmd);
  1811.   install_element (ENABLE_NODE, &vtysh_ping_cmd);
  1812.   install_element (ENABLE_NODE, &vtysh_traceroute_cmd);
  1813.   install_element (ENABLE_NODE, &vtysh_telnet_cmd);
  1814.   install_element (ENABLE_NODE, &vtysh_telnet_port_cmd);
  1815.   install_element (ENABLE_NODE, &vtysh_start_shell_cmd);
  1816.   install_element (ENABLE_NODE, &vtysh_start_bash_cmd);
  1817.   install_element (ENABLE_NODE, &vtysh_start_zsh_cmd);
  1819.   install_element (CONFIG_NODE, &vtysh_log_stdout_cmd);
  1820.   install_element (CONFIG_NODE, &no_vtysh_log_stdout_cmd);
  1821.   install_element (CONFIG_NODE, &vtysh_log_file_cmd);
  1822.   install_element (CONFIG_NODE, &no_vtysh_log_file_cmd);
  1823.   install_element (CONFIG_NODE, &vtysh_log_syslog_cmd);
  1824.   install_element (CONFIG_NODE, &no_vtysh_log_syslog_cmd);
  1825.   install_element (CONFIG_NODE, &vtysh_log_trap_cmd);
  1826.   install_element (CONFIG_NODE, &no_vtysh_log_trap_cmd);
  1827.   install_element (CONFIG_NODE, &vtysh_log_record_priority_cmd);
  1828.   install_element (CONFIG_NODE, &no_vtysh_log_record_priority_cmd);
  1829. }