blob: e24b03665bec2d9c5dabb41bae5bfc493be6fe4c [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>
Lorenzo Colitti98089522014-10-09 22:29:45 +090022#include <netinet/in6.h>
junyulaic4e591a2018-11-26 22:36:10 +090023#include <stdio.h>
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090024#include <sys/uio.h>
25
26#include <gtest/gtest.h>
Lorenzo Colitti66deecd2019-01-04 12:27:27 +090027
28#include "netutils/ifc.h"
Lorenzo Colitti72f770c2019-01-04 15:00:28 +090029#include "tun_interface.h"
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090030
31extern "C" {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +090032#include "clatd.h"
junyulaic4e591a2018-11-26 22:36:10 +090033#include "config.h"
Lorenzo Colitti72f770c2019-01-04 15:00:28 +090034#include "getaddr.h"
Lorenzo Colitti98de5952019-01-20 11:45:03 +090035#include "netutils/checksum.h"
junyulaic4e591a2018-11-26 22:36:10 +090036#include "translate.h"
Lorenzo Colitti72f770c2019-01-04 15:00:28 +090037#include "tun.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));
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900588 Global_Clatd_Config.default_pdp_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 Global_Clatd_Config.ipv4_local_prefixlen = 29;
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900635
636 // Create an interface for configure_tun_ip to configure and bring up.
637 TunInterface v4Iface;
638 ASSERT_EQ(0, v4Iface.init());
639 struct tun_data tunnel = makeTunData();
640 strlcpy(tunnel.device4, v4Iface.name().c_str(), sizeof(tunnel.device4));
641
Maciej Żenczykowskib00113f2020-01-18 23:56:14 -0800642 configure_tun_ip(&tunnel, "192.0.2.1" /* v4_addr */, 1472);
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900643 EXPECT_EQ(inet_addr("192.0.2.1"), Global_Clatd_Config.ipv4_local_subnet.s_addr);
644
645 union anyip *ip = getinterface_ip(v4Iface.name().c_str(), AF_INET);
646 ASSERT_NE(nullptr, ip);
647 EXPECT_EQ(inet_addr("192.0.2.1"), ip->ip4.s_addr);
648 free(ip);
649
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900650 v4Iface.destroy();
651}
652
Lorenzo Colitti98089522014-10-09 22:29:45 +0900653TEST_F(ClatdTest, DataSanitycheck) {
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900654 // Sanity checks the data.
Brian Carlstromfcac4102014-02-24 20:03:01 -0800655 uint8_t v4_header[] = { IPV4_UDP_HEADER };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900656 ASSERT_EQ(sizeof(struct iphdr), sizeof(v4_header)) << "Test IPv4 header: incorrect length\n";
657
Brian Carlstromfcac4102014-02-24 20:03:01 -0800658 uint8_t v6_header[] = { IPV6_UDP_HEADER };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900659 ASSERT_EQ(sizeof(struct ip6_hdr), sizeof(v6_header)) << "Test IPv6 header: incorrect length\n";
660
Brian Carlstromfcac4102014-02-24 20:03:01 -0800661 uint8_t udp_header[] = { UDP_HEADER };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900662 ASSERT_EQ(sizeof(struct udphdr), sizeof(udp_header)) << "Test UDP header: incorrect length\n";
663
664 // Sanity checks check_packet.
665 struct udphdr *udp;
Brian Carlstromfcac4102014-02-24 20:03:01 -0800666 uint8_t v4_udp_packet[] = { IPV4_UDP_HEADER UDP_HEADER PAYLOAD };
junyulaic4e591a2018-11-26 22:36:10 +0900667 udp = (struct udphdr *)(v4_udp_packet + sizeof(struct iphdr));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900668 fix_udp_checksum(v4_udp_packet);
669 ASSERT_EQ(kUdpV4Checksum, udp->check) << "UDP/IPv4 packet checksum sanity check\n";
670 check_packet(v4_udp_packet, sizeof(v4_udp_packet), "UDP/IPv4 packet sanity check");
671
Brian Carlstromfcac4102014-02-24 20:03:01 -0800672 uint8_t v6_udp_packet[] = { IPV6_UDP_HEADER UDP_HEADER PAYLOAD };
junyulaic4e591a2018-11-26 22:36:10 +0900673 udp = (struct udphdr *)(v6_udp_packet + sizeof(struct ip6_hdr));
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900674 fix_udp_checksum(v6_udp_packet);
675 ASSERT_EQ(kUdpV6Checksum, udp->check) << "UDP/IPv6 packet checksum sanity check\n";
676 check_packet(v6_udp_packet, sizeof(v6_udp_packet), "UDP/IPv6 packet sanity check");
677
Brian Carlstromfcac4102014-02-24 20:03:01 -0800678 uint8_t ipv4_ping[] = { IPV4_ICMP_HEADER IPV4_PING PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900679 check_packet(ipv4_ping, sizeof(ipv4_ping), "IPv4 ping sanity check");
680
Brian Carlstromfcac4102014-02-24 20:03:01 -0800681 uint8_t ipv6_ping[] = { IPV6_ICMPV6_HEADER IPV6_PING PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900682 check_packet(ipv6_ping, sizeof(ipv6_ping), "IPv6 ping sanity check");
683
684 // Sanity checks reassemble_packet.
Brian Carlstromfcac4102014-02-24 20:03:01 -0800685 uint8_t reassembled[MAXMTU];
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900686 size_t total_length = sizeof(reassembled);
junyulaic4e591a2018-11-26 22:36:10 +0900687 reassemble_packet(kIPv4Fragments, kIPv4FragLengths, ARRAYSIZE(kIPv4Fragments), reassembled,
688 &total_length, "Reassembly sanity check");
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900689 check_packet(reassembled, total_length, "IPv4 Reassembled packet is valid");
690 ASSERT_EQ(sizeof(kReassembledIPv4), total_length) << "IPv4 reassembly sanity check: length\n";
junyulaic4e591a2018-11-26 22:36:10 +0900691 ASSERT_TRUE(!is_ipv4_fragment((struct iphdr *)reassembled))
692 << "Sanity check: reassembled packet is a fragment!\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900693 check_data_matches(kReassembledIPv4, reassembled, total_length, "IPv4 reassembly sanity check");
694
695 total_length = sizeof(reassembled);
junyulaic4e591a2018-11-26 22:36:10 +0900696 reassemble_packet(kIPv6Fragments, kIPv6FragLengths, ARRAYSIZE(kIPv6Fragments), reassembled,
697 &total_length, "IPv6 reassembly sanity check");
698 ASSERT_TRUE(!is_ipv6_fragment((struct ip6_hdr *)reassembled, total_length))
699 << "Sanity check: reassembled packet is a fragment!\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900700 check_packet(reassembled, total_length, "IPv6 Reassembled packet is valid");
701}
702
703TEST_F(ClatdTest, PseudoChecksum) {
704 uint32_t pseudo_checksum;
705
junyulaic4e591a2018-11-26 22:36:10 +0900706 uint8_t v4_header[] = { IPV4_UDP_HEADER };
Brian Carlstromfcac4102014-02-24 20:03:01 -0800707 uint8_t v4_pseudo_header[] = { IPV4_PSEUDOHEADER(v4_header, UDP_LEN) };
junyulaic4e591a2018-11-26 22:36:10 +0900708 pseudo_checksum = ipv4_pseudo_header_checksum((struct iphdr *)v4_header, UDP_LEN);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900709 EXPECT_EQ(ip_checksum_finish(pseudo_checksum),
710 ip_checksum(v4_pseudo_header, sizeof(v4_pseudo_header)))
junyulaic4e591a2018-11-26 22:36:10 +0900711 << "ipv4_pseudo_header_checksum incorrect\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900712
junyulaic4e591a2018-11-26 22:36:10 +0900713 uint8_t v6_header[] = { IPV6_UDP_HEADER };
Brian Carlstromfcac4102014-02-24 20:03:01 -0800714 uint8_t v6_pseudo_header[] = { IPV6_PSEUDOHEADER(v6_header, IPPROTO_UDP, UDP_LEN) };
junyulaic4e591a2018-11-26 22:36:10 +0900715 pseudo_checksum = ipv6_pseudo_header_checksum((struct ip6_hdr *)v6_header, UDP_LEN, IPPROTO_UDP);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900716 EXPECT_EQ(ip_checksum_finish(pseudo_checksum),
717 ip_checksum(v6_pseudo_header, sizeof(v6_pseudo_header)))
junyulaic4e591a2018-11-26 22:36:10 +0900718 << "ipv6_pseudo_header_checksum incorrect\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900719}
720
721TEST_F(ClatdTest, TransportChecksum) {
junyulaic4e591a2018-11-26 22:36:10 +0900722 uint8_t udphdr[] = { UDP_HEADER };
Brian Carlstromfcac4102014-02-24 20:03:01 -0800723 uint8_t payload[] = { PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900724 EXPECT_EQ(kUdpPartialChecksum, ip_checksum_add(0, udphdr, sizeof(udphdr)))
junyulaic4e591a2018-11-26 22:36:10 +0900725 << "UDP partial checksum\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900726 EXPECT_EQ(kPayloadPartialChecksum, ip_checksum_add(0, payload, sizeof(payload)))
junyulaic4e591a2018-11-26 22:36:10 +0900727 << "Payload partial checksum\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900728
junyulaic4e591a2018-11-26 22:36:10 +0900729 uint8_t ip[] = { IPV4_UDP_HEADER };
730 uint8_t ip6[] = { IPV6_UDP_HEADER };
731 uint32_t ipv4_pseudo_sum = ipv4_pseudo_header_checksum((struct iphdr *)ip, UDP_LEN);
732 uint32_t ipv6_pseudo_sum =
733 ipv6_pseudo_header_checksum((struct ip6_hdr *)ip6, UDP_LEN, IPPROTO_UDP);
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900734
Maciej Żenczykowskif037e772019-11-10 23:02:13 -0800735 EXPECT_NE(0, ipv4_pseudo_sum);
736 EXPECT_NE(0, ipv6_pseudo_sum);
737 EXPECT_EQ(0x3ad0U, ipv4_pseudo_sum % 0xFFFF) << "IPv4 pseudo-checksum sanity check\n";
738 EXPECT_EQ(0x644dU, ipv6_pseudo_sum % 0xFFFF) << "IPv6 pseudo-checksum sanity check\n";
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900739 EXPECT_EQ(
740 kUdpV4Checksum,
741 ip_checksum_finish(ipv4_pseudo_sum + kUdpPartialChecksum + kPayloadPartialChecksum))
742 << "Unexpected UDP/IPv4 checksum\n";
743 EXPECT_EQ(
744 kUdpV6Checksum,
745 ip_checksum_finish(ipv6_pseudo_sum + kUdpPartialChecksum + kPayloadPartialChecksum))
746 << "Unexpected UDP/IPv6 checksum\n";
747
748 EXPECT_EQ(kUdpV6Checksum,
749 ip_checksum_adjust(kUdpV4Checksum, ipv4_pseudo_sum, ipv6_pseudo_sum))
750 << "Adjust IPv4/UDP checksum to IPv6\n";
751 EXPECT_EQ(kUdpV4Checksum,
752 ip_checksum_adjust(kUdpV6Checksum, ipv6_pseudo_sum, ipv4_pseudo_sum))
753 << "Adjust IPv6/UDP checksum to IPv4\n";
754}
755
756TEST_F(ClatdTest, AdjustChecksum) {
757 struct checksum_data {
758 uint16_t checksum;
759 uint32_t old_hdr_sum;
760 uint32_t new_hdr_sum;
761 uint16_t result;
762 } DATA[] = {
763 { 0x1423, 0xb8ec, 0x2d757, 0xf5b5 },
764 { 0xf5b5, 0x2d757, 0xb8ec, 0x1423 },
765 { 0xdd2f, 0x5555, 0x3285, 0x0000 },
766 { 0x1215, 0x5560, 0x15560 + 20, 0x1200 },
767 { 0xd0c7, 0x3ad0, 0x2644b, 0xa74a },
768 };
Chih-Hung Hsieh05ff5082014-08-26 10:46:26 -0700769 unsigned i = 0;
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900770
771 for (i = 0; i < ARRAYSIZE(DATA); i++) {
772 struct checksum_data *data = DATA + i;
773 uint16_t result = ip_checksum_adjust(data->checksum, data->old_hdr_sum, data->new_hdr_sum);
774 EXPECT_EQ(result, data->result)
775 << "Incorrect checksum" << std::showbase << std::hex
776 << "\n Expected: " << data->result
777 << "\n Actual: " << result
778 << "\n checksum=" << data->checksum
779 << " old_sum=" << data->old_hdr_sum << " new_sum=" << data->new_hdr_sum << "\n";
780 }
781}
782
783TEST_F(ClatdTest, Translate) {
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900784 // This test uses hardcoded packets so the clatd address must be fixed.
785 inet_pton(AF_INET6, kIPv6LocalAddr, &Global_Clatd_Config.ipv6_local_subnet);
786
Brian Carlstromfcac4102014-02-24 20:03:01 -0800787 uint8_t udp_ipv4[] = { IPV4_UDP_HEADER UDP_HEADER PAYLOAD };
788 uint8_t udp_ipv6[] = { IPV6_UDP_HEADER UDP_HEADER PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900789 fix_udp_checksum(udp_ipv4);
790 fix_udp_checksum(udp_ipv6);
791 check_translated_packet(udp_ipv4, sizeof(udp_ipv4), udp_ipv6, sizeof(udp_ipv6),
792 "UDP/IPv4 -> UDP/IPv6 translation");
793 check_translated_packet(udp_ipv6, sizeof(udp_ipv6), udp_ipv4, sizeof(udp_ipv4),
794 "UDP/IPv6 -> UDP/IPv4 translation");
795
Brian Carlstromfcac4102014-02-24 20:03:01 -0800796 uint8_t ipv4_ping[] = { IPV4_ICMP_HEADER IPV4_PING PAYLOAD };
797 uint8_t ipv6_ping[] = { IPV6_ICMPV6_HEADER IPV6_PING PAYLOAD };
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900798 check_translated_packet(ipv4_ping, sizeof(ipv4_ping), ipv6_ping, sizeof(ipv6_ping),
799 "ICMP->ICMPv6 translation");
800 check_translated_packet(ipv6_ping, sizeof(ipv6_ping), ipv4_ping, sizeof(ipv4_ping),
801 "ICMPv6->ICMP translation");
802}
803
804TEST_F(ClatdTest, Fragmentation) {
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900805 // This test uses hardcoded packets so the clatd address must be fixed.
806 inet_pton(AF_INET6, kIPv6LocalAddr, &Global_Clatd_Config.ipv6_local_subnet);
807
junyulaic4e591a2018-11-26 22:36:10 +0900808 check_fragment_translation(kIPv4Fragments, kIPv4FragLengths, kIPv6Fragments, kIPv6FragLengths,
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900809 ARRAYSIZE(kIPv4Fragments), "IPv4->IPv6 fragment translation");
810
junyulaic4e591a2018-11-26 22:36:10 +0900811 check_fragment_translation(kIPv6Fragments, kIPv6FragLengths, kIPv4Fragments, kIPv4FragLengths,
Lorenzo Colittif3beefc2014-02-14 13:19:27 +0900812 ARRAYSIZE(kIPv6Fragments), "IPv6->IPv4 fragment translation");
813}
Lorenzo Colitti98089522014-10-09 22:29:45 +0900814
Lorenzo Colittibaa3c6a2020-06-02 01:55:12 +0900815// picks a random interface ID that is checksum neutral with the IPv4 address and the NAT64 prefix
816void gen_random_iid(struct in6_addr *myaddr, struct in_addr *ipv4_local_subnet,
817 struct in6_addr *plat_subnet) {
818 // Fill last 8 bytes of IPv6 address with random bits.
819 arc4random_buf(&myaddr->s6_addr[8], 8);
820
821 // Make the IID checksum-neutral. That is, make it so that:
822 // checksum(Local IPv4 | Remote IPv4) = checksum(Local IPv6 | Remote IPv6)
823 // in other words (because remote IPv6 = NAT64 prefix | Remote IPv4):
824 // checksum(Local IPv4) = checksum(Local IPv6 | NAT64 prefix)
825 // Do this by adjusting the two bytes in the middle of the IID.
826
827 uint16_t middlebytes = (myaddr->s6_addr[11] << 8) + myaddr->s6_addr[12];
828
829 uint32_t c1 = ip_checksum_add(0, ipv4_local_subnet, sizeof(*ipv4_local_subnet));
830 uint32_t c2 = ip_checksum_add(0, plat_subnet, sizeof(*plat_subnet)) +
831 ip_checksum_add(0, myaddr, sizeof(*myaddr));
832
833 uint16_t delta = ip_checksum_adjust(middlebytes, c1, c2);
834 myaddr->s6_addr[11] = delta >> 8;
835 myaddr->s6_addr[12] = delta & 0xff;
836}
837
Lorenzo Colitti98089522014-10-09 22:29:45 +0900838void check_translate_checksum_neutral(const uint8_t *original, size_t original_len,
839 size_t expected_len, const char *msg) {
840 uint8_t translated[MAXMTU];
841 size_t translated_len = sizeof(translated);
842 do_translate_packet(original, original_len, translated, &translated_len, msg);
843 EXPECT_EQ(expected_len, translated_len) << msg << ": Translated packet length incorrect\n";
844 // do_translate_packet already checks packets for validity and verifies the checksum.
junyulaic4e591a2018-11-26 22:36:10 +0900845 int original_check = get_transport_checksum(original);
Lorenzo Colitti98089522014-10-09 22:29:45 +0900846 int translated_check = get_transport_checksum(translated);
847 ASSERT_NE(-1, original_check);
848 ASSERT_NE(-1, translated_check);
849 ASSERT_EQ(original_check, translated_check)
junyulaic4e591a2018-11-26 22:36:10 +0900850 << "Not checksum neutral: original and translated checksums differ\n";
Lorenzo Colitti98089522014-10-09 22:29:45 +0900851}
852
853TEST_F(ClatdTest, TranslateChecksumNeutral) {
854 // Generate a random clat IPv6 address and check that translation is checksum-neutral.
Lorenzo Colitti98089522014-10-09 22:29:45 +0900855 ASSERT_TRUE(inet_pton(AF_INET6, "2001:db8:1:2:f076:ae99:124e:aa54",
856 &Global_Clatd_Config.ipv6_local_subnet));
Lorenzo Colittibaa3c6a2020-06-02 01:55:12 +0900857
858 gen_random_iid(&Global_Clatd_Config.ipv6_local_subnet, &Global_Clatd_Config.ipv4_local_subnet,
859 &Global_Clatd_Config.plat_subnet);
860
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900861 ASSERT_NE(htonl((uint32_t)0x00000464), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[3]);
junyulaic4e591a2018-11-26 22:36:10 +0900862 ASSERT_NE((uint32_t)0, Global_Clatd_Config.ipv6_local_subnet.s6_addr32[3]);
Lorenzo Colitti98089522014-10-09 22:29:45 +0900863
864 // Check that translating UDP packets is checksum-neutral. First, IPv4.
865 uint8_t udp_ipv4[] = { IPV4_UDP_HEADER UDP_HEADER PAYLOAD };
866 fix_udp_checksum(udp_ipv4);
867 check_translate_checksum_neutral(udp_ipv4, sizeof(udp_ipv4), sizeof(udp_ipv4) + 20,
868 "UDP/IPv4 -> UDP/IPv6 checksum neutral");
869
870 // Now try IPv6.
871 uint8_t udp_ipv6[] = { IPV6_UDP_HEADER UDP_HEADER PAYLOAD };
872 // The test packet uses the static IID, not the random IID. Fix up the source address.
junyulaic4e591a2018-11-26 22:36:10 +0900873 struct ip6_hdr *ip6 = (struct ip6_hdr *)udp_ipv6;
Lorenzo Colitti98089522014-10-09 22:29:45 +0900874 memcpy(&ip6->ip6_src, &Global_Clatd_Config.ipv6_local_subnet, sizeof(ip6->ip6_src));
875 fix_udp_checksum(udp_ipv6);
876 check_translate_checksum_neutral(udp_ipv4, sizeof(udp_ipv4), sizeof(udp_ipv4) + 20,
877 "UDP/IPv4 -> UDP/IPv6 checksum neutral");
878}
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900879
880TEST_F(ClatdTest, GetInterfaceIp) {
881 union anyip *ip = getinterface_ip(sTun.name().c_str(), AF_INET6);
882 ASSERT_NE(nullptr, ip);
883 in6_addr expected = sTun.srcAddr();
884 in6_addr actual = ip->ip6;
885 expect_ipv6_addr_equal(&expected, &actual);
886}
887
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900888void expectSocketBound(int ifindex, int sock) {
889 // Check that the packet socket is bound to the interface. We can't check the socket filter
890 // because there is no way to fetch it from the kernel.
891 sockaddr_ll sll;
892 socklen_t len = sizeof(sll);
893 ASSERT_EQ(0, getsockname(sock, reinterpret_cast<sockaddr *>(&sll), &len));
894 EXPECT_EQ(htons(ETH_P_IPV6), sll.sll_protocol);
895 EXPECT_EQ(ifindex, sll.sll_ifindex);
896}
897
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900898TEST_F(ClatdTest, ConfigureIpv6Address) {
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900899 struct tun_data tunnel = makeTunData();
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900900
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900901 ASSERT_TRUE(IN6_IS_ADDR_UNSPECIFIED(&Global_Clatd_Config.ipv6_local_subnet));
902
903 const char *addrStr = "2001:db8::f00";
904 in6_addr addr;
905 ASSERT_EQ(1, inet_pton(AF_INET6, addrStr, &addr));
906 ASSERT_EQ(1, configure_clat_ipv6_address(&tunnel, sTun.name().c_str(), addrStr));
907
908 EXPECT_EQ(htonl(0x20010db8), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[0]);
909 EXPECT_EQ(htonl(0x00000000), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[1]);
910 EXPECT_EQ(htonl(0x00000000), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[2]);
911 EXPECT_EQ(htonl(0x00000f00), Global_Clatd_Config.ipv6_local_subnet.s6_addr32[3]);
912
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900913 // Check that the packet socket is bound to the interface. We can't check the socket filter
914 // because there is no way to fetch it from the kernel.
915 sockaddr_ll sll;
916 socklen_t len = sizeof(sll);
917 ASSERT_EQ(0, getsockname(tunnel.read_fd6, reinterpret_cast<sockaddr *>(&sll), &len));
918 EXPECT_EQ(htons(ETH_P_IPV6), sll.sll_protocol);
919 EXPECT_EQ(sll.sll_ifindex, sTun.ifindex());
Lorenzo Colittif0fac862019-01-11 18:10:11 +0900920
921 expectSocketBound(sTun.ifindex(), tunnel.read_fd6);
922
923 freeTunData(&tunnel);
Lorenzo Colitti72f770c2019-01-04 15:00:28 +0900924}
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900925
926TEST_F(ClatdTest, Ipv6AddressChanged) {
927 // Configure the clat IPv6 address.
928 struct tun_data tunnel = {
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900929 .read_fd6 = socket(AF_PACKET, SOCK_DGRAM, htons(ETH_P_IPV6)),
Nick Desaulniers75f4dc62019-11-19 09:37:42 -0800930 .write_fd6 = socket(AF_INET6, SOCK_RAW | SOCK_NONBLOCK, IPPROTO_RAW),
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900931 };
932 const char *ifname = sTun.name().c_str();
Lorenzo Colittibaa3c6a2020-06-02 01:55:12 +0900933
934 in6_addr myaddr = sTun.srcAddr();
935 gen_random_iid(&myaddr, &Global_Clatd_Config.ipv4_local_subnet, &Global_Clatd_Config.plat_subnet);
936 char addrstr[INET6_ADDRSTRLEN];
937 ASSERT_NE(nullptr, inet_ntop(AF_INET6, &myaddr, addrstr, sizeof(addrstr)));
938
939 ASSERT_EQ(1, configure_clat_ipv6_address(&tunnel, ifname, addrstr));
Lorenzo Colitti66deecd2019-01-04 12:27:27 +0900940 EXPECT_EQ(0, ipv6_address_changed(ifname));
941 EXPECT_EQ(0, ipv6_address_changed(ifname));
942
943 // Change the IP address on the tun interface to a new prefix.
944 char srcaddr[INET6_ADDRSTRLEN];
945 char dstaddr[INET6_ADDRSTRLEN];
946 ASSERT_NE(nullptr, inet_ntop(AF_INET6, &sTun.srcAddr(), srcaddr, sizeof(srcaddr)));
947 ASSERT_NE(nullptr, inet_ntop(AF_INET6, &sTun.dstAddr(), dstaddr, sizeof(dstaddr)));
948 EXPECT_EQ(0, ifc_del_address(ifname, srcaddr, 64));
949 EXPECT_EQ(0, ifc_del_address(ifname, dstaddr, 64));
950
951 // Check that we can tell that the address has changed.
952 EXPECT_EQ(0, ifc_add_address(ifname, "2001:db8::1:2", 64));
953 EXPECT_EQ(1, ipv6_address_changed(ifname));
954 EXPECT_EQ(1, ipv6_address_changed(ifname));
955
956 // Restore the tun interface configuration.
957 sTun.destroy();
958 ASSERT_EQ(0, sTun.init());
959}