+ private void testRepeatedSmr() throws SocketTimeoutException, UnknownHostException {
+ cleanUP();
+ long timeout = ConfigIni.getInstance().getSmrTimeout();
+
+ final InstanceIdType iid = new InstanceIdType(1L);
+ final Eid eid1 = LispAddressUtil.asIpv4Eid("1.1.1.1", 1L);
+ final int expectedSmrs1 = 2;
+ final int expectedSmrs2 = 3;
+
+ /* set auth */
+ final Eid eid = LispAddressUtil.asIpv4PrefixBinaryEid("0.0.0.0/0", iid);
+ mapService.addAuthenticationKey(eid, NULL_AUTH_KEY);
+
+ /* add subscribers */
+ final String subscriberSrcRloc1 = "127.0.0.3";
+ final String subscriberSrcRloc2 = "127.0.0.4";
+ final Set<SubscriberRLOC> subscriberSet1 = Sets.newHashSet(newSubscriber(eid1, subscriberSrcRloc1),
+ newSubscriber(eid1, subscriberSrcRloc2));
+ mapService.addData(MappingOrigin.Southbound, eid1, SubKeys.SUBSCRIBERS, subscriberSet1);
+
+ final SocketReader reader1 = startSocketReader(subscriberSrcRloc1, 15000);
+ final SocketReader reader2 = startSocketReader(subscriberSrcRloc2, 15000);
+ sleepForSeconds(1);
+
+ /* add mapping */
+ final MappingRecord mapping1 = new MappingRecordBuilder()
+ .setEid(eid1).setTimestamp(System.currentTimeMillis()).setRecordTtl(1440).build();
+ mapService.addMapping(MappingOrigin.Northbound, eid1, null, new MappingData(mapping1));
+
+ sleepForMilliseconds((timeout * expectedSmrs1) - 1500);
+ final List<MapRequest> requests1 = processSmrPackets(reader1, subscriberSrcRloc1, expectedSmrs1);
+ final MapReply mapReply1 = lms.handleMapRequest(
+ new MapRequestBuilder(requests1.get(0))
+ .setItrRloc(Lists.newArrayList(new ItrRlocBuilder()
+ .setRloc(LispAddressUtil.asIpv4Rloc(subscriberSrcRloc1)).build()))
+ .setEidItem(Lists.newArrayList(new EidItemBuilder().setEid(eid1).build()))
+ .setSmrInvoked(true)
+ .setSmr(false).build());
+
+ // sleep to get 1 extra smr request
+ sleepForMilliseconds(timeout * 1);
+ final List<MapRequest> requests2 = processSmrPackets(reader2, subscriberSrcRloc2, expectedSmrs2);
+ final MapReply mapReply2 = lms.handleMapRequest(
+ new MapRequestBuilder(requests2.get(0))
+ .setItrRloc(Lists.newArrayList(new ItrRlocBuilder()
+ .setRloc(LispAddressUtil.asIpv4Rloc(subscriberSrcRloc2)).build()))
+ .setEidItem(Lists.newArrayList(new EidItemBuilder().setEid(eid1).build()))
+ .setSmrInvoked(true)
+ .setSmr(false).build());
+
+ sleepForSeconds(3);
+ assertEquals(expectedSmrs1, requests1.size());
+ assertEquals(expectedSmrs2, requests2.size());
+ assertEquals((long) mapReply1.getNonce(), (long) requests1.get(0).getNonce());
+ assertEquals((long) mapReply2.getNonce(), (long) requests2.get(0).getNonce());
+ assertNextBufferEmpty(reader1);
+ assertNextBufferEmpty(reader2);
+
+ reader1.stopReading();
+ reader2.stopReading();
+ }
+
+ private SocketReader startSocketReader(String address, int timeout) {
+ DatagramSocket receivingSocket = null;
+
+ try {
+ receivingSocket = new DatagramSocket(new InetSocketAddress(address, LispMessage.PORT_NUM));
+ } catch (SocketException e) {
+ LOG.error("Can't initialize socket for {}", address, e);
+ }
+ return SocketReader.startReadingInStandaloneThread(receivingSocket, timeout);
+ }
+
+ private List<MapRequest> processSmrPackets(SocketReader reader, String address, int expectedSmrs) {
+ InetAddress inetAddress = null;
+ try {
+ inetAddress = InetAddress.getByName(address);
+ } catch (UnknownHostException e) {
+ LOG.error("Unknown address {}.", address, e);
+ }
+
+ final List<MapRequest> requests = Lists.newArrayList();
+ byte[][] buffers = reader.getBuffers(expectedSmrs);
+ for (byte[] buf : buffers) {
+ MapRequest request = MapRequestSerializer.getInstance().deserialize(ByteBuffer.wrap(buf), inetAddress);
+ requests.add(request);
+ }
+ return requests;
+ }
+
+ private void assertNextBufferEmpty(SocketReader socketReader) {
+ assertTrue(isArrayEmpty(socketReader.getBuffers(1)[0]));
+ }
+
+ private boolean isArrayEmpty(byte[] byteArray) {
+ for (byte b : byteArray) {
+ if (b != 0) {
+ return false;
+ }
+ }
+ return true;
+ }
+
+ private static SubscriberRLOC newSubscriber(Eid srcEid, String srcRlocIp) {
+ final byte[] addressBinary = InetAddresses.forString(srcRlocIp).getAddress();
+ final int timeout = 5;
+ final Rloc srcRloc = new RlocBuilder().setAddress(new Ipv4BinaryBuilder()
+ .setIpv4Binary(new Ipv4AddressBinary(addressBinary)).build()).build();
+
+ return new SubscriberRLOC(srcRloc, srcEid, timeout);
+ }
+
+ private void testMultipleMappings() throws UnknownHostException {
+ final InstanceIdType iid = new InstanceIdType(1L);
+ final String prefix1 = "1.1.127.10/32"; // prefix from the intersection of NB and SB gaps
+ final String prefix2 = "1.1.200.255/32"; // prefix with existing mapping in NB
+ final String prefix3 = "1.3.255.255/32";
+
+ final MapRequest mapRequest = new MapRequestBuilder().setSmrInvoked(false).setEidItem(Lists.newArrayList(
+ new EidItemBuilder().setEid(LispAddressUtil.asIpv4PrefixBinaryEid(prefix1, iid))
+ .build(),
+ new EidItemBuilder().setEid(LispAddressUtil.asIpv4PrefixBinaryEid(prefix2, iid))
+ .build(),
+ new EidItemBuilder().setEid(LispAddressUtil.asIpv4PrefixBinaryEid(prefix3, iid))
+ .build()))
+ .build();
+ final MapReply mapReply = lms.handleMapRequest(mapRequest);
+
+ // expected result
+ final String resultPrefix1 = "1.1.64.0";
+ final Address resultNegMapping1 = new Ipv4PrefixBinaryBuilder()
+ .setIpv4AddressBinary(new Ipv4AddressBinary(InetAddress.getByName(resultPrefix1).getAddress()))
+ .setIpv4MaskLength((short) 18).build();
+
+ final String resultPrefix2 = "1.1.192.0";
+ final Address resultMapping2 = new Ipv4PrefixBinaryBuilder()
+ .setIpv4AddressBinary(new Ipv4AddressBinary(InetAddress.getByName(resultPrefix2).getAddress()))
+ .setIpv4MaskLength((short) 18).build();
+
+ final String resultPrefix3 = "1.3.0.0";
+ final Address resultNegMapping3 = new Ipv4PrefixBinaryBuilder()
+ .setIpv4AddressBinary(new Ipv4AddressBinary(InetAddress.getByName(resultPrefix3).getAddress()))
+ .setIpv4MaskLength((short) 16).build();
+
+ assertEquals(resultNegMapping1, mapReply.getMappingRecordItem().get(0).getMappingRecord().getEid()
+ .getAddress());
+ assertEquals(resultMapping2, mapReply.getMappingRecordItem().get(1).getMappingRecord().getEid()
+ .getAddress());
+ assertEquals(resultNegMapping3, mapReply.getMappingRecordItem().get(2).getMappingRecord().getEid()
+ .getAddress());
+ }
+
+ /**
+ * Tests a negative mapping from an intersection of gaps in northbound and southbound.
+ */
+ private void testGapIntersection() throws UnknownHostException {
+ final InstanceIdType iid = new InstanceIdType(1L);
+
+ // request an Eid from a gap between mappings
+ final MapRequest mapRequest = new MapRequestBuilder().setSmrInvoked(false).setEidItem(Lists.newArrayList(
+ new EidItemBuilder().setEid(LispAddressUtil.asIpv4PrefixBinaryEid("1.1.127.10/32", iid))
+ .build()))
+ .build();
+ final MapReply mapReply = lms.handleMapRequest(mapRequest);
+
+ // expected negative mapping
+ final Address resultNegMapping = new Ipv4PrefixBinaryBuilder()
+ .setIpv4AddressBinary(new Ipv4AddressBinary(InetAddress.getByName("1.1.64.0").getAddress()))
+ .setIpv4MaskLength((short) 18).build();
+ assertEquals(resultNegMapping, mapReply.getMappingRecordItem().get(0).getMappingRecord().getEid()
+ .getAddress());
+ }
+
+ private void insertMappings() {
+ cleanUP();
+ mapService.setLookupPolicy(IMappingService.LookupPolicy.NB_AND_SB);
+
+ final InstanceIdType iid = new InstanceIdType(1L);
+ final String prefixNbLeft = "1.2.0.0/16";
+ final String prefixNbRight = "1.1.128.0/17";
+ final String prefixSbLeft = "1.1.32.0/19";
+ final String prefixSbRight = "1.0.0.0/8";
+
+ final MappingRecord mapRecordNbLeft = newMappingRecord(prefixNbLeft, iid);
+ final MappingRecord mapRecordNbRight = newMappingRecord(prefixNbRight, iid);
+ final MappingRecord mapRecordSbLeft = newMappingRecord(prefixSbLeft, iid);
+ final MappingRecord mapRecordSbRight = newMappingRecord(prefixSbRight, iid);
+
+ /* set auth */
+ final Eid eid = LispAddressUtil.asIpv4PrefixBinaryEid("0.0.0.0/0", iid);
+ mapService.addAuthenticationKey(eid, NULL_AUTH_KEY);
+
+ mapService.addMapping(MappingOrigin.Northbound, mapRecordNbLeft.getEid(), null,
+ new MappingData(mapRecordNbLeft));
+ mapService.addMapping(MappingOrigin.Northbound, mapRecordNbRight.getEid(), null,
+ new MappingData(mapRecordNbRight));
+ mapService.addMapping(MappingOrigin.Southbound, mapRecordSbLeft.getEid(), null,
+ new MappingData(mapRecordSbLeft, System.currentTimeMillis()));
+ mapService.addMapping(MappingOrigin.Southbound, mapRecordSbRight.getEid(), null,
+ new MappingData(mapRecordSbRight, System.currentTimeMillis()));
+
+ restartSocket();
+ sleepForSeconds(2);
+ }
+
+ /**
+ * Creates a new MappingRecord object.
+ *
+ * @param prefix The Eid prefix
+ * @param iid VNI
+ * @return new MappingRecord object
+ */
+ private MappingRecord newMappingRecord(String prefix, InstanceIdType iid) {
+ final Eid prefixBinary = LispAddressUtil.asIpv4PrefixBinaryEid(prefix, iid);
+ return new MappingRecordBuilder()
+ .setEid(prefixBinary)
+ .setLocatorRecord(Lists.newArrayList(new LocatorRecordBuilder()
+ .setRloc(LispAddressUtil.asIpv4Rloc("2.2.2.2")).setLocatorId("loc_id").build()))
+ .setTimestamp(System.currentTimeMillis()).setRecordTtl(1440).build();
+ }