Bump MRI upstreams
[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 com.google.common.base.Splitter;
11 import com.google.common.net.InetAddresses;
12 import java.net.InetAddress;
13 import java.net.UnknownHostException;
14 import java.util.ArrayList;
15 import java.util.Iterator;
16 import java.util.List;
17 import java.util.Set;
18 import org.opendaylight.lispflowmapping.lisp.type.LispMessage;
19 import org.opendaylight.lispflowmapping.lisp.util.LispAddressStringifier;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.SimpleAddress;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.Address;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.ApplicationData;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.DistinguishedName;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.KeyValueAddress;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinary;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.IpAddressBinaryBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv4AddressBinary;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.inet.binary.types.rev160303.Ipv6AddressBinary;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv4Binary;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.binary.address.types.rev160504.augmented.lisp.address.address.Ipv6Binary;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.AddMapping;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.SiteId;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapping._record.container.MappingRecord;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.mapregisternotification.MapRegister;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.rloc.container.Rloc;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddress;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.transport.address.TransportAddressBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.EidUri;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.MappingOrigin;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.Mapping;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.mappingservice.rev150906.db.instance.MappingBuilder;
46 import org.opendaylight.yangtools.yang.common.Uint16;
47 import org.slf4j.Logger;
48 import org.slf4j.LoggerFactory;
49
50 public final class LispNotificationHelper {
51     private static final Logger LOG = LoggerFactory.getLogger(LispNotificationHelper.class);
52     private static final Splitter COLON_SPLITTER = Splitter.on(':');
53
54     // Utility class, should not be instantiated
55     private LispNotificationHelper() {
56     }
57
58     public static TransportAddress getTransportAddressFromRloc(Rloc rloc) {
59         TransportAddressBuilder tab = new TransportAddressBuilder();
60         Address address = rloc.getAddress();
61         // once we switch to using Ipv4Binary and Ipv6Binary by default for RLOCs, we will only need to keep the below
62         // for backwards compatibility reasons, but the default will be to just return the reference for the binary
63         // object, yey!
64         if (address instanceof Ipv4) {
65             String ipv4 = ((Ipv4) address).getIpv4().getValue();
66             tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv4).getAddress()));
67             tab.setPort(new PortNumber(LispMessage.PORT_NUMBER));
68         } else if (address instanceof Ipv6) {
69             String ipv6 = ((Ipv6) address).getIpv6().getValue();
70             tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv6).getAddress()));
71             tab.setPort(new PortNumber(LispMessage.PORT_NUMBER));
72         } else if (address instanceof Ipv4Binary) {
73             Ipv4AddressBinary ipv6 = ((Ipv4Binary) address).getIpv4Binary();
74             tab.setIpAddress(new IpAddressBinary(ipv6));
75             tab.setPort(new PortNumber(LispMessage.PORT_NUMBER));
76         } else if (address instanceof Ipv6Binary) {
77             Ipv6AddressBinary ipv6 = ((Ipv6Binary) address).getIpv6Binary();
78             tab.setIpAddress(new IpAddressBinary(ipv6));
79             tab.setPort(new PortNumber(LispMessage.PORT_NUMBER));
80         } else if (address instanceof KeyValueAddress) {
81             SimpleAddress sa = ((KeyValueAddress) address).getKeyValueAddress().getValue();
82             if (sa.getDistinguishedNameType() != null) {
83                 final Iterator<String> it = COLON_SPLITTER.split(sa.getDistinguishedNameType().getValue()).iterator();
84                 String ip = it.next();
85                 int port = Integer.valueOf(it.next());
86
87                 tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ip).getAddress()));
88                 tab.setPort(new PortNumber(Uint16.valueOf(port)));
89             }
90         } else if (address instanceof DistinguishedName) {
91             DistinguishedName dname = (DistinguishedName) address;
92             final Iterator<String> it = COLON_SPLITTER.split(dname.getDistinguishedName().getValue()).iterator();
93             String ip = it.next();
94             int port = Integer.valueOf(it.next());
95
96             tab.setIpAddress(IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ip).getAddress()));
97             tab.setPort(new PortNumber(Uint16.valueOf(port)));
98         } else if (address instanceof ApplicationData) {
99             ApplicationData appData = (ApplicationData) address;
100             tab.setIpAddress(getIpAddressBinary(appData.getApplicationData().getAddress().getIpAddress()));
101             tab.setPort(new PortNumber(appData.getApplicationData().getLocalPortLow().getValue()));
102         }
103         return tab.build();
104     }
105
106     public static InetAddress getAddressByName(String ipAddress) {
107         try {
108             return InetAddress.getByName(ipAddress);
109         } catch (UnknownHostException e) {
110             LOG.debug("Unknown host {}", ipAddress, e);
111             return null;
112         }
113     }
114
115     public static List<Mapping> getMapping(AddMapping mapRegisterNotification) {
116         List<Mapping> mappings = new ArrayList<>();
117         for (int i = 0; i < mapRegisterNotification.getMapRegister().getMappingRecordItem().size(); i++) {
118             MappingRecord record = mapRegisterNotification.getMapRegister().getMappingRecordItem().get(i)
119                     .getMappingRecord();
120             MappingBuilder mb = new MappingBuilder();
121             mb.setEidUri(new EidUri(LispAddressStringifier.getURIString(
122                     record.getEid())));
123             mb.setOrigin(MappingOrigin.Southbound);
124             mb.setSiteId(getSiteId(mapRegisterNotification.getMapRegister()));
125             mb.setMappingRecord(record);
126             mappings.add(mb.build());
127         }
128         return mappings;
129     }
130
131     public static Set<SiteId> getSiteId(MapRegister mapRegister) {
132         if (mapRegister.getXtrSiteIdPresent()) {
133             return Set.of(new SiteId(mapRegister.getSiteId()));
134         } else {
135             return null;
136         }
137     }
138
139     // We will see if we need to keep this method post full binary-migration, and if yes, will probably move to
140     // LispAddressUtil (and add tests!)
141     private static IpAddressBinary getIpAddressBinary(IpAddress address) {
142         if (address.getIpv4Address() != null) {
143             String ipv4 = address.getIpv4Address().getValue();
144             return IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv4).getAddress());
145         } else if (address.getIpv6Address() != null) {
146             String ipv6 = address.getIpv6Address().getValue();
147             return IpAddressBinaryBuilder.getDefaultInstance(InetAddresses.forString(ipv6).getAddress());
148         }
149         return null;
150     }
151 }