14414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey/* 24414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Copyright (C) 2011 The Android Open Source Project 34414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * 44414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Licensed under the Apache License, Version 2.0 (the "License"); 54414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * you may not use this file except in compliance with the License. 64414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * You may obtain a copy of the License at 74414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * 84414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * http://www.apache.org/licenses/LICENSE-2.0 94414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * 104414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Unless required by applicable law or agreed to in writing, software 114414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * distributed under the License is distributed on an "AS IS" BASIS, 124414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 134414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * See the License for the specific language governing permissions and 144414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * limitations under the License. 154414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey */ 164414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 171059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkeypackage com.android.internal.net; 184414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 19ada23ed56addb7c06a489c7464ae228a46588b95Lorenzo Colittiimport static android.net.NetworkStats.DEFAULT_NETWORK_NO; 20c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chenimport static android.net.NetworkStats.METERED_NO; 211f7e05eda687279ac9deb894f42ac927bd471ad2Jeff Davidsonimport static android.net.NetworkStats.ROAMING_NO; 228fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkeyimport static android.net.NetworkStats.SET_ALL; 23b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkeyimport static android.net.NetworkStats.SET_DEFAULT; 24b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkeyimport static android.net.NetworkStats.SET_FOREGROUND; 25fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkeyimport static android.net.NetworkStats.TAG_NONE; 26fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkeyimport static android.net.NetworkStats.UID_ALL; 278568db534118fc14cc28100306d51626464ff319Jesse Wilsonimport static com.android.server.NetworkManagementSocketTagger.kernelToTag; 284a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport static org.junit.Assert.assertEquals; 294a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport static org.junit.Assert.fail; 304414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 314414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport android.content.res.Resources; 324414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport android.net.NetworkStats; 332d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkeyimport android.net.TrafficStats; 344a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport android.support.test.InstrumentationRegistry; 35e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichiimport android.support.test.filters.SmallTest; 364a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport android.support.test.runner.AndroidJUnit4; 374414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 38c86013be936f36c61fa0cdfad95141f6de3c6ef7Lorenzo Colittiimport com.android.frameworks.tests.net.R; 394414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 404414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.File; 414414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.FileOutputStream; 42fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkeyimport java.io.FileWriter; 434414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.InputStream; 444414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.OutputStream; 454414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 464414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport libcore.io.IoUtils; 474414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport libcore.io.Streams; 484414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 494a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport org.junit.runner.RunWith; 504a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport org.junit.After; 514a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport org.junit.Before; 524a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichiimport org.junit.Test; 534a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi 544414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey/** 551059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey * Tests for {@link NetworkStatsFactory}. 564414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey */ 574a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi@RunWith(AndroidJUnit4.class) 58e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi@SmallTest 594a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichipublic class NetworkStatsFactoryTest { 604414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey private File mTestProc; 611059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey private NetworkStatsFactory mFactory; 624414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 634a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Before 644414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void setUp() throws Exception { 654a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi mTestProc = new File(InstrumentationRegistry.getContext().getFilesDir(), "proc"); 66fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey if (mTestProc.exists()) { 67a63ba59260cd1bb3f5c16e395ace45a61f1d4461Jeff Sharkey IoUtils.deleteContents(mTestProc); 68fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 69fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey 70828f1b45fb302f68b5eb8934bf474177fa249c7cChenbo Feng mFactory = new NetworkStatsFactory(mTestProc, false); 714414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 724414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 734a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @After 744414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void tearDown() throws Exception { 751059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey mFactory = null; 764414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 77fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey if (mTestProc.exists()) { 78a63ba59260cd1bb3f5c16e395ace45a61f1d4461Jeff Sharkey IoUtils.deleteContents(mTestProc); 79fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 804414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 814414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 824a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Test 834414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void testNetworkStatsDetail() throws Exception { 84e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi final NetworkStats stats = parseDetailedStats(R.raw.xt_qtaguid_typical); 854414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 86163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertEquals(70, stats.size()); 87163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "wlan0", 0, SET_DEFAULT, 0x0, 18621L, 2898L); 88163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "wlan0", 10011, SET_DEFAULT, 0x0, 35777L, 5718L); 89163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "wlan0", 10021, SET_DEFAULT, 0x7fffff01, 562386L, 49228L); 90163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "rmnet1", 10021, SET_DEFAULT, 0x30100000, 219110L, 227423L); 91163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "rmnet2", 10001, SET_DEFAULT, 0x0, 1125899906842624L, 984L); 924414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 934414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 944a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Test 954414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void testKernelTags() throws Exception { 964414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(0, kernelToTag("0x0000000000000000")); 974414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(0x32, kernelToTag("0x0000003200000000")); 984414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(2147483647, kernelToTag("0x7fffffff00000000")); 994414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(0, kernelToTag("0x0000000000000000")); 1004414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(2147483136, kernelToTag("0x7FFFFE0000000000")); 1012d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey 1022d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(0, kernelToTag("0x0")); 1032d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(0, kernelToTag("0xf00d")); 1042d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(1, kernelToTag("0x100000000")); 1052d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(14438007, kernelToTag("0xdc4e7700000000")); 1062d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(TrafficStats.TAG_SYSTEM_DOWNLOAD, kernelToTag("0xffffff0100000000")); 1074414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 1084414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 1094a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Test 110b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey public void testNetworkStatsWithSet() throws Exception { 111e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi final NetworkStats stats = parseDetailedStats(R.raw.xt_qtaguid_typical); 112163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertEquals(70, stats.size()); 113c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen assertStatsEntry(stats, "rmnet1", 10021, SET_DEFAULT, 0x30100000, 219110L, 578L, 227423L, 114c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen 676L); 115163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "rmnet1", 10021, SET_FOREGROUND, 0x30100000, 742L, 3L, 1265L, 3L); 11693afbf258f554e980b58c3c67d55ebbb3e4eb915Chenbo Feng } 11793afbf258f554e980b58c3c67d55ebbb3e4eb915Chenbo Feng 11893afbf258f554e980b58c3c67d55ebbb3e4eb915Chenbo Feng @Test 119ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey public void testNetworkStatsSingle() throws Exception { 120e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi stageFile(R.raw.xt_qtaguid_iface_typical, file("net/xt_qtaguid/iface_stat_all")); 121ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey 122e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey final NetworkStats stats = mFactory.readNetworkStatsSummaryDev(); 123ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey assertEquals(6, stats.size()); 1248fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkey assertStatsEntry(stats, "rmnet0", UID_ALL, SET_ALL, TAG_NONE, 2112L, 24L, 700L, 10L); 1258fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkey assertStatsEntry(stats, "test1", UID_ALL, SET_ALL, TAG_NONE, 6L, 8L, 10L, 12L); 1268fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkey assertStatsEntry(stats, "test2", UID_ALL, SET_ALL, TAG_NONE, 1L, 2L, 3L, 4L); 127ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey } 128ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey 1294a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Test 130e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey public void testNetworkStatsXt() throws Exception { 131e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi stageFile(R.raw.xt_qtaguid_iface_fmt_typical, file("net/xt_qtaguid/iface_stat_fmt")); 132e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey 133e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey final NetworkStats stats = mFactory.readNetworkStatsSummaryXt(); 134e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey assertEquals(3, stats.size()); 135e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey assertStatsEntry(stats, "rmnet0", UID_ALL, SET_ALL, TAG_NONE, 6824L, 16L, 5692L, 10L); 136c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen assertStatsEntry(stats, "rmnet1", UID_ALL, SET_ALL, TAG_NONE, 11153922L, 8051L, 190226L, 137c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen 2468L); 138e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey assertStatsEntry(stats, "rmnet2", UID_ALL, SET_ALL, TAG_NONE, 4968L, 35L, 3081L, 39L); 139e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey } 140e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey 1414a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Test 142e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi public void testDoubleClatAccounting() throws Exception { 143e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi NetworkStatsFactory.noteStackedIface("v4-wlan0", "wlan0"); 144e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 145e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // xt_qtaguid_with_clat_simple is a synthetic file that simulates 146e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // - 213 received 464xlat packets of size 200 bytes 147e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // - 41 sent 464xlat packets of size 100 bytes 148e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // - no other traffic on base interface for root uid. 149e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi NetworkStats stats = parseDetailedStats(R.raw.xt_qtaguid_with_clat_simple); 150e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertEquals(4, stats.size()); 151e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 152e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 10060, SET_DEFAULT, 0x0, 46860L, 4920L); 153e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 0, SET_DEFAULT, 0x0, 0L, 0L); 154e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 155e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi stats = parseDetailedStats(R.raw.xt_qtaguid_with_clat); 156e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertEquals(42, stats.size()); 157e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 158e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 0, SET_DEFAULT, 0x0, 356L, 276L); 159e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 1000, SET_DEFAULT, 0x0, 30812L, 2310L); 160e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 10102, SET_DEFAULT, 0x0, 10022L, 3330L); 161e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 10060, SET_DEFAULT, 0x0, 9532772L, 254112L); 162e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 0, SET_DEFAULT, 0x0, 15229L, 5766L); 163e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 1000, SET_DEFAULT, 0x0, 6126L, 2013L); 164e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 10013, SET_DEFAULT, 0x0, 0L, 144L); 165e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 10018, SET_DEFAULT, 0x0, 5980263L, 167667L); 166e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 10060, SET_DEFAULT, 0x0, 134356L, 8705L); 167e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 10079, SET_DEFAULT, 0x0, 10926L, 1507L); 168e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 10102, SET_DEFAULT, 0x0, 25038L, 8245L); 169e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 10103, SET_DEFAULT, 0x0, 0L, 192L); 170e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "dummy0", 0, SET_DEFAULT, 0x0, 0L, 168L); 171e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "lo", 0, SET_DEFAULT, 0x0, 1288L, 1288L); 172e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 173b6a920124f28422877f59bfb32719099a0067d76Remi NGUYEN VAN NetworkStatsFactory.clearStackedIfaces(); 174e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi } 175e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 1764a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi @Test 177e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi public void testDoubleClatAccounting100MBDownload() throws Exception { 178e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // Downloading 100mb from an ipv4 only destination in a foreground activity 179e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 180e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi long appRxBytesBefore = 328684029L; 181e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi long appRxBytesAfter = 439237478L; 182e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertEquals("App traffic should be ~100MB", 110553449, appRxBytesAfter - appRxBytesBefore); 183e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 184e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi long rootRxBytesBefore = 1394011L; 185e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi long rootRxBytesAfter = 1398634L; 186e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertEquals("UID 0 traffic should be ~0", 4623, rootRxBytesAfter - rootRxBytesBefore); 187e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 188e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi NetworkStatsFactory.noteStackedIface("v4-wlan0", "wlan0"); 189e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi NetworkStats stats; 190e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 191e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // Stats snapshot before the download 192e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi stats = parseDetailedStats(R.raw.xt_qtaguid_with_clat_100mb_download_before); 193e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 10106, SET_FOREGROUND, 0x0, appRxBytesBefore, 5199872L); 194e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 0, SET_DEFAULT, 0x0, rootRxBytesBefore, 647888L); 195e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 196e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi // Stats snapshot after the download 197e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi stats = parseDetailedStats(R.raw.xt_qtaguid_with_clat_100mb_download_after); 198e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "v4-wlan0", 10106, SET_FOREGROUND, 0x0, appRxBytesAfter, 7867488L); 199e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi assertStatsEntry(stats, "wlan0", 0, SET_DEFAULT, 0x0, rootRxBytesAfter, 647587L); 200e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 201b6a920124f28422877f59bfb32719099a0067d76Remi NGUYEN VAN NetworkStatsFactory.clearStackedIfaces(); 202e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi } 203e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 2044414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey /** 2054414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Copy a {@link Resources#openRawResource(int)} into {@link File} for 2064414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * testing purposes. 2074414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey */ 2084414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey private void stageFile(int rawId, File file) throws Exception { 2094414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey new File(file.getParent()).mkdirs(); 2104414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey InputStream in = null; 2114414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey OutputStream out = null; 2124414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey try { 2134a0c5d7ef7144280fe8a209a871bbd4ef90d6368Hugo Benichi in = InstrumentationRegistry.getContext().getResources().openRawResource(rawId); 2144414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey out = new FileOutputStream(file); 2154414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey Streams.copy(in, out); 2164414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } finally { 2174414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey IoUtils.closeQuietly(in); 2184414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey IoUtils.closeQuietly(out); 2194414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 2204414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 2214414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 222fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey private void stageLong(long value, File file) throws Exception { 223fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey new File(file.getParent()).mkdirs(); 224fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey FileWriter out = null; 225fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey try { 226fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey out = new FileWriter(file); 227fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey out.write(Long.toString(value)); 228fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } finally { 229fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey IoUtils.closeQuietly(out); 230fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 231fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 232fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey 233e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi private File file(String path) throws Exception { 234e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi return new File(mTestProc, path); 235e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi } 236e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 237e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi private NetworkStats parseDetailedStats(int resourceId) throws Exception { 238e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi stageFile(resourceId, file("net/xt_qtaguid/stats")); 239e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi return mFactory.readNetworkStatsDetail(); 240e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi } 241e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi 242b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey private static void assertStatsEntry(NetworkStats stats, String iface, int uid, int set, 243b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey int tag, long rxBytes, long txBytes) { 244ada23ed56addb7c06a489c7464ae228a46588b95Lorenzo Colitti final int i = stats.findIndex(iface, uid, set, tag, METERED_NO, ROAMING_NO, 245ada23ed56addb7c06a489c7464ae228a46588b95Lorenzo Colitti DEFAULT_NETWORK_NO); 246e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi if (i < 0) { 247e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi fail(String.format("no NetworkStats for (iface: %s, uid: %d, set: %d, tag: %d)", 248e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi iface, uid, set, tag)); 249e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi } 250b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey final NetworkStats.Entry entry = stats.getValues(i, null); 251b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected rxBytes", rxBytes, entry.rxBytes); 252b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected txBytes", txBytes, entry.txBytes); 253b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey } 254b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey 255b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey private static void assertStatsEntry(NetworkStats stats, String iface, int uid, int set, 256b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey int tag, long rxBytes, long rxPackets, long txBytes, long txPackets) { 257ada23ed56addb7c06a489c7464ae228a46588b95Lorenzo Colitti final int i = stats.findIndex(iface, uid, set, tag, METERED_NO, ROAMING_NO, 258ada23ed56addb7c06a489c7464ae228a46588b95Lorenzo Colitti DEFAULT_NETWORK_NO); 259e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi if (i < 0) { 260e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi fail(String.format("no NetworkStats for (iface: %s, uid: %d, set: %d, tag: %d)", 261e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi iface, uid, set, tag)); 262e1bb3a14d1ee1aba2f92e5bccf659774699eaffeHugo Benichi } 263fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey final NetworkStats.Entry entry = stats.getValues(i, null); 264b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected rxBytes", rxBytes, entry.rxBytes); 265b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected rxPackets", rxPackets, entry.rxPackets); 266b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected txBytes", txBytes, entry.txBytes); 267b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected txPackets", txPackets, entry.txPackets); 2684414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 2694414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey} 270