Merge "[DK1]Add TCP polling mechanism"
This commit is contained in:
Chiachang Wang
@@ -0,0 +1,243 @@
|
||||
/*
|
||||
* Copyright (C) 2022 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.server.connectivity;
|
||||
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertThrows;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.mockito.ArgumentMatchers.any;
|
||||
import static org.mockito.Mockito.doNothing;
|
||||
import static org.mockito.Mockito.doReturn;
|
||||
|
||||
import android.content.Context;
|
||||
import android.content.res.Resources;
|
||||
import android.net.INetd;
|
||||
import android.net.MarkMaskParcel;
|
||||
import android.os.Build;
|
||||
import android.os.HandlerThread;
|
||||
import android.test.suitebuilder.annotation.SmallTest;
|
||||
|
||||
import com.android.connectivity.resources.R;
|
||||
import com.android.testutils.DevSdkIgnoreRule;
|
||||
import com.android.testutils.DevSdkIgnoreRunner;
|
||||
|
||||
import libcore.util.HexEncoding;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.MockitoAnnotations;
|
||||
|
||||
import java.nio.ByteBuffer;
|
||||
import java.nio.ByteOrder;
|
||||
|
||||
@RunWith(DevSdkIgnoreRunner.class)
|
||||
@SmallTest
|
||||
@DevSdkIgnoreRule.IgnoreUpTo(Build.VERSION_CODES.R)
|
||||
public class KeepaliveTrackerTest {
|
||||
private static final int[] TEST_SUPPORTED_KEEPALIVES = {1, 3, 0, 0, 0, 0, 0, 0, 0};
|
||||
private static final int TEST_NETID = 0xA85;
|
||||
private static final int TEST_NETID_FWMARK = 0x0A85;
|
||||
private static final int OTHER_NETID = 0x1A85;
|
||||
private static final int NETID_MASK = 0xffff;
|
||||
private static final int SUPPORTED_SLOT_COUNT = 2;
|
||||
private KeepaliveTracker mKeepaliveTracker;
|
||||
private HandlerThread mHandlerThread;
|
||||
|
||||
@Mock INetd mNetd;
|
||||
@Mock KeepaliveTracker.Dependencies mDependencies;
|
||||
@Mock Context mCtx;
|
||||
@Mock Resources mResources;
|
||||
|
||||
// Hexadecimal representation of a SOCK_DIAG response with tcp info.
|
||||
private static final String SOCK_DIAG_TCP_INET_HEX =
|
||||
// struct nlmsghdr.
|
||||
"14010000" + // length = 276
|
||||
"1400" + // type = SOCK_DIAG_BY_FAMILY
|
||||
"0301" + // flags = NLM_F_REQUEST | NLM_F_DUMP
|
||||
"00000000" + // seqno
|
||||
"00000000" + // pid (0 == kernel)
|
||||
// struct inet_diag_req_v2
|
||||
"02" + // family = AF_INET
|
||||
"06" + // state
|
||||
"00" + // timer
|
||||
"00" + // retrans
|
||||
// inet_diag_sockid
|
||||
"DEA5" + // idiag_sport = 42462
|
||||
"71B9" + // idiag_dport = 47473
|
||||
"0a006402000000000000000000000000" + // idiag_src = 10.0.100.2
|
||||
"08080808000000000000000000000000" + // idiag_dst = 8.8.8.8
|
||||
"00000000" + // idiag_if
|
||||
"34ED000076270000" + // idiag_cookie = 43387759684916
|
||||
"00000000" + // idiag_expires
|
||||
"00000000" + // idiag_rqueue
|
||||
"00000000" + // idiag_wqueue
|
||||
"00000000" + // idiag_uid
|
||||
"00000000" + // idiag_inode
|
||||
// rtattr
|
||||
"0500" + // len = 5
|
||||
"0800" + // type = 8
|
||||
"00000000" + // data
|
||||
"0800" + // len = 8
|
||||
"0F00" + // type = 15(INET_DIAG_MARK)
|
||||
"850A0C00" + // data, socket mark=789125
|
||||
"AC00" + // len = 172
|
||||
"0200" + // type = 2(INET_DIAG_INFO)
|
||||
// tcp_info
|
||||
"01" + // state = TCP_ESTABLISHED
|
||||
"00" + // ca_state = TCP_CA_OPEN
|
||||
"05" + // retransmits = 5
|
||||
"00" + // probes = 0
|
||||
"00" + // backoff = 0
|
||||
"07" + // option = TCPI_OPT_WSCALE|TCPI_OPT_SACK|TCPI_OPT_TIMESTAMPS
|
||||
"88" + // wscale = 8
|
||||
"00" + // delivery_rate_app_limited = 0
|
||||
"4A911B00" + // rto = 1806666
|
||||
"00000000" + // ato = 0
|
||||
"2E050000" + // sndMss = 1326
|
||||
"18020000" + // rcvMss = 536
|
||||
"00000000" + // unsacked = 0
|
||||
"00000000" + // acked = 0
|
||||
"00000000" + // lost = 0
|
||||
"00000000" + // retrans = 0
|
||||
"00000000" + // fackets = 0
|
||||
"BB000000" + // lastDataSent = 187
|
||||
"00000000" + // lastAckSent = 0
|
||||
"BB000000" + // lastDataRecv = 187
|
||||
"BB000000" + // lastDataAckRecv = 187
|
||||
"DC050000" + // pmtu = 1500
|
||||
"30560100" + // rcvSsthresh = 87600
|
||||
"3E2C0900" + // rttt = 601150
|
||||
"1F960400" + // rttvar = 300575
|
||||
"78050000" + // sndSsthresh = 1400
|
||||
"0A000000" + // sndCwnd = 10
|
||||
"A8050000" + // advmss = 1448
|
||||
"03000000" + // reordering = 3
|
||||
"00000000" + // rcvrtt = 0
|
||||
"30560100" + // rcvspace = 87600
|
||||
"00000000" + // totalRetrans = 0
|
||||
"53AC000000000000" + // pacingRate = 44115
|
||||
"FFFFFFFFFFFFFFFF" + // maxPacingRate = 18446744073709551615
|
||||
"0100000000000000" + // bytesAcked = 1
|
||||
"0000000000000000" + // bytesReceived = 0
|
||||
"0A000000" + // SegsOut = 10
|
||||
"00000000" + // SegsIn = 0
|
||||
"00000000" + // NotSentBytes = 0
|
||||
"3E2C0900" + // minRtt = 601150
|
||||
"00000000" + // DataSegsIn = 0
|
||||
"00000000" + // DataSegsOut = 0
|
||||
"0000000000000000"; // deliverRate = 0
|
||||
private static final String SOCK_DIAG_NO_TCP_INET_HEX =
|
||||
// struct nlmsghdr
|
||||
"14000000" // length = 20
|
||||
+ "0300" // type = NLMSG_DONE
|
||||
+ "0301" // flags = NLM_F_REQUEST | NLM_F_DUMP
|
||||
+ "00000000" // seqno
|
||||
+ "00000000" // pid (0 == kernel)
|
||||
// struct inet_diag_req_v2
|
||||
+ "02" // family = AF_INET
|
||||
+ "06" // state
|
||||
+ "00" // timer
|
||||
+ "00"; // retrans
|
||||
private static final byte[] SOCK_DIAG_NO_TCP_INET_BYTES =
|
||||
HexEncoding.decode(SOCK_DIAG_NO_TCP_INET_HEX.toCharArray(), false);
|
||||
private static final String TEST_RESPONSE_HEX =
|
||||
SOCK_DIAG_TCP_INET_HEX + SOCK_DIAG_NO_TCP_INET_HEX;
|
||||
private static final byte[] TEST_RESPONSE_BYTES =
|
||||
HexEncoding.decode(TEST_RESPONSE_HEX.toCharArray(), false);
|
||||
|
||||
@Before
|
||||
public void setup() throws Exception {
|
||||
MockitoAnnotations.initMocks(this);
|
||||
|
||||
doReturn(mNetd).when(mDependencies).getNetd();
|
||||
doReturn(makeMarkMaskParcel(NETID_MASK, TEST_NETID_FWMARK)).when(mNetd)
|
||||
.getFwmarkForNetwork(TEST_NETID);
|
||||
|
||||
doReturn(TEST_SUPPORTED_KEEPALIVES).when(mDependencies).getSupportedKeepalives();
|
||||
doReturn(mResources).when(mDependencies).newConnectivityResources();
|
||||
mockResource();
|
||||
doNothing().when(mDependencies).sendRequest(any(), any());
|
||||
|
||||
mHandlerThread = new HandlerThread("KeepaliveTrackerTest");
|
||||
mHandlerThread.start();
|
||||
|
||||
mKeepaliveTracker = new KeepaliveTracker(mCtx, mHandlerThread.getThreadHandler(),
|
||||
mDependencies);
|
||||
}
|
||||
|
||||
private void mockResource() {
|
||||
doReturn(SUPPORTED_SLOT_COUNT).when(mResources).getInteger(
|
||||
R.integer.config_reservedPrivilegedKeepaliveSlots);
|
||||
doReturn(SUPPORTED_SLOT_COUNT).when(mResources).getInteger(
|
||||
R.integer.config_allowedUnprivilegedKeepalivePerUid);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsAnyTcpSocketConnected_runOnNonHandlerThread() throws Exception {
|
||||
setupResponseWithSocketExisting();
|
||||
assertThrows(IllegalStateException.class,
|
||||
() -> mKeepaliveTracker.isAnyTcpSocketConnected(TEST_NETID));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsAnyTcpSocketConnected_withTargetNetId() throws Exception {
|
||||
setupResponseWithSocketExisting();
|
||||
mHandlerThread.getThreadHandler().post(
|
||||
() -> assertTrue(mKeepaliveTracker.isAnyTcpSocketConnected(TEST_NETID)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsAnyTcpSocketConnected_withIncorrectNetId() throws Exception {
|
||||
setupResponseWithSocketExisting();
|
||||
mHandlerThread.getThreadHandler().post(
|
||||
() -> assertFalse(mKeepaliveTracker.isAnyTcpSocketConnected(OTHER_NETID)));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIsAnyTcpSocketConnected_noSocketExists() throws Exception {
|
||||
setupResponseWithoutSocketExisting();
|
||||
mHandlerThread.getThreadHandler().post(
|
||||
() -> assertFalse(mKeepaliveTracker.isAnyTcpSocketConnected(TEST_NETID)));
|
||||
}
|
||||
|
||||
private void setupResponseWithSocketExisting() throws Exception {
|
||||
final ByteBuffer tcpBufferV6 = getByteBuffer(TEST_RESPONSE_BYTES);
|
||||
final ByteBuffer tcpBufferV4 = getByteBuffer(TEST_RESPONSE_BYTES);
|
||||
doReturn(tcpBufferV6, tcpBufferV4).when(mDependencies).recvSockDiagResponse(any());
|
||||
}
|
||||
|
||||
private void setupResponseWithoutSocketExisting() throws Exception {
|
||||
final ByteBuffer tcpBufferV6 = getByteBuffer(SOCK_DIAG_NO_TCP_INET_BYTES);
|
||||
final ByteBuffer tcpBufferV4 = getByteBuffer(SOCK_DIAG_NO_TCP_INET_BYTES);
|
||||
doReturn(tcpBufferV6, tcpBufferV4).when(mDependencies).recvSockDiagResponse(any());
|
||||
}
|
||||
|
||||
private MarkMaskParcel makeMarkMaskParcel(final int mask, final int mark) {
|
||||
final MarkMaskParcel parcel = new MarkMaskParcel();
|
||||
parcel.mask = mask;
|
||||
parcel.mark = mark;
|
||||
return parcel;
|
||||
}
|
||||
|
||||
private ByteBuffer getByteBuffer(final byte[] bytes) {
|
||||
final ByteBuffer buffer = ByteBuffer.wrap(bytes);
|
||||
buffer.order(ByteOrder.nativeOrder());
|
||||
return buffer;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user