static bool test_function_generate_df_flag(void)
{
struct ipv6hdr hdr;
bool success = true;
hdr.payload_len = cpu_to_be16(4); /* packet length is 44. */
success &= assert_equals_u16(1, generate_df_flag(&hdr), "Length < 88 bytes");
hdr.payload_len = cpu_to_be16(48); /* packet length is 88. */
success &= assert_equals_u16(1, generate_df_flag(&hdr), "Length = 88 bytes");
hdr.payload_len = cpu_to_be16(500); /* packet length is 540. */
success &= assert_equals_u16(0, generate_df_flag(&hdr), "88 < Len < 1280");
hdr.payload_len = cpu_to_be16(1240); /* packet length is 1280. */
success &= assert_equals_u16(0, generate_df_flag(&hdr), "Len = 1280");
hdr.payload_len = cpu_to_be16(4000); /* packet length is 4040. */
success &= assert_equals_u16(1, generate_df_flag(&hdr), "Len > 1280");
return success;
}
static bool test_function_icmp6_minimum_mtu(void)
{
struct xlation state = { .jool.global = config };
int i;
bool success = true;
/*
* I'm assuming the default plateaus list has 3 elements or more.
* (so I don't have to reallocate mtu_plateaus)
*/
config->cfg.mtu_plateaus[0] = 5000;
config->cfg.mtu_plateaus[1] = 4000;
config->cfg.mtu_plateaus[2] = 500;
config->cfg.mtu_plateau_count = 2;
/* Simple tests */
success &= ASSERT_UINT(1320, min_mtu(1300, 3000, 3000, 2000), "min(1300, 3000, 3000)");
success &= ASSERT_UINT(1321, min_mtu(3001, 1301, 3001, 2001), "min(3001, 1301, 3001)");
success &= ASSERT_UINT(1302, min_mtu(3002, 3002, 1302, 2002), "min(3002, 3002, 1302)");
if (!success)
return false;
/* Lowest MTU is illegal on IPv6. */
success &= ASSERT_UINT(1280, min_mtu(100, 200, 200, 150), "min(100, 200, 200)");
success &= ASSERT_UINT(1280, min_mtu(200, 100, 200, 150), "min(200, 100, 200)");
success &= ASSERT_UINT(1280, min_mtu(200, 200, 100, 150), "min(200, 200, 100)");
/* Test plateaus (pkt is min). */
for (i = 5500; i > 5000 && success; --i)
success &= ASSERT_UINT(5020, min_mtu(0, 6000, 6000, i), "min(%d, 6000, 6000)", i);
for (i = 5000; i > 4000 && success; --i)
success &= ASSERT_UINT(4020, min_mtu(0, 6000, 6000, i), "min(%d, 6000, 6000)", i);
for (i = 4000; i >= 0 && success; --i)
success &= ASSERT_UINT(1280, min_mtu(0, 6000, 6000, i), "min(%d, 6000, 6000)", i);
/* Test plateaus (in/out is min). */
success &= ASSERT_UINT(1420, min_mtu(0, 1400, 5500, 4500), "min(4000,1400,5500)");
success &= ASSERT_UINT(1400, min_mtu(0, 5500, 1400, 4500), "min(4000,5500,1400)");
/* Plateaus and illegal MTU at the same time. */
success &= ASSERT_UINT(1280, min_mtu(0, 700, 700, 1000), "min(500, 700, 700)");
success &= ASSERT_UINT(1280, min_mtu(0, 1, 700, 1000), "min(500, 1, 700)");
success &= ASSERT_UINT(1280, min_mtu(0, 700, 1, 1000), "min(500, 700, 1)");
return success;
}
#undef min_mtu
static bool test_function_icmp4_to_icmp6_param_prob(void)
{
struct icmphdr hdr4;
struct icmp6hdr hdr6;
bool success = true;
hdr4.type = ICMP_PARAMETERPROB;
hdr4.code = ICMP_PTR_INDICATES_ERROR;
hdr4.icmp4_unused = cpu_to_be32(0x08000000U);
success &= ASSERT_INT(0, icmp4_to_icmp6_param_prob(&hdr4, &hdr6), "func result 1");
success &= ASSERT_UINT(ICMPV6_HDR_FIELD, hdr6.icmp6_code, "code");
success &= ASSERT_UINT(7, be32_to_cpu(hdr6.icmp6_pointer), "pointer");
hdr4.icmp4_unused = cpu_to_be32(0x05000000U);
success &= ASSERT_INT(-EINVAL, icmp4_to_icmp6_param_prob(&hdr4, &hdr6), "func result 2");
return success;
}
static bool test_function_generate_ipv4_id(void)
{
struct frag_hdr hdr;
__be16 attempt_1, attempt_2, attempt_3;
bool success = true;
attempt_1 = generate_ipv4_id(NULL);
attempt_2 = generate_ipv4_id(NULL);
attempt_3 = generate_ipv4_id(NULL);
/*
* At least one of the attempts should be nonzero,
* otherwise the random would be sucking major ****.
*/
success &= ASSERT_BOOL(true, (attempt_1 | attempt_2 | attempt_3) != 0, "No frag");
hdr.identification = 0;
success &= ASSERT_BE16(0, generate_ipv4_id(&hdr), "Simplest id");
hdr.identification = cpu_to_be32(0x0000abcdU);
success &= ASSERT_BE16(0xabcd, generate_ipv4_id(&hdr), "No overflow");
hdr.identification = cpu_to_be32(0x12345678U);
success &= ASSERT_BE16(0x5678, generate_ipv4_id(&hdr), "Overflow");
return success;
}
static bool test_function_generate_df_flag(void)
{
struct packet pkt;
struct sk_buff *skb;
bool success = true;
skb = alloc_skb(1500, GFP_ATOMIC);
if (!skb)
return false;
pkt.skb = skb;
skb_put(skb, 1000);
success &= ASSERT_UINT(0, generate_df_flag(&pkt), "Len < 1260");
skb_put(skb, 260);
success &= ASSERT_UINT(0, generate_df_flag(&pkt), "Len = 1260");
skb_put(skb, 200);
success &= ASSERT_UINT(1, generate_df_flag(&pkt), "Len > 1260");
kfree_skb(skb);
return success;
}
/**
* By the way. This test kind of looks like it should test more combinations of headers.
* But that'd be testing the header iterator, not the build_protocol_field() function.
* Please look elsewhere for that.
*/
static bool test_function_build_protocol_field(void)
{
struct ipv6hdr *ip6_hdr;
struct ipv6_opt_hdr *hop_by_hop_hdr;
struct ipv6_opt_hdr *routing_hdr;
struct ipv6_opt_hdr *dest_options_hdr;
struct icmp6hdr *icmp6_hdr;
ip6_hdr = kmalloc(sizeof(*ip6_hdr) + 8 + 16 + 24 + sizeof(struct tcphdr), GFP_ATOMIC);
if (!ip6_hdr) {
log_err("Could not allocate a test packet.");
goto failure;
}
/* Just ICMP. */
ip6_hdr->nexthdr = NEXTHDR_ICMP;
ip6_hdr->payload_len = cpu_to_be16(sizeof(*icmp6_hdr));
if (!ASSERT_UINT(IPPROTO_ICMP, ttp64_xlat_proto(ip6_hdr), "Just ICMP"))
goto failure;
/* Skippable headers then ICMP. */
ip6_hdr->nexthdr = NEXTHDR_HOP;
ip6_hdr->payload_len = cpu_to_be16(8 + 16 + 24 + sizeof(*icmp6_hdr));
hop_by_hop_hdr = (struct ipv6_opt_hdr *) (ip6_hdr + 1);
hop_by_hop_hdr->nexthdr = NEXTHDR_ROUTING;
hop_by_hop_hdr->hdrlen = 0; /* the hdrlen field does not include the first 8 octets. */
routing_hdr = (struct ipv6_opt_hdr *) (((unsigned char *) hop_by_hop_hdr) + 8);
routing_hdr->nexthdr = NEXTHDR_DEST;
routing_hdr->hdrlen = 1;
dest_options_hdr = (struct ipv6_opt_hdr *) (((unsigned char *) routing_hdr) + 16);
dest_options_hdr->nexthdr = NEXTHDR_ICMP;
dest_options_hdr->hdrlen = 2;
if (!ASSERT_UINT(IPPROTO_ICMP, ttp64_xlat_proto(ip6_hdr), "Skippable then ICMP"))
goto failure;
/* Skippable headers then something else */
dest_options_hdr->nexthdr = NEXTHDR_TCP;
ip6_hdr->payload_len = cpu_to_be16(8 + 16 + 24 + sizeof(struct tcphdr));
if (!ASSERT_UINT(IPPROTO_TCP, ttp64_xlat_proto(ip6_hdr), "Skippable then TCP"))
goto failure;
kfree(ip6_hdr);
return true;
failure:
kfree(ip6_hdr);
return false;
}
static bool test_function_has_nonzero_segments_left(void)
{
struct ipv6hdr *ip6_hdr;
struct ipv6_rt_hdr *routing_hdr;
struct frag_hdr *fragment_hdr;
__u32 offset;
bool success = true;
ip6_hdr = kmalloc(sizeof(*ip6_hdr) + sizeof(*fragment_hdr) + sizeof(*routing_hdr), GFP_ATOMIC);
if (!ip6_hdr) {
log_err("Could not allocate a test packet.");
return false;
}
ip6_hdr->payload_len = cpu_to_be16(sizeof(*fragment_hdr) + sizeof(*routing_hdr));
/* No extension headers. */
ip6_hdr->nexthdr = NEXTHDR_TCP;
success &= ASSERT_BOOL(false, has_nonzero_segments_left(ip6_hdr, &offset), "No extension headers");
if (!success)
goto end;
/* Routing header with nonzero segments left. */
ip6_hdr->nexthdr = NEXTHDR_ROUTING;
routing_hdr = (struct ipv6_rt_hdr *) (ip6_hdr + 1);
routing_hdr->segments_left = 12;
success &= ASSERT_BOOL(true, has_nonzero_segments_left(ip6_hdr, &offset), "Nonzero left - result");
success &= ASSERT_UINT(40 + 3, offset, "Nonzero left - offset");
if (!success)
goto end;
/* Routing header with zero segments left. */
routing_hdr->segments_left = 0;
success &= ASSERT_BOOL(false, has_nonzero_segments_left(ip6_hdr, &offset), "Zero left");
if (!success)
goto end;
/*
* Fragment header, then routing header with nonzero segments left
* (further test the out parameter).
*/
ip6_hdr->nexthdr = NEXTHDR_FRAGMENT;
fragment_hdr = (struct frag_hdr *) (ip6_hdr + 1);
fragment_hdr->nexthdr = NEXTHDR_ROUTING;
routing_hdr = (struct ipv6_rt_hdr *) (fragment_hdr + 1);
routing_hdr->segments_left = 24;
success &= ASSERT_BOOL(true, has_nonzero_segments_left(ip6_hdr, &offset), "Two headers - result");
success &= ASSERT_UINT(40 + 8 + 3, offset, "Two headers - offset");
/* Fall through. */
end:
kfree(ip6_hdr);
return success;
}
static bool test_function_icmp4_minimum_mtu(void)
{
bool success = true;
success &= ASSERT_UINT(2, be16_to_cpu(minimum(2, 4, 6)), "First is min");
success &= ASSERT_UINT(8, be16_to_cpu(minimum(10, 8, 12)), "Second is min");
success &= ASSERT_UINT(14, be16_to_cpu(minimum(16, 18, 14)), "Third is min");
return success;
}
