2 * Copyright (c) 2014 Contextream, Inc. and others. All rights reserved.
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
8 package org.opendaylight.lispflowmapping.lisp.util;
10 import java.net.Inet4Address;
11 import java.net.Inet6Address;
12 import java.net.InetAddress;
13 import java.net.UnknownHostException;
14 import java.nio.ByteBuffer;
16 import org.apache.commons.lang3.exception.ExceptionUtils;
17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpPrefix;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.SimpleAddress;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.Address;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.InstanceId;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv4Prefix;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.Ipv6Prefix;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.lisp.address.types.rev151105.lisp.address.address.SourceDestKey;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.lfm.lisp.proto.rev151105.eid.container.Eid;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
30 public class MaskUtil {
31 private static final Logger LOG = LoggerFactory.getLogger(MaskUtil.class);
32 private static final short IPV4_MAX_MASK = 32;
33 private static final short IPV6_MAX_MASK = 128;
35 public static boolean isMaskable(Address address) {
36 if (address instanceof Ipv4Prefix || address instanceof Ipv6Prefix || address instanceof SourceDestKey) {
38 } else if (address instanceof InstanceId) {
39 return isMaskable(((InstanceId)address).getInstanceId().getAddress());
44 public static boolean isMaskable(SimpleAddress address) {
45 if (address.getIpPrefix() != null) {
51 public static Eid normalize(Eid eid, short mask) {
52 Address address = eid.getAddress();
54 if (address instanceof Ipv4Prefix) {
55 String[] v4prefix = String.valueOf(((Ipv4Prefix)address).getIpv4Prefix().getValue()).split("/");
56 InetAddress normalized = normalizeIP(Inet4Address.getByName(v4prefix[0]), mask);
57 return LispAddressUtil.asIpv4PrefixEid(eid, (Inet4Address)normalized, mask);
58 } else if (address instanceof Ipv6Prefix) {
59 String[] v6prefix = String.valueOf(((Ipv6Prefix)address).getIpv6Prefix().getValue()).split("/");
60 InetAddress normalized = normalizeIP(Inet6Address.getByName(v6prefix[0]), mask);
61 return LispAddressUtil.asIpv6PrefixEid(eid, (Inet6Address)normalized, mask);
62 } else if (address instanceof InstanceId) {
63 // TODO - not absolutely necessary, but should be implemented
66 } catch (UnknownHostException e) {
67 LOG.trace("Failed to normalize eid {} with mask {}: {}", eid, mask, ExceptionUtils.getStackTrace(e));
72 public static Eid normalize(Eid eid) {
73 Address address = eid.getAddress();
75 if (address instanceof Ipv4Prefix) {
76 String[] v4prefix = String.valueOf(((Ipv4Prefix)address).getIpv4Prefix().getValue()).split("/");
77 short mask = Short.parseShort(v4prefix[1]);
78 InetAddress normalized = normalizeIP(Inet4Address.getByName(v4prefix[0]), mask);
79 return LispAddressUtil.asIpv4PrefixEid(eid, (Inet4Address)normalized, mask);
80 } else if (address instanceof Ipv6Prefix) {
81 String[] v6prefix = String.valueOf(((Ipv6Prefix)address).getIpv6Prefix().getValue()).split("/");
82 short mask = Short.parseShort(v6prefix[1]);
83 InetAddress normalized = normalizeIP(Inet6Address.getByName(v6prefix[0]), mask);
84 return LispAddressUtil.asIpv6PrefixEid(eid, (Inet6Address)normalized, mask);
85 } else if (address instanceof InstanceId) {
86 // TODO - not absolutely necessary, but should be implemented
89 } catch (UnknownHostException e) {
90 LOG.trace("Failed to normalize eid {}: {}", eid, ExceptionUtils.getStackTrace(e));
95 private static InetAddress normalizeIP(InetAddress address, int mask) throws UnknownHostException {
96 ByteBuffer byteRepresentation = ByteBuffer.wrap(address.getAddress());
98 for (int i = 0; i < byteRepresentation.array().length; i++) {
100 byteRepresentation.put(i, (byte) (b & byteRepresentation.get(i)));
101 } else if (mask > 0) {
102 byteRepresentation.put(i, (byte) ((byte) (b << (8 - mask)) & byteRepresentation.get(i)));
104 byteRepresentation.put(i, (byte) (0 & byteRepresentation.get(i)));
109 return InetAddress.getByAddress(byteRepresentation.array());
112 public static int getMaxMask(Address address) {
113 if (address instanceof Ipv4 || address instanceof Ipv4Prefix) {
114 return IPV4_MAX_MASK;
115 } else if (address instanceof Ipv6 || address instanceof Ipv6Prefix) {
116 return IPV6_MAX_MASK;
122 public static short getMaskForAddress(SimpleAddress address) {
123 if (address.getIpPrefix() == null) {
126 return getMaskForIpPrefix(address.getIpPrefix());
129 public static short getMaskForIpPrefix(IpPrefix prefix) {
130 return Short.parseShort(String.valueOf(prefix.getValue()).split("/")[1]);
133 public static String getAddressStringForIpPrefix(IpPrefix prefix) {
134 return String.valueOf(prefix.getValue()).split("/")[0];
137 public static String getAddressStringForIpv4Prefix(Ipv4Prefix prefix) {
138 return String.valueOf(prefix.getIpv4Prefix().getValue()).split("/")[0];
141 public static String getAddressStringForIpv6Prefix(Ipv6Prefix prefix) {
142 return String.valueOf(prefix.getIpv6Prefix().getValue()).split("/")[0];
145 public static short getMaskForAddress(Address address) {
146 if (address instanceof Ipv4) {
147 return IPV4_MAX_MASK;
148 } else if (address instanceof Ipv6) {
149 return IPV6_MAX_MASK;
150 } else if (address instanceof Ipv4Prefix) {
151 String[] prefix = String.valueOf(((Ipv4Prefix)address).getIpv4Prefix().getValue()).split("/"); // TODO use something more efficient
152 return Short.parseShort(prefix[1]);
153 } else if (address instanceof Ipv6Prefix) {
154 String[] prefix = String.valueOf(((Ipv6Prefix)address).getIpv6Prefix().getValue()).split("/"); // TODO use something more efficient
155 return Short.parseShort(prefix[1]);
156 } else if (address instanceof InstanceId) {
157 return getMaskForAddress(((InstanceId)address).getInstanceId().getAddress());