Browse Source
* In the case of multiple network cards, a valid host is returned * In the case of multiple network cards, a valid host is returned * In the case of multiple network cards, a valid host is returned * In the case of multiple network cards, a valid host is returned * fix LOCAL_ADDRESS not assigned value * fix LOCAL_ADDRESS not assigned value * fix LOCAL_ADDRESS not assigned value * fix lock error * fix lock error * end * end * end * end * code smell * code smell * code smell * code smell * code smell * code smell * code smell * lowering * messy * small change * compile error Co-authored-by: dailidong <dailidong66@gmail.com> Co-authored-by: gabry.wu <gabrywu@apache.org>pull/3/MERGE
CalvinKirs
4 years ago
committed by
GitHub
17 changed files with 309 additions and 68 deletions
@ -0,0 +1,209 @@ |
|||||||
|
/* |
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||||
|
* contributor license agreements. See the NOTICE file distributed with |
||||||
|
* this work for additional information regarding copyright ownership. |
||||||
|
* The ASF licenses this file to You 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. |
||||||
|
*/ |
||||||
|
package org.apache.dolphinscheduler.common.utils; |
||||||
|
|
||||||
|
import org.slf4j.Logger; |
||||||
|
import org.slf4j.LoggerFactory; |
||||||
|
|
||||||
|
import java.io.IOException; |
||||||
|
import java.net.*; |
||||||
|
import java.util.Enumeration; |
||||||
|
import java.util.LinkedList; |
||||||
|
import java.util.List; |
||||||
|
import java.util.Optional; |
||||||
|
import java.util.regex.Pattern; |
||||||
|
|
||||||
|
import static java.util.Collections.emptyList; |
||||||
|
|
||||||
|
/** |
||||||
|
* NetUtils |
||||||
|
*/ |
||||||
|
public class NetUtils { |
||||||
|
|
||||||
|
|
||||||
|
private NetUtils() { |
||||||
|
throw new IllegalStateException("Utility class"); |
||||||
|
} |
||||||
|
|
||||||
|
private static Logger logger = LoggerFactory.getLogger(NetUtils.class); |
||||||
|
|
||||||
|
private static final Pattern IP_PATTERN = Pattern.compile("\\d{1,3}(\\.\\d{1,3}){3,5}$"); |
||||||
|
|
||||||
|
private static String ANY_HOST_VALUE = "0.0.0.0"; |
||||||
|
|
||||||
|
private static String LOCAL_HOST_VALUE = "127.0.0.1"; |
||||||
|
|
||||||
|
private static InetAddress LOCAL_ADDRESS = null; |
||||||
|
|
||||||
|
private static volatile String HOST_ADDRESS; |
||||||
|
|
||||||
|
public static String getHost() { |
||||||
|
if (HOST_ADDRESS != null) { |
||||||
|
return HOST_ADDRESS; |
||||||
|
} |
||||||
|
|
||||||
|
InetAddress address = getLocalAddress(); |
||||||
|
if (address != null) { |
||||||
|
HOST_ADDRESS = address.getHostAddress(); |
||||||
|
return HOST_ADDRESS; |
||||||
|
} |
||||||
|
return LOCAL_HOST_VALUE; |
||||||
|
} |
||||||
|
|
||||||
|
private static InetAddress getLocalAddress() { |
||||||
|
if (null != LOCAL_ADDRESS) { |
||||||
|
return LOCAL_ADDRESS; |
||||||
|
} |
||||||
|
return getLocalAddress0(); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Find first valid IP from local network card |
||||||
|
* |
||||||
|
* @return first valid local IP |
||||||
|
*/ |
||||||
|
private static synchronized InetAddress getLocalAddress0() { |
||||||
|
if (null != LOCAL_ADDRESS) { |
||||||
|
return LOCAL_ADDRESS; |
||||||
|
} |
||||||
|
InetAddress localAddress = null; |
||||||
|
NetworkInterface networkInterface = findNetworkInterface(); |
||||||
|
Enumeration<InetAddress> addresses = networkInterface.getInetAddresses(); |
||||||
|
while (addresses.hasMoreElements()) { |
||||||
|
Optional<InetAddress> addressOp = toValidAddress(addresses.nextElement()); |
||||||
|
if (addressOp.isPresent()) { |
||||||
|
try { |
||||||
|
if (addressOp.get().isReachable(100)) { |
||||||
|
LOCAL_ADDRESS = addressOp.get(); |
||||||
|
return LOCAL_ADDRESS; |
||||||
|
} |
||||||
|
} catch (IOException e) { |
||||||
|
logger.warn("test address id reachable io exception", e); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
try { |
||||||
|
localAddress = InetAddress.getLocalHost(); |
||||||
|
} catch (UnknownHostException e) { |
||||||
|
logger.warn("InetAddress get LocalHost exception", e); |
||||||
|
} |
||||||
|
Optional<InetAddress> addressOp = toValidAddress(localAddress); |
||||||
|
if (addressOp.isPresent()) { |
||||||
|
LOCAL_ADDRESS = addressOp.get(); |
||||||
|
} |
||||||
|
return LOCAL_ADDRESS; |
||||||
|
} |
||||||
|
|
||||||
|
private static Optional<InetAddress> toValidAddress(InetAddress address) { |
||||||
|
if (address instanceof Inet6Address) { |
||||||
|
Inet6Address v6Address = (Inet6Address) address; |
||||||
|
if (isPreferIPV6Address()) { |
||||||
|
return Optional.ofNullable(normalizeV6Address(v6Address)); |
||||||
|
} |
||||||
|
} |
||||||
|
if (isValidV4Address(address)) { |
||||||
|
return Optional.of(address); |
||||||
|
} |
||||||
|
return Optional.empty(); |
||||||
|
} |
||||||
|
|
||||||
|
private static InetAddress normalizeV6Address(Inet6Address address) { |
||||||
|
String addr = address.getHostAddress(); |
||||||
|
int i = addr.lastIndexOf('%'); |
||||||
|
if (i > 0) { |
||||||
|
try { |
||||||
|
return InetAddress.getByName(addr.substring(0, i) + '%' + address.getScopeId()); |
||||||
|
} catch (UnknownHostException e) { |
||||||
|
logger.debug("Unknown IPV6 address: ", e); |
||||||
|
} |
||||||
|
} |
||||||
|
return address; |
||||||
|
} |
||||||
|
|
||||||
|
public static boolean isValidV4Address(InetAddress address) { |
||||||
|
|
||||||
|
if (address == null || address.isLoopbackAddress()) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
String name = address.getHostAddress(); |
||||||
|
return (name != null |
||||||
|
&& IP_PATTERN.matcher(name).matches() |
||||||
|
&& !ANY_HOST_VALUE.equals(name) |
||||||
|
&& !LOCAL_HOST_VALUE.equals(name)); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Check if an ipv6 address |
||||||
|
* |
||||||
|
* @return true if it is reachable |
||||||
|
*/ |
||||||
|
private static boolean isPreferIPV6Address() { |
||||||
|
return Boolean.getBoolean("java.net.preferIPv6Addresses"); |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Get the suitable {@link NetworkInterface} |
||||||
|
* |
||||||
|
* @return If no {@link NetworkInterface} is available , return <code>null</code> |
||||||
|
*/ |
||||||
|
private static NetworkInterface findNetworkInterface() { |
||||||
|
|
||||||
|
List<NetworkInterface> validNetworkInterfaces = emptyList(); |
||||||
|
|
||||||
|
try { |
||||||
|
validNetworkInterfaces = getValidNetworkInterfaces(); |
||||||
|
} catch (SocketException e) { |
||||||
|
logger.warn("ValidNetworkInterfaces exception", e); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
return validNetworkInterfaces.get(0); |
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* Get the valid {@link NetworkInterface network interfaces} |
||||||
|
* |
||||||
|
* @throws SocketException SocketException if an I/O error occurs. |
||||||
|
*/ |
||||||
|
private static List<NetworkInterface> getValidNetworkInterfaces() throws SocketException { |
||||||
|
List<NetworkInterface> validNetworkInterfaces = new LinkedList<>(); |
||||||
|
Enumeration<NetworkInterface> interfaces = NetworkInterface.getNetworkInterfaces(); |
||||||
|
while (interfaces.hasMoreElements()) { |
||||||
|
NetworkInterface networkInterface = interfaces.nextElement(); |
||||||
|
if (ignoreNetworkInterface(networkInterface)) { // ignore
|
||||||
|
continue; |
||||||
|
} |
||||||
|
validNetworkInterfaces.add(networkInterface); |
||||||
|
} |
||||||
|
return validNetworkInterfaces; |
||||||
|
} |
||||||
|
|
||||||
|
/** |
||||||
|
* @param networkInterface {@link NetworkInterface} |
||||||
|
* @return if the specified {@link NetworkInterface} should be ignored, return <code>true</code> |
||||||
|
* @throws SocketException SocketException if an I/O error occurs. |
||||||
|
*/ |
||||||
|
public static boolean ignoreNetworkInterface(NetworkInterface networkInterface) throws SocketException { |
||||||
|
return networkInterface == null |
||||||
|
|| networkInterface.isLoopback() |
||||||
|
|| networkInterface.isVirtual() |
||||||
|
|| !networkInterface.isUp(); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,57 @@ |
|||||||
|
/* |
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one or more |
||||||
|
* contributor license agreements. See the NOTICE file distributed with |
||||||
|
* this work for additional information regarding copyright ownership. |
||||||
|
* The ASF licenses this file to You 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. |
||||||
|
*/ |
||||||
|
package org.apache.dolphinscheduler.common.utils; |
||||||
|
|
||||||
|
import org.junit.Test; |
||||||
|
|
||||||
|
import java.net.InetAddress; |
||||||
|
|
||||||
|
import static org.junit.Assert.*; |
||||||
|
import static org.mockito.Mockito.mock; |
||||||
|
import static org.mockito.Mockito.when; |
||||||
|
|
||||||
|
/** |
||||||
|
* NetUtilsTest |
||||||
|
*/ |
||||||
|
public class NetUtilsTest { |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testGetLocalHost() { |
||||||
|
assertNotNull(NetUtils.getHost()); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void testIsValidAddress() { |
||||||
|
assertFalse(NetUtils.isValidV4Address(null)); |
||||||
|
InetAddress address = mock(InetAddress.class); |
||||||
|
when(address.isLoopbackAddress()).thenReturn(true); |
||||||
|
assertFalse(NetUtils.isValidV4Address(address)); |
||||||
|
address = mock(InetAddress.class); |
||||||
|
when(address.getHostAddress()).thenReturn("localhost"); |
||||||
|
assertFalse(NetUtils.isValidV4Address(address)); |
||||||
|
address = mock(InetAddress.class); |
||||||
|
when(address.getHostAddress()).thenReturn("0.0.0.0"); |
||||||
|
assertFalse(NetUtils.isValidV4Address(address)); |
||||||
|
address = mock(InetAddress.class); |
||||||
|
when(address.getHostAddress()).thenReturn("127.0.0.1"); |
||||||
|
assertFalse(NetUtils.isValidV4Address(address)); |
||||||
|
address = mock(InetAddress.class); |
||||||
|
when(address.getHostAddress()).thenReturn("1.2.3.4"); |
||||||
|
assertTrue(NetUtils.isValidV4Address(address)); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue