aboutsummaryrefslogtreecommitdiffstats
path: root/datarouter-node/src/main/java/org/onap/dmaap/datarouter/node/IsFrom.java
blob: 5cdb3445ecdfdeee09a6c50d29ce4747c9b00162 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*******************************************************************************
 * ============LICENSE_START==================================================
 * * org.onap.dmaap
 * * ===========================================================================
 * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
 * * ===========================================================================
 * * Licensed under the Apache License, Version 2.0 (the "License");
 * * you may not use this file except in compliance with the License.
 * * You may obtain a copy of the License at
 * *
 *  *      http://www.apache.org/licenses/LICENSE-2.0
 * *
 *  * Unless required by applicable law or agreed to in writing, software
 * * distributed under the License is distributed on an "AS IS" BASIS,
 * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * * See the License for the specific language governing permissions and
 * * limitations under the License.
 * * ============LICENSE_END====================================================
 * *
 * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
 * *
 ******************************************************************************/


package org.onap.dmaap.datarouter.node;

import com.att.eelf.configuration.EELFLogger;
import com.att.eelf.configuration.EELFManager;
import java.io.IOException;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Arrays;

/**
 * Determine if an IP address is from a machine.
 */
public class IsFrom {

    private static final EELFLogger logger = EELFManager.getInstance().getLogger(IsFrom.class);
    private long nextcheck;
    private String[] ips;
    private String fqdn;

    /**
     * Create an IsFrom for the specified fully qualified domain name.
     */
    public IsFrom(String fqdn) {
        this.fqdn = fqdn;
    }

    /**
     * Configure the JVM DNS cache to have a 10 second TTL.  This needs to be called very very early or it won't have
     * any effect.
     */
    public static void setDNSCache() {
        java.security.Security.setProperty("networkaddress.cache.ttl", "10");
    }

    /**
     * Check if an IP address matches.  If it has been more than 10 seconds since DNS was last checked for changes to
     * the IP address(es) of this FQDN, check again.  Then check if the specified IP address belongs to the FQDN.
     */
    public synchronized boolean isFrom(String ip) {
        long now = System.currentTimeMillis();
        if (now > nextcheck) {
            nextcheck = now + 10000;
            ArrayList<String> hostAddrArray = new ArrayList<>();
            try {
                InetAddress[] addrs = InetAddress.getAllByName(fqdn);
                for (InetAddress addr : addrs) {
                    hostAddrArray.add(addr.getHostAddress());
                }
            } catch (UnknownHostException e) {
                logger.error("IsFrom: UnknownHostEx: " + e.toString(), e);
            }
            ips = hostAddrArray.toArray(new String[0]);
            logger.debug("IsFrom: DNS ENTRIES FOR FQDN " + fqdn + " : " + Arrays.toString(ips));
        }
        for (String ipAddr : ips) {
            if (ipAddr.equals(ip)) {
                return true;
            }
        }
        return false;
    }

    synchronized boolean isReachable(String ip) {
        try {
            if (InetAddress.getByName(ip).isReachable(1000)) {
                return true;
            }
        } catch (UnknownHostException e) {
            logger.error("IsFrom: UnknownHostEx: " + e, e);
        } catch (IOException e) {
            logger.error("IsFrom: Failed to parse IP : " + ip + " : " + e, e);
        }
        return false;
    }

    /**
     * Return the fully qualified domain name.
     */
    public String toString() {
        return (fqdn);
    }
}