HADOOP-6663. BlockDecompressorStream get EOF exception when decompressing the file compressed from empty file. Contributed by Kang Xiao.
git-svn-id: https://svn.apache.org/repos/asf/hadoop/common/trunk@1027312 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
parent
625e4086e4
commit
17f9e57f7c
|
@ -272,6 +272,9 @@ Trunk (unreleased changes)
|
||||||
|
|
||||||
HADOOP-6933. TestListFiles is flaky. (Todd Lipcon via tomwhite)
|
HADOOP-6933. TestListFiles is flaky. (Todd Lipcon via tomwhite)
|
||||||
|
|
||||||
|
HADOOP-6663. BlockDecompressorStream get EOF exception when decompressing
|
||||||
|
the file compressed from empty file. (Kang Xiao via tomwhite)
|
||||||
|
|
||||||
Release 0.21.1 - Unreleased
|
Release 0.21.1 - Unreleased
|
||||||
|
|
||||||
IMPROVEMENTS
|
IMPROVEMENTS
|
||||||
|
|
|
@ -75,6 +75,12 @@ public class BlockDecompressorStream extends DecompressorStream {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
noUncompressedBytes = 0;
|
noUncompressedBytes = 0;
|
||||||
|
// EOF if originalBlockSize is 0
|
||||||
|
// This will occur only when decompressing previous compressed empty file
|
||||||
|
if (originalBlockSize == 0) {
|
||||||
|
eof = true;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
int n = 0;
|
int n = 0;
|
||||||
|
|
|
@ -0,0 +1,229 @@
|
||||||
|
/**
|
||||||
|
* Licensed to the Apache Software Foundation (ASF) under one
|
||||||
|
* or more contributor license agreements. See the NOTICE file
|
||||||
|
* distributed with this work for additional information
|
||||||
|
* regarding copyright ownership. The ASF licenses this file
|
||||||
|
* to you under the Apache License, Version 2.0 (the
|
||||||
|
* "License"); you may not use this file except in compliance
|
||||||
|
* with the License. You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
package org.apache.hadoop.io.compress;
|
||||||
|
|
||||||
|
import java.io.ByteArrayInputStream;
|
||||||
|
import java.io.ByteArrayOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
|
||||||
|
import org.apache.hadoop.conf.Configuration;
|
||||||
|
|
||||||
|
import org.junit.Test;
|
||||||
|
import static org.junit.Assert.*;
|
||||||
|
|
||||||
|
public class TestBlockDecompressorStream {
|
||||||
|
|
||||||
|
private byte[] buf;
|
||||||
|
private ByteArrayInputStream bytesIn;
|
||||||
|
private ByteArrayOutputStream bytesOut;
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testRead() throws IOException {
|
||||||
|
// compress empty stream
|
||||||
|
bytesOut = new ByteArrayOutputStream();
|
||||||
|
BlockCompressorStream blockCompressorStream =
|
||||||
|
new BlockCompressorStream(bytesOut,
|
||||||
|
new FakeCompressor(), 1024, 0);
|
||||||
|
// close without any write
|
||||||
|
blockCompressorStream.close();
|
||||||
|
|
||||||
|
// check compressed output
|
||||||
|
buf = bytesOut.toByteArray();
|
||||||
|
assertEquals("empty file compressed output size is not 4", 4, buf.length);
|
||||||
|
|
||||||
|
// use compressed output as input for decompression
|
||||||
|
bytesIn = new ByteArrayInputStream(buf);
|
||||||
|
|
||||||
|
// get decompression stream
|
||||||
|
BlockDecompressorStream blockDecompressorStream =
|
||||||
|
new BlockDecompressorStream(bytesIn, new FakeDecompressor(), 1024);
|
||||||
|
try {
|
||||||
|
assertEquals("return value is not -1",
|
||||||
|
-1 , blockDecompressorStream.read());
|
||||||
|
} catch (IOException e) {
|
||||||
|
fail("unexpected IOException : " + e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A fake compressor
|
||||||
|
* Its input and output is the same.
|
||||||
|
*/
|
||||||
|
class FakeCompressor implements Compressor{
|
||||||
|
|
||||||
|
private boolean finish;
|
||||||
|
private boolean finished;
|
||||||
|
int nread;
|
||||||
|
int nwrite;
|
||||||
|
|
||||||
|
byte [] userBuf;
|
||||||
|
int userBufOff;
|
||||||
|
int userBufLen;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int compress(byte[] b, int off, int len) throws IOException {
|
||||||
|
int n = Math.min(len, userBufLen);
|
||||||
|
if (userBuf != null && b != null)
|
||||||
|
System.arraycopy(userBuf, userBufOff, b, off, n);
|
||||||
|
userBufOff += n;
|
||||||
|
userBufLen -= n;
|
||||||
|
nwrite += n;
|
||||||
|
|
||||||
|
if (finish && userBufLen <= 0)
|
||||||
|
finished = true;
|
||||||
|
|
||||||
|
return n;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void end() {
|
||||||
|
// nop
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void finish() {
|
||||||
|
finish = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean finished() {
|
||||||
|
return finished;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getBytesRead() {
|
||||||
|
return nread;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public long getBytesWritten() {
|
||||||
|
return nwrite;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean needsInput() {
|
||||||
|
return userBufLen <= 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void reset() {
|
||||||
|
finish = false;
|
||||||
|
finished = false;
|
||||||
|
nread = 0;
|
||||||
|
nwrite = 0;
|
||||||
|
userBuf = null;
|
||||||
|
userBufOff = 0;
|
||||||
|
userBufLen = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setDictionary(byte[] b, int off, int len) {
|
||||||
|
// nop
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setInput(byte[] b, int off, int len) {
|
||||||
|
nread += len;
|
||||||
|
userBuf = b;
|
||||||
|
userBufOff = off;
|
||||||
|
userBufLen = len;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void reinit(Configuration conf) {
|
||||||
|
// nop
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A fake decompressor, just like FakeCompressor
|
||||||
|
* Its input and output is the same.
|
||||||
|
*/
|
||||||
|
class FakeDecompressor implements Decompressor {
|
||||||
|
|
||||||
|
private boolean finish;
|
||||||
|
private boolean finished;
|
||||||
|
int nread;
|
||||||
|
int nwrite;
|
||||||
|
|
||||||
|
byte [] userBuf;
|
||||||
|
int userBufOff;
|
||||||
|
int userBufLen;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public int decompress(byte[] b, int off, int len) throws IOException {
|
||||||
|
int n = Math.min(len, userBufLen);
|
||||||
|
if (userBuf != null && b != null)
|
||||||
|
System.arraycopy(userBuf, userBufOff, b, off, n);
|
||||||
|
userBufOff += n;
|
||||||
|
userBufLen -= n;
|
||||||
|
nwrite += n;
|
||||||
|
|
||||||
|
if (finish && userBufLen <= 0)
|
||||||
|
finished = true;
|
||||||
|
|
||||||
|
return n;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void end() {
|
||||||
|
// nop
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean finished() {
|
||||||
|
return finished;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean needsDictionary() {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean needsInput() {
|
||||||
|
return userBufLen <= 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void reset() {
|
||||||
|
finish = false;
|
||||||
|
finished = false;
|
||||||
|
nread = 0;
|
||||||
|
nwrite = 0;
|
||||||
|
userBuf = null;
|
||||||
|
userBufOff = 0;
|
||||||
|
userBufLen = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setDictionary(byte[] b, int off, int len) {
|
||||||
|
// nop
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public void setInput(byte[] b, int off, int len) {
|
||||||
|
nread += len;
|
||||||
|
userBuf = b;
|
||||||
|
userBufOff = off;
|
||||||
|
userBufLen = len;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue