154cf1da20
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.
This applies a previous change, this time using DevSdkIgnoreRunner,
as SdkSuppress does not work as expected in branches that have a
codename 2 versions above the SDK version, as is the case in AOSP.
Bug: 187935317
Test: atest ConnectivityCoverageTests on R
atest FrameworksNetTests on S
Change-Id: I5808d34459a4c19238baf8873afa7cf942828460
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 com.android.testutils.DevSdkIgnoreRule;
|
|
import com.android.testutils.DevSdkIgnoreRunner;
|
|
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
|
|
import java.net.InetAddress;
|
|
import java.nio.ByteBuffer;
|
|
|
|
@RunWith(DevSdkIgnoreRunner.class)
|
|
@DevSdkIgnoreRule.IgnoreUpTo(Build.VERSION_CODES.R)
|
|
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());
|
|
}
|
|
}
|