9bf2c1592fa9ff3286576bb745d71ecead315423
[one.git] / 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', 'one', 'one_enable_disable')
54 SimpleMapping('lisp_map_resolver', 'one map-resolver', 'one_add_del_map_resolver')
55 SimpleMapping('lisp_map_server', 'one map-server', 'one_add_del_map_server')
56 SimpleMapping('lisp_local_eid', 'one eid-table', 'one_add_del_local_eid')
57 SimpleMapping('lisp_remote_mapping', 'one remote-mapping', 'one_add_del_remote_mapping')
58 SimpleMapping('lisp_pitr', 'one pitr ls', 'one_pitr_set_locator_set locator-set')
59 SimpleMapping('lisp_adjacency', 'one adjacency', 'one_add_del_adjacency')
60 SimpleMapping('lisp_map_request_mode', 'one map-request mode', 'one_map_request_mode')
61 SimpleMapping('set_if_ip', 'set int ip address', 'sw_interface_add_del_address')
62 SimpleMapping('lisp_rloc_probe_state', 'one rloc-probe', 'one_rloc_probe_enable_disable')
63 SimpleMapping('lisp_map_register_state', 'one map-register', 'one_map_register_enable_disable')
64 SimpleMapping('lisp_stats', 'one statistics', 'one_stats_enable_disable')
65 SimpleMapping('lisp_l2_arp', 'one l2 arp', 'one_add_del_l2_arp_entry')
66 SimpleMapping('lisp_nsh_set_ls', 'one nsh-mapping', 'one_nsh_set_locator_set')
67
68 SimpleMapping('ipsec_sad_add_del_entry', 'TODO', 'ipsec_sad_add_del_entry')
69 SimpleMapping('ipsec_spd_add_del', 'todo', 'ipsec_spd_add_del')
70 SimpleMapping('ipsec_interface_add_del_spd' , 'todo', 'ipsec_interface_add_del_spd')
71 SimpleMapping('ipsec_spd_add_del_entry', 'todo', 'ipsec_spd_add_del_entry')
72
73 CustomMapping('lisp_eid_map_bd',
74               'one eid-table map vni {0} bd {1}',
75               'one_eid_table_add_del_map vni {0} bd_index {1}')
76 CustomMapping('lisp_eid_map_vrf',
77               'one eid-table map vni {0} vrf {1}',
78               'one_eid_table_add_del_map vni {0} vrf {1}')
79 CustomMapping('set_if_l2_bridge', 'set interface l2 bridge {0} {1}',
80               'sw_interface_set_l2_bridge {0} bd_id {1}')
81 CustomMapping('set_if_ip_table', 'set interface ip table {0} {1}',
82               'sw_interface_set_table {0} vrf {1}')
83 CustomMapping('lisp_locator_set_with_locator',
84               'one locator-set add {0} iface {1} p {2} w {3}',
85               'one_add_del_locator_set locator-set {0} iface {1} p {2} w {3}')
86 CustomMapping('create_host_iface',
87     'create host-interface name {0}\n'
88     'set int state host-{0} up\n'
89     'set int ip address host-{0} {1}',
90
91     'af_packet_create name {0}\n'
92     'sw_interface_set_flags host-{0} admin-up link-up\n'
93     'sw_interface_add_del_address host-{0} {1}')
94
95 CustomMapping('create_host_iface_vrf',
96     'create host-interface name {0}\n'
97     'set int state host-{0} up\n'
98     'set interface ip table host-{0} {2}\n'
99     'set int ip address host-{0} {1}',
100
101     'af_packet_create name {0}\n'
102     'sw_interface_set_flags host-{0} admin-up link-up\n'
103     'sw_interface_set_table host-{0} vrf {2}\n'
104     'sw_interface_add_del_address host-{0} {1}')
105
106 CustomMapping('create_host_iface_vrf_v6',
107     'create host-interface name {0}\n'
108     'set int state host-{0} up\n'
109     'set interface ip6 table host-{0} {2}\n'
110     'set int ip address host-{0} {1}',
111
112     'af_packet_create name {0}\n'
113     'sw_interface_set_flags host-{0} admin-up link-up\n'
114     'sw_interface_set_table host-{0} vrf {2} ipv6\n'
115     'sw_interface_add_del_address host-{0} {1}')
116
117 RepeatableLocators('lisp_ls_multiple_locs',
118                    'one locator-set add',
119                    'one_add_del_locator_set locator-set')
120