1 | |
|
2 | |
|
3 | |
|
4 | |
|
5 | |
|
6 | |
|
7 | |
|
8 | |
|
9 | |
|
10 | |
|
11 | |
|
12 | |
|
13 | |
|
14 | |
|
15 | |
|
16 | |
|
17 | |
|
18 | |
|
19 | |
|
20 | |
|
21 | |
|
22 | |
|
23 | |
|
24 | |
|
25 | |
|
26 | |
|
27 | |
|
28 | |
|
29 | |
|
30 | |
|
31 | |
|
32 | |
|
33 | |
|
34 | |
|
35 | |
|
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 | |
|
54 | |
|
55 | |
|
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 | |
|
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 | |
} |