add proxies to phistank_checkurl.py

TheHive-Project · Dec 4, 2024 · 7b664cc · 7b664cc
1 parent 6a0c1ff
commit 7b664cc
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/analyzers/PhishTank/phishtank_checkurl.py b/analyzers/PhishTank/phishtank_checkurl.py
@@ -8,14 +8,15 @@ class PhishtankAnalyzer(Analyzer):
 
     def __init__(self):
         Analyzer.__init__(self)
-        self.phishtank_key = self.get_param('config.key', None,
-                                            'Missing PhishTank API key')
+        self.phishtank_key = self.get_param('config.key', None, 'Missing PhishTank API key')
+        self.proxies = {'http': self.get_param('config.proxy_http', None),
+                        'https': self.get_param('config.proxy_https', None)}
 
     def phishtank_checkurl(self, data):
         url = 'https://checkurl.phishtank.com/checkurl/'
         postheaders = {"User-Agent": "phishtank/cortex"}
         postdata = {'url': data, 'format': 'json', 'app_key': self.phishtank_key}
-        r = requests.post(url, headers=postheaders, data=postdata)
+        r = requests.post(url, headers=postheaders, data=postdata, proxies=self.proxies)
         return r.json()
 
     def summary(self, raw):