httpURLConnection.getErrorStream() can return null so handle it.

jopenlibs · Apr 24, 2023 · d65c09e · d65c09e
1 parent 8b9fcfa
commit d65c09e
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 12 deletions.
diff --git a/src/main/java/io/github/jopenlibs/vault/rest/Rest.java b/src/main/java/io/github/jopenlibs/vault/rest/Rest.java
@@ -545,17 +545,7 @@ private byte[] responseBodyBytes(final URLConnection connection) throws RestExce
                     inputStream = httpURLConnection.getErrorStream();
                 }
             }
-            final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
-            // getErrorStream() can return null so handle it.
-            if(inputStream != null) {
-                int bytesRead;
-                final byte[] bytes = new byte[16384];
-                while ((bytesRead = inputStream.read(bytes, 0, bytes.length)) != -1) {
-                    byteArrayOutputStream.write(bytes, 0, bytesRead);
-                }
-            }
-            byteArrayOutputStream.flush();
-            return byteArrayOutputStream.toByteArray();
+            return handleResponseInputStream(inputStream);
         } catch (IOException e) {
             return new byte[0];
         }
@@ -589,4 +579,31 @@ private int connectionStatus(final URLConnection connection) throws IOException,
         return statusCode;
     }
 
-}
+    /**
+     * <p>This method </p>
+     *
+     * @param inputStream The input stream from the connection.
+     * @return The body payload, downloaded from the HTTP connection response
+     */
+    protected byte[] handleResponseInputStream(final InputStream inputStream) {
+        try {
+            // getErrorStream() can return null so handle it.
+            if (inputStream != null) {
+                final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
+
+                int bytesRead;
+                final byte[] bytes = new byte[16384];
+                while ((bytesRead = inputStream.read(bytes, 0, bytes.length)) != -1) {
+                    byteArrayOutputStream.write(bytes, 0, bytesRead);
+                }
+
+                byteArrayOutputStream.flush();
+                return byteArrayOutputStream.toByteArray();
+            } else {
+                return new byte[0];
+            }
+        } catch (IOException e) {
+            return new byte[0];
+        }
+    }
+}
diff --git a/src/test/java/io/github/jopenlibs/vault/rest/GetTests.java b/src/test/java/io/github/jopenlibs/vault/rest/GetTests.java
@@ -192,4 +192,15 @@ public void testGet_RetrievesResponseBodyWhenStatusIs418() throws RestException
                 responseBody.contains("User-Agent"));
     }
 
+
+    /**
+     * <p>Verify that response body does not cause NPE when input stream is null.</p>
+     */
+    @Test
+    public void test_handleResponseInputStream() {
+        final Rest rest = new Rest();
+        final byte[] result = rest.handleResponseInputStream(null);
+        assertEquals(0, result.length);
+    }
+
 }