Browse Source

fix issue in IOUtils.toByteArrayWithMaxLength

git-svn-id: https://svn.apache.org/repos/asf/poi/trunk@1898856 13f79535-47bb-0310-9956-ffa450edef68
tags/REL_5_2_2
PJ Fanning 2 years ago
parent
commit
e7f9cd277d
1 changed files with 5 additions and 5 deletions
  1. 5
    5
      poi/src/main/java/org/apache/poi/util/IOUtils.java

+ 5
- 5
poi/src/main/java/org/apache/poi/util/IOUtils.java View File

@@ -171,7 +171,7 @@ public final class IOUtils {
* @throws RecordFormatException If the requested length is invalid.
*/
public static byte[] toByteArray(InputStream stream, final int length, final int maxLength) throws IOException {
return toByteArray(stream, length, maxLength, true);
return toByteArray(stream, length, maxLength, true, length != Integer.MAX_VALUE);
}

/**
@@ -188,11 +188,11 @@ public final class IOUtils {
* @since POI 5.2.1
*/
public static byte[] toByteArrayWithMaxLength(InputStream stream, final int maxLength) throws IOException {
return toByteArray(stream, maxLength, maxLength, false);
return toByteArray(stream, maxLength, maxLength, false, false);
}

private static byte[] toByteArray(InputStream stream, final int length, final int maxLength,
final boolean checkEOFException) throws IOException {
final boolean checkEOFException, final boolean isLengthKnown) throws IOException {
if (length < 0 || maxLength < 0) {
throw new RecordFormatException("Can't allocate an array of length < 0");
}
@@ -202,8 +202,8 @@ public final class IOUtils {
}

final int derivedLen = Math.min(length, derivedMaxLength);
try (UnsynchronizedByteArrayOutputStream baos =
new UnsynchronizedByteArrayOutputStream(derivedLen == Integer.MAX_VALUE ? 4096 : derivedLen)) {
final int bufferLen = isLengthKnown ? derivedLen : 4096;
try (UnsynchronizedByteArrayOutputStream baos = new UnsynchronizedByteArrayOutputStream(bufferLen)) {
byte[] buffer = new byte[4096];
int totalBytes = 0, readBytes;
do {

Loading…
Cancel
Save