1/*
2 * Copyright (C) 2010 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
17package android.net;
18
19import java.lang.reflect.Method;
20import java.net.InetAddress;
21
22import android.net.IpPrefix;
23import android.net.RouteInfo;
24import android.os.Parcel;
25
26import junit.framework.TestCase;
27import android.test.suitebuilder.annotation.SmallTest;
28
29public class RouteInfoTest extends TestCase {
30
31    private InetAddress Address(String addr) {
32        return InetAddress.parseNumericAddress(addr);
33    }
34
35    private IpPrefix Prefix(String prefix) {
36        return new IpPrefix(prefix);
37    }
38
39    @SmallTest
40    public void testConstructor() {
41        RouteInfo r;
42
43        // Invalid input.
44        try {
45            r = new RouteInfo((IpPrefix) null, null, "rmnet0");
46            fail("Expected RuntimeException:  destination and gateway null");
47        } catch(RuntimeException e) {}
48
49        // Null destination is default route.
50        r = new RouteInfo((IpPrefix) null, Address("2001:db8::1"), null);
51        assertEquals(Prefix("::/0"), r.getDestination());
52        assertEquals(Address("2001:db8::1"), r.getGateway());
53        assertNull(r.getInterface());
54
55        r = new RouteInfo((IpPrefix) null, Address("192.0.2.1"), "wlan0");
56        assertEquals(Prefix("0.0.0.0/0"), r.getDestination());
57        assertEquals(Address("192.0.2.1"), r.getGateway());
58        assertEquals("wlan0", r.getInterface());
59
60        // Null gateway sets gateway to unspecified address (why?).
61        r = new RouteInfo(Prefix("2001:db8:beef:cafe::/48"), null, "lo");
62        assertEquals(Prefix("2001:db8:beef::/48"), r.getDestination());
63        assertEquals(Address("::"), r.getGateway());
64        assertEquals("lo", r.getInterface());
65
66        r = new RouteInfo(Prefix("192.0.2.5/24"), null);
67        assertEquals(Prefix("192.0.2.0/24"), r.getDestination());
68        assertEquals(Address("0.0.0.0"), r.getGateway());
69        assertNull(r.getInterface());
70    }
71
72    public void testMatches() {
73        class PatchedRouteInfo {
74            private final RouteInfo mRouteInfo;
75
76            public PatchedRouteInfo(IpPrefix destination, InetAddress gateway, String iface) {
77                mRouteInfo = new RouteInfo(destination, gateway, iface);
78            }
79
80            public boolean matches(InetAddress destination) {
81                return mRouteInfo.matches(destination);
82            }
83        }
84
85        PatchedRouteInfo r;
86
87        r = new PatchedRouteInfo(Prefix("2001:db8:f00::ace:d00d/127"), null, "rmnet0");
88        assertTrue(r.matches(Address("2001:db8:f00::ace:d00c")));
89        assertTrue(r.matches(Address("2001:db8:f00::ace:d00d")));
90        assertFalse(r.matches(Address("2001:db8:f00::ace:d00e")));
91        assertFalse(r.matches(Address("2001:db8:f00::bad:d00d")));
92        assertFalse(r.matches(Address("2001:4868:4860::8888")));
93
94        r = new PatchedRouteInfo(Prefix("192.0.2.0/23"), null, "wlan0");
95        assertTrue(r.matches(Address("192.0.2.43")));
96        assertTrue(r.matches(Address("192.0.3.21")));
97        assertFalse(r.matches(Address("192.0.0.21")));
98        assertFalse(r.matches(Address("8.8.8.8")));
99
100        PatchedRouteInfo ipv6Default = new PatchedRouteInfo(Prefix("::/0"), null, "rmnet0");
101        assertTrue(ipv6Default.matches(Address("2001:db8::f00")));
102        assertFalse(ipv6Default.matches(Address("192.0.2.1")));
103
104        PatchedRouteInfo ipv4Default = new PatchedRouteInfo(Prefix("0.0.0.0/0"), null, "rmnet0");
105        assertTrue(ipv4Default.matches(Address("255.255.255.255")));
106        assertTrue(ipv4Default.matches(Address("192.0.2.1")));
107        assertFalse(ipv4Default.matches(Address("2001:db8::f00")));
108    }
109
110    private void assertAreEqual(Object o1, Object o2) {
111        assertTrue(o1.equals(o2));
112        assertTrue(o2.equals(o1));
113    }
114
115    private void assertAreNotEqual(Object o1, Object o2) {
116        assertFalse(o1.equals(o2));
117        assertFalse(o2.equals(o1));
118    }
119
120    public void testEquals() {
121        // IPv4
122        RouteInfo r1 = new RouteInfo(Prefix("2001:db8:ace::/48"), Address("2001:db8::1"), "wlan0");
123        RouteInfo r2 = new RouteInfo(Prefix("2001:db8:ace::/48"), Address("2001:db8::1"), "wlan0");
124        assertAreEqual(r1, r2);
125
126        RouteInfo r3 = new RouteInfo(Prefix("2001:db8:ace::/49"), Address("2001:db8::1"), "wlan0");
127        RouteInfo r4 = new RouteInfo(Prefix("2001:db8:ace::/48"), Address("2001:db8::2"), "wlan0");
128        RouteInfo r5 = new RouteInfo(Prefix("2001:db8:ace::/48"), Address("2001:db8::1"), "rmnet0");
129        assertAreNotEqual(r1, r3);
130        assertAreNotEqual(r1, r4);
131        assertAreNotEqual(r1, r5);
132
133        // IPv6
134        r1 = new RouteInfo(Prefix("192.0.2.0/25"), Address("192.0.2.1"), "wlan0");
135        r2 = new RouteInfo(Prefix("192.0.2.0/25"), Address("192.0.2.1"), "wlan0");
136        assertAreEqual(r1, r2);
137
138        r3 = new RouteInfo(Prefix("192.0.2.0/24"), Address("192.0.2.1"), "wlan0");
139        r4 = new RouteInfo(Prefix("192.0.2.0/25"), Address("192.0.2.2"), "wlan0");
140        r5 = new RouteInfo(Prefix("192.0.2.0/25"), Address("192.0.2.1"), "rmnet0");
141        assertAreNotEqual(r1, r3);
142        assertAreNotEqual(r1, r4);
143        assertAreNotEqual(r1, r5);
144
145        // Interfaces (but not destinations or gateways) can be null.
146        r1 = new RouteInfo(Prefix("192.0.2.0/25"), Address("192.0.2.1"), null);
147        r2 = new RouteInfo(Prefix("192.0.2.0/25"), Address("192.0.2.1"), null);
148        r3 = new RouteInfo(Prefix("192.0.2.0/24"), Address("192.0.2.1"), "wlan0");
149        assertAreEqual(r1, r2);
150        assertAreNotEqual(r1, r3);
151    }
152
153    public void testHostAndDefaultRoutes() {
154      RouteInfo r;
155
156      r = new RouteInfo(Prefix("0.0.0.0/0"), Address("0.0.0.0"), "wlan0");
157      assertFalse(r.isHostRoute());
158      assertTrue(r.isDefaultRoute());
159      assertTrue(r.isIPv4Default());
160      assertFalse(r.isIPv6Default());
161
162      r = new RouteInfo(Prefix("::/0"), Address("::"), "wlan0");
163      assertFalse(r.isHostRoute());
164      assertTrue(r.isDefaultRoute());
165      assertFalse(r.isIPv4Default());
166      assertTrue(r.isIPv6Default());
167
168      r = new RouteInfo(Prefix("192.0.2.0/24"), null, "wlan0");
169      assertFalse(r.isHostRoute());
170      assertFalse(r.isDefaultRoute());
171      assertFalse(r.isIPv4Default());
172      assertFalse(r.isIPv6Default());
173
174      r = new RouteInfo(Prefix("2001:db8::/48"), null, "wlan0");
175      assertFalse(r.isHostRoute());
176      assertFalse(r.isDefaultRoute());
177      assertFalse(r.isIPv4Default());
178      assertFalse(r.isIPv6Default());
179
180      r = new RouteInfo(Prefix("192.0.2.0/32"), Address("0.0.0.0"), "wlan0");
181      assertTrue(r.isHostRoute());
182      assertFalse(r.isDefaultRoute());
183      assertFalse(r.isIPv4Default());
184      assertFalse(r.isIPv6Default());
185
186      r = new RouteInfo(Prefix("2001:db8::/128"), Address("::"), "wlan0");
187      assertTrue(r.isHostRoute());
188      assertFalse(r.isDefaultRoute());
189      assertFalse(r.isIPv4Default());
190      assertFalse(r.isIPv6Default());
191
192      r = new RouteInfo(Prefix("192.0.2.0/32"), null, "wlan0");
193      assertTrue(r.isHostRoute());
194      assertFalse(r.isDefaultRoute());
195      assertFalse(r.isIPv4Default());
196      assertFalse(r.isIPv6Default());
197
198      r = new RouteInfo(Prefix("2001:db8::/128"), null, "wlan0");
199      assertTrue(r.isHostRoute());
200      assertFalse(r.isDefaultRoute());
201      assertFalse(r.isIPv4Default());
202      assertFalse(r.isIPv6Default());
203
204      r = new RouteInfo(Prefix("::/128"), Address("fe80::"), "wlan0");
205      assertTrue(r.isHostRoute());
206      assertFalse(r.isDefaultRoute());
207      assertFalse(r.isIPv4Default());
208      assertFalse(r.isIPv6Default());
209
210      r = new RouteInfo(Prefix("0.0.0.0/32"), Address("192.0.2.1"), "wlan0");
211      assertTrue(r.isHostRoute());
212      assertFalse(r.isDefaultRoute());
213      assertFalse(r.isIPv4Default());
214      assertFalse(r.isIPv6Default());
215    }
216
217    public void testTruncation() {
218      LinkAddress l;
219      RouteInfo r;
220
221      l = new LinkAddress("192.0.2.5/30");
222      r = new RouteInfo(l, Address("192.0.2.1"), "wlan0");
223      assertEquals("192.0.2.4", r.getDestination().getAddress().getHostAddress());
224
225      l = new LinkAddress("2001:db8:1:f::5/63");
226      r = new RouteInfo(l, Address("2001:db8:5::1"), "wlan0");
227      assertEquals("2001:db8:1:e::", r.getDestination().getAddress().getHostAddress());
228    }
229
230    // Make sure that creating routes to multicast addresses doesn't throw an exception. Even though
231    // there's nothing we can do with them, we don't want to crash if, e.g., someone calls
232    // requestRouteToHostAddress("230.0.0.0", MOBILE_HIPRI);
233    public void testMulticastRoute() {
234      RouteInfo r;
235      r = new RouteInfo(Prefix("230.0.0.0/32"), Address("192.0.2.1"), "wlan0");
236      r = new RouteInfo(Prefix("ff02::1/128"), Address("2001:db8::1"), "wlan0");
237      // No exceptions? Good.
238    }
239
240    public RouteInfo passThroughParcel(RouteInfo r) {
241        Parcel p = Parcel.obtain();
242        RouteInfo r2 = null;
243        try {
244            r.writeToParcel(p, 0);
245            p.setDataPosition(0);
246            r2 = RouteInfo.CREATOR.createFromParcel(p);
247        } finally {
248            p.recycle();
249        }
250        assertNotNull(r2);
251        return r2;
252    }
253
254    public void assertParcelingIsLossless(RouteInfo r) {
255      RouteInfo r2 = passThroughParcel(r);
256      assertEquals(r, r2);
257    }
258
259    public void testParceling() {
260        RouteInfo r;
261
262        r = new RouteInfo(Prefix("::/0"), Address("2001:db8::"), null);
263        assertParcelingIsLossless(r);
264
265        r = new RouteInfo(Prefix("192.0.2.0/24"), null, "wlan0");
266        assertParcelingIsLossless(r);
267    }
268}
269