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