New year bug

Author: Prabhu Subramanian

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="appthreat-vulnerability-db",
-    version="1.6.2",
+    version="1.6.3",
     author="Team AppThreat",
     author_email="[email protected]",
     description="AppThreat's vulnerability database and package search library with a built-in file based storage. CVE, GitHub, npm are the primary sources of vulnerabilities.",

vdb/lib/db.py

@@ -57,6 +57,8 @@ def store(db, datas):
     :param datas: Data list to store
     :return: Stored packed documents
     """
+    if datas is None:
+        return None
     global index_data, vendor_index_data
     docs = storage.store(datas, db_file=db["db_file"], index_file=db["index_file"])
     # Re-read the index

vdb/lib/nvd.py

@@ -55,6 +55,8 @@ def download_all(self, local_store=True):
         data_list = []
         for y in range(now.year, int(start_year) - 1, -1):
             data = self.fetch(y)
+            if not data:
+                continue
             if local_store:
                 self.store(data)
             data_list += data
@@ -72,7 +74,11 @@ def fetch(self, year):
         url = config.nvd_url % dict(year=year)
         LOG.debug("Download NVD CVE from {}".format(url))
         with tempfile.NamedTemporaryFile() as tf:
-            r = requests.get(url, stream=True)
+            try:
+                r = requests.get(url, stream=True)
+            except Exception:
+                logging.warning(f"Exception while downloading NVD feed from {url}")
+                return None
             for chunk in r.iter_content(chunk_size=download_chunk_size):
                 tf.write(chunk)
             tf.flush()