MAPREDUCE-2631. Potential resource leaks in BinaryProtocol$TeeOutputStream.java. Contributed by Sunil G.

This commit is contained in:
Naganarasimha 2016-10-28 05:50:13 +05:30
parent 5877f20f9c
commit 28660f51af
3 changed files with 35 additions and 7 deletions

View File

@ -24,6 +24,7 @@
import org.apache.hadoop.classification.InterfaceAudience;
import org.apache.hadoop.classification.InterfaceStability;
import org.apache.hadoop.io.IOUtils;
import org.apache.hadoop.util.DataChecksum;
/**
* A Checksum output stream.
@ -60,8 +61,11 @@ public void close() throws IOException {
return;
}
closed = true;
finish();
out.close();
try {
finish();
} finally {
IOUtils.closeStream(out);
}
}
/**

View File

@ -36,6 +36,7 @@
import org.apache.commons.logging.LogFactory;
import org.apache.hadoop.io.BytesWritable;
import org.apache.hadoop.io.DataOutputBuffer;
import org.apache.hadoop.io.IOUtils;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.io.Writable;
import org.apache.hadoop.io.WritableComparable;
@ -200,8 +201,8 @@ private static class TeeOutputStream extends FilterOutputStream {
file = new FileOutputStream(filename);
}
public void write(byte b[], int off, int len) throws IOException {
file.write(b,off,len);
out.write(b,off,len);
file.write(b, off, len);
out.write(b, off, len);
}
public void write(int b) throws IOException {
@ -215,9 +216,12 @@ public void flush() throws IOException {
}
public void close() throws IOException {
flush();
file.close();
out.close();
try {
flush();
} finally {
IOUtils.closeStream(file);
IOUtils.closeStream(out);
}
}
}

View File

@ -22,7 +22,13 @@
import org.apache.hadoop.io.DataInputBuffer;
import org.apache.hadoop.io.DataOutputBuffer;
import org.junit.Test;
import org.mockito.Mockito;
import static org.junit.Assert.fail;
import java.io.IOException;
import java.io.OutputStream;
import static org.junit.Assert.assertEquals;
public class TestIFileStreams {
@ -99,4 +105,18 @@ public void testBadLength() throws Exception {
fail("Did not detect bad data in checksum");
}
@Test
public void testCloseStreamOnException() throws Exception {
OutputStream outputStream = Mockito.mock(OutputStream.class);
IFileOutputStream ifos = new IFileOutputStream(outputStream);
Mockito.doThrow(new IOException("Dummy Exception")).when(outputStream)
.flush();
try {
ifos.close();
fail("IOException is not thrown");
} catch (IOException ioe) {
assertEquals("Dummy Exception", ioe.getMessage());
}
Mockito.verify(outputStream).close();
}
}