xref: /linux/tools/testing/selftests/net/lib/gro.c (revision ff1cb3ad2abce4d03eaf3e8d5f38a7a5dfd36e79)
18888bf4fSJakub Kicinski // SPDX-License-Identifier: GPL-2.0
28888bf4fSJakub Kicinski /*
38888bf4fSJakub Kicinski  * This testsuite provides conformance testing for GRO coalescing.
48888bf4fSJakub Kicinski  *
58888bf4fSJakub Kicinski  * Test cases:
68888bf4fSJakub Kicinski  *
78888bf4fSJakub Kicinski  * data_*:
88888bf4fSJakub Kicinski  *  Data packets of the same size and same header setup with correct
98888bf4fSJakub Kicinski  *  sequence numbers coalesce. The one exception being the last data
108888bf4fSJakub Kicinski  *  packet coalesced: it can be smaller than the rest and coalesced
118888bf4fSJakub Kicinski  *  as long as it is in the same flow.
128888bf4fSJakub Kicinski  *   - data_same:    same size packets coalesce
138888bf4fSJakub Kicinski  *   - data_lrg_sml: large then small coalesces
148888bf4fSJakub Kicinski  *   - data_sml_lrg: small then large doesn't coalesce
158888bf4fSJakub Kicinski  *
168888bf4fSJakub Kicinski  * ack:
178888bf4fSJakub Kicinski  *  Pure ACK does not coalesce.
188888bf4fSJakub Kicinski  *
198888bf4fSJakub Kicinski  * flags_*:
208888bf4fSJakub Kicinski  *  No packets with PSH, SYN, URG, RST, CWR set will be coalesced.
218888bf4fSJakub Kicinski  *   - flags_psh, flags_syn, flags_rst, flags_urg, flags_cwr
228888bf4fSJakub Kicinski  *
238888bf4fSJakub Kicinski  * tcp_*:
248888bf4fSJakub Kicinski  *  Packets with incorrect checksum, non-consecutive seqno and
258888bf4fSJakub Kicinski  *  different TCP header options shouldn't coalesce. Nit: given that
268888bf4fSJakub Kicinski  *  some extension headers have paddings, such as timestamp, headers
278888bf4fSJakub Kicinski  *  that are padded differently would not be coalesced.
288888bf4fSJakub Kicinski  *   - tcp_csum: incorrect checksum
298888bf4fSJakub Kicinski  *   - tcp_seq:  non-consecutive sequence numbers
308888bf4fSJakub Kicinski  *   - tcp_ts:   different timestamps
318888bf4fSJakub Kicinski  *   - tcp_opt:  different TCP options
328888bf4fSJakub Kicinski  *
338888bf4fSJakub Kicinski  * ip_*:
348888bf4fSJakub Kicinski  *  Packets with different (ECN, TTL, TOS) header, IP options or
358888bf4fSJakub Kicinski  *  IP fragments shouldn't coalesce.
368888bf4fSJakub Kicinski  *   - ip_ecn, ip_tos:            shared between IPv4/IPv6
378888bf4fSJakub Kicinski  *   - ip_ttl, ip_opt, ip_frag4:  IPv4 only
388888bf4fSJakub Kicinski  *   - ip_id_df*:                 IPv4 IP ID field coalescing tests
398888bf4fSJakub Kicinski  *   - ip_frag6, ip_v6ext_*:      IPv6 only
408888bf4fSJakub Kicinski  *
418888bf4fSJakub Kicinski  * large_*:
428888bf4fSJakub Kicinski  *  Packets larger than GRO_MAX_SIZE packets shouldn't coalesce.
438888bf4fSJakub Kicinski  *   - large_max: exceeding max size
448888bf4fSJakub Kicinski  *   - large_rem: remainder handling
458888bf4fSJakub Kicinski  *
46ba5d4128SJakub Kicinski  * single, capacity:
47ba5d4128SJakub Kicinski  *  Boring cases used to test coalescing machinery itself and stats
48ba5d4128SJakub Kicinski  *  more than protocol behavior.
49ba5d4128SJakub Kicinski  *
508888bf4fSJakub Kicinski  * MSS is defined as 4096 - header because if it is too small
518888bf4fSJakub Kicinski  * (i.e. 1500 MTU - header), it will result in many packets,
528888bf4fSJakub Kicinski  * increasing the "large" test case's flakiness. This is because
538888bf4fSJakub Kicinski  * due to time sensitivity in the coalescing window, the receiver
548888bf4fSJakub Kicinski  * may not coalesce all of the packets.
558888bf4fSJakub Kicinski  *
568888bf4fSJakub Kicinski  * Note the timing issue applies to all of the test cases, so some
578888bf4fSJakub Kicinski  * flakiness is to be expected.
588888bf4fSJakub Kicinski  *
598888bf4fSJakub Kicinski  */
608888bf4fSJakub Kicinski 
618888bf4fSJakub Kicinski #define _GNU_SOURCE
628888bf4fSJakub Kicinski 
638888bf4fSJakub Kicinski #include <arpa/inet.h>
648888bf4fSJakub Kicinski #include <errno.h>
658888bf4fSJakub Kicinski #include <error.h>
668888bf4fSJakub Kicinski #include <getopt.h>
678888bf4fSJakub Kicinski #include <linux/filter.h>
688888bf4fSJakub Kicinski #include <linux/if_packet.h>
698888bf4fSJakub Kicinski #include <linux/ipv6.h>
70f26d43acSJakub Kicinski #include <linux/net_tstamp.h>
718888bf4fSJakub Kicinski #include <net/ethernet.h>
728888bf4fSJakub Kicinski #include <net/if.h>
738888bf4fSJakub Kicinski #include <netinet/in.h>
748888bf4fSJakub Kicinski #include <netinet/ip.h>
758888bf4fSJakub Kicinski #include <netinet/ip6.h>
768888bf4fSJakub Kicinski #include <netinet/tcp.h>
778888bf4fSJakub Kicinski #include <stdbool.h>
788888bf4fSJakub Kicinski #include <stddef.h>
798888bf4fSJakub Kicinski #include <stdio.h>
808888bf4fSJakub Kicinski #include <stdarg.h>
818888bf4fSJakub Kicinski #include <string.h>
82f26d43acSJakub Kicinski #include <time.h>
838888bf4fSJakub Kicinski #include <unistd.h>
848888bf4fSJakub Kicinski 
858888bf4fSJakub Kicinski #include "kselftest.h"
868888bf4fSJakub Kicinski #include "ksft.h"
878888bf4fSJakub Kicinski 
888888bf4fSJakub Kicinski #define DPORT 8000
898888bf4fSJakub Kicinski #define SPORT 1500
908888bf4fSJakub Kicinski #define PAYLOAD_LEN 100
918888bf4fSJakub Kicinski #define NUM_PACKETS 4
928888bf4fSJakub Kicinski #define START_SEQ 100
938888bf4fSJakub Kicinski #define START_ACK 100
948888bf4fSJakub Kicinski #define ETH_P_NONE 0
958888bf4fSJakub Kicinski #define TOTAL_HDR_LEN (ETH_HLEN + sizeof(struct ipv6hdr) + sizeof(struct tcphdr))
968888bf4fSJakub Kicinski #define MSS (4096 - sizeof(struct tcphdr) - sizeof(struct ipv6hdr))
978888bf4fSJakub Kicinski #define MAX_PAYLOAD (IP_MAXPACKET - sizeof(struct tcphdr) - sizeof(struct ipv6hdr))
988888bf4fSJakub Kicinski #define NUM_LARGE_PKT (MAX_PAYLOAD / MSS)
998888bf4fSJakub Kicinski #define MAX_HDR_LEN (ETH_HLEN + sizeof(struct ipv6hdr) + sizeof(struct tcphdr))
1008888bf4fSJakub Kicinski #define MIN_EXTHDR_SIZE 8
1018888bf4fSJakub Kicinski #define EXT_PAYLOAD_1 "\x00\x00\x00\x00\x00\x00"
1028888bf4fSJakub Kicinski #define EXT_PAYLOAD_2 "\x11\x11\x11\x11\x11\x11"
1038888bf4fSJakub Kicinski 
1048888bf4fSJakub Kicinski #define ipv6_optlen(p)  (((p)->hdrlen+1) << 3) /* calculate IPv6 extension header len */
1058888bf4fSJakub Kicinski #define BUILD_BUG_ON(condition) ((void)sizeof(char[1 - 2*!!(condition)]))
1068888bf4fSJakub Kicinski 
1078888bf4fSJakub Kicinski enum flush_id_case {
1088888bf4fSJakub Kicinski 	FLUSH_ID_DF1_INC,
1098888bf4fSJakub Kicinski 	FLUSH_ID_DF1_FIXED,
1108888bf4fSJakub Kicinski 	FLUSH_ID_DF0_INC,
1118888bf4fSJakub Kicinski 	FLUSH_ID_DF0_FIXED,
1128888bf4fSJakub Kicinski 	FLUSH_ID_DF1_INC_FIXED,
1138888bf4fSJakub Kicinski 	FLUSH_ID_DF1_FIXED_INC,
1148888bf4fSJakub Kicinski };
1158888bf4fSJakub Kicinski 
1168888bf4fSJakub Kicinski static const char *addr6_src = "fdaa::2";
1178888bf4fSJakub Kicinski static const char *addr6_dst = "fdaa::1";
1188888bf4fSJakub Kicinski static const char *addr4_src = "192.168.1.200";
1198888bf4fSJakub Kicinski static const char *addr4_dst = "192.168.1.100";
1208888bf4fSJakub Kicinski static int proto = -1;
1218888bf4fSJakub Kicinski static uint8_t src_mac[ETH_ALEN], dst_mac[ETH_ALEN];
1228888bf4fSJakub Kicinski static char *testname = "data";
1238888bf4fSJakub Kicinski static char *ifname = "eth0";
1248888bf4fSJakub Kicinski static char *smac = "aa:00:00:00:00:02";
1258888bf4fSJakub Kicinski static char *dmac = "aa:00:00:00:00:01";
1268888bf4fSJakub Kicinski static bool verbose;
1278888bf4fSJakub Kicinski static bool tx_socket = true;
1288888bf4fSJakub Kicinski static int tcp_offset = -1;
1298888bf4fSJakub Kicinski static int total_hdr_len = -1;
1308888bf4fSJakub Kicinski static int ethhdr_proto = -1;
1318888bf4fSJakub Kicinski static bool ipip;
132f26d43acSJakub Kicinski static uint64_t txtime_ns;
133ba5d4128SJakub Kicinski static int num_flows = 4;
134*ff1cb3adSJakub Kicinski static bool order_check;
135ba5d4128SJakub Kicinski 
136ba5d4128SJakub Kicinski #define CAPACITY_PAYLOAD_LEN 200
137f26d43acSJakub Kicinski 
138f26d43acSJakub Kicinski #define TXTIME_DELAY_MS 5
1398888bf4fSJakub Kicinski 
1408888bf4fSJakub Kicinski static void vlog(const char *fmt, ...)
1418888bf4fSJakub Kicinski {
1428888bf4fSJakub Kicinski 	va_list args;
1438888bf4fSJakub Kicinski 
1448888bf4fSJakub Kicinski 	if (verbose) {
1458888bf4fSJakub Kicinski 		va_start(args, fmt);
1468888bf4fSJakub Kicinski 		vfprintf(stderr, fmt, args);
1478888bf4fSJakub Kicinski 		va_end(args);
1488888bf4fSJakub Kicinski 	}
1498888bf4fSJakub Kicinski }
1508888bf4fSJakub Kicinski 
1518888bf4fSJakub Kicinski static void setup_sock_filter(int fd)
1528888bf4fSJakub Kicinski {
1538888bf4fSJakub Kicinski 	const int dport_off = tcp_offset + offsetof(struct tcphdr, dest);
1548888bf4fSJakub Kicinski 	const int ethproto_off = offsetof(struct ethhdr, h_proto);
1558888bf4fSJakub Kicinski 	int optlen = 0;
1568888bf4fSJakub Kicinski 	int ipproto_off, opt_ipproto_off;
1578888bf4fSJakub Kicinski 	int next_off;
1588888bf4fSJakub Kicinski 
1598888bf4fSJakub Kicinski 	if (ipip)
1608888bf4fSJakub Kicinski 		next_off = sizeof(struct iphdr) + offsetof(struct iphdr, protocol);
1618888bf4fSJakub Kicinski 	else if (proto == PF_INET)
1628888bf4fSJakub Kicinski 		next_off = offsetof(struct iphdr, protocol);
1638888bf4fSJakub Kicinski 	else
1648888bf4fSJakub Kicinski 		next_off = offsetof(struct ipv6hdr, nexthdr);
1658888bf4fSJakub Kicinski 	ipproto_off = ETH_HLEN + next_off;
1668888bf4fSJakub Kicinski 
1678888bf4fSJakub Kicinski 	/* Overridden later if exthdrs are used: */
1688888bf4fSJakub Kicinski 	opt_ipproto_off = ipproto_off;
1698888bf4fSJakub Kicinski 
1708888bf4fSJakub Kicinski 	if (strcmp(testname, "ip_opt") == 0) {
1718888bf4fSJakub Kicinski 		optlen = sizeof(struct ip_timestamp);
1728888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_frag6") == 0 ||
1738888bf4fSJakub Kicinski 		   strcmp(testname, "ip_v6ext_same") == 0 ||
1748888bf4fSJakub Kicinski 		   strcmp(testname, "ip_v6ext_diff") == 0) {
1758888bf4fSJakub Kicinski 		BUILD_BUG_ON(sizeof(struct ip6_hbh) > MIN_EXTHDR_SIZE);
1768888bf4fSJakub Kicinski 		BUILD_BUG_ON(sizeof(struct ip6_dest) > MIN_EXTHDR_SIZE);
1778888bf4fSJakub Kicinski 		BUILD_BUG_ON(sizeof(struct ip6_frag) > MIN_EXTHDR_SIZE);
1788888bf4fSJakub Kicinski 
1798888bf4fSJakub Kicinski 		/* same size for HBH and Fragment extension header types */
1808888bf4fSJakub Kicinski 		optlen = MIN_EXTHDR_SIZE;
1818888bf4fSJakub Kicinski 		opt_ipproto_off = ETH_HLEN + sizeof(struct ipv6hdr)
1828888bf4fSJakub Kicinski 			+ offsetof(struct ip6_ext, ip6e_nxt);
1838888bf4fSJakub Kicinski 	}
1848888bf4fSJakub Kicinski 
1858888bf4fSJakub Kicinski 	/* this filter validates the following:
1868888bf4fSJakub Kicinski 	 *	- packet is IPv4/IPv6 according to the running test.
1878888bf4fSJakub Kicinski 	 *	- packet is TCP. Also handles the case of one extension header and then TCP.
1888888bf4fSJakub Kicinski 	 *	- checks the packet tcp dport equals to DPORT. Also handles the case of one
1898888bf4fSJakub Kicinski 	 *	  extension header and then TCP.
1908888bf4fSJakub Kicinski 	 */
1918888bf4fSJakub Kicinski 	struct sock_filter filter[] = {
1928888bf4fSJakub Kicinski 			BPF_STMT(BPF_LD  + BPF_H   + BPF_ABS, ethproto_off),
1938888bf4fSJakub Kicinski 			BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, ntohs(ethhdr_proto), 0, 9),
1948888bf4fSJakub Kicinski 			BPF_STMT(BPF_LD  + BPF_B   + BPF_ABS, ipproto_off),
1958888bf4fSJakub Kicinski 			BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, IPPROTO_TCP, 2, 0),
1968888bf4fSJakub Kicinski 			BPF_STMT(BPF_LD  + BPF_B   + BPF_ABS, opt_ipproto_off),
1978888bf4fSJakub Kicinski 			BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, IPPROTO_TCP, 0, 5),
1988888bf4fSJakub Kicinski 			BPF_STMT(BPF_LD  + BPF_H   + BPF_ABS, dport_off),
1998888bf4fSJakub Kicinski 			BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, DPORT, 2, 0),
2008888bf4fSJakub Kicinski 			BPF_STMT(BPF_LD  + BPF_H   + BPF_ABS, dport_off + optlen),
2018888bf4fSJakub Kicinski 			BPF_JUMP(BPF_JMP + BPF_JEQ + BPF_K, DPORT, 0, 1),
2028888bf4fSJakub Kicinski 			BPF_STMT(BPF_RET + BPF_K, 0xFFFFFFFF),
2038888bf4fSJakub Kicinski 			BPF_STMT(BPF_RET + BPF_K, 0),
2048888bf4fSJakub Kicinski 	};
2058888bf4fSJakub Kicinski 
2068888bf4fSJakub Kicinski 	struct sock_fprog bpf = {
2078888bf4fSJakub Kicinski 		.len = ARRAY_SIZE(filter),
2088888bf4fSJakub Kicinski 		.filter = filter,
2098888bf4fSJakub Kicinski 	};
2108888bf4fSJakub Kicinski 
2118888bf4fSJakub Kicinski 	if (setsockopt(fd, SOL_SOCKET, SO_ATTACH_FILTER, &bpf, sizeof(bpf)) < 0)
2128888bf4fSJakub Kicinski 		error(1, errno, "error setting filter");
2138888bf4fSJakub Kicinski }
2148888bf4fSJakub Kicinski 
2158888bf4fSJakub Kicinski static uint32_t checksum_nofold(void *data, size_t len, uint32_t sum)
2168888bf4fSJakub Kicinski {
2178888bf4fSJakub Kicinski 	uint16_t *words = data;
2188888bf4fSJakub Kicinski 	int i;
2198888bf4fSJakub Kicinski 
2208888bf4fSJakub Kicinski 	for (i = 0; i < len / 2; i++)
2218888bf4fSJakub Kicinski 		sum += words[i];
2228888bf4fSJakub Kicinski 	if (len & 1)
2238888bf4fSJakub Kicinski 		sum += ((char *)data)[len - 1];
2248888bf4fSJakub Kicinski 	return sum;
2258888bf4fSJakub Kicinski }
2268888bf4fSJakub Kicinski 
2278888bf4fSJakub Kicinski static uint16_t checksum_fold(void *data, size_t len, uint32_t sum)
2288888bf4fSJakub Kicinski {
2298888bf4fSJakub Kicinski 	sum = checksum_nofold(data, len, sum);
2308888bf4fSJakub Kicinski 	while (sum > 0xFFFF)
2318888bf4fSJakub Kicinski 		sum = (sum & 0xFFFF) + (sum >> 16);
2328888bf4fSJakub Kicinski 	return ~sum;
2338888bf4fSJakub Kicinski }
2348888bf4fSJakub Kicinski 
2358888bf4fSJakub Kicinski static uint16_t tcp_checksum(void *buf, int payload_len)
2368888bf4fSJakub Kicinski {
2378888bf4fSJakub Kicinski 	struct pseudo_header6 {
2388888bf4fSJakub Kicinski 		struct in6_addr saddr;
2398888bf4fSJakub Kicinski 		struct in6_addr daddr;
2408888bf4fSJakub Kicinski 		uint16_t protocol;
2418888bf4fSJakub Kicinski 		uint16_t payload_len;
2428888bf4fSJakub Kicinski 	} ph6;
2438888bf4fSJakub Kicinski 	struct pseudo_header4 {
2448888bf4fSJakub Kicinski 		struct in_addr saddr;
2458888bf4fSJakub Kicinski 		struct in_addr daddr;
2468888bf4fSJakub Kicinski 		uint16_t protocol;
2478888bf4fSJakub Kicinski 		uint16_t payload_len;
2488888bf4fSJakub Kicinski 	} ph4;
2498888bf4fSJakub Kicinski 	uint32_t sum = 0;
2508888bf4fSJakub Kicinski 
2518888bf4fSJakub Kicinski 	if (proto == PF_INET6) {
2528888bf4fSJakub Kicinski 		if (inet_pton(AF_INET6, addr6_src, &ph6.saddr) != 1)
2538888bf4fSJakub Kicinski 			error(1, errno, "inet_pton6 source ip pseudo");
2548888bf4fSJakub Kicinski 		if (inet_pton(AF_INET6, addr6_dst, &ph6.daddr) != 1)
2558888bf4fSJakub Kicinski 			error(1, errno, "inet_pton6 dest ip pseudo");
2568888bf4fSJakub Kicinski 		ph6.protocol = htons(IPPROTO_TCP);
2578888bf4fSJakub Kicinski 		ph6.payload_len = htons(sizeof(struct tcphdr) + payload_len);
2588888bf4fSJakub Kicinski 
2598888bf4fSJakub Kicinski 		sum = checksum_nofold(&ph6, sizeof(ph6), 0);
2608888bf4fSJakub Kicinski 	} else if (proto == PF_INET) {
2618888bf4fSJakub Kicinski 		if (inet_pton(AF_INET, addr4_src, &ph4.saddr) != 1)
2628888bf4fSJakub Kicinski 			error(1, errno, "inet_pton source ip pseudo");
2638888bf4fSJakub Kicinski 		if (inet_pton(AF_INET, addr4_dst, &ph4.daddr) != 1)
2648888bf4fSJakub Kicinski 			error(1, errno, "inet_pton dest ip pseudo");
2658888bf4fSJakub Kicinski 		ph4.protocol = htons(IPPROTO_TCP);
2668888bf4fSJakub Kicinski 		ph4.payload_len = htons(sizeof(struct tcphdr) + payload_len);
2678888bf4fSJakub Kicinski 
2688888bf4fSJakub Kicinski 		sum = checksum_nofold(&ph4, sizeof(ph4), 0);
2698888bf4fSJakub Kicinski 	}
2708888bf4fSJakub Kicinski 
2718888bf4fSJakub Kicinski 	return checksum_fold(buf, sizeof(struct tcphdr) + payload_len, sum);
2728888bf4fSJakub Kicinski }
2738888bf4fSJakub Kicinski 
2748888bf4fSJakub Kicinski static void read_MAC(uint8_t *mac_addr, char *mac)
2758888bf4fSJakub Kicinski {
2768888bf4fSJakub Kicinski 	if (sscanf(mac, "%hhx:%hhx:%hhx:%hhx:%hhx:%hhx",
2778888bf4fSJakub Kicinski 		   &mac_addr[0], &mac_addr[1], &mac_addr[2],
2788888bf4fSJakub Kicinski 		   &mac_addr[3], &mac_addr[4], &mac_addr[5]) != 6)
2798888bf4fSJakub Kicinski 		error(1, 0, "sscanf");
2808888bf4fSJakub Kicinski }
2818888bf4fSJakub Kicinski 
2828888bf4fSJakub Kicinski static void fill_datalinklayer(void *buf)
2838888bf4fSJakub Kicinski {
2848888bf4fSJakub Kicinski 	struct ethhdr *eth = buf;
2858888bf4fSJakub Kicinski 
2868888bf4fSJakub Kicinski 	memcpy(eth->h_dest, dst_mac, ETH_ALEN);
2878888bf4fSJakub Kicinski 	memcpy(eth->h_source, src_mac, ETH_ALEN);
2888888bf4fSJakub Kicinski 	eth->h_proto = ethhdr_proto;
2898888bf4fSJakub Kicinski }
2908888bf4fSJakub Kicinski 
2918888bf4fSJakub Kicinski static void fill_networklayer(void *buf, int payload_len, int protocol)
2928888bf4fSJakub Kicinski {
2938888bf4fSJakub Kicinski 	struct ipv6hdr *ip6h = buf;
2948888bf4fSJakub Kicinski 	struct iphdr *iph = buf;
2958888bf4fSJakub Kicinski 
2968888bf4fSJakub Kicinski 	if (proto == PF_INET6) {
2978888bf4fSJakub Kicinski 		memset(ip6h, 0, sizeof(*ip6h));
2988888bf4fSJakub Kicinski 
2998888bf4fSJakub Kicinski 		ip6h->version = 6;
3008888bf4fSJakub Kicinski 		ip6h->payload_len = htons(sizeof(struct tcphdr) + payload_len);
3018888bf4fSJakub Kicinski 		ip6h->nexthdr = protocol;
3028888bf4fSJakub Kicinski 		ip6h->hop_limit = 8;
3038888bf4fSJakub Kicinski 		if (inet_pton(AF_INET6, addr6_src, &ip6h->saddr) != 1)
3048888bf4fSJakub Kicinski 			error(1, errno, "inet_pton source ip6");
3058888bf4fSJakub Kicinski 		if (inet_pton(AF_INET6, addr6_dst, &ip6h->daddr) != 1)
3068888bf4fSJakub Kicinski 			error(1, errno, "inet_pton dest ip6");
3078888bf4fSJakub Kicinski 	} else if (proto == PF_INET) {
3088888bf4fSJakub Kicinski 		memset(iph, 0, sizeof(*iph));
3098888bf4fSJakub Kicinski 
3108888bf4fSJakub Kicinski 		iph->version = 4;
3118888bf4fSJakub Kicinski 		iph->ihl = 5;
3128888bf4fSJakub Kicinski 		iph->ttl = 8;
3138888bf4fSJakub Kicinski 		iph->protocol	= protocol;
3148888bf4fSJakub Kicinski 		iph->tot_len = htons(sizeof(struct tcphdr) +
3158888bf4fSJakub Kicinski 				payload_len + sizeof(struct iphdr));
3168888bf4fSJakub Kicinski 		iph->frag_off = htons(0x4000); /* DF = 1, MF = 0 */
3178888bf4fSJakub Kicinski 		if (inet_pton(AF_INET, addr4_src, &iph->saddr) != 1)
3188888bf4fSJakub Kicinski 			error(1, errno, "inet_pton source ip");
3198888bf4fSJakub Kicinski 		if (inet_pton(AF_INET, addr4_dst, &iph->daddr) != 1)
3208888bf4fSJakub Kicinski 			error(1, errno, "inet_pton dest ip");
3218888bf4fSJakub Kicinski 		iph->check = checksum_fold(buf, sizeof(struct iphdr), 0);
3228888bf4fSJakub Kicinski 	}
3238888bf4fSJakub Kicinski }
3248888bf4fSJakub Kicinski 
3258888bf4fSJakub Kicinski static void fill_transportlayer(void *buf, int seq_offset, int ack_offset,
3268888bf4fSJakub Kicinski 				int payload_len, int fin)
3278888bf4fSJakub Kicinski {
3288888bf4fSJakub Kicinski 	struct tcphdr *tcph = buf;
3298888bf4fSJakub Kicinski 
3308888bf4fSJakub Kicinski 	memset(tcph, 0, sizeof(*tcph));
3318888bf4fSJakub Kicinski 
3328888bf4fSJakub Kicinski 	tcph->source = htons(SPORT);
3338888bf4fSJakub Kicinski 	tcph->dest = htons(DPORT);
3348888bf4fSJakub Kicinski 	tcph->seq = ntohl(START_SEQ + seq_offset);
3358888bf4fSJakub Kicinski 	tcph->ack_seq = ntohl(START_ACK + ack_offset);
3368888bf4fSJakub Kicinski 	tcph->ack = 1;
3378888bf4fSJakub Kicinski 	tcph->fin = fin;
3388888bf4fSJakub Kicinski 	tcph->doff = 5;
3398888bf4fSJakub Kicinski 	tcph->window = htons(TCP_MAXWIN);
3408888bf4fSJakub Kicinski 	tcph->urg_ptr = 0;
3418888bf4fSJakub Kicinski 	tcph->check = tcp_checksum(tcph, payload_len);
3428888bf4fSJakub Kicinski }
3438888bf4fSJakub Kicinski 
3448888bf4fSJakub Kicinski static void write_packet(int fd, char *buf, int len, struct sockaddr_ll *daddr)
3458888bf4fSJakub Kicinski {
346f26d43acSJakub Kicinski 	char control[CMSG_SPACE(sizeof(uint64_t))];
347f26d43acSJakub Kicinski 	struct msghdr msg = {};
348f26d43acSJakub Kicinski 	struct iovec iov = {};
349f26d43acSJakub Kicinski 	struct cmsghdr *cm;
3508888bf4fSJakub Kicinski 	int ret = -1;
3518888bf4fSJakub Kicinski 
352f26d43acSJakub Kicinski 	iov.iov_base = buf;
353f26d43acSJakub Kicinski 	iov.iov_len = len;
354f26d43acSJakub Kicinski 
355f26d43acSJakub Kicinski 	msg.msg_iov = &iov;
356f26d43acSJakub Kicinski 	msg.msg_iovlen = 1;
357f26d43acSJakub Kicinski 	msg.msg_name = daddr;
358f26d43acSJakub Kicinski 	msg.msg_namelen = sizeof(*daddr);
359f26d43acSJakub Kicinski 
360f26d43acSJakub Kicinski 	if (txtime_ns) {
361f26d43acSJakub Kicinski 		memset(control, 0, sizeof(control));
362f26d43acSJakub Kicinski 		msg.msg_control = control;
363f26d43acSJakub Kicinski 		msg.msg_controllen = sizeof(control);
364f26d43acSJakub Kicinski 
365f26d43acSJakub Kicinski 		cm = CMSG_FIRSTHDR(&msg);
366f26d43acSJakub Kicinski 		cm->cmsg_level = SOL_SOCKET;
367f26d43acSJakub Kicinski 		cm->cmsg_type = SCM_TXTIME;
368f26d43acSJakub Kicinski 		cm->cmsg_len = CMSG_LEN(sizeof(uint64_t));
369f26d43acSJakub Kicinski 		memcpy(CMSG_DATA(cm), &txtime_ns, sizeof(txtime_ns));
370f26d43acSJakub Kicinski 	}
371f26d43acSJakub Kicinski 
372f26d43acSJakub Kicinski 	ret = sendmsg(fd, &msg, 0);
3738888bf4fSJakub Kicinski 	if (ret == -1)
374f26d43acSJakub Kicinski 		error(1, errno, "sendmsg failure");
3758888bf4fSJakub Kicinski 	if (ret != len)
376f26d43acSJakub Kicinski 		error(1, 0, "sendmsg wrong length: %d vs %d", ret, len);
3778888bf4fSJakub Kicinski }
3788888bf4fSJakub Kicinski 
3798888bf4fSJakub Kicinski static void create_packet(void *buf, int seq_offset, int ack_offset,
3808888bf4fSJakub Kicinski 			  int payload_len, int fin)
3818888bf4fSJakub Kicinski {
3828888bf4fSJakub Kicinski 	memset(buf, 0, total_hdr_len);
3838888bf4fSJakub Kicinski 	memset(buf + total_hdr_len, 'a', payload_len);
3848888bf4fSJakub Kicinski 
3858888bf4fSJakub Kicinski 	fill_transportlayer(buf + tcp_offset, seq_offset, ack_offset,
3868888bf4fSJakub Kicinski 			    payload_len, fin);
3878888bf4fSJakub Kicinski 
3888888bf4fSJakub Kicinski 	if (ipip) {
3898888bf4fSJakub Kicinski 		fill_networklayer(buf + ETH_HLEN, payload_len + sizeof(struct iphdr),
3908888bf4fSJakub Kicinski 				  IPPROTO_IPIP);
3918888bf4fSJakub Kicinski 		fill_networklayer(buf + ETH_HLEN + sizeof(struct iphdr),
3928888bf4fSJakub Kicinski 				  payload_len, IPPROTO_TCP);
3938888bf4fSJakub Kicinski 	} else {
3948888bf4fSJakub Kicinski 		fill_networklayer(buf + ETH_HLEN, payload_len, IPPROTO_TCP);
3958888bf4fSJakub Kicinski 	}
3968888bf4fSJakub Kicinski 
3978888bf4fSJakub Kicinski 	fill_datalinklayer(buf);
3988888bf4fSJakub Kicinski }
3998888bf4fSJakub Kicinski 
400ba5d4128SJakub Kicinski static void create_capacity_packet(void *buf, int flow_id, int pkt_idx, int psh)
401ba5d4128SJakub Kicinski {
402ba5d4128SJakub Kicinski 	int seq_offset = pkt_idx * CAPACITY_PAYLOAD_LEN;
403ba5d4128SJakub Kicinski 	struct tcphdr *tcph;
404ba5d4128SJakub Kicinski 
405ba5d4128SJakub Kicinski 	create_packet(buf, seq_offset, 0, CAPACITY_PAYLOAD_LEN, 0);
406ba5d4128SJakub Kicinski 
407ba5d4128SJakub Kicinski 	/* Customize for this flow id */
408ba5d4128SJakub Kicinski 	memset(buf + total_hdr_len, 'a' + flow_id, CAPACITY_PAYLOAD_LEN);
409ba5d4128SJakub Kicinski 
410ba5d4128SJakub Kicinski 	tcph = buf + tcp_offset;
411ba5d4128SJakub Kicinski 	tcph->source = htons(SPORT + flow_id);
412ba5d4128SJakub Kicinski 	tcph->psh = psh;
413ba5d4128SJakub Kicinski 	tcph->check = 0;
414ba5d4128SJakub Kicinski 	tcph->check = tcp_checksum(tcph, CAPACITY_PAYLOAD_LEN);
415ba5d4128SJakub Kicinski }
416ba5d4128SJakub Kicinski 
417ba5d4128SJakub Kicinski /* Send a capacity test, 2 packets per flow, all first packets then all second:
418ba5d4128SJakub Kicinski  *  A1 B1 C1 D1 ... A2 B2 C2 D2 ...
419ba5d4128SJakub Kicinski  */
420ba5d4128SJakub Kicinski static void send_capacity(int fd, struct sockaddr_ll *daddr)
421ba5d4128SJakub Kicinski {
422ba5d4128SJakub Kicinski 	static char buf[MAX_HDR_LEN + CAPACITY_PAYLOAD_LEN];
423ba5d4128SJakub Kicinski 	int pkt_size = total_hdr_len + CAPACITY_PAYLOAD_LEN;
424ba5d4128SJakub Kicinski 	int i;
425ba5d4128SJakub Kicinski 
426ba5d4128SJakub Kicinski 	/* Send first packet of each flow (no PSH) */
427ba5d4128SJakub Kicinski 	for (i = 0; i < num_flows; i++) {
428ba5d4128SJakub Kicinski 		create_capacity_packet(buf, i, 0, 0);
429ba5d4128SJakub Kicinski 		write_packet(fd, buf, pkt_size, daddr);
430ba5d4128SJakub Kicinski 	}
431ba5d4128SJakub Kicinski 
432ba5d4128SJakub Kicinski 	/* Send second packet of each flow (with PSH to flush) */
433ba5d4128SJakub Kicinski 	for (i = 0; i < num_flows; i++) {
434ba5d4128SJakub Kicinski 		create_capacity_packet(buf, i, 1, 1);
435ba5d4128SJakub Kicinski 		write_packet(fd, buf, pkt_size, daddr);
436ba5d4128SJakub Kicinski 	}
437ba5d4128SJakub Kicinski }
438ba5d4128SJakub Kicinski 
4398888bf4fSJakub Kicinski #ifndef TH_CWR
4408888bf4fSJakub Kicinski #define TH_CWR 0x80
4418888bf4fSJakub Kicinski #endif
4428888bf4fSJakub Kicinski static void set_flags(struct tcphdr *tcph, int payload_len, int psh, int syn,
4438888bf4fSJakub Kicinski 		      int rst, int urg, int cwr)
4448888bf4fSJakub Kicinski {
4458888bf4fSJakub Kicinski 	tcph->psh = psh;
4468888bf4fSJakub Kicinski 	tcph->syn = syn;
4478888bf4fSJakub Kicinski 	tcph->rst = rst;
4488888bf4fSJakub Kicinski 	tcph->urg = urg;
4498888bf4fSJakub Kicinski 	if (cwr)
4508888bf4fSJakub Kicinski 		tcph->th_flags |= TH_CWR;
4518888bf4fSJakub Kicinski 	else
4528888bf4fSJakub Kicinski 		tcph->th_flags &= ~TH_CWR;
4538888bf4fSJakub Kicinski 	tcph->check = 0;
4548888bf4fSJakub Kicinski 	tcph->check = tcp_checksum(tcph, payload_len);
4558888bf4fSJakub Kicinski }
4568888bf4fSJakub Kicinski 
4578888bf4fSJakub Kicinski /* send extra flags of the (NUM_PACKETS / 2) and (NUM_PACKETS / 2 - 1)
4588888bf4fSJakub Kicinski  * pkts, not first and not last pkt
4598888bf4fSJakub Kicinski  */
4608888bf4fSJakub Kicinski static void send_flags(int fd, struct sockaddr_ll *daddr, int psh, int syn,
4618888bf4fSJakub Kicinski 		       int rst, int urg, int cwr)
4628888bf4fSJakub Kicinski {
4638888bf4fSJakub Kicinski 	static char flag_buf[2][MAX_HDR_LEN + PAYLOAD_LEN];
4648888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
4658888bf4fSJakub Kicinski 	int payload_len, pkt_size, i;
4668888bf4fSJakub Kicinski 	struct tcphdr *tcph;
4678888bf4fSJakub Kicinski 	int flag[2];
4688888bf4fSJakub Kicinski 
4698888bf4fSJakub Kicinski 	payload_len = PAYLOAD_LEN * (psh || cwr);
4708888bf4fSJakub Kicinski 	pkt_size = total_hdr_len + payload_len;
4718888bf4fSJakub Kicinski 	flag[0] = NUM_PACKETS / 2;
4728888bf4fSJakub Kicinski 	flag[1] = NUM_PACKETS / 2 - 1;
4738888bf4fSJakub Kicinski 
4748888bf4fSJakub Kicinski 	/* Create and configure packets with flags
4758888bf4fSJakub Kicinski 	 */
4768888bf4fSJakub Kicinski 	for (i = 0; i < 2; i++) {
4778888bf4fSJakub Kicinski 		if (flag[i] > 0) {
4788888bf4fSJakub Kicinski 			create_packet(flag_buf[i], flag[i] * payload_len, 0,
4798888bf4fSJakub Kicinski 				      payload_len, 0);
4808888bf4fSJakub Kicinski 			tcph = (struct tcphdr *)(flag_buf[i] + tcp_offset);
4818888bf4fSJakub Kicinski 			set_flags(tcph, payload_len, psh, syn, rst, urg, cwr);
4828888bf4fSJakub Kicinski 		}
4838888bf4fSJakub Kicinski 	}
4848888bf4fSJakub Kicinski 
4858888bf4fSJakub Kicinski 	for (i = 0; i < NUM_PACKETS + 1; i++) {
4868888bf4fSJakub Kicinski 		if (i == flag[0]) {
4878888bf4fSJakub Kicinski 			write_packet(fd, flag_buf[0], pkt_size, daddr);
4888888bf4fSJakub Kicinski 			continue;
4898888bf4fSJakub Kicinski 		} else if (i == flag[1] && cwr) {
4908888bf4fSJakub Kicinski 			write_packet(fd, flag_buf[1], pkt_size, daddr);
4918888bf4fSJakub Kicinski 			continue;
4928888bf4fSJakub Kicinski 		}
4938888bf4fSJakub Kicinski 		create_packet(buf, i * PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
4948888bf4fSJakub Kicinski 		write_packet(fd, buf, total_hdr_len + PAYLOAD_LEN, daddr);
4958888bf4fSJakub Kicinski 	}
4968888bf4fSJakub Kicinski }
4978888bf4fSJakub Kicinski 
4988888bf4fSJakub Kicinski /* Test for data of same length, smaller than previous
4998888bf4fSJakub Kicinski  * and of different lengths
5008888bf4fSJakub Kicinski  */
5018888bf4fSJakub Kicinski static void send_data_pkts(int fd, struct sockaddr_ll *daddr,
5028888bf4fSJakub Kicinski 			   int payload_len1, int payload_len2)
5038888bf4fSJakub Kicinski {
5048888bf4fSJakub Kicinski 	static char buf[ETH_HLEN + IP_MAXPACKET];
5058888bf4fSJakub Kicinski 
5068888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, payload_len1, 0);
5078888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len + payload_len1, daddr);
5088888bf4fSJakub Kicinski 	create_packet(buf, payload_len1, 0, payload_len2, 0);
5098888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len + payload_len2, daddr);
5108888bf4fSJakub Kicinski }
5118888bf4fSJakub Kicinski 
5128888bf4fSJakub Kicinski /* If incoming segments make tracked segment length exceed
5138888bf4fSJakub Kicinski  * legal IP datagram length, do not coalesce
5148888bf4fSJakub Kicinski  */
5158888bf4fSJakub Kicinski static void send_large(int fd, struct sockaddr_ll *daddr, int remainder)
5168888bf4fSJakub Kicinski {
5178888bf4fSJakub Kicinski 	static char pkts[NUM_LARGE_PKT][TOTAL_HDR_LEN + MSS];
5188888bf4fSJakub Kicinski 	static char last[TOTAL_HDR_LEN + MSS];
5198888bf4fSJakub Kicinski 	static char new_seg[TOTAL_HDR_LEN + MSS];
5208888bf4fSJakub Kicinski 	int i;
5218888bf4fSJakub Kicinski 
5228888bf4fSJakub Kicinski 	for (i = 0; i < NUM_LARGE_PKT; i++)
5238888bf4fSJakub Kicinski 		create_packet(pkts[i], i * MSS, 0, MSS, 0);
5248888bf4fSJakub Kicinski 	create_packet(last, NUM_LARGE_PKT * MSS, 0, remainder, 0);
5258888bf4fSJakub Kicinski 	create_packet(new_seg, (NUM_LARGE_PKT + 1) * MSS, 0, remainder, 0);
5268888bf4fSJakub Kicinski 
5278888bf4fSJakub Kicinski 	for (i = 0; i < NUM_LARGE_PKT; i++)
5288888bf4fSJakub Kicinski 		write_packet(fd, pkts[i], total_hdr_len + MSS, daddr);
5298888bf4fSJakub Kicinski 	write_packet(fd, last, total_hdr_len + remainder, daddr);
5308888bf4fSJakub Kicinski 	write_packet(fd, new_seg, total_hdr_len + remainder, daddr);
5318888bf4fSJakub Kicinski }
5328888bf4fSJakub Kicinski 
5338888bf4fSJakub Kicinski /* Pure acks and dup acks don't coalesce */
5348888bf4fSJakub Kicinski static void send_ack(int fd, struct sockaddr_ll *daddr)
5358888bf4fSJakub Kicinski {
5368888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN];
5378888bf4fSJakub Kicinski 
5388888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, 0, 0);
5398888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len, daddr);
5408888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len, daddr);
5418888bf4fSJakub Kicinski 	create_packet(buf, 0, 1, 0, 0);
5428888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len, daddr);
5438888bf4fSJakub Kicinski }
5448888bf4fSJakub Kicinski 
5458888bf4fSJakub Kicinski static void recompute_packet(char *buf, char *no_ext, int extlen)
5468888bf4fSJakub Kicinski {
5478888bf4fSJakub Kicinski 	struct tcphdr *tcphdr = (struct tcphdr *)(buf + tcp_offset);
5488888bf4fSJakub Kicinski 	struct ipv6hdr *ip6h = (struct ipv6hdr *)(buf + ETH_HLEN);
5498888bf4fSJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buf + ETH_HLEN);
5508888bf4fSJakub Kicinski 
5518888bf4fSJakub Kicinski 	memmove(buf, no_ext, total_hdr_len);
5528888bf4fSJakub Kicinski 	memmove(buf + total_hdr_len + extlen,
5538888bf4fSJakub Kicinski 		no_ext + total_hdr_len, PAYLOAD_LEN);
5548888bf4fSJakub Kicinski 
5558888bf4fSJakub Kicinski 	tcphdr->doff = tcphdr->doff + (extlen / 4);
5568888bf4fSJakub Kicinski 	tcphdr->check = 0;
5578888bf4fSJakub Kicinski 	tcphdr->check = tcp_checksum(tcphdr, PAYLOAD_LEN + extlen);
5588888bf4fSJakub Kicinski 	if (proto == PF_INET) {
5598888bf4fSJakub Kicinski 		iph->tot_len = htons(ntohs(iph->tot_len) + extlen);
5608888bf4fSJakub Kicinski 		iph->check = 0;
5618888bf4fSJakub Kicinski 		iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
5628888bf4fSJakub Kicinski 
5638888bf4fSJakub Kicinski 		if (ipip) {
5648888bf4fSJakub Kicinski 			iph += 1;
5658888bf4fSJakub Kicinski 			iph->tot_len = htons(ntohs(iph->tot_len) + extlen);
5668888bf4fSJakub Kicinski 			iph->check = 0;
5678888bf4fSJakub Kicinski 			iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
5688888bf4fSJakub Kicinski 		}
5698888bf4fSJakub Kicinski 	} else {
5708888bf4fSJakub Kicinski 		ip6h->payload_len = htons(ntohs(ip6h->payload_len) + extlen);
5718888bf4fSJakub Kicinski 	}
5728888bf4fSJakub Kicinski }
5738888bf4fSJakub Kicinski 
5748888bf4fSJakub Kicinski static void tcp_write_options(char *buf, int kind, int ts)
5758888bf4fSJakub Kicinski {
5768888bf4fSJakub Kicinski 	struct tcp_option_ts {
5778888bf4fSJakub Kicinski 		uint8_t kind;
5788888bf4fSJakub Kicinski 		uint8_t len;
5798888bf4fSJakub Kicinski 		uint32_t tsval;
5808888bf4fSJakub Kicinski 		uint32_t tsecr;
5818888bf4fSJakub Kicinski 	} *opt_ts = (void *)buf;
5828888bf4fSJakub Kicinski 	struct tcp_option_window {
5838888bf4fSJakub Kicinski 		uint8_t kind;
5848888bf4fSJakub Kicinski 		uint8_t len;
5858888bf4fSJakub Kicinski 		uint8_t shift;
5868888bf4fSJakub Kicinski 	} *opt_window = (void *)buf;
5878888bf4fSJakub Kicinski 
5888888bf4fSJakub Kicinski 	switch (kind) {
5898888bf4fSJakub Kicinski 	case TCPOPT_NOP:
5908888bf4fSJakub Kicinski 		buf[0] = TCPOPT_NOP;
5918888bf4fSJakub Kicinski 		break;
5928888bf4fSJakub Kicinski 	case TCPOPT_WINDOW:
5938888bf4fSJakub Kicinski 		memset(opt_window, 0, sizeof(struct tcp_option_window));
5948888bf4fSJakub Kicinski 		opt_window->kind = TCPOPT_WINDOW;
5958888bf4fSJakub Kicinski 		opt_window->len = TCPOLEN_WINDOW;
5968888bf4fSJakub Kicinski 		opt_window->shift = 0;
5978888bf4fSJakub Kicinski 		break;
5988888bf4fSJakub Kicinski 	case TCPOPT_TIMESTAMP:
5998888bf4fSJakub Kicinski 		memset(opt_ts, 0, sizeof(struct tcp_option_ts));
6008888bf4fSJakub Kicinski 		opt_ts->kind = TCPOPT_TIMESTAMP;
6018888bf4fSJakub Kicinski 		opt_ts->len = TCPOLEN_TIMESTAMP;
6028888bf4fSJakub Kicinski 		opt_ts->tsval = ts;
6038888bf4fSJakub Kicinski 		opt_ts->tsecr = 0;
6048888bf4fSJakub Kicinski 		break;
6058888bf4fSJakub Kicinski 	default:
6068888bf4fSJakub Kicinski 		error(1, 0, "unimplemented TCP option");
6078888bf4fSJakub Kicinski 		break;
6088888bf4fSJakub Kicinski 	}
6098888bf4fSJakub Kicinski }
6108888bf4fSJakub Kicinski 
6118888bf4fSJakub Kicinski /* TCP with options is always a permutation of {TS, NOP, NOP}.
6128888bf4fSJakub Kicinski  * Implement different orders to verify coalescing stops.
6138888bf4fSJakub Kicinski  */
6148888bf4fSJakub Kicinski static void add_standard_tcp_options(char *buf, char *no_ext, int ts, int order)
6158888bf4fSJakub Kicinski {
6168888bf4fSJakub Kicinski 	switch (order) {
6178888bf4fSJakub Kicinski 	case 0:
6188888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len, TCPOPT_NOP, 0);
6198888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len + 1, TCPOPT_NOP, 0);
6208888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len + 2 /* two NOP opts */,
6218888bf4fSJakub Kicinski 				  TCPOPT_TIMESTAMP, ts);
6228888bf4fSJakub Kicinski 		break;
6238888bf4fSJakub Kicinski 	case 1:
6248888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len, TCPOPT_NOP, 0);
6258888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len + 1,
6268888bf4fSJakub Kicinski 				  TCPOPT_TIMESTAMP, ts);
6278888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len + 1 + TCPOLEN_TIMESTAMP,
6288888bf4fSJakub Kicinski 				  TCPOPT_NOP, 0);
6298888bf4fSJakub Kicinski 		break;
6308888bf4fSJakub Kicinski 	case 2:
6318888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len, TCPOPT_TIMESTAMP, ts);
6328888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len + TCPOLEN_TIMESTAMP + 1,
6338888bf4fSJakub Kicinski 				  TCPOPT_NOP, 0);
6348888bf4fSJakub Kicinski 		tcp_write_options(buf + total_hdr_len + TCPOLEN_TIMESTAMP + 2,
6358888bf4fSJakub Kicinski 				  TCPOPT_NOP, 0);
6368888bf4fSJakub Kicinski 		break;
6378888bf4fSJakub Kicinski 	default:
6388888bf4fSJakub Kicinski 		error(1, 0, "unknown order");
6398888bf4fSJakub Kicinski 		break;
6408888bf4fSJakub Kicinski 	}
6418888bf4fSJakub Kicinski 	recompute_packet(buf, no_ext, TCPOLEN_TSTAMP_APPA);
6428888bf4fSJakub Kicinski }
6438888bf4fSJakub Kicinski 
6448888bf4fSJakub Kicinski /* Packets with invalid checksum don't coalesce. */
6458888bf4fSJakub Kicinski static void send_changed_checksum(int fd, struct sockaddr_ll *daddr)
6468888bf4fSJakub Kicinski {
6478888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
6488888bf4fSJakub Kicinski 	struct tcphdr *tcph = (struct tcphdr *)(buf + tcp_offset);
6498888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN;
6508888bf4fSJakub Kicinski 
6518888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
6528888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
6538888bf4fSJakub Kicinski 
6548888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
6558888bf4fSJakub Kicinski 	tcph->check = tcph->check - 1;
6568888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
6578888bf4fSJakub Kicinski }
6588888bf4fSJakub Kicinski 
6598888bf4fSJakub Kicinski  /* Packets with non-consecutive sequence number don't coalesce.*/
6608888bf4fSJakub Kicinski static void send_changed_seq(int fd, struct sockaddr_ll *daddr)
6618888bf4fSJakub Kicinski {
6628888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
6638888bf4fSJakub Kicinski 	struct tcphdr *tcph = (struct tcphdr *)(buf + tcp_offset);
6648888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN;
6658888bf4fSJakub Kicinski 
6668888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
6678888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
6688888bf4fSJakub Kicinski 
6698888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
6708888bf4fSJakub Kicinski 	tcph->seq = ntohl(htonl(tcph->seq) + 1);
6718888bf4fSJakub Kicinski 	tcph->check = 0;
6728888bf4fSJakub Kicinski 	tcph->check = tcp_checksum(tcph, PAYLOAD_LEN);
6738888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
6748888bf4fSJakub Kicinski }
6758888bf4fSJakub Kicinski 
6768888bf4fSJakub Kicinski  /* Packet with different timestamp option or different timestamps
6778888bf4fSJakub Kicinski   * don't coalesce.
6788888bf4fSJakub Kicinski   */
6798888bf4fSJakub Kicinski static void send_changed_ts(int fd, struct sockaddr_ll *daddr)
6808888bf4fSJakub Kicinski {
6818888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
6828888bf4fSJakub Kicinski 	static char extpkt[sizeof(buf) + TCPOLEN_TSTAMP_APPA];
6838888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN + TCPOLEN_TSTAMP_APPA;
6848888bf4fSJakub Kicinski 
6858888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
6868888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt, buf, 0, 0);
6878888bf4fSJakub Kicinski 	write_packet(fd, extpkt, pkt_size, daddr);
6888888bf4fSJakub Kicinski 
6898888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
6908888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt, buf, 0, 0);
6918888bf4fSJakub Kicinski 	write_packet(fd, extpkt, pkt_size, daddr);
6928888bf4fSJakub Kicinski 
6938888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 2, 0, PAYLOAD_LEN, 0);
6948888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt, buf, 100, 0);
6958888bf4fSJakub Kicinski 	write_packet(fd, extpkt, pkt_size, daddr);
6968888bf4fSJakub Kicinski 
6978888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 3, 0, PAYLOAD_LEN, 0);
6988888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt, buf, 100, 1);
6998888bf4fSJakub Kicinski 	write_packet(fd, extpkt, pkt_size, daddr);
7008888bf4fSJakub Kicinski 
7018888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 4, 0, PAYLOAD_LEN, 0);
7028888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt, buf, 100, 2);
7038888bf4fSJakub Kicinski 	write_packet(fd, extpkt, pkt_size, daddr);
7048888bf4fSJakub Kicinski }
7058888bf4fSJakub Kicinski 
7068888bf4fSJakub Kicinski /* Packet with different tcp options don't coalesce. */
7078888bf4fSJakub Kicinski static void send_diff_opt(int fd, struct sockaddr_ll *daddr)
7088888bf4fSJakub Kicinski {
7098888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
7108888bf4fSJakub Kicinski 	static char extpkt1[sizeof(buf) + TCPOLEN_TSTAMP_APPA];
7118888bf4fSJakub Kicinski 	static char extpkt2[sizeof(buf) + TCPOLEN_MAXSEG];
7128888bf4fSJakub Kicinski 	int extpkt1_size = total_hdr_len + PAYLOAD_LEN + TCPOLEN_TSTAMP_APPA;
7138888bf4fSJakub Kicinski 	int extpkt2_size = total_hdr_len + PAYLOAD_LEN + TCPOLEN_MAXSEG;
7148888bf4fSJakub Kicinski 
7158888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
7168888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt1, buf, 0, 0);
7178888bf4fSJakub Kicinski 	write_packet(fd, extpkt1, extpkt1_size, daddr);
7188888bf4fSJakub Kicinski 
7198888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
7208888bf4fSJakub Kicinski 	add_standard_tcp_options(extpkt1, buf, 0, 0);
7218888bf4fSJakub Kicinski 	write_packet(fd, extpkt1, extpkt1_size, daddr);
7228888bf4fSJakub Kicinski 
7238888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 2, 0, PAYLOAD_LEN, 0);
7248888bf4fSJakub Kicinski 	tcp_write_options(extpkt2 + MAX_HDR_LEN, TCPOPT_NOP, 0);
7258888bf4fSJakub Kicinski 	tcp_write_options(extpkt2 + MAX_HDR_LEN + 1, TCPOPT_WINDOW, 0);
7268888bf4fSJakub Kicinski 	recompute_packet(extpkt2, buf, TCPOLEN_WINDOW + 1);
7278888bf4fSJakub Kicinski 	write_packet(fd, extpkt2, extpkt2_size, daddr);
7288888bf4fSJakub Kicinski }
7298888bf4fSJakub Kicinski 
7308888bf4fSJakub Kicinski static void add_ipv4_ts_option(void *buf, void *optpkt)
7318888bf4fSJakub Kicinski {
7328888bf4fSJakub Kicinski 	struct ip_timestamp *ts = (struct ip_timestamp *)(optpkt + tcp_offset);
7338888bf4fSJakub Kicinski 	int optlen = sizeof(struct ip_timestamp);
7348888bf4fSJakub Kicinski 	struct iphdr *iph;
7358888bf4fSJakub Kicinski 
7368888bf4fSJakub Kicinski 	if (optlen % 4)
7378888bf4fSJakub Kicinski 		error(1, 0, "ipv4 timestamp length is not a multiple of 4B");
7388888bf4fSJakub Kicinski 
7398888bf4fSJakub Kicinski 	ts->ipt_code = IPOPT_TS;
7408888bf4fSJakub Kicinski 	ts->ipt_len = optlen;
7418888bf4fSJakub Kicinski 	ts->ipt_ptr = 5;
7428888bf4fSJakub Kicinski 	ts->ipt_flg = IPOPT_TS_TSONLY;
7438888bf4fSJakub Kicinski 
7448888bf4fSJakub Kicinski 	memcpy(optpkt, buf, tcp_offset);
7458888bf4fSJakub Kicinski 	memcpy(optpkt + tcp_offset + optlen, buf + tcp_offset,
7468888bf4fSJakub Kicinski 	       sizeof(struct tcphdr) + PAYLOAD_LEN);
7478888bf4fSJakub Kicinski 
7488888bf4fSJakub Kicinski 	iph = (struct iphdr *)(optpkt + ETH_HLEN);
7498888bf4fSJakub Kicinski 	iph->ihl = 5 + (optlen / 4);
7508888bf4fSJakub Kicinski 	iph->tot_len = htons(ntohs(iph->tot_len) + optlen);
7518888bf4fSJakub Kicinski 	iph->check = 0;
7528888bf4fSJakub Kicinski 	iph->check = checksum_fold(iph, sizeof(struct iphdr) + optlen, 0);
7538888bf4fSJakub Kicinski }
7548888bf4fSJakub Kicinski 
7558888bf4fSJakub Kicinski static void add_ipv6_exthdr(void *buf, void *optpkt, __u8 exthdr_type, char *ext_payload)
7568888bf4fSJakub Kicinski {
7578888bf4fSJakub Kicinski 	struct ipv6_opt_hdr *exthdr = (struct ipv6_opt_hdr *)(optpkt + tcp_offset);
7588888bf4fSJakub Kicinski 	struct ipv6hdr *iph = (struct ipv6hdr *)(optpkt + ETH_HLEN);
7598888bf4fSJakub Kicinski 	char *exthdr_payload_start = (char *)(exthdr + 1);
7608888bf4fSJakub Kicinski 
7618888bf4fSJakub Kicinski 	exthdr->hdrlen = 0;
7628888bf4fSJakub Kicinski 	exthdr->nexthdr = IPPROTO_TCP;
7638888bf4fSJakub Kicinski 
7648888bf4fSJakub Kicinski 	memcpy(exthdr_payload_start, ext_payload, MIN_EXTHDR_SIZE - sizeof(*exthdr));
7658888bf4fSJakub Kicinski 
7668888bf4fSJakub Kicinski 	memcpy(optpkt, buf, tcp_offset);
7678888bf4fSJakub Kicinski 	memcpy(optpkt + tcp_offset + MIN_EXTHDR_SIZE, buf + tcp_offset,
7688888bf4fSJakub Kicinski 		sizeof(struct tcphdr) + PAYLOAD_LEN);
7698888bf4fSJakub Kicinski 
7708888bf4fSJakub Kicinski 	iph->nexthdr = exthdr_type;
7718888bf4fSJakub Kicinski 	iph->payload_len = htons(ntohs(iph->payload_len) + MIN_EXTHDR_SIZE);
7728888bf4fSJakub Kicinski }
7738888bf4fSJakub Kicinski 
7748888bf4fSJakub Kicinski static void fix_ip4_checksum(struct iphdr *iph)
7758888bf4fSJakub Kicinski {
7768888bf4fSJakub Kicinski 	iph->check = 0;
7778888bf4fSJakub Kicinski 	iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
7788888bf4fSJakub Kicinski }
7798888bf4fSJakub Kicinski 
7808888bf4fSJakub Kicinski static void send_flush_id_case(int fd, struct sockaddr_ll *daddr,
7818888bf4fSJakub Kicinski 			       enum flush_id_case tcase)
7828888bf4fSJakub Kicinski {
7838888bf4fSJakub Kicinski 	static char buf1[MAX_HDR_LEN + PAYLOAD_LEN];
7848888bf4fSJakub Kicinski 	static char buf2[MAX_HDR_LEN + PAYLOAD_LEN];
7858888bf4fSJakub Kicinski 	static char buf3[MAX_HDR_LEN + PAYLOAD_LEN];
7868888bf4fSJakub Kicinski 	bool send_three = false;
7878888bf4fSJakub Kicinski 	struct iphdr *iph1;
7888888bf4fSJakub Kicinski 	struct iphdr *iph2;
7898888bf4fSJakub Kicinski 	struct iphdr *iph3;
7908888bf4fSJakub Kicinski 
7918888bf4fSJakub Kicinski 	iph1 = (struct iphdr *)(buf1 + ETH_HLEN);
7928888bf4fSJakub Kicinski 	iph2 = (struct iphdr *)(buf2 + ETH_HLEN);
7938888bf4fSJakub Kicinski 	iph3 = (struct iphdr *)(buf3 + ETH_HLEN);
7948888bf4fSJakub Kicinski 
7958888bf4fSJakub Kicinski 	create_packet(buf1, 0, 0, PAYLOAD_LEN, 0);
7968888bf4fSJakub Kicinski 	create_packet(buf2, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
7978888bf4fSJakub Kicinski 	create_packet(buf3, PAYLOAD_LEN * 2, 0, PAYLOAD_LEN, 0);
7988888bf4fSJakub Kicinski 
7998888bf4fSJakub Kicinski 	switch (tcase) {
8008888bf4fSJakub Kicinski 	case FLUSH_ID_DF1_INC: /* DF=1, Incrementing - should coalesce */
8018888bf4fSJakub Kicinski 		iph1->frag_off |= htons(IP_DF);
8028888bf4fSJakub Kicinski 		iph1->id = htons(8);
8038888bf4fSJakub Kicinski 
8048888bf4fSJakub Kicinski 		iph2->frag_off |= htons(IP_DF);
8058888bf4fSJakub Kicinski 		iph2->id = htons(9);
8068888bf4fSJakub Kicinski 		break;
8078888bf4fSJakub Kicinski 
8088888bf4fSJakub Kicinski 	case FLUSH_ID_DF1_FIXED: /* DF=1, Fixed - should coalesce */
8098888bf4fSJakub Kicinski 		iph1->frag_off |= htons(IP_DF);
8108888bf4fSJakub Kicinski 		iph1->id = htons(8);
8118888bf4fSJakub Kicinski 
8128888bf4fSJakub Kicinski 		iph2->frag_off |= htons(IP_DF);
8138888bf4fSJakub Kicinski 		iph2->id = htons(8);
8148888bf4fSJakub Kicinski 		break;
8158888bf4fSJakub Kicinski 
8168888bf4fSJakub Kicinski 	case FLUSH_ID_DF0_INC: /* DF=0, Incrementing - should coalesce */
8178888bf4fSJakub Kicinski 		iph1->frag_off &= ~htons(IP_DF);
8188888bf4fSJakub Kicinski 		iph1->id = htons(8);
8198888bf4fSJakub Kicinski 
8208888bf4fSJakub Kicinski 		iph2->frag_off &= ~htons(IP_DF);
8218888bf4fSJakub Kicinski 		iph2->id = htons(9);
8228888bf4fSJakub Kicinski 		break;
8238888bf4fSJakub Kicinski 
8248888bf4fSJakub Kicinski 	case FLUSH_ID_DF0_FIXED: /* DF=0, Fixed - should coalesce */
8258888bf4fSJakub Kicinski 		iph1->frag_off &= ~htons(IP_DF);
8268888bf4fSJakub Kicinski 		iph1->id = htons(8);
8278888bf4fSJakub Kicinski 
8288888bf4fSJakub Kicinski 		iph2->frag_off &= ~htons(IP_DF);
8298888bf4fSJakub Kicinski 		iph2->id = htons(8);
8308888bf4fSJakub Kicinski 		break;
8318888bf4fSJakub Kicinski 
8328888bf4fSJakub Kicinski 	case FLUSH_ID_DF1_INC_FIXED: /* DF=1, two packets incrementing, and
8338888bf4fSJakub Kicinski 				      * one fixed - should coalesce only the
8348888bf4fSJakub Kicinski 				      * first two packets
8358888bf4fSJakub Kicinski 				      */
8368888bf4fSJakub Kicinski 		iph1->frag_off |= htons(IP_DF);
8378888bf4fSJakub Kicinski 		iph1->id = htons(8);
8388888bf4fSJakub Kicinski 
8398888bf4fSJakub Kicinski 		iph2->frag_off |= htons(IP_DF);
8408888bf4fSJakub Kicinski 		iph2->id = htons(9);
8418888bf4fSJakub Kicinski 
8428888bf4fSJakub Kicinski 		iph3->frag_off |= htons(IP_DF);
8438888bf4fSJakub Kicinski 		iph3->id = htons(9);
8448888bf4fSJakub Kicinski 		send_three = true;
8458888bf4fSJakub Kicinski 		break;
8468888bf4fSJakub Kicinski 
8478888bf4fSJakub Kicinski 	case FLUSH_ID_DF1_FIXED_INC: /* DF=1, two packets fixed, and one
8488888bf4fSJakub Kicinski 				      * incrementing - should coalesce only
8498888bf4fSJakub Kicinski 				      * the first two packets
8508888bf4fSJakub Kicinski 				      */
8518888bf4fSJakub Kicinski 		iph1->frag_off |= htons(IP_DF);
8528888bf4fSJakub Kicinski 		iph1->id = htons(8);
8538888bf4fSJakub Kicinski 
8548888bf4fSJakub Kicinski 		iph2->frag_off |= htons(IP_DF);
8558888bf4fSJakub Kicinski 		iph2->id = htons(8);
8568888bf4fSJakub Kicinski 
8578888bf4fSJakub Kicinski 		iph3->frag_off |= htons(IP_DF);
8588888bf4fSJakub Kicinski 		iph3->id = htons(9);
8598888bf4fSJakub Kicinski 		send_three = true;
8608888bf4fSJakub Kicinski 		break;
8618888bf4fSJakub Kicinski 	}
8628888bf4fSJakub Kicinski 
8638888bf4fSJakub Kicinski 	fix_ip4_checksum(iph1);
8648888bf4fSJakub Kicinski 	fix_ip4_checksum(iph2);
8658888bf4fSJakub Kicinski 	write_packet(fd, buf1, total_hdr_len + PAYLOAD_LEN, daddr);
8668888bf4fSJakub Kicinski 	write_packet(fd, buf2, total_hdr_len + PAYLOAD_LEN, daddr);
8678888bf4fSJakub Kicinski 
8688888bf4fSJakub Kicinski 	if (send_three) {
8698888bf4fSJakub Kicinski 		fix_ip4_checksum(iph3);
8708888bf4fSJakub Kicinski 		write_packet(fd, buf3, total_hdr_len + PAYLOAD_LEN, daddr);
8718888bf4fSJakub Kicinski 	}
8728888bf4fSJakub Kicinski }
8738888bf4fSJakub Kicinski 
8748888bf4fSJakub Kicinski static void send_ipv6_exthdr(int fd, struct sockaddr_ll *daddr, char *ext_data1, char *ext_data2)
8758888bf4fSJakub Kicinski {
8768888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
8778888bf4fSJakub Kicinski 	static char exthdr_pck[sizeof(buf) + MIN_EXTHDR_SIZE];
8788888bf4fSJakub Kicinski 
8798888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
8808888bf4fSJakub Kicinski 	add_ipv6_exthdr(buf, exthdr_pck, IPPROTO_DSTOPTS, ext_data1);
8818888bf4fSJakub Kicinski 	write_packet(fd, exthdr_pck, total_hdr_len + PAYLOAD_LEN + MIN_EXTHDR_SIZE, daddr);
8828888bf4fSJakub Kicinski 
8838888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 1, 0, PAYLOAD_LEN, 0);
8848888bf4fSJakub Kicinski 	add_ipv6_exthdr(buf, exthdr_pck, IPPROTO_DSTOPTS, ext_data2);
8858888bf4fSJakub Kicinski 	write_packet(fd, exthdr_pck, total_hdr_len + PAYLOAD_LEN + MIN_EXTHDR_SIZE, daddr);
8868888bf4fSJakub Kicinski }
8878888bf4fSJakub Kicinski 
8888888bf4fSJakub Kicinski /* IPv4 options shouldn't coalesce */
8898888bf4fSJakub Kicinski static void send_ip_options(int fd, struct sockaddr_ll *daddr)
8908888bf4fSJakub Kicinski {
8918888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
8928888bf4fSJakub Kicinski 	static char optpkt[sizeof(buf) + sizeof(struct ip_timestamp)];
8938888bf4fSJakub Kicinski 	int optlen = sizeof(struct ip_timestamp);
8948888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN + optlen;
8958888bf4fSJakub Kicinski 
8968888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
8978888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len + PAYLOAD_LEN, daddr);
8988888bf4fSJakub Kicinski 
8998888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 1, 0, PAYLOAD_LEN, 0);
9008888bf4fSJakub Kicinski 	add_ipv4_ts_option(buf, optpkt);
9018888bf4fSJakub Kicinski 	write_packet(fd, optpkt, pkt_size, daddr);
9028888bf4fSJakub Kicinski 
9038888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 2, 0, PAYLOAD_LEN, 0);
9048888bf4fSJakub Kicinski 	write_packet(fd, buf, total_hdr_len + PAYLOAD_LEN, daddr);
9058888bf4fSJakub Kicinski }
9068888bf4fSJakub Kicinski 
9078888bf4fSJakub Kicinski /*  IPv4 fragments shouldn't coalesce */
9088888bf4fSJakub Kicinski static void send_fragment4(int fd, struct sockaddr_ll *daddr)
9098888bf4fSJakub Kicinski {
9108888bf4fSJakub Kicinski 	static char buf[IP_MAXPACKET];
9118888bf4fSJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buf + ETH_HLEN);
9128888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN;
9138888bf4fSJakub Kicinski 
9148888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
9158888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9168888bf4fSJakub Kicinski 
9178888bf4fSJakub Kicinski 	/* Once fragmented, packet would retain the total_len.
9188888bf4fSJakub Kicinski 	 * Tcp header is prepared as if rest of data is in follow-up frags,
9198888bf4fSJakub Kicinski 	 * but follow up frags aren't actually sent.
9208888bf4fSJakub Kicinski 	 */
9218888bf4fSJakub Kicinski 	memset(buf + total_hdr_len, 'a', PAYLOAD_LEN * 2);
9228888bf4fSJakub Kicinski 	fill_transportlayer(buf + tcp_offset, PAYLOAD_LEN, 0, PAYLOAD_LEN * 2, 0);
9238888bf4fSJakub Kicinski 	fill_networklayer(buf + ETH_HLEN, PAYLOAD_LEN, IPPROTO_TCP);
9248888bf4fSJakub Kicinski 	fill_datalinklayer(buf);
9258888bf4fSJakub Kicinski 
9268888bf4fSJakub Kicinski 	iph->frag_off = htons(0x6000); // DF = 1, MF = 1
9278888bf4fSJakub Kicinski 	iph->check = 0;
9288888bf4fSJakub Kicinski 	iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
9298888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9308888bf4fSJakub Kicinski }
9318888bf4fSJakub Kicinski 
9328888bf4fSJakub Kicinski /* IPv4 packets with different ttl don't coalesce.*/
9338888bf4fSJakub Kicinski static void send_changed_ttl(int fd, struct sockaddr_ll *daddr)
9348888bf4fSJakub Kicinski {
9358888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN;
9368888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
9378888bf4fSJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buf + ETH_HLEN);
9388888bf4fSJakub Kicinski 
9398888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
9408888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9418888bf4fSJakub Kicinski 
9428888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
9438888bf4fSJakub Kicinski 	iph->ttl = 7;
9448888bf4fSJakub Kicinski 	iph->check = 0;
9458888bf4fSJakub Kicinski 	iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
9468888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9478888bf4fSJakub Kicinski }
9488888bf4fSJakub Kicinski 
9498888bf4fSJakub Kicinski /* Packets with different tos don't coalesce.*/
9508888bf4fSJakub Kicinski static void send_changed_tos(int fd, struct sockaddr_ll *daddr)
9518888bf4fSJakub Kicinski {
9528888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN;
9538888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
9548888bf4fSJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buf + ETH_HLEN);
9558888bf4fSJakub Kicinski 	struct ipv6hdr *ip6h = (struct ipv6hdr *)(buf + ETH_HLEN);
9568888bf4fSJakub Kicinski 
9578888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
9588888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9598888bf4fSJakub Kicinski 
9608888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
9618888bf4fSJakub Kicinski 	if (proto == PF_INET) {
9628888bf4fSJakub Kicinski 		iph->tos = 1;
9638888bf4fSJakub Kicinski 		iph->check = 0;
9648888bf4fSJakub Kicinski 		iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
9658888bf4fSJakub Kicinski 	} else if (proto == PF_INET6) {
9668888bf4fSJakub Kicinski 		ip6h->priority = 0xf;
9678888bf4fSJakub Kicinski 	}
9688888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9698888bf4fSJakub Kicinski }
9708888bf4fSJakub Kicinski 
9718888bf4fSJakub Kicinski /* Packets with different ECN don't coalesce.*/
9728888bf4fSJakub Kicinski static void send_changed_ECN(int fd, struct sockaddr_ll *daddr)
9738888bf4fSJakub Kicinski {
9748888bf4fSJakub Kicinski 	int pkt_size = total_hdr_len + PAYLOAD_LEN;
9758888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
9768888bf4fSJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buf + ETH_HLEN);
9778888bf4fSJakub Kicinski 
9788888bf4fSJakub Kicinski 	create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
9798888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9808888bf4fSJakub Kicinski 
9818888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN, 0, PAYLOAD_LEN, 0);
9828888bf4fSJakub Kicinski 	if (proto == PF_INET) {
9838888bf4fSJakub Kicinski 		buf[ETH_HLEN + 1] ^= 0x2; // ECN set to 10
9848888bf4fSJakub Kicinski 		iph->check = 0;
9858888bf4fSJakub Kicinski 		iph->check = checksum_fold(iph, sizeof(struct iphdr), 0);
9868888bf4fSJakub Kicinski 	} else {
9878888bf4fSJakub Kicinski 		buf[ETH_HLEN + 1] ^= 0x20; // ECN set to 10
9888888bf4fSJakub Kicinski 	}
9898888bf4fSJakub Kicinski 	write_packet(fd, buf, pkt_size, daddr);
9908888bf4fSJakub Kicinski }
9918888bf4fSJakub Kicinski 
9928888bf4fSJakub Kicinski /* IPv6 fragments and packets with extensions don't coalesce.*/
9938888bf4fSJakub Kicinski static void send_fragment6(int fd, struct sockaddr_ll *daddr)
9948888bf4fSJakub Kicinski {
9958888bf4fSJakub Kicinski 	static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
9968888bf4fSJakub Kicinski 	static char extpkt[MAX_HDR_LEN + PAYLOAD_LEN +
9978888bf4fSJakub Kicinski 			   sizeof(struct ip6_frag)];
9988888bf4fSJakub Kicinski 	struct ipv6hdr *ip6h = (struct ipv6hdr *)(buf + ETH_HLEN);
9998888bf4fSJakub Kicinski 	struct ip6_frag *frag = (void *)(extpkt + tcp_offset);
10008888bf4fSJakub Kicinski 	int extlen = sizeof(struct ip6_frag);
10018888bf4fSJakub Kicinski 	int bufpkt_len = total_hdr_len + PAYLOAD_LEN;
10028888bf4fSJakub Kicinski 	int extpkt_len = bufpkt_len + extlen;
10038888bf4fSJakub Kicinski 	int i;
10048888bf4fSJakub Kicinski 
10058888bf4fSJakub Kicinski 	for (i = 0; i < 2; i++) {
10068888bf4fSJakub Kicinski 		create_packet(buf, PAYLOAD_LEN * i, 0, PAYLOAD_LEN, 0);
10078888bf4fSJakub Kicinski 		write_packet(fd, buf, bufpkt_len, daddr);
10088888bf4fSJakub Kicinski 	}
10098888bf4fSJakub Kicinski 	sleep(1);
10108888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 2, 0, PAYLOAD_LEN, 0);
10118888bf4fSJakub Kicinski 	memset(extpkt, 0, extpkt_len);
10128888bf4fSJakub Kicinski 
10138888bf4fSJakub Kicinski 	ip6h->nexthdr = IPPROTO_FRAGMENT;
10148888bf4fSJakub Kicinski 	ip6h->payload_len = htons(ntohs(ip6h->payload_len) + extlen);
10158888bf4fSJakub Kicinski 	frag->ip6f_nxt = IPPROTO_TCP;
10168888bf4fSJakub Kicinski 
10178888bf4fSJakub Kicinski 	memcpy(extpkt, buf, tcp_offset);
10188888bf4fSJakub Kicinski 	memcpy(extpkt + tcp_offset + extlen, buf + tcp_offset,
10198888bf4fSJakub Kicinski 	       sizeof(struct tcphdr) + PAYLOAD_LEN);
10208888bf4fSJakub Kicinski 	write_packet(fd, extpkt, extpkt_len, daddr);
10218888bf4fSJakub Kicinski 
10228888bf4fSJakub Kicinski 	create_packet(buf, PAYLOAD_LEN * 3, 0, PAYLOAD_LEN, 0);
10238888bf4fSJakub Kicinski 	write_packet(fd, buf, bufpkt_len, daddr);
10248888bf4fSJakub Kicinski }
10258888bf4fSJakub Kicinski 
10268888bf4fSJakub Kicinski static void bind_packetsocket(int fd)
10278888bf4fSJakub Kicinski {
10288888bf4fSJakub Kicinski 	struct sockaddr_ll daddr = {};
10298888bf4fSJakub Kicinski 
10308888bf4fSJakub Kicinski 	daddr.sll_family = AF_PACKET;
10318888bf4fSJakub Kicinski 	daddr.sll_protocol = ethhdr_proto;
10328888bf4fSJakub Kicinski 	daddr.sll_ifindex = if_nametoindex(ifname);
10338888bf4fSJakub Kicinski 	if (daddr.sll_ifindex == 0)
10348888bf4fSJakub Kicinski 		error(1, errno, "if_nametoindex");
10358888bf4fSJakub Kicinski 
10368888bf4fSJakub Kicinski 	if (bind(fd, (void *)&daddr, sizeof(daddr)) < 0)
10378888bf4fSJakub Kicinski 		error(1, errno, "could not bind socket");
10388888bf4fSJakub Kicinski }
10398888bf4fSJakub Kicinski 
10408888bf4fSJakub Kicinski static void set_timeout(int fd)
10418888bf4fSJakub Kicinski {
10428888bf4fSJakub Kicinski 	struct timeval timeout;
10438888bf4fSJakub Kicinski 
10448888bf4fSJakub Kicinski 	timeout.tv_sec = 3;
10458888bf4fSJakub Kicinski 	timeout.tv_usec = 0;
10468888bf4fSJakub Kicinski 	if (setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (char *)&timeout,
10478888bf4fSJakub Kicinski 		       sizeof(timeout)) < 0)
10488888bf4fSJakub Kicinski 		error(1, errno, "cannot set timeout, setsockopt failed");
10498888bf4fSJakub Kicinski }
10508888bf4fSJakub Kicinski 
10518888bf4fSJakub Kicinski static void set_rcvbuf(int fd)
10528888bf4fSJakub Kicinski {
10538888bf4fSJakub Kicinski 	int bufsize = 1 * 1024 * 1024; /* 1 MB */
10548888bf4fSJakub Kicinski 
10558888bf4fSJakub Kicinski 	if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &bufsize, sizeof(bufsize)))
10568888bf4fSJakub Kicinski 		error(1, errno, "cannot set rcvbuf size, setsockopt failed");
10578888bf4fSJakub Kicinski }
10588888bf4fSJakub Kicinski 
10598888bf4fSJakub Kicinski static void recv_error(int fd, int rcv_errno)
10608888bf4fSJakub Kicinski {
10618888bf4fSJakub Kicinski 	struct tpacket_stats stats;
10628888bf4fSJakub Kicinski 	socklen_t len;
10638888bf4fSJakub Kicinski 
10648888bf4fSJakub Kicinski 	len = sizeof(stats);
10658888bf4fSJakub Kicinski 	if (getsockopt(fd, SOL_PACKET, PACKET_STATISTICS, &stats, &len))
10668888bf4fSJakub Kicinski 		error(1, errno, "can't get stats");
10678888bf4fSJakub Kicinski 
10688888bf4fSJakub Kicinski 	fprintf(stderr, "Socket stats: packets=%u, drops=%u\n",
10698888bf4fSJakub Kicinski 		stats.tp_packets, stats.tp_drops);
10708888bf4fSJakub Kicinski 	error(1, rcv_errno, "could not receive");
10718888bf4fSJakub Kicinski }
10728888bf4fSJakub Kicinski 
10738888bf4fSJakub Kicinski static void check_recv_pkts(int fd, int *correct_payload,
10748888bf4fSJakub Kicinski 			    int correct_num_pkts)
10758888bf4fSJakub Kicinski {
10768888bf4fSJakub Kicinski 	static char buffer[IP_MAXPACKET + ETH_HLEN + 1];
10778888bf4fSJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buffer + ETH_HLEN);
10788888bf4fSJakub Kicinski 	struct ipv6hdr *ip6h = (struct ipv6hdr *)(buffer + ETH_HLEN);
10798888bf4fSJakub Kicinski 	struct tcphdr *tcph;
10808888bf4fSJakub Kicinski 	bool bad_packet = false;
10818888bf4fSJakub Kicinski 	int tcp_ext_len = 0;
10828888bf4fSJakub Kicinski 	int ip_ext_len = 0;
10838888bf4fSJakub Kicinski 	int pkt_size = -1;
10848888bf4fSJakub Kicinski 	int data_len = 0;
10858888bf4fSJakub Kicinski 	int num_pkt = 0;
10868888bf4fSJakub Kicinski 	int i;
10878888bf4fSJakub Kicinski 
10888888bf4fSJakub Kicinski 	vlog("Expected {");
10898888bf4fSJakub Kicinski 	for (i = 0; i < correct_num_pkts; i++)
10908888bf4fSJakub Kicinski 		vlog("%d ", correct_payload[i]);
10918888bf4fSJakub Kicinski 	vlog("}, Total %d packets\nReceived {", correct_num_pkts);
10928888bf4fSJakub Kicinski 
10938888bf4fSJakub Kicinski 	while (1) {
10948888bf4fSJakub Kicinski 		ip_ext_len = 0;
10958888bf4fSJakub Kicinski 		pkt_size = recv(fd, buffer, IP_MAXPACKET + ETH_HLEN + 1, 0);
10968888bf4fSJakub Kicinski 		if (pkt_size < 0)
10978888bf4fSJakub Kicinski 			recv_error(fd, errno);
10988888bf4fSJakub Kicinski 
10998888bf4fSJakub Kicinski 		if (iph->version == 4)
11008888bf4fSJakub Kicinski 			ip_ext_len = (iph->ihl - 5) * 4;
11018888bf4fSJakub Kicinski 		else if (ip6h->version == 6 && ip6h->nexthdr != IPPROTO_TCP)
11028888bf4fSJakub Kicinski 			ip_ext_len = MIN_EXTHDR_SIZE;
11038888bf4fSJakub Kicinski 
11048888bf4fSJakub Kicinski 		tcph = (struct tcphdr *)(buffer + tcp_offset + ip_ext_len);
11058888bf4fSJakub Kicinski 
11068888bf4fSJakub Kicinski 		if (tcph->fin)
11078888bf4fSJakub Kicinski 			break;
11088888bf4fSJakub Kicinski 
11098888bf4fSJakub Kicinski 		tcp_ext_len = (tcph->doff - 5) * 4;
11108888bf4fSJakub Kicinski 		data_len = pkt_size - total_hdr_len - tcp_ext_len - ip_ext_len;
11118888bf4fSJakub Kicinski 		/* Min ethernet frame payload is 46(ETH_ZLEN - ETH_HLEN) by RFC 802.3.
11128888bf4fSJakub Kicinski 		 * Ipv4/tcp packets without at least 6 bytes of data will be padded.
11138888bf4fSJakub Kicinski 		 * Packet sockets are protocol agnostic, and will not trim the padding.
11148888bf4fSJakub Kicinski 		 */
11158888bf4fSJakub Kicinski 		if (pkt_size == ETH_ZLEN && iph->version == 4) {
11168888bf4fSJakub Kicinski 			data_len = ntohs(iph->tot_len)
11178888bf4fSJakub Kicinski 				- sizeof(struct tcphdr) - sizeof(struct iphdr);
11188888bf4fSJakub Kicinski 		}
11198888bf4fSJakub Kicinski 		vlog("%d ", data_len);
11208888bf4fSJakub Kicinski 		if (data_len != correct_payload[num_pkt]) {
11218888bf4fSJakub Kicinski 			vlog("[!=%d]", correct_payload[num_pkt]);
11228888bf4fSJakub Kicinski 			bad_packet = true;
11238888bf4fSJakub Kicinski 		}
11248888bf4fSJakub Kicinski 		num_pkt++;
11258888bf4fSJakub Kicinski 	}
11268888bf4fSJakub Kicinski 	vlog("}, Total %d packets.\n", num_pkt);
11278888bf4fSJakub Kicinski 	if (num_pkt != correct_num_pkts)
11288888bf4fSJakub Kicinski 		error(1, 0, "incorrect number of packets");
11298888bf4fSJakub Kicinski 	if (bad_packet)
11308888bf4fSJakub Kicinski 		error(1, 0, "incorrect packet geometry");
11318888bf4fSJakub Kicinski 
11328888bf4fSJakub Kicinski 	printf("Test succeeded\n\n");
11338888bf4fSJakub Kicinski }
11348888bf4fSJakub Kicinski 
1135ba5d4128SJakub Kicinski static void check_capacity_pkts(int fd)
1136ba5d4128SJakub Kicinski {
1137ba5d4128SJakub Kicinski 	static char buffer[IP_MAXPACKET + ETH_HLEN + 1];
1138ba5d4128SJakub Kicinski 	struct iphdr *iph = (struct iphdr *)(buffer + ETH_HLEN);
1139ba5d4128SJakub Kicinski 	struct ipv6hdr *ip6h = (struct ipv6hdr *)(buffer + ETH_HLEN);
1140*ff1cb3adSJakub Kicinski 	int num_pkt = 0, num_coal = 0, pkt_idx;
1141ba5d4128SJakub Kicinski 	const char *fail_reason = NULL;
1142ba5d4128SJakub Kicinski 	int flow_order[num_flows * 2];
1143ba5d4128SJakub Kicinski 	int coalesced[num_flows];
1144ba5d4128SJakub Kicinski 	struct tcphdr *tcph;
1145ba5d4128SJakub Kicinski 	int ip_ext_len = 0;
1146ba5d4128SJakub Kicinski 	int total_data = 0;
1147ba5d4128SJakub Kicinski 	int pkt_size = -1;
1148ba5d4128SJakub Kicinski 	int data_len = 0;
1149ba5d4128SJakub Kicinski 	int flow_id;
1150ba5d4128SJakub Kicinski 	int sport;
1151ba5d4128SJakub Kicinski 
1152ba5d4128SJakub Kicinski 	memset(coalesced, 0, sizeof(coalesced));
1153ba5d4128SJakub Kicinski 	memset(flow_order, -1, sizeof(flow_order));
1154ba5d4128SJakub Kicinski 
1155ba5d4128SJakub Kicinski 	while (total_data < num_flows * CAPACITY_PAYLOAD_LEN * 2) {
1156ba5d4128SJakub Kicinski 		ip_ext_len = 0;
1157ba5d4128SJakub Kicinski 		pkt_size = recv(fd, buffer, IP_MAXPACKET + ETH_HLEN + 1, 0);
1158ba5d4128SJakub Kicinski 		if (pkt_size < 0)
1159ba5d4128SJakub Kicinski 			recv_error(fd, errno);
1160ba5d4128SJakub Kicinski 
1161ba5d4128SJakub Kicinski 		if (iph->version == 4)
1162ba5d4128SJakub Kicinski 			ip_ext_len = (iph->ihl - 5) * 4;
1163ba5d4128SJakub Kicinski 		else if (ip6h->version == 6 && ip6h->nexthdr != IPPROTO_TCP)
1164ba5d4128SJakub Kicinski 			ip_ext_len = MIN_EXTHDR_SIZE;
1165ba5d4128SJakub Kicinski 
1166ba5d4128SJakub Kicinski 		tcph = (struct tcphdr *)(buffer + tcp_offset + ip_ext_len);
1167ba5d4128SJakub Kicinski 
1168ba5d4128SJakub Kicinski 		/* FIN packet terminates reception */
1169ba5d4128SJakub Kicinski 		if (tcph->fin)
1170ba5d4128SJakub Kicinski 			break;
1171ba5d4128SJakub Kicinski 
1172ba5d4128SJakub Kicinski 		sport = ntohs(tcph->source);
1173ba5d4128SJakub Kicinski 		flow_id = sport - SPORT;
1174ba5d4128SJakub Kicinski 
1175ba5d4128SJakub Kicinski 		if (flow_id < 0 || flow_id >= num_flows) {
1176ba5d4128SJakub Kicinski 			vlog("Invalid flow_id %d from sport %d\n",
1177ba5d4128SJakub Kicinski 			     flow_id, sport);
1178ba5d4128SJakub Kicinski 			fail_reason = fail_reason ?: "invalid packet";
1179ba5d4128SJakub Kicinski 			continue;
1180ba5d4128SJakub Kicinski 		}
1181ba5d4128SJakub Kicinski 
1182ba5d4128SJakub Kicinski 		/* Calculate payload length */
1183ba5d4128SJakub Kicinski 		if (pkt_size == ETH_ZLEN && iph->version == 4) {
1184ba5d4128SJakub Kicinski 			data_len = ntohs(iph->tot_len)
1185ba5d4128SJakub Kicinski 				- sizeof(struct tcphdr) - sizeof(struct iphdr);
1186ba5d4128SJakub Kicinski 		} else {
1187ba5d4128SJakub Kicinski 			data_len = pkt_size - total_hdr_len - ip_ext_len;
1188ba5d4128SJakub Kicinski 		}
1189ba5d4128SJakub Kicinski 
1190ba5d4128SJakub Kicinski 		flow_order[num_pkt] = flow_id;
1191ba5d4128SJakub Kicinski 		coalesced[flow_id] = data_len;
1192ba5d4128SJakub Kicinski 
1193ba5d4128SJakub Kicinski 		if (data_len == CAPACITY_PAYLOAD_LEN * 2) {
1194ba5d4128SJakub Kicinski 			num_coal++;
1195ba5d4128SJakub Kicinski 		} else {
1196ba5d4128SJakub Kicinski 			vlog("Pkt %d: flow %d, sport %d, len %d (expected %d)\n",
1197ba5d4128SJakub Kicinski 			     num_pkt, flow_id, sport, data_len,
1198ba5d4128SJakub Kicinski 			     CAPACITY_PAYLOAD_LEN * 2);
1199ba5d4128SJakub Kicinski 			fail_reason = fail_reason ?: "not coalesced";
1200ba5d4128SJakub Kicinski 		}
1201ba5d4128SJakub Kicinski 
1202ba5d4128SJakub Kicinski 		num_pkt++;
1203ba5d4128SJakub Kicinski 		total_data += data_len;
1204ba5d4128SJakub Kicinski 	}
1205ba5d4128SJakub Kicinski 
1206*ff1cb3adSJakub Kicinski 	/* Check flow ordering. We expect to see all non-coalesced first segs
1207*ff1cb3adSJakub Kicinski 	 * then interleaved coalesced and non-coalesced second frames.
1208*ff1cb3adSJakub Kicinski 	 */
1209*ff1cb3adSJakub Kicinski 	pkt_idx = 0;
1210*ff1cb3adSJakub Kicinski 	for (flow_id = 0; order_check && flow_id < num_flows; flow_id++) {
1211*ff1cb3adSJakub Kicinski 		bool coaled = coalesced[flow_id] > CAPACITY_PAYLOAD_LEN;
1212*ff1cb3adSJakub Kicinski 
1213*ff1cb3adSJakub Kicinski 		if (coaled)
1214*ff1cb3adSJakub Kicinski 			continue;
1215*ff1cb3adSJakub Kicinski 
1216*ff1cb3adSJakub Kicinski 		if (flow_order[pkt_idx] != flow_id) {
1217*ff1cb3adSJakub Kicinski 			vlog("Flow order mismatch (non-coalesced) at position %d: expected flow %d, got flow %d\n",
1218*ff1cb3adSJakub Kicinski 			     pkt_idx, flow_id, flow_order[pkt_idx]);
1219*ff1cb3adSJakub Kicinski 			fail_reason = fail_reason ?: "bad packet order (1)";
1220*ff1cb3adSJakub Kicinski 		}
1221*ff1cb3adSJakub Kicinski 		pkt_idx++;
1222*ff1cb3adSJakub Kicinski 	}
1223*ff1cb3adSJakub Kicinski 	for (flow_id = 0; order_check && flow_id < num_flows; flow_id++) {
1224*ff1cb3adSJakub Kicinski 		bool coaled = coalesced[flow_id] > CAPACITY_PAYLOAD_LEN;
1225*ff1cb3adSJakub Kicinski 
1226*ff1cb3adSJakub Kicinski 		if (flow_order[pkt_idx] != flow_id) {
1227*ff1cb3adSJakub Kicinski 			vlog("Flow order mismatch at position %d: expected flow %d, got flow %d, coalesced: %d\n",
1228*ff1cb3adSJakub Kicinski 			     pkt_idx, flow_id, flow_order[pkt_idx], coaled);
1229*ff1cb3adSJakub Kicinski 			fail_reason = fail_reason ?: "bad packet order (2)";
1230*ff1cb3adSJakub Kicinski 		}
1231*ff1cb3adSJakub Kicinski 		pkt_idx++;
1232*ff1cb3adSJakub Kicinski 	}
1233*ff1cb3adSJakub Kicinski 
1234ba5d4128SJakub Kicinski 	if (!fail_reason) {
1235ba5d4128SJakub Kicinski 		vlog("All %d flows coalesced correctly\n", num_flows);
1236ba5d4128SJakub Kicinski 		printf("Test succeeded\n\n");
1237ba5d4128SJakub Kicinski 	} else {
1238ba5d4128SJakub Kicinski 		printf("FAILED\n");
1239ba5d4128SJakub Kicinski 	}
1240ba5d4128SJakub Kicinski 
1241ba5d4128SJakub Kicinski 	/* Always print stats for external validation */
1242ba5d4128SJakub Kicinski 	printf("STATS: received=%d wire=%d coalesced=%d\n",
1243ba5d4128SJakub Kicinski 	       num_pkt, num_pkt + num_coal, num_coal);
1244ba5d4128SJakub Kicinski 
1245ba5d4128SJakub Kicinski 	if (fail_reason)
1246ba5d4128SJakub Kicinski 		error(1, 0, "capacity test failed %s", fail_reason);
1247ba5d4128SJakub Kicinski }
1248ba5d4128SJakub Kicinski 
12498888bf4fSJakub Kicinski static void gro_sender(void)
12508888bf4fSJakub Kicinski {
1251f26d43acSJakub Kicinski 	int bufsize = 4 * 1024 * 1024; /* 4 MB */
12528888bf4fSJakub Kicinski 	const int fin_delay_us = 100 * 1000;
12538888bf4fSJakub Kicinski 	static char fin_pkt[MAX_HDR_LEN];
12548888bf4fSJakub Kicinski 	struct sockaddr_ll daddr = {};
12558888bf4fSJakub Kicinski 	int txfd = -1;
12568888bf4fSJakub Kicinski 
12578888bf4fSJakub Kicinski 	txfd = socket(PF_PACKET, SOCK_RAW, IPPROTO_RAW);
12588888bf4fSJakub Kicinski 	if (txfd < 0)
12598888bf4fSJakub Kicinski 		error(1, errno, "socket creation");
12608888bf4fSJakub Kicinski 
1261f26d43acSJakub Kicinski 	if (setsockopt(txfd, SOL_SOCKET, SO_SNDBUF, &bufsize, sizeof(bufsize)))
1262f26d43acSJakub Kicinski 		error(1, errno, "cannot set sndbuf size, setsockopt failed");
1263f26d43acSJakub Kicinski 
1264f26d43acSJakub Kicinski 	/* Enable SO_TXTIME unless test case generates more than one flow
1265f26d43acSJakub Kicinski 	 * SO_TXTIME could result in qdisc layer sorting the packets at sender.
1266f26d43acSJakub Kicinski 	 */
1267ba5d4128SJakub Kicinski 	if (strcmp(testname, "single") && strcmp(testname, "capacity")) {
1268f26d43acSJakub Kicinski 		struct sock_txtime so_txtime = { .clockid = CLOCK_MONOTONIC, };
1269f26d43acSJakub Kicinski 		struct timespec ts;
1270f26d43acSJakub Kicinski 
1271f26d43acSJakub Kicinski 		if (setsockopt(txfd, SOL_SOCKET, SO_TXTIME,
1272f26d43acSJakub Kicinski 			       &so_txtime, sizeof(so_txtime)))
1273f26d43acSJakub Kicinski 			error(1, errno, "setsockopt SO_TXTIME");
1274f26d43acSJakub Kicinski 
1275f26d43acSJakub Kicinski 		if (clock_gettime(CLOCK_MONOTONIC, &ts))
1276f26d43acSJakub Kicinski 			error(1, errno, "clock_gettime");
1277f26d43acSJakub Kicinski 
1278f26d43acSJakub Kicinski 		txtime_ns = ts.tv_sec * 1000000000ULL + ts.tv_nsec;
1279f26d43acSJakub Kicinski 		txtime_ns += TXTIME_DELAY_MS * 1000000ULL;
1280f26d43acSJakub Kicinski 	}
1281f26d43acSJakub Kicinski 
12828888bf4fSJakub Kicinski 	memset(&daddr, 0, sizeof(daddr));
12838888bf4fSJakub Kicinski 	daddr.sll_ifindex = if_nametoindex(ifname);
12848888bf4fSJakub Kicinski 	if (daddr.sll_ifindex == 0)
12858888bf4fSJakub Kicinski 		error(1, errno, "if_nametoindex");
12868888bf4fSJakub Kicinski 	daddr.sll_family = AF_PACKET;
12878888bf4fSJakub Kicinski 	memcpy(daddr.sll_addr, dst_mac, ETH_ALEN);
12888888bf4fSJakub Kicinski 	daddr.sll_halen = ETH_ALEN;
12898888bf4fSJakub Kicinski 	create_packet(fin_pkt, PAYLOAD_LEN * 2, 0, 0, 1);
12908888bf4fSJakub Kicinski 
12918888bf4fSJakub Kicinski 	/* data sub-tests */
12928888bf4fSJakub Kicinski 	if (strcmp(testname, "data_same") == 0) {
12938888bf4fSJakub Kicinski 		send_data_pkts(txfd, &daddr, PAYLOAD_LEN, PAYLOAD_LEN);
12948888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
12958888bf4fSJakub Kicinski 	} else if (strcmp(testname, "data_lrg_sml") == 0) {
12968888bf4fSJakub Kicinski 		send_data_pkts(txfd, &daddr, PAYLOAD_LEN, PAYLOAD_LEN / 2);
12978888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
12988888bf4fSJakub Kicinski 	} else if (strcmp(testname, "data_sml_lrg") == 0) {
12998888bf4fSJakub Kicinski 		send_data_pkts(txfd, &daddr, PAYLOAD_LEN / 2, PAYLOAD_LEN);
13008888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13018888bf4fSJakub Kicinski 
13028888bf4fSJakub Kicinski 	/* ack test */
13038888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ack") == 0) {
13048888bf4fSJakub Kicinski 		send_ack(txfd, &daddr);
13058888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13068888bf4fSJakub Kicinski 
13078888bf4fSJakub Kicinski 	/* flags sub-tests */
13088888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_psh") == 0) {
13098888bf4fSJakub Kicinski 		send_flags(txfd, &daddr, 1, 0, 0, 0, 0);
13108888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13118888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_syn") == 0) {
13128888bf4fSJakub Kicinski 		send_flags(txfd, &daddr, 0, 1, 0, 0, 0);
13138888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13148888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_rst") == 0) {
13158888bf4fSJakub Kicinski 		send_flags(txfd, &daddr, 0, 0, 1, 0, 0);
13168888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13178888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_urg") == 0) {
13188888bf4fSJakub Kicinski 		send_flags(txfd, &daddr, 0, 0, 0, 1, 0);
13198888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13208888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_cwr") == 0) {
13218888bf4fSJakub Kicinski 		send_flags(txfd, &daddr, 0, 0, 0, 0, 1);
13228888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13238888bf4fSJakub Kicinski 
13248888bf4fSJakub Kicinski 	/* tcp sub-tests */
13258888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_csum") == 0) {
13268888bf4fSJakub Kicinski 		send_changed_checksum(txfd, &daddr);
13278888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13288888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13298888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_seq") == 0) {
13308888bf4fSJakub Kicinski 		send_changed_seq(txfd, &daddr);
13318888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13328888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13338888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_ts") == 0) {
13348888bf4fSJakub Kicinski 		send_changed_ts(txfd, &daddr);
13358888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13368888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13378888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_opt") == 0) {
13388888bf4fSJakub Kicinski 		send_diff_opt(txfd, &daddr);
13398888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13408888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13418888bf4fSJakub Kicinski 
13428888bf4fSJakub Kicinski 	/* ip sub-tests - shared between IPv4 and IPv6 */
13438888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_ecn") == 0) {
13448888bf4fSJakub Kicinski 		send_changed_ECN(txfd, &daddr);
13458888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13468888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_tos") == 0) {
13478888bf4fSJakub Kicinski 		send_changed_tos(txfd, &daddr);
13488888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13498888bf4fSJakub Kicinski 
13508888bf4fSJakub Kicinski 	/* ip sub-tests - IPv4 only */
13518888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_ttl") == 0) {
13528888bf4fSJakub Kicinski 		send_changed_ttl(txfd, &daddr);
13538888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13548888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_opt") == 0) {
13558888bf4fSJakub Kicinski 		send_ip_options(txfd, &daddr);
13568888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13578888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13588888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_frag4") == 0) {
13598888bf4fSJakub Kicinski 		send_fragment4(txfd, &daddr);
13608888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13618888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13628888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_inc") == 0) {
13638888bf4fSJakub Kicinski 		send_flush_id_case(txfd, &daddr, FLUSH_ID_DF1_INC);
13648888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13658888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13668888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_fixed") == 0) {
13678888bf4fSJakub Kicinski 		send_flush_id_case(txfd, &daddr, FLUSH_ID_DF1_FIXED);
13688888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13698888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13708888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df0_inc") == 0) {
13718888bf4fSJakub Kicinski 		send_flush_id_case(txfd, &daddr, FLUSH_ID_DF0_INC);
13728888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13738888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13748888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df0_fixed") == 0) {
13758888bf4fSJakub Kicinski 		send_flush_id_case(txfd, &daddr, FLUSH_ID_DF0_FIXED);
13768888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13778888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13788888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_inc_fixed") == 0) {
13798888bf4fSJakub Kicinski 		send_flush_id_case(txfd, &daddr, FLUSH_ID_DF1_INC_FIXED);
13808888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13818888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13828888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_fixed_inc") == 0) {
13838888bf4fSJakub Kicinski 		send_flush_id_case(txfd, &daddr, FLUSH_ID_DF1_FIXED_INC);
13848888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13858888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13868888bf4fSJakub Kicinski 
13878888bf4fSJakub Kicinski 	/* ip sub-tests - IPv6 only */
13888888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_frag6") == 0) {
13898888bf4fSJakub Kicinski 		send_fragment6(txfd, &daddr);
13908888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13918888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13928888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_v6ext_same") == 0) {
13938888bf4fSJakub Kicinski 		send_ipv6_exthdr(txfd, &daddr, EXT_PAYLOAD_1, EXT_PAYLOAD_1);
13948888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13958888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
13968888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_v6ext_diff") == 0) {
13978888bf4fSJakub Kicinski 		send_ipv6_exthdr(txfd, &daddr, EXT_PAYLOAD_1, EXT_PAYLOAD_2);
13988888bf4fSJakub Kicinski 		usleep(fin_delay_us);
13998888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
14008888bf4fSJakub Kicinski 
14018888bf4fSJakub Kicinski 	/* large sub-tests */
14028888bf4fSJakub Kicinski 	} else if (strcmp(testname, "large_max") == 0) {
14038888bf4fSJakub Kicinski 		int offset = (proto == PF_INET && !ipip) ? 20 : 0;
14048888bf4fSJakub Kicinski 		int remainder = (MAX_PAYLOAD + offset) % MSS;
14058888bf4fSJakub Kicinski 
14068888bf4fSJakub Kicinski 		send_large(txfd, &daddr, remainder);
14078888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
14088888bf4fSJakub Kicinski 	} else if (strcmp(testname, "large_rem") == 0) {
14098888bf4fSJakub Kicinski 		int offset = (proto == PF_INET && !ipip) ? 20 : 0;
14108888bf4fSJakub Kicinski 		int remainder = (MAX_PAYLOAD + offset) % MSS;
14118888bf4fSJakub Kicinski 
14128888bf4fSJakub Kicinski 		send_large(txfd, &daddr, remainder + 1);
14138888bf4fSJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
1414ba5d4128SJakub Kicinski 
1415ba5d4128SJakub Kicinski 	/* machinery sub-tests */
1416ba5d4128SJakub Kicinski 	} else if (strcmp(testname, "single") == 0) {
1417ba5d4128SJakub Kicinski 		static char buf[MAX_HDR_LEN + PAYLOAD_LEN];
1418ba5d4128SJakub Kicinski 
1419ba5d4128SJakub Kicinski 		create_packet(buf, 0, 0, PAYLOAD_LEN, 0);
1420ba5d4128SJakub Kicinski 		write_packet(txfd, buf, total_hdr_len + PAYLOAD_LEN, &daddr);
1421ba5d4128SJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
1422ba5d4128SJakub Kicinski 	} else if (strcmp(testname, "capacity") == 0) {
1423ba5d4128SJakub Kicinski 		send_capacity(txfd, &daddr);
1424ba5d4128SJakub Kicinski 		usleep(fin_delay_us);
1425ba5d4128SJakub Kicinski 		write_packet(txfd, fin_pkt, total_hdr_len, &daddr);
1426ba5d4128SJakub Kicinski 
14278888bf4fSJakub Kicinski 	} else {
14288888bf4fSJakub Kicinski 		error(1, 0, "Unknown testcase: %s", testname);
14298888bf4fSJakub Kicinski 	}
14308888bf4fSJakub Kicinski 
14318888bf4fSJakub Kicinski 	if (close(txfd))
14328888bf4fSJakub Kicinski 		error(1, errno, "socket close");
14338888bf4fSJakub Kicinski }
14348888bf4fSJakub Kicinski 
14358888bf4fSJakub Kicinski static void gro_receiver(void)
14368888bf4fSJakub Kicinski {
14378888bf4fSJakub Kicinski 	static int correct_payload[NUM_PACKETS];
14388888bf4fSJakub Kicinski 	int rxfd = -1;
14398888bf4fSJakub Kicinski 
14408888bf4fSJakub Kicinski 	rxfd = socket(PF_PACKET, SOCK_RAW, htons(ETH_P_NONE));
14418888bf4fSJakub Kicinski 	if (rxfd < 0)
14428888bf4fSJakub Kicinski 		error(1, 0, "socket creation");
14438888bf4fSJakub Kicinski 	setup_sock_filter(rxfd);
14448888bf4fSJakub Kicinski 	set_timeout(rxfd);
14458888bf4fSJakub Kicinski 	set_rcvbuf(rxfd);
14468888bf4fSJakub Kicinski 	bind_packetsocket(rxfd);
14478888bf4fSJakub Kicinski 
14488888bf4fSJakub Kicinski 	ksft_ready();
14498888bf4fSJakub Kicinski 
14508888bf4fSJakub Kicinski 	memset(correct_payload, 0, sizeof(correct_payload));
14518888bf4fSJakub Kicinski 
14528888bf4fSJakub Kicinski 	/* data sub-tests */
14538888bf4fSJakub Kicinski 	if (strcmp(testname, "data_same") == 0) {
14548888bf4fSJakub Kicinski 		printf("pure data packet of same size: ");
14558888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
14568888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
14578888bf4fSJakub Kicinski 	} else if (strcmp(testname, "data_lrg_sml") == 0) {
14588888bf4fSJakub Kicinski 		printf("large data packets followed by a smaller one: ");
14598888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 1.5;
14608888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
14618888bf4fSJakub Kicinski 	} else if (strcmp(testname, "data_sml_lrg") == 0) {
14628888bf4fSJakub Kicinski 		printf("small data packets followed by a larger one: ");
14638888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN / 2;
14648888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
14658888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
14668888bf4fSJakub Kicinski 
14678888bf4fSJakub Kicinski 	/* ack test */
14688888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ack") == 0) {
14698888bf4fSJakub Kicinski 		printf("duplicate ack and pure ack: ");
14708888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
14718888bf4fSJakub Kicinski 
14728888bf4fSJakub Kicinski 	/* flags sub-tests */
14738888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_psh") == 0) {
14748888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 3;
14758888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN * 2;
14768888bf4fSJakub Kicinski 		printf("psh flag ends coalescing: ");
14778888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
14788888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_syn") == 0) {
14798888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
14808888bf4fSJakub Kicinski 		correct_payload[1] = 0;
14818888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN * 2;
14828888bf4fSJakub Kicinski 		printf("syn flag ends coalescing: ");
14838888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
14848888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_rst") == 0) {
14858888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
14868888bf4fSJakub Kicinski 		correct_payload[1] = 0;
14878888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN * 2;
14888888bf4fSJakub Kicinski 		printf("rst flag ends coalescing: ");
14898888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
14908888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_urg") == 0) {
14918888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
14928888bf4fSJakub Kicinski 		correct_payload[1] = 0;
14938888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN * 2;
14948888bf4fSJakub Kicinski 		printf("urg flag ends coalescing: ");
14958888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
14968888bf4fSJakub Kicinski 	} else if (strcmp(testname, "flags_cwr") == 0) {
14978888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
14988888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN * 2;
14998888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN * 2;
15008888bf4fSJakub Kicinski 		printf("cwr flag ends coalescing: ");
15018888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
15028888bf4fSJakub Kicinski 
15038888bf4fSJakub Kicinski 	/* tcp sub-tests */
15048888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_csum") == 0) {
15058888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15068888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15078888bf4fSJakub Kicinski 		printf("changed checksum does not coalesce: ");
15088888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15098888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_seq") == 0) {
15108888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15118888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15128888bf4fSJakub Kicinski 		printf("Wrong Seq number doesn't coalesce: ");
15138888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15148888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_ts") == 0) {
15158888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15168888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15178888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN;
15188888bf4fSJakub Kicinski 		correct_payload[3] = PAYLOAD_LEN;
15198888bf4fSJakub Kicinski 		printf("Different timestamp doesn't coalesce: ");
15208888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 4);
15218888bf4fSJakub Kicinski 	} else if (strcmp(testname, "tcp_opt") == 0) {
15228888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15238888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15248888bf4fSJakub Kicinski 		printf("Different options doesn't coalesce: ");
15258888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15268888bf4fSJakub Kicinski 
15278888bf4fSJakub Kicinski 	/* ip sub-tests - shared between IPv4 and IPv6 */
15288888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_ecn") == 0) {
15298888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15308888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15318888bf4fSJakub Kicinski 		printf("different ECN doesn't coalesce: ");
15328888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15338888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_tos") == 0) {
15348888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15358888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15368888bf4fSJakub Kicinski 		printf("different tos doesn't coalesce: ");
15378888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15388888bf4fSJakub Kicinski 
15398888bf4fSJakub Kicinski 	/* ip sub-tests - IPv4 only */
15408888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_ttl") == 0) {
15418888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15428888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15438888bf4fSJakub Kicinski 		printf("different ttl doesn't coalesce: ");
15448888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15458888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_opt") == 0) {
15468888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15478888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15488888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN;
15498888bf4fSJakub Kicinski 		printf("ip options doesn't coalesce: ");
15508888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
15518888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_frag4") == 0) {
15528888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
15538888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15548888bf4fSJakub Kicinski 		printf("fragmented ip4 doesn't coalesce: ");
15558888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15568888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_inc") == 0) {
15578888bf4fSJakub Kicinski 		printf("DF=1, Incrementing - should coalesce: ");
15588888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15598888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
15608888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_fixed") == 0) {
15618888bf4fSJakub Kicinski 		printf("DF=1, Fixed - should coalesce: ");
15628888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15638888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
15648888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df0_inc") == 0) {
15658888bf4fSJakub Kicinski 		printf("DF=0, Incrementing - should coalesce: ");
15668888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15678888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
15688888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df0_fixed") == 0) {
15698888bf4fSJakub Kicinski 		printf("DF=0, Fixed - should coalesce: ");
15708888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15718888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
15728888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_inc_fixed") == 0) {
15738888bf4fSJakub Kicinski 		printf("DF=1, 2 Incrementing and one fixed - should coalesce only first 2 packets: ");
15748888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15758888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15768888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15778888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_id_df1_fixed_inc") == 0) {
15788888bf4fSJakub Kicinski 		printf("DF=1, 2 Fixed and one incrementing - should coalesce only first 2 packets: ");
15798888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15808888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15818888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
15828888bf4fSJakub Kicinski 
15838888bf4fSJakub Kicinski 	/* ip sub-tests - IPv6 only */
15848888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_frag6") == 0) {
15858888bf4fSJakub Kicinski 		/* GRO doesn't check for ipv6 hop limit when flushing.
15868888bf4fSJakub Kicinski 		 * Hence no corresponding test to the ipv4 case.
15878888bf4fSJakub Kicinski 		 */
15888888bf4fSJakub Kicinski 		printf("fragmented ip6 doesn't coalesce: ");
15898888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15908888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
15918888bf4fSJakub Kicinski 		correct_payload[2] = PAYLOAD_LEN;
15928888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
15938888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_v6ext_same") == 0) {
15948888bf4fSJakub Kicinski 		printf("ipv6 with ext header does coalesce: ");
15958888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN * 2;
15968888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
15978888bf4fSJakub Kicinski 	} else if (strcmp(testname, "ip_v6ext_diff") == 0) {
15988888bf4fSJakub Kicinski 		printf("ipv6 with ext header with different payloads doesn't coalesce: ");
15998888bf4fSJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
16008888bf4fSJakub Kicinski 		correct_payload[1] = PAYLOAD_LEN;
16018888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
16028888bf4fSJakub Kicinski 
16038888bf4fSJakub Kicinski 	/* large sub-tests */
16048888bf4fSJakub Kicinski 	} else if (strcmp(testname, "large_max") == 0) {
16058888bf4fSJakub Kicinski 		int offset = (proto == PF_INET && !ipip) ? 20 : 0;
16068888bf4fSJakub Kicinski 		int remainder = (MAX_PAYLOAD + offset) % MSS;
16078888bf4fSJakub Kicinski 
16088888bf4fSJakub Kicinski 		correct_payload[0] = (MAX_PAYLOAD + offset);
16098888bf4fSJakub Kicinski 		correct_payload[1] = remainder;
16108888bf4fSJakub Kicinski 		printf("Shouldn't coalesce if exceed IP max pkt size: ");
16118888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 2);
16128888bf4fSJakub Kicinski 	} else if (strcmp(testname, "large_rem") == 0) {
16138888bf4fSJakub Kicinski 		int offset = (proto == PF_INET && !ipip) ? 20 : 0;
16148888bf4fSJakub Kicinski 		int remainder = (MAX_PAYLOAD + offset) % MSS;
16158888bf4fSJakub Kicinski 
16168888bf4fSJakub Kicinski 		/* last segment sent individually, doesn't start new segment */
16178888bf4fSJakub Kicinski 		correct_payload[0] = (MAX_PAYLOAD + offset) - remainder;
16188888bf4fSJakub Kicinski 		correct_payload[1] = remainder + 1;
16198888bf4fSJakub Kicinski 		correct_payload[2] = remainder + 1;
16208888bf4fSJakub Kicinski 		printf("last segment sent individually: ");
16218888bf4fSJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 3);
1622ba5d4128SJakub Kicinski 
1623ba5d4128SJakub Kicinski 	/* machinery sub-tests */
1624ba5d4128SJakub Kicinski 	} else if (strcmp(testname, "single") == 0) {
1625ba5d4128SJakub Kicinski 		printf("single data packet: ");
1626ba5d4128SJakub Kicinski 		correct_payload[0] = PAYLOAD_LEN;
1627ba5d4128SJakub Kicinski 		check_recv_pkts(rxfd, correct_payload, 1);
1628ba5d4128SJakub Kicinski 	} else if (strcmp(testname, "capacity") == 0) {
1629ba5d4128SJakub Kicinski 		check_capacity_pkts(rxfd);
1630ba5d4128SJakub Kicinski 
16318888bf4fSJakub Kicinski 	} else {
16328888bf4fSJakub Kicinski 		error(1, 0, "Test case error: unknown testname %s", testname);
16338888bf4fSJakub Kicinski 	}
16348888bf4fSJakub Kicinski 
16358888bf4fSJakub Kicinski 	if (close(rxfd))
16368888bf4fSJakub Kicinski 		error(1, 0, "socket close");
16378888bf4fSJakub Kicinski }
16388888bf4fSJakub Kicinski 
16398888bf4fSJakub Kicinski static void parse_args(int argc, char **argv)
16408888bf4fSJakub Kicinski {
16418888bf4fSJakub Kicinski 	static const struct option opts[] = {
16428888bf4fSJakub Kicinski 		{ "daddr", required_argument, NULL, 'd' },
16438888bf4fSJakub Kicinski 		{ "dmac", required_argument, NULL, 'D' },
16448888bf4fSJakub Kicinski 		{ "iface", required_argument, NULL, 'i' },
16458888bf4fSJakub Kicinski 		{ "ipv4", no_argument, NULL, '4' },
16468888bf4fSJakub Kicinski 		{ "ipv6", no_argument, NULL, '6' },
16478888bf4fSJakub Kicinski 		{ "ipip", no_argument, NULL, 'e' },
1648ba5d4128SJakub Kicinski 		{ "num-flows", required_argument, NULL, 'n' },
16498888bf4fSJakub Kicinski 		{ "rx", no_argument, NULL, 'r' },
16508888bf4fSJakub Kicinski 		{ "saddr", required_argument, NULL, 's' },
16518888bf4fSJakub Kicinski 		{ "smac", required_argument, NULL, 'S' },
16528888bf4fSJakub Kicinski 		{ "test", required_argument, NULL, 't' },
1653*ff1cb3adSJakub Kicinski 		{ "order-check", no_argument, NULL, 'o' },
16548888bf4fSJakub Kicinski 		{ "verbose", no_argument, NULL, 'v' },
16558888bf4fSJakub Kicinski 		{ 0, 0, 0, 0 }
16568888bf4fSJakub Kicinski 	};
16578888bf4fSJakub Kicinski 	int c;
16588888bf4fSJakub Kicinski 
1659*ff1cb3adSJakub Kicinski 	while ((c = getopt_long(argc, argv, "46d:D:ei:n:rs:S:t:ov", opts, NULL)) != -1) {
16608888bf4fSJakub Kicinski 		switch (c) {
16618888bf4fSJakub Kicinski 		case '4':
16628888bf4fSJakub Kicinski 			proto = PF_INET;
16638888bf4fSJakub Kicinski 			ethhdr_proto = htons(ETH_P_IP);
16648888bf4fSJakub Kicinski 			break;
16658888bf4fSJakub Kicinski 		case '6':
16668888bf4fSJakub Kicinski 			proto = PF_INET6;
16678888bf4fSJakub Kicinski 			ethhdr_proto = htons(ETH_P_IPV6);
16688888bf4fSJakub Kicinski 			break;
16698888bf4fSJakub Kicinski 		case 'e':
16708888bf4fSJakub Kicinski 			ipip = true;
16718888bf4fSJakub Kicinski 			proto = PF_INET;
16728888bf4fSJakub Kicinski 			ethhdr_proto = htons(ETH_P_IP);
16738888bf4fSJakub Kicinski 			break;
16748888bf4fSJakub Kicinski 		case 'd':
16758888bf4fSJakub Kicinski 			addr4_dst = addr6_dst = optarg;
16768888bf4fSJakub Kicinski 			break;
16778888bf4fSJakub Kicinski 		case 'D':
16788888bf4fSJakub Kicinski 			dmac = optarg;
16798888bf4fSJakub Kicinski 			break;
16808888bf4fSJakub Kicinski 		case 'i':
16818888bf4fSJakub Kicinski 			ifname = optarg;
16828888bf4fSJakub Kicinski 			break;
1683ba5d4128SJakub Kicinski 		case 'n':
1684ba5d4128SJakub Kicinski 			num_flows = atoi(optarg);
1685ba5d4128SJakub Kicinski 			break;
16868888bf4fSJakub Kicinski 		case 'r':
16878888bf4fSJakub Kicinski 			tx_socket = false;
16888888bf4fSJakub Kicinski 			break;
16898888bf4fSJakub Kicinski 		case 's':
16908888bf4fSJakub Kicinski 			addr4_src = addr6_src = optarg;
16918888bf4fSJakub Kicinski 			break;
16928888bf4fSJakub Kicinski 		case 'S':
16938888bf4fSJakub Kicinski 			smac = optarg;
16948888bf4fSJakub Kicinski 			break;
16958888bf4fSJakub Kicinski 		case 't':
16968888bf4fSJakub Kicinski 			testname = optarg;
16978888bf4fSJakub Kicinski 			break;
1698*ff1cb3adSJakub Kicinski 		case 'o':
1699*ff1cb3adSJakub Kicinski 			order_check = true;
1700*ff1cb3adSJakub Kicinski 			break;
17018888bf4fSJakub Kicinski 		case 'v':
17028888bf4fSJakub Kicinski 			verbose = true;
17038888bf4fSJakub Kicinski 			break;
17048888bf4fSJakub Kicinski 		default:
17058888bf4fSJakub Kicinski 			error(1, 0, "%s invalid option %c\n", __func__, c);
17068888bf4fSJakub Kicinski 			break;
17078888bf4fSJakub Kicinski 		}
17088888bf4fSJakub Kicinski 	}
17098888bf4fSJakub Kicinski }
17108888bf4fSJakub Kicinski 
17118888bf4fSJakub Kicinski int main(int argc, char **argv)
17128888bf4fSJakub Kicinski {
17138888bf4fSJakub Kicinski 	parse_args(argc, argv);
17148888bf4fSJakub Kicinski 
17158888bf4fSJakub Kicinski 	if (ipip) {
17168888bf4fSJakub Kicinski 		tcp_offset = ETH_HLEN + sizeof(struct iphdr) * 2;
17178888bf4fSJakub Kicinski 		total_hdr_len = tcp_offset + sizeof(struct tcphdr);
17188888bf4fSJakub Kicinski 	} else if (proto == PF_INET) {
17198888bf4fSJakub Kicinski 		tcp_offset = ETH_HLEN + sizeof(struct iphdr);
17208888bf4fSJakub Kicinski 		total_hdr_len = tcp_offset + sizeof(struct tcphdr);
17218888bf4fSJakub Kicinski 	} else if (proto == PF_INET6) {
17228888bf4fSJakub Kicinski 		tcp_offset = ETH_HLEN + sizeof(struct ipv6hdr);
17238888bf4fSJakub Kicinski 		total_hdr_len = MAX_HDR_LEN;
17248888bf4fSJakub Kicinski 	} else {
17258888bf4fSJakub Kicinski 		error(1, 0, "Protocol family is not ipv4 or ipv6");
17268888bf4fSJakub Kicinski 	}
17278888bf4fSJakub Kicinski 
17288888bf4fSJakub Kicinski 	read_MAC(src_mac, smac);
17298888bf4fSJakub Kicinski 	read_MAC(dst_mac, dmac);
17308888bf4fSJakub Kicinski 
17318888bf4fSJakub Kicinski 	if (tx_socket) {
17328888bf4fSJakub Kicinski 		gro_sender();
17338888bf4fSJakub Kicinski 	} else {
17348888bf4fSJakub Kicinski 		/* Only the receiver exit status determines test success. */
17358888bf4fSJakub Kicinski 		gro_receiver();
17368888bf4fSJakub Kicinski 		fprintf(stderr, "Gro::%s test passed.\n", testname);
17378888bf4fSJakub Kicinski 	}
17388888bf4fSJakub Kicinski 
17398888bf4fSJakub Kicinski 	return 0;
17408888bf4fSJakub Kicinski }
1741