Merge changes from topic "ipsec-sync-with-pi"
am: ff904be7a9
Change-Id: Ic91ed8f496163c560c6f1fd306bc74e5fecb61c2
This commit is contained in:
@@ -30,6 +30,7 @@ import static org.mockito.Mockito.mock;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.test.mock.MockContext;
|
||||
import android.support.test.filters.SmallTest;
|
||||
import android.support.test.runner.AndroidJUnit4;
|
||||
import android.system.Os;
|
||||
@@ -70,11 +71,17 @@ public class IpSecManagerTest {
|
||||
|
||||
private IpSecService mMockIpSecService;
|
||||
private IpSecManager mIpSecManager;
|
||||
private MockContext mMockContext = new MockContext() {
|
||||
@Override
|
||||
public String getOpPackageName() {
|
||||
return "fooPackage";
|
||||
}
|
||||
};
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
mMockIpSecService = mock(IpSecService.class);
|
||||
mIpSecManager = new IpSecManager(mMockIpSecService);
|
||||
mIpSecManager = new IpSecManager(mMockContext, mMockIpSecService);
|
||||
}
|
||||
|
||||
/*
|
||||
@@ -255,7 +262,7 @@ public class IpSecManagerTest {
|
||||
new IpSecTunnelInterfaceResponse(IpSecManager.Status.OK, resourceId, intfName);
|
||||
when(mMockIpSecService.createTunnelInterface(
|
||||
eq(VTI_LOCAL_ADDRESS.getHostAddress()), eq(GOOGLE_DNS_4.getHostAddress()),
|
||||
anyObject(), anyObject()))
|
||||
anyObject(), anyObject(), anyString()))
|
||||
.thenReturn(dummyResponse);
|
||||
|
||||
IpSecManager.IpSecTunnelInterface tunnelIntf = mIpSecManager.createIpSecTunnelInterface(
|
||||
@@ -273,7 +280,7 @@ public class IpSecManagerTest {
|
||||
assertEquals(VTI_INTF_NAME, tunnelIntf.getInterfaceName());
|
||||
|
||||
tunnelIntf.close();
|
||||
verify(mMockIpSecService).deleteTunnelInterface(eq(DUMMY_RESOURCE_ID));
|
||||
verify(mMockIpSecService).deleteTunnelInterface(eq(DUMMY_RESOURCE_ID), anyString());
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -281,12 +288,16 @@ public class IpSecManagerTest {
|
||||
IpSecManager.IpSecTunnelInterface tunnelIntf =
|
||||
createAndValidateVti(DUMMY_RESOURCE_ID, VTI_INTF_NAME);
|
||||
|
||||
tunnelIntf.addAddress(VTI_INNER_ADDRESS);
|
||||
tunnelIntf.addAddress(VTI_INNER_ADDRESS.getAddress(),
|
||||
VTI_INNER_ADDRESS.getPrefixLength());
|
||||
verify(mMockIpSecService)
|
||||
.addAddressToTunnelInterface(eq(DUMMY_RESOURCE_ID), eq(VTI_INNER_ADDRESS));
|
||||
.addAddressToTunnelInterface(
|
||||
eq(DUMMY_RESOURCE_ID), eq(VTI_INNER_ADDRESS), anyString());
|
||||
|
||||
tunnelIntf.removeAddress(VTI_INNER_ADDRESS);
|
||||
tunnelIntf.removeAddress(VTI_INNER_ADDRESS.getAddress(),
|
||||
VTI_INNER_ADDRESS.getPrefixLength());
|
||||
verify(mMockIpSecService)
|
||||
.addAddressToTunnelInterface(eq(DUMMY_RESOURCE_ID), eq(VTI_INNER_ADDRESS));
|
||||
.addAddressToTunnelInterface(
|
||||
eq(DUMMY_RESOURCE_ID), eq(VTI_INNER_ADDRESS), anyString());
|
||||
}
|
||||
}
|
||||
@@ -27,6 +27,7 @@ import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import android.app.AppOpsManager;
|
||||
import android.content.Context;
|
||||
import android.net.INetd;
|
||||
import android.net.IpSecAlgorithm;
|
||||
@@ -40,6 +41,7 @@ import android.net.Network;
|
||||
import android.net.NetworkUtils;
|
||||
import android.os.Binder;
|
||||
import android.os.ParcelFileDescriptor;
|
||||
import android.test.mock.MockContext;
|
||||
import android.support.test.filters.SmallTest;
|
||||
import android.system.Os;
|
||||
|
||||
@@ -48,6 +50,7 @@ import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Ignore;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.junit.runners.Parameterized;
|
||||
@@ -92,7 +95,28 @@ public class IpSecServiceParameterizedTest {
|
||||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7F
|
||||
};
|
||||
|
||||
Context mMockContext;
|
||||
AppOpsManager mMockAppOps = mock(AppOpsManager.class);
|
||||
|
||||
MockContext mMockContext = new MockContext() {
|
||||
@Override
|
||||
public Object getSystemService(String name) {
|
||||
switch(name) {
|
||||
case Context.APP_OPS_SERVICE:
|
||||
return mMockAppOps;
|
||||
default:
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void enforceCallingOrSelfPermission(String permission, String message) {
|
||||
if (permission == android.Manifest.permission.MANAGE_IPSEC_TUNNELS) {
|
||||
return;
|
||||
}
|
||||
throw new SecurityException("Unavailable permission requested");
|
||||
}
|
||||
};
|
||||
|
||||
INetd mMockNetd;
|
||||
IpSecService.IpSecServiceConfiguration mMockIpSecSrvConfig;
|
||||
IpSecService mIpSecService;
|
||||
@@ -114,13 +138,22 @@ public class IpSecServiceParameterizedTest {
|
||||
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
mMockContext = mock(Context.class);
|
||||
mMockNetd = mock(INetd.class);
|
||||
mMockIpSecSrvConfig = mock(IpSecService.IpSecServiceConfiguration.class);
|
||||
mIpSecService = new IpSecService(mMockContext, mMockIpSecSrvConfig);
|
||||
|
||||
// Injecting mock netd
|
||||
when(mMockIpSecSrvConfig.getNetdInstance()).thenReturn(mMockNetd);
|
||||
// A package granted the AppOp for MANAGE_IPSEC_TUNNELS will be MODE_ALLOWED.
|
||||
when(mMockAppOps.noteOp(anyInt(), anyInt(), eq("blessedPackage")))
|
||||
.thenReturn(AppOpsManager.MODE_ALLOWED);
|
||||
// A system package will not be granted the app op, so this should fall back to
|
||||
// a permissions check, which should pass.
|
||||
when(mMockAppOps.noteOp(anyInt(), anyInt(), eq("systemPackage")))
|
||||
.thenReturn(AppOpsManager.MODE_DEFAULT);
|
||||
// A mismatch between the package name and the UID will return MODE_IGNORED.
|
||||
when(mMockAppOps.noteOp(anyInt(), anyInt(), eq("badPackage")))
|
||||
.thenReturn(AppOpsManager.MODE_IGNORED);
|
||||
}
|
||||
|
||||
@Test
|
||||
@@ -232,7 +265,7 @@ public class IpSecServiceParameterizedTest {
|
||||
addAuthAndCryptToIpSecConfig(ipSecConfig);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
assertEquals(IpSecManager.Status.OK, createTransformResp.status);
|
||||
|
||||
verify(mMockNetd)
|
||||
@@ -267,7 +300,7 @@ public class IpSecServiceParameterizedTest {
|
||||
ipSecConfig.setAuthenticatedEncryption(AEAD_ALGO);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
assertEquals(IpSecManager.Status.OK, createTransformResp.status);
|
||||
|
||||
verify(mMockNetd)
|
||||
@@ -301,12 +334,12 @@ public class IpSecServiceParameterizedTest {
|
||||
addAuthAndCryptToIpSecConfig(ipSecConfig);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
assertEquals(IpSecManager.Status.OK, createTransformResp.status);
|
||||
|
||||
// Attempting to create transform a second time with the same SPIs should throw an error...
|
||||
try {
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
fail("IpSecService should have thrown an error for reuse of SPI");
|
||||
} catch (IllegalStateException expected) {
|
||||
}
|
||||
@@ -314,7 +347,7 @@ public class IpSecServiceParameterizedTest {
|
||||
// ... even if the transform is deleted
|
||||
mIpSecService.deleteTransform(createTransformResp.resourceId);
|
||||
try {
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
fail("IpSecService should have thrown an error for reuse of SPI");
|
||||
} catch (IllegalStateException expected) {
|
||||
}
|
||||
@@ -327,7 +360,7 @@ public class IpSecServiceParameterizedTest {
|
||||
addAuthAndCryptToIpSecConfig(ipSecConfig);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
IpSecService.UserRecord userRecord =
|
||||
mIpSecService.mUserResourceTracker.getUserRecord(Os.getuid());
|
||||
assertEquals(1, userRecord.mSpiQuotaTracker.mCurrent);
|
||||
@@ -351,7 +384,7 @@ public class IpSecServiceParameterizedTest {
|
||||
addAuthAndCryptToIpSecConfig(ipSecConfig);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
mIpSecService.deleteTransform(createTransformResp.resourceId);
|
||||
|
||||
verify(mMockNetd, times(1))
|
||||
@@ -398,7 +431,7 @@ public class IpSecServiceParameterizedTest {
|
||||
addAuthAndCryptToIpSecConfig(ipSecConfig);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
|
||||
IpSecService.UserRecord userRecord =
|
||||
mIpSecService.mUserResourceTracker.getUserRecord(Os.getuid());
|
||||
@@ -435,7 +468,7 @@ public class IpSecServiceParameterizedTest {
|
||||
addAuthAndCryptToIpSecConfig(ipSecConfig);
|
||||
|
||||
IpSecTransformResponse createTransformResp =
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder());
|
||||
mIpSecService.createTransform(ipSecConfig, new Binder(), "blessedPackage");
|
||||
ParcelFileDescriptor pfd = ParcelFileDescriptor.fromSocket(new Socket());
|
||||
|
||||
int resourceId = createTransformResp.resourceId;
|
||||
@@ -460,10 +493,10 @@ public class IpSecServiceParameterizedTest {
|
||||
}
|
||||
|
||||
private IpSecTunnelInterfaceResponse createAndValidateTunnel(
|
||||
String localAddr, String remoteAddr) {
|
||||
String localAddr, String remoteAddr, String pkgName) {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
mIpSecService.createTunnelInterface(
|
||||
mSourceAddr, mDestinationAddr, fakeNetwork, new Binder());
|
||||
mSourceAddr, mDestinationAddr, fakeNetwork, new Binder(), pkgName);
|
||||
|
||||
assertNotNull(createTunnelResp);
|
||||
assertEquals(IpSecManager.Status.OK, createTunnelResp.status);
|
||||
@@ -473,7 +506,7 @@ public class IpSecServiceParameterizedTest {
|
||||
@Test
|
||||
public void testCreateTunnelInterface() throws Exception {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr);
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr, "blessedPackage");
|
||||
|
||||
// Check that we have stored the tracking object, and retrieve it
|
||||
IpSecService.UserRecord userRecord =
|
||||
@@ -495,12 +528,12 @@ public class IpSecServiceParameterizedTest {
|
||||
@Test
|
||||
public void testDeleteTunnelInterface() throws Exception {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr);
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr, "blessedPackage");
|
||||
|
||||
IpSecService.UserRecord userRecord =
|
||||
mIpSecService.mUserResourceTracker.getUserRecord(Os.getuid());
|
||||
|
||||
mIpSecService.deleteTunnelInterface(createTunnelResp.resourceId);
|
||||
mIpSecService.deleteTunnelInterface(createTunnelResp.resourceId, "blessedPackage");
|
||||
|
||||
// Verify quota and RefcountedResource objects cleaned up
|
||||
assertEquals(0, userRecord.mTunnelQuotaTracker.mCurrent);
|
||||
@@ -516,7 +549,7 @@ public class IpSecServiceParameterizedTest {
|
||||
@Test
|
||||
public void testTunnelInterfaceBinderDeath() throws Exception {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr);
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr, "blessedPackage");
|
||||
|
||||
IpSecService.UserRecord userRecord =
|
||||
mIpSecService.mUserResourceTracker.getUserRecord(Os.getuid());
|
||||
@@ -539,22 +572,35 @@ public class IpSecServiceParameterizedTest {
|
||||
|
||||
@Test
|
||||
public void testAddRemoveAddressFromTunnelInterface() throws Exception {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr);
|
||||
for (String pkgName : new String[]{"blessedPackage", "systemPackage"}) {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr, pkgName);
|
||||
mIpSecService.addAddressToTunnelInterface(
|
||||
createTunnelResp.resourceId, mLocalInnerAddress, pkgName);
|
||||
verify(mMockNetd, times(1))
|
||||
.interfaceAddAddress(
|
||||
eq(createTunnelResp.interfaceName),
|
||||
eq(mLocalInnerAddress.getAddress().getHostAddress()),
|
||||
eq(mLocalInnerAddress.getPrefixLength()));
|
||||
mIpSecService.removeAddressFromTunnelInterface(
|
||||
createTunnelResp.resourceId, mLocalInnerAddress, pkgName);
|
||||
verify(mMockNetd, times(1))
|
||||
.interfaceDelAddress(
|
||||
eq(createTunnelResp.interfaceName),
|
||||
eq(mLocalInnerAddress.getAddress().getHostAddress()),
|
||||
eq(mLocalInnerAddress.getPrefixLength()));
|
||||
mIpSecService.deleteTunnelInterface(createTunnelResp.resourceId, pkgName);
|
||||
}
|
||||
}
|
||||
|
||||
mIpSecService.addAddressToTunnelInterface(createTunnelResp.resourceId, mLocalInnerAddress);
|
||||
verify(mMockNetd)
|
||||
.interfaceAddAddress(
|
||||
eq(createTunnelResp.interfaceName),
|
||||
eq(mLocalInnerAddress.getAddress().getHostAddress()),
|
||||
eq(mLocalInnerAddress.getPrefixLength()));
|
||||
|
||||
mIpSecService.removeAddressFromTunnelInterface(
|
||||
createTunnelResp.resourceId, mLocalInnerAddress);
|
||||
verify(mMockNetd)
|
||||
.interfaceDelAddress(
|
||||
eq(createTunnelResp.interfaceName),
|
||||
eq(mLocalInnerAddress.getAddress().getHostAddress()),
|
||||
eq(mLocalInnerAddress.getPrefixLength()));
|
||||
@Ignore
|
||||
@Test
|
||||
public void testAddTunnelFailsForBadPackageName() throws Exception {
|
||||
try {
|
||||
IpSecTunnelInterfaceResponse createTunnelResp =
|
||||
createAndValidateTunnel(mSourceAddr, mDestinationAddr, "badPackage");
|
||||
fail("Expected a SecurityException for badPackage.");
|
||||
} catch (SecurityException expected) {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user