4 from vpp_papi.vpp_serializer import VPPType, VPPEnumType
5 from vpp_papi.vpp_serializer import VPPUnionType, VPPMessage
6 from socket import inet_pton, AF_INET, AF_INET6
11 class TestAddType(unittest.TestCase):
14 un = VPPUnionType('test_union',
18 b = un.pack({'is_int': 0x12345678})
19 nt, size = un.unpack(b)
20 self.assertEqual(len(b), size)
21 self.assertEqual(nt.is_bool, 0x12)
22 self.assertEqual(nt.is_int, 0x12345678)
24 def test_address(self):
25 af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0],
28 ip4 = VPPType('vl_api_ip4_address_t', [['u8', 'address', 4]])
29 ip6 = VPPType('vl_api_ip6_address_t', [['u8', 'address', 16]])
30 VPPUnionType('vl_api_address_union_t',
31 [["vl_api_ip4_address_t", "ip4"],
32 ["vl_api_ip6_address_t", "ip6"]])
34 address = VPPType('vl_api_address_t',
35 [['vl_api_address_family_t', 'af'],
36 ['vl_api_address_union_t', 'un']])
38 va_address_list = VPPType('list_addresses',
40 ['vl_api_address_t', 'addresses',
43 message_with_va_address_list = VPPType('msg_with_vla',
48 b = ip4.pack({'address': inet_pton(AF_INET, '1.1.1.1')})
49 self.assertEqual(len(b), 4)
50 nt, size = ip4.unpack(b)
51 self.assertEqual(nt.address, inet_pton(AF_INET, '1.1.1.1'))
53 b = ip6.pack({'address': inet_pton(AF_INET6, '1::1')})
54 self.assertEqual(len(b), 16)
56 b = address.pack({'af': af.ADDRESS_IP4,
59 {'address': inet_pton(AF_INET, '2.2.2.2')}}})
60 self.assertEqual(len(b), 20)
62 nt, size = address.unpack(b)
63 self.assertEqual(nt.af, af.ADDRESS_IP4)
64 self.assertEqual(nt.un.ip4.address,
65 inet_pton(AF_INET, '2.2.2.2'))
66 self.assertEqual(nt.un.ip6.address,
67 inet_pton(AF_INET6, '0202:0202::'))
72 address_list.append({'af': af.ADDRESS_IP4,
75 {'address': inet_pton(AF_INET, '2.2.2.2')}}})
76 b = va_address_list.pack({'count': len(address_list),
77 'addresses': address_list})
78 self.assertEqual(len(b), 81)
80 nt, size = va_address_list.unpack(b)
81 self.assertEqual(nt.addresses[0].un.ip4.address,
82 inet_pton(AF_INET, '2.2.2.2'))
84 b = message_with_va_address_list.pack({'vla_address':
85 {'count': len(address_list),
86 'addresses': address_list},
88 self.assertEqual(len(b), 82)
89 nt, size = message_with_va_address_list.unpack(b)
90 self.assertEqual(nt.is_cool, 100)
92 def test_arrays(self):
95 # 2. Fixed list of variable length sub type
96 # 3. Variable length type
98 s = VPPType('str', [['u32', 'length'],
99 ['u8', 'string', 0, 'length']])
101 ip4 = VPPType('ip4_address', [['u8', 'address', 4]])
102 listip4 = VPPType('list_ip4_t', [['ip4_address', 'addresses', 4]])
103 valistip4 = VPPType('list_ip4_t',
105 ['ip4_address', 'addresses', 0, 'count']])
107 valistip4_legacy = VPPType('list_ip4_t',
109 ['ip4_address', 'addresses', 0]])
113 addresses.append({'address': inet_pton(AF_INET, '2.2.2.2')})
114 b = listip4.pack({'addresses': addresses})
115 self.assertEqual(len(b), 16)
116 nt, size = listip4.unpack(b)
117 self.assertEqual(nt.addresses[0].address,
118 inet_pton(AF_INET, '2.2.2.2'))
120 b = valistip4.pack({'count': len(addresses), 'addresses': addresses})
121 self.assertEqual(len(b), 17)
123 nt, size = valistip4.unpack(b)
124 self.assertEqual(nt.count, 4)
125 self.assertEqual(nt.addresses[0].address,
126 inet_pton(AF_INET, '2.2.2.2'))
128 b = valistip4_legacy.pack({'foo': 1, 'addresses': addresses})
129 self.assertEqual(len(b), 17)
130 nt, size = valistip4_legacy.unpack(b)
131 self.assertEqual(len(nt.addresses), 4)
132 self.assertEqual(nt.addresses[0].address,
133 inet_pton(AF_INET, '2.2.2.2'))
135 string = 'foobar foobar'
136 b = s.pack({'length': len(string), 'string': string})
137 nt, size = s.unpack(b)
138 self.assertEqual(len(b), size)
140 def test_string(self):
141 s = VPPType('str', [['u32', 'length'],
142 ['u8', 'string', 0, 'length']])
145 b = s.pack({'length': len(string), 'string': string})
146 nt, size = s.unpack(b)
147 self.assertEqual(len(b), size)
149 def test_message(self):
150 foo = VPPMessage('foo', [['u16', '_vl_msg_id'],
151 ['u8', 'client_index'],
153 {"crc": "0x559b9f3c"}])
154 b = foo.pack({'_vl_msg_id': 1, 'client_index': 5,
156 nt, size = foo.unpack(b)
157 self.assertEqual(len(b), size)
158 self.assertEqual(nt.something, 200)
162 fib_mpls_label = VPPType('vl_api_fib_mpls_label_t',
163 [['u8', 'is_uniform'],
168 label_stack = {'is_uniform': 0,
173 b = fib_mpls_label.pack(label_stack)
174 self.assertEqual(len(b), 7)
176 fib_path = VPPType('vl_api_fib_path_t',
177 [['u32', 'sw_if_index'],
180 ['u8', 'preference'],
183 ['u8', 'is_udp_encap'],
184 ['u8', 'is_unreach'],
185 ['u8', 'is_prohibit'],
186 ['u8', 'is_resolve_host'],
187 ['u8', 'is_resolve_attached'],
189 ['u8', 'is_source_lookup'],
191 ['u8', 'next_hop', 16],
192 ['u32', 'next_hop_id'],
194 ['u32', 'via_label'],
196 ['vl_api_fib_mpls_label_t', 'label_stack', 16]])
197 label_stack_list = []
199 label_stack_list.append(label_stack)
201 paths = {'is_udp_encap': 0,
202 'next_hop': b'\x10\x02\x02\xac',
206 'next_hop_id': 4294967295,
207 'label_stack': label_stack_list,
209 'sw_if_index': 4294967295,
212 b = fib_path.pack(paths)
213 self.assertEqual(len(b), (7*16) + 49)
215 abf_policy = VPPType('vl_api_abf_policy_t',
216 [['u32', 'policy_id'],
217 ['u32', 'acl_index'],
219 ['vl_api_fib_path_t', 'paths', 0, 'n_paths']])
227 b = abf_policy.pack(policy)
228 self.assertEqual(len(b), (7*16) + 49 + 9)
230 abf_policy_add_del = VPPMessage('abf_policy_add_del',
231 [['u16', '_vl_msg_id'],
232 ['u32', 'client_index'],
235 ['vl_api_abf_policy_t', 'policy']])
237 b = abf_policy_add_del.pack({'is_add': 1,
242 nt, size = abf_policy_add_del.unpack(b)
243 self.assertEqual(nt.policy.paths[0].next_hop,
244 b'\x10\x02\x02\xac\x00\x00\x00\x00'
245 b'\x00\x00\x00\x00\x00\x00\x00\x00')
249 bier_table_id = VPPType('vl_api_bier_table_id_t',
251 ['u8', 'bt_sub_domain'],
252 ['u8', 'bt_hdr_len_id']])
254 bier_imp_add = VPPMessage('bier_imp_add',
255 [['u32', 'client_index'],
257 ['vl_api_bier_table_id_t', 'bi_tbl_id'],
259 ['u8', 'bi_n_bytes'],
260 ['u8', 'bi_bytes', 0, 'bi_n_bytes']])
262 table_id = {'bt_set': 0,
268 b = bier_imp_add.pack({'bi_tbl_id': table_id,
269 'bi_n_bytes': len(bibytes),
270 'bi_bytes': bibytes})
272 self.assertEqual(len(b), 20)
275 if __name__ == '__main__':