Bug 5270: Fix merging
[lispflowmapping.git] / mappingservice / implementation / src / main / java / org / opendaylight / lispflowmapping / implementation / util / LispNotificationHelper.java
index 5c773e48fff6e5896cd2c064bccceef4f79cc989..5df1ab1c34b8e388b08bd6a46a80aeec1a10635b 100644 (file)
@@ -7,64 +7,75 @@
  */
 package org.opendaylight.lispflowmapping.implementation.util;
 
-import java.net.Inet4Address;
 import java.net.InetAddress;
 import java.net.UnknownHostException;
+import java.util.ArrayList;
+import java.util.List;
 
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispaddress.LispAddressContainer;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.lispaddress.lispaddresscontainer.address.Ipv4;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.mapregisternotification.MapRegister;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.mapregisternotification.MapRegisterBuilder;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.maprequestnotification.MapRequest;
-import org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.maprequestnotification.MapRequestBuilder;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv6Address;
+import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
+import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.AddMapping;
+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.mapregisternotification.MapRegister;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
+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.EidUri;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.SiteId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddressBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.SimpleAddress;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.Address;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.ApplicationData;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.DistinguishedName;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.KeyValueAddress;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
 
-public class LispNotificationHelper {
+public final class LispNotificationHelper {
+    protected static final Logger LOG = LoggerFactory.getLogger(LispNotificationHelper.class);
 
-    public static MapRegister convertMapRegister(org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.MapRegister mapRegister) {
-        return new MapRegisterBuilder().setAuthenticationData(mapRegister.getAuthenticationData())
-                .setEidToLocatorRecord(mapRegister.getEidToLocatorRecord()).setKeyId(mapRegister.getKeyId()).setNonce(mapRegister.getNonce())
-                .setProxyMapReply(mapRegister.isProxyMapReply()).setWantMapNotify(mapRegister.isWantMapNotify()).build();
+    // Utility class, should not be instantiated
+    private LispNotificationHelper() {
     }
 
-    public static MapRequest convertMapRequest(org.opendaylight.yang.gen.v1.lispflowmapping.rev131031.MapRequest mapRequest) {
-        return new MapRequestBuilder().setAuthoritative(mapRequest.isAuthoritative()).setEidRecord(mapRequest.getEidRecord())
-                .setItrRloc(mapRequest.getItrRloc()).setMapDataPresent(mapRequest.isMapDataPresent()).setMapReply(mapRequest.getMapReply())
-                .setNonce(mapRequest.getNonce()).setPitr(mapRequest.isPitr()).setProbe(mapRequest.isProbe()).setSmr(mapRequest.isSmr())
-                .setSmrInvoked(mapRequest.isSmrInvoked()).setSourceEid(mapRequest.getSourceEid()).build();
-    }
-
-    public static IpAddress getIpAddressFromInetAddress(InetAddress inetAddress) {
-        if (inetAddress == null) {
-            inetAddress = InetAddress.getLoopbackAddress();
-        }
-        if (inetAddress instanceof Inet4Address) {
-            return new IpAddress(new Ipv4Address(inetAddress.getHostAddress()));
-        } else {
-            return new IpAddress(new Ipv6Address(inetAddress.getHostAddress()));
-        }
-    }
-
-    public static InetAddress getInetAddressFromIpAddress(IpAddress ipAddress) {
-        InetAddress address = null;
-        if (ipAddress != null) {
-            if (ipAddress.getIpv4Address() != null) {
-                address = getAddressByName(ipAddress.getIpv4Address().getValue());
-            } else if (ipAddress.getIpv6Address() != null) {
-                address = getAddressByName(ipAddress.getIpv6Address().getValue());
+    public static TransportAddress getTransportAddressFromRloc(Rloc rloc) {
+        TransportAddressBuilder tab = new TransportAddressBuilder();
+        Address address = rloc.getAddress();
+        if (address instanceof Ipv4) {
+            tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((Ipv4) address).getIpv4().getValue()));
+            tab.setPort(new PortNumber(LispMessage.PORT_NUM));
+        } else if (address instanceof Ipv6) {
+            tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((Ipv6) address).getIpv6().getValue()));
+            tab.setPort(new PortNumber(LispMessage.PORT_NUM));
+        } else if (address instanceof KeyValueAddress) {
+            SimpleAddress sa = ((KeyValueAddress) address).getKeyValueAddress().getValue();
+            if (sa.getDistinguishedNameType() != null) {
+                String value = sa.getDistinguishedNameType().getValue();
+                String ip = value.split(":")[0];
+                int port = Integer.valueOf(value.split(":")[1]);
+                tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
+                tab.setPort(new PortNumber(port));
             }
-        }
-        if (address == null) {
-            address = InetAddress.getLoopbackAddress();
-        }
-        return address;
-    }
+        } else if (address instanceof DistinguishedName) {
+            DistinguishedName dname = (DistinguishedName) address;
+            String value = dname.getDistinguishedName().getValue();
+            String ip = value.split(":")[0];
+            int port = Integer.valueOf(value.split(":")[1]);
 
-    public static InetAddress getInetAddressFromContainer(LispAddressContainer container) {
-        Ipv4Address ipAddress = ((Ipv4) LispAFIConvertor.toAFI(container)).getIpv4Address();
-        return getAddressByName(ipAddress.getValue());
+            tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
+            tab.setPort(new PortNumber(port));
+        } else if (address instanceof ApplicationData) {
+            ApplicationData appData = (ApplicationData) address;
+            tab.setIpAddress(appData.getApplicationData().getAddress().getIpAddress());
+            tab.setPort(new PortNumber(appData.getApplicationData().getLocalPortLow()));
+        }
+        return tab.build();
     }
 
     public static InetAddress getAddressByName(String IPAddress) {
@@ -72,6 +83,33 @@ public class LispNotificationHelper {
             InetAddress address = InetAddress.getByName(IPAddress);
             return address;
         } catch (UnknownHostException e) {
+            LOG.debug("Unknown host {}", IPAddress, e);
+            return null;
+        }
+    }
+
+    public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
+        List<Mapping> mappings = new ArrayList<Mapping>();
+        for (int i=0; i<mapRegisterNotification.getMapRegister().getMappingRecordItem().size(); i++) {
+            MappingRecord record = mapRegisterNotification.getMapRegister().getMappingRecordItem().get(i).getMappingRecord();
+            MappingBuilder mb = new MappingBuilder();
+            mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(
+                    record.getEid())));
+            mb.setOrigin(MappingOrigin.Southbound);
+            mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
+            mb.setMappingRecord(record);
+            mappings.add(mb.build());
+        }
+        return mappings;
+    }
+
+    public static List<SiteId> getSiteId(MapRegister mapRegister) {
+        if (mapRegister.isXtrSiteIdPresent()) {
+            List<SiteId> siteIds = new ArrayList<SiteId>();
+            SiteId siteId = new SiteId(mapRegister.getSiteId());
+            siteIds.add(siteId);
+            return siteIds;
+        } else {
             return null;
         }
     }