datarouter-node/src/main/java/org/onap/dmaap/datarouter/node/IsFrom.java

   1 /*******************************************************************************
   2  * ============LICENSE_START==================================================
   3  * * org.onap.dmaap
   4  * * ===========================================================================
   5  * * Copyright © 2017 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  * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  21  * *
  22  ******************************************************************************/
  23
  24
  25 package org.onap.dmaap.datarouter.node;
  26
  27 import com.att.eelf.configuration.EELFLogger;
  28 import com.att.eelf.configuration.EELFManager;
  29 import java.io.IOException;
  30 import java.net.InetAddress;
  31 import java.net.UnknownHostException;
  32 import java.util.ArrayList;
  33 import java.util.Arrays;
  34
  35 /**
  36  * Determine if an IP address is from a machine.
  37  */
  38 public class IsFrom {
  39
  40     private static EELFLogger logger = EELFManager.getInstance().getLogger(IsFrom.class);
  41     private long nextcheck;
  42     private String[] ips;
  43     private String fqdn;
  44
  45     /**
  46      * Create an IsFrom for the specified fully qualified domain name.
  47      */
  48     public IsFrom(String fqdn) {
  49         this.fqdn = fqdn;
  50     }
  51
  52     /**
  53      * Configure the JVM DNS cache to have a 10 second TTL.  This needs to be called very very early or it won't have
  54      * any effect.
  55      */
  56     public static void setDNSCache() {
  57         java.security.Security.setProperty("networkaddress.cache.ttl", "10");
  58     }
  59
  60     /**
  61      * Check if an IP address matches.  If it has been more than 10 seconds since DNS was last checked for changes to
  62      * the IP address(es) of this FQDN, check again.  Then check if the specified IP address belongs to the FQDN.
  63      */
  64     public synchronized boolean isFrom(String ip) {
  65         long now = System.currentTimeMillis();
  66         if (now > nextcheck) {
  67             nextcheck = now + 10000;
  68             ArrayList<String> hostAddrArray = new ArrayList<>();
  69             try {
  70                 InetAddress[] addrs = InetAddress.getAllByName(fqdn);
  71                 for (InetAddress addr : addrs) {
  72                     hostAddrArray.add(addr.getHostAddress());
  73                 }
  74             } catch (UnknownHostException e) {
  75                 logger.error("IsFrom: UnknownHostEx: " + e.toString(), e);
  76             }
  77             ips = hostAddrArray.toArray(new String[0]);
  78             logger.info("IsFrom: DNS ENTRIES FOR FQDN " + fqdn + " : " + Arrays.toString(ips));
  79         }
  80         for (String ipAddr : ips) {
  81             if (ipAddr.equals(ip)) {
  82                 return true;
  83             }
  84         }
  85         return false;
  86     }
  87
  88     synchronized boolean isReachable(String ip) {
  89         try {
  90             if (InetAddress.getByName(ip).isReachable(1000)) {
  91                 return true;
  92             }
  93         } catch (UnknownHostException e) {
  94             logger.error("IsFrom: UnknownHostEx: " + e.toString(), e);
  95         } catch (IOException e) {
  96             logger.error("IsFrom: Failed to parse IP : " + ip + " : " + e.toString(), e);
  97         }
  98         return false;
  99     }
 100
 101     /**
 102      * Return the fully qualified domain name.
 103      */
 104     public String toString() {
 105         return (fqdn);
 106     }
 107 }