93 lines
3.1 KiB
Java
93 lines
3.1 KiB
Java
package app.fedilab.fedilabtube.client;
|
|
/* Copyright 2020 Thomas Schneider
|
|
*
|
|
* This file is a part of TubeLab
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it under the terms of the
|
|
* GNU General Public License as published by the Free Software Foundation; either version 3 of the
|
|
* License, or (at your option) any later version.
|
|
*
|
|
* TubeLab is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
|
|
* the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
|
|
* Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along with TubeLab; if not,
|
|
* see <http://www.gnu.org/licenses>. */
|
|
|
|
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.SSLSocket;
|
|
import javax.net.ssl.SSLSocketFactory;
|
|
|
|
|
|
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;
|
|
}
|
|
|
|
|
|
@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 instanceof SSLSocket)) {
|
|
((SSLSocket) socket).setEnabledProtocols(new String[]{"TLSv1.1", "TLSv1.2",});
|
|
}
|
|
return socket;
|
|
}
|
|
} |