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