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 class LispNotificationHelper {
  39     public static TransportAddress getTransportAddressFromRloc(Rloc rloc) {
  40         TransportAddressBuilder tab = new TransportAddressBuilder();
  41         Address address = rloc.getAddress();
  42         if (address instanceof Ipv4) {
  43             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((Ipv4) address).getIpv4().getValue()));
  44             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  45         } else if (address instanceof Ipv6) {
  46             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((Ipv6) address).getIpv6().getValue()));
  47             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  48         } else if (address instanceof KeyValueAddress) {
  49             SimpleAddress sa = ((KeyValueAddress) address).getKeyValueAddress().getValue();
  50             if (sa.getDistinguishedNameType() != null) {
  51                 String value = sa.getDistinguishedNameType().getValue();
  52                 String ip = value.split(":")[0];
  53                 int port = Integer.valueOf(value.split(":")[1]);
  54                 tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
  55                 tab.setPort(new PortNumber(port));
  56             }
  57         } else if (address instanceof DistinguishedName) {
  58             DistinguishedName dname = (DistinguishedName) address;
  59             String value = dname.getDistinguishedName().getValue();
  60             String ip = value.split(":")[0];
  61             int port = Integer.valueOf(value.split(":")[1]);
  62
  63             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
  64             tab.setPort(new PortNumber(port));
  65         } else if (address instanceof ApplicationData) {
  66             ApplicationData appData = (ApplicationData) address;
  67             tab.setIpAddress(appData.getApplicationData().getAddress().getIpAddress());
  68             tab.setPort(new PortNumber(appData.getApplicationData().getLocalPortLow()));
  69         }
  70         return tab.build();
  71     }
  72
  73     public static InetAddress getAddressByName(String IPAddress) {
  74         try {
  75             InetAddress address = InetAddress.getByName(IPAddress);
  76             return address;
  77         } catch (UnknownHostException e) {
  78             return null;
  79         }
  80     }
  81
  82     public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
  83         List<Mapping> mappings = new ArrayList<Mapping>();
  84         for (int i=0; i<mapRegisterNotification.getMapRegister().getMappingRecordItem().size(); i++) {
  85             MappingRecord record = mapRegisterNotification.getMapRegister().getMappingRecordItem().get(i).getMappingRecord();
  86             MappingBuilder mb = new MappingBuilder();
  87             mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(
  88                     record.getEid())));
  89             mb.setOrigin(MappingOrigin.Southbound);
  90             mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
  91             mb.setMappingRecord(record);
  92             mappings.add(mb.build());
  93         }
  94         return mappings;
  95     }
  96
  97     public static List<SiteId> getSiteId(MapRegister mapRegister) {
  98         if (mapRegister.isXtrSiteIdPresent()) {
  99             List<SiteId> siteIds = new ArrayList<SiteId>();
 100             SiteId siteId = new SiteId(mapRegister.getSiteId());
 101             siteIds.add(siteId);
 102             return siteIds;
 103         } else {
 104             return null;
 105         }
 106     }
 107 }