Mercurial > 510Connectbot
diff src/com/jcraft/jzlib/ZInputStream.java @ 0:0ce5cc452d02
initial version
author | Carl Byington <carl@five-ten-sg.com> |
---|---|
date | Thu, 22 May 2014 10:41:19 -0700 |
parents | |
children | 46c2115ae1c8 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/src/com/jcraft/jzlib/ZInputStream.java Thu May 22 10:41:19 2014 -0700 @@ -0,0 +1,160 @@ +/* -*-mode:java; c-basic-offset:2; indent-tabs-mode:nil -*- */ +/* +Copyright (c) 2001 Lapo Luchini. + +Redistribution and use in source and binary forms, with or without +modification, are permitted provided that the following conditions are met: + + 1. Redistributions of source code must retain the above copyright notice, + this list of conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in + the documentation and/or other materials provided with the distribution. + + 3. The names of the authors may not be used to endorse or promote products + derived from this software without specific prior written permission. + +THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, +INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND +FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS +OR ANY CONTRIBUTORS TO THIS SOFTWARE BE LIABLE FOR ANY DIRECT, INDIRECT, +INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT +LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, +OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF +LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING +NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, +EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ +/* + * This program is based on zlib-1.1.3, so all credit should go authors + * Jean-loup Gailly(jloup@gzip.org) and Mark Adler(madler@alumni.caltech.edu) + * and contributors of zlib. + */ + +package com.jcraft.jzlib; +import java.io.*; + +public class ZInputStream extends FilterInputStream { + + protected ZStream z = new ZStream(); + protected int bufsize = 512; + protected int flush = JZlib.Z_NO_FLUSH; + protected byte[] buf = new byte[bufsize], + buf1 = new byte[1]; + protected boolean compress; + + protected InputStream in = null; + + public ZInputStream(InputStream in) { + this(in, false); + } + public ZInputStream(InputStream in, boolean nowrap) { + super(in); + this.in = in; + z.inflateInit(nowrap); + compress = false; + z.next_in = buf; + z.next_in_index = 0; + z.avail_in = 0; + } + + public ZInputStream(InputStream in, int level) { + super(in); + this.in = in; + z.deflateInit(level); + compress = true; + z.next_in = buf; + z.next_in_index = 0; + z.avail_in = 0; + } + + /*public int available() throws IOException { + return inf.finished() ? 0 : 1; + }*/ + + public int read() throws IOException { + if (read(buf1, 0, 1) == -1) + return (-1); + + return (buf1[0] & 0xFF); + } + + private boolean nomoreinput = false; + + public int read(byte[] b, int off, int len) throws IOException { + if (len == 0) + return (0); + + int err; + z.next_out = b; + z.next_out_index = off; + z.avail_out = len; + + do { + if ((z.avail_in == 0) && (!nomoreinput)) { // if buffer is empty and more input is avaiable, refill it + z.next_in_index = 0; + z.avail_in = in.read(buf, 0, bufsize); //(bufsize<z.avail_out ? bufsize : z.avail_out)); + + if (z.avail_in == -1) { + z.avail_in = 0; + nomoreinput = true; + } + } + + if (compress) + err = z.deflate(flush); + else + err = z.inflate(flush); + + if (nomoreinput && (err == JZlib.Z_BUF_ERROR)) + return (-1); + + if (err != JZlib.Z_OK && err != JZlib.Z_STREAM_END) + throw new ZStreamException((compress ? "de" : "in") + "flating: " + z.msg); + + if ((nomoreinput || err == JZlib.Z_STREAM_END) && (z.avail_out == len)) + return (-1); + } + while (z.avail_out == len && err == JZlib.Z_OK); + + //System.err.print("("+(len-z.avail_out)+")"); + return (len - z.avail_out); + } + + public long skip(long n) throws IOException { + int len = 512; + + if (n < len) + len = (int)n; + + byte[] tmp = new byte[len]; + return ((long)read(tmp)); + } + + public int getFlushMode() { + return (flush); + } + + public void setFlushMode(int flush) { + this.flush = flush; + } + + /** + * Returns the total number of bytes input so far. + */ + public long getTotalIn() { + return z.total_in; + } + + /** + * Returns the total number of bytes output so far. + */ + public long getTotalOut() { + return z.total_out; + } + + public void close() throws IOException { + in.close(); + } +}