Coverage Report - com.sun.grizzly.filter.SSLEchoAsyncWriteQueueFilter
 
Classes in this File Line Coverage Branch Coverage Complexity
SSLEchoAsyncWriteQueueFilter
92 %
22/24
100 %
4/4
2,6
SSLEchoAsyncWriteQueueFilter$SSLWritePreProcessor
79 %
11/14
50 %
2/4
2,6
 
 1  
 /*
 2  
  * 
 3  
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 4  
  * 
 5  
  * Copyright 2007-2008 Sun Microsystems, Inc. All rights reserved.
 6  
  * 
 7  
  * The contents of this file are subject to the terms of either the GNU
 8  
  * General Public License Version 2 only ("GPL") or the Common Development
 9  
  * and Distribution License("CDDL") (collectively, the "License").  You
 10  
  * may not use this file except in compliance with the License. You can obtain
 11  
  * a copy of the License at https://glassfish.dev.java.net/public/CDDL+GPL.html
 12  
  * or glassfish/bootstrap/legal/LICENSE.txt.  See the License for the specific
 13  
  * language governing permissions and limitations under the License.
 14  
  * 
 15  
  * When distributing the software, include this License Header Notice in each
 16  
  * file and include the License file at glassfish/bootstrap/legal/LICENSE.txt.
 17  
  * Sun designates this particular file as subject to the "Classpath" exception
 18  
  * as provided by Sun in the GPL Version 2 section of the License file that
 19  
  * accompanied this code.  If applicable, add the following below the License
 20  
  * Header, with the fields enclosed by brackets [] replaced by your own
 21  
  * identifying information: "Portions Copyrighted [year]
 22  
  * [name of copyright owner]"
 23  
  * 
 24  
  * Contributor(s):
 25  
  * 
 26  
  * If you wish your version of this file to be governed by only the CDDL or
 27  
  * only the GPL Version 2, indicate your decision by adding "[Contributor]
 28  
  * elects to include this software in this distribution under the [CDDL or GPL
 29  
  * Version 2] license."  If you don't indicate a single choice of license, a
 30  
  * recipient has the option to distribute your version of this file under
 31  
  * either the CDDL, the GPL Version 2 or to extend the choice of license to
 32  
  * its licensees as provided above.  However, if you add GPL Version 2 code
 33  
  * and therefore, elected the GPL Version 2 license, then the option applies
 34  
  * only if the new code is made subject to such option by the copyright
 35  
  * holder.
 36  
  *
 37  
  */
 38  
 
 39  
 package com.sun.grizzly.filter;
 40  
 
 41  
 import com.sun.grizzly.Context;
 42  
 import com.sun.grizzly.Context.AttributeScope;
 43  
 import com.sun.grizzly.ProtocolFilter;
 44  
 import com.sun.grizzly.async.AsyncQueueDataProcessor;
 45  
 import com.sun.grizzly.util.AttributeHolder;
 46  
 import com.sun.grizzly.util.WorkerThread;
 47  
 import java.io.IOException;
 48  
 import java.nio.ByteBuffer;
 49  
 import javax.net.ssl.SSLEngine;
 50  
 import javax.net.ssl.SSLEngineResult;
 51  
 
 52  
 /**
 53  
  * SSL echo filter, which uses {@link AsyncQueueWriter}
 54  
  *
 55  
  * @author Alexey Stashok
 56  
  */
 57  2
 public class SSLEchoAsyncWriteQueueFilter implements ProtocolFilter {
 58  
     private static final String SSL_WRITE_PREPROCESSOR = "SSL_WRITE_PREPROCESSOR";
 59  
     
 60  
     public boolean execute(Context ctx) throws IOException {
 61  53652
         final WorkerThread workerThread = ((WorkerThread) Thread.currentThread());
 62  53652
         ByteBuffer buffer = workerThread.getByteBuffer();
 63  53652
         buffer.flip();
 64  53652
         if (buffer.hasRemaining()) {
 65  
             // Store incoming data in byte[]
 66  53468
             byte[] data = new byte[buffer.remaining()];
 67  53468
             int position = buffer.position();
 68  53468
             buffer.get(data);
 69  53468
             buffer.position(position);
 70  
             try {
 71  53468
                 SSLEngine sslEngine = workerThread.getSSLEngine();
 72  53468
                 AttributeHolder attributes = ctx.getAttributeHolderByScope(AttributeScope.CONNECTION);
 73  
                 
 74  53468
                 SSLWritePreProcessor preProcessor = 
 75  
                         (SSLWritePreProcessor) attributes.getAttribute(SSL_WRITE_PREPROCESSOR);
 76  53468
                 if (preProcessor == null) {
 77  92
                     preProcessor = new SSLWritePreProcessor(sslEngine, workerThread.getOutputBB());
 78  92
                     workerThread.setOutputBB(null);
 79  92
                     attributes.setAttribute(SSL_WRITE_PREPROCESSOR, preProcessor);
 80  
                 }
 81  
                 
 82  53468
                 ctx.getAsyncQueueWritable().writeToAsyncQueue(buffer, null, preProcessor, true);
 83  0
             } catch (IOException ex) {
 84  0
                 throw ex;
 85  53468
             }
 86  
         }
 87  
         
 88  53652
         buffer.clear();
 89  53652
         return false;
 90  
     }
 91  
     
 92  
     public boolean postExecute(Context ctx) throws IOException {
 93  53652
         return true;
 94  
     }
 95  
     
 96  2
     private static class SSLWritePreProcessor implements AsyncQueueDataProcessor {
 97  
         private SSLEngine sslEngine;
 98  
         private ByteBuffer securedOutputBuffer;
 99  
 
 100  92
         public SSLWritePreProcessor(SSLEngine sslEngine, ByteBuffer securedOutputBuffer) {
 101  92
             this.sslEngine = sslEngine;
 102  92
             this.securedOutputBuffer = securedOutputBuffer;
 103  92
         }
 104  
         
 105  
         public ByteBuffer getInternalByteBuffer() {
 106  106936
             return securedOutputBuffer;
 107  
         }
 108  
 
 109  
         public void process(ByteBuffer byteBuffer) throws IOException {
 110  53468
             if (!byteBuffer.hasRemaining() || securedOutputBuffer.hasRemaining()) return;
 111  
             
 112  53468
             securedOutputBuffer.clear();
 113  
             
 114  
             try {
 115  53468
                 SSLEngineResult result = sslEngine.wrap(byteBuffer, securedOutputBuffer);
 116  53468
                 securedOutputBuffer.flip();
 117  0
             } catch (Exception e) {
 118  0
                 securedOutputBuffer.position(securedOutputBuffer.limit());
 119  0
                 throw new IOException(e.getMessage());
 120  53468
             }
 121  53468
         }
 122  
     }
 123  
 }