org.apache.commons.net.util.SubnetUtils.SubnetInfo Java Examples

@Override
public boolean isValid(String value, ConstraintValidatorContext context) {
    if (value == null) {
        return true;
    }
    if (value.isEmpty()) {
        return subnetType.equals(SubnetType.RFC_1918_COMPLIANT_ONLY_OR_EMPTY);
    }
    try {


        SubnetInfo info = new SubnetUtils(value).getInfo();
        if (!info.getAddress().equals(info.getNetworkAddress())) {
            return false;
        }
        switch (subnetType) {
            case CUSTOM:
                return true;
            case RFC_1918_COMPLIANT_ONLY:
            default:
                return isRfc1918CompliantSubnet(info);
        }
    } catch (IllegalArgumentException exception) {
        return false;
    }
}
 

private String calculateSubnet(String stackName, Vpc vpc, Iterable<String> subnetCidrs) {
    SubnetInfo vpcInfo = new SubnetUtils(vpc.getCidrBlock()).getInfo();
    String[] cidrParts = vpcInfo.getCidrSignature().split("/");
    int netmask = Integer.parseInt(cidrParts[cidrParts.length - 1]);
    int netmaskBits = CIDR_PREFIX - netmask;
    if (netmaskBits <= 0) {
        throw new CloudConnectorException("The selected VPC has to be in a bigger CIDR range than /24");
    }
    int numberOfSubnets = Double.valueOf(Math.pow(2, netmaskBits)).intValue();
    int targetSubnet = 0;
    if (stackName != null) {
        byte[] b = stackName.getBytes(Charset.forName("UTF-8"));
        for (byte ascii : b) {
            targetSubnet += ascii;
        }
    }
    targetSubnet = Long.valueOf(targetSubnet % numberOfSubnets).intValue();
    String cidr = getSubnetCidrInRange(vpc, subnetCidrs, targetSubnet, numberOfSubnets);
    if (cidr == null) {
        cidr = getSubnetCidrInRange(vpc, subnetCidrs, 0, targetSubnet);
    }
    if (cidr == null) {
        throw new CloudConnectorException("Cannot find non-overlapping CIDR range");
    }
    return cidr;
}
 

private void validate(APIAddIpRangeByNetworkCidrMsg msg) {
    try {
        SubnetUtils utils = new SubnetUtils(msg.getNetworkCidr());
        utils.setInclusiveHostCount(false);
        SubnetInfo subnet = utils.getInfo();
        if (subnet.getAddressCount() == 0) {
            throw new ApiMessageInterceptionException(argerr("%s is not an allowed network cidr, because it doesn't have usable ip range", msg.getNetworkCidr()));
        }

        if (msg.getGateway() != null && !(msg.getGateway().equals(subnet.getLowAddress()) || msg.getGateway().equals(subnet.getHighAddress()))) {
            throw new ApiMessageInterceptionException(argerr("%s is not the first or last address of the cidr %s", msg.getGateway(), msg.getNetworkCidr()));
        }
    } catch (IllegalArgumentException e) {
        throw new ApiMessageInterceptionException(argerr("%s is not a valid network cidr", msg.getNetworkCidr()));
    }

    if (msg.getIpRangeType() == null) {
        msg.setIpRangeType(IpRangeType.Normal.toString());
    }

    IpRangeInventory ipr = IpRangeInventory.fromMessage(msg);
    validate(ipr);
}
 

private static List<SubnetInfo> convertToSubnetInfos(List<String> cidrBlocks) {
  return cidrBlocks.stream().map(x -> {
    SubnetUtils subnetUtils = new SubnetUtils(x);
    subnetUtils.setInclusiveHostCount(true);
    return subnetUtils.getInfo();
  }).collect(toList());
}
 

private boolean isRfc1918CompliantSubnet(SubnetInfo info) {
    Ip ip = new Ip(info.getAddress());
    long addr = toLong(info.getAddress());
    long lowerAddr = toLong(info.getLowAddress()) - 1;
    if (addr != lowerAddr) {
        return false;
    }
    return (new Ip("10.0.0.0").compareTo(ip) <= 0 && new Ip("10.255.255.255").compareTo(ip) >= 0)
            || (new Ip("172.16.0.0").compareTo(ip) <= 0 && new Ip("172.31.255.255").compareTo(ip) >= 0)
            || (new Ip("192.168.0.0").compareTo(ip) <= 0 && new Ip("192.168.255.255").compareTo(ip) >= 0);
}
 

private String getSubnetCidrInRange(Vpc vpc, Iterable<String> subnetCidrs, int start, int end) {
    SubnetInfo vpcInfo = new SubnetUtils(vpc.getCidrBlock()).getInfo();
    String lowProbe = incrementIp(vpcInfo.getLowAddress());
    String highProbe = new SubnetUtils(toSubnetCidr(lowProbe)).getInfo().getHighAddress();
    // start from the target subnet
    for (int i = 0; i < start - 1; i++) {
        lowProbe = incrementIp(lowProbe);
        highProbe = incrementIp(highProbe);
    }
    boolean foundProbe = false;
    for (int i = start; i < end; i++) {
        boolean overlapping = false;
        for (String subnetCidr : subnetCidrs) {
            SubnetInfo subnetInfo = new SubnetUtils(subnetCidr).getInfo();
            if (isInRange(lowProbe, subnetInfo) || isInRange(highProbe, subnetInfo)) {
                overlapping = true;
                break;
            }
        }
        if (overlapping) {
            lowProbe = incrementIp(lowProbe);
            highProbe = incrementIp(highProbe);
        } else {
            foundProbe = true;
            break;
        }
    }
    if (foundProbe && isInRange(highProbe, vpcInfo)) {
        String subnet = toSubnetCidr(lowProbe);
        LOGGER.debug("The following subnet cidr found: {} for VPC: {}", subnet, vpc.getVpcId());
        return subnet;
    } else {
        return null;
    }
}
 

/**
 * Gets the SubnetInfo block from the input parameters
 * @param _session
 * @param argStruct
 * @return
 * @throws cfmRunTimeException
 */
protected SubnetInfo	getInfo(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	String ipaddress	= getNamedStringParam(argStruct, "ip", null);
	if ( ipaddress == null )
		throwException(_session, "Missing ip paramter" );
	
	if ( ipaddress.lastIndexOf("/") == -1 ){
		String mask	= getNamedStringParam(argStruct, "mask", null);
		if ( mask == null )
			throwException(_session, "Since IP address is not in CIDR notation, please provide a mask" );
		
		return new SubnetUtils(ipaddress, mask).getInfo();
	}else
		return new SubnetUtils(ipaddress).getInfo();
}
 

/**
 * Gets the SubnetInfo block from the input parameters
 * @param _session
 * @param argStruct
 * @return
 * @throws cfmRunTimeException
 */
protected SubnetInfo	getInfo(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	String ipaddress	= getNamedStringParam(argStruct, "ip", null);
	if ( ipaddress == null )
		throwException(_session, "Missing ip paramter" );
	
	if ( ipaddress.lastIndexOf("/") == -1 ){
		String mask	= getNamedStringParam(argStruct, "mask", null);
		if ( mask == null )
			throwException(_session, "Since IP address is not in CIDR notation, please provide a mask" );
		
		return new SubnetUtils(ipaddress, mask).getInfo();
	}else
		return new SubnetUtils(ipaddress).getInfo();
}
 

public static IpRangeInventory fromMessage(APIAddIpRangeByNetworkCidrMsg msg) {
    SubnetUtils utils = new SubnetUtils(msg.getNetworkCidr());
    SubnetInfo subnet = utils.getInfo();

    IpRangeInventory ipr = new IpRangeInventory();
    ipr.setNetworkCidr(msg.getNetworkCidr());
    ipr.setName(msg.getName());
    ipr.setDescription(msg.getDescription());

    ipr.setIpRangeType(IpRangeType.valueOf(msg.getIpRangeType()));

    if (ipr.getIpRangeType() == IpRangeType.Normal) {
        String lowAddress = NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getLowAddress()));
        String highAddress = NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getHighAddress()));
        if (msg.getGateway() == null || msg.getGateway().equals(lowAddress)) {
            ipr.setGateway(lowAddress);
            ipr.setStartIp(NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getLowAddress()) + 1));
            ipr.setEndIp(subnet.getHighAddress());
        } else if (msg.getGateway().equals(highAddress)) {
            ipr.setGateway(highAddress);
            ipr.setStartIp(lowAddress);
            ipr.setEndIp(NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getHighAddress()) - 1));
        }
    } else {
        ipr.setGateway(NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getLowAddress())));
        ipr.setStartIp(NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getLowAddress()) - 1));
        ipr.setEndIp(NetworkUtils.longToIpv4String(NetworkUtils.ipv4StringToLong(subnet.getHighAddress()) + 1));
    }
    ipr.setNetmask(subnet.getNetmask());
    ipr.setPrefixLen(NetworkUtils.getPrefixLengthFromNetwork(subnet.getNetmask()));
    ipr.setL3NetworkUuid(msg.getL3NetworkUuid());
    ipr.setUuid(msg.getResourceUuid());
    ipr.setIpVersion(IPv6Constants.IPv4);
    return ipr;
}
 

/**
 * Return an InetAddress for each interface that matches the
 * given subnet specified using CIDR notation.
 *
 * @param subnet subnet specified using CIDR notation
 * @param returnSubinterfaces
 *            whether to return IPs associated with subinterfaces
 * @throws IllegalArgumentException if subnet is invalid
 */
public static List<InetAddress> getIPs(String subnet,
    boolean returnSubinterfaces) {
  List<InetAddress> addrs = new ArrayList<InetAddress>();
  SubnetInfo subnetInfo = new SubnetUtils(subnet).getInfo();
  Enumeration<NetworkInterface> nifs;

  try {
    nifs = NetworkInterface.getNetworkInterfaces();
  } catch (SocketException e) {
    LOG.error("Unable to get host interfaces", e);
    return addrs;
  }

  while (nifs.hasMoreElements()) {
    NetworkInterface nif = nifs.nextElement();
    // NB: adding addresses even if the nif is not up
    addMatchingAddrs(nif, subnetInfo, addrs);

    if (!returnSubinterfaces) {
      continue;
    }
    Enumeration<NetworkInterface> subNifs = nif.getSubInterfaces();
    while (subNifs.hasMoreElements()) {
      addMatchingAddrs(subNifs.nextElement(), subnetInfo, addrs);
    }
  }
  return addrs;
}
 

/**
 * Add all addresses associated with the given nif in the
 * given subnet to the given list.
 */
private static void addMatchingAddrs(NetworkInterface nif,
    SubnetInfo subnetInfo, List<InetAddress> addrs) {
  Enumeration<InetAddress> ifAddrs = nif.getInetAddresses();
  while (ifAddrs.hasMoreElements()) {
    InetAddress ifAddr = ifAddrs.nextElement();
    if (subnetInfo.isInRange(ifAddr.getHostAddress())) {
      addrs.add(ifAddr);
    }
  }
}
 

/**
 * Return an InetAddress for each interface that matches the
 * given subnet specified using CIDR notation.
 *
 * @param subnet subnet specified using CIDR notation
 * @param returnSubinterfaces
 *            whether to return IPs associated with subinterfaces
 * @throws IllegalArgumentException if subnet is invalid
 */
public static List<InetAddress> getIPs(String subnet,
    boolean returnSubinterfaces) {
  List<InetAddress> addrs = new ArrayList<InetAddress>();
  SubnetInfo subnetInfo = new SubnetUtils(subnet).getInfo();
  Enumeration<NetworkInterface> nifs;

  try {
    nifs = NetworkInterface.getNetworkInterfaces();
  } catch (SocketException e) {
    LOG.error("Unable to get host interfaces", e);
    return addrs;
  }

  while (nifs.hasMoreElements()) {
    NetworkInterface nif = nifs.nextElement();
    // NB: adding addresses even if the nif is not up
    addMatchingAddrs(nif, subnetInfo, addrs);

    if (!returnSubinterfaces) {
      continue;
    }
    Enumeration<NetworkInterface> subNifs = nif.getSubInterfaces();
    while (subNifs.hasMoreElements()) {
      addMatchingAddrs(subNifs.nextElement(), subnetInfo, addrs);
    }
  }
  return addrs;
}
 

/**
 * Add all addresses associated with the given nif in the
 * given subnet to the given list.
 */
private static void addMatchingAddrs(NetworkInterface nif,
    SubnetInfo subnetInfo, List<InetAddress> addrs) {
  Enumeration<InetAddress> ifAddrs = nif.getInetAddresses();
  while (ifAddrs.hasMoreElements()) {
    InetAddress ifAddr = ifAddrs.nextElement();
    if (subnetInfo.isInRange(ifAddr.getHostAddress())) {
      addrs.add(ifAddr);
    }
  }
}
 

String getSpoofAddr(String addr) {
	for (SubnetInfo info : subnets) {
		if (info.isInRange(addr)) {
			return info.getAddress();
		}
	}
	return defaultAddr;
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	return new cfStringData( info.getBroadcastAddress() );
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	return new cfStringData( info.getHighAddress() );
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	return new cfStringData( info.getLowAddress() );
}
 

private CIDRMatch(AccessPrivilege accessPrivilege, SubnetInfo subnetInfo) {
  super(accessPrivilege);
  this.subnetInfo = subnetInfo;
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	return new cfStringData( info.getNetworkAddress() );
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	return new cfStringData( info.getCidrSignature() );
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	return new cfNumberData( info.getAddressCount() );
}
 

private CIDRMatch(AccessPrivilege accessPrivilege, SubnetInfo subnetInfo) {
  super(accessPrivilege);
  this.subnetInfo = subnetInfo;
}
 

private boolean isInRange(String address, SubnetInfo subnetInfo) {
    int low = InetAddresses.coerceToInteger(InetAddresses.forString(subnetInfo.getLowAddress()));
    int high = InetAddresses.coerceToInteger(InetAddresses.forString(subnetInfo.getHighAddress()));
    int currentAddress = InetAddresses.coerceToInteger(InetAddresses.forString(address));
    return low <= currentAddress && currentAddress <= high;
}
 

private static LoadingCache<List<String>, List<SubnetInfo>> cache() {
  return CacheBuilder.newBuilder().expireAfterAccess(10, MINUTES).build(
      CacheLoader.from(x -> convertToSubnetInfos(x)));
}
 

public cfData execute(cfSession _session, cfArgStructData argStruct ) throws cfmRunTimeException {
	SubnetInfo	info	= getInfo( _session, argStruct );
	
	String checkip	= getNamedStringParam(argStruct, "checkip", null);
	if ( checkip == null )
		throwException(_session, "Missing checkip paramter" );

	
	return cfBooleanData.getcfBooleanData( info.isInRange(checkip) );
}