Coverage Report - org.mule.transport.ssl.SslMessageReceiver
 
Classes in this File Line Coverage Branch Coverage Complexity
SslMessageReceiver
0%
0/30
0%
0/8
0
SslMessageReceiver$SslWorker
0%
0/8
N/A
0
 
 1  
 /*
 2  
  * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 3  
  * The software in this package is published under the terms of the CPAL v1.0
 4  
  * license, a copy of which has been included with this distribution in the
 5  
  * LICENSE.txt file.
 6  
  */
 7  
 package org.mule.transport.ssl;
 8  
 
 9  
 import org.mule.api.MuleMessage;
 10  
 import org.mule.api.construct.FlowConstruct;
 11  
 import org.mule.api.endpoint.InboundEndpoint;
 12  
 import org.mule.api.lifecycle.CreateException;
 13  
 import org.mule.api.transport.Connector;
 14  
 import org.mule.config.i18n.CoreMessages;
 15  
 import org.mule.transport.AbstractMessageReceiver;
 16  
 import org.mule.transport.ConnectException;
 17  
 import org.mule.transport.tcp.TcpMessageReceiver;
 18  
 import org.mule.util.StringUtils;
 19  
 
 20  
 import java.io.IOException;
 21  
 import java.net.Socket;
 22  
 import java.security.cert.Certificate;
 23  
 
 24  
 import javax.net.ssl.HandshakeCompletedEvent;
 25  
 import javax.net.ssl.HandshakeCompletedListener;
 26  
 import javax.net.ssl.SSLPeerUnverifiedException;
 27  
 import javax.net.ssl.SSLSocket;
 28  
 import javax.resource.spi.work.Work;
 29  
 
 30  
 import edu.emory.mathcs.backport.java.util.concurrent.CountDownLatch;
 31  
 import edu.emory.mathcs.backport.java.util.concurrent.TimeUnit;
 32  
 
 33  
 
 34  0
 public class SslMessageReceiver extends TcpMessageReceiver implements HandshakeCompletedListener
 35  
 {
 36  
 
 37  
     // We must wait for handshake to complete before sending message, as the callback
 38  
     // sets important properties. The wait period is arbitrary, but the two threads
 39  
     // are approximately synchronized (handshake completes before/at same time as
 40  
     // message is received) so value should not be critical
 41  0
     private CountDownLatch handshakeComplete = new CountDownLatch(1);
 42  
 
 43  
     private Certificate[] peerCertificateChain;
 44  
     private Certificate[] localCertificateChain;
 45  
 
 46  
     public SslMessageReceiver(Connector connector, FlowConstruct flowConstruct, InboundEndpoint endpoint)
 47  
             throws CreateException
 48  
     {
 49  0
         super(connector, flowConstruct, endpoint);
 50  0
     }
 51  
 
 52  
     @Override
 53  
     protected void doConnect() throws ConnectException
 54  
     {
 55  0
         checkKeyStore();
 56  0
         super.doConnect();
 57  0
     }
 58  
 
 59  
     protected void checkKeyStore() throws ConnectException
 60  
     {
 61  0
         SslConnector sslConnector = (SslConnector) connector;
 62  0
         String keyStore = sslConnector.getKeyStore();
 63  0
         if (StringUtils.isBlank(keyStore))
 64  
         {
 65  0
             throw new ConnectException(CoreMessages.objectIsNull("tls-key-store"), this);
 66  
         }
 67  0
     }
 68  
 
 69  
     @Override
 70  
     protected Work createWork(Socket socket) throws IOException
 71  
     {
 72  0
         return new SslWorker(socket, this);
 73  
     }
 74  
 
 75  
     private void preRoute(MuleMessage message) throws Exception
 76  
     {
 77  0
         long sslHandshakeTimeout = ((SslConnector) getConnector()).getSslHandshakeTimeout();
 78  0
         boolean rc = handshakeComplete.await(sslHandshakeTimeout, TimeUnit.MILLISECONDS);
 79  0
         if (rc == false)
 80  
         {
 81  0
             throw new IllegalStateException("Handshake did not complete");
 82  
         }
 83  
 
 84  0
         if (peerCertificateChain != null)
 85  
         {
 86  0
             message.setOutboundProperty(SslConnector.PEER_CERTIFICATES, peerCertificateChain);
 87  
         }
 88  0
         if (localCertificateChain != null)
 89  
         {
 90  0
             message.setOutboundProperty(SslConnector.LOCAL_CERTIFICATES, localCertificateChain);
 91  
         }
 92  0
     }
 93  
 
 94  
     public void handshakeCompleted(HandshakeCompletedEvent event)
 95  
     {
 96  
         try
 97  
         {
 98  0
             localCertificateChain = event.getLocalCertificates();
 99  
             try
 100  
             {
 101  0
                 peerCertificateChain = event.getPeerCertificates();
 102  
             }
 103  0
             catch (SSLPeerUnverifiedException e)
 104  
             {
 105  0
                 logger.debug("Cannot get peer certificate chain: "+ e.getMessage());
 106  0
             }
 107  
         }
 108  
         finally
 109  
         {
 110  0
             handshakeComplete.countDown();
 111  0
         }
 112  0
     }
 113  
 
 114  
     protected class SslWorker extends TcpWorker
 115  
     {
 116  
         public SslWorker(Socket socket, AbstractMessageReceiver receiver) throws IOException
 117  0
         {
 118  0
             super(socket, receiver);
 119  0
             ((SSLSocket) socket).addHandshakeCompletedListener(SslMessageReceiver.this);
 120  0
         }
 121  
 
 122  
         @Override
 123  
         protected void preRouteMuleMessage(MuleMessage message) throws Exception
 124  
         {
 125  0
             super.preRouteMuleMessage(message);
 126  
 
 127  0
             preRoute(message);
 128  0
         }
 129  
 
 130  
         @Override
 131  
         protected void shutdownSocket() throws IOException
 132  
         {
 133  
             // SSL Sockets don't support shutdownSocket
 134  0
         }
 135  
     }
 136  
 
 137  
 }