support concated stream

davies · Dec 17, 2015 · cc1fa94 · cc1fa94
1 parent b69d567
commit cc1fa94
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/java/net/jpountz/lz4/LZ4BlockInputStream.java b/src/java/net/jpountz/lz4/LZ4BlockInputStream.java
@@ -90,6 +90,9 @@ public LZ4BlockInputStream(InputStream in) {
 
   @Override
   public int available() throws IOException {
+    if (o == originalLen) {
+      refill();
+    }
     return originalLen - o;
   }
 
@@ -147,7 +150,12 @@ public long skip(long n) throws IOException {
   }
 
   private void refill() throws IOException {
-    readFully(compressedBuffer, HEADER_LENGTH);
+    try {
+      readFully(compressedBuffer, HEADER_LENGTH);
+    } catch (EOFException e) {
+      finished = true;
+      return;
+    }
     for (int i = 0; i < MAGIC_LENGTH; ++i) {
       if (compressedBuffer[i] != MAGIC[i]) {
         throw new IOException("Stream is corrupted");
@@ -175,7 +183,7 @@ private void refill() throws IOException {
       if (check != 0) {
         throw new IOException("Stream is corrupted");
       }
-      finished = true;
+      refill();
       return;
     }
     if (buffer.length < originalLen) {