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
  65 CustomMapping('lisp_eid_map_bd',
  66               'lisp eid-table map vni {0} bd {1}',
  67               'lisp_eid_table_add_del_map vni {0} bd_index {1}')
  68 CustomMapping('lisp_eid_map_vrf',
  69               'lisp eid-table map vni {0} vrf {1}',
  70               'lisp_eid_table_add_del_map vni {0} vrf {1}')
  71 CustomMapping('set_if_l2_bridge', 'set interface l2 bridge {0} {1}',
  72               'sw_interface_set_l2_bridge {0} bd_id {1}')
  73 CustomMapping('set_if_ip_table', 'set interface ip table {0} {1}',
  74               'sw_interface_set_table {0} vrf {1}')
  75 CustomMapping('lisp_locator_set_with_locator',
  76               'lisp locator-set add {0} iface {1} p {2} w {3}',
  77               'lisp_add_del_locator_set locator-set {0} iface {1} p {2} w {3}')
  78 CustomMapping('create_host_iface',
  79     'create host-interface name {0}\n'
  80     'set int state host-{0} up\n'
  81     'set int ip address host-{0} {1}',
  82
  83     'af_packet_create name {0}\n'
  84     'sw_interface_set_flags host-{0} admin-up link-up\n'
  85     'sw_interface_add_del_address host-{0} {1}')
  86
  87 CustomMapping('create_host_iface_vrf',
  88     'create host-interface name {0}\n'
  89     'set int state host-{0} up\n'
  90     'set interface ip table host-{0} {2}\n'
  91     'set int ip address host-{0} {1}',
  92
  93     'af_packet_create name {0}\n'
  94     'sw_interface_set_flags host-{0} admin-up link-up\n'
  95     'sw_interface_set_table host-{0} vrf {2}\n'
  96     'sw_interface_add_del_address host-{0} {1}')
  97
  98 CustomMapping('create_host_iface_vrf_v6',
  99     'create host-interface name {0}\n'
 100     'set int state host-{0} up\n'
 101     'set interface ip6 table host-{0} {2}\n'
 102     'set int ip address host-{0} {1}',
 103
 104     'af_packet_create name {0}\n'
 105     'sw_interface_set_flags host-{0} admin-up link-up\n'
 106     'sw_interface_set_table host-{0} vrf {2} ipv6\n'
 107     'sw_interface_add_del_address host-{0} {1}')
 108
 109 RepeatableLocators('lisp_ls_multiple_locs',
 110                    'lisp locator-set add',
 111                    'lisp_add_del_locator_set locator-set')
 112