mirror of
https://github.com/Athou/commafeed.git
synced 2026-03-21 21:37:29 +00:00
269 lines
9.1 KiB
Java
269 lines
9.1 KiB
Java
package com.commafeed.backend;
|
|
|
|
import java.io.IOException;
|
|
import java.security.SecureRandom;
|
|
import java.security.cert.CertificateException;
|
|
import java.security.cert.X509Certificate;
|
|
|
|
import javax.net.ssl.KeyManager;
|
|
import javax.net.ssl.SSLContext;
|
|
import javax.net.ssl.SSLException;
|
|
import javax.net.ssl.SSLSession;
|
|
import javax.net.ssl.SSLSocket;
|
|
import javax.net.ssl.TrustManager;
|
|
import javax.net.ssl.X509TrustManager;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
import org.apache.commons.lang.StringUtils;
|
|
import org.apache.http.Header;
|
|
import org.apache.http.HttpEntity;
|
|
import org.apache.http.HttpHeaders;
|
|
import org.apache.http.HttpHost;
|
|
import org.apache.http.HttpResponse;
|
|
import org.apache.http.HttpStatus;
|
|
import org.apache.http.client.ClientProtocolException;
|
|
import org.apache.http.client.HttpClient;
|
|
import org.apache.http.client.HttpResponseException;
|
|
import org.apache.http.client.methods.HttpGet;
|
|
import org.apache.http.client.methods.HttpUriRequest;
|
|
import org.apache.http.client.params.CookiePolicy;
|
|
import org.apache.http.client.params.HttpClientParams;
|
|
import org.apache.http.conn.ClientConnectionManager;
|
|
import org.apache.http.conn.scheme.Scheme;
|
|
import org.apache.http.conn.scheme.SchemeRegistry;
|
|
import org.apache.http.conn.ssl.SSLSocketFactory;
|
|
import org.apache.http.conn.ssl.X509HostnameVerifier;
|
|
import org.apache.http.impl.client.DecompressingHttpClient;
|
|
import org.apache.http.impl.client.DefaultHttpClient;
|
|
import org.apache.http.impl.client.DefaultHttpRequestRetryHandler;
|
|
import org.apache.http.impl.client.SystemDefaultHttpClient;
|
|
import org.apache.http.params.HttpConnectionParams;
|
|
import org.apache.http.params.HttpParams;
|
|
import org.apache.http.params.HttpProtocolParams;
|
|
import org.apache.http.protocol.BasicHttpContext;
|
|
import org.apache.http.protocol.ExecutionContext;
|
|
import org.apache.http.protocol.HttpContext;
|
|
import org.apache.http.util.EntityUtils;
|
|
|
|
/**
|
|
* Smart HTTP getter: handles gzip, ssl, last modified and etag headers
|
|
*
|
|
*/
|
|
@Slf4j
|
|
public class HttpGetter {
|
|
|
|
private static final String USER_AGENT = "CommaFeed/1.0 (http://www.commafeed.com)";
|
|
private static final String ACCEPT_LANGUAGE = "en";
|
|
private static final String PRAGMA_NO_CACHE = "No-cache";
|
|
private static final String CACHE_CONTROL_NO_CACHE = "no-cache";
|
|
private static final String UTF8 = "UTF-8";
|
|
private static final String HTTPS = "https";
|
|
|
|
private static SSLContext SSL_CONTEXT = null;
|
|
static {
|
|
try {
|
|
SSL_CONTEXT = SSLContext.getInstance("TLS");
|
|
SSL_CONTEXT.init(new KeyManager[0], new TrustManager[] { new DefaultTrustManager() }, new SecureRandom());
|
|
} catch (Exception e) {
|
|
log.error("Could not configure ssl context");
|
|
}
|
|
}
|
|
|
|
private static final X509HostnameVerifier VERIFIER = new DefaultHostnameVerifier();
|
|
|
|
public HttpResult getBinary(String url, int timeout) throws ClientProtocolException, IOException, NotModifiedException {
|
|
return getBinary(url, null, null, timeout);
|
|
}
|
|
|
|
/**
|
|
*
|
|
* @param url
|
|
* the url to retrive
|
|
* @param lastModified
|
|
* header we got last time we queried that url, or null
|
|
* @param eTag
|
|
* header we got last time we queried that url, or null
|
|
* @return
|
|
* @throws ClientProtocolException
|
|
* @throws IOException
|
|
* @throws NotModifiedException
|
|
* if the url hasn't changed since we asked for it last time
|
|
*/
|
|
public HttpResult getBinary(String url, String lastModified, String eTag, int timeout) throws ClientProtocolException, IOException,
|
|
NotModifiedException {
|
|
HttpResult result = null;
|
|
long start = System.currentTimeMillis();
|
|
|
|
HttpClient client = newClient(timeout);
|
|
try {
|
|
HttpGet httpget = new HttpGet(url);
|
|
HttpContext context = new BasicHttpContext();
|
|
|
|
httpget.addHeader(HttpHeaders.ACCEPT_LANGUAGE, ACCEPT_LANGUAGE);
|
|
httpget.addHeader(HttpHeaders.PRAGMA, PRAGMA_NO_CACHE);
|
|
httpget.addHeader(HttpHeaders.CACHE_CONTROL, CACHE_CONTROL_NO_CACHE);
|
|
httpget.addHeader(HttpHeaders.USER_AGENT, USER_AGENT);
|
|
|
|
if (lastModified != null) {
|
|
httpget.addHeader(HttpHeaders.IF_MODIFIED_SINCE, lastModified);
|
|
}
|
|
if (eTag != null) {
|
|
httpget.addHeader(HttpHeaders.IF_NONE_MATCH, eTag);
|
|
}
|
|
|
|
HttpResponse response = null;
|
|
try {
|
|
response = client.execute(httpget, context);
|
|
int code = response.getStatusLine().getStatusCode();
|
|
if (code == HttpStatus.SC_NOT_MODIFIED) {
|
|
throw new NotModifiedException("'304 - not modified' http code received");
|
|
} else if (code >= 300) {
|
|
throw new HttpResponseException(code, "Server returned HTTP error code " + code);
|
|
}
|
|
|
|
} catch (HttpResponseException e) {
|
|
if (e.getStatusCode() == HttpStatus.SC_NOT_MODIFIED) {
|
|
throw new NotModifiedException("'304 - not modified' http code received");
|
|
} else {
|
|
throw e;
|
|
}
|
|
}
|
|
Header lastModifiedHeader = response.getFirstHeader(HttpHeaders.LAST_MODIFIED);
|
|
String lastModifiedHeaderValue = lastModifiedHeader == null ? null : StringUtils.trimToNull(lastModifiedHeader.getValue());
|
|
if (lastModifiedHeaderValue != null && StringUtils.equals(lastModified, lastModifiedHeaderValue)) {
|
|
throw new NotModifiedException("lastModifiedHeader is the same");
|
|
}
|
|
|
|
Header eTagHeader = response.getFirstHeader(HttpHeaders.ETAG);
|
|
String eTagHeaderValue = eTagHeader == null ? null : StringUtils.trimToNull(eTagHeader.getValue());
|
|
if (eTag != null && StringUtils.equals(eTag, eTagHeaderValue)) {
|
|
throw new NotModifiedException("eTagHeader is the same");
|
|
}
|
|
|
|
HttpEntity entity = response.getEntity();
|
|
byte[] content = null;
|
|
String contentType = null;
|
|
if (entity != null) {
|
|
content = EntityUtils.toByteArray(entity);
|
|
if (entity.getContentType() != null) {
|
|
contentType = entity.getContentType().getValue();
|
|
}
|
|
}
|
|
HttpUriRequest req = (HttpUriRequest) context.getAttribute(ExecutionContext.HTTP_REQUEST);
|
|
HttpHost host = (HttpHost) context.getAttribute(ExecutionContext.HTTP_TARGET_HOST);
|
|
String urlAfterRedirect = req.getURI().isAbsolute() ? req.getURI().toString() : host.toURI() + req.getURI();
|
|
|
|
long duration = System.currentTimeMillis() - start;
|
|
result = new HttpResult(content, contentType, lastModifiedHeaderValue, eTagHeaderValue, duration, urlAfterRedirect);
|
|
} finally {
|
|
client.getConnectionManager().shutdown();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
public static class HttpResult {
|
|
|
|
private byte[] content;
|
|
private String contentType;
|
|
private String lastModifiedSince;
|
|
private String eTag;
|
|
private long duration;
|
|
private String urlAfterRedirect;
|
|
|
|
public HttpResult(byte[] content, String contentType, String lastModifiedSince, String eTag, long duration, String urlAfterRedirect) {
|
|
this.content = content;
|
|
this.contentType = contentType;
|
|
this.lastModifiedSince = lastModifiedSince;
|
|
this.eTag = eTag;
|
|
this.duration = duration;
|
|
this.urlAfterRedirect = urlAfterRedirect;
|
|
}
|
|
|
|
public byte[] getContent() {
|
|
return content;
|
|
}
|
|
|
|
public String getContentType() {
|
|
return contentType;
|
|
}
|
|
|
|
public String getLastModifiedSince() {
|
|
return lastModifiedSince;
|
|
}
|
|
|
|
public String geteTag() {
|
|
return eTag;
|
|
}
|
|
|
|
public long getDuration() {
|
|
return duration;
|
|
}
|
|
|
|
public String getUrlAfterRedirect() {
|
|
return urlAfterRedirect;
|
|
}
|
|
}
|
|
|
|
public static HttpClient newClient(int timeout) {
|
|
DefaultHttpClient client = new SystemDefaultHttpClient();
|
|
|
|
SSLSocketFactory ssf = new SSLSocketFactory(SSL_CONTEXT, VERIFIER);
|
|
ClientConnectionManager ccm = client.getConnectionManager();
|
|
SchemeRegistry sr = ccm.getSchemeRegistry();
|
|
sr.register(new Scheme(HTTPS, 443, ssf));
|
|
|
|
HttpParams params = client.getParams();
|
|
HttpClientParams.setCookiePolicy(params, CookiePolicy.IGNORE_COOKIES);
|
|
HttpProtocolParams.setContentCharset(params, UTF8);
|
|
HttpConnectionParams.setConnectionTimeout(params, timeout);
|
|
HttpConnectionParams.setSoTimeout(params, timeout);
|
|
client.setHttpRequestRetryHandler(new DefaultHttpRequestRetryHandler(0, false));
|
|
return new DecompressingHttpClient(client);
|
|
}
|
|
|
|
public static class NotModifiedException extends Exception {
|
|
private static final long serialVersionUID = 1L;
|
|
|
|
public NotModifiedException(String message) {
|
|
super(message);
|
|
}
|
|
|
|
}
|
|
|
|
private static class DefaultTrustManager implements X509TrustManager {
|
|
@Override
|
|
public void checkClientTrusted(X509Certificate[] arg0, String arg1) throws CertificateException {
|
|
}
|
|
|
|
@Override
|
|
public void checkServerTrusted(X509Certificate[] arg0, String arg1) throws CertificateException {
|
|
}
|
|
|
|
@Override
|
|
public X509Certificate[] getAcceptedIssuers() {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
private static class DefaultHostnameVerifier implements X509HostnameVerifier {
|
|
|
|
@Override
|
|
public void verify(String string, SSLSocket ssls) throws IOException {
|
|
}
|
|
|
|
@Override
|
|
public void verify(String string, X509Certificate xc) throws SSLException {
|
|
}
|
|
|
|
@Override
|
|
public void verify(String string, String[] strings, String[] strings1) throws SSLException {
|
|
}
|
|
|
|
@Override
|
|
public boolean verify(String string, SSLSession ssls) {
|
|
return true;
|
|
}
|
|
};
|
|
}
|