mappingservice/implementation/src/main/java/org/opendaylight/lispflowmapping/implementation/util/LispNotificationHelper.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.implementation.util;
   9
  10 import java.net.Inet4Address;
  11 import java.net.InetAddress;
  12 import java.net.UnknownHostException;
  13 import java.util.ArrayList;
  14 import java.util.List;
  15
  16 import org.opendaylight.lispflowmapping.implementation.serializer.LispMessage;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.AddMapping;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.EidToLocatorRecord;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafApplicationDataAddress;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispAFIAddress;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispDistinguishedNameAddress;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.LispAddressContainer;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.distinguishedname.DistinguishedName;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafkeyvalue.LcafKeyValueAddressAddr;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.PrimitiveAddress;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapnotifynotification.MapNotify;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapnotifynotification.MapNotifyBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapregisternotification.MapRegister;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapregisternotification.MapRegisterBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapreplynotification.MapReply;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapreplynotification.MapReplyBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequestnotification.MapRequest;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequestnotification.MapRequestBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddress;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddressBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.EidUri;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.MappingOrigin;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.SiteId;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.db.instance.Mapping;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.db.instance.MappingBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddressBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
  44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv6Address;
  45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
  46
  47 public class LispNotificationHelper {
  48
  49     public static MapRegister convertMapRegister(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRegister mapRegister) {
  50         return new MapRegisterBuilder().setAuthenticationData(mapRegister.getAuthenticationData())
  51                 .setEidToLocatorRecord(mapRegister.getEidToLocatorRecord()).setKeyId(mapRegister.getKeyId()).setNonce(mapRegister.getNonce())
  52                 .setProxyMapReply(mapRegister.isProxyMapReply()).setWantMapNotify(mapRegister.isWantMapNotify())
  53                 .setXtrSiteIdPresent(mapRegister.isXtrSiteIdPresent()).setXtrId(mapRegister.getXtrId()).setSiteId(mapRegister.getSiteId()).build();
  54     }
  55
  56     public static MapNotify convertMapNotify(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapNotify mapNotify) {
  57         return new MapNotifyBuilder().setAuthenticationData(mapNotify.getAuthenticationData())
  58                 .setEidToLocatorRecord(mapNotify.getEidToLocatorRecord()).setKeyId(mapNotify.getKeyId()).setNonce(mapNotify.getNonce())
  59                 .setXtrSiteIdPresent(mapNotify.isXtrSiteIdPresent()).setXtrId(mapNotify.getXtrId()).setSiteId(mapNotify.getSiteId()).build();
  60     }
  61
  62     public static MapRequest convertMapRequest(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRequest mapRequest) {
  63         return new MapRequestBuilder().setAuthoritative(mapRequest.isAuthoritative()).setEidRecord(mapRequest.getEidRecord())
  64                 .setItrRloc(mapRequest.getItrRloc()).setMapDataPresent(mapRequest.isMapDataPresent()).setMapReply(mapRequest.getMapReply())
  65                 .setNonce(mapRequest.getNonce()).setPitr(mapRequest.isPitr()).setProbe(mapRequest.isProbe()).setSmr(mapRequest.isSmr())
  66                 .setSmrInvoked(mapRequest.isSmrInvoked()).setSourceEid(mapRequest.getSourceEid()).build();
  67     }
  68
  69     public static MapReply convertMapReply(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapReply mapReply) {
  70         return new MapReplyBuilder().setEchoNonceEnabled(mapReply.isEchoNonceEnabled()).setEidToLocatorRecord(mapReply.getEidToLocatorRecord())
  71                 .setNonce(mapReply.getNonce()).setProbe(mapReply.isProbe()).setSecurityEnabled(mapReply.isSecurityEnabled()).build();
  72     }
  73
  74     public static IpAddress getIpAddressFromInetAddress(InetAddress inetAddress) {
  75         if (inetAddress == null) {
  76             inetAddress = InetAddress.getLoopbackAddress();
  77         }
  78         if (inetAddress instanceof Inet4Address) {
  79             return new IpAddress(new Ipv4Address(inetAddress.getHostAddress()));
  80         } else {
  81             return new IpAddress(new Ipv6Address(inetAddress.getHostAddress()));
  82         }
  83     }
  84
  85     public static InetAddress getInetAddressFromIpAddress(IpAddress ipAddress) {
  86         InetAddress address = null;
  87         if (ipAddress != null) {
  88             if (ipAddress.getIpv4Address() != null) {
  89                 address = getAddressByName(ipAddress.getIpv4Address().getValue());
  90             } else if (ipAddress.getIpv6Address() != null) {
  91                 address = getAddressByName(ipAddress.getIpv6Address().getValue());
  92             }
  93         }
  94         if (address == null) {
  95             address = InetAddress.getLoopbackAddress();
  96         }
  97         return address;
  98     }
  99
 100     public static TransportAddress getTransportAddressFromContainer(LispAddressContainer container) {
 101         TransportAddressBuilder tab = new TransportAddressBuilder();
 102         LispAFIAddress address = LispAFIConvertor.toAFI(container);
 103         if (address instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address) {
 104             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address) address).getIpv4Address().getValue()));
 105             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
 106         } else if (address instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv6.Ipv6Address) {
 107             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv6.Ipv6Address) address).getIpv6Address().getValue()));
 108             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
 109         } else if (address instanceof LcafKeyValueAddressAddr) {
 110             PrimitiveAddress primitiveAddress = ((LcafKeyValueAddressAddr) address).getValue().getPrimitiveAddress();
 111             if (primitiveAddress instanceof LispDistinguishedNameAddress) {
 112                 String value = ((LispDistinguishedNameAddress) primitiveAddress).getDistinguishedName();
 113                 String ip = value.split(":")[0];
 114                 int port = Integer.valueOf(value.split(":")[1]);
 115                 tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
 116                 tab.setPort(new PortNumber(port));
 117             }
 118         } else if (address instanceof DistinguishedName) {
 119             DistinguishedName dname = (DistinguishedName) address;
 120             String value = dname.getDistinguishedName();
 121             String ip = value.split(":")[0];
 122             int port = Integer.valueOf(value.split(":")[1]);
 123
 124             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
 125             tab.setPort(new PortNumber(port));
 126         } else if (address instanceof LcafApplicationDataAddress) {
 127             LcafApplicationDataAddress appData = (LcafApplicationDataAddress) address;
 128             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) appData.getAddress().getPrimitiveAddress()).getIpv4Address()
 129                     .getIpv4Address().getValue()));
 130             tab.setPort(new PortNumber(appData.getLocalPortLow()));
 131         }
 132         return tab.build();
 133     }
 134
 135     public static InetAddress getAddressByName(String IPAddress) {
 136         try {
 137             InetAddress address = InetAddress.getByName(IPAddress);
 138             return address;
 139         } catch (UnknownHostException e) {
 140             return null;
 141         }
 142     }
 143
 144     public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
 145         List<Mapping> mappings = new ArrayList<Mapping>();
 146         for (int i=0; i<mapRegisterNotification.getMapRegister().getEidToLocatorRecord().size(); i++) {
 147             EidToLocatorRecord record = mapRegisterNotification.getMapRegister().getEidToLocatorRecord().get(i);
 148             MappingBuilder mb = new MappingBuilder();
 149             mb.setEid(new EidUri(LispAddressStringifier.getURIString(
 150                     record.getLispAddressContainer(), record.getMaskLength())));
 151             mb.setOrigin(MappingOrigin.Southbound);
 152             mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
 153             mb.setRecordTtl(record.getRecordTtl());
 154             mb.setMaskLength(record.getMaskLength());
 155             mb.setMapVersion(record.getMapVersion());
 156             mb.setAction(record.getAction());
 157             mb.setAuthoritative(record.isAuthoritative());
 158             mb.setLispAddressContainer(record.getLispAddressContainer());
 159             mb.setLocatorRecord(record.getLocatorRecord());
 160             mappings.add(mb.build());
 161         }
 162         return mappings;
 163     }
 164
 165     public static List<SiteId> getSiteId(MapRegister mapRegister) {
 166         if (mapRegister.isXtrSiteIdPresent()) {
 167             List<SiteId> siteIds = new ArrayList<SiteId>();
 168             SiteId siteId = new SiteId(mapRegister.getSiteId());
 169             siteIds.add(siteId);
 170             return siteIds;
 171         } else {
 172             return null;
 173         }
 174     }
 175 }