cadi/core/src/main/java/org/onap/aaf/cadi/util/NetMask.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.cadi.util;
  23
  24 /*
  25  * NetMask - a class to quickly validate whether a given IP is part of a mask, as defined by bytes or standard String format.
  26  *
  27  * Needs the IPV6 Mask Builder.
  28  */
  29 public class NetMask {
  30     private long mask;
  31
  32     public NetMask(byte[] inBytes) {
  33         mask = derive(inBytes);
  34     }
  35
  36     public NetMask(String string) throws MaskFormatException {
  37         mask = derive(string,true);
  38     }
  39
  40     public boolean isInNet(byte[] inBytes) {
  41         long addr = derive(inBytes);
  42         return (mask & addr) == addr;
  43     }
  44
  45     public boolean isInNet(String str) {
  46         long addr;
  47         try {
  48             addr = derive(str,false);
  49             return (mask & addr) == addr;
  50         } catch (MaskFormatException e) {
  51             // will not hit this code;
  52             return false;
  53         }
  54     }
  55
  56     public static long derive(byte[] inBytes) {
  57         long addr = 0L;
  58         int offset = inBytes.length*8;
  59         for (int i=0;i<inBytes.length;++i) {
  60             addr&=(inBytes[i]<<offset);
  61             offset-=8;
  62         }
  63         return addr;
  64     }
  65
  66     public static long derive(String str, boolean check) throws MaskFormatException {
  67         long rv=0L;
  68         int idx=str.indexOf(':');
  69         int slash = str.indexOf('/');
  70
  71         if (idx<0) { // Not IPV6, so it's IPV4... Is there a mask of 123/254?
  72             idx=str.indexOf('.');
  73             int offset = 24;
  74             int end = slash>=0?slash:str.length();
  75             int bits = slash>=0?Integer.parseInt(str.substring(slash+1)):32;
  76             if (check && bits>32) {
  77                 throw new MaskFormatException("Invalid Mask Offset in IPV4 Address");
  78             }
  79             int prev = 0;
  80             long lbyte;
  81             while (prev<end) {
  82                 if (idx<0) {
  83                     idx = end;
  84                 }
  85                 lbyte = Long.parseLong(str.substring(prev, idx));
  86                 if (check && (lbyte>255 || lbyte<0)) {
  87                     throw new MaskFormatException("Invalid Byte in IPV4 Address");
  88                 }
  89                 rv|=lbyte<<offset;
  90                 prev = ++idx;
  91                 idx=str.indexOf('.',prev);
  92                 offset-=8;
  93             }
  94             rv|=0x00000000FFFFFFFFL>>bits;
  95         }
  96         return rv;
  97     }
  98
  99 }