Fixed issue with hosting ISP

qtfkwk · Nov 26, 2015 · c8cf0ec · c8cf0ec
1 parent 594bbea
commit c8cf0ec
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/config/constants.py b/config/constants.py
@@ -19,6 +19,7 @@
 build = "0.6.0"
 
 # Update Information
+url_test = "http://www.toolswatch.org"
 url = "http://www.toolswatch.org/vfeed/"
 db = "vfeed.db"
 db_compressed = "vfeed.db.tgz"

diff --git a/lib/core/update.py b/lib/core/update.py
@@ -6,14 +6,15 @@
 import sys
 import urllib2
 import tarfile
-from config.constants import db, db_compressed, url, update_status
+from config.constants import db, db_compressed, url, url_test, update_status
 from lib.common.utils import checksum
 
 
 class Update(object):
     def __init__(self):
         self.db = db
         self.db_compressed = db_compressed
+        self.url_test = url_test
         self.db_url = url
         self.db_update = update_status
         self.db_download = self.db_url + self.db_compressed
@@ -28,7 +29,7 @@ def update(self):
 
         print "[+] Checking connectivity to", self.db_url
         try:
-            if urllib2.urlopen(self.db_url):
+            if urllib2.urlopen(self.url_test):
                 if not os.path.isfile(self.db):
                     print "[+] New install. Downloading the Correlated Vulnerability Database."
                     self.download(self.remote_db)