86 lines
2.6 KiB
Java
86 lines
2.6 KiB
Java
package fr.gouv.etalab.mastodon.client;
|
|
|
|
|
|
import java.io.IOException;
|
|
import java.net.InetAddress;
|
|
import java.net.Socket;
|
|
import java.security.KeyManagementException;
|
|
import java.security.NoSuchAlgorithmException;
|
|
|
|
import javax.net.ssl.SSLContext;
|
|
import javax.net.ssl.SSLEngine;
|
|
import javax.net.ssl.SSLSocket;
|
|
import javax.net.ssl.SSLSocketFactory;
|
|
|
|
/**
|
|
* Created by Thomas on 29/08/2017.
|
|
*
|
|
*/
|
|
|
|
public class TLSSocketFactory extends SSLSocketFactory {
|
|
|
|
private SSLSocketFactory sSLSocketFactory;
|
|
private SSLContext sslContext;
|
|
|
|
public TLSSocketFactory() throws KeyManagementException, NoSuchAlgorithmException {
|
|
|
|
sslContext = SSLContext.getInstance("TLS");
|
|
sslContext.init(null, null, null);
|
|
sSLSocketFactory = sslContext.getSocketFactory();
|
|
}
|
|
|
|
public SSLContext getSSLContext(){
|
|
return this.sslContext;
|
|
}
|
|
|
|
public SSLEngine getSSLEngine(){
|
|
return this.sslContext.createSSLEngine();
|
|
}
|
|
|
|
@Override
|
|
public String[] getDefaultCipherSuites() {
|
|
return sSLSocketFactory.getDefaultCipherSuites();
|
|
}
|
|
|
|
@Override
|
|
public String[] getSupportedCipherSuites() {
|
|
return sSLSocketFactory.getSupportedCipherSuites();
|
|
}
|
|
|
|
@Override
|
|
public Socket createSocket() throws IOException {
|
|
return enableTLSOnSocket(sSLSocketFactory.createSocket());
|
|
}
|
|
|
|
@Override
|
|
public Socket createSocket(Socket s, String host, int port, boolean autoClose) throws IOException {
|
|
return enableTLSOnSocket(sSLSocketFactory.createSocket(s, host, port, autoClose));
|
|
}
|
|
|
|
@Override
|
|
public Socket createSocket(String host, int port) throws IOException {
|
|
return enableTLSOnSocket(sSLSocketFactory.createSocket(host, port));
|
|
}
|
|
|
|
@Override
|
|
public Socket createSocket(String host, int port, InetAddress localHost, int localPort) throws IOException {
|
|
return enableTLSOnSocket(sSLSocketFactory.createSocket(host, port, localHost, localPort));
|
|
}
|
|
|
|
@Override
|
|
public Socket createSocket(InetAddress host, int port) throws IOException {
|
|
return enableTLSOnSocket(sSLSocketFactory.createSocket(host, port));
|
|
}
|
|
|
|
@Override
|
|
public Socket createSocket(InetAddress address, int port, InetAddress localAddress, int localPort) throws IOException {
|
|
return enableTLSOnSocket(sSLSocketFactory.createSocket(address, port, localAddress, localPort));
|
|
}
|
|
|
|
private Socket enableTLSOnSocket(Socket socket) {
|
|
if(socket != null && (socket instanceof SSLSocket)) {
|
|
((SSLSocket)socket).setEnabledProtocols(new String[] {"TLSv1.1", "TLSv1.2"});
|
|
}
|
|
return socket;
|
|
}
|
|
} |