提交 6c7a0295 编写于 作者: B bae

6687968: PNGImageReader leaks native memory through an Inflater.

Reviewed-by: igor, prr
上级 6d2b098e
......@@ -616,11 +616,16 @@ public class PNGImageReader extends ImageReader {
private static String inflate(byte[] b) throws IOException {
InputStream bais = new ByteArrayInputStream(b);
InputStream iis = new InflaterInputStream(bais);
StringBuilder sb = new StringBuilder(80);
int c;
try {
while ((c = iis.read()) != -1) {
sb.append((char)c);
}
} finally {
iis.close();
}
return sb.toString();
}
......@@ -1244,13 +1249,26 @@ public class PNGImageReader extends ImageReader {
destinationBands = param.getDestinationBands();
destinationOffset = param.getDestinationOffset();
}
Inflater inf = null;
try {
stream.seek(imageStartPosition);
Enumeration e = new PNGImageDataEnumeration(stream);
InputStream is = new SequenceInputStream(e);
is = new InflaterInputStream(is, new Inflater());
/* InflaterInputStream uses an Inflater instance which consumes
* native (non-GC visible) resources. This is normally implicitly
* freed when the stream is closed. However since the
* InflaterInputStream wraps a client-supplied input stream,
* we cannot close it.
* But the app may depend on GC finalization to close the stream.
* Therefore to ensure timely freeing of native resources we
* explicitly create the Inflater instance and free its resources
* when we are done with the InflaterInputStream by calling
* inf.end();
*/
inf = new Inflater();
is = new InflaterInputStream(is, inf);
is = new BufferedInputStream(is);
this.pixelStream = new DataInputStream(is);
......@@ -1283,6 +1301,10 @@ public class PNGImageReader extends ImageReader {
}
} catch (IOException e) {
throw new IIOException("Error reading PNG image data", e);
} finally {
if (inf != null) {
inf.end();
}
}
}
......
......@@ -244,6 +244,7 @@ final class IDATOutputStream extends ImageOutputStreamImpl {
}
public void finish() throws IOException {
try {
if (!def.finished()) {
def.finish();
while (!def.finished()) {
......@@ -251,6 +252,9 @@ final class IDATOutputStream extends ImageOutputStreamImpl {
}
}
finishChunk();
} finally {
def.end();
}
}
protected void finalize() throws Throwable {
......@@ -928,7 +932,7 @@ public class PNGImageWriter extends ImageWriter {
// Use sourceXOffset, etc.
private void write_IDAT(RenderedImage image) throws IOException {
IDATOutputStream ios = new IDATOutputStream(stream, 32768);
try {
if (metadata.IHDR_interlaceMethod == 1) {
for (int i = 0; i < 7; i++) {
encodePass(ios, image,
......@@ -943,9 +947,10 @@ public class PNGImageWriter extends ImageWriter {
} else {
encodePass(ios, image, 0, 0, 1, 1);
}
} finally {
ios.finish();
}
}
private void writeIEND() throws IOException {
ChunkStream cs = new ChunkStream(PNGImageReader.IEND_TYPE, stream);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册