KickJava   Java API By Example, From Geeks To Geeks.

Java > Open Source Codes > org > apache > coyote > http11 > filters > BufferedInputFilter


1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one or more
3  * contributor license agreements. See the NOTICE file distributed with
4  * this work for additional information regarding copyright ownership.
5  * The ASF licenses this file to You under the Apache License, Version 2.0
6  * (the "License"); you may not use this file except in compliance with
7  * the License. You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  */
17
18 package org.apache.coyote.http11.filters;
19
20 import java.io.IOException JavaDoc;
21 import org.apache.coyote.Request;
22 import org.apache.coyote.InputBuffer;
23 import org.apache.coyote.http11.InputFilter;
24 import org.apache.tomcat.util.buf.ByteChunk;
25
26 /**
27  * Input filter responsible for reading and buffering the request body, so that
28  * it does not interfere with client SSL handshake messages.
29  */
30 public class BufferedInputFilter implements InputFilter {
31
32     // -------------------------------------------------------------- Constants
33
34     private static final String JavaDoc ENCODING_NAME = "buffered";
35     private static final ByteChunk ENCODING = new ByteChunk();
36
37
38     // ----------------------------------------------------- Instance Variables
39
40     private ByteChunk buffered = null;
41     private ByteChunk tempRead = new ByteChunk(1024);
42     private InputBuffer buffer;
43     private boolean hasRead = false;
44
45
46     // ----------------------------------------------------- Static Initializer
47
48     static {
49         ENCODING.setBytes(ENCODING_NAME.getBytes(), 0, ENCODING_NAME.length());
50     }
51
52
53     // --------------------------------------------------------- Public Methods
54
55
56     /**
57      * Set the buffering limit. This should be reset every time the buffer is
58      * used.
59      */
60     public void setLimit(int limit) {
61         if (buffered == null) {
62             buffered = new ByteChunk(4048);
63             buffered.setLimit(limit);
64         }
65     }
66
67
68     // ---------------------------------------------------- InputBuffer Methods
69
70
71     /**
72      * Reads the request body and buffers it.
73      */
74     public void setRequest(Request request) {
75         // save off the Request body
76 try {
77             while (buffer.doRead(tempRead, request) >= 0) {
78                 buffered.append(tempRead);
79                 tempRead.recycle();
80             }
81         } catch(IOException JavaDoc iex) {
82             // Ignore
83 }
84     }
85
86     /**
87      * Fills the given ByteChunk with the buffered request body.
88      */
89     public int doRead(ByteChunk chunk, Request request) throws IOException JavaDoc {
90         if (hasRead || buffered.getLength() <= 0) {
91             return -1;
92         } else {
93             chunk.setBytes(buffered.getBytes(), buffered.getStart(),
94                            buffered.getLength());
95             hasRead = true;
96         }
97         return chunk.getLength();
98     }
99
100     public void setBuffer(InputBuffer buffer) {
101         this.buffer = buffer;
102     }
103
104     public void recycle() {
105         if (buffered.getBuffer().length > 65536) {
106             buffered = null;
107         } else {
108             buffered.recycle();
109         }
110         tempRead.recycle();
111         hasRead = false;
112         buffer = null;
113     }
114
115     public ByteChunk getEncodingName() {
116         return ENCODING;
117     }
118
119     public long end() throws IOException JavaDoc {
120         return 0;
121     }
122
123 }
124
Popular Tags