tests/data_plane/vpp_lite_topo/scripts/cmd_mappings.py

   1
   2 mappings = {}
   3
   4 class SimpleMapping(object):
   5
   6   def __init__(self, cmd, cli, vat):
   7     if cmd in mappings:
   8       raise Exception('{} already in cmd db!'.format(cmd))
   9
  10     self.cmd = cmd
  11     self.cli = cli
  12     self.vat = vat
  13     mappings[cmd] = self
  14
  15   def generate(self, mode, args):
  16     s = ''
  17     # simply append arguments string to right command
  18     if mode == 'vat':
  19       s = self.vat + ' ' + args
  20     else:
  21       s = self.cli + ' ' + args
  22     return s
  23
  24
  25 class CustomMapping(SimpleMapping):
  26
  27   def generate(self, mode, args):
  28     s = ''
  29     if mode == 'vat':
  30       s = self.vat
  31     else:
  32       s = self.cli
  33
  34     args = args.split(' ')
  35     return s.format(*args)
  36
  37
  38 class RepeatableLocators(SimpleMapping):
  39
  40   def generate(self, mode, args):
  41     name = args[:args.index(' ')]  # first word is ls name
  42     locs = args[args.index(' '):]
  43
  44     if mode == 'vat':
  45       s = self.vat
  46     else:
  47       s = self.cli
  48
  49     s = s + ' ' + name + locs
  50     return s
  51
  52
  53 SimpleMapping('lisp_state', 'lisp', 'lisp_enable_disable')
  54 SimpleMapping('lisp_map_resolver', 'lisp map-resolver', 'lisp_add_del_map_resolver')
  55 SimpleMapping('lisp_map_server', 'lisp map-server', 'lisp_add_del_map_server')
  56 SimpleMapping('lisp_local_eid', 'lisp eid-table', 'lisp_add_del_local_eid')
  57 SimpleMapping('lisp_remote_mapping', 'lisp remote-mapping', 'lisp_add_del_remote_mapping')
  58 SimpleMapping('lisp_pitr', 'lisp pitr ls', 'lisp_pitr_set_locator_set locator-set')
  59 SimpleMapping('lisp_adjacency', 'lisp adjacency', 'lisp_add_del_adjacency')
  60 SimpleMapping('lisp_map_request_mode', 'lisp map-request mode', 'lisp_map_request_mode')
  61 SimpleMapping('set_if_ip', 'set int ip address', 'sw_interface_add_del_address')
  62 SimpleMapping('lisp_rloc_probe_state', 'lisp rloc-probe', 'lisp_rloc_probe_enable_disable')
  63 SimpleMapping('lisp_map_register_state', 'lisp map-register', 'lisp_map_register_enable_disable')
  64 SimpleMapping('lisp_stats', 'one statistics', 'one_stats_enable_disable')
  65
  66 CustomMapping('lisp_eid_map_bd',
  67               'lisp eid-table map vni {0} bd {1}',
  68               'lisp_eid_table_add_del_map vni {0} bd_index {1}')
  69 CustomMapping('lisp_eid_map_vrf',
  70               'lisp eid-table map vni {0} vrf {1}',
  71               'lisp_eid_table_add_del_map vni {0} vrf {1}')
  72 CustomMapping('set_if_l2_bridge', 'set interface l2 bridge {0} {1}',
  73               'sw_interface_set_l2_bridge {0} bd_id {1}')
  74 CustomMapping('set_if_ip_table', 'set interface ip table {0} {1}',
  75               'sw_interface_set_table {0} vrf {1}')
  76 CustomMapping('lisp_locator_set_with_locator',
  77               'lisp locator-set add {0} iface {1} p {2} w {3}',
  78               'lisp_add_del_locator_set locator-set {0} iface {1} p {2} w {3}')
  79 CustomMapping('create_host_iface',
  80     'create host-interface name {0}\n'
  81     'set int state host-{0} up\n'
  82     'set int ip address host-{0} {1}',
  83
  84     'af_packet_create name {0}\n'
  85     'sw_interface_set_flags host-{0} admin-up link-up\n'
  86     'sw_interface_add_del_address host-{0} {1}')
  87
  88 CustomMapping('create_host_iface_vrf',
  89     'create host-interface name {0}\n'
  90     'set int state host-{0} up\n'
  91     'set interface ip table host-{0} {2}\n'
  92     'set int ip address host-{0} {1}',
  93
  94     'af_packet_create name {0}\n'
  95     'sw_interface_set_flags host-{0} admin-up link-up\n'
  96     'sw_interface_set_table host-{0} vrf {2}\n'
  97     'sw_interface_add_del_address host-{0} {1}')
  98
  99 CustomMapping('create_host_iface_vrf_v6',
 100     'create host-interface name {0}\n'
 101     'set int state host-{0} up\n'
 102     'set interface ip6 table host-{0} {2}\n'
 103     'set int ip address host-{0} {1}',
 104
 105     'af_packet_create name {0}\n'
 106     'sw_interface_set_flags host-{0} admin-up link-up\n'
 107     'sw_interface_set_table host-{0} vrf {2} ipv6\n'
 108     'sw_interface_add_del_address host-{0} {1}')
 109
 110 RepeatableLocators('lisp_ls_multiple_locs',
 111                    'lisp locator-set add',
 112                    'lisp_add_del_locator_set locator-set')
 113