• 目录
• Zebra源码分析和改造三-添加定制命令

Zebra源码分析和改造三-添加定制命令

一 视图介绍

由上面几篇文章分析可见，所有的命令都是包含在node中的，根据Cisco或者H3常见路由器或者交换机的CLI格式可见，一个node就对应着一个视图（View）。常用的视图包括：普通视图，管理视图，文件系统视图，配置视图，以及接口配置视图和VLAN视图等。

在Zebra-VTYSH源码中，实现了的有Enable视图和配置视图。如下图所示：

/ # vtysh   

Copyright 2010-2011 IBM Co., Ltd.  

CLI> enable   
CLI#   
  clear        Reset functions  
  configure    Configuration from vty interface  
  copy         Copy from one file to another  
  debug        Debugging functions (see also 'undebug')  
  disable      Turn off privileged mode command  
  end          End current mode and down to previous mode  
  exit         Exit current mode and down to previous mode  
  list         Print command list  
  no           Negate a command or set its defaults  
  ping         send echo messages  
  quit         Exit current mode and down to previous mode  
  show         Show running system information  
  start-shell  Start UNIX shell  
  telnet       Open a telnet connection  
  terminal     Set terminal line parameters  
  traceroute   Trace route to destination  
  undebug      Disable debugging functions (see also 'debug')  
  write        Write running configuration to memory, network, or terminal  
CLI# configure terminal   
CLI(config)#   
  access-list    Add an access list entry  
  bgp            BGP information  
  debug          Debugging functions (see also 'undebug')  
  device-config  Device configuration  
  dump           Dump packet  
  enable         Modify enable password parameters  
  end            End current mode and down to previous mode  
  exit           Exit current mode and down to previous mode  
  hostname       Set system's network name  
  interface      Select an interface to configure  
  ip             IP information  
  ipv6           IPv6 information  
  key            Authentication key management  
  list           Print command list  
  log            Logging control  
  no             Negate a command or set its defaults  
  password       Assign the terminal connection password  
  route-map      Create route-map or enter route-map command mode  
  router         Enable a routing process  
  system-config  System and management configuration  
  username  
  write          Write running configuration to memory, network, or terminal  
CLI(config)# system-config   
CLI(config-system)#   
  access                   Set CPE access ND flag  
  admin-idle-time          Set system idle time  
  admin-psw                Set system administrator password  
  admin-username           Set system administrator username  
  connection-mode          Set network connection mode : static and dynamic  
  datetime                 Set date time (format:2000-01-01 00:00:00)  
  default-gateway          Set system's network default gateway  
  dns-server-1             Set system network DNS server 1  
  dns-server-2             Set system network DNS server 2  
  exit                     Exit current mode and down to previous mode  
  factory-defaults         Restore ALL configure to factory default values( 0: reset all 1: reset with network parameters unchanged)  
  hostname                 Set system's network name  
  image-upgrade            Upgrade image via ftp method  
  ip                       Set system ip address and netmask  
  list                     Print command list  
  managment-ip-range       Set management IP range and netmask  
  managment-ip-range-flag  Set management IP range service flag  
  mgr-vlan-id              Set management VLAN ID  
  ntpserver                Set NTP server  
  quit                     Exit current mode and down to previous mode  
  reset                    Reset system  
  snmp-refresh-time        Set SNMP service refresh time cycle  
  snmp-rwcommunicty        Set SNMP read/write community  
  snmp-service             Set SNMP service enable or disable  
  snmp-trap-ip1            Set SNMP trap ip 1 address  
  snmp-trap-ip2            Set SNMP trap ip 2 address  
  snmp-trap1-ip-flag       Set SNMP trap ip 1 service flag(enable/disable)  
  snmp-trap2-ip-flag       Set SNMP trap ip 2 service flag(enable/disable)  
  ssh                      Set ssh service port and timeout values  
  ssh-service              Set ssh service flag  
  telnet                   Set telnet PORT  
  telnet-service           Set telnet service flag  
  timesync                 Set time sync service flag  
  timezone                 Set time zone (0:ShangHai,1:ChongQing)  
CLI(config-system)# quit  
CLI(config)# device-config   
CLI(config-device)#   
  exit                       Exit current mode and down to previous mode  
  list                       Print command list  
  port-mirror-analysis-port  Device configuration: Set analysis port(1: eth1 2: eth2)  
  port-mirror-flag           Device configuration: Enable or disable port mirror service(0:disable,1:enable)  
  port-mirror-packet         Device configuration: Set packet type to be mirrored(1:Import & Export 2: Import 3: Export)  
  port-mirror-port           Device configuration:Set port to be mirrored  
  port1-rate                 Device configuration: set duplex mode and import/export/broadcast/unkown/multicast rate limit.  
  port2-rate                 Device configuration: set duplex mode and import/export/broadcast/unkown/multicast rate limit.  
  quit                       Exit current mode and down to previous mode  
CLI(config-device)#   
CLI(config-device)#

如果想要添加自己的命令，可以在原有的视图上增加（也就是在原有的node中增加commands），或者新开自己的视图，然后在新视图中添加自己的commands。

二 添加命令

进入vtysh目录中，查看vtysh_main.c文件的main函数，也就是和vtysh初始化相关的一切都在这里，基本上在这里可以完成你需要的一些基本命令。

在函数vtysh_init_vty（）中，有个

/* Initialize command interface. Install basic nodes and commands. */  
Void cmd_init (int terminal)

的函数，就是负责初始化command接口，安装node和命令的。

比如你就可以添加自己的视图如下：

/*Added by xyang*/  
    install_element (CONFIG_NODE, &vtysh_sysconfig_cmd);  
    install_element (CONFIG_NODE, &vtysh_devconfig_cmd);

（其中，安装的system和device配置的视图）

/*Added by xyang 
* system config node* 
*/  
DEFUN (system_config,  
       vtysh_sysconfig_cmd,  
       "system-config",  
       SYS_CFG_STR  
       "\n")  
{    
  //vty_out (vty, "testing by xyang.%s", VTY_NEWLINE);  

   vty->node = SYSCONFIG_NODE;  

  return CMD_SUCCESS;  
}  
DEFUN (device_config,  
       vtysh_devconfig_cmd,  
       "device-config",  
       DEV_CFG_STR  
       "\n")  
{    

  //vty_out (vty, "testing by xyang.%s", VTY_NEWLINE);  

   vty->node = DEVCONFIG_NODE;  

  return CMD_SUCCESS;  
}  
DEFUN定义为：  
/* DEFUN for vty command interafce. Little bit hacky ;-). */  
#define DEFUN(funcname, cmdname, cmdstr, helpstr) \  
  int funcname (struct cmd_element *, struct vty *, int, char **); \  
  struct cmd_element cmdname = \  
  { \  
    cmdstr, \  
    funcname, \  
    helpstr \  
  }; \  
  int funcname \  
  (struct cmd_element *self, struct vty *vty, int argc, char **argv)

SYSCONFIG_NODE和DEVCONFIG_NODE要添加进enum node_type{}中去。

最后就要在init_cmd的最后加进自己的command了

比如

/*add commands to system config node 
  * added by xyang @ 2012-02-01* 
  */  
    /*management network settings*/   

    install_element (SYSCONFIG_NODE, &vtysh_system_cfg_ip_cmd);//ip and subnet mask

其中，函数指针需要定义先：

DEFUN (vtysh_system_cfg_ip,  
     vtysh_system_cfg_ip_cmd,  
     "ip ADDRESS NETMASK",  
     "Set system ip address and netmask\n")  
{  
    applyCfg(argv[0],"IPADDR");  
    applyCfg(argv[1],"NETMASK");  
    system(NETWORK_SETTING_SCRIPT);   
    return CMD_SUCCESS;  
}

这样，基本上完成了添加node和命令的任务了。

其他Zebra-VTYH自带的命令如果不想要的话删除掉就行了。