Merge "Merge "apf: Documentation improvements and code optimizations for... (62a5229d) · Commits · e / os / android_frameworks_base

packages/NetworkStack/src/android/net/apf/ApfFilter.java

+63 −27

Original line number	Original line	Diff line number	Diff line
	@@ -74,6 +74,7 @@ import java.net.SocketException;
	import java.net.UnknownHostException;		import java.net.UnknownHostException;
	import java.nio.BufferUnderflowException;		import java.nio.BufferUnderflowException;
	import java.nio.ByteBuffer;		import java.nio.ByteBuffer;
			import java.nio.ByteOrder;
	import java.util.ArrayList;		import java.util.ArrayList;
	import java.util.Arrays;		import java.util.Arrays;

	@@ -282,6 +283,7 @@ public class ApfFilter {
	private static final byte[] ETH_BROADCAST_MAC_ADDRESS =		private static final byte[] ETH_BROADCAST_MAC_ADDRESS =
	{(byte) 0xff, (byte) 0xff, (byte) 0xff, (byte) 0xff, (byte) 0xff, (byte) 0xff };		{(byte) 0xff, (byte) 0xff, (byte) 0xff, (byte) 0xff, (byte) 0xff, (byte) 0xff };
	// TODO: Make these offsets relative to end of link-layer header; don't include ETH_HEADER_LEN.		// TODO: Make these offsets relative to end of link-layer header; don't include ETH_HEADER_LEN.
			private static final int IPV4_TOTAL_LENGTH_OFFSET = ETH_HEADER_LEN + 2;
	private static final int IPV4_FRAGMENT_OFFSET_OFFSET = ETH_HEADER_LEN + 6;		private static final int IPV4_FRAGMENT_OFFSET_OFFSET = ETH_HEADER_LEN + 6;
	// Endianness is not an issue for this constant because the APF interpreter always operates in		// Endianness is not an issue for this constant because the APF interpreter always operates in
	// network byte order.		// network byte order.
	@@ -881,10 +883,23 @@ public class ApfFilter {

	protected final TcpKeepaliveAckData mPacket;		protected final TcpKeepaliveAckData mPacket;
	protected final byte[] mSrcDstAddr;		protected final byte[] mSrcDstAddr;
			protected final byte[] mPortSeqAckFingerprint;

	TcpKeepaliveAck(final TcpKeepaliveAckData packet, final byte[] srcDstAddr) {		TcpKeepaliveAck(final TcpKeepaliveAckData packet, final byte[] srcDstAddr) {
	mPacket = packet;		mPacket = packet;
	mSrcDstAddr = srcDstAddr;		mSrcDstAddr = srcDstAddr;
			mPortSeqAckFingerprint = generatePortSeqAckFingerprint(mPacket.srcPort,
			mPacket.dstPort, mPacket.seq, mPacket.ack);
			}

			static byte[] generatePortSeqAckFingerprint(int srcPort, int dstPort, int seq, int ack) {
			final ByteBuffer fp = ByteBuffer.allocate(12);
			fp.order(ByteOrder.BIG_ENDIAN);
			fp.putShort((short) srcPort);
			fp.putShort((short) dstPort);
			fp.putInt(seq);
			fp.putInt(ack);
			return fp.array();
	}		}

	static byte[] concatArrays(final byte[]... arr) {		static byte[] concatArrays(final byte[]... arr) {
	@@ -919,10 +934,6 @@ public class ApfFilter {

	private class TcpKeepaliveAckV4 extends TcpKeepaliveAck {		private class TcpKeepaliveAckV4 extends TcpKeepaliveAck {
	private static final int IPV4_SRC_ADDR_OFFSET = IP_HEADER_OFFSET + 12;		private static final int IPV4_SRC_ADDR_OFFSET = IP_HEADER_OFFSET + 12;
	private static final int IPV4_TCP_SRC_PORT_OFFSET = 0;
	private static final int IPV4_TCP_DST_PORT_OFFSET = 2;
	private static final int IPV4_TCP_SEQ_OFFSET = 4;
	private static final int IPV4_TCP_ACK_OFFSET = 8;

	TcpKeepaliveAckV4(final TcpKeepalivePacketDataParcelable sentKeepalivePacket) {		TcpKeepaliveAckV4(final TcpKeepalivePacketDataParcelable sentKeepalivePacket) {
	this(new TcpKeepaliveAckData(sentKeepalivePacket));		this(new TcpKeepaliveAckData(sentKeepalivePacket));
	@@ -934,12 +945,12 @@ public class ApfFilter {
	@Override		@Override
	void generateFilterLocked(ApfGenerator gen) throws IllegalInstructionException {		void generateFilterLocked(ApfGenerator gen) throws IllegalInstructionException {
	final String nextFilterLabel = "keepalive_ack" + getUniqueNumberLocked();		final String nextFilterLabel = "keepalive_ack" + getUniqueNumberLocked();
	gen.addLoad8(Register.R0, IPV4_PROTOCOL_OFFSET);
	gen.addJumpIfR0NotEquals(IPPROTO_TCP, nextFilterLabel);
	gen.addLoadImmediate(Register.R0, ETH_HEADER_LEN + IPV4_SRC_ADDR_OFFSET);		gen.addLoadImmediate(Register.R0, ETH_HEADER_LEN + IPV4_SRC_ADDR_OFFSET);
	gen.addJumpIfBytesNotEqual(Register.R0, mSrcDstAddr, nextFilterLabel);		gen.addJumpIfBytesNotEqual(Register.R0, mSrcDstAddr, nextFilterLabel);

	// Pass the packet if it's not zero-sized :		// Skip to the next filter if it's not zero-sized :
			// TCP_HEADER_SIZE + IPV4_HEADER_SIZE - ipv4_total_length == 0
	// Load the IP header size into R1		// Load the IP header size into R1
	gen.addLoadFromMemory(Register.R1, gen.IPV4_HEADER_SIZE_MEMORY_SLOT);		gen.addLoadFromMemory(Register.R1, gen.IPV4_HEADER_SIZE_MEMORY_SLOT);
	// Load the TCP header size into R0 (it's indexed by R1)		// Load the TCP header size into R0 (it's indexed by R1)
	@@ -947,27 +958,18 @@ public class ApfFilter {
	// Size offset is in the top nibble, but it must be multiplied by 4, and the two		// Size offset is in the top nibble, but it must be multiplied by 4, and the two
	// top bits of the low nibble are guaranteed to be zeroes. Right-shift R0 by 2.		// top bits of the low nibble are guaranteed to be zeroes. Right-shift R0 by 2.
	gen.addRightShift(2);		gen.addRightShift(2);
	// R0 += R1 -> R0 contains TCP + IP headers lenght		// R0 += R1 -> R0 contains TCP + IP headers length
	gen.addAddR1();
	// Add the Ethernet header length to R0.
	gen.addLoadImmediate(Register.R1, ETH_HEADER_LEN);
	gen.addAddR1();		gen.addAddR1();
	// Compare total length of headers to the size of the packet.		// Load IPv4 total length
	gen.addLoadFromMemory(Register.R1, gen.PACKET_SIZE_MEMORY_SLOT);		gen.addLoad16(Register.R1, IPV4_TOTAL_LENGTH_OFFSET);
	gen.addNeg(Register.R0);		gen.addNeg(Register.R0);
	gen.addAddR1();		gen.addAddR1();
	gen.addJumpIfR0NotEquals(0, nextFilterLabel);		gen.addJumpIfR0NotEquals(0, nextFilterLabel);

	// Add IPv4 header length		// Add IPv4 header length
	gen.addLoadFromMemory(Register.R1, gen.IPV4_HEADER_SIZE_MEMORY_SLOT);		gen.addLoadFromMemory(Register.R1, gen.IPV4_HEADER_SIZE_MEMORY_SLOT);
	gen.addLoad16Indexed(Register.R0, ETH_HEADER_LEN + IPV4_TCP_SRC_PORT_OFFSET);		gen.addLoadImmediate(Register.R0, ETH_HEADER_LEN);
	gen.addJumpIfR0NotEquals(mPacket.srcPort, nextFilterLabel);		gen.addAddR1();
	gen.addLoad16Indexed(Register.R0, ETH_HEADER_LEN + IPV4_TCP_DST_PORT_OFFSET);		gen.addJumpIfBytesNotEqual(Register.R0, mPortSeqAckFingerprint, nextFilterLabel);
	gen.addJumpIfR0NotEquals(mPacket.dstPort, nextFilterLabel);
	gen.addLoad32Indexed(Register.R0, ETH_HEADER_LEN + IPV4_TCP_SEQ_OFFSET);
	gen.addJumpIfR0NotEquals(mPacket.seq, nextFilterLabel);
	gen.addLoad32Indexed(Register.R0, ETH_HEADER_LEN + IPV4_TCP_ACK_OFFSET);
	gen.addJumpIfR0NotEquals(mPacket.ack, nextFilterLabel);

	maybeSetupCounter(gen, Counter.DROPPED_IPV4_KEEPALIVE_ACK);		maybeSetupCounter(gen, Counter.DROPPED_IPV4_KEEPALIVE_ACK);
	gen.addJump(mCountAndDropLabel);		gen.addJump(mCountAndDropLabel);
	@@ -1169,9 +1171,10 @@ public class ApfFilter {
	gen.addJumpIfR0Equals(broadcastAddr, mCountAndDropLabel);		gen.addJumpIfR0Equals(broadcastAddr, mCountAndDropLabel);
	}		}

	// If any keepalive filters,		// If any keepalive filter matches, drop
	generateKeepaliveFilter(gen);		generateV4KeepaliveFilters(gen);

			// Otherwise, this is an IPv4 unicast, pass
	// If L2 broadcast packet, drop.		// If L2 broadcast packet, drop.
	// TODO: can we invert this condition to fall through to the common pass case below?		// TODO: can we invert this condition to fall through to the common pass case below?
	maybeSetupCounter(gen, Counter.PASSED_IPV4_UNICAST);		maybeSetupCounter(gen, Counter.PASSED_IPV4_UNICAST);
	@@ -1180,7 +1183,7 @@ public class ApfFilter {
	maybeSetupCounter(gen, Counter.DROPPED_IPV4_L2_BROADCAST);		maybeSetupCounter(gen, Counter.DROPPED_IPV4_L2_BROADCAST);
	gen.addJump(mCountAndDropLabel);		gen.addJump(mCountAndDropLabel);
	} else {		} else {
	generateKeepaliveFilter(gen);		generateV4KeepaliveFilters(gen);
	}		}

	// Otherwise, pass		// Otherwise, pass
	@@ -1188,12 +1191,25 @@ public class ApfFilter {
	gen.addJump(mCountAndPassLabel);		gen.addJump(mCountAndPassLabel);
	}		}

	private void generateKeepaliveFilter(ApfGenerator gen) throws IllegalInstructionException {		private void generateV4KeepaliveFilters(ApfGenerator gen) throws IllegalInstructionException {
			final String skipV4KeepaliveFilter = "skip_v4_keepalive_filter";
			final boolean haveV4KeepaliveAcks = NetworkStackUtils.any(mKeepaliveAcks,
			ack -> ack instanceof TcpKeepaliveAckV4);

			// If no keepalive acks
			if (!haveV4KeepaliveAcks) return;

			// If not tcp, skip keepalive filters
			gen.addLoad8(Register.R0, IPV4_PROTOCOL_OFFSET);
			gen.addJumpIfR0NotEquals(IPPROTO_TCP, skipV4KeepaliveFilter);

	// Drop IPv4 Keepalive acks		// Drop IPv4 Keepalive acks
	for (int i = 0; i < mKeepaliveAcks.size(); ++i) {		for (int i = 0; i < mKeepaliveAcks.size(); ++i) {
	final TcpKeepaliveAck ack = mKeepaliveAcks.valueAt(i);		final TcpKeepaliveAck ack = mKeepaliveAcks.valueAt(i);
	if (ack instanceof TcpKeepaliveAckV4) ack.generateFilterLocked(gen);		if (ack instanceof TcpKeepaliveAckV4) ack.generateFilterLocked(gen);
	}		}

			gen.defineLabel(skipV4KeepaliveFilter);
	}		}

	/**		/**
	@@ -1244,11 +1260,14 @@ public class ApfFilter {
	maybeSetupCounter(gen, Counter.DROPPED_IPV6_NON_ICMP_MULTICAST);		maybeSetupCounter(gen, Counter.DROPPED_IPV6_NON_ICMP_MULTICAST);
	gen.addLoad8(Register.R0, IPV6_DEST_ADDR_OFFSET);		gen.addLoad8(Register.R0, IPV6_DEST_ADDR_OFFSET);
	gen.addJumpIfR0Equals(0xff, mCountAndDropLabel);		gen.addJumpIfR0Equals(0xff, mCountAndDropLabel);
			// If any keepalive filter matches, drop
			generateV6KeepaliveFilters(gen);
	// Not multicast. Pass.		// Not multicast. Pass.
	maybeSetupCounter(gen, Counter.PASSED_IPV6_UNICAST_NON_ICMP);		maybeSetupCounter(gen, Counter.PASSED_IPV6_UNICAST_NON_ICMP);
	gen.addJump(mCountAndPassLabel);		gen.addJump(mCountAndPassLabel);
	gen.defineLabel(skipIPv6MulticastFilterLabel);		gen.defineLabel(skipIPv6MulticastFilterLabel);
	} else {		} else {
			generateV6KeepaliveFilters(gen);
	// If not ICMPv6, pass.		// If not ICMPv6, pass.
	maybeSetupCounter(gen, Counter.PASSED_IPV6_NON_ICMP);		maybeSetupCounter(gen, Counter.PASSED_IPV6_NON_ICMP);
	gen.addJumpIfR0NotEquals(IPPROTO_ICMPV6, mCountAndPassLabel);		gen.addJumpIfR0NotEquals(IPPROTO_ICMPV6, mCountAndPassLabel);
	@@ -1272,12 +1291,27 @@ public class ApfFilter {
	maybeSetupCounter(gen, Counter.DROPPED_IPV6_MULTICAST_NA);		maybeSetupCounter(gen, Counter.DROPPED_IPV6_MULTICAST_NA);
	gen.addJump(mCountAndDropLabel);		gen.addJump(mCountAndDropLabel);
	gen.defineLabel(skipUnsolicitedMulticastNALabel);		gen.defineLabel(skipUnsolicitedMulticastNALabel);
			}

			private void generateV6KeepaliveFilters(ApfGenerator gen) throws IllegalInstructionException {
			final String skipV6KeepaliveFilter = "skip_v6_keepalive_filter";
			final boolean haveV6KeepaliveAcks = NetworkStackUtils.any(mKeepaliveAcks,
			ack -> ack instanceof TcpKeepaliveAckV6);

			// If no keepalive acks
			if (!haveV6KeepaliveAcks) return;

			// If not tcp, skip keepalive filters
			gen.addLoad8(Register.R0, IPV6_NEXT_HEADER_OFFSET);
			gen.addJumpIfR0NotEquals(IPPROTO_TCP, skipV6KeepaliveFilter);

	// Drop IPv6 Keepalive acks		// Drop IPv6 Keepalive acks
	for (int i = 0; i < mKeepaliveAcks.size(); ++i) {		for (int i = 0; i < mKeepaliveAcks.size(); ++i) {
	final TcpKeepaliveAck ack = mKeepaliveAcks.valueAt(i);		final TcpKeepaliveAck ack = mKeepaliveAcks.valueAt(i);
	if (ack instanceof TcpKeepaliveAckV6) ack.generateFilterLocked(gen);		if (ack instanceof TcpKeepaliveAckV6) ack.generateFilterLocked(gen);
	}		}

			gen.defineLabel(skipV6KeepaliveFilter);
	}		}

	/**		/**
	@@ -1294,6 +1328,8 @@ public class ApfFilter {
	* <li>Pass all non-IPv4 and non-IPv6 packets,		* <li>Pass all non-IPv4 and non-IPv6 packets,
	* <li>Drop IPv6 ICMPv6 NAs to ff02::1.		* <li>Drop IPv6 ICMPv6 NAs to ff02::1.
	* <li>Drop IPv6 ICMPv6 RSs.		* <li>Drop IPv6 ICMPv6 RSs.
			* <li>Filter IPv4 packets (see generateIPv4FilterLocked())
			* <li>Filter IPv6 packets (see generateIPv6FilterLocked())
	* <li>Let execution continue off the end of the program for IPv6 ICMPv6 packets. This allows		* <li>Let execution continue off the end of the program for IPv6 ICMPv6 packets. This allows
	* insertion of RA filters here, or if there aren't any, just passes the packets.		* insertion of RA filters here, or if there aren't any, just passes the packets.
	* </ul>		* </ul>
	@@ -1737,7 +1773,7 @@ public class ApfFilter {
	}		}
	pw.decreaseIndent();		pw.decreaseIndent();

	pw.println("Keepalive filter:");		pw.println("Keepalive filters:");
	pw.increaseIndent();		pw.increaseIndent();
	for (int i = 0; i < mKeepaliveAcks.size(); ++i) {		for (int i = 0; i < mKeepaliveAcks.size(); ++i) {
	final TcpKeepaliveAck keepaliveAck = mKeepaliveAcks.valueAt(i);		final TcpKeepaliveAck keepaliveAck = mKeepaliveAcks.valueAt(i);

packages/NetworkStack/src/android/net/apf/ApfGenerator.java

+5 −5

packages/NetworkStack/src/android/net/util/NetworkStackUtils.java

+16 −0

Original line number	Original line	Diff line number	Diff line
	@@ -17,10 +17,13 @@
	package android.net.util;		package android.net.util;

	import android.annotation.NonNull;		import android.annotation.NonNull;
			import android.util.SparseArray;

	import java.io.FileDescriptor;		import java.io.FileDescriptor;
	import java.io.IOException;		import java.io.IOException;
	import java.util.List;		import java.util.List;
			import java.util.function.Predicate;


	/**		/**
	* Collection of utilities for the network stack.		* Collection of utilities for the network stack.
	@@ -65,4 +68,17 @@ public class NetworkStackUtils {
	}		}
	return array;		return array;
	}		}

			/**
			* @return True if there exists at least one element in the sparse array for which
			* condition {@code predicate}
			*/
			public static <T> boolean any(SparseArray<T> array, Predicate<T> predicate) {
			for (int i = 0; i < array.size(); ++i) {
			if (predicate.test(array.valueAt(i))) {
			return true;
			}
			}
			return false;
			}
	}		}

packages/NetworkStack/tests/src/android/net/apf/ApfTest.java

+1 −1

File changed.

Contains only whitespace changes.

Original line number	Original line	Diff line number	Diff line
	@@ -108,7 +108,7 @@ public class ApfGenerator {
	private String mLabel;		private String mLabel;
	// When mOpcode == Opcodes.JNEBS:		// When mOpcode == Opcodes.JNEBS:
	private byte[] mCompareBytes;		private byte[] mCompareBytes;
	// Offset in bytes from the begining of this program. Set by {@link ApfGenerator#generate}.		// Offset in bytes from the beginning of this program. Set by {@link ApfGenerator#generate}.
	int offset;		int offset;

	Instruction(Opcodes opcode, Register register) {		Instruction(Opcodes opcode, Register register) {
	@@ -431,7 +431,7 @@ public class ApfGenerator {

	/**		/**
	* Add an instruction to the end of the program to load the byte at offset {@code offset}		* Add an instruction to the end of the program to load the byte at offset {@code offset}
	* bytes from the begining of the packet into {@code register}.		* bytes from the beginning of the packet into {@code register}.
	*/		*/
	public ApfGenerator addLoad8(Register register, int offset) {		public ApfGenerator addLoad8(Register register, int offset) {
	Instruction instruction = new Instruction(Opcodes.LDB, register);		Instruction instruction = new Instruction(Opcodes.LDB, register);
	@@ -442,7 +442,7 @@ public class ApfGenerator {

	/**		/**
	* Add an instruction to the end of the program to load 16-bits at offset {@code offset}		* Add an instruction to the end of the program to load 16-bits at offset {@code offset}
	* bytes from the begining of the packet into {@code register}.		* bytes from the beginning of the packet into {@code register}.
	*/		*/
	public ApfGenerator addLoad16(Register register, int offset) {		public ApfGenerator addLoad16(Register register, int offset) {
	Instruction instruction = new Instruction(Opcodes.LDH, register);		Instruction instruction = new Instruction(Opcodes.LDH, register);
	@@ -453,7 +453,7 @@ public class ApfGenerator {

	/**		/**
	* Add an instruction to the end of the program to load 32-bits at offset {@code offset}		* Add an instruction to the end of the program to load 32-bits at offset {@code offset}
	* bytes from the begining of the packet into {@code register}.		* bytes from the beginning of the packet into {@code register}.
	*/		*/
	public ApfGenerator addLoad32(Register register, int offset) {		public ApfGenerator addLoad32(Register register, int offset) {
	Instruction instruction = new Instruction(Opcodes.LDW, register);		Instruction instruction = new Instruction(Opcodes.LDW, register);
	@@ -464,7 +464,7 @@ public class ApfGenerator {

	/**		/**
	* Add an instruction to the end of the program to load a byte from the packet into		* Add an instruction to the end of the program to load a byte from the packet into
	* {@code register}. The offset of the loaded byte from the begining of the packet is		* {@code register}. The offset of the loaded byte from the beginning of the packet is
	* the sum of {@code offset} and the value in register R1.		* the sum of {@code offset} and the value in register R1.
	*/		*/
	public ApfGenerator addLoad8Indexed(Register register, int offset) {		public ApfGenerator addLoad8Indexed(Register register, int offset) {