Merge "Rename static final variable 'logger' to 'LOG'"
[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / lisp / MapResolver.java
index 2ce297b2837334445e09d00165a0d93666104ac8..c198ac2be1a80ecb2a795d12cb0626c4b4de6d39 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2013 Contextream, Inc. and others.  All rights reserved.
+ * Copyright (c) 2014 Contextream, Inc. and others.  All rights reserved.
  *
  * This program and the accompanying materials are made available under the
  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
@@ -9,49 +9,73 @@
 package org.opendaylight.lispflowmapping.implementation.lisp;
 
 import java.util.ArrayList;
-import java.util.Map;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Map.Entry;
 
+import org.opendaylight.lispflowmapping.implementation.config.ConfigIni;
 import org.opendaylight.lispflowmapping.implementation.dao.MappingServiceKeyUtil;
-import org.opendaylight.lispflowmapping.implementation.util.MaskUtil;
+import org.opendaylight.lispflowmapping.implementation.util.DAOMappingUtil;
+import org.opendaylight.lispflowmapping.implementation.util.LispAFIConvertor;
+import org.opendaylight.lispflowmapping.implementation.util.LispNotificationHelper;
 import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
 import org.opendaylight.lispflowmapping.interfaces.dao.IMappingServiceKey;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceRLOC;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceValue;
-import org.opendaylight.lispflowmapping.interfaces.lisp.IMapReplyHandler;
+import org.opendaylight.lispflowmapping.interfaces.dao.MappingEntry;
+import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceRLOCGroup;
+import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceSubscriberRLOC;
+import org.opendaylight.lispflowmapping.interfaces.lisp.IMapRequestResultHandler;
 import org.opendaylight.lispflowmapping.interfaces.lisp.IMapResolverAsync;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.EidToLocatorRecord.Action;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.LcafTrafficEngineeringAddress;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.MapRequest;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidrecords.EidRecord;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecord;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecord.Action;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecordBuilder;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lcaftrafficengineeringaddress.Hops;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispaddress.LispAddressContainer;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispaddress.LispAddressContainerBuilder;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.locatorrecords.LocatorRecord;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.locatorrecords.LocatorRecordBuilder;
 import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.mapreplymessage.MapReplyBuilder;
+import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.maprequest.ItrRloc;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
-public class MapResolver implements IMapResolverAsync {
+public class MapResolver extends AbstractLispComponent implements IMapResolverAsync {
+
     private static final int TTL_RLOC_TIMED_OUT = 1;
     private static final int TTL_NO_RLOC_KNOWN = 15;
-    private ILispDAO dao;
-    private volatile boolean shouldAuthenticate;
-    private volatile boolean shouldIterateMask;
-    protected static final Logger logger = LoggerFactory.getLogger(MapResolver.class);
+    protected static final Logger LOG = LoggerFactory.getLogger(MapResolver.class);
+
+    private static final ConfigIni configIni = new ConfigIni();
+    private static final String elpPolicy = configIni.getElpPolicy();
 
     public MapResolver(ILispDAO dao) {
         this(dao, true, true);
     }
 
     public MapResolver(ILispDAO dao, boolean authenticate, boolean iterateMask) {
-        this.dao = dao;
-        this.shouldAuthenticate = authenticate;
-        this.shouldIterateMask = iterateMask;
+        super(dao, authenticate, iterateMask);
     }
 
-    public void handleMapRequest(MapRequest request, IMapReplyHandler callback) {
+    public void handleMapRequest(MapRequest request, boolean smr, IMapRequestResultHandler callback) {
         if (dao == null) {
-            logger.warn("handleMapRequest called while dao is uninitialized");
+            LOG.warn("handleMapRequest called while dao is uninitialized");
+            return;
+        }
+        if (request.isPitr()) {
+            if (request.getEidRecord().size() > 0) {
+                EidRecord eid = request.getEidRecord().get(0);
+                Object result = DAOMappingUtil.getLocatorsSpecificByEidRecord(eid, dao, ADDRESS_SUBKEY, shouldIterateMask());
+                if (result != null && result instanceof MappingServiceRLOCGroup) {
+                    MappingServiceRLOCGroup locatorsGroup = (MappingServiceRLOCGroup) result;
+                    if (locatorsGroup != null && locatorsGroup.getRecords().size() > 0) {
+                        callback.handleNonProxyMapRequest(request,
+                                LispNotificationHelper.getTransportAddressFromContainer(locatorsGroup.getRecords().get(0).getLispAddressContainer()));
+                    }
+                }
+            }
+
         } else {
             MapReplyBuilder builder = new MapReplyBuilder();
             builder.setEchoNonceEnabled(false);
@@ -61,101 +85,137 @@ public class MapResolver implements IMapResolverAsync {
             builder.setEidToLocatorRecord(new ArrayList<EidToLocatorRecord>());
             for (EidRecord eid : request.getEidRecord()) {
                 EidToLocatorRecordBuilder recordBuilder = new EidToLocatorRecordBuilder();
+                Entry<IMappingServiceKey, List<MappingServiceRLOCGroup>> mapping = DAOMappingUtil.getMappingForEidRecord(eid, dao);
                 recordBuilder.setRecordTtl(0);
                 recordBuilder.setAction(Action.NoAction);
                 recordBuilder.setAuthoritative(false);
                 recordBuilder.setMapVersion((short) 0);
-                recordBuilder.setMaskLength(eid.getMask());
-                recordBuilder.setLispAddressContainer(eid.getLispAddressContainer());
+                recordBuilder.setMaskLength((short) mapping.getKey().getMask());
+                recordBuilder.setLispAddressContainer(mapping.getKey().getEID());
                 recordBuilder.setLocatorRecord(new ArrayList<LocatorRecord>());
-                IMappingServiceKey key = MappingServiceKeyUtil.generateMappingServiceKey(eid.getLispAddressContainer(), eid.getMask());
-                Map<String, ?> locators = dao.get(key);
-                if (shouldIterateMask() && locators == null && MaskUtil.isMaskable(key.getEID().getAddress())) {
-                    locators = findMaskLocators(key);
-                }
-                if (locators != null) {
-                    MappingServiceValue value = (MappingServiceValue) locators.get("value");
-
-                    if (value.getRlocs() != null && value.getRlocs().size() > 0) {
-                        addLocators(recordBuilder, value);
-                    } else {
-                        recordBuilder
-                                .setAction(org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecord.Action.NativelyForward);
-
-                        if (didContainRLOCs(value)) {
-                            recordBuilder.setRecordTtl(TTL_RLOC_TIMED_OUT);
-                        } else {
-                            recordBuilder.setRecordTtl(TTL_NO_RLOC_KNOWN);
+                List<MappingServiceRLOCGroup> locators = mapping.getValue();
+                if (locators != null && locators.size() > 0) {
+                    List<ItrRloc> itrRlocs = request.getItrRloc();
+                    addLocatorGroups(recordBuilder, locators, itrRlocs);
+                    if (itrRlocs != null && itrRlocs.size() > 0) {
+                        LispAddressContainer itrRloc = itrRlocs.get(0).getLispAddressContainer();
+                        MappingServiceSubscriberRLOC subscriberRloc = new MappingServiceSubscriberRLOC(itrRloc);
+                        HashSet<MappingServiceSubscriberRLOC> subscribers = getSubscribers(mapping.getKey().getEID(), mapping.getKey().getMask());
+                        if (subscribers == null) {
+                            subscribers = new HashSet<MappingServiceSubscriberRLOC>();
+                        } else if (subscribers.contains(subscriberRloc)) {
+                            /*
+                             * If there is an entry already for this
+                             * subscriberRloc, remove it, so that it gets the
+                             * new timestamp
+                             */
+                            subscribers.remove(subscriberRloc);
+                        }
+                        if (smr) {
+                            IMappingServiceKey key = MappingServiceKeyUtil.generateMappingServiceKey(mapping.getKey().getEID(),
+                                    mapping.getKey().getMask());
+                            LOG.trace("Adding new subscriber: " + subscriberRloc.toString());
+                            subscribers.add(subscriberRloc);
+                            dao.put(key, new MappingEntry<HashSet<MappingServiceSubscriberRLOC>>(SUBSCRIBERS_SUBKEY, subscribers));
                         }
                     }
                 } else {
                     recordBuilder
                             .setAction(org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecord.Action.NativelyForward);
-                    recordBuilder.setRecordTtl(TTL_NO_RLOC_KNOWN);
+                    if (shouldAuthenticate() && getPassword(eid.getLispAddressContainer(), eid.getMask()) != null) {
+                        recordBuilder.setRecordTtl(TTL_RLOC_TIMED_OUT);
+                    } else {
+                        recordBuilder.setRecordTtl(TTL_NO_RLOC_KNOWN);
+
+                    }
                 }
                 builder.getEidToLocatorRecord().add(recordBuilder.build());
             }
-            callback.handleMapReply(builder.build());
-        }
-    }
-
-    private boolean didContainRLOCs(MappingServiceValue value) {
-        return value.getKey() == null;
-    }
 
-    private Map<String, ?> findMaskLocators(IMappingServiceKey key) {
-        int mask = key.getMask();
-        while (mask > 0) {
-            key = MappingServiceKeyUtil.generateMappingServiceKey(
-                    new LispAddressContainerBuilder().setAddress(MaskUtil.normalize(key.getEID().getAddress(), mask)).build(), mask);
-            mask--;
-            Map<String, ?> locators = dao.get(key);
-            if (locators != null) {
-                return locators;
-            }
+            callback.handleMapReply(builder.build());
         }
-        return null;
     }
 
-    private void addLocators(EidToLocatorRecordBuilder recordBuilder, MappingServiceValue value) {
-        for (MappingServiceRLOC rloc : value.getRlocs()) {
-            addLocator(recordBuilder, rloc);
+    private void addLocatorGroups(EidToLocatorRecordBuilder recordBuilder, List<MappingServiceRLOCGroup> rlocs, List<ItrRloc> itrRlocs) {
+        for (MappingServiceRLOCGroup rloc : rlocs) {
+            addLocators(recordBuilder, rloc, itrRlocs);
             recordBuilder.setRecordTtl(rloc.getTtl());
         }
     }
 
-    private void addLocator(EidToLocatorRecordBuilder recordBuilder, MappingServiceRLOC locatorObject) {
+    private void addLocators(EidToLocatorRecordBuilder recordBuilder, MappingServiceRLOCGroup locatorObject, List<ItrRloc> itrRlocs) {
         if (locatorObject == null) {
             return;
         }
+
+        recordBuilder.setAction(locatorObject.getAction());
+        recordBuilder.setAuthoritative(locatorObject.isAuthoritative());
+        recordBuilder.setRecordTtl(locatorObject.getTtl());
+
         try {
-            recordBuilder.getLocatorRecord().add(
-                    new LocatorRecordBuilder().setLocalLocator(locatorObject.getRecord().isLocalLocator())
-                            .setRlocProbed(locatorObject.getRecord().isRlocProbed()).setWeight(locatorObject.getRecord().getWeight())
-                            .setPriority(locatorObject.getRecord().getPriority()).setMulticastWeight(locatorObject.getRecord().getMulticastWeight())
-                            .setMulticastPriority(locatorObject.getRecord().getMulticastPriority()).setRouted(true)
-                            .setLispAddressContainer(locatorObject.getRecord().getLispAddressContainer()).build());
-            recordBuilder.setAction(locatorObject.getAction());
-            recordBuilder.setAuthoritative(locatorObject.isAuthoritative());
-            recordBuilder.setRecordTtl(locatorObject.getTtl());
+            for (LocatorRecord record : locatorObject.getRecords()) {
+                LispAddressContainer container = record.getLispAddressContainer();
+
+                // For non-ELP RLOCs, or when ELP policy is default, just add the locator and be done
+                if ((!(container.getAddress() instanceof LcafTrafficEngineeringAddress)) || elpPolicy.equalsIgnoreCase("default")) {
+                    recordBuilder.getLocatorRecord().add(
+                            new LocatorRecordBuilder().setLocalLocator(record.isLocalLocator()).setRlocProbed(record.isRlocProbed())
+                                    .setWeight(record.getWeight()).setPriority(record.getPriority()).setMulticastWeight(record.getMulticastWeight())
+                                    .setMulticastPriority(record.getMulticastPriority()).setRouted(true)
+                                    .setLispAddressContainer(container).build());
+                    continue;
+                }
+
+                LispAddressContainer nextHop = getNextELPHop((LcafTrafficEngineeringAddress) container.getAddress(), itrRlocs);
+                if (nextHop != null) {
+                    java.lang.Short priority = record.getPriority();
+                    if (elpPolicy.equalsIgnoreCase("both")) {
+                        recordBuilder.getLocatorRecord().add(
+                                new LocatorRecordBuilder().setLocalLocator(record.isLocalLocator()).setRlocProbed(record.isRlocProbed())
+                                        .setWeight(record.getWeight()).setPriority(record.getPriority()).setMulticastWeight(record.getMulticastWeight())
+                                        .setMulticastPriority(record.getMulticastPriority()).setRouted(true)
+                                        .setLispAddressContainer(container).build());
+                        // Make the priority of the added simple locator lower so that ELP is used by default if
+                        // the xTR understands ELP.  Exclude 255, since that means don't use for unicast forwarding
+                        // XXX Complex cases like several ELPs with different priorities are not handled
+                        if (priority != 254 || priority !=255) {
+                            priority++;
+                        }
+                    }
+                    // Build and add the simple RLOC
+                    recordBuilder.getLocatorRecord().add(
+                            new LocatorRecordBuilder().setLocalLocator(record.isLocalLocator()).setRlocProbed(record.isRlocProbed())
+                            .setWeight(record.getWeight()).setPriority(priority).setMulticastWeight(record.getMulticastWeight())
+                            .setMulticastPriority(record.getMulticastPriority()).setRouted(true)
+                            .setLispAddressContainer(nextHop).build());
+                }
+            }
         } catch (ClassCastException cce) {
         }
     }
 
-    public boolean shouldIterateMask() {
-        return shouldIterateMask;
-    }
-
-    public boolean shouldAuthenticate() {
-        return shouldAuthenticate;
-    }
-
-    public void setShouldIterateMask(boolean shouldIterateMask) {
-        this.shouldIterateMask = shouldIterateMask;
-    }
+    private LispAddressContainer getNextELPHop(LcafTrafficEngineeringAddress elp, List<ItrRloc> itrRlocs) {
+        LispAddressContainer nextHop = null;
+        List<Hops> hops = elp.getHops();
+
+        if (hops != null && hops.size() > 0) {
+            // By default we return the first hop
+            nextHop = LispAFIConvertor.toContainer(LispAFIConvertor.toAFIfromPrimitive(hops.get(0).getHop().getPrimitiveAddress()));
+            for (Hops hop : hops) {
+                LispAddressContainer hopContainer = LispAFIConvertor.toContainer(LispAFIConvertor.toAFIfromPrimitive(hop.getHop().getPrimitiveAddress()));
+                for (ItrRloc itrRloc : itrRlocs) {
+                    if (itrRloc.getLispAddressContainer().equals(hopContainer)) {
+                        int i = hops.indexOf(hop);
+                        if (i < hops.size() - 1) {
+                            nextHop = LispAFIConvertor.toContainer(LispAFIConvertor.toAFIfromPrimitive(hops.get(i+1).getHop().getPrimitiveAddress()));
+                            return nextHop;
+                        }
+                    }
+                }
+            }
+        }
 
-    public void setShouldAuthenticate(boolean shouldAuthenticate) {
-        this.shouldAuthenticate = shouldAuthenticate;
+        return nextHop;
     }
 
 }