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