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