fix: prevent NPE

resolves #112
jeremylong · Dec 15, 2023 · eb4f1c4 · eb4f1c4
1 parent c2fb945
commit eb4f1c4
Showing 1 changed file with 39 additions and 33 deletions.
diff --git a/...clients/src/main/java/io/github/jeremylong/openvulnerability/client/nvd/NvdCveClient.java b/...clients/src/main/java/io/github/jeremylong/openvulnerability/client/nvd/NvdCveClient.java
@@ -319,37 +319,43 @@ public Collection<DefCveItem> next() {
         RateLimitedCall call;
         try {
             call = getCompletedFuture();
-            SimpleHttpResponse response = call.getResponse();
-            if (response.getCode() == 200) {
-                LOG.debug("Content-Type Received: {}", response.getContentType());
-                json = response.getBodyText();
-                // resolve issue #20
-                if (json == null && response.getBody().isBytes()) {
-                    json = new String(response.getBodyBytes(), StandardCharsets.UTF_8);
-                }
-
-                CveApiJson20 current;
-                try {
-                    current = objectMapper.readValue(json, CveApiJson20.class);
-                    this.indexesToRetrieve.remove(call.getStartIndex());
-                } catch (JsonProcessingException e) {
+            if (call == null) {
+                if (hasNext()) {
                     return next();
                 }
-                this.totalAvailable = current.getTotalResults();
-                lastUpdated = findLastUpdated(lastUpdated, current.getVulnerabilities());
-                if (firstCall) {
-                    firstCall = false;
-                    queueCalls();
-                }
-                if (futures.isEmpty() && !indexesToRetrieve.isEmpty()) {
-                    queueUnsuccessful();
-                }
-                return current.getVulnerabilities();
             } else {
-                lastStatusCode = response.getCode();
-                LOG.debug("Status Code: {}", lastStatusCode);
-                LOG.debug("Response: {}", response.getBodyText());
-                throw new NvdApiException("NVD Returned Status Code: " + lastStatusCode);
+                SimpleHttpResponse response = call.getResponse();
+                if (response.getCode() == 200) {
+                    LOG.debug("Content-Type Received: {}", response.getContentType());
+                    json = response.getBodyText();
+                    // resolve issue #20
+                    if (json == null && response.getBody().isBytes()) {
+                        json = new String(response.getBodyBytes(), StandardCharsets.UTF_8);
+                    }
+
+                    CveApiJson20 current;
+                    try {
+                        current = objectMapper.readValue(json, CveApiJson20.class);
+                        this.indexesToRetrieve.remove(call.getStartIndex());
+                    } catch (JsonProcessingException e) {
+                        return next();
+                    }
+                    this.totalAvailable = current.getTotalResults();
+                    lastUpdated = findLastUpdated(lastUpdated, current.getVulnerabilities());
+                    if (firstCall) {
+                        firstCall = false;
+                        queueCalls();
+                    }
+                    if (futures.isEmpty() && !indexesToRetrieve.isEmpty()) {
+                        queueUnsuccessful();
+                    }
+                    return current.getVulnerabilities();
+                } else {
+                    lastStatusCode = response.getCode();
+                    LOG.debug("Status Code: {}", lastStatusCode);
+                    LOG.debug("Response: {}", response.getBodyText());
+                    throw new NvdApiException("NVD Returned Status Code: " + lastStatusCode);
+                }
             }
         } catch (InterruptedException e) {
             Thread.currentThread().interrupt();
@@ -362,8 +368,8 @@ public Collection<DefCveItem> next() {
                 return next();
             }
             close();
-            return null;
         }
+        return null;
     }
 
     /**
@@ -389,17 +395,17 @@ public ZonedDateTime getLastUpdated() {
     }
 
     private RateLimitedCall getCompletedFuture() throws InterruptedException, ExecutionException {
-        boolean notFound = futures.size() > 0;
         Future<RateLimitedCall> result = null;
-        while (notFound) {
+        while (result == null && !futures.isEmpty()) {
             for (Future<RateLimitedCall> future : futures) {
                 if (future.isDone()) {
                     result = future;
-                    notFound = false;
                     break;
                 }
             }
-            Thread.sleep(500);
+            if (result == null) {
+                Thread.sleep(500);
+            }
         }
         if (result != null) {
             futures.remove(result);