moparisthebest
/
jDnsProxy
1
0
Fork 0
Code Issues Pull Requests Releases 1 Wiki Activity

Replace HttpURLConnection with Apache HttpClient

This commit is contained in:
Jordan Doyle 2018-03-22 01:37:40 +00:00
parent 081295197c
commit 18616eae52
No known key found for this signature in database
GPG Key ID: D0CD7855EB8063D2
2 changed files with 60 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pom.xml
View File

@ -69,6 +69,11 @@
<artifactId>java-pinning-java7</artifactId>
<version>1.1.0</version>
</dependency>
<dependency>
<groupId>org.apache.httpcomponents</groupId>
<artifactId>httpclient</artifactId>
<version>4.5.5</version>
</dependency>
</dependencies>
<build>
<finalName>${project.artifactId}</finalName>

110
src/main/java/com/moparisthebest/dns/resolve/HttpResolver.java
View File

@ -1,79 +1,79 @@
package com.moparisthebest.dns.resolve;
import com.moparisthebest.dns.dto.Packet;
import org.apache.http.HttpEntity;
import org.apache.http.HttpHeaders;
import org.apache.http.HttpHost;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.config.RequestConfig;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.util.EntityUtils;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLSocketFactory;
import java.io.*;
import java.net.*;
import java.util.concurrent.TimeUnit;
import static com.moparisthebest.dns.Util.readPacket;
public class HttpResolver extends AbstractQueueProcessingResolver {
private final OpenConnection openConnection;
private final int connectTimeout;
private final int readTimeout = 4000;
interface OpenConnection {
HttpURLConnection open() throws Exception;
}
private final HttpClient client;
private final HttpPost request;
public HttpResolver(final int maxRetries, final String name, final URL url, final int connectTimeout, final Proxy proxy, final SSLSocketFactory sslSocketFactory) {
super(maxRetries, name);
this.connectTimeout = connectTimeout;
if(proxy == null && sslSocketFactory == null) {
openConnection = () -> (HttpURLConnection) url.openConnection();
} else if(proxy != null && sslSocketFactory == null) {
openConnection = () -> (HttpURLConnection) url.openConnection(proxy);
} else if(proxy == null
//&& sslSocketFactory != null
) {
openConnection = () -> {
final HttpsURLConnection conn = (HttpsURLConnection) url.openConnection();
conn.setSSLSocketFactory(sslSocketFactory);
return conn;
};
} else //if(proxy != null && sslSocketFactory != null)
{
openConnection = () -> {
final HttpsURLConnection conn = (HttpsURLConnection) url.openConnection(proxy);
conn.setSSLSocketFactory(sslSocketFactory);
return conn;
};
HttpHost proxyHost;
if (proxy != null && proxy.type().equals(Proxy.Type.HTTP)) {
final InetSocketAddress socketAddress = (InetSocketAddress) proxy.address();
final InetAddress inetAddress = socketAddress.getAddress();
proxyHost = new HttpHost(inetAddress, socketAddress.getPort());
} else if (proxy != null && proxy.type().equals(Proxy.Type.SOCKS)) {
throw new IllegalArgumentException("Socks hosts aren't supported");
} else {
proxyHost = null;
}
RequestConfig requestConfig = RequestConfig.custom()
.setConnectionRequestTimeout(connectTimeout)
.setConnectTimeout(connectTimeout)
.setSocketTimeout(connectTimeout)
.build();
this.client = HttpClients.custom()
// .setSSLSocketFactory(new SSLConnectionSocketFactory(sslSocketFactory, (s, sslSession) -> s.equalsIgnoreCase(name)))
.setProxy(proxyHost)
.setConnectionTimeToLive(365, TimeUnit.DAYS)
.setDefaultRequestConfig(requestConfig)
.build();
try {
this.request = new HttpPost(url.toURI());
this.request.setHeader(HttpHeaders.CONTENT_TYPE, "application/dns-udpwireformat");
this.request.setHeader(HttpHeaders.ACCEPT, "application/dns-udpwireformat");
} catch (URISyntaxException e) {
throw new IllegalArgumentException("Failed to parse upstream DNS URL", e);
}
}
@Override
public Packet resolve(final Packet request) throws Exception {
final HttpURLConnection conn = openConnection.open();
final HttpPost req = (HttpPost) this.request.clone();
req.setEntity(new ByteArrayEntity(request.getBuf().array()));
conn.setConnectTimeout(connectTimeout);
conn.setReadTimeout(readTimeout);
conn.setUseCaches(false);
conn.setDoInput(true);
conn.setDoOutput(true);
final HttpResponse response = this.client.execute(req);
conn.setRequestMethod("POST");
conn.setRequestProperty("Content-Type", "application/dns-udpwireformat");
conn.setRequestProperty("Accept", "application/dns-udpwireformat");
final int responseCode = response.getStatusLine().getStatusCode();
// System.out.println("responseCode: " + responseCode);
//conn.connect();
final HttpEntity responseEntity = response.getEntity();
try (OutputStream os = conn.getOutputStream()) {
final long contentLength = responseEntity.getContentLength();
// System.out.println("contentLength: " + contentLength);
os.write(request.getBuf().array());
os.flush();
try (InputStream is = conn.getInputStream();
DataInputStream dis = new DataInputStream(is);) {
final int responseCode = conn.getResponseCode();
//System.out.println("responseCode: " + responseCode);
final int contentLength = conn.getContentLength();
//System.out.println("contentLength: " + contentLength);
return readPacket(contentLength, dis);
}
}
return new Packet(EntityUtils.toByteArray(responseEntity));
}
}