diff options
| -rw-r--r-- | services/core/java/com/android/server/connectivity/Vpn.java | 13 | ||||
| -rw-r--r-- | tests/net/java/com/android/server/ConnectivityServiceTest.java | 18 |
2 files changed, 15 insertions, 16 deletions
diff --git a/services/core/java/com/android/server/connectivity/Vpn.java b/services/core/java/com/android/server/connectivity/Vpn.java index a4df43c71eb5..0360609d9833 100644 --- a/services/core/java/com/android/server/connectivity/Vpn.java +++ b/services/core/java/com/android/server/connectivity/Vpn.java @@ -17,7 +17,6 @@ package com.android.server.connectivity; import static android.Manifest.permission.BIND_VPN_SERVICE; -import static android.net.ConnectivityManager.NETID_UNSET; import static android.net.NetworkCapabilities.NET_CAPABILITY_NOT_METERED; import static android.net.RouteInfo.RTN_THROW; import static android.net.RouteInfo.RTN_UNREACHABLE; @@ -1129,17 +1128,17 @@ public class Vpn { } /** - * Return netId of current running VPN network. + * Return Network of current running VPN network. * - * @return a netId if there is a running VPN network or NETID_UNSET if there is no running VPN + * @return a Network if there is a running VPN network or null if there is no running VPN * network or network is null. */ - public synchronized int getNetId() { + public synchronized Network getNetwork() { final NetworkAgent agent = mNetworkAgent; - if (null == agent) return NETID_UNSET; + if (null == agent) return null; final Network network = agent.getNetwork(); - if (null == network) return NETID_UNSET; - return network.getNetId(); + if (null == network) return null; + return network; } private LinkProperties makeLinkProperties() { diff --git a/tests/net/java/com/android/server/ConnectivityServiceTest.java b/tests/net/java/com/android/server/ConnectivityServiceTest.java index 0bbc74cd4dab..8cb33479832e 100644 --- a/tests/net/java/com/android/server/ConnectivityServiceTest.java +++ b/tests/net/java/com/android/server/ConnectivityServiceTest.java @@ -1178,11 +1178,6 @@ public class ConnectivityServiceTest { } @Override - public int getNetId() { - return (mMockNetworkAgent == null) ? NETID_UNSET : mMockNetworkAgent.getNetwork().netId; - } - - @Override public int getActiveVpnType() { return mVpnType; } @@ -1206,10 +1201,12 @@ public class ConnectivityServiceTest { mNetworkCapabilities); mMockNetworkAgent.waitForIdle(TIMEOUT_MS); - verify(mMockNetd, times(1)).networkAddUidRanges(eq(mMockVpn.getNetId()), + final int expectedNetId = mMockVpn.getNetwork() == null ? NETID_UNSET + : mMockVpn.getNetwork().getNetId(); + verify(mMockNetd, times(1)).networkAddUidRanges(eq(expectedNetId), eq(toUidRangeStableParcels(uids))); verify(mMockNetd, never()) - .networkRemoveUidRanges(eq(mMockVpn.getNetId()), any()); + .networkRemoveUidRanges(eq(expectedNetId), any()); mAgentRegistered = true; updateState(NetworkInfo.DetailedState.CONNECTED, "registerAgent"); mNetworkCapabilities.set(mMockNetworkAgent.getNetworkCapabilities()); @@ -9744,11 +9741,14 @@ public class ConnectivityServiceTest { exemptUidCaptor.capture()); assertContainsExactly(exemptUidCaptor.getValue(), Process.VPN_UID, exemptUid); + final int expectedNetId = mMockVpn.getNetwork() == null ? NETID_UNSET + : mMockVpn.getNetwork().getNetId(); + if (add) { - inOrder.verify(mMockNetd, times(1)).networkAddUidRanges(eq(mMockVpn.getNetId()), + inOrder.verify(mMockNetd, times(1)).networkAddUidRanges(eq(expectedNetId), eq(toUidRangeStableParcels(vpnRanges))); } else { - inOrder.verify(mMockNetd, times(1)).networkRemoveUidRanges(eq(mMockVpn.getNetId()), + inOrder.verify(mMockNetd, times(1)).networkRemoveUidRanges(eq(expectedNetId), eq(toUidRangeStableParcels(vpnRanges))); } |