01: /*
02: * Licensed to the Apache Software Foundation (ASF) under one
03: * or more contributor license agreements. See the NOTICE file
04: * distributed with this work for additional information
05: * regarding copyright ownership. The ASF licenses this file
06: * to you under the Apache License, Version 2.0 (the
07: * "License"); you may not use this file except in compliance
08: * with the License. You may obtain a copy of the License at
09: *
10: * http://www.apache.org/licenses/LICENSE-2.0
11: *
12: * Unless required by applicable law or agreed to in writing,
13: * software distributed under the License is distributed on an
14: * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15: * KIND, either express or implied. See the License for the
16: * specific language governing permissions and limitations
17: * under the License.
18: *
19: */
20: package org.apache.mina.filter.codec.statemachine;
21:
22: import org.apache.mina.common.IoBuffer;
23: import org.apache.mina.filter.codec.ProtocolDecoderException;
24: import org.apache.mina.filter.codec.ProtocolDecoderOutput;
25:
26: /**
27: * A {@link DecodingState} which consumes all received bytes until a configured
28: * number of read bytes has been reached. Please note that this state can
29: * produce the buffer with less data if the associated session has been
30: * closed unexpectedly.
31: *
32: * @author The Apache MINA Project (dev@mina.apache.org)
33: * @version $Rev: 601996 $, $Date: 2007-12-06 22:06:18 -0700 (Thu, 06 Dec 2007) $
34: */
35: public abstract class ConsumeToEndOfSessionDecodingState implements
36: DecodingState {
37:
38: private IoBuffer buffer;
39: private final int maxLength;
40:
41: public ConsumeToEndOfSessionDecodingState(int maxLength) {
42: this .maxLength = maxLength;
43: }
44:
45: public DecodingState decode(IoBuffer in, ProtocolDecoderOutput out)
46: throws Exception {
47: if (buffer == null) {
48: buffer = IoBuffer.allocate(256).setAutoExpand(true);
49: }
50:
51: if (buffer.position() + in.remaining() > maxLength) {
52: throw new ProtocolDecoderException("Received data exceeds "
53: + maxLength + " byte(s).");
54: }
55: buffer.put(in);
56: return this ;
57: }
58:
59: public DecodingState finishDecode(ProtocolDecoderOutput out)
60: throws Exception {
61: try {
62: buffer.flip();
63: return finishDecode(buffer, out);
64: } finally {
65: buffer = null;
66: }
67: }
68:
69: protected abstract DecodingState finishDecode(IoBuffer readData,
70: ProtocolDecoderOutput out) throws Exception;
71: }
|