fix: avoid NPE when initializing SingleCertValidatingFactory with cla…

…sspath resource and contextclassloader is null
pgjdbc · Jul 4, 2020 · 0b7575a · 0b7575a
1 parent aa70d2f
commit 0b7575a
Showing 1 changed file with 20 additions and 2 deletions.
diff --git a/pgjdbc/src/main/java/org/postgresql/ssl/SingleCertValidatingFactory.java b/pgjdbc/src/main/java/org/postgresql/ssl/SingleCertValidatingFactory.java
@@ -97,8 +97,26 @@ public SingleCertValidatingFactory(String sslFactoryArg) throws GeneralSecurityE
         in = new BufferedInputStream(new FileInputStream(path));
       } else if (sslFactoryArg.startsWith(CLASSPATH_PREFIX)) {
         String path = sslFactoryArg.substring(CLASSPATH_PREFIX.length());
-        in = new BufferedInputStream(
-            Thread.currentThread().getContextClassLoader().getResourceAsStream(path));
+        ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
+        InputStream inputStream;
+        if (classLoader != null) {
+          inputStream = classLoader.getResourceAsStream(path);
+          if (inputStream == null) {
+            throw new IllegalArgumentException(
+                GT.tr("Unable to find resource {0} via Thread contextClassLoader {1}",
+                    path, classLoader)
+            );
+          }
+        } else {
+          inputStream = getClass().getResourceAsStream(path);
+          if (inputStream == null) {
+            throw new IllegalArgumentException(
+                GT.tr("Unable to find resource {0} via class {1} ClassLoader {2}",
+                    path, getClass(), getClass().getClassLoader())
+            );
+          }
+        }
+        in = new BufferedInputStream(inputStream);
       } else if (sslFactoryArg.startsWith(ENV_PREFIX)) {
         String name = sslFactoryArg.substring(ENV_PREFIX.length());
         String cert = System.getenv(name);