Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.dapr.utils.NetworkUtils Maven / Gradle / Ivy
/*
* Copyright 2021 The Dapr Authors
* 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.
*/
package io.dapr.utils;
import io.dapr.config.Properties;
import io.grpc.ClientInterceptor;
import io.grpc.ManagedChannel;
import io.grpc.ManagedChannelBuilder;
import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.util.regex.Pattern;
/**
* Utility methods for network, internal to Dapr SDK.
*/
public final class NetworkUtils {
private static final long RETRY_WAIT_MILLISECONDS = 1000;
// Thanks to https://ihateregex.io/expr/ipv6/
private static final String IPV6_REGEX = "(([0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|([0-9a-fA-F]{1,4}:){1,7}:|"
+ "([0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|"
+ "([0-9a-fA-F]{1,4}:){1,5}(:[0-9a-fA-F]{1,4}){1,2}|"
+ "([0-9a-fA-F]{1,4}:){1,4}(:[0-9a-fA-F]{1,4}){1,3}|([0-9a-fA-F]{1,4}:){1,3}(:[0-9a-fA-F]{1,4}){1,4}|"
+ "([0-9a-fA-F]{1,4}:){1,2}(:[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:((:[0-9a-fA-F]{1,4}){1,6})|"
+ ":((:[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(:[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|"
+ "::(ffff(:0{1,4}){0,1}:){0,1}((25[0-5]|(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\\.){3,3}(25[0-5]|(2[0-4]|"
+ "1{0,1}[0-9]){0,1}[0-9])|([0-9a-fA-F]{1,4}:){1,4}:((25[0-5]|"
+ "(2[0-4]|1{0,1}[0-9]){0,1}[0-9])\\.){3,3}(25[0-5]|"
+ "(2[0-4]|1{0,1}[0-9]){0,1}[0-9]))";
private static final Pattern IPV6_PATTERN = Pattern.compile(IPV6_REGEX, Pattern.CASE_INSENSITIVE);
// Don't accept "?" to avoid ambiguity with ?tls=true
private static final String GRPC_ENDPOINT_FILENAME_REGEX_PART = "[^\0\\?]+";
private static final String GRPC_ENDPOINT_HOSTNAME_REGEX_PART = "(([A-Za-z0-9_\\-\\.]+)|(\\[" + IPV6_REGEX + "\\]))";
private static final String GRPC_ENDPOINT_DNS_AUTHORITY_REGEX_PART =
"(?dns://)(?" + GRPC_ENDPOINT_HOSTNAME_REGEX_PART + ":[0-9]+)?/";
private static final String GRPC_ENDPOINT_PARAM_REGEX_PART = "(\\?(?((unix:)|(unix://)|(unix-abstract:))" + GRPC_ENDPOINT_FILENAME_REGEX_PART + ")";
private static final String GRPC_ENDPOINT_VSOCKET_REGEX_PART =
"(?vsock:" + GRPC_ENDPOINT_HOSTNAME_REGEX_PART + ":[0-9]+)";
private static final String GRPC_ENDPOINT_HOST_REGEX_PART =
"((?http://)|(?https://)|(?dns:)|(" + GRPC_ENDPOINT_DNS_AUTHORITY_REGEX_PART + "))?"
+ "(?" + GRPC_ENDPOINT_HOSTNAME_REGEX_PART + ")?+"
+ "(:(?[0-9]+))?";
private static final String GRPC_ENDPOINT_REGEX = "^("
+ "(" + GRPC_ENDPOINT_HOST_REGEX_PART + ")|"
+ "(" + GRPC_ENDPOINT_SOCKET_REGEX_PART + ")|"
+ "(" + GRPC_ENDPOINT_VSOCKET_REGEX_PART + ")"
+ ")" + GRPC_ENDPOINT_PARAM_REGEX_PART + "$";
private static final Pattern GRPC_ENDPOINT_PATTERN = Pattern.compile(GRPC_ENDPOINT_REGEX, Pattern.CASE_INSENSITIVE);
private NetworkUtils() {
}
/**
* Tries to connect to a socket, retrying every 1 second.
*
* @param host Host to connect to.
* @param port Port to connect to.
* @param timeoutInMilliseconds Timeout in milliseconds to give up trying.
* @throws InterruptedException If retry is interrupted.
*/
public static void waitForSocket(String host, int port, int timeoutInMilliseconds) throws InterruptedException {
long started = System.currentTimeMillis();
callWithRetry(() -> {
try {
try (Socket socket = new Socket()) {
// timeout cannot be negative.
// zero timeout means infinite, so 1 is the practical minimum.
int remainingTimeout = (int) Math.max(1, timeoutInMilliseconds - (System.currentTimeMillis() - started));
socket.connect(new InetSocketAddress(host, port), remainingTimeout);
}
} catch (IOException e) {
throw new RuntimeException(e);
}
}, timeoutInMilliseconds);
}
/**
* Creates a GRPC managed channel.
*
* @param interceptors Optional interceptors to add to the channel.
* @return GRPC managed channel to communicate with the sidecar.
*/
public static ManagedChannel buildGrpcManagedChannel(ClientInterceptor... interceptors) {
var settings = GrpcEndpointSettings.parse();
ManagedChannelBuilder builder = ManagedChannelBuilder.forTarget(settings.endpoint)
.userAgent(Version.getSdkVersion());
if (!settings.secure) {
builder = builder.usePlaintext();
}
if (interceptors != null && interceptors.length > 0) {
builder = builder.intercept(interceptors);
}
return builder.build();
}
// Not private to allow unit testing
static final class GrpcEndpointSettings {
final String endpoint;
final boolean secure;
private GrpcEndpointSettings(String endpoint, boolean secure) {
this.endpoint = endpoint;
this.secure = secure;
}
static GrpcEndpointSettings parse() {
String address = Properties.SIDECAR_IP.get();
int port = Properties.GRPC_PORT.get();
boolean secure = false;
String grpcEndpoint = Properties.GRPC_ENDPOINT.get();
if ((grpcEndpoint != null) && !grpcEndpoint.isEmpty()) {
var matcher = GRPC_ENDPOINT_PATTERN.matcher(grpcEndpoint);
if (!matcher.matches()) {
throw new IllegalArgumentException("Illegal gRPC endpoint: " + grpcEndpoint);
}
var parsedHost = matcher.group("hostname");
if (parsedHost != null) {
address = parsedHost;
}
var https = matcher.group("https") != null;
var http = matcher.group("http") != null;
secure = https;
String parsedPort = matcher.group("port");
if (parsedPort != null) {
port = Integer.parseInt(parsedPort);
} else {
// This implements default port as 80 for http for backwards compatibility.
port = http ? 80 : 443;
}
String parsedParam = matcher.group("param");
if ((http || https) && (parsedParam != null)) {
throw new IllegalArgumentException("Query params is not supported in HTTP URI for gRPC endpoint.");
}
if (parsedParam != null) {
secure = parsedParam.equalsIgnoreCase("tls=true");
}
var authorityEndpoint = matcher.group("authorityEndpoint");
if (authorityEndpoint != null) {
return new GrpcEndpointSettings(String.format("dns://%s/%s:%d", authorityEndpoint, address, port), secure);
}
var socket = matcher.group("socket");
if (socket != null) {
return new GrpcEndpointSettings(socket, secure);
}
var vsocket = matcher.group("vsocket");
if (vsocket != null) {
return new GrpcEndpointSettings(vsocket, secure);
}
}
return new GrpcEndpointSettings(String.format("dns:///%s:%d", address, port), secure);
}
}
private static void callWithRetry(Runnable function, long retryTimeoutMilliseconds) throws InterruptedException {
long started = System.currentTimeMillis();
while (true) {
Throwable exception;
try {
function.run();
return;
} catch (Exception e) {
exception = e;
} catch (AssertionError e) {
exception = e;
}
long elapsed = System.currentTimeMillis() - started;
if (elapsed >= retryTimeoutMilliseconds) {
if (exception instanceof RuntimeException) {
throw (RuntimeException) exception;
}
throw new RuntimeException(exception);
}
long remaining = retryTimeoutMilliseconds - elapsed;
Thread.sleep(Math.min(remaining, RETRY_WAIT_MILLISECONDS));
}
}
/**
* Retrieve loopback address for the host.
*
* @return The loopback address String
*/
public static String getHostLoopbackAddress() {
return InetAddress.getLoopbackAddress().getHostAddress();
}
static boolean isIPv6(String ip) {
return IPV6_PATTERN.matcher(ip).matches();
}
}
