blob: d6d7e66aa82ff8f6570dc6d64605363b5081c5b5 [file] [log] [blame]
Lorenzo Colittif3beefc2014-02-14 13:19:27 +09001/*
2 * Copyright 2014 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 *
16 * clatd_test.cpp - unit tests for clatd
17 */
18
19#include <iostream>
20
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090021#include <arpa/inet.h>
Maciej Żenczykowski50303532020-06-02 14:46:45 -070022#include <linux/if_packet.h>
Lorenzo Colitti98089522014-10-09 22:29:45 +090023#include <netinet/in6.h>
junyulaic4e591a2018-11-26 22:36:10 +090024#include <stdio.h>
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090025#include <sys/uio.h>
26
27#include <gtest/gtest.h>
Lorenzo Colitti66deecd2019-01-04 12:27:27 +090028
29#include "netutils/ifc.h"
Lorenzo Colitti72f770c2019-01-04 15:00:28 +090030#include "tun_interface.h"
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090031
32extern "C" {
Lorenzo Colitti0cd5aa52021-12-09 15:05:52 +090033#include "checksum.h"
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090034#include "clatd.h"
junyulaic4e591a2018-11-26 22:36:10 +090035#include "config.h"
Lorenzo Colitti72f770c2019-01-04 15:00:28 +090036#include "getaddr.h"
junyulaic4e591a2018-11-26 22:36:10 +090037#include "translate.h"
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090038}
39
40// For convenience.
41#define ARRAYSIZE(x) sizeof((x)) / sizeof((x)[0])
42
Lorenzo Colitti72f770c2019-01-04 15:00:28 +090043using android::net::TunInterface;
44
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090045// Default translation parameters.
junyulaic4e591a2018-11-26 22:36:10 +090046static const char kIPv4LocalAddr[] = "192.0.0.4";
47static const char kIPv6LocalAddr[] = "2001:db8:0:b11::464";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090048static const char kIPv6PlatSubnet[] = "64:ff9b::";
49
junyulaic4e591a2018-11-26 22:36:10 +090050// clang-format off
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090051// Test packet portions. Defined as macros because it's easy to concatenate them to make packets.
52#define IPV4_HEADER(p, c1, c2) \
53 0x45, 0x00, 0, 41, /* Version=4, IHL=5, ToS=0x80, len=41 */ \
54 0x00, 0x00, 0x40, 0x00, /* ID=0x0000, flags=IP_DF, offset=0 */ \
55 55, (p), (c1), (c2), /* TTL=55, protocol=p, checksum=c1,c2 */ \
56 192, 0, 0, 4, /* Src=192.0.0.4 */ \
57 8, 8, 8, 8, /* Dst=8.8.8.8 */
58#define IPV4_UDP_HEADER IPV4_HEADER(IPPROTO_UDP, 0x73, 0xb0)
59#define IPV4_ICMP_HEADER IPV4_HEADER(IPPROTO_ICMP, 0x73, 0xc0)
60
61#define IPV6_HEADER(p) \
62 0x60, 0x00, 0, 0, /* Version=6, tclass=0x00, flowlabel=0 */ \
63 0, 21, (p), 55, /* plen=11, nxthdr=p, hlim=55 */ \
64 0x20, 0x01, 0x0d, 0xb8, /* Src=2001:db8:0:b11::464 */ \
65 0x00, 0x00, 0x0b, 0x11, \
66 0x00, 0x00, 0x00, 0x00, \
67 0x00, 0x00, 0x04, 0x64, \
68 0x00, 0x64, 0xff, 0x9b, /* Dst=64:ff9b::8.8.8.8 */ \
69 0x00, 0x00, 0x00, 0x00, \
70 0x00, 0x00, 0x00, 0x00, \
71 0x08, 0x08, 0x08, 0x08,
72#define IPV6_UDP_HEADER IPV6_HEADER(IPPROTO_UDP)
73#define IPV6_ICMPV6_HEADER IPV6_HEADER(IPPROTO_ICMPV6)
74
75#define UDP_LEN 21
76#define UDP_HEADER \
77 0xc8, 0x8b, 0, 53, /* Port 51339->53 */ \
78 0x00, UDP_LEN, 0, 0, /* Length 21, checksum empty for now */
79
80#define PAYLOAD 'H', 'e', 'l', 'l', 'o', ' ', 0x4e, 0xb8, 0x96, 0xe7, 0x95, 0x8c, 0x00
81
82#define IPV4_PING \
83 0x08, 0x00, 0x88, 0xd0, /* Type 8, code 0, checksum 0x88d0 */ \
84 0xd0, 0x0d, 0x00, 0x03, /* ID=0xd00d, seq=3 */
85
86#define IPV6_PING \
87 0x80, 0x00, 0xc3, 0x42, /* Type 128, code 0, checksum 0xc342 */ \
88 0xd0, 0x0d, 0x00, 0x03, /* ID=0xd00d, seq=3 */
89
90// Macros to return pseudo-headers from packets.
91#define IPV4_PSEUDOHEADER(ip, tlen) \
92 ip[12], ip[13], ip[14], ip[15], /* Source address */ \
93 ip[16], ip[17], ip[18], ip[19], /* Destination address */ \
94 0, ip[9], /* 0, protocol */ \
95 ((tlen) >> 16) & 0xff, (tlen) & 0xff, /* Transport length */
96
97#define IPV6_PSEUDOHEADER(ip6, protocol, tlen) \
98 ip6[8], ip6[9], ip6[10], ip6[11], /* Source address */ \
99 ip6[12], ip6[13], ip6[14], ip6[15], \
100 ip6[16], ip6[17], ip6[18], ip6[19], \
101 ip6[20], ip6[21], ip6[22], ip6[23], \
102 ip6[24], ip6[25], ip6[26], ip6[27], /* Destination address */ \
103 ip6[28], ip6[29], ip6[30], ip6[31], \
104 ip6[32], ip6[33], ip6[34], ip6[35], \
105 ip6[36], ip6[37], ip6[38], ip6[39], \
106 ((tlen) >> 24) & 0xff, /* Transport length */ \
107 ((tlen) >> 16) & 0xff, \
108 ((tlen) >> 8) & 0xff, \
109 (tlen) & 0xff, \
110 0, 0, 0, (protocol),
111
112// A fragmented DNS request.
Brian Carlstromfcac4102014-02-24 20:03:01 -0800113static const uint8_t kIPv4Frag1[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900114 0x45, 0x00, 0x00, 0x24, 0xfe, 0x47, 0x20, 0x00, 0x40, 0x11,
115 0x8c, 0x6d, 0xc0, 0x00, 0x00, 0x04, 0x08, 0x08, 0x08, 0x08,
116 0x14, 0x5d, 0x00, 0x35, 0x00, 0x29, 0x68, 0xbb, 0x50, 0x47,
117 0x01, 0x00, 0x00, 0x01, 0x00, 0x00
118};
Brian Carlstromfcac4102014-02-24 20:03:01 -0800119static const uint8_t kIPv4Frag2[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900120 0x45, 0x00, 0x00, 0x24, 0xfe, 0x47, 0x20, 0x02, 0x40, 0x11,
121 0x8c, 0x6b, 0xc0, 0x00, 0x00, 0x04, 0x08, 0x08, 0x08, 0x08,
122 0x00, 0x00, 0x00, 0x00, 0x04, 0x69, 0x70, 0x76, 0x34, 0x06,
123 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65
124};
Brian Carlstromfcac4102014-02-24 20:03:01 -0800125static const uint8_t kIPv4Frag3[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900126 0x45, 0x00, 0x00, 0x1d, 0xfe, 0x47, 0x00, 0x04, 0x40, 0x11,
127 0xac, 0x70, 0xc0, 0x00, 0x00, 0x04, 0x08, 0x08, 0x08, 0x08,
128 0x03, 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00, 0x01
129};
Brian Carlstromfcac4102014-02-24 20:03:01 -0800130static const uint8_t *kIPv4Fragments[] = { kIPv4Frag1, kIPv4Frag2, kIPv4Frag3 };
131static const size_t kIPv4FragLengths[] = { sizeof(kIPv4Frag1), sizeof(kIPv4Frag2),
132 sizeof(kIPv4Frag3) };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900133
Brian Carlstromfcac4102014-02-24 20:03:01 -0800134static const uint8_t kIPv6Frag1[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900135 0x60, 0x00, 0x00, 0x00, 0x00, 0x18, 0x2c, 0x40, 0x20, 0x01,
136 0x0d, 0xb8, 0x00, 0x00, 0x0b, 0x11, 0x00, 0x00, 0x00, 0x00,
137 0x00, 0x00, 0x04, 0x64, 0x00, 0x64, 0xff, 0x9b, 0x00, 0x00,
138 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x08, 0x08, 0x08, 0x08,
139 0x11, 0x00, 0x00, 0x01, 0x00, 0x00, 0xfe, 0x47, 0x14, 0x5d,
140 0x00, 0x35, 0x00, 0x29, 0xeb, 0x91, 0x50, 0x47, 0x01, 0x00,
141 0x00, 0x01, 0x00, 0x00
142};
143
Brian Carlstromfcac4102014-02-24 20:03:01 -0800144static const uint8_t kIPv6Frag2[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900145 0x60, 0x00, 0x00, 0x00, 0x00, 0x18, 0x2c, 0x40, 0x20, 0x01,
146 0x0d, 0xb8, 0x00, 0x00, 0x0b, 0x11, 0x00, 0x00, 0x00, 0x00,
147 0x00, 0x00, 0x04, 0x64, 0x00, 0x64, 0xff, 0x9b, 0x00, 0x00,
148 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x08, 0x08, 0x08, 0x08,
149 0x11, 0x00, 0x00, 0x11, 0x00, 0x00, 0xfe, 0x47, 0x00, 0x00,
150 0x00, 0x00, 0x04, 0x69, 0x70, 0x76, 0x34, 0x06, 0x67, 0x6f,
151 0x6f, 0x67, 0x6c, 0x65
152};
153
Brian Carlstromfcac4102014-02-24 20:03:01 -0800154static const uint8_t kIPv6Frag3[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900155 0x60, 0x00, 0x00, 0x00, 0x00, 0x11, 0x2c, 0x40, 0x20, 0x01,
156 0x0d, 0xb8, 0x00, 0x00, 0x0b, 0x11, 0x00, 0x00, 0x00, 0x00,
157 0x00, 0x00, 0x04, 0x64, 0x00, 0x64, 0xff, 0x9b, 0x00, 0x00,
158 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x08, 0x08, 0x08, 0x08,
159 0x11, 0x00, 0x00, 0x20, 0x00, 0x00, 0xfe, 0x47, 0x03, 0x63,
160 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00, 0x01
161};
Brian Carlstromfcac4102014-02-24 20:03:01 -0800162static const uint8_t *kIPv6Fragments[] = { kIPv6Frag1, kIPv6Frag2, kIPv6Frag3 };
163static const size_t kIPv6FragLengths[] = { sizeof(kIPv6Frag1), sizeof(kIPv6Frag2),
164 sizeof(kIPv6Frag3) };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900165
Brian Carlstromfcac4102014-02-24 20:03:01 -0800166static const uint8_t kReassembledIPv4[] = {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900167 0x45, 0x00, 0x00, 0x3d, 0xfe, 0x47, 0x00, 0x00, 0x40, 0x11,
168 0xac, 0x54, 0xc0, 0x00, 0x00, 0x04, 0x08, 0x08, 0x08, 0x08,
169 0x14, 0x5d, 0x00, 0x35, 0x00, 0x29, 0x68, 0xbb, 0x50, 0x47,
170 0x01, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
171 0x04, 0x69, 0x70, 0x76, 0x34, 0x06, 0x67, 0x6f, 0x6f, 0x67,
172 0x6c, 0x65, 0x03, 0x63, 0x6f, 0x6d, 0x00, 0x00, 0x01, 0x00,
173 0x01
174};
junyulaic4e591a2018-11-26 22:36:10 +0900175// clang-format on
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900176
177// Expected checksums.
178static const uint32_t kUdpPartialChecksum = 0xd5c8;
179static const uint32_t kPayloadPartialChecksum = 0x31e9c;
180static const uint16_t kUdpV4Checksum = 0xd0c7;
181static const uint16_t kUdpV6Checksum = 0xa74a;
182
Brian Carlstromfcac4102014-02-24 20:03:01 -0800183uint8_t ip_version(const uint8_t *packet) {
184 uint8_t version = packet[0] >> 4;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900185 return version;
186}
187
188int is_ipv4_fragment(struct iphdr *ip) {
189 // A packet is a fragment if its fragment offset is nonzero or if the MF flag is set.
190 return ntohs(ip->frag_off) & (IP_OFFMASK | IP_MF);
191}
192
193int is_ipv6_fragment(struct ip6_hdr *ip6, size_t len) {
194 if (ip6->ip6_nxt != IPPROTO_FRAGMENT) {
195 return 0;
196 }
junyulaic4e591a2018-11-26 22:36:10 +0900197 struct ip6_frag *frag = (struct ip6_frag *)(ip6 + 1);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900198 return len >= sizeof(*ip6) + sizeof(*frag) &&
junyulaic4e591a2018-11-26 22:36:10 +0900199 (frag->ip6f_offlg & (IP6F_OFF_MASK | IP6F_MORE_FRAG));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900200}
201
202int ipv4_fragment_offset(struct iphdr *ip) {
203 return ntohs(ip->frag_off) & IP_OFFMASK;
204}
205
206int ipv6_fragment_offset(struct ip6_frag *frag) {
207 return ntohs((frag->ip6f_offlg & IP6F_OFF_MASK) >> 3);
208}
209
Brian Carlstromfcac4102014-02-24 20:03:01 -0800210void check_packet(const uint8_t *packet, size_t len, const char *msg) {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900211 void *payload;
junyulaic4e591a2018-11-26 22:36:10 +0900212 size_t payload_length = 0;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900213 uint32_t pseudo_checksum = 0;
junyulaic4e591a2018-11-26 22:36:10 +0900214 uint8_t protocol = 0;
215 int version = ip_version(packet);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900216 switch (version) {
217 case 4: {
junyulaic4e591a2018-11-26 22:36:10 +0900218 struct iphdr *ip = (struct iphdr *)packet;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900219 ASSERT_GE(len, sizeof(*ip)) << msg << ": IPv4 packet shorter than IPv4 header\n";
220 EXPECT_EQ(5, ip->ihl) << msg << ": Unsupported IP header length\n";
221 EXPECT_EQ(len, ntohs(ip->tot_len)) << msg << ": Incorrect IPv4 length\n";
222 EXPECT_EQ(0, ip_checksum(ip, sizeof(*ip))) << msg << ": Incorrect IP checksum\n";
223 protocol = ip->protocol;
junyulaic4e591a2018-11-26 22:36:10 +0900224 payload = ip + 1;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900225 if (!is_ipv4_fragment(ip)) {
junyulaic4e591a2018-11-26 22:36:10 +0900226 payload_length = len - sizeof(*ip);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900227 pseudo_checksum = ipv4_pseudo_header_checksum(ip, payload_length);
228 }
229 ASSERT_TRUE(protocol == IPPROTO_TCP || protocol == IPPROTO_UDP || protocol == IPPROTO_ICMP)
junyulaic4e591a2018-11-26 22:36:10 +0900230 << msg << ": Unsupported IPv4 protocol " << protocol << "\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900231 break;
232 }
233 case 6: {
junyulaic4e591a2018-11-26 22:36:10 +0900234 struct ip6_hdr *ip6 = (struct ip6_hdr *)packet;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900235 ASSERT_GE(len, sizeof(*ip6)) << msg << ": IPv6 packet shorter than IPv6 header\n";
236 EXPECT_EQ(len - sizeof(*ip6), htons(ip6->ip6_plen)) << msg << ": Incorrect IPv6 length\n";
237
238 if (ip6->ip6_nxt == IPPROTO_FRAGMENT) {
junyulaic4e591a2018-11-26 22:36:10 +0900239 struct ip6_frag *frag = (struct ip6_frag *)(ip6 + 1);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900240 ASSERT_GE(len, sizeof(*ip6) + sizeof(*frag))
junyulaic4e591a2018-11-26 22:36:10 +0900241 << msg << ": IPv6 fragment: short fragment header\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900242 protocol = frag->ip6f_nxt;
junyulaic4e591a2018-11-26 22:36:10 +0900243 payload = frag + 1;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900244 // Even though the packet has a Fragment header, it might not be a fragment.
245 if (!is_ipv6_fragment(ip6, len)) {
246 payload_length = len - sizeof(*ip6) - sizeof(*frag);
247 }
248 } else {
249 // Since there are no extension headers except Fragment, this must be the payload.
junyulaic4e591a2018-11-26 22:36:10 +0900250 protocol = ip6->ip6_nxt;
251 payload = ip6 + 1;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900252 payload_length = len - sizeof(*ip6);
253 }
254 ASSERT_TRUE(protocol == IPPROTO_TCP || protocol == IPPROTO_UDP || protocol == IPPROTO_ICMPV6)
junyulaic4e591a2018-11-26 22:36:10 +0900255 << msg << ": Unsupported IPv6 next header " << protocol;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900256 if (payload_length) {
257 pseudo_checksum = ipv6_pseudo_header_checksum(ip6, payload_length, protocol);
258 }
259 break;
260 }
261 default:
262 FAIL() << msg << ": Unsupported IP version " << version << "\n";
263 return;
264 }
265
266 // If we understand the payload, verify the checksum.
267 if (payload_length) {
268 uint16_t checksum;
junyulaic4e591a2018-11-26 22:36:10 +0900269 switch (protocol) {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900270 case IPPROTO_UDP:
271 case IPPROTO_TCP:
272 case IPPROTO_ICMPV6:
273 checksum = ip_checksum_finish(ip_checksum_add(pseudo_checksum, payload, payload_length));
274 break;
275 case IPPROTO_ICMP:
276 checksum = ip_checksum(payload, payload_length);
277 break;
278 default:
279 checksum = 0; // Don't check.
280 break;
281 }
282 EXPECT_EQ(0, checksum) << msg << ": Incorrect transport checksum\n";
283 }
284
285 if (protocol == IPPROTO_UDP) {
junyulaic4e591a2018-11-26 22:36:10 +0900286 struct udphdr *udp = (struct udphdr *)payload;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900287 EXPECT_NE(0, udp->check) << msg << ": UDP checksum 0 should be 0xffff";
288 // If this is not a fragment, check the UDP length field.
289 if (payload_length) {
290 EXPECT_EQ(payload_length, ntohs(udp->len)) << msg << ": Incorrect UDP length\n";
291 }
292 }
293}
294
Brian Carlstromfcac4102014-02-24 20:03:01 -0800295void reassemble_packet(const uint8_t **fragments, const size_t lengths[], int numpackets,
296 uint8_t *reassembled, size_t *reassembled_len, const char *msg) {
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900297 struct iphdr *ip = nullptr;
298 struct ip6_hdr *ip6 = nullptr;
junyulaic4e591a2018-11-26 22:36:10 +0900299 size_t total_length, pos = 0;
Ben Cheng932614e2014-04-02 17:00:26 -0700300 uint8_t protocol = 0;
junyulaic4e591a2018-11-26 22:36:10 +0900301 uint8_t version = ip_version(fragments[0]);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900302
303 for (int i = 0; i < numpackets; i++) {
Brian Carlstromfcac4102014-02-24 20:03:01 -0800304 const uint8_t *packet = fragments[i];
junyulaic4e591a2018-11-26 22:36:10 +0900305 int len = lengths[i];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900306 int headersize, payload_offset;
307
308 ASSERT_EQ(ip_version(packet), version) << msg << ": Inconsistent fragment versions\n";
309 check_packet(packet, len, "Fragment sanity check");
310
311 switch (version) {
312 case 4: {
junyulaic4e591a2018-11-26 22:36:10 +0900313 struct iphdr *ip_orig = (struct iphdr *)packet;
314 headersize = sizeof(*ip_orig);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900315 ASSERT_TRUE(is_ipv4_fragment(ip_orig))
junyulaic4e591a2018-11-26 22:36:10 +0900316 << msg << ": IPv4 fragment #" << i + 1 << " not a fragment\n";
317 ASSERT_EQ(pos, ipv4_fragment_offset(ip_orig) * 8 + ((i != 0) ? sizeof(*ip) : 0))
318 << msg << ": IPv4 fragment #" << i + 1 << ": inconsistent offset\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900319
junyulaic4e591a2018-11-26 22:36:10 +0900320 headersize = sizeof(*ip_orig);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900321 payload_offset = headersize;
322 if (pos == 0) {
junyulaic4e591a2018-11-26 22:36:10 +0900323 ip = (struct iphdr *)reassembled;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900324 }
325 break;
326 }
327 case 6: {
junyulaic4e591a2018-11-26 22:36:10 +0900328 struct ip6_hdr *ip6_orig = (struct ip6_hdr *)packet;
329 struct ip6_frag *frag = (struct ip6_frag *)(ip6_orig + 1);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900330 ASSERT_TRUE(is_ipv6_fragment(ip6_orig, len))
junyulaic4e591a2018-11-26 22:36:10 +0900331 << msg << ": IPv6 fragment #" << i + 1 << " not a fragment\n";
332 ASSERT_EQ(pos, ipv6_fragment_offset(frag) * 8 + ((i != 0) ? sizeof(*ip6) : 0))
333 << msg << ": IPv6 fragment #" << i + 1 << ": inconsistent offset\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900334
junyulaic4e591a2018-11-26 22:36:10 +0900335 headersize = sizeof(*ip6_orig);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900336 payload_offset = sizeof(*ip6_orig) + sizeof(*frag);
337 if (pos == 0) {
junyulaic4e591a2018-11-26 22:36:10 +0900338 ip6 = (struct ip6_hdr *)reassembled;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900339 protocol = frag->ip6f_nxt;
340 }
341 break;
342 }
343 default:
344 FAIL() << msg << ": Invalid IP version << " << version;
345 }
346
347 // If this is the first fragment, copy the header.
348 if (pos == 0) {
junyulaic4e591a2018-11-26 22:36:10 +0900349 ASSERT_LT(headersize, (int)*reassembled_len) << msg << ": Reassembly buffer too small\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900350 memcpy(reassembled, packet, headersize);
351 total_length = headersize;
352 pos += headersize;
353 }
354
355 // Copy the payload.
356 int payload_length = len - payload_offset;
357 total_length += payload_length;
Ben Cheng932614e2014-04-02 17:00:26 -0700358 ASSERT_LT(total_length, *reassembled_len) << msg << ": Reassembly buffer too small\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900359 memcpy(reassembled + pos, packet + payload_offset, payload_length);
360 pos += payload_length;
361 }
362
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900363 // Fix up the reassembled headers to reflect fragmentation and length (and IPv4 checksum).
364 ASSERT_EQ(total_length, pos) << msg << ": Reassembled packet length incorrect\n";
365 if (ip) {
366 ip->frag_off &= ~htons(IP_MF);
367 ip->tot_len = htons(total_length);
junyulaic4e591a2018-11-26 22:36:10 +0900368 ip->check = 0;
369 ip->check = ip_checksum(ip, sizeof(*ip));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900370 ASSERT_FALSE(is_ipv4_fragment(ip)) << msg << ": reassembled IPv4 packet is a fragment!\n";
371 }
372 if (ip6) {
junyulaic4e591a2018-11-26 22:36:10 +0900373 ip6->ip6_nxt = protocol;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900374 ip6->ip6_plen = htons(total_length - sizeof(*ip6));
375 ASSERT_FALSE(is_ipv6_fragment(ip6, ip6->ip6_plen))
junyulaic4e591a2018-11-26 22:36:10 +0900376 << msg << ": reassembled IPv6 packet is a fragment!\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900377 }
378
379 *reassembled_len = total_length;
380}
381
Lorenzo Colitti98089522014-10-09 22:29:45 +0900382void check_data_matches(const void *expected, const void *actual, size_t len, const char *msg) {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900383 if (memcmp(expected, actual, len)) {
384 // Hex dump, 20 bytes per line, one space between bytes (1 byte = 3 chars), indented by 4.
385 int hexdump_len = len * 3 + (len / 20 + 1) * 5;
386 char expected_hexdump[hexdump_len], actual_hexdump[hexdump_len];
387 unsigned pos = 0;
388 for (unsigned i = 0; i < len; i++) {
389 if (i % 20 == 0) {
George Burgess IV2dcca6f2016-02-24 15:55:54 -0800390 snprintf(expected_hexdump + pos, hexdump_len - pos, "\n ");
391 snprintf(actual_hexdump + pos, hexdump_len - pos, "\n ");
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900392 pos += 4;
393 }
junyulaic4e591a2018-11-26 22:36:10 +0900394 snprintf(expected_hexdump + pos, hexdump_len - pos, " %02x", ((uint8_t *)expected)[i]);
395 snprintf(actual_hexdump + pos, hexdump_len - pos, " %02x", ((uint8_t *)actual)[i]);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900396 pos += 3;
397 }
Lorenzo Colitti98089522014-10-09 22:29:45 +0900398 FAIL() << msg << ": Data doesn't match"
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900399 << "\n Expected:" << (char *) expected_hexdump
400 << "\n Actual:" << (char *) actual_hexdump << "\n";
401 }
402}
403
junyulaic4e591a2018-11-26 22:36:10 +0900404void fix_udp_checksum(uint8_t *packet) {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900405 uint32_t pseudo_checksum;
Brian Carlstromfcac4102014-02-24 20:03:01 -0800406 uint8_t version = ip_version(packet);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900407 struct udphdr *udp;
408 switch (version) {
409 case 4: {
junyulaic4e591a2018-11-26 22:36:10 +0900410 struct iphdr *ip = (struct iphdr *)packet;
411 udp = (struct udphdr *)(ip + 1);
412 pseudo_checksum = ipv4_pseudo_header_checksum(ip, ntohs(udp->len));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900413 break;
414 }
415 case 6: {
junyulaic4e591a2018-11-26 22:36:10 +0900416 struct ip6_hdr *ip6 = (struct ip6_hdr *)packet;
417 udp = (struct udphdr *)(ip6 + 1);
418 pseudo_checksum = ipv6_pseudo_header_checksum(ip6, ntohs(udp->len), IPPROTO_UDP);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900419 break;
420 }
421 default:
422 FAIL() << "unsupported IP version" << version << "\n";
423 return;
junyulaic4e591a2018-11-26 22:36:10 +0900424 }
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900425
426 udp->check = 0;
427 udp->check = ip_checksum_finish(ip_checksum_add(pseudo_checksum, udp, ntohs(udp->len)));
428}
429
Lorenzo Colittice140882014-06-02 21:20:40 +0900430// Testing stub for send_rawv6. The real version uses sendmsg() with a
431// destination IPv6 address, and attempting to call that on our test socketpair
432// fd results in EINVAL.
junyulaic4e591a2018-11-26 22:36:10 +0900433extern "C" void send_rawv6(int fd, clat_packet out, int iov_len) { writev(fd, out, iov_len); }
Lorenzo Colittice140882014-06-02 21:20:40 +0900434
Brian Carlstromfcac4102014-02-24 20:03:01 -0800435void do_translate_packet(const uint8_t *original, size_t original_len, uint8_t *out, size_t *outlen,
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900436 const char *msg) {
437 int fds[2];
438 if (socketpair(AF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0, fds)) {
439 abort();
440 }
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900441
442 char foo[512];
443 snprintf(foo, sizeof(foo), "%s: Invalid original packet", msg);
444 check_packet(original, original_len, foo);
445
Lorenzo Colittie24982e2014-06-02 15:49:36 +0900446 int read_fd, write_fd;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900447 uint16_t expected_proto;
448 int version = ip_version(original);
449 switch (version) {
450 case 4:
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900451 expected_proto = htons(ETH_P_IPV6);
junyulaic4e591a2018-11-26 22:36:10 +0900452 read_fd = fds[1];
453 write_fd = fds[0];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900454 break;
455 case 6:
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900456 expected_proto = htons(ETH_P_IP);
junyulaic4e591a2018-11-26 22:36:10 +0900457 read_fd = fds[0];
458 write_fd = fds[1];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900459 break;
460 default:
461 FAIL() << msg << ": Unsupported IP version " << version << "\n";
462 break;
463 }
464
Lorenzo Colittie24982e2014-06-02 15:49:36 +0900465 translate_packet(write_fd, (version == 4), original, original_len);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900466
Lorenzo Colitti98089522014-10-09 22:29:45 +0900467 snprintf(foo, sizeof(foo), "%s: Invalid translated packet", msg);
Lorenzo Colittice140882014-06-02 21:20:40 +0900468 if (version == 6) {
469 // Translating to IPv4. Expect a tun header.
470 struct tun_pi new_tun_header;
471 struct iovec iov[] = {
472 { &new_tun_header, sizeof(new_tun_header) },
junyulaic4e591a2018-11-26 22:36:10 +0900473 { out, *outlen },
Lorenzo Colittice140882014-06-02 21:20:40 +0900474 };
junyulaic4e591a2018-11-26 22:36:10 +0900475
Lorenzo Colittice140882014-06-02 21:20:40 +0900476 int len = readv(read_fd, iov, 2);
junyulaic4e591a2018-11-26 22:36:10 +0900477 if (len > (int)sizeof(new_tun_header)) {
478 ASSERT_LT((size_t)len, *outlen) << msg << ": Translated packet buffer too small\n";
Lorenzo Colittice140882014-06-02 21:20:40 +0900479 EXPECT_EQ(expected_proto, new_tun_header.proto) << msg << "Unexpected tun proto\n";
480 *outlen = len - sizeof(new_tun_header);
Lorenzo Colitti98089522014-10-09 22:29:45 +0900481 check_packet(out, *outlen, msg);
Lorenzo Colittice140882014-06-02 21:20:40 +0900482 } else {
Lorenzo Colitti98089522014-10-09 22:29:45 +0900483 FAIL() << msg << ": Packet was not translated: len=" << len;
Lorenzo Colittice140882014-06-02 21:20:40 +0900484 *outlen = 0;
485 }
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900486 } else {
Lorenzo Colittice140882014-06-02 21:20:40 +0900487 // Translating to IPv6. Expect raw packet.
488 *outlen = read(read_fd, out, *outlen);
Lorenzo Colitti98089522014-10-09 22:29:45 +0900489 check_packet(out, *outlen, msg);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900490 }
491}
492
junyulaic4e591a2018-11-26 22:36:10 +0900493void check_translated_packet(const uint8_t *original, size_t original_len, const uint8_t *expected,
494 size_t expected_len, const char *msg) {
Brian Carlstromfcac4102014-02-24 20:03:01 -0800495 uint8_t translated[MAXMTU];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900496 size_t translated_len = sizeof(translated);
497 do_translate_packet(original, original_len, translated, &translated_len, msg);
498 EXPECT_EQ(expected_len, translated_len) << msg << ": Translated packet length incorrect\n";
499 check_data_matches(expected, translated, translated_len, msg);
500}
501
Brian Carlstromfcac4102014-02-24 20:03:01 -0800502void check_fragment_translation(const uint8_t *original[], const size_t original_lengths[],
503 const uint8_t *expected[], const size_t expected_lengths[],
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900504 int numfragments, const char *msg) {
505 for (int i = 0; i < numfragments; i++) {
506 // Check that each of the fragments translates as expected.
507 char frag_msg[512];
508 snprintf(frag_msg, sizeof(frag_msg), "%s: fragment #%d", msg, i + 1);
junyulaic4e591a2018-11-26 22:36:10 +0900509 check_translated_packet(original[i], original_lengths[i], expected[i], expected_lengths[i],
510 frag_msg);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900511 }
512
513 // Sanity check that reassembling the original and translated fragments produces valid packets.
Brian Carlstromfcac4102014-02-24 20:03:01 -0800514 uint8_t reassembled[MAXMTU];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900515 size_t reassembled_len = sizeof(reassembled);
516 reassemble_packet(original, original_lengths, numfragments, reassembled, &reassembled_len, msg);
517 check_packet(reassembled, reassembled_len, msg);
518
Brian Carlstromfcac4102014-02-24 20:03:01 -0800519 uint8_t translated[MAXMTU];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900520 size_t translated_len = sizeof(translated);
521 do_translate_packet(reassembled, reassembled_len, translated, &translated_len, msg);
522 check_packet(translated, translated_len, msg);
523}
524
Lorenzo Colitti98089522014-10-09 22:29:45 +0900525int get_transport_checksum(const uint8_t *packet) {
526 struct iphdr *ip;
527 struct ip6_hdr *ip6;
528 uint8_t protocol;
529 const void *payload;
530
531 int version = ip_version(packet);
532 switch (version) {
533 case 4:
junyulaic4e591a2018-11-26 22:36:10 +0900534 ip = (struct iphdr *)packet;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900535 if (is_ipv4_fragment(ip)) {
junyulaic4e591a2018-11-26 22:36:10 +0900536 return -1;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900537 }
538 protocol = ip->protocol;
junyulaic4e591a2018-11-26 22:36:10 +0900539 payload = ip + 1;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900540 break;
541 case 6:
junyulaic4e591a2018-11-26 22:36:10 +0900542 ip6 = (struct ip6_hdr *)packet;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900543 protocol = ip6->ip6_nxt;
junyulaic4e591a2018-11-26 22:36:10 +0900544 payload = ip6 + 1;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900545 break;
546 default:
547 return -1;
548 }
549
550 switch (protocol) {
551 case IPPROTO_UDP:
junyulaic4e591a2018-11-26 22:36:10 +0900552 return ((struct udphdr *)payload)->check;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900553
554 case IPPROTO_TCP:
junyulaic4e591a2018-11-26 22:36:10 +0900555 return ((struct tcphdr *)payload)->check;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900556
557 case IPPROTO_FRAGMENT:
558 default:
559 return -1;
560 }
561}
562
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900563static tun_data makeTunData() {
564 // Create some fake but realistic-looking sockets so update_clat_ipv6_address doesn't balk.
565 return {
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900566 .read_fd6 = socket(AF_PACKET, SOCK_DGRAM, htons(ETH_P_IPV6)),
Nick Desaulniers75f4dc62019-11-19 09:37:42 -0800567 .write_fd6 = socket(AF_INET6, SOCK_RAW | SOCK_NONBLOCK, IPPROTO_RAW),
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900568 .fd4 = socket(AF_UNIX, SOCK_DGRAM, 0),
569 };
570}
571
572void freeTunData(tun_data *tunnel) {
573 close(tunnel->write_fd6);
574 close(tunnel->read_fd6);
575 close(tunnel->fd4);
576}
577
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900578struct clat_config Global_Clatd_Config;
579
580class ClatdTest : public ::testing::Test {
581 protected:
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900582 static TunInterface sTun;
583
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900584 virtual void SetUp() {
585 inet_pton(AF_INET, kIPv4LocalAddr, &Global_Clatd_Config.ipv4_local_subnet);
586 inet_pton(AF_INET6, kIPv6PlatSubnet, &Global_Clatd_Config.plat_subnet);
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900587 memset(&Global_Clatd_Config.ipv6_local_subnet, 0, sizeof(in6_addr));
Maciej Żenczykowskiba667df2020-06-02 01:41:54 -0700588 Global_Clatd_Config.native_ipv6_interface = const_cast<char *>(sTun.name().c_str());
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900589 }
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900590
591 // Static because setting up the tun interface takes about 40ms.
592 static void SetUpTestCase() { ASSERT_EQ(0, sTun.init()); }
593
594 // Closing the socket removes the interface and IP addresses.
595 static void TearDownTestCase() { sTun.destroy(); }
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900596};
597
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900598TunInterface ClatdTest::sTun;
599
Lorenzo Colitti98089522014-10-09 22:29:45 +0900600void expect_ipv6_addr_equal(struct in6_addr *expected, struct in6_addr *actual) {
601 if (!IN6_ARE_ADDR_EQUAL(expected, actual)) {
602 char expected_str[INET6_ADDRSTRLEN], actual_str[INET6_ADDRSTRLEN];
603 inet_ntop(AF_INET6, expected, expected_str, sizeof(expected_str));
604 inet_ntop(AF_INET6, actual, actual_str, sizeof(actual_str));
605 FAIL()
606 << "Unexpected IPv6 address:: "
607 << "\n Expected: " << expected_str
608 << "\n Actual: " << actual_str
609 << "\n";
610 }
611}
612
613TEST_F(ClatdTest, TestIPv6PrefixEqual) {
614 EXPECT_TRUE(ipv6_prefix_equal(&Global_Clatd_Config.plat_subnet,
615 &Global_Clatd_Config.plat_subnet));
616 EXPECT_FALSE(ipv6_prefix_equal(&Global_Clatd_Config.plat_subnet,
617 &Global_Clatd_Config.ipv6_local_subnet));
618
619 struct in6_addr subnet2 = Global_Clatd_Config.ipv6_local_subnet;
620 EXPECT_TRUE(ipv6_prefix_equal(&Global_Clatd_Config.ipv6_local_subnet, &subnet2));
621 EXPECT_TRUE(ipv6_prefix_equal(&subnet2, &Global_Clatd_Config.ipv6_local_subnet));
622
623 subnet2.s6_addr[6] = 0xff;
624 EXPECT_FALSE(ipv6_prefix_equal(&Global_Clatd_Config.ipv6_local_subnet, &subnet2));
625 EXPECT_FALSE(ipv6_prefix_equal(&subnet2, &Global_Clatd_Config.ipv6_local_subnet));
626}
627
Maciej Żenczykowskib00113f2020-01-18 23:56:14 -0800628TEST_F(ClatdTest, DetectMtu) {
629 // ::1 with bottom 32 bits set to 1 is still ::1 which routes via lo with mtu of 64KiB
630 ASSERT_EQ(detect_mtu(&in6addr_loopback, htonl(1), 0 /*MARK_UNSET*/), 65536);
631}
632
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900633TEST_F(ClatdTest, ConfigureTunIpManual) {
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900634 // Create an interface for configure_tun_ip to configure and bring up.
635 TunInterface v4Iface;
636 ASSERT_EQ(0, v4Iface.init());
637 struct tun_data tunnel = makeTunData();
638 strlcpy(tunnel.device4, v4Iface.name().c_str(), sizeof(tunnel.device4));
639
Maciej Żenczykowskib00113f2020-01-18 23:56:14 -0800640 configure_tun_ip(&tunnel, "192.0.2.1" /* v4_addr */, 1472);
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900641 EXPECT_EQ(inet_addr("192.0.2.1"), Global_Clatd_Config.ipv4_local_subnet.s_addr);
642
643 union anyip *ip = getinterface_ip(v4Iface.name().c_str(), AF_INET);
644 ASSERT_NE(nullptr, ip);
645 EXPECT_EQ(inet_addr("192.0.2.1"), ip->ip4.s_addr);
646 free(ip);
647
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900648 v4Iface.destroy();
649}
650
Lorenzo Colitti98089522014-10-09 22:29:45 +0900651TEST_F(ClatdTest, DataSanitycheck) {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900652 // Sanity checks the data.
Brian Carlstromfcac4102014-02-24 20:03:01 -0800653 uint8_t v4_header[] = { IPV4_UDP_HEADER };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900654 ASSERT_EQ(sizeof(struct iphdr), sizeof(v4_header)) << "Test IPv4 header: incorrect length\n";
655
Brian Carlstromfcac4102014-02-24 20:03:01 -0800656 uint8_t v6_header[] = { IPV6_UDP_HEADER };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900657 ASSERT_EQ(sizeof(struct ip6_hdr), sizeof(v6_header)) << "Test IPv6 header: incorrect length\n";
658
Brian Carlstromfcac4102014-02-24 20:03:01 -0800659 uint8_t udp_header[] = { UDP_HEADER };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900660 ASSERT_EQ(sizeof(struct udphdr), sizeof(udp_header)) << "Test UDP header: incorrect length\n";
661
662 // Sanity checks check_packet.
663 struct udphdr *udp;
Brian Carlstromfcac4102014-02-24 20:03:01 -0800664 uint8_t v4_udp_packet[] = { IPV4_UDP_HEADER UDP_HEADER PAYLOAD };
junyulaic4e591a2018-11-26 22:36:10 +0900665 udp = (struct udphdr *)(v4_udp_packet + sizeof(struct iphdr));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900666 fix_udp_checksum(v4_udp_packet);
667 ASSERT_EQ(kUdpV4Checksum, udp->check) << "UDP/IPv4 packet checksum sanity check\n";
668 check_packet(v4_udp_packet, sizeof(v4_udp_packet), "UDP/IPv4 packet sanity check");
669
Brian Carlstromfcac4102014-02-24 20:03:01 -0800670 uint8_t v6_udp_packet[] = { IPV6_UDP_HEADER UDP_HEADER PAYLOAD };
junyulaic4e591a2018-11-26 22:36:10 +0900671 udp = (struct udphdr *)(v6_udp_packet + sizeof(struct ip6_hdr));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900672 fix_udp_checksum(v6_udp_packet);
673 ASSERT_EQ(kUdpV6Checksum, udp->check) << "UDP/IPv6 packet checksum sanity check\n";
674 check_packet(v6_udp_packet, sizeof(v6_udp_packet), "UDP/IPv6 packet sanity check");
675
Brian Carlstromfcac4102014-02-24 20:03:01 -0800676 uint8_t ipv4_ping[] = { IPV4_ICMP_HEADER IPV4_PING PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900677 check_packet(ipv4_ping, sizeof(ipv4_ping), "IPv4 ping sanity check");
678
Brian Carlstromfcac4102014-02-24 20:03:01 -0800679 uint8_t ipv6_ping[] = { IPV6_ICMPV6_HEADER IPV6_PING PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900680 check_packet(ipv6_ping, sizeof(ipv6_ping), "IPv6 ping sanity check");
681
682 // Sanity checks reassemble_packet.
Brian Carlstromfcac4102014-02-24 20:03:01 -0800683 uint8_t reassembled[MAXMTU];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900684 size_t total_length = sizeof(reassembled);
junyulaic4e591a2018-11-26 22:36:10 +0900685 reassemble_packet(kIPv4Fragments, kIPv4FragLengths, ARRAYSIZE(kIPv4Fragments), reassembled,
686 &total_length, "Reassembly sanity check");
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900687 check_packet(reassembled, total_length, "IPv4 Reassembled packet is valid");
688 ASSERT_EQ(sizeof(kReassembledIPv4), total_length) << "IPv4 reassembly sanity check: length\n";
junyulaic4e591a2018-11-26 22:36:10 +0900689 ASSERT_TRUE(!is_ipv4_fragment((struct iphdr *)reassembled))
690 << "Sanity check: reassembled packet is a fragment!\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900691 check_data_matches(kReassembledIPv4, reassembled, total_length, "IPv4 reassembly sanity check");
692
693 total_length = sizeof(reassembled);
junyulaic4e591a2018-11-26 22:36:10 +0900694 reassemble_packet(kIPv6Fragments, kIPv6FragLengths, ARRAYSIZE(kIPv6Fragments), reassembled,
695 &total_length, "IPv6 reassembly sanity check");
696 ASSERT_TRUE(!is_ipv6_fragment((struct ip6_hdr *)reassembled, total_length))
697 << "Sanity check: reassembled packet is a fragment!\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900698 check_packet(reassembled, total_length, "IPv6 Reassembled packet is valid");
699}
700
701TEST_F(ClatdTest, PseudoChecksum) {
702 uint32_t pseudo_checksum;
703
junyulaic4e591a2018-11-26 22:36:10 +0900704 uint8_t v4_header[] = { IPV4_UDP_HEADER };
Brian Carlstromfcac4102014-02-24 20:03:01 -0800705 uint8_t v4_pseudo_header[] = { IPV4_PSEUDOHEADER(v4_header, UDP_LEN) };
junyulaic4e591a2018-11-26 22:36:10 +0900706 pseudo_checksum = ipv4_pseudo_header_checksum((struct iphdr *)v4_header, UDP_LEN);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900707 EXPECT_EQ(ip_checksum_finish(pseudo_checksum),
708 ip_checksum(v4_pseudo_header, sizeof(v4_pseudo_header)))
junyulaic4e591a2018-11-26 22:36:10 +0900709 << "ipv4_pseudo_header_checksum incorrect\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900710
junyulaic4e591a2018-11-26 22:36:10 +0900711 uint8_t v6_header[] = { IPV6_UDP_HEADER };
Brian Carlstromfcac4102014-02-24 20:03:01 -0800712 uint8_t v6_pseudo_header[] = { IPV6_PSEUDOHEADER(v6_header, IPPROTO_UDP, UDP_LEN) };
junyulaic4e591a2018-11-26 22:36:10 +0900713 pseudo_checksum = ipv6_pseudo_header_checksum((struct ip6_hdr *)v6_header, UDP_LEN, IPPROTO_UDP);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900714 EXPECT_EQ(ip_checksum_finish(pseudo_checksum),
715 ip_checksum(v6_pseudo_header, sizeof(v6_pseudo_header)))
junyulaic4e591a2018-11-26 22:36:10 +0900716 << "ipv6_pseudo_header_checksum incorrect\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900717}
718
719TEST_F(ClatdTest, TransportChecksum) {
junyulaic4e591a2018-11-26 22:36:10 +0900720 uint8_t udphdr[] = { UDP_HEADER };
Brian Carlstromfcac4102014-02-24 20:03:01 -0800721 uint8_t payload[] = { PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900722 EXPECT_EQ(kUdpPartialChecksum, ip_checksum_add(0, udphdr, sizeof(udphdr)))
junyulaic4e591a2018-11-26 22:36:10 +0900723 << "UDP partial checksum\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900724 EXPECT_EQ(kPayloadPartialChecksum, ip_checksum_add(0, payload, sizeof(payload)))
junyulaic4e591a2018-11-26 22:36:10 +0900725 << "Payload partial checksum\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900726
junyulaic4e591a2018-11-26 22:36:10 +0900727 uint8_t ip[] = { IPV4_UDP_HEADER };
728 uint8_t ip6[] = { IPV6_UDP_HEADER };
729 uint32_t ipv4_pseudo_sum = ipv4_pseudo_header_checksum((struct iphdr *)ip, UDP_LEN);
730 uint32_t ipv6_pseudo_sum =
731 ipv6_pseudo_header_checksum((struct ip6_hdr *)ip6, UDP_LEN, IPPROTO_UDP);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900732
Maciej Żenczykowskif037e772019-11-10 23:02:13 -0800733 EXPECT_NE(0, ipv4_pseudo_sum);
734 EXPECT_NE(0, ipv6_pseudo_sum);
735 EXPECT_EQ(0x3ad0U, ipv4_pseudo_sum % 0xFFFF) << "IPv4 pseudo-checksum sanity check\n";
736 EXPECT_EQ(0x644dU, ipv6_pseudo_sum % 0xFFFF) << "IPv6 pseudo-checksum sanity check\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900737 EXPECT_EQ(
738 kUdpV4Checksum,
739 ip_checksum_finish(ipv4_pseudo_sum + kUdpPartialChecksum + kPayloadPartialChecksum))
740 << "Unexpected UDP/IPv4 checksum\n";
741 EXPECT_EQ(
742 kUdpV6Checksum,
743 ip_checksum_finish(ipv6_pseudo_sum + kUdpPartialChecksum + kPayloadPartialChecksum))
744 << "Unexpected UDP/IPv6 checksum\n";
745
746 EXPECT_EQ(kUdpV6Checksum,
747 ip_checksum_adjust(kUdpV4Checksum, ipv4_pseudo_sum, ipv6_pseudo_sum))
748 << "Adjust IPv4/UDP checksum to IPv6\n";
749 EXPECT_EQ(kUdpV4Checksum,
750 ip_checksum_adjust(kUdpV6Checksum, ipv6_pseudo_sum, ipv4_pseudo_sum))
751 << "Adjust IPv6/UDP checksum to IPv4\n";
752}
753
754TEST_F(ClatdTest, AdjustChecksum) {
755 struct checksum_data {
756 uint16_t checksum;
757 uint32_t old_hdr_sum;
758 uint32_t new_hdr_sum;
759 uint16_t result;
760 } DATA[] = {
761 { 0x1423, 0xb8ec, 0x2d757, 0xf5b5 },
762 { 0xf5b5, 0x2d757, 0xb8ec, 0x1423 },
763 { 0xdd2f, 0x5555, 0x3285, 0x0000 },
764 { 0x1215, 0x5560, 0x15560 + 20, 0x1200 },
765 { 0xd0c7, 0x3ad0, 0x2644b, 0xa74a },
766 };
Chih-Hung Hsieh05ff5082014-08-26 10:46:26 -0700767 unsigned i = 0;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900768
769 for (i = 0; i < ARRAYSIZE(DATA); i++) {
770 struct checksum_data *data = DATA + i;
771 uint16_t result = ip_checksum_adjust(data->checksum, data->old_hdr_sum, data->new_hdr_sum);
772 EXPECT_EQ(result, data->result)
773 << "Incorrect checksum" << std::showbase << std::hex
774 << "\n Expected: " << data->result
775 << "\n Actual: " << result
776 << "\n checksum=" << data->checksum
777 << " old_sum=" << data->old_hdr_sum << " new_sum=" << data->new_hdr_sum << "\n";
778 }
779}
780
781TEST_F(ClatdTest, Translate) {
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900782 // This test uses hardcoded packets so the clatd address must be fixed.
783 inet_pton(AF_INET6, kIPv6LocalAddr, &Global_Clatd_Config.ipv6_local_subnet);
784
Brian Carlstromfcac4102014-02-24 20:03:01 -0800785 uint8_t udp_ipv4[] = { IPV4_UDP_HEADER UDP_HEADER PAYLOAD };
786 uint8_t udp_ipv6[] = { IPV6_UDP_HEADER UDP_HEADER PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900787 fix_udp_checksum(udp_ipv4);
788 fix_udp_checksum(udp_ipv6);
789 check_translated_packet(udp_ipv4, sizeof(udp_ipv4), udp_ipv6, sizeof(udp_ipv6),
790 "UDP/IPv4 -> UDP/IPv6 translation");
791 check_translated_packet(udp_ipv6, sizeof(udp_ipv6), udp_ipv4, sizeof(udp_ipv4),
792 "UDP/IPv6 -> UDP/IPv4 translation");
793
Brian Carlstromfcac4102014-02-24 20:03:01 -0800794 uint8_t ipv4_ping[] = { IPV4_ICMP_HEADER IPV4_PING PAYLOAD };
795 uint8_t ipv6_ping[] = { IPV6_ICMPV6_HEADER IPV6_PING PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900796 check_translated_packet(ipv4_ping, sizeof(ipv4_ping), ipv6_ping, sizeof(ipv6_ping),
797 "ICMP->ICMPv6 translation");
798 check_translated_packet(ipv6_ping, sizeof(ipv6_ping), ipv4_ping, sizeof(ipv4_ping),
799 "ICMPv6->ICMP translation");
800}
801
802TEST_F(ClatdTest, Fragmentation) {
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900803 // This test uses hardcoded packets so the clatd address must be fixed.
804 inet_pton(AF_INET6, kIPv6LocalAddr, &Global_Clatd_Config.ipv6_local_subnet);
805
junyulaic4e591a2018-11-26 22:36:10 +0900806 check_fragment_translation(kIPv4Fragments, kIPv4FragLengths, kIPv6Fragments, kIPv6FragLengths,
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900807 ARRAYSIZE(kIPv4Fragments), "IPv4->IPv6 fragment translation");
808
junyulaic4e591a2018-11-26 22:36:10 +0900809 check_fragment_translation(kIPv6Fragments, kIPv6FragLengths, kIPv4Fragments, kIPv4FragLengths,
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900810 ARRAYSIZE(kIPv6Fragments), "IPv6->IPv4 fragment translation");
811}
Lorenzo Colitti98089522014-10-09 22:29:45 +0900812
Lorenzo Colittibaa3c6a2020-06-02 01:55:12 +0900813// picks a random interface ID that is checksum neutral with the IPv4 address and the NAT64 prefix
814void gen_random_iid(struct in6_addr *myaddr, struct in_addr *ipv4_local_subnet,
815 struct in6_addr *plat_subnet) {
816 // Fill last 8 bytes of IPv6 address with random bits.
817 arc4random_buf(&myaddr->s6_addr[8], 8);
818
819 // Make the IID checksum-neutral. That is, make it so that:
820 // checksum(Local IPv4 | Remote IPv4) = checksum(Local IPv6 | Remote IPv6)
821 // in other words (because remote IPv6 = NAT64 prefix | Remote IPv4):
822 // checksum(Local IPv4) = checksum(Local IPv6 | NAT64 prefix)
823 // Do this by adjusting the two bytes in the middle of the IID.
824
825 uint16_t middlebytes = (myaddr->s6_addr[11] << 8) + myaddr->s6_addr[12];
826
827 uint32_t c1 = ip_checksum_add(0, ipv4_local_subnet, sizeof(*ipv4_local_subnet));
828 uint32_t c2 = ip_checksum_add(0, plat_subnet, sizeof(*plat_subnet)) +
829 ip_checksum_add(0, myaddr, sizeof(*myaddr));
830
831 uint16_t delta = ip_checksum_adjust(middlebytes, c1, c2);
832 myaddr->s6_addr[11] = delta >> 8;
833 myaddr->s6_addr[12] = delta & 0xff;
834}
835
Lorenzo Colitti98089522014-10-09 22:29:45 +0900836void check_translate_checksum_neutral(const uint8_t *original, size_t original_len,
837 size_t expected_len, const char *msg) {
838 uint8_t translated[MAXMTU];
839 size_t translated_len = sizeof(translated);
840 do_translate_packet(original, original_len, translated, &translated_len, msg);
841 EXPECT_EQ(expected_len, translated_len) << msg << ": Translated packet length incorrect\n";
842 // do_translate_packet already checks packets for validity and verifies the checksum.
junyulaic4e591a2018-11-26 22:36:10 +0900843 int original_check = get_transport_checksum(original);
Lorenzo Colitti98089522014-10-09 22:29:45 +0900844 int translated_check = get_transport_checksum(translated);
845 ASSERT_NE(-1, original_check);
846 ASSERT_NE(-1, translated_check);
847 ASSERT_EQ(original_check, translated_check)
junyulaic4e591a2018-11-26 22:36:10 +0900848 << "Not checksum neutral: original and translated checksums differ\n";
Lorenzo Colitti98089522014-10-09 22:29:45 +0900849}
850
851TEST_F(ClatdTest, TranslateChecksumNeutral) {
852 // Generate a random clat IPv6 address and check that translation is checksum-neutral.
Lorenzo Colitti98089522014-10-09 22:29:45 +0900853 ASSERT_TRUE(inet_pton(AF_INET6, "2001:db8:1:2:f076:ae99:124e:aa54",
854 &Global_Clatd_Config.ipv6_local_subnet));
Lorenzo Colittibaa3c6a2020-06-02 01:55:12 +0900855
856 gen_random_iid(&Global_Clatd_Config.ipv6_local_subnet, &Global_Clatd_Config.ipv4_local_subnet,
857 &Global_Clatd_Config.plat_subnet);
858
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900859 ASSERT_NE(htonl((uint32_t)0x00000464), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[3]);
junyulaic4e591a2018-11-26 22:36:10 +0900860 ASSERT_NE((uint32_t)0, Global_Clatd_Config.ipv6_local_subnet.s6_addr32[3]);
Lorenzo Colitti98089522014-10-09 22:29:45 +0900861
862 // Check that translating UDP packets is checksum-neutral. First, IPv4.
863 uint8_t udp_ipv4[] = { IPV4_UDP_HEADER UDP_HEADER PAYLOAD };
864 fix_udp_checksum(udp_ipv4);
865 check_translate_checksum_neutral(udp_ipv4, sizeof(udp_ipv4), sizeof(udp_ipv4) + 20,
866 "UDP/IPv4 -> UDP/IPv6 checksum neutral");
867
868 // Now try IPv6.
869 uint8_t udp_ipv6[] = { IPV6_UDP_HEADER UDP_HEADER PAYLOAD };
870 // The test packet uses the static IID, not the random IID. Fix up the source address.
junyulaic4e591a2018-11-26 22:36:10 +0900871 struct ip6_hdr *ip6 = (struct ip6_hdr *)udp_ipv6;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900872 memcpy(&ip6->ip6_src, &Global_Clatd_Config.ipv6_local_subnet, sizeof(ip6->ip6_src));
873 fix_udp_checksum(udp_ipv6);
874 check_translate_checksum_neutral(udp_ipv4, sizeof(udp_ipv4), sizeof(udp_ipv4) + 20,
875 "UDP/IPv4 -> UDP/IPv6 checksum neutral");
876}
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900877
878TEST_F(ClatdTest, GetInterfaceIp) {
879 union anyip *ip = getinterface_ip(sTun.name().c_str(), AF_INET6);
880 ASSERT_NE(nullptr, ip);
881 in6_addr expected = sTun.srcAddr();
882 in6_addr actual = ip->ip6;
883 expect_ipv6_addr_equal(&expected, &actual);
884}
885
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900886void expectSocketBound(int ifindex, int sock) {
887 // Check that the packet socket is bound to the interface. We can't check the socket filter
888 // because there is no way to fetch it from the kernel.
889 sockaddr_ll sll;
890 socklen_t len = sizeof(sll);
891 ASSERT_EQ(0, getsockname(sock, reinterpret_cast<sockaddr *>(&sll), &len));
892 EXPECT_EQ(htons(ETH_P_IPV6), sll.sll_protocol);
893 EXPECT_EQ(ifindex, sll.sll_ifindex);
894}
895
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900896TEST_F(ClatdTest, ConfigureIpv6Address) {
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900897 struct tun_data tunnel = makeTunData();
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900898
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900899 ASSERT_TRUE(IN6_IS_ADDR_UNSPECIFIED(&Global_Clatd_Config.ipv6_local_subnet));
900
901 const char *addrStr = "2001:db8::f00";
902 in6_addr addr;
903 ASSERT_EQ(1, inet_pton(AF_INET6, addrStr, &addr));
904 ASSERT_EQ(1, configure_clat_ipv6_address(&tunnel, sTun.name().c_str(), addrStr));
905
906 EXPECT_EQ(htonl(0x20010db8), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[0]);
907 EXPECT_EQ(htonl(0x00000000), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[1]);
908 EXPECT_EQ(htonl(0x00000000), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[2]);
909 EXPECT_EQ(htonl(0x00000f00), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[3]);
910
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900911 // Check that the packet socket is bound to the interface. We can't check the socket filter
912 // because there is no way to fetch it from the kernel.
913 sockaddr_ll sll;
914 socklen_t len = sizeof(sll);
915 ASSERT_EQ(0, getsockname(tunnel.read_fd6, reinterpret_cast<sockaddr *>(&sll), &len));
916 EXPECT_EQ(htons(ETH_P_IPV6), sll.sll_protocol);
917 EXPECT_EQ(sll.sll_ifindex, sTun.ifindex());
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900918
919 expectSocketBound(sTun.ifindex(), tunnel.read_fd6);
920
921 freeTunData(&tunnel);
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900922}
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900923
924TEST_F(ClatdTest, Ipv6AddressChanged) {
925 // Configure the clat IPv6 address.
926 struct tun_data tunnel = {
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900927 .read_fd6 = socket(AF_PACKET, SOCK_DGRAM, htons(ETH_P_IPV6)),
Nick Desaulniers75f4dc62019-11-19 09:37:42 -0800928 .write_fd6 = socket(AF_INET6, SOCK_RAW | SOCK_NONBLOCK, IPPROTO_RAW),
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900929 };
930 const char *ifname = sTun.name().c_str();
Lorenzo Colittibaa3c6a2020-06-02 01:55:12 +0900931
932 in6_addr myaddr = sTun.srcAddr();
933 gen_random_iid(&myaddr, &Global_Clatd_Config.ipv4_local_subnet, &Global_Clatd_Config.plat_subnet);
934 char addrstr[INET6_ADDRSTRLEN];
935 ASSERT_NE(nullptr, inet_ntop(AF_INET6, &myaddr, addrstr, sizeof(addrstr)));
936
937 ASSERT_EQ(1, configure_clat_ipv6_address(&tunnel, ifname, addrstr));
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900938 EXPECT_EQ(0, ipv6_address_changed(ifname));
939 EXPECT_EQ(0, ipv6_address_changed(ifname));
940
941 // Change the IP address on the tun interface to a new prefix.
942 char srcaddr[INET6_ADDRSTRLEN];
943 char dstaddr[INET6_ADDRSTRLEN];
944 ASSERT_NE(nullptr, inet_ntop(AF_INET6, &sTun.srcAddr(), srcaddr, sizeof(srcaddr)));
945 ASSERT_NE(nullptr, inet_ntop(AF_INET6, &sTun.dstAddr(), dstaddr, sizeof(dstaddr)));
946 EXPECT_EQ(0, ifc_del_address(ifname, srcaddr, 64));
947 EXPECT_EQ(0, ifc_del_address(ifname, dstaddr, 64));
948
949 // Check that we can tell that the address has changed.
950 EXPECT_EQ(0, ifc_add_address(ifname, "2001:db8::1:2", 64));
951 EXPECT_EQ(1, ipv6_address_changed(ifname));
952 EXPECT_EQ(1, ipv6_address_changed(ifname));
953
954 // Restore the tun interface configuration.
955 sTun.destroy();
956 ASSERT_EQ(0, sTun.init());
957}