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
  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
  18 import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
  19 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.AddMapping;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping.record.container.MappingRecord;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapregisternotification.MapRegister;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddress;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddressBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.EidUri;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddressBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
  33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.SimpleAddress;
  34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.Address;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.ApplicationData;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.DistinguishedName;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6;
  39 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.KeyValueAddress;
  40 import org.slf4j.Logger;
  41 import org.slf4j.LoggerFactory;
  42
  43 public final class LispNotificationHelper {
  44     protected static final Logger LOG = LoggerFactory.getLogger(LispNotificationHelper.class);
  45     private static final Splitter COLON_SPLITTER = Splitter.on(':');
  46
  47     // Utility class, should not be instantiated
  48     private LispNotificationHelper() {
  49     }
  50
  51     public static TransportAddress getTransportAddressFromRloc(Rloc rloc) {
  52         TransportAddressBuilder tab = new TransportAddressBuilder();
  53         Address address = rloc.getAddress();
  54         if (address instanceof Ipv4) {
  55             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((Ipv4) address).getIpv4().getValue()));
  56             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  57         } else if (address instanceof Ipv6) {
  58             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((Ipv6) address).getIpv6().getValue()));
  59             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  60         } else if (address instanceof KeyValueAddress) {
  61             SimpleAddress sa = ((KeyValueAddress) address).getKeyValueAddress().getValue();
  62             if (sa.getDistinguishedNameType() != null) {
  63                 final Iterator<String> it = COLON_SPLITTER.split(sa.getDistinguishedNameType().getValue()).iterator();
  64                 String ip = it.next();
  65                 int port = Integer.valueOf(it.next());
  66
  67                 tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
  68                 tab.setPort(new PortNumber(port));
  69             }
  70         } else if (address instanceof DistinguishedName) {
  71             DistinguishedName dname = (DistinguishedName) address;
  72             final Iterator<String> it = COLON_SPLITTER.split(dname.getDistinguishedName().getValue()).iterator();
  73             String ip = it.next();
  74             int port = Integer.valueOf(it.next());
  75
  76             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
  77             tab.setPort(new PortNumber(port));
  78         } else if (address instanceof ApplicationData) {
  79             ApplicationData appData = (ApplicationData) address;
  80             tab.setIpAddress(appData.getApplicationData().getAddress().getIpAddress());
  81             tab.setPort(new PortNumber(appData.getApplicationData().getLocalPortLow()));
  82         }
  83         return tab.build();
  84     }
  85
  86     public static InetAddress getAddressByName(String IPAddress) {
  87         try {
  88             InetAddress address = InetAddress.getByName(IPAddress);
  89             return address;
  90         } catch (UnknownHostException e) {
  91             LOG.debug("Unknown host {}", IPAddress, e);
  92             return null;
  93         }
  94     }
  95
  96     public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
  97         List<Mapping> mappings = new ArrayList<Mapping>();
  98         for (int i=0; i<mapRegisterNotification.getMapRegister().getMappingRecordItem().size(); i++) {
  99             MappingRecord record = mapRegisterNotification.getMapRegister().getMappingRecordItem().get(i).getMappingRecord();
 100             MappingBuilder mb = new MappingBuilder();
 101             mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(
 102                     record.getEid())));
 103             mb.setOrigin(MappingOrigin.Southbound);
 104             mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
 105             mb.setMappingRecord(record);
 106             mappings.add(mb.build());
 107         }
 108         return mappings;
 109     }
 110
 111     public static List<SiteId> getSiteId(MapRegister mapRegister) {
 112         if (mapRegister.isXtrSiteIdPresent()) {
 113             List<SiteId> siteIds = new ArrayList<SiteId>();
 114             SiteId siteId = new SiteId(mapRegister.getSiteId());
 115             siteIds.add(siteId);
 116             return siteIds;
 117         } else {
 118             return null;
 119         }
 120     }
 121 }