Prepare the YANG models for RESTCONF
[lispflowmapping.git] / 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.Inet4Address;
11 import java.net.InetAddress;
12 import java.net.UnknownHostException;
13
14 import org.opendaylight.lispflowmapping.implementation.serializer.LispMessage;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LcafApplicationDataAddress;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispAFIAddress;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispDistinguishedNameAddress;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.LispIpv4Address;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.LispAddressContainer;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.distinguishedname.DistinguishedName;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.lcafkeyvalue.LcafKeyValueAddressAddr;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispsimpleaddress.PrimitiveAddress;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapregisternotification.MapRegister;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapregisternotification.MapRegisterBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapreplynotification.MapReply;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.mapreplynotification.MapReplyBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequestnotification.MapRequest;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.maprequestnotification.MapRequestBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddress;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.transportaddress.TransportAddressBuilder;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddressBuilder;
33 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
34 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv6Address;
35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.PortNumber;
36
37 public class LispNotificationHelper {
38
39     public static MapRegister convertMapRegister(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRegister mapRegister) {
40         return new MapRegisterBuilder().setAuthenticationData(mapRegister.getAuthenticationData())
41                 .setEidToLocatorRecord(mapRegister.getEidToLocatorRecord()).setKeyId(mapRegister.getKeyId()).setNonce(mapRegister.getNonce())
42                 .setProxyMapReply(mapRegister.isProxyMapReply()).setWantMapNotify(mapRegister.isWantMapNotify())
43                 .setXtrSiteIdPresent(mapRegister.isXtrSiteIdPresent()).setXtrId(mapRegister.getXtrId()).setSiteId(mapRegister.getSiteId()).build();
44     }
45
46     public static MapRequest convertMapRequest(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapRequest mapRequest) {
47         return new MapRequestBuilder().setAuthoritative(mapRequest.isAuthoritative()).setEidRecord(mapRequest.getEidRecord())
48                 .setItrRloc(mapRequest.getItrRloc()).setMapDataPresent(mapRequest.isMapDataPresent()).setMapReply(mapRequest.getMapReply())
49                 .setNonce(mapRequest.getNonce()).setPitr(mapRequest.isPitr()).setProbe(mapRequest.isProbe()).setSmr(mapRequest.isSmr())
50                 .setSmrInvoked(mapRequest.isSmrInvoked()).setSourceEid(mapRequest.getSourceEid()).build();
51     }
52
53     public static MapReply convertMapReply(org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.MapReply mapReply) {
54         return new MapReplyBuilder().setEchoNonceEnabled(mapReply.isEchoNonceEnabled()).setEidToLocatorRecord(mapReply.getEidToLocatorRecord())
55                 .setNonce(mapReply.getNonce()).setProbe(mapReply.isProbe()).setSecurityEnabled(mapReply.isSecurityEnabled()).build();
56     }
57
58     public static IpAddress getIpAddressFromInetAddress(InetAddress inetAddress) {
59         if (inetAddress == null) {
60             inetAddress = InetAddress.getLoopbackAddress();
61         }
62         if (inetAddress instanceof Inet4Address) {
63             return new IpAddress(new Ipv4Address(inetAddress.getHostAddress()));
64         } else {
65             return new IpAddress(new Ipv6Address(inetAddress.getHostAddress()));
66         }
67     }
68
69     public static InetAddress getInetAddressFromIpAddress(IpAddress ipAddress) {
70         InetAddress address = null;
71         if (ipAddress != null) {
72             if (ipAddress.getIpv4Address() != null) {
73                 address = getAddressByName(ipAddress.getIpv4Address().getValue());
74             } else if (ipAddress.getIpv6Address() != null) {
75                 address = getAddressByName(ipAddress.getIpv6Address().getValue());
76             }
77         }
78         if (address == null) {
79             address = InetAddress.getLoopbackAddress();
80         }
81         return address;
82     }
83
84     public static TransportAddress getTransportAddressFromContainer(LispAddressContainer container) {
85         TransportAddressBuilder tab = new TransportAddressBuilder();
86         LispAFIAddress address = LispAFIConvertor.toAFI(container);
87         if (address instanceof org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address) {
88             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.control.plane.rev150314.lispaddress.lispaddresscontainer.address.ipv4.Ipv4Address) address).getIpv4Address().getValue()));
89             tab.setPort(new PortNumber(LispMessage.PORT_NUM));
90         } else if (address instanceof LcafKeyValueAddressAddr) {
91             PrimitiveAddress primitiveAddress = ((LcafKeyValueAddressAddr) address).getValue().getPrimitiveAddress();
92             if (primitiveAddress instanceof LispDistinguishedNameAddress) {
93                 String value = ((LispDistinguishedNameAddress) primitiveAddress).getDistinguishedName();
94                 String ip = value.split(":")[0];
95                 int port = Integer.valueOf(value.split(":")[1]);
96                 tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
97                 tab.setPort(new PortNumber(port));
98             }
99         } else if (address instanceof DistinguishedName) {
100             DistinguishedName dname = (DistinguishedName) address;
101             String value = dname.getDistinguishedName();
102             String ip = value.split(":")[0];
103             int port = Integer.valueOf(value.split(":")[1]);
104
105             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(ip));
106             tab.setPort(new PortNumber(port));
107         } else if (address instanceof LcafApplicationDataAddress) {
108             LcafApplicationDataAddress appData = (LcafApplicationDataAddress) address;
109             tab.setIpAddress(IpAddressBuilder.getDefaultInstance(((LispIpv4Address) appData.getAddress().getPrimitiveAddress()).getIpv4Address()
110                     .getValue()));
111             tab.setPort(new PortNumber(appData.getLocalPort()));
112         }
113         return tab.build();
114     }
115
116     public static InetAddress getAddressByName(String IPAddress) {
117         try {
118             InetAddress address = InetAddress.getByName(IPAddress);
119             return address;
120         } catch (UnknownHostException e) {
121             return null;
122         }
123     }
124 }