##### 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;2,6
SSLEchoAsyncWriteQueueFilter$SSLWritePreProcessor |
79 % | 11/14
50 % | 2/4
| 2.6;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 | 52961 |
             final WorkerThread workerThread = ((WorkerThread) Thread.currentThread());
62 | 52961 |
             ByteBuffer buffer = workerThread.getByteBuffer();
63 | 52961 |
             buffer.flip();
64 | 52961 |
             if (buffer.hasRemaining()) {
65 |  |
                 // Store incoming data in byte[]
66 | 52741 |
                 byte[] data = new byte[buffer.remaining()];
67 | 52741 |
                 int position = buffer.position();
68 | 52741 |
                 buffer.get(data);
69 | 52741 |
                 buffer.position(position);
70 |  |
                 try {
71 | 52741 |
                     SSLEngine sslEngine = workerThread.getSSLEngine();
72 | 52741 |
                     AttributeHolder attributes = ctx.getAttributeHolderByScope(AttributeScope.CONNECTION);
73 |  |
74 | 52741 |
                     SSLWritePreProcessor preProcessor =
75 |  |
                             (SSLWritePreProcessor) attributes.getAttribute(SSL_WRITE_PREPROCESSOR);
76 | 52741 |
                     if (preProcessor == null) {
77 | 110 |
                         preProcessor = new SSLWritePreProcessor(sslEngine, workerThread.getOutputBB());
78 | 110 |
                         workerThread.setOutputBB(null);
79 | 110 |
                         attributes.setAttribute(SSL_WRITE_PREPROCESSOR, preProcessor);
80 |  |
                     }
81 |  |
82 | 52741 |
                     ctx.getAsyncQueueWritable().writeToAsyncQueue(buffer, null, preProcessor, true);
83 | 0 |
                 } catch (IOException ex) {
84 | 0 |
                     throw ex;
85 | 52741 |
                 }
86 |  |
             }
87 |  |
88 | 52961 |
             buffer.clear();
89 | 52961 |
             return false;
90 |  |
         }
91 |  |
92 |  |
         public boolean postExecute(Context ctx) throws IOException {
93 | 52961 |
             return true;
94 |  |
         }
95 |  |
96 | 2 |
         private static class SSLWritePreProcessor implements AsyncQueueDataProcessor {
97 |  |
             private SSLEngine sslEngine;
98 |  |
             private ByteBuffer securedOutputBuffer;
99 |  |
100 | 110 |
             public SSLWritePreProcessor(SSLEngine sslEngine, ByteBuffer securedOutputBuffer) {
101 | 110 |
                 this.sslEngine = sslEngine;
102 | 110 |
                 this.securedOutputBuffer = securedOutputBuffer;
103 | 110 |
             }
104 |  |
105 |  |
             public ByteBuffer getInternalByteBuffer() {
106 | 105482 |
                 return securedOutputBuffer;
107 |  |
             }
108 |  |
109 |  |
             public void process(ByteBuffer byteBuffer) throws IOException {
110 | 52741 |
                 if (!byteBuffer.hasRemaining() || securedOutputBuffer.hasRemaining()) return;
111 |  |
112 | 52741 |
                 securedOutputBuffer.clear();
113 |  |
114 |  |
                 try {
115 | 52741 |
                     SSLEngineResult result = sslEngine.wrap(byteBuffer, securedOutputBuffer);
116 | 52741 |
                     securedOutputBuffer.flip();
117 | 0 |
                 } catch (Exception e) {
118 | 0 |
                     securedOutputBuffer.position(securedOutputBuffer.limit());
119 | 0 |
                     throw new IOException(e.getMessage());
120 | 52741 |
                 }
121 | 52741 |
             }
122 |  |
         }
123 |  |
     }