1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
/*
* Copyright (C) 2019 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.net.module.util;
import static com.android.testutils.DevSdkIgnoreRuleKt.SC_V2;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import android.net.InetAddresses;
import android.net.IpPrefix;
import android.net.RouteInfo;
import androidx.test.runner.AndroidJUnit4;
import com.android.testutils.DevSdkIgnoreRule;
import com.android.testutils.DevSdkIgnoreRule.IgnoreUpTo;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import java.net.InetAddress;
import java.util.ArrayList;
import java.util.List;
@RunWith(AndroidJUnit4.class)
public final class NetUtilsTest {
@Rule
public final DevSdkIgnoreRule ignoreRule = new DevSdkIgnoreRule();
private static final InetAddress V4_ADDR1 = toInetAddress("75.208.7.1");
private static final InetAddress V4_ADDR2 = toInetAddress("75.208.7.2");
private static final InetAddress V6_ADDR1 = toInetAddress("2001:0db8:85a3::8a2e:0370:7334");
private static final InetAddress V6_ADDR2 = toInetAddress("2001:0db8:85a3::8a2e:0370:7335");
private static final InetAddress V4_GATEWAY = toInetAddress("75.208.8.1");
private static final InetAddress V6_GATEWAY = toInetAddress("fe80::6:0000:613");
private static final InetAddress V4_DEST = toInetAddress("75.208.8.15");
private static final InetAddress V6_DEST = toInetAddress("2001:db8:cafe::123");
private static final RouteInfo V4_EXPECTED = new RouteInfo(new IpPrefix("75.208.8.0/24"),
V4_GATEWAY, "wlan0");
private static final RouteInfo V6_EXPECTED = new RouteInfo(new IpPrefix("2001:db8:cafe::/64"),
V6_GATEWAY, "wlan0");
private static InetAddress toInetAddress(String addr) {
return InetAddresses.parseNumericAddress(addr);
}
@Test
public void testAddressTypeMatches() {
assertTrue(NetUtils.addressTypeMatches(V4_ADDR1, V4_ADDR2));
assertTrue(NetUtils.addressTypeMatches(V6_ADDR1, V6_ADDR2));
assertFalse(NetUtils.addressTypeMatches(V4_ADDR1, V6_ADDR1));
assertFalse(NetUtils.addressTypeMatches(V6_ADDR1, V4_ADDR1));
}
@Test
public void testSelectBestRoute() {
final List<RouteInfo> routes = new ArrayList<>();
RouteInfo route = NetUtils.selectBestRoute(null, V4_DEST);
assertNull(route);
route = NetUtils.selectBestRoute(routes, null);
assertNull(route);
route = NetUtils.selectBestRoute(routes, V4_DEST);
assertNull(route);
routes.add(V4_EXPECTED);
// "75.208.0.0/16" is not an expected result since it is not the longest prefix.
routes.add(new RouteInfo(new IpPrefix("75.208.0.0/16"), V4_GATEWAY, "wlan0"));
routes.add(new RouteInfo(new IpPrefix("75.208.7.0/24"), V4_GATEWAY, "wlan0"));
routes.add(V6_EXPECTED);
// "2001:db8::/32" is not an expected result since it is not the longest prefix.
routes.add(new RouteInfo(new IpPrefix("2001:db8::/32"), V6_GATEWAY, "wlan0"));
routes.add(new RouteInfo(new IpPrefix("2001:db8:beef::/64"), V6_GATEWAY, "wlan0"));
// Verify expected v4 route is selected
route = NetUtils.selectBestRoute(routes, V4_DEST);
assertEquals(V4_EXPECTED, route);
// Verify expected v6 route is selected
route = NetUtils.selectBestRoute(routes, V6_DEST);
assertEquals(V6_EXPECTED, route);
// Remove expected v4 route
routes.remove(V4_EXPECTED);
route = NetUtils.selectBestRoute(routes, V4_DEST);
assertNotEquals(V4_EXPECTED, route);
// Remove expected v6 route
routes.remove(V6_EXPECTED);
route = NetUtils.selectBestRoute(routes, V4_DEST);
assertNotEquals(V6_EXPECTED, route);
}
@Test @IgnoreUpTo(SC_V2)
public void testSelectBestRouteWithExcludedRoutes() {
final List<RouteInfo> routes = new ArrayList<>();
routes.add(V4_EXPECTED);
routes.add(new RouteInfo(new IpPrefix("75.208.0.0/16"), V4_GATEWAY, "wlan0"));
routes.add(new RouteInfo(new IpPrefix("75.208.7.0/24"), V4_GATEWAY, "wlan0"));
routes.add(V6_EXPECTED);
routes.add(new RouteInfo(new IpPrefix("2001:db8::/32"), V6_GATEWAY, "wlan0"));
routes.add(new RouteInfo(new IpPrefix("2001:db8:beef::/64"), V6_GATEWAY, "wlan0"));
// After adding excluded v4 route with longer prefix, expected result is null.
routes.add(new RouteInfo(new IpPrefix("75.208.8.0/28"), null /* gateway */, "wlan0",
RouteInfo.RTN_THROW));
RouteInfo route = NetUtils.selectBestRoute(routes, V4_DEST);
assertNull(route);
// After adding excluded v6 route with longer prefix, expected result is null.
routes.add(new RouteInfo(new IpPrefix("2001:db8:cafe::/96"), null /* gateway */, "wlan0",
RouteInfo.RTN_THROW));
route = NetUtils.selectBestRoute(routes, V6_DEST);
assertNull(route);
}
}
|
