LineageOS/android_packages_modules_Connectivity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support testing upload/download packets

Browse Source 
This is a preparing CL to test the packets sent from clients and the
packets replied from remote side.

Bug: 183166581
Test: atest EthernetTetheringTest
Change-Id: I898de024c0d9ecf13d81023035242c4abf856702
This commit is contained in:
markchien
2021-11-26 16:10:12 +08:00
parent d8d887bd39
commit 13e5a48cb7
1 changed files with 29 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
Tethering/tests/integration/src/android/net/TetheringTester.java
View File

@@ -30,7 +30,7 @@ import java.net.Inet4Address;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.util.Random; import java.util.Random;
import java.util.concurrent.TimeoutException; import java.util.concurrent.TimeoutException;
import java.util.function.Function; import java.util.function.Predicate;
/** /**
* A class simulate tethered client. When caller create TetheringTester, it would connect to * A class simulate tethered client. When caller create TetheringTester, it would connect to
@@ -129,27 +129,43 @@ public final class TetheringTester {
mDownstreamReader.sendResponse(packet); mDownstreamReader.sendResponse(packet);
} }
private DhcpPacket getNextDhcpPacket() { private DhcpPacket getNextDhcpPacket() throws Exception {
return getNextMatchedPacket((p) -> { final byte[] packet = getNextMatchedPacket((p) -> {
// Test whether this is DHCP packet.
try { try {
return DhcpPacket.decodeFullPacket(p, p.length, DhcpPacket.ENCAP_L2); DhcpPacket.decodeFullPacket(p, p.length, DhcpPacket.ENCAP_L2);
} catch (DhcpPacket.ParseException e) { } catch (DhcpPacket.ParseException e) {
// Not a DHCP packet. Continue. // Not a DHCP packet.
return false;
} }
return null; return true;
}); });
return packet == null ? null :
DhcpPacket.decodeFullPacket(packet, packet.length, DhcpPacket.ENCAP_L2);
} }
private <R> R getNextMatchedPacket(Function<byte[], R> match) { public void sendPacket(ByteBuffer packet) throws Exception {
byte[] packet; mDownstreamReader.sendResponse(packet);
R result; }
while ((packet = mDownstreamReader.popPacket(PACKET_READ_TIMEOUT_MS)) != null) {
result = match.apply(packet);
if (result != null) return result; public byte[] getNextMatchedPacket(Predicate<byte[]> filter) {
return mDownstreamReader.poll(PACKET_READ_TIMEOUT_MS, filter);
}
public static class RemoteResponder {
final TapPacketReader mUpstreamReader;
public RemoteResponder(TapPacketReader reader) {
mUpstreamReader = reader;
} }
return null; public void sendPacket(ByteBuffer packet) throws Exception {
mUpstreamReader.sendResponse(packet);
}
public byte[] getNextMatchedPacket(Predicate<byte[]> filter) throws Exception {
return mUpstreamReader.poll(PACKET_READ_TIMEOUT_MS, filter);
}
} }
} }