1 /*
2  * Copyright (C) 2015 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 
17 package android.net;
18 
19 import static junit.framework.Assert.assertEquals;
20 
21 import android.os.Build;
22 
23 import androidx.test.filters.SmallTest;
24 
25 import com.android.testutils.DevSdkIgnoreRule;
26 import com.android.testutils.DevSdkIgnoreRunner;
27 
28 import org.junit.Test;
29 import org.junit.runner.RunWith;
30 
31 import java.math.BigInteger;
32 import java.util.TreeSet;
33 
34 @RunWith(DevSdkIgnoreRunner.class)
35 @SmallTest
36 @DevSdkIgnoreRule.IgnoreUpTo(Build.VERSION_CODES.R)
37 public class NetworkUtilsTest {
38     @Test
testRoutedIPv4AddressCount()39     public void testRoutedIPv4AddressCount() {
40         final TreeSet<IpPrefix> set = new TreeSet<>(IpPrefix.lengthComparator());
41         // No routes routes to no addresses.
42         assertEquals(0, NetworkUtils.routedIPv4AddressCount(set));
43 
44         set.add(new IpPrefix("0.0.0.0/0"));
45         assertEquals(1l << 32, NetworkUtils.routedIPv4AddressCount(set));
46 
47         set.add(new IpPrefix("20.18.0.0/16"));
48         set.add(new IpPrefix("20.18.0.0/24"));
49         set.add(new IpPrefix("20.18.0.0/8"));
50         // There is a default route, still covers everything
51         assertEquals(1l << 32, NetworkUtils.routedIPv4AddressCount(set));
52 
53         set.clear();
54         set.add(new IpPrefix("20.18.0.0/24"));
55         set.add(new IpPrefix("20.18.0.0/8"));
56         // The 8-length includes the 24-length prefix
57         assertEquals(1l << 24, NetworkUtils.routedIPv4AddressCount(set));
58 
59         set.add(new IpPrefix("10.10.10.126/25"));
60         // The 8-length does not include this 25-length prefix
61         assertEquals((1l << 24) + (1 << 7), NetworkUtils.routedIPv4AddressCount(set));
62 
63         set.clear();
64         set.add(new IpPrefix("1.2.3.4/32"));
65         set.add(new IpPrefix("1.2.3.4/32"));
66         set.add(new IpPrefix("1.2.3.4/32"));
67         set.add(new IpPrefix("1.2.3.4/32"));
68         assertEquals(1l, NetworkUtils.routedIPv4AddressCount(set));
69 
70         set.add(new IpPrefix("1.2.3.5/32"));
71         set.add(new IpPrefix("1.2.3.6/32"));
72 
73         set.add(new IpPrefix("1.2.3.7/32"));
74         set.add(new IpPrefix("1.2.3.8/32"));
75         set.add(new IpPrefix("1.2.3.9/32"));
76         set.add(new IpPrefix("1.2.3.0/32"));
77         assertEquals(7l, NetworkUtils.routedIPv4AddressCount(set));
78 
79         // 1.2.3.4/30 eats 1.2.3.{4-7}/32
80         set.add(new IpPrefix("1.2.3.4/30"));
81         set.add(new IpPrefix("6.2.3.4/28"));
82         set.add(new IpPrefix("120.2.3.4/16"));
83         assertEquals(7l - 4 + 4 + 16 + 65536, NetworkUtils.routedIPv4AddressCount(set));
84     }
85 
86     @Test
testRoutedIPv6AddressCount()87     public void testRoutedIPv6AddressCount() {
88         final TreeSet<IpPrefix> set = new TreeSet<>(IpPrefix.lengthComparator());
89         // No routes routes to no addresses.
90         assertEquals(BigInteger.ZERO, NetworkUtils.routedIPv6AddressCount(set));
91 
92         set.add(new IpPrefix("::/0"));
93         assertEquals(BigInteger.ONE.shiftLeft(128), NetworkUtils.routedIPv6AddressCount(set));
94 
95         set.add(new IpPrefix("1234:622a::18/64"));
96         set.add(new IpPrefix("add4:f00:80:f7:1111::6adb/96"));
97         set.add(new IpPrefix("add4:f00:80:f7:1111::6adb/8"));
98         // There is a default route, still covers everything
99         assertEquals(BigInteger.ONE.shiftLeft(128), NetworkUtils.routedIPv6AddressCount(set));
100 
101         set.clear();
102         set.add(new IpPrefix("add4:f00:80:f7:1111::6adb/96"));
103         set.add(new IpPrefix("add4:f00:80:f7:1111::6adb/8"));
104         // The 8-length includes the 96-length prefix
105         assertEquals(BigInteger.ONE.shiftLeft(120), NetworkUtils.routedIPv6AddressCount(set));
106 
107         set.add(new IpPrefix("10::26/64"));
108         // The 8-length does not include this 64-length prefix
109         assertEquals(BigInteger.ONE.shiftLeft(120).add(BigInteger.ONE.shiftLeft(64)),
110                 NetworkUtils.routedIPv6AddressCount(set));
111 
112         set.clear();
113         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad4/128"));
114         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad4/128"));
115         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad4/128"));
116         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad4/128"));
117         assertEquals(BigInteger.ONE, NetworkUtils.routedIPv6AddressCount(set));
118 
119         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad5/128"));
120         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad6/128"));
121         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad7/128"));
122         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad8/128"));
123         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad9/128"));
124         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad0/128"));
125         assertEquals(BigInteger.valueOf(7), NetworkUtils.routedIPv6AddressCount(set));
126 
127         // add4:f00:80:f7:1111::6ad4/126 eats add4:f00:8[:f7:1111::6ad{4-7}/128
128         set.add(new IpPrefix("add4:f00:80:f7:1111::6ad4/126"));
129         set.add(new IpPrefix("d00d:f00:80:f7:1111::6ade/124"));
130         set.add(new IpPrefix("f00b:a33::/112"));
131         assertEquals(BigInteger.valueOf(7l - 4 + 4 + 16 + 65536),
132                 NetworkUtils.routedIPv6AddressCount(set));
133     }
134 }
135