def test_string(self): fixed_string = VPPType('fixed_string', [['string', 'name', 16]]) b = fixed_string.pack({'name': 'foobar'}) self.assertEqual(len(b), 16) # Ensure string is nul terminated self.assertEqual(b.decode('ascii')[6], '\x00') nt, size = fixed_string.unpack(b) self.assertEqual(size, 16) self.assertEqual(nt.name, 'foobar') # Empty string b = fixed_string.pack({'name': ''}) self.assertEqual(len(b), 16) nt, size = fixed_string.unpack(b) self.assertEqual(size, 16) self.assertEqual(nt.name, '') # String too long with self.assertRaises(VPPSerializerValueError): b = fixed_string.pack({'name': 'foobarfoobar1234'}) variable_string = VPPType('variable_string', [['string', 'name', 0]]) b = variable_string.pack({'name': 'foobar'}) self.assertEqual(len(b), 4 + len('foobar')) nt, size = variable_string.unpack(b) self.assertEqual(size, 4 + len('foobar')) self.assertEqual(nt.name, 'foobar') self.assertEqual(len(nt.name), len('foobar'))
def test_string(self): fixed_string = VPPType("fixed_string", [["string", "name", 16]]) b = fixed_string.pack({"name": "foobar"}) self.assertEqual(len(b), 16) # Ensure string is nul terminated self.assertEqual(b.decode("ascii")[6], "\x00") nt, size = fixed_string.unpack(b) self.assertEqual(size, 16) self.assertEqual(nt.name, "foobar") # Empty string b = fixed_string.pack({"name": ""}) self.assertEqual(len(b), 16) nt, size = fixed_string.unpack(b) self.assertEqual(size, 16) self.assertEqual(nt.name, "") # String too long with self.assertRaises(VPPSerializerValueError): b = fixed_string.pack({"name": "foobarfoobar1234"}) variable_string = VPPType("variable_string", [["string", "name", 0]]) b = variable_string.pack({"name": "foobar"}) self.assertEqual(len(b), 4 + len("foobar")) nt, size = variable_string.unpack(b) self.assertEqual(size, 4 + len("foobar")) self.assertEqual(nt.name, "foobar") self.assertEqual(len(nt.name), len("foobar"))
def test_defaults(self): default_type = VPPType("default_type_t", [["u16", "mtu", { "default": 1500, "limit": 0 }]]) without_default_type = VPPType("without_default_type_t", [["u16", "mtu"]]) b = default_type.pack({}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 1500) # distinguish between parameter 0 and parameter not passed b = default_type.pack({"mtu": 0}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 0) # Ensure that basetypes does not inherit default b = without_default_type.pack({}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 0) # default enum type VPPEnumType( "vl_api_enum_t", [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }], ) default_with_enum = VPPType( "default_enum_type_t", [["u16", "mtu"], ["vl_api_enum_t", "e", { "default": 1 }]], ) b = default_with_enum.pack({}) self.assertEqual(len(b), 6) nt, size = default_with_enum.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.e, 1)
def test_string(self): s = VPPType("str", [["u32", "length"], ["u8", "string", 0, "length"]]) string = "" b = s.pack({"length": len(string), "string": string.encode("utf-8")}) nt, size = s.unpack(b) self.assertEqual(len(b), size)
def test_string(self): s = VPPType('str', [['u32', 'length'], ['u8', 'string', 0, 'length']]) string = '' b = s.pack({'length': len(string), 'string': string}) nt, size = s.unpack(b) self.assertEqual(len(b), size)
def test_arrays(self): # Test cases # 1. Fixed list # 2. Fixed list of variable length sub type # 3. Variable length type # s = VPPType('str', [['u32', 'length'], ['u8', 'string', 0, 'length']]) ip4 = VPPType('ip4_address', [['u8', 'address', 4]]) listip4 = VPPType('list_ip4_t', [['ip4_address', 'addresses', 4]]) valistip4 = VPPType('list_ip4_t', [['u8', 'count'], ['ip4_address', 'addresses', 0, 'count']]) valistip4_legacy = VPPType('list_ip4_t', [['u8', 'foo'], ['ip4_address', 'addresses', 0]]) addresses = [] for i in range(4): addresses.append({'address': inet_pton(AF_INET, '2.2.2.2')}) b = listip4.pack({'addresses': addresses}) self.assertEqual(len(b), 16) nt, size = listip4.unpack(b) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, '2.2.2.2')) b = valistip4.pack({'count': len(addresses), 'addresses': addresses}) self.assertEqual(len(b), 17) nt, size = valistip4.unpack(b) self.assertEqual(nt.count, 4) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, '2.2.2.2')) b = valistip4_legacy.pack({'foo': 1, 'addresses': addresses}) self.assertEqual(len(b), 17) nt, size = valistip4_legacy.unpack(b) self.assertEqual(len(nt.addresses), 4) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, '2.2.2.2')) string = 'foobar foobar' b = s.pack({'length': len(string), 'string': string.encode('utf-8')}) nt, size = s.unpack(b) self.assertEqual(len(b), size)
def test_string(self): s = VPPType('str', [['u32', 'length'], ['u8', 'string', 0, 'length']]) string = '' b = s.pack({'length': len(string), 'string': string.encode('utf-8')}) nt, size = s.unpack(b) self.assertEqual(len(b), size)
def test_defaults(self): default_type = VPPType('default_type_t', [['u16', 'mtu', { 'default': 1500, 'limit': 0 }]]) without_default_type = VPPType('without_default_type_t', [['u16', 'mtu']]) b = default_type.pack({}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 1500) # distinguish between parameter 0 and parameter not passed b = default_type.pack({'mtu': 0}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 0) # Ensure that basetypes does not inherit default b = without_default_type.pack({}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 0) # default enum type VPPEnumType('vl_api_enum_t', [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }]) default_with_enum = VPPType( 'default_enum_type_t', [['u16', 'mtu'], ['vl_api_enum_t', 'e', { 'default': 1 }]]) b = default_with_enum.pack({}) self.assertEqual(len(b), 6) nt, size = default_with_enum.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.e, 1)
def test_arrays(self): # Test cases # 1. Fixed list # 2. Fixed list of variable length sub type # 3. Variable length type # s = VPPType("str", [["u32", "length"], ["u8", "string", 0, "length"]]) ip4 = VPPType("ip4_address", [["u8", "address", 4]]) listip4 = VPPType("list_ip4_t", [["ip4_address", "addresses", 4]]) valistip4 = VPPType( "list_ip4_t", [["u8", "count"], ["ip4_address", "addresses", 0, "count"]]) valistip4_legacy = VPPType( "list_ip4_t", [["u8", "foo"], ["ip4_address", "addresses", 0]]) addresses = [] for i in range(4): addresses.append({"address": inet_pton(AF_INET, "2.2.2.2")}) b = listip4.pack({"addresses": addresses}) self.assertEqual(len(b), 16) nt, size = listip4.unpack(b) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, "2.2.2.2")) b = valistip4.pack({"count": len(addresses), "addresses": addresses}) self.assertEqual(len(b), 17) nt, size = valistip4.unpack(b) self.assertEqual(nt.count, 4) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, "2.2.2.2")) b = valistip4_legacy.pack({"foo": 1, "addresses": addresses}) self.assertEqual(len(b), 17) nt, size = valistip4_legacy.unpack(b) self.assertEqual(len(nt.addresses), 4) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, "2.2.2.2")) string = "foobar foobar" b = s.pack({"length": len(string), "string": string.encode("utf-8")}) nt, size = s.unpack(b) self.assertEqual(len(b), size)
def test_defaults(self): default_type = VPPType('default_type_t', [['u16', 'mtu', {'default': 1500, 'limit': 0}]]) b = default_type.pack({}) self.assertEqual(len(b), 2) nt, size = default_type.unpack(b) self.assertEqual(len(b), size) self.assertEqual(nt.mtu, 1500)
def test_arrays(self): # Test cases # 1. Fixed list # 2. Fixed list of variable length sub type # 3. Variable length type # ip4 = VPPType('ip4_address', [['u8', 'address', 4]]) listip4 = VPPType('list_ip4_t', [['ip4_address', 'addresses', 4]]) valistip4 = VPPType( 'list_ip4_t', [['u8', 'count'], ['ip4_address', 'addresses', 0, 'count']]) valistip4_legacy = VPPType( 'list_ip4_t', [['u8', 'foo'], ['ip4_address', 'addresses', 0]]) addresses = [] for i in range(4): addresses.append({'address': inet_pton(AF_INET, '2.2.2.2')}) b = listip4.pack({'addresses': addresses}) self.assertEqual(len(b), 16) nt = listip4.unpack(b) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, '2.2.2.2')) b = valistip4.pack({'count': len(addresses), 'addresses': addresses}) self.assertEqual(len(b), 17) nt = valistip4.unpack(b) self.assertEqual(nt.count, 4) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, '2.2.2.2')) b = valistip4_legacy.pack({'foo': 1, 'addresses': addresses}) self.assertEqual(len(b), 17) nt = valistip4_legacy.unpack(b) self.assertEqual(len(nt.addresses), 4) self.assertEqual(nt.addresses[0].address, inet_pton(AF_INET, '2.2.2.2'))
def test_lisp(self): VPPEnumType('vl_api_eid_type_t', [["EID_TYPE_API_PREFIX", 0], ["EID_TYPE_API_MAC", 1], ["EID_TYPE_API_NSH", 2], {"enumtype": "u32"}]) VPPTypeAlias('vl_api_mac_address_t', {'type': 'u8', 'length': 6}) VPPType('vl_api_nsh_t', [["u32", "spi"], ["u8", "si"]]) VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], {"enumtype": "u32"}]) VPPTypeAlias('vl_api_ip4_address_t', {'type': 'u8', 'length': 4}) VPPTypeAlias('vl_api_ip6_address_t', {'type': 'u8', 'length': 16}) VPPUnionType('vl_api_address_union_t', [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]]) VPPType('vl_api_address_t', [['vl_api_address_family_t', 'af'], ['vl_api_address_union_t', 'un']]) VPPType('vl_api_prefix_t', [['vl_api_address_t', 'address'], ['u8', 'len']]) VPPUnionType('vl_api_eid_address_t', [["vl_api_prefix_t", "prefix"], ["vl_api_mac_address_t", "mac"], ["vl_api_nsh_t", "nsh"]]) eid = VPPType('vl_api_eid_t', [["vl_api_eid_type_t", "type"], ["vl_api_eid_address_t", "address"]]) b = eid.pack({'type':1, 'address': { 'mac': MACAddress('aa:bb:cc:dd:ee:ff')}}) self.assertEqual(len(b), 25) nt, size = eid.unpack(b) self.assertEqual(str(nt.address.mac), 'aa:bb:cc:dd:ee:ff') self.assertIsNone(nt.address.prefix)
def test_address(self): af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }]) ip4 = VPPType('vl_api_ip4_address_t', [['u8', 'address', 4]]) ip6 = VPPType('vl_api_ip6_address_t', [['u8', 'address', 16]]) VPPUnionType( 'vl_api_address_union_t', [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]]) address = VPPType('address', [['vl_api_address_family_t', 'af'], ['vl_api_address_union_t', 'un']]) b = ip4.pack({'address': inet_pton(AF_INET, '1.1.1.1')}) self.assertEqual(len(b), 4) nt = ip4.unpack(b) self.assertEqual(nt.address, inet_pton(AF_INET, '1.1.1.1')) b = ip6.pack({'address': inet_pton(AF_INET6, '1::1')}) self.assertEqual(len(b), 16) b = address.pack({ 'af': af.ADDRESS_IP4, 'un': { 'ip4': { 'address': inet_pton(AF_INET, '2.2.2.2') } } }) self.assertEqual(len(b), 20) nt = address.unpack(b) self.assertEqual(nt.af, af.ADDRESS_IP4) self.assertEqual(nt.un.ip4.address, inet_pton(AF_INET, '2.2.2.2')) self.assertEqual(nt.un.ip6.address, inet_pton(AF_INET6, '0202:0202::'))
def test_address(self): af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }]) ip4 = VPPType('vl_api_ip4_address_t', [['u8', 'address', 4]]) ip6 = VPPType('vl_api_ip6_address_t', [['u8', 'address', 16]]) VPPUnionType( 'vl_api_address_union_t', [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]]) address = VPPType('vl_api_address_t', [['vl_api_address_family_t', 'af'], ['vl_api_address_union_t', 'un']]) va_address_list = VPPType( 'list_addresses', [['u8', 'count'], ['vl_api_address_t', 'addresses', 0, 'count']]) message_with_va_address_list = VPPType( 'msg_with_vla', [['list_addresses', 'vla_address'], ['u8', 'is_cool']]) b = ip4.pack({'address': inet_pton(AF_INET, '1.1.1.1')}) self.assertEqual(len(b), 4) nt, size = ip4.unpack(b) self.assertEqual(nt.address, inet_pton(AF_INET, '1.1.1.1')) b = ip6.pack({'address': inet_pton(AF_INET6, '1::1')}) self.assertEqual(len(b), 16) b = address.pack({ 'af': af.ADDRESS_IP4, 'un': { 'ip4': { 'address': inet_pton(AF_INET, '2.2.2.2') } } }) self.assertEqual(len(b), 20) nt, size = address.unpack(b) self.assertEqual(nt.af, af.ADDRESS_IP4) self.assertEqual(nt.un.ip4.address, inet_pton(AF_INET, '2.2.2.2')) self.assertEqual(nt.un.ip6.address, inet_pton(AF_INET6, '0202:0202::')) # List of addresses address_list = [] for i in range(4): address_list.append({ 'af': af.ADDRESS_IP4, 'un': { 'ip4': { 'address': inet_pton(AF_INET, '2.2.2.2') } } }) b = va_address_list.pack({ 'count': len(address_list), 'addresses': address_list }) self.assertEqual(len(b), 81) nt, size = va_address_list.unpack(b) self.assertEqual(nt.addresses[0].un.ip4.address, inet_pton(AF_INET, '2.2.2.2')) b = message_with_va_address_list.pack({ 'vla_address': { 'count': len(address_list), 'addresses': address_list }, 'is_cool': 100 }) self.assertEqual(len(b), 82) nt, size = message_with_va_address_list.unpack(b) self.assertEqual(nt.is_cool, 100)
def test_address(self): af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], {"enumtype": "u32"}]) ip4 = VPPTypeAlias('vl_api_ip4_address_t', {'type': 'u8', 'length': 4}) ip6 = VPPTypeAlias('vl_api_ip6_address_t', {'type': 'u8', 'length': 16}) VPPUnionType('vl_api_address_union_t', [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]]) address = VPPType('vl_api_address_t', [['vl_api_address_family_t', 'af'], ['vl_api_address_union_t', 'un']]) va_address_list = VPPType('list_addresses', [['u8', 'count'], ['vl_api_address_t', 'addresses', 0, 'count']]) message_with_va_address_list = VPPType('msg_with_vla', [['list_addresses', 'vla_address'], ['u8', 'is_cool']]) b = ip4.pack(inet_pton(AF_INET, '1.1.1.1')) self.assertEqual(len(b), 4) nt, size = ip4.unpack(b) self.assertEqual(str(nt), '1.1.1.1') b = ip6.pack(inet_pton(AF_INET6, '1::1')) self.assertEqual(len(b), 16) b = address.pack({'af': af.ADDRESS_IP4, 'un': {'ip4': inet_pton(AF_INET, '2.2.2.2')}}) self.assertEqual(len(b), 20) nt, size = address.unpack(b) self.assertEqual(str(nt), '2.2.2.2') # List of addresses address_list = [] for i in range(4): address_list.append({'af': af.ADDRESS_IP4, 'un': {'ip4': inet_pton(AF_INET, '2.2.2.2')}}) b = va_address_list.pack({'count': len(address_list), 'addresses': address_list}) self.assertEqual(len(b), 81) nt, size = va_address_list.unpack(b) self.assertEqual(str(nt.addresses[0]), '2.2.2.2') b = message_with_va_address_list.pack({'vla_address': {'count': len(address_list), 'addresses': address_list}, 'is_cool': 100}) self.assertEqual(len(b), 82) nt, size = message_with_va_address_list.unpack(b) self.assertEqual(nt.is_cool, 100)
def test_address_with_prefix(self): af = VPPEnumType('vl_api_address_family_t', [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }]) ip4 = VPPTypeAlias('vl_api_ip4_address_t', {'type': 'u8', 'length': 4}) ip6 = VPPTypeAlias('vl_api_ip6_address_t', { 'type': 'u8', 'length': 16 }) VPPUnionType( 'vl_api_address_union_t', [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]]) address = VPPType('vl_api_address_t', [['vl_api_address_family_t', 'af'], ['vl_api_address_union_t', 'un']]) prefix = VPPType('vl_api_prefix_t', [['vl_api_address_t', 'address'], ['u8', 'len']]) prefix4 = VPPType('vl_api_ip4_prefix_t', [['vl_api_ip4_address_t', 'address'], ['u8', 'len']]) prefix6 = VPPType('vl_api_ip6_prefix_t', [['vl_api_ip6_address_t', 'address'], ['u8', 'len']]) address_with_prefix = VPPTypeAlias('vl_api_address_with_prefix_t', {'type': 'vl_api_prefix_t'}) address4_with_prefix = VPPTypeAlias('vl_api_ip4_address_with_prefix_t', {'type': 'vl_api_ip4_prefix_t'}) address6_with_prefix = VPPTypeAlias('vl_api_ip6_address_with_prefix_t', {'type': 'vl_api_ip6_prefix_t'}) awp_type = VPPType('foobar_t', [['vl_api_address_with_prefix_t', 'address']]) # address with prefix b = address_with_prefix.pack(IPv4Interface('2.2.2.2/24')) self.assertEqual(len(b), 21) nt, size = address_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Interface)) self.assertEqual(str(nt), '2.2.2.2/24') b = address_with_prefix.pack(IPv6Interface('2::2/64')) self.assertEqual(len(b), 21) nt, size = address_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv6Interface)) self.assertEqual(str(nt), '2::2/64') b = address_with_prefix.pack(IPv4Network('2.2.2.2/24', strict=False)) self.assertEqual(len(b), 21) nt, size = address_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Interface)) self.assertEqual(str(nt), '2.2.2.0/24') b = address4_with_prefix.pack('2.2.2.2/24') self.assertEqual(len(b), 5) nt, size = address4_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Interface)) self.assertEqual(str(nt), '2.2.2.2/24') b = address4_with_prefix.pack(IPv4Interface('2.2.2.2/24')) self.assertEqual(len(b), 5) b = address6_with_prefix.pack('2::2/64') self.assertEqual(len(b), 17) nt, size = address6_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv6Interface)) self.assertEqual(str(nt), '2::2/64') b = address6_with_prefix.pack(IPv6Interface('2::2/64')) self.assertEqual(len(b), 17) b = prefix.pack('192.168.10.0/24') self.assertEqual(len(b), 21) nt, size = prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Network)) self.assertEqual(str(nt), '192.168.10.0/24') b = awp_type.pack({'address': '1.2.3.4/24'}) self.assertEqual(len(b), 21) nt, size = awp_type.unpack(b) self.assertTrue(isinstance(nt.address, IPv4Interface)) self.assertEqual(str(nt.address), '1.2.3.4/24') b = awp_type.pack({'address': IPv4Interface('1.2.3.4/24')}) self.assertEqual(len(b), 21) nt, size = awp_type.unpack(b) self.assertTrue(isinstance(nt.address, IPv4Interface)) self.assertEqual(str(nt.address), '1.2.3.4/24')
def test_lisp(self): VPPEnumType( "vl_api_eid_type_t", [ ["EID_TYPE_API_PREFIX", 0], ["EID_TYPE_API_MAC", 1], ["EID_TYPE_API_NSH", 2], { "enumtype": "u32" }, ], ) VPPTypeAlias("vl_api_mac_address_t", {"type": "u8", "length": 6}) VPPType("vl_api_nsh_t", [["u32", "spi"], ["u8", "si"]]) VPPEnumType( "vl_api_address_family_t", [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }], ) VPPTypeAlias("vl_api_ip4_address_t", {"type": "u8", "length": 4}) VPPTypeAlias("vl_api_ip6_address_t", {"type": "u8", "length": 16}) VPPUnionType( "vl_api_address_union_t", [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]], ) VPPType( "vl_api_address_t", [["vl_api_address_family_t", "af"], ["vl_api_address_union_t", "un"]], ) VPPType("vl_api_prefix_t", [["vl_api_address_t", "address"], ["u8", "len"]]) VPPUnionType( "vl_api_eid_address_t", [ ["vl_api_prefix_t", "prefix"], ["vl_api_mac_address_t", "mac"], ["vl_api_nsh_t", "nsh"], ], ) eid = VPPType( "vl_api_eid_t", [["vl_api_eid_type_t", "type"], ["vl_api_eid_address_t", "address"]], ) b = eid.pack({ "type": 1, "address": { "mac": MACAddress("aa:bb:cc:dd:ee:ff") } }) self.assertEqual(len(b), 25) nt, size = eid.unpack(b) self.assertEqual(str(nt.address.mac), "aa:bb:cc:dd:ee:ff") self.assertIsNone(nt.address.prefix)
def test_address(self): af = VPPEnumType( "vl_api_address_family_t", [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }], ) aff = VPPEnumFlagType( "vl_api_address_family_flag_t", [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }], ) ip4 = VPPTypeAlias("vl_api_ip4_address_t", {"type": "u8", "length": 4}) ip6 = VPPTypeAlias("vl_api_ip6_address_t", { "type": "u8", "length": 16 }) VPPUnionType( "vl_api_address_union_t", [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]], ) address = VPPType( "vl_api_address_t", [["vl_api_address_family_t", "af"], ["vl_api_address_union_t", "un"]], ) prefix = VPPType("vl_api_prefix_t", [["vl_api_address_t", "address"], ["u8", "len"]]) va_address_list = VPPType( "list_addresses", [["u8", "count"], ["vl_api_address_t", "addresses", 0, "count"]], ) message_with_va_address_list = VPPType( "msg_with_vla", [["list_addresses", "vla_address"], ["u8", "is_cool"]]) b = ip4.pack(inet_pton(AF_INET, "1.1.1.1")) self.assertEqual(len(b), 4) nt, size = ip4.unpack(b) self.assertEqual(str(nt), "1.1.1.1") b = ip6.pack(inet_pton(AF_INET6, "1::1")) self.assertEqual(len(b), 16) b = address.pack({ "af": af.ADDRESS_IP4, "un": { "ip4": inet_pton(AF_INET, "2.2.2.2") } }) self.assertEqual(len(b), 20) nt, size = address.unpack(b) self.assertEqual(str(nt), "2.2.2.2") # List of addresses address_list = [] for i in range(4): address_list.append({ "af": af.ADDRESS_IP4, "un": { "ip4": inet_pton(AF_INET, "2.2.2.2") } }) b = va_address_list.pack({ "count": len(address_list), "addresses": address_list }) self.assertEqual(len(b), 81) nt, size = va_address_list.unpack(b) self.assertEqual(str(nt.addresses[0]), "2.2.2.2") b = message_with_va_address_list.pack({ "vla_address": { "count": len(address_list), "addresses": address_list }, "is_cool": 100, }) self.assertEqual(len(b), 82) nt, size = message_with_va_address_list.unpack(b) self.assertEqual(nt.is_cool, 100)
def test_address_with_prefix(self): af = VPPEnumType( "vl_api_address_family_t", [["ADDRESS_IP4", 0], ["ADDRESS_IP6", 1], { "enumtype": "u32" }], ) ip4 = VPPTypeAlias("vl_api_ip4_address_t", {"type": "u8", "length": 4}) ip6 = VPPTypeAlias("vl_api_ip6_address_t", { "type": "u8", "length": 16 }) VPPUnionType( "vl_api_address_union_t", [["vl_api_ip4_address_t", "ip4"], ["vl_api_ip6_address_t", "ip6"]], ) address = VPPType( "vl_api_address_t", [["vl_api_address_family_t", "af"], ["vl_api_address_union_t", "un"]], ) prefix = VPPType("vl_api_prefix_t", [["vl_api_address_t", "address"], ["u8", "len"]]) prefix4 = VPPType("vl_api_ip4_prefix_t", [["vl_api_ip4_address_t", "address"], ["u8", "len"]]) prefix6 = VPPType("vl_api_ip6_prefix_t", [["vl_api_ip6_address_t", "address"], ["u8", "len"]]) address_with_prefix = VPPTypeAlias("vl_api_address_with_prefix_t", {"type": "vl_api_prefix_t"}) address4_with_prefix = VPPTypeAlias("vl_api_ip4_address_with_prefix_t", {"type": "vl_api_ip4_prefix_t"}) address6_with_prefix = VPPTypeAlias("vl_api_ip6_address_with_prefix_t", {"type": "vl_api_ip6_prefix_t"}) awp_type = VPPType("foobar_t", [["vl_api_address_with_prefix_t", "address"]]) # address with prefix b = address_with_prefix.pack(IPv4Interface("2.2.2.2/24")) self.assertEqual(len(b), 21) nt, size = address_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Interface)) self.assertEqual(str(nt), "2.2.2.2/24") b = address_with_prefix.pack(IPv6Interface("2::2/64")) self.assertEqual(len(b), 21) nt, size = address_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv6Interface)) self.assertEqual(str(nt), "2::2/64") b = address_with_prefix.pack(IPv4Network("2.2.2.2/24", strict=False)) self.assertEqual(len(b), 21) nt, size = address_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Interface)) self.assertEqual(str(nt), "2.2.2.0/24") b = address4_with_prefix.pack("2.2.2.2/24") self.assertEqual(len(b), 5) nt, size = address4_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Interface)) self.assertEqual(str(nt), "2.2.2.2/24") b = address4_with_prefix.pack(IPv4Interface("2.2.2.2/24")) self.assertEqual(len(b), 5) b = address6_with_prefix.pack("2::2/64") self.assertEqual(len(b), 17) nt, size = address6_with_prefix.unpack(b) self.assertTrue(isinstance(nt, IPv6Interface)) self.assertEqual(str(nt), "2::2/64") b = address6_with_prefix.pack(IPv6Interface("2::2/64")) self.assertEqual(len(b), 17) b = prefix.pack("192.168.10.0/24") self.assertEqual(len(b), 21) nt, size = prefix.unpack(b) self.assertTrue(isinstance(nt, IPv4Network)) self.assertEqual(str(nt), "192.168.10.0/24") b = awp_type.pack({"address": "1.2.3.4/24"}) self.assertEqual(len(b), 21) nt, size = awp_type.unpack(b) self.assertTrue(isinstance(nt.address, IPv4Interface)) self.assertEqual(str(nt.address), "1.2.3.4/24") b = awp_type.pack({"address": IPv4Interface("1.2.3.4/24")}) self.assertEqual(len(b), 21) nt, size = awp_type.unpack(b) self.assertTrue(isinstance(nt.address, IPv4Interface)) self.assertEqual(str(nt.address), "1.2.3.4/24")