package com.subgraph.orchid.sockets.sslengine;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.Socket;
import java.net.SocketAddress;
import java.net.SocketException;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import javax.net.ssl.HandshakeCompletedEvent;
import javax.net.ssl.HandshakeCompletedListener;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLEngine;
import javax.net.ssl.SSLSession;
import javax.net.ssl.SSLSocket;
public class SSLEngineSSLSocket extends SSLSocket implements HandshakeCallbackHandler {
private final SSLEngine engine;
private final SSLEngineManager manager;
private Socket socket;
private InputStream inputStream;
private OutputStream outputStream;
private final List<HandshakeCompletedListener> listenerList;
public SSLEngineSSLSocket(Socket socket, SSLContext context) throws IOException {
this.engine = createSSLEngine(context);
this.socket = socket;
this.manager = new SSLEngineManager(engine, this, socket.getInputStream(), socket.getOutputStream());
this.listenerList = new CopyOnWriteArrayList<HandshakeCompletedListener>();
}
private static SSLEngine createSSLEngine(SSLContext context) {
final SSLEngine engine = context.createSSLEngine();
engine.setUseClientMode(true);
return engine;
}
@Override
public String[] getSupportedCipherSuites() {
return engine.getSupportedCipherSuites();
}
@Override
public String[] getEnabledCipherSuites() {
return engine.getEnabledCipherSuites();
}
@Override
public void setEnabledCipherSuites(String[] suites) {
engine.setEnabledCipherSuites(suites);
}
@Override
public String[] getSupportedProtocols() {
return engine.getSupportedProtocols();
}
@Override
public String[] getEnabledProtocols() {
return engine.getEnabledProtocols();
}
@Override
public void setEnabledProtocols(String[] protocols) {
engine.setEnabledProtocols(protocols);
}
@Override
public SSLSession getSession() {
return engine.getSession();
}
@Override
public void addHandshakeCompletedListener(
HandshakeCompletedListener listener) {
listenerList.add(listener);
}
@Override
public void removeHandshakeCompletedListener(
HandshakeCompletedListener listener) {
listenerList.remove(listener);
}
@Override
public void startHandshake() throws IOException {
manager.startHandshake();
}
@Override
public void setUseClientMode(boolean mode) {
engine.setUseClientMode(mode);
}
@Override
public boolean getUseClientMode() {
return engine.getUseClientMode();
}
@Override
public void setNeedClientAuth(boolean need) {
engine.setNeedClientAuth(need);
}
@Override
public boolean getNeedClientAuth() {
return engine.getNeedClientAuth();
}
@Override
public void setWantClientAuth(boolean want) {
engine.setWantClientAuth(want);
}
@Override
public boolean getWantClientAuth() {
return engine.getWantClientAuth();
}
@Override
public void connect(SocketAddress endpoint) throws IOException {
throw new IOException("Socket is already connected");
}
@Override
public void connect(SocketAddress endpoint, int timeout) throws IOException {
throw new IOException("Socket is already connected");
}
@Override
public void bind(SocketAddress bindpoint) throws IOException {
throw new IOException("Socket is already connected");
}
@Override
public InetAddress getInetAddress() {
return socket.getInetAddress();
}
@Override
public InetAddress getLocalAddress() {
return socket.getLocalAddress();
}
@Override
public int getPort() {
return socket.getPort();
}
@Override
public int getLocalPort() {
return socket.getLocalPort();
}
@Override
public SocketAddress getRemoteSocketAddress() {
return socket.getRemoteSocketAddress();
}
@Override
public SocketAddress getLocalSocketAddress() {
return socket.getLocalSocketAddress();
}
@Override
public void setTcpNoDelay(boolean on) throws SocketException {
socket.setTcpNoDelay(on);
}
@Override
public boolean getTcpNoDelay() throws SocketException {
return socket.getTcpNoDelay();
}
@Override
public void setSoLinger(boolean on, int linger) throws SocketException {
socket.setSoLinger(on, linger);
}
@Override
public int getSoLinger() throws SocketException {
return socket.getSoLinger();
}
@Override
public void setOOBInline(boolean on) throws SocketException {
socket.setOOBInline(on);
}
@Override
public boolean getOOBInline() throws SocketException {
return socket.getOOBInline();
}
@Override
public synchronized void setSoTimeout(int timeout) throws SocketException {
socket.setSoTimeout(timeout);
}
@Override
public synchronized int getSoTimeout() throws SocketException {
return socket.getSoTimeout();
}
@Override
public synchronized void setSendBufferSize(int size) throws SocketException {
socket.setSendBufferSize(size);
}
@Override
public synchronized int getSendBufferSize() throws SocketException {
return socket.getSendBufferSize();
}
@Override
public synchronized void setReceiveBufferSize(int size)
throws SocketException {
socket.setReceiveBufferSize(size);
}
@Override
public synchronized int getReceiveBufferSize() throws SocketException {
return socket.getReceiveBufferSize();
}
@Override
public void setKeepAlive(boolean on) throws SocketException {
socket.setKeepAlive(on);
}
@Override
public boolean getKeepAlive() throws SocketException {
return socket.getKeepAlive();
}
@Override
public void setTrafficClass(int tc) throws SocketException {
socket.setTrafficClass(tc);
}
@Override
public int getTrafficClass() throws SocketException {
return socket.getTrafficClass();
}
@Override
public void setReuseAddress(boolean on) throws SocketException {
socket.setReuseAddress(on);
}
@Override
public boolean getReuseAddress() throws SocketException {
return socket.getReuseAddress();
}
@Override
public void shutdownInput() throws IOException {
throw new UnsupportedOperationException("shutdownInput() not supported on SSL Sockets");
}
@Override
public void shutdownOutput() throws IOException {
throw new UnsupportedOperationException("shutdownOutput() not supported on SSL Sockets");
}
@Override
public boolean isInputShutdown() {
return socket.isInputShutdown();
}
@Override
public boolean isOutputShutdown() {
return socket.isOutputShutdown();
}
@Override
public void setEnableSessionCreation(boolean flag) {
engine.setEnableSessionCreation(flag);
}
@Override
public boolean getEnableSessionCreation() {
return engine.getEnableSessionCreation();
}
@Override
public synchronized InputStream getInputStream() throws IOException {
if(inputStream == null) {
inputStream = new SSLEngineInputStream(manager);
}
return inputStream;
}
@Override
public OutputStream getOutputStream() throws IOException {
if(outputStream == null) {
outputStream = new SSLEngineOutputStream(manager);
}
return outputStream;
}
public void handshakeCompleted() {
if(listenerList.isEmpty()) {
return;
}
final HandshakeCompletedEvent event = new HandshakeCompletedEvent(this, engine.getSession());
for(HandshakeCompletedListener listener: listenerList) {
listener.handshakeCompleted(event);
}
}
}