1 /* LimitedLengthInputStream.java --
2 Copyright (C) 2005 Free Software Foundation, Inc.
4 This file is part of GNU Classpath.
6 GNU Classpath is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2, or (at your option)
11 GNU Classpath is distributed in the hope that it will be useful, but
12 WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with GNU Classpath; see the file COPYING. If not, write to the
18 Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
21 Linking this library statically or dynamically with other modules is
22 making a combined work based on this library. Thus, the terms and
23 conditions of the GNU General Public License cover the whole
26 As a special exception, the copyright holders of this library give you
27 permission to link this library with independent modules to produce an
28 executable, regardless of the license terms of these independent
29 modules, and to copy and distribute the resulting executable under
30 terms of your choice, provided that you also meet, for each linked
31 independent module, the terms and conditions of the license of that
32 module. An independent module is a module which is not derived from
33 or based on this library. If you modify this library, you may extend
34 this exception to your version of the library, but you are not
35 obligated to do so. If you do not wish to do so, delete this
36 exception statement from your version. */
39 package gnu
.java
.net
.protocol
.http
;
41 import java
.io
.IOException
;
42 import java
.io
.InputStream
;
45 * InputStream that limits the total number of bytes that can be read
46 * from an underlying stream. In addition to limiting the number of
47 * bytes read, close() is not propagated to the underlying stream.
49 * @author David Daney (ddaney@avtrex.com)
51 class LimitedLengthInputStream
54 private long remainingLen
;
55 private boolean restrictLen
;
56 private HTTPConnection connection
;
58 private InputStream in
;
59 private boolean doClose
;
62 private void handleClose()
81 * @param in the underlying stream
83 * @param maxLen the maximum number of bytes to read
85 * @param restrictLen if true the number of bytes that can be read
86 * from this stream will be limited to maxLen, otherwise the number
87 * of bytes is not restricted.
89 * @param con the HTTPConnection associated with this stream
91 * @param doClose if true con will be closed when finished reading,
92 * else it will be placed back in the connection pool.
95 LimitedLengthInputStream(InputStream in
,
104 this.remainingLen
= maxLen
;
105 this.restrictLen
= restrictLen
;
106 this.connection
= con
;
107 this.doClose
= doClose
;
112 throw new IllegalArgumentException();
113 else if (maxLen
== 0)
114 handleClose(); // Nothing to do, release the connection.
118 public synchronized int read()
132 if (0 == remainingLen
)
145 public int read(byte[] buffer
)
148 return read(buffer
, 0, buffer
.length
);
151 public synchronized int read(byte[] buffer
, int offset
, int length
)
157 if (restrictLen
&& length
> remainingLen
)
158 length
= (int) remainingLen
;
160 int r
= in
.read(buffer
, offset
, length
);
165 if (restrictLen
&& r
> 0)
168 if (0 == remainingLen
)
174 public synchronized long skip(long n
)
181 if (restrictLen
&& n
> remainingLen
)
189 if (0 == remainingLen
)
195 public synchronized int available()
201 int a
= in
.available();
202 if (restrictLen
&& a
> remainingLen
)
203 a
= (int)remainingLen
;
207 public synchronized void close()
213 // If we get to here, the stream was not fully read. Just throw