Skip to content

Commit

Permalink
ok http client can acessible
Browse files Browse the repository at this point in the history
  • Loading branch information
asep.rojali committed Jun 23, 2023
1 parent 04191cb commit 4b2b7e6
Showing 1 changed file with 109 additions and 112 deletions.
221 changes: 109 additions & 112 deletions src/main/java/io/github/avew/VewHttp.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@
import lombok.extern.slf4j.Slf4j;
import okhttp3.*;
import okhttp3.Authenticator;
import okhttp3.logging.HttpLoggingInterceptor;
import org.apache.commons.lang.StringUtils;
import org.jetbrains.annotations.NotNull;
import retrofit2.Retrofit;
import retrofit2.adapter.rxjava.RxJavaCallAdapterFactory;

Expand All @@ -17,7 +17,6 @@
import java.security.KeyManagementException;
import java.security.NoSuchAlgorithmException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.function.Consumer;
import java.util.regex.Matcher;
Expand All @@ -32,112 +31,26 @@ public class VewHttp {
private CustomHttpConfig config;
private List<String> masking = new ArrayList<>();
private boolean debug;
private OkHttpClient client;

public VewHttp(CustomHttpConfig config, List<String> masking, boolean debug) {
this.config = config;
this.masking = masking;
this.debug = debug;
}


public Retrofit.Builder builder() throws RuntimeException {
return builder(e -> {
});
public VewHttp(CustomHttpConfig config, OkHttpClient client, List<String> masking, boolean debug) {
this.config = config;
this.masking = masking;
this.debug = debug;
this.client = client;
}


public Retrofit.Builder builder(Consumer<OkHttpClient.Builder> configureClient) throws RuntimeException {
public Retrofit.Builder builder() throws RuntimeException {
try {

OkHttpClient.Builder httpClient = new OkHttpClient().newBuilder();

httpClient.connectTimeout(config.getCustomTimeout().getConnectTimeout(), config.getCustomTimeout().getConnectTimeoutUnit());
httpClient.readTimeout(config.getCustomTimeout().getReadTimeout(), config.getCustomTimeout().getReadTimeoutUnit());
httpClient.writeTimeout(config.getCustomTimeout().getWriteTimeout(), config.getCustomTimeout().getWriteTimeoutUnit());
httpClient.retryOnConnectionFailure(config.isRetryConnectionFailure());
httpClient.addInterceptor(chain -> {
Request original = chain.request();
Request.Builder request = original.newBuilder();
if (config.getAgent() != null)
request.header("User-Agent", config.getAgent());
if (config.isRetryConnectionFailure()) request.header("Connection", "close");
return chain.proceed(request.build());
});
httpClient.addNetworkInterceptor(customNetworkInterceptors());

ConnectionSpec connectionSpecClearText = new ConnectionSpec.Builder(ConnectionSpec.CLEARTEXT)
.build();
switch (config.getTLS()) {
case TLS1_1:
ConnectionSpec tls11 = new ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS)
.tlsVersions(TlsVersion.TLS_1_0, TlsVersion.TLS_1_1)
.build();

httpClient.connectionSpecs(List.of(tls11, connectionSpecClearText));
break;
case TLS1_2:
ConnectionSpec tls12 = new ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS)
.tlsVersions(TlsVersion.TLS_1_2)
.build();
httpClient.connectionSpecs(List.of(tls12, connectionSpecClearText));
break;
case TLS1_3:
ConnectionSpec tls13 = new ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS)
.tlsVersions(TlsVersion.TLS_1_3)
.build();
httpClient.connectionSpecs(List.of(tls13, connectionSpecClearText));
break;
}

this.setTrustManager(httpClient);

if (config.getProxy() != null) {
CustomProxy proxy = config.getProxy();
if (proxy.isProxy()) {
log.info("CONNECTION TO {} USE PROXY", config.getUrl());

if (proxy.isAuth()) {
log.info("CONNECTION TO {} USE PROXY WITH AUTH", config.getUrl());
Authenticator proxyAuthenticator = (route, response) -> {
String credential = Credentials.basic(proxy.getUsername(), proxy.getPassword());
return response.request().newBuilder()
.header("Proxy-Authorization", credential)
.build();
};
httpClient.proxyAuthenticator(proxyAuthenticator);
}

httpClient.proxySelector(new ProxySelector() {
@Override
public List<Proxy> select(URI uri) {

// Host
final String host = uri.getHost();

String noProxyHost = proxy.getUrlSkip().stream().collect(Collectors.joining(","));
if (StringUtils.contains(noProxyHost, host)) {
return List.of(Proxy.NO_PROXY);
} else {
// Add Proxy
return List.of(new Proxy(Proxy.Type.HTTP,
new InetSocketAddress(proxy.getHost(), proxy.getPort())));
}
}

@Override
public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
throw new UnsupportedOperationException("Proxy Not supported yet.");
}
});
}
}

if (configureClient != null) configureClient.accept(httpClient);
OkHttpClient client = httpClient.build();

return new Retrofit.Builder()
.baseUrl(config.getUrl())
.client(client)
.client(client == null ? setupClient() : client)
.addCallAdapterFactory(RxJavaCallAdapterFactory.create());

} catch (Exception ex) {
Expand All @@ -146,6 +59,94 @@ public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
}
}

@NotNull
public OkHttpClient setupClient() throws NoSuchAlgorithmException, KeyManagementException {
OkHttpClient.Builder httpClient = new OkHttpClient().newBuilder();

httpClient.connectTimeout(config.getCustomTimeout().getConnectTimeout(), config.getCustomTimeout().getConnectTimeoutUnit());
httpClient.readTimeout(config.getCustomTimeout().getReadTimeout(), config.getCustomTimeout().getReadTimeoutUnit());
httpClient.writeTimeout(config.getCustomTimeout().getWriteTimeout(), config.getCustomTimeout().getWriteTimeoutUnit());
httpClient.retryOnConnectionFailure(config.isRetryConnectionFailure());
httpClient.addInterceptor(chain -> {
Request original = chain.request();
Request.Builder request = original.newBuilder();
if (config.getAgent() != null)
request.header("User-Agent", config.getAgent());
if (config.isRetryConnectionFailure()) request.header("Connection", "close");
return chain.proceed(request.build());
});
httpClient.addNetworkInterceptor(customNetworkInterceptors());

ConnectionSpec connectionSpecClearText = new ConnectionSpec.Builder(ConnectionSpec.CLEARTEXT)
.build();
switch (config.getTLS()) {
case TLS1_1:
ConnectionSpec tls11 = new ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS)
.tlsVersions(TlsVersion.TLS_1_0, TlsVersion.TLS_1_1)
.build();

httpClient.connectionSpecs(List.of(tls11, connectionSpecClearText));
break;
case TLS1_2:
ConnectionSpec tls12 = new ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS)
.tlsVersions(TlsVersion.TLS_1_2)
.build();
httpClient.connectionSpecs(List.of(tls12, connectionSpecClearText));
break;
case TLS1_3:
ConnectionSpec tls13 = new ConnectionSpec.Builder(ConnectionSpec.MODERN_TLS)
.tlsVersions(TlsVersion.TLS_1_3)
.build();
httpClient.connectionSpecs(List.of(tls13, connectionSpecClearText));
break;
}

this.setTrustManager(httpClient);

if (config.getProxy() != null) {
CustomProxy proxy = config.getProxy();
if (proxy.isProxy()) {
log.info("CONNECTION TO {} USE PROXY", config.getUrl());

if (proxy.isAuth()) {
log.info("CONNECTION TO {} USE PROXY WITH AUTH", config.getUrl());
Authenticator proxyAuthenticator = (route, response) -> {
String credential = Credentials.basic(proxy.getUsername(), proxy.getPassword());
return response.request().newBuilder()
.header("Proxy-Authorization", credential)
.build();
};
httpClient.proxyAuthenticator(proxyAuthenticator);
}

httpClient.proxySelector(new ProxySelector() {
@Override
public List<Proxy> select(URI uri) {

// Host
final String host = uri.getHost();

String noProxyHost = proxy.getUrlSkip().stream().collect(Collectors.joining(","));
if (StringUtils.contains(noProxyHost, host)) {
return List.of(Proxy.NO_PROXY);
} else {
// Add Proxy
return List.of(new Proxy(Proxy.Type.HTTP,
new InetSocketAddress(proxy.getHost(), proxy.getPort())));
}
}

@Override
public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
throw new UnsupportedOperationException("Proxy Not supported yet.");
}
});
}
}

return httpClient.build();
}

private void setTrustManager(OkHttpClient.Builder httpClient) throws NoSuchAlgorithmException, KeyManagementException {
/* create a trust manager that does not validate certificate chains */
TrustManager[] trustAllCerts = new TrustManager[]{
Expand Down Expand Up @@ -177,24 +178,20 @@ public java.security.cert.X509Certificate[] getAcceptedIssuers() {
}

private String maskingLog(List<String> words, String msg) {
List<String> maskPatterns;
if (config.isMasking()) {
maskPatterns = words.stream().map(s -> "\"([" + s + "\"]+)\"\\s*:\\s*\"([^\"]+)\",?").collect(Collectors.toList());
} else {
maskPatterns = new ArrayList<>();
}

Pattern multilinePattern = Pattern.compile(String.join("|", maskPatterns), Pattern.MULTILINE);
StringBuilder sb = new StringBuilder(msg);
Matcher matcher = multilinePattern.matcher(sb);
while (matcher.find()) {
IntStream.rangeClosed(1, matcher.groupCount()).forEach(group -> {
if (matcher.group(group) != null) {
IntStream.range(matcher.start(group), matcher.end(group)).forEach(i -> sb.setCharAt(i, '*'));
}
});
if (config.isMasking()) {
List<String> maskPatterns = words.stream().map(s -> "\"([" + s + "\"]+)\"\\s*:\\s*\"([^\"]+)\",?").collect(Collectors.toList());
Pattern multilinePattern = Pattern.compile(String.join("|", maskPatterns), Pattern.MULTILINE);
Matcher matcher = multilinePattern.matcher(sb);
while (matcher.find()) {
IntStream.rangeClosed(1, matcher.groupCount()).forEach(group -> {
if (matcher.group(group) != null) {
IntStream.range(matcher.start(group), matcher.end(group)).forEach(i -> sb.setCharAt(i, '*'));
}
});
}
return sb.toString();
}

return sb.toString();
}

Expand Down

0 comments on commit 4b2b7e6

Please sign in to comment.