Skip to content

Commit

Permalink
[bug-67579] do not close stream when reading XSSFWorkbook based on st…
Browse files Browse the repository at this point in the history
…ream

git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1912700 13f79535-47bb-0310-9956-ffa450edef68
  • Loading branch information
pjfanning committed Oct 3, 2023
1 parent 5d36e0c commit 105966c
Show file tree
Hide file tree
Showing 4 changed files with 75 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -132,6 +132,7 @@ public ZipPackage() {
try (ZipArchiveThresholdInputStream zis = ZipHelper.openZipStream(in)) {
this.zipArchive = new ZipInputStreamZipEntrySource(zis);
} catch (final IOException | RuntimeException e) {
IOUtils.closeQuietly(in);
throw e;
}
}
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
/* ====================================================================
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.poi.openxml4j.opc.internal;

import java.io.FilterInputStream;
import java.io.InputStream;

final class NoCloseInputStream extends FilterInputStream {
NoCloseInputStream(InputStream stream) {
super(stream);
}

@Override
public void close() {}
}
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,7 @@ public static ZipArchiveThresholdInputStream openZipStream(InputStream stream) t
verifyZipHeader(checkedStream);

// Open as a proper zip stream
return new ZipArchiveThresholdInputStream(new ZipArchiveInputStream(checkedStream));
return new ZipArchiveThresholdInputStream(new ZipArchiveInputStream(new NoCloseInputStream(checkedStream)));
}

/**
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,10 @@ Licensed to the Apache Software Foundation (ASF) under one or more

package org.apache.poi.xssf.usermodel;

import org.apache.commons.compress.archivers.ArchiveEntry;
import org.apache.commons.compress.archivers.zip.ZipArchiveEntry;
import org.apache.commons.compress.archivers.zip.ZipArchiveInputStream;
import org.apache.commons.compress.archivers.zip.ZipArchiveOutputStream;
import org.apache.commons.io.output.NullOutputStream;
import org.apache.commons.io.output.UnsynchronizedByteArrayOutputStream;
import org.apache.poi.POIDataSamples;
Expand Down Expand Up @@ -77,14 +81,17 @@ Licensed to the Apache Software Foundation (ASF) under one or more
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.util.Date;
import java.util.Iterator;
import java.util.List;
import java.util.Locale;
import java.util.zip.CRC32;

import static org.apache.poi.hssf.HSSFTestDataSamples.openSampleFileStream;
import static org.apache.poi.xssf.XSSFTestDataSamples.getSampleFile;
import static org.apache.poi.xssf.XSSFTestDataSamples.openSampleWorkbook;
import static org.apache.poi.xssf.XSSFTestDataSamples.writeOut;
import static org.apache.poi.xssf.XSSFTestDataSamples.writeOutAndReadBack;
Expand Down Expand Up @@ -1440,6 +1447,42 @@ void testGithub321() throws Exception {
}
}

@Test
void readFromZipStream() throws IOException {
File tempFile = TempFile.createTempFile("poitest", ".zip");
try {
try (ZipArchiveOutputStream zos = new ZipArchiveOutputStream(tempFile)) {
File f1 = getSampleFile("github-321.xlsx");
File f2 = getSampleFile("48495.xlsx");
ArchiveEntry e1 = zos.createArchiveEntry(f1, "github-321.xlsx");
zos.putArchiveEntry(e1);
try (InputStream s = Files.newInputStream(f1.toPath())) {
IOUtils.copy(s, zos);
}
zos.closeArchiveEntry();
ArchiveEntry e2 = zos.createArchiveEntry(f2, "48495.xlsx");
zos.putArchiveEntry(e2);
try (InputStream s = Files.newInputStream(f2.toPath())) {
IOUtils.copy(s, zos);
}
zos.closeArchiveEntry();
zos.finish();
}
int count = 0;
try (ZipArchiveInputStream zis = new ZipArchiveInputStream(Files.newInputStream(tempFile.toPath()))) {
ZipArchiveEntry entry;
while ((entry = zis.getNextZipEntry()) != null) {
XSSFWorkbook wb = new XSSFWorkbook(zis);
assertNotNull(wb);
count++;
}
}
assertEquals(2, count);
} finally {
tempFile.delete();
}
}

private static void expectFormattedContent(Cell cell, String value) {
assertEquals(value, new DataFormatter().formatCellValue(cell),
"Cell " + ref(cell) + " has wrong formatted content.");
Expand Down

0 comments on commit 105966c

Please sign in to comment.