src/vnet/map/test.c - fdio/vpp - Gitiles

 /*
  * test.c : MAP unit tests
  *
  * Copyright (c) 2016 Cisco and/or its affiliates.
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at:
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 #include <assert.h>
 #include "map.h"

 static map_domain_t *
 get_domain(ip4_address_t * ip4_prefix, u8 ip4_prefix_len,
 	   ip6_address_t * ip6_prefix, u8 ip6_prefix_len,
 	   ip6_address_t * ip6_src, u8 ip6_src_len,
 	   u8 ea_bits_len, u8 psid_offset,
 	   u8 psid_length, u16 mtu, u8 flags)
 {
   map_domain_t * d = malloc(sizeof(*d));
   u8 suffix_len;

   /* EA bits must be within the first 64 bits */
   if (ea_bits_len > 0 && (ip6_prefix_len + ea_bits_len) > 64)
     return NULL;

   /* Init domain struct */
   d->ip4_prefix.as_u32 = ip4_prefix->as_u32;
   d->ip4_prefix_len = ip4_prefix_len;
   d->ip6_prefix = *ip6_prefix;
   d->ip6_prefix_len = ip6_prefix_len;
   d->ip6_src = *ip6_src;
   d->ip6_src_len = ip6_src_len;
   d->ea_bits_len = ea_bits_len;
   d->psid_offset = psid_offset;
   d->psid_length = psid_length;
   d->mtu = mtu;
   d->flags = flags;

   /* How many, and which bits to grab from the IPv4 DA */
   if (ip4_prefix_len + ea_bits_len < 32)
     {
       if (!(flags & MAP_DOMAIN_TRANSLATION))
         d->flags |= MAP_DOMAIN_PREFIX;
       d->suffix_shift = 32 - ip4_prefix_len - ea_bits_len;
       suffix_len = ea_bits_len;
     }
   else
     {
       d->suffix_shift = 0;
       suffix_len = 32 - ip4_prefix_len;
     }
   d->suffix_mask = (1 << suffix_len) - 1;

   d->psid_shift = 16 - psid_length - psid_offset;
   d->psid_mask = (1 << d->psid_length) - 1;

   if (ip6_prefix_len + suffix_len + d->psid_length > 64)
     return NULL;

   d->ea_shift = 64 - ip6_prefix_len - suffix_len - d->psid_length;

   return d;
 }


 /*
  * VPP-340:
  * map_add_domain ip4-pfx 20.0.0.0/8 ip6-pfx 2001:db8::/40 ip6-src 2001:db8:ffff::/96 ea-bits-len 24 psid-offset 0 psid-len 0 map-t
  * IPv4 src = 100.0.0.1
  * IPv4 dst = 20.169.201.219
  * UDP dest port = 1232
  * IPv6 src = 2001:db8:ffff::6400:1
  * IPv6 dst = a9c9:dfb8::14a9:c9db:0
  * a9c9:dfb8::14a9:c9db:0 != 2001:db8:a9:c9db:0:14a9:c9db:0
  */
 static void
 test_map_t_destaddr (void)
 {
   ip4_address_t ip4_prefix;
   ip6_address_t ip6_prefix;
   ip6_address_t ip6_src;

   ip4_prefix.as_u32 = clib_host_to_net_u32(0x14000000);
   ip6_prefix.as_u64[0] = clib_host_to_net_u64(0x20010db800000000);
   ip6_prefix.as_u64[1] = 0;
   ip6_src.as_u64[0] = clib_host_to_net_u64(0x20010db8ffff0000);
   map_domain_t * d = get_domain (&ip4_prefix, 8, &ip6_prefix, 40, &ip6_src, 96, 24, 0, 0, 0, MAP_DOMAIN_TRANSLATION);

   ip6_address_t dst6;

   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db800a9c9db);
   assert(dst6.as_u64[1] == 0x000014a9c9db0000);
 }

 /*
  * VPP-228
  * ip4-pfx 20.0.0.0/8
  * ip6-pfx 2001:db8::/<n>
  * ip6-src 2001:db8:ffff::1
  * ea-bits-len 16 psid-offset 6 psid-len 8
  * 20.169.201.219 port 1232
  */
 static void
 test_map_eabits (void)
 {
   ip4_address_t ip4_prefix;
   ip6_address_t ip6_prefix;
   ip6_address_t ip6_src;
   ip6_address_t dst6;

   ip4_prefix.as_u32 = clib_host_to_net_u32(0x14000000);
   ip6_prefix.as_u64[0] = clib_host_to_net_u64(0x20010db800000000);
   ip6_prefix.as_u64[1] = 0;
   ip6_src.as_u64[0] = clib_host_to_net_u64(0x20010db8ffff0000);
   ip6_src.as_u64[1] = clib_host_to_net_u64(0x0000000000000001);
   map_domain_t * d = get_domain (&ip4_prefix, 16, &ip6_prefix, 48, &ip6_src,
 				 128, 16, 6, 8, 0, 0);
   assert(!d);

   //20.0.0.0/8	2001:db8::/32	4	2001:db8:a000::14a0:0:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 4, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a0000000);
   assert(dst6.as_u64[1] == 0x000014a000000000);

   //20.0.0.0/8	2001:db8::/32	8	2001:db8:a900::14a9:0:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 8, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a9000000);
   assert(dst6.as_u64[1] == 0x000014a900000000);

   //20.0.0.0/8	2001:db8::/32	10	2001:db8:a9c0::14a9:c000:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 10, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a9c00000);
   assert(dst6.as_u64[1] == 0x000014a9c0000000);

   //20.0.0.0/8	2001:db8::/32	16	2001:db8:a9c9::14a9:c900:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 16, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a9c90000);
   assert(dst6.as_u64[1] == 0x000014a9c9000000);

   //20.0.0.0/8	2001:db8::/32	20	2001:db8:a9c9:d000:0:14a9:c9d0:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 20, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a9c9d000);
   assert(dst6.as_u64[1] == 0x000014a9c9d00000);

   //20.0.0.0/8	2001:db8::/32	23	2001:db8:a9c9:da00:0:14a9:c9da:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 23, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a9c9da00);
   assert(dst6.as_u64[1] == 0x000014a9c9da0000);

   //20.169.201.0/24	2001:db8::/32	7	2001:db8:da00::14a9:c9da:0
   d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
 		  128, 7, 0, 0, 0, 0);
   dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
   dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
   assert(dst6.as_u64[0] == 0x20010db8a8000000);
   assert(dst6.as_u64[1] == 0x000014a800000000);
 }

 #define foreach_test_case			\
   _(map_t_destaddr)				\
   _(map_eabits)

 static void
 run_tests (void)
 {
 #define _(_test_name)				\
   test_ ## _test_name ();

   foreach_test_case
 #undef _
 }

 int main()
 {
   run_tests ();
   return 0;
 }
	/*
	* test.c : MAP unit tests
	*
	* Copyright (c) 2016 Cisco and/or its affiliates.
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at:
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	#include <assert.h>
	#include "map.h"

	static map_domain_t *
	get_domain(ip4_address_t * ip4_prefix, u8 ip4_prefix_len,
	ip6_address_t * ip6_prefix, u8 ip6_prefix_len,
	ip6_address_t * ip6_src, u8 ip6_src_len,
	u8 ea_bits_len, u8 psid_offset,
	u8 psid_length, u16 mtu, u8 flags)
	{
	map_domain_t * d = malloc(sizeof(*d));
	u8 suffix_len;

	/* EA bits must be within the first 64 bits */
	if (ea_bits_len > 0 && (ip6_prefix_len + ea_bits_len) > 64)
	return NULL;

	/* Init domain struct */
	d->ip4_prefix.as_u32 = ip4_prefix->as_u32;
	d->ip4_prefix_len = ip4_prefix_len;
	d->ip6_prefix = *ip6_prefix;
	d->ip6_prefix_len = ip6_prefix_len;
	d->ip6_src = *ip6_src;
	d->ip6_src_len = ip6_src_len;
	d->ea_bits_len = ea_bits_len;
	d->psid_offset = psid_offset;
	d->psid_length = psid_length;
	d->mtu = mtu;
	d->flags = flags;

	/* How many, and which bits to grab from the IPv4 DA */
	if (ip4_prefix_len + ea_bits_len < 32)
	{
	if (!(flags & MAP_DOMAIN_TRANSLATION))
	d->flags \|= MAP_DOMAIN_PREFIX;
	d->suffix_shift = 32 - ip4_prefix_len - ea_bits_len;
	suffix_len = ea_bits_len;
	}
	else
	{
	d->suffix_shift = 0;
	suffix_len = 32 - ip4_prefix_len;
	}
	d->suffix_mask = (1 << suffix_len) - 1;

	d->psid_shift = 16 - psid_length - psid_offset;
	d->psid_mask = (1 << d->psid_length) - 1;

	if (ip6_prefix_len + suffix_len + d->psid_length > 64)
	return NULL;

	d->ea_shift = 64 - ip6_prefix_len - suffix_len - d->psid_length;

	return d;
	}


	/*
	* VPP-340:
	* map_add_domain ip4-pfx 20.0.0.0/8 ip6-pfx 2001:db8::/40 ip6-src 2001:db8:ffff::/96 ea-bits-len 24 psid-offset 0 psid-len 0 map-t
	* IPv4 src = 100.0.0.1
	* IPv4 dst = 20.169.201.219
	* UDP dest port = 1232
	* IPv6 src = 2001:db8:ffff::6400:1
	* IPv6 dst = a9c9:dfb8::14a9:c9db:0
	* a9c9:dfb8::14a9:c9db:0 != 2001:db8:a9:c9db:0:14a9:c9db:0
	*/
	static void
	test_map_t_destaddr (void)
	{
	ip4_address_t ip4_prefix;
	ip6_address_t ip6_prefix;
	ip6_address_t ip6_src;

	ip4_prefix.as_u32 = clib_host_to_net_u32(0x14000000);
	ip6_prefix.as_u64[0] = clib_host_to_net_u64(0x20010db800000000);
	ip6_prefix.as_u64[1] = 0;
	ip6_src.as_u64[0] = clib_host_to_net_u64(0x20010db8ffff0000);
	map_domain_t * d = get_domain (&ip4_prefix, 8, &ip6_prefix, 40, &ip6_src, 96, 24, 0, 0, 0, MAP_DOMAIN_TRANSLATION);

	ip6_address_t dst6;

	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db800a9c9db);
	assert(dst6.as_u64[1] == 0x000014a9c9db0000);
	}

	/*
	* VPP-228
	* ip4-pfx 20.0.0.0/8
	* ip6-pfx 2001:db8::/<n>
	* ip6-src 2001:db8:ffff::1
	* ea-bits-len 16 psid-offset 6 psid-len 8
	* 20.169.201.219 port 1232
	*/
	static void
	test_map_eabits (void)
	{
	ip4_address_t ip4_prefix;
	ip6_address_t ip6_prefix;
	ip6_address_t ip6_src;
	ip6_address_t dst6;

	ip4_prefix.as_u32 = clib_host_to_net_u32(0x14000000);
	ip6_prefix.as_u64[0] = clib_host_to_net_u64(0x20010db800000000);
	ip6_prefix.as_u64[1] = 0;
	ip6_src.as_u64[0] = clib_host_to_net_u64(0x20010db8ffff0000);
	ip6_src.as_u64[1] = clib_host_to_net_u64(0x0000000000000001);
	map_domain_t * d = get_domain (&ip4_prefix, 16, &ip6_prefix, 48, &ip6_src,
	128, 16, 6, 8, 0, 0);
	assert(!d);

	//20.0.0.0/8 2001:db8::/32 4 2001:db8:a000::14a0:0:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 4, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a0000000);
	assert(dst6.as_u64[1] == 0x000014a000000000);

	//20.0.0.0/8 2001:db8::/32 8 2001:db8:a900::14a9:0:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 8, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a9000000);
	assert(dst6.as_u64[1] == 0x000014a900000000);

	//20.0.0.0/8 2001:db8::/32 10 2001:db8:a9c0::14a9:c000:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 10, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a9c00000);
	assert(dst6.as_u64[1] == 0x000014a9c0000000);

	//20.0.0.0/8 2001:db8::/32 16 2001:db8:a9c9::14a9:c900:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 16, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a9c90000);
	assert(dst6.as_u64[1] == 0x000014a9c9000000);

	//20.0.0.0/8 2001:db8::/32 20 2001:db8:a9c9:d000:0:14a9:c9d0:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 20, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a9c9d000);
	assert(dst6.as_u64[1] == 0x000014a9c9d00000);

	//20.0.0.0/8 2001:db8::/32 23 2001:db8:a9c9:da00:0:14a9:c9da:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 23, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a9c9da00);
	assert(dst6.as_u64[1] == 0x000014a9c9da0000);

	//20.169.201.0/24 2001:db8::/32 7 2001:db8:da00::14a9:c9da:0
	d = get_domain (&ip4_prefix, 8, &ip6_prefix, 32, &ip6_src,
	128, 7, 0, 0, 0, 0);
	dst6.as_u64[0] = map_get_pfx(d, 0x14a9c9db, 1232);
	dst6.as_u64[1] = map_get_sfx(d, 0x14a9c9db, 1232);
	assert(dst6.as_u64[0] == 0x20010db8a8000000);
	assert(dst6.as_u64[1] == 0x000014a800000000);
	}

	#define foreach_test_case \
	_(map_t_destaddr) \
	_(map_eabits)

	static void
	run_tests (void)
	{
	#define _(_test_name) \
	test_ ## _test_name ();

	foreach_test_case
	#undef _
	}

	int main()
	{
	run_tests ();
	return 0;
	}