Merge "Support specific client address configuration" into rvc-dev am: ad6540d6ac (d6ca1051) · Commits · e / os / android_packages_modules_NetworkStack

common/networkstackclient/aidl_api/networkstack-aidl-interfaces/current/android/net/dhcp/DhcpServingParamsParcel.aidl

+1 −0

Original line number	Diff line number	Diff line
		@@ -25,4 +25,5 @@ parcelable DhcpServingParamsParcel {
		long dhcpLeaseTimeSecs;
		int linkMtu;
		boolean metered;
		int clientAddr;
		}

common/networkstackclient/src/android/net/dhcp/DhcpServingParamsParcel.aidl

+1 −0

Original line number	Diff line number	Diff line
		@@ -26,5 +26,6 @@ parcelable DhcpServingParamsParcel {
		long dhcpLeaseTimeSecs;
		int linkMtu;
		boolean metered;
		int clientAddr;
		}

src/android/net/dhcp/DhcpLeaseRepository.java

+12 −4

Original line number	Diff line number	Diff line
		@@ -80,6 +80,8 @@ class DhcpLeaseRepository {
		private int mSubnetMask;
		private int mNumAddresses;
		private long mLeaseTimeMs;
		@Nullable
		private Inet4Address mClientAddr;

		/**
		* Next timestamp when committed or declined leases should be checked for expired ones. This
		@@ -128,21 +130,24 @@ class DhcpLeaseRepository {
		private final LinkedHashMap<Inet4Address, Long> mDeclinedAddrs = new LinkedHashMap<>();

		DhcpLeaseRepository(@NonNull IpPrefix prefix, @NonNull Set<Inet4Address> reservedAddrs,
		long leaseTimeMs, @NonNull SharedLog log, @NonNull Clock clock) {
		updateParams(prefix, reservedAddrs, leaseTimeMs);
		long leaseTimeMs, @Nullable Inet4Address clientAddr, @NonNull SharedLog log,
		@NonNull Clock clock) {
		mLog = log;
		mClock = clock;
		mClientAddr = clientAddr;
		updateParams(prefix, reservedAddrs, leaseTimeMs, clientAddr);
		}

		public void updateParams(@NonNull IpPrefix prefix, @NonNull Set<Inet4Address> reservedAddrs,
		long leaseTimeMs) {
		long leaseTimeMs, @Nullable Inet4Address clientAddr) {
		mPrefix = prefix;
		mReservedAddrs = Collections.unmodifiableSet(new HashSet<>(reservedAddrs));
		mPrefixLength = prefix.getPrefixLength();
		mSubnetMask = prefixLengthToV4NetmaskIntHTH(mPrefixLength);
		mSubnetAddr = inet4AddressToIntHTH((Inet4Address) prefix.getAddress()) & mSubnetMask;
		mNumAddresses = 1 << (IPV4_ADDR_BITS - prefix.getPrefixLength());
		mNumAddresses = clientAddr != null ? 1 : 1 << (IPV4_ADDR_BITS - prefix.getPrefixLength());
		mLeaseTimeMs = leaseTimeMs;
		mClientAddr = clientAddr;

		cleanMap(mDeclinedAddrs);
		if (cleanMap(mCommittedLeases)) {
		@@ -514,6 +519,9 @@ class DhcpLeaseRepository {
		* address (with the ordering in {@link #getAddrIndex(int)}) is returned.
		*/
		private int getValidAddress(int addr) {
		// Only mClientAddr is valid if static client address is enforced.
		if (mClientAddr != null) return inet4AddressToIntHTH(mClientAddr);

		final int lastByteMask = 0xff;
		int addrIndex = getAddrIndex(addr); // 0-based index of the address in the subnet

src/android/net/dhcp/DhcpServer.java

+4 −3

Original line number	Diff line number	Diff line
		@@ -205,8 +205,8 @@ public class DhcpServer extends IDhcpServer.Stub {
		@NonNull SharedLog log, @NonNull Clock clock) {
		return new DhcpLeaseRepository(
		DhcpServingParams.makeIpPrefix(servingParams.serverAddr),
		servingParams.excludedAddrs,
		servingParams.dhcpLeaseTimeSecs * 1000, log.forSubComponent(REPO_TAG), clock);
		servingParams.excludedAddrs, servingParams.dhcpLeaseTimeSecs * 1000,
		servingParams.clientAddr, log.forSubComponent(REPO_TAG), clock);
		}

		@Override
		@@ -351,7 +351,8 @@ public class DhcpServer extends IDhcpServer.Stub {
		mLeaseRepo.updateParams(
		DhcpServingParams.makeIpPrefix(mServingParams.serverAddr),
		params.excludedAddrs,
		params.dhcpLeaseTimeSecs);
		params.dhcpLeaseTimeSecs,
		params.clientAddr);

		cb = pair.second;
		break;

src/android/net/dhcp/DhcpServingParams.java

+26 −2

Original line number	Diff line number	Diff line
		@@ -84,6 +84,12 @@ public class DhcpServingParams {
		*/
		public final boolean metered;

		/**
		* Client inet address. This will be the only address offered by DhcpServer if set.
		*/
		@Nullable
		public final Inet4Address clientAddr;

		/**
		* Checked exception thrown when some parameters used to build {@link DhcpServingParams} are
		* missing or invalid.
		@@ -97,7 +103,7 @@ public class DhcpServingParams {
		private DhcpServingParams(@NonNull LinkAddress serverAddr,
		@NonNull Set<Inet4Address> defaultRouters,
		@NonNull Set<Inet4Address> dnsServers, @NonNull Set<Inet4Address> excludedAddrs,
		long dhcpLeaseTimeSecs, int linkMtu, boolean metered) {
		long dhcpLeaseTimeSecs, int linkMtu, boolean metered, Inet4Address clientAddr) {
		this.serverAddr = serverAddr;
		this.defaultRouters = defaultRouters;
		this.dnsServers = dnsServers;
		@@ -105,6 +111,7 @@ public class DhcpServingParams {
		this.dhcpLeaseTimeSecs = dhcpLeaseTimeSecs;
		this.linkMtu = linkMtu;
		this.metered = metered;
		this.clientAddr = clientAddr;
		}

		/**
		@@ -119,6 +126,11 @@ public class DhcpServingParams {
		final LinkAddress serverAddr = new LinkAddress(
		intToInet4AddressHTH(parcel.serverAddr),
		parcel.serverAddrPrefixLength);
		Inet4Address clientAddr = null;
		if (parcel.clientAddr != 0) {
		clientAddr = intToInet4AddressHTH(parcel.clientAddr);
		}

		return new Builder()
		.setServerAddr(serverAddr)
		.setDefaultRouters(toInet4AddressSet(parcel.defaultRouters))
		@@ -127,6 +139,7 @@ public class DhcpServingParams {
		.setDhcpLeaseTimeSecs(parcel.dhcpLeaseTimeSecs)
		.setLinkMtu(parcel.linkMtu)
		.setMetered(parcel.metered)
		.setClientAddr(clientAddr)
		.build();
		}

		@@ -181,6 +194,7 @@ public class DhcpServingParams {
		private long mDhcpLeaseTimeSecs;
		private int mLinkMtu = MTU_UNSET;
		private boolean mMetered;
		private Inet4Address mClientAddr;

		/**
		* Set the server address and served prefix for the DHCP server.
		@@ -304,6 +318,16 @@ public class DhcpServingParams {
		return this;
		}

		/**
		* Set the client address.
		*
		* <p>If not set, the default value is null.
		*/
		public Builder setClientAddr(@Nullable Inet4Address clientAddr) {
		this.mClientAddr = clientAddr;
		return this;
		}

		/**
		* Create a new {@link DhcpServingParams} instance based on parameters set in the builder.
		*
		@@ -358,7 +382,7 @@ public class DhcpServingParams {
		Collections.unmodifiableSet(new HashSet<>(mDefaultRouters)),
		Collections.unmodifiableSet(new HashSet<>(mDnsServers)),
		Collections.unmodifiableSet(excl),
		mDhcpLeaseTimeSecs, mLinkMtu, mMetered);
		mDhcpLeaseTimeSecs, mLinkMtu, mMetered, mClientAddr);
		}
		}