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.LispAFIConvertor;
  17 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.AddMapping;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.EidToLocatorRecord;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafApplicationDataAddress;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispAFIAddress;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispDistinguishedNameAddress;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.LispAddressContainer;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.distinguishedname.DistinguishedName;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafkeyvalue.LcafKeyValueAddressAddr;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.PrimitiveAddress;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapregisternotification.MapRegister;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddress;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddressBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.EidUri;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.MappingOrigin;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.SiteId;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.db.instance.Mapping;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mapping.database.rev150314.db.instance.MappingBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddressBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
  37
  38 public class LispNotificationHelper {
  39     public static TransportAddress getTransportAddressFromContainer(LispAddressContainer container) {
  40         TransportAddressBuilder tab = new TransportAddressBuilder();
  41         LispAFIAddress address = LispAFIConvertor.toAFI(container);
  42         if (address instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address) {
  43             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address) address).getIpv4Address().getValue()));
  44             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  45         } else if (address instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv6.Ipv6Address) {
  46             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv6.Ipv6Address) address).getIpv6Address().getValue()));
  47             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
  48         } else if (address instanceof LcafKeyValueAddressAddr) {
  49             PrimitiveAddress primitiveAddress = ((LcafKeyValueAddressAddr) address).getValue().getPrimitiveAddress();
  50             if (primitiveAddress instanceof LispDistinguishedNameAddress) {
  51                 String value = ((LispDistinguishedNameAddress) primitiveAddress).getDistinguishedName();
  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();
  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 LcafApplicationDataAddress) {
  66             LcafApplicationDataAddress appData = (LcafApplicationDataAddress) address;
  67             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.primitiveaddress.Ipv4) appData.getAddress().getPrimitiveAddress()).getIpv4Address()
  68                     .getIpv4Address().getValue()));
  69             tab.setPort(new PortNumber(appData.getLocalPortLow()));
  70         }
  71         return tab.build();
  72     }
  73
  74     public static InetAddress getAddressByName(String IPAddress) {
  75         try {
  76             InetAddress address = InetAddress.getByName(IPAddress);
  77             return address;
  78         } catch (UnknownHostException e) {
  79             return null;
  80         }
  81     }
  82
  83     public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
  84         List<Mapping> mappings = new ArrayList<Mapping>();
  85         for (int i=0; i<mapRegisterNotification.getMapRegister().getEidToLocatorRecord().size(); i++) {
  86             EidToLocatorRecord record = mapRegisterNotification.getMapRegister().getEidToLocatorRecord().get(i);
  87             MappingBuilder mb = new MappingBuilder();
  88             mb.setEid(new EidUri(LispAddressStringifier.getURIString(
  89                     record.getLispAddressContainer(), record.getMaskLength())));
  90             mb.setOrigin(MappingOrigin.Southbound);
  91             mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
  92             mb.setRecordTtl(record.getRecordTtl());
  93             mb.setMaskLength(record.getMaskLength());
  94             mb.setMapVersion(record.getMapVersion());
  95             mb.setAction(record.getAction());
  96             mb.setAuthoritative(record.isAuthoritative());
  97             mb.setLispAddressContainer(record.getLispAddressContainer());
  98             mb.setLocatorRecord(record.getLocatorRecord());
  99             mappings.add(mb.build());
 100         }
 101         return mappings;
 102     }
 103
 104     public static List<SiteId> getSiteId(MapRegister mapRegister) {
 105         if (mapRegister.isXtrSiteIdPresent()) {
 106             List<SiteId> siteIds = new ArrayList<SiteId>();
 107             SiteId siteId = new SiteId(mapRegister.getSiteId());
 108             siteIds.add(siteId);
 109             return siteIds;
 110         } else {
 111             return null;
 112         }
 113     }
 114 }