/*
- * Copyright (c) 2014 Contextream, Inc. and others. All rights reserved.
+ * Copyright (c) 2014, 2017 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,
package org.opendaylight.lispflowmapping.implementation.lisp;
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Maps;
+import com.google.common.util.concurrent.ThreadFactoryBuilder;
import java.net.InetAddress;
import java.net.NetworkInterface;
import java.net.SocketException;
import java.util.ArrayList;
+import java.util.Arrays;
import java.util.Enumeration;
+import java.util.HashSet;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
import java.util.Set;
-
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.ThreadFactory;
+import java.util.concurrent.TimeUnit;
import org.apache.commons.lang3.BooleanUtils;
-import org.apache.commons.lang3.exception.ExceptionUtils;
-import org.opendaylight.lispflowmapping.implementation.authentication.LispAuthenticationUtil;
-import org.opendaylight.lispflowmapping.implementation.config.ConfigIni;
-import org.opendaylight.lispflowmapping.implementation.mapcache.TopologyMapCache;
-import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
-import org.opendaylight.lispflowmapping.interfaces.dao.SubscriberRLOC;
+import org.opendaylight.controller.md.sal.binding.api.NotificationService;
+import org.opendaylight.lispflowmapping.config.ConfigIni;
+import org.opendaylight.lispflowmapping.implementation.util.LoggingUtil;
+import org.opendaylight.lispflowmapping.implementation.util.MSNotificationInputUtil;
+import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
+import org.opendaylight.lispflowmapping.interfaces.dao.Subscriber;
import org.opendaylight.lispflowmapping.interfaces.lisp.IMapNotifyHandler;
import org.opendaylight.lispflowmapping.interfaces.lisp.IMapServerAsync;
-import org.opendaylight.lispflowmapping.lisp.util.LcafSourceDestHelper;
-import org.opendaylight.lispflowmapping.lisp.util.LispAFIConvertor;
+import org.opendaylight.lispflowmapping.interfaces.lisp.ISmrNotificationListener;
+import org.opendaylight.lispflowmapping.interfaces.lisp.SmrEvent;
+import org.opendaylight.lispflowmapping.interfaces.mappingservice.IMappingService;
+import org.opendaylight.lispflowmapping.lisp.authentication.LispAuthenticationUtil;
+import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
+import org.opendaylight.lispflowmapping.lisp.type.MappingData;
+import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
+import org.opendaylight.lispflowmapping.lisp.util.LispAddressUtil;
import org.opendaylight.lispflowmapping.lisp.util.MapNotifyBuilderHelper;
import org.opendaylight.lispflowmapping.lisp.util.MapRequestUtil;
-import org.opendaylight.lispflowmapping.lisp.util.MaskUtil;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.MapRegister;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidrecords.EidRecord;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidrecords.EidRecordBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecord;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.eidtolocatorrecords.EidToLocatorRecordBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lispaddress.LispAddressContainer;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.lispaddress.lispaddresscontainer.address.LcafSourceDest;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.mapnotifymessage.MapNotifyBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev150820.maprequestnotification.MapRequestBuilder;
+import org.opendaylight.lispflowmapping.lisp.util.MappingRecordUtil;
+import org.opendaylight.lispflowmapping.lisp.util.SourceDestKeyHelper;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.SourceDestKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinary;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.MapRegister;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.list.EidItem;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.list.EidItemBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapnotifymessage.MapNotifyBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.authkey.container.MappingAuthkey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItem;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.list.MappingRecordItemBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.maprequestnotification.MapRequestBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddress;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddressBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChanged;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.OdlMappingserviceListener;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-public class MapServer extends AbstractLispComponent implements IMapServerAsync {
+public class MapServer implements IMapServerAsync, OdlMappingserviceListener, ISmrNotificationListener {
protected static final Logger LOG = LoggerFactory.getLogger(MapServer.class);
+ private static final byte[] ALL_ZEROES_XTR_ID = new byte[] {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0 ,0};
+ private IMappingService mapService;
+ private boolean subscriptionService;
+ private IMapNotifyHandler notifyHandler;
+ private NotificationService notificationService;
+ private ListenerRegistration<MapServer> mapServerListenerRegistration;
+ private SmrScheduler scheduler;
- private static final ConfigIni configIni = new ConfigIni();
- private static final boolean overwriteConfig = configIni.mappingOverwriteIsSet();
- private boolean overwrite;
-
- public MapServer(ILispDAO dao) {
- this(dao, overwriteConfig);
- }
-
- public MapServer(ILispDAO dao, boolean overwrite) {
- this(dao, overwrite, true);
+ public MapServer(IMappingService mapService, boolean subscriptionService,
+ IMapNotifyHandler notifyHandler, NotificationService notificationService) {
+ Preconditions.checkNotNull(mapService);
+ this.mapService = mapService;
+ this.subscriptionService = subscriptionService;
+ this.notifyHandler = notifyHandler;
+ this.notificationService = notificationService;
+ if (notificationService != null) {
+ notificationService.registerNotificationListener(this);
+ }
+ scheduler = new SmrScheduler();
}
- public MapServer(ILispDAO dao, boolean overwrite, boolean authenticate) {
- this(dao, overwrite, authenticate, true);
+ @Override
+ public void setSubscriptionService(boolean subscriptionService) {
+ this.subscriptionService = subscriptionService;
}
- public MapServer(ILispDAO dao, boolean overwrite, boolean authenticate, boolean iterateAuthenticationMask) {
- super(dao, authenticate, iterateAuthenticationMask);
- this.overwrite = overwrite;
- }
+ @SuppressWarnings("unchecked")
+ public void handleMapRegister(MapRegister mapRegister) {
+ boolean mappingUpdated = false;
+ boolean oldMappingRemoved = false;
+ boolean merge = ConfigIni.getInstance().mappingMergeIsSet() && mapRegister.isMergeEnabled();
+ Set<Subscriber> subscribers = null;
+ MappingRecord oldMapping;
- private static InetAddress getLocalAddress() {
- try {
- Enumeration<NetworkInterface> interfaces = NetworkInterface.getNetworkInterfaces();
- while (interfaces.hasMoreElements()) {
- NetworkInterface current = interfaces.nextElement();
- LOG.debug("Interface " + current.toString());
- if (!current.isUp() || current.isLoopback() || current.isVirtual())
- continue;
- Enumeration<InetAddress> addresses = current.getInetAddresses();
- while (addresses.hasMoreElements()) {
- InetAddress current_addr = addresses.nextElement();
- // Skip loopback and link local addresses
- if (current_addr.isLoopbackAddress() || current_addr.isLinkLocalAddress())
- continue;
- LOG.debug(current_addr.getHostAddress());
- return current_addr;
- }
+ if (merge) {
+ if (!mapRegister.isXtrSiteIdPresent() || mapRegister.getXtrId() == null) {
+ LOG.error("Merge bit is set in Map-Register, but xTR-ID is not present. Will not merge.");
+ merge = false;
+ } else if (Arrays.equals(mapRegister.getXtrId().getValue(), ALL_ZEROES_XTR_ID)) {
+ LOG.warn("Merge bit is set in Map-Register, but xTR-ID is all zeroes.");
}
- } catch (SocketException se) {
}
- return null;
- }
- public void handleMapRegister(MapRegister mapRegister, boolean smr, IMapNotifyHandler callback) {
- if (dao == null) {
- LOG.warn("handleMapRegister called while dao is uninitialized");
- return;
- }
- boolean failed = false;
- String password = null;
- for (EidToLocatorRecord eidRecord : mapRegister.getEidToLocatorRecord()) {
- if (shouldAuthenticate()) {
- password = TopologyMapCache.getAuthenticationKey(eidRecord.getLispAddressContainer(), dao, shouldIterateMask());
- if (!LispAuthenticationUtil.validate(mapRegister, password)) {
- LOG.warn("Authentication failed");
- failed = true;
- break;
- }
+ for (MappingRecordItem record : mapRegister.getMappingRecordItem()) {
+ MappingRecord mapping = record.getMappingRecord();
+ Eid eid = mapping.getEid();
+ MappingData mappingData = new MappingData(mapping, System.currentTimeMillis());
+ mappingData.setMergeEnabled(merge);
+ mappingData.setXtrId(mapRegister.getXtrId());
+
+ oldMapping = getMappingRecord(mapService.getMapping(MappingOrigin.Southbound, eid));
+ mapService.addMapping(MappingOrigin.Southbound, eid, getSiteId(mapRegister), mappingData);
+ if (oldMapping != null && MappingRecordUtil.isNegativeMapping(oldMapping)) {
+ mapService.removeMapping(MappingOrigin.Southbound, oldMapping.getEid());
+ oldMappingRemoved = true;
}
- boolean mappingChanged = TopologyMapCache.addMapping(eidRecord, smr, dao, shouldIterateMask(),
- shouldOverwrite());
- if (smr && mappingChanged) {
- Set<SubscriberRLOC> subscribers = TopologyMapCache.getSubscribers(
- eidRecord.getLispAddressContainer(), dao);
- sendSmrs(eidRecord, subscribers, callback);
+
+ if (subscriptionService) {
+ MappingRecord newMapping = merge
+ ? getMappingRecord(mapService.getMapping(MappingOrigin.Southbound, eid)) : mapping;
+
+ if (mappingChanged(oldMapping, newMapping)) {
+ if (LOG.isDebugEnabled()) {
+ LOG.debug("Mapping update occured for {} SMRs will be sent for its subscribers.",
+ LispAddressStringifier.getString(mapping.getEid()));
+ }
+ subscribers = getSubscribers(eid);
+ if (oldMapping != null && !oldMapping.getEid().equals(eid)) {
+ subscribers = addParentSubscribers(eid, subscribers);
+ }
+ LoggingUtil.logSubscribers(LOG, eid, subscribers);
+ handleSmr(eid, subscribers);
+ if (oldMapping != null && oldMappingRemoved && !oldMapping.getEid().equals(eid)) {
+ subscribers = getSubscribers(oldMapping.getEid());
+ LoggingUtil.logSubscribers(LOG, oldMapping.getEid(), subscribers);
+ handleSmr(oldMapping.getEid(), subscribers);
+ }
+ mappingUpdated = true;
+ }
}
}
- if (!failed) {
+ if (BooleanUtils.isTrue(mapRegister.isWantMapNotify())) {
+ LOG.trace("MapRegister wants MapNotify");
MapNotifyBuilder builder = new MapNotifyBuilder();
- if (BooleanUtils.isTrue(mapRegister.isWantMapNotify())) {
- LOG.trace("MapRegister wants MapNotify");
+ List<TransportAddress> rlocs = null;
+ if (merge) {
+ Set<IpAddressBinary> notifyRlocs = new HashSet<IpAddressBinary>();
+ List<MappingRecordItem> mergedMappings = new ArrayList<MappingRecordItem>();
+ for (MappingRecordItem record : mapRegister.getMappingRecordItem()) {
+ MappingRecord mapping = record.getMappingRecord();
+ MappingRecord currentRecord = getMappingRecord(mapService.getMapping(MappingOrigin.Southbound,
+ mapping.getEid()));
+ mergedMappings.add(new MappingRecordItemBuilder().setMappingRecord(currentRecord).build());
+ Set<IpAddressBinary> sourceRlocs = (Set<IpAddressBinary>) mapService.getData(
+ MappingOrigin.Southbound, mapping.getEid(), SubKeys.SRC_RLOCS);
+ if (sourceRlocs != null) {
+ notifyRlocs.addAll(sourceRlocs);
+ }
+ }
+ MapNotifyBuilderHelper.setFromMapRegisterAndMappingRecordItems(builder, mapRegister, mergedMappings);
+ // send map-notify to merge group only when mapping record is changed
+ if (mappingUpdated) {
+ rlocs = getTransportAddresses(notifyRlocs);
+ }
+ } else {
MapNotifyBuilderHelper.setFromMapRegister(builder, mapRegister);
- if (shouldAuthenticate()) {
+ }
+ List<MappingRecordItem> mappings = builder.getMappingRecordItem();
+ if (mappings != null && mappings.get(0) != null && mappings.get(0).getMappingRecord() != null
+ && mappings.get(0).getMappingRecord().getEid() != null) {
+ MappingAuthkey authkey = mapService.getAuthenticationKey(mappings.get(0).getMappingRecord().getEid());
+ if (authkey != null) {
builder.setAuthenticationData(LispAuthenticationUtil.createAuthenticationData(builder.build(),
- password));
+ authkey.getKeyString()));
}
- callback.handleMapNotify(builder.build());
}
+ notifyHandler.handleMapNotify(builder.build(), rlocs);
+ }
+ }
+
+ private static List<TransportAddress> getTransportAddresses(Set<IpAddressBinary> addresses) {
+ List<TransportAddress> rlocs = new ArrayList<TransportAddress>();
+ for (IpAddressBinary address : addresses) {
+ TransportAddressBuilder tab = new TransportAddressBuilder();
+ tab.setIpAddress(address);
+ tab.setPort(new PortNumber(LispMessage.PORT_NUM));
+ rlocs.add(tab.build());
}
+ return rlocs;
}
- public String getAuthenticationKey(LispAddressContainer address) {
- return TopologyMapCache.getAuthenticationKey(address, dao, shouldIterateMask());
+ private static SiteId getSiteId(MapRegister mapRegister) {
+ return (mapRegister.getSiteId() != null) ? new SiteId(mapRegister.getSiteId()) : null;
}
- public void removeAuthenticationKey(LispAddressContainer address) {
- TopologyMapCache.removeAuthenticationKey(address, dao);
+ private static MappingRecord getMappingRecord(MappingData mappingData) {
+ return (mappingData != null) ? mappingData.getRecord() : null;
}
- public void removeMapping(LispAddressContainer address, boolean smr, IMapNotifyHandler callback) {
- if (smr) {
- // mapping is removed before first SMR is sent to avoid inconsistent replies
- TopologyMapCache.removeMapping(address, dao, shouldOverwrite());
- Set<SubscriberRLOC> subscribers = TopologyMapCache.getSubscribers(address, dao);
- sendSmrs(new EidToLocatorRecordBuilder().setLispAddressContainer(address).
- setMaskLength(MaskUtil.getMaskForAddress(address)).build(), subscribers, callback);
- TopologyMapCache.removeSubscribers(address, dao, shouldOverwrite());
- } else {
- TopologyMapCache.removeMapping(address, dao, shouldOverwrite());
- TopologyMapCache.removeSubscribers(address, dao, shouldOverwrite());
+ @Override
+ public void onMappingChanged(MappingChanged notification) {
+ LOG.trace("MappingChanged event of type: `{}'", notification.getChangeType());
+ if (subscriptionService) {
+ Eid eid = notification.getMappingRecord().getEid();
+ Set<Subscriber> subscribers = MSNotificationInputUtil.toSubscriberSet(notification.getSubscriberItem());
+ LoggingUtil.logSubscribers(LOG, eid, subscribers);
+ if (mapService.isMaster()) {
+ sendSmrs(eid, subscribers);
+ if (eid.getAddress() instanceof SourceDestKey) {
+ Set<Subscriber> dstSubscribers = MSNotificationInputUtil.toSubscriberSetFromDst(
+ notification.getDstSubscriberItem());
+ LoggingUtil.logSubscribers(LOG, SourceDestKeyHelper.getDstBinary(eid), dstSubscribers);
+ sendSmrs(SourceDestKeyHelper.getDstBinary(eid), dstSubscribers);
+ }
+ }
+ }
+ }
+
+ private static boolean mappingChanged(MappingRecord oldMapping, MappingRecord newMapping) {
+ // We only check for fields we care about
+ // XXX: This code needs to be checked and updated when the YANG model for MappingRecord is modified
+ Preconditions.checkNotNull(newMapping, "The new mapping should never be null");
+ if (oldMapping == null) {
+ LOG.trace("mappingChanged(): old mapping is null");
+ return true;
+ } else if (!Objects.equals(oldMapping.getEid(), newMapping.getEid())) {
+ LOG.trace("mappingChanged(): EID");
+ return true;
+ } else if (!Objects.equals(oldMapping.getLocatorRecord(), newMapping.getLocatorRecord())) {
+ LOG.trace("mappingChanged(): RLOC");
+ return true;
+ } else if (!Objects.equals(oldMapping.getAction(), newMapping.getAction())) {
+ LOG.trace("mappingChanged(): action");
+ return true;
+ } else if (!Objects.equals(oldMapping.getRecordTtl(), newMapping.getRecordTtl())) {
+ LOG.trace("mappingChanged(): TTL");
+ return true;
+ } else if (!Objects.equals(oldMapping.getMapVersion(), newMapping.getMapVersion())) {
+ LOG.trace("mappingChanged(): mapping version");
+ return true;
}
+ return false;
}
- private void sendSmrs(EidToLocatorRecord record, Set<SubscriberRLOC> subscribers, IMapNotifyHandler callback) {
- LispAddressContainer eid = record.getLispAddressContainer();
- handleSmr(record, subscribers, callback);
+ private void handleSmr(Eid eid, Set<Subscriber> subscribers) {
+ sendSmrs(eid, subscribers);
// For SrcDst LCAF also send SMRs to Dst prefix
- if (eid.getAddress() instanceof LcafSourceDest) {
- LispAddressContainer dstAddr = LispAFIConvertor.toContainer(LcafSourceDestHelper.getDstAfi(eid));
- short dstMask = LcafSourceDestHelper.getDstMask(eid);
- subscribers = TopologyMapCache.getSubscribers(dstAddr, dao);
- EidToLocatorRecord newRecord = new EidToLocatorRecordBuilder().setAction(record.getAction()).
- setAuthoritative(record.isAuthoritative()).setLocatorRecord(record.getLocatorRecord()).
- setMapVersion(record.getMapVersion()).setRecordTtl(record.getRecordTtl()).
- setLispAddressContainer(dstAddr).setMaskLength(dstMask).build();
- handleSmr(newRecord, subscribers, callback);
+ if (eid.getAddress() instanceof SourceDestKey) {
+ Eid dstAddr = SourceDestKeyHelper.getDstBinary(eid);
+ Set<Subscriber> dstSubs = getSubscribers(dstAddr);
+ sendSmrs(dstAddr, dstSubs);
}
}
- private void handleSmr(EidToLocatorRecord record, Set<SubscriberRLOC> subscribers, IMapNotifyHandler callback) {
+ private void sendSmrs(Eid eid, Set<Subscriber> subscribers) {
if (subscribers == null) {
return;
}
- MapRequestBuilder mrb = MapRequestUtil.prepareSMR(record.getLispAddressContainer(),
- LispAFIConvertor.toContainer(getLocalAddress()));
+ final MapRequestBuilder mrb = MapRequestUtil.prepareSMR(eid, LispAddressUtil.toRloc(getLocalAddress()));
LOG.trace("Built SMR packet: " + mrb.build().toString());
- for (SubscriberRLOC subscriber : subscribers) {
- if (subscriber.timedOut()) {
- LOG.trace("Lazy removing expired subscriber entry " + subscriber.toString());
- subscribers.remove(subscriber);
+
+ scheduler.scheduleSmrs(mrb, subscribers.iterator());
+ }
+
+ @SuppressWarnings("unchecked")
+ private Set<Subscriber> getSubscribers(Eid address) {
+ return (Set<Subscriber>) mapService.getData(MappingOrigin.Southbound, address, SubKeys.SUBSCRIBERS);
+ }
+
+ private Set<Subscriber> addParentSubscribers(Eid eid, Set<Subscriber> subscribers) {
+ Eid parentPrefix = mapService.getParentPrefix(eid);
+ if (parentPrefix == null) {
+ return subscribers;
+ }
+
+ Set<Subscriber> parentSubscribers = getSubscribers(parentPrefix);
+ if (parentSubscribers != null) {
+ if (subscribers != null) {
+ subscribers.addAll(parentSubscribers);
} else {
- try {
- // The address stored in the SMR's EID record is used as Source EID in the SMR-invoked Map-Request.
- // To ensure consistent behavior it is set to the value used to originally request a given mapping
- mrb.setEidRecord(new ArrayList<EidRecord>());
- mrb.getEidRecord().add(new EidRecordBuilder()
- .setMask((short)MaskUtil.getMaxMask(LispAFIConvertor.toAFI(subscriber.getSrcEid())))
- .setLispAddressContainer(subscriber.getSrcEid()).build());
- callback.handleSMR(mrb.build(), subscriber.getSrcRloc());
- } catch (Exception e) {
- LOG.error("Errors encountered while handling SMR:" + ExceptionUtils.getStackTrace(e));
- }
+ subscribers = parentSubscribers;
}
}
- TopologyMapCache.addSubscribers(record.getLispAddressContainer(), subscribers, dao);
+ return subscribers;
}
- public boolean shouldOverwrite() {
- return overwrite;
+ private void addSubscribers(Eid address, Set<Subscriber> subscribers) {
+ mapService.addData(MappingOrigin.Southbound, address, SubKeys.SUBSCRIBERS, subscribers);
}
- public void setOverwrite(boolean overwrite) {
- this.overwrite = overwrite;
+ private static InetAddress getLocalAddress() {
+ try {
+ Enumeration<NetworkInterface> interfaces = NetworkInterface.getNetworkInterfaces();
+ while (interfaces.hasMoreElements()) {
+ NetworkInterface current = interfaces.nextElement();
+ LOG.trace("Interface " + current.toString());
+ if (!current.isUp() || current.isLoopback() || current.isVirtual()) {
+ continue;
+ }
+ Enumeration<InetAddress> addresses = current.getInetAddresses();
+ while (addresses.hasMoreElements()) {
+ InetAddress currentAddr = addresses.nextElement();
+ // Skip loopback and link local addresses
+ if (currentAddr.isLoopbackAddress() || currentAddr.isLinkLocalAddress()) {
+ continue;
+ }
+ LOG.debug(currentAddr.getHostAddress());
+ return currentAddr;
+ }
+ }
+ } catch (SocketException se) {
+ LOG.debug("Caught socket exception", se);
+ }
+ return null;
}
@Override
- public void addAuthenticationKey(LispAddressContainer address, String key) {
- TopologyMapCache.addAuthenticationKey(address, key, dao);
+ public void onSmrInvokedReceived(SmrEvent event) {
+ scheduler.smrReceived(event);
}
+ /**
+ * Task scheduler is responsible for resending SMR messages to a subscriber (xTR)
+ * {@value ConfigIni#LISP_SMR_RETRY_COUNT} times, or until {@link ISmrNotificationListener#onSmrInvokedReceived}
+ * is triggered.
+ */
+ private class SmrScheduler {
+ final int cpuCores = Runtime.getRuntime().availableProcessors();
+ private final ThreadFactory threadFactory = new ThreadFactoryBuilder()
+ .setNameFormat("smr-executor-%d").build();
+ private final ScheduledExecutorService executor = Executors.newScheduledThreadPool(cpuCores * 2, threadFactory);
+ private final Map<IpAddressBinary, Map<Eid, ScheduledFuture<?>>> subscriberFutureMap = Maps.newConcurrentMap();
+
+ void scheduleSmrs(MapRequestBuilder mrb, Iterator<Subscriber> subscribers) {
+ // Using Iterator ensures that we don't get a ConcurrentModificationException when removing a Subscriber
+ // from a Set.
+ while (subscribers.hasNext()) {
+ Subscriber subscriber = subscribers.next();
+ if (subscriber.timedOut()) {
+ LOG.debug("Lazy removing expired subscriber entry " + subscriber.getString());
+ subscribers.remove();
+ } else {
+ final Eid srcEid = mrb.getSourceEid().getEid();
+ final ScheduledFuture<?> future = executor.scheduleAtFixedRate(new CancellableRunnable(
+ mrb, subscriber), 0L, ConfigIni.getInstance().getSmrTimeout(), TimeUnit.MILLISECONDS);
+ final IpAddressBinary subscriberAddress = LispAddressUtil
+ .addressBinaryFromAddress(subscriber.getSrcRloc().getAddress());
+
+ if (subscriberFutureMap.containsKey(subscriberAddress)) {
+ subscriberFutureMap.get(subscriberAddress).put(srcEid, future);
+ } else {
+ final Map<Eid, ScheduledFuture<?>> eidFutureMap = Maps.newConcurrentMap();
+ eidFutureMap.put(srcEid, future);
+ subscriberFutureMap.put(subscriberAddress, eidFutureMap);
+ }
+ }
+ }
+ }
+
+ void smrReceived(SmrEvent event) {
+ final List<IpAddressBinary> subscriberAddressList = event.getSubscriberAddressList();
+ for (IpAddressBinary subscriberAddress : subscriberAddressList) {
+ final Map<Eid, ScheduledFuture<?>> eidFutureMap = subscriberFutureMap.get(subscriberAddress);
+ if (eidFutureMap != null) {
+ final ScheduledFuture<?> future = eidFutureMap.get(event.getEid());
+ if (future != null && !future.isCancelled()) {
+ future.cancel(true);
+ LOG.debug("SMR-invoked MapRequest received, scheduled task for subscriber {}, EID {} with"
+ + " nonce {} has been cancelled", subscriberAddress.toString(),
+ LispAddressStringifier.getString(event.getEid()), event.getNonce());
+ eidFutureMap.remove(event.getEid());
+ }
+ if (eidFutureMap.isEmpty()) {
+ subscriberFutureMap.remove(subscriberAddress);
+ }
+ }
+ }
+ }
+
+ private final class CancellableRunnable implements Runnable {
+ private MapRequestBuilder mrb;
+ private Subscriber subscriber;
+ private int executionCount = 1;
+
+ CancellableRunnable(MapRequestBuilder mrb, Subscriber subscriber) {
+ this.mrb = mrb;
+ this.subscriber = subscriber;
+ }
+
+ @SuppressWarnings("checkstyle:IllegalCatch")
+ @Override
+ public void run() {
+ final IpAddressBinary subscriberAddress = LispAddressUtil
+ .addressBinaryFromAddress(subscriber.getSrcRloc().getAddress());
+ final Eid srcEid = mrb.getSourceEid().getEid();
+
+ try {
+ // The address stored in the SMR's EID record is used as Source EID in the SMR-invoked
+ // Map-Request. To ensure consistent behavior it is set to the value used to originally request
+ // a given mapping.
+ if (executionCount <= ConfigIni.getInstance().getSmrRetryCount()) {
+ synchronized (mrb) {
+ mrb.setEidItem(new ArrayList<EidItem>());
+ mrb.getEidItem().add(new EidItemBuilder().setEid(subscriber.getSrcEid()).build());
+ notifyHandler.handleSMR(mrb.build(), subscriber.getSrcRloc());
+ if (LOG.isTraceEnabled()) {
+ LOG.trace("Attempt #{} to send SMR for EID {} to subscriber {}, source EID {}",
+ executionCount,
+ LispAddressStringifier.getString(mrb.getSourceEid().getEid()),
+ LispAddressStringifier.getString(subscriber.getSrcRloc()),
+ LispAddressStringifier.getString(mrb.getEidItem().get(0).getEid()));
+ }
+ }
+ } else {
+ LOG.trace("Cancelling execution of a SMR Map-Request after {} failed attempts.",
+ executionCount - 1);
+ cancelAndRemove(subscriberAddress, srcEid);
+ return;
+ }
+ } catch (Exception e) {
+ LOG.error("Errors encountered while handling SMR:", e);
+ cancelAndRemove(subscriberAddress, srcEid);
+ return;
+ }
+ executionCount++;
+ }
+
+ private void cancelAndRemove(IpAddressBinary subscriberAddress, Eid eid) {
+ final Map<Eid, ScheduledFuture<?>> eidFutureMap = subscriberFutureMap.get(subscriberAddress);
+ if (eidFutureMap == null) {
+ LOG.warn("Couldn't find subscriber {} in SMR scheduler internal list", subscriberAddress);
+ return;
+ }
+
+ if (eidFutureMap.containsKey(eid)) {
+ ScheduledFuture<?> eidFuture = eidFutureMap.get(eid);
+ eidFutureMap.remove(eid);
+ eidFuture.cancel(false);
+ }
+ if (eidFutureMap.isEmpty()) {
+ subscriberFutureMap.remove(subscriberAddress);
+ }
+ }
+ }
+ }
}
Code Review / lispflowmapping.git / blobdiff