05dd1ba540
The tests are packaged together with other tests in a common coverage suite, so need to be individually annotated so that they are skipped before S. They do not need to pass on platforms older than S, because such platforms will not updated with the code that is tested. Bug: 187935317 Test: atest ConnectivityCoverageTests on R Change-Id: I10da45ae023e7f1e8cd11ec857b84346892a3fa8
223 lines
8.7 KiB
Java
223 lines
8.7 KiB
Java
/*
|
|
* Copyright (C) 2020 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 android.net;
|
|
|
|
import static com.android.testutils.ParcelUtils.assertParcelingIsLossless;
|
|
|
|
import static org.junit.Assert.assertArrayEquals;
|
|
import static org.junit.Assert.assertEquals;
|
|
import static org.junit.Assert.assertTrue;
|
|
import static org.junit.Assert.fail;
|
|
|
|
import android.net.util.KeepalivePacketDataUtil;
|
|
import android.os.Build;
|
|
import android.util.Log;
|
|
|
|
import androidx.test.filters.SdkSuppress;
|
|
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.junit.runners.JUnit4;
|
|
|
|
import java.net.InetAddress;
|
|
import java.nio.ByteBuffer;
|
|
|
|
@RunWith(JUnit4.class)
|
|
@SdkSuppress(minSdkVersion = Build.VERSION_CODES.S, codeName = "S")
|
|
public final class KeepalivePacketDataUtilTest {
|
|
private static final byte[] IPV4_KEEPALIVE_SRC_ADDR = {10, 0, 0, 1};
|
|
private static final byte[] IPV4_KEEPALIVE_DST_ADDR = {10, 0, 0, 5};
|
|
|
|
private Log.TerribleFailureHandler mOriginalHandler;
|
|
|
|
@Before
|
|
public void setUp() {
|
|
// Terrible failures are logged when using deprecated methods on newer platforms
|
|
mOriginalHandler = Log.setWtfHandler((tag, what, sys) ->
|
|
Log.e(tag, "Terrible failure in test", what));
|
|
}
|
|
|
|
@After
|
|
public void tearDown() {
|
|
Log.setWtfHandler(mOriginalHandler);
|
|
}
|
|
|
|
@Test
|
|
public void testFromTcpKeepaliveStableParcelable() throws Exception {
|
|
final int srcPort = 1234;
|
|
final int dstPort = 4321;
|
|
final int seq = 0x11111111;
|
|
final int ack = 0x22222222;
|
|
final int wnd = 8000;
|
|
final int wndScale = 2;
|
|
final int tos = 4;
|
|
final int ttl = 64;
|
|
TcpKeepalivePacketData resultData = null;
|
|
final TcpKeepalivePacketDataParcelable testInfo = new TcpKeepalivePacketDataParcelable();
|
|
testInfo.srcAddress = IPV4_KEEPALIVE_SRC_ADDR;
|
|
testInfo.srcPort = srcPort;
|
|
testInfo.dstAddress = IPV4_KEEPALIVE_DST_ADDR;
|
|
testInfo.dstPort = dstPort;
|
|
testInfo.seq = seq;
|
|
testInfo.ack = ack;
|
|
testInfo.rcvWnd = wnd;
|
|
testInfo.rcvWndScale = wndScale;
|
|
testInfo.tos = tos;
|
|
testInfo.ttl = ttl;
|
|
try {
|
|
resultData = KeepalivePacketDataUtil.fromStableParcelable(testInfo);
|
|
} catch (InvalidPacketException e) {
|
|
fail("InvalidPacketException: " + e);
|
|
}
|
|
|
|
assertEquals(InetAddress.getByAddress(testInfo.srcAddress), resultData.getSrcAddress());
|
|
assertEquals(InetAddress.getByAddress(testInfo.dstAddress), resultData.getDstAddress());
|
|
assertEquals(testInfo.srcPort, resultData.getSrcPort());
|
|
assertEquals(testInfo.dstPort, resultData.getDstPort());
|
|
assertEquals(testInfo.seq, resultData.tcpSeq);
|
|
assertEquals(testInfo.ack, resultData.tcpAck);
|
|
assertEquals(testInfo.rcvWnd, resultData.tcpWindow);
|
|
assertEquals(testInfo.rcvWndScale, resultData.tcpWindowScale);
|
|
assertEquals(testInfo.tos, resultData.ipTos);
|
|
assertEquals(testInfo.ttl, resultData.ipTtl);
|
|
|
|
assertParcelingIsLossless(resultData);
|
|
|
|
final byte[] packet = resultData.getPacket();
|
|
// IP version and IHL
|
|
assertEquals(packet[0], 0x45);
|
|
// TOS
|
|
assertEquals(packet[1], tos);
|
|
// TTL
|
|
assertEquals(packet[8], ttl);
|
|
// Source IP address.
|
|
byte[] ip = new byte[4];
|
|
ByteBuffer buf = ByteBuffer.wrap(packet, 12, 4);
|
|
buf.get(ip);
|
|
assertArrayEquals(ip, IPV4_KEEPALIVE_SRC_ADDR);
|
|
// Destination IP address.
|
|
buf = ByteBuffer.wrap(packet, 16, 4);
|
|
buf.get(ip);
|
|
assertArrayEquals(ip, IPV4_KEEPALIVE_DST_ADDR);
|
|
|
|
buf = ByteBuffer.wrap(packet, 20, 12);
|
|
// Source port.
|
|
assertEquals(buf.getShort(), srcPort);
|
|
// Destination port.
|
|
assertEquals(buf.getShort(), dstPort);
|
|
// Sequence number.
|
|
assertEquals(buf.getInt(), seq);
|
|
// Ack.
|
|
assertEquals(buf.getInt(), ack);
|
|
// Window size.
|
|
buf = ByteBuffer.wrap(packet, 34, 2);
|
|
assertEquals(buf.getShort(), wnd >> wndScale);
|
|
}
|
|
|
|
//TODO: add ipv6 test when ipv6 supported
|
|
|
|
@Test
|
|
public void testToTcpKeepaliveStableParcelable() throws Exception {
|
|
final int srcPort = 1234;
|
|
final int dstPort = 4321;
|
|
final int sequence = 0x11111111;
|
|
final int ack = 0x22222222;
|
|
final int wnd = 48_000;
|
|
final int wndScale = 2;
|
|
final int tos = 4;
|
|
final int ttl = 64;
|
|
final TcpKeepalivePacketDataParcelable testInfo = new TcpKeepalivePacketDataParcelable();
|
|
testInfo.srcAddress = IPV4_KEEPALIVE_SRC_ADDR;
|
|
testInfo.srcPort = srcPort;
|
|
testInfo.dstAddress = IPV4_KEEPALIVE_DST_ADDR;
|
|
testInfo.dstPort = dstPort;
|
|
testInfo.seq = sequence;
|
|
testInfo.ack = ack;
|
|
testInfo.rcvWnd = wnd;
|
|
testInfo.rcvWndScale = wndScale;
|
|
testInfo.tos = tos;
|
|
testInfo.ttl = ttl;
|
|
TcpKeepalivePacketData testData = null;
|
|
TcpKeepalivePacketDataParcelable resultData = null;
|
|
testData = KeepalivePacketDataUtil.fromStableParcelable(testInfo);
|
|
resultData = KeepalivePacketDataUtil.toStableParcelable(testData);
|
|
assertArrayEquals(resultData.srcAddress, IPV4_KEEPALIVE_SRC_ADDR);
|
|
assertArrayEquals(resultData.dstAddress, IPV4_KEEPALIVE_DST_ADDR);
|
|
assertEquals(resultData.srcPort, srcPort);
|
|
assertEquals(resultData.dstPort, dstPort);
|
|
assertEquals(resultData.seq, sequence);
|
|
assertEquals(resultData.ack, ack);
|
|
assertEquals(resultData.rcvWnd, wnd);
|
|
assertEquals(resultData.rcvWndScale, wndScale);
|
|
assertEquals(resultData.tos, tos);
|
|
assertEquals(resultData.ttl, ttl);
|
|
|
|
final String expected = ""
|
|
+ "android.net.TcpKeepalivePacketDataParcelable{srcAddress: [10, 0, 0, 1],"
|
|
+ " srcPort: 1234, dstAddress: [10, 0, 0, 5], dstPort: 4321, seq: 286331153,"
|
|
+ " ack: 572662306, rcvWnd: 48000, rcvWndScale: 2, tos: 4, ttl: 64}";
|
|
assertEquals(expected, resultData.toString());
|
|
}
|
|
|
|
@Test
|
|
public void testParseTcpKeepalivePacketData() throws Exception {
|
|
final int srcPort = 1234;
|
|
final int dstPort = 4321;
|
|
final int sequence = 0x11111111;
|
|
final int ack = 0x22222222;
|
|
final int wnd = 4800;
|
|
final int wndScale = 2;
|
|
final int tos = 4;
|
|
final int ttl = 64;
|
|
final TcpKeepalivePacketDataParcelable testParcel = new TcpKeepalivePacketDataParcelable();
|
|
testParcel.srcAddress = IPV4_KEEPALIVE_SRC_ADDR;
|
|
testParcel.srcPort = srcPort;
|
|
testParcel.dstAddress = IPV4_KEEPALIVE_DST_ADDR;
|
|
testParcel.dstPort = dstPort;
|
|
testParcel.seq = sequence;
|
|
testParcel.ack = ack;
|
|
testParcel.rcvWnd = wnd;
|
|
testParcel.rcvWndScale = wndScale;
|
|
testParcel.tos = tos;
|
|
testParcel.ttl = ttl;
|
|
|
|
final KeepalivePacketData testData =
|
|
KeepalivePacketDataUtil.fromStableParcelable(testParcel);
|
|
final TcpKeepalivePacketDataParcelable parsedParcelable =
|
|
KeepalivePacketDataUtil.parseTcpKeepalivePacketData(testData);
|
|
final TcpKeepalivePacketData roundTripData =
|
|
KeepalivePacketDataUtil.fromStableParcelable(parsedParcelable);
|
|
|
|
// Generated packet is the same, but rcvWnd / wndScale will differ if scale is non-zero
|
|
assertTrue(testData.getPacket().length > 0);
|
|
assertArrayEquals(testData.getPacket(), roundTripData.getPacket());
|
|
|
|
testParcel.rcvWndScale = 0;
|
|
final KeepalivePacketData noScaleTestData =
|
|
KeepalivePacketDataUtil.fromStableParcelable(testParcel);
|
|
final TcpKeepalivePacketDataParcelable noScaleParsedParcelable =
|
|
KeepalivePacketDataUtil.parseTcpKeepalivePacketData(noScaleTestData);
|
|
final TcpKeepalivePacketData noScaleRoundTripData =
|
|
KeepalivePacketDataUtil.fromStableParcelable(noScaleParsedParcelable);
|
|
assertEquals(noScaleTestData, noScaleRoundTripData);
|
|
assertTrue(noScaleTestData.getPacket().length > 0);
|
|
assertArrayEquals(noScaleTestData.getPacket(), noScaleRoundTripData.getPacket());
|
|
}
|
|
}
|