mappingservice/integrationtest/src/test/java/org/opendaylight/lispflowmapping/integrationtest/MappingServiceIntegrationTest.java

   1 /*
   2  * Copyright (c) 2014 Contextream, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.lispflowmapping.integrationtest;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.junit.Assert.fail;
  14 import static org.ops4j.pax.exam.CoreOptions.options;
  15
  16 import java.io.BufferedReader;
  17 import java.io.IOException;
  18 import java.io.InputStream;
  19 import java.io.InputStreamReader;
  20 import java.io.OutputStreamWriter;
  21 import java.net.DatagramPacket;
  22 import java.net.DatagramSocket;
  23 import java.net.HttpURLConnection;
  24 import java.net.InetAddress;
  25 import java.net.InetSocketAddress;
  26 import java.net.MalformedURLException;
  27 import java.net.SocketException;
  28 import java.net.SocketTimeoutException;
  29 import java.net.URL;
  30 import java.net.UnknownHostException;
  31 import java.nio.ByteBuffer;
  32 import java.nio.charset.Charset;
  33 import java.util.ArrayList;
  34 import java.util.Arrays;
  35
  36 import javax.inject.Inject;
  37
  38 import org.apache.commons.codec.binary.Base64;
  39 import org.codehaus.jettison.json.JSONException;
  40 import org.codehaus.jettison.json.JSONObject;
  41 import org.codehaus.jettison.json.JSONTokener;
  42 import org.junit.After;
  43 import org.junit.Before;
  44 import org.junit.Test;
  45 import org.junit.runner.RunWith;
  46 import org.opendaylight.controller.sal.binding.api.NotificationListener;
  47 import org.opendaylight.lispflowmapping.implementation.LispMappingService;
  48 import org.opendaylight.lispflowmapping.implementation.serializer.LispMessage;
  49 import org.opendaylight.lispflowmapping.implementation.serializer.MapNotifySerializer;
  50 import org.opendaylight.lispflowmapping.implementation.serializer.MapRegisterSerializer;
  51 import org.opendaylight.lispflowmapping.implementation.serializer.MapReplySerializer;
  52 import org.opendaylight.lispflowmapping.implementation.serializer.MapRequestSerializer;
  53 import org.opendaylight.lispflowmapping.implementation.util.LispAFIConvertor;
  54 import org.opendaylight.lispflowmapping.interfaces.lisp.IFlowMapping;
  55 import org.opendaylight.lispflowmapping.type.AddressFamilyNumberEnum;
  56 import org.opendaylight.lispflowmapping.type.LispCanonicalAddressFormatEnum;
  57 import org.opendaylight.lispflowmapping.type.sbplugin.IConfigLispSouthboundPlugin;
  58 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.EidToLocatorRecord.Action;
  59 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafApplicationDataAddress;
  60 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafListAddress;
  61 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafSegmentAddress;
  62 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafTrafficEngineeringAddress;
  63 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispAFIAddress;
  64 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispIpv4Address;
  65 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispMacAddress;
  66 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapNotify;
  67 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRegister;
  68 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapReply;
  69 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRequest;
  70 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.ReencapHop;
  71 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.XtrRequestMapping;
  72 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.eidrecords.EidRecord;
  73 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.eidrecords.EidRecordBuilder;
  74 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.eidtolocatorrecords.EidToLocatorRecord;
  75 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.eidtolocatorrecords.EidToLocatorRecordBuilder;
  76 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafkeyvalueaddress.KeyBuilder;
  77 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafkeyvalueaddress.ValueBuilder;
  78 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcaflistaddress.Addresses;
  79 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcaflistaddress.AddressesBuilder;
  80 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafsegmentaddress.AddressBuilder;
  81 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafsourcedestaddress.DstAddressBuilder;
  82 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafsourcedestaddress.SrcAddressBuilder;
  83 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcaftrafficengineeringaddress.Hops;
  84 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcaftrafficengineeringaddress.HopsBuilder;
  85 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.LispAddressContainer;
  86 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.LcafKeyValue;
  87 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.LcafSourceDest;
  88 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address;
  89 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafapplicationdata.LcafApplicationDataAddr;
  90 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafapplicationdata.LcafApplicationDataAddrBuilder;
  91 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafkeyvalue.LcafKeyValueAddressAddrBuilder;
  92 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcaflist.LcafListAddrBuilder;
  93 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafsegment.LcafSegmentAddrBuilder;
  94 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafsourcedest.LcafSourceDestAddr;
  95 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafsourcedest.LcafSourceDestAddrBuilder;
  96 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcaftrafficengineering.LcafTrafficEngineeringAddrBuilder;
  97 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.no.NoAddressBuilder;
  98 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.locatorrecords.LocatorRecord;
  99 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.locatorrecords.LocatorRecordBuilder;
 100 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapregisternotification.MapRegisterBuilder;
 101 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequest.ItrRloc;
 102 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequest.ItrRlocBuilder;
 103 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequest.SourceEidBuilder;
 104 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequestnotification.MapRequestBuilder;
 105 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.reencaphop.Hop;
 106 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.reencaphop.HopBuilder;
 107 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
 108 import org.ops4j.pax.exam.Configuration;
 109 import org.ops4j.pax.exam.Option;
 110 import org.ops4j.pax.exam.junit.PaxExam;
 111 import org.ops4j.pax.exam.util.Filter;
 112 import org.osgi.framework.Bundle;
 113 import org.osgi.framework.BundleContext;
 114 import org.osgi.framework.InvalidSyntaxException;
 115 import org.osgi.framework.ServiceReference;
 116 import org.slf4j.Logger;
 117 import org.slf4j.LoggerFactory;
 118
 119 @RunWith(PaxExam.class)
 120 public class MappingServiceIntegrationTest {
 121
 122     protected static final Logger LOG = LoggerFactory.getLogger(MappingServiceIntegrationTest.class);
 123     private byte[] mapRequestPacket;
 124     private byte[] mapRegisterPacketWithNotify;
 125     private byte[] mapRegisterPacketWithoutNotify;
 126     String lispBindAddress = "127.0.0.1";
 127     String ourAddress = "127.0.0.2";
 128     private LispAFIAddress locatorEid;
 129     private DatagramSocket socket;
 130     private byte[] mapRegisterPacketWithAuthenticationAndMapNotify;
 131
 132     public static final String ODL = "org.opendaylight.controller";
 133     public static final String YANG = "org.opendaylight.yangtools";
 134     public static final String JERSEY = "com.sun.jersey";
 135     private static final int MAX_SERVICE_LOAD_RETRIES = 45;
 136     private static final int MAX_NOTIFICATION_RETRYS = 20;
 137
 138     @After
 139     public void after() {
 140         if (socket != null) {
 141             socket.close();
 142         }
 143         if (connection != null) {
 144             connection.disconnect();
 145         }
 146     }
 147
 148     @Before
 149     public void before() throws Exception {
 150         areWeReady();
 151         locatorEid = LispAFIConvertor.asIPAfiAddress("4.3.2.1");
 152         socket = initSocket(socket, LispMessage.PORT_NUM);
 153
 154         // SRC: 127.0.0.1:58560 to 127.0.0.1:4342
 155         // LISP(Type = 8 - Encapsulated)
 156         // IP: 192.168.136.10 -> 153.16.254.1
 157         // UDP: 56756
 158         // LISP(Type = 1 Map-Request
 159         // Record Count: 1
 160         // ITR-RLOC count: 0
 161         // Source EID AFI: 0
 162         // Source EID not present
 163         // Nonce: 0x3d8d2acd39c8d608
 164         // ITR-RLOC AFI=1 Address=192.168.136.10
 165         // Record 1: 153.16.254.1/32
 166         mapRequestPacket = extractWSUdpByteArray(new String("0000   00 00 00 00 00 00 00 00 00 00 00 00 08 00 45 00 " //
 167                 + "0010   00 58 00 00 40 00 40 11 3c 93 7f 00 00 01 7f 00 "
 168                 + "0020   00 01 e4 c0 10 f6 00 44 fe 57 80 00 00 00 45 00 "
 169                 + "0030   00 3c d4 31 00 00 ff 11 56 f3 7f 00 00 02 99 10 "
 170                 + "0040   fe 01 dd b4 10 f6 00 28 ef 3a 10 00 00 01 3d 8d "
 171                 + "0050   2a cd 39 c8 d6 08 00 01 01 02 03 04 00 01 7f 00 00 02 00 20 " //
 172                 + "0060   00 01 99 10 fe 01"));
 173
 174         // IP: 192.168.136.10 -> 128.223.156.35
 175         // UDP: 49289 -> 4342
 176         // LISP(Type = 3 Map-Register, P=1, M=1
 177         // Record Counter: 1
 178         // Nonce: 0
 179         // Key ID: 0x0001
 180         // AuthDataLength: 20 Data:
 181         // e8:f5:0b:c5:c5:f2:b0:21:27:a8:21:41:04:f3:46:5a:a5:68:89:ec
 182         // EID prefix: 153.16.254.1/32 (EID=0x9910FE01), TTL: 10, Authoritative,
 183         // No-Action
 184         // Local RLOC: 192.168.136.10 (RLOC=0xC0A8880A), Reachable,
 185         // Priority/Weight: 1/100, Multicast Priority/Weight:
 186         // 255/0
 187         //
 188
 189         mapRegisterPacketWithAuthenticationAndMapNotify = extractWSUdpByteArray(new String("0000   00 50 56 ee d1 4f 00 0c 29 7a ce 79 08 00 45 00 " //
 190                 + "0010   00 5c 00 00 40 00 40 11 d4 db c0 a8 88 0a 80 df "
 191                 + "0020   9c 23 d6 40 10 f6 00 48 59 a4 38 00 01 01 00 00 "
 192                 + "0030   00 00 00 00 00 00 00 01 00 14 0e a4 c6 d8 a4 06 "
 193                 + "0040   71 7c 33 a4 5c 4a 83 1c de 74 53 03 0c ad 00 00 "
 194                 + "0050   00 0a 01 20 10 00 00 00 00 01 99 10 fe 01 01 64 " //
 195                 + "0060   ff 00 00 05 00 01 c0 a8 88 0a"));
 196
 197         // IP: 192.168.136.10 -> 128.223.156.35
 198         // UDP: 49289 -> 4342
 199         // LISP(Type = 3 Map-Register, P=1, M=1
 200         // Record Counter: 1
 201         // Nonce: 7
 202         // Key ID: 0x0000 NO AUTHENTICATION!!
 203         // AuthDataLength: 00 Data:
 204         // EID prefix: 153.16.254.1/32 (EID=0x9910FE01), TTL: 10, Authoritative,
 205         // No-Action
 206         // Local RLOC: 192.168.136.10 (RLOC=0xC0A8880A), Reachable,
 207         // Priority/Weight: 1/100, Multicast Priority/Weight:
 208         // 255/0
 209         //
 210
 211         mapRegisterPacketWithNotify = extractWSUdpByteArray(new String("0000   00 50 56 ee d1 4f 00 0c 29 7a ce 79 08 00 45 00 " //
 212                 + "0010   00 5c 00 00 40 00 40 11 d4 db c0 a8 88 0a 80 df "
 213                 + "0020   9c 23 d6 40 10 f6 00 48 59 a4 38 00 01 01 00 00 "
 214                 + "0030   00 00 00 00 00 07 00 00 00 14 0e a4 c6 d8 a4 06 "
 215                 + "0040   71 7c 33 a4 5c 4a 83 1c de 74 53 03 0c ad 00 00 "
 216                 + "0050   00 0a 01 20 10 00 00 00 00 01 99 10 fe 01 01 64 " //
 217                 + "0060   ff 00 00 05 00 01 c0 a8 88 0a"));
 218
 219         // IP: 192.168.136.10 -> 128.223.156.35
 220         // UDP: 49289 -> 4342
 221         // LISP(Type = 3 Map-Register, P=1, M=1
 222         // Record Counter: 1
 223         // Nonce: 7
 224         // Key ID: 0x0000 NO AUTHENTICATION!!
 225         // AuthDataLength: 00 Data:
 226         // EID prefix: 153.16.254.1/32 (EID=0x9910FE01), TTL: 10, Authoritative,
 227         // No-Action
 228         // Local RLOC: 192.168.136.10 (RLOC=0xC0A8880A), Reachable,
 229         // Priority/Weight: 1/100, Multicast Priority/Weight:
 230         // 255/0
 231         //
 232
 233         mapRegisterPacketWithoutNotify = extractWSUdpByteArray(new String("0000   00 50 56 ee d1 4f 00 0c 29 7a ce 79 08 00 45 00 " //
 234                 + "0010   00 5c 00 00 40 00 40 11 d4 db c0 a8 88 0a 80 df "
 235                 + "0020   9c 23 d6 40 10 f6 00 48 59 a4 38 00 00 01 00 00 "
 236                 + "0030   00 00 00 00 00 07 00 00 00 14 0e a4 c6 d8 a4 06 "
 237                 + "0040   71 7c 33 a4 5c 4a 83 1c de 74 53 03 0c ad 00 00 "
 238                 + "0050   00 0a 01 20 10 00 00 00 00 01 99 10 fe 01 01 64 " //
 239                 + "0060   ff 00 00 05 00 01 c0 a8 88 0a"));
 240     }
 241
 242     @Inject
 243     private BundleContext bc;
 244     private HttpURLConnection connection;
 245     protected static boolean notificationCalled;
 246
 247     @Inject @Filter(timeout=10000)
 248     private IFlowMapping lms;
 249
 250     @Inject @Filter(timeout=10000)
 251     private IConfigLispSouthboundPlugin configLispPlugin;
 252
 253
 254     // Configure the OSGi container
 255     @Configuration
 256     public Option[] config() {
 257         return options(MappingServiceTestHelper.mappingServiceBundlesWithClusterDAO());
 258     }
 259
 260     @Test
 261     public void testSimpleUsage() throws Exception {
 262         mapRequestSimple();
 263         mapRegisterWithMapNotify();
 264         mapRegisterWithMapNotifyAndMapRequest();
 265         registerAndQuery__MAC();
 266         mapRequestMapRegisterAndMapRequest();
 267         mapRegisterWithAuthenticationWithoutConfiguringAKey();
 268         mapRegisterWithoutMapNotify();
 269     }
 270
 271     @Test
 272     public void testLCAFs() throws Exception {
 273         registerAndQuery__SrcDestLCAF();
 274         registerAndQuery__SrcDestLCAFOverlap();
 275         registerAndQuery__KeyValueLCAF();
 276         registerAndQuery__ListLCAF();
 277         registerAndQuery__ApplicationData();
 278         registerAndQuery__TrafficEngineering();
 279         registerAndQuery__SegmentLCAF();
 280     }
 281
 282     @Test
 283     public void testMask() throws Exception {
 284         //testPasswordExactMatch();                     TODO commented because it needs NB
 285         //testPasswordMaskMatch();                      TODO commented because it needs NB
 286         eidPrefixLookupIPv4();
 287         eidPrefixLookupIPv6();
 288     }
 289 /*
 290     @Test
 291     public void testNorthbound() throws Exception {
 292         northboundAddKey();
 293         northboundAddMapping();
 294         northboundDeleteMapping();
 295         northboundRetrieveKey();
 296         northboundRetrieveMapping();
 297         northboundRetrieveSourceDestKey();
 298         northboundRetrieveSourceDestMapping();
 299     }
 300 */
 301     @Test
 302     public void testOverWriting() throws Exception {
 303         //testMapRegisterDosntOverwritesOtherSubKeys(); TODO weird failure, needs debug
 304         testMapRegisterOverwritesSameSubkey();
 305         testMapRegisterOverwritesNoSubkey();
 306         testMapRegisterDoesntOverwritesNoSubkey();
 307     }
 308
 309     @Test
 310     public void testTimeOuts() throws Exception {
 311         mapRequestMapRegisterAndMapRequestTestTimeout();
 312         //mapRequestMapRegisterAndMapRequestTestNativelyForwardTimeoutResponse();   TODO commented because it needs NB
 313     }
 314
 315 //    @Test
 316 //    public void testNonProxy() throws Throwable {
 317 //        testSimpleNonProxy();
 318 //        testNonProxyOtherPort();
 319 //        testRecievingNonProxyOnXtrPort();
 320 //    }
 321
 322     @Test
 323     public void testSmr() throws Exception {
 324         registerQueryRegisterWithSmr();
 325     }
 326
 327     // ------------------------------- Simple Tests ---------------------------
 328
 329     public void mapRequestSimple() throws SocketTimeoutException {
 330         cleanUP();
 331
 332         // This Map-Request is sent from a source port different from 4342
 333         // We close and bind the socket on the correct port
 334         if (socket != null) {
 335             socket.close();
 336         }
 337         socket = initSocket(socket, 56756);
 338
 339         sendPacket(mapRequestPacket);
 340         ByteBuffer readBuf = ByteBuffer.wrap(receivePacket().getData());
 341         MapReply reply = MapReplySerializer.getInstance().deserialize(readBuf);
 342         assertEquals(4435248268955932168L, reply.getNonce().longValue());
 343
 344     }
 345
 346     public void mapRegisterWithMapNotify() throws SocketTimeoutException {
 347         cleanUP();
 348         sendPacket(mapRegisterPacketWithNotify);
 349         MapNotify reply = receiveMapNotify();
 350         assertEquals(7, reply.getNonce().longValue());
 351     }
 352
 353     public void mapRegisterWithMapNotifyAndMapRequest() throws SocketTimeoutException {
 354         cleanUP();
 355         LispAFIAddress eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 356
 357         MapReply mapReply = registerAddressAndQuery(eid, 32);
 358
 359         assertEquals(4, mapReply.getNonce().longValue());
 360         assertEquals(LispAFIConvertor.toContainer(locatorEid), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0)
 361                 .getLispAddressContainer());
 362
 363     }
 364
 365     public void registerAndQuery__MAC() throws SocketTimeoutException {
 366         cleanUP();
 367         String macAddress = "01:02:03:04:05:06";
 368
 369         MapReply reply = registerAddressAndQuery(LispAFIConvertor.asMacAfiAddress(macAddress));
 370
 371         assertTrue(true);
 372         LispAFIAddress addressFromNetwork = LispAFIConvertor.toAFI(reply.getEidToLocatorRecord().get(0).getLispAddressContainer());
 373         assertTrue(addressFromNetwork instanceof LispMacAddress);
 374         String macAddressFromReply = ((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.mac.MacAddress) addressFromNetwork).getMacAddress().getValue();
 375
 376         assertEquals(macAddress, macAddressFromReply);
 377     }
 378
 379     public void mapRequestMapRegisterAndMapRequest() throws SocketTimeoutException {
 380         cleanUP();
 381         LispAFIAddress eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 382         MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
 383         mapRequestBuilder.setNonce((long) 4);
 384         mapRequestBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(
 385                 LispAFIConvertor.toContainer(new NoAddressBuilder().setAfi((short) 0).build())).build());
 386         mapRequestBuilder.setEidRecord(new ArrayList<EidRecord>());
 387         mapRequestBuilder.getEidRecord().add(
 388                 new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(eid)).build());
 389         mapRequestBuilder.setItrRloc(new ArrayList<ItrRloc>());
 390         mapRequestBuilder.getItrRloc().add(
 391                 new ItrRlocBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
 392         sendMapRequest(mapRequestBuilder.build());
 393         MapReply mapReply = receiveMapReply();
 394         assertEquals(4, mapReply.getNonce().longValue());
 395         assertEquals(0, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
 396         MapRegisterBuilder mapRegisterbuilder = new MapRegisterBuilder();
 397         mapRegisterbuilder.setWantMapNotify(true);
 398         mapRegisterbuilder.setNonce((long) 8);
 399         EidToLocatorRecordBuilder etlrBuilder = new EidToLocatorRecordBuilder();
 400         etlrBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
 401         etlrBuilder.setMaskLength((short) 32);
 402         etlrBuilder.setRecordTtl(254);
 403         LocatorRecordBuilder recordBuilder = new LocatorRecordBuilder();
 404         recordBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
 405         etlrBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
 406         etlrBuilder.getLocatorRecord().add(recordBuilder.build());
 407         mapRegisterbuilder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
 408         mapRegisterbuilder.getEidToLocatorRecord().add(etlrBuilder.build());
 409         sendMapRegister(mapRegisterbuilder.build());
 410         MapNotify mapNotify = receiveMapNotify();
 411         assertEquals(8, mapNotify.getNonce().longValue());
 412         sendMapRequest(mapRequestBuilder.build());
 413         mapReply = receiveMapReply();
 414         assertEquals(4, mapReply.getNonce().longValue());
 415         assertEquals(recordBuilder.getLispAddressContainer(), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0)
 416                 .getLispAddressContainer());
 417
 418     }
 419
 420     public void testMapRegisterDosntOverwritesOtherSubKeys() throws SocketTimeoutException {
 421         cleanUP();
 422         LispAFIAddress eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 423         LispAFIAddress rloc1Value = LispAFIConvertor.asIPAfiAddress("4.3.2.1");
 424         LispAFIAddress rloc1 = LispAFIConvertor.asKeyValue("subkey1", LispAFIConvertor.toPrimitive(rloc1Value));
 425         LispAFIAddress rloc2Value = LispAFIConvertor.asIPAfiAddress("4.3.2.2");
 426         LispAFIAddress rloc2 = LispAFIConvertor.asKeyValue("subkey2", LispAFIConvertor.toPrimitive(rloc2Value));
 427         MapReply mapReply = sendMapRegisterTwiceWithDiffrentValues(eid, rloc1, rloc2);
 428         assertEquals(2, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
 429         assertEquals(LispAFIConvertor.toContainer(rloc2), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0).getLispAddressContainer());
 430         assertEquals(LispAFIConvertor.toContainer(rloc1), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(1).getLispAddressContainer());
 431     }
 432
 433     public void testMapRegisterOverwritesSameSubkey() throws SocketTimeoutException {
 434         cleanUP();
 435         LispAFIAddress eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 436         LispAFIAddress rloc1Value = LispAFIConvertor.asIPAfiAddress("4.3.2.1");
 437         LispAFIAddress rloc1 = LispAFIConvertor.asKeyValue("subkey", LispAFIConvertor.toPrimitive(rloc1Value));
 438         LispAFIAddress rloc2Value = LispAFIConvertor.asIPAfiAddress("4.3.2.2");
 439         LispAFIAddress rloc2 = LispAFIConvertor.asKeyValue("subkey", LispAFIConvertor.toPrimitive(rloc2Value));
 440         MapReply mapReply = sendMapRegisterTwiceWithDiffrentValues(eid, rloc1, rloc2);
 441         assertEquals(1, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
 442         assertEquals(LispAFIConvertor.toContainer(rloc2), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0).getLispAddressContainer());
 443     }
 444
 445     public void testMapRegisterOverwritesNoSubkey() throws SocketTimeoutException {
 446         cleanUP();
 447         lms.setOverwrite(true);
 448         LispAFIAddress eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 449         LispAFIAddress rloc1Value = LispAFIConvertor.asIPAfiAddress("4.3.2.1");
 450         LispAFIAddress rloc2Value = LispAFIConvertor.asIPAfiAddress("4.3.2.2");
 451         MapReply mapReply = sendMapRegisterTwiceWithDiffrentValues(eid, rloc1Value, rloc2Value);
 452         assertEquals(1, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
 453         assertEquals(LispAFIConvertor.toContainer(rloc2Value), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0)
 454                 .getLispAddressContainer());
 455     }
 456
 457     public void testMapRegisterDoesntOverwritesNoSubkey() throws SocketTimeoutException {
 458         cleanUP();
 459         lms.setOverwrite(false);
 460         LispAFIAddress eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 461         LispAFIAddress rloc1Value = LispAFIConvertor.asIPAfiAddress("4.3.2.1");
 462         LispAFIAddress rloc2Value = LispAFIConvertor.asIPAfiAddress("4.3.2.2");
 463         MapReply mapReply = sendMapRegisterTwiceWithDiffrentValues(eid, rloc1Value, rloc2Value);
 464         assertEquals(2, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
 465         LispAddressContainer rloc1ReturnValueContainer = mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0).getLispAddressContainer();
 466         LispAddressContainer rloc2ReturnValueContainer = mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(1).getLispAddressContainer();
 467         assertTrue((LispAFIConvertor.toContainer(rloc1Value).equals(rloc1ReturnValueContainer) && LispAFIConvertor.toContainer(rloc2Value).equals(
 468                 rloc2ReturnValueContainer))
 469                 || (LispAFIConvertor.toContainer(rloc1Value).equals(rloc2ReturnValueContainer) && LispAFIConvertor.toContainer(rloc2Value).equals(
 470                         rloc1ReturnValueContainer)));
 471     }
 472
 473     private MapReply sendMapRegisterTwiceWithDiffrentValues(LispAFIAddress eid, LispAFIAddress rloc1, LispAFIAddress rloc2)
 474             throws SocketTimeoutException {
 475         MapRegister mb = createMapRegister(eid, rloc1);
 476         MapNotify mapNotify = lms.handleMapRegister(mb, false);
 477         MapRequest mr = createMapRequest(eid);
 478         MapReply mapReply = lms.handleMapRequest(mr);
 479         assertEquals(mb.getEidToLocatorRecord().get(0).getLocatorRecord().get(0).getLispAddressContainer(), mapReply.getEidToLocatorRecord().get(0)
 480                 .getLocatorRecord().get(0).getLispAddressContainer());
 481         mb = createMapRegister(eid, rloc2);
 482         mapNotify = lms.handleMapRegister(mb, false);
 483         assertEquals(8, mapNotify.getNonce().longValue());
 484         mr = createMapRequest(eid);
 485         sendMapRequest(mr);
 486         mapReply = lms.handleMapRequest(mr);
 487         return mapReply;
 488     }
 489
 490     public void mapRegisterWithAuthenticationWithoutConfiguringAKey() throws SocketTimeoutException {
 491         cleanUP();
 492         sendPacket(mapRegisterPacketWithAuthenticationAndMapNotify);
 493         try {
 494             receivePacket(3000);
 495             // If didn't timeout then fail:
 496             fail();
 497         } catch (SocketTimeoutException ste) {
 498         }
 499     }
 500
 501     public void mapRegisterWithoutMapNotify() {
 502         cleanUP();
 503         sendPacket(mapRegisterPacketWithoutNotify);
 504         try {
 505             receivePacket(3000);
 506             // If didn't timeout then fail:
 507             fail();
 508         } catch (SocketTimeoutException ste) {
 509         }
 510     }
 511
 512     public void registerQueryRegisterWithSmr() throws SocketTimeoutException {
 513         cleanUP();
 514         lms.setShouldUseSmr(true);
 515
 516         sendPacket(mapRegisterPacketWithNotify);
 517         receiveMapNotify();
 518
 519         sendPacket(mapRequestPacket);
 520         try {
 521             Thread.sleep(1000);
 522         } catch (InterruptedException e) {
 523         }
 524
 525         mapRegisterPacketWithoutNotify[mapRegisterPacketWithoutNotify.length - 1] += 1;
 526         sendPacket(mapRegisterPacketWithoutNotify);
 527
 528         ByteBuffer readBuf = ByteBuffer.wrap(receivePacket().getData());
 529         MapRequest smr = MapRequestSerializer.getInstance().deserialize(readBuf);
 530         assertTrue(smr.isSmr());
 531         LispAddressContainer smrEid = smr.getEidRecord().get(0).getLispAddressContainer();
 532         assertTrue(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("153.16.254.1")).equals(smrEid));
 533     }
 534
 535     // --------------------- Northbound Tests ---------------------------
 536
 537     private void northboundAddKey() throws Exception {
 538         cleanUP();
 539         LispIpv4Address address = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
 540         int mask = 32;
 541         String pass = "asdf";
 542
 543         URL url = createPutURL("key");
 544         String authKeyJSON = createAuthKeyJSON(pass, address, mask);
 545         callURL("PUT", "application/json", "text/plain", authKeyJSON, url);
 546
 547         String retrievedKey = lms.getAuthenticationKey(LispAFIConvertor.toContainer(address), mask);
 548
 549         // Check stored password matches the one sent
 550         assertEquals(pass, retrievedKey);
 551
 552     }
 553
 554     private void northboundRetrieveSourceDestKey() throws Exception {
 555         cleanUP();
 556         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 address1 = (org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) LispAFIConvertor
 557                 .toPrimitive(LispAFIConvertor.asIPAfiAddress("10.0.0.1"));
 558         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 address2 = (org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) LispAFIConvertor
 559                 .toPrimitive(LispAFIConvertor.asIPAfiAddress("10.0.0.2"));
 560         int mask1 = 32;
 561         int mask2 = 32;
 562         LcafSourceDestAddr sourceDestAddress = new LcafSourceDestAddrBuilder().setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode())
 563                 .setLcafType((short) LispCanonicalAddressFormatEnum.SOURCE_DEST.getLispCode())
 564                 .setSrcAddress(new SrcAddressBuilder().setPrimitiveAddress(address1).build()).setSrcMaskLength((short) mask1)
 565                 .setDstAddress(new DstAddressBuilder().setPrimitiveAddress(address2).build()).setDstMaskLength((short) mask2).build();
 566         String pass = "asdf";
 567
 568         lms.addAuthenticationKey(LispAFIConvertor.toContainer(sourceDestAddress), mask1, pass);
 569
 570         // URL url = createGetKeyIPv4URL(address1, mask1);
 571         URL url = createGetKeySourceDestURL(address1.getIpv4Address().getAfi(),
 572                 ((LispIpv4Address) LispAFIConvertor.toAFIfromPrimitive(sourceDestAddress.getSrcAddress().getPrimitiveAddress())).getIpv4Address().getValue(), sourceDestAddress.getSrcMaskLength(),
 573                 ((LispIpv4Address) LispAFIConvertor.toAFIfromPrimitive(sourceDestAddress.getDstAddress().getPrimitiveAddress())).getIpv4Address().getValue(), sourceDestAddress.getDstMaskLength());
 574         String reply = callURL("GET", null, "application/json", null, url);
 575         JSONTokener jt = new JSONTokener(reply);
 576         JSONObject json = new JSONObject(jt);
 577
 578         // test that the password matches what was we expected.
 579         assertEquals(pass, json.get("key"));
 580
 581     }
 582
 583     private void northboundRetrieveKey() throws Exception {
 584         cleanUP();
 585         LispIpv4Address address = LispAFIConvertor.asIPAfiAddress("10.0.0.1");
 586         int mask = 32;
 587         String pass = "asdf";
 588
 589         lms.addAuthenticationKey(LispAFIConvertor.toContainer(address), mask, pass);
 590
 591         URL url = createGetKeyIPv4URL(address, mask);
 592         String reply = callURL("GET", null, "application/json", null, url);
 593         JSONTokener jt = new JSONTokener(reply);
 594         JSONObject json = new JSONObject(jt);
 595
 596         // test that the password matches what was we expected.
 597         assertEquals(pass, json.get("key"));
 598
 599     }
 600
 601     private String createAuthKeyJSON(String key, LispIpv4Address address, int mask) {
 602         return "{\"key\" : \"" + key + "\",\"maskLength\" : " + mask + ",\"address\" : " + "{\"ipAddress\" : \""
 603                 + address.getIpv4Address().getValue() + "\",\"afi\" : " + address.getAfi().shortValue() + "}}";
 604     }
 605
 606     private void northboundAddMapping() throws Exception {
 607         cleanUP();
 608         String pass = "asdf";
 609         LispIpv4Address eid = LispAFIConvertor.asIPAfiAddress("10.0.0.1");
 610         int mask = 32;
 611         LispIpv4Address rloc = LispAFIConvertor.asIPAfiAddress("20.0.0.2");
 612
 613         // NB add mapping always checks the key
 614         lms.addAuthenticationKey(LispAFIConvertor.toContainer(eid), mask, pass);
 615
 616         URL url = createPutURL("mapping");
 617         String mapRegisterJSON = createMapRegisterJSON(pass, eid, mask, rloc);
 618         callURL("PUT", "application/json", "text/plain", mapRegisterJSON, url);
 619
 620         // Retrieve the RLOC from the database
 621         MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
 622         mapRequestBuilder.setPitr(false);
 623         mapRequestBuilder.setEidRecord(new ArrayList<EidRecord>());
 624         mapRequestBuilder.getEidRecord().add(
 625                 new EidRecordBuilder().setMask((short) mask).setLispAddressContainer(LispAFIConvertor.toContainer(eid)).build());
 626         MapReply mapReply = lms.handleMapRequest(mapRequestBuilder.build());
 627
 628         LispIpv4Address retrievedRloc = (LispIpv4Address) LispAFIConvertor.toAFI(
 629                 mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0).getLispAddressContainer());
 630
 631         assertEquals(rloc.getIpv4Address().getValue(), retrievedRloc.getIpv4Address().getValue());
 632
 633     }
 634
 635     private String createMapRegisterJSON(String key, LispIpv4Address eid, int mask, LispIpv4Address rloc) {
 636         String jsonString = "{ " + "\"key\" : \"" + key + "\"," + "\"mapregister\" : " + "{ " + "\"proxyMapReply\" : false, "
 637                 + "\"eidToLocatorRecords\" : " + "[ " + "{ " + "\"authoritative\" : true," + "\"prefixGeneric\" : " + "{ " + "\"ipAddress\" : \""
 638                 + eid.getIpv4Address().getValue() + "\"," + "\"afi\" : " + eid.getAfi().shortValue() + "}," + "\"mapVersion\" : 0,"
 639                 + "\"maskLength\" : " + mask + ", " + "\"action\" : \"NoAction\"," + "\"locators\" : " + "[ " + "{ " + "\"multicastPriority\" : 1,"
 640                 + "\"locatorGeneric\" : " + "{ " + "\"ipAddress\" : \"" + rloc.getIpv4Address().getValue() + "\"," + "\"afi\" : "
 641                 + rloc.getAfi().shortValue() + "}, " + "\"routed\" : true," + "\"multicastWeight\" : 50," + "\"rlocProbed\" : false, "
 642                 + "\"localLocator\" : false, " + "\"priority\" : 1, " + "\"weight\" : 50 " + "} " + "], " + "\"recordTtl\" : 100" + "} " + "], "
 643                 + "\"nonce\" : 3," + "\"keyId\" : 0 " + "} " + "}";
 644
 645         return jsonString;
 646     }
 647
 648     private void northboundRetrieveMapping() throws Exception {
 649         cleanUP();
 650         LispIpv4Address eid = LispAFIConvertor.asIPAfiAddress("10.0.0.1");
 651         int mask = 32;
 652         LispIpv4Address rloc = LispAFIConvertor.asIPAfiAddress("20.0.0.2");
 653         // Insert mapping in the database
 654         MapRegisterBuilder mapRegister = new MapRegisterBuilder();
 655         EidToLocatorRecordBuilder etlr = new EidToLocatorRecordBuilder();
 656         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
 657         etlr.setMaskLength((short) mask);
 658         etlr.setRecordTtl(254);
 659         etlr.setAuthoritative(false);
 660         etlr.setAction(Action.NoAction);
 661         LocatorRecordBuilder record = new LocatorRecordBuilder();
 662         record.setLispAddressContainer(LispAFIConvertor.toContainer(rloc));
 663         record.setRouted(true);
 664         record.setRlocProbed(false);
 665         record.setLocalLocator(false);
 666         record.setPriority((short) 1);
 667         record.setWeight((short) 50);
 668         record.setMulticastPriority((short) 1);
 669         record.setMulticastWeight((short) 1);
 670         etlr.setLocatorRecord(new ArrayList<LocatorRecord>());
 671         etlr.getLocatorRecord().add(record.build());
 672         mapRegister.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
 673         mapRegister.getEidToLocatorRecord().add(etlr.build());
 674         lms.handleMapRegister(mapRegister.build(), false);
 675
 676         // Get mapping using NB interface. No IID used
 677         URL url = createGetMappingIPv4URL(0, eid, mask);
 678         String reply = callURL("GET", null, "application/json", null, url);
 679         JSONTokener jt = new JSONTokener(reply);
 680         JSONObject json = new JSONObject(jt);
 681
 682         // With just one locator, locators is not a JSONArray
 683         String rlocRetrieved = json.getJSONArray("locators").getJSONObject(0).getJSONObject("locatorGeneric").getString("ipAddress");
 684
 685         assertEquals(rloc.getIpv4Address().getValue(), rlocRetrieved);
 686
 687     }
 688
 689     private void northboundDeleteMapping() throws Exception {
 690         cleanUP();
 691         LispIpv4Address eid = LispAFIConvertor.asIPAfiAddress("10.0.0.1");
 692         int mask = 32;
 693         LispIpv4Address rloc = LispAFIConvertor.asIPAfiAddress("20.0.0.2");
 694         // Insert mapping in the database
 695         MapRegisterBuilder mapRegister = new MapRegisterBuilder();
 696         EidToLocatorRecordBuilder etlr = new EidToLocatorRecordBuilder();
 697         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
 698         etlr.setMaskLength((short) mask);
 699         etlr.setRecordTtl(254);
 700         etlr.setAuthoritative(false);
 701         etlr.setAction(Action.NoAction);
 702         LocatorRecordBuilder record = new LocatorRecordBuilder();
 703         record.setLispAddressContainer(LispAFIConvertor.toContainer(rloc));
 704         record.setRouted(true);
 705         record.setRlocProbed(false);
 706         record.setLocalLocator(false);
 707         record.setPriority((short) 1);
 708         record.setWeight((short) 50);
 709         record.setMulticastPriority((short) 1);
 710         record.setMulticastWeight((short) 1);
 711         etlr.setLocatorRecord(new ArrayList<LocatorRecord>());
 712         etlr.getLocatorRecord().add(record.build());
 713         mapRegister.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
 714         mapRegister.getEidToLocatorRecord().add(etlr.build());
 715         lms.handleMapRegister(mapRegister.build(), false);
 716
 717         // Delete mapping using NB interface. No IID used
 718         URL url = createDeleteMappingIPv4URL(0, eid, mask);
 719         String reply = callURL("DELETE", null, "application/json", null, url);
 720
 721         // Get mapping using NB interface. No IID used
 722         url = createGetMappingIPv4URL(0, eid, mask);
 723         reply = callURL("GET", null, "application/json", null, url);
 724         JSONTokener jt = new JSONTokener(reply);
 725         JSONObject json = new JSONObject(jt);
 726
 727         // With just one locator, locators is not a JSONArray
 728         assertEquals(json.getJSONArray("locators").length(), 0);
 729     }
 730
 731     private void northboundRetrieveSourceDestMapping() throws Exception {
 732         cleanUP();
 733         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 address1 = (org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) LispAFIConvertor
 734                 .toPrimitive(LispAFIConvertor.asIPAfiAddress("10.0.0.1"));
 735         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 address2 = (org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) LispAFIConvertor
 736                 .toPrimitive(LispAFIConvertor.asIPAfiAddress("10.0.0.2"));
 737         int mask1 = 32;
 738         int mask2 = 32;
 739         LcafSourceDestAddr sourceDestAddress = new LcafSourceDestAddrBuilder().setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode())
 740                 .setLcafType((short) LispCanonicalAddressFormatEnum.SOURCE_DEST.getLispCode())
 741                 .setSrcAddress(new SrcAddressBuilder().setPrimitiveAddress(address1).build()).setSrcMaskLength((short) mask1)
 742                 .setDstAddress(new DstAddressBuilder().setPrimitiveAddress(address2).build()).setDstMaskLength((short) mask2).build();
 743         LispIpv4Address rloc = LispAFIConvertor.asIPAfiAddress("20.0.0.2");
 744
 745         // Insert mapping in the database
 746         MapRegisterBuilder mapRegister = new MapRegisterBuilder();
 747         EidToLocatorRecordBuilder etlr = new EidToLocatorRecordBuilder();
 748         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(sourceDestAddress));
 749         etlr.setMaskLength((short) mask1);
 750         etlr.setRecordTtl(254);
 751         etlr.setAuthoritative(false);
 752         etlr.setAction(Action.NoAction);
 753         LocatorRecordBuilder record = new LocatorRecordBuilder();
 754         record.setLispAddressContainer(LispAFIConvertor.toContainer(rloc));
 755         record.setRouted(true);
 756         record.setRlocProbed(false);
 757         record.setLocalLocator(false);
 758         record.setPriority((short) 1);
 759         record.setWeight((short) 50);
 760         record.setMulticastPriority((short) 1);
 761         record.setMulticastWeight((short) 1);
 762         etlr.setLocatorRecord(new ArrayList<LocatorRecord>());
 763         etlr.getLocatorRecord().add(record.build());
 764         mapRegister.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
 765         mapRegister.getEidToLocatorRecord().add(etlr.build());
 766         lms.handleMapRegister(mapRegister.build(), false);
 767
 768         // Get mapping using NB interface. No IID used
 769         URL url = createGetMappingSourceDestURL(address1.getIpv4Address().getAfi(),
 770                 address1.getIpv4Address().getIpv4Address().getValue(),
 771                 mask1,
 772                 address2.getIpv4Address().getIpv4Address().getValue(),
 773                 mask2);
 774         String reply = callURL("GET", null, "application/json", null, url);
 775         JSONTokener jt = new JSONTokener(reply);
 776         JSONObject json = new JSONObject(jt);
 777
 778         // With just one locator, locators is not a JSONArray
 779         String rlocRetrieved = json.getJSONArray("locators").getJSONObject(0).getJSONObject("locatorGeneric").getString("ipAddress");
 780
 781         assertEquals(rloc.getIpv4Address().getValue(), rlocRetrieved);
 782
 783     }
 784
 785     private URL createGetKeyIPv4URL(LispIpv4Address address, int mask) throws MalformedURLException {
 786         String restUrl = String.format("http://localhost:8080/lispflowmapping/nb/v2/default/%s/0/%d/%s/%d", "key", address.getAfi().shortValue(),
 787                 address.getIpv4Address().getValue(), mask);
 788         URL url = new URL(restUrl);
 789         return url;
 790     }
 791
 792     private URL createGetKeySourceDestURL(int afi, String srcAddress, int srcMask, String dstAddress, int dstMask) throws MalformedURLException {
 793         String restUrl = String.format("http://localhost:8080/lispflowmapping/nb/v2/default/%s/0/%d/%s/%d/%s/%d", "key", afi, srcAddress, srcMask,
 794                 dstAddress, dstMask);
 795         URL url = new URL(restUrl);
 796         return url;
 797     }
 798
 799     private URL createGetMappingSourceDestURL(int afi, String srcAddress, int srcMask, String dstAddress, int dstMask) throws MalformedURLException {
 800         String restUrl = String.format("http://localhost:8080/lispflowmapping/nb/v2/default/%s/0/%d/%s/%d/%s/%d", "mapping", afi, srcAddress,
 801                 srcMask, dstAddress, dstMask);
 802         URL url = new URL(restUrl);
 803         return url;
 804     }
 805
 806     private URL createGetMappingIPv4URL(int iid, LispIpv4Address address, int mask) throws MalformedURLException {
 807         String restUrl = String.format("http://localhost:8080/lispflowmapping/nb/v2/default/%s/%d/%d/%s/%d", "mapping", iid, address.getAfi()
 808                 .shortValue(), address.getIpv4Address().getValue(), mask);
 809         URL url = new URL(restUrl);
 810         return url;
 811     }
 812
 813     private URL createDeleteMappingIPv4URL(int iid, LispIpv4Address address, int mask) throws MalformedURLException {
 814         String restUrl = String.format("http://localhost:8080/lispflowmapping/nb/v2/default/%s/%d/%d/%s/%d", "mapping", iid, address.getAfi()
 815                 .shortValue(), address.getIpv4Address().getValue(), mask);
 816         URL url = new URL(restUrl);
 817         return url;
 818     }
 819
 820     private URL createPutURL(String resource) throws MalformedURLException {
 821
 822         String restUrl = String.format("http://localhost:8080/lispflowmapping/nb/v2/default/%s", resource);
 823
 824         URL url = new URL(restUrl);
 825         return url;
 826     }
 827
 828     private String createAuthenticationString() {
 829         String authString = "admin:admin";
 830         byte[] authEncBytes = Base64.encodeBase64(authString.getBytes());
 831         String authStringEnc = new String(authEncBytes);
 832         return authStringEnc;
 833     }
 834
 835     private String callURL(String method, String content, String accept, String body, URL url) throws IOException, JSONException {
 836         String authStringEnc = createAuthenticationString();
 837         connection = (HttpURLConnection) url.openConnection();
 838         connection.setRequestMethod(method);
 839         connection.setRequestProperty("Authorization", "Basic " + authStringEnc);
 840         if (content != null) {
 841             connection.setRequestProperty("Content-Type", content);
 842         }
 843         if (accept != null) {
 844             connection.setRequestProperty("Accept", accept);
 845         }
 846         if (body != null) {
 847             // now add the request body
 848             connection.setDoOutput(true);
 849             OutputStreamWriter wr = new OutputStreamWriter(connection.getOutputStream());
 850             wr.write(body);
 851             wr.flush();
 852         }
 853         connection.connect();
 854
 855         // getting the result, first check response code
 856         Integer httpResponseCode = connection.getResponseCode();
 857
 858         if (httpResponseCode > 299) {
 859             LOG.trace("HTTP Address: " + url);
 860             LOG.trace("HTTP Response Code: " + httpResponseCode);
 861             fail();
 862         }
 863
 864         InputStream is = connection.getInputStream();
 865         BufferedReader rd = new BufferedReader(new InputStreamReader(is, Charset.forName("UTF-8")));
 866         StringBuilder sb = new StringBuilder();
 867         int cp;
 868         while ((cp = rd.read()) != -1) {
 869             sb.append((char) cp);
 870         }
 871         is.close();
 872         connection.disconnect();
 873         return (sb.toString());
 874     }
 875
 876     // timePeriod - in ms
 877     public void assertNoPacketReceived(int timePeriod) {
 878         try {
 879             receivePacket(timePeriod);
 880             // If didn't timeout then fail:
 881             fail();
 882         } catch (SocketTimeoutException ste) {
 883         }
 884     }
 885
 886     // ------------------------------- Mask Tests ---------------------------
 887
 888     public void eidPrefixLookupIPv4() throws SocketTimeoutException {
 889         cleanUP();
 890         runPrefixTest(LispAFIConvertor.asIPAfiAddress("1.2.3.4"), 16, LispAFIConvertor.asIPAfiAddress("1.2.3.2"),
 891                 LispAFIConvertor.asIPAfiAddress("1.1.1.1"), (byte) 32);
 892     }
 893
 894     public void eidPrefixLookupIPv6() throws SocketTimeoutException {
 895         cleanUP();
 896         runPrefixTest(LispAFIConvertor.asIPv6AfiAddress("1:2:3:4:5:6:7:8"), 64, LispAFIConvertor.asIPv6AfiAddress("1:2:3:4:5:1:2:3"),
 897                 LispAFIConvertor.asIPv6AfiAddress("1:2:3:1:2:3:1:2"), 128);
 898     }
 899
 900     private void runPrefixTest(LispAFIAddress registerEID, int registerdMask, LispAFIAddress matchedAddress, LispAFIAddress unMatchedAddress, int mask)
 901             throws SocketTimeoutException {
 902
 903         MapRegisterBuilder mapRegister = new MapRegisterBuilder();
 904         mapRegister.setWantMapNotify(true);
 905         mapRegister.setNonce((long) 8);
 906         mapRegister.setWantMapNotify(true);
 907         mapRegister.setKeyId((short) 0);
 908         mapRegister.setAuthenticationData(new byte[0]);
 909         mapRegister.setNonce((long) 8);
 910         mapRegister.setProxyMapReply(false);
 911         EidToLocatorRecordBuilder etlr = new EidToLocatorRecordBuilder();
 912         etlr.setRecordTtl(254);
 913         etlr.setAction(Action.NoAction);
 914         etlr.setAuthoritative(false);
 915         etlr.setMapVersion((short) 0);
 916         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(registerEID));
 917         etlr.setMaskLength((short) registerdMask);
 918         etlr.setRecordTtl(254);
 919         LocatorRecordBuilder record = new LocatorRecordBuilder();
 920         record.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
 921         record.setLocalLocator(false);
 922         record.setRlocProbed(false);
 923         record.setRouted(true);
 924         record.setMulticastPriority((short) 0);
 925         record.setMulticastWeight((short) 0);
 926         record.setPriority((short) 0);
 927         record.setWeight((short) 0);
 928         etlr.setLocatorRecord(new ArrayList<LocatorRecord>());
 929         etlr.getLocatorRecord().add(record.build());
 930         mapRegister.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
 931         mapRegister.getEidToLocatorRecord().add(etlr.build());
 932         sendMapRegister(mapRegister.build());
 933         MapNotify mapNotify = receiveMapNotify();
 934         assertEquals(8, mapNotify.getNonce().longValue());
 935         MapRequestBuilder mapRequest = new MapRequestBuilder();
 936         mapRequest.setNonce((long) 4);
 937         mapRequest.setSourceEid(new SourceEidBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
 938         mapRequest.setEidRecord(new ArrayList<EidRecord>());
 939         mapRequest.setAuthoritative(false);
 940         mapRequest.setMapDataPresent(false);
 941         mapRequest.setPitr(false);
 942         mapRequest.setProbe(false);
 943         mapRequest.setSmr(false);
 944         mapRequest.setSmrInvoked(false);
 945         mapRequest.getEidRecord().add(
 946                 new EidRecordBuilder().setMask((short) mask).setLispAddressContainer(LispAFIConvertor.toContainer(matchedAddress)).build());
 947         mapRequest.setItrRloc(new ArrayList<ItrRloc>());
 948         mapRequest.getItrRloc().add(
 949                 new ItrRlocBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
 950         sendMapRequest(mapRequest.build());
 951         MapReply mapReply = receiveMapReply();
 952         assertEquals(4, mapReply.getNonce().longValue());
 953         assertEquals(record.getLispAddressContainer(), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0).getLispAddressContainer());
 954         mapRequest.setEidRecord(new ArrayList<EidRecord>());
 955         mapRequest.getEidRecord().add(
 956                 new EidRecordBuilder().setMask((short) mask).setLispAddressContainer(LispAFIConvertor.toContainer(unMatchedAddress)).build());
 957         sendMapRequest(mapRequest.build());
 958         mapReply = receiveMapReply();
 959         assertEquals(0, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
 960     }
 961
 962     // This registers an IP with a MapRegister, then adds a password via the
 963     // northbound REST API
 964     // and checks that the password works
 965     public void testPasswordExactMatch() throws Exception {
 966         cleanUP();
 967         String ipString = "10.0.0.1";
 968         LispIpv4Address address = LispAFIConvertor.asIPAfiAddress(ipString);
 969         int mask = 32;
 970         String pass = "pass";
 971
 972         URL url = createPutURL("key");
 973
 974         String jsonAuthData = createAuthKeyJSON(pass, address, mask);
 975
 976         LOG.trace("Sending this JSON to LISP server: \n" + jsonAuthData);
 977         LOG.trace("Address: " + address);
 978
 979         byte[] expectedSha = new byte[] { (byte) 146, (byte) 234, (byte) 52, (byte) 247, (byte) 186, (byte) 232, (byte) 31, (byte) 249, (byte) 87,
 980                 (byte) 73, (byte) 234, (byte) 54, (byte) 225, (byte) 160, (byte) 129, (byte) 251, (byte) 73, (byte) 53, (byte) 196, (byte) 62 };
 981
 982         byte[] zeros = new byte[20];
 983
 984         callURL("PUT", "application/json", "text/plain", jsonAuthData, url);
 985
 986         // build a MapRegister
 987         MapRegisterBuilder mapRegister = new MapRegisterBuilder();
 988         mapRegister.setWantMapNotify(true);
 989         mapRegister.setNonce((long) 8);
 990         EidToLocatorRecordBuilder etlr = new EidToLocatorRecordBuilder();
 991         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(address));
 992         etlr.setMaskLength((short) mask);
 993         etlr.setRecordTtl(254);
 994         LocatorRecordBuilder record = new LocatorRecordBuilder();
 995         record.setLispAddressContainer(LispAFIConvertor.toContainer(locatorEid));
 996         etlr.setLocatorRecord(new ArrayList<LocatorRecord>());
 997         etlr.getLocatorRecord().add(record.build());
 998         mapRegister.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
 999         mapRegister.getEidToLocatorRecord().add(etlr.build());
1000
1001         mapRegister.setKeyId((short) 1); // LispKeyIDEnum.SHA1.getKeyID()
1002         mapRegister.setAuthenticationData(zeros);
1003
1004         sendMapRegister(mapRegister.build());
1005         assertNoPacketReceived(3000);
1006
1007         mapRegister.setAuthenticationData(expectedSha);
1008
1009         sendMapRegister(mapRegister.build());
1010
1011         assertMapNotifyRecieved();
1012     }
1013
1014     public void testPasswordMaskMatch() throws Exception {
1015         cleanUP();
1016         LispIpv4Address addressInRange = LispAFIConvertor.asIPAfiAddress("10.20.30.40");
1017         LispIpv4Address addressOutOfRange = LispAFIConvertor.asIPAfiAddress("20.40.30.40");
1018         LispIpv4Address range = LispAFIConvertor.asIPAfiAddress("10.20.30.0");
1019
1020         int mask = 32;
1021         String pass = "pass";
1022
1023         URL url = createPutURL("key");
1024         String jsonAuthData = createAuthKeyJSON(pass, range, 8);
1025
1026         callURL("PUT", "application/json", "text/plain", jsonAuthData, url);
1027         // build a MapRegister
1028         MapRegisterBuilder mapRegister = new MapRegisterBuilder();
1029
1030         mapRegister.setWantMapNotify(true);
1031         mapRegister.setNonce((long) 8);
1032         EidToLocatorRecordBuilder etlr = new EidToLocatorRecordBuilder();
1033         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(addressInRange));
1034         etlr.setMaskLength((short) mask);
1035         etlr.setRecordTtl(254);
1036         LocatorRecordBuilder record = new LocatorRecordBuilder();
1037         record.setLispAddressContainer(LispAFIConvertor.toContainer(locatorEid));
1038         record.setLispAddressContainer(LispAFIConvertor.toContainer(locatorEid));
1039         etlr.setLocatorRecord(new ArrayList<LocatorRecord>());
1040         etlr.getLocatorRecord().add(record.build());
1041         mapRegister.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
1042         mapRegister.getEidToLocatorRecord().add(etlr.build());
1043
1044         mapRegister.setKeyId((short) 1); // LispKeyIDEnum.SHA1.getKeyID()
1045         mapRegister
1046                 .setAuthenticationData(new byte[] { -15, -52, 38, -94, 125, -111, -68, -79, 68, 6, 101, 45, -1, 47, -4, -67, -113, 104, -110, -71 });
1047
1048         sendMapRegister(mapRegister.build());
1049
1050         assertMapNotifyRecieved();
1051
1052         etlr.setLispAddressContainer(LispAFIConvertor.toContainer(addressOutOfRange));
1053         mapRegister
1054                 .setAuthenticationData(new byte[] { -54, 68, -58, -91, -23, 22, -88, -31, 113, 39, 115, 78, -68, -123, -71, -14, -99, 67, -23, -73 });
1055
1056         sendMapRegister(mapRegister.build());
1057         assertNoPacketReceived(3000);
1058     }
1059
1060     private MapReply registerAddressAndQuery(LispAFIAddress eid) throws SocketTimeoutException {
1061         return registerAddressAndQuery(eid, -1);
1062     }
1063
1064     // takes an address, packs it in a MapRegister and sends it
1065     private void registerAddress(LispAFIAddress eid, int maskLength) throws SocketTimeoutException {
1066         MapRegisterBuilder mapRegisterBuilder = new MapRegisterBuilder();
1067         mapRegisterBuilder.setWantMapNotify(true);
1068         mapRegisterBuilder.setKeyId((short) 0);
1069         mapRegisterBuilder.setAuthenticationData(new byte[0]);
1070         mapRegisterBuilder.setNonce((long) 8);
1071         mapRegisterBuilder.setProxyMapReply(false);
1072         EidToLocatorRecordBuilder etlrBuilder = new EidToLocatorRecordBuilder();
1073         etlrBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
1074         if (maskLength != -1) {
1075             etlrBuilder.setMaskLength((short) maskLength);
1076         } else {
1077             etlrBuilder.setMaskLength((short) 0);
1078         }
1079         etlrBuilder.setRecordTtl(254);
1080         etlrBuilder.setAction(Action.NoAction);
1081         etlrBuilder.setAuthoritative(false);
1082         etlrBuilder.setMapVersion((short) 0);
1083         LocatorRecordBuilder recordBuilder = new LocatorRecordBuilder();
1084         recordBuilder.setLocalLocator(false);
1085         recordBuilder.setRlocProbed(false);
1086         recordBuilder.setRouted(true);
1087         recordBuilder.setMulticastPriority((short) 0);
1088         recordBuilder.setMulticastWeight((short) 0);
1089         recordBuilder.setPriority((short) 0);
1090         recordBuilder.setWeight((short) 0);
1091         recordBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(locatorEid));
1092         etlrBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
1093         etlrBuilder.getLocatorRecord().add(recordBuilder.build());
1094         mapRegisterBuilder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
1095         mapRegisterBuilder.getEidToLocatorRecord().add(etlrBuilder.build());
1096         sendMapRegister(mapRegisterBuilder.build());
1097         MapNotify mapNotify = receiveMapNotify();
1098         assertEquals(8, mapNotify.getNonce().longValue());
1099     }
1100
1101     private MapReply queryForAddress(LispAFIAddress eid, String srcEid) throws SocketTimeoutException {
1102         MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
1103         mapRequestBuilder.setNonce((long) 4);
1104         mapRequestBuilder.setEidRecord(new ArrayList<EidRecord>());
1105         mapRequestBuilder.getEidRecord().add(
1106                 new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(eid)).build());
1107         mapRequestBuilder.setItrRloc(new ArrayList<ItrRloc>());
1108         if (srcEid != null) {
1109             mapRequestBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(LispAFIConvertor.
1110                     toContainer(LispAFIConvertor.asIPAfiAddress(srcEid))).build());
1111         } else {
1112             mapRequestBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(LispAFIConvertor.
1113                     toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
1114         }
1115         mapRequestBuilder.getItrRloc().add(
1116                 new ItrRlocBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
1117         mapRequestBuilder.setAuthoritative(false);
1118         mapRequestBuilder.setMapDataPresent(false);
1119         mapRequestBuilder.setPitr(false);
1120         mapRequestBuilder.setProbe(false);
1121         mapRequestBuilder.setSmr(false);
1122         mapRequestBuilder.setSmrInvoked(false);
1123         sendMapRequest(mapRequestBuilder.build());
1124         return receiveMapReply();
1125     }
1126
1127     // takes an address, packs it in a MapRegister, sends it, returns the
1128     // MapReply
1129     private MapReply registerAddressAndQuery(LispAFIAddress eid, int maskLength) throws SocketTimeoutException {
1130         MapRegisterBuilder mapRegisterBuilder = new MapRegisterBuilder();
1131         mapRegisterBuilder.setWantMapNotify(true);
1132         mapRegisterBuilder.setKeyId((short) 0);
1133         mapRegisterBuilder.setAuthenticationData(new byte[0]);
1134         mapRegisterBuilder.setNonce((long) 8);
1135         mapRegisterBuilder.setProxyMapReply(false);
1136         EidToLocatorRecordBuilder etlrBuilder = new EidToLocatorRecordBuilder();
1137         etlrBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
1138         if (maskLength != -1) {
1139             etlrBuilder.setMaskLength((short) maskLength);
1140         } else {
1141             etlrBuilder.setMaskLength((short) 0);
1142         }
1143         etlrBuilder.setRecordTtl(254);
1144         etlrBuilder.setAction(Action.NoAction);
1145         etlrBuilder.setAuthoritative(false);
1146         etlrBuilder.setMapVersion((short) 0);
1147         LocatorRecordBuilder recordBuilder = new LocatorRecordBuilder();
1148         recordBuilder.setLocalLocator(false);
1149         recordBuilder.setRlocProbed(false);
1150         recordBuilder.setRouted(true);
1151         recordBuilder.setMulticastPriority((short) 0);
1152         recordBuilder.setMulticastWeight((short) 0);
1153         recordBuilder.setPriority((short) 0);
1154         recordBuilder.setWeight((short) 0);
1155         recordBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(locatorEid));
1156         etlrBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
1157         etlrBuilder.getLocatorRecord().add(recordBuilder.build());
1158         mapRegisterBuilder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
1159         mapRegisterBuilder.getEidToLocatorRecord().add(etlrBuilder.build());
1160         sendMapRegister(mapRegisterBuilder.build());
1161         MapNotify mapNotify = receiveMapNotify();
1162         assertEquals(8, mapNotify.getNonce().longValue());
1163         MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
1164         mapRequestBuilder.setNonce((long) 4);
1165         mapRequestBuilder.setEidRecord(new ArrayList<EidRecord>());
1166         mapRequestBuilder.getEidRecord().add(
1167                 new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(eid)).build());
1168         mapRequestBuilder.setItrRloc(new ArrayList<ItrRloc>());
1169         mapRequestBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress)))
1170                 .build());
1171         mapRequestBuilder.getItrRloc().add(
1172                 new ItrRlocBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
1173         mapRequestBuilder.setAuthoritative(false);
1174         mapRequestBuilder.setMapDataPresent(false);
1175         mapRequestBuilder.setPitr(false);
1176         mapRequestBuilder.setProbe(false);
1177         mapRequestBuilder.setSmr(false);
1178         mapRequestBuilder.setSmrInvoked(false);
1179         sendMapRequest(mapRequestBuilder.build());
1180         return receiveMapReply();
1181     }
1182
1183     // ------------------------------- LCAF Tests ---------------------------
1184
1185     @Test
1186     public void registerAndQuery__SrcDestLCAF() throws SocketTimeoutException {
1187         cleanUP();
1188         String ipString = "10.20.30.200";
1189         String macString = "01:02:03:04:05:06";
1190         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 addrToSend1 = LispAFIConvertor.asPrimitiveIPAfiAddress(ipString);
1191         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Mac addrToSend2 = LispAFIConvertor.asPrimitiveMacAfiAddress(macString);
1192         LcafSourceDestAddrBuilder builder = new LcafSourceDestAddrBuilder();
1193         builder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode());
1194         builder.setLcafType((short) LispCanonicalAddressFormatEnum.SOURCE_DEST.getLispCode());
1195         builder.setSrcMaskLength((short) 32);
1196         builder.setDstMaskLength((short) 0);
1197         builder.setSrcAddress(new SrcAddressBuilder().setPrimitiveAddress(addrToSend1).build());
1198         builder.setDstAddress(new DstAddressBuilder().setPrimitiveAddress(addrToSend2).build());
1199
1200         MapReply reply = registerAddressAndQuery(builder.build());
1201
1202         LispAddressContainer fromNetwork = reply.getEidToLocatorRecord().get(0).getLispAddressContainer();
1203         assertTrue(fromNetwork.getAddress() instanceof LcafSourceDest);
1204         LcafSourceDest sourceDestFromNetwork = (LcafSourceDest) fromNetwork.getAddress();
1205
1206         LispAFIAddress receivedAddr1 = LispAFIConvertor.toAFIfromPrimitive(sourceDestFromNetwork.getLcafSourceDestAddr().getSrcAddress().getPrimitiveAddress());
1207         LispAFIAddress receivedAddr2 = LispAFIConvertor.toAFIfromPrimitive(sourceDestFromNetwork.getLcafSourceDestAddr().getDstAddress().getPrimitiveAddress());
1208
1209         assertTrue(receivedAddr1 instanceof LispIpv4Address);
1210         assertTrue(receivedAddr2 instanceof LispMacAddress);
1211
1212         LispIpv4Address receivedIP = (LispIpv4Address) receivedAddr1;
1213         LispMacAddress receivedMAC = (LispMacAddress) receivedAddr2;
1214
1215         assertEquals(ipString, receivedIP.getIpv4Address().getValue());
1216         assertEquals(macString, receivedMAC.getMacAddress().getValue());
1217     }
1218
1219     @Test
1220     public void registerAndQuery__SrcDestLCAFOverlap() throws SocketTimeoutException {
1221         cleanUP();
1222         String ipString1 = "10.10.10.0";
1223         String ipString2 = "20.20.20.0";
1224         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 addrToSend1 = LispAFIConvertor.asPrimitiveIPAfiAddress(ipString1);
1225         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 addrToSend2 = LispAFIConvertor.asPrimitiveIPAfiAddress(ipString2);
1226         LcafSourceDestAddrBuilder builder = new LcafSourceDestAddrBuilder();
1227         builder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode());
1228         builder.setLcafType((short) LispCanonicalAddressFormatEnum.SOURCE_DEST.getLispCode());
1229         builder.setSrcMaskLength((short) 24);
1230         builder.setDstMaskLength((short) 24);
1231         builder.setSrcAddress(new SrcAddressBuilder().setPrimitiveAddress(addrToSend1).build());
1232         builder.setDstAddress(new DstAddressBuilder().setPrimitiveAddress(addrToSend2).build());
1233
1234         LcafSourceDestAddr srcDst = builder.build();
1235         registerAddress(LispAFIConvertor.asIPAfiAddress(ipString2), 24);
1236         registerAddress(srcDst, -1);
1237
1238         // exact match
1239         MapReply reply = queryForAddress(srcDst, null);
1240
1241         LispAddressContainer fromNetwork = reply.getEidToLocatorRecord().get(0).getLispAddressContainer();
1242         assertTrue(fromNetwork.getAddress() instanceof LcafSourceDest);
1243         LcafSourceDest sourceDestFromNetwork = (LcafSourceDest) fromNetwork.getAddress();
1244
1245         LispAFIAddress receivedAddr1 = LispAFIConvertor.toAFIfromPrimitive(sourceDestFromNetwork.getLcafSourceDestAddr().getSrcAddress().getPrimitiveAddress());
1246         LispAFIAddress receivedAddr2 = LispAFIConvertor.toAFIfromPrimitive(sourceDestFromNetwork.getLcafSourceDestAddr().getDstAddress().getPrimitiveAddress());
1247
1248         assertTrue(receivedAddr1 instanceof LispIpv4Address);
1249         assertTrue(receivedAddr2 instanceof LispIpv4Address);
1250
1251         LispIpv4Address receivedIP1 = (LispIpv4Address) receivedAddr1;
1252         LispIpv4Address receivedIP2 = (LispIpv4Address) receivedAddr2;
1253
1254         assertEquals(ipString1, receivedIP1.getIpv4Address().getValue());
1255         assertEquals(ipString2, receivedIP2.getIpv4Address().getValue());
1256
1257         // srcEid/dstEid match
1258         reply = queryForAddress(LispAFIConvertor.asIPAfiAddress("20.20.20.1"), "10.10.10.1");
1259         fromNetwork = reply.getEidToLocatorRecord().get(0).getLispAddressContainer();
1260         assertTrue(fromNetwork.getAddress() instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.Ipv4);
1261
1262         Ipv4Address ipAddr2 = ((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.Ipv4) fromNetwork.getAddress()).getIpv4Address();
1263         assertEquals(ipString2, ipAddr2.getIpv4Address().getValue());
1264
1265         // dstEid match only
1266         reply = queryForAddress(LispAFIConvertor.asIPAfiAddress("20.20.20.1"), "1.2.3.4");
1267         fromNetwork = reply.getEidToLocatorRecord().get(0).getLispAddressContainer();
1268         assertTrue(fromNetwork.getAddress() instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.Ipv4);
1269
1270         ipAddr2 = ((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.Ipv4) fromNetwork.getAddress()).getIpv4Address();
1271         assertEquals(ipString2, ipAddr2.getIpv4Address().getValue());
1272     }
1273
1274     @Test
1275     public void registerAndQuery__KeyValueLCAF() throws SocketTimeoutException {
1276         cleanUP();
1277         String ipString = "10.20.30.200";
1278         String macString = "01:02:03:04:05:06";
1279         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4 addrToSend1 = LispAFIConvertor.asPrimitiveIPAfiAddress(ipString);
1280         org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Mac addrToSend2 = LispAFIConvertor.asPrimitiveMacAfiAddress(macString);
1281         LcafKeyValueAddressAddrBuilder builder = new LcafKeyValueAddressAddrBuilder();
1282         builder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode());
1283         builder.setLcafType((short) LispCanonicalAddressFormatEnum.KEY_VALUE.getLispCode());
1284         builder.setKey(new KeyBuilder().setPrimitiveAddress(addrToSend1).build());
1285         builder.setValue(new ValueBuilder().setPrimitiveAddress(addrToSend2).build());
1286
1287         MapReply reply = registerAddressAndQuery(builder.build());
1288
1289         LispAddressContainer fromNetwork = reply.getEidToLocatorRecord().get(0).getLispAddressContainer();
1290         assertTrue(fromNetwork.getAddress() instanceof LcafKeyValue);
1291         LcafKeyValue keyValueFromNetwork = (LcafKeyValue) fromNetwork.getAddress();
1292
1293         LispAFIAddress receivedAddr1 = LispAFIConvertor.toAFIfromPrimitive(keyValueFromNetwork.getLcafKeyValueAddressAddr().getKey().getPrimitiveAddress());
1294         LispAFIAddress receivedAddr2 = LispAFIConvertor.toAFIfromPrimitive(keyValueFromNetwork.getLcafKeyValueAddressAddr().getValue().getPrimitiveAddress());
1295
1296         assertTrue(receivedAddr1 instanceof LispIpv4Address);
1297         assertTrue(receivedAddr2 instanceof LispMacAddress);
1298
1299         LispIpv4Address receivedIP = (LispIpv4Address) receivedAddr1;
1300         LispMacAddress receivedMAC = (LispMacAddress) receivedAddr2;
1301
1302         assertEquals(ipString, receivedIP.getIpv4Address().getValue());
1303         assertEquals(macString, receivedMAC.getMacAddress().getValue());
1304     }
1305
1306     public void registerAndQuery__ListLCAF() throws SocketTimeoutException {
1307         cleanUP();
1308         String macString = "01:02:03:04:05:06";
1309         String ipString = "10.20.255.30";
1310         LcafListAddrBuilder listbuilder = new LcafListAddrBuilder();
1311         listbuilder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode()).setLcafType((short) LispCanonicalAddressFormatEnum.LIST.getLispCode());
1312         listbuilder.setAddresses(new ArrayList<Addresses>());
1313         listbuilder.getAddresses().add(new AddressesBuilder().setPrimitiveAddress(LispAFIConvertor.toPrimitive(LispAFIConvertor.asIPAfiAddress(ipString))).build());
1314         listbuilder.getAddresses().add(new AddressesBuilder().setPrimitiveAddress(LispAFIConvertor.toPrimitive(LispAFIConvertor.asMacAfiAddress(macString))).build());
1315
1316         MapReply reply = registerAddressAndQuery(listbuilder.build());
1317
1318         LispAFIAddress receivedAddress = LispAFIConvertor.toAFI(reply.getEidToLocatorRecord().get(0).getLispAddressContainer());
1319
1320         assertTrue(receivedAddress instanceof LcafListAddress);
1321
1322         LcafListAddress listAddrFromNetwork = (LcafListAddress) receivedAddress;
1323         LispAFIAddress receivedAddr1 = LispAFIConvertor.toAFIfromPrimitive(listAddrFromNetwork.getAddresses().get(0).getPrimitiveAddress());
1324         LispAFIAddress receivedAddr2 = LispAFIConvertor.toAFIfromPrimitive(listAddrFromNetwork.getAddresses().get(1).getPrimitiveAddress());
1325
1326         assertTrue(receivedAddr1 instanceof LispIpv4Address);
1327         assertTrue(receivedAddr2 instanceof LispMacAddress);
1328
1329         assertEquals(macString, ((LispMacAddress) receivedAddr2).getMacAddress().getValue());
1330         assertEquals(ipString, ((LispIpv4Address) receivedAddr1).getIpv4Address().getValue());
1331     }
1332
1333     public void registerAndQuery__SegmentLCAF() throws SocketTimeoutException {
1334         cleanUP();
1335         String ipString = "10.20.255.30";
1336         int instanceId = 6;
1337         LcafSegmentAddrBuilder builder = new LcafSegmentAddrBuilder();
1338         builder.setInstanceId((long) instanceId);
1339         builder.setIidMaskLength((short) 32);
1340         builder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode()).setLcafType((short) LispCanonicalAddressFormatEnum.SEGMENT.getLispCode());
1341         builder.setAddress(new AddressBuilder().setPrimitiveAddress(LispAFIConvertor.toPrimitive(LispAFIConvertor.asIPAfiAddress(ipString))).build());
1342
1343         MapReply reply = registerAddressAndQuery(builder.build());
1344
1345         LispAFIAddress receivedAddress = LispAFIConvertor.toAFI(reply.getEidToLocatorRecord().get(0).getLispAddressContainer());
1346         assertTrue(receivedAddress instanceof LcafSegmentAddress);
1347
1348         LcafSegmentAddress segmentfromNetwork = (LcafSegmentAddress) receivedAddress;
1349         LispAFIAddress addrFromSegment = LispAFIConvertor.toAFIfromPrimitive(segmentfromNetwork.getAddress().getPrimitiveAddress());
1350         assertTrue(addrFromSegment instanceof LispIpv4Address);
1351         assertEquals(ipString, ((LispIpv4Address) addrFromSegment).getIpv4Address().getValue());
1352
1353         assertEquals(instanceId, segmentfromNetwork.getInstanceId().intValue());
1354     }
1355
1356     public void registerAndQuery__TrafficEngineering() throws SocketTimeoutException {
1357         cleanUP();
1358         String macString = "01:02:03:04:05:06";
1359         String ipString = "10.20.255.30";
1360         HopBuilder hopBuilder = new HopBuilder();
1361         hopBuilder.setPrimitiveAddress(LispAFIConvertor.toPrimitive(LispAFIConvertor.asIPAfiAddress(ipString)));
1362         Hop hop1 = hopBuilder.build();
1363         hopBuilder.setPrimitiveAddress(LispAFIConvertor.toPrimitive(LispAFIConvertor.asMacAfiAddress(macString)));
1364         Hop hop2 = hopBuilder.build();
1365         HopsBuilder hb = new HopsBuilder();
1366         hb.setHop(hop1);
1367         hb.setLookup(true);
1368         hb.setRLOCProbe(false);
1369         hb.setStrict(true);
1370         HopsBuilder hb2 = new HopsBuilder();
1371         hb2.setHop(hop2);
1372         hb2.setLookup(false);
1373         hb2.setRLOCProbe(true);
1374         hb2.setStrict(false);
1375         Hops hops1 = hb.build();
1376         Hops hops2 = hb2.build();
1377         LcafTrafficEngineeringAddrBuilder trafficBuilder = new LcafTrafficEngineeringAddrBuilder();
1378         trafficBuilder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode()).setLcafType(
1379                 (short) LispCanonicalAddressFormatEnum.TRAFFIC_ENGINEERING.getLispCode());
1380         trafficBuilder.setHops(new ArrayList<Hops>());
1381         trafficBuilder.getHops().add(hb.build());
1382         trafficBuilder.getHops().add(hb2.build());
1383
1384         MapReply reply = registerAddressAndQuery(trafficBuilder.build());
1385
1386         assertTrue(LispAFIConvertor.toAFI(reply.getEidToLocatorRecord().get(0).getLispAddressContainer()) instanceof LcafTrafficEngineeringAddress);
1387
1388         LcafTrafficEngineeringAddress receivedAddress = (LcafTrafficEngineeringAddress) LispAFIConvertor.toAFI(reply.getEidToLocatorRecord().get(0)
1389                 .getLispAddressContainer());
1390
1391         ReencapHop receivedHop1 = receivedAddress.getHops().get(0);
1392         ReencapHop receivedHop2 = receivedAddress.getHops().get(1);
1393
1394         assertEquals(true, hops1.isLookup());
1395         assertEquals(false, hops1.isRLOCProbe());
1396         assertEquals(true, hops1.isStrict());
1397
1398         assertEquals(false, hops2.isLookup());
1399         assertEquals(true, hops2.isRLOCProbe());
1400         assertEquals(false, hops2.isStrict());
1401
1402         assertTrue(receivedHop1.getHop().getPrimitiveAddress() instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4);
1403         assertTrue(receivedHop2.getHop().getPrimitiveAddress() instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Mac);
1404
1405         assertEquals(ipString, ((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) receivedHop1.getHop().getPrimitiveAddress()).getIpv4Address().getIpv4Address().getValue());
1406         assertEquals(macString, ((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Mac) receivedHop2.getHop().getPrimitiveAddress()).getMacAddress().getMacAddress().getValue());
1407     }
1408
1409     public void registerAndQuery__ApplicationData() throws SocketTimeoutException {
1410         cleanUP();
1411         String ipString = "1.2.3.4";
1412         short protocol = 1;
1413         int ipTOs = 2;
1414         int localPortLow = 3;
1415         int localPortHigh = 4;
1416         int remotePortLow = 4;
1417         int remotePortHigh = 5;
1418
1419         LcafApplicationDataAddrBuilder builder = new LcafApplicationDataAddrBuilder();
1420         builder.setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode()).setLcafType((short) LispCanonicalAddressFormatEnum.APPLICATION_DATA.getLispCode());
1421         builder.setIpTos(ipTOs);
1422         builder.setProtocol(protocol);
1423         builder.setLocalPortLow(new PortNumber(localPortLow));
1424         builder.setLocalPortHigh(new PortNumber(localPortHigh));
1425         builder.setRemotePortLow(new PortNumber(remotePortLow));
1426         builder.setRemotePortHigh(new PortNumber(remotePortHigh));
1427         builder.setAddress(new org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafapplicationdataaddress.AddressBuilder()
1428                 .setPrimitiveAddress(LispAFIConvertor.toPrimitive(LispAFIConvertor.asIPAfiAddress(ipString))).build());
1429
1430         LcafApplicationDataAddress addressToSend = builder.build();
1431
1432         MapReply reply = registerAddressAndQuery(addressToSend);
1433
1434         LispAFIAddress receivedAddress = LispAFIConvertor.toAFI(reply.getEidToLocatorRecord().get(0).getLispAddressContainer());
1435
1436         assertTrue(receivedAddress instanceof LcafApplicationDataAddress);
1437
1438         LcafApplicationDataAddress receivedApplicationDataAddress = (LcafApplicationDataAddress) receivedAddress;
1439         assertEquals(protocol, receivedApplicationDataAddress.getProtocol().intValue());
1440         assertEquals(ipTOs, receivedApplicationDataAddress.getIpTos().intValue());
1441         assertEquals(localPortLow, receivedApplicationDataAddress.getLocalPortLow().getValue().intValue());
1442         assertEquals(localPortHigh, receivedApplicationDataAddress.getLocalPortHigh().getValue().intValue());
1443         assertEquals(remotePortLow, receivedApplicationDataAddress.getRemotePortLow().getValue().intValue());
1444         assertEquals(remotePortHigh, receivedApplicationDataAddress.getRemotePortHigh().getValue().intValue());
1445
1446         LispIpv4Address ipAddressReceived = (LispIpv4Address) LispAFIConvertor.toAFIfromPrimitive(receivedApplicationDataAddress.getAddress().getPrimitiveAddress());
1447         assertEquals(ipString, ipAddressReceived.getIpv4Address().getValue());
1448     }
1449
1450     // ------------------- TimeOut Tests -----------
1451
1452     public void mapRequestMapRegisterAndMapRequestTestTimeout() throws SocketTimeoutException {
1453         cleanUP();
1454         LispIpv4Address eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
1455         MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
1456         mapRequestBuilder.setNonce((long) 4);
1457         mapRequestBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(
1458                 LispAFIConvertor.toContainer(new NoAddressBuilder().setAfi((short) 0).build())).build());
1459         mapRequestBuilder.setEidRecord(new ArrayList<EidRecord>());
1460         mapRequestBuilder.getEidRecord().add(
1461                 new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(eid)).build());
1462         mapRequestBuilder.setItrRloc(new ArrayList<ItrRloc>());
1463         mapRequestBuilder.getItrRloc().add(
1464                 new ItrRlocBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
1465         sendMapRequest(mapRequestBuilder.build());
1466         MapReply mapReply = receiveMapReply();
1467         assertEquals(4, mapReply.getNonce().longValue());
1468         assertEquals(0, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
1469         MapRegisterBuilder mapRegisterbuilder = new MapRegisterBuilder();
1470         mapRegisterbuilder.setWantMapNotify(true);
1471         mapRegisterbuilder.setNonce((long) 8);
1472         EidToLocatorRecordBuilder etlrBuilder = new EidToLocatorRecordBuilder();
1473         etlrBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
1474         etlrBuilder.setMaskLength((short) 32);
1475         etlrBuilder.setRecordTtl(254);
1476         LocatorRecordBuilder recordBuilder = new LocatorRecordBuilder();
1477         recordBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")));
1478         etlrBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
1479         etlrBuilder.getLocatorRecord().add(recordBuilder.build());
1480         mapRegisterbuilder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
1481         mapRegisterbuilder.getEidToLocatorRecord().add(etlrBuilder.build());
1482         sendMapRegister(mapRegisterbuilder.build());
1483         MapNotify mapNotify = receiveMapNotify();
1484         assertEquals(8, mapNotify.getNonce().longValue());
1485         sendMapRequest(mapRequestBuilder.build());
1486         mapReply = receiveMapReply();
1487         assertEquals(4, mapReply.getNonce().longValue());
1488         assertEquals(recordBuilder.getLispAddressContainer(), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0)
1489                 .getLispAddressContainer());
1490         causeEntryToBeCleaned();
1491         sendMapRequest(mapRequestBuilder.build());
1492         mapReply = receiveMapReply();
1493         assertEquals(0, mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().size());
1494     }
1495
1496     public void mapRequestMapRegisterAndMapRequestTestNativelyForwardTimeoutResponse() throws Exception {
1497         cleanUP();
1498         LispIpv4Address eid = LispAFIConvertor.asIPAfiAddress("1.2.3.4");
1499         MapRequest mapRequest = createMapRequest(eid);
1500
1501         testTTLBeforeRegister(mapRequest);
1502
1503         registerForTTL(eid);
1504
1505         testTTLAfterRegister(mapRequest);
1506
1507         causeEntryToBeCleaned();
1508         testTTLAfterClean(mapRequest);
1509
1510         northboundAddKey();
1511         testTTLAfterAutherize(mapRequest);
1512
1513     }
1514
1515     private void testTTLAfterClean(MapRequest mapRequest) throws SocketTimeoutException {
1516         MapReply mapReply;
1517         sendMapRequest(mapRequest);
1518         mapReply = receiveMapReply();
1519         assertCorrectMapReplyTTLAndAction(mapReply, 15, Action.NativelyForward);
1520     }
1521
1522     private void causeEntryToBeCleaned() {
1523         // TODO for the time being, to keep master and stable/lithium in sync, we need to remove the forceful
1524         // expiration of DAO entries. Once we're past this, we'll have to expose methods to setTimeUnit(TimeUnit)
1525         // and cleanOld() (expired) entries in IFlowMapping (and perhaps ILispDAO) and use them here.
1526         this.lms.clean();
1527     }
1528
1529     private void testTTLAfterRegister(MapRequest mapRequest) throws SocketTimeoutException {
1530         MapReply mapReply;
1531         sendMapRequest(mapRequest);
1532         mapReply = receiveMapReply();
1533         assertEquals(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress("4.3.2.1")), mapReply.getEidToLocatorRecord().get(0).getLocatorRecord().get(0)
1534                 .getLispAddressContainer());
1535         assertCorrectMapReplyTTLAndAction(mapReply, 254, Action.NoAction);
1536     }
1537
1538     private void registerForTTL(LispIpv4Address eid) throws SocketTimeoutException {
1539         MapRegister mapRegister = createMapRegister(eid);
1540         sendMapRegister(mapRegister);
1541         assertMapNotifyRecieved();
1542     }
1543
1544     private void testTTLBeforeRegister(MapRequest mapRequest) throws SocketTimeoutException {
1545         MapReply mapReply;
1546         sendMapRequest(mapRequest);
1547         mapReply = receiveMapReply();
1548         assertCorrectMapReplyTTLAndAction(mapReply, 15, Action.NativelyForward);
1549     }
1550
1551     private void testTTLAfterAutherize(MapRequest mapRequest) throws SocketTimeoutException {
1552         MapReply mapReply;
1553         sendMapRequest(mapRequest);
1554         mapReply = receiveMapReply();
1555         assertCorrectMapReplyTTLAndAction(mapReply, 1, Action.NativelyForward);
1556     }
1557
1558     private void assertCorrectMapReplyTTLAndAction(MapReply mapReply, int expectedTTL, Action expectedAction) {
1559         assertEquals(expectedTTL, mapReply.getEidToLocatorRecord().get(0).getRecordTtl().intValue());
1560         assertEquals(expectedAction, mapReply.getEidToLocatorRecord().get(0).getAction());
1561     }
1562
1563     private MapRegister createMapRegister(LispAFIAddress eid, LispAFIAddress rloc) {
1564         MapRegisterBuilder mapRegisterbuilder = new MapRegisterBuilder();
1565         mapRegisterbuilder.setWantMapNotify(true);
1566         mapRegisterbuilder.setNonce((long) 8);
1567         mapRegisterbuilder.setKeyId((short) 0);
1568         EidToLocatorRecordBuilder etlrBuilder = new EidToLocatorRecordBuilder();
1569         etlrBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(eid));
1570         etlrBuilder.setMaskLength((short) 24);
1571         etlrBuilder.setRecordTtl(254);
1572         etlrBuilder.setAuthoritative(false);
1573         etlrBuilder.setAction(Action.NoAction);
1574         LocatorRecordBuilder recordBuilder = new LocatorRecordBuilder();
1575         recordBuilder.setLispAddressContainer(LispAFIConvertor.toContainer(rloc));
1576         etlrBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
1577         etlrBuilder.getLocatorRecord().add(recordBuilder.build());
1578         mapRegisterbuilder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
1579         mapRegisterbuilder.getEidToLocatorRecord().add(etlrBuilder.build());
1580         MapRegister mapRegister = mapRegisterbuilder.build();
1581         return mapRegister;
1582     }
1583
1584     private MapRegister createMapRegister(LispIpv4Address eid) {
1585         return createMapRegister(eid, LispAFIConvertor.asIPAfiAddress("4.3.2.1"));
1586     }
1587
1588     private MapRequest createMapRequest(LispAFIAddress eid) {
1589         MapRequestBuilder mapRequestBuilder = new MapRequestBuilder();
1590         mapRequestBuilder.setNonce((long) 4);
1591         mapRequestBuilder.setPitr(false);
1592         mapRequestBuilder.setSourceEid(new SourceEidBuilder().setLispAddressContainer(
1593                 LispAFIConvertor.toContainer(new NoAddressBuilder().setAfi((short) 0).build())).build());
1594         mapRequestBuilder.setEidRecord(new ArrayList<EidRecord>());
1595         mapRequestBuilder.getEidRecord().add(
1596                 new EidRecordBuilder().setMask((short) 32).setLispAddressContainer(LispAFIConvertor.toContainer(eid)).build());
1597         mapRequestBuilder.setItrRloc(new ArrayList<ItrRloc>());
1598         mapRequestBuilder.getItrRloc().add(
1599                 new ItrRlocBuilder().setLispAddressContainer(LispAFIConvertor.toContainer(LispAFIConvertor.asIPAfiAddress(ourAddress))).build());
1600         MapRequest mr = mapRequestBuilder.build();
1601         return mr;
1602     }
1603
1604     public void testSimpleNonProxy() throws SocketTimeoutException, SocketException {
1605         cleanUP();
1606         String rloc = "127.0.0.3";
1607         int port = LispMessage.PORT_NUM;
1608         LispAFIAddress ipRloc = LispAFIConvertor.asIPAfiAddress(rloc);
1609         sendProxyMapRequest(rloc, port, ipRloc);
1610
1611     }
1612
1613     public void testNonProxyOtherPort() throws SocketTimeoutException, SocketException {
1614         cleanUP();
1615         String rloc = "127.0.0.3";
1616         int port = 4350;
1617         LcafApplicationDataAddr adLcaf = new LcafApplicationDataAddrBuilder()
1618                 .setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode())
1619                 .setLcafType((short) LispCanonicalAddressFormatEnum.APPLICATION_DATA.getLispCode())
1620                 .setAddress(
1621                         new org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafapplicationdataaddress.AddressBuilder().setPrimitiveAddress(
1622                                 LispAFIConvertor.asPrimitiveIPAfiAddress(rloc)).build()).setLocalPortLow(new PortNumber(port)).build();
1623         LOG.info("testNonProxyOtherPort:" + adLcaf.toString());
1624         sendProxyMapRequest(rloc, port, adLcaf);
1625
1626     }
1627
1628     public void testRecievingNonProxyOnXtrPort() throws SocketTimeoutException, SocketException, Throwable {
1629         cleanUP();
1630         configLispPlugin.shouldListenOnXtrPort(true);
1631         notificationCalled = false;
1632         final String eid = "10.10.10.10";
1633         String rloc = "127.0.0.3";
1634         int port = LispMessage.XTR_PORT_NUM;
1635         LcafApplicationDataAddr adLcaf = new LcafApplicationDataAddrBuilder()
1636                 .setAfi(AddressFamilyNumberEnum.LCAF.getIanaCode())
1637                 .setLcafType((short) LispCanonicalAddressFormatEnum.APPLICATION_DATA.getLispCode())
1638                 .setAddress(
1639                         new org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lcafapplicationdataaddress.AddressBuilder().setPrimitiveAddress(
1640                                 LispAFIConvertor.asPrimitiveIPAfiAddress(rloc)).build()).setLocalPortLow(new PortNumber(port)).build();
1641         final MapRequest mapRequest = createNonProxyMapRequest(eid, adLcaf);
1642         ((LispMappingService) lms).registerNotificationListener(XtrRequestMapping.class, new NotificationListener<XtrRequestMapping>() {
1643
1644             @Override
1645             public void onNotification(XtrRequestMapping notification) {
1646                 assertEquals(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.Ipv4) mapRequest.getEidRecord().get(0).getLispAddressContainer().getAddress()).getIpv4Address().getIpv4Address().getValue(),
1647                         eid);
1648                 notificationCalled = true;
1649                 LOG.warn("notification arrived");
1650             }
1651         });
1652         sendMapRequest(mapRequest, port);
1653         for (int i = 0; i < MAX_NOTIFICATION_RETRYS; i++) {
1654             if (notificationCalled) {
1655                 return;
1656             } else {
1657                 LOG.warn("notification hasn't arrived, sleeping...");
1658                 Thread.sleep(500);
1659             }
1660         }
1661
1662         fail("Notification hasn't arrived");
1663
1664     }
1665
1666     private void sendProxyMapRequest(String rloc, int port, LispAFIAddress adLcaf) throws SocketTimeoutException, SocketException {
1667         String eid = "10.1.0.1";
1668         MapRequest mapRequest = createNonProxyMapRequest(eid, adLcaf);
1669         sendMapRequest(mapRequest);
1670         DatagramSocket nonProxySocket = new DatagramSocket(new InetSocketAddress(rloc, port));
1671         MapRequest recievedMapRequest = receiveMapRequest(nonProxySocket);
1672         assertEquals(mapRequest.getNonce(), recievedMapRequest.getNonce());
1673         assertEquals(mapRequest.getSourceEid(), recievedMapRequest.getSourceEid());
1674         assertEquals(mapRequest.getItrRloc(), recievedMapRequest.getItrRloc());
1675         assertEquals(mapRequest.getEidRecord(), recievedMapRequest.getEidRecord());
1676         nonProxySocket.close();
1677     }
1678
1679     private MapRequest createNonProxyMapRequest(String eid, LispAFIAddress adLcaf) throws SocketTimeoutException {
1680         MapRegister mr = createMapRegister(LispAFIConvertor.asIPAfiAddress(eid));
1681         LocatorRecord record = new LocatorRecordBuilder(mr.getEidToLocatorRecord().get(0).getLocatorRecord().get(0)).setLispAddressContainer(
1682                 LispAFIConvertor.toContainer(adLcaf)).build();
1683         mr.getEidToLocatorRecord().get(0).getLocatorRecord().set(0, record);
1684         sendMapRegister(mr);
1685         assertMapNotifyRecieved();
1686         MapRequest mapRequest = createMapRequest(LispAFIConvertor.asIPAfiAddress(eid));
1687         MapRequestBuilder builder = new MapRequestBuilder(mapRequest);
1688         builder.setPitr(true);
1689         mapRequest = builder.build();
1690         return mapRequest;
1691     }
1692
1693     private void assertMapNotifyRecieved() throws SocketTimeoutException {
1694         receiveMapNotify();
1695     }
1696
1697     private MapReply receiveMapReply() throws SocketTimeoutException {
1698         return MapReplySerializer.getInstance().deserialize(ByteBuffer.wrap(receivePacket().getData()));
1699     }
1700
1701     private MapRequest receiveMapRequest(DatagramSocket datagramSocket) throws SocketTimeoutException {
1702         return MapRequestSerializer.getInstance().deserialize(ByteBuffer.wrap(receivePacket(datagramSocket, 30000).getData()));
1703     }
1704
1705     private MapNotify receiveMapNotify() throws SocketTimeoutException {
1706         return MapNotifySerializer.getInstance().deserialize(ByteBuffer.wrap(receivePacket().getData()));
1707     }
1708
1709     private void sendMapRequest(MapRequest mapRequest) {
1710         sendMapRequest(mapRequest, LispMessage.PORT_NUM);
1711     }
1712
1713     private void sendMapRequest(MapRequest mapRequest, int port) {
1714         sendPacket(MapRequestSerializer.getInstance().serialize(mapRequest).array(), port);
1715     }
1716
1717     private void sendMapRegister(MapRegister mapRegister) {
1718         sendPacket(MapRegisterSerializer.getInstance().serialize(mapRegister).array());
1719     }
1720
1721     private void sendPacket(byte[] bytesToSend) {
1722         sendPacket(bytesToSend, LispMessage.PORT_NUM);
1723     }
1724
1725     private void sendPacket(byte[] bytesToSend, int port) {
1726         try {
1727             DatagramPacket packet = new DatagramPacket(bytesToSend, bytesToSend.length);
1728             initPacketAddress(packet, port);
1729             LOG.trace("Sending packet to LispPlugin on socket, port {}", port);
1730             socket.send(packet);
1731         } catch (Throwable t) {
1732             fail();
1733         }
1734     }
1735
1736     private DatagramPacket receivePacket() throws SocketTimeoutException {
1737         return receivePacket(6000);
1738     }
1739
1740     private DatagramPacket receivePacket(int timeout) throws SocketTimeoutException {
1741         return receivePacket(socket, timeout);
1742     }
1743
1744     private DatagramPacket receivePacket(DatagramSocket receivedSocket, int timeout) throws SocketTimeoutException {
1745         try {
1746             byte[] buffer = new byte[4096];
1747             DatagramPacket receivePacket = new DatagramPacket(buffer, buffer.length);
1748             LOG.trace("Waiting for packet from socket...");
1749             receivedSocket.setSoTimeout(timeout);
1750             receivedSocket.receive(receivePacket);
1751             LOG.trace("Recieved packet from socket!");
1752             return receivePacket;
1753         } catch (SocketTimeoutException ste) {
1754             throw ste;
1755         } catch (Throwable t) {
1756             fail();
1757             return null;
1758         }
1759     }
1760
1761     private void initPacketAddress(DatagramPacket packet, int port) throws UnknownHostException {
1762         packet.setAddress(InetAddress.getByName(lispBindAddress));
1763         packet.setPort(port);
1764     }
1765
1766     private DatagramSocket initSocket(DatagramSocket socket, int port) {
1767         try {
1768             socket = new DatagramSocket(new InetSocketAddress(ourAddress, port));
1769         } catch (SocketException e) {
1770             e.printStackTrace();
1771             fail();
1772         }
1773         return socket;
1774     }
1775
1776     private byte[] extractWSUdpByteArray(String wiresharkHex) {
1777         final int HEADER_LEN = 42;
1778         byte[] res = new byte[1000];
1779         String[] split = wiresharkHex.split(" ");
1780         int counter = 0;
1781         for (String cur : split) {
1782             cur = cur.trim();
1783             if (cur.length() == 2) {
1784                 ++counter;
1785                 if (counter > HEADER_LEN) {
1786                     res[counter - HEADER_LEN - 1] = (byte) Integer.parseInt(cur, 16);
1787                 }
1788
1789             }
1790         }
1791         return Arrays.copyOf(res, counter - HEADER_LEN);
1792     }
1793
1794     private String stateToString(int state) {
1795         switch (state) {
1796         case Bundle.ACTIVE:
1797             return "ACTIVE";
1798         case Bundle.INSTALLED:
1799             return "INSTALLED";
1800         case Bundle.RESOLVED:
1801             return "RESOLVED";
1802         case Bundle.UNINSTALLED:
1803             return "UNINSTALLED";
1804         default:
1805             return "Not CONVERTED";
1806         }
1807     }
1808
1809     @SuppressWarnings({ "rawtypes", "unchecked" })
1810     private void areWeReady() throws InvalidSyntaxException {
1811         try {
1812             Thread.sleep(5000);
1813         } catch (InterruptedException e) {
1814         }
1815
1816         assertNotNull(bc);
1817         boolean debugit = false;
1818         Bundle b[] = bc.getBundles();
1819         for (Bundle element : b) {
1820             int state = element.getState();
1821             LOG.trace("Bundle[" + element.getBundleId() + "]:" + element.getSymbolicName() + ",v" + element.getVersion() + ", state:"
1822                     + stateToString(state));
1823             if (state != Bundle.ACTIVE && state != Bundle.RESOLVED) {
1824                 LOG.debug("Bundle:" + element.getSymbolicName() + " state:" + stateToString(state));
1825
1826                 // try {
1827                 // String host = element.getHeaders().get("FRAGMENT-HOST");
1828                 // if (host != null) {
1829                 // LOG.warn("Bundle " + element.getSymbolicName() +
1830                 // " is a fragment which is part of: " + host);
1831                 // LOG.warn("Required imports are: " +
1832                 // element.getHeaders().get("IMPORT-PACKAGE"));
1833                 // } else {
1834                 // element.start();
1835                 // }
1836                 // } catch (BundleException e) {
1837                 // LOG.error("BundleException:", e);
1838                 // fail();
1839                 // }
1840
1841                 debugit = true;
1842
1843             }
1844         }
1845         if (debugit) {
1846             LOG.warn(("Do some debugging because some bundle is unresolved"));
1847         }
1848         // assertNotNull(broker);
1849
1850         ServiceReference r = null;
1851         configLispPlugin.setLispAddress(lispBindAddress);
1852
1853         // Uncomment this code to Know which services were actually loaded to
1854         // BundleContext
1855
1856         /*
1857         for (ServiceReference sr : bc.getAllServiceReferences(null, null)) {
1858             LOG.info(sr.getBundle().getSymbolicName());
1859             LOG.info(sr.toString());
1860         }
1861         */
1862
1863         try {
1864             Thread.sleep(1000);
1865         } catch (InterruptedException e) {
1866         }
1867     }
1868
1869     private void cleanUP() {
1870         after();
1871         lms.clean();
1872         configLispPlugin.shouldListenOnXtrPort(false);
1873         socket = initSocket(socket, LispMessage.PORT_NUM);
1874
1875     }
1876
1877 }