From dfb39c8eb30148017f17b96d146f3f3d761b3b6d Mon Sep 17 00:00:00 2001
From: ekultek <thomas.perkins23@icloud.com>
Date: Tue, 7 Nov 2017 11:56:51 -0600
Subject: [PATCH] added a timeout to the WhoIs lookup, this will prevent the
 API from stopping us

---
 lib/attacks/whois_lookup/whois.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lib/attacks/whois_lookup/whois.py b/lib/attacks/whois_lookup/whois.py
index b107793..d303f7a 100644
--- a/lib/attacks/whois_lookup/whois.py
+++ b/lib/attacks/whois_lookup/whois.py
@@ -1,5 +1,6 @@
 import os
 import json
+import time
 import urllib2
 
 from base64 import b64decode
@@ -97,11 +98,15 @@ def whois_lookup_main(domain, **kwargs):
     """
     main function
     """
+    # sleep a little bit so that WhoIs doesn't stop us from making requests
     verbose = kwargs.get("verbose", False)
+    timeout = kwargs.get("timeout", None)
     domain = replace_http(domain)
     logger.info(set_color(
         "performing WhoIs lookup on given domain '{}'...".format(domain)
     ))
+    if timeout is not None:
+        time.sleep(timeout)
     raw_information = gather_raw_whois_info(domain)
     logger.info(set_color(
         "discovered raw information..."