package org.opendaylight.lispflowmapping.implementation.lisp;
+import java.net.InetAddress;
+import java.net.NetworkInterface;
+import java.net.SocketException;
import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.Enumeration;
+import java.util.Set;
import org.apache.commons.lang3.BooleanUtils;
+import org.apache.commons.lang3.exception.ExceptionUtils;
+import org.opendaylight.controller.md.sal.binding.api.NotificationService;
import org.opendaylight.lispflowmapping.implementation.authentication.LispAuthenticationUtil;
-import org.opendaylight.lispflowmapping.implementation.dao.MappingServiceKeyUtil;
-import org.opendaylight.lispflowmapping.implementation.util.MapNotifyBuilderHelper;
-import org.opendaylight.lispflowmapping.interfaces.dao.ILispDAO;
-import org.opendaylight.lispflowmapping.interfaces.dao.IMappingServiceKey;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingEntry;
-import org.opendaylight.lispflowmapping.interfaces.dao.MappingServiceRLOCGroup;
+import org.opendaylight.lispflowmapping.interfaces.dao.SubKeys;
+import org.opendaylight.lispflowmapping.interfaces.dao.SubscriberRLOC;
import org.opendaylight.lispflowmapping.interfaces.lisp.IMapNotifyHandler;
import org.opendaylight.lispflowmapping.interfaces.lisp.IMapServerAsync;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.MapRegister;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.eidtolocatorrecords.EidToLocatorRecord;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispaddress.LispAddressContainer;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispaddress.lispaddresscontainer.address.LcafKeyValue;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispsimpleaddress.primitiveaddress.DistinguishedName;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.locatorrecords.LocatorRecord;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.mapnotifymessage.MapNotifyBuilder;
+import org.opendaylight.lispflowmapping.interfaces.mappingservice.IMappingService;
+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.SourceDestKeyHelper;
+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.lisp.proto.rev151105.MapRegister;
+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.record.container.MappingRecord;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecordBuilder;
+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.maprequestnotification.MapRequestBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingChange;
+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.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.SiteId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.mapping.authkey.container.MappingAuthkey;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-public class MapServer extends AbstractLispComponent implements IMapServerAsync {
-
- protected static final Logger logger = LoggerFactory.getLogger(MapServer.class);
- private boolean overwrite;
-
- public MapServer(ILispDAO dao) {
- this(dao, true);
+import com.google.common.base.Preconditions;
+
+public class MapServer implements IMapServerAsync, OdlMappingserviceListener {
+
+ protected static final Logger LOG = LoggerFactory.getLogger(MapServer.class);
+ private IMappingService mapService;
+ private boolean authenticate;
+ private boolean subscriptionService;
+ private IMapNotifyHandler notifyHandler;
+ private NotificationService notificationService;
+
+ public MapServer(IMappingService mapService, boolean authenticate, boolean subscriptionService,
+ IMapNotifyHandler notifyHandler, NotificationService notificationService) {
+ Preconditions.checkNotNull(mapService);
+ this.mapService = mapService;
+ this.authenticate = authenticate;
+ this.subscriptionService = subscriptionService;
+ this.notifyHandler = notifyHandler;
+ this.notificationService = notificationService;
+ if (notificationService != null) {
+ notificationService.registerNotificationListener(this);
+ }
}
- public MapServer(ILispDAO dao, boolean overwrite) {
- this(dao, overwrite, true);
+ @Override
+ public void setSubscriptionService(boolean subscriptionService) {
+ this.subscriptionService = subscriptionService;
}
- public MapServer(ILispDAO dao, boolean overwrite, boolean authenticate) {
- this(dao, overwrite, authenticate, true);
+ @Override
+ public boolean shouldAuthenticate() {
+ return authenticate;
}
- public MapServer(ILispDAO dao, boolean overwrite, boolean authenticate, boolean iterateAuthenticationMask) {
- super(dao, authenticate, iterateAuthenticationMask);
- this.overwrite = overwrite;
+ @Override
+ public void setShouldAuthenticate(boolean shouldAuthenticate) {
+ authenticate = shouldAuthenticate;
}
- public void handleMapRegister(MapRegister mapRegister, IMapNotifyHandler callback) {
- if (dao == null) {
- logger.warn("handleMapRegister called while dao is uninitialized");
- } else {
- boolean failed = false;
- String password = null;
- for (EidToLocatorRecord eidRecord : mapRegister.getEidToLocatorRecord()) {
- if (shouldAuthenticate()) {
- password = getPassword(eidRecord.getLispAddressContainer(), eidRecord.getMaskLength());
- if (!LispAuthenticationUtil.validate(mapRegister, password)) {
- logger.warn("Authentication failed");
- failed = true;
- break;
- }
+ public void handleMapRegister(MapRegister mapRegister) {
+ boolean failed = false;
+ String password = null;
+ for (MappingRecordItem record : mapRegister.getMappingRecordItem()) {
+ MappingRecord mapping = record.getMappingRecord();
+ if (authenticate) {
+ MappingAuthkey authkey = mapService.getAuthenticationKey(mapping.getEid());
+ if (authkey != null) {
+ password = authkey.getKeyString();
}
- saveRlocs(eidRecord);
-
- }
- if (!failed) {
- MapNotifyBuilder builder = new MapNotifyBuilder();
- if (BooleanUtils.isTrue(mapRegister.isWantMapNotify())) {
- logger.trace("MapRegister wants MapNotify");
- MapNotifyBuilderHelper.setFromMapRegister(builder, mapRegister);
- if (shouldAuthenticate()) {
- builder.setAuthenticationData(LispAuthenticationUtil.createAuthenticationData(builder.build(), password));
- }
- callback.handleMapNotify(builder.build());
+ if (!LispAuthenticationUtil.validate(mapRegister, password)) {
+ LOG.warn("Authentication failed");
+ failed = true;
+ break;
}
}
+ MappingRecord oldMapping = (MappingRecord) mapService.getMapping(MappingOrigin.Southbound,
+ mapping.getEid());
+ mapService.addMapping(MappingOrigin.Southbound, mapping.getEid(), getSiteId(mapRegister), mapping);
+ if (subscriptionService && !mapping.equals(oldMapping)) {
+ LOG.debug("Sending SMRs for subscribers of {}", mapping.getEid());
+ Set<SubscriberRLOC> subscribers = getSubscribers(mapping.getEid());
+ sendSmrs(mapping, subscribers);
+ }
}
- }
-
- public void saveRlocs(EidToLocatorRecord eidRecord) {
- IMappingServiceKey key = MappingServiceKeyUtil.generateMappingServiceKey(eidRecord.getLispAddressContainer(), eidRecord.getMaskLength());
- Map<String, MappingServiceRLOCGroup> rlocGroups = new HashMap<String, MappingServiceRLOCGroup>();
- if (eidRecord.getLocatorRecord() != null) {
- for (LocatorRecord locatorRecord : eidRecord.getLocatorRecord()) {
- String subkey = getLocatorKey(locatorRecord);
- if (!rlocGroups.containsKey(subkey)) {
- rlocGroups.put(subkey, new MappingServiceRLOCGroup(eidRecord.getRecordTtl(), eidRecord.getAction(), eidRecord.isAuthoritative()));
+ if (!failed) {
+ MapNotifyBuilder builder = new MapNotifyBuilder();
+ if (BooleanUtils.isTrue(mapRegister.isWantMapNotify())) {
+ LOG.trace("MapRegister wants MapNotify");
+ MapNotifyBuilderHelper.setFromMapRegister(builder, mapRegister);
+ if (authenticate) {
+ builder.setAuthenticationData(LispAuthenticationUtil.createAuthenticationData(builder.build(),
+ password));
}
- rlocGroups.get(subkey).addRecord(locatorRecord);
+ notifyHandler.handleMapNotify(builder.build());
}
}
- List<MappingEntry<MappingServiceRLOCGroup>> entries = new ArrayList<>();
- for (String subkey : rlocGroups.keySet()) {
- entries.add(new MappingEntry<>(subkey, rlocGroups.get(subkey)));
- }
- dao.put(key, entries.toArray(new MappingEntry[entries.size()]));
}
- private String getLocatorKey(LocatorRecord locatorRecord) {
- if (locatorRecord.getLispAddressContainer().getAddress() instanceof LcafKeyValue) {
- LcafKeyValue keyVal = (LcafKeyValue) locatorRecord.getLispAddressContainer().getAddress();
- if (keyVal.getKey().getPrimitiveAddress() instanceof DistinguishedName) {
- return ((DistinguishedName) keyVal.getKey().getPrimitiveAddress()).getDistinguishedName();
+ private SiteId getSiteId(MapRegister mapRegister) {
+ return (mapRegister.getSiteId() != null) ? new SiteId(mapRegister.getSiteId()) : null;
+ }
+
+ @Override
+ public void onMappingChanged(MappingChanged notification) {
+ if (subscriptionService) {
+ sendSmrs(notification.getMappingRecord(), getSubscribers(notification.getMappingRecord().getEid()));
+ if (notification.getChangeType().equals(MappingChange.Removed)) {
+ removeSubscribers(notification.getMappingRecord().getEid());
}
}
- if (shouldOverwrite()) {
- return ADDRESS_SUBKEY;
- } else {
- return String.valueOf(locatorRecord.getLispAddressContainer().getAddress().hashCode());
- }
}
- public String getAuthenticationKey(LispAddressContainer address, int maskLen) {
- return getPassword(address, maskLen);
+ private void sendSmrs(MappingRecord record, Set<SubscriberRLOC> subscribers) {
+ Eid eid = record.getEid();
+ handleSmr(record, subscribers, notifyHandler);
+
+ // For SrcDst LCAF also send SMRs to Dst prefix
+ if (eid.getAddress() instanceof SourceDestKey) {
+ Eid dstAddr = SourceDestKeyHelper.getDst(eid);
+ subscribers = getSubscribers(dstAddr);
+ MappingRecord newRecord = new MappingRecordBuilder().setAction(record.getAction())
+ .setAuthoritative(record.isAuthoritative()).setLocatorRecord(record.getLocatorRecord())
+ .setMapVersion(record.getMapVersion()).setRecordTtl(record.getRecordTtl())
+ .setEid(dstAddr).build();
+ handleSmr(newRecord, subscribers, notifyHandler);
+ }
}
- public boolean removeAuthenticationKey(LispAddressContainer address, int maskLen) {
- IMappingServiceKey key = MappingServiceKeyUtil.generateMappingServiceKey(address, maskLen);
- dao.removeSpecific(key, PASSWORD_SUBKEY);
- return true;
+ private void handleSmr(MappingRecord record, Set<SubscriberRLOC> subscribers, IMapNotifyHandler callback) {
+ if (subscribers == null) {
+ return;
+ }
+ MapRequestBuilder mrb = MapRequestUtil.prepareSMR(record.getEid(), 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);
+ } 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.setEidItem(new ArrayList<EidItem>());
+ mrb.getEidItem().add(new EidItemBuilder().setEid(subscriber.getSrcEid()).build());
+ callback.handleSMR(mrb.build(), subscriber.getSrcRloc());
+ } catch (Exception e) {
+ LOG.error("Errors encountered while handling SMR:" + ExceptionUtils.getStackTrace(e));
+ }
+ }
+ }
+ addSubscribers(record.getEid(), subscribers);
}
- public boolean addAuthenticationKey(LispAddressContainer address, int maskLen, String key) {
- IMappingServiceKey mappingServiceKey = MappingServiceKeyUtil.generateMappingServiceKey(address, maskLen);
- dao.put(mappingServiceKey, new MappingEntry<String>(PASSWORD_SUBKEY, key));
- return true;
+ @SuppressWarnings("unchecked")
+ private Set<SubscriberRLOC> getSubscribers(Eid address) {
+ return (Set<SubscriberRLOC>) mapService.getData(MappingOrigin.Southbound, address, SubKeys.SUBSCRIBERS);
}
- public boolean shouldOverwrite() {
- return overwrite;
+ private void removeSubscribers(Eid address) {
+ mapService.removeData(MappingOrigin.Southbound, address, SubKeys.SUBSCRIBERS);
}
- public void setOverwrite(boolean overwrite) {
- this.overwrite = overwrite;
+ private void addSubscribers(Eid address, Set<SubscriberRLOC> subscribers) {
+ mapService.addData(MappingOrigin.Southbound, address, SubKeys.SUBSCRIBERS, subscribers);
}
+ 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;
+ }
+ }
+ } catch (SocketException se) {
+ }
+ return null;
+ }
}
Code Review / lispflowmapping.git / blobdiff