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 com.google.common.base.Splitter;
  11 import com.google.common.net.InetAddresses;
  12
  13 import java.net.InetAddress;
  14 import java.net.UnknownHostException;
  15 import java.util.ArrayList;
  16 import java.util.Iterator;
  17 import java.util.List;
  18
  19 import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
  20 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinary;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinaryBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv4AddressBinary;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv6AddressBinary;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv4Binary;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv6Binary;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.AddMapping;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapregisternotification.MapRegister;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddress;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddressBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.EidUri;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
  40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.SimpleAddress;
  41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.Address;
  42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.ApplicationData;
  43 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.DistinguishedName;
  44 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
  45 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6;
  46 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.KeyValueAddress;
  47 import org.slf4j.Logger;
  48 import org.slf4j.LoggerFactory;
  49
  50 public final class LispNotificationHelper {
  51     protected static final Logger LOG = LoggerFactory.getLogger(LispNotificationHelper.class);
  52     private static final Splitter COLON_SPLITTER = Splitter.on(':');
  53
  54     // Utility class, should not be instantiated
  55     private LispNotificationHelper() {
  56     }
  57
  58     public static TransportAddress getTransportAddressFromRloc(Rloc rloc) {
  59         TransportAddressBuilder tab = new TransportAddressBuilder();
  60         Address address = rloc.getAddress();
  61         // once we switch to using Ipv4Binary and Ipv6Binary by default for RLOCs, we will only need to keep the below
  62         // for backwards compatibility reasons, but the default will be to just return the reference for the binary
  63         // object, yey!
  64         if (address instanceof Ipv4) {
  65             String ipv4 = ((Ipv4) address).getIpv4().getValue();
  66             tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv4).getAddress()));
  67             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  68         } else if (address instanceof Ipv6) {
  69             String ipv6 = ((Ipv6) address).getIpv6().getValue();
  70             tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv6).getAddress()));
  71             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  72         } else if (address instanceof Ipv4Binary) {
  73             Ipv4AddressBinary ipv6 = ((Ipv4Binary) address).getIpv4Binary();
  74             tab.setIpAddress(new IpAddressBinary(ipv6));
  75             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  76         } else if (address instanceof Ipv6Binary) {
  77             Ipv6AddressBinary ipv6 = ((Ipv6Binary) address).getIpv6Binary();
  78             tab.setIpAddress(new IpAddressBinary(ipv6));
  79             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  80         } else if (address instanceof KeyValueAddress) {
  81             SimpleAddress sa = ((KeyValueAddress) address).getKeyValueAddress().getValue();
  82             if (sa.getDistinguishedNameType() != null) {
  83                 final Iterator<String> it = COLON_SPLITTER.split(sa.getDistinguishedNameType().getValue()).iterator();
  84                 String ip = it.next();
  85                 int port = Integer.valueOf(it.next());
  86
  87                 tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ip).getAddress()));
  88                 tab.setPort(new PortNumber(port));
  89             }
  90         } else if (address instanceof DistinguishedName) {
  91             DistinguishedName dname = (DistinguishedName) address;
  92             final Iterator<String> it = COLON_SPLITTER.split(dname.getDistinguishedName().getValue()).iterator();
  93             String ip = it.next();
  94             int port = Integer.valueOf(it.next());
  95
  96             tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ip).getAddress()));
  97             tab.setPort(new PortNumber(port));
  98         } else if (address instanceof ApplicationData) {
  99             ApplicationData appData = (ApplicationData) address;
 100             tab.setIpAddress(getIpAddressBinary(appData.getApplicationData().getAddress().getIpAddress()));
 101             tab.setPort(new PortNumber(appData.getApplicationData().getLocalPortLow()));
 102         }
 103         return tab.build();
 104     }
 105
 106     public static InetAddress getAddressByName(String IPAddress) {
 107         try {
 108             InetAddress address = InetAddress.getByName(IPAddress);
 109             return address;
 110         } catch (UnknownHostException e) {
 111             LOG.debug("Unknown host {}", IPAddress, e);
 112             return null;
 113         }
 114     }
 115
 116     public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
 117         List<Mapping> mappings = new ArrayList<Mapping>();
 118         for (int i=0; i<mapRegisterNotification.getMapRegister().getMappingRecordItem().size(); i++) {
 119             MappingRecord record = mapRegisterNotification.getMapRegister().getMappingRecordItem().get(i)
 120                     .getMappingRecord();
 121             MappingBuilder mb = new MappingBuilder();
 122             mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(
 123                     record.getEid())));
 124             mb.setOrigin(MappingOrigin.Southbound);
 125             mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
 126             mb.setMappingRecord(record);
 127             mappings.add(mb.build());
 128         }
 129         return mappings;
 130     }
 131
 132     public static List<SiteId> getSiteId(MapRegister mapRegister) {
 133         if (mapRegister.isXtrSiteIdPresent()) {
 134             List<SiteId> siteIds = new ArrayList<SiteId>();
 135             SiteId siteId = new SiteId(mapRegister.getSiteId());
 136             siteIds.add(siteId);
 137             return siteIds;
 138         } else {
 139             return null;
 140         }
 141     }
 142
 143     // We will see if we need to keep this method post full binary-migration, and if yes, will probably move to
 144     // LispAddressUtil (and add tests!)
 145     private static IpAddressBinary getIpAddressBinary(IpAddress address) {
 146         if (address.getIpv4Address() != null) {
 147             String ipv4 = address.getIpv4Address().getValue();
 148             return IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv4).getAddress());
 149         } else if (address.getIpv6Address() != null) {
 150             String ipv6 = address.getIpv6Address().getValue();
 151             return IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv6).getAddress());
 152         }
 153         return null;
 154     }
 155 }