1379747a4040118130ca35947aef6942b2cd1cd67Erik Kline/* 2379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * Copyright (C) 2015 The Android Open Source Project 3379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 4379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * Licensed under the Apache License, Version 2.0 (the "License"); 5379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * you may not use this file except in compliance with the License. 6379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * You may obtain a copy of the License at 7379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 8379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * http://www.apache.org/licenses/LICENSE-2.0 9379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 10379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * Unless required by applicable law or agreed to in writing, software 11379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * distributed under the License is distributed on an "AS IS" BASIS, 12379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * See the License for the specific language governing permissions and 14379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * limitations under the License. 15379747a4040118130ca35947aef6942b2cd1cd67Erik Kline */ 16379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 17379747a4040118130ca35947aef6942b2cd1cd67Erik Klinepackage com.android.server.connectivity; 18379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 19379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport static android.system.OsConstants.*; 20379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 2129f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colittiimport android.net.LinkAddress; 22379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.net.LinkProperties; 23379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.net.Network; 2487cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colittiimport android.net.NetworkUtils; 25379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.net.RouteInfo; 26619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkeyimport android.net.TrafficStats; 27379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.os.SystemClock; 28379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.system.ErrnoException; 29379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.system.Os; 30379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.system.StructTimeval; 31379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport android.text.TextUtils; 3229f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colittiimport android.util.Pair; 33379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 34379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport com.android.internal.util.IndentingPrintWriter; 35379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 36379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.io.Closeable; 37379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.io.FileDescriptor; 38379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.io.InterruptedIOException; 39379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.io.IOException; 40379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.Inet4Address; 41379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.Inet6Address; 42379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.InetAddress; 43379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.InetSocketAddress; 44379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.NetworkInterface; 45379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.SocketAddress; 46379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.SocketException; 47379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.net.UnknownHostException; 48379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.nio.ByteBuffer; 49379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.nio.charset.StandardCharsets; 50379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.util.concurrent.CountDownLatch; 51379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.util.concurrent.TimeUnit; 52379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.util.Arrays; 53106cdf6c94c455462316555b7ac59ec79c3b40faErik Klineimport java.util.ArrayList; 54379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.util.HashMap; 55106cdf6c94c455462316555b7ac59ec79c3b40faErik Klineimport java.util.List; 56379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.util.Map; 57379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport java.util.Random; 58379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 59379747a4040118130ca35947aef6942b2cd1cd67Erik Klineimport libcore.io.IoUtils; 60379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 61379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 62379747a4040118130ca35947aef6942b2cd1cd67Erik Kline/** 63379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * NetworkDiagnostics 64379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 65379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * A simple class to diagnose network connectivity fundamentals. Current 66379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * checks performed are: 67379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - ICMPv4/v6 echo requests for all routers 68379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - ICMPv4/v6 echo requests for all DNS servers 69379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - DNS UDP queries to all DNS servers 70379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 71379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * Currently unimplemented checks include: 72379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - report ARP/ND data about on-link neighbors 73379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - DNS TCP queries to all DNS servers 74379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - HTTP DIRECT and PROXY checks 75379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - port 443 blocking/TLS intercept checks 76379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - QUIC reachability checks 77379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * - MTU checks 78379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 79379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * The supplied timeout bounds the entire diagnostic process. Each specific 80379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * check class must implement this upper bound on measurements in whichever 81379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * manner is most appropriate and effective. 82379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * 83379747a4040118130ca35947aef6942b2cd1cd67Erik Kline * @hide 84379747a4040118130ca35947aef6942b2cd1cd67Erik Kline */ 85379747a4040118130ca35947aef6942b2cd1cd67Erik Klinepublic class NetworkDiagnostics { 86379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final String TAG = "NetworkDiagnostics"; 87379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 8887cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti private static final InetAddress TEST_DNS4 = NetworkUtils.numericToInetAddress("8.8.8.8"); 8987cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti private static final InetAddress TEST_DNS6 = NetworkUtils.numericToInetAddress( 9087cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti "2001:4860:4860::8888"); 9187cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti 92379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // For brevity elsewhere. 93379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final long now() { 94379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return SystemClock.elapsedRealtime(); 95379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 96379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 97379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Values from RFC 1035 section 4.1.1, names from <arpa/nameser.h>. 98379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Should be a member of DnsUdpCheck, but "compiler says no". 99379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public static enum DnsResponseCode { NOERROR, FORMERR, SERVFAIL, NXDOMAIN, NOTIMP, REFUSED }; 100379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 101379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final Network mNetwork; 102379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final LinkProperties mLinkProperties; 103379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final Integer mInterfaceIndex; 104379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 105379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final long mTimeoutMs; 106379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final long mStartTime; 107379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final long mDeadlineTime; 108379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 109379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // A counter, initialized to the total number of measurements, 110379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // so callers can wait for completion. 111379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final CountDownLatch mCountDownLatch; 112379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 113106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline public class Measurement { 114379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final String SUCCEEDED = "SUCCEEDED"; 115379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final String FAILED = "FAILED"; 116379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 117106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline private boolean succeeded; 118379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 119106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // Package private. TODO: investigate better encapsulation. 120106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline String description = ""; 121106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline long startTime; 122106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline long finishTime; 123106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline String result = ""; 124106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline Thread thread; 125106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 126106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline public boolean checkSucceeded() { return succeeded; } 127106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 128106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline void recordSuccess(String msg) { 129379747a4040118130ca35947aef6942b2cd1cd67Erik Kline maybeFixupTimes(); 130106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline succeeded = true; 131ed1b45968edb11ce8dc5c1d5a5db09d943e8d425Erik Kline result = SUCCEEDED + ": " + msg; 132379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mCountDownLatch != null) { 133379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mCountDownLatch.countDown(); 134379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 135379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 136379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 137106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline void recordFailure(String msg) { 138379747a4040118130ca35947aef6942b2cd1cd67Erik Kline maybeFixupTimes(); 139106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline succeeded = false; 140ed1b45968edb11ce8dc5c1d5a5db09d943e8d425Erik Kline result = FAILED + ": " + msg; 141379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mCountDownLatch != null) { 142379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mCountDownLatch.countDown(); 143379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 144379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 145379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 146379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private void maybeFixupTimes() { 147379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Allows the caller to just set success/failure and not worry 148379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // about also setting the correct finishing time. 149379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (finishTime == 0) { finishTime = now(); } 150379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 151379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // In cases where, for example, a failure has occurred before the 152379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // measurement even began, fixup the start time to reflect as much. 153379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (startTime == 0) { startTime = finishTime; } 154379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 155379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 156379747a4040118130ca35947aef6942b2cd1cd67Erik Kline @Override 157379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public String toString() { 158379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return description + ": " + result + " (" + (finishTime - startTime) + "ms)"; 159379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 160379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 161379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 162379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final Map<InetAddress, Measurement> mIcmpChecks = new HashMap<>(); 16329f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti private final Map<Pair<InetAddress, InetAddress>, Measurement> mExplicitSourceIcmpChecks = 16429f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti new HashMap<>(); 165379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final Map<InetAddress, Measurement> mDnsUdpChecks = new HashMap<>(); 166379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final String mDescription; 167379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 168379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 169379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public NetworkDiagnostics(Network network, LinkProperties lp, long timeoutMs) { 170379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mNetwork = network; 171379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mLinkProperties = lp; 172379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mInterfaceIndex = getInterfaceIndex(mLinkProperties.getInterfaceName()); 173379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mTimeoutMs = timeoutMs; 174379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mStartTime = now(); 175379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mDeadlineTime = mStartTime + mTimeoutMs; 176379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 17787cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // Hardcode measurements to TEST_DNS4 and TEST_DNS6 in order to test off-link connectivity. 17887cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // We are free to modify mLinkProperties with impunity because ConnectivityService passes us 17987cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // a copy and not the original object. It's easier to do it this way because we don't need 18087cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // to check whether the LinkProperties already contains these DNS servers because 18187cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // LinkProperties#addDnsServer checks for duplicates. 18287cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti if (mLinkProperties.isReachable(TEST_DNS4)) { 18387cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti mLinkProperties.addDnsServer(TEST_DNS4); 18487cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti } 18587cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // TODO: we could use mLinkProperties.isReachable(TEST_DNS6) here, because we won't set any 18687cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // DNS servers for which isReachable() is false, but since this is diagnostic code, be extra 18787cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti // careful. 18887cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti if (mLinkProperties.hasGlobalIPv6Address() || mLinkProperties.hasIPv6DefaultRoute()) { 18987cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti mLinkProperties.addDnsServer(TEST_DNS6); 19087cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti } 19187cfc70b732422d6927ad9fe97f1ad7ab65fd508Lorenzo Colitti 192379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (RouteInfo route : mLinkProperties.getRoutes()) { 193379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (route.hasGateway()) { 19429f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti InetAddress gateway = route.getGateway(); 19529f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti prepareIcmpMeasurement(gateway); 19629f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti if (route.isIPv6Default()) { 19729f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti prepareExplicitSourceIcmpMeasurements(gateway); 19829f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 199379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 200379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 201379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (InetAddress nameserver : mLinkProperties.getDnsServers()) { 202379747a4040118130ca35947aef6942b2cd1cd67Erik Kline prepareIcmpMeasurement(nameserver); 203379747a4040118130ca35947aef6942b2cd1cd67Erik Kline prepareDnsMeasurement(nameserver); 204379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 205379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 206379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mCountDownLatch = new CountDownLatch(totalMeasurementCount()); 207379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 208379747a4040118130ca35947aef6942b2cd1cd67Erik Kline startMeasurements(); 209379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 210379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mDescription = "ifaces{" + TextUtils.join(",", mLinkProperties.getAllInterfaceNames()) + "}" 211379747a4040118130ca35947aef6942b2cd1cd67Erik Kline + " index{" + mInterfaceIndex + "}" 212379747a4040118130ca35947aef6942b2cd1cd67Erik Kline + " network{" + mNetwork + "}" 213379747a4040118130ca35947aef6942b2cd1cd67Erik Kline + " nethandle{" + mNetwork.getNetworkHandle() + "}"; 214379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 215379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 216379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static Integer getInterfaceIndex(String ifname) { 217379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 218379747a4040118130ca35947aef6942b2cd1cd67Erik Kline NetworkInterface ni = NetworkInterface.getByName(ifname); 219379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return ni.getIndex(); 220379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (NullPointerException | SocketException e) { 221379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return null; 222379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 223379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 224379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 225379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private void prepareIcmpMeasurement(InetAddress target) { 226379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (!mIcmpChecks.containsKey(target)) { 227379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Measurement measurement = new Measurement(); 228379747a4040118130ca35947aef6942b2cd1cd67Erik Kline measurement.thread = new Thread(new IcmpCheck(target, measurement)); 229379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mIcmpChecks.put(target, measurement); 230379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 231379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 232379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 23329f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti private void prepareExplicitSourceIcmpMeasurements(InetAddress target) { 23429f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti for (LinkAddress l : mLinkProperties.getLinkAddresses()) { 23529f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti InetAddress source = l.getAddress(); 23629f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti if (source instanceof Inet6Address && l.isGlobalPreferred()) { 23729f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti Pair<InetAddress, InetAddress> srcTarget = new Pair<>(source, target); 23829f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti if (!mExplicitSourceIcmpChecks.containsKey(srcTarget)) { 23929f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti Measurement measurement = new Measurement(); 24029f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti measurement.thread = new Thread(new IcmpCheck(source, target, measurement)); 24129f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti mExplicitSourceIcmpChecks.put(srcTarget, measurement); 24229f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 24329f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 24429f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 24529f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 24629f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti 247379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private void prepareDnsMeasurement(InetAddress target) { 248379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (!mDnsUdpChecks.containsKey(target)) { 249379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Measurement measurement = new Measurement(); 250379747a4040118130ca35947aef6942b2cd1cd67Erik Kline measurement.thread = new Thread(new DnsUdpCheck(target, measurement)); 251379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mDnsUdpChecks.put(target, measurement); 252379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 253379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 254379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 255379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private int totalMeasurementCount() { 25629f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti return mIcmpChecks.size() + mExplicitSourceIcmpChecks.size() + mDnsUdpChecks.size(); 257379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 258379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 259379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private void startMeasurements() { 260379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (Measurement measurement : mIcmpChecks.values()) { 261379747a4040118130ca35947aef6942b2cd1cd67Erik Kline measurement.thread.start(); 262379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 26329f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti for (Measurement measurement : mExplicitSourceIcmpChecks.values()) { 26429f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti measurement.thread.start(); 26529f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 266379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (Measurement measurement : mDnsUdpChecks.values()) { 267379747a4040118130ca35947aef6942b2cd1cd67Erik Kline measurement.thread.start(); 268379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 269379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 270379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 271379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public void waitForMeasurements() { 272379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 273379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mCountDownLatch.await(mDeadlineTime - now(), TimeUnit.MILLISECONDS); 274379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (InterruptedException ignored) {} 275379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 276379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 277106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline public List<Measurement> getMeasurements() { 278106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // TODO: Consider moving waitForMeasurements() in here to minimize the 279106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // chance of caller errors. 280379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 281106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline ArrayList<Measurement> measurements = new ArrayList(totalMeasurementCount()); 282106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 283106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // Sort measurements IPv4 first. 284379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (Map.Entry<InetAddress, Measurement> entry : mIcmpChecks.entrySet()) { 285379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (entry.getKey() instanceof Inet4Address) { 286106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline measurements.add(entry.getValue()); 287106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 288106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 289106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline for (Map.Entry<Pair<InetAddress, InetAddress>, Measurement> entry : 290106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline mExplicitSourceIcmpChecks.entrySet()) { 291106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline if (entry.getKey().first instanceof Inet4Address) { 292106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline measurements.add(entry.getValue()); 293106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 294106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 295106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline for (Map.Entry<InetAddress, Measurement> entry : mDnsUdpChecks.entrySet()) { 296106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline if (entry.getKey() instanceof Inet4Address) { 297106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline measurements.add(entry.getValue()); 298379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 299379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 300106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 301106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // IPv6 measurements second. 302379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (Map.Entry<InetAddress, Measurement> entry : mIcmpChecks.entrySet()) { 303379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (entry.getKey() instanceof Inet6Address) { 304106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline measurements.add(entry.getValue()); 305379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 306379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 30729f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti for (Map.Entry<Pair<InetAddress, InetAddress>, Measurement> entry : 30829f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti mExplicitSourceIcmpChecks.entrySet()) { 309106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline if (entry.getKey().first instanceof Inet6Address) { 310106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline measurements.add(entry.getValue()); 311379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 312379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 313379747a4040118130ca35947aef6942b2cd1cd67Erik Kline for (Map.Entry<InetAddress, Measurement> entry : mDnsUdpChecks.entrySet()) { 314379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (entry.getKey() instanceof Inet6Address) { 315106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline measurements.add(entry.getValue()); 316379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 317379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 318106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 319106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline return measurements; 320106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 321106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 322106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline public void dump(IndentingPrintWriter pw) { 323106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline pw.println(TAG + ":" + mDescription); 324106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline final long unfinished = mCountDownLatch.getCount(); 325106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline if (unfinished > 0) { 326106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // This can't happen unless a caller forgets to call waitForMeasurements() 327106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline // or a measurement isn't implemented to correctly honor the timeout. 328106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline pw.println("WARNING: countdown wait incomplete: " 329106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline + unfinished + " unfinished measurements"); 330106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 331106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 332106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline pw.increaseIndent(); 333106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 334106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline String prefix; 335106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline for (Measurement m : getMeasurements()) { 336106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline prefix = m.checkSucceeded() ? "." : "F"; 337106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline pw.println(prefix + " " + m.toString()); 338106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline } 339106cdf6c94c455462316555b7ac59ec79c3b40faErik Kline 340379747a4040118130ca35947aef6942b2cd1cd67Erik Kline pw.decreaseIndent(); 341379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 342379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 343379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 344379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private class SimpleSocketCheck implements Closeable { 34529f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti protected final InetAddress mSource; // Usually null. 346379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected final InetAddress mTarget; 347379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected final int mAddressFamily; 348379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected final Measurement mMeasurement; 349379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected FileDescriptor mFileDescriptor; 350379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected SocketAddress mSocketAddress; 351379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 35229f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti protected SimpleSocketCheck( 35329f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti InetAddress source, InetAddress target, Measurement measurement) { 354379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement = measurement; 355379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 356379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (target instanceof Inet6Address) { 357379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Inet6Address targetWithScopeId = null; 358379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (target.isLinkLocalAddress() && mInterfaceIndex != null) { 359379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 360379747a4040118130ca35947aef6942b2cd1cd67Erik Kline targetWithScopeId = Inet6Address.getByAddress( 361379747a4040118130ca35947aef6942b2cd1cd67Erik Kline null, target.getAddress(), mInterfaceIndex); 362379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (UnknownHostException e) { 363379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure(e.toString()); 364379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 365379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 366379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mTarget = (targetWithScopeId != null) ? targetWithScopeId : target; 367379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mAddressFamily = AF_INET6; 368379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } else { 369379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mTarget = target; 370379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mAddressFamily = AF_INET; 371379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 37229f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti 37329f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti // We don't need to check the scope ID here because we currently only do explicit-source 37429f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti // measurements from global IPv6 addresses. 37529f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti mSource = source; 37629f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 37729f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti 37829f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti protected SimpleSocketCheck(InetAddress target, Measurement measurement) { 37929f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti this(null, target, measurement); 380379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 381379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 382379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected void setupSocket( 383379747a4040118130ca35947aef6942b2cd1cd67Erik Kline int sockType, int protocol, long writeTimeout, long readTimeout, int dstPort) 384379747a4040118130ca35947aef6942b2cd1cd67Erik Kline throws ErrnoException, IOException { 385619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkey final int oldTag = TrafficStats.getAndSetThreadStatsTag(TrafficStats.TAG_SYSTEM_PROBE); 386619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkey try { 387619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkey mFileDescriptor = Os.socket(mAddressFamily, sockType, protocol); 388619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkey } finally { 389619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkey TrafficStats.setThreadStatsTag(oldTag); 390619a511d4f5b5cc74b9a8ab853db386489e02e6fJeff Sharkey } 391379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Setting SNDTIMEO is purely for defensive purposes. 392379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.setsockoptTimeval(mFileDescriptor, 393379747a4040118130ca35947aef6942b2cd1cd67Erik Kline SOL_SOCKET, SO_SNDTIMEO, StructTimeval.fromMillis(writeTimeout)); 394379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.setsockoptTimeval(mFileDescriptor, 395379747a4040118130ca35947aef6942b2cd1cd67Erik Kline SOL_SOCKET, SO_RCVTIMEO, StructTimeval.fromMillis(readTimeout)); 396379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // TODO: Use IP_RECVERR/IPV6_RECVERR, pending OsContants availability. 397379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mNetwork.bindSocket(mFileDescriptor); 39829f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti if (mSource != null) { 39929f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti Os.bind(mFileDescriptor, mSource, 0); 40029f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 401379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.connect(mFileDescriptor, mTarget, dstPort); 402379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mSocketAddress = Os.getsockname(mFileDescriptor); 403379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 404379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 405379747a4040118130ca35947aef6942b2cd1cd67Erik Kline protected String getSocketAddressString() { 406379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // The default toString() implementation is not the prettiest. 407379747a4040118130ca35947aef6942b2cd1cd67Erik Kline InetSocketAddress inetSockAddr = (InetSocketAddress) mSocketAddress; 408379747a4040118130ca35947aef6942b2cd1cd67Erik Kline InetAddress localAddr = inetSockAddr.getAddress(); 409379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return String.format( 410379747a4040118130ca35947aef6942b2cd1cd67Erik Kline (localAddr instanceof Inet6Address ? "[%s]:%d" : "%s:%d"), 411379747a4040118130ca35947aef6942b2cd1cd67Erik Kline localAddr.getHostAddress(), inetSockAddr.getPort()); 412379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 413379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 414379747a4040118130ca35947aef6942b2cd1cd67Erik Kline @Override 415379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public void close() { 416379747a4040118130ca35947aef6942b2cd1cd67Erik Kline IoUtils.closeQuietly(mFileDescriptor); 417379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 418379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 419379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 420379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 421379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private class IcmpCheck extends SimpleSocketCheck implements Runnable { 422379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int TIMEOUT_SEND = 100; 423379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int TIMEOUT_RECV = 300; 424379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int ICMPV4_ECHO_REQUEST = 8; 425379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int ICMPV6_ECHO_REQUEST = 128; 426379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int PACKET_BUFSIZE = 512; 427379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final int mProtocol; 428379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final int mIcmpType; 429379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 43029f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti public IcmpCheck(InetAddress source, InetAddress target, Measurement measurement) { 43129f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti super(source, target, measurement); 432379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 433379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mAddressFamily == AF_INET6) { 434379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mProtocol = IPPROTO_ICMPV6; 435379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mIcmpType = ICMPV6_ECHO_REQUEST; 436379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description = "ICMPv6"; 437379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } else { 438379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mProtocol = IPPROTO_ICMP; 439379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mIcmpType = ICMPV4_ECHO_REQUEST; 440379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description = "ICMPv4"; 441379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 442379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 443379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description += " dst{" + mTarget.getHostAddress() + "}"; 444379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 445379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 44629f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti public IcmpCheck(InetAddress target, Measurement measurement) { 44729f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti this(null, target, measurement); 44829f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti } 44929f666688d73dcaf3f65b8124f34841927f70186Lorenzo Colitti 450379747a4040118130ca35947aef6942b2cd1cd67Erik Kline @Override 451379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public void run() { 452379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Check if this measurement has already failed during setup. 453379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mMeasurement.finishTime > 0) { 454379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // If the measurement failed during construction it didn't 455379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // decrement the countdown latch; do so here. 456379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mCountDownLatch.countDown(); 457379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return; 458379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 459379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 460379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 461379747a4040118130ca35947aef6942b2cd1cd67Erik Kline setupSocket(SOCK_DGRAM, mProtocol, TIMEOUT_SEND, TIMEOUT_RECV, 0); 462379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (ErrnoException | IOException e) { 463379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure(e.toString()); 464379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return; 465379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 466379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description += " src{" + getSocketAddressString() + "}"; 467379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 468379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Build a trivial ICMP packet. 469379747a4040118130ca35947aef6942b2cd1cd67Erik Kline final byte[] icmpPacket = { 470379747a4040118130ca35947aef6942b2cd1cd67Erik Kline (byte) mIcmpType, 0, 0, 0, 0, 0, 0, 0 // ICMP header 471379747a4040118130ca35947aef6942b2cd1cd67Erik Kline }; 472379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 473379747a4040118130ca35947aef6942b2cd1cd67Erik Kline int count = 0; 474379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.startTime = now(); 475379747a4040118130ca35947aef6942b2cd1cd67Erik Kline while (now() < mDeadlineTime - (TIMEOUT_SEND + TIMEOUT_RECV)) { 476379747a4040118130ca35947aef6942b2cd1cd67Erik Kline count++; 477379747a4040118130ca35947aef6942b2cd1cd67Erik Kline icmpPacket[icmpPacket.length - 1] = (byte) count; 478379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 479379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.write(mFileDescriptor, icmpPacket, 0, icmpPacket.length); 480379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (ErrnoException | InterruptedIOException e) { 481379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure(e.toString()); 482379747a4040118130ca35947aef6942b2cd1cd67Erik Kline break; 483379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 484379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 485379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 486379747a4040118130ca35947aef6942b2cd1cd67Erik Kline ByteBuffer reply = ByteBuffer.allocate(PACKET_BUFSIZE); 487379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.read(mFileDescriptor, reply); 488379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // TODO: send a few pings back to back to guesstimate packet loss. 489379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordSuccess("1/" + count); 490379747a4040118130ca35947aef6942b2cd1cd67Erik Kline break; 491379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (ErrnoException | InterruptedIOException e) { 492379747a4040118130ca35947aef6942b2cd1cd67Erik Kline continue; 493379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 494379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 495379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mMeasurement.finishTime == 0) { 496379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure("0/" + count); 497379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 498379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 499379747a4040118130ca35947aef6942b2cd1cd67Erik Kline close(); 500379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 501379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 502379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 503379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 504379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private class DnsUdpCheck extends SimpleSocketCheck implements Runnable { 505379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int TIMEOUT_SEND = 100; 506379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int TIMEOUT_RECV = 500; 507379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int DNS_SERVER_PORT = 53; 508379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int RR_TYPE_A = 1; 509379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int RR_TYPE_AAAA = 28; 510379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private static final int PACKET_BUFSIZE = 512; 511379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 512379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final Random mRandom = new Random(); 513379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 514379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Should be static, but the compiler mocks our puny, human attempts at reason. 515379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private String responseCodeStr(int rcode) { 516379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 517379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return DnsResponseCode.values()[rcode].toString(); 518379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (IndexOutOfBoundsException e) { 519379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return String.valueOf(rcode); 520379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 521379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 522379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 523379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private final int mQueryType; 524379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 525379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public DnsUdpCheck(InetAddress target, Measurement measurement) { 526379747a4040118130ca35947aef6942b2cd1cd67Erik Kline super(target, measurement); 527379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 528379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // TODO: Ideally, query the target for both types regardless of address family. 529379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mAddressFamily == AF_INET6) { 530379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mQueryType = RR_TYPE_AAAA; 531379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } else { 532379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mQueryType = RR_TYPE_A; 533379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 534379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 535379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description = "DNS UDP dst{" + mTarget.getHostAddress() + "}"; 536379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 537379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 538379747a4040118130ca35947aef6942b2cd1cd67Erik Kline @Override 539379747a4040118130ca35947aef6942b2cd1cd67Erik Kline public void run() { 540379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Check if this measurement has already failed during setup. 541379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mMeasurement.finishTime > 0) { 542379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // If the measurement failed during construction it didn't 543379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // decrement the countdown latch; do so here. 544379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mCountDownLatch.countDown(); 545379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return; 546379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 547379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 548379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 549379747a4040118130ca35947aef6942b2cd1cd67Erik Kline setupSocket(SOCK_DGRAM, IPPROTO_UDP, TIMEOUT_SEND, TIMEOUT_RECV, DNS_SERVER_PORT); 550379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (ErrnoException | IOException e) { 551379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure(e.toString()); 552379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return; 553379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 554379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description += " src{" + getSocketAddressString() + "}"; 555379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 556379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // This needs to be fixed length so it can be dropped into the pre-canned packet. 557a09b4d2a611a7606e8fc8c73a24bd941b6fc173fNarayan Kamath final String sixRandomDigits = String.valueOf(mRandom.nextInt(900000) + 100000); 558379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.description += " qtype{" + mQueryType + "}" 559379747a4040118130ca35947aef6942b2cd1cd67Erik Kline + " qname{" + sixRandomDigits + "-android-ds.metric.gstatic.com}"; 560379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 561379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // Build a trivial DNS packet. 562379747a4040118130ca35947aef6942b2cd1cd67Erik Kline final byte[] dnsPacket = getDnsQueryPacket(sixRandomDigits); 563379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 564379747a4040118130ca35947aef6942b2cd1cd67Erik Kline int count = 0; 565379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.startTime = now(); 566379747a4040118130ca35947aef6942b2cd1cd67Erik Kline while (now() < mDeadlineTime - (TIMEOUT_RECV + TIMEOUT_RECV)) { 567379747a4040118130ca35947aef6942b2cd1cd67Erik Kline count++; 568379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 569379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.write(mFileDescriptor, dnsPacket, 0, dnsPacket.length); 570379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (ErrnoException | InterruptedIOException e) { 571379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure(e.toString()); 572379747a4040118130ca35947aef6942b2cd1cd67Erik Kline break; 573379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 574379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 575379747a4040118130ca35947aef6942b2cd1cd67Erik Kline try { 576379747a4040118130ca35947aef6942b2cd1cd67Erik Kline ByteBuffer reply = ByteBuffer.allocate(PACKET_BUFSIZE); 577379747a4040118130ca35947aef6942b2cd1cd67Erik Kline Os.read(mFileDescriptor, reply); 578379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // TODO: more correct and detailed evaluation of the response, 579379747a4040118130ca35947aef6942b2cd1cd67Erik Kline // possibly adding the returned IP address(es) to the output. 580379747a4040118130ca35947aef6942b2cd1cd67Erik Kline final String rcodeStr = (reply.limit() > 3) 581379747a4040118130ca35947aef6942b2cd1cd67Erik Kline ? " " + responseCodeStr((int) (reply.get(3)) & 0x0f) 582379747a4040118130ca35947aef6942b2cd1cd67Erik Kline : ""; 583379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordSuccess("1/" + count + rcodeStr); 584379747a4040118130ca35947aef6942b2cd1cd67Erik Kline break; 585379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } catch (ErrnoException | InterruptedIOException e) { 586379747a4040118130ca35947aef6942b2cd1cd67Erik Kline continue; 587379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 588379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 589379747a4040118130ca35947aef6942b2cd1cd67Erik Kline if (mMeasurement.finishTime == 0) { 590379747a4040118130ca35947aef6942b2cd1cd67Erik Kline mMeasurement.recordFailure("0/" + count); 591379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 592379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 593379747a4040118130ca35947aef6942b2cd1cd67Erik Kline close(); 594379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 595379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 596379747a4040118130ca35947aef6942b2cd1cd67Erik Kline private byte[] getDnsQueryPacket(String sixRandomDigits) { 597379747a4040118130ca35947aef6942b2cd1cd67Erik Kline byte[] rnd = sixRandomDigits.getBytes(StandardCharsets.US_ASCII); 598379747a4040118130ca35947aef6942b2cd1cd67Erik Kline return new byte[] { 599379747a4040118130ca35947aef6942b2cd1cd67Erik Kline (byte) mRandom.nextInt(), (byte) mRandom.nextInt(), // [0-1] query ID 600379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 1, 0, // [2-3] flags; byte[2] = 1 for recursion desired (RD). 601379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, 1, // [4-5] QDCOUNT (number of queries) 602379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, 0, // [6-7] ANCOUNT (number of answers) 603379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, 0, // [8-9] NSCOUNT (number of name server records) 604379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, 0, // [10-11] ARCOUNT (number of additional records) 605379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 17, rnd[0], rnd[1], rnd[2], rnd[3], rnd[4], rnd[5], 606379747a4040118130ca35947aef6942b2cd1cd67Erik Kline '-', 'a', 'n', 'd', 'r', 'o', 'i', 'd', '-', 'd', 's', 607379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 6, 'm', 'e', 't', 'r', 'i', 'c', 608379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 7, 'g', 's', 't', 'a', 't', 'i', 'c', 609379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 3, 'c', 'o', 'm', 610379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, // null terminator of FQDN (root TLD) 611379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, (byte) mQueryType, // QTYPE 612379747a4040118130ca35947aef6942b2cd1cd67Erik Kline 0, 1 // QCLASS, set to 1 = IN (Internet) 613379747a4040118130ca35947aef6942b2cd1cd67Erik Kline }; 614379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 615379747a4040118130ca35947aef6942b2cd1cd67Erik Kline } 616379747a4040118130ca35947aef6942b2cd1cd67Erik Kline} 617